arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-12-22 22:38:08 +00:00
Code Issues Releases Wiki Activity

core/bitcoin: Replace TxInputType parameter in input_derive_script.

Browse Source
This commit is contained in:
Andrew Kozlik 2020-06-10 11:43:48 +02:00 committed by Andrew Kozlik
parent a901573ea2
commit 173bb7ed13
2 changed files with 18 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
core/src/apps/bitcoin/scripts.py
View File

@ -24,41 +24,43 @@ from .writers import (
if False:
from typing import List, Optional
from trezor.messages.TxInputType import EnumTypeInputScriptType
from .writers import Writer
def input_derive_script(
txi: TxInputType,
script_type: EnumTypeInputScriptType,
multisig: MultisigRedeemScriptType,
coin: CoinInfo,
hash_type: int,
pubkey: bytes,
signature: Optional[bytes],
) -> bytes:
if txi.script_type == InputScriptType.SPENDADDRESS:
if script_type == InputScriptType.SPENDADDRESS:
# p2pkh or p2sh
return input_script_p2pkh_or_p2sh(pubkey, signature, hash_type)
if txi.script_type == InputScriptType.SPENDP2SHWITNESS:
if script_type == InputScriptType.SPENDP2SHWITNESS:
# p2wpkh or p2wsh using p2sh
if txi.multisig:
if multisig:
# p2wsh in p2sh
pubkeys = multisig_get_pubkeys(txi.multisig)
pubkeys = multisig_get_pubkeys(multisig)
witness_script_hasher = utils.HashWriter(sha256())
write_output_script_multisig(witness_script_hasher, pubkeys, txi.multisig.m)
write_output_script_multisig(witness_script_hasher, pubkeys, multisig.m)
witness_script_hash = witness_script_hasher.get_digest()
return input_script_p2wsh_in_p2sh(witness_script_hash)
# p2wpkh in p2sh
return input_script_p2wpkh_in_p2sh(common.ecdsa_hash_pubkey(pubkey, coin))
elif txi.script_type == InputScriptType.SPENDWITNESS:
elif script_type == InputScriptType.SPENDWITNESS:
# native p2wpkh or p2wsh
return input_script_native_p2wpkh_or_p2wsh()
elif txi.script_type == InputScriptType.SPENDMULTISIG:
elif script_type == InputScriptType.SPENDMULTISIG:
# p2sh multisig
signature_index = multisig_pubkey_index(txi.multisig, pubkey)
signature_index = multisig_pubkey_index(multisig, pubkey)
return input_script_multisig(
txi.multisig, signature, signature_index, hash_type, coin
multisig, signature, signature_index, hash_type, coin
)
else:
raise wire.ProcessError("Invalid script type")

7
core/src/apps/bitcoin/sign_tx/bitcoin.py
View File

@ -467,7 +467,12 @@ class Bitcoin:
self, txi: TxInputType, pubkey: bytes, signature: bytes = None
) -> bytes:
return scripts.input_derive_script(
txi, self.coin, self.get_hash_type(), pubkey, signature
txi.script_type,
txi.multisig,
self.coin,
self.get_hash_type(),
pubkey,
signature,
)
# BIP-0143