Merge pull request #9939 from kfnta/fix_for_secure_partition

Fix for secure partition

Author: Cruz Monrreal

components/TARGET_PSA/services/crypto/COMPONENT_SPE/psa_crypto_partition.c

@@ -641,6 +641,7 @@ static void psa_asymmetric_operation(void)
 
                     psa_write(msg.handle, 1,
                               &signature_length, sizeof(signature_length));
+                    mbedtls_free(hash);
                     mbedtls_free(signature);
                     break;
                 }
@@ -663,6 +664,7 @@ static void psa_asymmetric_operation(void)
                     hash = mbedtls_calloc(1, msg.in_size[2]);
                     if (hash == NULL) {
                         status = PSA_ERROR_INSUFFICIENT_MEMORY;
+                        mbedtls_free(signature);
                         break;
                     }
 
@@ -1200,20 +1202,19 @@ static void psa_key_management_operation(void)
                 }
 
                 case PSA_GET_KEY_INFORMATION: {
-                    psa_key_type_t type;
-                    size_t bits;
+                    psa_key_type_t type = 0;
+                    size_t bits = 0;
 
-                    if (!psa_crypto_access_control_is_handle_permitted(psa_key_mng.handle,
-                                                                       partition_id)) {
+                    if (psa_crypto_access_control_is_handle_permitted(psa_key_mng.handle,
+                                                                      partition_id)) {
+                        status = psa_get_key_information(psa_key_mng.handle, &type, &bits);
+                    } else {
                         status = PSA_ERROR_INVALID_HANDLE;
-                        break;
                     }
 
-                    status = psa_get_key_information(psa_key_mng.handle,
-                                                     &type, &bits);
-                    if (msg.out_size[0] >= sizeof(psa_key_type_t))
-                        psa_write(msg.handle, 0,
-                                  &type, sizeof(psa_key_type_t));
+                    if (msg.out_size[0] >= sizeof(psa_key_type_t)) {
+                        psa_write(msg.handle, 0, &type, sizeof(psa_key_type_t));
+                    }
                     if (msg.out_size[1] >= sizeof(size_t)) {
                         psa_write(msg.handle, 1, &bits, sizeof(size_t));
                     }
@@ -1637,6 +1638,8 @@ void psa_crypto_generator_operations(void)
                                                 label,
                                                 msg.in_size[2],//label length
                                                 psa_crypto_ipc.capacity);
+                    mbedtls_free(label);
+                    mbedtls_free(salt);
 
                     break;
                 }
@@ -1666,7 +1669,7 @@ void psa_crypto_generator_operations(void)
                                                private_key,
                                                msg.in_size[1],//private_key length
                                                psa_crypto_ipc.alg);
-
+                    mbedtls_free(private_key);
                     break;
                 }