From ac5b32b8944485bfb1a517d09a6e77f4181effb9 Mon Sep 17 00:00:00 2001
From: Ryan Everett <ryan.everett@arm.com>
Date: Wed, 15 Nov 2023 16:26:01 +0000
Subject: [PATCH] Fix error handling for secure element keys in
 `psa_start_key_creation`

Signed-off-by: Ryan Everett <ryan.everett@arm.com>
---
 library/psa_crypto.c | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/library/psa_crypto.c b/library/psa_crypto.c
index bbd6b24ed4..4beda81124 100644
--- a/library/psa_crypto.c
+++ b/library/psa_crypto.c
@@ -1841,6 +1841,9 @@ static psa_status_t psa_start_key_creation(
 
         status = psa_copy_key_material_into_slot(
             slot, (uint8_t *) (&slot_number), sizeof(slot_number));
+        if (status != PSA_SUCCESS) {
+            return status;
+        }
     }
 
     if (*p_drv == NULL && method == PSA_KEY_CREATION_REGISTER) {