Ecosystem/p2pool
1
0
Fork 0
mirror of https://github.com/SChernykh/p2pool.git synced 2025-01-03 09:19:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Update crypto_tests.cpp

Browse source
This commit is contained in:
SChernykh 2021-09-14 11:30:08 +02:00
parent b35b4a9f76
commit 94adce4045
1 changed files with 36 additions and 51 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

87
tests/src/crypto_tests.cpp
View file

@ -23,59 +23,44 @@
namespace p2pool { namespace p2pool {
static void parse_hash(const std::string& s, hash& h) TEST(crypto, derivation)
{ {
for (size_t i = 0; i < HASH_SIZE; ++i) { // Run the tests twice to check how crypto cache works
uint8_t d[2]; for (int i = 0; i < 2; ++i) {
if (!from_hex(s[i * 2], d[0]) || !from_hex(s[i * 2 + 1], d[1])) { std::ifstream f("crypto_tests.txt");
h = {}; ASSERT_EQ(f.good() && f.is_open(), true);
break; do {
} std::string name;
h.h[i] = (d[0] << 4) | d[1]; f >> name;
if (name == "generate_key_derivation") {
hash key1, key2, derivation, expected_derivation;
std::string result_str;
f >> key1 >> key2 >> result_str;
const bool result = (result_str == "true");
if (result) {
f >> expected_derivation;
}
ASSERT_EQ(p2pool::generate_key_derivation(key1, key2, derivation), result);
if (result) {
ASSERT_EQ(derivation, expected_derivation);
}
}
else if (name == "derive_public_key") {
hash derivation, base, derived_key, expected_derived_key;
std::string result_str;
size_t output_index;
f >> derivation >> output_index >> base >> result_str;
const bool result = (result_str == "true");
if (result) {
f >> expected_derived_key;
}
ASSERT_EQ(derive_public_key(derivation, output_index, base, derived_key), result);
if (result) {
ASSERT_EQ(derived_key, expected_derived_key);
}
}
} while (!f.eof());
} }
} }
TEST(crypto, derivation)
{
std::ifstream f("crypto_tests.txt");
ASSERT_EQ(f.good() && f.is_open(), true);
do {
std::string name;
f >> name;
if (name == "generate_key_derivation") {
std::string key1_str, key2_str, result_str, expected_derivation_str;
f >> key1_str >> key2_str >> result_str;
const bool result = (result_str == "true");
if (result) {
f >> expected_derivation_str;
}
hash key1, key2, derivation, expected_derivation;
parse_hash(key1_str, key1);
parse_hash(key2_str, key2);
ASSERT_EQ(p2pool::generate_key_derivation(key1, key2, derivation), result);
if (result) {
parse_hash(expected_derivation_str, expected_derivation);
ASSERT_EQ(derivation, expected_derivation);
}
}
else if (name == "derive_public_key") {
std::string derivation_str, base_str, result_str, expected_derived_key_str;
size_t output_index;
f >> derivation_str >> output_index >> base_str >> result_str;
const bool result = (result_str == "true");
if (result) {
f >> expected_derived_key_str;
}
hash derivation, base, derived_key, expected_derived_key;
parse_hash(derivation_str, derivation);
parse_hash(base_str, base);
ASSERT_EQ(derive_public_key(derivation, output_index, base, derived_key), result);
if (result) {
parse_hash(expected_derived_key_str, expected_derived_key);
ASSERT_EQ(derived_key, expected_derived_key);
}
}
} while (!f.eof());
} }
}