commit 719c1324a124dcb7f6744f206e2c930020032629
author Paul Elliott <paul.elliott@arm.com> Mon Sep 13 18:27:22 2021 +0100
committer Paul Elliott <paul.elliott@arm.com> Sun Sep 19 18:43:16 2021 +0100
tree 69a42fe6ca3f158757da75cee6f90302b6991e3c
parent e58cb1e0cf2f517fb407e4808aee6bffd0a0b263

Add tag buffer size tests to finish buffer tests

Signed-off-by: Paul Elliott <paul.elliott@arm.com>

tests/suites/test_suite_psa_crypto.function

diff --git a/tests/suites/test_suite_psa_crypto.function b/tests/suites/test_suite_psa_crypto.function
index 5c27a59..eea0b68 100644
--- a/tests/suites/test_suite_psa_crypto.function
+++ b/tests/suites/test_suite_psa_crypto.function
@@ -3977,6 +3977,7 @@
 void aead_multipart_finish_buffer_test( int key_type_arg, data_t *key_data,
                                         int alg_arg,
                                         int finish_ciphertext_size_arg,
+                                        int tag_size_arg,
                                         data_t *nonce,
                                         data_t *additional_data,
                                         data_t *input_data,
@@ -3992,11 +3993,12 @@
     psa_status_t expected_status = expected_status_arg;
     unsigned char *ciphertext = NULL;
     unsigned char *finish_ciphertext = NULL;
+    unsigned char *tag_buffer = NULL;
     size_t ciphertext_size = 0;
     size_t ciphertext_length = 0;
     size_t finish_ciphertext_size = ( size_t ) finish_ciphertext_size_arg;
+    size_t tag_size = ( size_t ) tag_size_arg;
     size_t tag_length = 0;
-    uint8_t tag_buffer[PSA_AEAD_TAG_MAX_SIZE];
 
     PSA_ASSERT( psa_crypto_init( ) );
 
@@ -4017,6 +4019,8 @@
 
     ASSERT_ALLOC( finish_ciphertext, finish_ciphertext_size );
 
+    ASSERT_ALLOC( tag_buffer, tag_size );
+
     operation = psa_aead_operation_init( );
 
     status = psa_aead_encrypt_setup( &operation, key, alg );
@@ -4044,7 +4048,7 @@
     status = psa_aead_finish( &operation, finish_ciphertext,
                               finish_ciphertext_size,
                               &ciphertext_length, tag_buffer,
-                              PSA_AEAD_TAG_MAX_SIZE, &tag_length );
+                              tag_size, &tag_length );
 
     TEST_EQUAL( status, expected_status );