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