aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--include/libssh/priv.h97
-rw-r--r--src/sftp.c65
2 files changed, 38 insertions, 124 deletions
diff --git a/include/libssh/priv.h b/include/libssh/priv.h
index f399dc29..e53802f5 100644
--- a/include/libssh/priv.h
+++ b/include/libssh/priv.h
@@ -120,38 +120,39 @@ int gettimeofday(struct timeval *__p, void *__t);
#endif
#endif
-#ifdef __cplusplus
-extern "C" {
-#endif
+#define enter_function() (void)session
+#define leave_function() (void)session
#ifdef HAVE_SYS_TIME_H
#include <sys/time.h>
#endif
-/* error handling structure */
-struct error_struct {
- int error_code;
- char error_buffer[ERROR_BUFFERLEN];
-};
-
+/* forward declarations */
struct ssh_common_struct;
struct ssh_kex_struct;
-/* server data */
+int ssh_get_key_params(ssh_session session, ssh_key *privkey);
+/* LOGGING */
+#define SSH_LOG(session, priority, ...) \
+ ssh_log_common(&session->common, priority, __FUNCTION__, __VA_ARGS__)
+void ssh_log_common(struct ssh_common_struct *common,
+ int verbosity,
+ const char *function,
+ const char *format, ...) PRINTF_ATTRIBUTE(4, 5);
+void ssh_log_function(int verbosity,
+ const char *function,
+ const char *buffer);
-SSH_PACKET_CALLBACK(ssh_packet_disconnect_callback);
-SSH_PACKET_CALLBACK(ssh_packet_ignore_callback);
-int ssh_get_key_params(ssh_session session, ssh_key *privkey);
-/* client.c */
+/* ERROR HANDLING */
-int ssh_send_banner(ssh_session session, int is_server);
-SSH_PACKET_CALLBACK(ssh_packet_dh_reply);
-SSH_PACKET_CALLBACK(ssh_packet_newkeys);
-SSH_PACKET_CALLBACK(ssh_packet_service_accept);
+/* error handling structure */
+struct error_struct {
+ int error_code;
+ char error_buffer[ERROR_BUFFERLEN];
+};
-/* errors.c */
#define ssh_set_error(error, code, ...) \
_ssh_set_error(error, code, __FUNCTION__, __VA_ARGS__)
void _ssh_set_error(void *error,
@@ -167,6 +168,23 @@ void _ssh_set_error_oom(void *error, const char *function);
_ssh_set_error_invalid(error, __FUNCTION__)
void _ssh_set_error_invalid(void *error, const char *function);
+
+
+
+
+/* server data */
+
+
+SSH_PACKET_CALLBACK(ssh_packet_disconnect_callback);
+SSH_PACKET_CALLBACK(ssh_packet_ignore_callback);
+
+/* client.c */
+
+int ssh_send_banner(ssh_session session, int is_server);
+SSH_PACKET_CALLBACK(ssh_packet_dh_reply);
+SSH_PACKET_CALLBACK(ssh_packet_newkeys);
+SSH_PACKET_CALLBACK(ssh_packet_service_accept);
+
/* in crypt.c */
uint32_t packet_decrypt_len(ssh_session session,char *crypted);
int packet_decrypt(ssh_session session, void *packet,unsigned int len);
@@ -208,44 +226,9 @@ int match_hostname(const char *host, const char *pattern, unsigned int len);
int message_handle(ssh_session session, void *user, uint8_t type, ssh_buffer packet);
-#define _enter_function(sess) \
- do {\
- if((sess)->common.log_verbosity >= SSH_LOG_FUNCTIONS){ \
- ssh_log((sess),SSH_LOG_FUNCTIONS,"entering function %s line %d in " __FILE__ , __FUNCTION__,__LINE__);\
- (sess)->common.log_indent++; \
- } \
- } while(0)
-
-#define _leave_function(sess) \
- do { \
- if((sess)->common.log_verbosity >= SSH_LOG_FUNCTIONS){ \
- (sess)->common.log_indent--; \
- ssh_log((sess),SSH_LOG_FUNCTIONS,"leaving function %s line %d in " __FILE__ , __FUNCTION__,__LINE__);\
- }\
- } while(0)
-
-#ifdef DEBUG_CALLTRACE
-#define enter_function() _enter_function(session)
-#define leave_function() _leave_function(session)
-#else
-#define enter_function() (void)session
-#define leave_function() (void)session
-#endif
-
-
/* server.c */
SSH_PACKET_CALLBACK(ssh_packet_kexdh_init);
-/* LOGGING */
-#define SSH_LOG(session, priority, ...) \
- ssh_log_common(&session->common, priority, __FUNCTION__, __VA_ARGS__)
-void ssh_log_common(struct ssh_common_struct *common,
- int verbosity,
- const char *function,
- const char *format, ...) PRINTF_ATTRIBUTE(4, 5);
-void ssh_log_function(int verbosity,
- const char *function,
- const char *buffer);
/** Free memory space */
@@ -288,9 +271,5 @@ int my_gcry_dec2bn(bignum *bn, const char *data);
char *my_gcry_bn2dec(bignum bn);
#endif /* !HAVE_LIBGCRYPT */
-#ifdef __cplusplus
-}
-#endif
-
#endif /* _LIBSSH_PRIV_H */
-/* vim: set ts=2 sw=2 et cindent: */
+/* vim: set ts=4 sw=4 et cindent: */
diff --git a/src/sftp.c b/src/sftp.c
index 6000a9d8..a2241121 100644
--- a/src/sftp.c
+++ b/src/sftp.c
@@ -56,9 +56,6 @@
#ifdef WITH_SFTP
-#define sftp_enter_function() _enter_function(sftp->channel->session)
-#define sftp_leave_function() _leave_function(sftp->channel->session)
-
struct sftp_ext_struct {
unsigned int count;
char **name;
@@ -174,11 +171,8 @@ int sftp_server_init(sftp_session sftp){
ssh_buffer reply = NULL;
uint32_t version;
- sftp_enter_function();
-
packet = sftp_packet_read(sftp);
if (packet == NULL) {
- sftp_leave_function();
return -1;
}
@@ -188,7 +182,6 @@ int sftp_server_init(sftp_session sftp){
packet->type);
sftp_packet_free(packet);
- sftp_leave_function();
return -1;
}
@@ -204,20 +197,17 @@ int sftp_server_init(sftp_session sftp){
reply = ssh_buffer_new();
if (reply == NULL) {
ssh_set_error_oom(session);
- sftp_leave_function();
return -1;
}
if (buffer_add_u32(reply, ntohl(LIBSFTP_VERSION)) < 0) {
ssh_set_error_oom(session);
ssh_buffer_free(reply);
- sftp_leave_function();
return -1;
}
if (sftp_packet_write(sftp, SSH_FXP_VERSION, reply) < 0) {
ssh_buffer_free(reply);
- sftp_leave_function();
return -1;
}
ssh_buffer_free(reply);
@@ -230,7 +220,6 @@ int sftp_server_init(sftp_session sftp){
sftp->version=version;
}
- sftp_leave_function();
return 0;
}
#endif /* WITH_SERVER */
@@ -293,8 +282,6 @@ sftp_packet sftp_packet_read(sftp_session sftp) {
uint32_t size;
int r;
- sftp_enter_function();
-
packet = malloc(sizeof(struct sftp_packet_struct));
if (packet == NULL) {
ssh_set_error_oom(sftp->session);
@@ -312,7 +299,6 @@ sftp_packet sftp_packet_read(sftp_session sftp) {
if (r < 0) {
ssh_buffer_free(packet->payload);
SAFE_FREE(packet);
- sftp_leave_function();
return NULL;
}
buffer_add_data(packet->payload,buffer, r);
@@ -320,7 +306,6 @@ sftp_packet sftp_packet_read(sftp_session sftp) {
ssh_set_error(sftp->session, SSH_FATAL, "Short sftp packet!");
ssh_buffer_free(packet->payload);
SAFE_FREE(packet);
- sftp_leave_function();
return NULL;
}
@@ -330,7 +315,6 @@ sftp_packet sftp_packet_read(sftp_session sftp) {
/* TODO: check if there are cases where an error needs to be set here */
ssh_buffer_free(packet->payload);
SAFE_FREE(packet);
- sftp_leave_function();
return NULL;
}
buffer_add_data(packet->payload, buffer, r);
@@ -344,20 +328,17 @@ sftp_packet sftp_packet_read(sftp_session sftp) {
/* TODO: check if there are cases where an error needs to be set here */
ssh_buffer_free(packet->payload);
SAFE_FREE(packet);
- sftp_leave_function();
return NULL;
}
if(buffer_add_data(packet->payload,buffer,r)==SSH_ERROR){
ssh_buffer_free(packet->payload);
SAFE_FREE(packet);
- sftp_leave_function();
ssh_set_error_oom(sftp->session);
return NULL;
}
size -= r;
}
- sftp_leave_function();
return packet;
}
@@ -379,8 +360,6 @@ int sftp_get_error(sftp_session sftp) {
static sftp_message sftp_message_new(sftp_session sftp){
sftp_message msg = NULL;
- sftp_enter_function();
-
msg = malloc(sizeof(struct sftp_message_struct));
if (msg == NULL) {
ssh_set_error_oom(sftp->session);
@@ -396,35 +375,24 @@ static sftp_message sftp_message_new(sftp_session sftp){
}
msg->sftp = sftp;
- sftp_leave_function();
return msg;
}
static void sftp_message_free(sftp_message msg) {
- sftp_session sftp;
-
if (msg == NULL) {
return;
}
- sftp = msg->sftp;
- sftp_enter_function();
-
ssh_buffer_free(msg->payload);
SAFE_FREE(msg);
-
- sftp_leave_function();
}
static sftp_message sftp_get_message(sftp_packet packet) {
sftp_session sftp = packet->sftp;
sftp_message msg = NULL;
- sftp_enter_function();
-
msg = sftp_message_new(sftp);
if (msg == NULL) {
- sftp_leave_function();
return NULL;
}
@@ -437,7 +405,6 @@ static sftp_message sftp_get_message(sftp_packet packet) {
ssh_set_error(packet->sftp->session, SSH_FATAL,
"Unknown packet type %d", packet->type);
sftp_message_free(msg);
- sftp_leave_function();
return NULL;
}
@@ -445,7 +412,6 @@ static sftp_message sftp_get_message(sftp_packet packet) {
ssh_set_error(packet->sftp->session, SSH_FATAL,
"Invalid packet %d: no ID", packet->type);
sftp_message_free(msg);
- sftp_leave_function();
return NULL;
}
@@ -458,11 +424,9 @@ static sftp_message sftp_get_message(sftp_packet packet) {
buffer_get_rest_len(packet->payload)) < 0) {
ssh_set_error_oom(sftp->session);
sftp_message_free(msg);
- sftp_leave_function();
return NULL;
}
- sftp_leave_function();
return msg;
}
@@ -470,28 +434,22 @@ static int sftp_read_and_dispatch(sftp_session sftp) {
sftp_packet packet = NULL;
sftp_message msg = NULL;
- sftp_enter_function();
-
packet = sftp_packet_read(sftp);
if (packet == NULL) {
- sftp_leave_function();
return -1; /* something nasty happened reading the packet */
}
msg = sftp_get_message(packet);
sftp_packet_free(packet);
if (msg == NULL) {
- sftp_leave_function();
return -1;
}
if (sftp_enqueue(sftp, msg) < 0) {
sftp_message_free(msg);
- sftp_leave_function();
return -1;
}
- sftp_leave_function();
return 0;
}
@@ -514,31 +472,25 @@ int sftp_init(sftp_session sftp) {
char *ext_data = NULL;
uint32_t version = htonl(LIBSFTP_VERSION);
- sftp_enter_function();
-
buffer = ssh_buffer_new();
if (buffer == NULL) {
ssh_set_error_oom(sftp->session);
- sftp_leave_function();
return -1;
}
if (buffer_add_u32(buffer, version) < 0) {
ssh_set_error_oom(sftp->session);
ssh_buffer_free(buffer);
- sftp_leave_function();
return -1;
}
if (sftp_packet_write(sftp, SSH_FXP_INIT, buffer) < 0) {
ssh_buffer_free(buffer);
- sftp_leave_function();
return -1;
}
ssh_buffer_free(buffer);
packet = sftp_packet_read(sftp);
if (packet == NULL) {
- sftp_leave_function();
return -1;
}
@@ -546,7 +498,6 @@ int sftp_init(sftp_session sftp) {
ssh_set_error(sftp->session, SSH_FATAL,
"Received a %d messages instead of SSH_FXP_VERSION", packet->type);
sftp_packet_free(packet);
- sftp_leave_function();
return -1;
}
@@ -619,7 +570,6 @@ int sftp_init(sftp_session sftp) {
sftp->version = sftp->server_version = version;
- sftp_leave_function();
return 0;
}
@@ -1843,8 +1793,6 @@ int sftp_async_read_begin(sftp_file file, uint32_t len){
ssh_buffer buffer;
uint32_t id;
- sftp_enter_function();
-
buffer = ssh_buffer_new();
if (buffer == NULL) {
ssh_set_error_oom(sftp->session);
@@ -1868,7 +1816,6 @@ int sftp_async_read_begin(sftp_file file, uint32_t len){
file->offset += len; /* assume we'll read len bytes */
- sftp_leave_function();
return id;
}
@@ -1881,10 +1828,7 @@ int sftp_async_read(sftp_file file, void *data, uint32_t size, uint32_t id){
int err = SSH_OK;
uint32_t len;
- sftp_enter_function();
-
if (file->eof) {
- sftp_leave_function();
return 0;
}
@@ -1899,7 +1843,6 @@ int sftp_async_read(sftp_file file, void *data, uint32_t size, uint32_t id){
if (sftp_read_and_dispatch(sftp) < 0) {
/* something nasty has happened */
- sftp_leave_function();
return SSH_ERROR;
}
@@ -1911,20 +1854,17 @@ int sftp_async_read(sftp_file file, void *data, uint32_t size, uint32_t id){
status = parse_status_msg(msg);
sftp_message_free(msg);
if (status == NULL) {
- sftp_leave_function();
return -1;
}
sftp_set_error(sftp, status->status);
if (status->status != SSH_FX_EOF) {
ssh_set_error(sftp->session, SSH_REQUEST_DENIED,
"SFTP server : %s", status->errormsg);
- sftp_leave_function();
err = SSH_ERROR;
} else {
file->eof = 1;
}
status_msg_free(status);
- sftp_leave_function();
return err;
case SSH_FXP_DATA:
datastring = buffer_get_ssh_string(msg->payload);
@@ -1932,7 +1872,6 @@ int sftp_async_read(sftp_file file, void *data, uint32_t size, uint32_t id){
if (datastring == NULL) {
ssh_set_error(sftp->session, SSH_FATAL,
"Received invalid DATA packet from sftp server");
- sftp_leave_function();
return SSH_ERROR;
}
if (ssh_string_len(datastring) > size) {
@@ -1941,7 +1880,6 @@ int sftp_async_read(sftp_file file, void *data, uint32_t size, uint32_t id){
"%" PRIdS " and asked for %u",
ssh_string_len(datastring), size);
ssh_string_free(datastring);
- sftp_leave_function();
return SSH_ERROR;
}
len = ssh_string_len(datastring);
@@ -1949,16 +1887,13 @@ int sftp_async_read(sftp_file file, void *data, uint32_t size, uint32_t id){
file->offset = file->offset - (size - len);
memcpy(data, ssh_string_data(datastring), len);
ssh_string_free(datastring);
- sftp_leave_function();
return len;
default:
ssh_set_error(sftp->session,SSH_FATAL,"Received message %d during read!",msg->packet_type);
sftp_message_free(msg);
- sftp_leave_function();
return SSH_ERROR;
}
- sftp_leave_function();
return SSH_ERROR;
}