Merge pull request #7721 from SparkiDev/dilithium_fix_3

Dilithium: fixes
wolfSSL · Jul 10, 2024 · 2543674 · 2543674
2 parents 00e4215 + d1e26b4
commit 2543674
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/tests/api.c b/tests/api.c
@@ -34565,6 +34565,7 @@ static int test_wc_dilithium_make_key_from_seed(void)
     ExpectIntEQ(XMEMCMP(key->k, sk_87, sizeof(sk_87)), 0);
 #endif
 
+    wc_dilithium_free(key);
     XFREE(key, NULL, DYNAMIC_TYPE_TMP_BUFFER);
 #endif
     return EXPECT_RESULT();
@@ -36710,6 +36711,7 @@ static int test_wc_dilithium_verify_kats(void)
     ExpectIntEQ(res, 1);
 #endif
 
+    wc_dilithium_free(key);
     XFREE(key, NULL, DYNAMIC_TYPE_TMP_BUFFER);
 #endif
     return EXPECT_RESULT();

diff --git a/wolfcrypt/src/dilithium.c b/wolfcrypt/src/dilithium.c
@@ -2825,8 +2825,7 @@ static int dilithium_check_low(const sword32* a, sword32 hi)
     return ret;
 }
 
-#if (!defined(WOLFSSL_DILITHIUM_NO_VERIFY) && \
-     !defined(WOLFSSL_DILITHIUM_VERIFY_SMALL_MEM)) || \
+#if !defined(WOLFSSL_DILITHIUM_NO_VERIFY) || \
     (!defined(WOLFSSL_DILITHIUM_NO_SIGN) && \
      !defined(WOLFSSL_DILITHIUM_SIGN_SMALL_MEM))
 /* Check that the values of the vector are in range.