commit 298a2d6109acbaec2b97ad6ff2ec2fe8f553b1f5
author Gabor Mezei <gabor.mezei@arm.com> Mon Feb 07 18:21:39 2022 +0100
committer Gabor Mezei <gabor.mezei@arm.com> Mon Feb 21 15:43:52 2022 +0100
tree 3bf150bcad51ddae4b488a8c1ee38bff7decc675
parent 89c1a95f8f6f7281e9223fd0b59f8663f270658c

Use ASSERT_ALLOC

Change the calloc functions to ASSERT_ALLOC to check the
return value of calloc as well.

Signed-off-by: Gabor Mezei <gabor.mezei@arm.com>

tests/suites/test_suite_hkdf.function

diff --git a/tests/suites/test_suite_hkdf.function b/tests/suites/test_suite_hkdf.function
index feb1717..1e112ba 100644
--- a/tests/suites/test_suite_hkdf.function
+++ b/tests/suites/test_suite_hkdf.function
@@ -44,7 +44,7 @@
     TEST_ASSERT( md != NULL );
 
     output_prk_len = mbedtls_md_get_size( md );
-    output_prk = mbedtls_calloc( 1, output_prk_len );
+    ASSERT_ALLOC( output_prk, output_prk_len );
 
     ikm = mbedtls_test_unhexify_alloc( hex_ikm_string, &ikm_len );
     salt = mbedtls_test_unhexify_alloc( hex_salt_string, &salt_len );
@@ -113,7 +113,7 @@
     fake_md_info.type = MBEDTLS_MD_NONE;
     fake_md_info.size = hash_len;
 
-    prk = mbedtls_calloc( MBEDTLS_MD_MAX_SIZE, 1 );
+    ASSERT_ALLOC( prk, MBEDTLS_MD_MAX_SIZE);
     salt_len = 0;
     ikm_len = 0;