From 83cdd1c314af5d6326470254b3cdbb9d85182348 Mon Sep 17 00:00:00 2001
From: Takashi Kojo <kojo@wolfssl.com>
Date: Tue, 22 Sep 2020 07:30:21 +0900
Subject: [PATCH] fix NO_DH guard

---
 tests/api.c         | 2 +-
 wolfcrypt/src/evp.c | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/api.c b/tests/api.c
index d18334ef9..9c4d79966 100644
--- a/tests/api.c
+++ b/tests/api.c
@@ -32844,7 +32844,7 @@ static void test_wolfSSL_OCSP_get0_info()
 
 static void test_wolfSSL_EVP_PKEY_derive(void)
 {
-#ifdef OPENSSL_ALL
+#if defined(OPENSSL_ALL) && !defined(NO_DH)
     printf(testingFmt, "wolfSSL_EVP_PKEY_derive()");
 #if !defined(HAVE_FIPS) || (defined(HAVE_FIPS_VERSION) && (HAVE_FIPS_VERSION>2))
     EVP_PKEY_CTX *ctx;
diff --git a/wolfcrypt/src/evp.c b/wolfcrypt/src/evp.c
index 6df054318..c34d92327 100644
--- a/wolfcrypt/src/evp.c
+++ b/wolfcrypt/src/evp.c
@@ -6096,7 +6096,7 @@ int wolfSSL_EVP_PKEY_assign(WOLFSSL_EVP_PKEY *pkey, int type, void *key)
             ret = wolfSSL_EVP_PKEY_assign_EC_KEY(pkey, (WOLFSSL_EC_KEY*)key);
             break;
     #endif
-    #ifdef NO_DH
+    #ifndef NO_DH
          case EVP_PKEY_DH:
             ret = wolfSSL_EVP_PKEY_assign_DH(pkey, (WOLFSSL_DH*)key);
             break;