commit af4ecddd4febb8c241e23cea0129911892816cd0
author Mateusz Starzyk <mateusz.starzyk@mobica.com> Tue Jun 15 15:29:48 2021 +0200
committer Mateusz Starzyk <mateusz.starzyk@mobica.com> Wed Jun 16 14:34:09 2021 +0200
tree d0f4e656d07782382630eedb163d1db6127e91cd
parent 3d0bbeef0c15830177e13039ee433a5aae0242ee

Pass associated data split as check_multipart argument.

Signed-off-by: Mateusz Starzyk <mateusz.starzyk@mobica.com>

tests/suites/test_suite_gcm.function

diff --git a/tests/suites/test_suite_gcm.function b/tests/suites/test_suite_gcm.function
index c7942f8..fc01b40 100644
--- a/tests/suites/test_suite_gcm.function
+++ b/tests/suites/test_suite_gcm.function
@@ -11,12 +11,12 @@
                             const data_t *input,
                             const data_t *expected_output,
                             const data_t *tag,
-                            size_t n1 )
+                            size_t n1,
+                            size_t n1_add)
 {
     int ok = 0;
     uint8_t *output = NULL;
     size_t n2 = input->len - n1;
-    size_t n1_add = n1 < add->len ? add->len - n1 : add->len;
     size_t n2_add = add->len - n1_add;
     size_t olen;
 
@@ -105,6 +105,7 @@
     mbedtls_gcm_context ctx;
     size_t tag_len = tag_len_bits / 8;
     size_t n1;
+    size_t n1_add;
 
     mbedtls_gcm_init( &ctx );
 
@@ -123,11 +124,16 @@
         for( n1 = 0; n1 <= src_str->len; n1 += 1 )
         {
             mbedtls_test_set_step( n1 );
-            if( !check_multipart( &ctx, MBEDTLS_GCM_ENCRYPT,
-                                  iv_str, add_str, src_str,
-                                  dst, tag,
-                                  n1 ) )
-                goto exit;
+
+            for( n1_add = 0; n1_add <= add_str->len; n1_add += 1 )
+            {
+                mbedtls_test_set_step( n1_add );
+                if( !check_multipart( &ctx, MBEDTLS_GCM_ENCRYPT,
+                                    iv_str, add_str, src_str,
+                                    dst, tag,
+                                    n1, n1_add ) )
+                    goto exit;
+            }
         }
     }
 
@@ -148,6 +154,7 @@
     int ret;
     size_t tag_len = tag_len_bits / 8;
     size_t n1;
+    size_t n1_add;
 
     mbedtls_gcm_init( &ctx );
 
@@ -171,11 +178,15 @@
             for( n1 = 0; n1 <= src_str->len; n1 += 1 )
             {
                 mbedtls_test_set_step( n1 );
-                if( !check_multipart( &ctx, MBEDTLS_GCM_DECRYPT,
-                                      iv_str, add_str, src_str,
-                                      pt_result, tag_str,
-                                      n1 ) )
-                    goto exit;
+                for( n1_add = 0; n1_add <= add_str->len; n1_add += 1 )
+                {
+                    mbedtls_test_set_step( n1_add );
+                    if( !check_multipart( &ctx, MBEDTLS_GCM_DECRYPT,
+                                        iv_str, add_str, src_str,
+                                        pt_result, tag_str,
+                                        n1, n1_add ) )
+                        goto exit;
+                }
             }
         }
     }