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