mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-12-23 14:58:09 +00:00
133 lines
3.6 KiB
Python
Executable File
133 lines
3.6 KiB
Python
Executable File
#!/usr/bin/env python2
|
|
import json
|
|
import os
|
|
|
|
import collections
|
|
|
|
HEADER_TEMPLATE = """
|
|
// This file is automatically generated by coin_info.py -- DO NOT EDIT!
|
|
|
|
#ifndef __COIN_INFO_H__
|
|
#define __COIN_INFO_H__
|
|
|
|
#include "coins.h"
|
|
|
|
#if DEBUG_LINK
|
|
#define COINS_COUNT ({stable} + {debug})
|
|
#else
|
|
#define COINS_COUNT ({stable})
|
|
#endif
|
|
|
|
extern const CoinInfo coins[COINS_COUNT];
|
|
|
|
#endif
|
|
""".lstrip()
|
|
|
|
CODE_TEMPLATE = """
|
|
// This file is automatically generated by coin_info.py -- DO NOT EDIT!
|
|
|
|
#include "coins.h"
|
|
|
|
#include "curves.h"
|
|
#include "secp256k1.h"
|
|
|
|
const CoinInfo coins[COINS_COUNT] = {{
|
|
{stable}
|
|
#if DEBUG_LINK
|
|
{debug}
|
|
#endif
|
|
}};
|
|
""".lstrip()
|
|
|
|
|
|
def format_bool(value):
|
|
if value:
|
|
return "true"
|
|
else:
|
|
return "false"
|
|
|
|
|
|
def format_number(value):
|
|
if value is None:
|
|
value = 0
|
|
return str(value)
|
|
|
|
|
|
def format_string(value):
|
|
if value is None:
|
|
return "NULL"
|
|
else:
|
|
return json.dumps(value)
|
|
|
|
|
|
def format_hex(value):
|
|
if value is None:
|
|
value = "0"
|
|
return "0x{:08x}".format(int(value, 16))
|
|
|
|
|
|
def prepend_varint(string):
|
|
assert len(string) < 253
|
|
|
|
varint = "\"\\x{:02x}\"".format(len(string))
|
|
return "{} {}".format(varint, format_string(string))
|
|
|
|
|
|
def coin_to_struct(coin):
|
|
return collections.OrderedDict((
|
|
("coin_name", format_string(coin["coin_name"])),
|
|
("coin_shortcut", format_string(coin["coin_shortcut"])),
|
|
("maxfee_kb", format_number(coin["maxfee_kb"])),
|
|
("signed_message_header", prepend_varint(coin["signed_message_header"])), # noqa: E501
|
|
("has_address_type", format_bool(coin["address_type"] is not None)), # noqa: E501
|
|
("has_address_type_p2sh", format_bool(coin["address_type_p2sh"] is not None)), # noqa: E501
|
|
("has_segwit", format_bool(coin["segwit"])),
|
|
("has_forkid", format_bool(coin["forkid"] is not None)),
|
|
("force_bip143", format_bool(coin["force_bip143"])),
|
|
("decred", format_bool(coin["decred"])),
|
|
("address_type", format_number(coin["address_type"])),
|
|
("address_type_p2sh", format_number(coin["address_type_p2sh"])),
|
|
("xpub_magic", format_hex(coin["xpub_magic"])),
|
|
("xprv_magic", format_hex(coin["xprv_magic"])),
|
|
("forkid", format_number(coin["forkid"])),
|
|
("bech32_prefix", format_string(coin["bech32_prefix"])),
|
|
("coin_type", "({} | 0x80000000)".format(format_number(coin["bip44"]))), # noqa: E501
|
|
("curve_name", "{}_NAME".format(coin["curve_name"].upper())), # noqa: E501
|
|
("curve", "&{}_info".format(coin["curve_name"])),
|
|
))
|
|
|
|
|
|
def format_struct(struct):
|
|
return "{\n" + "\n".join(
|
|
"\t.{0} = {1},".format(member, value)
|
|
for member, value in struct.items()
|
|
) + "\n}"
|
|
|
|
|
|
def format_coin(coin):
|
|
return format_struct(coin_to_struct(coin))
|
|
|
|
|
|
def format_coins(coins):
|
|
return "\n".join("{},".format(format_coin(coin)) for coin in coins)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
os.chdir(os.path.abspath(os.path.dirname(__file__)))
|
|
|
|
coins = collections.defaultdict(list)
|
|
|
|
for coin in json.load(open("coins.json")):
|
|
firmware = coin["firmware"]
|
|
coins[firmware].append(coin)
|
|
|
|
with open("coin_info.h", "w+") as f:
|
|
f.write(HEADER_TEMPLATE.format(**{
|
|
k: format_number(len(v)) for k, v in coins.items()
|
|
}))
|
|
|
|
with open("coin_info.c", "w+") as f:
|
|
f.write(CODE_TEMPLATE.format(**{
|
|
k: format_coins(v) for k, v in coins.items()
|
|
}))
|