You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
470 lines
17 KiB
470 lines
17 KiB
from micropython import const
|
|
|
|
from trezor.crypto.hashlib import sha256
|
|
from trezor.crypto.curve import secp256k1
|
|
from trezor.crypto import base58, der
|
|
|
|
from trezor.messages.TxRequestSerializedType import TxRequestSerializedType
|
|
from trezor.messages.TxRequestDetailsType import TxRequestDetailsType
|
|
from trezor.messages import OutputScriptType
|
|
|
|
from apps.common import address_type
|
|
from apps.common import coins
|
|
from apps.wallet.sign_tx.addresses import *
|
|
from apps.wallet.sign_tx.helpers import *
|
|
from apps.wallet.sign_tx.segwit_bip143 import *
|
|
from apps.wallet.sign_tx.scripts import *
|
|
from apps.wallet.sign_tx.writers import *
|
|
from apps.wallet.sign_tx.tx_weight_calculator import *
|
|
|
|
# the number of bip32 levels used in a wallet (chain and address)
|
|
_BIP32_WALLET_DEPTH = const(2)
|
|
|
|
# the chain id used for change
|
|
_BIP32_CHANGE_CHAIN = const(1)
|
|
|
|
# the maximum allowed change address. this should be large enough for normal
|
|
# use and still allow to quickly brute-force the correct bip32 path
|
|
_BIP32_MAX_LAST_ELEMENT = const(1000000)
|
|
|
|
|
|
class SigningError(ValueError):
|
|
pass
|
|
|
|
|
|
# Transaction signing
|
|
# ===
|
|
# see https://github.com/trezor/trezor-mcu/blob/master/firmware/signing.c#L84
|
|
# for pseudo code overview
|
|
# ===
|
|
|
|
|
|
# Phase 1
|
|
# - check inputs, previous transactions, and outputs
|
|
# - ask for confirmations
|
|
# - check fee
|
|
async def check_tx_fee(tx: SignTx, root):
|
|
|
|
coin = coins.by_name(tx.coin_name)
|
|
|
|
# h_first is used to make sure the inputs and outputs streamed in Phase 1
|
|
# are the same as in Phase 2. it is thus not required to fully hash the
|
|
# tx, as the SignTx info is streamed only once
|
|
h_first = HashWriter(sha256) # not a real tx hash
|
|
bip143 = Bip143()
|
|
weight = TxWeightCalculator(tx.inputs_count, tx.outputs_count)
|
|
|
|
txo_bin = TxOutputBinType()
|
|
tx_req = TxRequest()
|
|
tx_req.details = TxRequestDetailsType()
|
|
|
|
total_in = 0 # sum of input amounts
|
|
segwit_in = 0 # sum of segwit input amounts
|
|
total_out = 0 # sum of output amounts
|
|
change_out = 0 # change output amount
|
|
wallet_path = [] # common prefix of input paths
|
|
segwit = {} # dict of booleans stating if input is segwit
|
|
|
|
for i in range(tx.inputs_count):
|
|
# STAGE_REQUEST_1_INPUT
|
|
txi = await request_tx_input(tx_req, i)
|
|
wallet_path = input_extract_wallet_path(txi, wallet_path)
|
|
write_tx_input_check(h_first, txi)
|
|
weight.add_input(txi)
|
|
bip143.add_prevouts(txi)
|
|
bip143.add_sequence(txi)
|
|
is_segwit = (txi.script_type == InputScriptType.SPENDWITNESS or
|
|
txi.script_type == InputScriptType.SPENDP2SHWITNESS)
|
|
if is_segwit:
|
|
if not coin.segwit:
|
|
raise SigningError(FailureType.DataError,
|
|
'Segwit not enabled on this coin')
|
|
if not txi.amount:
|
|
raise SigningError(FailureType.DataError,
|
|
'Segwit input without amount')
|
|
segwit[i] = True
|
|
segwit_in += txi.amount
|
|
total_in += txi.amount
|
|
elif txi.script_type == InputScriptType.SPENDADDRESS:
|
|
segwit[i] = False
|
|
total_in += await get_prevtx_output_value(
|
|
tx_req, txi.prev_hash, txi.prev_index)
|
|
else:
|
|
raise SigningError(FailureType.DataError,
|
|
'Wrong input script type')
|
|
|
|
for o in range(tx.outputs_count):
|
|
# STAGE_REQUEST_3_OUTPUT
|
|
txo = await request_tx_output(tx_req, o)
|
|
txo_bin.amount = txo.amount
|
|
txo_bin.script_pubkey = output_derive_script(txo, coin, root)
|
|
weight.add_output(txo_bin.script_pubkey)
|
|
if output_is_change(txo, wallet_path, segwit_in):
|
|
if change_out != 0:
|
|
raise SigningError(FailureType.ProcessError,
|
|
'Only one change output is valid')
|
|
change_out = txo.amount
|
|
elif not await confirm_output(txo, coin):
|
|
raise SigningError(FailureType.ActionCancelled,
|
|
'Output cancelled')
|
|
write_tx_output(h_first, txo_bin)
|
|
bip143.add_output(txo_bin)
|
|
total_out += txo_bin.amount
|
|
|
|
fee = total_in - total_out
|
|
if fee < 0:
|
|
raise SigningError(FailureType.NotEnoughFunds,
|
|
'Not enough funds')
|
|
|
|
# fee > (coin.maxfee per byte * tx size)
|
|
if fee > (coin.maxfee_kb / 1000) * (weight.get_total() / 4):
|
|
if not await confirm_feeoverthreshold(fee, coin):
|
|
raise SigningError(FailureType.ActionCancelled,
|
|
'Signing cancelled')
|
|
|
|
if not await confirm_total(total_out - change_out, fee, coin):
|
|
raise SigningError(FailureType.ActionCancelled,
|
|
'Total cancelled')
|
|
|
|
return h_first, bip143, segwit, total_in, wallet_path
|
|
|
|
|
|
async def sign_tx(tx: SignTx, root):
|
|
|
|
tx = sanitize_sign_tx(tx)
|
|
|
|
# Phase 1
|
|
|
|
h_first, bip143, segwit, authorized_in, wallet_path = await check_tx_fee(tx, root)
|
|
|
|
# Phase 2
|
|
# - sign inputs
|
|
# - check that nothing changed
|
|
|
|
coin = coins.by_name(tx.coin_name)
|
|
tx_ser = TxRequestSerializedType()
|
|
|
|
txo_bin = TxOutputBinType()
|
|
tx_req = TxRequest()
|
|
tx_req.details = TxRequestDetailsType()
|
|
tx_req.serialized = None
|
|
|
|
for i_sign in range(tx.inputs_count):
|
|
txi_sign = None
|
|
key_sign = None
|
|
key_sign_pub = None
|
|
|
|
if segwit[i_sign]:
|
|
# STAGE_REQUEST_SEGWIT_INPUT
|
|
txi_sign = await request_tx_input(tx_req, i_sign)
|
|
|
|
is_segwit = (txi_sign.script_type == InputScriptType.SPENDWITNESS or
|
|
txi_sign.script_type == InputScriptType.SPENDP2SHWITNESS)
|
|
if not is_segwit:
|
|
raise SigningError(FailureType.ProcessError,
|
|
'Transaction has changed during signing')
|
|
input_check_wallet_path(txi_sign, wallet_path)
|
|
|
|
key_sign = node_derive(root, txi_sign.address_n)
|
|
key_sign_pub = key_sign.public_key()
|
|
txi_sign.script_sig = input_derive_script(txi_sign, key_sign_pub)
|
|
w_txi = bytearray_with_cap(
|
|
7 + len(txi_sign.prev_hash) + 4 + len(txi_sign.script_sig) + 4)
|
|
if i_sign == 0: # serializing first input => prepend headers
|
|
write_bytes(w_txi, get_tx_header(tx, True))
|
|
write_tx_input(w_txi, txi_sign)
|
|
tx_ser.serialized_tx = w_txi
|
|
tx_req.serialized = tx_ser
|
|
|
|
else:
|
|
# hash of what we are signing with this input
|
|
h_sign = HashWriter(sha256)
|
|
# same as h_first, checked before signing the digest
|
|
h_second = HashWriter(sha256)
|
|
|
|
write_uint32(h_sign, tx.version)
|
|
write_varint(h_sign, tx.inputs_count)
|
|
|
|
for i in range(tx.inputs_count):
|
|
# STAGE_REQUEST_4_INPUT
|
|
txi = await request_tx_input(tx_req, i)
|
|
input_check_wallet_path(txi, wallet_path)
|
|
write_tx_input_check(h_second, txi)
|
|
if i == i_sign:
|
|
txi_sign = txi
|
|
key_sign = node_derive(root, txi.address_n)
|
|
key_sign_pub = key_sign.public_key()
|
|
txi_sign.script_sig = output_script_p2pkh(
|
|
ecdsa_hash_pubkey(key_sign_pub))
|
|
else:
|
|
txi.script_sig = bytes()
|
|
write_tx_input(h_sign, txi)
|
|
|
|
write_varint(h_sign, tx.outputs_count)
|
|
|
|
for o in range(tx.outputs_count):
|
|
# STAGE_REQUEST_4_OUTPUT
|
|
txo = await request_tx_output(tx_req, o)
|
|
txo_bin.amount = txo.amount
|
|
txo_bin.script_pubkey = output_derive_script(txo, coin, root)
|
|
write_tx_output(h_second, txo_bin)
|
|
write_tx_output(h_sign, txo_bin)
|
|
|
|
write_uint32(h_sign, tx.lock_time)
|
|
|
|
write_uint32(h_sign, 0x00000001) # SIGHASH_ALL hash_type
|
|
|
|
# check the control digests
|
|
if get_tx_hash(h_first, False) != get_tx_hash(h_second, False):
|
|
raise SigningError(FailureType.ProcessError,
|
|
'Transaction has changed during signing')
|
|
|
|
# compute the signature from the tx digest
|
|
signature = ecdsa_sign(key_sign, get_tx_hash(h_sign, True))
|
|
tx_ser.signature_index = i_sign
|
|
tx_ser.signature = signature
|
|
|
|
# serialize input with correct signature
|
|
txi_sign.script_sig = input_derive_script(
|
|
txi_sign, key_sign_pub, signature)
|
|
w_txi_sign = bytearray_with_cap(
|
|
5 + len(txi_sign.prev_hash) + 4 + len(txi_sign.script_sig) + 4)
|
|
if i_sign == 0: # serializing first input => prepend headers
|
|
write_bytes(w_txi_sign, get_tx_header(tx))
|
|
write_tx_input(w_txi_sign, txi_sign)
|
|
tx_ser.serialized_tx = w_txi_sign
|
|
|
|
tx_req.serialized = tx_ser
|
|
|
|
for o in range(tx.outputs_count):
|
|
# STAGE_REQUEST_5_OUTPUT
|
|
txo = await request_tx_output(tx_req, o)
|
|
txo_bin.amount = txo.amount
|
|
txo_bin.script_pubkey = output_derive_script(txo, coin, root)
|
|
|
|
# serialize output
|
|
w_txo_bin = bytearray_with_cap(
|
|
5 + 8 + 5 + len(txo_bin.script_pubkey) + 4)
|
|
if o == 0: # serializing first output => prepend outputs count
|
|
write_varint(w_txo_bin, tx.outputs_count)
|
|
write_tx_output(w_txo_bin, txo_bin)
|
|
|
|
tx_ser.signature_index = None
|
|
tx_ser.signature = None
|
|
tx_ser.serialized_tx = w_txo_bin
|
|
|
|
tx_req.serialized = tx_ser
|
|
|
|
any_segwit = True in segwit.values()
|
|
|
|
for i in range(tx.inputs_count):
|
|
if segwit[i]:
|
|
# STAGE_REQUEST_SEGWIT_WITNESS
|
|
txi = await request_tx_input(tx_req, i)
|
|
input_check_wallet_path(txi, wallet_path)
|
|
|
|
is_segwit = (txi.script_type == InputScriptType.SPENDWITNESS or
|
|
txi.script_type == InputScriptType.SPENDP2SHWITNESS)
|
|
if not is_segwit or txi.amount > authorized_in:
|
|
raise SigningError(FailureType.ProcessError,
|
|
'Transaction has changed during signing')
|
|
authorized_in -= txi.amount
|
|
|
|
key_sign = node_derive(root, txi.address_n)
|
|
key_sign_pub = key_sign.public_key()
|
|
bip143_hash = bip143.preimage_hash(tx, txi, ecdsa_hash_pubkey(key_sign_pub))
|
|
|
|
signature = ecdsa_sign(key_sign, bip143_hash)
|
|
witness = get_p2wpkh_witness(signature, key_sign_pub)
|
|
|
|
tx_ser.serialized_tx = witness
|
|
tx_ser.signature_index = i
|
|
tx_ser.signature = signature
|
|
elif any_segwit:
|
|
tx_ser.serialized_tx = bytearray(1) # empty witness for non-segwit inputs
|
|
tx_ser.signature_index = None
|
|
tx_ser.signature = None
|
|
|
|
tx_req.serialized = tx_ser
|
|
|
|
write_uint32(tx_ser.serialized_tx, tx.lock_time)
|
|
|
|
await request_tx_finish(tx_req)
|
|
|
|
|
|
async def get_prevtx_output_value(tx_req: TxRequest, prev_hash: bytes, prev_index: int) -> int:
|
|
total_out = 0 # sum of output amounts
|
|
|
|
# STAGE_REQUEST_2_PREV_META
|
|
tx = await request_tx_meta(tx_req, prev_hash)
|
|
|
|
txh = HashWriter(sha256)
|
|
|
|
write_uint32(txh, tx.version)
|
|
write_varint(txh, tx.inputs_cnt)
|
|
|
|
for i in range(tx.inputs_cnt):
|
|
# STAGE_REQUEST_2_PREV_INPUT
|
|
txi = await request_tx_input(tx_req, i, prev_hash)
|
|
write_tx_input(txh, txi)
|
|
|
|
write_varint(txh, tx.outputs_cnt)
|
|
|
|
for o in range(tx.outputs_cnt):
|
|
# STAGE_REQUEST_2_PREV_OUTPUT
|
|
txo_bin = await request_tx_output(tx_req, o, prev_hash)
|
|
write_tx_output(txh, txo_bin)
|
|
if o == prev_index:
|
|
total_out += txo_bin.amount
|
|
|
|
write_uint32(txh, tx.lock_time)
|
|
|
|
ofs = 0
|
|
while ofs < tx.extra_data_len:
|
|
size = min(1024, tx.extra_data_len - ofs)
|
|
data = await request_tx_extra_data(tx_req, ofs, size, prev_hash)
|
|
write_bytes(txh, data)
|
|
ofs += len(data)
|
|
|
|
if get_tx_hash(txh, True, True) != prev_hash:
|
|
raise SigningError(FailureType.ProcessError,
|
|
'Encountered invalid prev_hash')
|
|
|
|
return total_out
|
|
|
|
|
|
# TX Helpers
|
|
# ===
|
|
|
|
|
|
def get_tx_header(tx: SignTx, segwit=False):
|
|
w_txi = bytearray()
|
|
write_uint32(w_txi, tx.version)
|
|
if segwit:
|
|
write_varint(w_txi, 0x00) # segwit witness marker
|
|
write_varint(w_txi, 0x01) # segwit witness flag
|
|
write_varint(w_txi, tx.inputs_count)
|
|
return w_txi
|
|
|
|
|
|
def get_p2wpkh_witness(signature: bytes, pubkey: bytes):
|
|
w = bytearray_with_cap(1 + 5 + len(signature) + 1 + 5 + len(pubkey))
|
|
write_varint(w, 0x02) # num of segwit items, in P2WPKH it's always 2
|
|
append_signature_and_pubkey(w, pubkey, signature)
|
|
return w
|
|
|
|
|
|
# TX Outputs
|
|
# ===
|
|
|
|
|
|
def output_derive_script(o: TxOutputType, coin: CoinType, root) -> bytes:
|
|
|
|
if o.script_type == OutputScriptType.PAYTOOPRETURN:
|
|
if o.amount != 0:
|
|
raise SigningError(FailureType.DataError,
|
|
'OP_RETURN output with non-zero amount')
|
|
return output_script_paytoopreturn(o.op_return_data)
|
|
|
|
if o.address_n: # change output
|
|
if o.address:
|
|
raise SigningError(FailureType.DataError, 'Address in change output')
|
|
o.address = get_address_for_change(o, coin, root)
|
|
else:
|
|
if not o.address:
|
|
raise SigningError(FailureType.DataError, 'Missing address')
|
|
|
|
if coin.bech32_prefix and o.address.startswith(coin.bech32_prefix): # p2wpkh or p2wsh
|
|
witprog = decode_bech32_address(coin.bech32_prefix, o.address)
|
|
return output_script_native_p2wpkh_or_p2wsh(witprog)
|
|
|
|
raw_address = base58.decode_check(o.address)
|
|
|
|
if address_type.check(coin.address_type, raw_address): # p2pkh
|
|
pubkeyhash = address_type.strip(coin.address_type, raw_address)
|
|
return output_script_p2pkh(pubkeyhash)
|
|
|
|
elif address_type.check(coin.address_type_p2sh, raw_address): # p2sh
|
|
scripthash = address_type.strip(coin.address_type_p2sh, raw_address)
|
|
return output_script_p2sh(scripthash)
|
|
|
|
raise SigningError(FailureType.DataError, 'Invalid address type')
|
|
|
|
|
|
def get_address_for_change(o: TxOutputType, coin: CoinType, root):
|
|
if o.script_type == OutputScriptType.PAYTOADDRESS:
|
|
input_script_type = InputScriptType.SPENDADDRESS
|
|
elif o.script_type == OutputScriptType.PAYTOMULTISIG:
|
|
input_script_type = InputScriptType.SPENDMULTISIG
|
|
elif o.script_type == OutputScriptType.PAYTOWITNESS:
|
|
input_script_type = InputScriptType.SPENDWITNESS
|
|
elif o.script_type == OutputScriptType.PAYTOP2SHWITNESS:
|
|
input_script_type = InputScriptType.SPENDP2SHWITNESS
|
|
else:
|
|
raise SigningError(FailureType.DataError, 'Invalid script type')
|
|
return get_address(input_script_type, coin, node_derive(root, o.address_n))
|
|
|
|
|
|
def output_is_change(o: TxOutputType, wallet_path: list, segwit_in: int) -> bool:
|
|
address_n = o.address_n
|
|
is_segwit = (o.script_type == OutputScriptType.PAYTOWITNESS or
|
|
o.script_type == OutputScriptType.PAYTOP2SHWITNESS)
|
|
if is_segwit and o.amount > segwit_in:
|
|
# if the output is segwit, make sure it doesn't spend more than what the
|
|
# segwit inputs paid. this is to prevent user being tricked into
|
|
# creating ANYONECANSPEND outputs before full segwit activation.
|
|
return False
|
|
return (address_n is not None and wallet_path is not None
|
|
and wallet_path == address_n[:-_BIP32_WALLET_DEPTH]
|
|
and address_n[-2] == _BIP32_CHANGE_CHAIN
|
|
and address_n[-1] <= _BIP32_MAX_LAST_ELEMENT)
|
|
|
|
|
|
# Tx Inputs
|
|
# ===
|
|
|
|
|
|
def input_derive_script(i: TxInputType, pubkey: bytes, signature: bytes=None) -> bytes:
|
|
if i.script_type == InputScriptType.SPENDADDRESS:
|
|
return input_script_p2pkh_or_p2sh(pubkey, signature) # p2pkh or p2sh
|
|
if i.script_type == InputScriptType.SPENDP2SHWITNESS: # p2wpkh using p2sh
|
|
return input_script_p2wpkh_in_p2sh(ecdsa_hash_pubkey(pubkey))
|
|
elif i.script_type == InputScriptType.SPENDWITNESS: # native p2wpkh or p2wsh
|
|
return input_script_native_p2wpkh_or_p2wsh()
|
|
else:
|
|
raise SigningError(FailureType.ProcessError, 'Invalid script type')
|
|
|
|
|
|
def input_extract_wallet_path(txi: TxInputType, wallet_path: list) -> list:
|
|
if wallet_path is None:
|
|
return None # there was a mismatch in previous inputs
|
|
address_n = txi.address_n[:-_BIP32_WALLET_DEPTH]
|
|
if not address_n:
|
|
return None # input path is too short
|
|
if not wallet_path:
|
|
return address_n # this is the first input
|
|
if wallet_path == address_n:
|
|
return address_n # paths match
|
|
return None # paths don't match
|
|
|
|
|
|
def input_check_wallet_path(txi: TxInputType, wallet_path: list) -> list:
|
|
if wallet_path is None:
|
|
return # there was a mismatch in Phase 1, ignore it now
|
|
address_n = txi.address_n[:-_BIP32_WALLET_DEPTH]
|
|
if wallet_path != address_n:
|
|
raise SigningError(FailureType.ProcessError,
|
|
'Transaction has changed during signing')
|
|
|
|
|
|
def node_derive(root, address_n: list):
|
|
node = root.clone()
|
|
node.derive_path(address_n)
|
|
return node
|
|
|
|
|
|
def ecdsa_sign(node, digest: bytes) -> bytes:
|
|
sig = secp256k1.sign(node.private_key(), digest)
|
|
sigder = der.encode_seq((sig[1:33], sig[33:65]))
|
|
return sigder
|