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