validate parameters for importBlsKEyShare

1927fea3 · Oleh Nikolaiev · 8893e86d · 1927fea3
Unverified Commit 1927fea3 authored Sep 25, 2020 by Oleh Nikolaiev
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 1 deletion

SGXWalletServer.cpp SGXWalletServer.cpp +6 -1

No files found.
--- a/SGXWalletServer.cpp
+++ b/SGXWalletServer.cpp
@@ -193,7 +193,12 @@ SGXWalletServer::importBLSKeyShareImpl(const string &_keyShare, const string &_k
            throw SGXException(INVALID_BLS_NAME, "Invalid BLS key name");
        }

-        if (!checkHex(_keyShare)) {
+        string hashTmp = _keyShare;
+        if (hashTmp[0] == '0' && (hashTmp[1] == 'x' || hashTmp[1] == 'X')) {
+            hashTmp.erase(hashTmp.begin(), hashTmp.begin() + 2);
+        }
+
+        if (!checkHex(hashTmp)) {
            throw SGXException(INVALID_HEX, "Invalid BLS key share, please use hex");
        }