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