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     int i;
 869
 870     for( i = 0; i < i_data; i++ )
 871     {
 872         s[2*i+0] = hex[(p_data[i]>>4)&0xf];
 873         s[2*i+1] = hex[(p_data[i]   )&0xf];
 874     }
 875     s[2*i_data] = '\0';
 876 }
 877
 878 /**
 879  * Shrink the MTU down to a fixed packetization time (for audio).
 880  */
 881 static void
 882 rtp_set_ptime (sout_stream_id_t *id, unsigned ptime_ms, size_t bytes)
 883 {
 884     /* Samples per second */
 885     size_t spl = (id->i_clock_rate - 1) * ptime_ms / 1000 + 1;
 886     bytes *= id->i_channels;
 887     spl *= bytes;
 888
 889     if (spl < rtp_mtu (id)) /* MTU is big enough for ptime */
 890         id->i_mtu = 12 + spl;
 891     else /* MTU is too small for ptime, align to a sample boundary */
 892         id->i_mtu = 12 + (((id->i_mtu - 12) / bytes) * bytes);
 893 }
 894
 895 uint32_t rtp_compute_ts( const sout_stream_id_t *id, int64_t i_pts )
 896 {
 897     /* NOTE: this plays nice with offsets because the calculations are
 898      * linear. */
 899     return i_pts * (int64_t)id->i_clock_rate / CLOCK_FREQ;
 900 }
 901
 902 /** Add an ES as a new RTP stream */
 903 static sout_stream_id_t *Add( sout_stream_t *p_stream, es_format_t *p_fmt )
 904 {
 905     /* NOTE: As a special case, if we use a non-RTP
 906      * mux (TS/PS), then p_fmt is NULL. */
 907     sout_stream_sys_t *p_sys = p_stream->p_sys;
 908     char              *psz_sdp;
 909
 910     if (0 == p_sys->payload_bitmap)
 911     {
 912         msg_Err (p_stream, "too many RTP elementary streams");
 913         return NULL;
 914     }
 915
 916     sout_stream_id_t *id = malloc( sizeof( *id ) );
 917     if( unlikely(id == NULL) )
 918         return NULL;
 919     id->p_stream   = p_stream;
 920
 921     /* Look for free dymanic payload type */
 922     id->i_payload_type = 96 + clz32 (p_sys->payload_bitmap);
 923     assert (id->i_payload_type < 128);
 924
 925     vlc_rand_bytes (&id->i_sequence, sizeof (id->i_sequence));
 926     vlc_rand_bytes (id->ssrc, sizeof (id->ssrc));
 927
 928     id->psz_enc    = NULL;
 929     id->psz_fmtp   = NULL;
 930     id->i_clock_rate = 90000; /* most common case for video */
 931     id->i_channels = 0;
 932     if( p_fmt != NULL )
 933     {
 934         id->i_cat  = p_fmt->i_cat;
 935         if( p_fmt->i_cat == AUDIO_ES )
 936         {
 937             id->i_clock_rate = p_fmt->audio.i_rate;
 938             id->i_channels = p_fmt->audio.i_channels;
 939         }
 940         id->i_bitrate = p_fmt->i_bitrate/1000; /* Stream bitrate in kbps */
 941     }
 942     else
 943     {
 944         id->i_cat  = VIDEO_ES;
 945         id->i_bitrate = 0;
 946     }
 947
 948     id->i_mtu = var_InheritInteger( p_stream, "mtu" );
 949     if( id->i_mtu <= 12 + 16 )
 950         id->i_mtu = 576 - 20 - 8; /* pessimistic */
 951     msg_Dbg( p_stream, "maximum RTP packet size: %d bytes", id->i_mtu );
 952
 953     id->pf_packetize = NULL;
 954
 955 #ifdef HAVE_SRTP
 956     id->srtp = NULL;
 957 #endif
 958     vlc_mutex_init( &id->lock_sink );
 959     id->sinkc = 0;
 960     id->sinkv = NULL;
 961     id->rtsp_id = NULL;
 962     id->p_fifo = NULL;
 963     id->listen.fd = NULL;
 964
 965     id->i_caching =
 966         (int64_t)1000 * var_GetInteger( p_stream, SOUT_CFG_PREFIX "caching");
 967
 968 #ifdef HAVE_SRTP
 969     char *key = var_CreateGetNonEmptyString (p_stream, SOUT_CFG_PREFIX"key");
 970     if (key)
 971     {
 972         id->srtp = srtp_create (SRTP_ENCR_AES_CM, SRTP_AUTH_HMAC_SHA1, 10,
 973                                    SRTP_PRF_AES_CM, SRTP_RCC_MODE1);
 974         if (id->srtp == NULL)
 975         {
 976             free (key);
 977             goto error;
 978         }
 979
 980         char *salt = var_CreateGetNonEmptyString (p_stream, SOUT_CFG_PREFIX"salt");
 981         errno = srtp_setkeystring (id->srtp, key, salt ? salt : "");
 982         free (salt);
 983         free (key);
 984         if (errno)
 985         {
 986             msg_Err (p_stream, "bad SRTP key/salt combination (%m)");
 987             goto error;
 988         }
 989         id->i_sequence = 0; /* FIXME: awful hack for libvlc_srtp */
 990     }
 991 #endif
 992
 993     id->i_seq_sent_next = id->i_sequence;
 994
 995     if( p_sys->psz_destination != NULL )
 996     {
 997         /* Choose the port */
 998         uint16_t i_port = 0;
 999         if( p_fmt == NULL )
1000             ;
1001         else
1002         if( p_fmt->i_cat == AUDIO_ES && p_sys->i_port_audio > 0 )
1003             i_port = p_sys->i_port_audio;
1004         else
1005         if( p_fmt->i_cat == VIDEO_ES && p_sys->i_port_video > 0 )
1006             i_port = p_sys->i_port_video;
1007
1008         /* We do not need the ES lock (p_sys->lock_es) here, because
1009          * this is the only one thread that can *modify* the ES table.
1010          * The ES lock protects the other threads from our modifications
1011          * (TAB_APPEND, TAB_REMOVE). */
1012         for (int i = 0; i_port && (i < p_sys->i_es); i++)
1013              if (i_port == p_sys->es[i]->i_port)
1014                  i_port = 0; /* Port already in use! */
1015         for (uint16_t p = p_sys->i_port; i_port == 0; p += 2)
1016         {
1017             if (p == 0)
1018             {
1019                 msg_Err (p_stream, "too many RTP elementary streams");
1020                 goto error;
1021             }
1022             i_port = p;
1023             for (int i = 0; i_port && (i < p_sys->i_es); i++)
1024                  if (p == p_sys->es[i]->i_port)
1025                      i_port = 0;
1026         }
1027
1028         id->i_port = i_port;
1029
1030         int type = SOCK_STREAM;
1031
1032         switch( p_sys->proto )
1033         {
1034 #ifdef SOCK_DCCP
1035             case IPPROTO_DCCP:
1036             {
1037                 const char *code;
1038                 switch (id->i_cat)
1039                 {
1040                     case VIDEO_ES: code = "RTPV";     break;
1041                     case AUDIO_ES: code = "RTPARTPV"; break;
1042                     case SPU_ES:   code = "RTPTRTPV"; break;
1043                     default:       code = "RTPORTPV"; break;
1044                 }
1045                 var_SetString (p_stream, "dccp-service", code);
1046                 type = SOCK_DCCP;
1047             }   /* fall through */
1048 #endif
1049             case IPPROTO_TCP:
1050                 id->listen.fd = net_Listen( VLC_OBJECT(p_stream),
1051                                             p_sys->psz_destination, i_port,
1052                                             type, p_sys->proto );
1053                 if( id->listen.fd == NULL )
1054                 {
1055                     msg_Err( p_stream, "passive COMEDIA RTP socket failed" );
1056                     goto error;
1057                 }
1058                 if( vlc_clone( &id->listen.thread, rtp_listen_thread, id,
1059                                VLC_THREAD_PRIORITY_LOW ) )
1060                 {
1061                     net_ListenClose( id->listen.fd );
1062                     id->listen.fd = NULL;
1063                     goto error;
1064                 }
1065                 break;
1066
1067             default:
1068             {
1069                 int ttl = (p_sys->i_ttl >= 0) ? p_sys->i_ttl : -1;
1070                 int fd = net_ConnectDgram( p_stream, p_sys->psz_destination,
1071                                            i_port, ttl, p_sys->proto );
1072                 if( fd == -1 )
1073                 {
1074                     msg_Err( p_stream, "cannot create RTP socket" );
1075                     goto error;
1076                 }
1077                 /* Ignore any unexpected incoming packet (including RTCP-RR
1078                  * packets in case of rtcp-mux) */
1079                 setsockopt (fd, SOL_SOCKET, SO_RCVBUF, &(int){ 0 },
1080                             sizeof (int));
1081                 rtp_add_sink( id, fd, p_sys->rtcp_mux, NULL );
1082             }
1083         }
1084     }
1085
1086     if( p_fmt == NULL )
1087     {
1088         char *psz = var_GetNonEmptyString( p_stream, SOUT_CFG_PREFIX "mux" );
1089
1090         if( psz == NULL ) /* Uho! */
1091             ;
1092         else
1093         if( strncmp( psz, "ts", 2 ) == 0 )
1094         {
1095             id->i_payload_type = 33;
1096             id->psz_enc = "MP2T";
1097         }
1098         else
1099         {
1100             id->psz_enc = "MP2P";
1101         }
1102         free( psz );
1103     }
1104     else
1105     switch( p_fmt->i_codec )
1106     {
1107         case VLC_CODEC_MULAW:
1108             if( p_fmt->audio.i_channels == 1 && p_fmt->audio.i_rate == 8000 )
1109                 id->i_payload_type = 0;
1110             id->psz_enc = "PCMU";
1111             id->pf_packetize = rtp_packetize_split;
1112             rtp_set_ptime (id, 20, 1);
1113             break;
1114         case VLC_CODEC_ALAW:
1115             if( p_fmt->audio.i_channels == 1 && p_fmt->audio.i_rate == 8000 )
1116                 id->i_payload_type = 8;
1117             id->psz_enc = "PCMA";
1118             id->pf_packetize = rtp_packetize_split;
1119             rtp_set_ptime (id, 20, 1);
1120             break;
1121         case VLC_CODEC_S16B:
1122         case VLC_CODEC_S16L:
1123             if( p_fmt->audio.i_channels == 1 && p_fmt->audio.i_rate == 44100 )
1124             {
1125                 id->i_payload_type = 11;
1126             }
1127             else if( p_fmt->audio.i_channels == 2 &&
1128                      p_fmt->audio.i_rate == 44100 )
1129             {
1130                 id->i_payload_type = 10;
1131             }
1132             id->psz_enc = "L16";
1133             if( p_fmt->i_codec == VLC_CODEC_S16B )
1134                 id->pf_packetize = rtp_packetize_split;
1135             else
1136                 id->pf_packetize = rtp_packetize_swab;
1137             rtp_set_ptime (id, 20, 2);
1138             break;
1139         case VLC_CODEC_U8:
1140             id->psz_enc = "L8";
1141             id->pf_packetize = rtp_packetize_split;
1142             rtp_set_ptime (id, 20, 1);
1143             break;
1144         case VLC_CODEC_MPGA:
1145             id->i_payload_type = 14;
1146             id->psz_enc = "MPA";
1147             id->i_clock_rate = 90000; /* not 44100 */
1148             id->pf_packetize = rtp_packetize_mpa;
1149             break;
1150         case VLC_CODEC_MPGV:
1151             id->i_payload_type = 32;
1152             id->psz_enc = "MPV";
1153             id->pf_packetize = rtp_packetize_mpv;
1154             break;
1155         case VLC_CODEC_ADPCM_G726:
1156             switch( p_fmt->i_bitrate / 1000 )
1157             {
1158             case 16:
1159                 id->psz_enc = "G726-16";
1160                 id->pf_packetize = rtp_packetize_g726_16;
1161                 break;
1162             case 24:
1163                 id->psz_enc = "G726-24";
1164                 id->pf_packetize = rtp_packetize_g726_24;
1165                 break;
1166             case 32:
1167                 id->psz_enc = "G726-32";
1168                 id->pf_packetize = rtp_packetize_g726_32;
1169                 break;
1170             case 40:
1171                 id->psz_enc = "G726-40";
1172                 id->pf_packetize = rtp_packetize_g726_40;
1173                 break;
1174             default:
1175                 msg_Err( p_stream, "cannot add this stream (unsupported "
1176                          "G.726 bit rate: %u)", p_fmt->i_bitrate );
1177                 goto error;
1178             }
1179             break;
1180         case VLC_CODEC_A52:
1181             id->psz_enc = "ac3";
1182             id->pf_packetize = rtp_packetize_ac3;
1183             break;
1184         case VLC_CODEC_H263:
1185             id->psz_enc = "H263-1998";
1186             id->pf_packetize = rtp_packetize_h263;
1187             break;
1188         case VLC_CODEC_H264:
1189             id->psz_enc = "H264";
1190             id->pf_packetize = rtp_packetize_h264;
1191             id->psz_fmtp = NULL;
1192
1193             if( p_fmt->i_extra > 0 )
1194             {
1195                 uint8_t *p_buffer = p_fmt->p_extra;
1196                 int     i_buffer = p_fmt->i_extra;
1197                 char    *p_64_sps = NULL;
1198                 char    *p_64_pps = NULL;
1199                 char    hexa[6+1];
1200
1201                 while( i_buffer > 4 &&
1202                        p_buffer[0] == 0 && p_buffer[1] == 0 &&
1203                        p_buffer[2] == 0 && p_buffer[3] == 1 )
1204                 {
1205                     const int i_nal_type = p_buffer[4]&0x1f;
1206                     int i_offset;
1207                     int i_size      = 0;
1208
1209                     msg_Dbg( p_stream, "we found a startcode for NAL with TYPE:%d", i_nal_type );
1210
1211                     i_size = i_buffer;
1212                     for( i_offset = 4; i_offset+3 < i_buffer ; i_offset++)
1213                     {
1214                         if( !memcmp (p_buffer + i_offset, "\x00\x00\x00\x01", 4 ) )
1215                         {
1216                             /* we found another startcode */
1217                             i_size = i_offset;
1218                             break;
1219                         }
1220                     }
1221                     if( i_nal_type == 7 )
1222                     {
1223                         p_64_sps = vlc_b64_encode_binary( &p_buffer[4], i_size - 4 );
1224                         sprintf_hexa( hexa, &p_buffer[5], 3 );
1225                     }
1226                     else if( i_nal_type == 8 )
1227                     {
1228                         p_64_pps = vlc_b64_encode_binary( &p_buffer[4], i_size - 4 );
1229                     }
1230                     i_buffer -= i_size;
1231                     p_buffer += i_size;
1232                 }
1233                 /* */
1234                 if( p_64_sps && p_64_pps &&
1235                     ( asprintf( &id->psz_fmtp,
1236                                 "packetization-mode=1;profile-level-id=%s;"
1237                                 "sprop-parameter-sets=%s,%s;", hexa, p_64_sps,
1238                                 p_64_pps ) == -1 ) )
1239                     id->psz_fmtp = NULL;
1240                 free( p_64_sps );
1241                 free( p_64_pps );
1242             }
1243             if( !id->psz_fmtp )
1244                 id->psz_fmtp = strdup( "packetization-mode=1" );
1245             break;
1246
1247         case VLC_CODEC_MP4V:
1248         {
1249             id->psz_enc = "MP4V-ES";
1250             id->pf_packetize = rtp_packetize_split;
1251             if( p_fmt->i_extra > 0 )
1252             {
1253                 char hexa[2*p_fmt->i_extra +1];
1254                 sprintf_hexa( hexa, p_fmt->p_extra, p_fmt->i_extra );
1255                 if( asprintf( &id->psz_fmtp,
1256                               "profile-level-id=3; config=%s;", hexa ) == -1 )
1257                     id->psz_fmtp = NULL;
1258             }
1259             break;
1260         }
1261         case VLC_CODEC_MP4A:
1262         {
1263             if(!p_sys->b_latm)
1264             {
1265                 char hexa[2*p_fmt->i_extra +1];
1266
1267                 id->psz_enc = "mpeg4-generic";
1268                 id->pf_packetize = rtp_packetize_mp4a;
1269                 sprintf_hexa( hexa, p_fmt->p_extra, p_fmt->i_extra );
1270                 if( asprintf( &id->psz_fmtp,
1271                               "streamtype=5; profile-level-id=15; "
1272                               "mode=AAC-hbr; config=%s; SizeLength=13; "
1273                               "IndexLength=3; IndexDeltaLength=3; Profile=1;",
1274                               hexa ) == -1 )
1275                     id->psz_fmtp = NULL;
1276             }
1277             else
1278             {
1279                 char hexa[13];
1280                 int i;
1281                 unsigned char config[6];
1282                 unsigned int aacsrates[15] = {
1283                     96000, 88200, 64000, 48000, 44100, 32000, 24000, 22050,
1284                     16000, 12000, 11025, 8000, 7350, 0, 0 };
1285
1286                 for( i = 0; i < 15; i++ )
1287                     if( p_fmt->audio.i_rate == aacsrates[i] )
1288                         break;
1289
1290                 config[0]=0x40;
1291                 config[1]=0;
1292                 config[2]=0x20|i;
1293                 config[3]=p_fmt->audio.i_channels<<4;
1294                 config[4]=0x3f;
1295                 config[5]=0xc0;
1296
1297                 id->psz_enc = "MP4A-LATM";
1298                 id->pf_packetize = rtp_packetize_mp4a_latm;
1299                 sprintf_hexa( hexa, config, 6 );
1300                 if( asprintf( &id->psz_fmtp, "profile-level-id=15; "
1301                               "object=2; cpresent=0; config=%s", hexa ) == -1 )
1302                     id->psz_fmtp = NULL;
1303             }
1304             break;
1305         }
1306         case VLC_CODEC_AMR_NB:
1307             id->psz_enc = "AMR";
1308             id->psz_fmtp = strdup( "octet-align=1" );
1309             id->pf_packetize = rtp_packetize_amr;
1310             break;
1311         case VLC_CODEC_AMR_WB:
1312             id->psz_enc = "AMR-WB";
1313             id->psz_fmtp = strdup( "octet-align=1" );
1314             id->pf_packetize = rtp_packetize_amr;
1315             break;
1316         case VLC_CODEC_SPEEX:
1317             id->psz_enc = "SPEEX";
1318             id->pf_packetize = rtp_packetize_spx;
1319             break;
1320         case VLC_CODEC_ITU_T140:
1321             id->psz_enc = "t140" ;
1322             id->i_clock_rate = 1000;
1323             id->pf_packetize = rtp_packetize_t140;
1324             break;
1325
1326         default:
1327             msg_Err( p_stream, "cannot add this stream (unsupported "
1328                      "codec: %4.4s)", (char*)&p_fmt->i_codec );
1329             goto error;
1330     }
1331     if (id->i_payload_type >= 96)
1332         /* Mark dynamic payload type in use */
1333         p_sys->payload_bitmap &= ~(1 << (127 - id->i_payload_type));
1334
1335 #if 0 /* No payload formats sets this at the moment */
1336     int cscov = -1;
1337     if( cscov != -1 )
1338         cscov += 8 /* UDP */ + 12 /* RTP */;
1339     if( id->sinkc > 0 )
1340         net_SetCSCov( id->sinkv[0].rtp_fd, cscov, -1 );
1341 #endif
1342
1343     vlc_mutex_lock( &p_sys->lock_ts );
1344     id->b_ts_init = ( p_sys->i_npt_zero != VLC_TS_INVALID );
1345     vlc_mutex_unlock( &p_sys->lock_ts );
1346     if( id->b_ts_init )
1347         id->i_ts_offset = rtp_compute_ts( id, p_sys->i_pts_offset );
1348
1349     if( p_sys->rtsp != NULL )
1350         id->rtsp_id = RtspAddId( p_sys->rtsp, id,
1351                                  GetDWBE( id->ssrc ),
1352                                  p_sys->psz_destination,
1353                                  p_sys->i_ttl, id->i_port, id->i_port + 1 );
1354
1355     id->p_fifo = block_FifoNew();
1356     if( unlikely(id->p_fifo == NULL) )
1357         goto error;
1358     if( vlc_clone( &id->thread, ThreadSend, id, VLC_THREAD_PRIORITY_HIGHEST ) )
1359     {
1360         block_FifoRelease( id->p_fifo );
1361         id->p_fifo = NULL;
1362         goto error;
1363     }
1364
1365     /* Update p_sys context */
1366     vlc_mutex_lock( &p_sys->lock_es );
1367     TAB_APPEND( p_sys->i_es, p_sys->es, id );
1368     vlc_mutex_unlock( &p_sys->lock_es );
1369
1370     psz_sdp = SDPGenerate( p_stream, NULL );
1371
1372     vlc_mutex_lock( &p_sys->lock_sdp );
1373     free( p_sys->psz_sdp );
1374     p_sys->psz_sdp = psz_sdp;
1375     vlc_mutex_unlock( &p_sys->lock_sdp );
1376
1377     msg_Dbg( p_stream, "sdp=\n%s", p_sys->psz_sdp );
1378
1379     /* Update SDP (sap/file) */
1380     if( p_sys->b_export_sap ) SapSetup( p_stream );
1381     if( p_sys->psz_sdp_file != NULL ) FileSetup( p_stream );
1382
1383     return id;
1384
1385 error:
1386     Del( p_stream, id );
1387     return NULL;
1388 }
1389
1390 static int Del( sout_stream_t *p_stream, sout_stream_id_t *id )
1391 {
1392     sout_stream_sys_t *p_sys = p_stream->p_sys;
1393
1394     vlc_mutex_lock( &p_sys->lock_es );
1395     TAB_REMOVE( p_sys->i_es, p_sys->es, id );
1396     vlc_mutex_unlock( &p_sys->lock_es );
1397
1398     if( likely(id->p_fifo != NULL) )
1399     {
1400         vlc_cancel( id->thread );
1401         vlc_join( id->thread, NULL );
1402         block_FifoRelease( id->p_fifo );
1403     }
1404
1405     /* Release dynamic payload type */
1406     if (id->i_payload_type >= 96)
1407         p_sys->payload_bitmap |= 1 << (127 - id->i_payload_type);
1408
1409     free( id->psz_fmtp );
1410
1411     if( id->rtsp_id )
1412         RtspDelId( p_sys->rtsp, id->rtsp_id );
1413     if( id->listen.fd != NULL )
1414     {
1415         vlc_cancel( id->listen.thread );
1416         vlc_join( id->listen.thread, NULL );
1417         net_ListenClose( id->listen.fd );
1418     }
1419     /* Delete remaining sinks (incoming connections or explicit
1420      * outgoing dst=) */
1421     while( id->sinkc > 0 )
1422         rtp_del_sink( id, id->sinkv[0].rtp_fd );
1423 #ifdef HAVE_SRTP
1424     if( id->srtp != NULL )
1425         srtp_destroy( id->srtp );
1426 #endif
1427
1428     vlc_mutex_destroy( &id->lock_sink );
1429
1430     /* Update SDP (sap/file) */
1431     if( p_sys->b_export_sap && !p_sys->p_mux ) SapSetup( p_stream );
1432     if( p_sys->psz_sdp_file != NULL ) FileSetup( p_stream );
1433
1434     free( id );
1435     return VLC_SUCCESS;
1436 }
1437
1438 static int Send( sout_stream_t *p_stream, sout_stream_id_t *id,
1439                  block_t *p_buffer )
1440 {
1441     block_t *p_next;
1442
1443     assert( p_stream->p_sys->p_mux == NULL );
1444     (void)p_stream;
1445
1446     while( p_buffer != NULL )
1447     {
1448         p_next = p_buffer->p_next;
1449         if( id->pf_packetize( id, p_buffer ) )
1450             break;
1451
1452         block_Release( p_buffer );
1453         p_buffer = p_next;
1454     }
1455     return VLC_SUCCESS;
1456 }
1457
1458 /****************************************************************************
1459  * SAP:
1460  ****************************************************************************/
1461 static int SapSetup( sout_stream_t *p_stream )
1462 {
1463     sout_stream_sys_t *p_sys = p_stream->p_sys;
1464     sout_instance_t   *p_sout = p_stream->p_sout;
1465
1466     /* Remove the previous session */
1467     if( p_sys->p_session != NULL)
1468     {
1469         sout_AnnounceUnRegister( p_sout, p_sys->p_session);
1470         p_sys->p_session = NULL;
1471     }
1472
1473     if( ( p_sys->i_es > 0 || p_sys->p_mux ) && p_sys->psz_sdp && *p_sys->psz_sdp )
1474     {
1475         announce_method_t *p_method = sout_SAPMethod();
1476         p_sys->p_session = sout_AnnounceRegisterSDP( p_sout,
1477                                                      p_sys->psz_sdp,
1478                                                      p_sys->psz_destination,
1479                                                      p_method );
1480         sout_MethodRelease( p_method );
1481     }
1482
1483     return VLC_SUCCESS;
1484 }
1485
1486 /****************************************************************************
1487 * File:
1488 ****************************************************************************/
1489 static int FileSetup( sout_stream_t *p_stream )
1490 {
1491     sout_stream_sys_t *p_sys = p_stream->p_sys;
1492     FILE            *f;
1493
1494     if( p_sys->psz_sdp == NULL )
1495         return VLC_EGENERIC; /* too early */
1496
1497     if( ( f = vlc_fopen( p_sys->psz_sdp_file, "wt" ) ) == NULL )
1498     {
1499         msg_Err( p_stream, "cannot open file '%s' (%m)",
1500                  p_sys->psz_sdp_file );
1501         return VLC_EGENERIC;
1502     }
1503
1504     fputs( p_sys->psz_sdp, f );
1505     fclose( f );
1506
1507     return VLC_SUCCESS;
1508 }
1509
1510 /****************************************************************************
1511  * HTTP:
1512  ****************************************************************************/
1513 static int  HttpCallback( httpd_file_sys_t *p_args,
1514                           httpd_file_t *, uint8_t *p_request,
1515                           uint8_t **pp_data, int *pi_data );
1516
1517 static int HttpSetup( sout_stream_t *p_stream, const vlc_url_t *url)
1518 {
1519     sout_stream_sys_t *p_sys = p_stream->p_sys;
1520
1521     p_sys->p_httpd_host = httpd_HostNew( VLC_OBJECT(p_stream), url->psz_host,
1522                                          url->i_port > 0 ? url->i_port : 80 );
1523     if( p_sys->p_httpd_host )
1524     {
1525         p_sys->p_httpd_file = httpd_FileNew( p_sys->p_httpd_host,
1526                                              url->psz_path ? url->psz_path : "/",
1527                                              "application/sdp",
1528                                              NULL, NULL, NULL,
1529                                              HttpCallback, (void*)p_sys );
1530     }
1531     if( p_sys->p_httpd_file == NULL )
1532     {
1533         return VLC_EGENERIC;
1534     }
1535     return VLC_SUCCESS;
1536 }
1537
1538 static int  HttpCallback( httpd_file_sys_t *p_args,
1539                           httpd_file_t *f, uint8_t *p_request,
1540                           uint8_t **pp_data, int *pi_data )
1541 {
1542     VLC_UNUSED(f); VLC_UNUSED(p_request);
1543     sout_stream_sys_t *p_sys = (sout_stream_sys_t*)p_args;
1544
1545     vlc_mutex_lock( &p_sys->lock_sdp );
1546     if( p_sys->psz_sdp && *p_sys->psz_sdp )
1547     {
1548         *pi_data = strlen( p_sys->psz_sdp );
1549         *pp_data = malloc( *pi_data );
1550         memcpy( *pp_data, p_sys->psz_sdp, *pi_data );
1551     }
1552     else
1553     {
1554         *pp_data = NULL;
1555         *pi_data = 0;
1556     }
1557     vlc_mutex_unlock( &p_sys->lock_sdp );
1558
1559     return VLC_SUCCESS;
1560 }
1561
1562 /****************************************************************************
1563  * RTP send
1564  ****************************************************************************/
1565 static void* ThreadSend( void *data )
1566 {
1567 #ifdef WIN32
1568 # define ECONNREFUSED WSAECONNREFUSED
1569 # define ENOPROTOOPT  WSAENOPROTOOPT
1570 # define EHOSTUNREACH WSAEHOSTUNREACH
1571 # define ENETUNREACH  WSAENETUNREACH
1572 # define ENETDOWN     WSAENETDOWN
1573 # define ENOBUFS      WSAENOBUFS
1574 # define EAGAIN       WSAEWOULDBLOCK
1575 # define EWOULDBLOCK  WSAEWOULDBLOCK
1576 #endif
1577     sout_stream_id_t *id = data;
1578     unsigned i_caching = id->i_caching;
1579
1580     for (;;)
1581     {
1582         block_t *out = block_FifoGet( id->p_fifo );
1583         block_cleanup_push (out);
1584
1585 #ifdef HAVE_SRTP
1586         if( id->srtp )
1587         {   /* FIXME: this is awfully inefficient */
1588             size_t len = out->i_buffer;
1589             out = block_Realloc( out, 0, len + 10 );
1590             out->i_buffer = len;
1591
1592             int canc = vlc_savecancel ();
1593             int val = srtp_send( id->srtp, out->p_buffer, &len, len + 10 );
1594             vlc_restorecancel (canc);
1595             if( val )
1596             {
1597                 errno = val;
1598                 msg_Dbg( id->p_stream, "SRTP sending error: %m" );
1599                 block_Release( out );
1600                 out = NULL;
1601             }
1602             else
1603                 out->i_buffer = len;
1604         }
1605         if (out)
1606 #endif
1607             mwait (out->i_dts + i_caching);
1608         vlc_cleanup_pop ();
1609         if (out == NULL)
1610             continue;
1611
1612         ssize_t len = out->i_buffer;
1613         int canc = vlc_savecancel ();
1614
1615         vlc_mutex_lock( &id->lock_sink );
1616         unsigned deadc = 0; /* How many dead sockets? */
1617         int deadv[id->sinkc]; /* Dead sockets list */
1618
1619         for( int i = 0; i < id->sinkc; i++ )
1620         {
1621 #ifdef HAVE_SRTP
1622             if( !id->srtp ) /* FIXME: SRTCP support */
1623 #endif
1624                 SendRTCP( id->sinkv[i].rtcp, out );
1625
1626             if( send( id->sinkv[i].rtp_fd, out->p_buffer, len, 0 ) >= 0 )
1627                 continue;
1628             switch( net_errno )
1629             {
1630                 /* Soft errors (e.g. ICMP): */
1631                 case ECONNREFUSED: /* Port unreachable */
1632                 case ENOPROTOOPT:
1633 #ifdef EPROTO
1634                 case EPROTO:       /* Protocol unreachable */
1635 #endif
1636                 case EHOSTUNREACH: /* Host unreachable */
1637                 case ENETUNREACH:  /* Network unreachable */
1638                 case ENETDOWN:     /* Entire network down */
1639                     send( id->sinkv[i].rtp_fd, out->p_buffer, len, 0 );
1640                 /* Transient congestion: */
1641                 case ENOMEM: /* out of socket buffers */
1642                 case ENOBUFS:
1643                 case EAGAIN:
1644 #if (EAGAIN != EWOULDBLOCK)
1645                 case EWOULDBLOCK:
1646 #endif
1647                     continue;
1648             }
1649
1650             deadv[deadc++] = id->sinkv[i].rtp_fd;
1651         }
1652         id->i_seq_sent_next = ntohs(((uint16_t *) out->p_buffer)[1]) + 1;
1653         vlc_mutex_unlock( &id->lock_sink );
1654         block_Release( out );
1655
1656         for( unsigned i = 0; i < deadc; i++ )
1657         {
1658             msg_Dbg( id->p_stream, "removing socket %d", deadv[i] );
1659             rtp_del_sink( id, deadv[i] );
1660         }
1661         vlc_restorecancel (canc);
1662     }
1663     return NULL;
1664 }
1665
1666
1667 /* This thread dequeues incoming connections (DCCP streaming) */
1668 static void *rtp_listen_thread( void *data )
1669 {
1670     sout_stream_id_t *id = data;
1671
1672     assert( id->listen.fd != NULL );
1673
1674     for( ;; )
1675     {
1676         int fd = net_Accept( id->p_stream, id->listen.fd );
1677         if( fd == -1 )
1678             continue;
1679         int canc = vlc_savecancel( );
1680         rtp_add_sink( id, fd, true, NULL );
1681         vlc_restorecancel( canc );
1682     }
1683
1684     assert( 0 );
1685 }
1686
1687
1688 int rtp_add_sink( sout_stream_id_t *id, int fd, bool rtcp_mux, uint16_t *seq )
1689 {
1690     rtp_sink_t sink = { fd, NULL };
1691     sink.rtcp = OpenRTCP( VLC_OBJECT( id->p_stream ), fd, IPPROTO_UDP,
1692                           rtcp_mux );
1693     if( sink.rtcp == NULL )
1694         msg_Err( id->p_stream, "RTCP failed!" );
1695
1696     vlc_mutex_lock( &id->lock_sink );
1697     INSERT_ELEM( id->sinkv, id->sinkc, id->sinkc, sink );
1698     if( seq != NULL )
1699         *seq = id->i_seq_sent_next;
1700     vlc_mutex_unlock( &id->lock_sink );
1701     return VLC_SUCCESS;
1702 }
1703
1704 void rtp_del_sink( sout_stream_id_t *id, int fd )
1705 {
1706     rtp_sink_t sink = { fd, NULL };
1707
1708     /* NOTE: must be safe to use if fd is not included */
1709     vlc_mutex_lock( &id->lock_sink );
1710     for( int i = 0; i < id->sinkc; i++ )
1711     {
1712         if (id->sinkv[i].rtp_fd == fd)
1713         {
1714             sink = id->sinkv[i];
1715             REMOVE_ELEM( id->sinkv, id->sinkc, i );
1716             break;
1717         }
1718     }
1719     vlc_mutex_unlock( &id->lock_sink );
1720
1721     CloseRTCP( sink.rtcp );
1722     net_Close( sink.rtp_fd );
1723 }
1724
1725 uint16_t rtp_get_seq( sout_stream_id_t *id )
1726 {
1727     /* This will return values for the next packet. */
1728     uint16_t seq;
1729
1730     vlc_mutex_lock( &id->lock_sink );
1731     seq = id->i_seq_sent_next;
1732     vlc_mutex_unlock( &id->lock_sink );
1733
1734     return seq;
1735 }
1736
1737 /* Return a timestamp corresponding to packets being sent now, and that
1738  * can be passed to rtp_compute_ts() to get rtptime values for each ES. */
1739 int64_t rtp_get_ts( const sout_stream_t *p_stream )
1740 {
1741     sout_stream_sys_t *p_sys = p_stream->p_sys;
1742     mtime_t i_npt_zero;
1743     vlc_mutex_lock( &p_sys->lock_ts );
1744     i_npt_zero = p_sys->i_npt_zero;
1745     vlc_mutex_unlock( &p_sys->lock_ts );
1746
1747     if( i_npt_zero == VLC_TS_INVALID )
1748         return p_sys->i_pts_zero;
1749
1750     mtime_t now = mdate();
1751     if( now < i_npt_zero )
1752         return p_sys->i_pts_zero;
1753
1754     return p_sys->i_pts_zero + (now - i_npt_zero);
1755 }
1756
1757 void rtp_packetize_common( sout_stream_id_t *id, block_t *out,
1758                            int b_marker, int64_t i_pts )
1759 {
1760     if( !id->b_ts_init )
1761     {
1762         sout_stream_sys_t *p_sys = id->p_stream->p_sys;
1763         vlc_mutex_lock( &p_sys->lock_ts );
1764         if( p_sys->i_npt_zero == VLC_TS_INVALID )
1765         {
1766             /* This is the first packet of any ES. We initialize the
1767              * NPT=0 time reference, and the offset to match the
1768              * arbitrary PTS reference. */
1769             p_sys->i_npt_zero = i_pts + id->i_caching;
1770             p_sys->i_pts_offset = p_sys->i_pts_zero - i_pts;
1771         }
1772         vlc_mutex_unlock( &p_sys->lock_ts );
1773
1774         /* And in any case this is the first packet of this ES, so we
1775          * initialize the offset for this ES. */
1776         id->i_ts_offset = rtp_compute_ts( id, p_sys->i_pts_offset );
1777         id->b_ts_init = true;
1778     }
1779
1780     uint32_t i_timestamp = rtp_compute_ts( id, i_pts ) + id->i_ts_offset;
1781
1782     out->p_buffer[0] = 0x80;
1783     out->p_buffer[1] = (b_marker?0x80:0x00)|id->i_payload_type;
1784     out->p_buffer[2] = ( id->i_sequence >> 8)&0xff;
1785     out->p_buffer[3] = ( id->i_sequence     )&0xff;
1786     out->p_buffer[4] = ( i_timestamp >> 24 )&0xff;
1787     out->p_buffer[5] = ( i_timestamp >> 16 )&0xff;
1788     out->p_buffer[6] = ( i_timestamp >>  8 )&0xff;
1789     out->p_buffer[7] = ( i_timestamp       )&0xff;
1790
1791     memcpy( out->p_buffer + 8, id->ssrc, 4 );
1792
1793     out->i_buffer = 12;
1794     id->i_sequence++;
1795 }
1796
1797 void rtp_packetize_send( sout_stream_id_t *id, block_t *out )
1798 {
1799     block_FifoPut( id->p_fifo, out );
1800 }
1801
1802 /**
1803  * @return configured max RTP payload size (including payload type-specific
1804  * headers, excluding RTP and transport headers)
1805  */
1806 size_t rtp_mtu (const sout_stream_id_t *id)
1807 {
1808     return id->i_mtu - 12;
1809 }
1810
1811 /*****************************************************************************
1812  * Non-RTP mux
1813  *****************************************************************************/
1814
1815 /** Add an ES to a non-RTP muxed stream */
1816 static sout_stream_id_t *MuxAdd( sout_stream_t *p_stream, es_format_t *p_fmt )
1817 {
1818     sout_input_t      *p_input;
1819     sout_mux_t *p_mux = p_stream->p_sys->p_mux;
1820     assert( p_mux != NULL );
1821
1822     p_input = sout_MuxAddStream( p_mux, p_fmt );
1823     if( p_input == NULL )
1824     {
1825         msg_Err( p_stream, "cannot add this stream to the muxer" );
1826         return NULL;
1827     }
1828
1829     return (sout_stream_id_t *)p_input;
1830 }
1831
1832
1833 static int MuxSend( sout_stream_t *p_stream, sout_stream_id_t *id,
1834                     block_t *p_buffer )
1835 {
1836     sout_mux_t *p_mux = p_stream->p_sys->p_mux;
1837     assert( p_mux != NULL );
1838
1839     sout_MuxSendBuffer( p_mux, (sout_input_t *)id, p_buffer );
1840     return VLC_SUCCESS;
1841 }
1842
1843
1844 /** Remove an ES from a non-RTP muxed stream */
1845 static int MuxDel( sout_stream_t *p_stream, sout_stream_id_t *id )
1846 {
1847     sout_mux_t *p_mux = p_stream->p_sys->p_mux;
1848     assert( p_mux != NULL );
1849
1850     sout_MuxDeleteStream( p_mux, (sout_input_t *)id );
1851     return VLC_SUCCESS;
1852 }
1853
1854
1855 static ssize_t AccessOutGrabberWriteBuffer( sout_stream_t *p_stream,
1856                                             const block_t *p_buffer )
1857 {
1858     sout_stream_sys_t *p_sys = p_stream->p_sys;
1859     sout_stream_id_t *id = p_sys->es[0];
1860
1861     int64_t  i_dts = p_buffer->i_dts;
1862
1863     uint8_t         *p_data = p_buffer->p_buffer;
1864     size_t          i_data  = p_buffer->i_buffer;
1865     size_t          i_max   = id->i_mtu - 12;
1866
1867     size_t i_packet = ( p_buffer->i_buffer + i_max - 1 ) / i_max;
1868
1869     while( i_data > 0 )
1870     {
1871         size_t i_size;
1872
1873         /* output complete packet */
1874         if( p_sys->packet &&
1875             p_sys->packet->i_buffer + i_data > i_max )
1876         {
1877             rtp_packetize_send( id, p_sys->packet );
1878             p_sys->packet = NULL;
1879         }
1880
1881         if( p_sys->packet == NULL )
1882         {
1883             /* allocate a new packet */
1884             p_sys->packet = block_New( p_stream, id->i_mtu );
1885             rtp_packetize_common( id, p_sys->packet, 1, i_dts );
1886             p_sys->packet->i_dts = i_dts;
1887             p_sys->packet->i_length = p_buffer->i_length / i_packet;
1888             i_dts += p_sys->packet->i_length;
1889         }
1890
1891         i_size = __MIN( i_data,
1892                         (unsigned)(id->i_mtu - p_sys->packet->i_buffer) );
1893
1894         memcpy( &p_sys->packet->p_buffer[p_sys->packet->i_buffer],
1895                 p_data, i_size );
1896
1897         p_sys->packet->i_buffer += i_size;
1898         p_data += i_size;
1899         i_data -= i_size;
1900     }
1901
1902     return VLC_SUCCESS;
1903 }
1904
1905
1906 static ssize_t AccessOutGrabberWrite( sout_access_out_t *p_access,
1907                                       block_t *p_buffer )
1908 {
1909     sout_stream_t *p_stream = (sout_stream_t*)p_access->p_sys;
1910
1911     while( p_buffer )
1912     {
1913         block_t *p_next;
1914
1915         AccessOutGrabberWriteBuffer( p_stream, p_buffer );
1916
1917         p_next = p_buffer->p_next;
1918         block_Release( p_buffer );
1919         p_buffer = p_next;
1920     }
1921
1922     return VLC_SUCCESS;
1923 }
1924
1925
1926 static sout_access_out_t *GrabberCreate( sout_stream_t *p_stream )
1927 {
1928     sout_access_out_t *p_grab;
1929
1930     p_grab = vlc_object_create( p_stream->p_sout, sizeof( *p_grab ) );
1931     if( p_grab == NULL )
1932         return NULL;
1933
1934     p_grab->p_module    = NULL;
1935     p_grab->psz_access  = strdup( "grab" );
1936     p_grab->p_cfg       = NULL;
1937     p_grab->psz_path    = strdup( "" );
1938     p_grab->p_sys       = (sout_access_out_sys_t *)p_stream;
1939     p_grab->pf_seek     = NULL;
1940     p_grab->pf_write    = AccessOutGrabberWrite;
1941     vlc_object_attach( p_grab, p_stream );
1942     return p_grab;
1943 }