apps.common: add op_return to signtx

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

View File

@ -144,6 +144,7 @@ async def sign_tx(tx: SignTx, root):
'Only one change output is valid') 'Only one change output is valid')
change_out = txo.amount change_out = txo.amount
else: else:
if txo.script_type != OutputScriptType.PAYTOOPRETURN:
if not await confirm_output(txo, coin): if not await confirm_output(txo, coin):
raise SigningError(FailureType.ActionCancelled, raise SigningError(FailureType.ActionCancelled,
'Output cancelled') 'Output cancelled')
@ -319,6 +320,12 @@ def output_derive_script(o: TxOutputType, coin: CoinType, root) -> bytes:
elif o.script_type == OutputScriptType.PAYTOSCRIPTHASH: elif o.script_type == OutputScriptType.PAYTOSCRIPTHASH:
ra = output_paytoaddress_extract_raw_address(o, coin, root, p2sh=True) ra = output_paytoaddress_extract_raw_address(o, coin, root, p2sh=True)
return script_paytoscripthash_new(ra[1:]) 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: else:
raise SigningError(FailureType.SyntaxError, raise SigningError(FailureType.SyntaxError,
'Invalid output script type') 'Invalid output script type')
@ -421,6 +428,13 @@ def script_paytoscripthash_new(scripthash: bytes) -> bytearray:
s[22] = 0x87 # OP_EQUAL s[22] = 0x87 # OP_EQUAL
return s 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: def script_spendaddress_new(pubkey: bytes, signature: bytes) -> bytearray:
w = bytearray() w = bytearray()
write_op_push(w, len(signature) + 1) write_op_push(w, len(signature) + 1)