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