mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-11-25 17:09:44 +00:00
103 lines
4.3 KiB
Python
103 lines
4.3 KiB
Python
from common import *
|
|
|
|
from apps.wallet.sign_tx.signing import *
|
|
from apps.wallet.sign_tx.segwit_bip143 import *
|
|
from apps.common import coins
|
|
from trezor.messages.SignTx import SignTx
|
|
from trezor.messages.TxInputType import TxInputType
|
|
from trezor.messages.TxOutputType import TxOutputType
|
|
from trezor.messages import OutputScriptType
|
|
from trezor.crypto import bip32, bip39
|
|
|
|
|
|
class TestSegwitBip143NativeP2WPKH(unittest.TestCase):
|
|
# pylint: disable=C0301
|
|
|
|
tx = SignTx(coin_name='Bitcoin', version=1, lock_time=0x00000011, inputs_count=2, outputs_count=2)
|
|
inp1 = TxInputType(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 = TxInputType(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 = TxOutputType(address='1Cu32FVupVCgHkMMRJdYJugxwo2Aprgk7H', # derived
|
|
amount=0x0000000006b22c20,
|
|
script_type=OutputScriptType.PAYTOADDRESS,
|
|
multisig=None,
|
|
address_n=[])
|
|
out2 = TxOutputType(address='16TZ8J6Q5iZKBWizWzFAYnrsaox5Z5aBRV', # derived
|
|
amount=0x000000000d519390,
|
|
script_type=OutputScriptType.PAYTOADDRESS,
|
|
multisig=None,
|
|
address_n=[])
|
|
|
|
def test_prevouts(self):
|
|
|
|
bip143 = Bip143()
|
|
bip143.add_prevouts(self.inp1)
|
|
bip143.add_prevouts(self.inp2)
|
|
coin = coins.by_name(self.tx.coin_name)
|
|
self.assertEqual(hexlify(bip143.get_prevouts_hash(coin)), b'96b827c8483d4e9b96712b6713a7b68d6e8003a781feba36c31143470b4efd37')
|
|
|
|
def test_sequence(self):
|
|
|
|
bip143 = Bip143()
|
|
bip143.add_sequence(self.inp1)
|
|
bip143.add_sequence(self.inp2)
|
|
coin = coins.by_name(self.tx.coin_name)
|
|
self.assertEqual(hexlify(bip143.get_sequence_hash(coin)), b'52b0a642eea2fb7ae638c36f6252b6750293dbe574a806984b8e4d8548339a3b')
|
|
|
|
def test_outputs(self):
|
|
|
|
seed = bip39.seed('alcohol woman abuse must during monitor noble actual mixed trade anger aisle', '')
|
|
root = bip32.from_seed(seed, 'secp256k1')
|
|
coin = coins.by_name(self.tx.coin_name)
|
|
|
|
bip143 = Bip143()
|
|
|
|
for txo in [self.out1, self.out2]:
|
|
txo_bin = TxOutputBinType()
|
|
txo_bin.amount = txo.amount
|
|
txo_bin.script_pubkey = output_derive_script(txo, coin, root)
|
|
bip143.add_output(txo_bin)
|
|
|
|
self.assertEqual(hexlify(bip143.get_outputs_hash(coin)),
|
|
b'863ef3e1a92afbfdb97f31ad0fc7683ee943e9abcf2501590ff8f6551f47e5e5')
|
|
|
|
def test_preimage_testdata(self):
|
|
|
|
seed = bip39.seed('alcohol woman abuse must during monitor noble actual mixed trade anger aisle', '')
|
|
root = bip32.from_seed(seed, 'secp256k1')
|
|
coin = coins.by_name(self.tx.coin_name)
|
|
|
|
bip143 = Bip143()
|
|
bip143.add_prevouts(self.inp1)
|
|
bip143.add_prevouts(self.inp2)
|
|
bip143.add_sequence(self.inp1)
|
|
bip143.add_sequence(self.inp2)
|
|
|
|
for txo in [self.out1, self.out2]:
|
|
txo_bin = TxOutputBinType()
|
|
txo_bin.amount = txo.amount
|
|
txo_bin.script_pubkey = output_derive_script(txo, coin, root)
|
|
bip143.add_output(txo_bin)
|
|
|
|
# test data public key hash
|
|
# only for input 2 - input 1 is not segwit
|
|
result = bip143.preimage_hash(coin, self.tx, self.inp2, unhexlify('1d0f172a0ecb48aee1be1f2687d2963ae33f71a1'), 0x01)
|
|
self.assertEqual(hexlify(result), b'c37af31116d1b27caf68aae9e3ac82f1477929014d5b917657d0eb49478cb670')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|