Merge pull request #211 from ARMmbed/bug_fix_210

Fix memory allocation check in psa_save_generated_persistent_key (#211)
This commit is contained in:
Gilles Peskine 2018-11-21 15:51:07 +01:00 committed by GitHub
commit 83146e10bb
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -1117,6 +1117,8 @@ static psa_status_t psa_save_generated_persistent_key( psa_key_slot_t key,
size_t key_length; size_t key_length;
size_t data_size = PSA_KEY_EXPORT_MAX_SIZE( slot->type, bits ); size_t data_size = PSA_KEY_EXPORT_MAX_SIZE( slot->type, bits );
data = mbedtls_calloc( 1, data_size ); data = mbedtls_calloc( 1, data_size );
if( data == NULL )
return( PSA_ERROR_INSUFFICIENT_MEMORY );
/* Get key data in export format */ /* Get key data in export format */
status = psa_internal_export_key( slot, data, data_size, &key_length, 0 ); status = psa_internal_export_key( slot, data, data_size, &key_length, 0 );
if( status != PSA_SUCCESS ) if( status != PSA_SUCCESS )