Update test to cover move-decrypt sequence
Signed-off-by: Werner Lewis <werner.lewis@arm.com>
This commit is contained in:
parent
7656a373b6
commit
ec0193d019
1 changed files with 7 additions and 6 deletions
|
@ -464,7 +464,7 @@ void aes_misc_params( )
|
|||
void aes_ecb_copy_context( data_t * key_str, data_t * src_str )
|
||||
{
|
||||
unsigned char output1[16], output2[16], plain[16];
|
||||
mbedtls_aes_context ctx1, ctx2;
|
||||
mbedtls_aes_context ctx1, ctx2, ctx3;
|
||||
|
||||
// Set key and encrypt with original context
|
||||
mbedtls_aes_init( &ctx1 );
|
||||
|
@ -474,14 +474,15 @@ void aes_ecb_copy_context( data_t * key_str, data_t * src_str )
|
|||
src_str->x, output1 ) == 0 );
|
||||
|
||||
ctx2 = ctx1;
|
||||
TEST_ASSERT( mbedtls_aes_setkey_dec( &ctx1, key_str->x,
|
||||
key_str->len * 8 ) == 0 );
|
||||
ctx3 = ctx1;
|
||||
memset( &ctx1, 0, sizeof( ctx1 ) );
|
||||
|
||||
// Encrypt with copied context and decrypt
|
||||
// Encrypt and decrypt with copied context
|
||||
TEST_ASSERT( mbedtls_aes_crypt_ecb( &ctx2, MBEDTLS_AES_ENCRYPT,
|
||||
src_str->x, output2 ) == 0 );
|
||||
TEST_ASSERT( mbedtls_aes_setkey_dec( &ctx2, key_str->x,
|
||||
key_str->len * 8) == 0 );
|
||||
TEST_ASSERT( mbedtls_aes_crypt_ecb( &ctx2, MBEDTLS_AES_DECRYPT,
|
||||
TEST_ASSERT( mbedtls_aes_crypt_ecb( &ctx3, MBEDTLS_AES_DECRYPT,
|
||||
output1, plain ) == 0 );
|
||||
|
||||
TEST_ASSERT( mbedtls_test_hexcmp( output1, output2, 16, 16 ) == 0 );
|
||||
|
|
Loading…
Reference in a new issue