mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-12-23 23:08:14 +00:00
70 lines
1.8 KiB
Python
Executable File
70 lines
1.8 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
import json
|
|
import sys
|
|
from glob import glob
|
|
from hashlib import sha256
|
|
|
|
try:
|
|
opt = sys.argv[1]
|
|
except IndexError:
|
|
print("Usage: gen.py [core|mcu|check])")
|
|
sys.exit(1)
|
|
|
|
|
|
def c_bytes(h):
|
|
return "{ " + ", ".join(["0x%02x" % x for x in h]) + " }"
|
|
|
|
|
|
def gen_core(data):
|
|
print("_knownapps = {")
|
|
print(" # U2F")
|
|
for d in data:
|
|
if "u2f" in d:
|
|
url, label = d["u2f"], d["label"]
|
|
print(' "%s": {"label": "%s", "use_sign_count": True},' % (url, label))
|
|
print(" # WebAuthn")
|
|
for d in data:
|
|
if "webauthn" in d:
|
|
origin, label, use_sign_count = (
|
|
d["webauthn"],
|
|
d["label"],
|
|
d.get("use_sign_count", None),
|
|
)
|
|
if use_sign_count is None:
|
|
print(' "%s": {"label": "%s"},' % (origin, label))
|
|
else:
|
|
print(
|
|
' "%s": {"label": "%s", "use_sign_count": %s},'
|
|
% (origin, label, use_sign_count)
|
|
)
|
|
print("}")
|
|
|
|
|
|
def gen_mcu(data):
|
|
for d in data:
|
|
if "u2f" in d:
|
|
url, label = d["u2f"], d["label"]
|
|
h = sha256(url.encode()).digest()
|
|
print(
|
|
'\t{\n\t\t// U2F: %s\n\t\t%s,\n\t\t"%s"\n\t},'
|
|
% (url, c_bytes(h), label)
|
|
)
|
|
if "webauthn" in d:
|
|
origin, label = d["webauthn"], d["label"]
|
|
h = sha256(origin.encode()).digest()
|
|
print(
|
|
'\t{\n\t\t// WebAuthn: %s\n\t\t%s,\n\t\t"%s"\n\t},'
|
|
% (origin, c_bytes(h), label)
|
|
)
|
|
|
|
|
|
data = []
|
|
for fn in sorted(glob("apps/*.json")):
|
|
d = json.load(open(fn, "rt"))
|
|
data.append(d)
|
|
|
|
if opt == "core":
|
|
gen_core(data)
|
|
elif opt == "mcu":
|
|
gen_mcu(data)
|