Fix for algorithms other than hash-then-sign

Signed-off-by: gabor-mezei-arm <gabor.mezei@arm.com>
This commit is contained in:
gabor-mezei-arm 2021-05-03 16:26:20 +02:00
parent fc8d0aedd6
commit 9719a8450e
No known key found for this signature in database
GPG Key ID: 106F5A41ECC305BD

View File

@ -3024,11 +3024,12 @@ psa_status_t psa_sign_message_internal(
size_t *signature_length ) size_t *signature_length )
{ {
psa_status_t status = PSA_ERROR_CORRUPTION_DETECTED; psa_status_t status = PSA_ERROR_CORRUPTION_DETECTED;
size_t hash_length;
uint8_t hash[PSA_HASH_MAX_SIZE];
if ( PSA_ALG_IS_HASH_AND_SIGN( alg ) ) if ( PSA_ALG_IS_HASH_AND_SIGN( alg ) )
{ {
size_t hash_length;
uint8_t hash[PSA_HASH_MAX_SIZE];
status = psa_driver_wrapper_hash_compute( status = psa_driver_wrapper_hash_compute(
PSA_ALG_SIGN_GET_HASH( alg ), PSA_ALG_SIGN_GET_HASH( alg ),
input, input_length, input, input_length,
@ -3036,11 +3037,16 @@ psa_status_t psa_sign_message_internal(
if( status != PSA_SUCCESS ) if( status != PSA_SUCCESS )
return status; return status;
return psa_driver_wrapper_sign_hash(
attributes, key_buffer, key_buffer_size,
alg, hash, hash_length,
signature, signature_size, signature_length );
} }
return psa_driver_wrapper_sign_hash( return psa_driver_wrapper_sign_hash(
attributes, key_buffer, key_buffer_size, attributes, key_buffer, key_buffer_size,
alg, hash, hash_length, alg, input, input_length,
signature, signature_size, signature_length ); signature, signature_size, signature_length );
} }
@ -3068,11 +3074,12 @@ psa_status_t psa_verify_message_internal(
size_t signature_length ) size_t signature_length )
{ {
psa_status_t status = PSA_ERROR_CORRUPTION_DETECTED; psa_status_t status = PSA_ERROR_CORRUPTION_DETECTED;
size_t hash_length;
uint8_t hash[PSA_HASH_MAX_SIZE];
if ( PSA_ALG_IS_HASH_AND_SIGN( alg ) ) if ( PSA_ALG_IS_HASH_AND_SIGN( alg ) )
{ {
size_t hash_length;
uint8_t hash[PSA_HASH_MAX_SIZE];
status = psa_driver_wrapper_hash_compute( status = psa_driver_wrapper_hash_compute(
PSA_ALG_SIGN_GET_HASH( alg ), PSA_ALG_SIGN_GET_HASH( alg ),
input, input_length, input, input_length,
@ -3080,11 +3087,16 @@ psa_status_t psa_verify_message_internal(
if( status != PSA_SUCCESS ) if( status != PSA_SUCCESS )
return status; return status;
return psa_driver_wrapper_verify_hash(
attributes, key_buffer, key_buffer_size,
alg, hash, hash_length,
signature, signature_length );
} }
return psa_driver_wrapper_verify_hash( return psa_driver_wrapper_verify_hash(
attributes, key_buffer, key_buffer_size, attributes, key_buffer, key_buffer_size,
alg, hash, hash_length, alg, input, input_length,
signature, signature_length ); signature, signature_length );
} }