| # Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| # pylint: disable=W0212 |
| |
| import fcntl |
| import logging |
| import os |
| import psutil |
| |
| from pylib import cmd_helper |
| from pylib import constants |
| from pylib import valgrind_tools |
| |
| # TODO(jbudorick) Remove once telemetry gets switched over. |
| import pylib.android_commands |
| import pylib.device.device_utils |
| |
| |
| def _GetProcessStartTime(pid): |
| return psutil.Process(pid).create_time |
| |
| |
| class _FileLock(object): |
| """With statement-aware implementation of a file lock. |
| |
| File locks are needed for cross-process synchronization when the |
| multiprocessing Python module is used. |
| """ |
| def __init__(self, path): |
| self._fd = -1 |
| self._path = path |
| |
| def __enter__(self): |
| self._fd = os.open(self._path, os.O_RDONLY | os.O_CREAT) |
| if self._fd < 0: |
| raise Exception('Could not open file %s for reading' % self._path) |
| fcntl.flock(self._fd, fcntl.LOCK_EX) |
| |
| def __exit__(self, _exception_type, _exception_value, traceback): |
| fcntl.flock(self._fd, fcntl.LOCK_UN) |
| os.close(self._fd) |
| |
| |
| class Forwarder(object): |
| """Thread-safe class to manage port forwards from the device to the host.""" |
| |
| _DEVICE_FORWARDER_FOLDER = (constants.TEST_EXECUTABLE_DIR + |
| '/forwarder/') |
| _DEVICE_FORWARDER_PATH = (constants.TEST_EXECUTABLE_DIR + |
| '/forwarder/device_forwarder') |
| _LOCK_PATH = '/tmp/chrome.forwarder.lock' |
| # Defined in host_forwarder_main.cc |
| _HOST_FORWARDER_LOG = '/tmp/host_forwarder_log' |
| |
| _instance = None |
| |
| @staticmethod |
| def Map(port_pairs, device, tool=None): |
| """Runs the forwarder. |
| |
| Args: |
| port_pairs: A list of tuples (device_port, host_port) to forward. Note |
| that you can specify 0 as a device_port, in which case a |
| port will by dynamically assigned on the device. You can |
| get the number of the assigned port using the |
| DevicePortForHostPort method. |
| device: A DeviceUtils instance. |
| tool: Tool class to use to get wrapper, if necessary, for executing the |
| forwarder (see valgrind_tools.py). |
| |
| Raises: |
| Exception on failure to forward the port. |
| """ |
| # TODO(jbudorick) Remove once telemetry gets switched over. |
| if isinstance(device, pylib.android_commands.AndroidCommands): |
| device = pylib.device.device_utils.DeviceUtils(device) |
| if not tool: |
| tool = valgrind_tools.CreateTool(None, device) |
| with _FileLock(Forwarder._LOCK_PATH): |
| instance = Forwarder._GetInstanceLocked(tool) |
| instance._InitDeviceLocked(device, tool) |
| |
| device_serial = str(device) |
| redirection_commands = [ |
| ['--adb=' + constants.GetAdbPath(), |
| '--serial-id=' + device_serial, |
| '--map', str(device_port), str(host_port)] |
| for device_port, host_port in port_pairs] |
| logging.info('Forwarding using commands: %s', redirection_commands) |
| |
| for redirection_command in redirection_commands: |
| try: |
| (exit_code, output) = cmd_helper.GetCmdStatusAndOutput( |
| [instance._host_forwarder_path] + redirection_command) |
| except OSError as e: |
| if e.errno == 2: |
| raise Exception('Unable to start host forwarder. Make sure you have' |
| ' built host_forwarder.') |
| else: raise |
| if exit_code != 0: |
| Forwarder._KillDeviceLocked(device, tool) |
| raise Exception('%s exited with %d:\n%s' % ( |
| instance._host_forwarder_path, exit_code, '\n'.join(output))) |
| tokens = output.split(':') |
| if len(tokens) != 2: |
| raise Exception('Unexpected host forwarder output "%s", ' |
| 'expected "device_port:host_port"' % output) |
| device_port = int(tokens[0]) |
| host_port = int(tokens[1]) |
| serial_with_port = (device_serial, device_port) |
| instance._device_to_host_port_map[serial_with_port] = host_port |
| instance._host_to_device_port_map[host_port] = serial_with_port |
| logging.info('Forwarding device port: %d to host port: %d.', |
| device_port, host_port) |
| |
| @staticmethod |
| def UnmapDevicePort(device_port, device): |
| """Unmaps a previously forwarded device port. |
| |
| Args: |
| device: A DeviceUtils instance. |
| device_port: A previously forwarded port (through Map()). |
| """ |
| # TODO(jbudorick) Remove once telemetry gets switched over. |
| if isinstance(device, pylib.android_commands.AndroidCommands): |
| device = pylib.device.device_utils.DeviceUtils(device) |
| with _FileLock(Forwarder._LOCK_PATH): |
| Forwarder._UnmapDevicePortLocked(device_port, device) |
| |
| @staticmethod |
| def UnmapAllDevicePorts(device): |
| """Unmaps all the previously forwarded ports for the provided device. |
| |
| Args: |
| device: A DeviceUtils instance. |
| port_pairs: A list of tuples (device_port, host_port) to unmap. |
| """ |
| # TODO(jbudorick) Remove once telemetry gets switched over. |
| if isinstance(device, pylib.android_commands.AndroidCommands): |
| device = pylib.device.device_utils.DeviceUtils(device) |
| with _FileLock(Forwarder._LOCK_PATH): |
| if not Forwarder._instance: |
| return |
| adb_serial = str(device) |
| if adb_serial not in Forwarder._instance._initialized_devices: |
| return |
| port_map = Forwarder._GetInstanceLocked( |
| None)._device_to_host_port_map |
| for (device_serial, device_port) in port_map.keys(): |
| if adb_serial == device_serial: |
| Forwarder._UnmapDevicePortLocked(device_port, device) |
| # There are no more ports mapped, kill the device_forwarder. |
| tool = valgrind_tools.CreateTool(None, device) |
| Forwarder._KillDeviceLocked(device, tool) |
| |
| @staticmethod |
| def DevicePortForHostPort(host_port): |
| """Returns the device port that corresponds to a given host port.""" |
| with _FileLock(Forwarder._LOCK_PATH): |
| (_device_serial, device_port) = Forwarder._GetInstanceLocked( |
| None)._host_to_device_port_map.get(host_port) |
| return device_port |
| |
| @staticmethod |
| def RemoveHostLog(): |
| if os.path.exists(Forwarder._HOST_FORWARDER_LOG): |
| os.unlink(Forwarder._HOST_FORWARDER_LOG) |
| |
| @staticmethod |
| def GetHostLog(): |
| if not os.path.exists(Forwarder._HOST_FORWARDER_LOG): |
| return '' |
| with file(Forwarder._HOST_FORWARDER_LOG, 'r') as f: |
| return f.read() |
| |
| @staticmethod |
| def _GetInstanceLocked(tool): |
| """Returns the singleton instance. |
| |
| Note that the global lock must be acquired before calling this method. |
| |
| Args: |
| tool: Tool class to use to get wrapper, if necessary, for executing the |
| forwarder (see valgrind_tools.py). |
| """ |
| if not Forwarder._instance: |
| Forwarder._instance = Forwarder(tool) |
| return Forwarder._instance |
| |
| def __init__(self, tool): |
| """Constructs a new instance of Forwarder. |
| |
| Note that Forwarder is a singleton therefore this constructor should be |
| called only once. |
| |
| Args: |
| tool: Tool class to use to get wrapper, if necessary, for executing the |
| forwarder (see valgrind_tools.py). |
| """ |
| assert not Forwarder._instance |
| self._tool = tool |
| self._initialized_devices = set() |
| self._device_to_host_port_map = dict() |
| self._host_to_device_port_map = dict() |
| self._host_forwarder_path = os.path.join( |
| constants.GetOutDirectory(), 'host_forwarder') |
| assert os.path.exists(self._host_forwarder_path), 'Please build forwarder2' |
| self._device_forwarder_path_on_host = os.path.join( |
| constants.GetOutDirectory(), 'forwarder_dist') |
| self._InitHostLocked() |
| |
| @staticmethod |
| def _UnmapDevicePortLocked(device_port, device): |
| """Internal method used by UnmapDevicePort(). |
| |
| Note that the global lock must be acquired before calling this method. |
| """ |
| instance = Forwarder._GetInstanceLocked(None) |
| serial = str(device) |
| serial_with_port = (serial, device_port) |
| if not serial_with_port in instance._device_to_host_port_map: |
| logging.error('Trying to unmap non-forwarded port %d' % device_port) |
| return |
| redirection_command = ['--adb=' + constants.GetAdbPath(), |
| '--serial-id=' + serial, |
| '--unmap', str(device_port)] |
| (exit_code, output) = cmd_helper.GetCmdStatusAndOutput( |
| [instance._host_forwarder_path] + redirection_command) |
| if exit_code != 0: |
| logging.error('%s exited with %d:\n%s' % ( |
| instance._host_forwarder_path, exit_code, '\n'.join(output))) |
| host_port = instance._device_to_host_port_map[serial_with_port] |
| del instance._device_to_host_port_map[serial_with_port] |
| del instance._host_to_device_port_map[host_port] |
| |
| @staticmethod |
| def _GetPidForLock(): |
| """Returns the PID used for host_forwarder initialization. |
| |
| The PID of the "sharder" is used to handle multiprocessing. The "sharder" |
| is the initial process that forks that is the parent process. |
| """ |
| return os.getpgrp() |
| |
| def _InitHostLocked(self): |
| """Initializes the host forwarder daemon. |
| |
| Note that the global lock must be acquired before calling this method. This |
| method kills any existing host_forwarder process that could be stale. |
| """ |
| # See if the host_forwarder daemon was already initialized by a concurrent |
| # process or thread (in case multi-process sharding is not used). |
| pid_for_lock = Forwarder._GetPidForLock() |
| fd = os.open(Forwarder._LOCK_PATH, os.O_RDWR | os.O_CREAT) |
| with os.fdopen(fd, 'r+') as pid_file: |
| pid_with_start_time = pid_file.readline() |
| if pid_with_start_time: |
| (pid, process_start_time) = pid_with_start_time.split(':') |
| if pid == str(pid_for_lock): |
| if process_start_time == str(_GetProcessStartTime(pid_for_lock)): |
| return |
| self._KillHostLocked() |
| pid_file.seek(0) |
| pid_file.write( |
| '%s:%s' % (pid_for_lock, str(_GetProcessStartTime(pid_for_lock)))) |
| pid_file.truncate() |
| |
| def _InitDeviceLocked(self, device, tool): |
| """Initializes the device_forwarder daemon for a specific device (once). |
| |
| Note that the global lock must be acquired before calling this method. This |
| method kills any existing device_forwarder daemon on the device that could |
| be stale, pushes the latest version of the daemon (to the device) and starts |
| it. |
| |
| Args: |
| device: A DeviceUtils instance. |
| tool: Tool class to use to get wrapper, if necessary, for executing the |
| forwarder (see valgrind_tools.py). |
| """ |
| device_serial = str(device) |
| if device_serial in self._initialized_devices: |
| return |
| Forwarder._KillDeviceLocked(device, tool) |
| device.PushChangedFiles([( |
| self._device_forwarder_path_on_host, |
| Forwarder._DEVICE_FORWARDER_FOLDER)]) |
| cmd = '%s %s' % (tool.GetUtilWrapper(), Forwarder._DEVICE_FORWARDER_PATH) |
| device.RunShellCommand( |
| cmd, env={'LD_LIBRARY_PATH': Forwarder._DEVICE_FORWARDER_FOLDER}) |
| self._initialized_devices.add(device_serial) |
| |
| def _KillHostLocked(self): |
| """Kills the forwarder process running on the host. |
| |
| Note that the global lock must be acquired before calling this method. |
| """ |
| logging.info('Killing host_forwarder.') |
| (exit_code, output) = cmd_helper.GetCmdStatusAndOutput( |
| [self._host_forwarder_path, '--kill-server']) |
| if exit_code != 0: |
| (exit_code, output) = cmd_helper.GetCmdStatusAndOutput( |
| ['pkill', '-9', 'host_forwarder']) |
| if exit_code != 0: |
| raise Exception('%s exited with %d:\n%s' % ( |
| self._host_forwarder_path, exit_code, '\n'.join(output))) |
| |
| @staticmethod |
| def _KillDeviceLocked(device, tool): |
| """Kills the forwarder process running on the device. |
| |
| Note that the global lock must be acquired before calling this method. |
| |
| Args: |
| device: Instance of DeviceUtils for talking to the device. |
| tool: Wrapper tool (e.g. valgrind) that can be used to execute the device |
| forwarder (see valgrind_tools.py). |
| """ |
| logging.info('Killing device_forwarder.') |
| Forwarder._instance._initialized_devices.discard(str(device)) |
| if not device.FileExists(Forwarder._DEVICE_FORWARDER_PATH): |
| return |
| |
| cmd = '%s %s --kill-server' % (tool.GetUtilWrapper(), |
| Forwarder._DEVICE_FORWARDER_PATH) |
| device.RunShellCommand( |
| cmd, env={'LD_LIBRARY_PATH': Forwarder._DEVICE_FORWARDER_FOLDER}) |
