1 /*****************************************************************************
2 * udp.c: raw UDP & RTP input module
3 *****************************************************************************
4 * Copyright (C) 2001-2004 VideoLAN
5 * $Id: udp.c,v 1.29 2004/01/31 18:02:32 alexis Exp $
7 * Authors: Christophe Massiot <massiot@via.ecp.fr>
8 * Tristan Leteurtre <tooney@via.ecp.fr>
9 * Laurent Aimar <fenrir@via.ecp.fr>
11 * Reviewed: 23 October 2003, Jean-Paul Saman <jpsaman@wxs.nl>
13 * This program is free software; you can redistribute it and/or modify
14 * it under the terms of the GNU General Public License as published by
15 * the Free Software Foundation; either version 2 of the License, or
16 * (at your option) any later version.
18 * This program is distributed in the hope that it will be useful,
19 * but WITHOUT ANY WARRANTY; without even the implied warranty of
20 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
21 * GNU General Public License for more details.
23 * You should have received a copy of the GNU General Public License
24 * along with this program; if not, write to the Free Software
25 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111, USA.
26 *****************************************************************************/
28 /*****************************************************************************
30 *****************************************************************************/
34 #include <vlc/input.h>
38 /*****************************************************************************
40 *****************************************************************************/
41 #define CACHING_TEXT N_("Caching value in ms")
42 #define CACHING_LONGTEXT N_( \
43 "Allows you to modify the default caching value for udp streams. This " \
44 "value should be set in miliseconds units." )
46 static int Open ( vlc_object_t * );
47 static void Close( vlc_object_t * );
50 set_description( _("UDP/RTP input") );
52 add_integer( "udp-caching", DEFAULT_PTS_DELAY / 1000, NULL, CACHING_TEXT,
53 CACHING_LONGTEXT, VLC_TRUE );
55 set_capability( "access", 0 );
56 add_shortcut( "udp" );
57 add_shortcut( "udpstream" );
58 add_shortcut( "udp4" );
59 add_shortcut( "udp6" );
60 add_shortcut( "rtp" );
61 add_shortcut( "rtp4" );
62 add_shortcut( "rtp6" );
63 set_callbacks( Open, Close );
66 /*****************************************************************************
68 *****************************************************************************/
69 #define RTP_HEADER_LEN 12
71 static ssize_t Read ( input_thread_t *, byte_t *, size_t );
72 static ssize_t RTPRead ( input_thread_t *, byte_t *, size_t );
73 static ssize_t RTPChoose( input_thread_t *, byte_t *, size_t );
80 /*****************************************************************************
81 * Open: open the socket
82 *****************************************************************************/
83 static int Open( vlc_object_t *p_this )
85 input_thread_t *p_input = (input_thread_t *)p_this;
88 char * psz_name = strdup(p_input->psz_name);
89 char * psz_parser = psz_name;
90 char * psz_server_addr = "";
91 char * psz_server_port = "";
92 char * psz_bind_addr = "";
93 char * psz_bind_port = "";
94 int i_bind_port = 0, i_server_port = 0;
98 /* First set ipv4/ipv6 */
99 var_Create( p_input, "ipv4", VLC_VAR_BOOL | VLC_VAR_DOINHERIT );
100 var_Create( p_input, "ipv6", VLC_VAR_BOOL | VLC_VAR_DOINHERIT );
102 if( *p_input->psz_access )
104 /* Find out which shortcut was used */
105 if( !strncmp( p_input->psz_access, "udp4", 6 ) ||
106 !strncmp( p_input->psz_access, "rtp4", 6 ))
108 val.b_bool = VLC_TRUE;
109 var_Set( p_input, "ipv4", val );
111 val.b_bool = VLC_FALSE;
112 var_Set( p_input, "ipv6", val );
114 else if( !strncmp( p_input->psz_access, "udp6", 6 ) ||
115 !strncmp( p_input->psz_access, "rtp6", 6 ) )
117 val.b_bool = VLC_TRUE;
118 var_Set( p_input, "ipv6", val );
120 val.b_bool = VLC_FALSE;
121 var_Set( p_input, "ipv4", val );
125 /* Parse psz_name syntax :
126 * [serveraddr[:serverport]][@[bindaddr]:[bindport]] */
127 if( *psz_parser && *psz_parser != '@' )
130 psz_server_addr = psz_parser;
132 while( *psz_parser && *psz_parser != ':' && *psz_parser != '@' )
134 if( *psz_parser == '[' )
137 while( *psz_parser && *psz_parser != ']' )
145 if( *psz_parser == ':' )
147 /* Found server port */
148 *psz_parser++ = '\0'; /* Terminate server name */
149 psz_server_port = psz_parser;
151 while( *psz_parser && *psz_parser != '@' )
158 if( *psz_parser == '@' )
160 /* Found bind address or bind port */
161 *psz_parser++ = '\0'; /* Terminate server port or name if necessary */
163 if( *psz_parser && *psz_parser != ':' )
165 /* Found bind address */
166 psz_bind_addr = psz_parser;
168 while( *psz_parser && *psz_parser != ':' )
170 if( *psz_parser == '[' )
173 while( *psz_parser && *psz_parser != ']' )
182 if( *psz_parser == ':' )
184 /* Found bind port */
185 *psz_parser++ = '\0'; /* Terminate bind address if necessary */
186 psz_bind_port = psz_parser;
190 i_server_port = strtol( psz_server_port, NULL, 10 );
191 if( ( i_bind_port = strtol( psz_bind_port, NULL, 10 ) ) == 0 )
193 i_bind_port = config_GetInt( p_this, "server-port" );
196 msg_Dbg( p_input, "opening server=%s:%d local=%s:%d",
197 psz_server_addr, i_server_port, psz_bind_addr, i_bind_port );
199 p_sys = p_input->p_access_data = malloc( sizeof( access_sys_t ) );
202 msg_Err( p_input, "out of memory" );
206 p_sys->fd = net_OpenUDP( p_input, psz_bind_addr, i_bind_port,
207 psz_server_addr, i_server_port );
210 msg_Err( p_input, "cannot open socket" );
218 p_input->i_mtu = config_GetInt( p_this, "mtu" );
220 /* fill p_input fields */
221 p_input->pf_read = RTPChoose;
222 p_input->pf_set_program = input_SetProgram;
223 p_input->pf_set_area = NULL;
224 p_input->pf_seek = NULL;
226 vlc_mutex_lock( &p_input->stream.stream_lock );
227 p_input->stream.b_pace_control = VLC_FALSE;
228 p_input->stream.b_seekable = VLC_FALSE;
229 p_input->stream.p_selected_area->i_tell = 0;
230 p_input->stream.i_method = INPUT_METHOD_NETWORK;
231 vlc_mutex_unlock( &p_input->stream.stream_lock );
233 /* Update default_pts to a suitable value for udp access */
234 var_Create( p_input, "udp-caching", VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
235 var_Get( p_input, "udp-caching", &val );
236 p_input->i_pts_delay = val.i_int * 1000;
241 /*****************************************************************************
242 * Close: free unused data structures
243 *****************************************************************************/
244 static void Close( vlc_object_t *p_this )
246 input_thread_t *p_input = (input_thread_t *)p_this;
247 access_sys_t *p_sys = p_input->p_access_data;
249 msg_Info( p_input, "closing UDP target `%s'", p_input->psz_source );
251 net_Close( p_sys->fd );
256 /*****************************************************************************
257 * Read: read on a file descriptor, checking b_die periodically
258 *****************************************************************************/
259 static ssize_t Read( input_thread_t * p_input, byte_t * p_buffer, size_t i_len )
261 access_sys_t *p_sys = p_input->p_access_data;
263 return net_Read( p_input, p_sys->fd, p_buffer, i_len, VLC_FALSE );
266 /*****************************************************************************
267 * RTPRead : read from the network, and parse the RTP header
268 *****************************************************************************/
269 static ssize_t RTPRead( input_thread_t * p_input, byte_t * p_buffer,
277 byte_t * p_tmp_buffer = alloca( p_input->i_mtu );
279 /* Get the raw data from the socket.
280 * We first assume that RTP header size is the classic RTP_HEADER_LEN. */
281 ssize_t i_ret = Read( p_input, p_tmp_buffer, p_input->i_mtu );
283 if ( i_ret <= 0 ) return 0; /* i_ret is at least 1 */
285 /* Parse the header and make some verifications.
286 * See RFC 1889 & RFC 2250. */
288 i_rtp_version = ( p_tmp_buffer[0] & 0xC0 ) >> 6;
289 i_CSRC_count = ( p_tmp_buffer[0] & 0x0F );
290 i_payload_type = ( p_tmp_buffer[1] & 0x7F );
292 if ( i_rtp_version != 2 )
293 msg_Dbg( p_input, "RTP version is %u, should be 2", i_rtp_version );
295 if( i_payload_type == 14 )
299 else if( i_payload_type != 33 && i_payload_type != 32 )
301 msg_Dbg( p_input, "unsupported RTP payload type (%u)",
304 i_skip += RTP_HEADER_LEN + 4*i_CSRC_count;
306 /* A CSRC extension field is 32 bits in size (4 bytes) */
307 if ( i_ret < i_skip )
309 /* Packet is not big enough to hold the complete RTP_HEADER with
312 msg_Warn( p_input, "RTP input trashing %d bytes", i_ret - i_len );
316 /* Return the packet without the RTP header. */
319 if ( (size_t)i_ret > i_len )
321 /* This should NOT happen. */
322 msg_Warn( p_input, "RTP input trashing %d bytes", i_ret - i_len );
326 p_input->p_vlc->pf_memcpy( p_buffer, &p_tmp_buffer[i_skip], i_ret );
331 /*****************************************************************************
332 * RTPChoose : read from the network, and decide whether it's UDP or RTP
333 *****************************************************************************/
334 static ssize_t RTPChoose( input_thread_t * p_input, byte_t * p_buffer,
341 byte_t * p_tmp_buffer = alloca( p_input->i_mtu );
343 /* Get the raw data from the socket.
344 * We first assume that RTP header size is the classic RTP_HEADER_LEN. */
345 ssize_t i_ret = Read( p_input, p_tmp_buffer, p_input->i_mtu );
347 if ( i_ret <= 0 ) return 0; /* i_ret is at least 1 */
349 /* Check that it's not TS. */
350 if ( p_tmp_buffer[0] == 0x47 )
352 msg_Dbg( p_input, "detected TS over raw UDP" );
353 p_input->pf_read = Read;
354 p_input->p_vlc->pf_memcpy( p_buffer, p_tmp_buffer, i_ret );
358 /* Parse the header and make some verifications.
359 * See RFC 1889 & RFC 2250. */
361 i_rtp_version = ( p_tmp_buffer[0] & 0xC0 ) >> 6;
362 i_CSRC_count = ( p_tmp_buffer[0] & 0x0F );
363 i_payload_type = ( p_tmp_buffer[1] & 0x7F );
365 if ( i_rtp_version != 2 )
367 msg_Dbg( p_input, "no supported RTP header detected" );
368 p_input->pf_read = Read;
369 p_input->p_vlc->pf_memcpy( p_buffer, p_tmp_buffer, i_ret );
373 switch ( i_payload_type )
376 msg_Dbg( p_input, "detected TS over RTP" );
380 msg_Dbg( p_input, "detected MPEG audio over RTP" );
381 if( !p_input->psz_demux || *p_input->psz_demux == '\0' )
383 p_input->psz_demux = "mp3";
388 msg_Dbg( p_input, "detected MPEG video over RTP" );
392 msg_Dbg( p_input, "no RTP header detected" );
393 p_input->pf_read = Read;
394 p_input->p_vlc->pf_memcpy( p_buffer, p_tmp_buffer, i_ret );
398 p_input->pf_read = RTPRead;
400 /* A CSRC extension field is 32 bits in size (4 bytes) */
401 if( i_ret < RTP_HEADER_LEN + 4*i_CSRC_count )
403 /* Packet is not big enough to hold the complete RTP_HEADER with
406 msg_Warn( p_input, "RTP input trashing %d bytes", i_ret - i_len );
410 /* Return the packet without the RTP header. */
411 i_ret -= RTP_HEADER_LEN + 4*i_CSRC_count;
413 if ( (size_t)i_ret > i_len )
415 /* This should NOT happen. */
416 msg_Warn( p_input, "RTP input trashing %d bytes", i_ret - i_len );
420 p_input->p_vlc->pf_memcpy( p_buffer,
421 &p_tmp_buffer[RTP_HEADER_LEN + 4*i_CSRC_count],