commit 8bcd7cca94f9b7549cf0780d030582d16959a02b
author Gabor Mezei <gabor.mezei@arm.com> Fri Oct 14 16:18:10 2022 +0200
committer Gabor Mezei <gabor.mezei@arm.com> Fri Oct 14 16:18:10 2022 +0200
tree 15596ce47f22330d910c6ae6dbd7da90e8079532
parent 44ba2f6db919fc0e7008e4eb2c9b546e9c6b78c6

Use count instead of bytes in ASSERT_ALLOC argument

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

tests/suites/test_suite_mpi.function

diff --git a/tests/suites/test_suite_mpi.function b/tests/suites/test_suite_mpi.function
index 6ac36c4..18525ed 100644
--- a/tests/suites/test_suite_mpi.function
+++ b/tests/suites/test_suite_mpi.function
@@ -780,8 +780,8 @@
     TEST_EQUAL( limbs_X, limbs_Y );
     TEST_ASSERT( copy_limbs <= limbs );
 
-    ASSERT_ALLOC( X, len );
-    ASSERT_ALLOC( Y, len );
+    ASSERT_ALLOC( X, limbs );
+    ASSERT_ALLOC( Y, limbs );
 
     TEST_ASSERT( mbedtls_mpi_core_read_be( X, limbs, input_X->x, input_X->len )
                  == 0 );
@@ -844,19 +844,19 @@
     TEST_EQUAL( limbs_X, limbs_Y );
     TEST_ASSERT( copy_limbs <= limbs );
 
-    ASSERT_ALLOC( tmp_X, len );
-    ASSERT_ALLOC( tmp_Y, len );
+    ASSERT_ALLOC( tmp_X, limbs );
+    ASSERT_ALLOC( tmp_Y, limbs );
 
     TEST_ASSERT( mbedtls_mpi_core_read_be( tmp_X, limbs,
                                            input_X->x, input_X->len )
                  == 0 );
-    ASSERT_ALLOC( X, len );
+    ASSERT_ALLOC( X, limbs );
     memcpy( X, tmp_X, len );
 
     TEST_ASSERT( mbedtls_mpi_core_read_be( tmp_Y, limbs,
                                            input_Y->x, input_Y->len )
                  == 0 );
-    ASSERT_ALLOC( Y, len );
+    ASSERT_ALLOC( Y, limbs );
     memcpy( Y, tmp_Y, len );
 
     /* condition is false */
@@ -924,10 +924,10 @@
     TEST_EQUAL( limbs_X, limbs_Y );
     TEST_ASSERT( copy_limbs <= limbs );
 
-    ASSERT_ALLOC( X, len );
-    ASSERT_ALLOC( Y, len );
+    ASSERT_ALLOC( X, limbs );
+    ASSERT_ALLOC( Y, limbs );
 
-    ASSERT_ALLOC( buff_m, len );
+    ASSERT_ALLOC( buff_m, limbs );
     mbedtls_mpi_mod_modulus_init( &m );
     TEST_ASSERT( mbedtls_mpi_mod_modulus_setup(
                         &m, buff_m, copy_limbs,
@@ -1003,10 +1003,10 @@
     TEST_EQUAL( limbs_X, limbs_Y );
     TEST_ASSERT( copy_limbs <= limbs );
 
-    ASSERT_ALLOC( tmp_X, len );
-    ASSERT_ALLOC( tmp_Y, len );
+    ASSERT_ALLOC( tmp_X, limbs );
+    ASSERT_ALLOC( tmp_Y, limbs );
 
-    ASSERT_ALLOC( buff_m, input_len );
+    ASSERT_ALLOC( buff_m, copy_limbs );
     mbedtls_mpi_mod_modulus_init( &m );
     TEST_ASSERT( mbedtls_mpi_mod_modulus_setup(
                         &m, buff_m, copy_limbs,
@@ -1016,7 +1016,7 @@
 
     TEST_ASSERT( mbedtls_mpi_core_read_be( tmp_X, limbs, input_X->x, input_X->len )
                  == 0 );
-    ASSERT_ALLOC( X, len );
+    ASSERT_ALLOC( X, limbs );
     memcpy( X, tmp_X, len );
 
     TEST_ASSERT( mbedtls_mpi_core_read_be( tmp_Y, limbs, input_Y->x, input_Y->len )