From 10397d44db3c7ed17b194cdeaf3ba04e9602957c Mon Sep 17 00:00:00 2001 From: Andreas Schneider Date: Wed, 1 Apr 2009 21:30:53 +0000 Subject: [PATCH] Add one missing memory error check in crypt.c. git-svn-id: svn+ssh://svn.berlios.de/svnroot/repos/libssh/trunk@333 7dcaeef0-15fb-0310-b436-a5af3365683c --- libssh/crypt.c | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/libssh/crypt.c b/libssh/crypt.c index 9d8fa0f2..9148e84a 100644 --- a/libssh/crypt.c +++ b/libssh/crypt.c @@ -88,10 +88,14 @@ unsigned char * packet_encrypt(SSH_SESSION *session,void *data,u32 len){ #elif defined HAVE_LIBCRYPTO crypto->set_encrypt_key(crypto,session->current_crypto->encryptkey); #endif - out=malloc(len); + out = malloc(len); + if (out == NULL) { + return NULL; + } if(session->version==2){ ctx = hmac_init(session->current_crypto->encryptMAC,20,HMAC_SHA1); if (ctx == NULL) { + SAFE_FREE(out); return NULL; } hmac_update(ctx,(unsigned char *)&seq,sizeof(u32));