mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-12-23 23:08:14 +00:00
test_msg_signtx: proto -> messages
This commit is contained in:
Martin Milata
Tomas Susanka
parent
c9dc38c9f3
commit
75fa9dee77
@ -16,7 +16,7 @@
|
||||
|
||||
import pytest
|
||||
|
||||
from trezorlib import btc, messages as proto
|
||||
from trezorlib import btc, messages
|
||||
from trezorlib.exceptions import TrezorFailure
|
||||
from trezorlib.tools import H_, parse_path, tx_hash
|
||||
|
||||
@ -24,7 +24,7 @@ from ..common import MNEMONIC12
|
||||
from ..tx_cache import TxCache
|
||||
from .signtx import request_finished, request_input, request_meta, request_output
|
||||
|
||||
B = proto.ButtonRequestType
|
||||
B = messages.ButtonRequestType
|
||||
|
||||
TX_CACHE_TESTNET = TxCache("Testnet")
|
||||
TX_CACHE_MAINNET = TxCache("Bitcoin")
|
||||
@ -78,17 +78,17 @@ class TestMsgSigntx:
|
||||
# tx: d5f65ee80147b4bcc70b75e4bbf2d7382021b871bd8867ef8fa525ef50864882
|
||||
# input 0: 0.0039 BTC
|
||||
|
||||
inp1 = proto.TxInputType(
|
||||
inp1 = messages.TxInputType(
|
||||
address_n=parse_path("44h/0h/0h/0/0"),
|
||||
# amount=390000,
|
||||
prev_hash=TXHASH_d5f65e,
|
||||
prev_index=0,
|
||||
)
|
||||
|
||||
out1 = proto.TxOutputType(
|
||||
out1 = messages.TxOutputType(
|
||||
address="1MJ2tj2ThBE62zXbBYA5ZaN3fdve5CPAz1",
|
||||
amount=390000 - 10000,
|
||||
script_type=proto.OutputScriptType.PAYTOADDRESS,
|
||||
script_type=messages.OutputScriptType.PAYTOADDRESS,
|
||||
)
|
||||
|
||||
with client:
|
||||
@ -100,8 +100,8 @@ class TestMsgSigntx:
|
||||
request_input(1, TXHASH_d5f65e),
|
||||
request_output(0, TXHASH_d5f65e),
|
||||
request_output(0),
|
||||
proto.ButtonRequest(code=B.ConfirmOutput),
|
||||
proto.ButtonRequest(code=B.SignTx),
|
||||
messages.ButtonRequest(code=B.ConfirmOutput),
|
||||
messages.ButtonRequest(code=B.SignTx),
|
||||
request_input(0),
|
||||
request_output(0),
|
||||
request_output(0),
|
||||
@ -123,23 +123,23 @@ class TestMsgSigntx:
|
||||
|
||||
# tx: e5040e1bc1ae7667ffb9e5248e90b2fb93cd9150234151ce90e14ab2f5933bcd
|
||||
# input 0: 0.31 BTC
|
||||
inp1 = proto.TxInputType(
|
||||
inp1 = messages.TxInputType(
|
||||
address_n=parse_path("44'/1'/0'/0/0"),
|
||||
# amount=31000000,
|
||||
prev_hash=TXHASH_e5040e,
|
||||
prev_index=0,
|
||||
)
|
||||
|
||||
out1 = proto.TxOutputType(
|
||||
out1 = messages.TxOutputType(
|
||||
address="msj42CCGruhRsFrGATiUuh25dtxYtnpbTx",
|
||||
amount=30090000,
|
||||
script_type=proto.OutputScriptType.PAYTOADDRESS,
|
||||
script_type=messages.OutputScriptType.PAYTOADDRESS,
|
||||
)
|
||||
|
||||
out2 = proto.TxOutputType(
|
||||
out2 = messages.TxOutputType(
|
||||
address_n=parse_path("44'/1'/0'/1/0"),
|
||||
amount=900000,
|
||||
script_type=proto.OutputScriptType.PAYTOADDRESS,
|
||||
script_type=messages.OutputScriptType.PAYTOADDRESS,
|
||||
)
|
||||
|
||||
with client:
|
||||
@ -151,9 +151,9 @@ class TestMsgSigntx:
|
||||
request_output(0, TXHASH_e5040e),
|
||||
request_output(1, TXHASH_e5040e),
|
||||
request_output(0),
|
||||
proto.ButtonRequest(code=B.ConfirmOutput),
|
||||
messages.ButtonRequest(code=B.ConfirmOutput),
|
||||
request_output(1),
|
||||
proto.ButtonRequest(code=B.SignTx),
|
||||
messages.ButtonRequest(code=B.SignTx),
|
||||
request_input(0),
|
||||
request_output(0),
|
||||
request_output(1),
|
||||
@ -174,23 +174,23 @@ class TestMsgSigntx:
|
||||
def test_testnet_fee_too_high(self, client):
|
||||
# tx: 6f90f3c7cbec2258b0971056ef3fe34128dbde30daa9c0639a898f9977299d54
|
||||
# input 1: 10.00000000 BTC
|
||||
inp1 = proto.TxInputType(
|
||||
inp1 = messages.TxInputType(
|
||||
address_n=parse_path("44'/1'/0'/0/0"),
|
||||
# amount=1000000000,
|
||||
prev_hash=TXHASH_6f90f3,
|
||||
prev_index=1,
|
||||
)
|
||||
|
||||
out1 = proto.TxOutputType(
|
||||
out1 = messages.TxOutputType(
|
||||
address="mfiGQVPcRcaEvQPYDErR34DcCovtxYvUUV",
|
||||
amount=1000000000 - 500000000 - 100000000,
|
||||
script_type=proto.OutputScriptType.PAYTOADDRESS,
|
||||
script_type=messages.OutputScriptType.PAYTOADDRESS,
|
||||
)
|
||||
|
||||
out2 = proto.TxOutputType(
|
||||
out2 = messages.TxOutputType(
|
||||
address_n=parse_path("44'/1'/0'/1/0"),
|
||||
amount=500000000,
|
||||
script_type=proto.OutputScriptType.PAYTOADDRESS,
|
||||
script_type=messages.OutputScriptType.PAYTOADDRESS,
|
||||
)
|
||||
|
||||
with client:
|
||||
@ -203,10 +203,10 @@ class TestMsgSigntx:
|
||||
request_output(0, TXHASH_6f90f3),
|
||||
request_output(1, TXHASH_6f90f3),
|
||||
request_output(0),
|
||||
proto.ButtonRequest(code=B.ConfirmOutput),
|
||||
messages.ButtonRequest(code=B.ConfirmOutput),
|
||||
request_output(1),
|
||||
proto.ButtonRequest(code=B.FeeOverThreshold),
|
||||
proto.ButtonRequest(code=B.SignTx),
|
||||
messages.ButtonRequest(code=B.FeeOverThreshold),
|
||||
messages.ButtonRequest(code=B.SignTx),
|
||||
request_input(0),
|
||||
request_output(0),
|
||||
request_output(1),
|
||||
@ -227,7 +227,7 @@ class TestMsgSigntx:
|
||||
def test_one_two_fee(self, client):
|
||||
# tx: c275c333fd1b36bef4af316226c66a8b3693fbfcc081a5e16a2ae5fcb09e92bf
|
||||
|
||||
inp1 = proto.TxInputType(
|
||||
inp1 = messages.TxInputType(
|
||||
address_n=parse_path(
|
||||
"m/44'/0'/0'/0/5"
|
||||
), # 1GA9u9TfCG7SWmKCveBumdA1TZpfom6ZdJ
|
||||
@ -236,18 +236,18 @@ class TestMsgSigntx:
|
||||
prev_index=1,
|
||||
)
|
||||
|
||||
out1 = proto.TxOutputType(
|
||||
out1 = messages.TxOutputType(
|
||||
address_n=parse_path(
|
||||
"m/44'/0'/0'/1/3"
|
||||
), # 1EcL6AyfQTyWKGvXwNSfsWoYnD3whzVFdu
|
||||
amount=30000,
|
||||
script_type=proto.OutputScriptType.PAYTOADDRESS,
|
||||
script_type=messages.OutputScriptType.PAYTOADDRESS,
|
||||
)
|
||||
|
||||
out2 = proto.TxOutputType(
|
||||
out2 = messages.TxOutputType(
|
||||
address="1Up15Msx4sbvUCGm8Xgo2Zp5FQim3wE59",
|
||||
amount=10000,
|
||||
script_type=proto.OutputScriptType.PAYTOADDRESS,
|
||||
script_type=messages.OutputScriptType.PAYTOADDRESS,
|
||||
)
|
||||
|
||||
with client:
|
||||
@ -260,8 +260,8 @@ class TestMsgSigntx:
|
||||
request_output(1, TXHASH_50f6f1),
|
||||
request_output(0),
|
||||
request_output(1),
|
||||
proto.ButtonRequest(code=B.ConfirmOutput),
|
||||
proto.ButtonRequest(code=B.SignTx),
|
||||
messages.ButtonRequest(code=B.ConfirmOutput),
|
||||
messages.ButtonRequest(code=B.SignTx),
|
||||
request_input(0),
|
||||
request_output(0),
|
||||
request_output(1),
|
||||
@ -283,29 +283,29 @@ class TestMsgSigntx:
|
||||
# tx: d5f65ee80147b4bcc70b75e4bbf2d7382021b871bd8867ef8fa525ef50864882
|
||||
# input 0: 0.0039 BTC
|
||||
|
||||
inp1 = proto.TxInputType(
|
||||
inp1 = messages.TxInputType(
|
||||
address_n=parse_path("44'/0'/0'/0/0"),
|
||||
# amount=390000,
|
||||
prev_hash=TXHASH_d5f65e,
|
||||
prev_index=0,
|
||||
)
|
||||
|
||||
out1 = proto.TxOutputType(
|
||||
out1 = messages.TxOutputType(
|
||||
address="1MJ2tj2ThBE62zXbBYA5ZaN3fdve5CPAz1",
|
||||
amount=390000 - 80000 - 12000 - 10000,
|
||||
script_type=proto.OutputScriptType.PAYTOADDRESS,
|
||||
script_type=messages.OutputScriptType.PAYTOADDRESS,
|
||||
)
|
||||
|
||||
out2 = proto.TxOutputType(
|
||||
out2 = messages.TxOutputType(
|
||||
address="13uaUYn6XAooo88QvAqAVsiVvr2mAXutqP",
|
||||
amount=12000,
|
||||
script_type=proto.OutputScriptType.PAYTOADDRESS,
|
||||
script_type=messages.OutputScriptType.PAYTOADDRESS,
|
||||
)
|
||||
|
||||
out3 = proto.TxOutputType(
|
||||
out3 = messages.TxOutputType(
|
||||
address_n=parse_path("44'/0'/0'/1/0"),
|
||||
amount=80000,
|
||||
script_type=proto.OutputScriptType.PAYTOADDRESS,
|
||||
script_type=messages.OutputScriptType.PAYTOADDRESS,
|
||||
)
|
||||
|
||||
with client:
|
||||
@ -317,11 +317,11 @@ class TestMsgSigntx:
|
||||
request_input(1, TXHASH_d5f65e),
|
||||
request_output(0, TXHASH_d5f65e),
|
||||
request_output(0),
|
||||
proto.ButtonRequest(code=B.ConfirmOutput),
|
||||
messages.ButtonRequest(code=B.ConfirmOutput),
|
||||
request_output(1),
|
||||
proto.ButtonRequest(code=B.ConfirmOutput),
|
||||
messages.ButtonRequest(code=B.ConfirmOutput),
|
||||
request_output(2),
|
||||
proto.ButtonRequest(code=B.SignTx),
|
||||
messages.ButtonRequest(code=B.SignTx),
|
||||
request_input(0),
|
||||
request_output(0),
|
||||
request_output(1),
|
||||
@ -351,30 +351,30 @@ class TestMsgSigntx:
|
||||
# tx: 58497a7757224d1ff1941488d23087071103e5bf855f4c1c44e5c8d9d82ca46e
|
||||
# input 1: 0.0011 BTC
|
||||
|
||||
inp1 = proto.TxInputType(
|
||||
inp1 = messages.TxInputType(
|
||||
address_n=parse_path("44h/0h/0h/0/0"),
|
||||
# amount=100000,
|
||||
prev_hash=TXHASH_c6be22,
|
||||
prev_index=1,
|
||||
)
|
||||
|
||||
inp2 = proto.TxInputType(
|
||||
inp2 = messages.TxInputType(
|
||||
address_n=parse_path("44h/0h/0h/0/1"),
|
||||
# amount=110000,
|
||||
prev_hash=TXHASH_58497a,
|
||||
prev_index=1,
|
||||
)
|
||||
|
||||
out1 = proto.TxOutputType(
|
||||
out1 = messages.TxOutputType(
|
||||
address="15Jvu3nZNP7u2ipw2533Q9VVgEu2Lu9F2B",
|
||||
amount=210000 - 100000 - 10000,
|
||||
script_type=proto.OutputScriptType.PAYTOADDRESS,
|
||||
script_type=messages.OutputScriptType.PAYTOADDRESS,
|
||||
)
|
||||
|
||||
out2 = proto.TxOutputType(
|
||||
out2 = messages.TxOutputType(
|
||||
address_n=parse_path("44h/0h/0h/1/0"),
|
||||
amount=100000,
|
||||
script_type=proto.OutputScriptType.PAYTOADDRESS,
|
||||
script_type=messages.OutputScriptType.PAYTOADDRESS,
|
||||
)
|
||||
|
||||
with client:
|
||||
@ -391,9 +391,9 @@ class TestMsgSigntx:
|
||||
request_output(0, TXHASH_58497a),
|
||||
request_output(1, TXHASH_58497a),
|
||||
request_output(0),
|
||||
proto.ButtonRequest(code=B.ConfirmOutput),
|
||||
messages.ButtonRequest(code=B.ConfirmOutput),
|
||||
request_output(1),
|
||||
proto.ButtonRequest(code=B.SignTx),
|
||||
messages.ButtonRequest(code=B.SignTx),
|
||||
request_input(0),
|
||||
request_input(1),
|
||||
request_output(0),
|
||||
@ -431,16 +431,16 @@ class TestMsgSigntx:
|
||||
inputs = []
|
||||
for i in range(100):
|
||||
inputs.append(
|
||||
proto.TxInputType(
|
||||
messages.TxInputType(
|
||||
address_n=parse_path(f"44h/0h/0h/0/{i}"),
|
||||
prev_hash=TXHASH_4a7b7e,
|
||||
prev_index=i,
|
||||
)
|
||||
)
|
||||
out = proto.TxOutputType(
|
||||
out = messages.TxOutputType(
|
||||
address="19dvDdyxxptP9dGvozYe8BP6tgFV9L4jg5",
|
||||
amount=100 * 26000 - 15 * 10000,
|
||||
script_type=proto.OutputScriptType.PAYTOADDRESS,
|
||||
script_type=messages.OutputScriptType.PAYTOADDRESS,
|
||||
)
|
||||
_, serialized_tx = btc.sign_tx(
|
||||
client, "Bitcoin", inputs, [out], prev_txes=TX_CACHE_MAINNET
|
||||
@ -460,14 +460,14 @@ class TestMsgSigntx:
|
||||
# tx: 39a29e954977662ab3879c66fb251ef753e0912223a83d1dcb009111d28265e5
|
||||
# index 1: 0.0254 BTC
|
||||
|
||||
inp1 = proto.TxInputType(
|
||||
inp1 = messages.TxInputType(
|
||||
address_n=parse_path("44h/0h/1h/0/0"),
|
||||
# amount=100000,
|
||||
prev_hash=TXHASH_c63e24,
|
||||
prev_index=1,
|
||||
)
|
||||
|
||||
inp2 = proto.TxInputType(
|
||||
inp2 = messages.TxInputType(
|
||||
address_n=parse_path("44h/0h/1h/0/1"),
|
||||
# amount=2540000,
|
||||
prev_hash=TXHASH_39a29e,
|
||||
@ -477,10 +477,10 @@ class TestMsgSigntx:
|
||||
outputs = []
|
||||
cnt = 255
|
||||
for _ in range(cnt):
|
||||
out = proto.TxOutputType(
|
||||
out = messages.TxOutputType(
|
||||
address="1NwN6UduuVkJi6sw3gSiKZaCY5rHgVXC2h",
|
||||
amount=(100000 + 2540000 - 39000) // cnt,
|
||||
script_type=proto.OutputScriptType.PAYTOADDRESS,
|
||||
script_type=messages.OutputScriptType.PAYTOADDRESS,
|
||||
)
|
||||
outputs.append(out)
|
||||
|
||||
@ -502,14 +502,14 @@ class TestMsgSigntx:
|
||||
# tx: 39a29e954977662ab3879c66fb251ef753e0912223a83d1dcb009111d28265e5
|
||||
# index 1: 0.0254 BTC
|
||||
|
||||
inp1 = proto.TxInputType(
|
||||
inp1 = messages.TxInputType(
|
||||
address_n=parse_path("44h/0h/1h/0/0"),
|
||||
# amount=100000,
|
||||
prev_hash=TXHASH_c63e24,
|
||||
prev_index=1,
|
||||
)
|
||||
|
||||
inp2 = proto.TxInputType(
|
||||
inp2 = messages.TxInputType(
|
||||
address_n=parse_path("44h/0h/1h/0/1"),
|
||||
# amount=2540000,
|
||||
prev_hash=TXHASH_39a29e,
|
||||
@ -517,19 +517,19 @@ class TestMsgSigntx:
|
||||
)
|
||||
|
||||
outputs = [
|
||||
proto.TxOutputType(
|
||||
messages.TxOutputType(
|
||||
address="1NwN6UduuVkJi6sw3gSiKZaCY5rHgVXC2h",
|
||||
amount=500000,
|
||||
script_type=proto.OutputScriptType.PAYTOADDRESS,
|
||||
script_type=messages.OutputScriptType.PAYTOADDRESS,
|
||||
)
|
||||
]
|
||||
|
||||
cnt = 20
|
||||
for i in range(cnt):
|
||||
out = proto.TxOutputType(
|
||||
out = messages.TxOutputType(
|
||||
address_n=parse_path(f"44h/0h/1h/1/{i}"),
|
||||
amount=(100000 + 2540000 - 500000 - 39000) // cnt,
|
||||
script_type=proto.OutputScriptType.PAYTOADDRESS,
|
||||
script_type=messages.OutputScriptType.PAYTOADDRESS,
|
||||
)
|
||||
outputs.append(out)
|
||||
|
||||
@ -550,12 +550,12 @@ class TestMsgSigntx:
|
||||
request_output(0, TXHASH_39a29e),
|
||||
request_output(1, TXHASH_39a29e),
|
||||
request_output(0),
|
||||
proto.ButtonRequest(code=B.ConfirmOutput),
|
||||
messages.ButtonRequest(code=B.ConfirmOutput),
|
||||
]
|
||||
+ request_change_outputs
|
||||
+ [
|
||||
proto.ButtonRequest(code=B.SignTx),
|
||||
proto.ButtonRequest(code=B.SignTx),
|
||||
messages.ButtonRequest(code=B.SignTx),
|
||||
messages.ButtonRequest(code=B.SignTx),
|
||||
request_input(0),
|
||||
request_input(1),
|
||||
request_output(0),
|
||||
@ -581,17 +581,17 @@ class TestMsgSigntx:
|
||||
# tx: 1570416eb4302cf52979afd5e6909e37d8fdd874301f7cc87e547e509cb1caa6
|
||||
# input 0: 1.0 BTC
|
||||
|
||||
inp1 = proto.TxInputType(
|
||||
inp1 = messages.TxInputType(
|
||||
address_n=parse_path("44h/0h/0h/0/0"),
|
||||
# amount=100000000,
|
||||
prev_hash=TXHASH_157041,
|
||||
prev_index=0,
|
||||
)
|
||||
|
||||
out1 = proto.TxOutputType(
|
||||
out1 = messages.TxOutputType(
|
||||
address="1MJ2tj2ThBE62zXbBYA5ZaN3fdve5CPAz1",
|
||||
amount=100000000 - 510000,
|
||||
script_type=proto.OutputScriptType.PAYTOADDRESS,
|
||||
script_type=messages.OutputScriptType.PAYTOADDRESS,
|
||||
)
|
||||
|
||||
with client:
|
||||
@ -603,9 +603,9 @@ class TestMsgSigntx:
|
||||
request_output(0, TXHASH_157041),
|
||||
request_output(1, TXHASH_157041),
|
||||
request_output(0),
|
||||
proto.ButtonRequest(code=B.ConfirmOutput),
|
||||
proto.ButtonRequest(code=B.FeeOverThreshold),
|
||||
proto.ButtonRequest(code=B.SignTx),
|
||||
messages.ButtonRequest(code=B.ConfirmOutput),
|
||||
messages.ButtonRequest(code=B.FeeOverThreshold),
|
||||
messages.ButtonRequest(code=B.SignTx),
|
||||
request_input(0),
|
||||
request_output(0),
|
||||
request_output(0),
|
||||
@ -625,17 +625,17 @@ class TestMsgSigntx:
|
||||
# tx: d5f65ee80147b4bcc70b75e4bbf2d7382021b871bd8867ef8fa525ef50864882
|
||||
# input 0: 0.0039 BTC
|
||||
|
||||
inp1 = proto.TxInputType(
|
||||
inp1 = messages.TxInputType(
|
||||
address_n=parse_path("44h/0h/0h/0/0"),
|
||||
# amount=390000,
|
||||
prev_hash=TXHASH_d5f65e,
|
||||
prev_index=0,
|
||||
)
|
||||
|
||||
out1 = proto.TxOutputType(
|
||||
out1 = messages.TxOutputType(
|
||||
address="1MJ2tj2ThBE62zXbBYA5ZaN3fdve5CPAz1",
|
||||
amount=400000,
|
||||
script_type=proto.OutputScriptType.PAYTOADDRESS,
|
||||
script_type=messages.OutputScriptType.PAYTOADDRESS,
|
||||
)
|
||||
|
||||
with client:
|
||||
@ -647,8 +647,8 @@ class TestMsgSigntx:
|
||||
request_input(1, TXHASH_d5f65e),
|
||||
request_output(0, TXHASH_d5f65e),
|
||||
request_output(0),
|
||||
proto.ButtonRequest(code=B.ConfirmOutput),
|
||||
proto.Failure(code=proto.FailureType.NotEnoughFunds),
|
||||
messages.ButtonRequest(code=B.ConfirmOutput),
|
||||
messages.Failure(code=messages.FailureType.NotEnoughFunds),
|
||||
]
|
||||
)
|
||||
with pytest.raises(TrezorFailure, match="NotEnoughFunds"):
|
||||
@ -657,17 +657,17 @@ class TestMsgSigntx:
|
||||
)
|
||||
|
||||
def test_p2sh(self, client):
|
||||
inp1 = proto.TxInputType(
|
||||
inp1 = messages.TxInputType(
|
||||
address_n=parse_path("44h/0h/0h/0/0"),
|
||||
# amount=400000,
|
||||
prev_hash=TXHASH_54aa56,
|
||||
prev_index=1,
|
||||
)
|
||||
|
||||
out1 = proto.TxOutputType(
|
||||
out1 = messages.TxOutputType(
|
||||
address="3DKGE1pvPpBAgZj94MbCinwmksewUNNYVR", # p2sh
|
||||
amount=400000 - 10000,
|
||||
script_type=proto.OutputScriptType.PAYTOSCRIPTHASH,
|
||||
script_type=messages.OutputScriptType.PAYTOSCRIPTHASH,
|
||||
)
|
||||
|
||||
with client:
|
||||
@ -679,8 +679,8 @@ class TestMsgSigntx:
|
||||
request_output(0, TXHASH_54aa56),
|
||||
request_output(1, TXHASH_54aa56),
|
||||
request_output(0),
|
||||
proto.ButtonRequest(code=B.ConfirmOutput),
|
||||
proto.ButtonRequest(code=B.SignTx),
|
||||
messages.ButtonRequest(code=B.ConfirmOutput),
|
||||
messages.ButtonRequest(code=B.SignTx),
|
||||
request_input(0),
|
||||
request_output(0),
|
||||
request_output(0),
|
||||
@ -702,16 +702,16 @@ class TestMsgSigntx:
|
||||
# tx: 2bac7ad1dec654579a71ea9555463f63ac7b7df9d8ba67b4682bba4e514d0f0c:1
|
||||
# input 1: 411102528330 Satoshi
|
||||
|
||||
inp1 = proto.TxInputType(
|
||||
inp1 = messages.TxInputType(
|
||||
address_n=parse_path("m/44'/1'/0'/0/0"),
|
||||
amount=411102528330,
|
||||
prev_hash=TXHASH_2bac7a,
|
||||
prev_index=1,
|
||||
)
|
||||
out1 = proto.TxOutputType(
|
||||
out1 = messages.TxOutputType(
|
||||
address="mopZWqZZyQc3F2Sy33cvDtJchSAMsnLi7b", # seed allallall, bip32: m/44'/1'/0'/0/1
|
||||
amount=411102528330,
|
||||
script_type=proto.OutputScriptType.PAYTOADDRESS,
|
||||
script_type=messages.OutputScriptType.PAYTOADDRESS,
|
||||
)
|
||||
_, serialized_tx = btc.sign_tx(
|
||||
client, "Testnet", [inp1], [out1], prev_txes=TX_CACHE_TESTNET
|
||||
@ -723,30 +723,30 @@ class TestMsgSigntx:
|
||||
|
||||
@pytest.mark.setup_client(mnemonic=MNEMONIC12)
|
||||
def test_attack_change_outputs(self, client):
|
||||
inp1 = proto.TxInputType(
|
||||
inp1 = messages.TxInputType(
|
||||
address_n=parse_path("44h/0h/0h/0/0"),
|
||||
# amount=100000,
|
||||
prev_hash=TXHASH_c6be22,
|
||||
prev_index=1,
|
||||
)
|
||||
|
||||
inp2 = proto.TxInputType(
|
||||
inp2 = messages.TxInputType(
|
||||
address_n=parse_path("44h/0h/0h/0/1"),
|
||||
# amount=110000,
|
||||
prev_hash=TXHASH_58497a,
|
||||
prev_index=1,
|
||||
)
|
||||
|
||||
out1 = proto.TxOutputType(
|
||||
out1 = messages.TxOutputType(
|
||||
address="15Jvu3nZNP7u2ipw2533Q9VVgEu2Lu9F2B",
|
||||
amount=210000 - 100000 - 10000,
|
||||
script_type=proto.OutputScriptType.PAYTOADDRESS,
|
||||
script_type=messages.OutputScriptType.PAYTOADDRESS,
|
||||
)
|
||||
|
||||
out2 = proto.TxOutputType(
|
||||
out2 = messages.TxOutputType(
|
||||
address_n=parse_path("44h/0h/0h/1/0"),
|
||||
amount=100000,
|
||||
script_type=proto.OutputScriptType.PAYTOADDRESS,
|
||||
script_type=messages.OutputScriptType.PAYTOADDRESS,
|
||||
)
|
||||
|
||||
# Test if the transaction can be signed normally
|
||||
@ -773,7 +773,7 @@ class TestMsgSigntx:
|
||||
return msg
|
||||
|
||||
# Set up attack processors
|
||||
client.set_filter(proto.TxAck, attack_processor)
|
||||
client.set_filter(messages.TxAck, attack_processor)
|
||||
|
||||
with pytest.raises(
|
||||
TrezorFailure, match="Transaction has changed during signing"
|
||||
@ -793,23 +793,23 @@ class TestMsgSigntx:
|
||||
|
||||
# tx: e5040e1bc1ae7667ffb9e5248e90b2fb93cd9150234151ce90e14ab2f5933bcd
|
||||
# input 0: 0.31 BTC
|
||||
inp1 = proto.TxInputType(
|
||||
inp1 = messages.TxInputType(
|
||||
address_n=parse_path("44'/1'/0'/0/0"),
|
||||
# amount=31000000,
|
||||
prev_hash=TXHASH_e5040e,
|
||||
prev_index=0,
|
||||
)
|
||||
|
||||
out1 = proto.TxOutputType(
|
||||
out1 = messages.TxOutputType(
|
||||
address="msj42CCGruhRsFrGATiUuh25dtxYtnpbTx",
|
||||
amount=30090000,
|
||||
script_type=proto.OutputScriptType.PAYTOADDRESS,
|
||||
script_type=messages.OutputScriptType.PAYTOADDRESS,
|
||||
)
|
||||
|
||||
out2 = proto.TxOutputType(
|
||||
out2 = messages.TxOutputType(
|
||||
address_n=parse_path("44'/1'/0'/1/0"),
|
||||
amount=900000,
|
||||
script_type=proto.OutputScriptType.PAYTOADDRESS,
|
||||
script_type=messages.OutputScriptType.PAYTOADDRESS,
|
||||
)
|
||||
|
||||
run_attack = False
|
||||
@ -826,7 +826,7 @@ class TestMsgSigntx:
|
||||
return msg
|
||||
|
||||
# Set up attack processors
|
||||
client.set_filter(proto.TxAck, attack_processor)
|
||||
client.set_filter(messages.TxAck, attack_processor)
|
||||
|
||||
with pytest.raises(
|
||||
TrezorFailure, match="Transaction has changed during signing"
|
||||
@ -836,24 +836,24 @@ class TestMsgSigntx:
|
||||
)
|
||||
|
||||
def test_attack_change_input_address(self, client):
|
||||
inp1 = proto.TxInputType(
|
||||
inp1 = messages.TxInputType(
|
||||
address_n=parse_path("44'/1'/4'/0/0"),
|
||||
# moUJnmge8SRXuediK7bW6t4YfrPqbE6hD7
|
||||
prev_hash=TXHASH_d2dcda,
|
||||
prev_index=1,
|
||||
script_type=proto.InputScriptType.SPENDADDRESS,
|
||||
script_type=messages.InputScriptType.SPENDADDRESS,
|
||||
)
|
||||
|
||||
out1 = proto.TxOutputType(
|
||||
out1 = messages.TxOutputType(
|
||||
address="mwue7mokpBRAsJtHqEMcRPanYBmsSmYKvY",
|
||||
amount=100000,
|
||||
script_type=proto.OutputScriptType.PAYTOADDRESS,
|
||||
script_type=messages.OutputScriptType.PAYTOADDRESS,
|
||||
)
|
||||
|
||||
out2 = proto.TxOutputType(
|
||||
out2 = messages.TxOutputType(
|
||||
address_n=parse_path("44'/1'/4'/1/0"),
|
||||
amount=123400000 - 5000 - 100000,
|
||||
script_type=proto.OutputScriptType.PAYTOADDRESS,
|
||||
script_type=messages.OutputScriptType.PAYTOADDRESS,
|
||||
)
|
||||
|
||||
# Test if the transaction can be signed normally
|
||||
@ -878,7 +878,7 @@ class TestMsgSigntx:
|
||||
|
||||
return msg
|
||||
|
||||
client.set_filter(proto.TxAck, attack_processor)
|
||||
client.set_filter(messages.TxAck, attack_processor)
|
||||
# Now run the attack, must trigger the exception
|
||||
with client:
|
||||
client.set_expected_responses(
|
||||
@ -889,11 +889,11 @@ class TestMsgSigntx:
|
||||
request_output(0, TXHASH_d2dcda),
|
||||
request_output(1, TXHASH_d2dcda),
|
||||
request_output(0),
|
||||
proto.ButtonRequest(code=B.ConfirmOutput),
|
||||
messages.ButtonRequest(code=B.ConfirmOutput),
|
||||
request_output(1),
|
||||
proto.ButtonRequest(code=B.SignTx),
|
||||
messages.ButtonRequest(code=B.SignTx),
|
||||
request_input(0),
|
||||
proto.Failure(code=proto.FailureType.ProcessError),
|
||||
messages.Failure(code=messages.FailureType.ProcessError),
|
||||
]
|
||||
)
|
||||
# Now run the attack, must trigger the exception
|
||||
@ -902,7 +902,7 @@ class TestMsgSigntx:
|
||||
client, "Testnet", [inp1], [out1, out2], prev_txes=TX_CACHE_TESTNET,
|
||||
)
|
||||
|
||||
assert exc.value.code == proto.FailureType.ProcessError
|
||||
assert exc.value.code == messages.FailureType.ProcessError
|
||||
if client.features.model == "1":
|
||||
assert exc.value.message.endswith("Failed to compile input")
|
||||
else:
|
||||
@ -911,17 +911,17 @@ class TestMsgSigntx:
|
||||
)
|
||||
|
||||
def test_spend_coinbase(self, client):
|
||||
inp1 = proto.TxInputType(
|
||||
inp1 = messages.TxInputType(
|
||||
address_n=parse_path("44h/1h/0h/0/0"),
|
||||
# amount=390000,
|
||||
prev_hash=TXHASH_d6da21,
|
||||
prev_index=0,
|
||||
)
|
||||
|
||||
out1 = proto.TxOutputType(
|
||||
out1 = messages.TxOutputType(
|
||||
address="mm6FM31rM5Vc3sw5D7kztiBg3jHUzyqF1g",
|
||||
amount=2500278230 - 10000,
|
||||
script_type=proto.OutputScriptType.PAYTOADDRESS,
|
||||
script_type=messages.OutputScriptType.PAYTOADDRESS,
|
||||
)
|
||||
|
||||
with client:
|
||||
@ -932,8 +932,8 @@ class TestMsgSigntx:
|
||||
request_input(0, TXHASH_d6da21),
|
||||
request_output(0, TXHASH_d6da21),
|
||||
request_output(0),
|
||||
proto.ButtonRequest(code=B.ConfirmOutput),
|
||||
proto.ButtonRequest(code=B.SignTx),
|
||||
messages.ButtonRequest(code=B.ConfirmOutput),
|
||||
messages.ButtonRequest(code=B.SignTx),
|
||||
request_input(0),
|
||||
request_output(0),
|
||||
request_output(0),
|
||||
@ -955,29 +955,29 @@ class TestMsgSigntx:
|
||||
|
||||
# tx: e5040e1bc1ae7667ffb9e5248e90b2fb93cd9150234151ce90e14ab2f5933bcd
|
||||
# input 0: 0.31 BTC
|
||||
inp1 = proto.TxInputType(
|
||||
inp1 = messages.TxInputType(
|
||||
address_n=parse_path("44'/1'/0'/0/0"),
|
||||
# amount=31000000,
|
||||
prev_hash=TXHASH_e5040e,
|
||||
prev_index=0,
|
||||
)
|
||||
|
||||
out1 = proto.TxOutputType(
|
||||
out1 = messages.TxOutputType(
|
||||
address="msj42CCGruhRsFrGATiUuh25dtxYtnpbTx",
|
||||
amount=30090000,
|
||||
script_type=proto.OutputScriptType.PAYTOADDRESS,
|
||||
script_type=messages.OutputScriptType.PAYTOADDRESS,
|
||||
)
|
||||
|
||||
out_change1 = proto.TxOutputType(
|
||||
out_change1 = messages.TxOutputType(
|
||||
address_n=parse_path("44'/1'/0'/1/0"),
|
||||
amount=900000,
|
||||
script_type=proto.OutputScriptType.PAYTOADDRESS,
|
||||
script_type=messages.OutputScriptType.PAYTOADDRESS,
|
||||
)
|
||||
|
||||
out_change2 = proto.TxOutputType(
|
||||
out_change2 = messages.TxOutputType(
|
||||
address_n=parse_path("44'/1'/0'/1/1"),
|
||||
amount=10000,
|
||||
script_type=proto.OutputScriptType.PAYTOADDRESS,
|
||||
script_type=messages.OutputScriptType.PAYTOADDRESS,
|
||||
)
|
||||
|
||||
with client:
|
||||
@ -989,10 +989,10 @@ class TestMsgSigntx:
|
||||
request_output(0, TXHASH_e5040e),
|
||||
request_output(1, TXHASH_e5040e),
|
||||
request_output(0),
|
||||
proto.ButtonRequest(code=B.ConfirmOutput),
|
||||
messages.ButtonRequest(code=B.ConfirmOutput),
|
||||
request_output(1),
|
||||
request_output(2),
|
||||
proto.ButtonRequest(code=B.SignTx),
|
||||
messages.ButtonRequest(code=B.SignTx),
|
||||
request_input(0),
|
||||
request_output(0),
|
||||
request_output(1),
|
||||
@ -1017,24 +1017,24 @@ class TestMsgSigntx:
|
||||
|
||||
# tx: e5040e1bc1ae7667ffb9e5248e90b2fb93cd9150234151ce90e14ab2f5933bcd
|
||||
# input 0: 0.31 BTC
|
||||
inp1 = proto.TxInputType(
|
||||
inp1 = messages.TxInputType(
|
||||
address_n=parse_path("44'/1'/0'/0/0"),
|
||||
# amount=31000000,
|
||||
prev_hash=TXHASH_e5040e,
|
||||
prev_index=0,
|
||||
)
|
||||
|
||||
out1 = proto.TxOutputType(
|
||||
out1 = messages.TxOutputType(
|
||||
address="msj42CCGruhRsFrGATiUuh25dtxYtnpbTx",
|
||||
amount=30090000,
|
||||
script_type=proto.OutputScriptType.PAYTOADDRESS,
|
||||
script_type=messages.OutputScriptType.PAYTOADDRESS,
|
||||
)
|
||||
|
||||
# change on main chain is allowed => treated as a change
|
||||
out_change = proto.TxOutputType(
|
||||
out_change = messages.TxOutputType(
|
||||
address_n=parse_path("44'/1'/0'/0/0"),
|
||||
amount=900000,
|
||||
script_type=proto.OutputScriptType.PAYTOADDRESS,
|
||||
script_type=messages.OutputScriptType.PAYTOADDRESS,
|
||||
)
|
||||
|
||||
with client:
|
||||
@ -1046,9 +1046,9 @@ class TestMsgSigntx:
|
||||
request_output(0, TXHASH_e5040e),
|
||||
request_output(1, TXHASH_e5040e),
|
||||
request_output(0),
|
||||
proto.ButtonRequest(code=B.ConfirmOutput),
|
||||
messages.ButtonRequest(code=B.ConfirmOutput),
|
||||
request_output(1),
|
||||
proto.ButtonRequest(code=B.SignTx),
|
||||
messages.ButtonRequest(code=B.SignTx),
|
||||
request_input(0),
|
||||
request_output(0),
|
||||
request_output(1),
|
||||
@ -1074,22 +1074,22 @@ class TestMsgSigntx:
|
||||
assert len(prev_tx.bin_outputs) == 2
|
||||
|
||||
# vout[0] and vout[1] exist
|
||||
inp0 = proto.TxInputType(
|
||||
inp0 = messages.TxInputType(
|
||||
address_n=parse_path("44h/0h/0h/0/0"), prev_hash=TXHASH_157041, prev_index=0
|
||||
)
|
||||
inp1 = proto.TxInputType(
|
||||
inp1 = messages.TxInputType(
|
||||
address_n=parse_path("44h/0h/0h/0/1"), prev_hash=TXHASH_157041, prev_index=1
|
||||
)
|
||||
# vout[2] does not exist
|
||||
inp2 = proto.TxInputType(
|
||||
inp2 = messages.TxInputType(
|
||||
address_n=parse_path("44h/0h/0h/1/0"), prev_hash=TXHASH_157041, prev_index=2
|
||||
)
|
||||
|
||||
# try to spend the sum of existing vouts
|
||||
out1 = proto.TxOutputType(
|
||||
out1 = messages.TxOutputType(
|
||||
address="1MJ2tj2ThBE62zXbBYA5ZaN3fdve5CPAz1",
|
||||
amount=220160000,
|
||||
script_type=proto.OutputScriptType.PAYTOADDRESS,
|
||||
script_type=messages.OutputScriptType.PAYTOADDRESS,
|
||||
)
|
||||
|
||||
with pytest.raises(
|
||||
@ -1115,13 +1115,13 @@ class TestMsgSigntx:
|
||||
)
|
||||
@pytest.mark.skip_ui
|
||||
def test_prevtx_forbidden_fields(self, client, field, value):
|
||||
inp0 = proto.TxInputType(
|
||||
inp0 = messages.TxInputType(
|
||||
address_n=parse_path("44h/0h/0h/0/0"), prev_hash=TXHASH_157041, prev_index=0
|
||||
)
|
||||
out1 = proto.TxOutputType(
|
||||
out1 = messages.TxOutputType(
|
||||
address="1MJ2tj2ThBE62zXbBYA5ZaN3fdve5CPAz1",
|
||||
amount=1000,
|
||||
script_type=proto.OutputScriptType.PAYTOADDRESS,
|
||||
script_type=messages.OutputScriptType.PAYTOADDRESS,
|
||||
)
|
||||
|
||||
prev_tx = TX_CACHE_MAINNET[TXHASH_157041]
|
||||
@ -1140,16 +1140,16 @@ class TestMsgSigntx:
|
||||
)
|
||||
@pytest.mark.skip_ui
|
||||
def test_signtx_forbidden_fields(self, client, field, value):
|
||||
inp0 = proto.TxInputType(
|
||||
inp0 = messages.TxInputType(
|
||||
address_n=parse_path("44h/0h/0h/0/0"), prev_hash=TXHASH_157041, prev_index=0
|
||||
)
|
||||
out1 = proto.TxOutputType(
|
||||
out1 = messages.TxOutputType(
|
||||
address="1MJ2tj2ThBE62zXbBYA5ZaN3fdve5CPAz1",
|
||||
amount=1000,
|
||||
script_type=proto.OutputScriptType.PAYTOADDRESS,
|
||||
script_type=messages.OutputScriptType.PAYTOADDRESS,
|
||||
)
|
||||
|
||||
details = proto.SignTx()
|
||||
details = messages.SignTx()
|
||||
setattr(details, field, value)
|
||||
name = field.replace("_", " ")
|
||||
with pytest.raises(
|
||||
@ -1161,7 +1161,7 @@ class TestMsgSigntx:
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
"script_type",
|
||||
(proto.InputScriptType.SPENDADDRESS, proto.InputScriptType.EXTERNAL),
|
||||
(messages.InputScriptType.SPENDADDRESS, messages.InputScriptType.EXTERNAL),
|
||||
)
|
||||
@pytest.mark.skip_ui
|
||||
def test_incorrect_input_script_type(self, client, script_type):
|
||||
@ -1170,11 +1170,11 @@ class TestMsgSigntx:
|
||||
"030e669acac1f280d1ddf441cd2ba5e97417bf2689e4bbec86df4f831bf9f7ffd0"
|
||||
)
|
||||
|
||||
multisig = proto.MultisigRedeemScriptType(
|
||||
multisig = messages.MultisigRedeemScriptType(
|
||||
m=1,
|
||||
nodes=[
|
||||
btc.get_public_node(client, address_n, coin_name="Testnet").node,
|
||||
proto.HDNodeType(
|
||||
messages.HDNodeType(
|
||||
depth=0,
|
||||
fingerprint=0,
|
||||
child_num=0,
|
||||
@ -1184,7 +1184,7 @@ class TestMsgSigntx:
|
||||
],
|
||||
address_n=[],
|
||||
)
|
||||
inp1 = proto.TxInputType(
|
||||
inp1 = messages.TxInputType(
|
||||
address_n=address_n,
|
||||
prev_index=1,
|
||||
sequence=0xFFFFFFFF,
|
||||
@ -1192,16 +1192,16 @@ class TestMsgSigntx:
|
||||
multisig=multisig,
|
||||
prev_hash=TXHASH_e5040e,
|
||||
)
|
||||
out1 = proto.TxOutputType(
|
||||
out1 = messages.TxOutputType(
|
||||
address_n=address_n,
|
||||
amount=1000000 - 50000 - 10000,
|
||||
script_type=proto.OutputScriptType.PAYTOMULTISIG,
|
||||
script_type=messages.OutputScriptType.PAYTOMULTISIG,
|
||||
multisig=multisig,
|
||||
)
|
||||
out2 = proto.TxOutputType(
|
||||
out2 = messages.TxOutputType(
|
||||
address="mtkyndbpgv1G7nwggwKDVagRpxEJrwwyh6",
|
||||
amount=50000,
|
||||
script_type=proto.OutputScriptType.PAYTOADDRESS,
|
||||
script_type=messages.OutputScriptType.PAYTOADDRESS,
|
||||
)
|
||||
|
||||
with pytest.raises(
|
||||
@ -1213,7 +1213,10 @@ class TestMsgSigntx:
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
"script_type",
|
||||
(proto.OutputScriptType.PAYTOADDRESS, proto.OutputScriptType.PAYTOSCRIPTHASH),
|
||||
(
|
||||
messages.OutputScriptType.PAYTOADDRESS,
|
||||
messages.OutputScriptType.PAYTOSCRIPTHASH,
|
||||
),
|
||||
)
|
||||
@pytest.mark.skip_ui
|
||||
def test_incorrect_output_script_type(self, client, script_type):
|
||||
@ -1222,11 +1225,11 @@ class TestMsgSigntx:
|
||||
"030e669acac1f280d1ddf441cd2ba5e97417bf2689e4bbec86df4f831bf9f7ffd0"
|
||||
)
|
||||
|
||||
multisig = proto.MultisigRedeemScriptType(
|
||||
multisig = messages.MultisigRedeemScriptType(
|
||||
m=1,
|
||||
nodes=[
|
||||
btc.get_public_node(client, address_n, coin_name="Testnet").node,
|
||||
proto.HDNodeType(
|
||||
messages.HDNodeType(
|
||||
depth=0,
|
||||
fingerprint=0,
|
||||
child_num=0,
|
||||
@ -1236,23 +1239,23 @@ class TestMsgSigntx:
|
||||
],
|
||||
address_n=[],
|
||||
)
|
||||
inp1 = proto.TxInputType(
|
||||
inp1 = messages.TxInputType(
|
||||
address_n=address_n,
|
||||
prev_index=1,
|
||||
sequence=0xFFFFFFFF,
|
||||
script_type=proto.InputScriptType.SPENDADDRESS,
|
||||
script_type=messages.InputScriptType.SPENDADDRESS,
|
||||
prev_hash=TXHASH_e5040e,
|
||||
)
|
||||
out1 = proto.TxOutputType(
|
||||
out1 = messages.TxOutputType(
|
||||
address_n=address_n,
|
||||
amount=1000000 - 50000 - 10000,
|
||||
script_type=script_type, # incorrect script type
|
||||
multisig=multisig,
|
||||
)
|
||||
out2 = proto.TxOutputType(
|
||||
out2 = messages.TxOutputType(
|
||||
address="mtkyndbpgv1G7nwggwKDVagRpxEJrwwyh6",
|
||||
amount=50000,
|
||||
script_type=proto.OutputScriptType.PAYTOADDRESS,
|
||||
script_type=messages.OutputScriptType.PAYTOADDRESS,
|
||||
)
|
||||
|
||||
with pytest.raises(
|
||||
|
||||
Loading…
Reference in New Issue
Block a user