commit 20d2e06ca48f4babeb7308ba5637e32ee8c4bcd2
author Raef Coles <raef.coles@arm.com> Thu Oct 13 09:28:18 2022 +0100
committer Raef Coles <raef.coles@arm.com> Thu Oct 13 14:30:20 2022 +0100
tree b0cd26ddc031f0bb4a3e5787dfbaf3697161e8d7
parent 781f7bedb0a3c9d7d99c5d71a9dfac61966249d6

Add cleanup frees in LMS and LMOTS tests

Signed-off-by: Raef Coles <raef.coles@arm.com>

tests/suites/test_suite_lmots.function

diff --git a/tests/suites/test_suite_lmots.function b/tests/suites/test_suite_lmots.function
index c582ce7..1fec900 100644
--- a/tests/suites/test_suite_lmots.function
+++ b/tests/suites/test_suite_lmots.function
@@ -142,6 +142,8 @@
     }
 
 exit:
+    if( tmp_sig != NULL )
+        mbedtls_free( tmp_sig );
     mbedtls_lmots_public_free( &ctx );
 }
 /* END_CASE */

tests/suites/test_suite_lms.function

diff --git a/tests/suites/test_suite_lms.function b/tests/suites/test_suite_lms.function
index c0cf02c..b6ae309 100644
--- a/tests/suites/test_suite_lms.function
+++ b/tests/suites/test_suite_lms.function
@@ -138,6 +138,8 @@
     }
 
 exit:
+    if( tmp_sig != NULL )
+        mbedtls_free( tmp_sig );
     mbedtls_lms_public_free( &ctx );
 }
 /* END_CASE */
@@ -180,6 +182,8 @@
     }
 
 exit:
+    if( exported_pub_key != NULL )
+        mbedtls_free( exported_pub_key );
     mbedtls_lms_public_free( &ctx );
 }
 /* END_CASE */