commit 6158e283cc5a171b46bdbef88f5b01f66c4ec2f8
author gabor-mezei-arm <gabor.mezei@arm.com> Tue Jun 29 16:42:13 2021 +0200
committer gabor-mezei-arm <gabor.mezei@arm.com> Tue Jun 29 19:55:24 2021 +0200
tree c36b8227c49cb07f790fa2ec5adfb976d4a296f6
parent 58c17277754fc899a68f7ccd5c262c241b925a5c

Check the return status of the functions first

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

library/psa_crypto_cipher.c

diff --git a/library/psa_crypto_cipher.c b/library/psa_crypto_cipher.c
index 3701f72..6206237 100644
--- a/library/psa_crypto_cipher.c
+++ b/library/psa_crypto_cipher.c
@@ -482,16 +482,17 @@
                             output + operation.iv_length,
                             output_size - operation.iv_length,
                             &olength );
-    accumulated_length += olength;
     if( status != PSA_SUCCESS )
         goto exit;
 
+    accumulated_length += olength;
+
     status = cipher_finish( &operation, output + accumulated_length,
                             output_size - accumulated_length, &olength );
-    accumulated_length += olength;
     if( status != PSA_SUCCESS )
         goto exit;
 
+    accumulated_length += olength;
     *output_length = accumulated_length;
 
 exit:
@@ -531,17 +532,17 @@
     status = cipher_update( &operation, input + operation.iv_length,
                             input_length - operation.iv_length,
                             output, output_size, &olength );
-    accumulated_length = olength;
     if( status != PSA_SUCCESS )
         goto exit;
 
+    accumulated_length = olength;
 
     status = cipher_finish( &operation, output + accumulated_length,
                             output_size - accumulated_length, &olength );
-    accumulated_length += olength;
     if( status != PSA_SUCCESS )
         goto exit;
 
+    accumulated_length += olength;
     *output_length = accumulated_length;
 
 exit: