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 Rémi Denis-Courmont
   6  * $Id$
   7  *
   8  * Authors: Laurent Aimar <fenrir@via.ecp.fr>
   9  *
  10  * This program is free software; you can redistribute it and/or modify
  11  * it under the terms of the GNU General Public License as published by
  12  * the Free Software Foundation; either version 2 of the License, or
  13  * (at your option) any later version.
  14  *
  15  * This program is distributed in the hope that it will be useful,
  16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18  * GNU General Public License for more details.
  19  *
  20  * You should have received a copy of the GNU General Public License
  21  * along with this program; if not, write to the Free Software
  22  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  23  *****************************************************************************/
  24
  25 /*****************************************************************************
  26  * Preamble
  27  *****************************************************************************/
  28
  29 #include <vlc/vlc.h>
  30 #include <vlc_sout.h>
  31 #include <vlc_block.h>
  32
  33 #include <vlc_httpd.h>
  34 #include <vlc_url.h>
  35 #include <vlc_network.h>
  36 #include <vlc_charset.h>
  37 #include <vlc_strings.h>
  38
  39 #include "rtp.h"
  40
  41 #ifdef HAVE_UNISTD_H
  42 #   include <sys/types.h>
  43 #   include <unistd.h>
  44 #   include <fcntl.h>
  45 #   include <sys/stat.h>
  46 #endif
  47 #ifdef HAVE_LINUX_DCCP_H
  48 #   include <linux/dccp.h>
  49 #endif
  50 #ifndef IPPROTO_DCCP
  51 # define IPPROTO_DCCP 33
  52 #endif
  53 #ifndef IPPROTO_UDPLITE
  54 # define IPPROTO_UDPLITE 136
  55 #endif
  56
  57 #include <errno.h>
  58
  59 /*****************************************************************************
  60  * Module descriptor
  61  *****************************************************************************/
  62
  63 #define DST_TEXT N_("Destination")
  64 #define DST_LONGTEXT N_( \
  65     "This is the output URL that will be used." )
  66 #define SDP_TEXT N_("SDP")
  67 #define SDP_LONGTEXT N_( \
  68     "This allows you to specify how the SDP (Session Descriptor) for this RTP "\
  69     "session will be made available. You must use an url: http://location to " \
  70     "access the SDP via HTTP, rtsp://location for RTSP access, and sap:// " \
  71     "for the SDP to be announced via SAP." )
  72 #define MUX_TEXT N_("Muxer")
  73 #define MUX_LONGTEXT N_( \
  74     "This allows you to specify the muxer used for the streaming output. " \
  75     "Default is to use no muxer (standard RTP stream)." )
  76
  77 #define NAME_TEXT N_("Session name")
  78 #define NAME_LONGTEXT N_( \
  79     "This is the name of the session that will be announced in the SDP " \
  80     "(Session Descriptor)." )
  81 #define DESC_TEXT N_("Session descriptipn")
  82 #define DESC_LONGTEXT N_( \
  83     "This allows you to give a short description with details about the stream, " \
  84     "that will be announced in the SDP (Session Descriptor)." )
  85 #define URL_TEXT N_("Session URL")
  86 #define URL_LONGTEXT N_( \
  87     "This allows you to give an URL with more details about the stream " \
  88     "(often the website of the streaming organization), that will " \
  89     "be announced in the SDP (Session Descriptor)." )
  90 #define EMAIL_TEXT N_("Session email")
  91 #define EMAIL_LONGTEXT N_( \
  92     "This allows you to give a contact mail address for the stream, that will " \
  93     "be announced in the SDP (Session Descriptor)." )
  94 #define PHONE_TEXT N_("Session phone number")
  95 #define PHONE_LONGTEXT N_( \
  96     "This allows you to give a contact telephone number for the stream, that will " \
  97     "be announced in the SDP (Session Descriptor)." )
  98
  99 #define PORT_TEXT N_("Port")
 100 #define PORT_LONGTEXT N_( \
 101     "This allows you to specify the base port for the RTP streaming." )
 102 #define PORT_AUDIO_TEXT N_("Audio port")
 103 #define PORT_AUDIO_LONGTEXT N_( \
 104     "This allows you to specify the default audio port for the RTP streaming." )
 105 #define PORT_VIDEO_TEXT N_("Video port")
 106 #define PORT_VIDEO_LONGTEXT N_( \
 107     "This allows you to specify the default video port for the RTP streaming." )
 108
 109 #define TTL_TEXT N_("Hop limit (TTL)")
 110 #define TTL_LONGTEXT N_( \
 111     "This is the hop limit (also known as \"Time-To-Live\" or TTL) of " \
 112     "the multicast packets sent by the stream output (0 = use operating " \
 113     "system built-in default).")
 114
 115 #define DCCP_TEXT N_("DCCP transport")
 116 #define DCCP_LONGTEXT N_( \
 117     "This enables DCCP instead of UDP as a transport for RTP." )
 118 #define TCP_TEXT N_("TCP transport")
 119 #define TCP_LONGTEXT N_( \
 120     "This enables TCP instead of UDP as a transport for RTP." )
 121 #define UDP_LITE_TEXT N_("UDP-Lite transport")
 122 #define UDP_LITE_LONGTEXT N_( \
 123     "This enables UDP-Lite instead of UDP as a transport for RTP." )
 124
 125 #define RFC3016_TEXT N_("MP4A LATM")
 126 #define RFC3016_LONGTEXT N_( \
 127     "This allows you to stream MPEG4 LATM audio streams (see RFC3016)." )
 128
 129 static int  Open ( vlc_object_t * );
 130 static void Close( vlc_object_t * );
 131
 132 #define SOUT_CFG_PREFIX "sout-rtp-"
 133 #define MAX_EMPTY_BLOCKS 200
 134
 135 vlc_module_begin();
 136     set_shortname( _("RTP"));
 137     set_description( _("RTP stream output") );
 138     set_capability( "sout stream", 0 );
 139     add_shortcut( "rtp" );
 140     set_category( CAT_SOUT );
 141     set_subcategory( SUBCAT_SOUT_STREAM );
 142
 143     add_string( SOUT_CFG_PREFIX "dst", "", NULL, DST_TEXT,
 144                 DST_LONGTEXT, VLC_TRUE );
 145     add_string( SOUT_CFG_PREFIX "sdp", "", NULL, SDP_TEXT,
 146                 SDP_LONGTEXT, VLC_TRUE );
 147     add_string( SOUT_CFG_PREFIX "mux", "", NULL, MUX_TEXT,
 148                 MUX_LONGTEXT, VLC_TRUE );
 149
 150     add_string( SOUT_CFG_PREFIX "name", "", NULL, NAME_TEXT,
 151                 NAME_LONGTEXT, VLC_TRUE );
 152     add_string( SOUT_CFG_PREFIX "description", "", NULL, DESC_TEXT,
 153                 DESC_LONGTEXT, VLC_TRUE );
 154     add_string( SOUT_CFG_PREFIX "url", "", NULL, URL_TEXT,
 155                 URL_LONGTEXT, VLC_TRUE );
 156     add_string( SOUT_CFG_PREFIX "email", "", NULL, EMAIL_TEXT,
 157                 EMAIL_LONGTEXT, VLC_TRUE );
 158     add_string( SOUT_CFG_PREFIX "phone", "", NULL, PHONE_TEXT,
 159                 PHONE_LONGTEXT, VLC_TRUE );
 160
 161     add_integer( SOUT_CFG_PREFIX "port", 1234, NULL, PORT_TEXT,
 162                  PORT_LONGTEXT, VLC_TRUE );
 163     add_integer( SOUT_CFG_PREFIX "port-audio", 1230, NULL, PORT_AUDIO_TEXT,
 164                  PORT_AUDIO_LONGTEXT, VLC_TRUE );
 165     add_integer( SOUT_CFG_PREFIX "port-video", 1232, NULL, PORT_VIDEO_TEXT,
 166                  PORT_VIDEO_LONGTEXT, VLC_TRUE );
 167
 168     add_integer( SOUT_CFG_PREFIX "ttl", 0, NULL, TTL_TEXT,
 169                  TTL_LONGTEXT, VLC_TRUE );
 170
 171     add_bool( SOUT_CFG_PREFIX "dccp", 0, NULL,
 172               DCCP_TEXT, DCCP_LONGTEXT, VLC_FALSE );
 173     add_bool( SOUT_CFG_PREFIX "tcp", 0, NULL,
 174               TCP_TEXT, TCP_LONGTEXT, VLC_FALSE );
 175     add_bool( SOUT_CFG_PREFIX "udplite", 0, NULL,
 176               UDP_LITE_TEXT, UDP_LITE_LONGTEXT, VLC_FALSE );
 177
 178     add_bool( SOUT_CFG_PREFIX "mp4a-latm", 0, NULL, RFC3016_TEXT,
 179                  RFC3016_LONGTEXT, VLC_FALSE );
 180
 181     set_callbacks( Open, Close );
 182 vlc_module_end();
 183
 184 /*****************************************************************************
 185  * Exported prototypes
 186  *****************************************************************************/
 187 static const char *ppsz_sout_options[] = {
 188     "dst", "name", "port", "port-audio", "port-video", "*sdp", "ttl", "mux",
 189     "description", "url", "email", "phone",
 190     "dccp", "tcp", "udplite",
 191     "mp4a-latm", NULL
 192 };
 193
 194 static sout_stream_id_t *Add ( sout_stream_t *, es_format_t * );
 195 static int               Del ( sout_stream_t *, sout_stream_id_t * );
 196 static int               Send( sout_stream_t *, sout_stream_id_t *,
 197                                block_t* );
 198 static sout_stream_id_t *MuxAdd ( sout_stream_t *, es_format_t * );
 199 static int               MuxDel ( sout_stream_t *, sout_stream_id_t * );
 200 static int               MuxSend( sout_stream_t *, sout_stream_id_t *,
 201                                   block_t* );
 202
 203 static sout_access_out_t *GrabberCreate( sout_stream_t *p_sout );
 204 static void ThreadSend( vlc_object_t *p_this );
 205
 206 static void SDPHandleUrl( sout_stream_t *, char * );
 207
 208 static int SapSetup( sout_stream_t *p_stream );
 209 static int FileSetup( sout_stream_t *p_stream );
 210 static int HttpSetup( sout_stream_t *p_stream, vlc_url_t * );
 211
 212 struct sout_stream_sys_t
 213 {
 214     /* SDP */
 215     char    *psz_sdp;
 216     vlc_mutex_t  lock_sdp;
 217
 218     /* SDP to disk */
 219     vlc_bool_t b_export_sdp_file;
 220     char *psz_sdp_file;
 221
 222     /* SDP via SAP */
 223     vlc_bool_t b_export_sap;
 224     session_descriptor_t *p_session;
 225
 226     /* SDP via HTTP */
 227     httpd_host_t *p_httpd_host;
 228     httpd_file_t *p_httpd_file;
 229
 230     /* RTSP */
 231     rtsp_stream_t *rtsp;
 232
 233     /* */
 234     char     *psz_destination;
 235     uint8_t   proto;
 236     uint8_t   i_ttl;
 237     uint16_t  i_port;
 238     uint16_t  i_port_audio;
 239     uint16_t  i_port_video;
 240     vlc_bool_t b_latm;
 241
 242     /* when need to use a private one or when using muxer */
 243     int i_payload_type;
 244
 245     /* in case we do TS/PS over rtp */
 246     sout_mux_t        *p_mux;
 247     sout_access_out_t *p_grab;
 248     block_t           *packet;
 249
 250     /* */
 251     vlc_mutex_t      lock_es;
 252     int              i_es;
 253     sout_stream_id_t **es;
 254 };
 255
 256 typedef int (*pf_rtp_packetizer_t)( sout_stream_t *, sout_stream_id_t *,
 257                                     block_t * );
 258
 259 typedef struct rtp_sink_t
 260 {
 261     int rtp_fd;
 262     rtcp_sender_t *rtcp;
 263 } rtp_sink_t;
 264
 265 struct sout_stream_id_t
 266 {
 267     VLC_COMMON_MEMBERS
 268
 269     sout_stream_t *p_stream;
 270     /* rtp field */
 271     uint32_t    i_timestamp_start;
 272     uint16_t    i_sequence;
 273     uint8_t     i_payload_type;
 274     uint8_t     ssrc[4];
 275
 276     /* for sdp */
 277     char        *psz_rtpmap;
 278     char        *psz_fmtp;
 279     int          i_clock_rate;
 280     int          i_port;
 281     int          i_cat;
 282     int          i_bitrate;
 283
 284     /* Packetizer specific fields */
 285     pf_rtp_packetizer_t pf_packetize;
 286     int          i_mtu;
 287
 288     /* Packets sinks */
 289     vlc_mutex_t       lock_sink;
 290     int               sinkc;
 291     rtp_sink_t       *sinkv;
 292     rtsp_stream_id_t *rtsp_id;
 293     int              *listen_fd;
 294
 295     block_fifo_t     *p_fifo;
 296     int64_t           i_caching;
 297 };
 298
 299
 300 /*****************************************************************************
 301  * Open:
 302  *****************************************************************************/
 303 static int Open( vlc_object_t *p_this )
 304 {
 305     sout_stream_t       *p_stream = (sout_stream_t*)p_this;
 306     sout_instance_t     *p_sout = p_stream->p_sout;
 307     sout_stream_sys_t   *p_sys = NULL;
 308     config_chain_t      *p_cfg = NULL;
 309     char                *psz;
 310     vlc_bool_t          b_rtsp = VLC_FALSE;
 311
 312     config_ChainParse( p_stream, SOUT_CFG_PREFIX,
 313                        ppsz_sout_options, p_stream->p_cfg );
 314
 315     p_sys = malloc( sizeof( sout_stream_sys_t ) );
 316     if( p_sys == NULL )
 317         return VLC_ENOMEM;
 318
 319     p_sys->psz_destination = var_GetNonEmptyString( p_stream, SOUT_CFG_PREFIX "dst" );
 320
 321     p_sys->i_port       = var_GetInteger( p_stream, SOUT_CFG_PREFIX "port" );
 322     p_sys->i_port_audio = var_GetInteger( p_stream, SOUT_CFG_PREFIX "port-audio" );
 323     p_sys->i_port_video = var_GetInteger( p_stream, SOUT_CFG_PREFIX "port-video" );
 324
 325     p_sys->psz_sdp_file = NULL;
 326
 327     if( p_sys->i_port_audio == p_sys->i_port_video )
 328     {
 329         msg_Err( p_stream, "audio and video port cannot be the same" );
 330         p_sys->i_port_audio = 0;
 331         p_sys->i_port_video = 0;
 332     }
 333
 334     for( p_cfg = p_stream->p_cfg; p_cfg != NULL; p_cfg = p_cfg->p_next )
 335     {
 336         if( !strcmp( p_cfg->psz_name, "sdp" )
 337          && ( p_cfg->psz_value != NULL )
 338          && !strncasecmp( p_cfg->psz_value, "rtsp:", 5 ) )
 339         {
 340             b_rtsp = VLC_TRUE;
 341             break;
 342         }
 343     }
 344     if( !b_rtsp )
 345     {
 346         psz = var_GetNonEmptyString( p_stream, SOUT_CFG_PREFIX "sdp" );
 347         if( psz != NULL )
 348         {
 349             if( !strncasecmp( psz, "rtsp:", 5 ) )
 350                 b_rtsp = VLC_TRUE;
 351             free( psz );
 352         }
 353     }
 354
 355     /* Transport protocol */
 356     p_sys->proto = IPPROTO_UDP;
 357
 358     if( var_GetBool( p_stream, SOUT_CFG_PREFIX "dccp" ) )
 359     {
 360         p_sys->proto = IPPROTO_DCCP;
 361     }
 362 #if 0
 363     else
 364     if( var_GetBool( p_stream, SOUT_CFG_PREFIX "tcp" ) )
 365     {
 366         p_sys->proto = IPPROTO_TCP;
 367     }
 368     else
 369 #endif
 370     if( var_GetBool( p_stream, SOUT_CFG_PREFIX "udplite" ) )
 371         p_sys->proto = IPPROTO_UDPLITE;
 372
 373     if( ( p_sys->psz_destination == NULL ) && !b_rtsp )
 374     {
 375         msg_Err( p_stream, "missing destination and not in RTSP mode" );
 376         free( p_sys );
 377         return VLC_EGENERIC;
 378     }
 379
 380     p_sys->i_ttl = var_GetInteger( p_stream, SOUT_CFG_PREFIX "ttl" );
 381     if( p_sys->i_ttl == 0 )
 382     {
 383         /* Normally, we should let the default hop limit up to the core,
 384          * but we have to know it to build our SDP properly, which is why
 385          * we ask the core. FIXME: broken when neither sout-rtp-ttl nor
 386          * ttl are set. */
 387         p_sys->i_ttl = config_GetInt( p_stream, "ttl" );
 388     }
 389
 390     p_sys->b_latm = var_GetBool( p_stream, SOUT_CFG_PREFIX "mp4a-latm" );
 391
 392     p_sys->i_payload_type = 96;
 393     p_sys->i_es = 0;
 394     p_sys->es   = NULL;
 395     p_sys->rtsp = NULL;
 396     p_sys->psz_sdp = NULL;
 397
 398     p_sys->b_export_sap = VLC_FALSE;
 399     p_sys->b_export_sdp_file = VLC_FALSE;
 400     p_sys->p_session = NULL;
 401
 402     p_sys->p_httpd_host = NULL;
 403     p_sys->p_httpd_file = NULL;
 404
 405     p_stream->p_sys     = p_sys;
 406
 407     vlc_mutex_init( p_stream, &p_sys->lock_sdp );
 408     vlc_mutex_init( p_stream, &p_sys->lock_es );
 409
 410     psz = var_GetNonEmptyString( p_stream, SOUT_CFG_PREFIX "mux" );
 411     if( psz != NULL )
 412     {
 413         sout_stream_id_t *id;
 414
 415         /* Check muxer type */
 416         if( strncasecmp( psz, "ps", 2 )
 417          && strncasecmp( psz, "mpeg1", 5 )
 418          && strncasecmp( psz, "ts", 2 ) )
 419         {
 420             msg_Err( p_stream, "unsupported muxer type for RTP (only TS/PS)" );
 421             free( psz );
 422             vlc_mutex_destroy( &p_sys->lock_sdp );
 423             vlc_mutex_destroy( &p_sys->lock_es );
 424             free( p_sys );
 425             return VLC_EGENERIC;
 426         }
 427
 428         p_sys->p_grab = GrabberCreate( p_stream );
 429         p_sys->p_mux = sout_MuxNew( p_sout, psz, p_sys->p_grab );
 430         free( psz );
 431
 432         if( p_sys->p_mux == NULL )
 433         {
 434             msg_Err( p_stream, "cannot create muxer" );
 435             sout_AccessOutDelete( p_sys->p_grab );
 436             vlc_mutex_destroy( &p_sys->lock_sdp );
 437             vlc_mutex_destroy( &p_sys->lock_es );
 438             free( p_sys );
 439             return VLC_EGENERIC;
 440         }
 441
 442         id = Add( p_stream, NULL );
 443         if( id == NULL )
 444         {
 445             sout_MuxDelete( p_sys->p_mux );
 446             sout_AccessOutDelete( p_sys->p_grab );
 447             vlc_mutex_destroy( &p_sys->lock_sdp );
 448             vlc_mutex_destroy( &p_sys->lock_es );
 449             free( p_sys );
 450             return VLC_EGENERIC;
 451         }
 452
 453         p_sys->packet = NULL;
 454
 455         p_stream->pf_add  = MuxAdd;
 456         p_stream->pf_del  = MuxDel;
 457         p_stream->pf_send = MuxSend;
 458     }
 459     else
 460     {
 461         p_sys->p_mux    = NULL;
 462         p_sys->p_grab   = NULL;
 463
 464         p_stream->pf_add    = Add;
 465         p_stream->pf_del    = Del;
 466         p_stream->pf_send   = Send;
 467     }
 468
 469     psz = var_GetNonEmptyString( p_stream, SOUT_CFG_PREFIX "sdp" );
 470     if( psz != NULL )
 471     {
 472         config_chain_t *p_cfg;
 473
 474         SDPHandleUrl( p_stream, psz );
 475
 476         for( p_cfg = p_stream->p_cfg; p_cfg != NULL; p_cfg = p_cfg->p_next )
 477         {
 478             if( !strcmp( p_cfg->psz_name, "sdp" ) )
 479             {
 480                 if( p_cfg->psz_value == NULL || *p_cfg->psz_value == '\0' )
 481                     continue;
 482
 483                 /* needed both :sout-rtp-sdp= and rtp{sdp=} can be used */
 484                 if( !strcmp( p_cfg->psz_value, psz ) )
 485                     continue;
 486
 487                 SDPHandleUrl( p_stream, p_cfg->psz_value );
 488             }
 489         }
 490         free( psz );
 491     }
 492
 493     /* update p_sout->i_out_pace_nocontrol */
 494     p_stream->p_sout->i_out_pace_nocontrol++;
 495
 496     return VLC_SUCCESS;
 497 }
 498
 499 /*****************************************************************************
 500  * Close:
 501  *****************************************************************************/
 502 static void Close( vlc_object_t * p_this )
 503 {
 504     sout_stream_t     *p_stream = (sout_stream_t*)p_this;
 505     sout_stream_sys_t *p_sys = p_stream->p_sys;
 506
 507     /* update p_sout->i_out_pace_nocontrol */
 508     p_stream->p_sout->i_out_pace_nocontrol--;
 509
 510     if( p_sys->p_mux )
 511     {
 512         assert( p_sys->i_es == 1 );
 513         Del( p_stream, p_sys->es[0] );
 514
 515         sout_MuxDelete( p_sys->p_mux );
 516         sout_AccessOutDelete( p_sys->p_grab );
 517         if( p_sys->packet )
 518         {
 519             block_Release( p_sys->packet );
 520         }
 521         if( p_sys->b_export_sap )
 522         {
 523             p_sys->p_mux = NULL;
 524             SapSetup( p_stream );
 525         }
 526     }
 527
 528     if( p_sys->rtsp != NULL )
 529         RtspUnsetup( p_sys->rtsp );
 530
 531     vlc_mutex_destroy( &p_sys->lock_sdp );
 532     vlc_mutex_destroy( &p_sys->lock_es );
 533
 534     if( p_sys->p_httpd_file )
 535         httpd_FileDelete( p_sys->p_httpd_file );
 536
 537     if( p_sys->p_httpd_host )
 538         httpd_HostDelete( p_sys->p_httpd_host );
 539
 540     free( p_sys->psz_sdp );
 541
 542     if( p_sys->b_export_sdp_file )
 543     {
 544 #ifdef HAVE_UNISTD_H
 545         unlink( p_sys->psz_sdp_file );
 546 #endif
 547         free( p_sys->psz_sdp_file );
 548     }
 549     free( p_sys->psz_destination );
 550     free( p_sys );
 551 }
 552
 553 /*****************************************************************************
 554  * SDPHandleUrl:
 555  *****************************************************************************/
 556 static void SDPHandleUrl( sout_stream_t *p_stream, char *psz_url )
 557 {
 558     sout_stream_sys_t *p_sys = p_stream->p_sys;
 559     vlc_url_t url;
 560
 561     vlc_UrlParse( &url, psz_url, 0 );
 562     if( url.psz_protocol && !strcasecmp( url.psz_protocol, "http" ) )
 563     {
 564         if( p_sys->p_httpd_file )
 565         {
 566             msg_Err( p_stream, "you can use sdp=http:// only once" );
 567             goto out;
 568         }
 569
 570         if( HttpSetup( p_stream, &url ) )
 571         {
 572             msg_Err( p_stream, "cannot export SDP as HTTP" );
 573         }
 574     }
 575     else if( url.psz_protocol && !strcasecmp( url.psz_protocol, "rtsp" ) )
 576     {
 577         if( p_sys->rtsp != NULL )
 578         {
 579             msg_Err( p_stream, "you can use sdp=rtsp:// only once" );
 580             goto out;
 581         }
 582
 583         /* FIXME test if destination is multicast or no destination at all */
 584         p_sys->rtsp = RtspSetup( p_stream, &url );
 585         if( p_sys->rtsp == NULL )
 586         {
 587             msg_Err( p_stream, "cannot export SDP as RTSP" );
 588         }
 589
 590         if( p_sys->p_mux != NULL )
 591         {
 592             sout_stream_id_t *id = p_sys->es[0];
 593             id->rtsp_id = RtspAddId( p_sys->rtsp, id, 0,
 594                                      p_sys->psz_destination, p_sys->i_ttl,
 595                                      id->i_port, id->i_port + 1 );
 596         }
 597     }
 598     else if( ( url.psz_protocol && !strcasecmp( url.psz_protocol, "sap" ) ) ||
 599              ( url.psz_host && !strcasecmp( url.psz_host, "sap" ) ) )
 600     {
 601         p_sys->b_export_sap = VLC_TRUE;
 602         SapSetup( p_stream );
 603     }
 604     else if( url.psz_protocol && !strcasecmp( url.psz_protocol, "file" ) )
 605     {
 606         if( p_sys->b_export_sdp_file )
 607         {
 608             msg_Err( p_stream, "you can use sdp=file:// only once" );
 609             goto out;
 610         }
 611         p_sys->b_export_sdp_file = VLC_TRUE;
 612         psz_url = &psz_url[5];
 613         if( psz_url[0] == '/' && psz_url[1] == '/' )
 614             psz_url += 2;
 615         p_sys->psz_sdp_file = strdup( psz_url );
 616     }
 617     else
 618     {
 619         msg_Warn( p_stream, "unknown protocol for SDP (%s)",
 620                   url.psz_protocol );
 621     }
 622
 623 out:
 624     vlc_UrlClean( &url );
 625 }
 626
 627 /*****************************************************************************
 628  * SDPGenerate
 629  *****************************************************************************/
 630 /*static*/
 631 char *SDPGenerate( const sout_stream_t *p_stream, const char *rtsp_url )
 632 {
 633     const sout_stream_sys_t *p_sys = p_stream->p_sys;
 634     char *psz_sdp;
 635     struct sockaddr_storage dst;
 636     socklen_t dstlen;
 637     int i;
 638     /*
 639      * When we have a fixed destination (typically when we do multicast),
 640      * we need to put the actual port numbers in the SDP.
 641      * When there is no fixed destination, we only support RTSP unicast
 642      * on-demand setup, so we should rather let the clients decide which ports
 643      * to use.
 644      * When there is both a fixed destination and RTSP unicast, we need to
 645      * put port numbers used by the fixed destination, otherwise the SDP would
 646      * become totally incorrect for multicast use. It should be noted that
 647      * port numbers from SDP with RTSP are only "recommendation" from the
 648      * server to the clients (per RFC2326), so only broken clients will fail
 649      * to handle this properly. There is no solution but to use two differents
 650      * output chain with two different RTSP URLs if you need to handle this
 651      * scenario.
 652      */
 653     int inclport;
 654
 655     if( p_sys->psz_destination != NULL )
 656     {
 657         inclport = 1;
 658
 659         /* Oh boy, this is really ugly! (+ race condition on lock_es) */
 660         dstlen = sizeof( dst );
 661         if( p_sys->es[0]->listen_fd != NULL )
 662             getsockname( p_sys->es[0]->listen_fd[0],
 663                          (struct sockaddr *)&dst, &dstlen );
 664         else
 665             getpeername( p_sys->es[0]->sinkv[0].rtp_fd,
 666                          (struct sockaddr *)&dst, &dstlen );
 667     }
 668     else
 669     {
 670         inclport = 0;
 671
 672         /* Dummy destination address for RTSP */
 673         memset (&dst, 0, sizeof( struct sockaddr_in ) );
 674         dst.ss_family = AF_INET;
 675 #ifdef HAVE_SA_LEN
 676         dst.ss_len =
 677 #endif
 678         dstlen = sizeof( struct sockaddr_in );
 679     }
 680
 681     psz_sdp = vlc_sdp_Start( VLC_OBJECT( p_stream ), SOUT_CFG_PREFIX,
 682                              NULL, 0, (struct sockaddr *)&dst, dstlen );
 683     if( psz_sdp == NULL )
 684         return NULL;
 685
 686     /* TODO: a=source-filter */
 687
 688     if( rtsp_url != NULL )
 689         sdp_AddAttribute ( &psz_sdp, "control", "%s", rtsp_url );
 690
 691     /* FIXME: locking?! */
 692     for( i = 0; i < p_sys->i_es; i++ )
 693     {
 694         sout_stream_id_t *id = p_sys->es[i];
 695         const char *mime_major; /* major MIME type */
 696         const char *proto = "RTP/AVP"; /* protocol */
 697         const char *scode; /* DCCP service code */
 698
 699         switch( id->i_cat )
 700         {
 701             case VIDEO_ES:
 702                 mime_major = "video";
 703                 scode = "SC:RTPV";
 704                 break;
 705             case AUDIO_ES:
 706                 mime_major = "audio";
 707                 scode = "SC:RTPA";
 708                 break;
 709             case SPU_ES:
 710                 mime_major = "text";
 711                 scode = "SC:RTPT";
 712                 break;
 713             default:
 714                 mime_major = "application";
 715                 scode = "SC:RTPO";
 716                 break;
 717         }
 718
 719         if( rtsp_url == NULL )
 720         {
 721             switch( p_sys->proto )
 722             {
 723                 case IPPROTO_UDP:
 724                     break;
 725                 case IPPROTO_TCP:
 726                     proto = "TCP/RTP/AVP";
 727                     break;
 728                 case IPPROTO_DCCP:
 729                     proto = "DCCP/RTP/AVP";
 730                     break;
 731                 case IPPROTO_UDPLITE:
 732                     continue;
 733             }
 734         }
 735
 736         sdp_AddMedia( &psz_sdp, mime_major, proto, inclport * id->i_port,
 737                       id->i_payload_type, VLC_FALSE, id->i_bitrate,
 738                       id->psz_rtpmap, id->psz_fmtp);
 739
 740         if( rtsp_url != NULL )
 741         {
 742             assert( strlen( rtsp_url ) > 0 );
 743             vlc_bool_t addslash = ( rtsp_url[strlen( rtsp_url ) - 1] != '/' );
 744             sdp_AddAttribute ( &psz_sdp, "control",
 745                                addslash ? "%s/trackID=%u" : "%strackID=%u",
 746                                rtsp_url, i );
 747         }
 748         else
 749         {
 750             if( id->listen_fd != NULL )
 751                 sdp_AddAttribute( &psz_sdp, "setup", "passive" );
 752             if( p_sys->proto == IPPROTO_DCCP )
 753                 sdp_AddAttribute( &psz_sdp, "dccp-service-code", scode );
 754         }
 755     }
 756
 757     return psz_sdp;
 758 }
 759
 760 /*****************************************************************************
 761  * RTP mux
 762  *****************************************************************************/
 763 static int rtp_packetize_l16  ( sout_stream_t *, sout_stream_id_t *, block_t * );
 764 static int rtp_packetize_l8   ( sout_stream_t *, sout_stream_id_t *, block_t * );
 765 static int rtp_packetize_mpa  ( sout_stream_t *, sout_stream_id_t *, block_t * );
 766 static int rtp_packetize_mpv  ( sout_stream_t *, sout_stream_id_t *, block_t * );
 767 static int rtp_packetize_ac3  ( sout_stream_t *, sout_stream_id_t *, block_t * );
 768 static int rtp_packetize_split( sout_stream_t *, sout_stream_id_t *, block_t * );
 769 static int rtp_packetize_mp4a ( sout_stream_t *, sout_stream_id_t *, block_t * );
 770 static int rtp_packetize_mp4a_latm ( sout_stream_t *, sout_stream_id_t *, block_t * );
 771 static int rtp_packetize_h263 ( sout_stream_t *, sout_stream_id_t *, block_t * );
 772 static int rtp_packetize_h264 ( sout_stream_t *, sout_stream_id_t *, block_t * );
 773 static int rtp_packetize_amr  ( sout_stream_t *, sout_stream_id_t *, block_t * );
 774 static int rtp_packetize_t140 ( sout_stream_t *, sout_stream_id_t *, block_t * );
 775
 776 static void sprintf_hexa( char *s, uint8_t *p_data, int i_data )
 777 {
 778     static const char hex[16] = "0123456789abcdef";
 779     int i;
 780
 781     for( i = 0; i < i_data; i++ )
 782     {
 783         s[2*i+0] = hex[(p_data[i]>>4)&0xf];
 784         s[2*i+1] = hex[(p_data[i]   )&0xf];
 785     }
 786     s[2*i_data] = '\0';
 787 }
 788
 789
 790 /** Add an ES as a new RTP stream */
 791 static sout_stream_id_t *Add( sout_stream_t *p_stream, es_format_t *p_fmt )
 792 {
 793     /* NOTE: As a special case, if we use a non-RTP
 794      * mux (TS/PS), then p_fmt is NULL. */
 795     sout_stream_sys_t *p_sys = p_stream->p_sys;
 796     sout_stream_id_t  *id;
 797     int               i_port, cscov = -1;
 798     char              *psz_sdp;
 799
 800     id = vlc_object_create( p_stream, sizeof( sout_stream_id_t ) );
 801     if( id == NULL )
 802         return NULL;
 803     vlc_object_attach( id, p_stream );
 804
 805     /* Choose the port */
 806     i_port = 0;
 807     if( p_fmt == NULL )
 808         ;
 809     else
 810     if( p_fmt->i_cat == AUDIO_ES && p_sys->i_port_audio > 0 )
 811     {
 812         i_port = p_sys->i_port_audio;
 813         p_sys->i_port_audio = 0;
 814     }
 815     else
 816     if( p_fmt->i_cat == VIDEO_ES && p_sys->i_port_video > 0 )
 817     {
 818         i_port = p_sys->i_port_video;
 819         p_sys->i_port_video = 0;
 820     }
 821
 822     while( i_port == 0 )
 823     {
 824         if( p_sys->i_port != p_sys->i_port_audio
 825          && p_sys->i_port != p_sys->i_port_video )
 826         {
 827             i_port = p_sys->i_port;
 828             p_sys->i_port += 2;
 829             break;
 830         }
 831         p_sys->i_port += 2;
 832     }
 833
 834     id->p_stream   = p_stream;
 835
 836     id->i_timestamp_start = rand()&0xffffffff;
 837     id->i_sequence = rand()&0xffff;
 838     id->i_payload_type = p_sys->i_payload_type;
 839     id->ssrc[0] = rand()&0xff;
 840     id->ssrc[1] = rand()&0xff;
 841     id->ssrc[2] = rand()&0xff;
 842     id->ssrc[3] = rand()&0xff;
 843
 844     id->psz_rtpmap = NULL;
 845     id->psz_fmtp   = NULL;
 846     id->i_clock_rate = 90000; /* most common case */
 847     id->i_port     = i_port;
 848     if( p_fmt != NULL )
 849     {
 850         id->i_cat  = p_fmt->i_cat;
 851         id->i_bitrate = p_fmt->i_bitrate/1000; /* Stream bitrate in kbps */
 852     }
 853     else
 854     {
 855         id->i_cat  = VIDEO_ES;
 856         id->i_bitrate = 0;
 857     }
 858
 859     id->pf_packetize = NULL;
 860     id->i_mtu = config_GetInt( p_stream, "mtu" );
 861     if( id->i_mtu <= 12 + 16 )
 862         id->i_mtu = 576 - 20 - 8; /* pessimistic */
 863
 864     msg_Dbg( p_stream, "maximum RTP packet size: %d bytes", id->i_mtu );
 865
 866     vlc_mutex_init( p_stream, &id->lock_sink );
 867     id->sinkc = 0;
 868     id->sinkv = NULL;
 869     id->rtsp_id = NULL;
 870     id->p_fifo = NULL;
 871     id->listen_fd = NULL;
 872
 873     id->i_caching =
 874         (int64_t)1000 * var_GetInteger( p_stream, SOUT_CFG_PREFIX "caching");
 875
 876     if( p_sys->psz_destination != NULL )
 877         switch( p_sys->proto )
 878         {
 879             case IPPROTO_TCP:
 880             case IPPROTO_DCCP:
 881                 id->listen_fd = net_Listen( VLC_OBJECT(p_stream),
 882                                             p_sys->psz_destination, i_port,
 883                                             p_sys->proto );
 884                 if( id->listen_fd == NULL )
 885                 {
 886                     msg_Err( p_stream, "passive COMEDIA RTP socket failed" );
 887                     goto error;
 888                 }
 889                 break;
 890
 891             default:
 892             {
 893                 int ttl = (p_sys->i_ttl > 0) ? p_sys->i_ttl : -1;
 894                 int fd = net_ConnectDgram( p_stream, p_sys->psz_destination,
 895                                            i_port, ttl, p_sys->proto );
 896                 if( fd == -1 )
 897                 {
 898                     msg_Err( p_stream, "cannot create RTP socket" );
 899                     goto error;
 900                 }
 901                 rtp_add_sink( id, fd );
 902             }
 903         }
 904
 905     if( p_fmt == NULL )
 906     {
 907         char *psz = var_GetNonEmptyString( p_stream, SOUT_CFG_PREFIX "mux" );
 908
 909         if( psz == NULL ) /* Uho! */
 910             ;
 911         else
 912         if( strncmp( psz, "ts", 2 ) == 0 )
 913         {
 914             id->i_payload_type = 33;
 915             id->psz_rtpmap = strdup( "MP2T/90000" );
 916         }
 917         else
 918         {
 919             id->psz_rtpmap = strdup( "MP2P/90000" );
 920         }
 921     }
 922     else
 923     switch( p_fmt->i_codec )
 924     {
 925         case VLC_FOURCC( 's', '1', '6', 'b' ):
 926             if( p_fmt->audio.i_channels == 1 && p_fmt->audio.i_rate == 44100 )
 927             {
 928                 id->i_payload_type = 11;
 929             }
 930             else if( p_fmt->audio.i_channels == 2 &&
 931                      p_fmt->audio.i_rate == 44100 )
 932             {
 933                 id->i_payload_type = 10;
 934             }
 935             if( asprintf( &id->psz_rtpmap, "L16/%d/%d", p_fmt->audio.i_rate,
 936                           p_fmt->audio.i_channels ) == -1 )
 937                 id->psz_rtpmap = NULL;
 938             id->i_clock_rate = p_fmt->audio.i_rate;
 939             id->pf_packetize = rtp_packetize_l16;
 940             break;
 941         case VLC_FOURCC( 'u', '8', ' ', ' ' ):
 942             if( asprintf( &id->psz_rtpmap, "L8/%d/%d", p_fmt->audio.i_rate,
 943                           p_fmt->audio.i_channels ) == -1 )
 944                 id->psz_rtpmap = NULL;
 945             id->i_clock_rate = p_fmt->audio.i_rate;
 946             id->pf_packetize = rtp_packetize_l8;
 947             break;
 948         case VLC_FOURCC( 'm', 'p', 'g', 'a' ):
 949             id->i_payload_type = 14;
 950             id->psz_rtpmap = strdup( "MPA/90000" );
 951             id->pf_packetize = rtp_packetize_mpa;
 952             break;
 953         case VLC_FOURCC( 'm', 'p', 'g', 'v' ):
 954             id->i_payload_type = 32;
 955             id->psz_rtpmap = strdup( "MPV/90000" );
 956             id->pf_packetize = rtp_packetize_mpv;
 957             break;
 958         case VLC_FOURCC( 'a', '5', '2', ' ' ):
 959             id->psz_rtpmap = strdup( "ac3/90000" );
 960             id->pf_packetize = rtp_packetize_ac3;
 961             break;
 962         case VLC_FOURCC( 'H', '2', '6', '3' ):
 963             id->psz_rtpmap = strdup( "H263-1998/90000" );
 964             id->pf_packetize = rtp_packetize_h263;
 965             break;
 966         case VLC_FOURCC( 'h', '2', '6', '4' ):
 967             id->psz_rtpmap = strdup( "H264/90000" );
 968             id->pf_packetize = rtp_packetize_h264;
 969             id->psz_fmtp = NULL;
 970
 971             if( p_fmt->i_extra > 0 )
 972             {
 973                 uint8_t *p_buffer = p_fmt->p_extra;
 974                 int     i_buffer = p_fmt->i_extra;
 975                 char    *p_64_sps = NULL;
 976                 char    *p_64_pps = NULL;
 977                 char    hexa[6+1];
 978
 979                 while( i_buffer > 4 &&
 980                        p_buffer[0] == 0 && p_buffer[1] == 0 &&
 981                        p_buffer[2] == 0 && p_buffer[3] == 1 )
 982                 {
 983                     const int i_nal_type = p_buffer[4]&0x1f;
 984                     int i_offset;
 985                     int i_size      = 0;
 986
 987                     msg_Dbg( p_stream, "we found a startcode for NAL with TYPE:%d", i_nal_type );
 988
 989                     i_size = i_buffer;
 990                     for( i_offset = 4; i_offset+3 < i_buffer ; i_offset++)
 991                     {
 992                         if( !memcmp (p_buffer + i_offset, "\x00\x00\x00\x01", 4 ) )
 993                         {
 994                             /* we found another startcode */
 995                             i_size = i_offset;
 996                             break;
 997                         }
 998                     }
 999                     if( i_nal_type == 7 )
1000                     {
1001                         p_64_sps = vlc_b64_encode_binary( &p_buffer[4], i_size - 4 );
1002                         sprintf_hexa( hexa, &p_buffer[5], 3 );
1003                     }
1004                     else if( i_nal_type == 8 )
1005                     {
1006                         p_64_pps = vlc_b64_encode_binary( &p_buffer[4], i_size - 4 );
1007                     }
1008                     i_buffer -= i_size;
1009                     p_buffer += i_size;
1010                 }
1011                 /* */
1012                 if( p_64_sps && p_64_pps &&
1013                     ( asprintf( &id->psz_fmtp,
1014                                 "packetization-mode=1;profile-level-id=%s;"
1015                                 "sprop-parameter-sets=%s,%s;", hexa, p_64_sps,
1016                                 p_64_pps ) == -1 ) )
1017                     id->psz_fmtp = NULL;
1018                 free( p_64_sps );
1019                 free( p_64_pps );
1020             }
1021             if( !id->psz_fmtp )
1022                 id->psz_fmtp = strdup( "packetization-mode=1" );
1023             break;
1024
1025         case VLC_FOURCC( 'm', 'p', '4', 'v' ):
1026         {
1027             char hexa[2*p_fmt->i_extra +1];
1028
1029             id->psz_rtpmap = strdup( "MP4V-ES/90000" );
1030             id->pf_packetize = rtp_packetize_split;
1031             if( p_fmt->i_extra > 0 )
1032             {
1033                 sprintf_hexa( hexa, p_fmt->p_extra, p_fmt->i_extra );
1034                 if( asprintf( &id->psz_fmtp,
1035                               "profile-level-id=3; config=%s;", hexa ) == -1 )
1036                     id->psz_fmtp = NULL;
1037             }
1038             break;
1039         }
1040         case VLC_FOURCC( 'm', 'p', '4', 'a' ):
1041         {
1042             id->i_clock_rate = p_fmt->audio.i_rate;
1043
1044             if(!p_sys->b_latm)
1045             {
1046                 char hexa[2*p_fmt->i_extra +1];
1047
1048                 if( asprintf( &id->psz_rtpmap, "mpeg4-generic/%d",
1049                               p_fmt->audio.i_rate ) == -1 )
1050                     id->psz_rtpmap = NULL;
1051                 id->pf_packetize = rtp_packetize_mp4a;
1052                 sprintf_hexa( hexa, p_fmt->p_extra, p_fmt->i_extra );
1053                 if( asprintf( &id->psz_fmtp,
1054                               "streamtype=5; profile-level-id=15; "
1055                               "mode=AAC-hbr; config=%s; SizeLength=13; "
1056                               "IndexLength=3; IndexDeltaLength=3; Profile=1;",
1057                               hexa ) == -1 )
1058                     id->psz_fmtp = NULL;
1059             }
1060             else
1061             {
1062                 char hexa[13];
1063                 int i;
1064                 unsigned char config[6];
1065                 unsigned int aacsrates[15] = {
1066                     96000, 88200, 64000, 48000, 44100, 32000, 24000, 22050,
1067                     16000, 12000, 11025, 8000, 7350, 0, 0 };
1068
1069                 for( i = 0; i < 15; i++ )
1070                     if( p_fmt->audio.i_rate == aacsrates[i] )
1071                         break;
1072
1073                 config[0]=0x40;
1074                 config[1]=0;
1075                 config[2]=0x20|i;
1076                 config[3]=p_fmt->audio.i_channels<<4;
1077                 config[4]=0x3f;
1078                 config[5]=0xc0;
1079
1080                 if( asprintf( &id->psz_rtpmap, "MP4A-LATM/%d/%d",
1081                               p_fmt->audio.i_rate,
1082                               p_fmt->audio.i_channels ) == -1)
1083                     id->psz_rtpmap = NULL;
1084                 id->pf_packetize = rtp_packetize_mp4a_latm;
1085                 sprintf_hexa( hexa, config, 6 );
1086                 if( asprintf( &id->psz_fmtp, "profile-level-id=15; "
1087                               "object=2; cpresent=0; config=%s", hexa ) == -1 )
1088                     id->psz_fmtp = NULL;
1089             }
1090             break;
1091         }
1092         case VLC_FOURCC( 's', 'a', 'm', 'r' ):
1093             id->psz_rtpmap = strdup( p_fmt->audio.i_channels == 2 ?
1094                                      "AMR/8000/2" : "AMR/8000" );
1095             id->psz_fmtp = strdup( "octet-align=1" );
1096             id->i_clock_rate = p_fmt->audio.i_rate;
1097             id->pf_packetize = rtp_packetize_amr;
1098             break;
1099         case VLC_FOURCC( 's', 'a', 'w', 'b' ):
1100             id->psz_rtpmap = strdup( p_fmt->audio.i_channels == 2 ?
1101                                      "AMR-WB/16000/2" : "AMR-WB/16000" );
1102             id->psz_fmtp = strdup( "octet-align=1" );
1103             id->i_clock_rate = p_fmt->audio.i_rate;
1104             id->pf_packetize = rtp_packetize_amr;
1105             break;
1106         case VLC_FOURCC( 't', '1', '4', '0' ):
1107             id->psz_rtpmap = strdup( "t140/1000" );
1108             id->i_clock_rate = 1000;
1109             id->pf_packetize = rtp_packetize_t140;
1110             break;
1111
1112         default:
1113             msg_Err( p_stream, "cannot add this stream (unsupported "
1114                      "codec:%4.4s)", (char*)&p_fmt->i_codec );
1115             goto error;
1116     }
1117
1118     if( cscov != -1 )
1119         cscov += 8 /* UDP */ + 12 /* RTP */;
1120     if( id->sinkc > 0 )
1121         net_SetCSCov( id->sinkv[0].rtp_fd, cscov, -1 );
1122
1123     if( id->i_payload_type == p_sys->i_payload_type )
1124         p_sys->i_payload_type++;
1125
1126     if( p_sys->rtsp != NULL )
1127         id->rtsp_id = RtspAddId( p_sys->rtsp, id, p_sys->i_es,
1128                                  p_sys->psz_destination,
1129                                  p_sys->i_ttl, id->i_port, id->i_port + 1 );
1130
1131     id->p_fifo = block_FifoNew( p_stream );
1132     if( vlc_thread_create( id, "RTP send thread", ThreadSend,
1133                            VLC_THREAD_PRIORITY_HIGHEST, VLC_FALSE ) )
1134         goto error;
1135
1136     /* Update p_sys context */
1137     vlc_mutex_lock( &p_sys->lock_es );
1138     TAB_APPEND( p_sys->i_es, p_sys->es, id );
1139     vlc_mutex_unlock( &p_sys->lock_es );
1140
1141     psz_sdp = SDPGenerate( p_stream, NULL );
1142
1143     vlc_mutex_lock( &p_sys->lock_sdp );
1144     free( p_sys->psz_sdp );
1145     p_sys->psz_sdp = psz_sdp;
1146     vlc_mutex_unlock( &p_sys->lock_sdp );
1147
1148     msg_Dbg( p_stream, "sdp=\n%s", p_sys->psz_sdp );
1149
1150     /* Update SDP (sap/file) */
1151     if( p_sys->b_export_sap ) SapSetup( p_stream );
1152     if( p_sys->b_export_sdp_file ) FileSetup( p_stream );
1153
1154     return id;
1155
1156 error:
1157     Del( p_stream, id );
1158     return NULL;
1159 }
1160
1161 static int Del( sout_stream_t *p_stream, sout_stream_id_t *id )
1162 {
1163     sout_stream_sys_t *p_sys = p_stream->p_sys;
1164
1165     if( id->p_fifo != NULL )
1166     {
1167         vlc_object_kill( id );
1168         block_FifoWake( id->p_fifo );
1169         vlc_thread_join( id );
1170         block_FifoRelease( id->p_fifo );
1171     }
1172
1173     vlc_mutex_lock( &p_sys->lock_es );
1174     TAB_REMOVE( p_sys->i_es, p_sys->es, id );
1175     vlc_mutex_unlock( &p_sys->lock_es );
1176
1177     /* Release port */
1178     if( id->i_port > 0 )
1179     {
1180         if( id->i_cat == AUDIO_ES && p_sys->i_port_audio == 0 )
1181             p_sys->i_port_audio = id->i_port;
1182         else if( id->i_cat == VIDEO_ES && p_sys->i_port_video == 0 )
1183             p_sys->i_port_video = id->i_port;
1184     }
1185
1186     free( id->psz_rtpmap );
1187     free( id->psz_fmtp );
1188
1189     if( id->rtsp_id )
1190         RtspDelId( p_sys->rtsp, id->rtsp_id );
1191     if( id->sinkc > 0 )
1192         rtp_del_sink( id, id->sinkv[0].rtp_fd ); /* sink for explicit dst= */
1193     if( id->listen_fd != NULL )
1194         net_ListenClose( id->listen_fd );
1195
1196     vlc_mutex_destroy( &id->lock_sink );
1197
1198     /* Update SDP (sap/file) */
1199     if( p_sys->b_export_sap && !p_sys->p_mux ) SapSetup( p_stream );
1200     if( p_sys->b_export_sdp_file ) FileSetup( p_stream );
1201
1202     vlc_object_detach( id );
1203     vlc_object_destroy( id );
1204     return VLC_SUCCESS;
1205 }
1206
1207 static int Send( sout_stream_t *p_stream, sout_stream_id_t *id,
1208                  block_t *p_buffer )
1209 {
1210     block_t *p_next;
1211
1212     assert( p_stream->p_sys->p_mux == NULL );
1213
1214     while( p_buffer != NULL )
1215     {
1216         p_next = p_buffer->p_next;
1217         if( id->pf_packetize( p_stream, id, p_buffer ) )
1218             break;
1219
1220         block_Release( p_buffer );
1221         p_buffer = p_next;
1222     }
1223     return VLC_SUCCESS;
1224 }
1225
1226 /****************************************************************************
1227  * SAP:
1228  ****************************************************************************/
1229 static int SapSetup( sout_stream_t *p_stream )
1230 {
1231     sout_stream_sys_t *p_sys = p_stream->p_sys;
1232     sout_instance_t   *p_sout = p_stream->p_sout;
1233
1234     /* Remove the previous session */
1235     if( p_sys->p_session != NULL)
1236     {
1237         sout_AnnounceUnRegister( p_sout, p_sys->p_session);
1238         p_sys->p_session = NULL;
1239     }
1240
1241     if( ( p_sys->i_es > 0 || p_sys->p_mux ) && p_sys->psz_sdp && *p_sys->psz_sdp )
1242     {
1243         announce_method_t *p_method = sout_SAPMethod();
1244         p_sys->p_session = sout_AnnounceRegisterSDP( p_sout, SOUT_CFG_PREFIX,
1245                                                      p_sys->psz_sdp,
1246                                                      p_sys->psz_destination,
1247                                                      p_method );
1248         sout_MethodRelease( p_method );
1249     }
1250
1251     return VLC_SUCCESS;
1252 }
1253
1254 /****************************************************************************
1255 * File:
1256 ****************************************************************************/
1257 static int FileSetup( sout_stream_t *p_stream )
1258 {
1259     sout_stream_sys_t *p_sys = p_stream->p_sys;
1260     FILE            *f;
1261
1262     if( ( f = utf8_fopen( p_sys->psz_sdp_file, "wt" ) ) == NULL )
1263     {
1264         msg_Err( p_stream, "cannot open file '%s' (%m)",
1265                  p_sys->psz_sdp_file );
1266         return VLC_EGENERIC;
1267     }
1268
1269     fputs( p_sys->psz_sdp, f );
1270     fclose( f );
1271
1272     return VLC_SUCCESS;
1273 }
1274
1275 /****************************************************************************
1276  * HTTP:
1277  ****************************************************************************/
1278 static int  HttpCallback( httpd_file_sys_t *p_args,
1279                           httpd_file_t *, uint8_t *p_request,
1280                           uint8_t **pp_data, int *pi_data );
1281
1282 static int HttpSetup( sout_stream_t *p_stream, vlc_url_t *url)
1283 {
1284     sout_stream_sys_t *p_sys = p_stream->p_sys;
1285
1286     p_sys->p_httpd_host = httpd_HostNew( VLC_OBJECT(p_stream), url->psz_host,
1287                                          url->i_port > 0 ? url->i_port : 80 );
1288     if( p_sys->p_httpd_host )
1289     {
1290         p_sys->p_httpd_file = httpd_FileNew( p_sys->p_httpd_host,
1291                                              url->psz_path ? url->psz_path : "/",
1292                                              "application/sdp",
1293                                              NULL, NULL, NULL,
1294                                              HttpCallback, (void*)p_sys );
1295     }
1296     if( p_sys->p_httpd_file == NULL )
1297     {
1298         return VLC_EGENERIC;
1299     }
1300     return VLC_SUCCESS;
1301 }
1302
1303 static int  HttpCallback( httpd_file_sys_t *p_args,
1304                           httpd_file_t *f, uint8_t *p_request,
1305                           uint8_t **pp_data, int *pi_data )
1306 {
1307     sout_stream_sys_t *p_sys = (sout_stream_sys_t*)p_args;
1308
1309     vlc_mutex_lock( &p_sys->lock_sdp );
1310     if( p_sys->psz_sdp && *p_sys->psz_sdp )
1311     {
1312         *pi_data = strlen( p_sys->psz_sdp );
1313         *pp_data = malloc( *pi_data );
1314         memcpy( *pp_data, p_sys->psz_sdp, *pi_data );
1315     }
1316     else
1317     {
1318         *pp_data = NULL;
1319         *pi_data = 0;
1320     }
1321     vlc_mutex_unlock( &p_sys->lock_sdp );
1322
1323     return VLC_SUCCESS;
1324 }
1325
1326 /****************************************************************************
1327  * RTP send
1328  ****************************************************************************/
1329 static void ThreadSend( vlc_object_t *p_this )
1330 {
1331     sout_stream_id_t *id = (sout_stream_id_t *)p_this;
1332     unsigned i_caching = id->i_caching;
1333 #ifdef HAVE_TEE
1334     int fd[5] = { -1, -1, -1, -1, -1 };
1335
1336     if( pipe( fd ) )
1337         fd[0] = fd[1] = -1;
1338     else
1339     if( pipe( fd ) )
1340         fd[2] = fd[3] = -1;
1341     else
1342         fd[4] = open( "/dev/null", O_WRONLY );
1343 #endif
1344
1345     while( !id->b_die )
1346     {
1347         block_t *out = block_FifoGet( id->p_fifo );
1348         if( out == NULL )
1349             continue; /* Forced wakeup */
1350
1351         mtime_t  i_date = out->i_dts + i_caching;
1352         ssize_t  len = out->i_buffer;
1353
1354 #ifdef HAVE_TEE
1355         if( fd[4] != -1 )
1356             len = write( fd[1], out->p_buffer, len);
1357         if( len == -1 )
1358             continue; /* Uho - should not happen */
1359 #endif
1360         mwait( i_date );
1361
1362         vlc_mutex_lock( &id->lock_sink );
1363 #if 0
1364         unsigned deadc = 0; /* How many dead sockets? */
1365         int deadv[id->sinkc]; /* Dead sockets list */
1366 #endif
1367
1368         for( int i = 0; i < id->sinkc; i++ )
1369         {
1370             SendRTCP( id->sinkv[i].rtcp, out );
1371
1372 #ifdef HAVE_TEE
1373             tee( fd[0], fd[3], len, 0 );
1374             if( splice( fd[2], NULL, id->sinkv[i].rtp_fd, NULL, len,
1375                         SPLICE_F_NONBLOCK ) >= 0 )
1376                 continue;
1377
1378             /* splice failed */
1379             splice( fd[2], NULL, fd[4], NULL, len, 0 );
1380 #endif
1381             if( send( id->sinkv[i].rtp_fd, out->p_buffer, len, 0 ) < 0 )
1382                 /*deadv[deadc++] = id->sinkv[i].rtp_fd*/;
1383         }
1384         vlc_mutex_unlock( &id->lock_sink );
1385
1386         block_Release( out );
1387 #ifdef HAVE_TEE
1388         splice( fd[0], NULL, fd[4], NULL, len, 0 );
1389 #endif
1390
1391 #if 0
1392         for( unsigned i = 0; i < deadc; i++ )
1393         {
1394             msg_Dbg( id, "removing socket %d", deadv[i] );
1395             rtp_del_sink( id, deadv[i] );
1396         }
1397 #endif
1398
1399         /* Hopefully we won't overflow the SO_MAXCONN accept queue */
1400         while( id->listen_fd != NULL )
1401         {
1402             int fd = net_Accept( id, id->listen_fd, 0 );
1403             if( fd == -1 )
1404                 break;
1405             msg_Dbg( id, "adding socket %d", fd );
1406             rtp_add_sink( id, fd );
1407         }
1408     }
1409
1410 #ifdef HAVE_TEE
1411     for( unsigned i = 0; i < 5; i++ )
1412         close( fd[i] );
1413 #endif
1414 }
1415
1416 static inline void rtp_packetize_send( sout_stream_id_t *id, block_t *out )
1417 {
1418     block_FifoPut( id->p_fifo, out );
1419 }
1420
1421 int rtp_add_sink( sout_stream_id_t *id, int fd )
1422 {
1423     rtp_sink_t sink = { fd, NULL };
1424     sink.rtcp = OpenRTCP( VLC_OBJECT( id->p_stream ), fd, IPPROTO_UDP );
1425     if( sink.rtcp == NULL )
1426         msg_Err( id, "RTCP failed!" );
1427
1428     vlc_mutex_lock( &id->lock_sink );
1429     INSERT_ELEM( id->sinkv, id->sinkc, id->sinkc, sink );
1430     vlc_mutex_unlock( &id->lock_sink );
1431     return VLC_SUCCESS;
1432 }
1433
1434 void rtp_del_sink( sout_stream_id_t *id, int fd )
1435 {
1436     rtp_sink_t sink = { fd, NULL };
1437
1438     /* NOTE: must be safe to use if fd is not included */
1439     vlc_mutex_lock( &id->lock_sink );
1440     for( int i = 0; i < id->sinkc; i++ )
1441     {
1442         if (id->sinkv[i].rtp_fd == fd)
1443         {
1444             sink = id->sinkv[i];
1445             REMOVE_ELEM( id->sinkv, id->sinkc, i );
1446             break;
1447         }
1448     }
1449     vlc_mutex_unlock( &id->lock_sink );
1450
1451     CloseRTCP( sink.rtcp );
1452     net_Close( sink.rtp_fd );
1453 }
1454
1455 /****************************************************************************
1456  * rtp_packetize_*:
1457  ****************************************************************************/
1458 static void rtp_packetize_common( sout_stream_id_t *id, block_t *out,
1459                                   int b_marker, int64_t i_pts )
1460 {
1461     uint32_t i_timestamp = i_pts * (int64_t)id->i_clock_rate / I64C(1000000);
1462
1463     out->p_buffer[0] = 0x80;
1464     out->p_buffer[1] = (b_marker?0x80:0x00)|id->i_payload_type;
1465     out->p_buffer[2] = ( id->i_sequence >> 8)&0xff;
1466     out->p_buffer[3] = ( id->i_sequence     )&0xff;
1467     out->p_buffer[4] = ( i_timestamp >> 24 )&0xff;
1468     out->p_buffer[5] = ( i_timestamp >> 16 )&0xff;
1469     out->p_buffer[6] = ( i_timestamp >>  8 )&0xff;
1470     out->p_buffer[7] = ( i_timestamp       )&0xff;
1471
1472     memcpy( out->p_buffer + 8, id->ssrc, 4 );
1473
1474     out->i_buffer = 12;
1475     id->i_sequence++;
1476 }
1477
1478 static int rtp_packetize_mpa( sout_stream_t *p_stream, sout_stream_id_t *id,
1479                               block_t *in )
1480 {
1481     int     i_max   = id->i_mtu - 12 - 4; /* payload max in one packet */
1482     int     i_count = ( in->i_buffer + i_max - 1 ) / i_max;
1483
1484     uint8_t *p_data = in->p_buffer;
1485     int     i_data  = in->i_buffer;
1486     int     i;
1487
1488     for( i = 0; i < i_count; i++ )
1489     {
1490         int           i_payload = __MIN( i_max, i_data );
1491         block_t *out = block_New( p_stream, 16 + i_payload );
1492
1493         /* rtp common header */
1494         rtp_packetize_common( id, out, (i == i_count - 1)?1:0, in->i_pts );
1495         /* mbz set to 0 */
1496         out->p_buffer[12] = 0;
1497         out->p_buffer[13] = 0;
1498         /* fragment offset in the current frame */
1499         out->p_buffer[14] = ( (i*i_max) >> 8 )&0xff;
1500         out->p_buffer[15] = ( (i*i_max)      )&0xff;
1501         memcpy( &out->p_buffer[16], p_data, i_payload );
1502
1503         out->i_buffer   = 16 + i_payload;
1504         out->i_dts    = in->i_dts + i * in->i_length / i_count;
1505         out->i_length = in->i_length / i_count;
1506
1507         rtp_packetize_send( id, out );
1508
1509         p_data += i_payload;
1510         i_data -= i_payload;
1511     }
1512
1513     return VLC_SUCCESS;
1514 }
1515
1516 /* rfc2250 */
1517 static int rtp_packetize_mpv( sout_stream_t *p_stream, sout_stream_id_t *id,
1518                               block_t *in )
1519 {
1520     int     i_max   = id->i_mtu - 12 - 4; /* payload max in one packet */
1521     int     i_count = ( in->i_buffer + i_max - 1 ) / i_max;
1522
1523     uint8_t *p_data = in->p_buffer;
1524     int     i_data  = in->i_buffer;
1525     int     i;
1526     int     b_sequence_start = 0;
1527     int     i_temporal_ref = 0;
1528     int     i_picture_coding_type = 0;
1529     int     i_fbv = 0, i_bfc = 0, i_ffv = 0, i_ffc = 0;
1530     int     b_start_slice = 0;
1531
1532     /* preparse this packet to get some info */
1533     if( in->i_buffer > 4 )
1534     {
1535         uint8_t *p = p_data;
1536         int      i_rest = in->i_buffer;
1537
1538         for( ;; )
1539         {
1540             while( i_rest > 4 &&
1541                    ( p[0] != 0x00 || p[1] != 0x00 || p[2] != 0x01 ) )
1542             {
1543                 p++;
1544                 i_rest--;
1545             }
1546             if( i_rest <= 4 )
1547             {
1548                 break;
1549             }
1550             p += 3;
1551             i_rest -= 4;
1552
1553             if( *p == 0xb3 )
1554             {
1555                 /* sequence start code */
1556                 b_sequence_start = 1;
1557             }
1558             else if( *p == 0x00 && i_rest >= 4 )
1559             {
1560                 /* picture */
1561                 i_temporal_ref = ( p[1] << 2) |((p[2]>>6)&0x03);
1562                 i_picture_coding_type = (p[2] >> 3)&0x07;
1563
1564                 if( i_rest >= 4 && ( i_picture_coding_type == 2 ||
1565                                     i_picture_coding_type == 3 ) )
1566                 {
1567                     i_ffv = (p[3] >> 2)&0x01;
1568                     i_ffc = ((p[3]&0x03) << 1)|((p[4]>>7)&0x01);
1569                     if( i_rest > 4 && i_picture_coding_type == 3 )
1570                     {
1571                         i_fbv = (p[4]>>6)&0x01;
1572                         i_bfc = (p[4]>>3)&0x07;
1573                     }
1574                 }
1575             }
1576             else if( *p <= 0xaf )
1577             {
1578                 b_start_slice = 1;
1579             }
1580         }
1581     }
1582
1583     for( i = 0; i < i_count; i++ )
1584     {
1585         int           i_payload = __MIN( i_max, i_data );
1586         block_t *out = block_New( p_stream,
1587                                              16 + i_payload );
1588         uint32_t      h = ( i_temporal_ref << 16 )|
1589                           ( b_sequence_start << 13 )|
1590                           ( b_start_slice << 12 )|
1591                           ( i == i_count - 1 ? 1 << 11 : 0 )|
1592                           ( i_picture_coding_type << 8 )|
1593                           ( i_fbv << 7 )|( i_bfc << 4 )|( i_ffv << 3 )|i_ffc;
1594
1595         /* rtp common header */
1596         rtp_packetize_common( id, out, (i == i_count - 1)?1:0,
1597                               in->i_pts > 0 ? in->i_pts : in->i_dts );
1598
1599         /* MBZ:5 T:1 TR:10 AN:1 N:1 S:1 B:1 E:1 P:3 FBV:1 BFC:3 FFV:1 FFC:3 */
1600         out->p_buffer[12] = ( h >> 24 )&0xff;
1601         out->p_buffer[13] = ( h >> 16 )&0xff;
1602         out->p_buffer[14] = ( h >>  8 )&0xff;
1603         out->p_buffer[15] = ( h       )&0xff;
1604
1605         memcpy( &out->p_buffer[16], p_data, i_payload );
1606
1607         out->i_buffer   = 16 + i_payload;
1608         out->i_dts    = in->i_dts + i * in->i_length / i_count;
1609         out->i_length = in->i_length / i_count;
1610
1611         rtp_packetize_send( id, out );
1612
1613         p_data += i_payload;
1614         i_data -= i_payload;
1615     }
1616
1617     return VLC_SUCCESS;
1618 }
1619
1620 static int rtp_packetize_ac3( sout_stream_t *p_stream, sout_stream_id_t *id,
1621                               block_t *in )
1622 {
1623     int     i_max   = id->i_mtu - 12 - 2; /* payload max in one packet */
1624     int     i_count = ( in->i_buffer + i_max - 1 ) / i_max;
1625
1626     uint8_t *p_data = in->p_buffer;
1627     int     i_data  = in->i_buffer;
1628     int     i;
1629
1630     for( i = 0; i < i_count; i++ )
1631     {
1632         int           i_payload = __MIN( i_max, i_data );
1633         block_t *out = block_New( p_stream, 14 + i_payload );
1634
1635         /* rtp common header */
1636         rtp_packetize_common( id, out, (i == i_count - 1)?1:0, in->i_pts );
1637         /* unit count */
1638         out->p_buffer[12] = 1;
1639         /* unit header */
1640         out->p_buffer[13] = 0x00;
1641         /* data */
1642         memcpy( &out->p_buffer[14], p_data, i_payload );
1643
1644         out->i_buffer   = 14 + i_payload;
1645         out->i_dts    = in->i_dts + i * in->i_length / i_count;
1646         out->i_length = in->i_length / i_count;
1647
1648         rtp_packetize_send( id, out );
1649
1650         p_data += i_payload;
1651         i_data -= i_payload;
1652     }
1653
1654     return VLC_SUCCESS;
1655 }
1656
1657 static int rtp_packetize_split( sout_stream_t *p_stream, sout_stream_id_t *id,
1658                                 block_t *in )
1659 {
1660     int     i_max   = id->i_mtu - 12; /* payload max in one packet */
1661     int     i_count = ( in->i_buffer + i_max - 1 ) / i_max;
1662
1663     uint8_t *p_data = in->p_buffer;
1664     int     i_data  = in->i_buffer;
1665     int     i;
1666
1667     for( i = 0; i < i_count; i++ )
1668     {
1669         int           i_payload = __MIN( i_max, i_data );
1670         block_t *out = block_New( p_stream, 12 + i_payload );
1671
1672         /* rtp common header */
1673         rtp_packetize_common( id, out, ((i == i_count - 1)?1:0),
1674                               (in->i_pts > 0 ? in->i_pts : in->i_dts) );
1675         memcpy( &out->p_buffer[12], p_data, i_payload );
1676
1677         out->i_buffer   = 12 + i_payload;
1678         out->i_dts    = in->i_dts + i * in->i_length / i_count;
1679         out->i_length = in->i_length / i_count;
1680
1681         rtp_packetize_send( id, out );
1682
1683         p_data += i_payload;
1684         i_data -= i_payload;
1685     }
1686
1687     return VLC_SUCCESS;
1688 }
1689
1690 /* rfc3016 */
1691 static int rtp_packetize_mp4a_latm( sout_stream_t *p_stream, sout_stream_id_t *id,
1692                                 block_t *in )
1693 {
1694     int     i_max   = id->i_mtu - 14;              /* payload max in one packet */
1695     int     latmhdrsize = in->i_buffer / 0xff + 1;
1696     int     i_count = ( in->i_buffer + i_max - 1 ) / i_max;
1697
1698     uint8_t *p_data = in->p_buffer, *p_header = NULL;
1699     int     i_data  = in->i_buffer;
1700     int     i;
1701
1702     for( i = 0; i < i_count; i++ )
1703     {
1704         int     i_payload = __MIN( i_max, i_data );
1705         block_t *out;
1706
1707         if( i != 0 )
1708             latmhdrsize = 0;
1709         out = block_New( p_stream, 12 + latmhdrsize + i_payload );
1710
1711         /* rtp common header */
1712         rtp_packetize_common( id, out, ((i == i_count - 1) ? 1 : 0),
1713                               (in->i_pts > 0 ? in->i_pts : in->i_dts) );
1714
1715         if( i == 0 )
1716         {
1717             int tmp = in->i_buffer;
1718
1719             p_header=out->p_buffer+12;
1720             while( tmp > 0xfe )
1721             {
1722                 *p_header = 0xff;
1723                 p_header++;
1724                 tmp -= 0xff;
1725             }
1726             *p_header = tmp;
1727         }
1728
1729         memcpy( &out->p_buffer[12+latmhdrsize], p_data, i_payload );
1730
1731         out->i_buffer   = 12 + latmhdrsize + i_payload;
1732         out->i_dts    = in->i_dts + i * in->i_length / i_count;
1733         out->i_length = in->i_length / i_count;
1734
1735         rtp_packetize_send( id, out );
1736
1737         p_data += i_payload;
1738         i_data -= i_payload;
1739     }
1740
1741     return VLC_SUCCESS;
1742 }
1743
1744 static int rtp_packetize_l16( sout_stream_t *p_stream, sout_stream_id_t *id,
1745                               block_t *in )
1746 {
1747     int     i_max   = id->i_mtu - 12; /* payload max in one packet */
1748     int     i_count = ( in->i_buffer + i_max - 1 ) / i_max;
1749
1750     uint8_t *p_data = in->p_buffer;
1751     int     i_data  = in->i_buffer;
1752     int     i_packet = 0;
1753
1754     while( i_data > 0 )
1755     {
1756         int           i_payload = (__MIN( i_max, i_data )/4)*4;
1757         block_t *out = block_New( p_stream, 12 + i_payload );
1758
1759         /* rtp common header */
1760         rtp_packetize_common( id, out, 0,
1761                               (in->i_pts > 0 ? in->i_pts : in->i_dts) );
1762         memcpy( &out->p_buffer[12], p_data, i_payload );
1763
1764         out->i_buffer   = 12 + i_payload;
1765         out->i_dts    = in->i_dts + i_packet * in->i_length / i_count;
1766         out->i_length = in->i_length / i_count;
1767
1768         rtp_packetize_send( id, out );
1769
1770         p_data += i_payload;
1771         i_data -= i_payload;
1772         i_packet++;
1773     }
1774
1775     return VLC_SUCCESS;
1776 }
1777
1778 static int rtp_packetize_l8( sout_stream_t *p_stream, sout_stream_id_t *id,
1779                              block_t *in )
1780 {
1781     int     i_max   = id->i_mtu - 12; /* payload max in one packet */
1782     int     i_count = ( in->i_buffer + i_max - 1 ) / i_max;
1783
1784     uint8_t *p_data = in->p_buffer;
1785     int     i_data  = in->i_buffer;
1786     int     i_packet = 0;
1787
1788     while( i_data > 0 )
1789     {
1790         int           i_payload = (__MIN( i_max, i_data )/2)*2;
1791         block_t *out = block_New( p_stream, 12 + i_payload );
1792
1793         /* rtp common header */
1794         rtp_packetize_common( id, out, 0,
1795                               (in->i_pts > 0 ? in->i_pts : in->i_dts) );
1796         memcpy( &out->p_buffer[12], p_data, i_payload );
1797
1798         out->i_buffer   = 12 + i_payload;
1799         out->i_dts    = in->i_dts + i_packet * in->i_length / i_count;
1800         out->i_length = in->i_length / i_count;
1801
1802         rtp_packetize_send( id, out );
1803
1804         p_data += i_payload;
1805         i_data -= i_payload;
1806         i_packet++;
1807     }
1808
1809     return VLC_SUCCESS;
1810 }
1811
1812 static int rtp_packetize_mp4a( sout_stream_t *p_stream, sout_stream_id_t *id,
1813                                block_t *in )
1814 {
1815     int     i_max   = id->i_mtu - 16; /* payload max in one packet */
1816     int     i_count = ( in->i_buffer + i_max - 1 ) / i_max;
1817
1818     uint8_t *p_data = in->p_buffer;
1819     int     i_data  = in->i_buffer;
1820     int     i;
1821
1822     for( i = 0; i < i_count; i++ )
1823     {
1824         int           i_payload = __MIN( i_max, i_data );
1825         block_t *out = block_New( p_stream, 16 + i_payload );
1826
1827         /* rtp common header */
1828         rtp_packetize_common( id, out, ((i == i_count - 1)?1:0),
1829                               (in->i_pts > 0 ? in->i_pts : in->i_dts) );
1830         /* AU headers */
1831         /* AU headers length (bits) */
1832         out->p_buffer[12] = 0;
1833         out->p_buffer[13] = 2*8;
1834         /* for each AU length 13 bits + idx 3bits, */
1835         out->p_buffer[14] = ( in->i_buffer >> 5 )&0xff;
1836         out->p_buffer[15] = ( (in->i_buffer&0xff)<<3 )|0;
1837
1838         memcpy( &out->p_buffer[16], p_data, i_payload );
1839
1840         out->i_buffer   = 16 + i_payload;
1841         out->i_dts    = in->i_dts + i * in->i_length / i_count;
1842         out->i_length = in->i_length / i_count;
1843
1844         rtp_packetize_send( id, out );
1845
1846         p_data += i_payload;
1847         i_data -= i_payload;
1848     }
1849
1850     return VLC_SUCCESS;
1851 }
1852
1853
1854 /* rfc2429 */
1855 #define RTP_H263_HEADER_SIZE (2)  // plen = 0
1856 #define RTP_H263_PAYLOAD_START (14)  // plen = 0
1857 static int rtp_packetize_h263( sout_stream_t *p_stream, sout_stream_id_t *id,
1858                                block_t *in )
1859 {
1860     uint8_t *p_data = in->p_buffer;
1861     int     i_data  = in->i_buffer;
1862     int     i;
1863     int     i_max   = id->i_mtu - 12 - RTP_H263_HEADER_SIZE; /* payload max in one packet */
1864     int     i_count;
1865     int     b_p_bit;
1866     int     b_v_bit = 0; // no pesky error resilience
1867     int     i_plen = 0; // normally plen=0 for PSC packet
1868     int     i_pebit = 0; // because plen=0
1869     uint16_t h;
1870
1871     if( i_data < 2 )
1872     {
1873         return VLC_EGENERIC;
1874     }
1875     if( p_data[0] || p_data[1] )
1876     {
1877         return VLC_EGENERIC;
1878     }
1879     /* remove 2 leading 0 bytes */
1880     p_data += 2;
1881     i_data -= 2;
1882     i_count = ( i_data + i_max - 1 ) / i_max;
1883
1884     for( i = 0; i < i_count; i++ )
1885     {
1886         int      i_payload = __MIN( i_max, i_data );
1887         block_t *out = block_New( p_stream,
1888                                   RTP_H263_PAYLOAD_START + i_payload );
1889         b_p_bit = (i == 0) ? 1 : 0;
1890         h = ( b_p_bit << 10 )|
1891             ( b_v_bit << 9  )|
1892             ( i_plen  << 3  )|
1893               i_pebit;
1894
1895         /* rtp common header */
1896         //b_m_bit = 1; // always contains end of frame
1897         rtp_packetize_common( id, out, (i == i_count - 1)?1:0,
1898                               in->i_pts > 0 ? in->i_pts : in->i_dts );
1899
1900         /* h263 header */
1901         out->p_buffer[12] = ( h >>  8 )&0xff;
1902         out->p_buffer[13] = ( h       )&0xff;
1903         memcpy( &out->p_buffer[RTP_H263_PAYLOAD_START], p_data, i_payload );
1904
1905         out->i_buffer = RTP_H263_PAYLOAD_START + i_payload;
1906         out->i_dts    = in->i_dts + i * in->i_length / i_count;
1907         out->i_length = in->i_length / i_count;
1908
1909         rtp_packetize_send( id, out );
1910
1911         p_data += i_payload;
1912         i_data -= i_payload;
1913     }
1914
1915     return VLC_SUCCESS;
1916 }
1917
1918 /* rfc3984 */
1919 static int
1920 rtp_packetize_h264_nal( sout_stream_t *p_stream, sout_stream_id_t *id,
1921                         const uint8_t *p_data, int i_data, int64_t i_pts,
1922                         int64_t i_dts, vlc_bool_t b_last, int64_t i_length )
1923 {
1924     const int i_max = id->i_mtu - 12; /* payload max in one packet */
1925     int i_nal_hdr;
1926     int i_nal_type;
1927
1928     if( i_data < 5 )
1929         return VLC_SUCCESS;
1930
1931     i_nal_hdr = p_data[3];
1932     i_nal_type = i_nal_hdr&0x1f;
1933
1934     /* Skip start code */
1935     p_data += 3;
1936     i_data -= 3;
1937
1938     /* */
1939     if( i_data <= i_max )
1940     {
1941         /* Single NAL unit packet */
1942         block_t *out = block_New( p_stream, 12 + i_data );
1943         out->i_dts    = i_dts;
1944         out->i_length = i_length;
1945
1946         /* */
1947         rtp_packetize_common( id, out, b_last, i_pts );
1948         out->i_buffer = 12 + i_data;
1949
1950         memcpy( &out->p_buffer[12], p_data, i_data );
1951
1952         rtp_packetize_send( id, out );
1953     }
1954     else
1955     {
1956         /* FU-A Fragmentation Unit without interleaving */
1957         const int i_count = ( i_data-1 + i_max-2 - 1 ) / (i_max-2);
1958         int i;
1959
1960         p_data++;
1961         i_data--;
1962
1963         for( i = 0; i < i_count; i++ )
1964         {
1965             const int i_payload = __MIN( i_data, i_max-2 );
1966             block_t *out = block_New( p_stream, 12 + 2 + i_payload );
1967             out->i_dts    = i_dts + i * i_length / i_count;
1968             out->i_length = i_length / i_count;
1969
1970             /* */
1971             rtp_packetize_common( id, out, (b_last && i_payload == i_data), i_pts );
1972             out->i_buffer = 14 + i_payload;
1973
1974             /* FU indicator */
1975             out->p_buffer[12] = 0x00 | (i_nal_hdr & 0x60) | 28;
1976             /* FU header */
1977             out->p_buffer[13] = ( i == 0 ? 0x80 : 0x00 ) | ( (i == i_count-1) ? 0x40 : 0x00 )  | i_nal_type;
1978             memcpy( &out->p_buffer[14], p_data, i_payload );
1979
1980             rtp_packetize_send( id, out );
1981
1982             i_data -= i_payload;
1983             p_data += i_payload;
1984         }
1985     }
1986     return VLC_SUCCESS;
1987 }
1988
1989 static int rtp_packetize_h264( sout_stream_t *p_stream, sout_stream_id_t *id,
1990                                block_t *in )
1991 {
1992     const uint8_t *p_buffer = in->p_buffer;
1993     int i_buffer = in->i_buffer;
1994
1995     while( i_buffer > 4 && ( p_buffer[0] != 0 || p_buffer[1] != 0 || p_buffer[2] != 1 ) )
1996     {
1997         i_buffer--;
1998         p_buffer++;
1999     }
2000
2001     /* Split nal units */
2002     while( i_buffer > 4 )
2003     {
2004         int i_offset;
2005         int i_size = i_buffer;
2006         int i_skip = i_buffer;
2007
2008         /* search nal end */
2009         for( i_offset = 4; i_offset+2 < i_buffer ; i_offset++)
2010         {
2011             if( p_buffer[i_offset] == 0 && p_buffer[i_offset+1] == 0 && p_buffer[i_offset+2] == 1 )
2012             {
2013                 /* we found another startcode */
2014                 i_size = i_offset - ( p_buffer[i_offset-1] == 0 ? 1 : 0);
2015                 i_skip = i_offset;
2016                 break;
2017             }
2018         }
2019         /* TODO add STAP-A to remove a lot of overhead with small slice/sei/... */
2020         rtp_packetize_h264_nal( p_stream, id, p_buffer, i_size,
2021                                 (in->i_pts > 0 ? in->i_pts : in->i_dts), in->i_dts,
2022                                 (i_size >= i_buffer), in->i_length * i_size / in->i_buffer );
2023
2024         i_buffer -= i_skip;
2025         p_buffer += i_skip;
2026     }
2027     return VLC_SUCCESS;
2028 }
2029
2030 static int rtp_packetize_amr( sout_stream_t *p_stream, sout_stream_id_t *id,
2031                               block_t *in )
2032 {
2033     int     i_max   = id->i_mtu - 14; /* payload max in one packet */
2034     int     i_count = ( in->i_buffer + i_max - 1 ) / i_max;
2035
2036     uint8_t *p_data = in->p_buffer;
2037     int     i_data  = in->i_buffer;
2038     int     i;
2039
2040     /* Only supports octet-aligned mode */
2041     for( i = 0; i < i_count; i++ )
2042     {
2043         int           i_payload = __MIN( i_max, i_data );
2044         block_t *out = block_New( p_stream, 14 + i_payload );
2045
2046         /* rtp common header */
2047         rtp_packetize_common( id, out, ((i == i_count - 1)?1:0),
2048                               (in->i_pts > 0 ? in->i_pts : in->i_dts) );
2049         /* Payload header */
2050         out->p_buffer[12] = 0xF0; /* CMR */
2051         out->p_buffer[13] = p_data[0]&0x7C; /* ToC */ /* FIXME: frame type */
2052
2053         /* FIXME: are we fed multiple frames ? */
2054         memcpy( &out->p_buffer[14], p_data+1, i_payload-1 );
2055
2056         out->i_buffer   = 14 + i_payload-1;
2057         out->i_dts    = in->i_dts + i * in->i_length / i_count;
2058         out->i_length = in->i_length / i_count;
2059
2060         rtp_packetize_send( id, out );
2061
2062         p_data += i_payload;
2063         i_data -= i_payload;
2064     }
2065
2066     return VLC_SUCCESS;
2067 }
2068
2069 static int rtp_packetize_t140( sout_stream_t *p_stream, sout_stream_id_t *id,
2070                                block_t *in )
2071 {
2072     const size_t   i_max  = id->i_mtu - 12;
2073     const uint8_t *p_data = in->p_buffer;
2074     size_t         i_data = in->i_buffer;
2075
2076     for( unsigned i_packet = 0; i_data > 0; i_packet++ )
2077     {
2078         size_t i_payload = i_data;
2079
2080         /* Make sure we stop on an UTF-8 character boundary
2081          * (assuming the input is valid UTF-8) */
2082         if( i_data > i_max )
2083         {
2084             i_payload = i_max;
2085
2086             while( ( p_data[i_payload] & 0xC0 ) == 0x80 )
2087             {
2088                 if( i_payload == 0 )
2089                     return VLC_SUCCESS; /* fishy input! */
2090
2091                 i_payload--;
2092             }
2093         }
2094
2095         block_t *out = block_New( p_stream, 12 + i_payload );
2096         if( out == NULL )
2097             return VLC_SUCCESS;
2098
2099         rtp_packetize_common( id, out, 0, in->i_pts + i_packet );
2100         memcpy( out->p_buffer + 12, p_data, i_payload );
2101
2102         out->i_buffer = 12 + i_payload;
2103         out->i_dts    = out->i_pts;
2104         out->i_length = 0;
2105
2106         rtp_packetize_send( id, out );
2107
2108         p_data += i_payload;
2109         i_data -= i_payload;
2110     }
2111
2112     return VLC_SUCCESS;
2113 }
2114
2115 /*****************************************************************************
2116  * Non-RTP mux
2117  *****************************************************************************/
2118
2119 /** Add an ES to a non-RTP muxed stream */
2120 static sout_stream_id_t *MuxAdd( sout_stream_t *p_stream, es_format_t *p_fmt )
2121 {
2122     sout_input_t      *p_input;
2123     sout_mux_t *p_mux = p_stream->p_sys->p_mux;
2124     assert( p_mux != NULL );
2125
2126     p_input = sout_MuxAddStream( p_mux, p_fmt );
2127     if( p_input == NULL )
2128     {
2129         msg_Err( p_stream, "cannot add this stream to the muxer" );
2130         return NULL;
2131     }
2132
2133     return (sout_stream_id_t *)p_input;
2134 }
2135
2136
2137 static int MuxSend( sout_stream_t *p_stream, sout_stream_id_t *id,
2138                     block_t *p_buffer )
2139 {
2140     sout_mux_t *p_mux = p_stream->p_sys->p_mux;
2141     assert( p_mux != NULL );
2142
2143     sout_MuxSendBuffer( p_mux, (sout_input_t *)id, p_buffer );
2144     return VLC_SUCCESS;
2145 }
2146
2147
2148 /** Remove an ES from a non-RTP muxed stream */
2149 static int MuxDel( sout_stream_t *p_stream, sout_stream_id_t *id )
2150 {
2151     sout_mux_t *p_mux = p_stream->p_sys->p_mux;
2152     assert( p_mux != NULL );
2153
2154     sout_MuxDeleteStream( p_mux, (sout_input_t *)id );
2155     return VLC_SUCCESS;
2156 }
2157
2158
2159 static int AccessOutGrabberWriteBuffer( sout_stream_t *p_stream,
2160                                         const block_t *p_buffer )
2161 {
2162     sout_stream_sys_t *p_sys = p_stream->p_sys;
2163     sout_stream_id_t *id = p_sys->es[0];
2164
2165     int64_t  i_dts = p_buffer->i_dts;
2166
2167     uint8_t         *p_data = p_buffer->p_buffer;
2168     unsigned int    i_data  = p_buffer->i_buffer;
2169     unsigned int    i_max   = id->i_mtu - 12;
2170
2171     unsigned i_packet = ( p_buffer->i_buffer + i_max - 1 ) / i_max;
2172
2173     while( i_data > 0 )
2174     {
2175         unsigned int i_size;
2176
2177         /* output complete packet */
2178         if( p_sys->packet &&
2179             p_sys->packet->i_buffer + i_data > i_max )
2180         {
2181             rtp_packetize_send( id, p_sys->packet );
2182             p_sys->packet = NULL;
2183         }
2184
2185         if( p_sys->packet == NULL )
2186         {
2187             /* allocate a new packet */
2188             p_sys->packet = block_New( p_stream, id->i_mtu );
2189             rtp_packetize_common( id, p_sys->packet, 1, i_dts );
2190             p_sys->packet->i_dts = i_dts;
2191             p_sys->packet->i_length = p_buffer->i_length / i_packet;
2192             i_dts += p_sys->packet->i_length;
2193         }
2194
2195         i_size = __MIN( i_data,
2196                         (unsigned)(id->i_mtu - p_sys->packet->i_buffer) );
2197
2198         memcpy( &p_sys->packet->p_buffer[p_sys->packet->i_buffer],
2199                 p_data, i_size );
2200
2201         p_sys->packet->i_buffer += i_size;
2202         p_data += i_size;
2203         i_data -= i_size;
2204     }
2205
2206     return VLC_SUCCESS;
2207 }
2208
2209
2210 static int AccessOutGrabberWrite( sout_access_out_t *p_access,
2211                                   block_t *p_buffer )
2212 {
2213     sout_stream_t *p_stream = (sout_stream_t*)p_access->p_sys;
2214
2215     while( p_buffer )
2216     {
2217         block_t *p_next;
2218
2219         AccessOutGrabberWriteBuffer( p_stream, p_buffer );
2220
2221         p_next = p_buffer->p_next;
2222         block_Release( p_buffer );
2223         p_buffer = p_next;
2224     }
2225
2226     return VLC_SUCCESS;
2227 }
2228
2229
2230 static sout_access_out_t *GrabberCreate( sout_stream_t *p_stream )
2231 {
2232     sout_access_out_t *p_grab;
2233
2234     p_grab = vlc_object_create( p_stream->p_sout, sizeof( *p_grab ) );
2235     if( p_grab == NULL )
2236         return NULL;
2237
2238     p_grab->p_module    = NULL;
2239     p_grab->p_sout      = p_stream->p_sout;
2240     p_grab->psz_access  = strdup( "grab" );
2241     p_grab->p_cfg       = NULL;
2242     p_grab->psz_path    = strdup( "" );
2243     p_grab->p_sys       = (sout_access_out_sys_t *)p_stream;
2244     p_grab->pf_seek     = NULL;
2245     p_grab->pf_write    = AccessOutGrabberWrite;
2246     return p_grab;
2247 }