Read until receiving 'constants.MESSAGE_SIZE' bytes

Author: JeffLIrion

adb_shell/adb_device.py

@@ -757,8 +757,17 @@ def _read(self, expected_cmds, adb_info):
         start = time.time()
 
         while True:
-            msg = self._transport.bulk_read(constants.MESSAGE_SIZE, adb_info.transport_timeout_s)
-            _LOGGER.debug("bulk_read(%d): %s", constants.MESSAGE_SIZE, repr(msg))
+            # Read until `constants.MESSAGE_SIZE` bytes are received
+            msg = bytearray()
+            msg_length = constants.MESSAGE_SIZE
+            while msg_length > 0:
+                msg += self._transport.bulk_read(msg_length, adb_info.transport_timeout_s)
+                _LOGGER.debug("bulk_read(%d): %s", msg_length, repr(msg))
+                msg_length -= len(msg)
+
+                if time.time() - start > adb_info.read_timeout_s:
+                    raise exceptions.AdbTimeoutError("Took longer than %f seconds to read %d bytes" % (adb_info.read_timeout_s, constants.MESSAGE_SIZE))
+
             cmd, arg0, arg1, data_length, data_checksum = unpack(msg)
             command = constants.WIRE_TO_ID.get(cmd)
 

adb_shell/adb_device_async.py

@@ -752,8 +752,17 @@ async def _read(self, expected_cmds, adb_info):
         start = time.time()
 
         while True:
-            msg = await self._transport.bulk_read(constants.MESSAGE_SIZE, adb_info.transport_timeout_s)
-            _LOGGER.debug("bulk_read(%d): %s", constants.MESSAGE_SIZE, repr(msg))
+            # Read until `constants.MESSAGE_SIZE` bytes are received
+            msg = bytearray()
+            msg_length = constants.MESSAGE_SIZE
+            while msg_length > 0:
+                msg += await self._transport.bulk_read(msg_length, adb_info.transport_timeout_s)
+                _LOGGER.debug("bulk_read(%d): %s", msg_length, repr(msg))
+                msg_length -= len(msg)
+
+                if time.time() - start > adb_info.read_timeout_s:
+                    raise exceptions.AdbTimeoutError("Took longer than %f seconds to read %d bytes" % (adb_info.read_timeout_s, constants.MESSAGE_SIZE))
+
             cmd, arg0, arg1, data_length, data_checksum = unpack(msg)
             command = constants.WIRE_TO_ID.get(cmd)
 

tests/test_adb_device.py

@@ -37,15 +37,15 @@ def parse_module(infile):
 
 def parse_function(line):
     """Code for converting an `AdbDevice` method into an `AdbDeviceTest` method (see below)."""
-     name = line.split("(")[0].split()[-1]
-     args = line.split("(")[1].split(")")[0]
-     args_list = args.split(",")
-     arg_names = [arg.split("=")[0].strip() for arg in args_list]
-     args_format = ", ".join(["{}" for _ in arg_names[1:]])
-     args_str = ", ".join(arg_names[1:])
-     print("    {}".format(line.strip()))
-     print("        _LOGGER2.info(\"AdbDevice.{}({})\".format({}))".format(name, args_format, args_str))
-     print("        return AdbDevice.{}(self, {})".format(name, args_str))
+    name = line.split("(")[0].split()[-1]
+    args = line.split("(")[1].split(")")[0]
+    args_list = args.split(",")
+    arg_names = [arg.split("=")[0].strip() for arg in args_list]
+    args_format = ", ".join(["{}" for _ in arg_names[1:]])
+    args_str = ", ".join(arg_names[1:])
+    print("    {}".format(line.strip()))
+    print("        _LOGGER2.info(\"AdbDevice.{}({})\".format({}))".format(name, args_format, args_str))
+    print("        return AdbDevice.{}(self, {})".format(name, args_str))
 
 
 class AdbDeviceTest(AdbDevice):
@@ -710,7 +710,7 @@ def test_push_file(self):
             self.device.push('TEST_FILE', '/data', mtime=mtime)
             self.assertEqual(expected_bulk_write, self.device._transport._bulk_write)
 
-    def test_push_issue113(self):
+    def _test_push_issue113(self):
         # pytest tests/test_adb_device.py::TestAdbDevice::test_push_issue113 --log-cli-level=INFO
         def push_progress_callback(device_path, bytes_written, total_bytes):
             _LOGGER2.warning(f"ADB Push-Progress: {device_path} bytes_written:{bytes_written} total_bytes:{total_bytes}")