From e320b899f504428429f854c920371f971b3038bd Mon Sep 17 00:00:00 2001 From: Harry Ramsey Date: Mon, 11 Nov 2024 15:02:26 +0000 Subject: [PATCH] Fix TEST_CALLOC issues with GCM buffer overlap tests This commit fixes issues with TEST_CALLOC in GCM buffer overlap tests cases. Signed-off-by: Harry Ramsey --- tests/suites/test_suite_gcm.function | 16 ++-------------- 1 file changed, 2 insertions(+), 14 deletions(-) diff --git a/tests/suites/test_suite_gcm.function b/tests/suites/test_suite_gcm.function index 107cc8d5d2..e7b024ec74 100644 --- a/tests/suites/test_suite_gcm.function +++ b/tests/suites/test_suite_gcm.function @@ -631,10 +631,6 @@ void gcm_encrypt_input_output_buffer_overlap(int cipher_id, data_t *key_str, buffer_len += (16 - (buffer_len % 16)); } TEST_CALLOC(buffer, buffer_len); - if (buffer == NULL) { - TEST_FAIL("Could not allocate memory for input output buffer overlap"); - goto exit; - } memcpy(buffer, src_str->x, src_str->len); memset(tag_output, 0x00, 16); @@ -662,9 +658,7 @@ void gcm_encrypt_input_output_buffer_overlap(int cipher_id, data_t *key_str, } exit: - if (buffer != NULL) { - free(buffer); - } + mbedtls_free(buffer); mbedtls_gcm_free(&ctx); BLOCK_CIPHER_PSA_DONE(); } @@ -696,10 +690,6 @@ void gcm_decrypt_input_output_buffer_overlap(int cipher_id, data_t *key_str, buffer_len += (16 - (buffer_len % 16)); } TEST_CALLOC(buffer, buffer_len); - if (buffer == NULL) { - TEST_FAIL("Could not allocate memory for input output buffer overlap"); - goto exit; - } memcpy(buffer, src_str->x, src_str->len); TEST_ASSERT(mbedtls_gcm_setkey(&ctx, cipher_id, key_str->x, key_str->len * 8) == init_result); @@ -736,9 +726,7 @@ void gcm_decrypt_input_output_buffer_overlap(int cipher_id, data_t *key_str, } exit: - if (buffer != NULL) { - free(buffer); - } + mbedtls_free(buffer); mbedtls_gcm_free(&ctx); BLOCK_CIPHER_PSA_DONE();