commit 7c7fb877c655d3ee1655a03e46a434ab5a0db42d
author Przemyslaw Stekiel <przemyslaw.stekiel@mobica.com> Tue Oct 26 14:32:10 2021 +0200
committer Przemyslaw Stekiel <przemyslaw.stekiel@mobica.com> Sun Nov 14 20:03:23 2021 +0100
tree a29c5e9685a26e7ff56e6f199658fe5198a6277a
parent 2290afc2d4419c34fd58f0894670d01cb756ac3c

ssl_client2, ssl_server2: add check for psa memory leaks

Signed-off-by: Przemyslaw Stekiel <przemyslaw.stekiel@mobica.com>

programs/ssl/ssl_client2.c

diff --git a/programs/ssl/ssl_client2.c b/programs/ssl/ssl_client2.c
index 3937981..3cad338 100644
--- a/programs/ssl/ssl_client2.c
+++ b/programs/ssl/ssl_client2.c
@@ -19,6 +19,11 @@
 
 #include "ssl_test_lib.h"
 
+#if defined(MBEDTLS_USE_PSA_CRYPTO)
+#define SKIP_LIBRARY_HEADERS
+#include "test/psa_crypto_helpers.h"
+#endif
+
 #if defined(MBEDTLS_SSL_TEST_IMPOSSIBLE)
 int main( void )
 {
@@ -3041,7 +3046,15 @@
 #endif
 
 #if defined(MBEDTLS_USE_PSA_CRYPTO)
+
     mbedtls_psa_crypto_free( );
+    const char* message = mbedtls_test_helper_is_psa_leaking();
+    if( message )
+    {
+        if( ret == 0 )
+            ret = 1;
+        mbedtls_printf( "PSA memory leak detected: %s\n",  message);
+    }
 #endif
 
 #if defined(MBEDTLS_TEST_HOOKS)