shinys000114/libssh
1
0
Fork 0
mirror of https://git.libssh.org/projects/libssh.git synced 2026-02-11 02:38:09 +09:00
Code Issues Packages Projects Releases Wiki Activity

Improve sftp_read().

Browse Source 
git-svn-id: svn+ssh://svn.berlios.de/svnroot/repos/libssh/trunk@601 7dcaeef0-15fb-0310-b436-a5af3365683c
This commit is contained in:
Andreas Schneider
2009-04-24 09:05:52 +00:00
parent 10ddf310a6
commit 8ba57619fd
1 changed files with 171 additions and 127 deletions
Download Patch File Download Diff File Expand all files Collapse all files

298
libssh/sftp.c
View File

@@ -1352,67 +1352,93 @@ int sftp_closedir(SFTP_DIR *dir){
/* Open a file on the server. */ /* Open a file on the server. */
SFTP_FILE *sftp_open(SFTP_SESSION *sftp, const char *file, int flags, mode_t mode){ SFTP_FILE *sftp_open(SFTP_SESSION *sftp, const char *file, int flags, mode_t mode){
SFTP_FILE *handle; SFTP_MESSAGE *msg = NULL;
SFTP_MESSAGE *msg=NULL; STATUS_MESSAGE *status;
STATUS_MESSAGE *status; SFTP_ATTRIBUTES attr;
SFTP_ATTRIBUTES attr; SFTP_FILE *handle;
u32 sftp_flags = 0; STRING *filename;
u32 id=sftp_get_new_id(sftp); BUFFER *buffer;
BUFFER *buffer=buffer_new(); u32 sftp_flags = 0;
STRING *filename; u32 id;
ZERO_STRUCT(attr); buffer = buffer_new();
attr.permissions = mode; if (buffer == NULL) {
attr.flags = SSH_FILEXFER_ATTR_PERMISSIONS; return NULL;
}
if(flags == O_RDONLY) filename = string_from_char(file);
sftp_flags|=SSH_FXF_READ; // if any of the other flag is set, READ should not be set initialy if (filename == NULL) {
if(flags & O_WRONLY) buffer_free(buffer);
sftp_flags |= SSH_FXF_WRITE; return NULL;
if(flags & O_RDWR) }
sftp_flags|=(SSH_FXF_WRITE | SSH_FXF_READ);
if(flags & O_CREAT) ZERO_STRUCT(attr);
sftp_flags |=SSH_FXF_CREAT; attr.permissions = mode;
if(flags & O_TRUNC) attr.flags = SSH_FILEXFER_ATTR_PERMISSIONS;
sftp_flags |=SSH_FXF_TRUNC;
if(flags & O_EXCL) if (flags == O_RDONLY)
sftp_flags |= SSH_FXF_EXCL; sftp_flags |= SSH_FXF_READ; /* if any of the other flag is set,
buffer_add_u32(buffer,id); READ should not be set initialy */
filename=string_from_char(file); if (flags & O_WRONLY)
buffer_add_ssh_string(buffer,filename); sftp_flags |= SSH_FXF_WRITE;
free(filename); if (flags & O_RDWR)
buffer_add_u32(buffer,htonl(sftp_flags)); sftp_flags |= (SSH_FXF_WRITE | SSH_FXF_READ);
if (buffer_add_attributes(buffer, &attr) < 0) { if (flags & O_CREAT)
buffer_free(buffer); sftp_flags |= SSH_FXF_CREAT;
if (flags & O_TRUNC)
sftp_flags |= SSH_FXF_TRUNC;
if (flags & O_EXCL)
sftp_flags |= SSH_FXF_EXCL;
id = sftp_get_new_id(sftp);
if (buffer_add_u32(buffer, id) < 0 ||
buffer_add_ssh_string(buffer, filename) < 0) {
buffer_free(buffer);
string_free(filename);
return NULL;
}
string_free(filename);
if (buffer_add_u32(buffer, htonl(sftp_flags)) < 0 ||
buffer_add_attributes(buffer, &attr) < 0 ||
sftp_packet_write(sftp, SSH_FXP_OPEN, buffer) < 0) {
buffer_free(buffer);
return NULL;
}
buffer_free(buffer);
while (msg == NULL) {
if (sftp_read_and_dispatch(sftp) < 0) {
/* something nasty has happened */
return NULL; return NULL;
} }
sftp_packet_write(sftp,SSH_FXP_OPEN,buffer); msg = sftp_dequeue(sftp, id);
buffer_free(buffer); }
while(!msg){
if(sftp_read_and_dispatch(sftp)) switch (msg->packet_type) {
/* something nasty has happened */ case SSH_FXP_STATUS:
return NULL; status = parse_status_msg(msg);
msg=sftp_dequeue(sftp,id); sftp_message_free(msg);
} if (status == NULL) {
switch (msg->packet_type){ return NULL;
case SSH_FXP_STATUS: }
status=parse_status_msg(msg); sftp_set_error(sftp, status->status);
sftp_message_free(msg); ssh_set_error(sftp->session, SSH_REQUEST_DENIED,
if(!status) "SFTP server: %s", status->errormsg);
return NULL; status_msg_free(status);
sftp_set_error(sftp, status->status);
ssh_set_error(sftp->session,SSH_REQUEST_DENIED,"sftp server : %s",status->errormsg); return NULL;
status_msg_free(status); case SSH_FXP_HANDLE:
return NULL; handle = parse_handle_msg(msg);
case SSH_FXP_HANDLE: sftp_message_free(msg);
handle=parse_handle_msg(msg); return handle;
sftp_message_free(msg); default:
return handle; ssh_set_error(sftp->session, SSH_FATAL,
default: "Received message %d during open!", msg->packet_type);
ssh_set_error(sftp->session,SSH_FATAL,"Received message %d during open!",msg->packet_type); sftp_message_free(msg);
sftp_message_free(msg); }
}
return NULL; return NULL;
} }
void sftp_file_set_nonblocking(SFTP_FILE *handle){ void sftp_file_set_nonblocking(SFTP_FILE *handle){
@@ -1423,79 +1449,97 @@ void sftp_file_set_blocking(SFTP_FILE *handle){
} }
/* Read from a file using an opened sftp file handle. */ /* Read from a file using an opened sftp file handle. */
ssize_t sftp_read(SFTP_FILE *handle, void *buf, size_t count){ ssize_t sftp_read(SFTP_FILE *handle, void *buf, size_t count) {
SFTP_MESSAGE *msg=NULL; SFTP_SESSION *sftp = handle->sftp;
STATUS_MESSAGE *status; SFTP_MESSAGE *msg = NULL;
SFTP_SESSION *sftp=handle->sftp; STATUS_MESSAGE *status;
STRING *datastring; STRING *datastring;
int id; BUFFER *buffer;
int err=0; int id;
BUFFER *buffer;
if(handle->eof) if (handle->eof) {
return 0; return 0;
buffer=buffer_new(); }
id=sftp_get_new_id(handle->sftp);
buffer_add_u32(buffer,id); buffer = buffer_new();
buffer_add_ssh_string(buffer,handle->handle); if (buffer == NULL) {
buffer_add_u64(buffer,htonll(handle->offset)); return -1;
buffer_add_u32(buffer,htonl(count)); }
sftp_packet_write(handle->sftp,SSH_FXP_READ,buffer); id = sftp_get_new_id(handle->sftp);
if (buffer_add_u32(buffer, id) < 0 ||
buffer_add_ssh_string(buffer, handle->handle) < 0 ||
buffer_add_u64(buffer, htonll(handle->offset)) < 0 ||
buffer_add_u32(buffer,htonl(count)) < 0 ||
sftp_packet_write(handle->sftp, SSH_FXP_READ, buffer) < 0) {
buffer_free(buffer); buffer_free(buffer);
while(!msg){ return -1;
if (handle->nonblocking){ }
if(channel_poll(handle->sftp->channel,0)==0){ buffer_free(buffer);
/* we cannot block */
return 0; while (msg == NULL) {
} if (handle->nonblocking) {
} if (channel_poll(handle->sftp->channel, 0) == 0) {
if(sftp_read_and_dispatch(handle->sftp)) /* we cannot block */
/* something nasty has happened */ return 0;
return -1; }
msg=sftp_dequeue(handle->sftp,id);
} }
switch (msg->packet_type){ if (sftp_read_and_dispatch(handle->sftp) < 0) {
case SSH_FXP_STATUS: /* something nasty has happened */
status=parse_status_msg(msg); return -1;
sftp_message_free(msg); }
if(!status) msg = sftp_dequeue(handle->sftp, id);
return -1; }
sftp_set_error(sftp, status->status);
switch (status->status) { switch (msg->packet_type) {
case SSH_FX_EOF: case SSH_FXP_STATUS:
handle->eof=1; status = parse_status_msg(msg);
status_msg_free(status); sftp_message_free(msg);
return err ? err : 0; if (status == NULL) {
default: return -1;
break; }
} sftp_set_error(sftp, status->status);
ssh_set_error(sftp->session,SSH_REQUEST_DENIED,"sftp server : %s",status->errormsg); switch (status->status) {
status_msg_free(status); case SSH_FX_EOF:
return -1; handle->eof = 1;
case SSH_FXP_DATA: status_msg_free(status);
datastring=buffer_get_ssh_string(msg->payload); return 0;
sftp_message_free(msg);
if(!datastring){
ssh_set_error(sftp->session,SSH_FATAL,"Received invalid DATA packet from sftp server");
return -1;
}
if(string_len(datastring) > count){
ssh_set_error(sftp->session, SSH_FATAL,
"Received a too big DATA packet from sftp server: %zu and asked for %zu",
string_len(datastring), count);
free(datastring);
return -1;
}
count = string_len(datastring);
handle->offset+= count;
memcpy(buf, datastring->string, count);
free(datastring);
return count;
default: default:
ssh_set_error(sftp->session,SSH_FATAL,"Received message %d during read!",msg->packet_type); break;
sftp_message_free(msg); }
return -1; ssh_set_error(sftp->session,SSH_REQUEST_DENIED,
} "SFTP server: %s", status->errormsg);
return -1; /* not reached */ status_msg_free(status);
return -1;
case SSH_FXP_DATA:
datastring = buffer_get_ssh_string(msg->payload);
sftp_message_free(msg);
if (datastring == NULL) {
ssh_set_error(sftp->session, SSH_FATAL,
"Received invalid DATA packet from sftp server");
return -1;
}
if (string_len(datastring) > count) {
ssh_set_error(sftp->session, SSH_FATAL,
"Received a too big DATA packet from sftp server: "
"%zu and asked for %zu",
string_len(datastring), count);
string_free(datastring);
return -1;
}
count = string_len(datastring);
handle->offset += count;
memcpy(buf, datastring->string, count);
string_free(datastring);
return count;
default:
ssh_set_error(sftp->session, SSH_FATAL,
"Received message %d during read!", msg->packet_type);
sftp_message_free(msg);
return -1;
}
return -1; /* not reached */
} }
/* Start an asynchronous read from a file using an opened sftp file handle. */ /* Start an asynchronous read from a file using an opened sftp file handle. */