From 25bb6eef964e680137c3167dc0e0f9dbf2aafb60 Mon Sep 17 00:00:00 2001
From: Anderson Toshiyuki Sasaki <ansasaki@redhat.com>
Date: Fri, 14 Jun 2019 16:40:08 +0200
Subject: [PATCH] dh-gex: Check return code ssh_dh_keypair_get_keys()

Signed-off-by: Anderson Toshiyuki Sasaki <ansasaki@redhat.com>
Reviewed-by: Andreas Schneider <asn@cryptomilk.org>
---
 src/dh-gex.c | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/dh-gex.c b/src/dh-gex.c
index 994a5cbc..b20bf2b6 100644
--- a/src/dh-gex.c
+++ b/src/dh-gex.c
@@ -194,8 +194,13 @@ SSH_PACKET_CALLBACK(ssh_packet_client_dhgex_group)
     if (rc == SSH_ERROR) {
         goto error;
     }
+
     rc = ssh_dh_keypair_get_keys(session->next_crypto->dh_ctx,
                                  DH_CLIENT_KEYPAIR, NULL, &pubkey);
+    if (rc != SSH_OK) {
+        goto error;
+    }
+
     rc = ssh_buffer_pack(session->out_buffer,
                          "bB",
                          SSH2_MSG_KEX_DH_GEX_INIT,