commit 258ae07fb01909a5582001f4d29f4cf2eb5b7265
author gabor-mezei-arm <gabor.mezei@arm.com> Fri Jun 25 15:25:38 2021 +0200
committer gabor-mezei-arm <gabor.mezei@arm.com> Tue Jun 29 19:49:00 2021 +0200
tree eab274f11f7e85b7fcaf16b8ced628fca2c1f7c0
parent e5ff8f430c4747801dcedce6558fab0a57ddfdfc

Add checks for buffer size

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 ed43085..4d5bf20 100644
--- a/library/psa_crypto_cipher.c
+++ b/library/psa_crypto_cipher.c
@@ -532,6 +532,12 @@
     if( status != PSA_SUCCESS )
         goto exit;
 
+    if( output_size < accumulated_length )
+    {
+        status = PSA_ERROR_BUFFER_TOO_SMALL;
+        goto exit;
+    }
+
     status = cipher_finish( &operation, output + accumulated_length,
                             output_size - accumulated_length, &olength );
     accumulated_length += olength;