328 lines
15 KiB
C++
328 lines
15 KiB
C++
// Copyright (c) 2023-2024, The Monero Project
|
|
//
|
|
// All rights reserved.
|
|
//
|
|
// Redistribution and use in source and binary forms, with or without modification, are
|
|
// permitted provided that the following conditions are met:
|
|
//
|
|
// 1. Redistributions of source code must retain the above copyright notice, this list of
|
|
// conditions and the following disclaimer.
|
|
//
|
|
// 2. Redistributions in binary form must reproduce the above copyright notice, this list
|
|
// of conditions and the following disclaimer in the documentation and/or other
|
|
// materials provided with the distribution.
|
|
//
|
|
// 3. Neither the name of the copyright holder nor the names of its contributors may be
|
|
// used to endorse or promote products derived from this software without specific
|
|
// prior written permission.
|
|
//
|
|
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY
|
|
// EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
|
|
// MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL
|
|
// THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
|
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
|
// PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
|
// INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
|
|
// STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF
|
|
// THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
|
|
#include "gtest/gtest.h"
|
|
|
|
#include "carrot_impl/address_device_ram_borrowed.h"
|
|
#include "carrot_impl/carrot_tx_builder_inputs.h"
|
|
#include "carrot_mock_helpers.h"
|
|
#include "fake_pruned_blockchain.h"
|
|
#include "fcmp_pp/prove.h"
|
|
#include "wallet/tx_builder.h"
|
|
|
|
#undef MONERO_DEFAULT_LOG_CATEGORY
|
|
#define MONERO_DEFAULT_LOG_CATEGORY "unit_tests.wallet_scanning"
|
|
|
|
//----------------------------------------------------------------------------------------------------------------------
|
|
TEST(wallet_scanning, positive_smallout_main_addr_all_types_outputs)
|
|
{
|
|
// Test that wallet can scan and recover enotes of following type:
|
|
// a. pre-ringct coinbase
|
|
// b. pre-ringct
|
|
// c. ringct coinbase
|
|
// d. ringct long-amount
|
|
// e. ringct short-amount
|
|
// f. view-tagged ringct coinbase
|
|
// g. view-tagged pre-ringct (only possible in unmixable sweep txs)
|
|
// h. view-tagged ringct
|
|
// i. carrot v1 coinbase
|
|
// j. carrot v1 normal
|
|
// k. carrot v1 special
|
|
// l. carrot v1 internal (@TODO)
|
|
//
|
|
// All enotes are addressed to the main address in 2-out noin-coinbase txs or 1-out coinbase txs.
|
|
// We also don't test reorgs here.
|
|
|
|
// init blockchain
|
|
mock::fake_pruned_blockchain bc(0);
|
|
|
|
// generate wallet
|
|
tools::wallet2 w(cryptonote::MAINNET, /*kdf_rounds=*/1, /*unattended=*/true);
|
|
w.generate("", "");
|
|
const cryptonote::account_keys &acc_keys = w.get_account().get_keys();
|
|
const cryptonote::account_public_address main_addr = w.get_account().get_keys().m_account_address;
|
|
ASSERT_EQ(0, w.balance(0, true));
|
|
bc.init_wallet_for_starting_block(w); // needed b/c internal logic
|
|
|
|
uint64_t refresh_height = 0;
|
|
const auto wallet_process_new_blocks = [&w, &bc, &refresh_height]() -> boost::multiprecision::int128_t
|
|
{
|
|
const boost::multiprecision::int128_t old_balance = w.balance(0, true);
|
|
|
|
// note: doesn't handle reorgs
|
|
std::vector<cryptonote::block_complete_entry> block_entries;
|
|
std::vector<tools::wallet2::parsed_block> parsed_blocks;
|
|
bc.get_blocks_data(0, bc.height()-1, block_entries, parsed_blocks); //! @TODO: figure out why starting from refresh_height doesn't work
|
|
uint64_t blocks_added{};
|
|
auto output_tracker_cache = w.create_output_tracker_cache();
|
|
w.process_parsed_blocks(0, block_entries, parsed_blocks, blocks_added, output_tracker_cache);
|
|
|
|
// update refresh_height
|
|
refresh_height = bc.height();
|
|
|
|
// return amount of money received
|
|
return boost::multiprecision::int128_t(w.balance(0, true)) - old_balance;
|
|
};
|
|
|
|
// a. push block containing a pre-ringct coinbase output to wallet
|
|
bc.add_block(1, {}, main_addr);
|
|
|
|
// a. scan pre-ringct coinbase tx
|
|
auto balance_diff = wallet_process_new_blocks();
|
|
EXPECT_EQ(mock::fake_pruned_blockchain::miner_reward, balance_diff);
|
|
EXPECT_TRUE(verify_sals_of_recent_transfers());
|
|
|
|
// b. construct and push a pre-ringct tx
|
|
const rct::xmr_amount amount_b = rct::randXmrAmount(COIN);
|
|
{
|
|
const rct::xmr_amount fee = rct::randXmrAmount(COIN);
|
|
std::vector<cryptonote::tx_destination_entry> dests = {
|
|
cryptonote::tx_destination_entry(amount_b, acc_keys.m_account_address, false)};
|
|
cryptonote::transaction curr_tx = mock::construct_pre_carrot_tx_with_fake_inputs(
|
|
acc_keys,
|
|
w.m_subaddresses,
|
|
/*stripped_sources=*/{},
|
|
dests,
|
|
acc_keys.m_account_address,
|
|
fee,
|
|
/*hf_version=*/1);
|
|
ASSERT_FALSE(cryptonote::is_coinbase(curr_tx));
|
|
ASSERT_EQ(1, curr_tx.version);
|
|
ASSERT_EQ(rct::RCTTypeNull, curr_tx.rct_signatures.type);
|
|
ASSERT_EQ(typeid(cryptonote::txout_to_key), curr_tx.vout.at(0).target.type());
|
|
ASSERT_EQ(amount_b, curr_tx.vout.at(0).amount);
|
|
bc.add_block(1, {std::move(curr_tx)}, mock::null_addr);
|
|
}
|
|
|
|
// b. scan pre-ringct tx
|
|
balance_diff = wallet_process_new_blocks();
|
|
EXPECT_EQ(amount_b, balance_diff);
|
|
|
|
// c. construct and push a ringct coinbase tx
|
|
bc.add_block(HF_VERSION_DYNAMIC_FEE, {}, main_addr);
|
|
{
|
|
auto top_block = bc.get_parsed_block(bc.height() - 1);
|
|
const cryptonote::transaction &top_miner_tx = top_block.block.miner_tx;
|
|
ASSERT_EQ(2, top_miner_tx.version);
|
|
ASSERT_NE(0, top_miner_tx.vout.size());
|
|
ASSERT_EQ(rct::RCTTypeNull, top_miner_tx.rct_signatures.type);
|
|
ASSERT_EQ(0, top_miner_tx.signatures.size());
|
|
ASSERT_EQ(mock::fake_pruned_blockchain::miner_reward, top_miner_tx.vout.at(0).amount);
|
|
}
|
|
|
|
// c. scan ringct coinbase tx
|
|
balance_diff = wallet_process_new_blocks();
|
|
EXPECT_EQ(mock::fake_pruned_blockchain::miner_reward, balance_diff);
|
|
EXPECT_TRUE(verify_sals_of_recent_transfers());
|
|
|
|
// d. construct and push a ringct long-amount tx
|
|
const rct::xmr_amount amount_d = rct::randXmrAmount(COIN);
|
|
{
|
|
const rct::xmr_amount fee = rct::randXmrAmount(COIN);
|
|
std::vector<cryptonote::tx_destination_entry> dests = {
|
|
cryptonote::tx_destination_entry(amount_d, acc_keys.m_account_address, false)};
|
|
cryptonote::transaction curr_tx = mock::construct_pre_carrot_tx_with_fake_inputs(
|
|
acc_keys,
|
|
w.m_subaddresses,
|
|
/*stripped_sources=*/{},
|
|
dests,
|
|
acc_keys.m_account_address,
|
|
fee,
|
|
HF_VERSION_DYNAMIC_FEE);
|
|
ASSERT_FALSE(cryptonote::is_coinbase(curr_tx));
|
|
ASSERT_EQ(2, curr_tx.version);
|
|
ASSERT_EQ(rct::RCTTypeFull, curr_tx.rct_signatures.type);
|
|
ASSERT_EQ(typeid(cryptonote::txout_to_key), curr_tx.vout.at(0).target.type());
|
|
ASSERT_EQ(0, curr_tx.vout.at(0).amount);
|
|
bc.add_block(HF_VERSION_DYNAMIC_FEE, {std::move(curr_tx)}, mock::null_addr);
|
|
}
|
|
|
|
// d. scan ringct long-amount tx
|
|
balance_diff = wallet_process_new_blocks();
|
|
EXPECT_EQ(amount_d, balance_diff);
|
|
|
|
// e. construct and push a ringct short-amount tx
|
|
const rct::xmr_amount amount_e = rct::randXmrAmount(COIN);
|
|
{
|
|
const rct::xmr_amount fee = rct::randXmrAmount(COIN);
|
|
std::vector<cryptonote::tx_destination_entry> dests = {
|
|
cryptonote::tx_destination_entry(amount_e, acc_keys.m_account_address, false)};
|
|
cryptonote::transaction curr_tx = mock::construct_pre_carrot_tx_with_fake_inputs(
|
|
acc_keys,
|
|
w.m_subaddresses,
|
|
/*stripped_sources=*/{},
|
|
dests,
|
|
acc_keys.m_account_address,
|
|
fee,
|
|
HF_VERSION_SMALLER_BP);
|
|
ASSERT_FALSE(cryptonote::is_coinbase(curr_tx));
|
|
ASSERT_EQ(2, curr_tx.version);
|
|
ASSERT_EQ(rct::RCTTypeBulletproof2, curr_tx.rct_signatures.type);
|
|
ASSERT_EQ(typeid(cryptonote::txout_to_key), curr_tx.vout.at(0).target.type());
|
|
ASSERT_EQ(0, curr_tx.vout.at(0).amount);
|
|
bc.add_block(HF_VERSION_SMALLER_BP, {std::move(curr_tx)}, mock::null_addr);
|
|
}
|
|
|
|
// e. scan ringct short-amount tx
|
|
balance_diff = wallet_process_new_blocks();
|
|
EXPECT_EQ(amount_e, balance_diff);
|
|
|
|
// f. construct and push a view-tagged ringct coinbase tx
|
|
bc.add_block(HF_VERSION_VIEW_TAGS, {}, main_addr);
|
|
{
|
|
auto top_block = bc.get_parsed_block(bc.height() - 1);
|
|
const cryptonote::transaction &top_miner_tx = top_block.block.miner_tx;
|
|
ASSERT_EQ(2, top_miner_tx.version);
|
|
ASSERT_EQ(1, top_miner_tx.vout.size());
|
|
ASSERT_EQ(rct::RCTTypeNull, top_miner_tx.rct_signatures.type);
|
|
ASSERT_EQ(0, top_miner_tx.signatures.size());
|
|
ASSERT_EQ(typeid(cryptonote::txout_to_tagged_key), top_miner_tx.vout.at(0).target.type());
|
|
ASSERT_EQ(mock::fake_pruned_blockchain::miner_reward, top_miner_tx.vout.at(0).amount);
|
|
}
|
|
|
|
// f. scan view-tagged ringct coinbase tx
|
|
balance_diff = wallet_process_new_blocks();
|
|
EXPECT_EQ(mock::fake_pruned_blockchain::miner_reward, balance_diff);
|
|
EXPECT_TRUE(verify_sals_of_recent_transfers());
|
|
|
|
// g. construct and push a view-tagged pre-ringct (only possible in unmixable sweep txs) tx
|
|
const rct::xmr_amount amount_g = rct::randXmrAmount(COIN);
|
|
{
|
|
const rct::xmr_amount fee = rct::randXmrAmount(COIN);
|
|
std::vector<cryptonote::tx_destination_entry> dests = {
|
|
cryptonote::tx_destination_entry(amount_g, acc_keys.m_account_address, false)};
|
|
cryptonote::transaction curr_tx = mock::construct_pre_carrot_tx_with_fake_inputs(
|
|
acc_keys,
|
|
w.m_subaddresses,
|
|
/*stripped_sources=*/{},
|
|
dests,
|
|
acc_keys.m_account_address,
|
|
fee,
|
|
HF_VERSION_VIEW_TAGS,
|
|
/*sweep_unmixable_override=*/true);
|
|
ASSERT_FALSE(cryptonote::is_coinbase(curr_tx));
|
|
ASSERT_EQ(1, curr_tx.version);
|
|
ASSERT_EQ(rct::RCTTypeNull, curr_tx.rct_signatures.type);
|
|
ASSERT_EQ(typeid(cryptonote::txout_to_tagged_key), curr_tx.vout.at(0).target.type());
|
|
ASSERT_EQ(amount_g, curr_tx.vout.at(0).amount);
|
|
bc.add_block(HF_VERSION_VIEW_TAGS, {std::move(curr_tx)}, mock::null_addr);
|
|
}
|
|
|
|
// g. scan view-tagged pre-ringct (only possible in unmixable sweep txs) tx
|
|
balance_diff = wallet_process_new_blocks();
|
|
EXPECT_EQ(amount_g, balance_diff);
|
|
|
|
// h. construct and push a view-tagged ringct tx
|
|
const rct::xmr_amount amount_h = rct::randXmrAmount(COIN);
|
|
{
|
|
const rct::xmr_amount fee = rct::randXmrAmount(COIN);
|
|
std::vector<cryptonote::tx_destination_entry> dests = {
|
|
cryptonote::tx_destination_entry(amount_h, acc_keys.m_account_address, false)};
|
|
cryptonote::transaction curr_tx = mock::construct_pre_carrot_tx_with_fake_inputs(
|
|
acc_keys,
|
|
w.m_subaddresses,
|
|
/*stripped_sources=*/{},
|
|
dests,
|
|
acc_keys.m_account_address,
|
|
fee,
|
|
HF_VERSION_VIEW_TAGS);
|
|
ASSERT_FALSE(cryptonote::is_coinbase(curr_tx));
|
|
ASSERT_EQ(2, curr_tx.version);
|
|
ASSERT_EQ(rct::RCTTypeBulletproofPlus, curr_tx.rct_signatures.type);
|
|
ASSERT_EQ(typeid(cryptonote::txout_to_tagged_key), curr_tx.vout.at(0).target.type());
|
|
ASSERT_EQ(0, curr_tx.vout.at(0).amount);
|
|
bc.add_block(HF_VERSION_VIEW_TAGS, {std::move(curr_tx)}, mock::null_addr);
|
|
}
|
|
|
|
// h. scan ringct view-tagged ringct tx
|
|
balance_diff = wallet_process_new_blocks();
|
|
EXPECT_EQ(amount_h, balance_diff);
|
|
|
|
// i. construct and push a carrot v1 coinbase tx
|
|
bc.add_block(HF_VERSION_CARROT, {}, main_addr);
|
|
{
|
|
auto top_block = bc.get_parsed_block(bc.height() - 1);
|
|
const cryptonote::transaction &top_miner_tx = top_block.block.miner_tx;
|
|
ASSERT_EQ(2, top_miner_tx.version);
|
|
ASSERT_EQ(1, top_miner_tx.vout.size());
|
|
ASSERT_EQ(rct::RCTTypeNull, top_miner_tx.rct_signatures.type);
|
|
ASSERT_EQ(0, top_miner_tx.signatures.size());
|
|
ASSERT_EQ(typeid(cryptonote::txout_to_carrot_v1), top_miner_tx.vout.at(0).target.type());
|
|
ASSERT_EQ(mock::fake_pruned_blockchain::miner_reward, top_miner_tx.vout.at(0).amount);
|
|
}
|
|
|
|
// i. scan carrot v1 coinbase tx
|
|
balance_diff = wallet_process_new_blocks();
|
|
EXPECT_EQ(mock::fake_pruned_blockchain::miner_reward, balance_diff);
|
|
EXPECT_TRUE(verify_sals_of_recent_transfers());
|
|
|
|
// j. construct and push a carrot v1 normal tx
|
|
const rct::xmr_amount amount_j = rct::randXmrAmount(COIN);
|
|
{
|
|
std::vector<cryptonote::tx_destination_entry> dests = {
|
|
cryptonote::tx_destination_entry(amount_j, acc_keys.m_account_address, false)};
|
|
cryptonote::transaction curr_tx = mock::construct_carrot_pruned_transaction_fake_inputs(
|
|
{carrot::mock::convert_normal_payment_proposal_v1(dests.front())},
|
|
/*selfsend_payment_proposals=*/{},
|
|
acc_keys);
|
|
ASSERT_FALSE(cryptonote::is_coinbase(curr_tx));
|
|
ASSERT_EQ(2, curr_tx.version);
|
|
ASSERT_EQ(rct::RCTTypeFcmpPlusPlus, curr_tx.rct_signatures.type);
|
|
ASSERT_EQ(typeid(cryptonote::txout_to_carrot_v1), curr_tx.vout.at(0).target.type());
|
|
ASSERT_EQ(0, curr_tx.vout.at(0).amount);
|
|
bc.add_block(HF_VERSION_CARROT, {std::move(curr_tx)}, mock::null_addr);
|
|
}
|
|
|
|
// j. scan carrot v1 normal tx
|
|
balance_diff = wallet_process_new_blocks();
|
|
EXPECT_EQ(amount_j, balance_diff);
|
|
|
|
// k. construct and push a carrot v1 special tx
|
|
const rct::xmr_amount amount_k = rct::randXmrAmount(COIN);
|
|
{
|
|
std::vector<cryptonote::tx_destination_entry> dests = {
|
|
cryptonote::tx_destination_entry(amount_k, acc_keys.m_account_address, false)};
|
|
cryptonote::transaction curr_tx = mock::construct_carrot_pruned_transaction_fake_inputs(
|
|
/*normal_payment_proposals=*/{},
|
|
{{carrot::mock::convert_selfsend_payment_proposal_v1(dests.front()), {/*main*/}}},
|
|
acc_keys);
|
|
ASSERT_FALSE(cryptonote::is_coinbase(curr_tx));
|
|
ASSERT_EQ(2, curr_tx.version);
|
|
ASSERT_EQ(rct::RCTTypeFcmpPlusPlus, curr_tx.rct_signatures.type);
|
|
ASSERT_EQ(2, curr_tx.vout.size());
|
|
ASSERT_EQ(typeid(cryptonote::txout_to_carrot_v1), curr_tx.vout.at(0).target.type());
|
|
ASSERT_EQ(0, curr_tx.vout.at(0).amount);
|
|
bc.add_block(HF_VERSION_CARROT, {std::move(curr_tx)}, mock::null_addr);
|
|
}
|
|
|
|
// k. scan carrot v1 special tx
|
|
balance_diff = wallet_process_new_blocks();
|
|
EXPECT_EQ(amount_k, balance_diff);
|
|
}
|
|
//----------------------------------------------------------------------------------------------------------------------
|