Add key expansion for encrypt
Signed-off-by: Jerry Yu <jerry.h.yu@arm.com>
diff --git a/library/aesce.h b/library/aesce.h
index 2d5dde9..7fc0cfa 100644
--- a/library/aesce.h
+++ b/library/aesce.h
@@ -49,6 +49,20 @@
*/
int mbedtls_aesce_has_support(void);
+
+/**
+ * \brief Internal key expansion for encryption
+ *
+ * \param rk Destination buffer where the round keys are written
+ * \param key Encryption key
+ * \param bits Key size in bits (must be 128, 192 or 256)
+ *
+ * \return 0 if successful, or MBEDTLS_ERR_AES_INVALID_KEY_LENGTH
+ */
+int mbedtls_aesce_setkey_enc(unsigned char *rk,
+ const unsigned char *key,
+ size_t bits);
+
#ifdef __cplusplus
}
#endif