mirror/mbedtls
1
0
Fork 0
mirror of https://github.com/Mbed-TLS/mbedtls.git synced 2025-02-04 06:40:03 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add Cipher Encrypt Fail multi-part case

Browse Source 
Signed-off-by: Neil Armstrong <narmstrong@baylibre.com>
This commit is contained in:
Neil Armstrong 2022-02-07 15:19:29 +01:00
parent 3ee335dbe3
commit d8dba4e0aa
1 changed files with 41 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
tests/suites/test_suite_psa_crypto.function
View File

@ -3040,9 +3040,14 @@ void cipher_encrypt_fail( int alg_arg,
psa_key_type_t key_type = key_type_arg; psa_key_type_t key_type = key_type_arg;
psa_algorithm_t alg = alg_arg; psa_algorithm_t alg = alg_arg;
psa_status_t expected_status = expected_status_arg; psa_status_t expected_status = expected_status_arg;
unsigned char iv[PSA_CIPHER_IV_MAX_SIZE] = {0};
size_t iv_size = PSA_CIPHER_IV_MAX_SIZE;
size_t iv_length = 0;
unsigned char *output = NULL; unsigned char *output = NULL;
size_t output_buffer_size = 0; size_t output_buffer_size = 0;
size_t output_length = 0; size_t output_length = 0;
size_t function_output_length;
psa_cipher_operation_t operation = PSA_CIPHER_OPERATION_INIT;
psa_key_attributes_t attributes = PSA_KEY_ATTRIBUTES_INIT; psa_key_attributes_t attributes = PSA_KEY_ATTRIBUTES_INIT;
if ( PSA_ERROR_BAD_STATE != expected_status ) if ( PSA_ERROR_BAD_STATE != expected_status )
@ -3061,12 +3066,48 @@ void cipher_encrypt_fail( int alg_arg,
&key ) ); &key ) );
} }
/* Encrypt, one-shot */
status = psa_cipher_encrypt( key, alg, input->x, input->len, output, status = psa_cipher_encrypt( key, alg, input->x, input->len, output,
output_buffer_size, &output_length ); output_buffer_size, &output_length );
TEST_EQUAL( status, expected_status ); TEST_EQUAL( status, expected_status );
/* Encrypt, multi-part */
status = psa_cipher_encrypt_setup( &operation, key, alg );
if( status == PSA_SUCCESS )
{
if( alg != PSA_ALG_ECB_NO_PADDING )
{
PSA_ASSERT( psa_cipher_generate_iv( &operation,
iv, iv_size,
&iv_length ) );
}
status = psa_cipher_update( &operation, input->x, input->len,
output, output_buffer_size,
&function_output_length );
if( status == PSA_SUCCESS )
{
output_length += function_output_length;
status = psa_cipher_finish( &operation, output + output_length,
output_buffer_size - output_length,
&function_output_length );
TEST_EQUAL( status, expected_status );
}
else
{
TEST_EQUAL( status, expected_status );
}
}
else
{
TEST_EQUAL( status, expected_status );
}
exit: exit:
psa_cipher_abort( &operation );
mbedtls_free( output ); mbedtls_free( output );
psa_destroy_key( key ); psa_destroy_key( key );
PSA_DONE( ); PSA_DONE( );