| OLD | NEW |
| 1 # Copyright 2013 The Chromium Authors. All rights reserved. | 1 # Copyright 2013 The Chromium Authors. All rights reserved. |
| 2 # Use of this source code is governed by a BSD-style license that can be | 2 # Use of this source code is governed by a BSD-style license that can be |
| 3 # found in the LICENSE file. | 3 # found in the LICENSE file. |
| 4 | 4 |
| 5 """This module wraps Android's adb tool. | 5 """This module wraps Android's adb tool. |
| 6 | 6 |
| 7 This is a thin wrapper around the adb interface. Any additional complexity | 7 This is a thin wrapper around the adb interface. Any additional complexity |
| 8 should be delegated to a higher level (ex. DeviceUtils). | 8 should be delegated to a higher level (ex. DeviceUtils). |
| 9 """ | 9 """ |
| 10 | 10 |
| 11 import collections | 11 import collections |
| 12 import errno | 12 import errno |
| 13 import logging | 13 import logging |
| 14 import os | 14 import os |
| 15 import re |
| 15 | 16 |
| 16 from pylib import cmd_helper | 17 from pylib import cmd_helper |
| 17 from pylib import constants | 18 from pylib import constants |
| 18 from pylib.device import decorators | 19 from pylib.device import decorators |
| 19 from pylib.device import device_errors | 20 from pylib.device import device_errors |
| 20 from pylib.device import device_filter | |
| 21 from pylib.utils import timeout_retry | 21 from pylib.utils import timeout_retry |
| 22 | 22 |
| 23 | 23 |
| 24 _DEFAULT_TIMEOUT = 30 | 24 _DEFAULT_TIMEOUT = 30 |
| 25 _DEFAULT_RETRIES = 2 | 25 _DEFAULT_RETRIES = 2 |
| 26 | 26 |
| 27 _EMULATOR_RE = re.compile(r'^emulator-[0-9]+$') |
| 28 |
| 29 _READY_STATE = 'device' |
| 30 |
| 27 | 31 |
| 28 def _VerifyLocalFileExists(path): | 32 def _VerifyLocalFileExists(path): |
| 29 """Verifies a local file exists. | 33 """Verifies a local file exists. |
| 30 | 34 |
| 31 Args: | 35 Args: |
| 32 path: Path to the local file. | 36 path: Path to the local file. |
| 33 | 37 |
| 34 Raises: | 38 Raises: |
| 35 IOError: If the file doesn't exist. | 39 IOError: If the file doesn't exist. |
| 36 """ | 40 """ |
| (...skipping 116 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 153 def KillServer(cls, timeout=_DEFAULT_TIMEOUT, retries=_DEFAULT_RETRIES): | 157 def KillServer(cls, timeout=_DEFAULT_TIMEOUT, retries=_DEFAULT_RETRIES): |
| 154 cls._RunAdbCmd(['kill-server'], timeout=timeout, retries=retries) | 158 cls._RunAdbCmd(['kill-server'], timeout=timeout, retries=retries) |
| 155 | 159 |
| 156 @classmethod | 160 @classmethod |
| 157 def StartServer(cls, timeout=_DEFAULT_TIMEOUT, retries=_DEFAULT_RETRIES): | 161 def StartServer(cls, timeout=_DEFAULT_TIMEOUT, retries=_DEFAULT_RETRIES): |
| 158 # CPU affinity is used to reduce adb instability http://crbug.com/268450 | 162 # CPU affinity is used to reduce adb instability http://crbug.com/268450 |
| 159 cls._RunAdbCmd(['start-server'], timeout=timeout, retries=retries, | 163 cls._RunAdbCmd(['start-server'], timeout=timeout, retries=retries, |
| 160 cpu_affinity=0) | 164 cpu_affinity=0) |
| 161 | 165 |
| 162 @classmethod | 166 @classmethod |
| 163 def GetDevices(cls, filters=None, timeout=_DEFAULT_TIMEOUT, | 167 def GetDevices(cls, timeout=_DEFAULT_TIMEOUT, retries=_DEFAULT_RETRIES): |
| 164 retries=_DEFAULT_RETRIES): | |
| 165 """DEPRECATED. Refer to Devices(...) below.""" | 168 """DEPRECATED. Refer to Devices(...) below.""" |
| 166 # TODO(jbudorick): Remove this function once no more clients are using it. | 169 # TODO(jbudorick): Remove this function once no more clients are using it. |
| 167 return cls.Devices(filters=filters, timeout=timeout, retries=retries) | 170 return cls.Devices(timeout=timeout, retries=retries) |
| 168 | 171 |
| 169 @classmethod | 172 @classmethod |
| 170 def Devices(cls, filters=None, timeout=_DEFAULT_TIMEOUT, | 173 def Devices(cls, is_ready=True, timeout=_DEFAULT_TIMEOUT, |
| 171 retries=_DEFAULT_RETRIES): | 174 retries=_DEFAULT_RETRIES): |
| 172 """Get the list of active attached devices. | 175 """Get the list of active attached devices. |
| 173 | 176 |
| 174 Args: | 177 Args: |
| 175 filters: (optional) A list of binary functions that take an AdbWrapper | |
| 176 instance and a string description. Any device for which all provided | |
| 177 filter functions do not return True will not be included in the | |
| 178 returned list. | |
| 179 timeout: (optional) Timeout per try in seconds. | 178 timeout: (optional) Timeout per try in seconds. |
| 180 retries: (optional) Number of retries to attempt. | 179 retries: (optional) Number of retries to attempt. |
| 181 | 180 |
| 182 Yields: | 181 Yields: |
| 183 AdbWrapper instances. | 182 AdbWrapper instances. |
| 184 """ | 183 """ |
| 185 output = cls._RunAdbCmd(['devices'], timeout=timeout, retries=retries) | 184 output = cls._RunAdbCmd(['devices'], timeout=timeout, retries=retries) |
| 186 lines = (line.split() for line in output.splitlines()) | 185 lines = (line.split() for line in output.splitlines()) |
| 187 devices = (AdbWrapper(line[0]) for line in lines if len(line) == 2) | 186 return [AdbWrapper(line[0]) for line in lines |
| 188 | 187 if len(line) == 2 and (not is_ready or line[1] == _READY_STATE)] |
| 189 def matches_all_filters(device): | |
| 190 for f in filters or (): | |
| 191 if not f(device): | |
| 192 logging.info('Device %s failed filter %s', device.GetDeviceSerial(), | |
| 193 f.__name__) | |
| 194 return False | |
| 195 return True | |
| 196 | |
| 197 return [d for d in devices if matches_all_filters(d)] | |
| 198 | 188 |
| 199 def GetDeviceSerial(self): | 189 def GetDeviceSerial(self): |
| 200 """Gets the device serial number associated with this object. | 190 """Gets the device serial number associated with this object. |
| 201 | 191 |
| 202 Returns: | 192 Returns: |
| 203 Device serial number as a string. | 193 Device serial number as a string. |
| 204 """ | 194 """ |
| 205 return self._device_serial | 195 return self._device_serial |
| 206 | 196 |
| 207 def Push(self, local, remote, timeout=60*5, retries=_DEFAULT_RETRIES): | 197 def Push(self, local, remote, timeout=60*5, retries=_DEFAULT_RETRIES): |
| (...skipping 336 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 544 """Restarts the adbd daemon with root permissions, if possible. | 534 """Restarts the adbd daemon with root permissions, if possible. |
| 545 | 535 |
| 546 Args: | 536 Args: |
| 547 timeout: (optional) Timeout per try in seconds. | 537 timeout: (optional) Timeout per try in seconds. |
| 548 retries: (optional) Number of retries to attempt. | 538 retries: (optional) Number of retries to attempt. |
| 549 """ | 539 """ |
| 550 output = self._RunDeviceAdbCmd(['root'], timeout, retries) | 540 output = self._RunDeviceAdbCmd(['root'], timeout, retries) |
| 551 if 'cannot' in output: | 541 if 'cannot' in output: |
| 552 raise device_errors.AdbCommandFailedError( | 542 raise device_errors.AdbCommandFailedError( |
| 553 ['root'], output, device_serial=self._device_serial) | 543 ['root'], output, device_serial=self._device_serial) |
| 544 |
| 545 @property |
| 546 def is_emulator(self): |
| 547 return _EMULATOR_RE.match(self._device_serial) |
| 548 |
| 549 @property |
| 550 def is_ready(self): |
| 551 try: |
| 552 return self.GetState() == _READY_STATE |
| 553 except device_errors.CommandFailedError: |
| 554 return False |
| 555 |
| OLD | NEW |