diff --git a/src/apps/nem/transaction.py b/src/apps/nem/transaction.py new file mode 100644 index 0000000000..90cac69f75 --- /dev/null +++ b/src/apps/nem/transaction.py @@ -0,0 +1,86 @@ + +from .helpers import * +from .writers import * + + +def nem_transaction_create_transfer(network: int, timestamp: int, signer_public_key: bytes, fee: int, deadline: int, + recipient: str, amount: int, payload: bytearray = None, encrypted: bool = False, + mosaics: int = 0) -> bytearray: + + tx = _nem_transaction_write_common(NEM_TRANSACTION_TYPE_TRANSFER, + _nem_get_version(network, mosaics), + timestamp, + signer_public_key, + fee, + deadline) + + write_bytes_with_length(tx, bytearray(recipient)) + write_uint64(tx, amount) + + if payload: + # payload + payload size (u32) + encryption flag (u32) + write_uint32(tx, len(payload) + 2 * 4) + if encrypted: + write_uint32(tx, 0x02) + else: + write_uint32(tx, 0x01) + write_bytes_with_length(tx, payload) + else: + write_uint32(tx, 0) + + if mosaics: + write_uint32(tx, mosaics) + + return tx + + +def nem_transaction_create_provision_namespace(network: int, timestamp: int, signer_public_key: bytes, fee: int, + deadline: int, namespace: str, parent: str, rental_sink: str, + rental_fee: int) -> bytearray: + + tx = _nem_transaction_write_common(NEM_TRANSACTION_TYPE_PROVISION_NAMESPACE, + _nem_get_version(network), + timestamp, + signer_public_key, + fee, + deadline) + + write_bytes_with_length(tx, bytearray(rental_sink)) + write_uint64(tx, rental_fee) + write_bytes_with_length(tx, bytearray(namespace)) + if parent: + write_bytes_with_length(tx, bytearray(parent)) + else: + write_uint32(tx, 0xffffffff) + + return tx + + +def nem_transaction_write_mosaic(w: bytearray, namespace: str, mosaic: str, quantity: int): + identifier_length = 4 + len(namespace) + 4 + len(mosaic) + # indentifier length (u32) + quantity (u64) + identifier size + write_uint32(w, 4 + 8 + identifier_length) + write_uint32(w, identifier_length) + write_bytes_with_length(w, bytearray(namespace)) + write_bytes_with_length(w, bytearray(mosaic)) + write_uint64(w, quantity) + + +def _nem_transaction_write_common(tx_type: int, version: int, timestamp: int, signer: bytes, fee: int, deadline: int)\ + -> bytearray: + ret = bytearray() + write_uint32(ret, tx_type) + write_uint32(ret, version) + write_uint32(ret, timestamp) + + write_bytes_with_length(ret, bytearray(signer)) + write_uint64(ret, fee) + write_uint32(ret, deadline) + + return ret + + +def _nem_get_version(network, mosaics=None) -> int: + if mosaics: + return network << 24 | 2 + return network << 24 | 1 diff --git a/tests/test_apps.nem.transaction.provision.py b/tests/test_apps.nem.transaction.provision.py index a2c6aa87f2..9ef13edf59 100644 --- a/tests/test_apps.nem.transaction.provision.py +++ b/tests/test_apps.nem.transaction.provision.py @@ -1,6 +1,6 @@ from common import * -from apps.nem.sign_tx import * +from apps.nem.transaction import * from trezor.crypto import hashlib diff --git a/tests/test_apps.nem.transaction.transfer.py b/tests/test_apps.nem.transaction.transfer.py index c5e7201c47..6bb9791c14 100644 --- a/tests/test_apps.nem.transaction.transfer.py +++ b/tests/test_apps.nem.transaction.transfer.py @@ -1,6 +1,6 @@ from common import * -from apps.nem.sign_tx import * +from apps.nem.transaction import * from trezor.crypto import hashlib