arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-12-18 20:38:10 +00:00
Code Issues Releases Wiki Activity

nem: typos

Browse Source
This commit is contained in:
Tomas Susanka 2018-03-26 13:25:46 +02:00 committed by Jan Pochyla
parent 10a52d8688
commit 1f7ab29613
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/apps/nem/signing.py
View File

@ -13,9 +13,9 @@ async def nem_sign_tx(ctx, msg: NEMSignTx):
node = await seed.derive_node(ctx, msg.transaction.address_n, NEM_CURVE)
payload, encrypted = _get_payload(msg, node)
public_key = _get_public_key(node)
# 0x01 prefix is not part of the actual public key, hence removed
public_key = node.public_key()[1:]
_validate_network(msg.transaction.network)
tx = nem_transaction_create_transfer(
msg.transaction.network,
@ -59,8 +59,18 @@ def _get_payload(msg: NEMSignTx, node) -> [bytes, bool]:
return payload, encrypted
def _get_public_key(node) -> bytes:
# 0x01 prefix is not part of the actual public key, hence removed
return node.public_key()[1:]
def _nem_encrypt(node, public_key: bytes, payload: bytes) -> bytes:
salt = random.bytes(helpers.NEM_SALT_SIZE)
iv = random.bytes(helpers.AES_BLOCK_SIZE)
encrypted = node.nem_encrypt(public_key, iv, salt, payload)
return iv + salt + encrypted
def _validate_network(network):
if network not in [NEM_NETWORK_MAINNET, NEM_NETWORK_TESTNET, NEM_NETWORK_MIJIN]:
raise ValueError('Invalid NEM network')