trezor-firmware/common/protob/check.py

#!/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)