SKALE-3228

SEKManager.cpp

@@ -70,36 +70,23 @@ void create_test_key() {
 }
 
 
-shared_ptr <vector<uint8_t>> check_and_set_SEK(const string &SEK) {
+void validate_SEK() {
 
+    shared_ptr <string> test_key_ptr = LevelDB::getLevelDb()->readString("TEST_KEY");
+    vector <uint8_t> encr_test_key(BUF_LEN, 0);
     vector<char> decr_key(BUF_LEN, 0);
+    uint64_t len = 0;
     vector<char> errMsg(BUF_LEN, 0);
 
     int err_status = 0;
 
-    auto encrypted_SEK = make_shared < vector < uint8_t >> (BUF_LEN, 0);
-
-    uint32_t l = 0;
-
-    sgx_status_t status = trustedSetSEK_backup(eid, &err_status, errMsg.data(), encrypted_SEK->data(), &l, SEK.c_str());
-
-    encrypted_SEK->resize(l);
-
-    HANDLE_TRUSTED_FUNCTION_ERROR(status, err_status, errMsg.data());
-
-
-    shared_ptr <string> test_key_ptr = LevelDB::getLevelDb()->readString("TEST_KEY");
-    vector <uint8_t> encr_test_key(BUF_LEN, 0);
-
-    uint64_t len = 0;
-
     if (!hex2carray(test_key_ptr->c_str(), &len, encr_test_key.data(),
                     BUF_LEN)) {
         spdlog::error("Corrupt test key is LevelDB");
         exit(-1);
     }
 
-    status = trustedDecryptKeyAES(eid, &err_status, errMsg.data(), encr_test_key.data(), len, decr_key.data());
+    sgx_status_t status = trustedDecryptKeyAES(eid, &err_status, errMsg.data(), encr_test_key.data(), len, decr_key.data());
 
     HANDLE_TRUSTED_FUNCTION_ERROR(status, err_status, errMsg.data());
 
@@ -111,6 +98,27 @@ shared_ptr <vector<uint8_t>> check_and_set_SEK(const string &SEK) {
         spdlog::error("Then run sgxwallet using backup flag");
         exit(-1);
     }
+}
+
+
+shared_ptr <vector<uint8_t>> check_and_set_SEK(const string &SEK) {
+
+    vector<char> decr_key(BUF_LEN, 0);
+    vector<char> errMsg(BUF_LEN, 0);
+    int err_status = 0;
+
+    auto encrypted_SEK = make_shared < vector < uint8_t >> (BUF_LEN, 0);
+
+    uint32_t l = 0;
+
+    sgx_status_t status = trustedSetSEK_backup(eid, &err_status, errMsg.data(), encrypted_SEK->data(), &l,
+                                               SEK.c_str());
+
+    encrypted_SEK->resize(l);
+
+    HANDLE_TRUSTED_FUNCTION_ERROR(status, err_status, errMsg.data());
+
+    validate_SEK();
 
     return encrypted_SEK;
 }
@@ -163,6 +171,8 @@ void gen_SEK() {
     LevelDB::getLevelDb()->writeDataUnique("SEK", hexEncrKey.data());
 
     create_test_key();
+
+    validate_SEK();
 }
 
 void setSEK(shared_ptr <string> hex_encrypted_SEK) {
@@ -185,6 +195,7 @@ void setSEK(shared_ptr <string> hex_encrypted_SEK) {
 
     HANDLE_TRUSTED_FUNCTION_ERROR(status, err_status, errMsg.data());
 
+    validate_SEK();
 
 }