SKALE-3228

secure_enclave/AESUtils.c

@@ -41,7 +41,7 @@ int AES_encrypt(char *message, uint8_t *encr_message, uint64_t encrLen) {
         return -2;
     }
 
-    uint64_t len = strlen(message);
+    uint64_t len = strlen(message) + 1;
 
     if (len + SGX_AESGCM_MAC_SIZE + SGX_AESGCM_IV_SIZE > encrLen ) {
         LOG_ERROR("Output buffer too small");

secure_enclave/AESUtils.h

@@ -27,7 +27,10 @@
 sgx_aes_gcm_128bit_key_t AES_key;
 
 int AES_encrypt(char *message, uint8_t *encr_message, uint64_t encrLen);
-
 int AES_decrypt(uint8_t *encr_message, uint64_t length, char *message, uint64_t msgLen) ;
 
+int AES_encrypt_dh(char *message, uint8_t *encr_message, uint64_t encrLen);
+int AES_decrypt_dh(uint8_t *encr_message, uint64_t length, char *message, uint64_t msgLen) ;
+
+
 #endif //SGXD_AESUTILS_H

secure_enclave/secure_enclave.c

@@ -242,7 +242,7 @@ void sealHexSEK(int *errStatus, char *errString,
     CHECK_STATE(strnlen(sek_hex, 33) == 32)
     
 
-    uint64_t plaintextLen = strlen(sek_hex + 1);
+    uint64_t plaintextLen = strlen(sek_hex) + 1;
     
     uint64_t sealedLen = sgx_calc_sealed_data_size(0, plaintextLen);