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/client.c | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) (limited to 'libssh/client.c') diff --git a/libssh/client.c b/libssh/client.c index b79c605..76645e5 100644 --- a/libssh/client.c +++ b/libssh/client.c @@ -213,7 +213,13 @@ static int dh_handshake(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) { + leave_function(); + return SSH_ERROR; + } + session->dh_handshake_state=DH_STATE_FINISHED; leave_function(); return SSH_OK; -- cgit