mirror of
https://github.com/SChernykh/p2pool.git
synced 2024-12-22 19:39:22 +00:00
276 lines
7.2 KiB
C++
276 lines
7.2 KiB
C++
/*
|
|
* This file is part of the Monero P2Pool <https://github.com/SChernykh/p2pool>
|
|
* Copyright (c) 2021-2023 SChernykh <https://github.com/SChernykh>
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, version 3.
|
|
*
|
|
* This program is distributed in the hope that it will be useful, but
|
|
* WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include "common.h"
|
|
#include "keccak.h"
|
|
#include "merkle.h"
|
|
#include "gtest/gtest.h"
|
|
|
|
namespace p2pool {
|
|
|
|
TEST(merkle, tree)
|
|
{
|
|
hash input[10];
|
|
uint8_t data[] = "data 0";
|
|
|
|
for (size_t i = 0; i < 10; ++i, ++data[sizeof(data) - 2]) {
|
|
keccak(data, sizeof(data) - 1, input[i].h);
|
|
}
|
|
|
|
hash root;
|
|
std::vector<hash> hashes(1, input[0]);
|
|
|
|
auto check_full_tree = [&hashes, &root]() {
|
|
std::vector<std::vector<hash>> tree;
|
|
merkle_hash_full_tree(hashes, tree);
|
|
|
|
ASSERT_GE(tree.size(), 1);
|
|
|
|
const std::vector<hash>& tree_root = tree.back();
|
|
ASSERT_EQ(tree_root.size(), 1);
|
|
ASSERT_EQ(tree_root[0], root);
|
|
|
|
ASSERT_EQ(tree[0], hashes);
|
|
|
|
if (tree.size() > 1) {
|
|
ASSERT_LE(tree[1].size(), hashes.size());
|
|
ASSERT_GE(tree[1].size() * 2, hashes.size());
|
|
|
|
const size_t spill_size = tree[1].size() * 2 - hashes.size();
|
|
for (size_t i = 0; i < spill_size; ++i) {
|
|
ASSERT_EQ(tree[1][i], hashes[i]);
|
|
}
|
|
for (size_t i = spill_size, j = spill_size; i < tree[1].size(); ++i, j += 2) {
|
|
hash tmp;
|
|
keccak(hashes[j].h, HASH_SIZE * 2, tmp.h);
|
|
ASSERT_EQ(tmp, tree[1][i]);
|
|
}
|
|
}
|
|
|
|
for (size_t i = tree.size() - 1; i > 1; --i) {
|
|
ASSERT_EQ(tree[i].size() * 2, tree[i - 1].size());
|
|
for (size_t j = 0; j < tree[i].size(); ++j) {
|
|
hash tmp;
|
|
keccak(tree[i - 1][j * 2].h, HASH_SIZE * 2, tmp.h);
|
|
ASSERT_EQ(tmp, tree[i][j]);
|
|
}
|
|
}
|
|
|
|
for (size_t i = 0, n = hashes.size(); i < n; ++i) {
|
|
const hash& h = hashes[i];
|
|
std::vector<std::pair<bool, hash>> proof;
|
|
|
|
ASSERT_TRUE(get_merkle_proof(tree, h, proof));
|
|
ASSERT_TRUE(verify_merkle_proof(h, proof, root));
|
|
|
|
std::vector<hash> proof2;
|
|
proof2.reserve(proof.size());
|
|
|
|
for (const auto& p : proof) {
|
|
proof2.emplace_back(p.second);
|
|
}
|
|
|
|
ASSERT_TRUE(verify_merkle_proof(h, proof2, i, n, root));
|
|
}
|
|
};
|
|
|
|
// 1 leaf
|
|
merkle_hash(hashes, root);
|
|
ASSERT_EQ(root, input[0]);
|
|
check_full_tree();
|
|
|
|
// 2 leaves
|
|
hashes.push_back(input[1]);
|
|
merkle_hash(hashes, root);
|
|
|
|
hash check[8];
|
|
keccak(input[0].h, HASH_SIZE * 2, check[0].h);
|
|
ASSERT_EQ(root, check[0]);
|
|
check_full_tree();
|
|
|
|
// 3 leaves
|
|
hashes.push_back(input[2]);
|
|
merkle_hash(hashes, root);
|
|
|
|
keccak(input[1].h, HASH_SIZE * 2, check[1].h);
|
|
check[0] = input[0];
|
|
keccak(check[0].h, HASH_SIZE * 2, check[0].h);
|
|
ASSERT_EQ(root, check[0]);
|
|
check_full_tree();
|
|
|
|
// 4 leaves
|
|
hashes.push_back(input[3]);
|
|
merkle_hash(hashes, root);
|
|
|
|
keccak(input[0].h, HASH_SIZE * 2, check[0].h);
|
|
keccak(input[2].h, HASH_SIZE * 2, check[1].h);
|
|
keccak(check[0].h, HASH_SIZE * 2, check[0].h);
|
|
ASSERT_EQ(root, check[0]);
|
|
check_full_tree();
|
|
|
|
// 5 leaves
|
|
hashes.push_back(input[4]);
|
|
merkle_hash(hashes, root);
|
|
|
|
check[0] = input[0];
|
|
check[1] = input[1];
|
|
check[2] = input[2];
|
|
keccak(input[3].h, HASH_SIZE * 2, check[3].h);
|
|
keccak(check[0].h, HASH_SIZE * 2, check[0].h);
|
|
keccak(check[2].h, HASH_SIZE * 2, check[1].h);
|
|
keccak(check[0].h, HASH_SIZE * 2, check[0].h);
|
|
ASSERT_EQ(root, check[0]);
|
|
check_full_tree();
|
|
|
|
// 6 leaves
|
|
hashes.push_back(input[5]);
|
|
merkle_hash(hashes, root);
|
|
|
|
check[0] = input[0];
|
|
check[1] = input[1];
|
|
keccak(input[2].h, HASH_SIZE * 2, check[2].h);
|
|
keccak(input[4].h, HASH_SIZE * 2, check[3].h);
|
|
keccak(check[0].h, HASH_SIZE * 2, check[0].h);
|
|
keccak(check[2].h, HASH_SIZE * 2, check[1].h);
|
|
keccak(check[0].h, HASH_SIZE * 2, check[0].h);
|
|
ASSERT_EQ(root, check[0]);
|
|
check_full_tree();
|
|
|
|
// 7 leaves
|
|
hashes.push_back(input[6]);
|
|
merkle_hash(hashes, root);
|
|
|
|
check[0] = input[0];
|
|
keccak(input[1].h, HASH_SIZE * 2, check[1].h);
|
|
keccak(input[3].h, HASH_SIZE * 2, check[2].h);
|
|
keccak(input[5].h, HASH_SIZE * 2, check[3].h);
|
|
keccak(check[0].h, HASH_SIZE * 2, check[0].h);
|
|
keccak(check[2].h, HASH_SIZE * 2, check[1].h);
|
|
keccak(check[0].h, HASH_SIZE * 2, check[0].h);
|
|
ASSERT_EQ(root, check[0]);
|
|
check_full_tree();
|
|
|
|
// 8 leaves
|
|
hashes.push_back(input[7]);
|
|
merkle_hash(hashes, root);
|
|
|
|
keccak(input[0].h, HASH_SIZE * 2, check[0].h);
|
|
keccak(input[2].h, HASH_SIZE * 2, check[1].h);
|
|
keccak(input[4].h, HASH_SIZE * 2, check[2].h);
|
|
keccak(input[6].h, HASH_SIZE * 2, check[3].h);
|
|
keccak(check[0].h, HASH_SIZE * 2, check[0].h);
|
|
keccak(check[2].h, HASH_SIZE * 2, check[1].h);
|
|
keccak(check[0].h, HASH_SIZE * 2, check[0].h);
|
|
ASSERT_EQ(root, check[0]);
|
|
check_full_tree();
|
|
|
|
// 9 leaves
|
|
hashes.push_back(input[8]);
|
|
merkle_hash(hashes, root);
|
|
|
|
for (size_t i = 0; i < 7; ++i) {
|
|
check[i] = input[i];
|
|
}
|
|
keccak(input[7].h, HASH_SIZE * 2, check[7].h);
|
|
|
|
keccak(check[0].h, HASH_SIZE * 2, check[0].h);
|
|
keccak(check[2].h, HASH_SIZE * 2, check[1].h);
|
|
keccak(check[4].h, HASH_SIZE * 2, check[2].h);
|
|
keccak(check[6].h, HASH_SIZE * 2, check[3].h);
|
|
|
|
keccak(check[0].h, HASH_SIZE * 2, check[0].h);
|
|
keccak(check[2].h, HASH_SIZE * 2, check[1].h);
|
|
|
|
keccak(check[0].h, HASH_SIZE * 2, check[0].h);
|
|
|
|
ASSERT_EQ(root, check[0]);
|
|
check_full_tree();
|
|
|
|
// 10 leaves
|
|
hashes.push_back(input[9]);
|
|
merkle_hash(hashes, root);
|
|
|
|
for (size_t i = 0; i < 6; ++i) {
|
|
check[i] = input[i];
|
|
}
|
|
keccak(input[6].h, HASH_SIZE * 2, check[6].h);
|
|
keccak(input[8].h, HASH_SIZE * 2, check[7].h);
|
|
|
|
keccak(check[0].h, HASH_SIZE * 2, check[0].h);
|
|
keccak(check[2].h, HASH_SIZE * 2, check[1].h);
|
|
keccak(check[4].h, HASH_SIZE * 2, check[2].h);
|
|
keccak(check[6].h, HASH_SIZE * 2, check[3].h);
|
|
|
|
keccak(check[0].h, HASH_SIZE * 2, check[0].h);
|
|
keccak(check[2].h, HASH_SIZE * 2, check[1].h);
|
|
|
|
keccak(check[0].h, HASH_SIZE * 2, check[0].h);
|
|
|
|
ASSERT_EQ(root, check[0]);
|
|
check_full_tree();
|
|
}
|
|
|
|
TEST(merkle, aux_slot)
|
|
{
|
|
hash id;
|
|
|
|
ASSERT_EQ(get_aux_slot(id, 0, 0), 0U);
|
|
ASSERT_EQ(get_aux_slot(id, 0, 1), 0U);
|
|
ASSERT_EQ(get_aux_slot(id, 0, 2), 0U);
|
|
ASSERT_EQ(get_aux_slot(id, 0, 3), 0U);
|
|
ASSERT_EQ(get_aux_slot(id, 0, 4), 0U);
|
|
ASSERT_EQ(get_aux_slot(id, 0, 5), 1U);
|
|
ASSERT_EQ(get_aux_slot(id, 0, 6), 0U);
|
|
ASSERT_EQ(get_aux_slot(id, 0, 7), 5U);
|
|
ASSERT_EQ(get_aux_slot(id, 0, 8), 0U);
|
|
ASSERT_EQ(get_aux_slot(id, 0, 9), 6U);
|
|
|
|
ASSERT_EQ(get_aux_slot(id, 0, std::numeric_limits<uint32_t>::max()), 2389612776U);
|
|
ASSERT_EQ(get_aux_slot(id, 1, std::numeric_limits<uint32_t>::max()), 1080669337U);
|
|
}
|
|
|
|
TEST(merkle, aux_nonce)
|
|
{
|
|
std::vector<hash> aux_id;
|
|
uint32_t nonce;
|
|
|
|
ASSERT_TRUE(find_aux_nonce(aux_id, nonce));
|
|
ASSERT_EQ(nonce, 0);
|
|
|
|
uint8_t data[] = "aux0";
|
|
|
|
const uint32_t nonces[] = { 0, 0, 0, 7, 16, 56, 1, 287, 1423, 1074 };
|
|
hash h;
|
|
|
|
for (size_t i = 0; i < 10; ++i, ++data[sizeof(data) - 2]) {
|
|
keccak(data, sizeof(data) - 1, h.h);
|
|
aux_id.push_back(h);
|
|
|
|
ASSERT_TRUE(find_aux_nonce(aux_id, nonce));
|
|
ASSERT_EQ(nonce, nonces[i]);
|
|
}
|
|
|
|
h = aux_id.front();
|
|
|
|
aux_id.clear();
|
|
aux_id.push_back(h);
|
|
aux_id.push_back(h);
|
|
|
|
ASSERT_FALSE(find_aux_nonce(aux_id, nonce));
|
|
}
|
|
|
|
}
|