mirror of
https://codeberg.org/wownero/wownero-lws
synced 2026-01-10 15:45:15 -08:00
Add support for subaddress lookahead (#195)
This commit is contained in:
Lee *!* Clagett
committed by
Lee *!* Clagett
Lee *!* Clagett
parent
e8b889e95f
commit
16111cae2c
@@ -30,6 +30,9 @@
|
||||
|
||||
#include <boost/filesystem/operations.hpp>
|
||||
#include "common/util.h" // monero/src/
|
||||
#include "db/string.h"
|
||||
#include "error.h"
|
||||
#include "framework.test.h"
|
||||
|
||||
namespace lws { namespace db { namespace test
|
||||
{
|
||||
@@ -68,3 +71,201 @@ namespace lws { namespace db { namespace test
|
||||
return lws::account{make_db_account(pubs, key), {}, {}};
|
||||
}
|
||||
}}} // lws // db // test
|
||||
|
||||
LWS_CASE("lws::db::storage")
|
||||
{
|
||||
lws::db::account_address account_address{};
|
||||
crypto::secret_key view{};
|
||||
crypto::generate_keys(account_address.spend_public, view);
|
||||
crypto::generate_keys(account_address.view_public, view);
|
||||
const std::string address = lws::db::address_string(account_address);
|
||||
const std::string viewkey = epee::to_hex::string(epee::as_byte_span(unwrap(unwrap(view))));
|
||||
|
||||
const lws::db::address_index lookahead{
|
||||
lws::db::major_index(2), lws::db::minor_index(2)
|
||||
};
|
||||
|
||||
SETUP("Database with account")
|
||||
{
|
||||
lws::db::test::cleanup_db on_scope_exit{};
|
||||
lws::db::storage db = lws::db::test::get_fresh_db();
|
||||
EXPECT(db.add_account(account_address, view));
|
||||
|
||||
const lws::db::block_info last_block =
|
||||
MONERO_UNWRAP(MONERO_UNWRAP(db.start_read()).get_last_block());
|
||||
|
||||
const auto get_account = [&db, &account_address] () -> lws::db::account
|
||||
{
|
||||
return MONERO_UNWRAP(MONERO_UNWRAP(db.start_read()).get_account(account_address)).second;
|
||||
};
|
||||
|
||||
SECTION("rollback lookahead_fail via rescan")
|
||||
{
|
||||
EXPECT(db.import_request(account_address, last_block.id, lookahead));
|
||||
EXPECT(db.accept_requests(lws::db::request::import_scan, {std::addressof(account_address), 1}, 10));
|
||||
|
||||
const auto block_failed = lws::db::block_id(to_uint(last_block.id) + 1);
|
||||
const auto update =
|
||||
db.update_lookahead(account_address, block_failed, {lws::db::major_index(10), lws::db::minor_index(10)}, 10);
|
||||
EXPECT(update == -1);
|
||||
EXPECT(get_account().lookahead_fail == block_failed);
|
||||
|
||||
EXPECT(db.rescan(last_block.id, {std::addressof(account_address), 1}));
|
||||
EXPECT(get_account().lookahead_fail == lws::db::block_id(0));
|
||||
}
|
||||
|
||||
SECTION("rollback lookahead_fail via import")
|
||||
{
|
||||
EXPECT(db.import_request(account_address, last_block.id, lookahead));
|
||||
EXPECT(db.accept_requests(lws::db::request::import_scan, {std::addressof(account_address), 1}, 10));
|
||||
|
||||
const auto block_failed = lws::db::block_id(to_uint(last_block.id) + 1);
|
||||
const auto update =
|
||||
db.update_lookahead(account_address, block_failed, {lws::db::major_index(10), lws::db::minor_index(10)}, 10);
|
||||
EXPECT(update == -1);
|
||||
EXPECT(get_account().lookahead_fail == block_failed);
|
||||
|
||||
EXPECT(db.import_request(account_address, last_block.id, lookahead));
|
||||
EXPECT(db.accept_requests(lws::db::request::import_scan, {std::addressof(account_address), 1}, 10));
|
||||
EXPECT(get_account().lookahead_fail == lws::db::block_id(0));
|
||||
}
|
||||
|
||||
const auto add_output = [&] ()
|
||||
{
|
||||
auto account = get_account();
|
||||
const lws::db::transaction_link link{
|
||||
lws::db::block_id(to_uint(last_block.id) + 1), crypto::rand<crypto::hash>()
|
||||
};
|
||||
const crypto::public_key tx_public = []() {
|
||||
crypto::secret_key secret;
|
||||
crypto::public_key out;
|
||||
crypto::generate_keys(out, secret);
|
||||
return out;
|
||||
}();
|
||||
const crypto::hash tx_prefix = crypto::rand<crypto::hash>();
|
||||
const crypto::public_key pub = crypto::rand<crypto::public_key>();
|
||||
const rct::key ringct = crypto::rand<rct::key>();
|
||||
const auto extra =
|
||||
lws::db::extra(lws::db::extra::coinbase_output | lws::db::extra::ringct_output);
|
||||
const auto payment_id_ = crypto::rand<lws::db::output::payment_id_>();
|
||||
const crypto::key_image image = crypto::rand<crypto::key_image>();
|
||||
|
||||
lws::account real_account{account, {}, {}};
|
||||
real_account.add_out(
|
||||
lws::db::output{
|
||||
link,
|
||||
lws::db::output::spend_meta_{
|
||||
lws::db::output_id{500, 30},
|
||||
std::uint64_t(40000),
|
||||
std::uint32_t(16),
|
||||
std::uint32_t(2),
|
||||
tx_public
|
||||
},
|
||||
std::uint64_t(7000),
|
||||
std::uint64_t(4670),
|
||||
tx_prefix,
|
||||
pub,
|
||||
ringct,
|
||||
{0, 0, 0, 0, 0, 0, 0},
|
||||
lws::db::pack(extra, sizeof(crypto::hash)),
|
||||
payment_id_,
|
||||
std::uint64_t(100),
|
||||
lws::db::address_index{lws::db::major_index(2), lws::db::minor_index(10)}
|
||||
}
|
||||
);
|
||||
|
||||
{
|
||||
std::vector<crypto::hash> hashes{
|
||||
last_block.hash,
|
||||
crypto::rand<crypto::hash>(),
|
||||
crypto::rand<crypto::hash>(),
|
||||
crypto::rand<crypto::hash>(),
|
||||
crypto::rand<crypto::hash>(),
|
||||
crypto::rand<crypto::hash>()
|
||||
};
|
||||
|
||||
const auto thing = db.update(last_block.id, epee::to_span(hashes), {std::addressof(real_account), 1}, {});
|
||||
if (!thing)
|
||||
std::cout << thing.error().message() << std::endl;
|
||||
}
|
||||
};
|
||||
|
||||
|
||||
SECTION("Lookahead with outputs")
|
||||
{
|
||||
add_output();
|
||||
const auto scan_height = get_account().scan_height;
|
||||
|
||||
EXPECT(MONERO_UNWRAP(MONERO_UNWRAP(db.start_read()).get_subaddresses(lws::db::account_id(1))).empty());
|
||||
EXPECT(db.import_request(account_address, scan_height, lookahead));
|
||||
EXPECT(db.accept_requests(lws::db::request::import_scan, {std::addressof(account_address), 1}, 18));
|
||||
|
||||
const std::vector<lws::db::subaddress_dict> expected_range{
|
||||
{lws::db::major_index(0), {{lws::db::index_range{lws::db::minor_index(0), lws::db::minor_index(1)}}}},
|
||||
{lws::db::major_index(1), {{lws::db::index_range{lws::db::minor_index(0), lws::db::minor_index(1)}}}},
|
||||
{lws::db::major_index(2), {{lws::db::index_range{lws::db::minor_index(0), lws::db::minor_index(11)}}}},
|
||||
{lws::db::major_index(3), {{lws::db::index_range{lws::db::minor_index(0), lws::db::minor_index(1)}}}},
|
||||
};
|
||||
EXPECT(MONERO_UNWRAP(MONERO_UNWRAP(db.start_read()).get_subaddresses(lws::db::account_id(1))) == expected_range);
|
||||
|
||||
SECTION("shrink lookahead")
|
||||
{
|
||||
const lws::db::address_index test1{
|
||||
lws::db::major_index(3), lws::db::minor_index(2)
|
||||
};
|
||||
const lws::db::address_index test2{
|
||||
lws::db::major_index(2), lws::db::minor_index(3)
|
||||
};
|
||||
const lws::db::address_index shrink{
|
||||
lws::db::major_index(1), lws::db::minor_index(1)
|
||||
};
|
||||
EXPECT(!db.shrink_lookahead(account_address, test1));
|
||||
EXPECT(!db.shrink_lookahead(account_address, test2));
|
||||
EXPECT(db.shrink_lookahead(account_address, lookahead));
|
||||
|
||||
EXPECT(MONERO_UNWRAP(MONERO_UNWRAP(db.start_read()).get_subaddresses(lws::db::account_id(1))) == expected_range);
|
||||
EXPECT(get_account().lookahead == lookahead);
|
||||
EXPECT(MONERO_UNWRAP(MONERO_UNWRAP(db.start_read()).get_subaddresses(lws::db::account_id(1))) == expected_range);
|
||||
|
||||
EXPECT(db.shrink_lookahead(account_address, shrink));
|
||||
EXPECT(get_account().lookahead == shrink);
|
||||
EXPECT(MONERO_UNWRAP(MONERO_UNWRAP(db.start_read()).get_subaddresses(lws::db::account_id(1))) == expected_range);
|
||||
}
|
||||
}
|
||||
|
||||
SECTION("Lookahead failure with outputs")
|
||||
{
|
||||
add_output();
|
||||
const auto scan_height = get_account().scan_height;
|
||||
|
||||
EXPECT(MONERO_UNWRAP(MONERO_UNWRAP(db.start_read()).get_subaddresses(lws::db::account_id(1))).empty());
|
||||
EXPECT(db.import_request(account_address, scan_height, lookahead));
|
||||
EXPECT(db.accept_requests(lws::db::request::import_scan, {std::addressof(account_address), 1}, 17));
|
||||
|
||||
const std::vector<lws::db::subaddress_dict> expected_range{
|
||||
{lws::db::major_index(0), {{lws::db::index_range{lws::db::minor_index(0), lws::db::minor_index(1)}}}},
|
||||
{lws::db::major_index(1), {{lws::db::index_range{lws::db::minor_index(0), lws::db::minor_index(1)}}}},
|
||||
{lws::db::major_index(2), {{lws::db::index_range{lws::db::minor_index(0), lws::db::minor_index(11)}}}}
|
||||
};
|
||||
|
||||
EXPECT(MONERO_UNWRAP(MONERO_UNWRAP(db.start_read()).get_subaddresses(lws::db::account_id(1))) == expected_range);
|
||||
|
||||
SECTION("shrink lookahead")
|
||||
{
|
||||
const lws::db::address_index test1{
|
||||
lws::db::major_index(3), lws::db::minor_index(2)
|
||||
};
|
||||
const lws::db::address_index test2{
|
||||
lws::db::major_index(2), lws::db::minor_index(3)
|
||||
};
|
||||
const lws::db::address_index shrink{
|
||||
lws::db::major_index(1), lws::db::minor_index(1)
|
||||
};
|
||||
EXPECT(!db.shrink_lookahead(account_address, test1));
|
||||
EXPECT(!db.shrink_lookahead(account_address, test2));
|
||||
EXPECT(!db.shrink_lookahead(account_address, lookahead));
|
||||
EXPECT(!db.shrink_lookahead(account_address, shrink));
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user