commit 42373bdde7829b90eeead41e3e8ef4f255617589
author gabor-mezei-arm <gabor.mezei@arm.com> Tue Jun 29 16:41:25 2021 +0200
committer gabor-mezei-arm <gabor.mezei@arm.com> Tue Jun 29 16:49:51 2021 +0200
tree 0cc3c3b0d7e11e021b345d681da46036fcf96e10
parent 6f8d43df200c5b09134dedfe3fceafbef3e00fa5

Add buffer overflow check

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 8e63898..66e0426 100644
--- a/library/psa_crypto_cipher.c
+++ b/library/psa_crypto_cipher.c
@@ -393,6 +393,9 @@
         status = mbedtls_to_psa_error(
             mbedtls_cipher_update( &operation->ctx.cipher, input,
                                    input_length, output, output_length ) );
+
+        if( *output_length > output_size )
+            return PSA_ERROR_CORRUPTION_DETECTED;
     }
 
     return( status );