From 9b81b0b5055e63d0089ff855a9a1fa30dee6a919 Mon Sep 17 00:00:00 2001
From: Andras Fekete <andras@wolfssl.com>
Date: Wed, 9 Aug 2023 10:44:59 -0400
Subject: [PATCH] Add in clamp for mp_sqrtmod_prime

---
 wolfcrypt/src/ecc.c | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/wolfcrypt/src/ecc.c b/wolfcrypt/src/ecc.c
index 32426b9c7..bdbee0bcb 100644
--- a/wolfcrypt/src/ecc.c
+++ b/wolfcrypt/src/ecc.c
@@ -14710,6 +14710,12 @@ int mp_sqrtmod_prime(mp_int* n, mp_int* prime, mp_int* ret)
       /* Z = Z + 1 */
       if (res == MP_OKAY)
         res = mp_add_d(Z, 1, Z);
+
+      if ((res == MP_OKAY) && (mp_cmp(Z,prime) == MP_EQ)) {
+        /* This is to clamp the loop in case 'prime' is not really prime */
+        res = MP_VAL;
+        break;
+      }
     }
 
     /* C = Z ^ Q mod prime */