1 files changed, 4 insertions, 1 deletions
diff --git a/libssh/client.c b/libssh/client.c
index f8d95d9b..a7e695f4 100644
--- a/libssh/client.c
+++ b/libssh/client.c
@@ -313,7 +313,10 @@ static int dh_handshake(SSH_SESSION *session) {
         goto error;
       }
 
-      generate_session_keys(session);
+      if (generate_session_keys(session) < 0) {
+        rc = SSH_ERROR;
+        goto error;
+      }
 
       /* Verify the host's signature. FIXME do it sooner */
       signature = session->dh_server_signature;