From e4fdb63fed3d62d707c91358c3111c01543db61e Mon Sep 17 00:00:00 2001
From: John Safranek <john@wolfssl.com>
Date: Mon, 29 Mar 2021 14:40:40 -0700
Subject: [PATCH] Skip the small key DH test for SP and FFDHE builds.

---
 wolfcrypt/test/test.c | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/wolfcrypt/test/test.c b/wolfcrypt/test/test.c
index 65944246d..18782359d 100644
--- a/wolfcrypt/test/test.c
+++ b/wolfcrypt/test/test.c
@@ -16142,9 +16142,9 @@ static int dh_generate_test(WC_RNG *rng)
 {
     int    ret = 0;
     DhKey  smallKey;
-    byte   p[2] = { 1, 7 }; /* 263 in decimal */
+    byte   p[2] = { 0, 5 };
     byte   g[2] = { 0, 2 };
-#if !defined(WOLFSSL_SP_MATH)
+#if !defined(WOLFSSL_SP_MATH) && !defined(HAVE_FFDHE)
 #ifdef WOLFSSL_DH_CONST
     /* the table for constant DH lookup will round to the lowest byte size 21 */
     byte   priv[21];
@@ -16192,7 +16192,7 @@ static int dh_generate_test(WC_RNG *rng)
         ERROR_OUT(-8017, exit_gen_test);
     }
 
-#if !defined(WOLFSSL_SP_MATH)
+#if !defined(WOLFSSL_SP_MATH) && !defined(HAVE_FFDHE)
     /* Use API. */
     ret = wc_DhGenerateKeyPair(&smallKey, rng, priv, &privSz, pub, &pubSz);
 #if defined(WOLFSSL_ASYNC_CRYPT)