wallet2: fixes for export/import output flow

- only allow offline wallets to import outputs
- don't import empty outputs
- export subaddress indexes when exporting outputs
This commit is contained in:
j-berman 2022-08-15 19:50:34 -07:00 committed by moneromooo-monero
parent 692f1d4e3d
commit 1e912ecd8a
No known key found for this signature in database
GPG key ID: 686F07454D6CEFC3
4 changed files with 75 additions and 20 deletions

View file

@ -6606,7 +6606,7 @@ bool wallet2::sign_tx(unsigned_tx_set &exported_txs, std::vector<wallet2::pendin
{ {
if (!exported_txs.new_transfers.second.empty()) if (!exported_txs.new_transfers.second.empty())
import_outputs(exported_txs.new_transfers); import_outputs(exported_txs.new_transfers);
else if (exported_txs.transfers.second.empty()) else if (!exported_txs.transfers.second.empty())
import_outputs(exported_txs.transfers); import_outputs(exported_txs.transfers);
// sign the transactions // sign the transactions
@ -13132,6 +13132,8 @@ std::pair<uint64_t, std::vector<tools::wallet2::exported_transfer_details>> wall
etd.m_flags.m_key_image_partial = td.m_key_image_partial; etd.m_flags.m_key_image_partial = td.m_key_image_partial;
etd.m_amount = td.m_amount; etd.m_amount = td.m_amount;
etd.m_additional_tx_keys = get_additional_tx_pub_keys_from_extra(td.m_tx); etd.m_additional_tx_keys = get_additional_tx_pub_keys_from_extra(td.m_tx);
etd.m_subaddr_index_major = td.m_subaddr_index.major;
etd.m_subaddr_index_minor = td.m_subaddr_index.minor;
outs.push_back(etd); outs.push_back(etd);
} }
@ -13162,7 +13164,7 @@ size_t wallet2::import_outputs(const std::pair<uint64_t, std::vector<tools::wall
{ {
PERF_TIMER(import_outputs); PERF_TIMER(import_outputs);
THROW_WALLET_EXCEPTION_IF(watch_only(), error::wallet_internal_error, "Hot wallets cannot import outputs"); THROW_WALLET_EXCEPTION_IF(!m_offline, error::wallet_internal_error, "Hot wallets cannot import outputs");
THROW_WALLET_EXCEPTION_IF(outputs.first > m_transfers.size(), error::wallet_internal_error, THROW_WALLET_EXCEPTION_IF(outputs.first > m_transfers.size(), error::wallet_internal_error,
"Imported outputs omit more outputs that we know of"); "Imported outputs omit more outputs that we know of");
@ -13230,6 +13232,8 @@ size_t wallet2::import_outputs(const std::pair<uint64_t, std::vector<tools::wall
{ {
PERF_TIMER(import_outputs); PERF_TIMER(import_outputs);
THROW_WALLET_EXCEPTION_IF(!m_offline, error::wallet_internal_error, "Hot wallets cannot import outputs");
THROW_WALLET_EXCEPTION_IF(outputs.first > m_transfers.size(), error::wallet_internal_error, THROW_WALLET_EXCEPTION_IF(outputs.first > m_transfers.size(), error::wallet_internal_error,
"Imported outputs omit more outputs that we know of. Try using export_outputs all."); "Imported outputs omit more outputs that we know of. Try using export_outputs all.");

View file

@ -406,6 +406,8 @@ private:
BEGIN_SERIALIZE_OBJECT() BEGIN_SERIALIZE_OBJECT()
VERSION_FIELD(1) VERSION_FIELD(1)
if (version < 1)
return false;
FIELD(m_pubkey) FIELD(m_pubkey)
VARINT_FIELD(m_internal_output_index) VARINT_FIELD(m_internal_output_index)
VARINT_FIELD(m_global_output_index) VARINT_FIELD(m_global_output_index)

View file

@ -35,12 +35,18 @@ from __future__ import print_function
from framework.daemon import Daemon from framework.daemon import Daemon
from framework.wallet import Wallet from framework.wallet import Wallet
SEED = 'velvet lymph giddy number token physics poetry unquoted nibs useful sabotage limits benches lifestyle eden nitrogen anvil fewest avoid batch vials washing fences goat unquoted'
STANDARD_ADDRESS = '42ey1afDFnn4886T7196doS9GPMzexD9gXpsZJDwVjeRVdFCSoHnv7KPbBeGpzJBzHRCAs9UxqeoyFQMYbqSWYTfJJQAWDm'
SUBADDRESS = '84QRUYawRNrU3NN1VpFRndSukeyEb3Xpv8qZjjsoJZnTYpDYceuUTpog13D7qPxpviS7J29bSgSkR11hFFoXWk2yNdsR9WF'
class ColdSigningTest(): class ColdSigningTest():
def run_test(self): def run_test(self):
self.reset() self.reset()
self.create(0) self.create(0)
self.mine() self.mine()
self.transfer() self.transfer()
self.self_transfer_to_subaddress()
self.transfer_after_empty_export_import()
def reset(self): def reset(self):
print('Resetting blockchain') print('Resetting blockchain')
@ -57,17 +63,15 @@ class ColdSigningTest():
try: self.hot_wallet.close_wallet() try: self.hot_wallet.close_wallet()
except: pass except: pass
self.cold_wallet = Wallet(idx = 1) self.cold_wallet = Wallet(idx = 5)
# close the wallet if any, will throw if none is loaded # close the wallet if any, will throw if none is loaded
try: self.cold_wallet.close_wallet() try: self.cold_wallet.close_wallet()
except: pass except: pass
seed = 'velvet lymph giddy number token physics poetry unquoted nibs useful sabotage limits benches lifestyle eden nitrogen anvil fewest avoid batch vials washing fences goat unquoted' res = self.cold_wallet.restore_deterministic_wallet(seed = SEED)
res = self.cold_wallet.restore_deterministic_wallet(seed = seed)
self.cold_wallet.set_daemon('127.0.0.1:11111', ssl_support = "disabled")
spend_key = self.cold_wallet.query_key("spend_key").key spend_key = self.cold_wallet.query_key("spend_key").key
view_key = self.cold_wallet.query_key("view_key").key view_key = self.cold_wallet.query_key("view_key").key
res = self.hot_wallet.generate_from_keys(viewkey = view_key, address = '42ey1afDFnn4886T7196doS9GPMzexD9gXpsZJDwVjeRVdFCSoHnv7KPbBeGpzJBzHRCAs9UxqeoyFQMYbqSWYTfJJQAWDm') res = self.hot_wallet.generate_from_keys(viewkey = view_key, address = STANDARD_ADDRESS)
ok = False ok = False
try: res = self.hot_wallet.query_key("spend_key") try: res = self.hot_wallet.query_key("spend_key")
@ -79,28 +83,30 @@ class ColdSigningTest():
assert ok assert ok
assert self.cold_wallet.query_key("view_key").key == view_key assert self.cold_wallet.query_key("view_key").key == view_key
assert self.cold_wallet.get_address().address == self.hot_wallet.get_address().address assert self.cold_wallet.get_address().address == self.hot_wallet.get_address().address
assert self.cold_wallet.get_address().address == STANDARD_ADDRESS
def mine(self): def mine(self):
print("Mining some blocks") print("Mining some blocks")
daemon = Daemon() daemon = Daemon()
wallet = Wallet() wallet = Wallet()
daemon.generateblocks('42ey1afDFnn4886T7196doS9GPMzexD9gXpsZJDwVjeRVdFCSoHnv7KPbBeGpzJBzHRCAs9UxqeoyFQMYbqSWYTfJJQAWDm', 80) daemon.generateblocks(STANDARD_ADDRESS, 80)
wallet.refresh() wallet.refresh()
def transfer(self): def export_import(self):
daemon = Daemon()
print("Creating transaction in hot wallet")
dst = {'address': '42ey1afDFnn4886T7196doS9GPMzexD9gXpsZJDwVjeRVdFCSoHnv7KPbBeGpzJBzHRCAs9UxqeoyFQMYbqSWYTfJJQAWDm', 'amount': 1000000000000}
self.hot_wallet.refresh() self.hot_wallet.refresh()
res = self.hot_wallet.export_outputs() res = self.hot_wallet.export_outputs()
self.cold_wallet.import_outputs(res.outputs_data_hex) self.cold_wallet.import_outputs(res.outputs_data_hex)
res = self.cold_wallet.export_key_images(True) res = self.cold_wallet.export_key_images(True)
self.hot_wallet.import_key_images(res.signed_key_images, offset = res.offset) self.hot_wallet.import_key_images(res.signed_key_images, offset = res.offset)
def create_tx(self, destination_addr):
daemon = Daemon()
dst = {'address': destination_addr, 'amount': 1000000000000}
self.export_import()
res = self.hot_wallet.transfer([dst], ring_size = 16, get_tx_key = False) res = self.hot_wallet.transfer([dst], ring_size = 16, get_tx_key = False)
assert len(res.tx_hash) == 32*2 assert len(res.tx_hash) == 32*2
txid = res.tx_hash txid = res.tx_hash
@ -125,11 +131,11 @@ class ColdSigningTest():
assert desc.unlock_time == 0 assert desc.unlock_time == 0
assert desc.payment_id in ['', '0000000000000000'] assert desc.payment_id in ['', '0000000000000000']
assert desc.change_amount == desc.amount_in - 1000000000000 - fee assert desc.change_amount == desc.amount_in - 1000000000000 - fee
assert desc.change_address == '42ey1afDFnn4886T7196doS9GPMzexD9gXpsZJDwVjeRVdFCSoHnv7KPbBeGpzJBzHRCAs9UxqeoyFQMYbqSWYTfJJQAWDm' assert desc.change_address == STANDARD_ADDRESS
assert desc.fee == fee assert desc.fee == fee
assert len(desc.recipients) == 1 assert len(desc.recipients) == 1
rec = desc.recipients[0] rec = desc.recipients[0]
assert rec.address == '42ey1afDFnn4886T7196doS9GPMzexD9gXpsZJDwVjeRVdFCSoHnv7KPbBeGpzJBzHRCAs9UxqeoyFQMYbqSWYTfJJQAWDm' assert rec.address == destination_addr
assert rec.amount == 1000000000000 assert rec.amount == 1000000000000
res = self.cold_wallet.sign_transfer(unsigned_txset) res = self.cold_wallet.sign_transfer(unsigned_txset)
@ -148,7 +154,7 @@ class ColdSigningTest():
assert len([x for x in (res['pending'] if 'pending' in res else []) if x.txid == txid]) == 1 assert len([x for x in (res['pending'] if 'pending' in res else []) if x.txid == txid]) == 1
assert len([x for x in (res['out'] if 'out' in res else []) if x.txid == txid]) == 0 assert len([x for x in (res['out'] if 'out' in res else []) if x.txid == txid]) == 0
daemon.generateblocks('42ey1afDFnn4886T7196doS9GPMzexD9gXpsZJDwVjeRVdFCSoHnv7KPbBeGpzJBzHRCAs9UxqeoyFQMYbqSWYTfJJQAWDm', 1) daemon.generateblocks(STANDARD_ADDRESS, 1)
self.hot_wallet.refresh() self.hot_wallet.refresh()
res = self.hot_wallet.get_transfers() res = self.hot_wallet.get_transfers()
@ -160,6 +166,47 @@ class ColdSigningTest():
res = self.cold_wallet.get_tx_key(txid) res = self.cold_wallet.get_tx_key(txid)
assert len(res.tx_key) == 64 assert len(res.tx_key) == 64
self.export_import()
def transfer(self):
print("Creating transaction in hot wallet")
self.create_tx(STANDARD_ADDRESS)
res = self.cold_wallet.get_address()
assert len(res['addresses']) == 1
assert res['addresses'][0].address == STANDARD_ADDRESS
assert res['addresses'][0].used
res = self.hot_wallet.get_address()
assert len(res['addresses']) == 1
assert res['addresses'][0].address == STANDARD_ADDRESS
assert res['addresses'][0].used
def self_transfer_to_subaddress(self):
print("Self-spending to subaddress in hot wallet")
self.create_tx(SUBADDRESS)
res = self.cold_wallet.get_address()
assert len(res['addresses']) == 2
assert res['addresses'][0].address == STANDARD_ADDRESS
assert res['addresses'][0].used
assert res['addresses'][1].address == SUBADDRESS
assert res['addresses'][1].used
res = self.hot_wallet.get_address()
assert len(res['addresses']) == 2
assert res['addresses'][0].address == STANDARD_ADDRESS
assert res['addresses'][0].used
assert res['addresses'][1].address == SUBADDRESS
assert res['addresses'][1].used
def transfer_after_empty_export_import(self):
print("Creating transaction in hot wallet after empty export & import")
start_len = len(self.hot_wallet.get_transfers()['in'])
self.export_import()
assert start_len == len(self.hot_wallet.get_transfers()['in'])
self.create_tx(STANDARD_ADDRESS)
assert start_len == len(self.hot_wallet.get_transfers()['in']) - 1
class Guard: class Guard:
def __enter__(self): def __enter__(self):

View file

@ -40,8 +40,9 @@ except:
N_MONERODS = 4 N_MONERODS = 4
# 4 wallets connected to the main offline monerod # 4 wallets connected to the main offline monerod
# a wallet connected to the first local online monerod # 1 wallet connected to the first local online monerod
N_WALLETS = 5 # 1 offline wallet
N_WALLETS = 6
WALLET_DIRECTORY = builddir + "/functional-tests-directory" WALLET_DIRECTORY = builddir + "/functional-tests-directory"
FUNCTIONAL_TESTS_DIRECTORY = builddir + "/tests/functional_tests" FUNCTIONAL_TESTS_DIRECTORY = builddir + "/tests/functional_tests"
@ -61,6 +62,7 @@ wallet_extra = [
["--daemon-port", "18180"], ["--daemon-port", "18180"],
["--daemon-port", "18180"], ["--daemon-port", "18180"],
["--daemon-port", "18182"], ["--daemon-port", "18182"],
["--offline"],
] ]
command_lines = [] command_lines = []