Skip to content

S
sgxwallet
Unverified Commit 7e2ac473 authored by kladko's avatar kladko
Browse files
Options

SKALE-3067-cleanup-sgx

parent bf6cefc5
Hide whitespace changes
Inline Side-by-side
Showing with 249 additions and 159 deletions
secure_enclave/DKGUtils.cpp
View file @ 7e2ac473
...@@ -49,19 +49,19 @@ string stringFromFr(const libff::alt_bn128_Fr &_el) { ...@@ -49,19 +49,19 @@ string stringFromFr(const libff::alt_bn128_Fr &_el) {
try { try {
mpz_t t; mpz_t t;
mpz_init(t); mpz_init(t);
_el.as_bigint().to_mpz(t); _el.as_bigint().to_mpz(t);
char arr[mpz_sizeinbase(t, 10) + 2]; char arr[mpz_sizeinbase(t, 10) + 2];
char *tmp = mpz_get_str(arr, 10, t); char *tmp = mpz_get_str(arr, 10, t);
mpz_clear(t); mpz_clear(t);
return string(tmp); return string(tmp);
} catch (exception& e) { } catch (exception &e) {
LOG_ERROR(e.what()); LOG_ERROR(e.what());
return ""; return "";
} catch (...) { } catch (...) {
...@@ -119,27 +119,28 @@ string ConvertG2ToString(const libff::alt_bn128_G2 &elem, int base = 10, const s ...@@ -119,27 +119,28 @@ string ConvertG2ToString(const libff::alt_bn128_G2 &elem, int base = 10, const s
} }
vector <libff::alt_bn128_Fr> SplitStringToFr(const char *coeffs, const char symbol) { vector <libff::alt_bn128_Fr> SplitStringToFr(const char *coeffs, const char symbol) {
vector <libff::alt_bn128_Fr> tokens; vector <libff::alt_bn128_Fr> tokens;
try { try {
string str(coeffs); string str(coeffs);
string delim; string delim;
delim.push_back(symbol); delim.push_back(symbol);
size_t prev = 0, pos = 0; size_t prev = 0, pos = 0;
do { do {
pos = str.find(delim, prev); pos = str.find(delim, prev);
if (pos == string::npos) pos = str.length(); if (pos == string::npos) pos = str.length();
string token = str.substr(prev, pos - prev); string token = str.substr(prev, pos - prev);
if (!token.empty()) { if (!token.empty()) {
libff::alt_bn128_Fr coeff(token.c_str()); libff::alt_bn128_Fr coeff(token.c_str());
tokens.push_back(coeff); tokens.push_back(coeff);
} }
prev = pos + delim.length(); prev = pos + delim.length();
} while (pos < str.length() && prev < str.length()); } while (pos < str.length() && prev < str.length());
return tokens; return tokens;
} catch (exception& e) { } catch (exception &e) {
LOG_ERROR(e.what()); LOG_ERROR(e.what());
return tokens; return tokens;
} catch (...) { } catch (...) {
...@@ -150,25 +151,25 @@ vector <libff::alt_bn128_Fr> SplitStringToFr(const char *coeffs, const char symb ...@@ -150,25 +151,25 @@ vector <libff::alt_bn128_Fr> SplitStringToFr(const char *coeffs, const char symb
int gen_dkg_poly(char *secret, unsigned _t) { int gen_dkg_poly(char *secret, unsigned _t) {
try { try {
string result; string result;
for (size_t i = 0; i < _t; ++i) { for (size_t i = 0; i < _t; ++i) {
libff::alt_bn128_Fr cur_coef = libff::alt_bn128_Fr::random_element(); libff::alt_bn128_Fr cur_coef = libff::alt_bn128_Fr::random_element();
while (i == _t - 1 && cur_coef == libff::alt_bn128_Fr::zero()) { while (i == _t - 1 && cur_coef == libff::alt_bn128_Fr::zero()) {
cur_coef = libff::alt_bn128_Fr::random_element(); cur_coef = libff::alt_bn128_Fr::random_element();
}
result += stringFromFr(cur_coef);
result += ":";
} }
result += stringFromFr(cur_coef); strncpy(secret, result.c_str(), result.length() + 1);
result += ":";
}
strncpy(secret, result.c_str(), result.length() + 1);
if (strlen(secret) == 0) { if (strlen(secret) == 0) {
return 1; return 1;
} }
return 0; return 0;
} catch (exception& e) { } catch (exception &e) {
LOG_ERROR(e.what()); LOG_ERROR(e.what());
return 1; return 1;
} catch (...) { } catch (...) {
...@@ -179,207 +180,296 @@ int gen_dkg_poly(char *secret, unsigned _t) { ...@@ -179,207 +180,296 @@ int gen_dkg_poly(char *secret, unsigned _t) {
} }
libff::alt_bn128_Fr PolynomialValue(const vector <libff::alt_bn128_Fr> &pol, libff::alt_bn128_Fr point, unsigned _t) { libff::alt_bn128_Fr PolynomialValue(const vector <libff::alt_bn128_Fr> &pol, libff::alt_bn128_Fr point, unsigned _t) {
libff::alt_bn128_Fr value = libff::alt_bn128_Fr::zero(); libff::alt_bn128_Fr value = libff::alt_bn128_Fr::zero();
libff::alt_bn128_Fr pow = libff::alt_bn128_Fr::one(); try {
for (unsigned i = 0; i < pol.size(); ++i) {
value += pol[i] * pow;
pow *= point;
}
return value; libff::alt_bn128_Fr pow = libff::alt_bn128_Fr::one();
for (unsigned i = 0; i < pol.size(); ++i) {
value += pol[i] * pow;
pow *= point;
}
return value;
} catch (exception &e) {
LOG_ERROR(e.what());
return value;
} catch (...) {
LOG_ERROR("Unknown throwable");
return value;
}
} }
void calc_secret_shares(const char *decrypted_coeffs, void calc_secret_shares(const char *decrypted_coeffs,
char *secret_shares, // calculates secret shares in base 10 to a string secret_shares, char *secret_shares, // calculates secret shares in base 10 to a string secret_shares,
unsigned _t, unsigned _n) { // separated by ":" unsigned _t, unsigned _n) { // separated by ":"
// calculate for each node a list of secret values that will be used for verification // calculate for each node a list of secret values that will be used for verification
string result; string result;
char symbol = ':'; char symbol = ':';
vector <libff::alt_bn128_Fr> poly = SplitStringToFr(decrypted_coeffs, symbol);
for (size_t i = 0; i < _n; ++i) { try {
libff::alt_bn128_Fr secret_share = PolynomialValue(poly, libff::alt_bn128_Fr(i + 1), _t);
result += ConvertToString(secret_share); vector <libff::alt_bn128_Fr> poly = SplitStringToFr(decrypted_coeffs, symbol);
result += ":";
for (size_t i = 0; i < _n; ++i) {
libff::alt_bn128_Fr secret_share = PolynomialValue(poly, libff::alt_bn128_Fr(i + 1), _t);
result += ConvertToString(secret_share);
result += ":";
}
strncpy(secret_shares, result.c_str(), result.length() + 1);
} catch (exception &e) {
LOG_ERROR(e.what());
return;
} catch (...) {
LOG_ERROR("Unknown throwable");
return;
} }
strncpy(secret_shares, result.c_str(), result.length() + 1);
} }
int calc_secret_share(const char *decrypted_coeffs, char *s_share, int calc_secret_share(const char *decrypted_coeffs, char *s_share,
unsigned _t, unsigned _n, unsigned ind) { unsigned _t, unsigned _n, unsigned ind) {
char symbol = ':'; try {
vector <libff::alt_bn128_Fr> poly = SplitStringToFr(decrypted_coeffs, symbol); char symbol = ':';
if (poly.size() != _t) { vector <libff::alt_bn128_Fr> poly = SplitStringToFr(decrypted_coeffs, symbol);
return 1; if (poly.size() != _t) {
} return 1;
}
libff::alt_bn128_Fr secret_share = PolynomialValue(poly, libff::alt_bn128_Fr(ind), _t); libff::alt_bn128_Fr secret_share = PolynomialValue(poly, libff::alt_bn128_Fr(ind), _t);
string cur_share = ConvertToString(secret_share, 16); string cur_share = ConvertToString(secret_share, 16);
int n_zeroes = 64 - cur_share.size(); int n_zeroes = 64 - cur_share.size();
cur_share.insert(0, n_zeroes, '0'); cur_share.insert(0, n_zeroes, '0');
strncpy(s_share, cur_share.c_str(), cur_share.length() + 1); strncpy(s_share, cur_share.c_str(), cur_share.length() + 1);
return 0; return 0;
} catch (exception &e) {
LOG_ERROR(e.what());
return 1;
} catch (...) {
LOG_ERROR("Unknown throwable");
return 1;
}
} }
void calc_secret_shareG2_old(const char *decrypted_coeffs, char *s_shareG2, void calc_secret_shareG2_old(const char *decrypted_coeffs, char *s_shareG2,
unsigned _t, unsigned ind) { unsigned _t, unsigned ind) {
char symbol = ':'; try {
vector <libff::alt_bn128_Fr> poly = SplitStringToFr(decrypted_coeffs, symbol); char symbol = ':';
vector <libff::alt_bn128_Fr> poly = SplitStringToFr(decrypted_coeffs, symbol);
libff::alt_bn128_Fr secret_share = PolynomialValue(poly, libff::alt_bn128_Fr(ind), _t); libff::alt_bn128_Fr secret_share = PolynomialValue(poly, libff::alt_bn128_Fr(ind), _t);
libff::alt_bn128_G2 secret_shareG2 = secret_share * libff::alt_bn128_G2::one(); libff::alt_bn128_G2 secret_shareG2 = secret_share * libff::alt_bn128_G2::one();
string secret_shareG2_str = ConvertG2ToString(secret_shareG2); string secret_shareG2_str = ConvertG2ToString(secret_shareG2);
strncpy(s_shareG2, secret_shareG2_str.c_str(), secret_shareG2_str.length() + 1); strncpy(s_shareG2, secret_shareG2_str.c_str(), secret_shareG2_str.length() + 1);
} catch (exception &e) {
LOG_ERROR(e.what());
} catch (...) {
LOG_ERROR("Unknown throwable");
}
} }
int calc_secret_shareG2(const char *s_share, char *s_shareG2) { int calc_secret_shareG2(const char *s_share, char *s_shareG2) {
mpz_t share; try {
mpz_init(share);
if (mpz_set_str(share, s_share, 16) == -1) {
mpz_clear(share);
return 1;
}
char arr[mpz_sizeinbase(share, 10) + 2]; mpz_t share;
char *share_str = mpz_get_str(arr, 10, share); mpz_init(share);
if (mpz_set_str(share, s_share, 16) == -1) {
mpz_clear(share);
return 1;
}
libff::alt_bn128_Fr secret_share(share_str); char arr[mpz_sizeinbase(share, 10) + 2];
char *share_str = mpz_get_str(arr, 10, share);
libff::alt_bn128_G2 secret_shareG2 = secret_share * libff::alt_bn128_G2::one(); libff::alt_bn128_Fr secret_share(share_str);
secret_shareG2.to_affine_coordinates(); libff::alt_bn128_G2 secret_shareG2 = secret_share * libff::alt_bn128_G2::one();
string secret_shareG2_str = ConvertG2ToString(secret_shareG2); secret_shareG2.to_affine_coordinates();
strncpy(s_shareG2, secret_shareG2_str.c_str(), secret_shareG2_str.length() + 1); string secret_shareG2_str = ConvertG2ToString(secret_shareG2);
mpz_clear(share); strncpy(s_shareG2, secret_shareG2_str.c_str(), secret_shareG2_str.length() + 1);
return 0; mpz_clear(share);
return 0;
} catch (exception &e) {
LOG_ERROR(e.what());
return 1;
} catch (...) {
LOG_ERROR("Unknown throwable");
return 1;
}
} }
int calc_public_shares(const char *decrypted_coeffs, char *public_shares, int calc_public_shares(const char *decrypted_coeffs, char *public_shares,
unsigned _t) { unsigned _t) {
// calculate for each node a list of public shares try {
string result; // calculate for each node a list of public shares
char symbol = ':'; string result;
vector <libff::alt_bn128_Fr> poly = SplitStringToFr(decrypted_coeffs, symbol); char symbol = ':';
if (poly.size() != _t) { vector <libff::alt_bn128_Fr> poly = SplitStringToFr(decrypted_coeffs, symbol);
if (poly.size() != _t) {
return 1;
}
for (size_t i = 0; i < _t; ++i) {
libff::alt_bn128_G2 pub_share = poly.at(i) * libff::alt_bn128_G2::one();
pub_share.to_affine_coordinates();
string pub_share_str = ConvertG2ToString(pub_share);
result += pub_share_str + ",";
}
strncpy(public_shares, result.c_str(), result.length());
return 0;
} catch (exception &e) {
LOG_ERROR(e.what());
return 1;
} catch (...) {
LOG_ERROR("Unknown throwable");
return 1; return 1;
} }
for (size_t i = 0; i < _t; ++i) {
libff::alt_bn128_G2 pub_share = poly.at(i) * libff::alt_bn128_G2::one();
pub_share.to_affine_coordinates();
string pub_share_str = ConvertG2ToString(pub_share);
result += pub_share_str + ",";
}
strncpy(public_shares, result.c_str(), result.length());
return 0;
} }
string ConvertHexToDec(string hex_str) { string ConvertHexToDec(string hex_str) {
mpz_t dec;
mpz_init(dec);
if (mpz_set_str(dec, hex_str.c_str(), 16) == -1) { try {
mpz_t dec;
mpz_init(dec);
if (mpz_set_str(dec, hex_str.c_str(), 16) == -1) {
mpz_clear(dec);
return "";
}
char arr[mpz_sizeinbase(dec, 10) + 2];
char *result = mpz_get_str(arr, 10, dec);
mpz_clear(dec); mpz_clear(dec);
return "false";
}
char arr[mpz_sizeinbase(dec, 10) + 2]; return result;
char *result = mpz_get_str(arr, 10, dec);
mpz_clear(dec); } catch (exception &e) {
LOG_ERROR(e.what());
return "";
} catch (...) {
LOG_ERROR("Unknown throwable");
return "";
}
return result;
} }
int Verification(char *public_shares, mpz_t decr_secret_share, int _t, int ind) { int Verification(char *public_shares, mpz_t decr_secret_share, int _t, int ind) {
string pub_shares_str = public_shares;
try {
vector <libff::alt_bn128_G2> pub_shares;
uint64_t share_length = 256;
uint8_t coord_length = 64;
for (size_t i = 0; i < _t; ++i) {
libff::alt_bn128_G2 pub_share;
uint64_t pos0 = share_length * i; string pub_shares_str = public_shares;
string x_c0_str = ConvertHexToDec(pub_shares_str.substr(pos0, coord_length)); vector <libff::alt_bn128_G2> pub_shares;
string x_c1_str = ConvertHexToDec(pub_shares_str.substr(pos0 + coord_length, coord_length)); uint64_t share_length = 256;
string y_c0_str = ConvertHexToDec(pub_shares_str.substr(pos0 + 2 * coord_length, coord_length)); uint8_t coord_length = 64;
string y_c1_str = ConvertHexToDec(pub_shares_str.substr(pos0 + 3 * coord_length, coord_length));
if (x_c0_str == "false" || x_c1_str == "false" || y_c0_str == "false" || y_c1_str == "false") { for (size_t i = 0; i < _t; ++i) {
return 2; libff::alt_bn128_G2 pub_share;
uint64_t pos0 = share_length * i;
string x_c0_str = ConvertHexToDec(pub_shares_str.substr(pos0, coord_length));
string x_c1_str = ConvertHexToDec(pub_shares_str.substr(pos0 + coord_length, coord_length));
string y_c0_str = ConvertHexToDec(pub_shares_str.substr(pos0 + 2 * coord_length, coord_length));
string y_c1_str = ConvertHexToDec(pub_shares_str.substr(pos0 + 3 * coord_length, coord_length));
if (x_c0_str == "" || x_c1_str == "" || y_c0_str == "" || y_c1_str == "") {
return 2;
}
pub_share.X.c0 = libff::alt_bn128_Fq(x_c0_str.c_str());
pub_share.X.c1 = libff::alt_bn128_Fq(x_c1_str.c_str());
pub_share.Y.c0 = libff::alt_bn128_Fq(y_c0_str.c_str());
pub_share.Y.c1 = libff::alt_bn128_Fq(y_c1_str.c_str());
pub_share.Z = libff::alt_bn128_Fq2::one();
pub_shares.push_back(pub_share);
} }
pub_share.X.c0 = libff::alt_bn128_Fq(x_c0_str.c_str());
pub_share.X.c1 = libff::alt_bn128_Fq(x_c1_str.c_str());
pub_share.Y.c0 = libff::alt_bn128_Fq(y_c0_str.c_str());
pub_share.Y.c1 = libff::alt_bn128_Fq(y_c1_str.c_str());
pub_share.Z = libff::alt_bn128_Fq2::one();
pub_shares.push_back(pub_share); libff::alt_bn128_G2 val = libff::alt_bn128_G2::zero();
} for (int i = 0; i < _t; ++i) {
val = val + power(libff::alt_bn128_Fr(ind + 1), i) * pub_shares[i];
}
libff::alt_bn128_G2 val = libff::alt_bn128_G2::zero(); char arr[mpz_sizeinbase(decr_secret_share, 10) + 2];
for (int i = 0; i < _t; ++i) { char *tmp = mpz_get_str(arr, 10, decr_secret_share);
val = val + power(libff::alt_bn128_Fr(ind + 1), i) * pub_shares[i];
}
char arr[mpz_sizeinbase(decr_secret_share, 10) + 2]; libff::alt_bn128_Fr sshare(tmp);
char *tmp = mpz_get_str(arr, 10, decr_secret_share);
libff::alt_bn128_Fr sshare(tmp); libff::alt_bn128_G2 val2 = sshare * libff::alt_bn128_G2::one();
libff::alt_bn128_G2 val2 = sshare * libff::alt_bn128_G2::one(); memset(public_shares, 0, strlen(public_shares));
strncpy(public_shares, tmp, strlen(tmp));
memset(public_shares, 0, strlen(public_shares)); val.to_affine_coordinates();
strncpy(public_shares, tmp, strlen(tmp)); val2.to_affine_coordinates();
strncpy(public_shares, ConvertToString(val.X.c0).c_str(), ConvertToString(val.X.c0).length());
strncpy(public_shares + ConvertToString(val.X.c0).length(), ":", 1);
strncpy(public_shares + ConvertToString(val.X.c0).length() + 1, ConvertToString(val2.X.c0).c_str(),
ConvertToString(val2.X.c0).length());
val.to_affine_coordinates(); return (val == sshare * libff::alt_bn128_G2::one());
val2.to_affine_coordinates();
strncpy(public_shares, ConvertToString(val.X.c0).c_str(), ConvertToString(val.X.c0).length());
strncpy(public_shares + ConvertToString(val.X.c0).length(), ":", 1);
strncpy(public_shares + ConvertToString(val.X.c0).length() + 1, ConvertToString(val2.X.c0).c_str(),
ConvertToString(val2.X.c0).length());
return (val == sshare * libff::alt_bn128_G2::one()); } catch (exception &e) {
LOG_ERROR(e.what());
return 0;
} catch (...) {
LOG_ERROR("Unknown throwable");
return 0;
}
} }
int calc_bls_public_key(char *skey_hex, char *pub_key) { int calc_bls_public_key(char *skey_hex, char *pub_key) {
try {
mpz_t skey;
mpz_init(skey);
if (mpz_set_str(skey, skey_hex, 16) == -1) {
mpz_clear(skey);
return 1;
}
mpz_t skey; char skey_dec[mpz_sizeinbase(skey, 10) + 2];
mpz_init(skey); mpz_get_str(skey_dec, 10, skey);
if (mpz_set_str(skey, skey_hex, 16) == -1) {
mpz_clear(skey);
return 1;
}
char skey_dec[mpz_sizeinbase(skey, 10) + 2]; libff::alt_bn128_Fr bls_skey(skey_dec);
mpz_get_str(skey_dec, 10, skey);
libff::alt_bn128_Fr bls_skey(skey_dec); libff::alt_bn128_G2 public_key = bls_skey * libff::alt_bn128_G2::one();
public_key.to_affine_coordinates();
libff::alt_bn128_G2 public_key = bls_skey * libff::alt_bn128_G2::one(); string result = ConvertG2ToString(public_key);
public_key.to_affine_coordinates();
string result = ConvertG2ToString(public_key); strncpy(pub_key, result.c_str(), result.length());
strncpy(pub_key, result.c_str(), result.length()); mpz_clear(skey);
mpz_clear(skey); return 0;
return 0; } catch (exception &e) {
LOG_ERROR(e.what());
return 1;
} catch (...) {
LOG_ERROR("Unknown throwable");
return 1;
}
} }
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment