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>
diff --git a/library/aes.c b/library/aes.c
index 5493c31..39cf332 100644
--- a/library/aes.c
+++ b/library/aes.c
@@ -1815,6 +1815,13 @@
(mode == MBEDTLS_AES_DECRYPT) ? "dec" : "enc");
}
+#if defined(MBEDTLS_AES_ONLY_128_BIT_KEY_LENGTH)
+ if (keybits > 128) {
+ mbedtls_printf("skipped\n");
+ continue;
+ }
+#endif /* MBEDTLS_AES_ONLY_128_BIT_KEY_LENGTH */
+
memset(buf, 0, 16);
if (mode == MBEDTLS_AES_DECRYPT) {
@@ -1872,6 +1879,13 @@
(mode == MBEDTLS_AES_DECRYPT) ? "dec" : "enc");
}
+#if defined(MBEDTLS_AES_ONLY_128_BIT_KEY_LENGTH)
+ if (keybits > 128) {
+ mbedtls_printf("skipped\n");
+ continue;
+ }
+#endif /* MBEDTLS_AES_ONLY_128_BIT_KEY_LENGTH */
+
memset(iv, 0, 16);
memset(prv, 0, 16);
memset(buf, 0, 16);
@@ -1941,6 +1955,13 @@
(mode == MBEDTLS_AES_DECRYPT) ? "dec" : "enc");
}
+#if defined(MBEDTLS_AES_ONLY_128_BIT_KEY_LENGTH)
+ if (keybits > 128) {
+ mbedtls_printf("skipped\n");
+ continue;
+ }
+#endif /* MBEDTLS_AES_ONLY_128_BIT_KEY_LENGTH */
+
memcpy(iv, aes_test_cfb128_iv, 16);
memcpy(key, aes_test_cfb128_key[u], keybits / 8);
@@ -2000,6 +2021,13 @@
(mode == MBEDTLS_AES_DECRYPT) ? "dec" : "enc");
}
+#if defined(MBEDTLS_AES_ONLY_128_BIT_KEY_LENGTH)
+ if (keybits > 128) {
+ mbedtls_printf("skipped\n");
+ continue;
+ }
+#endif /* MBEDTLS_AES_ONLY_128_BIT_KEY_LENGTH */
+
memcpy(iv, aes_test_ofb_iv, 16);
memcpy(key, aes_test_ofb_key[u], keybits / 8);