commit 814fffbd72bfa2bcc3bd4716c03ecf9e0110113b
author Paul Elliott <paul.elliott@arm.com> Mon Aug 16 18:20:36 2021 +0100
committer Paul Elliott <paul.elliott@arm.com> Mon Aug 16 18:44:42 2021 +0100
tree 70c18aa8c783fcf014207a0c7f5f297594b67c3d
parent 91b021e4c70e0ac7e9352f97d9c55cb29f4572d0

Remove overly strict final checks

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

library/psa_crypto.c

diff --git a/library/psa_crypto.c b/library/psa_crypto.c
index 95f9740..e40e370 100644
--- a/library/psa_crypto.c
+++ b/library/psa_crypto.c
@@ -3749,11 +3749,8 @@
     return( status );
 }
 
-static psa_status_t psa_aead_final_checks( psa_aead_operation_t *operation,
-                                           size_t output_size )
+static psa_status_t psa_aead_final_checks( psa_aead_operation_t *operation )
 {
-    size_t finish_output_size;
-
     if( operation->id == 0 || !operation->nonce_set )
         return( PSA_ERROR_BAD_STATE );
 
@@ -3761,13 +3758,6 @@
                                    operation->body_remaining != 0 ) )
         return( PSA_ERROR_INVALID_ARGUMENT );
 
-    finish_output_size = operation->is_encrypt ?
-         PSA_AEAD_FINISH_OUTPUT_SIZE( operation->key_type, operation->alg ) :
-         PSA_AEAD_VERIFY_OUTPUT_SIZE( operation->key_type, operation->alg );
-
-    if( output_size < finish_output_size )
-        return( PSA_ERROR_BUFFER_TOO_SMALL );
-
     return( PSA_SUCCESS );
 }
 
@@ -3785,7 +3775,7 @@
     *ciphertext_length = 0;
     *tag_length = tag_size;
 
-    status = psa_aead_final_checks( operation, ciphertext_size );
+    status = psa_aead_final_checks( operation );
 
     if( status != PSA_SUCCESS )
         goto exit;
@@ -3831,7 +3821,7 @@
 
     *plaintext_length = 0;
 
-    status = psa_aead_final_checks( operation, plaintext_size );
+    status = psa_aead_final_checks( operation );
 
     if( status != PSA_SUCCESS )
         goto exit;