mirror of
https://github.com/yuzu-emu/mbedtls.git
synced 2024-12-26 08:35:36 +00:00
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:
commit
83146e10bb
|
@ -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 )
|
||||||
|
|
Loading…
Reference in a new issue