Fixing

secure_enclave/BLSUtils.cpp

@@ -70,8 +70,33 @@ libff::alt_bn128_Fr *keyFromString(const char* _keyString) {
 
 bool check_key(int *err_status, char *err_string, const char *_keyString) {
 
+
     libff::init_alt_bn128_params();
 
+
+
+    uint64_t keyLen = strnlen(_keyString, MAX_KEY_LENGTH);
+
+    // check that key is zero terminated string
+
+    if (keyLen == MAX_KEY_LENGTH) {
+        snprintf(err_string, MAX_ERR_LEN, "keyLen != MAX_KEY_LENGTH");
+        return false;
+    }
+
+    *err_status = -2;
+
+    // check that key is padded with 0s
+
+    for (int i = keyLen; i < MAX_KEY_LENGTH; i++) {
+        if (_keyString[i] != 0) {
+            snprintf(err_string, BUF_LEN,"Unpadded key");
+            return false;
+        }
+    }
+
+
+
   *err_status = -1;
 
 
@@ -107,6 +132,10 @@ bool check_key(int *err_status, char *err_string, const char *_keyString) {
 bool sign(const char *_keyString, const char* _hashXString, const char* _hashYString,
        char sig[BUF_LEN]) {
 
+
+    libff::init_alt_bn128_params();
+
+
     auto key = keyFromString(_keyString);
 
          libff::alt_bn128_Fq hashX(_hashXString);

secure_enclave/secure_enclave.c

@@ -120,28 +120,10 @@ void generate_ecdsa_key(int *err_status, char *err_string,
 void encrypt_key(int *err_status, char *err_string, char *key,
                  uint8_t *encrypted_key, uint32_t *enc_len) {
 
+
   *err_status = -1;
   memset(err_string, 0, BUF_LEN);
 
-  uint64_t keyLen = strnlen(key, MAX_KEY_LENGTH);
-
-  // check that key is zero terminated string
-
-  if (keyLen == MAX_KEY_LENGTH) {
-    snprintf(err_string, MAX_ERR_LEN, "keyLen != MAX_KEY_LENGTH");
-    return;
-  }
-
-  *err_status = -2;
-
-  // check that key is padded with 0s
-
-  for (int i = keyLen; i < MAX_KEY_LENGTH; i++) {
-    if (key[i] != 0) {
-      snprintf(err_string, BUF_LEN,"Unpadded key");
-      return;
-    }
-  }
 
   *err_status = -3;
 
@@ -208,6 +190,7 @@ void encrypt_key(int *err_status, char *err_string, char *key,
 void decrypt_key(int *err_status, char *err_string, uint8_t *encrypted_key,
                  uint32_t enc_len, char* key) {
 
+
   uint32_t decLen;
 
   *err_status = -9;
@@ -262,6 +245,8 @@ void bls_sign_message(int *err_status, char *err_string,  uint8_t *encrypted_key
   char key[BUF_LEN];
   char sig[BUF_LEN];
 
+
+
   decrypt_key(err_status, err_string, encrypted_key, enc_len, key);
 
   if (err_status != 0) {

testw.cpp

@@ -88,6 +88,7 @@ TEST_CASE( "BLS sign test", "[bls-sign]" ) {
   REQUIRE(status == SGX_SUCCESS);
   REQUIRE(err_status == 0);
 
+
   printf("Encrypt key completed with status: %d %s \n", err_status, errMsg);
   printf(" Encrypted key len %d\n", enc_len);