1 files changed, 9 insertions, 1 deletions

diff --git a/kexdhc.c b/kexdhc.c
index af259f16a..ad3975f09 100644
--- a/kexdhc.c
+++ b/kexdhc.c
@@ -1,4 +1,4 @@
-/* $OpenBSD: kexdhc.c,v 1.18 2015/01/26 06:10:03 djm Exp $ */
+/* $OpenBSD: kexdhc.c,v 1.19 2016/05/02 10:26:04 djm Exp $ */
 /*
  * Copyright (c) 2001 Markus Friedl.  All rights reserved.
  *
@@ -63,8 +63,15 @@ kexdh_client(struct ssh *ssh)
                 kex->dh = dh_new_group1();
                 break;
         case KEX_DH_GRP14_SHA1:
+        case KEX_DH_GRP14_SHA256:
                 kex->dh = dh_new_group14();
                 break;
+        case KEX_DH_GRP16_SHA512:
+                kex->dh = dh_new_group16();
+                break;
+        case KEX_DH_GRP18_SHA512:
+                kex->dh = dh_new_group18();
+                break;
         default:
                 r = SSH_ERR_INVALID_ARGUMENT;
                 goto out;
@@ -164,6 +171,7 @@ input_kex_dh(int type, u_int32_t seq, void *ctxt)
         /* calc and verify H */
         hashlen = sizeof(hash);
         if ((r = kex_dh_hash(
+            kex->hash_alg,
             kex->client_version_string,
             kex->server_version_string,
             sshbuf_ptr(kex->my), sshbuf_len(kex->my),