From 69c10a41c72d00dc1e1b3c7a938a52e13eeafad8 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Manuel=20P=C3=A9gouri=C3=A9-Gonnard?= Date: Tue, 6 Jul 2021 12:05:23 +0200 Subject: [PATCH] Fix memory leak on failure path in test code MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: Manuel Pégourié-Gonnard --- programs/ssl/ssl_test_common_source.c | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/programs/ssl/ssl_test_common_source.c b/programs/ssl/ssl_test_common_source.c index d30c36ea56..7b85c7633f 100644 --- a/programs/ssl/ssl_test_common_source.c +++ b/programs/ssl/ssl_test_common_source.c @@ -167,7 +167,8 @@ int ssl_check_record( mbedtls_ssl_context const *ssl, if( ret != ret_repeated ) { mbedtls_printf( "mbedtls_ssl_check_record() returned inconsistent results.\n" ); - return( -1 ); + ret = -1; + goto cleanup; } switch( ret ) @@ -178,29 +179,34 @@ int ssl_check_record( mbedtls_ssl_context const *ssl, case MBEDTLS_ERR_SSL_INVALID_RECORD: if( opt.debug_level > 1 ) mbedtls_printf( "mbedtls_ssl_check_record() detected invalid record.\n" ); + ret = 0; break; case MBEDTLS_ERR_SSL_INVALID_MAC: if( opt.debug_level > 1 ) mbedtls_printf( "mbedtls_ssl_check_record() detected unauthentic record.\n" ); + ret = 0; break; case MBEDTLS_ERR_SSL_UNEXPECTED_RECORD: if( opt.debug_level > 1 ) mbedtls_printf( "mbedtls_ssl_check_record() detected unexpected record.\n" ); + ret = 0; break; default: mbedtls_printf( "mbedtls_ssl_check_record() failed fatally with -%#04x.\n", (unsigned int) -ret ); - return( -1 ); + ret = -1; + goto cleanup; } /* Regardless of the outcome, forward the record to the stack. */ } +cleanup: mbedtls_free( tmp_buf ); - return( 0 ); + return( ret ); } int recv_cb( void *ctx, unsigned char *buf, size_t len )