micropython protobuf support message print

Signed-off-by: Adam BZH <adam@onekey.so>

Author: 424778940z

core/src/apps/base.py

@@ -2,14 +2,13 @@
 
 import storage.cache
 import storage.device
-from trezor import config, loop, ui, utils, wire, workflow
+from trezor import config, loop, protobuf, ui, utils, wire, workflow
 from trezor.enums import MessageType
 from trezor.messages import Success, UnlockPath
 
 from . import workflow_handlers
 
 if TYPE_CHECKING:
-    from trezor import protobuf
     from trezor.messages import (
         Features,
         Initialize,

core/src/trezor/protobuf.py

@@ -41,3 +41,47 @@ def dump_message_buffer(msg: MessageType) -> bytearray:
     buffer = bytearray(encoded_length(msg))
     encode(buffer, msg)
     return buffer
+
+
+def print_message(msg: MessageType, indent: int = 0, drop_none: bool = False):
+    if indent == 0:
+        print(
+            f"========================= BEGIN --- Msg: {msg.MESSAGE_NAME} Type: {msg.MESSAGE_WIRE_TYPE} ========================="
+        )
+
+    try:
+        for key, value in msg.__dict__.items():
+            if value:
+                if type(value) is type(
+                    msg
+                ):  # can't figure out what exact type due to shitty ffi designs from upstream
+                    print(
+                        f"{'    ' * (indent)}{str(key)}:{value.MESSAGE_NAME}:{type(value)} = "
+                    )
+                    print_message(value, indent + 1)
+                else:
+                    print(f"{'    ' * (indent)}{str(key)}:{type(value)} = ", end="")
+                    if type(value) is not bytes:
+                        print(
+                            str(value)
+                        )  # have to use otherwise may corrupt the output
+                    else:
+                        print("".join(f"{x:02x}" for x in value))
+            elif not drop_none:
+                print(f"{'    ' * (indent)}{str(key)}:{type(value)} = {str(value)}")
+            else:
+                continue
+
+    except Exception as ex:
+        from traceback import print_exception
+
+        print(
+            f"Error while handling Msg: {msg.MESSAGE_NAME} Type: {msg.MESSAGE_WIRE_TYPE}"
+        )
+        print_exception(ex)
+        pass
+
+    if indent == 0:
+        print(
+            f"========================= END --- Msg: {msg.MESSAGE_NAME} Type: {msg.MESSAGE_WIRE_TYPE} ========================="
+        )

core/src/trezor/wire/__init__.py

@@ -341,15 +341,19 @@ async def write(self, msg: protobuf.MessageType) -> None:
         if __debug__:
             log.debug(
                 __name__,
-                "%s:%x write: %s",
+                "%s:%x write: %s %s",
                 self.iface.iface_num(),
                 self.sid,
                 msg.MESSAGE_NAME,
+                msg.MESSAGE_WIRE_TYPE,
             )
 
         # cannot write message without wire type
         assert msg.MESSAGE_WIRE_TYPE is not None
 
+        if __debug__:
+            protobuf.print_message(msg, drop_none=True)
+
         msg_size = protobuf.encoded_length(msg)
 
         if msg_size <= len(self.buffer):