WIP tools/mprelp: When receiving binary data, loop until all data is read.

pi-anl · dpgeorge · commit 7d80767fe020 · 2019-10-18T13:16:28.000+11:00
Flush input buffer when a command begins
diff --git a/tools/mprepl.py b/tools/mprepl.py
@@ -86,6 +86,8 @@ def rdinto(self, buf):
         return self.fin.readinto(buf)
     def begin(self, type):
         micropython.kbd_intr(-1)
+        while(self.fin.any()):
+            self.fin.read(1)
         self.fout.write(bytearray([0x18, type]))
     def end(self):
         micropython.kbd_intr(3)
@@ -103,10 +105,19 @@ def wr_int32(self, i):
         self.fout.write(struct.pack('<i', i))
     def rd_bytes(self):
         n = struct.unpack('<H', self.rd(2))[0]
-        return self.rd(n)
+        buf = bytearray(n)
+        mv = memoryview(buf)
+        r = 0
+        while r<n:
+            r += self.rdinto(mv[r:])
+        return buf
     def rd_bytes_into(self, buf):
         n = struct.unpack('<H', self.rd(2))[0]
-        return self.rdinto(memoryview(buf)[:n])
+        mv = memoryview(buf)
+        r = 0
+        while r < n:
+            r += self.rdinto(mv[r:])
+        return r
     def wr_bytes(self, b):
         self.fout.write(struct.pack('<H', len(b)))
         self.fout.write(b)
