From f3546b50fd2a95dbf0a4db7f5c24ae044342ac60 Mon Sep 17 00:00:00 2001
From: Anthony Hu <anthony@wolfssl.com>
Date: Wed, 23 Nov 2022 17:56:56 +0000
Subject: [PATCH] Conform to pre-existing pattern.

---
 src/internal.c | 13 +++++++------
 1 file changed, 7 insertions(+), 6 deletions(-)

diff --git a/src/internal.c b/src/internal.c
index 744175c18..feb30d93b 100644
--- a/src/internal.c
+++ b/src/internal.c
@@ -5833,17 +5833,18 @@ int DhGenKeyPair(WOLFSSL* ssl, DhKey* dhKey,
 #endif
 
 #if defined(HAVE_PK_CALLBACKS)
+    ret = NOT_COMPILED_IN;
     if (ssl && ssl->ctx && ssl->ctx->DhGenerateKeyPairCb) {
         ret = ssl->ctx->DhGenerateKeyPairCb(dhKey, ssl->rng, priv, privSz,
                                             pub, pubSz);
-        if (ret != NOT_COMPILED_IN)
-            return ret;
     }
+    if (ret == NOT_COMPILED_IN)
 #endif
-
-    PRIVATE_KEY_UNLOCK();
-    ret = wc_DhGenerateKeyPair(dhKey, ssl->rng, priv, privSz, pub, pubSz);
-    PRIVATE_KEY_LOCK();
+    {
+        PRIVATE_KEY_UNLOCK();
+        ret = wc_DhGenerateKeyPair(dhKey, ssl->rng, priv, privSz, pub, pubSz);
+        PRIVATE_KEY_LOCK();
+    }
 
     /* Handle async pending response */
 #ifdef WOLFSSL_ASYNC_CRYPT