yuzu/mbedtls
1
0
Fork 0
mirror of https://github.com/yuzu-emu/mbedtls.git synced 2025-01-23 13:31:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix test output size

Browse source 
1. set output size to safe value
2. set output size correctly
3. check correct length of actual output
This commit is contained in:
Nir Sonnenschein 2018-06-04 16:31:13 +03:00 committed by itayzafrir
parent 553b8f39e9
commit d70bc48630
1 changed files with 12 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
tests/suites/test_suite_psa_crypto.function
View file

@ -638,14 +638,16 @@ void asymmetric_encrypt( int key_type_arg, char *key_hex,
unsigned char *input_data = NULL;
size_t input_size;
unsigned char *output = NULL;
size_t output_size = 4096;
size_t output_size = 0;
size_t output_length = 0;
unsigned char *output2 = NULL;
size_t output2_size = 4096;
size_t output2_size = 0;
size_t output2_length = 0;
key_data = unhexify_alloc( key_hex, &key_size );
TEST_ASSERT( key_data != NULL );
output_size = key_size;
output2_size = key_size;
input_data = unhexify_alloc( input_hex, &input_size );
TEST_ASSERT( input_data != NULL );
output = mbedtls_calloc( 1, output_size );
@ -704,19 +706,19 @@ void asymmetric_encrypt_fail( int key_type_arg, char *key_hex,
unsigned char *input_data = NULL;
size_t input_size;
unsigned char *output = NULL;
size_t output_size = 4096;
size_t output_size = 0;
size_t output_length = 0;
psa_status_t actual_status;
psa_status_t expected_status = expected_status_arg;
key_data = unhexify_alloc( key_hex, &key_size );
TEST_ASSERT( key_data != NULL );
output_size = key_size;
input_data = unhexify_alloc( input_hex, &input_size );
TEST_ASSERT( input_data != NULL );
output = mbedtls_calloc( 1, output_size );
TEST_ASSERT( output != NULL );
TEST_ASSERT( psa_crypto_init( ) == PSA_SUCCESS );
TEST_ASSERT( psa_import_key( slot, key_type,
@ -756,11 +758,13 @@ void asymmetric_decrypt( int key_type_arg, char *key_hex,
unsigned char *expected_data = NULL;
size_t expected_data_size;
unsigned char *output = NULL;
size_t output_size = 4096;
size_t output_size = 0;
size_t output_length = 0;
psa_key_policy_t policy = {0};
key_data = unhexify_alloc( key_hex, &key_size );
TEST_ASSERT( key_data != NULL );
output_size = key_size;
input_data = unhexify_alloc( input_hex, &input_size );
TEST_ASSERT( input_data != NULL );
expected_data = unhexify_alloc( expected_hex, &expected_data_size );
@ -768,7 +772,6 @@ void asymmetric_decrypt( int key_type_arg, char *key_hex,
output = mbedtls_calloc( 1, output_size );
TEST_ASSERT( output != NULL );
TEST_ASSERT( psa_crypto_init( ) == PSA_SUCCESS );
TEST_ASSERT( psa_import_key( slot, key_type,
@ -782,7 +785,7 @@ void asymmetric_decrypt( int key_type_arg, char *key_hex,
output_size,
&output_length) == PSA_SUCCESS );
TEST_ASSERT( ((size_t)expected_size) == output_length );
TEST_ASSERT( memcmp( expected_data, output, (output_length/8) ) == 0 );
TEST_ASSERT( memcmp( expected_data, output, (output_length) ) == 0 );
exit:
psa_destroy_key( slot );
@ -811,19 +814,19 @@ void asymmetric_decrypt_fail( int key_type_arg, char *key_hex,
unsigned char *input_data = NULL;
size_t input_size;
unsigned char *output = NULL;
size_t output_size = 4096;
size_t output_size = 0;
size_t output_length = 0;
psa_status_t actual_status;
psa_status_t expected_status = expected_status_arg;
key_data = unhexify_alloc( key_hex, &key_size );
TEST_ASSERT( key_data != NULL );
output_size = key_size;
input_data = unhexify_alloc( input_hex, &input_size );
TEST_ASSERT( input_data != NULL );
output = mbedtls_calloc( 1, output_size );
TEST_ASSERT( output != NULL );
TEST_ASSERT( psa_crypto_init( ) == PSA_SUCCESS );
TEST_ASSERT( psa_import_key( slot, key_type,