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