experiments/mtp.py: Update to use client

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

Author: marcan

proxyclient/experiments/mtp.py

@@ -11,37 +11,113 @@
 from m1n1.hw.dockchannel import DockChannel
 from m1n1.fw.smc import SMCClient, SMCError
 from m1n1.shell import run_shell
+from m1n1.fw.mtp import *
 
 from construct import *
 
+mon = RegMonitor(u)
+
+#mon.add(0x23b700000, 0x10000)
+mon.add(0x23d28c000, 0x4000)
+mon.poll()
+mon.poll()
+mon.poll()
+mon.add(0x24e400000, 0x4000)
+mon.add(0x24e808000, 0x14000)
+
+smc_addr = u.adt["arm-io/smc"].get_reg(0)[0]
+smc = SMCClient(u, smc_addr)
+smc.start()
+smc.start_ep(0x20)
+smc.verbose = 0
+
 p.dapf_init_all()
 
 dart = DART8110.from_adt(u, "/arm-io/dart-mtp", iova_range=(0x8000, 0x100000))
 
 dart.regs.TCR[1].set(BYPASS_DAPF=0, BYPASS_DART=0, TRANSLATE_ENABLE=1)
 
-mtp_addr = u.adt["/arm-io/mtp"].get_reg(0)[0]
-mtp = StandardASC(u, mtp_addr, dart, stream=1)
-mtp.verbose = 3
-mtp.allow_phys = True
-mtp.start()
-
 irq_base = u.adt["/arm-io/dockchannel-mtp"].get_reg(1)[0]
 fifo_base = u.adt["/arm-io/dockchannel-mtp"].get_reg(2)[0]
 dc = DockChannel(u, irq_base, fifo_base, 1)
 
-for i in range(128):
-    mtp.work()
+node = u.adt["/arm-io/dockchannel-mtp/mtp-transport"]
 
-dc.write(bytes.fromhex("08110c00010000008000020000000000b4010000c2ecf1ff"))
-dc.write(bytes.fromhex("08110c00020000008000020000000000b4020000c1ebf1ff"))
+while dc.rx_count:
+    dc.read(dc.rx_count)
+
+mtp_addr = u.adt["/arm-io/mtp"].get_reg(0)[0]
+mtp = StandardASC(u, mtp_addr, dart, stream=1)
+mtp.allow_phys = True
+print("pre start")
+mon.poll()
+mtp.start()
+print("started")
+mon.poll()
+print("ok")
 
 def poll():
-    v = dc.read_all()
-    if v:
-        chexdump(v)
     mtp.work()
+    mp.work_pending()
+
+# 0x40: device reset
+# 0x42:
+
+# 0 -> mbox?
+# 2 -> dapf?
+# 3 -> dart?
+# 3 -> dockchannel?
+# 5 -> mbox?
+def reset(i):
+    reg = 0x23d28c000 + i*8
+    p.set32(reg, 1<<10)
+    p.set32(reg, 1<<31)
+    p.clear32(reg, 1<<31)
+    p.clear32(reg, 1<<10)
+
+try:
+
+    mp = MTPProtocol(u, node, mtp, dc, smc)
+
+    mp.wait_init("keyboard")
+    mp.wait_init("multi_touch")
+
+    mtp.stop()
+    mtp.start()
+    mon.poll()
+
+    #for i in range(256):
+        #if i in (0x40, 0x42):
+            #continue
+        #m = UnkDeviceControlMsg()
+        #m.command = i
+        #for args in (b"", b"\x00", b"\x01", b"\x02",
+                     #b"\x01\x00", b"\x01\x01", b"\x01\x02",
+                     #b"\x00\x01", b"\x00\x02", b"\x00\x00",
+                     #b"\x00\x00\x00",
+                     #b"\x00\x00\x00\x00",
+                     #b"\x00\x00\x00\x00\x00",
+                     #b"\x00\x00\x00\x00\x00\x00",
+                     #b"\x00\x00\x00\x00\x00\x00\x00",
+                     #b"\x00\x00\x00\x00\x00\x00\x00\x00",):
+            #m.args = args
+            #print(f"{m.command:#x} {m.args.hex()}")
+            #mp.comm.device_control(m)
+
+    mon.poll()
+    mtp.stop()
+    mon.poll()
+    mtp.start()
+
+    mon.poll()
+    mtp.stop(1)
+    #reset(1)
+    #p.dapf_init_all()
+
+    #mtp.boot()
 
-mtp.stop()
+    run_shell(locals(), poll_func=poll)
 
-#run_shell(locals(), poll_func=poll)
+finally:
+    #mtp.stop()
+    p.reboot()