1
0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-11-09 17:10:17 +00:00
trezor-firmware/firmware/coin_info.py
2018-04-03 18:40:41 +02:00

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()
}))