lib/libssh
1
0
Fork 0
mirror of https://git.libssh.org/projects/libssh.git synced 2025-11-30 13:01:23 +03:00
Code Activity

sftp: Reformat sftp_fstatvfs()

Browse Source 
Signed-off-by: Andreas Schneider <asn@cryptomilk.org>
This commit is contained in:
Andreas Schneider
2018-08-31 19:07:32 +02:00
parent f1e84d5e67
commit 5914ea7c75
1 changed files with 84 additions and 83 deletions
Download Patch File Download Diff File Expand all files Collapse all files

167
src/sftp.c
View File

@@ -2903,100 +2903,101 @@ done:
return rc; return rc;
} }
sftp_statvfs_t sftp_fstatvfs(sftp_file file) { sftp_statvfs_t sftp_fstatvfs(sftp_file file)
sftp_status_message status = NULL; {
sftp_message msg = NULL; sftp_status_message status = NULL;
sftp_session sftp; sftp_message msg = NULL;
ssh_string ext; sftp_session sftp;
ssh_buffer buffer; ssh_string ext;
uint32_t id; ssh_buffer buffer;
int rc; uint32_t id;
int rc;
if (file == NULL) { if (file == NULL) {
return NULL; return NULL;
}
sftp = file->sftp;
buffer = ssh_buffer_new();
if (buffer == NULL) {
ssh_set_error_oom(sftp->session);
return NULL;
}
ext = ssh_string_from_char("fstatvfs@openssh.com");
if (ext == NULL) {
ssh_set_error_oom(sftp->session);
ssh_buffer_free(buffer);
return NULL;
}
rc = ssh_buffer_allocate_size(buffer,
sizeof(uint32_t) * 3 +
ssh_string_len(ext) +
ssh_string_len(file->handle));
if (rc < 0) {
ssh_set_error_oom(sftp->session);
ssh_buffer_free(buffer);
ssh_string_free(ext);
return NULL;
}
id = sftp_get_new_id(sftp);
if (ssh_buffer_add_u32(buffer, htonl(id)) < 0 ||
ssh_buffer_add_ssh_string(buffer, ext) < 0 ||
ssh_buffer_add_ssh_string(buffer, file->handle) < 0) {
ssh_set_error_oom(sftp->session);
ssh_buffer_free(buffer);
ssh_string_free(ext);
return NULL;
}
if (sftp_packet_write(sftp, SSH_FXP_EXTENDED, buffer) < 0) {
ssh_buffer_free(buffer);
ssh_string_free(ext);
return NULL;
}
ssh_buffer_free(buffer);
ssh_string_free(ext);
while (msg == NULL) {
if (sftp_read_and_dispatch(sftp) < 0) {
return NULL;
} }
msg = sftp_dequeue(sftp, id); sftp = file->sftp;
}
if (msg->packet_type == SSH_FXP_EXTENDED_REPLY) { buffer = ssh_buffer_new();
sftp_statvfs_t buf = sftp_parse_statvfs(sftp, msg->payload); if (buffer == NULL) {
sftp_message_free(msg); ssh_set_error_oom(sftp->session);
if (buf == NULL) { return NULL;
return NULL;
} }
return buf; ext = ssh_string_from_char("fstatvfs@openssh.com");
} else if (msg->packet_type == SSH_FXP_STATUS) { /* bad response (error) */ if (ext == NULL) {
status = parse_status_msg(msg); ssh_set_error_oom(sftp->session);
sftp_message_free(msg); ssh_buffer_free(buffer);
if (status == NULL) { return NULL;
return NULL;
} }
ssh_set_error(sftp->session, SSH_REQUEST_DENIED,
"SFTP server: %s", status->errormsg);
status_msg_free(status);
} else { /* this shouldn't happen */
ssh_set_error(sftp->session, SSH_FATAL,
"Received message %d when attempting to set stats", msg->packet_type);
sftp_message_free(msg);
}
return NULL; rc = ssh_buffer_allocate_size(buffer,
sizeof(uint32_t) * 3 +
ssh_string_len(ext) +
ssh_string_len(file->handle));
if (rc < 0) {
ssh_set_error_oom(sftp->session);
ssh_buffer_free(buffer);
ssh_string_free(ext);
return NULL;
}
id = sftp_get_new_id(sftp);
if (ssh_buffer_add_u32(buffer, htonl(id)) < 0 ||
ssh_buffer_add_ssh_string(buffer, ext) < 0 ||
ssh_buffer_add_ssh_string(buffer, file->handle) < 0) {
ssh_set_error_oom(sftp->session);
ssh_buffer_free(buffer);
ssh_string_free(ext);
return NULL;
}
if (sftp_packet_write(sftp, SSH_FXP_EXTENDED, buffer) < 0) {
ssh_buffer_free(buffer);
ssh_string_free(ext);
return NULL;
}
ssh_buffer_free(buffer);
ssh_string_free(ext);
while (msg == NULL) {
if (sftp_read_and_dispatch(sftp) < 0) {
return NULL;
}
msg = sftp_dequeue(sftp, id);
}
if (msg->packet_type == SSH_FXP_EXTENDED_REPLY) {
sftp_statvfs_t buf = sftp_parse_statvfs(sftp, msg->payload);
sftp_message_free(msg);
if (buf == NULL) {
return NULL;
}
return buf;
} else if (msg->packet_type == SSH_FXP_STATUS) { /* bad response (error) */
status = parse_status_msg(msg);
sftp_message_free(msg);
if (status == NULL) {
return NULL;
}
ssh_set_error(sftp->session, SSH_REQUEST_DENIED,
"SFTP server: %s", status->errormsg);
status_msg_free(status);
} else { /* this shouldn't happen */
ssh_set_error(sftp->session, SSH_FATAL,
"Received message %d when attempting to set stats", msg->packet_type);
sftp_message_free(msg);
}
return NULL;
} }
void sftp_statvfs_free(sftp_statvfs_t statvfs) { void sftp_statvfs_free(sftp_statvfs_t statvfs) {
if (statvfs == NULL) { if (statvfs == NULL) {
return; return;
} }
SAFE_FREE(statvfs); SAFE_FREE(statvfs);
} }
/* another code written by Nick */ /* another code written by Nick */