1
1

priv: Start to remove enter_function.

Этот коммит содержится в:
Andreas Schneider 2011-09-17 18:31:08 +02:00
родитель d7fa15df83
Коммит 33754c391f
2 изменённых файлов: 40 добавлений и 126 удалений

Просмотреть файл

@ -120,38 +120,39 @@ int gettimeofday(struct timeval *__p, void *__t);
#endif #endif
#endif #endif
#ifdef __cplusplus #define enter_function() (void)session
extern "C" { #define leave_function() (void)session
#endif
#ifdef HAVE_SYS_TIME_H #ifdef HAVE_SYS_TIME_H
#include <sys/time.h> #include <sys/time.h>
#endif #endif
/* forward declarations */
struct ssh_common_struct;
struct ssh_kex_struct;
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);
/* ERROR HANDLING */
/* error handling structure */ /* error handling structure */
struct error_struct { struct error_struct {
int error_code; int error_code;
char error_buffer[ERROR_BUFFERLEN]; char error_buffer[ERROR_BUFFERLEN];
}; };
struct ssh_common_struct;
struct ssh_kex_struct;
/* server data */
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 */
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);
/* errors.c */
#define ssh_set_error(error, code, ...) \ #define ssh_set_error(error, code, ...) \
_ssh_set_error(error, code, __FUNCTION__, __VA_ARGS__) _ssh_set_error(error, code, __FUNCTION__, __VA_ARGS__)
void _ssh_set_error(void *error, 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__) _ssh_set_error_invalid(error, __FUNCTION__)
void _ssh_set_error_invalid(void *error, const char *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 */ /* in crypt.c */
uint32_t packet_decrypt_len(ssh_session session,char *crypted); uint32_t packet_decrypt_len(ssh_session session,char *crypted);
int packet_decrypt(ssh_session session, void *packet,unsigned int len); 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); 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 */ /* server.c */
SSH_PACKET_CALLBACK(ssh_packet_kexdh_init); 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 */ /** Free memory space */
@ -288,9 +271,5 @@ int my_gcry_dec2bn(bignum *bn, const char *data);
char *my_gcry_bn2dec(bignum bn); char *my_gcry_bn2dec(bignum bn);
#endif /* !HAVE_LIBGCRYPT */ #endif /* !HAVE_LIBGCRYPT */
#ifdef __cplusplus
}
#endif
#endif /* _LIBSSH_PRIV_H */ #endif /* _LIBSSH_PRIV_H */
/* vim: set ts=2 sw=2 et cindent: */ /* vim: set ts=4 sw=4 et cindent: */

Просмотреть файл

@ -56,9 +56,6 @@
#ifdef WITH_SFTP #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 { struct sftp_ext_struct {
unsigned int count; unsigned int count;
char **name; char **name;
@ -174,11 +171,8 @@ int sftp_server_init(sftp_session sftp){
ssh_buffer reply = NULL; ssh_buffer reply = NULL;
uint32_t version; uint32_t version;
sftp_enter_function();
packet = sftp_packet_read(sftp); packet = sftp_packet_read(sftp);
if (packet == NULL) { if (packet == NULL) {
sftp_leave_function();
return -1; return -1;
} }
@ -188,7 +182,6 @@ int sftp_server_init(sftp_session sftp){
packet->type); packet->type);
sftp_packet_free(packet); sftp_packet_free(packet);
sftp_leave_function();
return -1; return -1;
} }
@ -204,20 +197,17 @@ int sftp_server_init(sftp_session sftp){
reply = ssh_buffer_new(); reply = ssh_buffer_new();
if (reply == NULL) { if (reply == NULL) {
ssh_set_error_oom(session); ssh_set_error_oom(session);
sftp_leave_function();
return -1; return -1;
} }
if (buffer_add_u32(reply, ntohl(LIBSFTP_VERSION)) < 0) { if (buffer_add_u32(reply, ntohl(LIBSFTP_VERSION)) < 0) {
ssh_set_error_oom(session); ssh_set_error_oom(session);
ssh_buffer_free(reply); ssh_buffer_free(reply);
sftp_leave_function();
return -1; return -1;
} }
if (sftp_packet_write(sftp, SSH_FXP_VERSION, reply) < 0) { if (sftp_packet_write(sftp, SSH_FXP_VERSION, reply) < 0) {
ssh_buffer_free(reply); ssh_buffer_free(reply);
sftp_leave_function();
return -1; return -1;
} }
ssh_buffer_free(reply); ssh_buffer_free(reply);
@ -230,7 +220,6 @@ int sftp_server_init(sftp_session sftp){
sftp->version=version; sftp->version=version;
} }
sftp_leave_function();
return 0; return 0;
} }
#endif /* WITH_SERVER */ #endif /* WITH_SERVER */
@ -293,8 +282,6 @@ sftp_packet sftp_packet_read(sftp_session sftp) {
uint32_t size; uint32_t size;
int r; int r;
sftp_enter_function();
packet = malloc(sizeof(struct sftp_packet_struct)); packet = malloc(sizeof(struct sftp_packet_struct));
if (packet == NULL) { if (packet == NULL) {
ssh_set_error_oom(sftp->session); ssh_set_error_oom(sftp->session);
@ -312,7 +299,6 @@ sftp_packet sftp_packet_read(sftp_session sftp) {
if (r < 0) { if (r < 0) {
ssh_buffer_free(packet->payload); ssh_buffer_free(packet->payload);
SAFE_FREE(packet); SAFE_FREE(packet);
sftp_leave_function();
return NULL; return NULL;
} }
buffer_add_data(packet->payload,buffer, r); 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_set_error(sftp->session, SSH_FATAL, "Short sftp packet!");
ssh_buffer_free(packet->payload); ssh_buffer_free(packet->payload);
SAFE_FREE(packet); SAFE_FREE(packet);
sftp_leave_function();
return NULL; 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 */ /* TODO: check if there are cases where an error needs to be set here */
ssh_buffer_free(packet->payload); ssh_buffer_free(packet->payload);
SAFE_FREE(packet); SAFE_FREE(packet);
sftp_leave_function();
return NULL; return NULL;
} }
buffer_add_data(packet->payload, buffer, r); 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 */ /* TODO: check if there are cases where an error needs to be set here */
ssh_buffer_free(packet->payload); ssh_buffer_free(packet->payload);
SAFE_FREE(packet); SAFE_FREE(packet);
sftp_leave_function();
return NULL; return NULL;
} }
if(buffer_add_data(packet->payload,buffer,r)==SSH_ERROR){ if(buffer_add_data(packet->payload,buffer,r)==SSH_ERROR){
ssh_buffer_free(packet->payload); ssh_buffer_free(packet->payload);
SAFE_FREE(packet); SAFE_FREE(packet);
sftp_leave_function();
ssh_set_error_oom(sftp->session); ssh_set_error_oom(sftp->session);
return NULL; return NULL;
} }
size -= r; size -= r;
} }
sftp_leave_function();
return packet; return packet;
} }
@ -379,8 +360,6 @@ int sftp_get_error(sftp_session sftp) {
static sftp_message sftp_message_new(sftp_session sftp){ static sftp_message sftp_message_new(sftp_session sftp){
sftp_message msg = NULL; sftp_message msg = NULL;
sftp_enter_function();
msg = malloc(sizeof(struct sftp_message_struct)); msg = malloc(sizeof(struct sftp_message_struct));
if (msg == NULL) { if (msg == NULL) {
ssh_set_error_oom(sftp->session); ssh_set_error_oom(sftp->session);
@ -396,35 +375,24 @@ static sftp_message sftp_message_new(sftp_session sftp){
} }
msg->sftp = sftp; msg->sftp = sftp;
sftp_leave_function();
return msg; return msg;
} }
static void sftp_message_free(sftp_message msg) { static void sftp_message_free(sftp_message msg) {
sftp_session sftp;
if (msg == NULL) { if (msg == NULL) {
return; return;
} }
sftp = msg->sftp;
sftp_enter_function();
ssh_buffer_free(msg->payload); ssh_buffer_free(msg->payload);
SAFE_FREE(msg); SAFE_FREE(msg);
sftp_leave_function();
} }
static sftp_message sftp_get_message(sftp_packet packet) { static sftp_message sftp_get_message(sftp_packet packet) {
sftp_session sftp = packet->sftp; sftp_session sftp = packet->sftp;
sftp_message msg = NULL; sftp_message msg = NULL;
sftp_enter_function();
msg = sftp_message_new(sftp); msg = sftp_message_new(sftp);
if (msg == NULL) { if (msg == NULL) {
sftp_leave_function();
return NULL; return NULL;
} }
@ -437,7 +405,6 @@ static sftp_message sftp_get_message(sftp_packet packet) {
ssh_set_error(packet->sftp->session, SSH_FATAL, ssh_set_error(packet->sftp->session, SSH_FATAL,
"Unknown packet type %d", packet->type); "Unknown packet type %d", packet->type);
sftp_message_free(msg); sftp_message_free(msg);
sftp_leave_function();
return NULL; return NULL;
} }
@ -445,7 +412,6 @@ static sftp_message sftp_get_message(sftp_packet packet) {
ssh_set_error(packet->sftp->session, SSH_FATAL, ssh_set_error(packet->sftp->session, SSH_FATAL,
"Invalid packet %d: no ID", packet->type); "Invalid packet %d: no ID", packet->type);
sftp_message_free(msg); sftp_message_free(msg);
sftp_leave_function();
return NULL; return NULL;
} }
@ -458,11 +424,9 @@ static sftp_message sftp_get_message(sftp_packet packet) {
buffer_get_rest_len(packet->payload)) < 0) { buffer_get_rest_len(packet->payload)) < 0) {
ssh_set_error_oom(sftp->session); ssh_set_error_oom(sftp->session);
sftp_message_free(msg); sftp_message_free(msg);
sftp_leave_function();
return NULL; return NULL;
} }
sftp_leave_function();
return msg; return msg;
} }
@ -470,28 +434,22 @@ static int sftp_read_and_dispatch(sftp_session sftp) {
sftp_packet packet = NULL; sftp_packet packet = NULL;
sftp_message msg = NULL; sftp_message msg = NULL;
sftp_enter_function();
packet = sftp_packet_read(sftp); packet = sftp_packet_read(sftp);
if (packet == NULL) { if (packet == NULL) {
sftp_leave_function();
return -1; /* something nasty happened reading the packet */ return -1; /* something nasty happened reading the packet */
} }
msg = sftp_get_message(packet); msg = sftp_get_message(packet);
sftp_packet_free(packet); sftp_packet_free(packet);
if (msg == NULL) { if (msg == NULL) {
sftp_leave_function();
return -1; return -1;
} }
if (sftp_enqueue(sftp, msg) < 0) { if (sftp_enqueue(sftp, msg) < 0) {
sftp_message_free(msg); sftp_message_free(msg);
sftp_leave_function();
return -1; return -1;
} }
sftp_leave_function();
return 0; return 0;
} }
@ -514,31 +472,25 @@ int sftp_init(sftp_session sftp) {
char *ext_data = NULL; char *ext_data = NULL;
uint32_t version = htonl(LIBSFTP_VERSION); uint32_t version = htonl(LIBSFTP_VERSION);
sftp_enter_function();
buffer = ssh_buffer_new(); buffer = ssh_buffer_new();
if (buffer == NULL) { if (buffer == NULL) {
ssh_set_error_oom(sftp->session); ssh_set_error_oom(sftp->session);
sftp_leave_function();
return -1; return -1;
} }
if (buffer_add_u32(buffer, version) < 0) { if (buffer_add_u32(buffer, version) < 0) {
ssh_set_error_oom(sftp->session); ssh_set_error_oom(sftp->session);
ssh_buffer_free(buffer); ssh_buffer_free(buffer);
sftp_leave_function();
return -1; return -1;
} }
if (sftp_packet_write(sftp, SSH_FXP_INIT, buffer) < 0) { if (sftp_packet_write(sftp, SSH_FXP_INIT, buffer) < 0) {
ssh_buffer_free(buffer); ssh_buffer_free(buffer);
sftp_leave_function();
return -1; return -1;
} }
ssh_buffer_free(buffer); ssh_buffer_free(buffer);
packet = sftp_packet_read(sftp); packet = sftp_packet_read(sftp);
if (packet == NULL) { if (packet == NULL) {
sftp_leave_function();
return -1; return -1;
} }
@ -546,7 +498,6 @@ int sftp_init(sftp_session sftp) {
ssh_set_error(sftp->session, SSH_FATAL, ssh_set_error(sftp->session, SSH_FATAL,
"Received a %d messages instead of SSH_FXP_VERSION", packet->type); "Received a %d messages instead of SSH_FXP_VERSION", packet->type);
sftp_packet_free(packet); sftp_packet_free(packet);
sftp_leave_function();
return -1; return -1;
} }
@ -619,7 +570,6 @@ int sftp_init(sftp_session sftp) {
sftp->version = sftp->server_version = version; sftp->version = sftp->server_version = version;
sftp_leave_function();
return 0; return 0;
} }
@ -1843,8 +1793,6 @@ int sftp_async_read_begin(sftp_file file, uint32_t len){
ssh_buffer buffer; ssh_buffer buffer;
uint32_t id; uint32_t id;
sftp_enter_function();
buffer = ssh_buffer_new(); buffer = ssh_buffer_new();
if (buffer == NULL) { if (buffer == NULL) {
ssh_set_error_oom(sftp->session); 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 */ file->offset += len; /* assume we'll read len bytes */
sftp_leave_function();
return id; 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; int err = SSH_OK;
uint32_t len; uint32_t len;
sftp_enter_function();
if (file->eof) { if (file->eof) {
sftp_leave_function();
return 0; 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) { if (sftp_read_and_dispatch(sftp) < 0) {
/* something nasty has happened */ /* something nasty has happened */
sftp_leave_function();
return SSH_ERROR; 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); status = parse_status_msg(msg);
sftp_message_free(msg); sftp_message_free(msg);
if (status == NULL) { if (status == NULL) {
sftp_leave_function();
return -1; return -1;
} }
sftp_set_error(sftp, status->status); sftp_set_error(sftp, status->status);
if (status->status != SSH_FX_EOF) { if (status->status != SSH_FX_EOF) {
ssh_set_error(sftp->session, SSH_REQUEST_DENIED, ssh_set_error(sftp->session, SSH_REQUEST_DENIED,
"SFTP server : %s", status->errormsg); "SFTP server : %s", status->errormsg);
sftp_leave_function();
err = SSH_ERROR; err = SSH_ERROR;
} else { } else {
file->eof = 1; file->eof = 1;
} }
status_msg_free(status); status_msg_free(status);
sftp_leave_function();
return err; return err;
case SSH_FXP_DATA: case SSH_FXP_DATA:
datastring = buffer_get_ssh_string(msg->payload); 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) { if (datastring == NULL) {
ssh_set_error(sftp->session, SSH_FATAL, ssh_set_error(sftp->session, SSH_FATAL,
"Received invalid DATA packet from sftp server"); "Received invalid DATA packet from sftp server");
sftp_leave_function();
return SSH_ERROR; return SSH_ERROR;
} }
if (ssh_string_len(datastring) > size) { 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", "%" PRIdS " and asked for %u",
ssh_string_len(datastring), size); ssh_string_len(datastring), size);
ssh_string_free(datastring); ssh_string_free(datastring);
sftp_leave_function();
return SSH_ERROR; return SSH_ERROR;
} }
len = ssh_string_len(datastring); 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); file->offset = file->offset - (size - len);
memcpy(data, ssh_string_data(datastring), len); memcpy(data, ssh_string_data(datastring), len);
ssh_string_free(datastring); ssh_string_free(datastring);
sftp_leave_function();
return len; return len;
default: default:
ssh_set_error(sftp->session,SSH_FATAL,"Received message %d during read!",msg->packet_type); ssh_set_error(sftp->session,SSH_FATAL,"Received message %d during read!",msg->packet_type);
sftp_message_free(msg); sftp_message_free(msg);
sftp_leave_function();
return SSH_ERROR; return SSH_ERROR;
} }
sftp_leave_function();
return SSH_ERROR; return SSH_ERROR;
} }