mirror of
https://github.com/XRPLF/rippled.git
synced 2026-07-13 18:20:22 +00:00
428 lines
15 KiB
C++
428 lines
15 KiB
C++
#include <test/shamap/common.h>
|
|
#include <test/unit_test/SuiteJournal.h>
|
|
|
|
#include <xrpl/basics/Blob.h>
|
|
#include <xrpl/basics/Buffer.h>
|
|
#include <xrpl/basics/SHAMapHash.h>
|
|
#include <xrpl/basics/base_uint.h>
|
|
#include <xrpl/beast/unit_test/suite.h>
|
|
#include <xrpl/beast/utility/Journal.h>
|
|
#include <xrpl/beast/utility/Zero.h>
|
|
#include <xrpl/shamap/SHAMap.h>
|
|
#include <xrpl/shamap/SHAMapInnerNode.h>
|
|
#include <xrpl/shamap/SHAMapItem.h>
|
|
#include <xrpl/shamap/SHAMapLeafNode.h>
|
|
#include <xrpl/shamap/SHAMapMissingNode.h>
|
|
#include <xrpl/shamap/SHAMapTreeNode.h>
|
|
|
|
#include <algorithm>
|
|
#include <array>
|
|
#include <cstdint>
|
|
#include <memory>
|
|
#include <type_traits>
|
|
#include <utility>
|
|
#include <vector>
|
|
|
|
namespace xrpl::tests {
|
|
|
|
#ifndef __INTELLISENSE__
|
|
static_assert(std::is_nothrow_destructible<SHAMap>{});
|
|
static_assert(!std::is_default_constructible<SHAMap>{});
|
|
static_assert(!std::is_copy_constructible<SHAMap>{});
|
|
static_assert(!std::is_copy_assignable<SHAMap>{});
|
|
static_assert(!std::is_move_constructible<SHAMap>{});
|
|
static_assert(!std::is_move_assignable<SHAMap>{});
|
|
|
|
static_assert(std::is_nothrow_destructible<SHAMap::ConstIterator>{});
|
|
static_assert(std::is_copy_constructible<SHAMap::ConstIterator>{});
|
|
static_assert(std::is_copy_assignable<SHAMap::ConstIterator>{});
|
|
static_assert(std::is_move_constructible<SHAMap::ConstIterator>{});
|
|
static_assert(std::is_move_assignable<SHAMap::ConstIterator>{});
|
|
|
|
static_assert(std::is_nothrow_destructible<SHAMapItem>{});
|
|
static_assert(!std::is_default_constructible<SHAMapItem>{});
|
|
static_assert(!std::is_copy_constructible<SHAMapItem>{});
|
|
|
|
static_assert(std::is_nothrow_destructible<SHAMapNodeID>{});
|
|
static_assert(std::is_default_constructible<SHAMapNodeID>{});
|
|
static_assert(std::is_copy_constructible<SHAMapNodeID>{});
|
|
static_assert(std::is_copy_assignable<SHAMapNodeID>{});
|
|
static_assert(std::is_move_constructible<SHAMapNodeID>{});
|
|
static_assert(std::is_move_assignable<SHAMapNodeID>{});
|
|
|
|
static_assert(std::is_nothrow_destructible<SHAMapHash>{});
|
|
static_assert(std::is_default_constructible<SHAMapHash>{});
|
|
static_assert(std::is_copy_constructible<SHAMapHash>{});
|
|
static_assert(std::is_copy_assignable<SHAMapHash>{});
|
|
static_assert(std::is_move_constructible<SHAMapHash>{});
|
|
static_assert(std::is_move_assignable<SHAMapHash>{});
|
|
|
|
static_assert(std::is_nothrow_destructible<SHAMapTreeNode>{});
|
|
static_assert(!std::is_default_constructible<SHAMapTreeNode>{});
|
|
static_assert(!std::is_copy_constructible<SHAMapTreeNode>{});
|
|
static_assert(!std::is_copy_assignable<SHAMapTreeNode>{});
|
|
static_assert(!std::is_move_constructible<SHAMapTreeNode>{});
|
|
static_assert(!std::is_move_assignable<SHAMapTreeNode>{});
|
|
|
|
static_assert(std::is_nothrow_destructible<SHAMapInnerNode>{});
|
|
static_assert(!std::is_default_constructible<SHAMapInnerNode>{});
|
|
static_assert(!std::is_copy_constructible<SHAMapInnerNode>{});
|
|
static_assert(!std::is_copy_assignable<SHAMapInnerNode>{});
|
|
static_assert(!std::is_move_constructible<SHAMapInnerNode>{});
|
|
static_assert(!std::is_move_assignable<SHAMapInnerNode>{});
|
|
|
|
static_assert(std::is_nothrow_destructible<SHAMapLeafNode>{});
|
|
static_assert(!std::is_default_constructible<SHAMapLeafNode>{});
|
|
static_assert(!std::is_copy_constructible<SHAMapLeafNode>{});
|
|
static_assert(!std::is_copy_assignable<SHAMapLeafNode>{});
|
|
static_assert(!std::is_move_constructible<SHAMapLeafNode>{});
|
|
static_assert(!std::is_move_assignable<SHAMapLeafNode>{});
|
|
#endif
|
|
|
|
inline bool
|
|
operator==(SHAMapItem const& a, SHAMapItem const& b)
|
|
{
|
|
return a.key() == b.key();
|
|
}
|
|
inline bool
|
|
operator!=(SHAMapItem const& a, SHAMapItem const& b)
|
|
{
|
|
return a.key() != b.key();
|
|
}
|
|
inline bool
|
|
operator==(SHAMapItem const& a, uint256 const& b)
|
|
{
|
|
return a.key() == b;
|
|
}
|
|
inline bool
|
|
operator!=(SHAMapItem const& a, uint256 const& b)
|
|
{
|
|
return a.key() != b;
|
|
}
|
|
|
|
class SHAMap_test : public beast::unit_test::Suite
|
|
{
|
|
public:
|
|
static Buffer
|
|
intToVuc(int v)
|
|
{
|
|
Buffer vuc(32);
|
|
std::fill_n(vuc.data(), vuc.size(), static_cast<std::uint8_t>(v));
|
|
return vuc;
|
|
}
|
|
|
|
void
|
|
run() override
|
|
{
|
|
using beast::Severity;
|
|
test::SuiteJournal journal("SHAMap_test", *this);
|
|
|
|
run(true, journal);
|
|
run(false, journal);
|
|
}
|
|
|
|
void
|
|
run(bool backed, beast::Journal const& journal)
|
|
{
|
|
if (backed)
|
|
{
|
|
testcase("add/traverse backed");
|
|
}
|
|
else
|
|
{
|
|
testcase("add/traverse unbacked");
|
|
}
|
|
|
|
tests::TestNodeFamily f(journal);
|
|
|
|
// h3 and h4 differ only in the leaf, same terminal node (level 19)
|
|
constexpr uint256 kH1("092891fe4ef6cee585fdc6fda0e09eb4d386363158ec3321b8123e5a772c6ca7");
|
|
constexpr uint256 kH2("436ccbac3347baa1f1e53baeef1f43334da88f1f6d70d963b833afd6dfa289fe");
|
|
constexpr uint256 kH3("b92891fe4ef6cee585fdc6fda1e09eb4d386363158ec3321b8123e5a772c6ca8");
|
|
constexpr uint256 kH4("b92891fe4ef6cee585fdc6fda2e09eb4d386363158ec3321b8123e5a772c6ca8");
|
|
constexpr uint256 kH5("a92891fe4ef6cee585fdc6fda0e09eb4d386363158ec3321b8123e5a772c6ca7");
|
|
|
|
SHAMap sMap(SHAMapType::FREE, f);
|
|
sMap.invariants();
|
|
if (!backed)
|
|
sMap.setUnbacked();
|
|
|
|
auto i1 = makeShamapitem(kH1, intToVuc(1));
|
|
auto i2 = makeShamapitem(kH2, intToVuc(2));
|
|
auto i3 = makeShamapitem(kH3, intToVuc(3));
|
|
auto i4 = makeShamapitem(kH4, intToVuc(4));
|
|
auto i5 = makeShamapitem(kH5, intToVuc(5));
|
|
|
|
unexpected(!sMap.addItem(SHAMapNodeType::TnTransactionNm, makeShamapitem(*i2)), "no add");
|
|
sMap.invariants();
|
|
unexpected(!sMap.addItem(SHAMapNodeType::TnTransactionNm, makeShamapitem(*i1)), "no add");
|
|
sMap.invariants();
|
|
|
|
auto i = sMap.begin();
|
|
auto e = sMap.end();
|
|
unexpected(i == e || (*i != *i1), "bad traverse");
|
|
++i;
|
|
unexpected(i == e || (*i != *i2), "bad traverse");
|
|
++i;
|
|
unexpected(i != e, "bad traverse");
|
|
sMap.addItem(SHAMapNodeType::TnTransactionNm, makeShamapitem(*i4));
|
|
sMap.invariants();
|
|
sMap.delItem(i2->key());
|
|
sMap.invariants();
|
|
sMap.addItem(SHAMapNodeType::TnTransactionNm, makeShamapitem(*i3));
|
|
sMap.invariants();
|
|
i = sMap.begin();
|
|
e = sMap.end();
|
|
unexpected(i == e || (*i != *i1), "bad traverse");
|
|
++i;
|
|
unexpected(i == e || (*i != *i3), "bad traverse");
|
|
++i;
|
|
unexpected(i == e || (*i != *i4), "bad traverse");
|
|
++i;
|
|
unexpected(i != e, "bad traverse");
|
|
|
|
if (backed)
|
|
{
|
|
testcase("snapshot backed");
|
|
}
|
|
else
|
|
{
|
|
testcase("snapshot unbacked");
|
|
}
|
|
|
|
SHAMapHash const mapHash = sMap.getHash();
|
|
std::shared_ptr<SHAMap> const map2 = sMap.snapShot(false);
|
|
map2->invariants();
|
|
unexpected(sMap.getHash() != mapHash, "bad snapshot");
|
|
unexpected(map2->getHash() != mapHash, "bad snapshot");
|
|
|
|
SHAMap::Delta delta;
|
|
BEAST_EXPECT(sMap.compare(*map2, delta, 100));
|
|
BEAST_EXPECT(delta.empty());
|
|
|
|
unexpected(!sMap.delItem(sMap.begin()->key()), "bad mod");
|
|
sMap.invariants();
|
|
unexpected(sMap.getHash() == mapHash, "bad snapshot");
|
|
unexpected(map2->getHash() != mapHash, "bad snapshot");
|
|
|
|
BEAST_EXPECT(sMap.compare(*map2, delta, 100));
|
|
BEAST_EXPECT(delta.size() == 1);
|
|
BEAST_EXPECT(delta.begin()->first == kH1);
|
|
BEAST_EXPECT(delta.begin()->second.first == nullptr);
|
|
BEAST_EXPECT(delta.begin()->second.second->key() == kH1);
|
|
|
|
sMap.dump();
|
|
|
|
if (backed)
|
|
{
|
|
testcase("build/tear backed");
|
|
}
|
|
else
|
|
{
|
|
testcase("build/tear unbacked");
|
|
}
|
|
{
|
|
static constexpr std::array keys{
|
|
uint256(
|
|
"b92891fe4ef6cee585fdc6fda1e09eb4d386363158ec3321b8123e"
|
|
"5a772c6ca8"),
|
|
uint256(
|
|
"b92881fe4ef6cee585fdc6fda1e09eb4d386363158ec3321b8123e"
|
|
"5a772c6ca8"),
|
|
uint256(
|
|
"b92691fe4ef6cee585fdc6fda1e09eb4d386363158ec3321b8123e"
|
|
"5a772c6ca8"),
|
|
uint256(
|
|
"b92791fe4ef6cee585fdc6fda1e09eb4d386363158ec3321b8123e"
|
|
"5a772c6ca8"),
|
|
uint256(
|
|
"b91891fe4ef6cee585fdc6fda1e09eb4d386363158ec3321b8123e"
|
|
"5a772c6ca8"),
|
|
uint256(
|
|
"b99891fe4ef6cee585fdc6fda1e09eb4d386363158ec3321b8123e"
|
|
"5a772c6ca8"),
|
|
uint256(
|
|
"f22891fe4ef6cee585fdc6fda1e09eb4d386363158ec3321b8123e"
|
|
"5a772c6ca8"),
|
|
uint256(
|
|
"292891fe4ef6cee585fdc6fda1e09eb4d386363158ec3321b8123e"
|
|
"5a772c6ca8")};
|
|
|
|
static constexpr std::array kHashes{
|
|
uint256(
|
|
"B7387CFEA0465759ADC718E8C42B52D2309D179B326E239EB5075C"
|
|
"64B6281F7F"),
|
|
uint256(
|
|
"FBC195A9592A54AB44010274163CB6BA95F497EC5BA0A883184546"
|
|
"7FB2ECE266"),
|
|
uint256(
|
|
"4E7D2684B65DFD48937FFB775E20175C43AF0C94066F7D5679F51A"
|
|
"E756795B75"),
|
|
uint256(
|
|
"7A2F312EB203695FFD164E038E281839EEF06A1B99BFC263F3CECC"
|
|
"6C74F93E07"),
|
|
uint256(
|
|
"395A6691A372387A703FB0F2C6D2C405DAF307D0817F8F0E207596"
|
|
"462B0E3A3E"),
|
|
uint256(
|
|
"D044C0A696DE3169CC70AE216A1564D69DE96582865796142CE7D9"
|
|
"8A84D9DDE4"),
|
|
uint256(
|
|
"76DCC77C4027309B5A91AD164083264D70B77B5E43E08AEDA5EBF9"
|
|
"4361143615"),
|
|
uint256(
|
|
"DF4220E93ADC6F5569063A01B4DC79F8DB9553B6A3222ADE23DEA0"
|
|
"2BBE7230E5")};
|
|
|
|
SHAMap map(SHAMapType::FREE, f);
|
|
if (!backed)
|
|
map.setUnbacked();
|
|
|
|
BEAST_EXPECT(map.getHash() == beast::kZero);
|
|
for (int k = 0; k < keys.size(); ++k)
|
|
{
|
|
BEAST_EXPECT(map.addItem(
|
|
SHAMapNodeType::TnTransactionNm, makeShamapitem(keys[k], intToVuc(k))));
|
|
BEAST_EXPECT(map.getHash().asUInt256() == kHashes[k]);
|
|
map.invariants();
|
|
}
|
|
for (int k = keys.size() - 1; k >= 0; --k)
|
|
{
|
|
BEAST_EXPECT(map.getHash().asUInt256() == kHashes[k]);
|
|
BEAST_EXPECT(map.delItem(keys[k]));
|
|
map.invariants();
|
|
}
|
|
BEAST_EXPECT(map.getHash() == beast::kZero);
|
|
}
|
|
|
|
if (backed)
|
|
{
|
|
testcase("iterate backed");
|
|
}
|
|
else
|
|
{
|
|
testcase("iterate unbacked");
|
|
}
|
|
|
|
{
|
|
static constexpr std::array keys{
|
|
uint256(
|
|
"f22891fe4ef6cee585fdc6fda1e09eb4d386363158ec3321b8123e"
|
|
"5a772c6ca8"),
|
|
uint256(
|
|
"b99891fe4ef6cee585fdc6fda1e09eb4d386363158ec3321b8123e"
|
|
"5a772c6ca8"),
|
|
uint256(
|
|
"b92891fe4ef6cee585fdc6fda1e09eb4d386363158ec3321b8123e"
|
|
"5a772c6ca8"),
|
|
uint256(
|
|
"b92881fe4ef6cee585fdc6fda1e09eb4d386363158ec3321b8123e"
|
|
"5a772c6ca8"),
|
|
uint256(
|
|
"b92791fe4ef6cee585fdc6fda1e09eb4d386363158ec3321b8123e"
|
|
"5a772c6ca8"),
|
|
uint256(
|
|
"b92691fe4ef6cee585fdc6fda1e09eb4d386363158ec3321b8123e"
|
|
"5a772c6ca8"),
|
|
uint256(
|
|
"b91891fe4ef6cee585fdc6fda1e09eb4d386363158ec3321b8123e"
|
|
"5a772c6ca8"),
|
|
uint256(
|
|
"292891fe4ef6cee585fdc6fda1e09eb4d386363158ec3321b8123e"
|
|
"5a772c6ca8")};
|
|
|
|
tests::TestNodeFamily tf{journal};
|
|
SHAMap map{SHAMapType::FREE, tf};
|
|
if (!backed)
|
|
map.setUnbacked();
|
|
for (auto const& k : keys)
|
|
{
|
|
map.addItem(SHAMapNodeType::TnTransactionNm, makeShamapitem(k, intToVuc(0)));
|
|
map.invariants();
|
|
}
|
|
|
|
int h = 7;
|
|
for (auto const& k : map)
|
|
{
|
|
BEAST_EXPECT(k.key() == keys[h]);
|
|
--h;
|
|
}
|
|
}
|
|
}
|
|
};
|
|
|
|
class SHAMapPathProof_test : public beast::unit_test::Suite
|
|
{
|
|
void
|
|
run() override
|
|
{
|
|
test::SuiteJournal journal("SHAMapPathProof_test", *this);
|
|
|
|
tests::TestNodeFamily tf{journal};
|
|
SHAMap map{SHAMapType::FREE, tf};
|
|
map.setUnbacked();
|
|
|
|
uint256 key;
|
|
uint256 rootHash;
|
|
std::vector<Blob> goodPath;
|
|
|
|
for (unsigned char c = 1; c < 100; ++c)
|
|
{
|
|
uint256 k(c);
|
|
map.addItem(
|
|
SHAMapNodeType::TnAccountState, makeShamapitem(k, Slice{k.data(), k.size()}));
|
|
map.invariants();
|
|
|
|
auto root = map.getHash().asUInt256();
|
|
auto path = map.getProofPath(k);
|
|
BEAST_EXPECT(path);
|
|
if (!path)
|
|
break;
|
|
BEAST_EXPECT(map.verifyProofPath(root, k, *path));
|
|
if (c == 1)
|
|
{
|
|
// extra node
|
|
path->insert(path->begin(), path->front());
|
|
BEAST_EXPECT(!map.verifyProofPath(root, k, *path));
|
|
// wrong key
|
|
uint256 const wrongKey(c + 1);
|
|
BEAST_EXPECT(!map.getProofPath(wrongKey));
|
|
}
|
|
if (c == 99)
|
|
{
|
|
key = k;
|
|
rootHash = root;
|
|
goodPath = std::move(*path);
|
|
}
|
|
}
|
|
|
|
// still good
|
|
BEAST_EXPECT(map.verifyProofPath(rootHash, key, goodPath));
|
|
// empty path
|
|
std::vector<Blob> badPath;
|
|
BEAST_EXPECT(!map.verifyProofPath(rootHash, key, badPath));
|
|
// too long
|
|
badPath = goodPath;
|
|
badPath.push_back(goodPath.back());
|
|
BEAST_EXPECT(!map.verifyProofPath(rootHash, key, badPath));
|
|
// bad node
|
|
badPath.clear();
|
|
badPath.emplace_back(100, 100);
|
|
BEAST_EXPECT(!map.verifyProofPath(rootHash, key, badPath));
|
|
// bad node type
|
|
badPath.clear();
|
|
badPath.push_back(goodPath.front());
|
|
badPath.front().back()--; // change node type
|
|
BEAST_EXPECT(!map.verifyProofPath(rootHash, key, badPath));
|
|
// all inner
|
|
badPath.clear();
|
|
badPath = goodPath;
|
|
badPath.erase(badPath.begin());
|
|
BEAST_EXPECT(!map.verifyProofPath(rootHash, key, badPath));
|
|
}
|
|
};
|
|
|
|
BEAST_DEFINE_TESTSUITE(SHAMap, shamap, xrpl);
|
|
BEAST_DEFINE_TESTSUITE(SHAMapPathProof, shamap, xrpl);
|
|
} // namespace xrpl::tests
|