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