Fix AesSivCipher heap hint on cleanup.

wolfSSL · Oct 8, 2024 · e4f4274 · e4f4274
1 parent 4753e1c
commit e4f4274
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/wolfcrypt/src/aes.c b/wolfcrypt/src/aes.c
@@ -14033,6 +14033,13 @@ static WARN_UNUSED_RESULT int AesSivCipher(
         }
     }
 
+#ifndef WOLFSSL_SMALL_STACK
+    /* make aes has heap hint and isAllocated initialized for cleanup below */
+    if (ret != 0) {
+        XMEMSET(aes, 0, sizeof(Aes));
+    }
+#endif
+
     if (ret == 0 && dataSz > 0) {
         sivTmp[12] &= 0x7f;
         sivTmp[8] &= 0x7f;
@@ -14066,7 +14073,6 @@ static WARN_UNUSED_RESULT int AesSivCipher(
     if (aes != NULL)
 #endif
     {
-        aes->isAllocated = 0;
         wc_AesFree(aes);
     #ifdef WOLFSSL_SMALL_STACK
         XFREE(aes, NULL, DYNAMIC_TYPE_AES);