mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-12-26 00:08:10 +00:00
9fc5bb546b
Changes many fields to required -- as far as we were able to figure out, signing would fail if these fields aren't provided anyway, so this should not pose a compatibility problem. Co-authored-by: matejcik <ja@matejcik.cz>
44 lines
1.3 KiB
Python
44 lines
1.3 KiB
Python
from typing import TYPE_CHECKING
|
|
|
|
import trezorproto
|
|
|
|
decode = trezorproto.decode
|
|
encode = trezorproto.encode
|
|
encoded_length = trezorproto.encoded_length
|
|
type_for_name = trezorproto.type_for_name
|
|
type_for_wire = trezorproto.type_for_wire
|
|
|
|
if TYPE_CHECKING:
|
|
# XXX
|
|
# Note that MessageType "subclasses" are not true subclasses, but instead instances
|
|
# of the built-in metaclass MsgDef. MessageType instances are in fact instances of
|
|
# the built-in type Msg. That is why isinstance checks do not work, and instead the
|
|
# MessageTypeSubclass.is_type_of() method must be used.
|
|
from typing import TypeGuard, TypeVar
|
|
|
|
T = TypeVar("T", bound="MessageType")
|
|
|
|
class MessageType:
|
|
MESSAGE_NAME: str = "MessageType"
|
|
MESSAGE_WIRE_TYPE: int | None = None
|
|
|
|
@classmethod
|
|
def is_type_of(cls: type[T], msg: "MessageType") -> TypeGuard[T]:
|
|
"""Identify if the provided message belongs to this type."""
|
|
raise NotImplementedError
|
|
|
|
|
|
def load_message_buffer(
|
|
buffer: bytes,
|
|
msg_wire_type: int,
|
|
experimental_enabled: bool = True,
|
|
) -> MessageType:
|
|
msg_type = type_for_wire(msg_wire_type)
|
|
return decode(buffer, msg_type, experimental_enabled)
|
|
|
|
|
|
def dump_message_buffer(msg: MessageType) -> bytearray:
|
|
buffer = bytearray(encoded_length(msg))
|
|
encode(buffer, msg)
|
|
return buffer
|