m1n1.fw.mtp: Fix parsing of init messages

Signed-off-by: Hector Martin <[email protected]>

Author: marcan

proxyclient/m1n1/fw/mtp.py

@@ -20,13 +20,13 @@ class GPIOInit(ConstructClass):
 class InitBlock(ConstructClass):
     subcon = Struct(
         "type" / Int16ul,
-        "subtype" / Int16ul,
         "length" / Int16ul,
         "payload" / FixedSized(this.length,
                    Switch(this.type, {
                        0: HIDDescriptor,
                        1: GPIOInit,
-                       2: Bytes(0),
+                       2: GreedyBytes, # Unknown 6 bytes
+                       7: GreedyBytes, # Device name
                    }, default=GreedyBytes))
     )
 
@@ -37,6 +37,7 @@ class InitMsg(ConstructClass):
         "unk" / Const(0x00, Int8ul),
         "device_id" / Int8ul,
         "device_name" / PaddedString(16, "ascii"),
+        "more_packets" / Int16ul,
         "msg" / RepeatUntil(lambda obj, lst, ctx: lst[-1].type == 2, InitBlock)
     )
 
@@ -275,6 +276,9 @@ def report(self, msg):
             ackmsg.unk = 0
             ackmsg.msg = msg
             self.device_control(ackmsg)
+        else:
+            self.log("Unknown message!")
+            print(msg)
 
     def ack(self, retcode, msg):
         msg = DeviceControlAck.parse(msg)