arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-11-18 05:28:40 +00:00
Code Issues Releases Wiki Activity

python/btc: improve missing signature detection

Browse Source
This commit is contained in:
matejcik 2020-08-05 14:58:39 +02:00
parent 037bef3a38
commit efcaf7bda6
1 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
python/src/trezorlib/btc.py
View File

@ -200,10 +200,7 @@ def sign_tx(
res = client.call(signtx)
# Prepare structure for signatures
signatures = [
None if i.script_type != messages.InputScriptType.EXTERNAL else ""
for i in inputs
]
signatures = [None] * len(inputs)
serialized_tx = b""
def copy_tx_meta(tx):
@ -268,7 +265,8 @@ def sign_tx(
if not isinstance(res, messages.TxRequest):
raise exceptions.TrezorException("Unexpected message")
if None in signatures:
for i, sig in zip(inputs, signatures):
if i.script_type != messages.InputScriptType.EXTERNAL and sig is None:
raise exceptions.TrezorException("Some signatures are missing!")
return signatures, serialized_tx