commit 910c76b3d101a47fd39dbdadcbc1cabf37ed36de
author itayzafrir <itay.zafrir@arm.com> Wed Nov 21 16:03:21 2018 +0200
committer itayzafrir <itay.zafrir@arm.com> Wed Nov 21 16:10:33 2018 +0200
tree d2235548ae3adef435c000e31ad68c2d8ca07d2b
parent dbb83ac5f7b96077b21fc9fe72b2687986acf963

Check that memory allocation was successful in psa_save_generated_persistent_key

library/psa_crypto.c

diff --git a/library/psa_crypto.c b/library/psa_crypto.c
index 58cb738..05b8a8e 100644
--- a/library/psa_crypto.c
+++ b/library/psa_crypto.c
@@ -1117,6 +1117,8 @@
     size_t key_length;
     size_t data_size = PSA_KEY_EXPORT_MAX_SIZE( slot->type, bits );
     data = mbedtls_calloc( 1, data_size );
+    if( data == NULL )
+        return( PSA_ERROR_INSUFFICIENT_MEMORY );
     /* Get key data in export format */
     status = psa_internal_export_key( slot, data, data_size, &key_length, 0 );
     if( status != PSA_SUCCESS )