diff --git a/components/ssl/mbedtls/port/esp8266/include/aes_alt.h b/components/ssl/mbedtls/port/esp8266/include/aes_alt.h new file mode 100644 index 00000000..d7843ea8 --- /dev/null +++ b/components/ssl/mbedtls/port/esp8266/include/aes_alt.h @@ -0,0 +1,116 @@ +/** + * \file aes_alt.h + * + * \brief AES block cipher + * + * Copyright (C) 2006-2015, ARM Limited, All Rights Reserved + * SPDX-License-Identifier: Apache-2.0 + * + * Licensed under the Apache License, Version 2.0 (the "License"); you may + * not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT + * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + * + * + */ +#ifndef AES_ALT_H +#define AES_ALT_H + +#ifdef __cplusplus +extern "C" { +#endif + +#if defined(MBEDTLS_AES_ALT) +#include "esp_aes.h" + +typedef esp_aes_t mbedtls_aes_context; + +#define mbedtls_aes_init(_ctx) { } +#define mbedtls_aes_free(_ctx) { } +#define mbedtls_aes_setkey_enc(_ctx, _key, _keybits) esp_aes_set_encrypt_key(_ctx, _key, _keybits) +#define mbedtls_aes_setkey_dec(_ctx, _key, _keybits) esp_aes_set_decrypt_key(_ctx, _key, _keybits) + +#define mbedtls_aes_crypt_ecb(_ctx, _mode, _input, _output) \ +({ \ + int ret; \ + \ + if (_mode == MBEDTLS_AES_DECRYPT) \ + ret = esp_aes_decrypt_ecb(_ctx, _input, _output); \ + else if (_mode == MBEDTLS_AES_ENCRYPT) \ + ret = esp_aes_encrypt_ecb(_ctx, _input, _output); \ + else \ + ret = -1; \ + \ + ret; \ +}) + +#if defined(MBEDTLS_CIPHER_MODE_CBC) +#define mbedtls_aes_crypt_cbc(_ctx, _mode, _length, _iv, _input, _output) \ +({ \ + int ret; \ + \ + if (_mode == MBEDTLS_AES_DECRYPT) \ + ret = esp_aes_decrypt_cbc(_ctx, _input, _length, _output, _length, _iv); \ + else if (_mode == MBEDTLS_AES_ENCRYPT) \ + ret = esp_aes_encrypt_cbc(_ctx, _input, _length, _output, _length, _iv); \ + else \ + ret = -1; \ + \ + ret; \ +}) +#endif + +#if defined(MBEDTLS_CIPHER_MODE_CFB) +#define mbedtls_aes_crypt_cfb128(_ctx, _mode, _length, _iv_off, _iv, _input, _output) \ +({ \ + int ret; \ + \ + if (_mode == MBEDTLS_AES_DECRYPT) \ + ret = esp_aes_decrypt_cfb128(_ctx, _input, _length, _output, _length, _iv, _iv_off); \ + else if (_mode == MBEDTLS_AES_ENCRYPT) \ + ret = esp_aes_encrypt_cfb128(_ctx, _input, _length, _output, _length, _iv, _iv_off); \ + else \ + ret = -1; \ + \ + ret; \ +}) + +#define mbedtls_aes_crypt_cfb8(_ctx, _mode, _length, _iv, _input, _output) \ +({ \ + int ret; \ + \ + if (_mode == MBEDTLS_AES_DECRYPT) \ + ret = esp_aes_decrypt_cfb8(_ctx, _input, _length, _output, _length, _iv); \ + else if (_mode == MBEDTLS_AES_ENCRYPT) \ + ret = esp_aes_encrypt_cfb8(_ctx, _input, _length, _output, _length, _iv); \ + else \ + ret = -1; \ + \ + ret; \ +}) +#endif + +#if defined(MBEDTLS_CIPHER_MODE_CTR) +#define mbedtls_aes_crypt_ctr(_ctx, _length, _nc_off, _nonce_counter, \ + _stream_block, _input _output \ + \ + esp_aes_encrypt_ctr(_ctx, _nc_off, _nonce_counter, _length, _input, \ + _length, _output, _length) +#endif + +#define mbedtls_internal_aes_encrypt(_ctx, _input, _output) esp_aes_encrypt(_ctx, _input, 16, _output, 16) +#define mbedtls_internal_aes_decrypt(_ctx, _input, _output) esp_aes_decrypt(_ctx, _input, 16, _output, 16) +#endif /* MBEDTLS_AES_ALT */ + +#ifdef __cplusplus +} +#endif + +#endif diff --git a/components/ssl/mbedtls/port/esp8266/include/mbedtls/esp_config.h b/components/ssl/mbedtls/port/esp8266/include/mbedtls/esp_config.h index 3cae8c86..40bf850b 100644 --- a/components/ssl/mbedtls/port/esp8266/include/mbedtls/esp_config.h +++ b/components/ssl/mbedtls/port/esp8266/include/mbedtls/esp_config.h @@ -275,7 +275,6 @@ * digests and ciphers instead. * */ -//#define MBEDTLS_AES_ALT //#define MBEDTLS_ARC4_ALT //#define MBEDTLS_BLOWFISH_ALT //#define MBEDTLS_CAMELLIA_ALT @@ -292,6 +291,10 @@ //#define MBEDTLS_RSA_ALT //#define MBEDTLS_XTEA_ALT +#ifdef CONFIG_ESP_AES +#define MBEDTLS_AES_ALT +#endif + #ifdef CONFIG_ESP_SHA #define MBEDTLS_SHA1_ALT #define MBEDTLS_SHA256_ALT diff --git a/components/util/Kconfig b/components/util/Kconfig index afc66d25..57557912 100644 --- a/components/util/Kconfig +++ b/components/util/Kconfig @@ -12,4 +12,19 @@ config ESP_SHA Disabling the "assert" function at menuconfig can speed up the calculation. +config ESP_AES + bool "Enable Espressif AES" + default y + help + Enable Espressif AES ECB, CBC, CFB128, CFB8 & CRT for other components to + speed up process speed and save code size. + + ESP8285 is like ESP8266 + 1MB flash, but its internal I/O connection from CPU + core to flash is DIO not QIO, which makes it read flash data slower. + So the function will speed up ESP8285 obviously. + + The calculation uses "ibus_data" to speed up load data from instruction bus. + + Disabling the "assert" function at menuconfig can speed up the calculation. + endmenu # Util diff --git a/components/util/include/esp_aes.h b/components/util/include/esp_aes.h new file mode 100644 index 00000000..2d2bb891 --- /dev/null +++ b/components/util/include/esp_aes.h @@ -0,0 +1,230 @@ +// Copyright 2019-2020 Espressif Systems (Shanghai) PTE LTD +// +// Licensed under the Apache License, Version 2.0 (the "License"); +// you may not use this file except in compliance with the License. +// You may obtain a copy of the License at + +// http://www.apache.org/licenses/LICENSE-2.0 +// +// Unless required by applicable law or agreed to in writing, software +// distributed under the License is distributed on an "AS IS" BASIS, +// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +// See the License for the specific language governing permissions and +// limitations under the License. + +#pragma once + +#include +#include + +#ifdef __cplusplus +extern "C" { +#endif + +typedef struct esp_aes { + int32_t nr; /*!< The number of AES key bits */ + uint32_t *rk; /*!< The AES AES key */ + uint32_t buf[68]; /*!< The AES calculation cache */ +} esp_aes_t; + + +/** + * @brief Set AES encrypt key + * + * @param aes AES contex pointer + * @param p_key AES key data buffer + * @param keybits number of AES key bits + * + * @return 0 if success or fail + */ +int esp_aes_set_encrypt_key(esp_aes_t *aes, const void *p_key, size_t keybits); + +/** + * @brief Set AES decrypt key + * + * @param aes AES contex pointer + * @param p_key AES key data buffer + * @param keybits number of AES key bits + * + * @return 0 if success or fail + */ +int esp_aes_set_decrypt_key(esp_aes_t *aes, const void *key, size_t keybits); + +/** + * @brief AES normal encrypt calculation + * + * @param aes AES contex pointer + * @param p_src input data buffer + * @param slen input data length by bytes + * @param p_dst output data buffer + * @param dlen output data length by bytes + * + * @return 0 if success or fail + */ +int esp_aes_encrypt(esp_aes_t *aes, const void *p_src, size_t slen, void *p_dst, size_t dlen); + +/** + * @brief AES normal decrypt calculation + * + * @param aes AES contex pointer + * @param p_src input data buffer + * @param slen input data length by bytes + * @param p_dst output data buffer + * @param dlen output data length by bytes + * + * @return 0 if success or fail + */ +int esp_aes_decrypt(esp_aes_t *aes, const void *p_src, size_t slen, void *p_dst, size_t dlen); + +/** + * @brief AES-ECB encrypt calculation + * + * @param aes AES contex pointer + * @param p_src input data buffer + * @param p_dst output data buffer + * + * @return 0 if success or fail + */ +static inline int esp_aes_encrypt_ecb(esp_aes_t *aes, const void *p_src, void *p_dst) +{ + return esp_aes_encrypt(aes, p_src, 16, p_dst, 16); +} + +/** + * @brief AES-ECB decrypt calculation + * + * @param aes AES contex pointer + * @param p_src input data buffer + * @param p_dst output data buffer + * + * @return 0 if success or fail + */ +static inline int esp_aes_decrypt_ecb(esp_aes_t *aes, const void *p_src, void *p_dst) +{ + return esp_aes_decrypt(aes, p_src, 16, p_dst, 16); +} + +/** + * @brief AES-CBC encrypt calculation + * + * @param aes AES contex pointer + * @param p_src input data buffer + * @param slen input data length by bytes + * @param p_dst output data buffer + * @param dlen output data length by bytes + * @param p_iv initialization vector buffer + * + * @return 0 if success or fail + */ +int esp_aes_encrypt_cbc(esp_aes_t *aes, const void *p_src, size_t slen, void *p_dst, size_t dlen, void *p_iv); + +/** + * @brief AES-CBC decrypt calculation + * + * @param aes AES contex pointer + * @param p_src input data buffer + * @param slen input data length by bytes + * @param p_dst output data buffer + * @param dlen output data length by bytes + * @param p_iv initialization vector buffer + * + * @return 0 if success or fail + */ +int esp_aes_decrypt_cbc(esp_aes_t *aes, const void *p_src, size_t slen, void *p_dst, size_t dlen, void *p_iv); + +/** + * @brief AES-CFB128 encrypt calculation + * + * @param aes AES contex pointer + * @param p_src input data buffer + * @param slen input data length by bytes + * @param p_dst output data buffer + * @param dlen output data length by bytes + * @param p_iv initialization vector buffer + * @param iv_off initialization vector offset + * + * @return 0 if success or fail + */ +int esp_aes_encrypt_cfb128(esp_aes_t *aes, const void *p_src, size_t slen, void *p_dst, size_t dlen, void *p_iv, size_t *iv_off); + +/** + * @brief AES-CFB128 decrypt calculation + * + * @param aes AES contex pointer + * @param p_src input data buffer + * @param slen input data length by bytes + * @param p_dst output data buffer + * @param dlen output data length by bytes + * @param p_iv initialization vector buffer + * @param iv_off initialization vector offset + * + * @return 0 if success or fail + */ +int esp_aes_decrypt_cfb128(esp_aes_t *aes, const void *p_src, size_t slen, void *p_dst, size_t dlen, void *p_iv, size_t *iv_off); + +/** + * @brief AES-CFB8 encrypt calculation + * + * @param aes AES contex pointer + * @param p_src input data buffer + * @param slen input data length by bytes + * @param p_dst output data buffer + * @param dlen output data length by bytes + * @param p_iv initialization vector buffer + * + * @return 0 if success or fail + */ +int esp_aes_encrypt_cfb8(esp_aes_t *aes, const void *p_src, size_t slen, void *p_dst, size_t dlen, void *p_iv); + +/** + * @brief AES-CFB8 decrypt calculation + * + * @param aes AES contex pointer + * @param p_src input data buffer + * @param slen input data length by bytes + * @param p_dst output data buffer + * @param dlen output data length by bytes + * @param p_iv initialization vector buffer + * + * @return 0 if success or fail + */ +int esp_aes_decrypt_cfb8(esp_aes_t *aes, const void *p_src, size_t slen, void *p_dst, size_t dlen, void *p_iv); + +/** + * @brief AES-CTR encrypt calculation + * + * @param aes AES contex pointer + * @param nc_off offset in the current stream block + * @param p_nonce_counter 128-bit nonce and counter buffer + * @param p_stream_block current stream block buffer + * @param p_src input data buffer + * @param slen input data length by bytes + * @param p_dst output data buffer + * @param dlen output data length by bytes + * + * @return 0 if success or fail + */ +int esp_aes_encrypt_ctr(esp_aes_t *aes, size_t *nc_off, void *p_nonce_counter, void *p_stream_block, const void *p_src, size_t slen, void *p_dst, size_t dlen); + +/** + * @brief AES-CTR decrypt calculation + * + * @param aes AES contex pointer + * @param nc_off offset in the current stream block + * @param p_nonce_counter 128-bit nonce and counter buffer + * @param p_stream_block current stream block buffer + * @param p_src input data buffer + * @param slen input data length by bytes + * @param p_dst output data buffer + * @param dlen output data length by bytes + * + * @return 0 if success or fail + */ +static inline int esp_aes_decrypt_ctr(esp_aes_t *aes, size_t *nc_off, void *p_nonce_counter, void *p_stream_block, const void *p_src, size_t slen, void *p_dst, size_t dlen) +{ + return esp_aes_encrypt_ctr(aes, nc_off, p_nonce_counter, p_stream_block, p_src, slen, p_dst, dlen); +} + +#ifdef __cplusplus +} +#endif diff --git a/components/util/src/aes.c b/components/util/src/aes.c new file mode 100644 index 00000000..0c0bee8c --- /dev/null +++ b/components/util/src/aes.c @@ -0,0 +1,866 @@ +// Copyright 2018-2019 Espressif Systems (Shanghai) PTE LTD +// +// Licensed under the Apache License, Version 2.0 (the "License"); +// you may not use this file except in compliance with the License. +// You may obtain a copy of the License at + +// http://www.apache.org/licenses/LICENSE-2.0 +// +// Unless required by applicable law or agreed to in writing, software +// distributed under the License is distributed on an "AS IS" BASIS, +// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +// See the License for the specific language governing permissions and +// limitations under the License. + +#include +#include +#include +#include "esp_aes.h" +#include "ibus_data.h" + +/* + * 32-bit integer manipulation macros (little endian) + */ +#ifndef GET_UINT32_LE +#define GET_UINT32_LE(n, b, i) \ +{ \ + (n) = ((uint32_t)(b)[(i) + 0] << 0) \ + | ((uint32_t)(b)[(i) + 1] << 8) \ + | ((uint32_t)(b)[(i) + 2] << 16) \ + | ((uint32_t)(b)[(i) + 3] << 24); \ +} +#endif + +#ifndef PUT_UINT32_LE +#define PUT_UINT32_LE(n, b, i) \ +{ \ + (b)[(i) + 0] = (uint8_t)(((n) >> 0) & 0xFF); \ + (b)[(i) + 1] = (uint8_t)(((n) >> 8) & 0xFF); \ + (b)[(i) + 2] = (uint8_t)(((n) >> 16) & 0xFF); \ + (b)[(i) + 3] = (uint8_t)(((n) >> 24) & 0xFF); \ +} +#endif + +#undef AES_FROUND +#define AES_FROUND(X0,X1,X2,X3,Y0,Y1,Y2,Y3) \ +{ \ + X0 = *RK++ ^ s_aes_ft0[(Y0 >> 0) & 0xFF] \ + ^ s_aes_ft1[(Y1 >> 8) & 0xFF] \ + ^ s_aes_ft2[(Y2 >> 16) & 0xFF] \ + ^ s_aes_ft3[(Y3 >> 24) & 0xFF]; \ + \ + X1 = *RK++ ^ s_aes_ft0[(Y1 >> 0) & 0xFF] \ + ^ s_aes_ft1[(Y2 >> 8) & 0xFF] \ + ^ s_aes_ft2[(Y3 >> 16) & 0xFF] \ + ^ s_aes_ft3[(Y0 >> 24) & 0xFF]; \ + \ + X2 = *RK++ ^ s_aes_ft0[(Y2 >> 0) & 0xFF] \ + ^ s_aes_ft1[(Y3 >> 8) & 0xFF] \ + ^ s_aes_ft2[(Y0 >> 16) & 0xFF] \ + ^ s_aes_ft3[(Y1 >> 24) & 0xFF]; \ + \ + X3 = *RK++ ^ s_aes_ft0[(Y3 >> 0) & 0xFF] \ + ^ s_aes_ft1[(Y0 >> 8) & 0xFF] \ + ^ s_aes_ft2[(Y1 >> 16) & 0xFF] \ + ^ s_aes_ft3[(Y2 >> 24) & 0xFF]; \ +} + +#undef AES_RROUND +#define AES_RROUND(X0,X1,X2,X3,Y0,Y1,Y2,Y3) \ +{ \ + X0 = *RK++ ^ s_aes_rt0[(Y0 >> 0) & 0xFF] \ + ^ s_aes_rt1[(Y3 >> 8) & 0xFF] \ + ^ s_aes_rt2[(Y2 >> 16) & 0xFF] \ + ^ s_aes_rt3[(Y1 >> 24) & 0xFF]; \ + \ + X1 = *RK++ ^ s_aes_rt0[(Y1 >> 0) & 0xFF] \ + ^ s_aes_rt1[(Y0 >> 8) & 0xFF] \ + ^ s_aes_rt2[(Y3 >> 16) & 0xFF] \ + ^ s_aes_rt3[(Y2 >> 24) & 0xFF]; \ + \ + X2 = *RK++ ^ s_aes_rt0[(Y2 >> 0) & 0xFF] \ + ^ s_aes_rt1[(Y1 >> 8) & 0xFF] \ + ^ s_aes_rt2[(Y0 >> 16) & 0xFF] \ + ^ s_aes_rt3[(Y3 >> 24) & 0xFF]; \ + \ + X3 = *RK++ ^ s_aes_rt0[(Y3 >> 0) & 0xFF] \ + ^ s_aes_rt1[(Y2 >> 8) & 0xFF] \ + ^ s_aes_rt2[(Y1 >> 16) & 0xFF] \ + ^ s_aes_rt3[(Y0 >> 24) & 0xFF]; \ +} + +/* + * Forward S-box + */ +static const uint8_t s_aes_fsb[256] __attribute__((aligned(4))) = +{ + 0x63, 0x7C, 0x77, 0x7B, 0xF2, 0x6B, 0x6F, 0xC5, + 0x30, 0x01, 0x67, 0x2B, 0xFE, 0xD7, 0xAB, 0x76, + 0xCA, 0x82, 0xC9, 0x7D, 0xFA, 0x59, 0x47, 0xF0, + 0xAD, 0xD4, 0xA2, 0xAF, 0x9C, 0xA4, 0x72, 0xC0, + 0xB7, 0xFD, 0x93, 0x26, 0x36, 0x3F, 0xF7, 0xCC, + 0x34, 0xA5, 0xE5, 0xF1, 0x71, 0xD8, 0x31, 0x15, + 0x04, 0xC7, 0x23, 0xC3, 0x18, 0x96, 0x05, 0x9A, + 0x07, 0x12, 0x80, 0xE2, 0xEB, 0x27, 0xB2, 0x75, + 0x09, 0x83, 0x2C, 0x1A, 0x1B, 0x6E, 0x5A, 0xA0, + 0x52, 0x3B, 0xD6, 0xB3, 0x29, 0xE3, 0x2F, 0x84, + 0x53, 0xD1, 0x00, 0xED, 0x20, 0xFC, 0xB1, 0x5B, + 0x6A, 0xCB, 0xBE, 0x39, 0x4A, 0x4C, 0x58, 0xCF, + 0xD0, 0xEF, 0xAA, 0xFB, 0x43, 0x4D, 0x33, 0x85, + 0x45, 0xF9, 0x02, 0x7F, 0x50, 0x3C, 0x9F, 0xA8, + 0x51, 0xA3, 0x40, 0x8F, 0x92, 0x9D, 0x38, 0xF5, + 0xBC, 0xB6, 0xDA, 0x21, 0x10, 0xFF, 0xF3, 0xD2, + 0xCD, 0x0C, 0x13, 0xEC, 0x5F, 0x97, 0x44, 0x17, + 0xC4, 0xA7, 0x7E, 0x3D, 0x64, 0x5D, 0x19, 0x73, + 0x60, 0x81, 0x4F, 0xDC, 0x22, 0x2A, 0x90, 0x88, + 0x46, 0xEE, 0xB8, 0x14, 0xDE, 0x5E, 0x0B, 0xDB, + 0xE0, 0x32, 0x3A, 0x0A, 0x49, 0x06, 0x24, 0x5C, + 0xC2, 0xD3, 0xAC, 0x62, 0x91, 0x95, 0xE4, 0x79, + 0xE7, 0xC8, 0x37, 0x6D, 0x8D, 0xD5, 0x4E, 0xA9, + 0x6C, 0x56, 0xF4, 0xEA, 0x65, 0x7A, 0xAE, 0x08, + 0xBA, 0x78, 0x25, 0x2E, 0x1C, 0xA6, 0xB4, 0xC6, + 0xE8, 0xDD, 0x74, 0x1F, 0x4B, 0xBD, 0x8B, 0x8A, + 0x70, 0x3E, 0xB5, 0x66, 0x48, 0x03, 0xF6, 0x0E, + 0x61, 0x35, 0x57, 0xB9, 0x86, 0xC1, 0x1D, 0x9E, + 0xE1, 0xF8, 0x98, 0x11, 0x69, 0xD9, 0x8E, 0x94, + 0x9B, 0x1E, 0x87, 0xE9, 0xCE, 0x55, 0x28, 0xDF, + 0x8C, 0xA1, 0x89, 0x0D, 0xBF, 0xE6, 0x42, 0x68, + 0x41, 0x99, 0x2D, 0x0F, 0xB0, 0x54, 0xBB, 0x16 +}; + +/* + * Forward tables + */ +#define FT \ +\ + V(A5,63,63,C6), V(84,7C,7C,F8), V(99,77,77,EE), V(8D,7B,7B,F6), \ + V(0D,F2,F2,FF), V(BD,6B,6B,D6), V(B1,6F,6F,DE), V(54,C5,C5,91), \ + V(50,30,30,60), V(03,01,01,02), V(A9,67,67,CE), V(7D,2B,2B,56), \ + V(19,FE,FE,E7), V(62,D7,D7,B5), V(E6,AB,AB,4D), V(9A,76,76,EC), \ + V(45,CA,CA,8F), V(9D,82,82,1F), V(40,C9,C9,89), V(87,7D,7D,FA), \ + V(15,FA,FA,EF), V(EB,59,59,B2), V(C9,47,47,8E), V(0B,F0,F0,FB), \ + V(EC,AD,AD,41), V(67,D4,D4,B3), V(FD,A2,A2,5F), V(EA,AF,AF,45), \ + V(BF,9C,9C,23), V(F7,A4,A4,53), V(96,72,72,E4), V(5B,C0,C0,9B), \ + V(C2,B7,B7,75), V(1C,FD,FD,E1), V(AE,93,93,3D), V(6A,26,26,4C), \ + V(5A,36,36,6C), V(41,3F,3F,7E), V(02,F7,F7,F5), V(4F,CC,CC,83), \ + V(5C,34,34,68), V(F4,A5,A5,51), V(34,E5,E5,D1), V(08,F1,F1,F9), \ + V(93,71,71,E2), V(73,D8,D8,AB), V(53,31,31,62), V(3F,15,15,2A), \ + V(0C,04,04,08), V(52,C7,C7,95), V(65,23,23,46), V(5E,C3,C3,9D), \ + V(28,18,18,30), V(A1,96,96,37), V(0F,05,05,0A), V(B5,9A,9A,2F), \ + V(09,07,07,0E), V(36,12,12,24), V(9B,80,80,1B), V(3D,E2,E2,DF), \ + V(26,EB,EB,CD), V(69,27,27,4E), V(CD,B2,B2,7F), V(9F,75,75,EA), \ + V(1B,09,09,12), V(9E,83,83,1D), V(74,2C,2C,58), V(2E,1A,1A,34), \ + V(2D,1B,1B,36), V(B2,6E,6E,DC), V(EE,5A,5A,B4), V(FB,A0,A0,5B), \ + V(F6,52,52,A4), V(4D,3B,3B,76), V(61,D6,D6,B7), V(CE,B3,B3,7D), \ + V(7B,29,29,52), V(3E,E3,E3,DD), V(71,2F,2F,5E), V(97,84,84,13), \ + V(F5,53,53,A6), V(68,D1,D1,B9), V(00,00,00,00), V(2C,ED,ED,C1), \ + V(60,20,20,40), V(1F,FC,FC,E3), V(C8,B1,B1,79), V(ED,5B,5B,B6), \ + V(BE,6A,6A,D4), V(46,CB,CB,8D), V(D9,BE,BE,67), V(4B,39,39,72), \ + V(DE,4A,4A,94), V(D4,4C,4C,98), V(E8,58,58,B0), V(4A,CF,CF,85), \ + V(6B,D0,D0,BB), V(2A,EF,EF,C5), V(E5,AA,AA,4F), V(16,FB,FB,ED), \ + V(C5,43,43,86), V(D7,4D,4D,9A), V(55,33,33,66), V(94,85,85,11), \ + V(CF,45,45,8A), V(10,F9,F9,E9), V(06,02,02,04), V(81,7F,7F,FE), \ + V(F0,50,50,A0), V(44,3C,3C,78), V(BA,9F,9F,25), V(E3,A8,A8,4B), \ + V(F3,51,51,A2), V(FE,A3,A3,5D), V(C0,40,40,80), V(8A,8F,8F,05), \ + V(AD,92,92,3F), V(BC,9D,9D,21), V(48,38,38,70), V(04,F5,F5,F1), \ + V(DF,BC,BC,63), V(C1,B6,B6,77), V(75,DA,DA,AF), V(63,21,21,42), \ + V(30,10,10,20), V(1A,FF,FF,E5), V(0E,F3,F3,FD), V(6D,D2,D2,BF), \ + V(4C,CD,CD,81), V(14,0C,0C,18), V(35,13,13,26), V(2F,EC,EC,C3), \ + V(E1,5F,5F,BE), V(A2,97,97,35), V(CC,44,44,88), V(39,17,17,2E), \ + V(57,C4,C4,93), V(F2,A7,A7,55), V(82,7E,7E,FC), V(47,3D,3D,7A), \ + V(AC,64,64,C8), V(E7,5D,5D,BA), V(2B,19,19,32), V(95,73,73,E6), \ + V(A0,60,60,C0), V(98,81,81,19), V(D1,4F,4F,9E), V(7F,DC,DC,A3), \ + V(66,22,22,44), V(7E,2A,2A,54), V(AB,90,90,3B), V(83,88,88,0B), \ + V(CA,46,46,8C), V(29,EE,EE,C7), V(D3,B8,B8,6B), V(3C,14,14,28), \ + V(79,DE,DE,A7), V(E2,5E,5E,BC), V(1D,0B,0B,16), V(76,DB,DB,AD), \ + V(3B,E0,E0,DB), V(56,32,32,64), V(4E,3A,3A,74), V(1E,0A,0A,14), \ + V(DB,49,49,92), V(0A,06,06,0C), V(6C,24,24,48), V(E4,5C,5C,B8), \ + V(5D,C2,C2,9F), V(6E,D3,D3,BD), V(EF,AC,AC,43), V(A6,62,62,C4), \ + V(A8,91,91,39), V(A4,95,95,31), V(37,E4,E4,D3), V(8B,79,79,F2), \ + V(32,E7,E7,D5), V(43,C8,C8,8B), V(59,37,37,6E), V(B7,6D,6D,DA), \ + V(8C,8D,8D,01), V(64,D5,D5,B1), V(D2,4E,4E,9C), V(E0,A9,A9,49), \ + V(B4,6C,6C,D8), V(FA,56,56,AC), V(07,F4,F4,F3), V(25,EA,EA,CF), \ + V(AF,65,65,CA), V(8E,7A,7A,F4), V(E9,AE,AE,47), V(18,08,08,10), \ + V(D5,BA,BA,6F), V(88,78,78,F0), V(6F,25,25,4A), V(72,2E,2E,5C), \ + V(24,1C,1C,38), V(F1,A6,A6,57), V(C7,B4,B4,73), V(51,C6,C6,97), \ + V(23,E8,E8,CB), V(7C,DD,DD,A1), V(9C,74,74,E8), V(21,1F,1F,3E), \ + V(DD,4B,4B,96), V(DC,BD,BD,61), V(86,8B,8B,0D), V(85,8A,8A,0F), \ + V(90,70,70,E0), V(42,3E,3E,7C), V(C4,B5,B5,71), V(AA,66,66,CC), \ + V(D8,48,48,90), V(05,03,03,06), V(01,F6,F6,F7), V(12,0E,0E,1C), \ + V(A3,61,61,C2), V(5F,35,35,6A), V(F9,57,57,AE), V(D0,B9,B9,69), \ + V(91,86,86,17), V(58,C1,C1,99), V(27,1D,1D,3A), V(B9,9E,9E,27), \ + V(38,E1,E1,D9), V(13,F8,F8,EB), V(B3,98,98,2B), V(33,11,11,22), \ + V(BB,69,69,D2), V(70,D9,D9,A9), V(89,8E,8E,07), V(A7,94,94,33), \ + V(B6,9B,9B,2D), V(22,1E,1E,3C), V(92,87,87,15), V(20,E9,E9,C9), \ + V(49,CE,CE,87), V(FF,55,55,AA), V(78,28,28,50), V(7A,DF,DF,A5), \ + V(8F,8C,8C,03), V(F8,A1,A1,59), V(80,89,89,09), V(17,0D,0D,1A), \ + V(DA,BF,BF,65), V(31,E6,E6,D7), V(C6,42,42,84), V(B8,68,68,D0), \ + V(C3,41,41,82), V(B0,99,99,29), V(77,2D,2D,5A), V(11,0F,0F,1E), \ + V(CB,B0,B0,7B), V(FC,54,54,A8), V(D6,BB,BB,6D), V(3A,16,16,2C) + +#define V(a,b,c,d) 0x##a##b##c##d +static const uint32_t s_aes_ft0[256] = { FT }; +#undef V + +#define V(a,b,c,d) 0x##b##c##d##a +static const uint32_t s_aes_ft1[256] = { FT }; +#undef V + +#define V(a,b,c,d) 0x##c##d##a##b +static const uint32_t s_aes_ft2[256] = { FT }; +#undef V + +#define V(a,b,c,d) 0x##d##a##b##c +static const uint32_t s_aes_ft3[256] = { FT }; +#undef V + +#undef FT + +/* + * Reverse S-box + */ +static const uint8_t s_aes_rsb[256] __attribute__((aligned(4))) = +{ + 0x52, 0x09, 0x6A, 0xD5, 0x30, 0x36, 0xA5, 0x38, + 0xBF, 0x40, 0xA3, 0x9E, 0x81, 0xF3, 0xD7, 0xFB, + 0x7C, 0xE3, 0x39, 0x82, 0x9B, 0x2F, 0xFF, 0x87, + 0x34, 0x8E, 0x43, 0x44, 0xC4, 0xDE, 0xE9, 0xCB, + 0x54, 0x7B, 0x94, 0x32, 0xA6, 0xC2, 0x23, 0x3D, + 0xEE, 0x4C, 0x95, 0x0B, 0x42, 0xFA, 0xC3, 0x4E, + 0x08, 0x2E, 0xA1, 0x66, 0x28, 0xD9, 0x24, 0xB2, + 0x76, 0x5B, 0xA2, 0x49, 0x6D, 0x8B, 0xD1, 0x25, + 0x72, 0xF8, 0xF6, 0x64, 0x86, 0x68, 0x98, 0x16, + 0xD4, 0xA4, 0x5C, 0xCC, 0x5D, 0x65, 0xB6, 0x92, + 0x6C, 0x70, 0x48, 0x50, 0xFD, 0xED, 0xB9, 0xDA, + 0x5E, 0x15, 0x46, 0x57, 0xA7, 0x8D, 0x9D, 0x84, + 0x90, 0xD8, 0xAB, 0x00, 0x8C, 0xBC, 0xD3, 0x0A, + 0xF7, 0xE4, 0x58, 0x05, 0xB8, 0xB3, 0x45, 0x06, + 0xD0, 0x2C, 0x1E, 0x8F, 0xCA, 0x3F, 0x0F, 0x02, + 0xC1, 0xAF, 0xBD, 0x03, 0x01, 0x13, 0x8A, 0x6B, + 0x3A, 0x91, 0x11, 0x41, 0x4F, 0x67, 0xDC, 0xEA, + 0x97, 0xF2, 0xCF, 0xCE, 0xF0, 0xB4, 0xE6, 0x73, + 0x96, 0xAC, 0x74, 0x22, 0xE7, 0xAD, 0x35, 0x85, + 0xE2, 0xF9, 0x37, 0xE8, 0x1C, 0x75, 0xDF, 0x6E, + 0x47, 0xF1, 0x1A, 0x71, 0x1D, 0x29, 0xC5, 0x89, + 0x6F, 0xB7, 0x62, 0x0E, 0xAA, 0x18, 0xBE, 0x1B, + 0xFC, 0x56, 0x3E, 0x4B, 0xC6, 0xD2, 0x79, 0x20, + 0x9A, 0xDB, 0xC0, 0xFE, 0x78, 0xCD, 0x5A, 0xF4, + 0x1F, 0xDD, 0xA8, 0x33, 0x88, 0x07, 0xC7, 0x31, + 0xB1, 0x12, 0x10, 0x59, 0x27, 0x80, 0xEC, 0x5F, + 0x60, 0x51, 0x7F, 0xA9, 0x19, 0xB5, 0x4A, 0x0D, + 0x2D, 0xE5, 0x7A, 0x9F, 0x93, 0xC9, 0x9C, 0xEF, + 0xA0, 0xE0, 0x3B, 0x4D, 0xAE, 0x2A, 0xF5, 0xB0, + 0xC8, 0xEB, 0xBB, 0x3C, 0x83, 0x53, 0x99, 0x61, + 0x17, 0x2B, 0x04, 0x7E, 0xBA, 0x77, 0xD6, 0x26, + 0xE1, 0x69, 0x14, 0x63, 0x55, 0x21, 0x0C, 0x7D +}; + +/* + * Reverse tables + */ +#define RT \ +\ + V(50,A7,F4,51), V(53,65,41,7E), V(C3,A4,17,1A), V(96,5E,27,3A), \ + V(CB,6B,AB,3B), V(F1,45,9D,1F), V(AB,58,FA,AC), V(93,03,E3,4B), \ + V(55,FA,30,20), V(F6,6D,76,AD), V(91,76,CC,88), V(25,4C,02,F5), \ + V(FC,D7,E5,4F), V(D7,CB,2A,C5), V(80,44,35,26), V(8F,A3,62,B5), \ + V(49,5A,B1,DE), V(67,1B,BA,25), V(98,0E,EA,45), V(E1,C0,FE,5D), \ + V(02,75,2F,C3), V(12,F0,4C,81), V(A3,97,46,8D), V(C6,F9,D3,6B), \ + V(E7,5F,8F,03), V(95,9C,92,15), V(EB,7A,6D,BF), V(DA,59,52,95), \ + V(2D,83,BE,D4), V(D3,21,74,58), V(29,69,E0,49), V(44,C8,C9,8E), \ + V(6A,89,C2,75), V(78,79,8E,F4), V(6B,3E,58,99), V(DD,71,B9,27), \ + V(B6,4F,E1,BE), V(17,AD,88,F0), V(66,AC,20,C9), V(B4,3A,CE,7D), \ + V(18,4A,DF,63), V(82,31,1A,E5), V(60,33,51,97), V(45,7F,53,62), \ + V(E0,77,64,B1), V(84,AE,6B,BB), V(1C,A0,81,FE), V(94,2B,08,F9), \ + V(58,68,48,70), V(19,FD,45,8F), V(87,6C,DE,94), V(B7,F8,7B,52), \ + V(23,D3,73,AB), V(E2,02,4B,72), V(57,8F,1F,E3), V(2A,AB,55,66), \ + V(07,28,EB,B2), V(03,C2,B5,2F), V(9A,7B,C5,86), V(A5,08,37,D3), \ + V(F2,87,28,30), V(B2,A5,BF,23), V(BA,6A,03,02), V(5C,82,16,ED), \ + V(2B,1C,CF,8A), V(92,B4,79,A7), V(F0,F2,07,F3), V(A1,E2,69,4E), \ + V(CD,F4,DA,65), V(D5,BE,05,06), V(1F,62,34,D1), V(8A,FE,A6,C4), \ + V(9D,53,2E,34), V(A0,55,F3,A2), V(32,E1,8A,05), V(75,EB,F6,A4), \ + V(39,EC,83,0B), V(AA,EF,60,40), V(06,9F,71,5E), V(51,10,6E,BD), \ + V(F9,8A,21,3E), V(3D,06,DD,96), V(AE,05,3E,DD), V(46,BD,E6,4D), \ + V(B5,8D,54,91), V(05,5D,C4,71), V(6F,D4,06,04), V(FF,15,50,60), \ + V(24,FB,98,19), V(97,E9,BD,D6), V(CC,43,40,89), V(77,9E,D9,67), \ + V(BD,42,E8,B0), V(88,8B,89,07), V(38,5B,19,E7), V(DB,EE,C8,79), \ + V(47,0A,7C,A1), V(E9,0F,42,7C), V(C9,1E,84,F8), V(00,00,00,00), \ + V(83,86,80,09), V(48,ED,2B,32), V(AC,70,11,1E), V(4E,72,5A,6C), \ + V(FB,FF,0E,FD), V(56,38,85,0F), V(1E,D5,AE,3D), V(27,39,2D,36), \ + V(64,D9,0F,0A), V(21,A6,5C,68), V(D1,54,5B,9B), V(3A,2E,36,24), \ + V(B1,67,0A,0C), V(0F,E7,57,93), V(D2,96,EE,B4), V(9E,91,9B,1B), \ + V(4F,C5,C0,80), V(A2,20,DC,61), V(69,4B,77,5A), V(16,1A,12,1C), \ + V(0A,BA,93,E2), V(E5,2A,A0,C0), V(43,E0,22,3C), V(1D,17,1B,12), \ + V(0B,0D,09,0E), V(AD,C7,8B,F2), V(B9,A8,B6,2D), V(C8,A9,1E,14), \ + V(85,19,F1,57), V(4C,07,75,AF), V(BB,DD,99,EE), V(FD,60,7F,A3), \ + V(9F,26,01,F7), V(BC,F5,72,5C), V(C5,3B,66,44), V(34,7E,FB,5B), \ + V(76,29,43,8B), V(DC,C6,23,CB), V(68,FC,ED,B6), V(63,F1,E4,B8), \ + V(CA,DC,31,D7), V(10,85,63,42), V(40,22,97,13), V(20,11,C6,84), \ + V(7D,24,4A,85), V(F8,3D,BB,D2), V(11,32,F9,AE), V(6D,A1,29,C7), \ + V(4B,2F,9E,1D), V(F3,30,B2,DC), V(EC,52,86,0D), V(D0,E3,C1,77), \ + V(6C,16,B3,2B), V(99,B9,70,A9), V(FA,48,94,11), V(22,64,E9,47), \ + V(C4,8C,FC,A8), V(1A,3F,F0,A0), V(D8,2C,7D,56), V(EF,90,33,22), \ + V(C7,4E,49,87), V(C1,D1,38,D9), V(FE,A2,CA,8C), V(36,0B,D4,98), \ + V(CF,81,F5,A6), V(28,DE,7A,A5), V(26,8E,B7,DA), V(A4,BF,AD,3F), \ + V(E4,9D,3A,2C), V(0D,92,78,50), V(9B,CC,5F,6A), V(62,46,7E,54), \ + V(C2,13,8D,F6), V(E8,B8,D8,90), V(5E,F7,39,2E), V(F5,AF,C3,82), \ + V(BE,80,5D,9F), V(7C,93,D0,69), V(A9,2D,D5,6F), V(B3,12,25,CF), \ + V(3B,99,AC,C8), V(A7,7D,18,10), V(6E,63,9C,E8), V(7B,BB,3B,DB), \ + V(09,78,26,CD), V(F4,18,59,6E), V(01,B7,9A,EC), V(A8,9A,4F,83), \ + V(65,6E,95,E6), V(7E,E6,FF,AA), V(08,CF,BC,21), V(E6,E8,15,EF), \ + V(D9,9B,E7,BA), V(CE,36,6F,4A), V(D4,09,9F,EA), V(D6,7C,B0,29), \ + V(AF,B2,A4,31), V(31,23,3F,2A), V(30,94,A5,C6), V(C0,66,A2,35), \ + V(37,BC,4E,74), V(A6,CA,82,FC), V(B0,D0,90,E0), V(15,D8,A7,33), \ + V(4A,98,04,F1), V(F7,DA,EC,41), V(0E,50,CD,7F), V(2F,F6,91,17), \ + V(8D,D6,4D,76), V(4D,B0,EF,43), V(54,4D,AA,CC), V(DF,04,96,E4), \ + V(E3,B5,D1,9E), V(1B,88,6A,4C), V(B8,1F,2C,C1), V(7F,51,65,46), \ + V(04,EA,5E,9D), V(5D,35,8C,01), V(73,74,87,FA), V(2E,41,0B,FB), \ + V(5A,1D,67,B3), V(52,D2,DB,92), V(33,56,10,E9), V(13,47,D6,6D), \ + V(8C,61,D7,9A), V(7A,0C,A1,37), V(8E,14,F8,59), V(89,3C,13,EB), \ + V(EE,27,A9,CE), V(35,C9,61,B7), V(ED,E5,1C,E1), V(3C,B1,47,7A), \ + V(59,DF,D2,9C), V(3F,73,F2,55), V(79,CE,14,18), V(BF,37,C7,73), \ + V(EA,CD,F7,53), V(5B,AA,FD,5F), V(14,6F,3D,DF), V(86,DB,44,78), \ + V(81,F3,AF,CA), V(3E,C4,68,B9), V(2C,34,24,38), V(5F,40,A3,C2), \ + V(72,C3,1D,16), V(0C,25,E2,BC), V(8B,49,3C,28), V(41,95,0D,FF), \ + V(71,01,A8,39), V(DE,B3,0C,08), V(9C,E4,B4,D8), V(90,C1,56,64), \ + V(61,84,CB,7B), V(70,B6,32,D5), V(74,5C,6C,48), V(42,57,B8,D0) + +#define V(a,b,c,d) 0x##a##b##c##d +static const uint32_t s_aes_rt0[256] = { RT }; +#undef V + +#define V(a,b,c,d) 0x##b##c##d##a +static const uint32_t s_aes_rt1[256] = { RT }; +#undef V + +#define V(a,b,c,d) 0x##c##d##a##b +static const uint32_t s_aes_rt2[256] = { RT }; +#undef V + +#define V(a,b,c,d) 0x##d##a##b##c +static const uint32_t s_aes_rt3[256] = { RT }; +#undef V + +#undef RT + +/* + * Round constants + */ +static const uint32_t s_aes_rcon[10] = +{ + 0x00000001, 0x00000002, 0x00000004, 0x00000008, + 0x00000010, 0x00000020, 0x00000040, 0x00000080, + 0x0000001B, 0x00000036 +}; + +void esp_aes_init(esp_aes_t *aes) +{ + +} + +int esp_aes_set_encrypt_key(esp_aes_t *aes, const void *p_key, size_t keybits) +{ + unsigned int i; + uint32_t *RK; + const uint8_t *key = (const uint8_t *)p_key; + + switch (keybits) { + case 128: + aes->nr = 10; + break; + case 192: + aes->nr = 12; + break; + case 256: + aes->nr = 14; + break; + default : + return -EINVAL; + } + + aes->rk = RK = aes->buf; + + for (i = 0; i < (keybits >> 5); i++) { + GET_UINT32_LE(RK[i], key, i << 2); + } + + switch (aes->nr) { + case 10: + for (i = 0; i < 10; i++, RK += 4) { + RK[4] = RK[0] ^ s_aes_rcon[i] + ^ ((uint32_t)ESP_IBUS_GET_U8_DATA((RK[3] >> 8) & 0xFF, s_aes_fsb) << 0) + ^ ((uint32_t)ESP_IBUS_GET_U8_DATA((RK[3] >> 16) & 0xFF, s_aes_fsb) << 8) + ^ ((uint32_t)ESP_IBUS_GET_U8_DATA((RK[3] >> 24) & 0xFF, s_aes_fsb) << 16) + ^ ((uint32_t)ESP_IBUS_GET_U8_DATA((RK[3] >> 0) & 0xFF, s_aes_fsb) << 24); + + RK[5] = RK[1] ^ RK[4]; + RK[6] = RK[2] ^ RK[5]; + RK[7] = RK[3] ^ RK[6]; + } + break; + case 12: + for (i = 0; i < 8; i++, RK += 6) { + RK[6] = RK[0] ^ s_aes_rcon[i] + ^ ((uint32_t)ESP_IBUS_GET_U8_DATA((RK[5] >> 8) & 0xFF, s_aes_fsb) << 0) + ^ ((uint32_t)ESP_IBUS_GET_U8_DATA((RK[5] >> 16) & 0xFF, s_aes_fsb) << 8) + ^ ((uint32_t)ESP_IBUS_GET_U8_DATA((RK[5] >> 24) & 0xFF, s_aes_fsb) << 16) + ^ ((uint32_t)ESP_IBUS_GET_U8_DATA((RK[5] >> 0) & 0xFF, s_aes_fsb) << 24); + + RK[7] = RK[1] ^ RK[6]; + RK[8] = RK[2] ^ RK[7]; + RK[9] = RK[3] ^ RK[8]; + RK[10] = RK[4] ^ RK[9]; + RK[11] = RK[5] ^ RK[10]; + } + break; + case 14: + for (i = 0; i < 7; i++, RK += 8) { + RK[8] = RK[0] ^ s_aes_rcon[i] + ^ ((uint32_t)ESP_IBUS_GET_U8_DATA((RK[7] >> 8) & 0xFF, s_aes_fsb) << 0) + ^ ((uint32_t)ESP_IBUS_GET_U8_DATA((RK[7] >> 16) & 0xFF, s_aes_fsb) << 8) + ^ ((uint32_t)ESP_IBUS_GET_U8_DATA((RK[7] >> 24) & 0xFF, s_aes_fsb) << 16) + ^ ((uint32_t)ESP_IBUS_GET_U8_DATA((RK[7] >> 0) & 0xFF, s_aes_fsb) << 24); + + RK[9] = RK[1] ^ RK[8]; + RK[10] = RK[2] ^ RK[9]; + RK[11] = RK[3] ^ RK[10]; + + RK[12] = RK[4] ^ ((uint32_t)ESP_IBUS_GET_U8_DATA((RK[11] >> 0) & 0xFF, s_aes_fsb) << 0) + ^ ((uint32_t)ESP_IBUS_GET_U8_DATA((RK[11] >> 8) & 0xFF, s_aes_fsb) << 8) + ^ ((uint32_t)ESP_IBUS_GET_U8_DATA((RK[11] >> 16) & 0xFF, s_aes_fsb) << 16) + ^ ((uint32_t)ESP_IBUS_GET_U8_DATA((RK[11] >> 24) & 0xFF, s_aes_fsb) << 24); + + RK[13] = RK[5] ^ RK[12]; + RK[14] = RK[6] ^ RK[13]; + RK[15] = RK[7] ^ RK[14]; + } + break; + } + + return 0; +} + +int esp_aes_set_decrypt_key(esp_aes_t *aes, const void *key, size_t keybits) +{ + int i, j, ret; + esp_aes_t cty; + uint32_t *RK; + uint32_t *SK; + + esp_aes_init(&cty); + + aes->rk = RK = aes->buf; + + /* Also checks keybits */ + if ((ret = esp_aes_set_encrypt_key(&cty, key, keybits)) != 0) + return ret; + + aes->nr = cty.nr; + + SK = cty.rk + cty.nr * 4; + + *RK++ = *SK++; + *RK++ = *SK++; + *RK++ = *SK++; + *RK++ = *SK++; + + for (i = aes->nr - 1, SK -= 8; i > 0; i--, SK -= 8) { + for (j = 0; j < 4; j++, SK++) { + *RK++ = s_aes_rt0[ESP_IBUS_GET_U8_DATA((*SK >> 0) & 0xFF, s_aes_fsb)] + ^ s_aes_rt1[ESP_IBUS_GET_U8_DATA((*SK >> 8) & 0xFF, s_aes_fsb)] + ^ s_aes_rt2[ESP_IBUS_GET_U8_DATA((*SK >> 16) & 0xFF, s_aes_fsb)] + ^ s_aes_rt3[ESP_IBUS_GET_U8_DATA((*SK >> 24) & 0xFF, s_aes_fsb)]; + } + } + + *RK++ = *SK++; + *RK++ = *SK++; + *RK++ = *SK++; + *RK++ = *SK++; + + return ret; +} + +static void __esp_aes_encrypt(esp_aes_t *aes, const void *p_src, void *p_dst) +{ + int i; + uint32_t *RK, X0, X1, X2, X3, Y0, Y1, Y2, Y3; + const uint8_t *input = (const uint8_t *)p_src; + uint8_t *output = (uint8_t *)p_dst; + + RK = aes->rk; + + GET_UINT32_LE(X0, input, 0); X0 ^= *RK++; + GET_UINT32_LE(X1, input, 4); X1 ^= *RK++; + GET_UINT32_LE(X2, input, 8); X2 ^= *RK++; + GET_UINT32_LE(X3, input, 12); X3 ^= *RK++; + + for (i =(aes->nr >> 1)- 1; i > 0; i--) { + AES_FROUND(Y0, Y1, Y2, Y3, X0, X1, X2, X3); + AES_FROUND(X0, X1, X2, X3, Y0, Y1, Y2, Y3); + } + + AES_FROUND(Y0, Y1, Y2, Y3, X0, X1, X2, X3); + + X0 = *RK++ ^ ((uint32_t)ESP_IBUS_GET_U8_DATA((Y0 >> 0) & 0xFF, s_aes_fsb) << 0) + ^ ((uint32_t)ESP_IBUS_GET_U8_DATA((Y1 >> 8) & 0xFF, s_aes_fsb) << 8) + ^ ((uint32_t)ESP_IBUS_GET_U8_DATA((Y2 >> 16) & 0xFF, s_aes_fsb) << 16) + ^ ((uint32_t)ESP_IBUS_GET_U8_DATA((Y3 >> 24) & 0xFF, s_aes_fsb) << 24); + + X1 = *RK++ ^ ((uint32_t)ESP_IBUS_GET_U8_DATA((Y1 >> 0) & 0xFF, s_aes_fsb) << 0) + ^ ((uint32_t)ESP_IBUS_GET_U8_DATA((Y2 >> 8) & 0xFF, s_aes_fsb) << 8) + ^ ((uint32_t)ESP_IBUS_GET_U8_DATA((Y3 >> 16) & 0xFF, s_aes_fsb) << 16) + ^ ((uint32_t)ESP_IBUS_GET_U8_DATA((Y0 >> 24) & 0xFF, s_aes_fsb) << 24); + + X2 = *RK++ ^ ((uint32_t)ESP_IBUS_GET_U8_DATA((Y2 >> 0) & 0xFF, s_aes_fsb) << 0) + ^ ((uint32_t)ESP_IBUS_GET_U8_DATA((Y3 >> 8) & 0xFF, s_aes_fsb) << 8) + ^ ((uint32_t)ESP_IBUS_GET_U8_DATA((Y0 >> 16) & 0xFF, s_aes_fsb) << 16) + ^ ((uint32_t)ESP_IBUS_GET_U8_DATA((Y1 >> 24) & 0xFF, s_aes_fsb) << 24); + + X3 = *RK++ ^ ((uint32_t) ESP_IBUS_GET_U8_DATA((Y3 >> 0) & 0xFF, s_aes_fsb) << 0) + ^ ((uint32_t) ESP_IBUS_GET_U8_DATA((Y0 >> 8) & 0xFF, s_aes_fsb) << 8) + ^ ((uint32_t) ESP_IBUS_GET_U8_DATA((Y1 >> 16) & 0xFF, s_aes_fsb) << 16) + ^ ((uint32_t) ESP_IBUS_GET_U8_DATA((Y2 >> 24) & 0xFF, s_aes_fsb) << 24); + + PUT_UINT32_LE(X0, output, 0); + PUT_UINT32_LE(X1, output, 4); + PUT_UINT32_LE(X2, output, 8); + PUT_UINT32_LE(X3, output, 12); +} + +static int __esp_aes_decrypt(esp_aes_t *aes, const void *p_src, void *p_dst) +{ + int i; + uint32_t *RK, X0, X1, X2, X3, Y0, Y1, Y2, Y3; + const uint8_t *input = (const uint8_t *)p_src; + uint8_t *output = (uint8_t *)p_dst; + + RK = aes->rk; + + GET_UINT32_LE(X0, input, 0); X0 ^= *RK++; + GET_UINT32_LE(X1, input, 4); X1 ^= *RK++; + GET_UINT32_LE(X2, input, 8); X2 ^= *RK++; + GET_UINT32_LE(X3, input, 12); X3 ^= *RK++; + + for (i =(aes->nr >> 1)- 1; i > 0; i--) { + AES_RROUND(Y0, Y1, Y2, Y3, X0, X1, X2, X3); + AES_RROUND(X0, X1, X2, X3, Y0, Y1, Y2, Y3); + } + + AES_RROUND(Y0, Y1, Y2, Y3, X0, X1, X2, X3); + + X0 = *RK++ ^ ((uint32_t)ESP_IBUS_GET_U8_DATA((Y0 >> 0) & 0xFF, s_aes_rsb) << 0) + ^ ((uint32_t)ESP_IBUS_GET_U8_DATA((Y3 >> 8) & 0xFF, s_aes_rsb) << 8) + ^ ((uint32_t)ESP_IBUS_GET_U8_DATA((Y2 >> 16) & 0xFF, s_aes_rsb) << 16) + ^ ((uint32_t)ESP_IBUS_GET_U8_DATA((Y1 >> 24) & 0xFF, s_aes_rsb) << 24); + + X1 = *RK++ ^ ((uint32_t)ESP_IBUS_GET_U8_DATA((Y1 >> 0) & 0xFF, s_aes_rsb) << 0) + ^ ((uint32_t)ESP_IBUS_GET_U8_DATA((Y0 >> 8) & 0xFF, s_aes_rsb) << 8) + ^ ((uint32_t)ESP_IBUS_GET_U8_DATA((Y3 >> 16) & 0xFF, s_aes_rsb) << 16) + ^ ((uint32_t)ESP_IBUS_GET_U8_DATA((Y2 >> 24) & 0xFF, s_aes_rsb) << 24); + + X2 = *RK++ ^ ((uint32_t)ESP_IBUS_GET_U8_DATA((Y2 >> 0) & 0xFF, s_aes_rsb) << 0) + ^ ((uint32_t)ESP_IBUS_GET_U8_DATA((Y1 >> 8) & 0xFF, s_aes_rsb) << 8) + ^ ((uint32_t)ESP_IBUS_GET_U8_DATA((Y0 >> 16) & 0xFF, s_aes_rsb) << 16) + ^ ((uint32_t)ESP_IBUS_GET_U8_DATA((Y3 >> 24) & 0xFF, s_aes_rsb) << 24); + + X3 = *RK++ ^ ((uint32_t)ESP_IBUS_GET_U8_DATA((Y3 >> 0) & 0xFF, s_aes_rsb) << 0) + ^ ((uint32_t)ESP_IBUS_GET_U8_DATA((Y2 >> 8) & 0xFF, s_aes_rsb) << 8) + ^ ((uint32_t)ESP_IBUS_GET_U8_DATA((Y1 >> 16) & 0xFF, s_aes_rsb) << 16) + ^ ((uint32_t)ESP_IBUS_GET_U8_DATA((Y0 >> 24) & 0xFF, s_aes_rsb) << 24); + + PUT_UINT32_LE(X0, output, 0); + PUT_UINT32_LE(X1, output, 4); + PUT_UINT32_LE(X2, output, 8); + PUT_UINT32_LE(X3, output, 12); + + return 0; +} + +int esp_aes_encrypt(esp_aes_t *aes, + const void *p_src, + size_t slen, + void *p_dst, + size_t dlen) +{ + const uint8_t *input = (const uint8_t *)p_src; + uint8_t *output = (uint8_t *)p_dst; + + assert(aes); + assert(p_src); + assert(p_dst); + assert((slen > 0) && ((slen % 16) == 0)); + assert(dlen >= slen); + + for (size_t i = 0; i < slen; i += 16) + __esp_aes_encrypt(aes, input + i, output + i); + + return 0; +} + +int esp_aes_decrypt(esp_aes_t *aes, + const void *p_src, + size_t slen, + void *p_dst, + size_t dlen) +{ + const uint8_t *input = (const uint8_t *)p_src; + uint8_t *output = (uint8_t *)p_dst; + + assert(aes); + assert(p_src); + assert(p_dst); + assert((slen > 0) && ((slen % 16) == 0)); + assert(dlen >= slen); + + for (size_t i = 0; i < slen; i += 16) + __esp_aes_decrypt(aes, input + i, output + i); + + return 0; +} + +int esp_aes_encrypt_cbc(esp_aes_t *aes, + const void *p_src, + size_t slen, + void *p_dst, + size_t dlen, + void *p_iv) +{ + const uint8_t *input = (const uint8_t *)p_src; + uint8_t *iv = (uint8_t *)p_iv; + uint8_t *output = (uint8_t *)p_dst; + + assert(aes); + assert(p_src); + assert(p_dst); + assert(p_iv); + assert((slen > 0) && ((slen % 16) == 0)); + assert(dlen >= slen); + + for (int i = 0; i < slen; i += 16) { + for(int j = 0; j < 16; j++ ) + output[j] = input[j] ^ iv[j]; + + __esp_aes_encrypt(aes, output, output); + memcpy(iv, output, 16); + + input += 16; + output += 16; + } + + return( 0 ); +} + +int esp_aes_decrypt_cbc(esp_aes_t *aes, + const void *p_src, + size_t slen, + void *p_dst, + size_t dlen, + void *p_iv) +{ + const uint8_t *input = (const uint8_t *)p_src; + uint8_t *iv = (uint8_t *)p_iv; + uint8_t *output = (uint8_t *)p_dst; + + assert(aes); + assert(p_src); + assert(p_dst); + assert(p_iv); + assert((slen > 0) && ((slen % 16) == 0)); + assert(dlen >= slen); + + for (int i = 0; i < slen; i += 16) { + uint8_t temp[16]; + + memcpy(temp, input, 16); + __esp_aes_decrypt(aes, input, output); + + for (int j = 0; j < 16; j++) + output[j] = output[j] ^ iv[j]; + + memcpy(iv, temp, 16); + + input += 16; + output += 16; + } + + return( 0 ); +} + +int esp_aes_encrypt_cfb8(esp_aes_t *aes, + const void *p_src, + size_t slen, + void *p_dst, + size_t dlen, + void *p_iv) +{ + uint8_t c; + uint8_t ov[17]; + const uint8_t *input = (const uint8_t *)p_src; + uint8_t *iv = (uint8_t *)p_iv; + uint8_t *output = (uint8_t *)p_dst; + + for (int i = 0; i < slen; i++) { + memcpy(ov, iv, 16); + __esp_aes_encrypt(aes, iv, iv); + + c = *output++ = iv[0] ^ *input++; + ov[16] = c; + + memcpy(iv, ov + 1, 16); + } + + return 0; +} + +int esp_aes_decrypt_cfb8(esp_aes_t *aes, + const void *p_src, + size_t slen, + void *p_dst, + size_t dlen, + void *p_iv) +{ + uint8_t ov[17]; + const uint8_t *input = (const uint8_t *)p_src; + uint8_t *iv = (uint8_t *)p_iv; + uint8_t *output = (uint8_t *)p_dst; + + assert(aes); + assert(p_src); + assert(p_dst); + assert(p_iv); + assert((slen > 0) && ((slen % 16) == 0)); + assert(dlen >= slen); + + for (int i = 0; i < slen; i++) { + memcpy(ov, iv, 16); + __esp_aes_encrypt(aes, iv, iv); + + ov[16] = *input; + *output++ = iv[0] ^ *input++; + + memcpy(iv, ov + 1, 16); + } + + return 0; +} + +int esp_aes_encrypt_cfb128(esp_aes_t *aes, + const void *p_src, + size_t slen, + void *p_dst, + size_t dlen, + void *p_iv, + size_t *iv_off) +{ + size_t n; + const uint8_t *input = (const uint8_t *)p_src; + uint8_t *iv = (uint8_t *)p_iv; + uint8_t *output = (uint8_t *)p_dst; + + assert(aes); + assert(p_src); + assert(p_dst); + assert(p_iv); + assert(iv_off); + assert((slen > 0) && ((slen % 16) == 0)); + assert(dlen >= slen); + + n = *iv_off; + for (int i = 0; i < slen; i++) { + if (n == 0) + __esp_aes_encrypt(aes, iv, iv); + + iv[n] = *output++ = iv[n] ^ *input++; + + n = (n + 1) & 0x0F; + } + + *iv_off = n; + + return 0; +} + +int esp_aes_decrypt_cfb128(esp_aes_t *aes, + const void *p_src, + size_t slen, + void *p_dst, + size_t dlen, + void *p_iv, + size_t *iv_off) +{ + uint8_t c; + size_t n; + const uint8_t *input = (const uint8_t *)p_src; + uint8_t *iv = (uint8_t *)p_iv; + uint8_t *output = (uint8_t *)p_dst; + + assert(aes); + assert(p_src); + assert(p_dst); + assert(p_iv); + assert(iv_off); + assert((slen > 0) && ((slen % 16) == 0)); + assert(dlen >= slen); + + n = *iv_off; + for (int i = 0; i < slen; i++) { + if (n == 0) + __esp_aes_encrypt(aes, iv, iv); + + c = *input++; + *output++ = c ^ iv[n]; + iv[n] = c; + + n = (n + 1) & 0x0F; + } + + *iv_off = n; + + return 0; +} + +int esp_aes_encrypt_ctr(esp_aes_t *aes, + size_t *nc_off, + void *p_nonce_counter, + void *p_stream_block, + const void *p_src, + size_t slen, + void *p_dst, + size_t dlen) +{ + uint8_t c; + size_t n; + uint8_t *output = (uint8_t *)p_dst; + uint8_t *nonce_counter = (uint8_t *)p_nonce_counter; + uint8_t *stream_block = (uint8_t *)p_stream_block; + const uint8_t *input = (const uint8_t *)p_src; + + assert(aes); + assert(nc_off); + assert(p_nonce_counter); + assert(p_stream_block); + assert(p_src); + assert(p_dst); + assert((slen > 0) && ((slen % 16) == 0)); + assert(dlen >= slen); + + n = *nc_off; + for (int i = 0; i < slen; i++) { + if( n == 0 ) { + __esp_aes_encrypt(aes, nonce_counter, stream_block); + + for (int j = 16; j > 0; j--) + if (++nonce_counter[j - 1] != 0) + break; + } + c = *input++; + *output++ = c ^ stream_block[n]; + + n = (n + 1) & 0x0F; + } + + *nc_off = n; + + return 0; +} diff --git a/components/util/test/test_aes.c b/components/util/test/test_aes.c new file mode 100644 index 00000000..e34ebc12 --- /dev/null +++ b/components/util/test/test_aes.c @@ -0,0 +1,453 @@ +// Copyright 2018-2019 Espressif Systems (Shanghai) PTE LTD +// +// Licensed under the Apache License, Version 2.0 (the "License"); +// you may not use this file except in compliance with the License. +// You may obtain a copy of the License at + +// http://www.apache.org/licenses/LICENSE-2.0 +// +// Unless required by applicable law or agreed to in writing, software +// distributed under the License is distributed on an "AS IS" BASIS, +// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +// See the License for the specific language governing permissions and +// limitations under the License. + +#include +#include +#include +#include "esp_aes.h" +#include "unity.h" + +#define TEST_AES_COUNT 1 +#define TEST_AES_DEBUG 0 +#define TEST_AES_DEBUG_TIME 0 + +#if TEST_AES_DEBUG_TIME +#undef TEST_ASSERT_TRUE +#define TEST_ASSERT_TRUE(__e) ((__e) ? (void)0 : (void)0) +#endif + +static const uint32_t s_aes_ecb_result[3][4] = { + // keybits = 128 + { + 0x34eff23c, 0xb90aab82, 0x5781d455, 0x743e4cef, + }, + + // keybits = 192 + { + 0x8b47f409, 0x2752feb6, 0x8001f471, 0x76f0bb1d, + }, + + // keybits = 256 + { + 0x3d33ac44, 0x709b21d1, 0x7372cd80, 0x6bad3758, + } +}; + +static const uint32_t s_aes_cbc_result[3][8] = { + // keybits = 128 + { + 0xda3f4f4f, 0x9b96f894, 0xbfbbf671, 0x2da753f9, 0xad8a1a91, 0x958360a3, 0x857d1ae4, 0xda4c89e1, + }, + + // keybits = 192 + { + 0xe92a9715, 0xbe5983de, 0xefce73cf, 0x21ef8004, 0x31499c08, 0xb7753f3d, 0x54afc778, 0xc2a8209f, + }, + + // keybits = 256 + { + 0x2288383b, 0x8e541787, 0x3eb2c691, 0x76ae0dcd, 0x2a6f677a, 0xa1b0118d, 0x150ca144, 0x41def443, + } +}; + +static const uint32_t s_aes_cfb8_result[3][8] = { + // keybits = 128 + { + 0x6b5dabee, 0x41102894, 0x13e333be, 0x4fd60d9c, 0x2974dde7, 0xcaa73794, 0x1dc3d2d8, 0x3a210c38, + }, + + // keybits = 192 + { + 0xf3e04de7, 0xeca9fd81, 0xebf7ee9e, 0x7b6e7091, 0xa8bde865, 0xc150d7d9, 0xddaf6657, 0x77d6e3f0, + }, + + // keybits = 256 + { + 0xc510918d, 0xac9004da, 0xc23fea19, 0x6914df10, 0xa34d6dd5, 0x9ecc2cc7, 0x9941cd88, 0x1d472bb7, + } +}; + +static const uint32_t s_aes_cfb128_result[3][8] = { + // keybits = 128 + { + 0xfe2d65ee, 0x2d86806b, 0xea5d2e84, 0x10ebab93, 0x32f9354f, 0x0dcbc189, 0x087d75d8, 0xc456acd3, + }, + + // keybits = 192 + { + 0xfaaf66e7, 0x9c2cbe18, 0xa1f67fe2, 0x5f92ac9e, 0x1432de12, 0x4e115c08, 0x36d6a739, 0x5d8c1bce, + }, + + // keybits = 256 + { + 0x14e2318d, 0x185ac4df, 0xe6efc826, 0x65b796fb, 0x39f9a2e7, 0xffd706f4, 0x7ac79119, 0x47b99a97, + } +}; + +static const uint32_t s_aes_ctr_result[3][8] = { + // keybits = 128 + { + 0xfe2d65ee, 0x2d86806b, 0xea5d2e84, 0x10ebab93, 0x26a701de, 0x0e9d716f, 0x2d87c066, 0x2ff1f3af, + }, + + // keybits = 192 + { + 0xfaaf66e7, 0x9c2cbe18, 0xa1f67fe2, 0x5f92ac9e, 0xf7e0935e, 0xf0d3b9c1, 0x0f442d3b, 0xf7368de3, + }, + + // keybits = 256 + { + 0x14e2318d, 0x185ac4df, 0xe6efc826, 0x65b796fb, 0x9fef3b32, 0x742db24c, 0x08c56021, 0x5f5c8bf4, + } +}; + +TEST_CASE("Test AES-ECB", "[AES]") +{ + uint32_t buf[4]; + uint32_t output[4]; + uint8_t key[32]; + esp_aes_t ctx; + uint32_t encode_time = 0, decode_time = 0; + extern uint32_t esp_get_time(void); + + memset(key, 0x11, sizeof(key)); + + for (int cnt = 0; cnt < TEST_AES_COUNT; cnt++) { + int index = 0; + + for (int i = 128; i <= 256; i += 64) { + memset(buf, 11, sizeof(buf)); + memset(output, 0, sizeof(output)); + + uint32_t tmp = esp_get_time(); + + TEST_ASSERT_TRUE(esp_aes_set_encrypt_key(&ctx, key, i) == 0); + + TEST_ASSERT_TRUE(esp_aes_encrypt_ecb(&ctx, buf, output) == 0); + + encode_time += esp_get_time() - tmp; + +#if TEST_AES_DEBUG + for (int j = 0; j < 4; j++) + printf("0x%08x(0x%08x), ", output[j], s_aes_ecb_result[index][j]); + printf("\n"); +#endif + + TEST_ASSERT_TRUE(memcmp(output, s_aes_ecb_result[index++], sizeof(output)) == 0); + + tmp = esp_get_time(); + + TEST_ASSERT_TRUE(esp_aes_set_decrypt_key(&ctx, key, i) == 0); + + TEST_ASSERT_TRUE(esp_aes_decrypt_ecb(&ctx, output, buf) == 0); + + decode_time += esp_get_time() - tmp; + + memset(output, 11, sizeof(output)); + +#if TEST_AES_DEBUG + for (int j = 0; j < 4; j++) + printf("0x%08x(0x%08x), ", buf[j], output[j]); + printf("\n"); +#endif + + TEST_ASSERT_TRUE(memcmp(output, buf, sizeof(output)) == 0); + } + } + +#if TEST_AES_DEBUG_TIME + printf("AES-ECB test cost time totally encode %u us and decode %u us, once cost is about encode %u us and decode %u us\n", + encode_time, decode_time, encode_time / TEST_AES_COUNT, decode_time / TEST_AES_COUNT); +#endif +} + +// align: AES-ECB test cost time totally encode 370305 us and decode 905644 us, once cost is about encode 361 us and decode 884 us +// no align: AES-ECB test cost time totally encode 513219 us and decode 1729937 us, once cost is about encode 501 us and decode 1689 us + +TEST_CASE("Test AES-CBC", "[AES]") +{ + uint32_t buf[8]; + uint32_t output[8]; + uint8_t key[32]; + uint8_t iv[16]; + esp_aes_t ctx; + uint32_t encode_time = 0, decode_time = 0; + extern uint32_t esp_get_time(void); + + memset(key, 0x11, sizeof(key)); + + for (int cnt = 0; cnt < TEST_AES_COUNT; cnt++) { + int index = 0; + + for (int i = 128; i <= 256; i += 64) { + memset(buf, 11, sizeof(buf)); + memset(output, 0, sizeof(output)); + memset(iv, 0x11, sizeof(iv)); + + uint32_t tmp = esp_get_time(); + + TEST_ASSERT_TRUE(esp_aes_set_encrypt_key(&ctx, key, i) == 0); + + TEST_ASSERT_TRUE(esp_aes_encrypt_cbc(&ctx, buf, sizeof(buf), output, sizeof(output), iv) == 0); + + encode_time += esp_get_time() - tmp; + +#if TEST_AES_DEBUG + for (int j = 0; j < 8; j++) + printf("0x%08x(0x%08x), ", output[j], s_aes_cbc_result[index][j]); + printf("\n"); +#endif + + TEST_ASSERT_TRUE(memcmp(output, s_aes_cbc_result[index++], sizeof(output)) == 0); + + memset(iv, 0x11, sizeof(iv)); + + tmp = esp_get_time(); + + TEST_ASSERT_TRUE(esp_aes_set_decrypt_key(&ctx, key, i) == 0); + + TEST_ASSERT_TRUE(esp_aes_decrypt_cbc(&ctx, output, sizeof(output), buf, sizeof(buf), iv) == 0); + + decode_time += esp_get_time() - tmp; + + memset(output, 11, sizeof(output)); + +#if TEST_AES_DEBUG + for (int j = 0; j < 8; j++) + printf("0x%08x(0x%08x), ", buf[j], output[j]); + printf("\n"); +#endif + + TEST_ASSERT_TRUE(memcmp(output, buf, sizeof(output)) == 0); + } + } + +#if TEST_AES_DEBUG_TIME + printf("AES-CBC test cost time totally encode %u us and decode %u us, once cost is about encode %u us and decode %u us\n", + encode_time, decode_time, encode_time / TEST_AES_COUNT, decode_time / TEST_AES_COUNT); +#endif +} + +// align: AES-CBC test cost time totally encode 554920 us and decode 1098010 us, once cost is about encode 541 us and decode 1072 us +// no align: AES-CBC test cost time totally encode 730527 us and decode 2052009 us, once cost is about encode 713 us and decode 2003 us + + +TEST_CASE("Test AES-CFB8", "[AES]") +{ + uint32_t buf[8]; + uint32_t output[8]; + uint8_t key[32]; + uint8_t iv[16]; + esp_aes_t ctx; + uint32_t encode_time = 0, decode_time = 0; + extern uint32_t esp_get_time(void); + + memset(key, 0x11, sizeof(key)); + + for (int cnt = 0; cnt < TEST_AES_COUNT; cnt++) { + int index = 0; + + for (int i = 128; i <= 256; i += 64) { + memset(buf, 11, sizeof(buf)); + memset(output, 0, sizeof(output)); + memset(iv, 0x11, sizeof(iv)); + + uint32_t tmp = esp_get_time(); + + TEST_ASSERT_TRUE(esp_aes_set_encrypt_key(&ctx, key, i) == 0); + + TEST_ASSERT_TRUE(esp_aes_encrypt_cfb8(&ctx, buf, sizeof(buf), output, sizeof(output), iv) == 0); + + encode_time += esp_get_time() - tmp; + +#if TEST_AES_DEBUG + for (int j = 0; j < 8; j++) + printf("0x%08x(0x%08x), ", output[j], s_aes_cfb8_result[index][j]); + printf("\n"); +#endif + + TEST_ASSERT_TRUE(memcmp(output, s_aes_cfb8_result[index++], sizeof(output)) == 0); + + memset(iv, 0x11, sizeof(iv)); + + tmp = esp_get_time(); + + TEST_ASSERT_TRUE(esp_aes_set_encrypt_key(&ctx, key, i) == 0); + + TEST_ASSERT_TRUE(esp_aes_decrypt_cfb8(&ctx, output, sizeof(output), buf, sizeof(buf), iv) == 0); + + decode_time += esp_get_time() - tmp; + + memset(output, 11, sizeof(output)); + +#if TEST_AES_DEBUG + for (int j = 0; j < 8; j++) + printf("0x%08x(0x%08x), ", buf[j], output[j]); + printf("\n"); +#endif + TEST_ASSERT_TRUE(memcmp(output, buf, sizeof(output)) == 0); + } + } + +#if TEST_AES_DEBUG_TIME + printf("AES-CFB8 test cost time totally encode %u us and decode %u us, once cost is about encode %u us and decode %u us\n", + encode_time, decode_time, encode_time / TEST_AES_COUNT, decode_time / TEST_AES_COUNT); +#endif +} + +// align: AES-CFB8 test cost time totally encode 3951207 us and decode 3918394 us, once cost is about encode 3858 us and decode 3826 us +// no align: AES-CFB8 test cost time totally encode 5510528 us and decode 5481662 us, once cost is about encode 5381 us and decode 5353 us + + +TEST_CASE("Test AES-CFB128", "[AES]") +{ + uint32_t buf[8]; + uint32_t output[8]; + uint8_t key[32]; + uint8_t iv[16]; + size_t iv_off; + esp_aes_t ctx; + uint32_t encode_time = 0, decode_time = 0; + extern uint32_t esp_get_time(void); + + memset(key, 0x11, sizeof(key)); + + for (int cnt = 0; cnt < TEST_AES_COUNT; cnt++) { + int index = 0; + + for (int i = 128; i <= 256; i += 64) { + memset(buf, 11, sizeof(buf)); + memset(output, 0, sizeof(output)); + memset(iv, 0x11, sizeof(iv)); + iv_off = 0; + + uint32_t tmp = esp_get_time(); + + TEST_ASSERT_TRUE(esp_aes_set_encrypt_key(&ctx, key, i) == 0); + + TEST_ASSERT_TRUE(esp_aes_encrypt_cfb128(&ctx, buf, sizeof(buf), output, sizeof(output), iv, &iv_off) == 0); + + encode_time += esp_get_time() - tmp; + +#if TEST_AES_DEBUG + for (int j = 0; j < 8; j++) + printf("0x%08x(0x%08x), ", output[j], s_aes_cfb128_result[index][j]); + printf("\n"); +#endif + + TEST_ASSERT_TRUE(memcmp(output, s_aes_cfb128_result[index++], sizeof(output)) == 0); + + memset(iv, 0x11, sizeof(iv)); + iv_off = 0; + + tmp = esp_get_time(); + + TEST_ASSERT_TRUE(esp_aes_set_encrypt_key(&ctx, key, i) == 0); + + TEST_ASSERT_TRUE(esp_aes_decrypt_cfb128(&ctx, output, sizeof(output), buf, sizeof(buf), iv, &iv_off) == 0); + + decode_time += esp_get_time() - tmp; + + memset(output, 11, sizeof(output)); + +#if TEST_AES_DEBUG + for (int j = 0; j < 8; j++) + printf("0x%08x(0x%08x), ", buf[j], output[j]); + printf("\n"); +#endif + + TEST_ASSERT_TRUE(memcmp(output, buf, sizeof(buf)) == 0); + } + } + +#if TEST_AES_DEBUG_TIME + printf("AES-CFB128 test cost time totally encode %u us and decode %u us, once cost is about encode %u us and decode %u us\n", + encode_time, decode_time, encode_time / TEST_AES_COUNT, decode_time / TEST_AES_COUNT); +#endif +} + +// align: AES-CFB128 test cost time totally encode 468238 us and decode 423293 us, once cost is about encode 457 us and decode 413 us +// no align: AES-CFB128 test cost time totally encode 675645 us and decode 667750 us, once cost is about encode 659 us and decode 652 us + + +TEST_CASE("Test AES-CTR", "[AES]") +{ + uint32_t buf[8]; + uint32_t output[8]; + uint8_t key[32]; + uint8_t nonce_counter[16]; + uint8_t stream_block[16]; + size_t nc_off; + esp_aes_t ctx; + uint32_t encode_time = 0, decode_time = 0; + extern uint32_t esp_get_time(void); + + memset(key, 0x11, sizeof(key)); + + for (int cnt = 0; cnt < TEST_AES_COUNT; cnt++) { + int index = 0; + + for (int i = 128; i <= 256; i += 64) { + memset(buf, 11, sizeof(buf)); + memset(output, 0, sizeof(output)); + memset(nonce_counter, 0x11, sizeof(nonce_counter)); + nc_off = 0; + + uint32_t tmp = esp_get_time(); + + TEST_ASSERT_TRUE(esp_aes_set_encrypt_key(&ctx, key, i) == 0); + + TEST_ASSERT_TRUE(esp_aes_encrypt_ctr(&ctx, &nc_off, nonce_counter, stream_block, buf, sizeof(buf), output, sizeof(output)) == 0); + + encode_time += esp_get_time() - tmp; + +#if TEST_AES_DEBUG + for (int j = 0; j < 8; j++) + printf("0x%08x(0x%08x), ", output[j], s_aes_cfb128_result[index][j]); + printf("\n"); +#endif + + TEST_ASSERT_TRUE(memcmp(output, s_aes_ctr_result[index++], sizeof(output)) == 0); + + memset(nonce_counter, 0x11, sizeof(nonce_counter)); + nc_off = 0; + + tmp = esp_get_time(); + + TEST_ASSERT_TRUE(esp_aes_set_encrypt_key(&ctx, key, i) == 0); + + TEST_ASSERT_TRUE(esp_aes_decrypt_ctr(&ctx, &nc_off, nonce_counter, stream_block, output, sizeof(output), buf, sizeof(buf)) == 0); + + decode_time += esp_get_time() - tmp; + + memset(output, 11, sizeof(output)); + +#if TEST_AES_DEBUG + for (int j = 0; j < 8; j++) + printf("0x%08x(0x%08x), ", buf[j], output[j]); + printf("\n"); +#endif + + TEST_ASSERT_TRUE(memcmp(output, buf, sizeof(buf)) == 0); + } + } + +#if TEST_AES_DEBUG_TIME + printf("AES-CRT test cost time totally encode %u us and decode %u us, once cost is about encode %u us and decode %u us\n", + encode_time, decode_time, encode_time / TEST_AES_COUNT, decode_time / TEST_AES_COUNT); +#endif +} + +// align: AES-CRT test cost time totally encode 344845 us and decode 344421 us, once cost is about encode 336 us and decode 336 us +// no align: AES-CRT test cost time totally encode 639256 us and decode 635343 us, once cost is about encode 624 us and decode 620 us diff --git a/components/wpa_supplicant/src/crypto/aes-cbc.c b/components/wpa_supplicant/src/crypto/aes-cbc.c index 01620779..d0e4c822 100644 --- a/components/wpa_supplicant/src/crypto/aes-cbc.c +++ b/components/wpa_supplicant/src/crypto/aes-cbc.c @@ -13,6 +13,10 @@ * See README and COPYING for more details. */ +#include "sdkconfig.h" + +#ifndef CONFIG_ESP_AES + #include "crypto/includes.h" #include "crypto/common.h" @@ -86,3 +90,5 @@ aes_128_cbc_decrypt(const u8 *key, const u8 *iv, u8 *data, size_t data_len) aes_decrypt_deinit(ctx); return 0; } + +#endif /* CONFIG_ESP_AES */ diff --git a/components/wpa_supplicant/src/crypto/aes-internal-dec.c b/components/wpa_supplicant/src/crypto/aes-internal-dec.c index 46371c55..d532aae2 100644 --- a/components/wpa_supplicant/src/crypto/aes-internal-dec.c +++ b/components/wpa_supplicant/src/crypto/aes-internal-dec.c @@ -21,6 +21,10 @@ * See README and COPYING for more details. */ +#include "sdkconfig.h" + +#ifndef CONFIG_ESP_AES + #include "crypto/includes.h" #include "crypto/common.h" @@ -170,3 +174,41 @@ void aes_decrypt_deinit(void *ctx) os_memset(ctx, 0, AES_PRIV_SIZE); os_free(ctx); } + +#else /* CONFIG_ESP_AES */ + +#include +#include "esp_aes.h" +#include "esp_heap_caps.h" + +void *aes_decrypt_init(const uint8_t *key, size_t len) +{ + int ret; + esp_aes_t *ctx; + + ctx = heap_caps_malloc(sizeof(esp_aes_t), MALLOC_CAP_8BIT); + if (!ctx) + return NULL; + + ret = esp_aes_set_decrypt_key(ctx, key, len * 8); + if (ret) + goto fail; + + return ctx; + +fail: + heap_caps_free(ctx); + return NULL; +} + +void aes_decrypt(void *ctx, const uint8_t *crypt, uint8_t *plain) +{ + esp_aes_decrypt_ecb(ctx, crypt, plain); +} + +void aes_decrypt_deinit(void *ctx) +{ + heap_caps_free(ctx); +} + +#endif /* CONFIG_ESP_AES */ diff --git a/components/wpa_supplicant/src/crypto/aes-internal-enc.c b/components/wpa_supplicant/src/crypto/aes-internal-enc.c index 7b1080c4..0367a812 100644 --- a/components/wpa_supplicant/src/crypto/aes-internal-enc.c +++ b/components/wpa_supplicant/src/crypto/aes-internal-enc.c @@ -21,6 +21,10 @@ * See README and COPYING for more details. */ +#include "sdkconfig.h" + +#ifndef CONFIG_ESP_AES + #include "crypto/includes.h" #include "crypto/common.h" #include "crypto/crypto.h" @@ -132,3 +136,41 @@ void aes_encrypt_deinit(void *ctx) os_memset(ctx, 0, AES_PRIV_SIZE); os_free(ctx); } + +#else /* CONFIG_ESP_AES */ + +#include +#include "esp_aes.h" +#include "esp_heap_caps.h" + +void *aes_encrypt_init(const uint8_t *key, size_t len) +{ + int ret; + esp_aes_t *ctx; + + ctx = heap_caps_malloc(sizeof(esp_aes_t), MALLOC_CAP_8BIT); + if (!ctx) + return NULL; + + ret = esp_aes_set_encrypt_key(ctx, key, len * 8); + if (ret) + goto fail; + + return ctx; + +fail: + heap_caps_free(ctx); + return NULL; +} + +void aes_encrypt(void *ctx, const uint8_t *crypt, uint8_t *plain) +{ + esp_aes_encrypt_ecb(ctx, crypt, plain); +} + +void aes_encrypt_deinit(void *ctx) +{ + heap_caps_free(ctx); +} + +#endif /* CONFIG_ESP_AES */ diff --git a/components/wpa_supplicant/src/crypto/aes-internal.c b/components/wpa_supplicant/src/crypto/aes-internal.c index 9618239f..df91412f 100644 --- a/components/wpa_supplicant/src/crypto/aes-internal.c +++ b/components/wpa_supplicant/src/crypto/aes-internal.c @@ -54,6 +54,10 @@ * EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. */ +#include "sdkconfig.h" + +#ifndef CONFIG_ESP_AES + #define AES_SMALL_TABLES /* @@ -852,3 +856,5 @@ int rijndaelKeySetupEnc(u32 rk[], const u8 cipherKey[], int keyBits) return -1; } + +#endif /* CONFIG_ESP_AES */ diff --git a/components/wpa_supplicant/src/crypto/aes-unwrap.c b/components/wpa_supplicant/src/crypto/aes-unwrap.c index 4a92f1cd..aafbcfb1 100644 --- a/components/wpa_supplicant/src/crypto/aes-unwrap.c +++ b/components/wpa_supplicant/src/crypto/aes-unwrap.c @@ -13,6 +13,10 @@ * See README and COPYING for more details. */ +#include "sdkconfig.h" + +#ifndef CONFIG_ESP_AES + #include "crypto/includes.h" #include "crypto/common.h" @@ -78,3 +82,58 @@ aes_unwrap(const u8 *kek, int n, const u8 *cipher, u8 *plain) return 0; } + +#else +#include +#include "esp_aes.h" + +int aes_unwrap(const uint8_t *kek, int n, const uint8_t *cipher, uint8_t *plain) +{ + int ret; + uint8_t a[8], *r, b[16]; + esp_aes_t ctx; + + /* 1) Initialize variables. */ + memcpy(a, cipher, 8); + r = plain; + memcpy(r, cipher + 8, 8 * n); + + ret = esp_aes_set_decrypt_key(&ctx, kek, 128); + if (ret) + return ret; + + /* 2) Compute intermediate values. + * For j = 5 to 0 + * For i = n to 1 + * B = AES-1(K, (A ^ t) | R[i]) where t = n*j+i + * A = MSB(64, B) + * R[i] = LSB(64, B) + */ + for (int j = 5; j >= 0; j--) { + r = plain + (n - 1) * 8; + for (int i = n; i >= 1; i--) { + memcpy(b, a, 8); + b[7] ^= n * j + i; + + memcpy(b + 8, r, 8); + esp_aes_decrypt_ecb(&ctx, b, b); + memcpy(a, b, 8); + memcpy(r, b + 8, 8); + r -= 8; + } + } + + /* 3) Output results. + * + * These are already in @plain due to the location of temporary + * variables. Just verify that the IV matches with the expected value. + */ + for (int i = 0; i < 8; i++) { + if (a[i] != 0xa6) + return -1; + } + + return 0; +} + +#endif /* CONFIG_ESP_AES */ diff --git a/components/wpa_supplicant/src/crypto/aes-wrap.c b/components/wpa_supplicant/src/crypto/aes-wrap.c index 388dd97a..8083ac0b 100644 --- a/components/wpa_supplicant/src/crypto/aes-wrap.c +++ b/components/wpa_supplicant/src/crypto/aes-wrap.c @@ -7,6 +7,10 @@ * See README for more details. */ +#include "sdkconfig.h" + +#ifndef CONFIG_ESP_AES + #include "crypto/includes.h" #include "crypto/common.h" @@ -68,3 +72,54 @@ int aes_wrap(const u8 *kek, int n, const u8 *plain, u8 *cipher) return 0; } +#else +#include +#include "esp_aes.h" + +int aes_wrap(const uint8_t *kek, int n, const uint8_t *plain, uint8_t *cipher) +{ + int ret; + uint8_t *a, *r, b[16]; + esp_aes_t ctx; + + a = cipher; + r = cipher + 8; + + /* 1) Initialize variables. */ + memset(a, 0xa6, 8); + memcpy(r, plain, 8 * n); + + ret = esp_aes_set_encrypt_key(&ctx, kek, 128); + if (ret) + return ret; + + /* 2) Calculate intermediate values. + * For j = 0 to 5 + * For i=1 to n + * B = AES(K, A | R[i]) + * A = MSB(64, B) ^ t where t = (n*j)+i + * R[i] = LSB(64, B) + */ + for (int j = 0; j <= 5; j++) { + r = cipher + 8; + for (int i = 1; i <= n; i++) { + memcpy(b, a, 8); + memcpy(b + 8, r, 8); + esp_aes_encrypt_ecb(&ctx, b, b); + memcpy(a, b, 8); + a[7] ^= n * j + i; + memcpy(r, b + 8, 8); + r += 8; + } + } + + /* 3) Output the results. + * + * These are already in @cipher due to the location of temporary + * variables. + */ + + return 0; +} + +#endif /* CONFIG_ESP_AES */