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