commit e096da1af61efc6a35e9bcd4f2fd35690299f17e
author Jerry Yu <jerry.h.yu@arm.com> Tue Jan 10 17:07:01 2023 +0800
committer Jerry Yu <jerry.h.yu@arm.com> Tue Feb 07 17:11:52 2023 +0800
tree 87970b5d6c7ca919510526d0e61db04bb21e63c7
parent 3f2fb71072678d494b88e477c74273c3fa7fa654

Add inverse key function

Signed-off-by: Jerry Yu <jerry.h.yu@arm.com>

library/aes.c

diff --git a/library/aes.c b/library/aes.c
index f0ade21..6306fec 100644
--- a/library/aes.c
+++ b/library/aes.c
@@ -661,6 +661,16 @@
     }
 #endif
 
+#if defined(MBEDTLS_AESCE_C) && defined(MBEDTLS_HAVE_ARM64)
+    if (mbedtls_aesce_has_support()) {
+        mbedtls_aesce_inverse_key(
+            (unsigned char *) RK,
+            (const unsigned char *) (cty.buf + cty.rk_offset),
+            ctx->nr);
+        goto exit;
+    }
+#endif
+
     SK = cty.buf + cty.rk_offset + cty.nr * 4;
 
     *RK++ = *SK++;