From b6abf4596e246c0ba0e107a54162269ad5d02fae Mon Sep 17 00:00:00 2001
From: Takashi Kojo <kojo@wolfsssl.com>
Date: Thu, 29 Jun 2017 11:34:26 +0900
Subject: [PATCH] SetDhInternal in DH_compute_key

---
 src/ssl.c | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/ssl.c b/src/ssl.c
index 531be8bdc..ce056480c 100644
--- a/src/ssl.c
+++ b/src/ssl.c
@@ -22695,7 +22695,9 @@ int wolfSSL_DH_compute_key(unsigned char* key, WOLFSSL_BIGNUM* otherPub,
     else {
         privSz = wolfSSL_BN_bn2bin(dh->priv_key, priv);
         pubSz  = wolfSSL_BN_bn2bin(otherPub, pub);
-
+        if (dh->inSet == 0 && SetDhInternal(dh) != SSL_SUCCESS)
+                WOLFSSL_MSG("Bad DH set internal");
+                
         if (privSz <= 0 || pubSz <= 0)
             WOLFSSL_MSG("Bad BN2bin set");
         else if (wc_DhAgree((DhKey*)dh->internal, key, &keySz,