commit 732ca3221d907c00e4a8d5c974400cc927a30605
author Arto Kinnunen <arto.kinnunen@arm.com> Fri Apr 14 14:26:10 2023 +0800
committer Yanray Wang <yanray.wang@arm.com> Fri May 05 11:20:38 2023 +0800
tree cd51c0850542e9d03e3b93fff3ac324b6c60e58c
parent ca4ca9a2f8cbe40a2eaae6b6bfe8c95c8a068c81

AES: add macro of MBEDTLS_AES_ONLY_128_BIT_KEY_LENGTH

Add configuration option to support 128-bit key length only
in AES calculation.

Signed-off-by: Yanray Wang <yanray.wang@arm.com>

library/aes.c

diff --git a/library/aes.c b/library/aes.c
index 69da582..5493c31 100644
--- a/library/aes.c
+++ b/library/aes.c
@@ -563,8 +563,10 @@
 
     switch (keybits) {
         case 128: ctx->nr = 10; break;
+#if !defined(MBEDTLS_AES_ONLY_128_BIT_KEY_LENGTH)
         case 192: ctx->nr = 12; break;
         case 256: ctx->nr = 14; break;
+#endif /* !MBEDTLS_AES_ONLY_128_BIT_KEY_LENGTH */
         default: return MBEDTLS_ERR_AES_INVALID_KEY_LENGTH;
     }
 
@@ -610,6 +612,7 @@
             }
             break;
 
+#if !defined(MBEDTLS_AES_ONLY_128_BIT_KEY_LENGTH)
         case 12:
 
             for (i = 0; i < 8; i++, RK += 6) {
@@ -651,6 +654,7 @@
                 RK[15] = RK[7] ^ RK[14];
             }
             break;
+#endif /* !MBEDTLS_AES_ONLY_128_BIT_KEY_LENGTH */
     }
 
     return 0;