From 891539af6cbdbdfabb8d37fb491f7f8cadf9823c Mon Sep 17 00:00:00 2001 From: Andreas Schneider Date: Wed, 1 Apr 2009 21:24:16 +0000 Subject: Add memory error checks for crypto wrapper functions. git-svn-id: svn+ssh://svn.berlios.de/svnroot/repos/libssh/trunk@330 7dcaeef0-15fb-0310-b436-a5af3365683c --- libssh/server.c | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) (limited to 'libssh/server.c') diff --git a/libssh/server.c b/libssh/server.c index 65903de..6cc0087 100644 --- a/libssh/server.c +++ b/libssh/server.c @@ -307,7 +307,10 @@ static int dh_handshake_server(SSH_SESSION *session){ crypto_free(session->current_crypto); /* XXX later, include a function to change keys */ session->current_crypto=session->next_crypto; - session->next_crypto=crypto_new(); + session->next_crypto = crypto_new(); + if (session->next_crypto == NULL) { + return -1; + } return 0; } /* do the banner and key exchange */ -- cgit