1 /*****************************************************************************
2 * sftp.c: SFTP input module
3 *****************************************************************************
4 * Copyright (C) 2009 VLC authors and VideoLAN
7 * Authors: RĂ©mi Duraffort <ivoire@videolan.org>
9 * This program is free software; you can redistribute it and/or modify it
10 * under the terms of the GNU Lesser General Public License as published by
11 * the Free Software Foundation; either version 2.1 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 Lesser General Public License for more details.
19 * You should have received a copy of the GNU Lesser General Public License
20 * along with this program; if not, write to the Free Software Foundation,
21 * Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
22 *****************************************************************************/
24 /*****************************************************************************
26 *****************************************************************************/
31 #include <vlc_common.h>
32 #include <vlc_plugin.h>
36 #include <vlc_access.h>
37 #include <vlc_dialog.h>
38 #include <vlc_network.h>
42 #include <libssh2_sftp.h>
45 /*****************************************************************************
47 *****************************************************************************/
48 static int Open ( vlc_object_t
* );
49 static void Close( vlc_object_t
* );
51 #define PORT_TEXT N_("SFTP port")
52 #define PORT_LONGTEXT N_("SFTP port number to use on the server")
53 #define MTU_TEXT N_("Read size")
54 #define MTU_LONGTEXT N_("Size of the request for reading access")
57 set_shortname( "SFTP" )
58 set_description( N_("SFTP input") )
59 set_capability( "access", 0 )
60 set_category( CAT_INPUT
)
61 set_subcategory( SUBCAT_INPUT_ACCESS
)
62 add_integer( "sftp-readsize", 8192, MTU_TEXT
, MTU_LONGTEXT
, true )
63 add_integer( "sftp-port", 22, PORT_TEXT
, PORT_LONGTEXT
, true )
64 add_shortcut( "sftp" )
65 set_callbacks( Open
, Close
)
69 /*****************************************************************************
71 *****************************************************************************/
72 static block_t
* Block( access_t
* );
73 static int Seek( access_t
*, uint64_t );
74 static int Control( access_t
*, int, va_list );
80 LIBSSH2_SESSION
* ssh_session
;
81 LIBSSH2_SFTP
* sftp_session
;
82 LIBSSH2_SFTP_HANDLE
* file
;
90 * Connect to the sftp server and ask for a file
91 * @param p_this: the vlc_object
92 * @return VLC_SUCCESS if everything was fine
94 static int Open( vlc_object_t
* p_this
)
96 access_t
* p_access
= (access_t
*)p_this
;
98 char* psz_username
= NULL
;
99 char* psz_password
= NULL
;
106 if( !p_access
->psz_location
)
109 STANDARD_BLOCK_ACCESS_INIT
;
112 const char* path
= p_access
->psz_location
;
113 vlc_UrlParse( &url
, path
, 0 );
115 /* Check for some parameters */
116 if( EMPTY_STR( url
.psz_host
) )
118 msg_Err( p_access
, "You might give a non empty host" );
122 /* If the user name is empty, ask the user */
123 if( !EMPTY_STR( url
.psz_username
) && url
.psz_password
)
125 psz_username
= strdup( url
.psz_username
);
126 psz_password
= strdup( url
.psz_password
);
130 dialog_Login( p_access
, &psz_username
, &psz_password
,
131 _("SFTP authentication"),
132 _("Please enter a valid login and password for the sftp "
133 "connexion to %s"), url
.psz_host
);
134 if( EMPTY_STR(psz_username
) || !psz_password
)
138 if( url
.i_port
<= 0 )
139 i_port
= var_InheritInteger( p_access
, "sftp-port" );
144 /* Connect to the server using a regular socket */
145 p_sys
->i_socket
= net_Connect( p_access
, url
.psz_host
, i_port
, SOCK_STREAM
, 0 );
147 /* Create the ssh connexion and wait until the server answer */
148 if( ( p_sys
->ssh_session
= libssh2_session_init() ) == NULL
)
151 while( ( i_ret
= libssh2_session_startup( p_sys
->ssh_session
,
153 == LIBSSH2_ERROR_EAGAIN
);
157 msg_Err( p_access
, "Impossible to open the connection to %s:%i", url
.psz_host
, i_port
);
161 /* Set the socket in non-blocking mode */
162 libssh2_session_set_blocking( p_sys
->ssh_session
, 1 );
164 /* List the know hosts */
165 LIBSSH2_KNOWNHOSTS
*ssh_knownhosts
= libssh2_knownhost_init( p_sys
->ssh_session
);
166 if( !ssh_knownhosts
)
169 char *psz_home
= config_GetUserDir( VLC_HOME_DIR
);
170 char *psz_knownhosts_file
;
171 if( asprintf( &psz_knownhosts_file
, "%s/.ssh/known_hosts", psz_home
) != -1 )
173 libssh2_knownhost_readfile( ssh_knownhosts
, psz_knownhosts_file
,
174 LIBSSH2_KNOWNHOST_FILE_OPENSSH
);
175 free( psz_knownhosts_file
);
179 const char *fingerprint
= libssh2_session_hostkey( p_sys
->ssh_session
, &i_len
, &i_type
);
180 struct libssh2_knownhost
*host
;
181 int check
= libssh2_knownhost_check( ssh_knownhosts
, url
.psz_host
,
183 LIBSSH2_KNOWNHOST_TYPE_PLAIN
|
184 LIBSSH2_KNOWNHOST_KEYENC_RAW
,
187 libssh2_knownhost_free( ssh_knownhosts
);
189 /* Check that it does match or at least that the host is unknown */
192 case LIBSSH2_KNOWNHOST_CHECK_FAILURE
:
193 case LIBSSH2_KNOWNHOST_CHECK_NOTFOUND
:
194 msg_Dbg( p_access
, "Unable to check the remote host" );
196 case LIBSSH2_KNOWNHOST_CHECK_MATCH
:
197 msg_Dbg( p_access
, "Succesfuly matched the host" );
199 case LIBSSH2_KNOWNHOST_CHECK_MISMATCH
:
200 msg_Err( p_access
, "The host does not match !! The remote key changed !!" );
204 //TODO: ask for the available auth methods
206 /* send the login/password */
207 if( libssh2_userauth_password( p_sys
->ssh_session
, psz_username
, psz_password
) )
209 msg_Err( p_access
, "Authentication by password failed" );
213 /* Create the sftp session */
214 p_sys
->sftp_session
= libssh2_sftp_init( p_sys
->ssh_session
);
216 if( !p_sys
->sftp_session
)
218 msg_Err( p_access
, "Unable to initialize the SFTP session" );
222 /* Open the given file */
223 p_sys
->file
= libssh2_sftp_open( p_sys
->sftp_session
, url
.psz_path
, LIBSSH2_FXF_READ
, 0 );
226 msg_Err( p_access
, "Unable to open the remote file %s", url
.psz_path
);
230 /* Get some information */
231 LIBSSH2_SFTP_ATTRIBUTES attributes
;
232 if( libssh2_sftp_stat( p_sys
->sftp_session
, url
.psz_path
, &attributes
) )
234 msg_Err( p_access
, "Impossible to get information about the remote file %s", url
.psz_path
);
237 p_sys
->filesize
= attributes
.filesize
;
239 p_sys
->i_read_size
= var_InheritInteger( p_access
, "sftp-readsize" );
241 free( psz_password
);
242 free( psz_username
);
243 vlc_UrlClean( &url
);
247 if( p_sys
->ssh_session
)
248 libssh2_session_free( p_sys
->ssh_session
);
249 free( psz_password
);
250 free( psz_username
);
251 vlc_UrlClean( &url
);
257 /* Close: quit the module */
258 static void Close( vlc_object_t
* p_this
)
260 access_t
* p_access
= (access_t
*)p_this
;
261 access_sys_t
* p_sys
= p_access
->p_sys
;
263 libssh2_sftp_close_handle( p_sys
->file
);
264 libssh2_sftp_shutdown( p_sys
->sftp_session
);
266 libssh2_session_free( p_sys
->ssh_session
);
271 static block_t
* Block( access_t
* p_access
)
273 access_sys_t
*p_sys
= p_access
->p_sys
;
275 if( p_access
->info
.b_eof
)
278 /* Allocate the buffer we need */
279 size_t i_len
= __MIN( p_sys
->i_read_size
,
280 p_sys
->filesize
- p_access
->info
.i_pos
);
281 block_t
* p_block
= block_Alloc( i_len
);
285 /* Read the specified size */
286 ssize_t i_ret
= libssh2_sftp_read( p_access
->p_sys
->file
, (char*)p_block
->p_buffer
, i_len
);
290 block_Release( p_block
);
291 msg_Err( p_access
, "read failed" );
294 else if( i_ret
== 0 )
296 p_access
->info
.b_eof
= true;
297 block_Release( p_block
);
302 p_access
->info
.i_pos
+= i_ret
;
308 static int Seek( access_t
* p_access
, uint64_t i_pos
)
310 p_access
->info
.i_pos
= i_pos
;
311 p_access
->info
.b_eof
= false;
313 libssh2_sftp_seek( p_access
->p_sys
->file
, i_pos
);
318 static int Control( access_t
* p_access
, int i_query
, va_list args
)
325 case ACCESS_CAN_SEEK
:
326 pb_bool
= (bool*)va_arg( args
, bool* );
330 case ACCESS_CAN_FASTSEEK
:
331 pb_bool
= (bool*)va_arg( args
, bool* );
335 case ACCESS_CAN_PAUSE
:
336 case ACCESS_CAN_CONTROL_PACE
:
337 pb_bool
= (bool*)va_arg( args
, bool* );
341 case ACCESS_GET_SIZE
:
342 *va_arg( args
, uint64_t * ) = p_access
->p_sys
->filesize
;
345 case ACCESS_GET_PTS_DELAY
:
346 pi_64
= (int64_t*)va_arg( args
, int64_t* );
347 *pi_64
= INT64_C(1000)
348 * var_InheritInteger( p_access
, "network-caching" );
351 case ACCESS_SET_PAUSE_STATE
: