yuzu/mbedtls
1
0
Fork 0
mirror of https://github.com/yuzu-emu/mbedtls.git synced 2024-12-25 14:25:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

key_agreement_capacity: test the actual capacity as well

Browse source 
After testing that the advertized capacity is what the test data says,
read that many bytes to test that this is also actual capacity.
This commit is contained in:
Gilles Peskine 2018-10-25 22:36:12 +02:00
parent 10df341436
commit bf49197c9b
1 changed files with 16 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
tests/suites/test_suite_psa_crypto.function
View file

@ -3821,6 +3821,7 @@ void key_agreement_capacity( int alg_arg,
psa_crypto_generator_t generator = PSA_CRYPTO_GENERATOR_INIT; psa_crypto_generator_t generator = PSA_CRYPTO_GENERATOR_INIT;
psa_key_policy_t policy; psa_key_policy_t policy;
size_t actual_capacity; size_t actual_capacity;
unsigned char output[16];
TEST_ASSERT( psa_crypto_init( ) == PSA_SUCCESS ); TEST_ASSERT( psa_crypto_init( ) == PSA_SUCCESS );
@ -3836,10 +3837,25 @@ void key_agreement_capacity( int alg_arg,
peer_key_data->x, peer_key_data->len, peer_key_data->x, peer_key_data->len,
alg ) == PSA_SUCCESS ); alg ) == PSA_SUCCESS );
/* Test the advertized capacity. */
TEST_ASSERT( psa_get_generator_capacity( TEST_ASSERT( psa_get_generator_capacity(
&generator, &actual_capacity ) == PSA_SUCCESS ); &generator, &actual_capacity ) == PSA_SUCCESS );
TEST_ASSERT( actual_capacity == (size_t) expected_capacity_arg ); TEST_ASSERT( actual_capacity == (size_t) expected_capacity_arg );
/* Test the actual capacity by reading the output. */
while( actual_capacity > sizeof( output ) )
{
TEST_ASSERT( psa_generator_read( &generator,
output, sizeof( output ) ) ==
PSA_SUCCESS );
actual_capacity -= sizeof( output );
}
TEST_ASSERT( psa_generator_read( &generator,
output, actual_capacity ) ==
PSA_SUCCESS );
TEST_ASSERT( psa_generator_read( &generator, output, 1 ) ==
PSA_ERROR_INSUFFICIENT_CAPACITY );
exit: exit:
psa_generator_abort( &generator ); psa_generator_abort( &generator );
psa_destroy_key( our_key ); psa_destroy_key( our_key );