commit 1aca2605713e3242f72297932409dd58ecfb782f
author Manuel Pégourié-Gonnard <manuel.pegourie-gonnard@arm.com> Wed Dec 12 12:56:55 2018 +0100
committer Manuel Pégourié-Gonnard <manuel.pegourie-gonnard@arm.com> Tue Dec 18 12:22:40 2018 +0100
tree 60836a74ca52c0d64b4f107cab032b6c9bdca79f
parent 68e3dff3f1924f308610d32253eb34ccb78ba8be

Add parameter validation for mbedtls_aes_crypt_ecb()

library/aes.c

diff --git a/library/aes.c b/library/aes.c
index 4d9a56a..9f20744 100644
--- a/library/aes.c
+++ b/library/aes.c
@@ -1006,6 +1006,12 @@
                     const unsigned char input[16],
                     unsigned char output[16] )
 {
+    AES_VALIDATE_RET( ctx != NULL );
+    AES_VALIDATE_RET( input != NULL );
+    AES_VALIDATE_RET( output != NULL );
+    AES_VALIDATE_RET( mode == MBEDTLS_AES_ENCRYPT ||
+                      mode == MBEDTLS_AES_DECRYPT );
+
 #if defined(MBEDTLS_AESNI_C) && defined(MBEDTLS_HAVE_X86_64)
     if( mbedtls_aesni_has_support( MBEDTLS_AESNI_AES ) )
         return( mbedtls_aesni_crypt_ecb( ctx, mode, input, output ) );