mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-12-30 01:58:11 +00:00
38 lines
1.2 KiB
Python
Executable File
38 lines
1.2 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")
|
|
|
|
# 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:
|
|
prefix = EXPECTED_PREFIX_RE.search(fn).group(1).capitalize()
|
|
if prefix in ("Bitcoin", "Bootloader", "Common", "Crypto", "Management"):
|
|
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)
|