From 26aaf473db995db21ad2b6a8c887033c1e3ccb57 Mon Sep 17 00:00:00 2001
From: David Garske <david@wolfssl.com>
Date: Mon, 10 Aug 2020 12:40:29 -0700
Subject: [PATCH] Fix for `unit.test` error with RSA 1024-bit key gen when
 using `./configure --enable-keygen --enable-sp`. Issue started in PR #3119

---
 wolfcrypt/src/rsa.c | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/wolfcrypt/src/rsa.c b/wolfcrypt/src/rsa.c
index cb0c8a756..36b0e30ec 100644
--- a/wolfcrypt/src/rsa.c
+++ b/wolfcrypt/src/rsa.c
@@ -651,10 +651,16 @@ int wc_CheckRsaKey(RsaKey* key)
                 break;
     #endif /* WOLFSSL_SP_4096 */
                 default:
+                /* If using only single prcsision math then issue key size error, 
+                    otherwise fall-back to multi-precision math calculation */
+                #ifdef WOLFSSL_SP_MATH
                     ret = WC_KEY_SIZE_E;
+                #endif
+                    break;
         }
     }
 #endif /* WOLFSSL_HAVE_SP_RSA */
+
 #ifndef WOLFSSL_SP_MATH
     if (ret == 0) {
         if (mp_exptmod(k, &key->e, &key->n, tmp) != MP_OKAY)
@@ -764,8 +770,8 @@ int wc_CheckRsaKey(RsaKey* key)
 
     return ret;
 }
-#endif
-#endif
+#endif /* WOLFSSL_KEY_GEN && !WOLFSSL_NO_RSA_KEY_CHECK */
+#endif /* WOLFSSL_RSA_PUBLIC_ONLY */
 
 
 #if !defined(WC_NO_RSA_OAEP) || defined(WC_RSA_PSS)