1 files changed, 4 insertions, 1 deletions
diff --git a/libssh/server.c b/libssh/server.c
index 7e3eb448..2809a8ec 100644
--- a/libssh/server.c
+++ b/libssh/server.c
@@ -278,7 +278,10 @@ static int dh_handshake_server(SSH_SESSION *session){
       ssh_set_error(session,SSH_FATAL,"Could not create y number");
       return -1;
     }
-    dh_generate_f(session);
+    if (dh_generate_f(session) < 0) {
+      ssh_set_error(session,SSH_FATAL,"Could not create f number");
+      return -1;
+    }
     f=dh_get_f(session);
     switch(session->hostkeys){
         case TYPE_DSS: