git.sesse.net Git - vlc/blob - modules/stream_out/rtp.c

   1 /*****************************************************************************
   2  * rtp.c: rtp stream output module
   3  *****************************************************************************
   4  * Copyright (C) 2003-2004, 2010 the VideoLAN team
   5  * Copyright © 2007-2008 Rémi Denis-Courmont
   6  *
   7  * Authors: Laurent Aimar <fenrir@via.ecp.fr>
   8  *          Pierre Ynard
   9  *
  10  * This program is free software; you can redistribute it and/or modify
  11  * it under the terms of the GNU General Public License as published by
  12  * the Free Software Foundation; either version 2 of the License, or
  13  * (at your option) any later version.
  14  *
  15  * This program is distributed in the hope that it will be useful,
  16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18  * GNU General Public License for more details.
  19  *
  20  * You should have received a copy of the GNU General Public License
  21  * along with this program; if not, write to the Free Software
  22  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  23  *****************************************************************************/
  24
  25 /*****************************************************************************
  26  * Preamble
  27  *****************************************************************************/
  28
  29 #ifdef HAVE_CONFIG_H
  30 # include "config.h"
  31 #endif
  32
  33 #include <vlc_common.h>
  34 #include <vlc_plugin.h>
  35 #include <vlc_sout.h>
  36 #include <vlc_block.h>
  37
  38 #include <vlc_httpd.h>
  39 #include <vlc_url.h>
  40 #include <vlc_network.h>
  41 #include <vlc_fs.h>
  42 #include <vlc_rand.h>
  43 #ifdef HAVE_SRTP
  44 # include <srtp.h>
  45 # include <gcrypt.h>
  46 # include <vlc_gcrypt.h>
  47 #endif
  48
  49 #include "rtp.h"
  50
  51 #ifdef HAVE_UNISTD_H
  52 #   include <sys/types.h>
  53 #   include <unistd.h>
  54 #endif
  55 #ifdef HAVE_ARPA_INET_H
  56 #   include <arpa/inet.h>
  57 #endif
  58 #ifdef HAVE_LINUX_DCCP_H
  59 #   include <linux/dccp.h>
  60 #endif
  61 #ifndef IPPROTO_DCCP
  62 # define IPPROTO_DCCP 33
  63 #endif
  64 #ifndef IPPROTO_UDPLITE
  65 # define IPPROTO_UDPLITE 136
  66 #endif
  67
  68 #include <errno.h>
  69
  70 #include <assert.h>
  71
  72 /*****************************************************************************
  73  * Module descriptor
  74  *****************************************************************************/
  75
  76 #define DEST_TEXT N_("Destination")
  77 #define DEST_LONGTEXT N_( \
  78     "This is the output URL that will be used." )
  79 #define SDP_TEXT N_("SDP")
  80 #define SDP_LONGTEXT N_( \
  81     "This allows you to specify how the SDP (Session Descriptor) for this RTP "\
  82     "session will be made available. You must use a url: http://location to " \
  83     "access the SDP via HTTP, rtsp://location for RTSP access, and sap:// " \
  84     "for the SDP to be announced via SAP." )
  85 #define SAP_TEXT N_("SAP announcing")
  86 #define SAP_LONGTEXT N_("Announce this session with SAP.")
  87 #define MUX_TEXT N_("Muxer")
  88 #define MUX_LONGTEXT N_( \
  89     "This allows you to specify the muxer used for the streaming output. " \
  90     "Default is to use no muxer (standard RTP stream)." )
  91
  92 #define NAME_TEXT N_("Session name")
  93 #define NAME_LONGTEXT N_( \
  94     "This is the name of the session that will be announced in the SDP " \
  95     "(Session Descriptor)." )
  96 #define CAT_TEXT N_("Session category")
  97 #define CAT_LONGTEXT N_( \
  98   "This allows you to specify a category for the session, " \
  99   "that will be announced if you choose to use SAP." )
 100 #define DESC_TEXT N_("Session description")
 101 #define DESC_LONGTEXT N_( \
 102     "This allows you to give a short description with details about the stream, " \
 103     "that will be announced in the SDP (Session Descriptor)." )
 104 #define URL_TEXT N_("Session URL")
 105 #define URL_LONGTEXT N_( \
 106     "This allows you to give a URL with more details about the stream " \
 107     "(often the website of the streaming organization), that will " \
 108     "be announced in the SDP (Session Descriptor)." )
 109 #define EMAIL_TEXT N_("Session email")
 110 #define EMAIL_LONGTEXT N_( \
 111     "This allows you to give a contact mail address for the stream, that will " \
 112     "be announced in the SDP (Session Descriptor)." )
 113 #define PHONE_TEXT N_("Session phone number")
 114 #define PHONE_LONGTEXT N_( \
 115     "This allows you to give a contact telephone number for the stream, that will " \
 116     "be announced in the SDP (Session Descriptor)." )
 117
 118 #define PORT_TEXT N_("Port")
 119 #define PORT_LONGTEXT N_( \
 120     "This allows you to specify the base port for the RTP streaming." )
 121 #define PORT_AUDIO_TEXT N_("Audio port")
 122 #define PORT_AUDIO_LONGTEXT N_( \
 123     "This allows you to specify the default audio port for the RTP streaming." )
 124 #define PORT_VIDEO_TEXT N_("Video port")
 125 #define PORT_VIDEO_LONGTEXT N_( \
 126     "This allows you to specify the default video port for the RTP streaming." )
 127
 128 #define TTL_TEXT N_("Hop limit (TTL)")
 129 #define TTL_LONGTEXT N_( \
 130     "This is the hop limit (also known as \"Time-To-Live\" or TTL) of " \
 131     "the multicast packets sent by the stream output (-1 = use operating " \
 132     "system built-in default).")
 133
 134 #define RTCP_MUX_TEXT N_("RTP/RTCP multiplexing")
 135 #define RTCP_MUX_LONGTEXT N_( \
 136     "This sends and receives RTCP packet multiplexed over the same port " \
 137     "as RTP packets." )
 138
 139 #define CACHING_TEXT N_("Caching value (ms)")
 140 #define CACHING_LONGTEXT N_( \
 141     "Default caching value for outbound RTP streams. This " \
 142     "value should be set in milliseconds." )
 143
 144 #define PROTO_TEXT N_("Transport protocol")
 145 #define PROTO_LONGTEXT N_( \
 146     "This selects which transport protocol to use for RTP." )
 147
 148 #define SRTP_KEY_TEXT N_("SRTP key (hexadecimal)")
 149 #define SRTP_KEY_LONGTEXT N_( \
 150     "RTP packets will be integrity-protected and ciphered "\
 151     "with this Secure RTP master shared secret key. "\
 152     "This must be a 32-character-long hexadecimal string.")
 153
 154 #define SRTP_SALT_TEXT N_("SRTP salt (hexadecimal)")
 155 #define SRTP_SALT_LONGTEXT N_( \
 156     "Secure RTP requires a (non-secret) master salt value. " \
 157     "This must be a 28-character-long hexadecimal string.")
 158
 159 static const char *const ppsz_protos[] = {
 160     "dccp", "sctp", "tcp", "udp", "udplite",
 161 };
 162
 163 static const char *const ppsz_protocols[] = {
 164     "DCCP", "SCTP", "TCP", "UDP", "UDP-Lite",
 165 };
 166
 167 #define RFC3016_TEXT N_("MP4A LATM")
 168 #define RFC3016_LONGTEXT N_( \
 169     "This allows you to stream MPEG4 LATM audio streams (see RFC3016)." )
 170
 171 #define RTSP_TIMEOUT_TEXT N_( "RTSP session timeout (s)" )
 172 #define RTSP_TIMEOUT_LONGTEXT N_( "RTSP sessions will be closed after " \
 173     "not receiving any RTSP request for this long. Setting it to a " \
 174     "negative value or zero disables timeouts. The default is 60 (one " \
 175     "minute)." )
 176
 177 #define RTSP_USER_TEXT N_("Username")
 178 #define RTSP_USER_LONGTEXT N_("User name that will be " \
 179                               "requested to access the stream." )
 180 #define RTSP_PASS_TEXT N_("Password")
 181 #define RTSP_PASS_LONGTEXT N_("Password that will be " \
 182                               "requested to access the stream." )
 183
 184 static int  Open ( vlc_object_t * );
 185 static void Close( vlc_object_t * );
 186
 187 #define SOUT_CFG_PREFIX "sout-rtp-"
 188 #define MAX_EMPTY_BLOCKS 200
 189
 190 vlc_module_begin ()
 191     set_shortname( N_("RTP"))
 192     set_description( N_("RTP stream output") )
 193     set_capability( "sout stream", 0 )
 194     add_shortcut( "rtp", "vod" )
 195     set_category( CAT_SOUT )
 196     set_subcategory( SUBCAT_SOUT_STREAM )
 197
 198     add_string( SOUT_CFG_PREFIX "dst", "", DEST_TEXT,
 199                 DEST_LONGTEXT, true )
 200     add_string( SOUT_CFG_PREFIX "sdp", "", SDP_TEXT,
 201                 SDP_LONGTEXT, true )
 202     add_string( SOUT_CFG_PREFIX "mux", "", MUX_TEXT,
 203                 MUX_LONGTEXT, true )
 204     add_bool( SOUT_CFG_PREFIX "sap", false, SAP_TEXT, SAP_LONGTEXT,
 205               true )
 206
 207     add_string( SOUT_CFG_PREFIX "name", "", NAME_TEXT,
 208                 NAME_LONGTEXT, true )
 209     add_string( SOUT_CFG_PREFIX "cat", "", CAT_TEXT, CAT_LONGTEXT, true )
 210     add_string( SOUT_CFG_PREFIX "description", "", DESC_TEXT,
 211                 DESC_LONGTEXT, true )
 212     add_string( SOUT_CFG_PREFIX "url", "", URL_TEXT,
 213                 URL_LONGTEXT, true )
 214     add_string( SOUT_CFG_PREFIX "email", "", EMAIL_TEXT,
 215                 EMAIL_LONGTEXT, true )
 216     add_string( SOUT_CFG_PREFIX "phone", "", PHONE_TEXT,
 217                 PHONE_LONGTEXT, true )
 218
 219     add_string( SOUT_CFG_PREFIX "proto", "udp", PROTO_TEXT,
 220                 PROTO_LONGTEXT, false )
 221         change_string_list( ppsz_protos, ppsz_protocols, NULL )
 222     add_integer( SOUT_CFG_PREFIX "port", 5004, PORT_TEXT,
 223                  PORT_LONGTEXT, true )
 224     add_integer( SOUT_CFG_PREFIX "port-audio", 0, PORT_AUDIO_TEXT,
 225                  PORT_AUDIO_LONGTEXT, true )
 226     add_integer( SOUT_CFG_PREFIX "port-video", 0, PORT_VIDEO_TEXT,
 227                  PORT_VIDEO_LONGTEXT, true )
 228
 229     add_integer( SOUT_CFG_PREFIX "ttl", -1, TTL_TEXT,
 230                  TTL_LONGTEXT, true )
 231     add_bool( SOUT_CFG_PREFIX "rtcp-mux", false,
 232               RTCP_MUX_TEXT, RTCP_MUX_LONGTEXT, false )
 233     add_integer( SOUT_CFG_PREFIX "caching", DEFAULT_PTS_DELAY / 1000,
 234                  CACHING_TEXT, CACHING_LONGTEXT, true )
 235
 236 #ifdef HAVE_SRTP
 237     add_string( SOUT_CFG_PREFIX "key", "",
 238                 SRTP_KEY_TEXT, SRTP_KEY_LONGTEXT, false )
 239     add_string( SOUT_CFG_PREFIX "salt", "",
 240                 SRTP_SALT_TEXT, SRTP_SALT_LONGTEXT, false )
 241 #endif
 242
 243     add_bool( SOUT_CFG_PREFIX "mp4a-latm", false, RFC3016_TEXT,
 244                  RFC3016_LONGTEXT, false )
 245
 246     set_callbacks( Open, Close )
 247
 248     add_submodule ()
 249     set_shortname( N_("RTSP VoD" ) )
 250     set_description( N_("RTSP VoD server") )
 251     set_category( CAT_SOUT )
 252     set_subcategory( SUBCAT_SOUT_VOD )
 253     set_capability( "vod server", 10 )
 254     set_callbacks( OpenVoD, CloseVoD )
 255     add_shortcut( "rtsp" )
 256     add_integer( "rtsp-timeout", 60, RTSP_TIMEOUT_TEXT,
 257                  RTSP_TIMEOUT_LONGTEXT, true )
 258     add_string( "sout-rtsp-user", "",
 259                 RTSP_USER_TEXT, RTSP_USER_LONGTEXT, true )
 260     add_password( "sout-rtsp-pwd", "",
 261                   RTSP_PASS_TEXT, RTSP_PASS_LONGTEXT, true )
 262
 263 vlc_module_end ()
 264
 265 /*****************************************************************************
 266  * Exported prototypes
 267  *****************************************************************************/
 268 static const char *const ppsz_sout_options[] = {
 269     "dst", "name", "cat", "port", "port-audio", "port-video", "*sdp", "ttl",
 270     "mux", "sap", "description", "url", "email", "phone",
 271     "proto", "rtcp-mux", "caching",
 272 #ifdef HAVE_SRTP
 273     "key", "salt",
 274 #endif
 275     "mp4a-latm", NULL
 276 };
 277
 278 static sout_stream_id_t *Add ( sout_stream_t *, es_format_t * );
 279 static int               Del ( sout_stream_t *, sout_stream_id_t * );
 280 static int               Send( sout_stream_t *, sout_stream_id_t *,
 281                                block_t* );
 282 static sout_stream_id_t *MuxAdd ( sout_stream_t *, es_format_t * );
 283 static int               MuxDel ( sout_stream_t *, sout_stream_id_t * );
 284 static int               MuxSend( sout_stream_t *, sout_stream_id_t *,
 285                                   block_t* );
 286
 287 static sout_access_out_t *GrabberCreate( sout_stream_t *p_sout );
 288 static void* ThreadSend( void * );
 289 static void *rtp_listen_thread( void * );
 290
 291 static void SDPHandleUrl( sout_stream_t *, const char * );
 292
 293 static int SapSetup( sout_stream_t *p_stream );
 294 static int FileSetup( sout_stream_t *p_stream );
 295 static int HttpSetup( sout_stream_t *p_stream, const vlc_url_t * );
 296
 297 static int64_t rtp_init_ts( const vod_media_t *p_media,
 298                             const char *psz_vod_session );
 299
 300 struct sout_stream_sys_t
 301 {
 302     /* SDP */
 303     char    *psz_sdp;
 304     vlc_mutex_t  lock_sdp;
 305
 306     /* SDP to disk */
 307     char *psz_sdp_file;
 308
 309     /* SDP via SAP */
 310     bool b_export_sap;
 311     session_descriptor_t *p_session;
 312
 313     /* SDP via HTTP */
 314     httpd_host_t *p_httpd_host;
 315     httpd_file_t *p_httpd_file;
 316
 317     /* RTSP */
 318     rtsp_stream_t *rtsp;
 319
 320     /* RTSP NPT and timestamp computations */
 321     mtime_t      i_npt_zero;    /* when NPT=0 packet is sent */
 322     int64_t      i_pts_zero;    /* predicts PTS of NPT=0 packet */
 323     int64_t      i_pts_offset;  /* matches actual PTS to prediction */
 324     vlc_mutex_t  lock_ts;
 325
 326     /* */
 327     char     *psz_destination;
 328     uint16_t  i_port;
 329     uint16_t  i_port_audio;
 330     uint16_t  i_port_video;
 331     uint8_t   proto;
 332     bool      rtcp_mux;
 333     bool      b_latm;
 334
 335     /* VoD */
 336     vod_media_t *p_vod_media;
 337     char     *psz_vod_session;
 338
 339     /* in case we do TS/PS over rtp */
 340     sout_mux_t        *p_mux;
 341     sout_access_out_t *p_grab;
 342     block_t           *packet;
 343
 344     /* */
 345     vlc_mutex_t      lock_es;
 346     int              i_es;
 347     sout_stream_id_t **es;
 348 };
 349
 350 typedef struct rtp_sink_t
 351 {
 352     int rtp_fd;
 353     rtcp_sender_t *rtcp;
 354 } rtp_sink_t;
 355
 356 struct sout_stream_id_t
 357 {
 358     sout_stream_t *p_stream;
 359     /* rtp field */
 360     uint16_t    i_sequence;
 361     bool        b_first_packet;
 362     bool        b_ts_init;
 363     uint32_t    i_ts_offset;
 364     uint8_t     ssrc[4];
 365
 366     /* for rtsp */
 367     uint16_t    i_seq_sent_next;
 368
 369     /* for sdp */
 370     rtp_format_t rtp_fmt;
 371     int          i_port;
 372
 373     /* Packetizer specific fields */
 374     int                 i_mtu;
 375 #ifdef HAVE_SRTP
 376     srtp_session_t     *srtp;
 377 #endif
 378
 379     /* Packets sinks */
 380     vlc_thread_t      thread;
 381     vlc_mutex_t       lock_sink;
 382     int               sinkc;
 383     rtp_sink_t       *sinkv;
 384     rtsp_stream_id_t *rtsp_id;
 385     struct {
 386         int          *fd;
 387         vlc_thread_t  thread;
 388     } listen;
 389
 390     block_fifo_t     *p_fifo;
 391     int64_t           i_caching;
 392 };
 393
 394 /*****************************************************************************
 395  * Open:
 396  *****************************************************************************/
 397 static int Open( vlc_object_t *p_this )
 398 {
 399     sout_stream_t       *p_stream = (sout_stream_t*)p_this;
 400     sout_instance_t     *p_sout = p_stream->p_sout;
 401     sout_stream_sys_t   *p_sys = NULL;
 402     config_chain_t      *p_cfg = NULL;
 403     char                *psz;
 404     bool          b_rtsp = false;
 405
 406     config_ChainParse( p_stream, SOUT_CFG_PREFIX,
 407                        ppsz_sout_options, p_stream->p_cfg );
 408
 409     p_sys = malloc( sizeof( sout_stream_sys_t ) );
 410     if( p_sys == NULL )
 411         return VLC_ENOMEM;
 412
 413     p_sys->psz_destination = var_GetNonEmptyString( p_stream, SOUT_CFG_PREFIX "dst" );
 414
 415     p_sys->i_port       = var_GetInteger( p_stream, SOUT_CFG_PREFIX "port" );
 416     p_sys->i_port_audio = var_GetInteger( p_stream, SOUT_CFG_PREFIX "port-audio" );
 417     p_sys->i_port_video = var_GetInteger( p_stream, SOUT_CFG_PREFIX "port-video" );
 418     p_sys->rtcp_mux     = var_GetBool( p_stream, SOUT_CFG_PREFIX "rtcp-mux" );
 419
 420     if( p_sys->i_port_audio && p_sys->i_port_video == p_sys->i_port_audio )
 421     {
 422         msg_Err( p_stream, "audio and video RTP port must be distinct" );
 423         free( p_sys->psz_destination );
 424         free( p_sys );
 425         return VLC_EGENERIC;
 426     }
 427
 428     for( p_cfg = p_stream->p_cfg; p_cfg != NULL; p_cfg = p_cfg->p_next )
 429     {
 430         if( !strcmp( p_cfg->psz_name, "sdp" )
 431          && ( p_cfg->psz_value != NULL )
 432          && !strncasecmp( p_cfg->psz_value, "rtsp:", 5 ) )
 433         {
 434             b_rtsp = true;
 435             break;
 436         }
 437     }
 438     if( !b_rtsp )
 439     {
 440         psz = var_GetNonEmptyString( p_stream, SOUT_CFG_PREFIX "sdp" );
 441         if( psz != NULL )
 442         {
 443             if( !strncasecmp( psz, "rtsp:", 5 ) )
 444                 b_rtsp = true;
 445             free( psz );
 446         }
 447     }
 448
 449     /* Transport protocol */
 450     p_sys->proto = IPPROTO_UDP;
 451     psz = var_GetNonEmptyString (p_stream, SOUT_CFG_PREFIX"proto");
 452
 453     if ((psz == NULL) || !strcasecmp (psz, "udp"))
 454         (void)0; /* default */
 455     else
 456     if (!strcasecmp (psz, "dccp"))
 457     {
 458         p_sys->proto = IPPROTO_DCCP;
 459         p_sys->rtcp_mux = true; /* Force RTP/RTCP mux */
 460     }
 461 #if 0
 462     else
 463     if (!strcasecmp (psz, "sctp"))
 464     {
 465         p_sys->proto = IPPROTO_TCP;
 466         p_sys->rtcp_mux = true; /* Force RTP/RTCP mux */
 467     }
 468 #endif
 469 #if 0
 470     else
 471     if (!strcasecmp (psz, "tcp"))
 472     {
 473         p_sys->proto = IPPROTO_TCP;
 474         p_sys->rtcp_mux = true; /* Force RTP/RTCP mux */
 475     }
 476 #endif
 477     else
 478     if (!strcasecmp (psz, "udplite") || !strcasecmp (psz, "udp-lite"))
 479         p_sys->proto = IPPROTO_UDPLITE;
 480     else
 481         msg_Warn (p_this, "unknown or unsupported transport protocol \"%s\"",
 482                   psz);
 483     free (psz);
 484     var_Create (p_this, "dccp-service", VLC_VAR_STRING);
 485
 486     p_sys->p_vod_media = NULL;
 487     p_sys->psz_vod_session = NULL;
 488
 489     if (! strcmp(p_stream->psz_name, "vod"))
 490     {
 491         /* The VLM stops all instances before deleting a media, so this
 492          * reference will remain valid during the lifetime of the rtp
 493          * stream output. */
 494         p_sys->p_vod_media = var_InheritAddress(p_stream, "vod-media");
 495
 496         if (p_sys->p_vod_media != NULL)
 497         {
 498             p_sys->psz_vod_session = var_InheritString(p_stream, "vod-session");
 499             if (p_sys->psz_vod_session == NULL)
 500             {
 501                 msg_Err(p_stream, "missing VoD session");
 502                 free(p_sys);
 503                 return VLC_EGENERIC;
 504             }
 505
 506             const char *mux = vod_get_mux(p_sys->p_vod_media);
 507             var_SetString(p_stream, SOUT_CFG_PREFIX "mux", mux);
 508         }
 509     }
 510
 511     if( p_sys->psz_destination == NULL && !b_rtsp
 512         && p_sys->p_vod_media == NULL )
 513     {
 514         msg_Err( p_stream, "missing destination and not in RTSP mode" );
 515         free( p_sys );
 516         return VLC_EGENERIC;
 517     }
 518
 519     int i_ttl = var_GetInteger( p_stream, SOUT_CFG_PREFIX "ttl" );
 520     if( i_ttl != -1 )
 521     {
 522         var_Create( p_stream, "ttl", VLC_VAR_INTEGER );
 523         var_SetInteger( p_stream, "ttl", i_ttl );
 524     }
 525
 526     p_sys->b_latm = var_GetBool( p_stream, SOUT_CFG_PREFIX "mp4a-latm" );
 527
 528     /* NPT=0 time will be determined when we packetize the first packet
 529      * (of any ES). But we want to be able to report rtptime in RTSP
 530      * without waiting (and already did in the VoD case). So until then,
 531      * we use an arbitrary reference PTS for timestamp computations, and
 532      * then actual PTS will catch up using offsets. */
 533     p_sys->i_npt_zero = VLC_TS_INVALID;
 534     p_sys->i_pts_zero = rtp_init_ts(p_sys->p_vod_media,
 535                                     p_sys->psz_vod_session);
 536     p_sys->i_es = 0;
 537     p_sys->es   = NULL;
 538     p_sys->rtsp = NULL;
 539     p_sys->psz_sdp = NULL;
 540
 541     p_sys->b_export_sap = false;
 542     p_sys->p_session = NULL;
 543     p_sys->psz_sdp_file = NULL;
 544
 545     p_sys->p_httpd_host = NULL;
 546     p_sys->p_httpd_file = NULL;
 547
 548     p_stream->p_sys     = p_sys;
 549
 550     vlc_mutex_init( &p_sys->lock_sdp );
 551     vlc_mutex_init( &p_sys->lock_ts );
 552     vlc_mutex_init( &p_sys->lock_es );
 553
 554     psz = var_GetNonEmptyString( p_stream, SOUT_CFG_PREFIX "mux" );
 555     if( psz != NULL )
 556     {
 557         /* Check muxer type */
 558         if( strncasecmp( psz, "ps", 2 )
 559          && strncasecmp( psz, "mpeg1", 5 )
 560          && strncasecmp( psz, "ts", 2 ) )
 561         {
 562             msg_Err( p_stream, "unsupported muxer type for RTP (only TS/PS)" );
 563             free( psz );
 564             vlc_mutex_destroy( &p_sys->lock_sdp );
 565             vlc_mutex_destroy( &p_sys->lock_ts );
 566             vlc_mutex_destroy( &p_sys->lock_es );
 567             free( p_sys->psz_vod_session );
 568             free( p_sys->psz_destination );
 569             free( p_sys );
 570             return VLC_EGENERIC;
 571         }
 572
 573         p_sys->p_grab = GrabberCreate( p_stream );
 574         p_sys->p_mux = sout_MuxNew( p_sout, psz, p_sys->p_grab );
 575         free( psz );
 576
 577         if( p_sys->p_mux == NULL )
 578         {
 579             msg_Err( p_stream, "cannot create muxer" );
 580             sout_AccessOutDelete( p_sys->p_grab );
 581             vlc_mutex_destroy( &p_sys->lock_sdp );
 582             vlc_mutex_destroy( &p_sys->lock_ts );
 583             vlc_mutex_destroy( &p_sys->lock_es );
 584             free( p_sys->psz_vod_session );
 585             free( p_sys->psz_destination );
 586             free( p_sys );
 587             return VLC_EGENERIC;
 588         }
 589
 590         p_sys->packet = NULL;
 591
 592         p_stream->pf_add  = MuxAdd;
 593         p_stream->pf_del  = MuxDel;
 594         p_stream->pf_send = MuxSend;
 595     }
 596     else
 597     {
 598         p_sys->p_mux    = NULL;
 599         p_sys->p_grab   = NULL;
 600
 601         p_stream->pf_add    = Add;
 602         p_stream->pf_del    = Del;
 603         p_stream->pf_send   = Send;
 604     }
 605
 606     if( var_GetBool( p_stream, SOUT_CFG_PREFIX"sap" ) )
 607         SDPHandleUrl( p_stream, "sap" );
 608
 609     psz = var_GetNonEmptyString( p_stream, SOUT_CFG_PREFIX "sdp" );
 610     if( psz != NULL )
 611     {
 612         config_chain_t *p_cfg;
 613
 614         SDPHandleUrl( p_stream, psz );
 615
 616         for( p_cfg = p_stream->p_cfg; p_cfg != NULL; p_cfg = p_cfg->p_next )
 617         {
 618             if( !strcmp( p_cfg->psz_name, "sdp" ) )
 619             {
 620                 if( p_cfg->psz_value == NULL || *p_cfg->psz_value == '\0' )
 621                     continue;
 622
 623                 /* needed both :sout-rtp-sdp= and rtp{sdp=} can be used */
 624                 if( !strcmp( p_cfg->psz_value, psz ) )
 625                     continue;
 626
 627                 SDPHandleUrl( p_stream, p_cfg->psz_value );
 628             }
 629         }
 630         free( psz );
 631     }
 632
 633     /* update p_sout->i_out_pace_nocontrol */
 634     p_stream->p_sout->i_out_pace_nocontrol++;
 635
 636     if( p_sys->p_mux != NULL )
 637     {
 638         sout_stream_id_t *id = Add( p_stream, NULL );
 639         if( id == NULL )
 640         {
 641             Close( p_this );
 642             return VLC_EGENERIC;
 643         }
 644     }
 645
 646     return VLC_SUCCESS;
 647 }
 648
 649 /*****************************************************************************
 650  * Close:
 651  *****************************************************************************/
 652 static void Close( vlc_object_t * p_this )
 653 {
 654     sout_stream_t     *p_stream = (sout_stream_t*)p_this;
 655     sout_stream_sys_t *p_sys = p_stream->p_sys;
 656
 657     /* update p_sout->i_out_pace_nocontrol */
 658     p_stream->p_sout->i_out_pace_nocontrol--;
 659
 660     if( p_sys->p_mux )
 661     {
 662         assert( p_sys->i_es <= 1 );
 663
 664         sout_MuxDelete( p_sys->p_mux );
 665         if ( p_sys->i_es > 0 )
 666             Del( p_stream, p_sys->es[0] );
 667         sout_AccessOutDelete( p_sys->p_grab );
 668
 669         if( p_sys->packet )
 670         {
 671             block_Release( p_sys->packet );
 672         }
 673     }
 674
 675     if( p_sys->rtsp != NULL )
 676         RtspUnsetup( p_sys->rtsp );
 677
 678     vlc_mutex_destroy( &p_sys->lock_sdp );
 679     vlc_mutex_destroy( &p_sys->lock_ts );
 680     vlc_mutex_destroy( &p_sys->lock_es );
 681
 682     if( p_sys->p_httpd_file )
 683         httpd_FileDelete( p_sys->p_httpd_file );
 684
 685     if( p_sys->p_httpd_host )
 686         httpd_HostDelete( p_sys->p_httpd_host );
 687
 688     free( p_sys->psz_sdp );
 689
 690     if( p_sys->psz_sdp_file != NULL )
 691     {
 692 #ifdef HAVE_UNISTD_H
 693         unlink( p_sys->psz_sdp_file );
 694 #endif
 695         free( p_sys->psz_sdp_file );
 696     }
 697     free( p_sys->psz_vod_session );
 698     free( p_sys->psz_destination );
 699     free( p_sys );
 700 }
 701
 702 /*****************************************************************************
 703  * SDPHandleUrl:
 704  *****************************************************************************/
 705 static void SDPHandleUrl( sout_stream_t *p_stream, const char *psz_url )
 706 {
 707     sout_stream_sys_t *p_sys = p_stream->p_sys;
 708     vlc_url_t url;
 709
 710     vlc_UrlParse( &url, psz_url, 0 );
 711     if( url.psz_protocol && !strcasecmp( url.psz_protocol, "http" ) )
 712     {
 713         if( p_sys->p_httpd_file )
 714         {
 715             msg_Err( p_stream, "you can use sdp=http:// only once" );
 716             goto out;
 717         }
 718
 719         if( HttpSetup( p_stream, &url ) )
 720         {
 721             msg_Err( p_stream, "cannot export SDP as HTTP" );
 722         }
 723     }
 724     else if( url.psz_protocol && !strcasecmp( url.psz_protocol, "rtsp" ) )
 725     {
 726         if( p_sys->rtsp != NULL )
 727         {
 728             msg_Err( p_stream, "you can use sdp=rtsp:// only once" );
 729             goto out;
 730         }
 731
 732         if( url.psz_host != NULL && *url.psz_host )
 733         {
 734             msg_Warn( p_stream, "\"%s\" RTSP host might be ignored in "
 735                       "multiple-host configurations, use at your own risks.",
 736                       url.psz_host );
 737             msg_Info( p_stream, "Consider passing --rtsp-host=IP on the "
 738                                 "command line instead." );
 739
 740             var_Create( p_stream, "rtsp-host", VLC_VAR_STRING );
 741             var_SetString( p_stream, "rtsp-host", url.psz_host );
 742         }
 743         if( url.i_port != 0 )
 744         {
 745             /* msg_Info( p_stream, "Consider passing --rtsp-port=%u on "
 746                       "the command line instead.", url.i_port ); */
 747
 748             var_Create( p_stream, "rtsp-port", VLC_VAR_INTEGER );
 749             var_SetInteger( p_stream, "rtsp-port", url.i_port );
 750         }
 751
 752         p_sys->rtsp = RtspSetup( VLC_OBJECT(p_stream), NULL, url.psz_path );
 753         if( p_sys->rtsp == NULL )
 754             msg_Err( p_stream, "cannot export SDP as RTSP" );
 755     }
 756     else if( ( url.psz_protocol && !strcasecmp( url.psz_protocol, "sap" ) ) ||
 757              ( url.psz_host && !strcasecmp( url.psz_host, "sap" ) ) )
 758     {
 759         p_sys->b_export_sap = true;
 760         SapSetup( p_stream );
 761     }
 762     else if( url.psz_protocol && !strcasecmp( url.psz_protocol, "file" ) )
 763     {
 764         if( p_sys->psz_sdp_file != NULL )
 765         {
 766             msg_Err( p_stream, "you can use sdp=file:// only once" );
 767             goto out;
 768         }
 769         p_sys->psz_sdp_file = make_path( psz_url );
 770         if( p_sys->psz_sdp_file == NULL )
 771             goto out;
 772         FileSetup( p_stream );
 773     }
 774     else
 775     {
 776         msg_Warn( p_stream, "unknown protocol for SDP (%s)",
 777                   url.psz_protocol );
 778     }
 779
 780 out:
 781     vlc_UrlClean( &url );
 782 }
 783
 784 /*****************************************************************************
 785  * SDPGenerate
 786  *****************************************************************************/
 787 /*static*/
 788 char *SDPGenerate( sout_stream_t *p_stream, const char *rtsp_url )
 789 {
 790     sout_stream_sys_t *p_sys = p_stream->p_sys;
 791     char *psz_sdp = NULL;
 792     struct sockaddr_storage dst;
 793     socklen_t dstlen;
 794     int i;
 795     /*
 796      * When we have a fixed destination (typically when we do multicast),
 797      * we need to put the actual port numbers in the SDP.
 798      * When there is no fixed destination, we only support RTSP unicast
 799      * on-demand setup, so we should rather let the clients decide which ports
 800      * to use.
 801      * When there is both a fixed destination and RTSP unicast, we need to
 802      * put port numbers used by the fixed destination, otherwise the SDP would
 803      * become totally incorrect for multicast use. It should be noted that
 804      * port numbers from SDP with RTSP are only "recommendation" from the
 805      * server to the clients (per RFC2326), so only broken clients will fail
 806      * to handle this properly. There is no solution but to use two differents
 807      * output chain with two different RTSP URLs if you need to handle this
 808      * scenario.
 809      */
 810     int inclport;
 811
 812     vlc_mutex_lock( &p_sys->lock_es );
 813     if( unlikely(p_sys->i_es == 0 || (rtsp_url != NULL && !p_sys->es[0]->rtsp_id)) )
 814         goto out; /* hmm... */
 815
 816     if( p_sys->psz_destination != NULL )
 817     {
 818         inclport = 1;
 819
 820         /* Oh boy, this is really ugly! */
 821         dstlen = sizeof( dst );
 822         if( p_sys->es[0]->listen.fd != NULL )
 823             getsockname( p_sys->es[0]->listen.fd[0],
 824                          (struct sockaddr *)&dst, &dstlen );
 825         else
 826             getpeername( p_sys->es[0]->sinkv[0].rtp_fd,
 827                          (struct sockaddr *)&dst, &dstlen );
 828     }
 829     else
 830     {
 831         inclport = 0;
 832
 833         /* Check against URL format rtsp://[<ipv6>]:<port>/<path> */
 834         bool ipv6 = rtsp_url != NULL && strlen( rtsp_url ) > 7
 835                     && rtsp_url[7] == '[';
 836
 837         /* Dummy destination address for RTSP */
 838         dstlen = ipv6 ? sizeof( struct sockaddr_in6 )
 839                       : sizeof( struct sockaddr_in );
 840         memset (&dst, 0, dstlen);
 841         dst.ss_family = ipv6 ? AF_INET6 : AF_INET;
 842 #ifdef HAVE_SA_LEN
 843         dst.ss_len = dstlen;
 844 #endif
 845     }
 846
 847     psz_sdp = vlc_sdp_Start( VLC_OBJECT( p_stream ), SOUT_CFG_PREFIX,
 848                              NULL, 0, (struct sockaddr *)&dst, dstlen );
 849     if( psz_sdp == NULL )
 850         goto out;
 851
 852     /* TODO: a=source-filter */
 853     if( p_sys->rtcp_mux )
 854         sdp_AddAttribute( &psz_sdp, "rtcp-mux", NULL );
 855
 856     if( rtsp_url != NULL )
 857         sdp_AddAttribute ( &psz_sdp, "control", "%s", rtsp_url );
 858
 859     const char *proto = "RTP/AVP"; /* protocol */
 860     if( rtsp_url == NULL )
 861     {
 862         switch( p_sys->proto )
 863         {
 864             case IPPROTO_UDP:
 865                 break;
 866             case IPPROTO_TCP:
 867                 proto = "TCP/RTP/AVP";
 868                 break;
 869             case IPPROTO_DCCP:
 870                 proto = "DCCP/RTP/AVP";
 871                 break;
 872             case IPPROTO_UDPLITE:
 873                 return psz_sdp;
 874         }
 875     }
 876
 877     for( i = 0; i < p_sys->i_es; i++ )
 878     {
 879         sout_stream_id_t *id = p_sys->es[i];
 880         rtp_format_t *rtp_fmt = &id->rtp_fmt;
 881         const char *mime_major; /* major MIME type */
 882
 883         switch( rtp_fmt->cat )
 884         {
 885             case VIDEO_ES:
 886                 mime_major = "video";
 887                 break;
 888             case AUDIO_ES:
 889                 mime_major = "audio";
 890                 break;
 891             case SPU_ES:
 892                 mime_major = "text";
 893                 break;
 894             default:
 895                 continue;
 896         }
 897
 898         sdp_AddMedia( &psz_sdp, mime_major, proto, inclport * id->i_port,
 899                       rtp_fmt->payload_type, false, rtp_fmt->bitrate,
 900                       rtp_fmt->ptname, rtp_fmt->clock_rate, rtp_fmt->channels,
 901                       rtp_fmt->fmtp);
 902
 903         /* cf RFC4566 §5.14 */
 904         if( inclport && !p_sys->rtcp_mux && (id->i_port & 1) )
 905             sdp_AddAttribute ( &psz_sdp, "rtcp", "%u", id->i_port + 1 );
 906
 907         if( rtsp_url != NULL )
 908         {
 909             char *track_url = RtspAppendTrackPath( id->rtsp_id, rtsp_url );
 910             if( track_url != NULL )
 911             {
 912                 sdp_AddAttribute ( &psz_sdp, "control", "%s", track_url );
 913                 free( track_url );
 914             }
 915         }
 916         else
 917         {
 918             if( id->listen.fd != NULL )
 919                 sdp_AddAttribute( &psz_sdp, "setup", "passive" );
 920             if( p_sys->proto == IPPROTO_DCCP )
 921                 sdp_AddAttribute( &psz_sdp, "dccp-service-code",
 922                                   "SC:RTP%c",
 923                                   toupper( (unsigned char)mime_major[0] ) );
 924         }
 925     }
 926 out:
 927     vlc_mutex_unlock( &p_sys->lock_es );
 928     return psz_sdp;
 929 }
 930
 931 /*****************************************************************************
 932  * RTP mux
 933  *****************************************************************************/
 934
 935 /**
 936  * Shrink the MTU down to a fixed packetization time (for audio).
 937  */
 938 static void
 939 rtp_set_ptime (sout_stream_id_t *id, unsigned ptime_ms, size_t bytes)
 940 {
 941     /* Samples per second */
 942     size_t spl = (id->rtp_fmt.clock_rate - 1) * ptime_ms / 1000 + 1;
 943     bytes *= id->rtp_fmt.channels;
 944     spl *= bytes;
 945
 946     if (spl < rtp_mtu (id)) /* MTU is big enough for ptime */
 947         id->i_mtu = 12 + spl;
 948     else /* MTU is too small for ptime, align to a sample boundary */
 949         id->i_mtu = 12 + (((id->i_mtu - 12) / bytes) * bytes);
 950 }
 951
 952 uint32_t rtp_compute_ts( unsigned i_clock_rate, int64_t i_pts )
 953 {
 954     /* This is an overflow-proof way of doing:
 955      * return i_pts * (int64_t)i_clock_rate / CLOCK_FREQ;
 956      *
 957      * NOTE: this plays nice with offsets because the (equivalent)
 958      * calculations are linear. */
 959     lldiv_t q = lldiv(i_pts, CLOCK_FREQ);
 960     return q.quot * (int64_t)i_clock_rate
 961           + q.rem * (int64_t)i_clock_rate / CLOCK_FREQ;
 962 }
 963
 964 /** Add an ES as a new RTP stream */
 965 static sout_stream_id_t *Add( sout_stream_t *p_stream, es_format_t *p_fmt )
 966 {
 967     /* NOTE: As a special case, if we use a non-RTP
 968      * mux (TS/PS), then p_fmt is NULL. */
 969     sout_stream_sys_t *p_sys = p_stream->p_sys;
 970     char              *psz_sdp;
 971
 972     sout_stream_id_t *id = malloc( sizeof( *id ) );
 973     if( unlikely(id == NULL) )
 974         return NULL;
 975     id->p_stream   = p_stream;
 976
 977     id->i_mtu = var_InheritInteger( p_stream, "mtu" );
 978     if( id->i_mtu <= 12 + 16 )
 979         id->i_mtu = 576 - 20 - 8; /* pessimistic */
 980     msg_Dbg( p_stream, "maximum RTP packet size: %d bytes", id->i_mtu );
 981
 982 #ifdef HAVE_SRTP
 983     id->srtp = NULL;
 984 #endif
 985     vlc_mutex_init( &id->lock_sink );
 986     id->sinkc = 0;
 987     id->sinkv = NULL;
 988     id->rtsp_id = NULL;
 989     id->p_fifo = NULL;
 990     id->listen.fd = NULL;
 991
 992     id->b_first_packet = true;
 993     id->i_caching =
 994         (int64_t)1000 * var_GetInteger( p_stream, SOUT_CFG_PREFIX "caching");
 995
 996     vlc_rand_bytes (&id->i_sequence, sizeof (id->i_sequence));
 997     vlc_rand_bytes (id->ssrc, sizeof (id->ssrc));
 998
 999     bool format = false;
1000
1001     if (p_sys->p_vod_media != NULL)
1002     {
1003         id->rtp_fmt.ptname = NULL;
1004         uint32_t ssrc;
1005         int val = vod_init_id(p_sys->p_vod_media, p_sys->psz_vod_session,
1006                               p_fmt ? p_fmt->i_id : 0, id, &id->rtp_fmt,
1007                               &ssrc, &id->i_seq_sent_next);
1008         if (val == VLC_SUCCESS)
1009         {
1010             memcpy(id->ssrc, &ssrc, sizeof(id->ssrc));
1011             /* This is ugly, but id->i_seq_sent_next needs to be
1012              * initialized inside vod_init_id() to avoid race
1013              * conditions. */
1014             id->i_sequence = id->i_seq_sent_next;
1015         }
1016         /* vod_init_id() may fail either because the ES wasn't found in
1017          * the VoD media, or because the RTSP session is gone. In the
1018          * former case, id->rtp_fmt was left untouched. */
1019         format = (id->rtp_fmt.ptname != NULL);
1020     }
1021
1022     if (!format)
1023     {
1024         id->rtp_fmt.fmtp = NULL; /* don't free() garbage on error */
1025         char *psz = var_GetNonEmptyString( p_stream, SOUT_CFG_PREFIX "mux" );
1026         if (p_fmt == NULL && psz == NULL)
1027             goto error;
1028         int val = rtp_get_fmt(VLC_OBJECT(p_stream), p_fmt, psz, &id->rtp_fmt);
1029         free( psz );
1030         if (val != VLC_SUCCESS)
1031             goto error;
1032     }
1033
1034 #ifdef HAVE_SRTP
1035     char *key = var_GetNonEmptyString (p_stream, SOUT_CFG_PREFIX"key");
1036     if (key)
1037     {
1038         vlc_gcrypt_init ();
1039         id->srtp = srtp_create (SRTP_ENCR_AES_CM, SRTP_AUTH_HMAC_SHA1, 10,
1040                                    SRTP_PRF_AES_CM, SRTP_RCC_MODE1);
1041         if (id->srtp == NULL)
1042         {
1043             free (key);
1044             goto error;
1045         }
1046
1047         char *salt = var_GetNonEmptyString (p_stream, SOUT_CFG_PREFIX"salt");
1048         errno = srtp_setkeystring (id->srtp, key, salt ? salt : "");
1049         free (salt);
1050         free (key);
1051         if (errno)
1052         {
1053             msg_Err (p_stream, "bad SRTP key/salt combination (%m)");
1054             goto error;
1055         }
1056         id->i_sequence = 0; /* FIXME: awful hack for libvlc_srtp */
1057     }
1058 #endif
1059
1060     id->i_seq_sent_next = id->i_sequence;
1061
1062     int mcast_fd = -1;
1063     if( p_sys->psz_destination != NULL )
1064     {
1065         /* Choose the port */
1066         uint16_t i_port = 0;
1067         if( p_fmt == NULL )
1068             ;
1069         else
1070         if( p_fmt->i_cat == AUDIO_ES && p_sys->i_port_audio > 0 )
1071             i_port = p_sys->i_port_audio;
1072         else
1073         if( p_fmt->i_cat == VIDEO_ES && p_sys->i_port_video > 0 )
1074             i_port = p_sys->i_port_video;
1075
1076         /* We do not need the ES lock (p_sys->lock_es) here, because
1077          * this is the only one thread that can *modify* the ES table.
1078          * The ES lock protects the other threads from our modifications
1079          * (TAB_APPEND, TAB_REMOVE). */
1080         for (int i = 0; i_port && (i < p_sys->i_es); i++)
1081              if (i_port == p_sys->es[i]->i_port)
1082                  i_port = 0; /* Port already in use! */
1083         for (uint16_t p = p_sys->i_port; i_port == 0; p += 2)
1084         {
1085             if (p == 0)
1086             {
1087                 msg_Err (p_stream, "too many RTP elementary streams");
1088                 goto error;
1089             }
1090             i_port = p;
1091             for (int i = 0; i_port && (i < p_sys->i_es); i++)
1092                  if (p == p_sys->es[i]->i_port)
1093                      i_port = 0;
1094         }
1095
1096         id->i_port = i_port;
1097
1098         int type = SOCK_STREAM;
1099
1100         switch( p_sys->proto )
1101         {
1102 #ifdef SOCK_DCCP
1103             case IPPROTO_DCCP:
1104             {
1105                 const char *code;
1106                 switch (id->rtp_fmt.cat)
1107                 {
1108                     case VIDEO_ES: code = "RTPV";     break;
1109                     case AUDIO_ES: code = "RTPARTPV"; break;
1110                     case SPU_ES:   code = "RTPTRTPV"; break;
1111                     default:       code = "RTPORTPV"; break;
1112                 }
1113                 var_SetString (p_stream, "dccp-service", code);
1114                 type = SOCK_DCCP;
1115             }   /* fall through */
1116 #endif
1117             case IPPROTO_TCP:
1118                 id->listen.fd = net_Listen( VLC_OBJECT(p_stream),
1119                                             p_sys->psz_destination, i_port,
1120                                             type, p_sys->proto );
1121                 if( id->listen.fd == NULL )
1122                 {
1123                     msg_Err( p_stream, "passive COMEDIA RTP socket failed" );
1124                     goto error;
1125                 }
1126                 if( vlc_clone( &id->listen.thread, rtp_listen_thread, id,
1127                                VLC_THREAD_PRIORITY_LOW ) )
1128                 {
1129                     net_ListenClose( id->listen.fd );
1130                     id->listen.fd = NULL;
1131                     goto error;
1132                 }
1133                 break;
1134
1135             default:
1136             {
1137                 int fd = net_ConnectDgram( p_stream, p_sys->psz_destination,
1138                                            i_port, -1, p_sys->proto );
1139                 if( fd == -1 )
1140                 {
1141                     msg_Err( p_stream, "cannot create RTP socket" );
1142                     goto error;
1143                 }
1144                 /* Ignore any unexpected incoming packet (including RTCP-RR
1145                  * packets in case of rtcp-mux) */
1146                 setsockopt (fd, SOL_SOCKET, SO_RCVBUF, &(int){ 0 },
1147                             sizeof (int));
1148                 rtp_add_sink( id, fd, p_sys->rtcp_mux, NULL );
1149                 /* FIXME: test if this is multicast  */
1150                 mcast_fd = fd;
1151             }
1152         }
1153     }
1154
1155     if( p_fmt != NULL )
1156     switch( p_fmt->i_codec )
1157     {
1158         case VLC_CODEC_MULAW:
1159         case VLC_CODEC_ALAW:
1160         case VLC_CODEC_U8:
1161             rtp_set_ptime (id, 20, 1);
1162             break;
1163         case VLC_CODEC_S16B:
1164         case VLC_CODEC_S16L:
1165             rtp_set_ptime (id, 20, 2);
1166             break;
1167         default:
1168             break;
1169     }
1170
1171 #if 0 /* No payload formats sets this at the moment */
1172     int cscov = -1;
1173     if( cscov != -1 )
1174         cscov += 8 /* UDP */ + 12 /* RTP */;
1175     if( id->sinkc > 0 )
1176         net_SetCSCov( id->sinkv[0].rtp_fd, cscov, -1 );
1177 #endif
1178
1179     vlc_mutex_lock( &p_sys->lock_ts );
1180     id->b_ts_init = ( p_sys->i_npt_zero != VLC_TS_INVALID );
1181     vlc_mutex_unlock( &p_sys->lock_ts );
1182     if( id->b_ts_init )
1183         id->i_ts_offset = rtp_compute_ts( id->rtp_fmt.clock_rate,
1184                                           p_sys->i_pts_offset );
1185
1186     if( p_sys->rtsp != NULL )
1187         id->rtsp_id = RtspAddId( p_sys->rtsp, id, GetDWBE( id->ssrc ),
1188                                  id->rtp_fmt.clock_rate, mcast_fd );
1189
1190     id->p_fifo = block_FifoNew();
1191     if( unlikely(id->p_fifo == NULL) )
1192         goto error;
1193     if( vlc_clone( &id->thread, ThreadSend, id, VLC_THREAD_PRIORITY_HIGHEST ) )
1194     {
1195         block_FifoRelease( id->p_fifo );
1196         id->p_fifo = NULL;
1197         goto error;
1198     }
1199
1200     /* Update p_sys context */
1201     vlc_mutex_lock( &p_sys->lock_es );
1202     TAB_APPEND( p_sys->i_es, p_sys->es, id );
1203     vlc_mutex_unlock( &p_sys->lock_es );
1204
1205     psz_sdp = SDPGenerate( p_stream, NULL );
1206
1207     vlc_mutex_lock( &p_sys->lock_sdp );
1208     free( p_sys->psz_sdp );
1209     p_sys->psz_sdp = psz_sdp;
1210     vlc_mutex_unlock( &p_sys->lock_sdp );
1211
1212     msg_Dbg( p_stream, "sdp=\n%s", p_sys->psz_sdp );
1213
1214     /* Update SDP (sap/file) */
1215     if( p_sys->b_export_sap ) SapSetup( p_stream );
1216     if( p_sys->psz_sdp_file != NULL ) FileSetup( p_stream );
1217
1218     return id;
1219
1220 error:
1221     Del( p_stream, id );
1222     return NULL;
1223 }
1224
1225 static int Del( sout_stream_t *p_stream, sout_stream_id_t *id )
1226 {
1227     sout_stream_sys_t *p_sys = p_stream->p_sys;
1228
1229     vlc_mutex_lock( &p_sys->lock_es );
1230     TAB_REMOVE( p_sys->i_es, p_sys->es, id );
1231     vlc_mutex_unlock( &p_sys->lock_es );
1232
1233     if( likely(id->p_fifo != NULL) )
1234     {
1235         vlc_cancel( id->thread );
1236         vlc_join( id->thread, NULL );
1237         block_FifoRelease( id->p_fifo );
1238     }
1239
1240     free( id->rtp_fmt.fmtp );
1241
1242     if (p_sys->p_vod_media != NULL)
1243         vod_detach_id(p_sys->p_vod_media, p_sys->psz_vod_session, id);
1244     if( id->rtsp_id )
1245         RtspDelId( p_sys->rtsp, id->rtsp_id );
1246     if( id->listen.fd != NULL )
1247     {
1248         vlc_cancel( id->listen.thread );
1249         vlc_join( id->listen.thread, NULL );
1250         net_ListenClose( id->listen.fd );
1251     }
1252     /* Delete remaining sinks (incoming connections or explicit
1253      * outgoing dst=) */
1254     while( id->sinkc > 0 )
1255         rtp_del_sink( id, id->sinkv[0].rtp_fd );
1256 #ifdef HAVE_SRTP
1257     if( id->srtp != NULL )
1258         srtp_destroy( id->srtp );
1259 #endif
1260
1261     vlc_mutex_destroy( &id->lock_sink );
1262
1263     /* Update SDP (sap/file) */
1264     if( p_sys->b_export_sap ) SapSetup( p_stream );
1265     if( p_sys->psz_sdp_file != NULL ) FileSetup( p_stream );
1266
1267     free( id );
1268     return VLC_SUCCESS;
1269 }
1270
1271 static int Send( sout_stream_t *p_stream, sout_stream_id_t *id,
1272                  block_t *p_buffer )
1273 {
1274     block_t *p_next;
1275
1276     assert( p_stream->p_sys->p_mux == NULL );
1277     (void)p_stream;
1278
1279     while( p_buffer != NULL )
1280     {
1281         p_next = p_buffer->p_next;
1282
1283         /* Send a Vorbis/Theora Packed Configuration packet (RFC 5215 §3.1)
1284          * as the first packet of the stream */
1285         if (id->b_first_packet)
1286         {
1287             id->b_first_packet = false;
1288             if (!strcmp(id->rtp_fmt.ptname, "vorbis") ||
1289                 !strcmp(id->rtp_fmt.ptname, "theora"))
1290                 rtp_packetize_xiph_config(id, id->rtp_fmt.fmtp,
1291                                           p_buffer->i_pts);
1292         }
1293
1294         if( id->rtp_fmt.pf_packetize( id, p_buffer ) )
1295             break;
1296
1297         block_Release( p_buffer );
1298         p_buffer = p_next;
1299     }
1300     return VLC_SUCCESS;
1301 }
1302
1303 /****************************************************************************
1304  * SAP:
1305  ****************************************************************************/
1306 static int SapSetup( sout_stream_t *p_stream )
1307 {
1308     sout_stream_sys_t *p_sys = p_stream->p_sys;
1309     sout_instance_t   *p_sout = p_stream->p_sout;
1310
1311     /* Remove the previous session */
1312     if( p_sys->p_session != NULL)
1313     {
1314         sout_AnnounceUnRegister( p_sout, p_sys->p_session);
1315         p_sys->p_session = NULL;
1316     }
1317
1318     if( p_sys->i_es > 0 && p_sys->psz_sdp && *p_sys->psz_sdp )
1319         p_sys->p_session = sout_AnnounceRegisterSDP( p_sout,
1320                                                      p_sys->psz_sdp,
1321                                                      p_sys->psz_destination );
1322
1323     return VLC_SUCCESS;
1324 }
1325
1326 /****************************************************************************
1327 * File:
1328 ****************************************************************************/
1329 static int FileSetup( sout_stream_t *p_stream )
1330 {
1331     sout_stream_sys_t *p_sys = p_stream->p_sys;
1332     FILE            *f;
1333
1334     if( p_sys->psz_sdp == NULL )
1335         return VLC_EGENERIC; /* too early */
1336
1337     if( ( f = vlc_fopen( p_sys->psz_sdp_file, "wt" ) ) == NULL )
1338     {
1339         msg_Err( p_stream, "cannot open file '%s' (%m)",
1340                  p_sys->psz_sdp_file );
1341         return VLC_EGENERIC;
1342     }
1343
1344     fputs( p_sys->psz_sdp, f );
1345     fclose( f );
1346
1347     return VLC_SUCCESS;
1348 }
1349
1350 /****************************************************************************
1351  * HTTP:
1352  ****************************************************************************/
1353 static int  HttpCallback( httpd_file_sys_t *p_args,
1354                           httpd_file_t *, uint8_t *p_request,
1355                           uint8_t **pp_data, int *pi_data );
1356
1357 static int HttpSetup( sout_stream_t *p_stream, const vlc_url_t *url)
1358 {
1359     sout_stream_sys_t *p_sys = p_stream->p_sys;
1360
1361     p_sys->p_httpd_host = vlc_http_HostNew( VLC_OBJECT(p_stream) );
1362     if( p_sys->p_httpd_host )
1363     {
1364         p_sys->p_httpd_file = httpd_FileNew( p_sys->p_httpd_host,
1365                                              url->psz_path ? url->psz_path : "/",
1366                                              "application/sdp",
1367                                              NULL, NULL,
1368                                              HttpCallback, (void*)p_sys );
1369     }
1370     if( p_sys->p_httpd_file == NULL )
1371     {
1372         return VLC_EGENERIC;
1373     }
1374     return VLC_SUCCESS;
1375 }
1376
1377 static int  HttpCallback( httpd_file_sys_t *p_args,
1378                           httpd_file_t *f, uint8_t *p_request,
1379                           uint8_t **pp_data, int *pi_data )
1380 {
1381     VLC_UNUSED(f); VLC_UNUSED(p_request);
1382     sout_stream_sys_t *p_sys = (sout_stream_sys_t*)p_args;
1383
1384     vlc_mutex_lock( &p_sys->lock_sdp );
1385     if( p_sys->psz_sdp && *p_sys->psz_sdp )
1386     {
1387         *pi_data = strlen( p_sys->psz_sdp );
1388         *pp_data = malloc( *pi_data );
1389         memcpy( *pp_data, p_sys->psz_sdp, *pi_data );
1390     }
1391     else
1392     {
1393         *pp_data = NULL;
1394         *pi_data = 0;
1395     }
1396     vlc_mutex_unlock( &p_sys->lock_sdp );
1397
1398     return VLC_SUCCESS;
1399 }
1400
1401 /****************************************************************************
1402  * RTP send
1403  ****************************************************************************/
1404 static void* ThreadSend( void *data )
1405 {
1406 #ifdef WIN32
1407 # define ENOBUFS      WSAENOBUFS
1408 # define EAGAIN       WSAEWOULDBLOCK
1409 # define EWOULDBLOCK  WSAEWOULDBLOCK
1410 #endif
1411     sout_stream_id_t *id = data;
1412     unsigned i_caching = id->i_caching;
1413
1414     for (;;)
1415     {
1416         block_t *out = block_FifoGet( id->p_fifo );
1417         block_cleanup_push (out);
1418
1419 #ifdef HAVE_SRTP
1420         if( id->srtp )
1421         {   /* FIXME: this is awfully inefficient */
1422             size_t len = out->i_buffer;
1423             out = block_Realloc( out, 0, len + 10 );
1424             out->i_buffer = len;
1425
1426             int canc = vlc_savecancel ();
1427             int val = srtp_send( id->srtp, out->p_buffer, &len, len + 10 );
1428             vlc_restorecancel (canc);
1429             if( val )
1430             {
1431                 errno = val;
1432                 msg_Dbg( id->p_stream, "SRTP sending error: %m" );
1433                 block_Release( out );
1434                 out = NULL;
1435             }
1436             else
1437                 out->i_buffer = len;
1438         }
1439         if (out)
1440             mwait (out->i_dts + i_caching);
1441         vlc_cleanup_pop ();
1442         if (out == NULL)
1443             continue;
1444 #else
1445         mwait (out->i_dts + i_caching);
1446         vlc_cleanup_pop ();
1447 #endif
1448
1449         ssize_t len = out->i_buffer;
1450         int canc = vlc_savecancel ();
1451
1452         vlc_mutex_lock( &id->lock_sink );
1453         unsigned deadc = 0; /* How many dead sockets? */
1454         int deadv[id->sinkc]; /* Dead sockets list */
1455
1456         for( int i = 0; i < id->sinkc; i++ )
1457         {
1458 #ifdef HAVE_SRTP
1459             if( !id->srtp ) /* FIXME: SRTCP support */
1460 #endif
1461                 SendRTCP( id->sinkv[i].rtcp, out );
1462
1463             if( send( id->sinkv[i].rtp_fd, out->p_buffer, len, 0 ) == -1
1464              && net_errno != EAGAIN && net_errno != EWOULDBLOCK
1465              && net_errno != ENOBUFS && net_errno != ENOMEM )
1466             {
1467                 int type;
1468                 getsockopt( id->sinkv[i].rtp_fd, SOL_SOCKET, SO_TYPE,
1469                             &type, &(socklen_t){ sizeof(type) });
1470                 if( type == SOCK_DGRAM )
1471                     /* ICMP soft error: ignore and retry */
1472                     send( id->sinkv[i].rtp_fd, out->p_buffer, len, 0 );
1473                 else
1474                     /* Broken connection */
1475                     deadv[deadc++] = id->sinkv[i].rtp_fd;
1476             }
1477         }
1478         id->i_seq_sent_next = ntohs(((uint16_t *) out->p_buffer)[1]) + 1;
1479         vlc_mutex_unlock( &id->lock_sink );
1480         block_Release( out );
1481
1482         for( unsigned i = 0; i < deadc; i++ )
1483         {
1484             msg_Dbg( id->p_stream, "removing socket %d", deadv[i] );
1485             rtp_del_sink( id, deadv[i] );
1486         }
1487         vlc_restorecancel (canc);
1488     }
1489     return NULL;
1490 }
1491
1492
1493 /* This thread dequeues incoming connections (DCCP streaming) */
1494 static void *rtp_listen_thread( void *data )
1495 {
1496     sout_stream_id_t *id = data;
1497
1498     assert( id->listen.fd != NULL );
1499
1500     for( ;; )
1501     {
1502         int fd = net_Accept( id->p_stream, id->listen.fd );
1503         if( fd == -1 )
1504             continue;
1505         int canc = vlc_savecancel( );
1506         rtp_add_sink( id, fd, true, NULL );
1507         vlc_restorecancel( canc );
1508     }
1509
1510     assert( 0 );
1511 }
1512
1513
1514 int rtp_add_sink( sout_stream_id_t *id, int fd, bool rtcp_mux, uint16_t *seq )
1515 {
1516     rtp_sink_t sink = { fd, NULL };
1517     sink.rtcp = OpenRTCP( VLC_OBJECT( id->p_stream ), fd, IPPROTO_UDP,
1518                           rtcp_mux );
1519     if( sink.rtcp == NULL )
1520         msg_Err( id->p_stream, "RTCP failed!" );
1521
1522     vlc_mutex_lock( &id->lock_sink );
1523     INSERT_ELEM( id->sinkv, id->sinkc, id->sinkc, sink );
1524     if( seq != NULL )
1525         *seq = id->i_seq_sent_next;
1526     vlc_mutex_unlock( &id->lock_sink );
1527     return VLC_SUCCESS;
1528 }
1529
1530 void rtp_del_sink( sout_stream_id_t *id, int fd )
1531 {
1532     rtp_sink_t sink = { fd, NULL };
1533
1534     /* NOTE: must be safe to use if fd is not included */
1535     vlc_mutex_lock( &id->lock_sink );
1536     for( int i = 0; i < id->sinkc; i++ )
1537     {
1538         if (id->sinkv[i].rtp_fd == fd)
1539         {
1540             sink = id->sinkv[i];
1541             REMOVE_ELEM( id->sinkv, id->sinkc, i );
1542             break;
1543         }
1544     }
1545     vlc_mutex_unlock( &id->lock_sink );
1546
1547     CloseRTCP( sink.rtcp );
1548     net_Close( sink.rtp_fd );
1549 }
1550
1551 uint16_t rtp_get_seq( sout_stream_id_t *id )
1552 {
1553     /* This will return values for the next packet. */
1554     uint16_t seq;
1555
1556     vlc_mutex_lock( &id->lock_sink );
1557     seq = id->i_seq_sent_next;
1558     vlc_mutex_unlock( &id->lock_sink );
1559
1560     return seq;
1561 }
1562
1563 /* Return an arbitrary initial timestamp for RTP timestamp computations.
1564  * RFC 3550 states that the resulting initial RTP timestamps SHOULD be
1565  * random (although we use the same reference for all the ES as a
1566  * feature). In the VoD case, this function is called independently
1567  * from several parts of the code, so we need to always return the same
1568  * value. */
1569 static int64_t rtp_init_ts( const vod_media_t *p_media,
1570                             const char *psz_vod_session )
1571 {
1572     if (p_media == NULL || psz_vod_session == NULL)
1573         return mdate();
1574
1575     uint64_t i_ts_init;
1576     /* As per RFC 2326, session identifiers are at least 8 bytes long */
1577     strncpy((char *)&i_ts_init, psz_vod_session, sizeof(uint64_t));
1578     i_ts_init ^= (uintptr_t)p_media;
1579     /* Limit the timestamp to 48 bits, this is enough and allows us
1580      * to stay away from overflows */
1581     i_ts_init &= 0xFFFFFFFFFFFF;
1582     return i_ts_init;
1583 }
1584
1585 /* Return a timestamp corresponding to packets being sent now, and that
1586  * can be passed to rtp_compute_ts() to get rtptime values for each ES.
1587  * Also return the NPT corresponding to this timestamp. If the stream
1588  * output is not started, the initial timestamp that will be used with
1589  * the first packets for NPT=0 is returned instead. */
1590 int64_t rtp_get_ts( const sout_stream_t *p_stream, const sout_stream_id_t *id,
1591                     const vod_media_t *p_media, const char *psz_vod_session,
1592                     int64_t *p_npt )
1593 {
1594     if (p_npt != NULL)
1595         *p_npt = 0;
1596
1597     if (id != NULL)
1598         p_stream = id->p_stream;
1599
1600     if (p_stream == NULL)
1601         return rtp_init_ts(p_media, psz_vod_session);
1602
1603     sout_stream_sys_t *p_sys = p_stream->p_sys;
1604     mtime_t i_npt_zero;
1605     vlc_mutex_lock( &p_sys->lock_ts );
1606     i_npt_zero = p_sys->i_npt_zero;
1607     vlc_mutex_unlock( &p_sys->lock_ts );
1608
1609     if( i_npt_zero == VLC_TS_INVALID )
1610         return p_sys->i_pts_zero;
1611
1612     mtime_t now = mdate();
1613     if( now < i_npt_zero )
1614         return p_sys->i_pts_zero;
1615
1616     int64_t npt = now - i_npt_zero;
1617     if (p_npt != NULL)
1618         *p_npt = npt;
1619
1620     return p_sys->i_pts_zero + npt;
1621 }
1622
1623 void rtp_packetize_common( sout_stream_id_t *id, block_t *out,
1624                            int b_marker, int64_t i_pts )
1625 {
1626     if( !id->b_ts_init )
1627     {
1628         sout_stream_sys_t *p_sys = id->p_stream->p_sys;
1629         vlc_mutex_lock( &p_sys->lock_ts );
1630         if( p_sys->i_npt_zero == VLC_TS_INVALID )
1631         {
1632             /* This is the first packet of any ES. We initialize the
1633              * NPT=0 time reference, and the offset to match the
1634              * arbitrary PTS reference. */
1635             p_sys->i_npt_zero = i_pts + id->i_caching;
1636             p_sys->i_pts_offset = p_sys->i_pts_zero - i_pts;
1637         }
1638         vlc_mutex_unlock( &p_sys->lock_ts );
1639
1640         /* And in any case this is the first packet of this ES, so we
1641          * initialize the offset for this ES. */
1642         id->i_ts_offset = rtp_compute_ts( id->rtp_fmt.clock_rate,
1643                                           p_sys->i_pts_offset );
1644         id->b_ts_init = true;
1645     }
1646
1647     uint32_t i_timestamp = rtp_compute_ts( id->rtp_fmt.clock_rate, i_pts )
1648                            + id->i_ts_offset;
1649
1650     out->p_buffer[0] = 0x80;
1651     out->p_buffer[1] = (b_marker?0x80:0x00)|id->rtp_fmt.payload_type;
1652     out->p_buffer[2] = ( id->i_sequence >> 8)&0xff;
1653     out->p_buffer[3] = ( id->i_sequence     )&0xff;
1654     out->p_buffer[4] = ( i_timestamp >> 24 )&0xff;
1655     out->p_buffer[5] = ( i_timestamp >> 16 )&0xff;
1656     out->p_buffer[6] = ( i_timestamp >>  8 )&0xff;
1657     out->p_buffer[7] = ( i_timestamp       )&0xff;
1658
1659     memcpy( out->p_buffer + 8, id->ssrc, 4 );
1660
1661     out->i_buffer = 12;
1662     id->i_sequence++;
1663 }
1664
1665 void rtp_packetize_send( sout_stream_id_t *id, block_t *out )
1666 {
1667     block_FifoPut( id->p_fifo, out );
1668 }
1669
1670 /**
1671  * @return configured max RTP payload size (including payload type-specific
1672  * headers, excluding RTP and transport headers)
1673  */
1674 size_t rtp_mtu (const sout_stream_id_t *id)
1675 {
1676     return id->i_mtu - 12;
1677 }
1678
1679 /*****************************************************************************
1680  * Non-RTP mux
1681  *****************************************************************************/
1682
1683 /** Add an ES to a non-RTP muxed stream */
1684 static sout_stream_id_t *MuxAdd( sout_stream_t *p_stream, es_format_t *p_fmt )
1685 {
1686     sout_input_t      *p_input;
1687     sout_mux_t *p_mux = p_stream->p_sys->p_mux;
1688     assert( p_mux != NULL );
1689
1690     p_input = sout_MuxAddStream( p_mux, p_fmt );
1691     if( p_input == NULL )
1692     {
1693         msg_Err( p_stream, "cannot add this stream to the muxer" );
1694         return NULL;
1695     }
1696
1697     return (sout_stream_id_t *)p_input;
1698 }
1699
1700
1701 static int MuxSend( sout_stream_t *p_stream, sout_stream_id_t *id,
1702                     block_t *p_buffer )
1703 {
1704     sout_mux_t *p_mux = p_stream->p_sys->p_mux;
1705     assert( p_mux != NULL );
1706
1707     sout_MuxSendBuffer( p_mux, (sout_input_t *)id, p_buffer );
1708     return VLC_SUCCESS;
1709 }
1710
1711
1712 /** Remove an ES from a non-RTP muxed stream */
1713 static int MuxDel( sout_stream_t *p_stream, sout_stream_id_t *id )
1714 {
1715     sout_mux_t *p_mux = p_stream->p_sys->p_mux;
1716     assert( p_mux != NULL );
1717
1718     sout_MuxDeleteStream( p_mux, (sout_input_t *)id );
1719     return VLC_SUCCESS;
1720 }
1721
1722
1723 static ssize_t AccessOutGrabberWriteBuffer( sout_stream_t *p_stream,
1724                                             const block_t *p_buffer )
1725 {
1726     sout_stream_sys_t *p_sys = p_stream->p_sys;
1727     sout_stream_id_t *id = p_sys->es[0];
1728
1729     int64_t  i_dts = p_buffer->i_dts;
1730
1731     uint8_t         *p_data = p_buffer->p_buffer;
1732     size_t          i_data  = p_buffer->i_buffer;
1733     size_t          i_max   = id->i_mtu - 12;
1734
1735     size_t i_packet = ( p_buffer->i_buffer + i_max - 1 ) / i_max;
1736
1737     while( i_data > 0 )
1738     {
1739         size_t i_size;
1740
1741         /* output complete packet */
1742         if( p_sys->packet &&
1743             p_sys->packet->i_buffer + i_data > i_max )
1744         {
1745             rtp_packetize_send( id, p_sys->packet );
1746             p_sys->packet = NULL;
1747         }
1748
1749         if( p_sys->packet == NULL )
1750         {
1751             /* allocate a new packet */
1752             p_sys->packet = block_New( p_stream, id->i_mtu );
1753             rtp_packetize_common( id, p_sys->packet, 1, i_dts );
1754             p_sys->packet->i_dts = i_dts;
1755             p_sys->packet->i_length = p_buffer->i_length / i_packet;
1756             i_dts += p_sys->packet->i_length;
1757         }
1758
1759         i_size = __MIN( i_data,
1760                         (unsigned)(id->i_mtu - p_sys->packet->i_buffer) );
1761
1762         memcpy( &p_sys->packet->p_buffer[p_sys->packet->i_buffer],
1763                 p_data, i_size );
1764
1765         p_sys->packet->i_buffer += i_size;
1766         p_data += i_size;
1767         i_data -= i_size;
1768     }
1769
1770     return VLC_SUCCESS;
1771 }
1772
1773
1774 static ssize_t AccessOutGrabberWrite( sout_access_out_t *p_access,
1775                                       block_t *p_buffer )
1776 {
1777     sout_stream_t *p_stream = (sout_stream_t*)p_access->p_sys;
1778
1779     while( p_buffer )
1780     {
1781         block_t *p_next;
1782
1783         AccessOutGrabberWriteBuffer( p_stream, p_buffer );
1784
1785         p_next = p_buffer->p_next;
1786         block_Release( p_buffer );
1787         p_buffer = p_next;
1788     }
1789
1790     return VLC_SUCCESS;
1791 }
1792
1793
1794 static sout_access_out_t *GrabberCreate( sout_stream_t *p_stream )
1795 {
1796     sout_access_out_t *p_grab;
1797
1798     p_grab = vlc_object_create( p_stream, sizeof( *p_grab ) );
1799     if( p_grab == NULL )
1800         return NULL;
1801
1802     p_grab->p_module    = NULL;
1803     p_grab->psz_access  = strdup( "grab" );
1804     p_grab->p_cfg       = NULL;
1805     p_grab->psz_path    = strdup( "" );
1806     p_grab->p_sys       = (sout_access_out_sys_t *)p_stream;
1807     p_grab->pf_seek     = NULL;
1808     p_grab->pf_write    = AccessOutGrabberWrite;
1809     return p_grab;
1810 }