From 5aa9cf2fcf3fc062f4560df52e3d48d894e0ba6e Mon Sep 17 00:00:00 2001 From: Aris Adamantiadis Date: Fri, 23 Jul 2010 08:09:57 +0200 Subject: Add consistency check to ssh_set_callbacks --- libssh/callbacks.c | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) (limited to 'libssh') diff --git a/libssh/callbacks.c b/libssh/callbacks.c index 7553ca6..1568d51 100644 --- a/libssh/callbacks.c +++ b/libssh/callbacks.c @@ -28,10 +28,16 @@ int ssh_set_callbacks(ssh_session session, ssh_callbacks cb) { if (session == NULL || cb == NULL) { - return -1; + return SSH_ERROR; + } + enter_function(); + if(cb->size <= 0 || cb->size > 1024 * sizeof(void *)){ + ssh_set_error(session,SSH_FATAL, + "Invalid callback passed in (badly initialized)"); + leave_function(); + return SSH_ERROR; } - session->callbacks = cb; - + leave_function(); return 0; } -- cgit