1
0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-12-05 22:18:16 +00:00
trezor-firmware/core/tests/test_apps.bitcoin.segwit.bip143.native_p2wpkh.py

136 lines
4.5 KiB
Python
Raw Normal View History

from common import * # isort:skip
from trezor.crypto import bip39
from trezor.enums import InputScriptType, OutputScriptType
from trezor.messages import PrevOutput, SignTx, TxInput, TxOutput
from apps.bitcoin.common import SigHashType
from apps.bitcoin.scripts import output_derive_script
from apps.bitcoin.sign_tx.sig_hasher import BitcoinSigHasher
from apps.bitcoin.writers import get_tx_hash
from apps.common import coins
2020-07-23 14:20:54 +00:00
from apps.common.keychain import Keychain
from apps.common.paths import AlwaysMatchingSchema
2017-11-09 14:37:22 +00:00
class TestSegwitBip143NativeP2WPKH(unittest.TestCase):
# pylint: disable=C0301
2023-06-28 10:46:29 +00:00
tx = SignTx(
coin_name="Bitcoin",
version=1,
lock_time=0x00000011,
inputs_count=2,
outputs_count=2,
)
inp1 = TxInput(
address_n=[0],
# Trezor expects hash in reversed format
prev_hash=unhexlify(
"9f96ade4b41d5433f4eda31e1738ec2b36f6e7d1420d94a6af99801a88f7f7ff"
),
prev_index=0,
amount=625000000, # 6.25 btc
script_type=InputScriptType.SPENDWITNESS,
multisig=None,
sequence=0xFFFFFFEE,
)
inp2 = TxInput(
address_n=[1],
# Trezor expects hash in reversed format
prev_hash=unhexlify(
"8ac60eb9575db5b2d987e29f301b5b819ea83a5c6579d282d189cc04b8e151ef"
),
prev_index=1,
multisig=None,
amount=600000000, # 6 btc
script_type=InputScriptType.SPENDWITNESS,
sequence=0xFFFFFFFF,
)
out1 = TxOutput(
address="1Cu32FVupVCgHkMMRJdYJugxwo2Aprgk7H", # derived
amount=0x0000000006B22C20,
script_type=OutputScriptType.PAYTOADDRESS,
multisig=None,
address_n=[],
)
out2 = TxOutput(
address="16TZ8J6Q5iZKBWizWzFAYnrsaox5Z5aBRV", # derived
amount=0x000000000D519390,
script_type=OutputScriptType.PAYTOADDRESS,
multisig=None,
address_n=[],
)
2017-11-09 14:37:22 +00:00
def test_prevouts(self):
2018-07-02 07:42:11 +00:00
coin = coins.by_name(self.tx.coin_name)
sig_hasher = BitcoinSigHasher()
sig_hasher.add_input(self.inp1, b"")
sig_hasher.add_input(self.inp2, b"")
prevouts_hash = get_tx_hash(sig_hasher.h_prevouts, double=coin.sign_hash_double)
2023-06-28 10:46:29 +00:00
self.assertEqual(
hexlify(prevouts_hash),
b"96b827c8483d4e9b96712b6713a7b68d6e8003a781feba36c31143470b4efd37",
)
2017-11-09 14:37:22 +00:00
def test_sequence(self):
2018-07-02 07:42:11 +00:00
coin = coins.by_name(self.tx.coin_name)
sig_hasher = BitcoinSigHasher()
sig_hasher.add_input(self.inp1, b"")
sig_hasher.add_input(self.inp2, b"")
2023-06-28 10:46:29 +00:00
sequence_hash = get_tx_hash(
sig_hasher.h_sequences, double=coin.sign_hash_double
)
self.assertEqual(
hexlify(sequence_hash),
b"52b0a642eea2fb7ae638c36f6252b6750293dbe574a806984b8e4d8548339a3b",
)
2017-11-09 14:37:22 +00:00
def test_outputs(self):
coin = coins.by_name(self.tx.coin_name)
sig_hasher = BitcoinSigHasher()
for txo in [self.out1, self.out2]:
script_pubkey = output_derive_script(txo.address, coin)
txo_bin = PrevOutput(amount=txo.amount, script_pubkey=script_pubkey)
sig_hasher.add_output(txo_bin, script_pubkey)
outputs_hash = get_tx_hash(sig_hasher.h_outputs, double=coin.sign_hash_double)
2023-06-28 10:46:29 +00:00
self.assertEqual(
hexlify(outputs_hash),
b"863ef3e1a92afbfdb97f31ad0fc7683ee943e9abcf2501590ff8f6551f47e5e5",
)
2017-11-09 14:37:22 +00:00
def test_preimage_testdata(self):
2023-06-28 10:46:29 +00:00
seed = bip39.seed(
"alcohol woman abuse must during monitor noble actual mixed trade anger aisle",
"",
)
coin = coins.by_name(self.tx.coin_name)
sig_hasher = BitcoinSigHasher()
sig_hasher.add_input(self.inp1, b"")
sig_hasher.add_input(self.inp2, b"")
2017-11-09 14:37:22 +00:00
for txo in [self.out1, self.out2]:
script_pubkey = output_derive_script(txo.address, coin)
txo_bin = PrevOutput(amount=txo.amount, script_pubkey=script_pubkey)
sig_hasher.add_output(txo_bin, script_pubkey)
keychain = Keychain(seed, coin.curve_name, [AlwaysMatchingSchema])
node = keychain.derive(self.inp2.address_n)
# test data public key hash
2017-11-09 14:37:22 +00:00
# only for input 2 - input 1 is not segwit
2023-06-28 10:46:29 +00:00
result = sig_hasher.hash143(
self.inp2, [node.public_key()], 1, self.tx, coin, SigHashType.SIGHASH_ALL
)
self.assertEqual(
hexlify(result),
b"2fa3f1351618b2532228d7182d3221d95c21fd3d496e7e22e9ded873cf022a8b",
)
2023-06-28 10:46:29 +00:00
if __name__ == "__main__":
unittest.main()