commit 77b9cfcea92934dc2f48e9747c0a3b60f7d703bf
author Arto Kinnunen <arto.kinnunen@arm.com> Fri Aug 30 11:43:21 2019 +0300
committer Arto Kinnunen <arto.kinnunen@arm.com> Thu Oct 24 09:49:23 2019 +0300
tree 128eaf203909671a3482ebbe1970d0019b96f231
parent e2bf54d3d11b6282179eb4f082f16a63ca2da3e2

AES: add config option for 128-bit keys

Add configuration options to support only 128-bit key lengths in AES
calculation.

library/aesni.c

diff --git a/library/aesni.c b/library/aesni.c
index 062708b..c5873fc 100644
--- a/library/aesni.c
+++ b/library/aesni.c
@@ -327,6 +327,7 @@
 /*
  * Key expansion, 192-bit case
  */
+#if !defined(MBEDTLS_AES_ONLY_128_BIT_KEY_LENGTH)
 static void aesni_setkey_enc_192( unsigned char *rk,
                                   const unsigned char *key )
 {
@@ -380,10 +381,12 @@
          : "r" (rk), "r" (key)
          : "memory", "cc", "0" );
 }
+#endif /* !MBEDTLS_AES_ONLY_128_BIT_KEY_LENGTH */
 
 /*
  * Key expansion, 256-bit case
  */
+#if !defined(MBEDTLS_AES_ONLY_128_BIT_KEY_LENGTH)
 static void aesni_setkey_enc_256( unsigned char *rk,
                                   const unsigned char *key )
 {
@@ -446,6 +449,7 @@
          : "r" (rk), "r" (key)
          : "memory", "cc", "0" );
 }
+#endif /* !MBEDTLS_AES_ONLY_128_BIT_KEY_LENGTH */
 
 /*
  * Key expansion, wrapper
@@ -457,8 +461,10 @@
     switch( bits )
     {
         case 128: aesni_setkey_enc_128( rk, key ); break;
+#if !defined(MBEDTLS_AES_ONLY_128_BIT_KEY_LENGTH)
         case 192: aesni_setkey_enc_192( rk, key ); break;
         case 256: aesni_setkey_enc_256( rk, key ); break;
+#endif /* !MBEDTLS_AES_ONLY_128_BIT_KEY_LENGTH */
         default : return( MBEDTLS_ERR_AES_INVALID_KEY_LENGTH );
     }