1 files changed, 3 insertions, 1 deletions
diff --git a/libssh/client.c b/libssh/client.c
index cfcaf96..57cbd18 100644
--- a/libssh/client.c
+++ b/libssh/client.c
@@ -201,7 +201,9 @@ static int dh_handshake(SSH_SESSION *session) {
         goto error;
       }
 
-      dh_generate_x(session);
+      if (dh_generate_x(session) < 0) {
+        goto error;
+      }
       dh_generate_e(session);
 
       e = dh_get_e(session);