As yet untested T.140 RTP packetization
[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 static int rtp_packetize_t140 ( sout_stream_t *, sout_stream_id_t *, block_t * );
745
746 static void sprintf_hexa( char *s, uint8_t *p_data, int i_data )
747 {
748     static const char hex[16] = "0123456789abcdef";
749     int i;
750
751     for( i = 0; i < i_data; i++ )
752     {
753         s[2*i+0] = hex[(p_data[i]>>4)&0xf];
754         s[2*i+1] = hex[(p_data[i]   )&0xf];
755     }
756     s[2*i_data] = '\0';
757 }
758
759
760 /** Add an ES as a new RTP stream */
761 static sout_stream_id_t *Add( sout_stream_t *p_stream, es_format_t *p_fmt )
762 {
763     /* NOTE: As a special case, if we use a non-RTP
764      * mux (TS/PS), then p_fmt is NULL. */
765     sout_stream_sys_t *p_sys = p_stream->p_sys;
766     sout_stream_id_t  *id;
767     int               i_port, cscov = -1;
768     char              *psz_sdp;
769
770     id = vlc_object_create( p_stream, sizeof( sout_stream_id_t ) );
771     if( id == NULL )
772         return NULL;
773
774     /* Choose the port */
775     i_port = 0;
776     if( p_fmt == NULL )
777         ;
778     else
779     if( p_fmt->i_cat == AUDIO_ES && p_sys->i_port_audio > 0 )
780     {
781         i_port = p_sys->i_port_audio;
782         p_sys->i_port_audio = 0;
783     }
784     else
785     if( p_fmt->i_cat == VIDEO_ES && p_sys->i_port_video > 0 )
786     {
787         i_port = p_sys->i_port_video;
788         p_sys->i_port_video = 0;
789     }
790
791     while( i_port == 0 )
792     {
793         if( p_sys->i_port != p_sys->i_port_audio
794          && p_sys->i_port != p_sys->i_port_video )
795         {
796             i_port = p_sys->i_port;
797             p_sys->i_port += 2;
798             break;
799         }
800         p_sys->i_port += 2;
801     }
802
803     id->p_stream   = p_stream;
804
805     id->i_timestamp_start = rand()&0xffffffff;
806     id->i_sequence = rand()&0xffff;
807     id->i_payload_type = p_sys->i_payload_type;
808     id->ssrc[0] = rand()&0xff;
809     id->ssrc[1] = rand()&0xff;
810     id->ssrc[2] = rand()&0xff;
811     id->ssrc[3] = rand()&0xff;
812
813     id->psz_rtpmap = NULL;
814     id->psz_fmtp   = NULL;
815     id->i_clock_rate = 90000; /* most common case */
816     id->i_port     = i_port;
817     if( p_fmt != NULL )
818     {
819         id->i_cat  = p_fmt->i_cat;
820         id->i_bitrate = p_fmt->i_bitrate/1000; /* Stream bitrate in kbps */
821     }
822     else
823     {
824         id->i_cat  = VIDEO_ES;
825         id->i_bitrate = 0;
826     }
827
828     id->pf_packetize = NULL;
829     id->i_mtu = config_GetInt( p_stream, "mtu" );
830     if( id->i_mtu <= 12 + 16 )
831         id->i_mtu = 576 - 20 - 8; /* pessimistic */
832
833     msg_Dbg( p_stream, "maximum RTP packet size: %d bytes", id->i_mtu );
834
835     vlc_mutex_init( p_stream, &id->lock_sink );
836     id->sinkc = 0;
837     id->sinkv = NULL;
838     id->rtsp_id = NULL;
839
840     id->i_caching =
841         (int64_t)1000 * var_GetInteger( p_stream, SOUT_CFG_PREFIX "caching");
842     id->p_fifo = block_FifoNew( p_stream );
843
844     if( vlc_thread_create( id, "RTP send thread", ThreadSend,
845                            VLC_THREAD_PRIORITY_HIGHEST, VLC_FALSE ) )
846     {
847         vlc_mutex_destroy( &id->lock_sink );
848         vlc_object_destroy( id );
849         return NULL;
850     }
851
852     if( p_sys->psz_destination != NULL )
853     {
854         int ttl = (p_sys->i_ttl > 0) ? p_sys->i_ttl : -1;
855         int fd = net_ConnectDgram( p_stream, p_sys->psz_destination,
856                                    i_port, ttl, p_sys->proto );
857
858         if( fd == -1 )
859         {
860             msg_Err( p_stream, "cannot create RTP socket" );
861             vlc_thread_join( id );
862             vlc_mutex_destroy( &id->lock_sink );
863             vlc_object_destroy( id );
864             return NULL;
865         }
866         rtp_add_sink( id, fd );
867     }
868
869     if( p_fmt == NULL )
870     {
871         char *psz = var_GetNonEmptyString( p_stream, SOUT_CFG_PREFIX "mux" );
872
873         if( psz == NULL ) /* Uho! */
874             ;
875         else
876         if( strncmp( psz, "ts", 2 ) == 0 )
877         {
878             id->i_payload_type = 33;
879             id->psz_rtpmap = strdup( "MP2T/90000" );
880         }
881         else
882         {
883             id->psz_rtpmap = strdup( "MP2P/90000" );
884         }
885     }
886     else
887     switch( p_fmt->i_codec )
888     {
889         case VLC_FOURCC( 's', '1', '6', 'b' ):
890             if( p_fmt->audio.i_channels == 1 && p_fmt->audio.i_rate == 44100 )
891             {
892                 id->i_payload_type = 11;
893             }
894             else if( p_fmt->audio.i_channels == 2 &&
895                      p_fmt->audio.i_rate == 44100 )
896             {
897                 id->i_payload_type = 10;
898             }
899             if( asprintf( &id->psz_rtpmap, "L16/%d/%d", p_fmt->audio.i_rate,
900                           p_fmt->audio.i_channels ) == -1 )
901                 id->psz_rtpmap = NULL;
902             id->i_clock_rate = p_fmt->audio.i_rate;
903             id->pf_packetize = rtp_packetize_l16;
904             break;
905         case VLC_FOURCC( 'u', '8', ' ', ' ' ):
906             if( asprintf( &id->psz_rtpmap, "L8/%d/%d", p_fmt->audio.i_rate,
907                           p_fmt->audio.i_channels ) == -1 )
908                 id->psz_rtpmap = NULL;
909             id->i_clock_rate = p_fmt->audio.i_rate;
910             id->pf_packetize = rtp_packetize_l8;
911             break;
912         case VLC_FOURCC( 'm', 'p', 'g', 'a' ):
913             id->i_payload_type = 14;
914             id->psz_rtpmap = strdup( "MPA/90000" );
915             id->pf_packetize = rtp_packetize_mpa;
916             break;
917         case VLC_FOURCC( 'm', 'p', 'g', 'v' ):
918             id->i_payload_type = 32;
919             id->psz_rtpmap = strdup( "MPV/90000" );
920             id->pf_packetize = rtp_packetize_mpv;
921             break;
922         case VLC_FOURCC( 'a', '5', '2', ' ' ):
923             id->psz_rtpmap = strdup( "ac3/90000" );
924             id->pf_packetize = rtp_packetize_ac3;
925             break;
926         case VLC_FOURCC( 'H', '2', '6', '3' ):
927             id->psz_rtpmap = strdup( "H263-1998/90000" );
928             id->pf_packetize = rtp_packetize_h263;
929             break;
930         case VLC_FOURCC( 'h', '2', '6', '4' ):
931             id->psz_rtpmap = strdup( "H264/90000" );
932             id->pf_packetize = rtp_packetize_h264;
933             id->psz_fmtp = NULL;
934
935             if( p_fmt->i_extra > 0 )
936             {
937                 uint8_t *p_buffer = p_fmt->p_extra;
938                 int     i_buffer = p_fmt->i_extra;
939                 char    *p_64_sps = NULL;
940                 char    *p_64_pps = NULL;
941                 char    hexa[6+1];
942
943                 while( i_buffer > 4 &&
944                        p_buffer[0] == 0 && p_buffer[1] == 0 &&
945                        p_buffer[2] == 0 && p_buffer[3] == 1 )
946                 {
947                     const int i_nal_type = p_buffer[4]&0x1f;
948                     int i_offset;
949                     int i_size      = 0;
950
951                     msg_Dbg( p_stream, "we found a startcode for NAL with TYPE:%d", i_nal_type );
952
953                     i_size = i_buffer;
954                     for( i_offset = 4; i_offset+3 < i_buffer ; i_offset++)
955                     {
956                         if( !memcmp (p_buffer + i_offset, "\x00\x00\x00\x01", 4 ) )
957                         {
958                             /* we found another startcode */
959                             i_size = i_offset;
960                             break;
961                         }
962                     }
963                     if( i_nal_type == 7 )
964                     {
965                         p_64_sps = vlc_b64_encode_binary( &p_buffer[4], i_size - 4 );
966                         sprintf_hexa( hexa, &p_buffer[5], 3 );
967                     }
968                     else if( i_nal_type == 8 )
969                     {
970                         p_64_pps = vlc_b64_encode_binary( &p_buffer[4], i_size - 4 );
971                     }
972                     i_buffer -= i_size;
973                     p_buffer += i_size;
974                 }
975                 /* */
976                 if( p_64_sps && p_64_pps &&
977                     ( asprintf( &id->psz_fmtp,
978                                 "packetization-mode=1;profile-level-id=%s;"
979                                 "sprop-parameter-sets=%s,%s;", hexa, p_64_sps,
980                                 p_64_pps ) == -1 ) )
981                     id->psz_fmtp = NULL;
982                 free( p_64_sps );
983                 free( p_64_pps );
984             }
985             if( !id->psz_fmtp )
986                 id->psz_fmtp = strdup( "packetization-mode=1" );
987             break;
988
989         case VLC_FOURCC( 'm', 'p', '4', 'v' ):
990         {
991             char hexa[2*p_fmt->i_extra +1];
992
993             id->psz_rtpmap = strdup( "MP4V-ES/90000" );
994             id->pf_packetize = rtp_packetize_split;
995             if( p_fmt->i_extra > 0 )
996             {
997                 sprintf_hexa( hexa, p_fmt->p_extra, p_fmt->i_extra );
998                 if( asprintf( &id->psz_fmtp,
999                               "profile-level-id=3; config=%s;", hexa ) == -1 )
1000                     id->psz_fmtp = NULL;
1001             }
1002             break;
1003         }
1004         case VLC_FOURCC( 'm', 'p', '4', 'a' ):
1005         {
1006             id->i_clock_rate = p_fmt->audio.i_rate;
1007
1008             if(!p_sys->b_latm)
1009             {
1010                 char hexa[2*p_fmt->i_extra +1];
1011
1012                 if( asprintf( &id->psz_rtpmap, "mpeg4-generic/%d",
1013                               p_fmt->audio.i_rate ) == -1 )
1014                     id->psz_rtpmap = NULL;
1015                 id->pf_packetize = rtp_packetize_mp4a;
1016                 sprintf_hexa( hexa, p_fmt->p_extra, p_fmt->i_extra );
1017                 if( asprintf( &id->psz_fmtp,
1018                               "streamtype=5; profile-level-id=15; "
1019                               "mode=AAC-hbr; config=%s; SizeLength=13; "
1020                               "IndexLength=3; IndexDeltaLength=3; Profile=1;",
1021                               hexa ) == -1 )
1022                     id->psz_fmtp = NULL;
1023             }
1024             else
1025             {
1026                 char hexa[13];
1027                 int i;
1028                 unsigned char config[6];
1029                 unsigned int aacsrates[15] = {
1030                     96000, 88200, 64000, 48000, 44100, 32000, 24000, 22050,
1031                     16000, 12000, 11025, 8000, 7350, 0, 0 };
1032
1033                 for( i = 0; i < 15; i++ )
1034                     if( p_fmt->audio.i_rate == aacsrates[i] )
1035                         break;
1036
1037                 config[0]=0x40;
1038                 config[1]=0;
1039                 config[2]=0x20|i;
1040                 config[3]=p_fmt->audio.i_channels<<4;
1041                 config[4]=0x3f;
1042                 config[5]=0xc0;
1043
1044                 if( asprintf( &id->psz_rtpmap, "MP4A-LATM/%d/%d",
1045                               p_fmt->audio.i_rate,
1046                               p_fmt->audio.i_channels ) == -1)
1047                     id->psz_rtpmap = NULL;
1048                 id->pf_packetize = rtp_packetize_mp4a_latm;
1049                 sprintf_hexa( hexa, config, 6 );
1050                 if( asprintf( &id->psz_fmtp, "profile-level-id=15; "
1051                               "object=2; cpresent=0; config=%s", hexa ) == -1 )
1052                     id->psz_fmtp = NULL;
1053             }
1054             break;
1055         }
1056         case VLC_FOURCC( 's', 'a', 'm', 'r' ):
1057             id->psz_rtpmap = strdup( p_fmt->audio.i_channels == 2 ?
1058                                      "AMR/8000/2" : "AMR/8000" );
1059             id->psz_fmtp = strdup( "octet-align=1" );
1060             id->i_clock_rate = p_fmt->audio.i_rate;
1061             id->pf_packetize = rtp_packetize_amr;
1062             break;
1063         case VLC_FOURCC( 's', 'a', 'w', 'b' ):
1064             id->psz_rtpmap = strdup( p_fmt->audio.i_channels == 2 ?
1065                                      "AMR-WB/16000/2" : "AMR-WB/16000" );
1066             id->psz_fmtp = strdup( "octet-align=1" );
1067             id->i_clock_rate = p_fmt->audio.i_rate;
1068             id->pf_packetize = rtp_packetize_amr;
1069             break;
1070         case VLC_FOURCC( 't', '1', '4', '0' ):
1071             id->psz_rtpmap = strdup( "t140/1000" );
1072             id->i_clock_rate = 1000;
1073             id->pf_packetize = rtp_packetize_t140;
1074             break;
1075
1076         default:
1077             msg_Err( p_stream, "cannot add this stream (unsupported "
1078                      "codec:%4.4s)", (char*)&p_fmt->i_codec );
1079             if( id->sinkc > 0 )
1080                 rtp_del_sink( id, id->sinkv[0].rtp_fd );
1081             vlc_thread_join( id );
1082             vlc_mutex_destroy( &id->lock_sink );
1083             vlc_object_destroy( id );
1084             return NULL;
1085     }
1086
1087     if( cscov != -1 )
1088         cscov += 8 /* UDP */ + 12 /* RTP */;
1089     if( id->sinkc > 0 )
1090         net_SetCSCov( id->sinkv[0].rtp_fd, cscov, -1 );
1091
1092     if( id->i_payload_type == p_sys->i_payload_type )
1093         p_sys->i_payload_type++;
1094
1095     if( p_sys->rtsp != NULL )
1096         id->rtsp_id = RtspAddId( p_sys->rtsp, id, p_sys->i_es,
1097                                  p_sys->psz_destination,
1098                                  p_sys->i_ttl, id->i_port, id->i_port + 1 );
1099
1100     /* Update p_sys context */
1101     vlc_mutex_lock( &p_sys->lock_es );
1102     TAB_APPEND( p_sys->i_es, p_sys->es, id );
1103     vlc_mutex_unlock( &p_sys->lock_es );
1104
1105     psz_sdp = SDPGenerate( p_stream, NULL );
1106
1107     vlc_mutex_lock( &p_sys->lock_sdp );
1108     free( p_sys->psz_sdp );
1109     p_sys->psz_sdp = psz_sdp;
1110     vlc_mutex_unlock( &p_sys->lock_sdp );
1111
1112     p_sys->i_sdp_version++;
1113
1114     msg_Dbg( p_stream, "sdp=\n%s", p_sys->psz_sdp );
1115
1116     /* Update SDP (sap/file) */
1117     if( p_sys->b_export_sap ) SapSetup( p_stream );
1118     if( p_sys->b_export_sdp_file ) FileSetup( p_stream );
1119
1120     vlc_object_attach( id, p_stream );
1121     return id;
1122 }
1123
1124 static int Del( sout_stream_t *p_stream, sout_stream_id_t *id )
1125 {
1126     sout_stream_sys_t *p_sys = p_stream->p_sys;
1127
1128     vlc_object_kill( id );
1129
1130     vlc_mutex_lock( &p_sys->lock_es );
1131     TAB_REMOVE( p_sys->i_es, p_sys->es, id );
1132     vlc_mutex_unlock( &p_sys->lock_es );
1133
1134     /* Release port */
1135     if( id->i_port > 0 )
1136     {
1137         if( id->i_cat == AUDIO_ES && p_sys->i_port_audio == 0 )
1138             p_sys->i_port_audio = id->i_port;
1139         else if( id->i_cat == VIDEO_ES && p_sys->i_port_video == 0 )
1140             p_sys->i_port_video = id->i_port;
1141     }
1142
1143     free( id->psz_rtpmap );
1144     free( id->psz_fmtp );
1145
1146     if( id->rtsp_id )
1147         RtspDelId( p_sys->rtsp, id->rtsp_id );
1148     if( id->sinkc > 0 )
1149         rtp_del_sink( id, id->sinkv[0].rtp_fd ); /* sink for explicit dst= */
1150
1151     vlc_thread_join( id );
1152     vlc_mutex_destroy( &id->lock_sink );
1153     block_FifoRelease( id->p_fifo );
1154
1155     /* Update SDP (sap/file) */
1156     if( p_sys->b_export_sap && !p_sys->p_mux ) SapSetup( p_stream );
1157     if( p_sys->b_export_sdp_file ) FileSetup( p_stream );
1158
1159     vlc_object_detach( id );
1160     vlc_object_destroy( id );
1161     return VLC_SUCCESS;
1162 }
1163
1164 static int Send( sout_stream_t *p_stream, sout_stream_id_t *id,
1165                  block_t *p_buffer )
1166 {
1167     block_t *p_next;
1168
1169     assert( p_stream->p_sys->p_mux == NULL );
1170
1171     while( p_buffer != NULL )
1172     {
1173         p_next = p_buffer->p_next;
1174         if( id->pf_packetize( p_stream, id, p_buffer ) )
1175             break;
1176
1177         block_Release( p_buffer );
1178         p_buffer = p_next;
1179     }
1180     return VLC_SUCCESS;
1181 }
1182
1183 /****************************************************************************
1184  * SAP:
1185  ****************************************************************************/
1186 static int SapSetup( sout_stream_t *p_stream )
1187 {
1188     sout_stream_sys_t *p_sys = p_stream->p_sys;
1189     sout_instance_t   *p_sout = p_stream->p_sout;
1190     announce_method_t *p_method = sout_SAPMethod();
1191
1192     /* Remove the previous session */
1193     if( p_sys->p_session != NULL)
1194     {
1195         sout_AnnounceUnRegister( p_sout, p_sys->p_session);
1196         sout_AnnounceSessionDestroy( p_sys->p_session );
1197         p_sys->p_session = NULL;
1198     }
1199
1200     if( ( p_sys->i_es > 0 || p_sys->p_mux ) && p_sys->psz_sdp && *p_sys->psz_sdp )
1201     {
1202         p_sys->p_session = sout_AnnounceRegisterSDP( p_sout, SOUT_CFG_PREFIX,
1203                                                      p_sys->psz_sdp,
1204                                                      p_sys->psz_destination,
1205                                                      p_method );
1206     }
1207
1208     sout_MethodRelease( p_method );
1209     return VLC_SUCCESS;
1210 }
1211
1212 /****************************************************************************
1213 * File:
1214 ****************************************************************************/
1215 static int FileSetup( sout_stream_t *p_stream )
1216 {
1217     sout_stream_sys_t *p_sys = p_stream->p_sys;
1218     FILE            *f;
1219
1220     if( ( f = utf8_fopen( p_sys->psz_sdp_file, "wt" ) ) == NULL )
1221     {
1222         msg_Err( p_stream, "cannot open file '%s' (%s)",
1223                  p_sys->psz_sdp_file, strerror(errno) );
1224         return VLC_EGENERIC;
1225     }
1226
1227     fputs( p_sys->psz_sdp, f );
1228     fclose( f );
1229
1230     return VLC_SUCCESS;
1231 }
1232
1233 /****************************************************************************
1234  * HTTP:
1235  ****************************************************************************/
1236 static int  HttpCallback( httpd_file_sys_t *p_args,
1237                           httpd_file_t *, uint8_t *p_request,
1238                           uint8_t **pp_data, int *pi_data );
1239
1240 static int HttpSetup( sout_stream_t *p_stream, vlc_url_t *url)
1241 {
1242     sout_stream_sys_t *p_sys = p_stream->p_sys;
1243
1244     p_sys->p_httpd_host = httpd_HostNew( VLC_OBJECT(p_stream), url->psz_host,
1245                                          url->i_port > 0 ? url->i_port : 80 );
1246     if( p_sys->p_httpd_host )
1247     {
1248         p_sys->p_httpd_file = httpd_FileNew( p_sys->p_httpd_host,
1249                                              url->psz_path ? url->psz_path : "/",
1250                                              "application/sdp",
1251                                              NULL, NULL, NULL,
1252                                              HttpCallback, (void*)p_sys );
1253     }
1254     if( p_sys->p_httpd_file == NULL )
1255     {
1256         return VLC_EGENERIC;
1257     }
1258     return VLC_SUCCESS;
1259 }
1260
1261 static int  HttpCallback( httpd_file_sys_t *p_args,
1262                           httpd_file_t *f, uint8_t *p_request,
1263                           uint8_t **pp_data, int *pi_data )
1264 {
1265     sout_stream_sys_t *p_sys = (sout_stream_sys_t*)p_args;
1266
1267     vlc_mutex_lock( &p_sys->lock_sdp );
1268     if( p_sys->psz_sdp && *p_sys->psz_sdp )
1269     {
1270         *pi_data = strlen( p_sys->psz_sdp );
1271         *pp_data = malloc( *pi_data );
1272         memcpy( *pp_data, p_sys->psz_sdp, *pi_data );
1273     }
1274     else
1275     {
1276         *pp_data = NULL;
1277         *pi_data = 0;
1278     }
1279     vlc_mutex_unlock( &p_sys->lock_sdp );
1280
1281     return VLC_SUCCESS;
1282 }
1283
1284 /****************************************************************************
1285  * RTP send
1286  ****************************************************************************/
1287 static void ThreadSend( vlc_object_t *p_this )
1288 {
1289     sout_stream_id_t *id = (sout_stream_id_t *)p_this;
1290     unsigned i_caching = id->i_caching;
1291 #ifdef HAVE_TEE
1292     int fd[5] = { -1, -1, -1, -1, -1 };
1293
1294     if( pipe( fd ) )
1295         fd[0] = fd[1] = -1;
1296     else
1297     if( pipe( fd ) )
1298         fd[2] = fd[3] = -1;
1299     else
1300         fd[4] = open( "/dev/null", O_WRONLY );
1301 #endif
1302
1303     while( !id->b_die )
1304     {
1305         block_t *out = block_FifoGet( id->p_fifo );
1306         mtime_t  i_date = out->i_dts + i_caching;
1307         ssize_t  len = out->i_buffer;
1308
1309 #ifdef HAVE_TEE
1310         if( fd[4] != -1 )
1311             len = write( fd[1], out->p_buffer, len);
1312         if( len == -1 )
1313             continue; /* Uho - should not happen */
1314 #endif
1315         mwait( i_date );
1316
1317         vlc_mutex_lock( &id->lock_sink );
1318         for( int i = 0; i < id->sinkc; i++ )
1319         {
1320             SendRTCP( id->sinkv[i].rtcp, out );
1321
1322 #ifdef HAVE_TEE
1323             tee( fd[0], fd[3], len, 0 );
1324             if( splice( fd[2], NULL, id->sinkv[i].rtp_fd, NULL, len,
1325                         SPLICE_F_NONBLOCK ) >= 0 )
1326                 continue;
1327
1328             /* splice failed */
1329             splice( fd[2], NULL, fd[4], NULL, len, 0 );
1330 #endif
1331             send( id->sinkv[i].rtp_fd, out->p_buffer, len, 0 );
1332         }
1333         vlc_mutex_unlock( &id->lock_sink );
1334
1335         block_Release( out );
1336 #ifdef HAVE_TEE
1337         splice( fd[0], NULL, fd[4], NULL, len, 0 );
1338 #endif
1339     }
1340
1341 #ifdef HAVE_TEE
1342     for( unsigned i = 0; i < 5; i++ )
1343         close( fd[i] );
1344 #endif
1345 }
1346
1347 static inline void rtp_packetize_send( sout_stream_id_t *id, block_t *out )
1348 {
1349     block_FifoPut( id->p_fifo, out );
1350 }
1351
1352 int rtp_add_sink( sout_stream_id_t *id, int fd )
1353 {
1354     rtp_sink_t sink = { fd, NULL };
1355     sink.rtcp = OpenRTCP( VLC_OBJECT( id->p_stream ), fd, IPPROTO_UDP );
1356     if( sink.rtcp == NULL )
1357         msg_Err( id, "RTCP failed!" );
1358
1359     vlc_mutex_lock( &id->lock_sink );
1360     INSERT_ELEM( id->sinkv, id->sinkc, id->sinkc, sink );
1361     vlc_mutex_unlock( &id->lock_sink );
1362     return VLC_SUCCESS;
1363 }
1364
1365 void rtp_del_sink( sout_stream_id_t *id, int fd )
1366 {
1367     rtp_sink_t sink = { fd, NULL };
1368
1369     /* NOTE: must be safe to use if fd is not included */
1370     vlc_mutex_lock( &id->lock_sink );
1371     for( int i = 0; i < id->sinkc; i++ )
1372     {
1373         if (id->sinkv[i].rtp_fd == fd)
1374         {
1375             sink = id->sinkv[i];
1376             REMOVE_ELEM( id->sinkv, id->sinkc, i );
1377             break;
1378         }
1379     }
1380     vlc_mutex_unlock( &id->lock_sink );
1381
1382     CloseRTCP( sink.rtcp );
1383     net_Close( sink.rtp_fd );
1384 }
1385
1386 /****************************************************************************
1387  * rtp_packetize_*:
1388  ****************************************************************************/
1389 static void rtp_packetize_common( sout_stream_id_t *id, block_t *out,
1390                                   int b_marker, int64_t i_pts )
1391 {
1392     uint32_t i_timestamp = i_pts * (int64_t)id->i_clock_rate / I64C(1000000);
1393
1394     out->p_buffer[0] = 0x80;
1395     out->p_buffer[1] = (b_marker?0x80:0x00)|id->i_payload_type;
1396     out->p_buffer[2] = ( id->i_sequence >> 8)&0xff;
1397     out->p_buffer[3] = ( id->i_sequence     )&0xff;
1398     out->p_buffer[4] = ( i_timestamp >> 24 )&0xff;
1399     out->p_buffer[5] = ( i_timestamp >> 16 )&0xff;
1400     out->p_buffer[6] = ( i_timestamp >>  8 )&0xff;
1401     out->p_buffer[7] = ( i_timestamp       )&0xff;
1402
1403     memcpy( out->p_buffer + 8, id->ssrc, 4 );
1404
1405     out->i_buffer = 12;
1406     id->i_sequence++;
1407 }
1408
1409 static int rtp_packetize_mpa( sout_stream_t *p_stream, sout_stream_id_t *id,
1410                               block_t *in )
1411 {
1412     int     i_max   = id->i_mtu - 12 - 4; /* payload max in one packet */
1413     int     i_count = ( in->i_buffer + i_max - 1 ) / i_max;
1414
1415     uint8_t *p_data = in->p_buffer;
1416     int     i_data  = in->i_buffer;
1417     int     i;
1418
1419     for( i = 0; i < i_count; i++ )
1420     {
1421         int           i_payload = __MIN( i_max, i_data );
1422         block_t *out = block_New( p_stream, 16 + i_payload );
1423
1424         /* rtp common header */
1425         rtp_packetize_common( id, out, (i == i_count - 1)?1:0, in->i_pts );
1426         /* mbz set to 0 */
1427         out->p_buffer[12] = 0;
1428         out->p_buffer[13] = 0;
1429         /* fragment offset in the current frame */
1430         out->p_buffer[14] = ( (i*i_max) >> 8 )&0xff;
1431         out->p_buffer[15] = ( (i*i_max)      )&0xff;
1432         memcpy( &out->p_buffer[16], p_data, i_payload );
1433
1434         out->i_buffer   = 16 + i_payload;
1435         out->i_dts    = in->i_dts + i * in->i_length / i_count;
1436         out->i_length = in->i_length / i_count;
1437
1438         rtp_packetize_send( id, out );
1439
1440         p_data += i_payload;
1441         i_data -= i_payload;
1442     }
1443
1444     return VLC_SUCCESS;
1445 }
1446
1447 /* rfc2250 */
1448 static int rtp_packetize_mpv( sout_stream_t *p_stream, sout_stream_id_t *id,
1449                               block_t *in )
1450 {
1451     int     i_max   = id->i_mtu - 12 - 4; /* payload max in one packet */
1452     int     i_count = ( in->i_buffer + i_max - 1 ) / i_max;
1453
1454     uint8_t *p_data = in->p_buffer;
1455     int     i_data  = in->i_buffer;
1456     int     i;
1457     int     b_sequence_start = 0;
1458     int     i_temporal_ref = 0;
1459     int     i_picture_coding_type = 0;
1460     int     i_fbv = 0, i_bfc = 0, i_ffv = 0, i_ffc = 0;
1461     int     b_start_slice = 0;
1462
1463     /* preparse this packet to get some info */
1464     if( in->i_buffer > 4 )
1465     {
1466         uint8_t *p = p_data;
1467         int      i_rest = in->i_buffer;
1468
1469         for( ;; )
1470         {
1471             while( i_rest > 4 &&
1472                    ( p[0] != 0x00 || p[1] != 0x00 || p[2] != 0x01 ) )
1473             {
1474                 p++;
1475                 i_rest--;
1476             }
1477             if( i_rest <= 4 )
1478             {
1479                 break;
1480             }
1481             p += 3;
1482             i_rest -= 4;
1483
1484             if( *p == 0xb3 )
1485             {
1486                 /* sequence start code */
1487                 b_sequence_start = 1;
1488             }
1489             else if( *p == 0x00 && i_rest >= 4 )
1490             {
1491                 /* picture */
1492                 i_temporal_ref = ( p[1] << 2) |((p[2]>>6)&0x03);
1493                 i_picture_coding_type = (p[2] >> 3)&0x07;
1494
1495                 if( i_rest >= 4 && ( i_picture_coding_type == 2 ||
1496                                     i_picture_coding_type == 3 ) )
1497                 {
1498                     i_ffv = (p[3] >> 2)&0x01;
1499                     i_ffc = ((p[3]&0x03) << 1)|((p[4]>>7)&0x01);
1500                     if( i_rest > 4 && i_picture_coding_type == 3 )
1501                     {
1502                         i_fbv = (p[4]>>6)&0x01;
1503                         i_bfc = (p[4]>>3)&0x07;
1504                     }
1505                 }
1506             }
1507             else if( *p <= 0xaf )
1508             {
1509                 b_start_slice = 1;
1510             }
1511         }
1512     }
1513
1514     for( i = 0; i < i_count; i++ )
1515     {
1516         int           i_payload = __MIN( i_max, i_data );
1517         block_t *out = block_New( p_stream,
1518                                              16 + i_payload );
1519         uint32_t      h = ( i_temporal_ref << 16 )|
1520                           ( b_sequence_start << 13 )|
1521                           ( b_start_slice << 12 )|
1522                           ( i == i_count - 1 ? 1 << 11 : 0 )|
1523                           ( i_picture_coding_type << 8 )|
1524                           ( i_fbv << 7 )|( i_bfc << 4 )|( i_ffv << 3 )|i_ffc;
1525
1526         /* rtp common header */
1527         rtp_packetize_common( id, out, (i == i_count - 1)?1:0,
1528                               in->i_pts > 0 ? in->i_pts : in->i_dts );
1529
1530         /* 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 */
1531         out->p_buffer[12] = ( h >> 24 )&0xff;
1532         out->p_buffer[13] = ( h >> 16 )&0xff;
1533         out->p_buffer[14] = ( h >>  8 )&0xff;
1534         out->p_buffer[15] = ( h       )&0xff;
1535
1536         memcpy( &out->p_buffer[16], p_data, i_payload );
1537
1538         out->i_buffer   = 16 + i_payload;
1539         out->i_dts    = in->i_dts + i * in->i_length / i_count;
1540         out->i_length = in->i_length / i_count;
1541
1542         rtp_packetize_send( id, out );
1543
1544         p_data += i_payload;
1545         i_data -= i_payload;
1546     }
1547
1548     return VLC_SUCCESS;
1549 }
1550
1551 static int rtp_packetize_ac3( sout_stream_t *p_stream, sout_stream_id_t *id,
1552                               block_t *in )
1553 {
1554     int     i_max   = id->i_mtu - 12 - 2; /* payload max in one packet */
1555     int     i_count = ( in->i_buffer + i_max - 1 ) / i_max;
1556
1557     uint8_t *p_data = in->p_buffer;
1558     int     i_data  = in->i_buffer;
1559     int     i;
1560
1561     for( i = 0; i < i_count; i++ )
1562     {
1563         int           i_payload = __MIN( i_max, i_data );
1564         block_t *out = block_New( p_stream, 14 + i_payload );
1565
1566         /* rtp common header */
1567         rtp_packetize_common( id, out, (i == i_count - 1)?1:0, in->i_pts );
1568         /* unit count */
1569         out->p_buffer[12] = 1;
1570         /* unit header */
1571         out->p_buffer[13] = 0x00;
1572         /* data */
1573         memcpy( &out->p_buffer[14], p_data, i_payload );
1574
1575         out->i_buffer   = 14 + i_payload;
1576         out->i_dts    = in->i_dts + i * in->i_length / i_count;
1577         out->i_length = in->i_length / i_count;
1578
1579         rtp_packetize_send( id, out );
1580
1581         p_data += i_payload;
1582         i_data -= i_payload;
1583     }
1584
1585     return VLC_SUCCESS;
1586 }
1587
1588 static int rtp_packetize_split( sout_stream_t *p_stream, sout_stream_id_t *id,
1589                                 block_t *in )
1590 {
1591     int     i_max   = id->i_mtu - 12; /* payload max in one packet */
1592     int     i_count = ( in->i_buffer + i_max - 1 ) / i_max;
1593
1594     uint8_t *p_data = in->p_buffer;
1595     int     i_data  = in->i_buffer;
1596     int     i;
1597
1598     for( i = 0; i < i_count; i++ )
1599     {
1600         int           i_payload = __MIN( i_max, i_data );
1601         block_t *out = block_New( p_stream, 12 + i_payload );
1602
1603         /* rtp common header */
1604         rtp_packetize_common( id, out, ((i == i_count - 1)?1:0),
1605                               (in->i_pts > 0 ? in->i_pts : in->i_dts) );
1606         memcpy( &out->p_buffer[12], p_data, i_payload );
1607
1608         out->i_buffer   = 12 + i_payload;
1609         out->i_dts    = in->i_dts + i * in->i_length / i_count;
1610         out->i_length = in->i_length / i_count;
1611
1612         rtp_packetize_send( id, out );
1613
1614         p_data += i_payload;
1615         i_data -= i_payload;
1616     }
1617
1618     return VLC_SUCCESS;
1619 }
1620
1621 /* rfc3016 */
1622 static int rtp_packetize_mp4a_latm( sout_stream_t *p_stream, sout_stream_id_t *id,
1623                                 block_t *in )
1624 {
1625     int     i_max   = id->i_mtu - 14;              /* payload max in one packet */
1626     int     latmhdrsize = in->i_buffer / 0xff + 1;
1627     int     i_count = ( in->i_buffer + i_max - 1 ) / i_max;
1628
1629     uint8_t *p_data = in->p_buffer, *p_header = NULL;
1630     int     i_data  = in->i_buffer;
1631     int     i;
1632
1633     for( i = 0; i < i_count; i++ )
1634     {
1635         int     i_payload = __MIN( i_max, i_data );
1636         block_t *out;
1637
1638         if( i != 0 )
1639             latmhdrsize = 0;
1640         out = block_New( p_stream, 12 + latmhdrsize + i_payload );
1641
1642         /* rtp common header */
1643         rtp_packetize_common( id, out, ((i == i_count - 1) ? 1 : 0),
1644                               (in->i_pts > 0 ? in->i_pts : in->i_dts) );
1645
1646         if( i == 0 )
1647         {
1648             int tmp = in->i_buffer;
1649
1650             p_header=out->p_buffer+12;
1651             while( tmp > 0xfe )
1652             {
1653                 *p_header = 0xff;
1654                 p_header++;
1655                 tmp -= 0xff;
1656             }
1657             *p_header = tmp;
1658         }
1659
1660         memcpy( &out->p_buffer[12+latmhdrsize], p_data, i_payload );
1661
1662         out->i_buffer   = 12 + latmhdrsize + i_payload;
1663         out->i_dts    = in->i_dts + i * in->i_length / i_count;
1664         out->i_length = in->i_length / i_count;
1665
1666         rtp_packetize_send( id, out );
1667
1668         p_data += i_payload;
1669         i_data -= i_payload;
1670     }
1671
1672     return VLC_SUCCESS;
1673 }
1674
1675 static int rtp_packetize_l16( sout_stream_t *p_stream, sout_stream_id_t *id,
1676                               block_t *in )
1677 {
1678     int     i_max   = id->i_mtu - 12; /* payload max in one packet */
1679     int     i_count = ( in->i_buffer + i_max - 1 ) / i_max;
1680
1681     uint8_t *p_data = in->p_buffer;
1682     int     i_data  = in->i_buffer;
1683     int     i_packet = 0;
1684
1685     while( i_data > 0 )
1686     {
1687         int           i_payload = (__MIN( i_max, i_data )/4)*4;
1688         block_t *out = block_New( p_stream, 12 + i_payload );
1689
1690         /* rtp common header */
1691         rtp_packetize_common( id, out, 0,
1692                               (in->i_pts > 0 ? in->i_pts : in->i_dts) );
1693         memcpy( &out->p_buffer[12], p_data, i_payload );
1694
1695         out->i_buffer   = 12 + i_payload;
1696         out->i_dts    = in->i_dts + i_packet * in->i_length / i_count;
1697         out->i_length = in->i_length / i_count;
1698
1699         rtp_packetize_send( id, out );
1700
1701         p_data += i_payload;
1702         i_data -= i_payload;
1703         i_packet++;
1704     }
1705
1706     return VLC_SUCCESS;
1707 }
1708
1709 static int rtp_packetize_l8( sout_stream_t *p_stream, sout_stream_id_t *id,
1710                              block_t *in )
1711 {
1712     int     i_max   = id->i_mtu - 12; /* payload max in one packet */
1713     int     i_count = ( in->i_buffer + i_max - 1 ) / i_max;
1714
1715     uint8_t *p_data = in->p_buffer;
1716     int     i_data  = in->i_buffer;
1717     int     i_packet = 0;
1718
1719     while( i_data > 0 )
1720     {
1721         int           i_payload = (__MIN( i_max, i_data )/2)*2;
1722         block_t *out = block_New( p_stream, 12 + i_payload );
1723
1724         /* rtp common header */
1725         rtp_packetize_common( id, out, 0,
1726                               (in->i_pts > 0 ? in->i_pts : in->i_dts) );
1727         memcpy( &out->p_buffer[12], p_data, i_payload );
1728
1729         out->i_buffer   = 12 + i_payload;
1730         out->i_dts    = in->i_dts + i_packet * in->i_length / i_count;
1731         out->i_length = in->i_length / i_count;
1732
1733         rtp_packetize_send( id, out );
1734
1735         p_data += i_payload;
1736         i_data -= i_payload;
1737         i_packet++;
1738     }
1739
1740     return VLC_SUCCESS;
1741 }
1742
1743 static int rtp_packetize_mp4a( sout_stream_t *p_stream, sout_stream_id_t *id,
1744                                block_t *in )
1745 {
1746     int     i_max   = id->i_mtu - 16; /* payload max in one packet */
1747     int     i_count = ( in->i_buffer + i_max - 1 ) / i_max;
1748
1749     uint8_t *p_data = in->p_buffer;
1750     int     i_data  = in->i_buffer;
1751     int     i;
1752
1753     for( i = 0; i < i_count; i++ )
1754     {
1755         int           i_payload = __MIN( i_max, i_data );
1756         block_t *out = block_New( p_stream, 16 + i_payload );
1757
1758         /* rtp common header */
1759         rtp_packetize_common( id, out, ((i == i_count - 1)?1:0),
1760                               (in->i_pts > 0 ? in->i_pts : in->i_dts) );
1761         /* AU headers */
1762         /* AU headers length (bits) */
1763         out->p_buffer[12] = 0;
1764         out->p_buffer[13] = 2*8;
1765         /* for each AU length 13 bits + idx 3bits, */
1766         out->p_buffer[14] = ( in->i_buffer >> 5 )&0xff;
1767         out->p_buffer[15] = ( (in->i_buffer&0xff)<<3 )|0;
1768
1769         memcpy( &out->p_buffer[16], p_data, i_payload );
1770
1771         out->i_buffer   = 16 + i_payload;
1772         out->i_dts    = in->i_dts + i * in->i_length / i_count;
1773         out->i_length = in->i_length / i_count;
1774
1775         rtp_packetize_send( id, out );
1776
1777         p_data += i_payload;
1778         i_data -= i_payload;
1779     }
1780
1781     return VLC_SUCCESS;
1782 }
1783
1784
1785 /* rfc2429 */
1786 #define RTP_H263_HEADER_SIZE (2)  // plen = 0
1787 #define RTP_H263_PAYLOAD_START (14)  // plen = 0
1788 static int rtp_packetize_h263( sout_stream_t *p_stream, sout_stream_id_t *id,
1789                                block_t *in )
1790 {
1791     uint8_t *p_data = in->p_buffer;
1792     int     i_data  = in->i_buffer;
1793     int     i;
1794     int     i_max   = id->i_mtu - 12 - RTP_H263_HEADER_SIZE; /* payload max in one packet */
1795     int     i_count;
1796     int     b_p_bit;
1797     int     b_v_bit = 0; // no pesky error resilience
1798     int     i_plen = 0; // normally plen=0 for PSC packet
1799     int     i_pebit = 0; // because plen=0
1800     uint16_t h;
1801
1802     if( i_data < 2 )
1803     {
1804         return VLC_EGENERIC;
1805     }
1806     if( p_data[0] || p_data[1] )
1807     {
1808         return VLC_EGENERIC;
1809     }
1810     /* remove 2 leading 0 bytes */
1811     p_data += 2;
1812     i_data -= 2;
1813     i_count = ( i_data + i_max - 1 ) / i_max;
1814
1815     for( i = 0; i < i_count; i++ )
1816     {
1817         int      i_payload = __MIN( i_max, i_data );
1818         block_t *out = block_New( p_stream,
1819                                   RTP_H263_PAYLOAD_START + i_payload );
1820         b_p_bit = (i == 0) ? 1 : 0;
1821         h = ( b_p_bit << 10 )|
1822             ( b_v_bit << 9  )|
1823             ( i_plen  << 3  )|
1824               i_pebit;
1825
1826         /* rtp common header */
1827         //b_m_bit = 1; // always contains end of frame
1828         rtp_packetize_common( id, out, (i == i_count - 1)?1:0,
1829                               in->i_pts > 0 ? in->i_pts : in->i_dts );
1830
1831         /* h263 header */
1832         out->p_buffer[12] = ( h >>  8 )&0xff;
1833         out->p_buffer[13] = ( h       )&0xff;
1834         memcpy( &out->p_buffer[RTP_H263_PAYLOAD_START], p_data, i_payload );
1835
1836         out->i_buffer = RTP_H263_PAYLOAD_START + i_payload;
1837         out->i_dts    = in->i_dts + i * in->i_length / i_count;
1838         out->i_length = in->i_length / i_count;
1839
1840         rtp_packetize_send( id, out );
1841
1842         p_data += i_payload;
1843         i_data -= i_payload;
1844     }
1845
1846     return VLC_SUCCESS;
1847 }
1848
1849 /* rfc3984 */
1850 static int
1851 rtp_packetize_h264_nal( sout_stream_t *p_stream, sout_stream_id_t *id,
1852                         const uint8_t *p_data, int i_data, int64_t i_pts,
1853                         int64_t i_dts, vlc_bool_t b_last, int64_t i_length )
1854 {
1855     const int i_max = id->i_mtu - 12; /* payload max in one packet */
1856     int i_nal_hdr;
1857     int i_nal_type;
1858
1859     if( i_data < 5 )
1860         return VLC_SUCCESS;
1861
1862     i_nal_hdr = p_data[3];
1863     i_nal_type = i_nal_hdr&0x1f;
1864
1865     /* Skip start code */
1866     p_data += 3;
1867     i_data -= 3;
1868
1869     /* */
1870     if( i_data <= i_max )
1871     {
1872         /* Single NAL unit packet */
1873         block_t *out = block_New( p_stream, 12 + i_data );
1874         out->i_dts    = i_dts;
1875         out->i_length = i_length;
1876
1877         /* */
1878         rtp_packetize_common( id, out, b_last, i_pts );
1879         out->i_buffer = 12 + i_data;
1880
1881         memcpy( &out->p_buffer[12], p_data, i_data );
1882
1883         rtp_packetize_send( id, out );
1884     }
1885     else
1886     {
1887         /* FU-A Fragmentation Unit without interleaving */
1888         const int i_count = ( i_data-1 + i_max-2 - 1 ) / (i_max-2);
1889         int i;
1890
1891         p_data++;
1892         i_data--;
1893
1894         for( i = 0; i < i_count; i++ )
1895         {
1896             const int i_payload = __MIN( i_data, i_max-2 );
1897             block_t *out = block_New( p_stream, 12 + 2 + i_payload );
1898             out->i_dts    = i_dts + i * i_length / i_count;
1899             out->i_length = i_length / i_count;
1900
1901             /* */
1902             rtp_packetize_common( id, out, (b_last && i_payload == i_data), i_pts );
1903             out->i_buffer = 14 + i_payload;
1904
1905             /* FU indicator */
1906             out->p_buffer[12] = 0x00 | (i_nal_hdr & 0x60) | 28;
1907             /* FU header */
1908             out->p_buffer[13] = ( i == 0 ? 0x80 : 0x00 ) | ( (i == i_count-1) ? 0x40 : 0x00 )  | i_nal_type;
1909             memcpy( &out->p_buffer[14], p_data, i_payload );
1910
1911             rtp_packetize_send( id, out );
1912
1913             i_data -= i_payload;
1914             p_data += i_payload;
1915         }
1916     }
1917     return VLC_SUCCESS;
1918 }
1919
1920 static int rtp_packetize_h264( sout_stream_t *p_stream, sout_stream_id_t *id,
1921                                block_t *in )
1922 {
1923     const uint8_t *p_buffer = in->p_buffer;
1924     int i_buffer = in->i_buffer;
1925
1926     while( i_buffer > 4 && ( p_buffer[0] != 0 || p_buffer[1] != 0 || p_buffer[2] != 1 ) )
1927     {
1928         i_buffer--;
1929         p_buffer++;
1930     }
1931
1932     /* Split nal units */
1933     while( i_buffer > 4 )
1934     {
1935         int i_offset;
1936         int i_size = i_buffer;
1937         int i_skip = i_buffer;
1938
1939         /* search nal end */
1940         for( i_offset = 4; i_offset+2 < i_buffer ; i_offset++)
1941         {
1942             if( p_buffer[i_offset] == 0 && p_buffer[i_offset+1] == 0 && p_buffer[i_offset+2] == 1 )
1943             {
1944                 /* we found another startcode */
1945                 i_size = i_offset - ( p_buffer[i_offset-1] == 0 ? 1 : 0);
1946                 i_skip = i_offset;
1947                 break;
1948             }
1949         }
1950         /* TODO add STAP-A to remove a lot of overhead with small slice/sei/... */
1951         rtp_packetize_h264_nal( p_stream, id, p_buffer, i_size,
1952                                 (in->i_pts > 0 ? in->i_pts : in->i_dts), in->i_dts,
1953                                 (i_size >= i_buffer), in->i_length * i_size / in->i_buffer );
1954
1955         i_buffer -= i_skip;
1956         p_buffer += i_skip;
1957     }
1958     return VLC_SUCCESS;
1959 }
1960
1961 static int rtp_packetize_amr( sout_stream_t *p_stream, sout_stream_id_t *id,
1962                               block_t *in )
1963 {
1964     int     i_max   = id->i_mtu - 14; /* payload max in one packet */
1965     int     i_count = ( in->i_buffer + i_max - 1 ) / i_max;
1966
1967     uint8_t *p_data = in->p_buffer;
1968     int     i_data  = in->i_buffer;
1969     int     i;
1970
1971     /* Only supports octet-aligned mode */
1972     for( i = 0; i < i_count; i++ )
1973     {
1974         int           i_payload = __MIN( i_max, i_data );
1975         block_t *out = block_New( p_stream, 14 + i_payload );
1976
1977         /* rtp common header */
1978         rtp_packetize_common( id, out, ((i == i_count - 1)?1:0),
1979                               (in->i_pts > 0 ? in->i_pts : in->i_dts) );
1980         /* Payload header */
1981         out->p_buffer[12] = 0xF0; /* CMR */
1982         out->p_buffer[13] = p_data[0]&0x7C; /* ToC */ /* FIXME: frame type */
1983
1984         /* FIXME: are we fed multiple frames ? */
1985         memcpy( &out->p_buffer[14], p_data+1, i_payload-1 );
1986
1987         out->i_buffer   = 14 + i_payload-1;
1988         out->i_dts    = in->i_dts + i * in->i_length / i_count;
1989         out->i_length = in->i_length / i_count;
1990
1991         rtp_packetize_send( id, out );
1992
1993         p_data += i_payload;
1994         i_data -= i_payload;
1995     }
1996
1997     return VLC_SUCCESS;
1998 }
1999
2000 static int rtp_packetize_t140( sout_stream_t *p_stream, sout_stream_id_t *id,
2001                                block_t *in )
2002 {
2003     const size_t   i_max  = id->i_mtu - 12;
2004     const uint8_t *p_data = in->p_buffer;
2005     size_t         i_data = in->i_buffer;
2006
2007     for( unsigned i_packet = 0; i_data > 0; i_packet++ )
2008     {
2009         size_t i_payload = i_data;
2010
2011         /* Make sure we stop on an UTF-8 character boundary
2012          * (assuming the input is valid UTF-8) */
2013         if( i_data > i_max )
2014         {
2015             i_payload = i_max;
2016
2017             while( ( p_data[i_payload] & 0xC0 ) == 0x80 )
2018             {
2019                 if( i_payload == 0 )
2020                     return VLC_SUCCESS; /* fishy input! */
2021
2022                 i_payload--;
2023             }
2024         }
2025
2026         block_t *out = block_New( p_stream, 12 + i_payload );
2027         if( out == NULL )
2028             return VLC_SUCCESS;
2029
2030         rtp_packetize_common( id, out, 0, in->i_pts + i_packet );
2031         memcpy( out->p_buffer + 12, p_data, i_payload );
2032
2033         out->i_buffer = 12 + i_payload;
2034         out->i_dts    = out->i_pts;
2035         out->i_length = 0;
2036
2037         rtp_packetize_send( id, out );
2038
2039         p_data += i_payload;
2040         i_data -= i_payload;
2041     }
2042
2043     return VLC_SUCCESS;
2044 }
2045
2046 /*****************************************************************************
2047  * Non-RTP mux
2048  *****************************************************************************/
2049
2050 /** Add an ES to a non-RTP muxed stream */
2051 static sout_stream_id_t *MuxAdd( sout_stream_t *p_stream, es_format_t *p_fmt )
2052 {
2053     sout_input_t      *p_input;
2054     sout_mux_t *p_mux = p_stream->p_sys->p_mux;
2055     assert( p_mux != NULL );
2056
2057     p_input = sout_MuxAddStream( p_mux, p_fmt );
2058     if( p_input == NULL )
2059     {
2060         msg_Err( p_stream, "cannot add this stream to the muxer" );
2061         return NULL;
2062     }
2063
2064     return (sout_stream_id_t *)p_input;
2065 }
2066
2067
2068 static int MuxSend( sout_stream_t *p_stream, sout_stream_id_t *id,
2069                     block_t *p_buffer )
2070 {
2071     sout_mux_t *p_mux = p_stream->p_sys->p_mux;
2072     assert( p_mux != NULL );
2073
2074     sout_MuxSendBuffer( p_mux, (sout_input_t *)id, p_buffer );
2075     return VLC_SUCCESS;
2076 }
2077
2078
2079 /** Remove an ES from a non-RTP muxed stream */
2080 static int MuxDel( sout_stream_t *p_stream, sout_stream_id_t *id )
2081 {
2082     sout_mux_t *p_mux = p_stream->p_sys->p_mux;
2083     assert( p_mux != NULL );
2084
2085     sout_MuxDeleteStream( p_mux, (sout_input_t *)id );
2086     return VLC_SUCCESS;
2087 }
2088
2089
2090 static int AccessOutGrabberWriteBuffer( sout_stream_t *p_stream,
2091                                         const block_t *p_buffer )
2092 {
2093     sout_stream_sys_t *p_sys = p_stream->p_sys;
2094     sout_stream_id_t *id = p_sys->es[0];
2095
2096     int64_t  i_dts = p_buffer->i_dts;
2097
2098     uint8_t         *p_data = p_buffer->p_buffer;
2099     unsigned int    i_data  = p_buffer->i_buffer;
2100     unsigned int    i_max   = id->i_mtu - 12;
2101
2102     unsigned i_packet = ( p_buffer->i_buffer + i_max - 1 ) / i_max;
2103
2104     while( i_data > 0 )
2105     {
2106         unsigned int i_size;
2107
2108         /* output complete packet */
2109         if( p_sys->packet &&
2110             p_sys->packet->i_buffer + i_data > i_max )
2111         {
2112             rtp_packetize_send( id, p_sys->packet );
2113             p_sys->packet = NULL;
2114         }
2115
2116         if( p_sys->packet == NULL )
2117         {
2118             /* allocate a new packet */
2119             p_sys->packet = block_New( p_stream, id->i_mtu );
2120             rtp_packetize_common( id, p_sys->packet, 1, i_dts );
2121             p_sys->packet->i_dts = i_dts;
2122             p_sys->packet->i_length = p_buffer->i_length / i_packet;
2123             i_dts += p_sys->packet->i_length;
2124         }
2125
2126         i_size = __MIN( i_data,
2127                         (unsigned)(id->i_mtu - p_sys->packet->i_buffer) );
2128
2129         memcpy( &p_sys->packet->p_buffer[p_sys->packet->i_buffer],
2130                 p_data, i_size );
2131
2132         p_sys->packet->i_buffer += i_size;
2133         p_data += i_size;
2134         i_data -= i_size;
2135     }
2136
2137     return VLC_SUCCESS;
2138 }
2139
2140
2141 static int AccessOutGrabberWrite( sout_access_out_t *p_access,
2142                                   block_t *p_buffer )
2143 {
2144     sout_stream_t *p_stream = (sout_stream_t*)p_access->p_sys;
2145
2146     while( p_buffer )
2147     {
2148         block_t *p_next;
2149
2150         AccessOutGrabberWriteBuffer( p_stream, p_buffer );
2151
2152         p_next = p_buffer->p_next;
2153         block_Release( p_buffer );
2154         p_buffer = p_next;
2155     }
2156
2157     return VLC_SUCCESS;
2158 }
2159
2160
2161 static sout_access_out_t *GrabberCreate( sout_stream_t *p_stream )
2162 {
2163     sout_access_out_t *p_grab;
2164
2165     p_grab = vlc_object_create( p_stream->p_sout, sizeof( *p_grab ) );
2166     if( p_grab == NULL )
2167         return NULL;
2168
2169     p_grab->p_module    = NULL;
2170     p_grab->p_sout      = p_stream->p_sout;
2171     p_grab->psz_access  = strdup( "grab" );
2172     p_grab->p_cfg       = NULL;
2173     p_grab->psz_path    = strdup( "" );
2174     p_grab->p_sys       = (sout_access_out_sys_t *)p_stream;
2175     p_grab->pf_seek     = NULL;
2176     p_grab->pf_write    = AccessOutGrabberWrite;
2177     return p_grab;
2178 }