1 /*****************************************************************************
3 *****************************************************************************
4 * Copyright (C) 2001-2003 VideoLAN
5 * $Id: ftp.c,v 1.20 2003/07/31 23:44:49 fenrir Exp $
7 * Authors: Laurent Aimar <fenrir@via.ecp.fr>
9 * This program is free software; you can redistribute it and/or modify
10 * it under the terms of the GNU General Public License as published by
11 * the Free Software Foundation; either version 2 of the License, or
12 * (at your option) any later version.
14 * This program is distributed in the hope that it will be useful,
15 * but WITHOUT ANY WARRANTY; without even the implied warranty of
16 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17 * GNU General Public License for more details.
19 * You should have received a copy of the GNU General Public License
20 * along with this program; if not, write to the Free Software
21 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111, USA.
22 *****************************************************************************/
24 /*****************************************************************************
26 *****************************************************************************/
28 #include <sys/types.h>
36 #include <vlc/input.h>
38 #ifdef HAVE_SYS_TIME_H
39 # include <sys/time.h>
47 # include <winsock2.h>
48 # include <ws2tcpip.h>
50 # define IN_MULTICAST(a) IN_CLASSD(a)
53 # include <sys/socket.h>
54 # include <netinet/in.h>
56 # include <arpa/inet.h>
57 # elif defined( SYS_BEOS )
58 # include <net/netdb.h>
64 /*****************************************************************************
66 *****************************************************************************/
67 static int Open ( vlc_object_t * );
68 static void Close ( vlc_object_t * );
70 static ssize_t Read ( input_thread_t * p_input, byte_t * p_buffer,
72 static void Seek ( input_thread_t *, off_t );
75 static ssize_t NetRead ( input_thread_t *, input_socket_t *, byte_t *, size_t );
76 static void NetClose( input_thread_t *, input_socket_t *);
78 static int ftp_SendCommand( input_thread_t *, char *, ... );
79 static int ftp_ReadCommand( input_thread_t *, int *, char ** );
80 static int ftp_StartStream( input_thread_t *, off_t );
81 static int ftp_StopStream ( input_thread_t *);
83 /*****************************************************************************
85 *****************************************************************************/
86 #define CACHING_TEXT N_("Caching value in ms")
87 #define CACHING_LONGTEXT N_( \
88 "Allows you to modify the default caching value for ftp streams. This " \
89 "value should be set in miliseconds units." )
92 set_description( _("FTP input") );
93 set_capability( "access", 0 );
94 add_category_hint( "stream", NULL, VLC_FALSE );
95 add_integer( "ftp-caching", 2 * DEFAULT_PTS_DELAY / 1000, NULL,
96 CACHING_TEXT, CACHING_LONGTEXT, VLC_TRUE );
97 add_string( "ftp-user", "anonymous", NULL, "ftp user name", "ftp user name", VLC_FALSE );
98 add_string( "ftp-pwd", "anonymous@dummy.org", NULL, "ftp password", "ftp password, be careful with that option...", VLC_FALSE );
99 add_string( "ftp-account", "anonymous", NULL, "ftp account", "ftp account", VLC_FALSE );
100 add_shortcut( "ftp" );
101 set_callbacks( Open, Close );
104 /* url: [/]host[:port][/path] */
107 char *psz_server_addr;
119 static void ftp_ParseURL( url_t *, char * );
121 #define FREE( p ) if( p ) free( p )
123 typedef struct access_s
125 input_socket_t socket_cmd;
126 input_socket_t socket_data;
128 url_t url; /* connect to this server */
137 /****************************************************************************
138 ****************************************************************************
139 ******************* *******************
140 ******************* Main functions *******************
141 ******************* *******************
142 ****************************************************************************
143 ****************************************************************************/
145 /****************************************************************************
146 * Open: connect to ftp server and ask for file
147 ****************************************************************************/
148 static int Open( vlc_object_t *p_this )
150 input_thread_t *p_input = (input_thread_t*)p_this;
156 network_socket_t socket_desc;
160 char *psz_user, *psz_pwd, *psz_account;
164 /* *** allocate p_access_data *** */
165 p_input->p_access_data =
166 (void*)p_access = malloc( sizeof( access_t ) );
167 memset( p_access, 0, sizeof( access_t ) );
168 p_url = &p_access->url;
170 /* *** Parse URL and get server addr/port and path *** */
171 ftp_ParseURL( p_url, p_input->psz_name );
173 if( p_url->psz_server_addr == NULL ||
174 !( *p_url->psz_server_addr ) )
176 FREE( p_url->psz_private );
177 msg_Err( p_input, "invalid server name" );
180 if( p_url->i_server_port == 0 )
182 p_url->i_server_port = 21; /* default port */
185 /* 2: look at ip version ipv4/ipv6 */
187 if( config_GetInt( p_input, "ipv4" ) )
189 psz_network = "ipv4";
191 else if( config_GetInt( p_input, "ipv6" ) )
193 psz_network = "ipv6";
196 /* 3: Open a TCP connection with server *** */
197 msg_Dbg( p_input, "waiting for connection..." );
198 socket_desc.i_type = NETWORK_TCP;
199 socket_desc.psz_server_addr = p_url->psz_server_addr;
200 socket_desc.i_server_port = p_url->i_server_port;
201 socket_desc.psz_bind_addr = "";
202 socket_desc.i_bind_port = 0;
203 socket_desc.i_ttl = 0;
204 p_input->p_private = (void*)&socket_desc;
205 if( !( p_network = module_Need( p_input, "network", psz_network ) ) )
207 msg_Err( p_input, "failed to connect with server" );
208 FREE( p_access->url.psz_private );
209 FREE( p_input->p_access_data );
212 module_Unneed( p_input, p_network );
213 p_access->socket_cmd.i_handle = socket_desc.i_handle;
214 p_input->i_mtu = socket_desc.i_mtu;
216 "connection with \"%s:%d\" successful",
217 p_url->psz_server_addr,
218 p_url->i_server_port );
223 if( ftp_ReadCommand( p_input, &i_answer, NULL ) < 0)
225 msg_Err( p_input, "failed to get answer" );
228 if( i_answer / 100 != 1 )
234 if( i_answer / 100 != 2 )
236 msg_Err( p_input, "connection rejected" );
241 msg_Dbg( p_input, "connection accepted (%d)", i_answer );
244 psz_user = config_GetPsz( p_input, "ftp-user" );
245 if( ftp_SendCommand( p_input, "USER %s", psz_user ) < 0 )
252 if( ftp_ReadCommand( p_input, &i_answer, NULL ) < 0)
254 msg_Err( p_input, "failed to get answer" );
257 switch( i_answer / 100 )
260 msg_Dbg( p_input, "user accepted" );
263 msg_Dbg( p_input, "password needed" );
264 psz_pwd = config_GetPsz( p_input, "ftp-pwd" );
265 if( ftp_SendCommand( p_input, "PASS %s", psz_pwd ) < 0 )
271 if( ftp_ReadCommand( p_input, &i_answer, NULL ) < 0)
273 msg_Err( p_input, "failed to get answer" );
276 switch( i_answer / 100 )
279 msg_Dbg( p_input, "password accepted" );
282 msg_Dbg( p_input, "account needed" );
283 psz_account = config_GetPsz( p_input, "ftp-account" );
284 if( ftp_SendCommand( p_input, "ACCT %s", psz_account ) < 0 )
290 if( ftp_ReadCommand( p_input, &i_answer, NULL ) < 0)
292 msg_Err( p_input, "failed to get answer" );
295 if( i_answer / 100 != 2 )
297 msg_Err( p_input, "account rejected" );
302 msg_Dbg( p_input, "account accepted" );
306 msg_Err( p_input, "password rejected" );
311 msg_Err( p_input, "user rejected" );
315 if( ftp_SendCommand( p_input, "TYPE I" ) < 0 )
317 msg_Err( p_input, "cannot set binary transfert mode" );
320 if( ftp_ReadCommand( p_input, &i_answer, NULL ) != 2 )
322 msg_Err( p_input, "cannot set binary transfert mode" );
327 if( ftp_SendCommand( p_input, "SIZE %s", p_url->psz_path ) < 0 )
329 msg_Err( p_input, "cannot get file size" );
332 if( ftp_ReadCommand( p_input, &i_answer, &psz_arg ) != 2 )
334 msg_Err( p_input, "cannot get file size" );
339 p_access->i_filesize = atoll( psz_arg + 4 );
343 char *psz_parser = psz_arg + 4;
346 while( *psz_parser == ' ' || *psz_parser == '\t' ) psz_parser++;
348 if( *psz_parser == '-' ) sign = -1;
349 while( *psz_parser >= '0' && *psz_parser <= '9' )
351 i_size = i_size * 10 + *psz_parser++ - '0';
353 p_access->i_filesize = i_size * sign;
357 msg_Dbg( p_input, "file size: "I64Fd, p_access->i_filesize );
360 if( ftp_StartStream( p_input, 0 ) < 0 )
362 msg_Err( p_input, "cannot retrieve file" );
365 /* *** set exported functions *** */
366 p_input->pf_read = Read;
367 p_input->pf_seek = Seek;
368 p_input->pf_set_program = input_SetProgram;
369 p_input->pf_set_area = NULL;
371 p_input->p_private = NULL;
373 /* *** finished to set some variable *** */
374 vlc_mutex_lock( &p_input->stream.stream_lock );
375 p_input->stream.b_pace_control = 1;
376 p_input->stream.p_selected_area->i_tell = 0;
377 p_input->stream.b_seekable = 1;
378 p_input->stream.p_selected_area->i_size = p_access->i_filesize;
379 p_input->stream.i_method = INPUT_METHOD_NETWORK;
380 vlc_mutex_unlock( &p_input->stream.stream_lock );
382 /* Update default_pts to a suitable value for ftp access */
383 p_input->i_pts_delay = config_GetInt( p_input, "ftp-caching" ) * 1000;
388 NetClose( p_input, &p_access->socket_cmd );
389 FREE( p_access->url.psz_private );
390 FREE( p_input->p_access_data );
394 /*****************************************************************************
395 * Close: free unused data structures
396 *****************************************************************************/
397 static void Close( vlc_object_t *p_this )
399 input_thread_t *p_input = (input_thread_t *)p_this;
400 access_t *p_access = (access_t*)p_input->p_access_data;
402 msg_Dbg( p_input, "stopping stream" );
403 ftp_StopStream( p_input );
405 if( ftp_SendCommand( p_input, "QUIT" ) < 0 )
407 msg_Err( p_input, "cannot quit" );
411 ftp_ReadCommand( p_input, NULL, NULL );
415 NetClose( p_input, &p_access->socket_cmd );
418 FREE( p_access->url.psz_private );
421 /*****************************************************************************
422 * Seek: try to go at the right place
423 *****************************************************************************/
424 static void Seek( input_thread_t * p_input, off_t i_pos )
426 //access_t *p_access = (access_t*)p_input->p_access_data;
431 vlc_mutex_lock( &p_input->stream.stream_lock );
433 msg_Dbg( p_input, "seeking to "I64Fd, i_pos );
435 ftp_StopStream( p_input );
436 ftp_StartStream( p_input, i_pos );
438 p_input->stream.p_selected_area->i_tell = i_pos;
439 vlc_mutex_unlock( &p_input->stream.stream_lock );
442 static ssize_t Read ( input_thread_t * p_input, byte_t * p_buffer,
445 access_t *p_access = (access_t*)p_input->p_access_data;
448 i_data = NetRead( p_input, &p_access->socket_data, p_buffer, i_len );
453 static int ftp_SendCommand( input_thread_t *p_input, char *psz_fmt, ... )
455 access_t *p_access = (access_t*)p_input->p_access_data;
458 #if !defined(HAVE_VASPRINTF) || defined(SYS_DARWIN) || defined(SYS_BEOS)
462 va_start( args, psz_fmt );
464 #if defined(HAVE_VASPRINTF) && !defined(SYS_DARWIN) && !defined(SYS_BEOS)
465 vasprintf( &psz_buffer, psz_fmt, args );
467 i_size = strlen( psz_fmt ) + 2048;
468 psz_buffer = calloc( i_size, sizeof( char ) );
469 vsnprintf( psz_buffer, i_size, psz_fmt, args );
470 psz_buffer[i_size - 1] = 0;
472 if( !strncmp( psz_buffer, "PASS", 4 ) )
474 msg_Dbg( p_input, "ftp_SendCommand:\"PASS xxx\"" );
478 msg_Dbg( p_input, "ftp_SendCommand:\"%s\"", psz_buffer );
480 psz_buffer = realloc( psz_buffer, strlen( psz_buffer ) + 3 );
481 strcat( psz_buffer, "\r\n" );
482 if( send( p_access->socket_cmd.i_handle,
484 strlen( psz_buffer ),
488 msg_Err( p_input, "failed to send command" );
498 #define BLOCK_SIZE 1024
499 /* TODO support this s**t :
500 RFC 959 allows the client to send certain TELNET strings at any moment,
501 even in the middle of a request:
504 * \377\376x where x is one byte.
505 * \377\375x where x is one byte. The server is obliged to send \377\374x
506 * immediately after reading x.
507 * \377\374x where x is one byte.
508 * \377\373x where x is one byte. The server is obliged to send \377\376x
509 * immediately after reading x.
510 * \377x for any other byte x.
512 These strings are not part of the requests, except in the case \377\377,
513 where the request contains one \377. */
515 static int ftp_ReadCommand( input_thread_t *p_input,
516 int *pi_answer, char **ppsz_answer )
518 access_t *p_access = (access_t*)p_input->p_access_data;
526 i_buffer_size = BLOCK_SIZE + 1;
527 p_buffer = malloc( BLOCK_SIZE + 1);
532 i_read = NetRead( p_input, &p_access->socket_cmd,
533 p_buffer + i_buffer, BLOCK_SIZE );
534 if( i_read <= 0 || p_input->b_die || p_input->b_error )
537 if( pi_answer ) *pi_answer = 500;
538 if( ppsz_answer ) *ppsz_answer = NULL;
546 if( i_read < BLOCK_SIZE )
548 p_buffer[i_buffer] = '\0';
551 i_buffer_size += BLOCK_SIZE;
552 p_buffer = realloc( p_buffer, i_buffer_size );
560 i_answer = atoi( p_buffer );
562 if( pi_answer ) *pi_answer = i_answer;
565 *ppsz_answer = p_buffer;
571 return( i_answer / 100 );
575 if( pi_answer ) *pi_answer = 500;
576 if( ppsz_answer ) *ppsz_answer = NULL;
580 static int ftp_StartStream( input_thread_t *p_input, off_t i_start )
582 access_t *p_access = (access_t*)p_input->p_access_data;
586 char *psz_arg, *psz_parser;
591 network_socket_t socket_desc;
593 if( ftp_SendCommand( p_input, "PASV" ) < 0 )
595 msg_Err( p_input, "cannot set passive transfert mode" );
598 if( ftp_ReadCommand( p_input, &i_answer, &psz_arg ) != 2 )
600 msg_Err( p_input, "cannot set passive transfert mode" );
603 psz_parser = strchr( psz_arg, '(' );
604 if( !psz_parser || sscanf( psz_parser, "(%d,%d,%d,%d,%d,%d", &a1, &a2, &a3, &a4, &p1, &p2 ) < 6 )
607 msg_Err( p_input, "cannot get ip/port for passive transfert mode" );
612 sprintf( psz_ip, "%d.%d.%d.%d", a1, a2, a3, a4 );
613 i_port = p1 * 256 + p2;
614 msg_Dbg( p_input, "ip:%s port:%d", psz_ip, i_port );
616 if( ftp_SendCommand( p_input, "TYPE I" ) < 0 )
618 msg_Err( p_input, "cannot set binary transfert mode" );
621 if( ftp_ReadCommand( p_input, &i_answer, NULL ) != 2 )
623 msg_Err( p_input, "cannot set binary transfert mode" );
630 if( ftp_SendCommand( p_input, "REST "I64Fu, i_start ) < 0 )
632 msg_Err( p_input, "cannot set restart point" );
635 if( ftp_ReadCommand( p_input, &i_answer, NULL ) > 3 )
637 msg_Err( p_input, "cannot set restart point" );
642 msg_Dbg( p_input, "waiting for data connection..." );
643 socket_desc.i_type = NETWORK_TCP;
644 socket_desc.psz_server_addr = psz_ip;
645 socket_desc.i_server_port = i_port;
646 socket_desc.psz_bind_addr = "";
647 socket_desc.i_bind_port = 0;
648 socket_desc.i_ttl = 0;
649 p_input->p_private = (void*)&socket_desc;
650 if( !( p_network = module_Need( p_input, "network", "" ) ) )
652 msg_Err( p_input, "failed to connect with server" );
655 module_Unneed( p_input, p_network );
656 p_access->socket_data.i_handle = socket_desc.i_handle;
657 p_input->i_mtu = socket_desc.i_mtu;
659 "connection with \"%s:%d\" successful",
662 if( ftp_SendCommand( p_input, "RETR %s", p_access->url.psz_path ) < 0 )
664 msg_Err( p_input, "cannot retreive file" );
668 if( ftp_ReadCommand( p_input, &i_answer, NULL ) > 2 )
670 msg_Err( p_input, "cannot retreive file" );
677 static int ftp_StopStream ( input_thread_t *p_input)
679 access_t *p_access = (access_t*)p_input->p_access_data;
683 NetClose( p_input, &p_access->socket_data );
685 if( ftp_SendCommand( p_input, "ABOR" ) < 0 )
687 msg_Err( p_input, "cannot abord file" );
691 ftp_ReadCommand( p_input, &i_answer, NULL );
692 ftp_ReadCommand( p_input, &i_answer, NULL );
698 /****************************************************************************
700 ****************************************************************************/
701 static void ftp_ParseURL( url_t *p_url, char *psz_url )
704 char *psz_server_port;
706 p_url->psz_private = strdup( psz_url );
708 psz_parser = p_url->psz_private;
710 while( *psz_parser == '/' )
714 p_url->psz_server_addr = psz_parser;
716 while( *psz_parser &&
717 *psz_parser != ':' && *psz_parser != '/' && *psz_parser != '@' )
722 if( *psz_parser == ':' )
726 psz_server_port = psz_parser;
728 while( *psz_parser && *psz_parser != '/' )
735 psz_server_port = "";
738 if( *psz_parser == '@' )
745 p_url->psz_bind_addr = psz_parser;
747 while( *psz_parser && *psz_parser != ':' && *psz_parser != '/' )
752 if( *psz_parser == ':' )
756 psz_bind_port = psz_parser;
758 while( *psz_parser && *psz_parser != '/' )
769 p_url->i_bind_port = strtol( psz_bind_port, &psz_parser, 10 );
773 p_url->i_bind_port = 0;
778 p_url->psz_bind_addr = "";
779 p_url->i_bind_port = 0;
782 if( *psz_parser == '/' )
786 p_url->psz_path = psz_parser;
789 if( *psz_server_port )
791 p_url->i_server_port = strtol( psz_server_port, &psz_parser, 10 );
795 p_url->i_server_port = 0;
799 /*****************************************************************************
800 * Read: read on a file descriptor, checking b_die periodically
801 *****************************************************************************/
802 static ssize_t NetRead( input_thread_t *p_input,
803 input_socket_t *p_socket,
804 byte_t *p_buffer, size_t i_len )
810 struct timeval timeout;
815 /* Initialize file descriptor set */
817 FD_SET( p_socket->i_handle, &fds );
819 /* We'll wait 1 second if nothing happens */
823 /* Find if some data is available */
824 while( (i_ret = select( p_socket->i_handle + 1, &fds,
825 NULL, NULL, &timeout )) == 0
826 || (i_ret < 0 && errno == EINTR) )
829 FD_SET( p_socket->i_handle, &fds );
833 if( p_input->b_die || p_input->b_error )
841 msg_Err( p_input, "network select error (%s)", strerror(errno) );
845 i_recv = recv( p_socket->i_handle, p_buffer, i_len, 0 );
849 msg_Err( p_input, "recv failed (%s)", strerror(errno) );
857 static void NetClose( input_thread_t *p_input, input_socket_t *p_socket )
859 #if defined( UNDER_CE )
860 CloseHandle( (HANDLE)p_socket->i_handle );
861 #elif defined( WIN32 )
862 closesocket( p_socket->i_handle );
864 close( p_socket->i_handle );