From e2bc158b3832a58e50a3e9804fe6abd50c76b22b Mon Sep 17 00:00:00 2001
From: Yanray Wang <yanray.wang@arm.com>
Date: Mon, 8 May 2023 10:28:53 +0800
Subject: [PATCH] aesce.c: add macro of MBEDTLS_AES_ONLY_128_BIT_KEY_LENGTH

Signed-off-by: Yanray Wang <yanray.wang@arm.com>
---
 library/aesce.c | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/library/aesce.c b/library/aesce.c
index ff8c2e09cd..4db8d2a199 100644
--- a/library/aesce.c
+++ b/library/aesce.c
@@ -251,6 +251,7 @@ static void aesce_setkey_enc(unsigned char *rk,
             /* Do not write overflow words.*/
             continue;
         }
+#if !defined(MBEDTLS_AES_ONLY_128_BIT_KEY_LENGTH)
         switch (key_bit_length) {
             case 128:
                 break;
@@ -265,6 +266,7 @@ static void aesce_setkey_enc(unsigned char *rk,
                 rko[7] = rko[6] ^ rki[7];
                 break;
         }
+#endif /* !MBEDTLS_AES_ONLY_128_BIT_KEY_LENGTH */
     }
 }