mirror of
https://github.com/trezor/trezor-firmware.git
synced 2026-03-05 06:55:24 +01:00
48 lines
1.4 KiB
Python
Executable File
48 lines
1.4 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
import os
|
|
import re
|
|
import sys
|
|
from glob import glob
|
|
|
|
error = False
|
|
|
|
MYDIR = os.path.dirname(__file__)
|
|
|
|
EXPECTED_PREFIX_RE = re.compile(r"messages-(\w+)(?:-.*)?\.proto")
|
|
|
|
WITHOUT_PREFIX = [
|
|
"Bitcoin",
|
|
"Bootloader",
|
|
"Common",
|
|
"Crypto",
|
|
"Definitions",
|
|
"Management",
|
|
]
|
|
# Checking all protobuf files for their `enum` and `message` declarations
|
|
# and making sure their names start with expected prefix
|
|
for fn in sorted(glob(os.path.join(MYDIR, "messages-*.proto"))):
|
|
with open(fn, "rt") as f:
|
|
filename = EXPECTED_PREFIX_RE.search(fn)
|
|
assert filename
|
|
prefix = filename.group(1).capitalize()
|
|
if prefix in WITHOUT_PREFIX:
|
|
continue
|
|
if prefix == "Nem":
|
|
prefix = "NEM"
|
|
elif prefix == "Webauthn":
|
|
prefix = "WebAuthn"
|
|
for line in f:
|
|
line = line.strip().split(" ")
|
|
if line[0] not in ("enum", "message"):
|
|
continue
|
|
expected_prefixes = (prefix, f"Debug{prefix}")
|
|
if not line[1].startswith(expected_prefixes):
|
|
print("ERROR: protobuf message does not start with expected prefix")
|
|
print(f" file: {fn}")
|
|
print(f" message: {line[1]}")
|
|
print(f" expected prefixes: {expected_prefixes}")
|
|
error = True
|
|
|
|
if error:
|
|
sys.exit(1)
|