commit 0f06618db043b8867c0a5c9d14c89af19c0b5c5e
author Arto Kinnunen <arto.kinnunen@arm.com> Thu Apr 20 10:02:46 2023 +0800
committer Yanray Wang <yanray.wang@arm.com> Fri May 05 11:20:59 2023 +0800
tree 61ca0510525b88ae607241fc2a8716ba14463447
parent 732ca3221d907c00e4a8d5c974400cc927a30605

AES: skip 192bit and 256bit key in selftest if 128bit_only enabled

This includes:
 - aes.c
 - cmac.c
 - gcm.c
 - nist_kw.c

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

library/gcm.c

diff --git a/library/gcm.c b/library/gcm.c
index 71fcc35..e51367d 100644
--- a/library/gcm.c
+++ b/library/gcm.c
@@ -888,13 +888,20 @@
         int key_len = 128 + 64 * j;
 
         for (i = 0; i < MAX_TESTS; i++) {
-            mbedtls_gcm_init(&ctx);
-
             if (verbose != 0) {
                 mbedtls_printf("  AES-GCM-%3d #%d (%s): ",
                                key_len, i, "enc");
             }
 
+#if defined(MBEDTLS_AES_ONLY_128_BIT_KEY_LENGTH)
+            if (key_len > 128) {
+                mbedtls_printf("skipped\n");
+                continue;
+            }
+#endif /* MBEDTLS_AES_ONLY_128_BIT_KEY_LENGTH */
+
+            mbedtls_gcm_init(&ctx);
+
             ret = mbedtls_gcm_setkey(&ctx, cipher,
                                      key_test_data[key_index_test_data[i]],
                                      key_len);