void readFile(LIBSSH2_SESSION * session, LIBSSH2_SFTP * sftpSession, LIBSSH2_SFTP_HANDLE * fileHandle) {
size_t totalRead = 0;
if (!fileHandle) {
printSftpErrorAndDie(session, sftpSession);
}
LIBSSH2_SFTP_ATTRIBUTES fileAttributes;
libssh2_sftp_fstat(fileHandle, &fileAttributes);
size_t totalSize = (size_t)fileAttributes.filesize;
char * filedata = malloc(totalSize);
char * iter = filedata;
while (totalRead < totalSize) {
ssize_t readResult = libssh2_sftp_read(fileHandle, iter, fileAttributes.filesize);
if (readResult < 0) {
printSftpErrorAndDie(session, sftpSession);
}
iter += readResult;
totalRead += readResult;
printf("Read %ld bytes\n", readResult);
}
FILE * file = fopen(clientFilePath, "w");
fwrite(filedata, totalRead, sizeof(char), file);
printf("Wrote %lu bytes to %s\n", totalRead, clientFilePath);
fclose(file);
free(filedata);
libssh2_sftp_close(fileHandle);
}
int SyncTransferThread::do_download(QString remote_path, QString local_path)
{
qDebug() <<__FUNCTION__<<": "<<__LINE__<<":"<< __FILE__;
qDebug()<< "remote_path = "<< remote_path << " , local_path = " << local_path ;
int pcnt = 0 ;
int rlen , wlen ;
int file_size , tran_len = 0 ;
LIBSSH2_SFTP_HANDLE * sftp_handle ;
LIBSSH2_SFTP_ATTRIBUTES ssh2_sftp_attrib;
char buff[8192] = {0};
QString currFile;
sftp_handle = libssh2_sftp_open(this->ssh2_sftp,
GlobalOption::instance()->remote_codec->fromUnicode(remote_path), LIBSSH2_FXF_READ, 0);
if (sftp_handle == NULL) {
//TODO 错误消息通知用户。
qDebug()<<"open sftp file error :"<< libssh2_sftp_last_error(this->ssh2_sftp);
return -1 ;
}
memset(&ssh2_sftp_attrib,0,sizeof(ssh2_sftp_attrib));
libssh2_sftp_fstat(sftp_handle,&ssh2_sftp_attrib);
file_size = ssh2_sftp_attrib.filesize;
qDebug()<<" remote file size :"<< file_size ;
currFile = local_path.right(local_path.length() - this->local_base_path.length() - 1);
emit this->syncFileStarted(currFile, file_size);
// 本地编码 --> Qt 内部编码
QFile q_file(local_path);
if (!q_file.open(QIODevice::ReadWrite|QIODevice::Truncate)) {
//TODO 错误消息通知用户。
qDebug()<<"open local file error:"<< q_file.errorString() ;
} else {
//read remote file and then write to local file
while ((rlen = libssh2_sftp_read(sftp_handle, buff, sizeof(buff))) > 0) {
wlen = q_file.write( buff, rlen );
tran_len += wlen ;
qDebug()<<"Read len :"<<rlen <<" , write len: "<<wlen
<<" tran len: "<<tran_len ;
//my progress signal
if (file_size == 0) {
emit this->transfer_percent_changed(currFile, 100, tran_len, wlen);
} else {
pcnt = 100.0 *((double)tran_len / (double)file_size);
emit this->transfer_percent_changed(currFile, pcnt, tran_len, wlen);
}
}
q_file.flush();
q_file.close();
}
libssh2_sftp_close(sftp_handle);
q_debug()<<"syncDownload done.";
this->do_touch_local_file_with_time(local_path, SSHFileInfo(ssh2_sftp_attrib).lastModified());
emit this->syncFileStopped(currFile, 0);
return 0;
}
/* {{{ php_ssh2_sftp_stream_fstat
*/
static int php_ssh2_sftp_stream_fstat(php_stream *stream, php_stream_statbuf *ssb)
{
php_ssh2_sftp_handle_data *data = (php_ssh2_sftp_handle_data*)stream->abstract;
LIBSSH2_SFTP_ATTRIBUTES attrs;
if (libssh2_sftp_fstat(data->handle, &attrs)) {
return -1;
}
return php_ssh2_sftp_attr2ssb(ssb, &attrs);
}
int FSSftp::FStat ( int fd, FSStat* st, int* err, FSCInfo* info )
{
MutexLock lock( &mutex );
int ret = CheckSession( err, info );
if ( ret ) { return ret; }
if ( fd < 0 || fd >= MAX_FILES || !fileTable[fd] )
{
if ( err ) { *err = EINVAL; }
return -1;
}
try
{
SftpAttr attr;
int ret;
WHILE_EAGAIN_( ret, libssh2_sftp_fstat( fileTable[fd], &attr.attr ) );
CheckSFTP( ret );
st->mode = attr.Permissions();
st->size = attr.Size();
st->uid = attr.Uid();
st->gid = attr.Gid();
st->mtime = attr.MTime();
}
catch ( int e )
{
st->mode = 0;
if ( err ) { *err = e; }
return ( e == -2 ) ? -2 : -1;
}
return 0;
}
/* {{{ php_ssh2_sftp_stream_seek
*/
static int php_ssh2_sftp_stream_seek(php_stream *stream, off_t offset, int whence, off_t *newoffset)
{
php_ssh2_sftp_handle_data *data = (php_ssh2_sftp_handle_data*)stream->abstract;
switch (whence) {
case SEEK_END:
{
LIBSSH2_SFTP_ATTRIBUTES attrs;
if (libssh2_sftp_fstat(data->handle, &attrs)) {
return -1;
}
if ((attrs.flags & LIBSSH2_SFTP_ATTR_SIZE) == 0) {
return -1;
}
offset += attrs.filesize;
}
case SEEK_CUR:
{
off_t current_offset = libssh2_sftp_tell(data->handle);
if (current_offset < 0) {
return -1;
}
offset += current_offset;
}
}
libssh2_sftp_seek(data->handle, offset);
if (newoffset) {
*newoffset = offset;
}
return 0;
}
static int connect_to_ssh(BDRVSSHState *s, QDict *options,
int ssh_flags, int creat_mode, Error **errp)
{
int r, ret;
const char *host, *user, *path, *host_key_check;
int port;
if (!qdict_haskey(options, "host")) {
ret = -EINVAL;
error_setg(errp, "No hostname was specified");
goto err;
}
host = qdict_get_str(options, "host");
if (qdict_haskey(options, "port")) {
port = qdict_get_int(options, "port");
} else {
port = 22;
}
if (!qdict_haskey(options, "path")) {
ret = -EINVAL;
error_setg(errp, "No path was specified");
goto err;
}
path = qdict_get_str(options, "path");
if (qdict_haskey(options, "user")) {
user = qdict_get_str(options, "user");
} else {
user = g_get_user_name();
if (!user) {
error_setg_errno(errp, errno, "Can't get user name");
ret = -errno;
goto err;
}
}
if (qdict_haskey(options, "host_key_check")) {
host_key_check = qdict_get_str(options, "host_key_check");
} else {
host_key_check = "yes";
}
/* Construct the host:port name for inet_connect. */
g_free(s->hostport);
s->hostport = g_strdup_printf("%s:%d", host, port);
/* Open the socket and connect. */
s->sock = inet_connect(s->hostport, errp);
if (s->sock < 0) {
ret = -errno;
goto err;
}
/* Create SSH session. */
s->session = libssh2_session_init();
if (!s->session) {
ret = -EINVAL;
session_error_setg(errp, s, "failed to initialize libssh2 session");
goto err;
}
#if TRACE_LIBSSH2 != 0
libssh2_trace(s->session, TRACE_LIBSSH2);
#endif
r = libssh2_session_handshake(s->session, s->sock);
if (r != 0) {
ret = -EINVAL;
session_error_setg(errp, s, "failed to establish SSH session");
goto err;
}
/* Check the remote host's key against known_hosts. */
ret = check_host_key(s, host, port, host_key_check, errp);
if (ret < 0) {
goto err;
}
/* Authenticate. */
ret = authenticate(s, user, errp);
if (ret < 0) {
goto err;
}
/* Start SFTP. */
s->sftp = libssh2_sftp_init(s->session);
if (!s->sftp) {
session_error_setg(errp, s, "failed to initialize sftp handle");
ret = -EINVAL;
goto err;
}
/* Open the remote file. */
DPRINTF("opening file %s flags=0x%x creat_mode=0%o",
path, ssh_flags, creat_mode);
s->sftp_handle = libssh2_sftp_open(s->sftp, path, ssh_flags, creat_mode);
if (!s->sftp_handle) {
session_error_setg(errp, s, "failed to open remote file '%s'", path);
ret = -EINVAL;
goto err;
}
r = libssh2_sftp_fstat(s->sftp_handle, &s->attrs);
if (r < 0) {
sftp_error_setg(errp, s, "failed to read file attributes");
return -EINVAL;
}
/* Delete the options we've used; any not deleted will cause the
* block layer to give an error about unused options.
*/
qdict_del(options, "host");
qdict_del(options, "port");
qdict_del(options, "user");
qdict_del(options, "path");
qdict_del(options, "host_key_check");
return 0;
err:
if (s->sftp_handle) {
libssh2_sftp_close(s->sftp_handle);
}
s->sftp_handle = NULL;
if (s->sftp) {
libssh2_sftp_shutdown(s->sftp);
}
s->sftp = NULL;
if (s->session) {
libssh2_session_disconnect(s->session,
"from qemu ssh client: "
"error opening connection");
libssh2_session_free(s->session);
}
s->session = NULL;
return ret;
}
static int connect_to_ssh(BDRVSSHState *s, QDict *options,
int ssh_flags, int creat_mode, Error **errp)
{
int r, ret;
QemuOpts *opts = NULL;
Error *local_err = NULL;
const char *user, *path, *host_key_check;
long port = 0;
opts = qemu_opts_create(&ssh_runtime_opts, NULL, 0, &error_abort);
qemu_opts_absorb_qdict(opts, options, &local_err);
if (local_err) {
ret = -EINVAL;
error_propagate(errp, local_err);
goto err;
}
if (!ssh_process_legacy_socket_options(options, opts, errp)) {
ret = -EINVAL;
goto err;
}
path = qemu_opt_get(opts, "path");
if (!path) {
ret = -EINVAL;
error_setg(errp, "No path was specified");
goto err;
}
user = qemu_opt_get(opts, "user");
if (!user) {
user = g_get_user_name();
if (!user) {
error_setg_errno(errp, errno, "Can't get user name");
ret = -errno;
goto err;
}
}
host_key_check = qemu_opt_get(opts, "host_key_check");
if (!host_key_check) {
host_key_check = "yes";
}
/* Pop the config into our state object, Exit if invalid */
s->inet = ssh_config(s, options, errp);
if (!s->inet) {
ret = -EINVAL;
goto err;
}
if (qemu_strtol(s->inet->port, NULL, 10, &port) < 0) {
error_setg(errp, "Use only numeric port value");
ret = -EINVAL;
goto err;
}
/* Open the socket and connect. */
s->sock = inet_connect_saddr(s->inet, errp, NULL, NULL);
if (s->sock < 0) {
ret = -EIO;
goto err;
}
/* Create SSH session. */
s->session = libssh2_session_init();
if (!s->session) {
ret = -EINVAL;
session_error_setg(errp, s, "failed to initialize libssh2 session");
goto err;
}
#if TRACE_LIBSSH2 != 0
libssh2_trace(s->session, TRACE_LIBSSH2);
#endif
r = libssh2_session_handshake(s->session, s->sock);
if (r != 0) {
ret = -EINVAL;
session_error_setg(errp, s, "failed to establish SSH session");
goto err;
}
/* Check the remote host's key against known_hosts. */
ret = check_host_key(s, s->inet->host, port, host_key_check,
errp);
if (ret < 0) {
goto err;
}
/* Authenticate. */
ret = authenticate(s, user, errp);
if (ret < 0) {
goto err;
}
/* Start SFTP. */
s->sftp = libssh2_sftp_init(s->session);
if (!s->sftp) {
session_error_setg(errp, s, "failed to initialize sftp handle");
ret = -EINVAL;
goto err;
}
/* Open the remote file. */
DPRINTF("opening file %s flags=0x%x creat_mode=0%o",
path, ssh_flags, creat_mode);
s->sftp_handle = libssh2_sftp_open(s->sftp, path, ssh_flags, creat_mode);
if (!s->sftp_handle) {
session_error_setg(errp, s, "failed to open remote file '%s'", path);
ret = -EINVAL;
goto err;
}
qemu_opts_del(opts);
r = libssh2_sftp_fstat(s->sftp_handle, &s->attrs);
if (r < 0) {
sftp_error_setg(errp, s, "failed to read file attributes");
return -EINVAL;
}
return 0;
err:
if (s->sftp_handle) {
libssh2_sftp_close(s->sftp_handle);
}
s->sftp_handle = NULL;
if (s->sftp) {
libssh2_sftp_shutdown(s->sftp);
}
s->sftp = NULL;
if (s->session) {
libssh2_session_disconnect(s->session,
"from qemu ssh client: "
"error opening connection");
libssh2_session_free(s->session);
}
s->session = NULL;
qemu_opts_del(opts);
return ret;
}
