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

defs: refactor build_coins.py

Browse Source
This commit is contained in:
Pavol Rusnak 2018-05-30 11:51:25 +02:00
parent 714c33b142
commit aed1d7632f
No known key found for this signature in database
GPG Key ID: 91F3B339B9A02A3D
1 changed files with 29 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

49
defs/coins/tools/build_coins.py
View File

@ -4,17 +4,21 @@ import json
import glob import glob
import re import re
import os import os
from hashlib import sha256 import sys
from binascii import unhexlify
import ed25519 if '--defs' in sys.argv:
from PIL import Image from binascii import unhexlify
from hashlib import sha256
from trezorlib.protobuf import dump_message import ed25519
from coindef import CoinDef from PIL import Image
from trezorlib.protobuf import dump_message
from coindef import CoinDef
BUILD_DEFS = True
else:
BUILD_DEFS = False
def check_type(val, types, nullable=False, empty=False, regex=None, choice=None): def check_type(val, types, nullable=False, empty=False, regex=None, choice=None): # noqa:E501
# check nullable # check nullable
if nullable and val is None: if nullable and val is None:
return True return True
@ -47,9 +51,9 @@ def validate_coin(coin):
assert check_type(coin['coin_shortcut'], str, regex=r'^[A-Zt][A-Z][A-Z]+$') assert check_type(coin['coin_shortcut'], str, regex=r'^[A-Zt][A-Z][A-Z]+$')
assert check_type(coin['coin_label'], str, regex=r'^[A-Z]') assert check_type(coin['coin_label'], str, regex=r'^[A-Z]')
assert check_type(coin['website'], str, regex=r'^http.*[^/]$') assert check_type(coin['website'], str, regex=r'^http.*[^/]$')
assert check_type(coin['github'], str, regex=r'^https://github.com/.*[^/]$') assert check_type(coin['github'], str, regex=r'^https://github.com/.*[^/]$') # noqa:E501
assert check_type(coin['maintainer'], str) assert check_type(coin['maintainer'], str)
assert check_type(coin['curve_name'], str, choice=['secp256k1', 'secp256k1_decred', 'secp256k1_groestl']) assert check_type(coin['curve_name'], str, choice=['secp256k1', 'secp256k1_decred', 'secp256k1_groestl']) # noqa:E501
assert check_type(coin['address_type'], int) assert check_type(coin['address_type'], int)
assert check_type(coin['address_type_p2sh'], int) assert check_type(coin['address_type_p2sh'], int)
assert coin['address_type'] != coin['address_type_p2sh'] assert coin['address_type'] != coin['address_type_p2sh']
@ -66,7 +70,7 @@ def validate_coin(coin):
assert coin['xprv_magic'] != coin['xpub_magic_segwit_native'] assert coin['xprv_magic'] != coin['xpub_magic_segwit_native']
assert coin['xpub_magic'] != coin['xpub_magic_segwit_p2sh'] assert coin['xpub_magic'] != coin['xpub_magic_segwit_p2sh']
assert coin['xpub_magic'] != coin['xpub_magic_segwit_native'] assert coin['xpub_magic'] != coin['xpub_magic_segwit_native']
assert coin['xpub_magic_segwit_p2sh'] is None or coin['xpub_magic_segwit_native'] is None or coin['xpub_magic_segwit_p2sh'] != coin['xpub_magic_segwit_native'] assert coin['xpub_magic_segwit_p2sh'] is None or coin['xpub_magic_segwit_native'] is None or coin['xpub_magic_segwit_p2sh'] != coin['xpub_magic_segwit_native'] # noqa:E501
assert check_type(coin['slip44'], int) assert check_type(coin['slip44'], int)
assert check_type(coin['segwit'], bool) assert check_type(coin['segwit'], bool)
assert check_type(coin['decred'], bool) assert check_type(coin['decred'], bool)
@ -146,14 +150,18 @@ def convert_icon(icon):
def process_json(fn): def process_json(fn):
print(fn, end=' ... ') print(fn, end=' ... ')
j = json.load(open(fn)) j = json.load(open(fn))
i = Image.open(fn.replace('.json', '.png')) if BUILD_DEFS:
validate_coin(j) i = Image.open(fn.replace('.json', '.png'))
validate_icon(i) validate_coin(j)
ser = serialize(j, convert_icon(i)) validate_icon(i)
sig = sign(ser) ser = serialize(j, convert_icon(i))
definition = (sig + ser).hex() sig = sign(ser)
print('OK') definition = (sig + ser).hex()
return j, definition print('OK')
return j, definition
else:
print('OK')
return j, None
coins = {} coins = {}
@ -167,4 +175,5 @@ for fn in glob.glob(scriptdir + '/../*.json'):
json.dump(coins, open('coins.json', 'w'), indent=4, sort_keys=True) json.dump(coins, open('coins.json', 'w'), indent=4, sort_keys=True)
# json.dump(defs, open('coindefs.json', 'w'), indent=4, sort_keys=True) if BUILD_DEFS:
json.dump(defs, open('coindefs.json', 'w'), indent=4, sort_keys=True)