Merge pull request #8326 from gasbytes/patch-rng-health-test-heap-hint

Fix missing heap hint in RNG health test XMALLOC call
wolfSSL · Jan 27, 2025 · edd8355 · edd8355
2 parents f0fdc72 + d491e54
commit edd8355
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/wolfcrypt/src/random.c b/wolfcrypt/src/random.c
@@ -2184,7 +2184,7 @@ static int wc_RNG_HealthTestLocal(int reseed, void* heap, int devId)
 #endif
 
 #ifdef WOLFSSL_SMALL_STACK
-    check = (byte*)XMALLOC(RNG_HEALTH_TEST_CHECK_SIZE, NULL,
+    check = (byte*)XMALLOC(RNG_HEALTH_TEST_CHECK_SIZE, heap,
                            DYNAMIC_TYPE_TMP_BUFFER);
     if (check == NULL) {
         return MEMORY_E;
@@ -2304,7 +2304,7 @@ static int wc_RNG_HealthTestLocal(int reseed, void* heap, int devId)
     }
 
 #ifdef WOLFSSL_SMALL_STACK
-    XFREE(check, NULL, DYNAMIC_TYPE_TMP_BUFFER);
+    XFREE(check, heap, DYNAMIC_TYPE_TMP_BUFFER);
 #endif
 
     return ret;