From cf191a4d9647af496e07b7e496437e3073aea737 Mon Sep 17 00:00:00 2001
From: John Safranek <john@wolfssl.com>
Date: Mon, 2 Jul 2018 13:31:13 -0700
Subject: [PATCH] Fixed a memory leak in the wolfCrypt test for DH key
 generation.

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

diff --git a/wolfcrypt/test/test.c b/wolfcrypt/test/test.c
index 4a45faf87..52e49ba5b 100644
--- a/wolfcrypt/test/test.c
+++ b/wolfcrypt/test/test.c
@@ -11325,6 +11325,10 @@ static int dh_fips_generate_test(WC_RNG *rng)
 #endif /* HAVE_SELFTEST */
 
 #ifdef WOLFSSL_KEY_GEN
+    wc_FreeDhKey(&key);
+    ret = wc_InitDhKey_ex(&key, HEAP_HINT, devId);
+    if (ret != 0)
+        return -8231;
 
     ret = wc_DhGenerateParams(rng, 2048, &key);
     if (ret != 0) {