kopia lustrzana https://github.com/espressif/esp-idf
495 wiersze
16 KiB
C
495 wiersze
16 KiB
C
/* mbedTLS GCM test
|
|
*/
|
|
|
|
#include <string.h>
|
|
#include <stdio.h>
|
|
#include <stdbool.h>
|
|
#include <esp_system.h>
|
|
#include "mbedtls/aes.h"
|
|
#include "mbedtls/gcm.h"
|
|
#include "unity.h"
|
|
#include "sdkconfig.h"
|
|
#include "esp_heap_caps.h"
|
|
#include "test_utils.h"
|
|
#include "ccomp_timer.h"
|
|
#include "sys/param.h"
|
|
|
|
#if CONFIG_MBEDTLS_HARDWARE_AES
|
|
|
|
/*
|
|
Python example code for generating test vectors
|
|
|
|
import os, binascii
|
|
from cryptography.hazmat.primitives.ciphers.aead import AESGCM
|
|
|
|
def as_c_array(byte_arr):
|
|
hex_str = ''
|
|
for idx, byte in enumerate(byte_arr):
|
|
hex_str += "0x{:02x}, ".format(byte)
|
|
bytes_per_line = 8
|
|
if idx % bytes_per_line == bytes_per_line - 1:
|
|
hex_str += '\n'
|
|
|
|
return hex_str
|
|
|
|
key = b'\x44' * 16
|
|
iv = b'\xEE' * 16
|
|
data = b'\xAA' * 3200
|
|
aad = b'\x76' * 16
|
|
|
|
aesgcm = AESGCM(key)
|
|
|
|
ct = aesgcm.encrypt(iv, data, aad)
|
|
|
|
print(as_c_array(ct))
|
|
*/
|
|
|
|
TEST_CASE("mbedtls GCM stream test", "[aes-gcm]")
|
|
{
|
|
|
|
const unsigned SZ = 100;
|
|
mbedtls_gcm_context ctx;
|
|
uint8_t nonce[16];
|
|
uint8_t key[16];
|
|
uint8_t tag[16];
|
|
mbedtls_cipher_id_t cipher = MBEDTLS_CIPHER_ID_AES;
|
|
|
|
const uint8_t expected_cipher[] = {
|
|
0x03, 0x92, 0x13, 0x49, 0x1f, 0x1f, 0x24, 0x41,
|
|
0xe8, 0xeb, 0x89, 0x47, 0x50, 0x0a, 0xce, 0xa3,
|
|
0xc7, 0x1c, 0x10, 0x70, 0xb0, 0x89, 0x82, 0x5e,
|
|
0x0f, 0x4a, 0x23, 0xee, 0xd2, 0xfc, 0xff, 0x45,
|
|
0x61, 0x4c, 0xd1, 0xfb, 0x6d, 0xe2, 0xbe, 0x67,
|
|
0x6f, 0x94, 0x72, 0xa3, 0xe7, 0x04, 0x99, 0xb3,
|
|
0x4a, 0x46, 0xf9, 0x2b, 0xaf, 0xac, 0xa9, 0x0e,
|
|
0x43, 0x7e, 0x8b, 0xc4, 0xbf, 0x49, 0xa4, 0x83,
|
|
0x9c, 0x31, 0x11, 0x1c, 0x09, 0xac, 0x90, 0xdf,
|
|
0x00, 0x34, 0x08, 0xe5, 0x70, 0xa3, 0x7e, 0x4b,
|
|
0x36, 0x48, 0x5a, 0x3f, 0x28, 0xc7, 0x1c, 0xd9,
|
|
0x1b, 0x1b, 0x49, 0x96, 0xe9, 0x7c, 0xea, 0x54,
|
|
0x7c, 0x71, 0x29, 0x0d
|
|
};
|
|
const uint8_t expected_tag[] = {
|
|
0x35, 0x1c, 0x21, 0xc6, 0xbc, 0x6b, 0x18, 0x52,
|
|
0x90, 0xe1, 0xf2, 0x5b, 0xe1, 0xf6, 0x15, 0xee,
|
|
};
|
|
|
|
|
|
memset(nonce, 0x89, 16);
|
|
memset(key, 0x56, 16);
|
|
|
|
// allocate internal memory
|
|
uint8_t *chipertext = heap_caps_malloc(SZ, MALLOC_CAP_DMA | MALLOC_CAP_8BIT | MALLOC_CAP_INTERNAL);
|
|
uint8_t *plaintext = heap_caps_malloc(SZ, MALLOC_CAP_DMA | MALLOC_CAP_8BIT | MALLOC_CAP_INTERNAL);
|
|
uint8_t *decryptedtext = heap_caps_malloc(SZ, MALLOC_CAP_DMA | MALLOC_CAP_8BIT | MALLOC_CAP_INTERNAL);
|
|
|
|
TEST_ASSERT_NOT_NULL(chipertext);
|
|
TEST_ASSERT_NOT_NULL(plaintext);
|
|
TEST_ASSERT_NOT_NULL(decryptedtext);
|
|
|
|
memset(plaintext, 0xAB, SZ);
|
|
/* Test that all the end results are the same
|
|
no matter how many bytes we encrypt each call
|
|
*/
|
|
for (int bytes_to_process = 16; bytes_to_process < SZ; bytes_to_process = bytes_to_process + 16) {
|
|
memset(nonce, 0x89, 16);
|
|
memset(chipertext, 0x0, SZ);
|
|
memset(decryptedtext, 0x0, SZ);
|
|
memset(tag, 0x0, 16);
|
|
|
|
mbedtls_gcm_init(&ctx);
|
|
mbedtls_gcm_setkey(&ctx, cipher, key, 128);
|
|
mbedtls_gcm_starts( &ctx, MBEDTLS_AES_ENCRYPT, nonce, sizeof(nonce), NULL, 0 );
|
|
|
|
// Encrypt
|
|
for (int idx = 0; idx < SZ; idx = idx + bytes_to_process) {
|
|
// Limit length of last call to avoid exceeding buffer size
|
|
size_t length = (idx + bytes_to_process > SZ) ? (SZ - idx) : bytes_to_process;
|
|
mbedtls_gcm_update(&ctx, length, plaintext + idx, chipertext + idx );
|
|
}
|
|
mbedtls_gcm_finish( &ctx, tag, sizeof(tag) );
|
|
TEST_ASSERT_EQUAL_HEX8_ARRAY(expected_cipher, chipertext, SZ);
|
|
TEST_ASSERT_EQUAL_HEX8_ARRAY(expected_tag, tag, sizeof(tag));
|
|
|
|
// Decrypt
|
|
memset(nonce, 0x89, 16);
|
|
mbedtls_gcm_free( &ctx );
|
|
mbedtls_gcm_init(&ctx);
|
|
mbedtls_gcm_setkey(&ctx, cipher, key, 128);
|
|
mbedtls_gcm_starts( &ctx, MBEDTLS_AES_DECRYPT, nonce, sizeof(nonce), NULL, 0 );
|
|
|
|
for (int idx = 0; idx < SZ; idx = idx + bytes_to_process) {
|
|
// Limit length of last call to avoid exceeding buffer size
|
|
|
|
size_t length = (idx + bytes_to_process > SZ) ? (SZ - idx) : bytes_to_process;
|
|
mbedtls_gcm_update(&ctx, length, chipertext + idx, decryptedtext + idx );
|
|
}
|
|
mbedtls_gcm_finish( &ctx, tag, sizeof(tag) );
|
|
TEST_ASSERT_EQUAL_HEX8_ARRAY(plaintext, decryptedtext, SZ);
|
|
mbedtls_gcm_free( &ctx );
|
|
}
|
|
free(plaintext);
|
|
free(chipertext);
|
|
free(decryptedtext);
|
|
}
|
|
|
|
TEST_CASE("mbedtls AES GCM self-tests", "[aes-gcm]")
|
|
{
|
|
TEST_ASSERT_FALSE_MESSAGE(mbedtls_gcm_self_test(1), "AES GCM self-test should pass.");
|
|
}
|
|
|
|
typedef struct {
|
|
uint8_t *plaintext;
|
|
size_t plaintext_length;
|
|
uint32_t output_caps;
|
|
uint8_t *add_buf;
|
|
size_t add_length;
|
|
uint8_t *iv;
|
|
size_t iv_length;
|
|
uint8_t *key;
|
|
size_t key_bits;
|
|
size_t tag_len;
|
|
} aes_gcm_test_cfg_t;
|
|
|
|
typedef struct {
|
|
const uint8_t *expected_tag;
|
|
const uint8_t *ciphertext_last_block; // Last block of the chipertext
|
|
} aes_gcm_test_expected_res_t;
|
|
|
|
|
|
typedef enum {
|
|
AES_GCM_TEST_CRYPT_N_TAG,
|
|
AES_GCM_TEST_START_UPDATE_FINISH,
|
|
} aes_gcm_test_type_t;
|
|
|
|
static void aes_gcm_test(aes_gcm_test_cfg_t *cfg, aes_gcm_test_expected_res_t *res, aes_gcm_test_type_t aes_gcm_type)
|
|
{
|
|
mbedtls_cipher_id_t cipher = MBEDTLS_CIPHER_ID_AES;
|
|
mbedtls_gcm_context ctx;
|
|
|
|
uint8_t tag_buf_encrypt[16] = {};
|
|
uint8_t tag_buf_decrypt[16] = {};
|
|
uint8_t iv_buf[16] = {};
|
|
|
|
uint8_t *ciphertext = heap_caps_malloc(cfg->plaintext_length, cfg->output_caps);
|
|
uint8_t *output = heap_caps_malloc(cfg->plaintext_length, MALLOC_CAP_DMA | MALLOC_CAP_8BIT | MALLOC_CAP_INTERNAL);
|
|
|
|
if (cfg->plaintext_length != 0) {
|
|
TEST_ASSERT_NOT_NULL(ciphertext);
|
|
TEST_ASSERT_NOT_NULL(output);
|
|
}
|
|
|
|
memset(ciphertext, 0, cfg->plaintext_length);
|
|
memset(output, 0, cfg->plaintext_length);
|
|
memcpy(iv_buf, cfg->iv, cfg->iv_length);
|
|
|
|
mbedtls_gcm_init(&ctx);
|
|
mbedtls_gcm_setkey(&ctx, cipher, cfg->key, cfg->key_bits);
|
|
|
|
/* Encrypt and tag */
|
|
if (aes_gcm_type == AES_GCM_TEST_CRYPT_N_TAG) {
|
|
mbedtls_gcm_crypt_and_tag(&ctx, MBEDTLS_AES_ENCRYPT, cfg->plaintext_length, iv_buf, cfg->iv_length, cfg->add_buf, cfg->add_length, cfg->plaintext, ciphertext, cfg->tag_len, tag_buf_encrypt);
|
|
} else if (aes_gcm_type == AES_GCM_TEST_START_UPDATE_FINISH){
|
|
TEST_ASSERT(mbedtls_gcm_starts( &ctx, MBEDTLS_AES_ENCRYPT, iv_buf, cfg->iv_length, cfg->add_buf, cfg->add_length) == 0 );
|
|
TEST_ASSERT(mbedtls_gcm_update( &ctx, cfg->plaintext_length, cfg->plaintext, ciphertext) == 0 );
|
|
TEST_ASSERT(mbedtls_gcm_finish( &ctx, tag_buf_encrypt, cfg->tag_len) == 0 );
|
|
}
|
|
size_t offset = cfg->plaintext_length > 16 ? cfg->plaintext_length - 16 : 0;
|
|
/* Sanity check: make sure the last ciphertext block matches what we expect to see. */
|
|
TEST_ASSERT_EQUAL_HEX8_ARRAY(res->ciphertext_last_block, ciphertext + offset, MIN(16, cfg->plaintext_length));
|
|
TEST_ASSERT_EQUAL_HEX8_ARRAY(res->expected_tag, tag_buf_encrypt, cfg->tag_len);
|
|
|
|
|
|
/* Decrypt and authenticate */
|
|
if (aes_gcm_type == AES_GCM_TEST_CRYPT_N_TAG) {
|
|
TEST_ASSERT(mbedtls_gcm_auth_decrypt(&ctx, cfg->plaintext_length, iv_buf, cfg->iv_length, cfg->add_buf, cfg->add_length, res->expected_tag, cfg->tag_len, ciphertext, output) == 0);
|
|
} else if (aes_gcm_type == AES_GCM_TEST_START_UPDATE_FINISH){
|
|
TEST_ASSERT(mbedtls_gcm_starts( &ctx, MBEDTLS_AES_DECRYPT, iv_buf, cfg->iv_length, cfg->add_buf, cfg->add_length) == 0 );
|
|
TEST_ASSERT(mbedtls_gcm_update( &ctx, cfg->plaintext_length, ciphertext, output) == 0 );
|
|
TEST_ASSERT(mbedtls_gcm_finish( &ctx, tag_buf_decrypt, cfg->tag_len) == 0 );
|
|
|
|
/* mbedtls_gcm_auth_decrypt already checks tag so only needed for AES_GCM_TEST_START_UPDATE_FINISH */
|
|
TEST_ASSERT_EQUAL_HEX8_ARRAY(res->expected_tag, tag_buf_decrypt, cfg->tag_len);
|
|
}
|
|
|
|
TEST_ASSERT_EQUAL_HEX8_ARRAY(cfg->plaintext, output, cfg->plaintext_length);
|
|
|
|
free(ciphertext);
|
|
free(output);
|
|
}
|
|
|
|
|
|
|
|
TEST_CASE("mbedtls AES GCM", "[aes-gcm]")
|
|
{
|
|
uint8_t iv[16];
|
|
uint8_t key[16];
|
|
uint8_t add[30];
|
|
|
|
memset(iv, 0xB1, sizeof(iv));
|
|
memset(key, 0x27, sizeof(key));
|
|
memset(add, 0x90, sizeof(add));
|
|
|
|
size_t length[] = {10, 16, 500, 5000, 12345};
|
|
|
|
const uint8_t expected_last_block[][16] = {
|
|
|
|
{0x37, 0x99, 0x4b, 0x16, 0x5f, 0x8d, 0x27, 0xb1,
|
|
0x60, 0x72},
|
|
|
|
{0x37, 0x99, 0x4b, 0x16, 0x5f, 0x8d, 0x27, 0xb1,
|
|
0x60, 0x72, 0x9a, 0x81, 0x8d, 0x3c, 0x69, 0x66},
|
|
|
|
{0x9d, 0x7a, 0xac, 0x84,0xe3, 0x70, 0x43, 0x0f,
|
|
0xa7, 0x83, 0x43, 0xc9, 0x04, 0xf8, 0x7d, 0x48},
|
|
|
|
{0xee, 0xfd, 0xab, 0x2a, 0x09, 0x44, 0x41, 0x6a,
|
|
0x91, 0xb0, 0x74, 0x24, 0xee, 0x35, 0xb1, 0x39},
|
|
|
|
{0x51, 0xf7, 0x1f, 0x67, 0x1a, 0x4a, 0x12, 0x37,
|
|
0x60, 0x3b, 0x68, 0x01, 0x20, 0x4f, 0xf3, 0xd9},
|
|
};
|
|
|
|
const uint8_t expected_tag[][16] = {
|
|
|
|
{0x06, 0x4f, 0xb5, 0x91, 0x12, 0x24, 0xb4, 0x24,
|
|
0x0b, 0xc2, 0x85, 0x59, 0x6a, 0x7c, 0x1f, 0xc9},
|
|
|
|
{0x45, 0xc2, 0xa8, 0xfe, 0xff, 0x49, 0x1f, 0x45,
|
|
0x8e, 0x29, 0x74, 0x41, 0xed, 0x9b, 0x54, 0x28},
|
|
|
|
{0xe1, 0xf9, 0x40, 0xfa, 0x29, 0x6f, 0x30, 0xae,
|
|
0xb6, 0x9b, 0x33, 0xdb, 0x8a, 0xf9, 0x70, 0xc4},
|
|
|
|
{0x22, 0xe1, 0x22, 0x34, 0x0c, 0x91, 0x0b, 0xcf,
|
|
0xa3, 0x42, 0xe0, 0x48, 0xe6, 0xfe, 0x2e, 0x28},
|
|
|
|
{0xfb, 0xfe, 0x5a, 0xed, 0x26, 0x5c, 0x5e, 0x66,
|
|
0x4e, 0xb2, 0x48, 0xce, 0xe9, 0x88, 0x1c, 0xe0},
|
|
};
|
|
|
|
aes_gcm_test_cfg_t cfg = {
|
|
.output_caps = MALLOC_CAP_DMA | MALLOC_CAP_8BIT | MALLOC_CAP_INTERNAL,
|
|
.iv = iv,
|
|
.iv_length = sizeof(iv),
|
|
.key = key,
|
|
.key_bits = 8*sizeof(key),
|
|
.add_buf = add,
|
|
.add_length = sizeof(add),
|
|
.tag_len = 16
|
|
};
|
|
|
|
aes_gcm_test_expected_res_t res = {
|
|
};
|
|
|
|
for (int i = 0; i < sizeof(length)/sizeof(length[0]); i++) {
|
|
printf("Test AES-GCM with plaintext length = %d\n", length[i]);
|
|
uint8_t *input = heap_caps_malloc(length[i], MALLOC_CAP_DMA | MALLOC_CAP_8BIT | MALLOC_CAP_INTERNAL);
|
|
TEST_ASSERT(input != NULL || length[i] == 0);
|
|
memset(input, 0x36, length[i]);
|
|
|
|
cfg.plaintext = input;
|
|
cfg.plaintext_length = length[i];
|
|
res.expected_tag = expected_tag[i];
|
|
res.ciphertext_last_block = expected_last_block[i],
|
|
|
|
aes_gcm_test(&cfg, &res, AES_GCM_TEST_CRYPT_N_TAG);
|
|
aes_gcm_test(&cfg, &res, AES_GCM_TEST_START_UPDATE_FINISH);
|
|
|
|
free(input);
|
|
}
|
|
}
|
|
|
|
|
|
TEST_CASE("mbedtls AES GCM - Different add messages", "[aes-gcm]")
|
|
{
|
|
const unsigned CALL_SZ = 160;
|
|
uint8_t iv[16];
|
|
uint8_t key[16];
|
|
uint8_t *input = heap_caps_malloc(CALL_SZ, MALLOC_CAP_DMA | MALLOC_CAP_8BIT | MALLOC_CAP_INTERNAL);
|
|
TEST_ASSERT_NOT_NULL(input);
|
|
|
|
memset(input, 0x67, CALL_SZ);
|
|
memset(iv, 0xA2, sizeof(iv));
|
|
memset(key, 0x48, sizeof(key));
|
|
|
|
const uint8_t expected_last_block[] = {
|
|
0xcd, 0xb9, 0xad, 0x6f, 0xc9, 0x35, 0x21, 0x0d,
|
|
0xc9, 0x5d, 0xea, 0xd9, 0xf7, 0x1d, 0x43, 0xed
|
|
};
|
|
|
|
size_t add_len[] = {0, 10, 16, 500, 5000};
|
|
|
|
const uint8_t expected_tag[][16] = {
|
|
{0xe3, 0x91, 0xad, 0x40, 0x96, 0xb7, 0x8c, 0x53,
|
|
0x4d, 0x15, 0x7d, 0x55, 0x15, 0xdf, 0x10, 0x69},
|
|
|
|
{0xc2, 0x38, 0x36, 0xe9, 0x12, 0x72, 0x5b, 0x31,
|
|
0x0c, 0xde, 0xb5, 0xc9, 0x8c, 0xa3, 0xcb, 0xe7},
|
|
|
|
{0x57, 0x10, 0x22, 0x91, 0x65, 0xfa, 0x89, 0xba,
|
|
0x0a, 0x3e, 0xc1, 0x7c, 0x93, 0x6e, 0x35, 0xac},
|
|
|
|
{0x3c, 0x28, 0x03, 0xc2, 0x14, 0x40, 0xec, 0xb6,
|
|
0x25, 0xfb, 0xdd, 0x55, 0xa0, 0xb2, 0x47, 0x7b},
|
|
|
|
{0xfa, 0x66, 0x4a, 0x97, 0x2d, 0x02, 0x32, 0x5b,
|
|
0x92, 0x94, 0xf1, 0x00, 0x1c, 0xfa, 0xe3, 0x07}
|
|
};
|
|
|
|
aes_gcm_test_cfg_t cfg = {
|
|
.plaintext = input,
|
|
.plaintext_length = CALL_SZ,
|
|
.output_caps = MALLOC_CAP_DMA | MALLOC_CAP_8BIT | MALLOC_CAP_INTERNAL,
|
|
.iv = iv,
|
|
.iv_length = sizeof(iv),
|
|
.key = key,
|
|
.key_bits = 8*sizeof(key),
|
|
.tag_len = 16
|
|
};
|
|
|
|
aes_gcm_test_expected_res_t res = {
|
|
.ciphertext_last_block = expected_last_block,
|
|
};
|
|
|
|
for (int i = 0; i < sizeof(add_len)/sizeof(add_len[0]); i++) {
|
|
printf("Test AES-GCM with add length = %d\n", add_len[i]);
|
|
uint8_t *add = heap_caps_malloc(add_len[i], MALLOC_CAP_DMA | MALLOC_CAP_8BIT | MALLOC_CAP_INTERNAL);
|
|
TEST_ASSERT(add != NULL || add_len[i] == 0);
|
|
memset(add, 0x12, add_len[i]);
|
|
|
|
cfg.add_buf = add;
|
|
cfg.add_length = add_len[i];
|
|
res.expected_tag = expected_tag[i];
|
|
|
|
aes_gcm_test(&cfg, &res, AES_GCM_TEST_CRYPT_N_TAG);
|
|
aes_gcm_test(&cfg, &res, AES_GCM_TEST_START_UPDATE_FINISH);
|
|
|
|
free(add);
|
|
}
|
|
free(input);
|
|
}
|
|
|
|
|
|
|
|
TEST_CASE("mbedtls AES GCM performance, start, update, ret", "[aes-gcm]")
|
|
{
|
|
const unsigned CALL_SZ = 16*3200;
|
|
mbedtls_gcm_context ctx;
|
|
float elapsed_usec;
|
|
unsigned char tag_buf[16];
|
|
mbedtls_cipher_id_t cipher = MBEDTLS_CIPHER_ID_AES;
|
|
uint8_t iv[16];
|
|
uint8_t key[16];
|
|
uint8_t aad[16];
|
|
|
|
memset(iv, 0xEE, 16);
|
|
memset(key, 0x44, 16);
|
|
memset(aad, 0x76, 16);
|
|
|
|
// allocate internal memory
|
|
uint8_t *buf = heap_caps_malloc(CALL_SZ, MALLOC_CAP_DMA | MALLOC_CAP_8BIT | MALLOC_CAP_INTERNAL);
|
|
TEST_ASSERT_NOT_NULL(buf);
|
|
|
|
mbedtls_gcm_init(&ctx);
|
|
mbedtls_gcm_setkey( &ctx, cipher, key, 128);
|
|
|
|
ccomp_timer_start();
|
|
|
|
memset(buf, 0xAA, CALL_SZ);
|
|
|
|
TEST_ASSERT(mbedtls_gcm_starts( &ctx, MBEDTLS_AES_ENCRYPT, iv, sizeof(iv), aad, sizeof(aad) ) == 0 );
|
|
TEST_ASSERT(mbedtls_gcm_update( &ctx, CALL_SZ, buf, buf ) == 0 );
|
|
TEST_ASSERT(mbedtls_gcm_finish( &ctx, tag_buf, 16 ) == 0 );
|
|
|
|
elapsed_usec = ccomp_timer_stop();
|
|
|
|
/* Sanity check: make sure the last ciphertext block matches
|
|
what we expect to see.
|
|
*/
|
|
const uint8_t expected_last_block[] = {
|
|
0xd4, 0x25, 0x88, 0xd4, 0x32, 0x52, 0x3d, 0x6f,
|
|
0xae, 0x49, 0x19, 0xb5, 0x95, 0x01, 0xde, 0x7d,
|
|
};
|
|
|
|
const uint8_t expected_tag[] = {
|
|
0xf5, 0x10, 0x1f, 0x21, 0x5b, 0x07, 0x0d, 0x3f,
|
|
0xac, 0xc9, 0xd0, 0x42, 0x45, 0xef, 0xc7, 0xfa,
|
|
};
|
|
|
|
TEST_ASSERT_EQUAL_HEX8_ARRAY(expected_last_block, buf + CALL_SZ - 16 , 16);
|
|
TEST_ASSERT_EQUAL_HEX8_ARRAY(expected_tag, tag_buf, 16);
|
|
|
|
free(buf);
|
|
|
|
// bytes/usec = MB/sec
|
|
float mb_sec = CALL_SZ / elapsed_usec;
|
|
printf("GCM encryption rate %.3fMB/sec\n", mb_sec);
|
|
|
|
#ifdef CONFIG_MBEDTLS_HARDWARE_GCM
|
|
// Don't put a hard limit on software AES performance
|
|
TEST_PERFORMANCE_GREATER_THAN(AES_GCM_UPDATE_THROUGHPUT_MBSEC, "%.3fMB/sec", mb_sec);
|
|
#endif
|
|
}
|
|
|
|
|
|
TEST_CASE("mbedtls AES GCM performance, crypt-and-tag", "[aes-gcm]")
|
|
{
|
|
const unsigned CALL_SZ = 16*3200;
|
|
mbedtls_gcm_context ctx;
|
|
float elapsed_usec;
|
|
unsigned char tag_buf[16] = {};
|
|
mbedtls_cipher_id_t cipher = MBEDTLS_CIPHER_ID_AES;
|
|
uint8_t iv[16];
|
|
uint8_t key[16];
|
|
uint8_t aad[16];
|
|
|
|
memset(iv, 0xEE, 16);
|
|
memset(key, 0x44, 16);
|
|
memset(aad, 0x76, 16);
|
|
|
|
// allocate internal memory
|
|
uint8_t *buf = heap_caps_malloc(CALL_SZ, MALLOC_CAP_DMA | MALLOC_CAP_8BIT | MALLOC_CAP_INTERNAL);
|
|
TEST_ASSERT_NOT_NULL(buf);
|
|
|
|
mbedtls_gcm_init(&ctx);
|
|
mbedtls_gcm_setkey( &ctx, cipher, key, 128);
|
|
|
|
memset(buf, 0xAA, CALL_SZ);
|
|
|
|
ccomp_timer_start();
|
|
mbedtls_gcm_crypt_and_tag(&ctx, MBEDTLS_AES_ENCRYPT, CALL_SZ, iv, sizeof(iv), aad, sizeof(aad), buf, buf, 16, tag_buf);
|
|
|
|
elapsed_usec = ccomp_timer_stop();
|
|
|
|
/* Sanity check: make sure the last ciphertext block matches
|
|
what we expect to see.
|
|
*/
|
|
|
|
const uint8_t expected_last_block[] = {
|
|
0xd4, 0x25, 0x88, 0xd4, 0x32, 0x52, 0x3d, 0x6f,
|
|
0xae, 0x49, 0x19, 0xb5, 0x95, 0x01, 0xde, 0x7d,
|
|
};
|
|
|
|
const uint8_t expected_tag[] = {
|
|
0xf5, 0x10, 0x1f, 0x21, 0x5b, 0x07, 0x0d, 0x3f,
|
|
0xac, 0xc9, 0xd0, 0x42, 0x45, 0xef, 0xc7, 0xfa,
|
|
};
|
|
|
|
TEST_ASSERT_EQUAL_HEX8_ARRAY(expected_last_block, buf + CALL_SZ - 16 , 16);
|
|
TEST_ASSERT_EQUAL_HEX8_ARRAY(expected_tag, tag_buf, 16);
|
|
|
|
free(buf);
|
|
|
|
// bytes/usec = MB/sec
|
|
float mb_sec = CALL_SZ / elapsed_usec;
|
|
printf("GCM encryption rate %.3fMB/sec\n", mb_sec);
|
|
|
|
#ifdef CONFIG_MBEDTLS_HARDWARE_GCM
|
|
// Don't put a hard limit on software AES performance
|
|
TEST_PERFORMANCE_GREATER_THAN(AES_GCM_CRYPT_TAG_THROUGHPUT_MBSEC, "%.3fMB/sec", mb_sec);
|
|
#endif
|
|
}
|
|
|
|
#endif //CONFIG_MBEDTLS_HARDWARE_AES
|