1 /*****************************************************************************
2 * udp.c: raw UDP input module
3 *****************************************************************************
4 * Copyright (C) 2001-2005 VLC authors and VideoLAN
5 * Copyright (C) 2007 Remi Denis-Courmont
8 * Authors: Christophe Massiot <massiot@via.ecp.fr>
9 * Tristan Leteurtre <tooney@via.ecp.fr>
10 * Laurent Aimar <fenrir@via.ecp.fr>
11 * Jean-Paul Saman <jpsaman #_at_# m2x dot nl>
14 * Reviewed: 23 October 2003, Jean-Paul Saman <jpsaman _at_ videolan _dot_ org>
16 * This program is free software; you can redistribute it and/or modify it
17 * under the terms of the GNU Lesser General Public License as published by
18 * the Free Software Foundation; either version 2.1 of the License, or
19 * (at your option) any later version.
21 * This program is distributed in the hope that it will be useful,
22 * but WITHOUT ANY WARRANTY; without even the implied warranty of
23 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
24 * GNU Lesser General Public License for more details.
26 * You should have received a copy of the GNU Lesser General Public License
27 * along with this program; if not, write to the Free Software Foundation,
28 * Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
29 *****************************************************************************/
31 /*****************************************************************************
33 *****************************************************************************/
40 #include <vlc_common.h>
41 #include <vlc_plugin.h>
42 #include <vlc_access.h>
43 #include <vlc_network.h>
44 #include <vlc_block.h>
48 /*****************************************************************************
50 *****************************************************************************/
51 static int Open( vlc_object_t * );
52 static void Close( vlc_object_t * );
54 #define BUFFER_TEXT N_("Receive buffer")
55 #define BUFFER_LONGTEXT N_("UDP receive buffer size (bytes)" )
58 set_shortname( N_("UDP" ) )
59 set_description( N_("UDP input") )
60 set_category( CAT_INPUT )
61 set_subcategory( SUBCAT_INPUT_ACCESS )
63 add_obsolete_integer( "server-port" ) /* since 2.0.0 */
64 add_integer( "udp-buffer", 0x400000, BUFFER_TEXT, BUFFER_LONGTEXT, true )
66 set_capability( "access", 0 )
67 add_shortcut( "udp", "udpstream", "udp4", "udp6" )
69 set_callbacks( Open, Close )
80 /*****************************************************************************
82 *****************************************************************************/
83 static block_t *BlockUDP( access_t * );
84 static int Control( access_t *, int, va_list );
85 static void* ThreadRead( void *data );
87 /*****************************************************************************
88 * Open: open the socket
89 *****************************************************************************/
90 static int Open( vlc_object_t *p_this )
92 access_t *p_access = (access_t*)p_this;
94 char *psz_name = strdup( p_access->psz_location );
96 const char *psz_server_addr, *psz_bind_addr = "";
97 int i_bind_port = 1234, i_server_port = 0;
99 access_sys_t *sys = malloc( sizeof( *sys ) );
100 if( unlikely( sys == NULL ) )
103 p_access->p_sys = sys;
105 /* Set up p_access */
106 access_InitFields( p_access );
107 ACCESS_SET_CALLBACKS( NULL, BlockUDP, Control, NULL );
109 /* Parse psz_name syntax :
110 * [serveraddr[:serverport]][@[bindaddr]:[bindport]] */
111 psz_parser = strchr( psz_name, '@' );
112 if( psz_parser != NULL )
114 /* Found bind address and/or bind port */
115 *psz_parser++ = '\0';
116 psz_bind_addr = psz_parser;
118 if( psz_bind_addr[0] == '[' )
119 /* skips bracket'd IPv6 address */
120 psz_parser = strchr( psz_parser, ']' );
122 if( psz_parser != NULL )
124 psz_parser = strchr( psz_parser, ':' );
125 if( psz_parser != NULL )
127 *psz_parser++ = '\0';
128 i_bind_port = atoi( psz_parser );
133 psz_server_addr = psz_name;
134 psz_parser = ( psz_server_addr[0] == '[' )
135 ? strchr( psz_name, ']' ) /* skips bracket'd IPv6 address */
138 if( psz_parser != NULL )
140 psz_parser = strchr( psz_parser, ':' );
141 if( psz_parser != NULL )
143 *psz_parser++ = '\0';
144 i_server_port = atoi( psz_parser );
148 msg_Dbg( p_access, "opening server=%s:%d local=%s:%d",
149 psz_server_addr, i_server_port, psz_bind_addr, i_bind_port );
151 sys->fd = net_OpenDgram( p_access, psz_bind_addr, i_bind_port,
152 psz_server_addr, i_server_port, IPPROTO_UDP );
156 msg_Err( p_access, "cannot open socket" );
160 sys->fifo = block_FifoNew();
161 if( unlikely( sys->fifo == NULL ) )
163 net_Close( sys->fd );
167 sys->fifo_size = var_InheritInteger( p_access, "udp-buffer");
169 if( vlc_clone( &sys->thread, ThreadRead, p_access,
170 VLC_THREAD_PRIORITY_INPUT ) )
172 block_FifoRelease( sys->fifo );
173 net_Close( sys->fd );
182 /*****************************************************************************
183 * Close: free unused data structures
184 *****************************************************************************/
185 static void Close( vlc_object_t *p_this )
187 access_t *p_access = (access_t*)p_this;
188 access_sys_t *sys = p_access->p_sys;
190 vlc_cancel( sys->thread );
191 vlc_join( sys->thread, NULL );
192 block_FifoRelease( sys->fifo );
193 net_Close( sys->fd );
197 /*****************************************************************************
199 *****************************************************************************/
200 static int Control( access_t *p_access, int i_query, va_list args )
207 case ACCESS_CAN_SEEK:
208 case ACCESS_CAN_FASTSEEK:
209 case ACCESS_CAN_PAUSE:
210 case ACCESS_CAN_CONTROL_PACE:
211 pb_bool = (bool*)va_arg( args, bool* );
215 case ACCESS_GET_PTS_DELAY:
216 pi_64 = (int64_t*)va_arg( args, int64_t * );
217 *pi_64 = INT64_C(1000)
218 * var_InheritInteger(p_access, "network-caching");
227 /*****************************************************************************
229 *****************************************************************************/
230 static block_t *BlockUDP( access_t *p_access )
232 access_sys_t *sys = p_access->p_sys;
234 return block_FifoGet( sys->fifo );
237 /*****************************************************************************
238 * ThreadRead: Pull packets from socket as soon as possible.
239 *****************************************************************************/
240 static void* ThreadRead( void *data )
242 access_t *access = data;
243 access_sys_t *sys = access->p_sys;
250 block_FifoPace( sys->fifo, SIZE_MAX, sys->fifo_size );
252 pkt = block_Alloc( MTU );
253 if( unlikely( pkt == NULL ) )
256 block_cleanup_push( pkt );
257 len = net_Read( access, sys->fd, NULL, pkt->p_buffer, MTU, false );
262 block_Release( pkt );
269 pkt = block_Realloc( pkt, 0, len );
270 block_FifoPut( sys->fifo, pkt );
273 block_FifoWake( sys->fifo );