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