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 )
{
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 ) )
{
size_t hash_length;
uint8_t hash[PSA_HASH_MAX_SIZE];
status = psa_driver_wrapper_hash_compute(
PSA_ALG_SIGN_GET_HASH( alg ),
input, input_length,
@ -3036,12 +3037,17 @@ psa_status_t psa_sign_message_internal(
if( status != PSA_SUCCESS )
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(
attributes, key_buffer, key_buffer_size,
alg, input, input_length,
signature, signature_size, signature_length );
}
psa_status_t psa_sign_message( mbedtls_svc_key_id_t key,
@ -3068,11 +3074,12 @@ psa_status_t psa_verify_message_internal(
size_t signature_length )
{
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 ) )
{
size_t hash_length;
uint8_t hash[PSA_HASH_MAX_SIZE];
status = psa_driver_wrapper_hash_compute(
PSA_ALG_SIGN_GET_HASH( alg ),
input, input_length,
@ -3080,12 +3087,17 @@ psa_status_t psa_verify_message_internal(
if( status != PSA_SUCCESS )
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(
attributes, key_buffer, key_buffer_size,
alg, input, input_length,
signature, signature_length );
}
psa_status_t psa_verify_message( mbedtls_svc_key_id_t key,