1
0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-12-05 14:08:39 +00:00
trezor-firmware/core/tests/test_apps.bitcoin.segwit.signtx.native_p2wpkh.py

503 lines
19 KiB
Python
Raw Normal View History

from common import * # isort:skip
from trezor import wire
from trezor.crypto import bip39
from trezor.enums import AmountUnit, InputScriptType, OutputScriptType
from trezor.enums.RequestType import TXFINISHED, TXINPUT, TXMETA, TXOUTPUT
from trezor.messages import (
PrevInput,
PrevOutput,
PrevTx,
SignTx,
TxAckInput,
TxAckInputWrapper,
TxAckOutput,
TxAckOutputWrapper,
TxAckPrevInput,
TxAckPrevInputWrapper,
TxAckPrevMeta,
TxAckPrevOutput,
TxAckPrevOutputWrapper,
TxInput,
TxOutput,
TxRequest,
TxRequestDetailsType,
TxRequestSerializedType,
)
from trezor.utils import chunks
from apps.bitcoin.keychain import _get_schemas_for_coin
from apps.bitcoin.sign_tx import bitcoin, helpers
from apps.common import coins
2020-07-23 14:20:54 +00:00
from apps.common.keychain import Keychain
EMPTY_SERIALIZED = TxRequestSerializedType(serialized_tx=bytearray())
class TestSignSegwitTxNativeP2WPKH(unittest.TestCase):
# pylint: disable=C0301
def test_send_native_p2wpkh(self):
2023-06-28 10:46:29 +00:00
coin = coins.by_name("Testnet")
seed = bip39.seed(" ".join(["all"] * 12), "")
inp1 = TxInput(
# 49'/1'/0'/0/0" - tb1qqzv60m9ajw8drqulta4ld4gfx0rdh82un5s65s
address_n=[49 | 0x80000000, 1 | 0x80000000, 0 | 0x80000000, 0, 0],
amount=12300000,
2023-06-28 10:46:29 +00:00
prev_hash=unhexlify(
"09144602765ce3dd8f4329445b20e3684e948709c5cdcaf12da3bb079c99448a"
),
prev_index=0,
script_type=InputScriptType.SPENDWITNESS,
2023-06-28 10:46:29 +00:00
sequence=0xFFFFFFFF,
multisig=None,
)
2023-06-28 10:46:29 +00:00
ptx1 = PrevTx(
version=1, lock_time=0, inputs_count=1, outputs_count=2, extra_data_len=0
)
pinp1 = PrevInput(
script_sig=unhexlify("160014d16b8c0680c61fc6ed2e407455715055e41052f5"),
prev_hash=unhexlify(
"20912f98ea3ed849042efed0fdac8cb4fc301961c5988cba56902d8ffb61c337"
),
prev_index=0,
sequence=4294967295,
)
pout1 = PrevOutput(
script_pubkey=unhexlify("00140099a7ecbd938ed1839f5f6bf6d50933c6db9d5c"),
amount=12300000,
)
pout2 = PrevOutput(
script_pubkey=unhexlify("a91458b53ea7f832e8f096e896b8713a8c6df0e892ca87"),
amount=111145789,
)
out1 = TxOutput(
2023-06-28 10:46:29 +00:00
address="2N4Q5FhU2497BryFfUgbqkAJE87aKHUhXMp",
amount=5000000,
script_type=OutputScriptType.PAYTOADDRESS,
address_n=[],
2018-02-12 11:40:18 +00:00
multisig=None,
)
out2 = TxOutput(
2023-06-28 10:46:29 +00:00
address="tb1q694ccp5qcc0udmfwgp692u2s2hjpq5h407urtu",
script_type=OutputScriptType.PAYTOADDRESS,
amount=12300000 - 11000 - 5000000,
address_n=[],
2018-02-12 11:40:18 +00:00
multisig=None,
)
2023-06-28 10:46:29 +00:00
tx = SignTx(
coin_name="Testnet", version=1, lock_time=0, inputs_count=1, outputs_count=2
)
# precomputed tx weight is 142 = ceil(566 / 4)
fee_rate = 11000 / 142
messages = [
None,
# check fee
2023-06-28 10:46:29 +00:00
TxRequest(
request_type=TXINPUT,
details=TxRequestDetailsType(request_index=0, tx_hash=None),
serialized=EMPTY_SERIALIZED,
),
TxAckInput(tx=TxAckInputWrapper(input=inp1)),
helpers.UiConfirmForeignAddress(address_n=inp1.address_n),
True,
2023-06-28 10:46:29 +00:00
TxRequest(
request_type=TXOUTPUT,
details=TxRequestDetailsType(request_index=0, tx_hash=None),
serialized=EMPTY_SERIALIZED,
),
TxAckOutput(tx=TxAckOutputWrapper(output=out1)),
helpers.UiConfirmOutput(out1, coin, AmountUnit.BITCOIN, 0, False, [H_(49), H_(1), H_(0)]),
True,
2023-06-28 10:46:29 +00:00
TxRequest(
request_type=TXOUTPUT,
details=TxRequestDetailsType(request_index=1, tx_hash=None),
serialized=EMPTY_SERIALIZED,
),
TxAckOutput(tx=TxAckOutputWrapper(output=out2)),
helpers.UiConfirmOutput(out2, coin, AmountUnit.BITCOIN, 1, False, [H_(49), H_(1), H_(0)]),
True,
2023-06-28 10:46:29 +00:00
helpers.UiConfirmTotal(
12300000, 11000, fee_rate, coin, AmountUnit.BITCOIN, inp1.address_n[:3]
),
True,
# check prev tx
2023-06-28 10:46:29 +00:00
TxRequest(
request_type=TXINPUT,
details=TxRequestDetailsType(request_index=0, tx_hash=None),
serialized=EMPTY_SERIALIZED,
),
TxAckInput(tx=TxAckInputWrapper(input=inp1)),
2023-06-28 10:46:29 +00:00
TxRequest(
request_type=TXMETA,
details=TxRequestDetailsType(
request_index=None, tx_hash=inp1.prev_hash
),
serialized=EMPTY_SERIALIZED,
),
TxAckPrevMeta(tx=ptx1),
2023-06-28 10:46:29 +00:00
TxRequest(
request_type=TXINPUT,
details=TxRequestDetailsType(request_index=0, tx_hash=inp1.prev_hash),
serialized=EMPTY_SERIALIZED,
),
TxAckPrevInput(tx=TxAckPrevInputWrapper(input=pinp1)),
2023-06-28 10:46:29 +00:00
TxRequest(
request_type=TXOUTPUT,
details=TxRequestDetailsType(request_index=0, tx_hash=inp1.prev_hash),
serialized=EMPTY_SERIALIZED,
),
TxAckPrevOutput(tx=TxAckPrevOutputWrapper(output=pout1)),
2023-06-28 10:46:29 +00:00
TxRequest(
request_type=TXOUTPUT,
details=TxRequestDetailsType(request_index=1, tx_hash=inp1.prev_hash),
serialized=EMPTY_SERIALIZED,
),
TxAckPrevOutput(tx=TxAckPrevOutputWrapper(output=pout2)),
# sign tx
2023-06-28 10:46:29 +00:00
TxRequest(
request_type=TXINPUT,
details=TxRequestDetailsType(request_index=0, tx_hash=None),
serialized=TxRequestSerializedType(
# returned serialized header
serialized_tx=unhexlify("01000000000101"),
),
),
TxAckInput(tx=TxAckInputWrapper(input=inp1)),
2023-06-28 10:46:29 +00:00
TxRequest(
request_type=TXOUTPUT,
details=TxRequestDetailsType(request_index=0, tx_hash=None),
serialized=TxRequestSerializedType(
# returned serialized inp1
serialized_tx=unhexlify(
"8a44999c07bba32df1cacdc50987944e68e3205b4429438fdde35c76024614090000000000ffffffff02"
),
),
),
TxAckOutput(tx=TxAckOutputWrapper(output=out1)),
2023-06-28 10:46:29 +00:00
TxRequest(
request_type=TXOUTPUT,
details=TxRequestDetailsType(request_index=1, tx_hash=None),
serialized=TxRequestSerializedType(
# returned serialized out1
serialized_tx=unhexlify(
"404b4c000000000017a9147a55d61848e77ca266e79a39bfc85c580a6426c987"
),
signature_index=None,
signature=None,
),
),
TxAckOutput(tx=TxAckOutputWrapper(output=out2)),
# segwit
2023-06-28 10:46:29 +00:00
TxRequest(
request_type=TXINPUT,
details=TxRequestDetailsType(request_index=0, tx_hash=None),
serialized=TxRequestSerializedType(
# returned serialized out2
serialized_tx=unhexlify(
"a8386f0000000000160014d16b8c0680c61fc6ed2e407455715055e41052f5"
),
signature_index=None,
signature=None,
),
),
TxAckInput(tx=TxAckInputWrapper(input=inp1)),
2023-06-28 10:46:29 +00:00
TxRequest(
request_type=TXFINISHED,
details=TxRequestDetailsType(),
serialized=TxRequestSerializedType(
serialized_tx=unhexlify(
"02483045022100a7ca8f097525f9044e64376dc0a0f5d4aeb8d15d66808ba97979a0475b06b66502200597c8ebcef63e047f9aeef1a8001d3560470cf896c12f6990eec4faec599b950121033add1f0e8e3c3136f7428dd4a4de1057380bd311f5b0856e2269170b4ffa65bf00000000"
),
signature_index=0,
signature=unhexlify(
"3045022100a7ca8f097525f9044e64376dc0a0f5d4aeb8d15d66808ba97979a0475b06b66502200597c8ebcef63e047f9aeef1a8001d3560470cf896c12f6990eec4faec599b95"
),
),
),
]
ns = _get_schemas_for_coin(coin)
2020-07-23 14:20:54 +00:00
keychain = Keychain(seed, coin.curve_name, ns)
signer = bitcoin.Bitcoin(tx, keychain, coin, None).signer()
for request, response in chunks(messages, 2):
res = signer.send(request)
if isinstance(res, tuple):
_, res = res
self.assertEqual(res, response)
with self.assertRaises(StopIteration):
signer.send(None)
def test_send_native_p2wpkh_change(self):
2023-06-28 10:46:29 +00:00
coin = coins.by_name("Testnet")
seed = bip39.seed(" ".join(["all"] * 12), "")
inp1 = TxInput(
# 49'/1'/0'/0/0" - tb1qqzv60m9ajw8drqulta4ld4gfx0rdh82un5s65s
address_n=[49 | 0x80000000, 1 | 0x80000000, 0 | 0x80000000, 0, 0],
amount=12300000,
2023-06-28 10:46:29 +00:00
prev_hash=unhexlify(
"09144602765ce3dd8f4329445b20e3684e948709c5cdcaf12da3bb079c99448a"
),
prev_index=0,
script_type=InputScriptType.SPENDWITNESS,
2023-06-28 10:46:29 +00:00
sequence=0xFFFFFFFF,
multisig=None,
)
2023-06-28 10:46:29 +00:00
ptx1 = PrevTx(
version=1, lock_time=0, inputs_count=1, outputs_count=2, extra_data_len=0
)
pinp1 = PrevInput(
script_sig=unhexlify("160014d16b8c0680c61fc6ed2e407455715055e41052f5"),
prev_hash=unhexlify(
"20912f98ea3ed849042efed0fdac8cb4fc301961c5988cba56902d8ffb61c337"
),
prev_index=0,
sequence=4294967295,
)
pout1 = PrevOutput(
script_pubkey=unhexlify("00140099a7ecbd938ed1839f5f6bf6d50933c6db9d5c"),
amount=12300000,
)
pout2 = PrevOutput(
script_pubkey=unhexlify("a91458b53ea7f832e8f096e896b8713a8c6df0e892ca87"),
amount=111145789,
)
out1 = TxOutput(
2023-06-28 10:46:29 +00:00
address="2N4Q5FhU2497BryFfUgbqkAJE87aKHUhXMp",
amount=5000000,
script_type=OutputScriptType.PAYTOADDRESS,
address_n=[],
2018-02-12 11:40:18 +00:00
multisig=None,
)
out2 = TxOutput(
address=None,
# 49'/1'/0'/1/0" - 2N1LGaGg836mqSQqiuUBLfcyGBhyZbremDX
address_n=[49 | 0x80000000, 1 | 0x80000000, 0 | 0x80000000, 1, 0],
script_type=OutputScriptType.PAYTOWITNESS,
amount=12300000 - 11000 - 5000000,
2018-02-12 11:40:18 +00:00
multisig=None,
)
2023-06-28 10:46:29 +00:00
tx = SignTx(
coin_name="Testnet", version=1, lock_time=0, inputs_count=1, outputs_count=2
)
# precomputed tx weight is 142 = ceil(566 / 4)
fee_rate = 11000 / 142
messages = [
None,
# check fee
2023-06-28 10:46:29 +00:00
TxRequest(
request_type=TXINPUT,
details=TxRequestDetailsType(request_index=0, tx_hash=None),
serialized=EMPTY_SERIALIZED,
),
TxAckInput(tx=TxAckInputWrapper(input=inp1)),
helpers.UiConfirmForeignAddress(address_n=inp1.address_n),
True,
2023-06-28 10:46:29 +00:00
TxRequest(
request_type=TXOUTPUT,
details=TxRequestDetailsType(request_index=0, tx_hash=None),
serialized=EMPTY_SERIALIZED,
),
TxAckOutput(tx=TxAckOutputWrapper(output=out1)),
helpers.UiConfirmOutput(out1, coin, AmountUnit.BITCOIN, 0, False, [H_(49), H_(1), H_(0)]),
True,
2023-06-28 10:46:29 +00:00
TxRequest(
request_type=TXOUTPUT,
details=TxRequestDetailsType(request_index=1, tx_hash=None),
serialized=EMPTY_SERIALIZED,
),
TxAckOutput(tx=TxAckOutputWrapper(output=out2)),
helpers.UiConfirmForeignAddress(address_n=out2.address_n),
True,
2023-06-28 10:46:29 +00:00
helpers.UiConfirmTotal(
5000000 + 11000,
11000,
fee_rate,
coin,
AmountUnit.BITCOIN,
inp1.address_n[:3],
),
True,
# check prev tx
2023-06-28 10:46:29 +00:00
TxRequest(
request_type=TXINPUT,
details=TxRequestDetailsType(request_index=0, tx_hash=None),
serialized=EMPTY_SERIALIZED,
),
TxAckInput(tx=TxAckInputWrapper(input=inp1)),
2023-06-28 10:46:29 +00:00
TxRequest(
request_type=TXMETA,
details=TxRequestDetailsType(
request_index=None, tx_hash=inp1.prev_hash
),
serialized=EMPTY_SERIALIZED,
),
TxAckPrevMeta(tx=ptx1),
2023-06-28 10:46:29 +00:00
TxRequest(
request_type=TXINPUT,
details=TxRequestDetailsType(request_index=0, tx_hash=inp1.prev_hash),
serialized=EMPTY_SERIALIZED,
),
TxAckPrevInput(tx=TxAckPrevInputWrapper(input=pinp1)),
2023-06-28 10:46:29 +00:00
TxRequest(
request_type=TXOUTPUT,
details=TxRequestDetailsType(request_index=0, tx_hash=inp1.prev_hash),
serialized=EMPTY_SERIALIZED,
),
TxAckPrevOutput(tx=TxAckPrevOutputWrapper(output=pout1)),
2023-06-28 10:46:29 +00:00
TxRequest(
request_type=TXOUTPUT,
details=TxRequestDetailsType(request_index=1, tx_hash=inp1.prev_hash),
serialized=EMPTY_SERIALIZED,
),
TxAckPrevOutput(tx=TxAckPrevOutputWrapper(output=pout2)),
# sign tx
2023-06-28 10:46:29 +00:00
TxRequest(
request_type=TXINPUT,
details=TxRequestDetailsType(request_index=0, tx_hash=None),
serialized=TxRequestSerializedType(
# returned serialized header
serialized_tx=unhexlify("01000000000101"),
),
),
TxAckInput(tx=TxAckInputWrapper(input=inp1)),
2023-06-28 10:46:29 +00:00
TxRequest(
request_type=TXOUTPUT,
details=TxRequestDetailsType(request_index=0, tx_hash=None),
serialized=TxRequestSerializedType(
# returned serialized inp1
serialized_tx=unhexlify(
"8a44999c07bba32df1cacdc50987944e68e3205b4429438fdde35c76024614090000000000ffffffff02"
),
),
),
2017-11-20 16:41:23 +00:00
# the out has to be cloned not to send the same object which was modified
TxAckOutput(tx=TxAckOutputWrapper(output=TxOutput(**out1.__dict__))),
2023-06-28 10:46:29 +00:00
TxRequest(
request_type=TXOUTPUT,
details=TxRequestDetailsType(request_index=1, tx_hash=None),
serialized=TxRequestSerializedType(
# returned serialized out1
serialized_tx=unhexlify(
"404b4c000000000017a9147a55d61848e77ca266e79a39bfc85c580a6426c987"
),
signature_index=None,
signature=None,
),
),
TxAckOutput(tx=TxAckOutputWrapper(output=TxOutput(**out2.__dict__))),
# segwit
2023-06-28 10:46:29 +00:00
TxRequest(
request_type=TXINPUT,
details=TxRequestDetailsType(request_index=0, tx_hash=None),
serialized=TxRequestSerializedType(
# returned serialized out2
serialized_tx=unhexlify(
"a8386f0000000000160014d16b8c0680c61fc6ed2e407455715055e41052f5"
),
signature_index=None,
signature=None,
),
),
TxAckInput(tx=TxAckInputWrapper(input=inp1)),
2023-06-28 10:46:29 +00:00
TxRequest(
request_type=TXFINISHED,
details=TxRequestDetailsType(),
serialized=TxRequestSerializedType(
serialized_tx=unhexlify(
"02483045022100a7ca8f097525f9044e64376dc0a0f5d4aeb8d15d66808ba97979a0475b06b66502200597c8ebcef63e047f9aeef1a8001d3560470cf896c12f6990eec4faec599b950121033add1f0e8e3c3136f7428dd4a4de1057380bd311f5b0856e2269170b4ffa65bf00000000"
),
signature_index=0,
signature=unhexlify(
"3045022100a7ca8f097525f9044e64376dc0a0f5d4aeb8d15d66808ba97979a0475b06b66502200597c8ebcef63e047f9aeef1a8001d3560470cf896c12f6990eec4faec599b95"
),
),
),
]
ns = _get_schemas_for_coin(coin)
2020-07-23 14:20:54 +00:00
keychain = Keychain(seed, coin.curve_name, ns)
signer = bitcoin.Bitcoin(tx, keychain, coin, None).signer()
for request, expected_response in chunks(messages, 2):
response = signer.send(request)
if isinstance(response, tuple):
_, response = response
self.assertEqual(response, expected_response)
with self.assertRaises(StopIteration):
signer.send(None)
def test_send_native_invalid_address(self):
2023-06-28 10:46:29 +00:00
coin = coins.by_name("Testnet")
seed = bip39.seed(" ".join(["all"] * 12), "")
inp1 = TxInput(
# 49'/1'/0'/0/0" - tb1qqzv60m9ajw8drqulta4ld4gfx0rdh82un5s65s
address_n=[49 | 0x80000000, 1 | 0x80000000, 0 | 0x80000000, 0, 0],
amount=12300000,
2023-06-28 10:46:29 +00:00
prev_hash=unhexlify(
"09144602765ce3dd8f4329445b20e3684e948709c5cdcaf12da3bb079c99448a"
),
prev_index=0,
script_type=InputScriptType.SPENDWITNESS,
2023-06-28 10:46:29 +00:00
sequence=0xFFFFFFFF,
multisig=None,
)
out1 = TxOutput(
2023-06-28 10:46:29 +00:00
address="TB1Q694CCP5QCC0UDMFWGP692U2S2HJPQ5H407URTU", # Error: should be lower case
script_type=OutputScriptType.PAYTOADDRESS,
amount=12300000 - 11000 - 5000000,
address_n=[],
multisig=None,
)
2023-06-28 10:46:29 +00:00
tx = SignTx(
coin_name="Testnet", version=1, lock_time=0, inputs_count=1, outputs_count=1
)
messages = [
None,
# check fee
2023-06-28 10:46:29 +00:00
TxRequest(
request_type=TXINPUT,
details=TxRequestDetailsType(request_index=0, tx_hash=None),
serialized=EMPTY_SERIALIZED,
),
TxAckInput(tx=TxAckInputWrapper(input=inp1)),
helpers.UiConfirmForeignAddress(address_n=inp1.address_n),
True,
2023-06-28 10:46:29 +00:00
TxRequest(
request_type=TXOUTPUT,
details=TxRequestDetailsType(request_index=0, tx_hash=None),
serialized=EMPTY_SERIALIZED,
),
TxAckOutput(tx=TxAckOutputWrapper(output=out1)),
2023-06-28 10:46:29 +00:00
None,
]
ns = _get_schemas_for_coin(coin)
2020-07-23 14:20:54 +00:00
keychain = Keychain(seed, coin.curve_name, ns)
signer = bitcoin.Bitcoin(tx, keychain, coin, None).signer()
for request, expected_response in chunks(messages, 2):
if expected_response is None:
with self.assertRaises(wire.DataError):
signer.send(request)
else:
response = signer.send(request)
if isinstance(response, tuple):
_, response = response
self.assertEqual(response, expected_response)
2023-06-28 10:46:29 +00:00
if __name__ == "__main__":
unittest.main()