From 0c3e06858bc64e0f421476b006feb3ae8cb69647 Mon Sep 17 00:00:00 2001
From: Sascha Tommasone <sascha@tommasone.me>
Date: Sat, 6 Jul 2024 15:02:27 +0200
Subject: [PATCH] [Assignment-7] .

---
 7-SGX_Hands-on/src/enclave/enclave.c | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/7-SGX_Hands-on/src/enclave/enclave.c b/7-SGX_Hands-on/src/enclave/enclave.c
index 6cfdba3..2cd3a80 100644
--- a/7-SGX_Hands-on/src/enclave/enclave.c
+++ b/7-SGX_Hands-on/src/enclave/enclave.c
@@ -168,12 +168,15 @@ sgx_status_t generate_key_pair(uint8_t *sealed, uint32_t sealed_size) {
 
     // create ecc keypair
     if((status = sgx_ecc256_create_key_pair(&private, &public, ecc_handle)) != SGX_SUCCESS) {
-        sgx_ecc256_close_context(ecc_handle);
-        return status;
+        goto exit;
     }
 
-    // return status of sealing
-    return seal_key_pair(&private, &public, &sealed);
+    // seal keypair
+    status = seal_key_pair(&private, &public, &sealed);
+    
+    exit: ;
+    sgx_ecc256_close_context(ecc_handle);
+    return status;
 }
 
 sgx_status_t get_public_key(const uint8_t *sealed, uint32_t sealed_size, uint8_t *public) {