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.
159 lines
5.0 KiB
159 lines
5.0 KiB
import gc
|
|
from micropython import const
|
|
|
|
from trezor import log
|
|
|
|
from apps.monero.xmr import crypto
|
|
|
|
if False:
|
|
from typing import Dict, List, Optional, Tuple
|
|
from apps.monero.xmr.types import Ge25519, Sc25519
|
|
from apps.monero.xmr.credentials import AccountCreds
|
|
|
|
Subaddresses = Dict[bytes, Tuple[int, int]]
|
|
|
|
|
|
class State:
|
|
|
|
STEP_INIT = const(0)
|
|
STEP_INP = const(100)
|
|
STEP_PERM = const(200)
|
|
STEP_VINI = const(300)
|
|
STEP_ALL_IN = const(350)
|
|
STEP_OUT = const(400)
|
|
STEP_ALL_OUT = const(500)
|
|
STEP_SIGN = const(600)
|
|
|
|
def __init__(self, ctx):
|
|
from apps.monero.xmr.keccak_hasher import KeccakXmrArchive
|
|
from apps.monero.xmr.mlsag_hasher import PreMlsagHasher
|
|
|
|
self.ctx = ctx
|
|
|
|
"""
|
|
Account credentials
|
|
type: AccountCreds
|
|
- view private/public key
|
|
- spend private/public key
|
|
- and its corresponding address
|
|
"""
|
|
self.creds = None # type: Optional[AccountCreds]
|
|
|
|
# HMAC/encryption keys used to protect offloaded data
|
|
self.key_hmac = None # type: Optional[bytes]
|
|
self.key_enc = None # type: Optional[bytes]
|
|
|
|
"""
|
|
Transaction keys
|
|
- also denoted as r/R
|
|
- tx_priv is a random number
|
|
- tx_pub is equal to `r*G` or `r*D` for subaddresses
|
|
- for subaddresses the `r` is commonly denoted as `s`, however it is still just a random number
|
|
- the keys are used to derive the one time address and its keys (P = H(A*r)*G + B)
|
|
"""
|
|
self.tx_priv = None # type: Sc25519
|
|
self.tx_pub = None # type: Ge25519
|
|
|
|
"""
|
|
In some cases when subaddresses are used we need more tx_keys
|
|
(explained in step 1).
|
|
"""
|
|
self.need_additional_txkeys = False
|
|
|
|
# Connected client version
|
|
self.client_version = 0
|
|
self.hard_fork = 12
|
|
|
|
self.input_count = 0
|
|
self.output_count = 0
|
|
self.progress_total = 0
|
|
self.progress_cur = 0
|
|
|
|
self.output_change = None
|
|
self.fee = 0
|
|
self.tx_type = 0
|
|
|
|
# wallet sub-address major index
|
|
self.account_idx = 0
|
|
|
|
# contains additional tx keys if need_additional_tx_keys is True
|
|
self.additional_tx_private_keys = [] # type: List[Sc25519]
|
|
self.additional_tx_public_keys = [] # type: List[bytes]
|
|
|
|
# currently processed input/output index
|
|
self.current_input_index = -1
|
|
self.current_output_index = -1
|
|
self.is_processing_offloaded = False
|
|
|
|
# for pseudo_out recomputation from new mask
|
|
self.input_last_amount = 0
|
|
|
|
self.summary_inputs_money = 0
|
|
self.summary_outs_money = 0
|
|
|
|
# output commitments
|
|
self.output_pk_commitments = [] # type: List[bytes]
|
|
|
|
self.output_amounts = [] # type: List[int]
|
|
# output *range proof* masks. HP10+ makes them deterministic.
|
|
self.output_masks = [] # type: List[Sc25519]
|
|
|
|
# the range proofs are calculated in batches, this denotes the grouping
|
|
self.rsig_grouping = [] # type: List[int]
|
|
# is range proof computing offloaded or not
|
|
self.rsig_offload = False
|
|
|
|
# sum of all inputs' pseudo out masks
|
|
self.sumpouts_alphas = crypto.sc_0() # type: Sc25519
|
|
# sum of all output' pseudo out masks
|
|
self.sumout = crypto.sc_0() # type: Sc25519
|
|
|
|
self.subaddresses = {} # type: Subaddresses
|
|
|
|
# TX_EXTRA_NONCE extra field for tx.extra, due to sort_tx_extra()
|
|
self.extra_nonce = None
|
|
|
|
# contains an array where each item denotes the input's position
|
|
# (inputs are sorted by key images)
|
|
self.source_permutation = [] # type: List[int]
|
|
|
|
# Last key image seen. Used for input permutation correctness check
|
|
self.last_ki = None # type: Optional[bytes]
|
|
|
|
# Encryption key to release to host after protocol ends without error
|
|
self.opening_key = None # type: Optional[bytes]
|
|
|
|
# Step transition automaton
|
|
self.last_step = self.STEP_INIT
|
|
|
|
"""
|
|
Tx prefix hasher/hash. We use the hasher to incrementally hash and then
|
|
store the final hash in tx_prefix_hash.
|
|
See Monero-Trezor documentation section 3.3 for more details.
|
|
"""
|
|
self.tx_prefix_hasher = KeccakXmrArchive()
|
|
self.tx_prefix_hash = None # type: Optional[bytes]
|
|
|
|
"""
|
|
Full message hasher/hash that is to be signed using MLSAG.
|
|
Contains tx_prefix_hash.
|
|
See Monero-Trezor documentation section 3.3 for more details.
|
|
"""
|
|
self.full_message_hasher = PreMlsagHasher()
|
|
self.full_message = None # type: Optional[bytes]
|
|
|
|
def mem_trace(self, x=None, collect=False):
|
|
if __debug__:
|
|
log.debug(
|
|
__name__,
|
|
"Log trace: %s, ... F: %s A: %s",
|
|
x,
|
|
gc.mem_free(),
|
|
gc.mem_alloc(),
|
|
)
|
|
if collect:
|
|
gc.collect()
|
|
|
|
def change_address(self):
|
|
return self.output_change.addr if self.output_change else None
|