From ea5374c62d19c2a7b67648eede844b1ed464e276 Mon Sep 17 00:00:00 2001
From: Marco Oliverio <marco@wolfssl.com>
Date: Wed, 29 Dec 2021 14:43:12 +0100
Subject: [PATCH] wolfcrypt: aes: gcm: streaming api: add missing wc_AesFree()

---
 wolfcrypt/src/aes.c | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/wolfcrypt/src/aes.c b/wolfcrypt/src/aes.c
index cf013f6d2..748e90ee0 100644
--- a/wolfcrypt/src/aes.c
+++ b/wolfcrypt/src/aes.c
@@ -9415,6 +9415,10 @@ int wc_AesGcmDecryptFinal(Aes* aes, const byte* authTag, word32 authTagSz)
         }
     }
 
+    /* reset the state */
+    if (ret == 0)
+        wc_AesFree(aes);
+
     return ret;
 }
 #endif /* HAVE_AES_DECRYPT || HAVE_AESGCM_DECRYPT */
@@ -10384,6 +10388,7 @@ void wc_AesFree(Aes* aes)
     !defined(WOLFSSL_AESNI)
     if (aes->streamData != NULL) {
         XFREE(aes->streamData, aes->heap, DYNAMIC_TYPE_AES);
+        aes->streamData = NULL;
     }
 #endif