micropython protobuf support message print

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

Author: 424778940z

core/src/trezor/protobuf.py

@@ -41,3 +41,40 @@ 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) == 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) != bytes:
+                        print(str(value)) # have to use otherwise may corrupt the output
+                    else:
+                        print(''.join('{:02x}'.format(x) for x in value))
+            elif not drop_none:
+                print(f"{'    '* (indent)}{str(key)}:{type(value)} = {str(value)}")
+            else:
+                continue
+
+    except Exception as ex:
+        import sys
+
+        print(f"Error while handling Msg: {msg.MESSAGE_NAME} Type: {msg.MESSAGE_WIRE_TYPE}")
+        sys.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,18 @@ 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_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):