1
0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-12-12 09:28:10 +00:00
trezor-firmware/firmware/protob/messages_map.py
2018-04-02 21:47:15 +02:00

76 lines
2.3 KiB
Python
Executable File

#!/usr/bin/env python
from collections import defaultdict
from messages_pb2 import MessageType
from types_pb2 import wire_in, wire_out
from types_pb2 import wire_debug_in, wire_debug_out
from types_pb2 import wire_bootloader, wire_tiny
# len("MessageType_MessageType_") - len("_fields") == 17
TEMPLATE = "\t{{ {type} {dir} {msg_id:46} {fields:29} {process_func} }},"
LABELS = {
wire_in: "in messages",
wire_out: "out messages",
wire_debug_in: "debug in messages",
wire_debug_out: "debug out messages",
}
def handle_message(message, extension):
name = message.name
short_name = name.split("MessageType_", 1).pop()
assert(short_name != name)
interface = "d" if extension in (wire_debug_in, wire_debug_out) else "n"
direction = "i" if extension in (wire_in, wire_debug_in) else "o"
options = message.GetOptions()
bootloader = options.Extensions[wire_bootloader]
tiny = options.Extensions[wire_tiny] and direction == "i"
if getattr(options, 'deprecated', None):
return '\t// Message %s is deprecated' % short_name
if bootloader:
return '\t// Message %s is used in bootloader mode only' % short_name
if tiny:
return '\t// Message %s is used in tiny mode' % short_name
if direction == "i":
process_func = "(void (*)(void *)) fsm_msg%s" % short_name
else:
process_func = "0"
return TEMPLATE.format(
type="'%c'," % interface,
dir="'%c'," % direction,
msg_id="MessageType_%s," % name,
fields="%s_fields," % short_name,
process_func=process_func,
)
print("\t// This file is automatically generated"
"by messages_map.py -- DO NOT EDIT!")
messages = defaultdict(list)
for message in MessageType.DESCRIPTOR.values:
extensions = message.GetOptions().Extensions
for extension in (wire_in, wire_out, wire_debug_in, wire_debug_out):
if extensions[extension]:
messages[extension].append(message)
for extension in (wire_in, wire_out, wire_debug_in, wire_debug_out):
if extension == wire_debug_in:
print("\n#if DEBUG_LINK")
print("\n\t// {label}\n".format(label=LABELS[extension]))
for message in messages[extension]:
print(handle_message(message, extension))
if extension == wire_debug_out:
print("\n#endif")