From a2f802199d5ab217015b54dd9574c5d8bd73a2da Mon Sep 17 00:00:00 2001
From: John Safranek <john@wolfssl.com>
Date: Thu, 4 Mar 2021 14:10:48 -0800
Subject: [PATCH] DH key gen should call DH check key.

---
 wolfcrypt/src/dh.c | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/wolfcrypt/src/dh.c b/wolfcrypt/src/dh.c
index ac02125d1..d023164ad 100644
--- a/wolfcrypt/src/dh.c
+++ b/wolfcrypt/src/dh.c
@@ -1864,6 +1864,10 @@ int wc_DhGenerateKeyPair(DhKey* key, WC_RNG* rng,
     }
 #endif /* WOLFSSL_KCAPI_DH */
 
+    if (ret == 0) {
+        ret = wc_DhCheckKeyPair(key, pub, *pubSz, priv, *privSz);
+    }
+
     return ret;
 }