Fix memory leak in handshake

Make sure peer dilithium keys are properly freed. Signed-off-by: Tobias Frauenschläger <[email protected]>
wolfSSL · Feb 26, 2025 · 75d6307 · 75d6307
1 parent 491e70b
commit 75d6307
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/internal.c b/src/internal.c
@@ -8906,6 +8906,10 @@ void FreeHandshakeResources(WOLFSSL* ssl)
         FreeKey(ssl, DYNAMIC_TYPE_FALCON, (void**)&ssl->peerFalconKey);
         ssl->peerFalconKeyPresent = 0;
 #endif /* HAVE_FALCON */
+#if defined(HAVE_DILITHIUM)
+        FreeKey(ssl, DYNAMIC_TYPE_DILITHIUM, (void**)&ssl->peerDilithiumKey);
+        ssl->peerDilithiumKeyPresent = 0;
+#endif /* HAVE_DILITHIUM */
     }
 
 #ifdef HAVE_ECC