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

nem: unit test for address

Browse Source
This commit is contained in:
Tomas Susanka 2018-03-14 14:53:11 +01:00 committed by Jan Pochyla
parent 33ac5de848
commit a18b398747
3 changed files with 100 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/apps/nem/get_address.py
View File

@ -3,12 +3,10 @@ from apps.common import seed
from trezor.messages.NEMAddress import NEMAddress from trezor.messages.NEMAddress import NEMAddress
from .helpers import * from .helpers import *
_NEM_CURVE = 'ed25519-keccak'
async def nem_get_address(ctx, msg): async def nem_get_address(ctx, msg):
network = nem_validate_network(msg.network) network = nem_validate_network(msg.network)
node = await seed.derive_node(ctx, msg.address_n, _NEM_CURVE) node = await seed.derive_node(ctx, msg.address_n, NEM_CURVE)
address = node.nem_address(network) address = node.nem_address(network)
if msg.show_display: if msg.show_display:

11
src/apps/nem/helpers.py
View File

@ -1,14 +1,15 @@
from micropython import const from micropython import const
_NEM_NETWORK_MAINNET = const(0x68) NEM_NETWORK_MAINNET = const(0x68)
_NEM_NETWORK_TESTNET = const(0x98) NEM_NETWORK_TESTNET = const(0x98)
_NEM_NETWORK_MIJIN = const(0x60) NEM_NETWORK_MIJIN = const(0x60)
NEM_CURVE = 'ed25519-keccak'
def nem_validate_network(network): def nem_validate_network(network):
if network in (_NEM_NETWORK_MAINNET, _NEM_NETWORK_TESTNET, _NEM_NETWORK_MIJIN): if network in (NEM_NETWORK_MAINNET, NEM_NETWORK_TESTNET, NEM_NETWORK_MIJIN):
return network return network
if network is None: if network is None:
return _NEM_NETWORK_MAINNET return NEM_NETWORK_MAINNET
raise ValueError('Invalid NEM network') raise ValueError('Invalid NEM network')

93
tests/test_apps.nem.address.py Normal file
View File

@ -0,0 +1,93 @@
from common import *
from ubinascii import unhexlify
from trezor.crypto import bip32
from apps.nem.helpers import NEM_NETWORK_MAINNET, NEM_CURVE
class TestNemAddress(unittest.TestCase):
# test vectors from https://raw.githubusercontent.com/NemProject/nem-test-vectors/master/1.test-keys.dat
def test_addresses(self):
# private key, public key, address
test_cases = [
('575dbb3062267eff57c970a336ebbc8fbcfe12c5bd3ed7bc11eb0481d7704ced',
'c5f54ba980fcbb657dbaaa42700539b207873e134d2375efeab5f1ab52f87844',
'NDD2CT6LQLIYQ56KIXI3ENTM6EK3D44P5JFXJ4R4'),
('5b0e3fa5d3b49a79022d7c1e121ba1cbbf4db5821f47ab8c708ef88defc29bfe',
'96eb2a145211b1b7ab5f0d4b14f8abc8d695c7aee31a3cfc2d4881313c68eea3',
'NABHFGE5ORQD3LE4O6B7JUFN47ECOFBFASC3SCAC'),
('738ba9bb9110aea8f15caa353aca5653b4bdfca1db9f34d0efed2ce1325aeeda',
'2d8425e4ca2d8926346c7a7ca39826acd881a8639e81bd68820409c6e30d142a',
'NAVOZX4HDVOAR4W6K4WJHWPD3MOFU27DFHC7KZOZ'),
('e8bf9bc0f35c12d8c8bf94dd3a8b5b4034f1063948e3cc5304e55e31aa4b95a6',
'4feed486777ed38e44c489c7c4e93a830e4c4a907fa19a174e630ef0f6ed0409',
'NBZ6JK5YOCU6UPSSZ5D3G27UHAPHTY5HDQMGE6TT'),
('c325ea529674396db5675939e7988883d59a5fc17a28ca977e3ba85370232a83',
'83ee32e4e145024d29bca54f71fa335a98b3e68283f1a3099c4d4ae113b53e54',
'NCQW2P5DNZ5BBXQVGS367DQ4AHC3RXOEVGRCLY6V'),
('a811cb7a80a7227ae61f6da536534ee3c2744e3c7e4b85f3e0df3c6a9c5613df',
'6d34c04f3a0e42f0c3c6f50e475ae018cfa2f56df58c481ad4300424a6270cbb',
'NA5IG3XFXZHIPJ5QLKX2FBJPEZYPMBPPK2ZRC3EH'),
('9c66de1ec77f4dfaaebdf9c8bc599ca7e8e6f0bc71390ffee2c9dd3f3619242a',
'a8fefd72a3b833dc7c7ed7d57ed86906dac22f88f1f4331873eb2da3152a3e77',
'NAABHVFJDBM74XMJJ52R7QN2MTTG2ZUXPQS62QZ7'),
('c56bc16ecf727878c15e24f4ae68569600ac7b251218a44ef50ce54175776edc',
'c92f761e6d83d20068fd46fe4bd5b97f4c6ba05d23180679b718d1f3e4fb066e',
'NCLK3OLMHR3F2E3KSBUIZ4K5PNWUDN37MLSJBJZP'),
('9dd73599283882fa1561ddfc9be5830b5dd453c90465d3fe5eeb646a3606374e',
'eaf16a4833e59370a04ccd5c63395058de34877b48c17174c71db5ed37b537ed',
'ND3AHW4VTI5R5QE5V44KIGPRU5FBJ5AFUCJXOY5H'),
('d9639dc6f49dad02a42fd8c217f1b1b4f8ce31ccd770388b645e639c72ff24fa',
'0f74a2f537cd9c986df018994dde75bdeee05e35eb9fe27adf506ca8475064f7',
'NCTZ4YAP43ONK3UYTASQVNDMBO24ZHJE65F3QPYE'),
('efc1992cd50b70ca55ac12c07aa5d026a8b78ffe28a7dbffc9228b26e02c38c1',
'2ebff201255f6cf948c78f528658b99a7c13ac791942fa22d59af610558111f5',
'NDQ2TMCMXBSFPZQPE2YKH6XLC24HD6LUMN6Z4GIC'),
('143a815e92e43f3ed1a921ee48cd143931b88b7c3d8e1e981f743c2a5be3c5ba',
'419ed11d48730e4ae2c93f0ea4df853b8d578713a36dab227517cf965861af4e',
'NA32IDDW2C53BDSBJNFL3Z6UU3J5CJZJMCZDXCF4'),
('bc1a082f5ac6fdd3a83ade211e5986ac0551bad6c7da96727ec744e5df963e2a',
'a160e6f9112233a7ce94202ed7a4443e1dac444b5095f9fecbb965fba3f92cac',
'NADUCEQLC3FTGB25GTA5HOUTB53CBVQNVOIP7NTJ'),
('4e47b4c6f4c7886e49ec109c61f4af5cfbb1637283218941d55a7f9fe1053f72',
'fbb91b16df828e21a9802980a44fc757c588bc1382a4cea429d6fa2ae0333f56',
'NBAF3BFLLPWH33MYE6VUPP5T6DQBZBKIDEQKZQOE'),
('efc4389da48ce49f85365cfa578c746530e9eac42db1b64ec346119b1becd347',
'2232f24dda0f2ded3ecd831210d4e8521a096b50cadd5a34f3f7083374e1ec12',
'NBOGTK2I2ATOGGD7ZFJHROG5MWL7XCKAUKSWIVSA'),
('bdba57c78ca7da16a3360efd13f06276284db8c40351de7fcd38ba0c35ac754d',
'c334c6c0dad5aaa2a0d0fb4c6032cb6a0edd96bf61125b5ea9062d5a00ee0eee',
'NCLERTEFYXKLK7RA4MVACEFMXMK3P7QMWTM7FBW2'),
('20694c1ec3c4a311bcdb29ed2edc428f6d4f9a4c429ad6a5bf3222084e35695f',
'518c4de412efa93de06a55947d11f697639443916ec8fcf04ebc3e6d17d0bd93',
'NB5V4BPIJHXVONO7UGMJDPFARMFA73BOBNOOYCOV'),
('e0d4f3760ac107b33c22c2cac24ab2f520b282684f5f66a4212ff95d926323ce',
'b3d16f4ead9de67c290144da535a0ed2504b03c05e5f1ceb8c7863762f786857',
'NC4PBAO5TPCAVQKBVOC4F6DMZP3CFSQBU46PSKBD'),
('efa9afc617412093c9c7a7c211a5332dd556f941e1a88c494ec860608610eea2',
'7e7716e4cebceb731d6f1fd28676f34888e9a0000fcfa1471db1c616c2ddf559',
'NCFW2LPXIWLBWAQN2QVIWEOD7IVDO3HQBD2OU56K'),
('d98499b3db61944684ce06a91735af4e14105338473fcf6ebe2b0bcada3dfd21',
'114171230ad6f8522a000cdc73fbc5c733b30bb71f2b146ccbdf34499f79a810',
'NCUKWDY3J3THKQHAKOK5ALF6ANJQABZHCH7VN6DP')
]
for test in test_cases:
private_key = bytearray(reversed(unhexlify(test[0])))
node = bip32.HDNode(
depth=0,
fingerprint=0,
child_num=0,
chain_code=bytearray(32),
private_key=private_key,
curve_name=NEM_CURVE
)
self.assertEqual(node.nem_address(NEM_NETWORK_MAINNET), test[2])
# public key is prepended with 1, removing
self.assertEqual(node.public_key()[1:], unhexlify(test[1]))
if __name__ == '__main__':
unittest.main()