mirror of
				https://github.com/cuberite/polarssl.git
				synced 2025-11-03 20:22:59 -05:00 
			
		
		
		
	add test scenarios to decrypt and encrypt input and compare with given output
This commit is contained in:
		
							parent
							
								
									89e0f468bf
								
							
						
					
					
						commit
						b152d4d8b6
					
				@ -1365,7 +1365,7 @@ static psa_status_t psa_cipher_setup(psa_cipher_operation_t *operation,
 | 
			
		||||
 | 
			
		||||
    operation->key_set = 1;
 | 
			
		||||
    operation->alg = alg;
 | 
			
		||||
    operation->block_size = PSA_BLOCK_CIPHER_BLOCK_SIZE(key_type);
 | 
			
		||||
    operation->block_size = PSA_ALG_IS_BLOCK_CIPHER( alg ) ? PSA_BLOCK_CIPHER_BLOCK_SIZE( key_type ) : 1;
 | 
			
		||||
    if( PSA_ALG_IS_BLOCK_CIPHER( alg ) || ( alg == PSA_ALG_CTR ) )
 | 
			
		||||
    {
 | 
			
		||||
        operation->iv_size = PSA_BLOCK_CIPHER_BLOCK_SIZE( key_type );
 | 
			
		||||
 | 
			
		||||
@ -107,7 +107,11 @@ key_lifetime_set_fail:1:PSA_KEY_LIFETIME_PERSISTENT+1:PSA_ERROR_INVALID_ARGUMENT
 | 
			
		||||
 | 
			
		||||
PSA Symmetric encryption: AES-128
 | 
			
		||||
depends_on:MBEDTLS_AES_C:MBEDTLS_CIPHER_MODE_CBC
 | 
			
		||||
cipher_test_positive:PSA_ALG_CBC_BASE:PSA_KEY_TYPE_AES:"2b7e151628aed2a6abf7158809cf4f3c":"6bc1bee22e409f96e93d7e117393172a"
 | 
			
		||||
cipher_test_encrypt:PSA_ALG_CBC_BASE:PSA_KEY_TYPE_AES:"2b7e151628aed2a6abf7158809cf4f3c":"6bc1bee22e409f96e93d7e117393172a":"a076ec9dfbe47d52afc357336f20743b"
 | 
			
		||||
 | 
			
		||||
PSA Symmetric encryption: AES-128
 | 
			
		||||
depends_on:MBEDTLS_AES_C:MBEDTLS_CIPHER_MODE_CBC
 | 
			
		||||
cipher_test_decrypt:PSA_ALG_CBC_BASE:PSA_KEY_TYPE_AES:"2b7e151628aed2a6abf7158809cf4f3c":"396ee84fb75fdbb5c2b13c7fe5a654aa":"49e4e66c89a86b67758df89db9ad6955"
 | 
			
		||||
 | 
			
		||||
PSA Symmetric encryption/decryption: AES-128
 | 
			
		||||
depends_on:MBEDTLS_AES_C:MBEDTLS_CIPHER_MODE_CBC
 | 
			
		||||
 | 
			
		||||
@ -538,11 +538,10 @@ void cipher_test_positive( int alg_arg, int key_type_arg,
 | 
			
		||||
    unsigned char *key = NULL;
 | 
			
		||||
    size_t key_size;
 | 
			
		||||
    unsigned char iv[16] = {0};
 | 
			
		||||
    size_t iv_size = 16;
 | 
			
		||||
    size_t iv_length = 0;
 | 
			
		||||
    unsigned char *input = NULL;
 | 
			
		||||
    size_t input_size = 0;
 | 
			
		||||
    unsigned char *output ;
 | 
			
		||||
    unsigned char *output;
 | 
			
		||||
    unsigned char *expected_output;
 | 
			
		||||
    size_t output_size = 0;
 | 
			
		||||
    size_t output_length = 0;
 | 
			
		||||
    psa_cipher_operation_t operation;
 | 
			
		||||
@ -553,6 +552,11 @@ void cipher_test_positive( int alg_arg, int key_type_arg,
 | 
			
		||||
 | 
			
		||||
    input = unhexify_alloc( input_hex, &input_size );
 | 
			
		||||
    TEST_ASSERT( input != NULL );
 | 
			
		||||
 | 
			
		||||
    expected_output = unhexify_alloc( output_hex, &output_size );
 | 
			
		||||
    TEST_ASSERT( expected_output != NULL );
 | 
			
		||||
 | 
			
		||||
    memset( iv, 0x2a, sizeof( iv ) );
 | 
			
		||||
 
 | 
			
		||||
    TEST_ASSERT( psa_crypto_init( ) == PSA_SUCCESS );
 | 
			
		||||
 | 
			
		||||
@ -561,10 +565,9 @@ void cipher_test_positive( int alg_arg, int key_type_arg,
 | 
			
		||||
 | 
			
		||||
    TEST_ASSERT( psa_encrypt_setup( &operation, key_slot, alg ) == PSA_SUCCESS );
 | 
			
		||||
 | 
			
		||||
    TEST_ASSERT( psa_encrypt_generate_iv( &operation, iv,
 | 
			
		||||
                                        iv_size, &iv_length) == PSA_SUCCESS );
 | 
			
		||||
    TEST_ASSERT( psa_encrypt_set_iv( &operation, iv,
 | 
			
		||||
                                        sizeof( iv ) ) == PSA_SUCCESS );
 | 
			
		||||
 | 
			
		||||
	output_size = input_size;
 | 
			
		||||
	output = mbedtls_calloc(0, output_size);
 | 
			
		||||
 | 
			
		||||
    TEST_ASSERT( psa_cipher_update( &operation, input, input_size,
 | 
			
		||||
@ -575,6 +578,9 @@ void cipher_test_positive( int alg_arg, int key_type_arg,
 | 
			
		||||
 | 
			
		||||
    TEST_ASSERT( psa_cipher_abort( &operation ) == PSA_SUCCESS );
 | 
			
		||||
 | 
			
		||||
    TEST_ASSERT( input_size == output_size );
 | 
			
		||||
    TEST_ASSERT( memcmp( expected_output, output, output_size ) == 0 );
 | 
			
		||||
 | 
			
		||||
exit:
 | 
			
		||||
    mbedtls_free( key );
 | 
			
		||||
    mbedtls_free( input );
 | 
			
		||||
@ -583,6 +589,70 @@ exit:
 | 
			
		||||
}
 | 
			
		||||
/* END_CASE */
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
/* BEGIN_CASE */
 | 
			
		||||
void cipher_test_decrypt( int alg_arg, int key_type_arg, 
 | 
			
		||||
                           char *key_hex,
 | 
			
		||||
                           char *input_hex, char *output_hex )
 | 
			
		||||
{
 | 
			
		||||
    int key_slot = 1;
 | 
			
		||||
    psa_key_type_t key_type = key_type_arg;
 | 
			
		||||
    psa_algorithm_t alg = alg_arg;
 | 
			
		||||
    unsigned char *key = NULL;
 | 
			
		||||
    size_t key_size;
 | 
			
		||||
    unsigned char iv[16] = {0};
 | 
			
		||||
    unsigned char *input = NULL;
 | 
			
		||||
    size_t input_size = 0;
 | 
			
		||||
    unsigned char *output;
 | 
			
		||||
    unsigned char *expected_output;
 | 
			
		||||
    size_t output_size = 0;
 | 
			
		||||
    size_t output_length = 0;
 | 
			
		||||
    psa_cipher_operation_t operation;
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
    key = unhexify_alloc( key_hex, &key_size );
 | 
			
		||||
    TEST_ASSERT( key != NULL );
 | 
			
		||||
 | 
			
		||||
    input = unhexify_alloc( input_hex, &input_size );
 | 
			
		||||
    TEST_ASSERT( input != NULL );
 | 
			
		||||
 | 
			
		||||
    expected_output = unhexify_alloc( output_hex, &output_size );
 | 
			
		||||
    TEST_ASSERT( expected_output != NULL );
 | 
			
		||||
 | 
			
		||||
    memset( iv, 0x2a, sizeof( iv ) );
 | 
			
		||||
 
 | 
			
		||||
    TEST_ASSERT( psa_crypto_init( ) == PSA_SUCCESS );
 | 
			
		||||
 | 
			
		||||
    TEST_ASSERT( psa_import_key( key_slot, key_type,
 | 
			
		||||
                                 key, key_size ) == PSA_SUCCESS );
 | 
			
		||||
 | 
			
		||||
    TEST_ASSERT( psa_decrypt_setup( &operation, key_slot, alg ) == PSA_SUCCESS );
 | 
			
		||||
 | 
			
		||||
    TEST_ASSERT( psa_encrypt_set_iv( &operation, iv,
 | 
			
		||||
                                        sizeof( iv ) ) == PSA_SUCCESS );
 | 
			
		||||
 | 
			
		||||
	output = mbedtls_calloc(0, output_size);
 | 
			
		||||
 | 
			
		||||
    TEST_ASSERT( psa_cipher_update( &operation, input, input_size,
 | 
			
		||||
                                    output, output_size, 
 | 
			
		||||
                                    &output_length) == PSA_SUCCESS );
 | 
			
		||||
	TEST_ASSERT( psa_cipher_finish( &operation, output + output_length,
 | 
			
		||||
                                    &output_length) == PSA_SUCCESS );
 | 
			
		||||
 | 
			
		||||
    TEST_ASSERT( psa_cipher_abort( &operation ) == PSA_SUCCESS );
 | 
			
		||||
 | 
			
		||||
    TEST_ASSERT( input_size == output_size );
 | 
			
		||||
    TEST_ASSERT( memcmp( expected_output, output, output_size ) == 0 );
 | 
			
		||||
 | 
			
		||||
exit:
 | 
			
		||||
    mbedtls_free( key );
 | 
			
		||||
    mbedtls_free( input );
 | 
			
		||||
    psa_destroy_key( key_slot );
 | 
			
		||||
    mbedtls_psa_crypto_free( );
 | 
			
		||||
}
 | 
			
		||||
/* END_CASE */
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
/* BEGIN_CASE */
 | 
			
		||||
void cipher_test_verify_output( int alg_arg, int key_type_arg, 
 | 
			
		||||
                           char *key_hex,
 | 
			
		||||
 | 
			
		||||
		Loading…
	
	
			
			x
			
			
		
	
		Reference in New Issue
	
	Block a user