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

apps.common: add op_return to signtx

Browse Source
This commit is contained in:
Pavol Rusnak 2016-11-16 01:39:31 +01:00
parent 357a081b53
commit ea12087de7
No known key found for this signature in database
GPG Key ID: 91F3B339B9A02A3D
1 changed files with 17 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
src/apps/common/signtx.py
View File

@ -144,9 +144,10 @@ async def sign_tx(tx: SignTx, root):
'Only one change output is valid')
change_out = txo.amount
else:
if not await confirm_output(txo, coin):
raise SigningError(FailureType.ActionCancelled,
'Output cancelled')
if txo.script_type != OutputScriptType.PAYTOOPRETURN:
if not await confirm_output(txo, coin):
raise SigningError(FailureType.ActionCancelled,
'Output cancelled')
txo_bin.amount = txo.amount
txo_bin.script_pubkey = output_derive_script(txo, coin, root)
write_tx_output(h_first, txo_bin)
@ -319,6 +320,12 @@ def output_derive_script(o: TxOutputType, coin: CoinType, root) -> bytes:
elif o.script_type == OutputScriptType.PAYTOSCRIPTHASH:
ra = output_paytoaddress_extract_raw_address(o, coin, root, p2sh=True)
return script_paytoscripthash_new(ra[1:])
elif o.script_type == OutputScriptType.PAYTOOPRETURN:
if o.amount == 0:
return script_paytoopreturn_new(o.op_return_data)
else:
raise SigningError(FailureType.SyntaxError,
'OP_RETURN output with non-zero amount')
else:
raise SigningError(FailureType.SyntaxError,
'Invalid output script type')
@ -421,6 +428,13 @@ def script_paytoscripthash_new(scripthash: bytes) -> bytearray:
s[22] = 0x87 # OP_EQUAL
return s
def script_paytoopreturn_new(data: bytes) -> bytearray:
w = bytearray()
w.append(0x6A) # OP_RETURN
write_op_push(w, len(data))
w.extend(data)
return w
def script_spendaddress_new(pubkey: bytes, signature: bytes) -> bytearray:
w = bytearray()
write_op_push(w, len(signature) + 1)