# Copyright 2013 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.
+
"""Brings in Chrome Android's android_commands module, which itself is a
thin(ish) wrapper around adb."""
import os
import shutil
import stat
-import sys
+from telemetry.core import platform
from telemetry.core import util
-from telemetry.core.platform.profiler import android_prebuilt_profiler_helper
+from telemetry.util import support_binaries
# This is currently a thin wrapper around Chrome Android's
# build scripts, located in chrome/build/android. This file exists mainly to
# deal with locating the module.
util.AddDirToPythonPath(util.GetChromiumSrcDir(), 'build', 'android')
+from pylib import android_commands # pylint: disable=F0401
+from pylib import constants # pylint: disable=F0401
try:
- from pylib import android_commands # pylint: disable=F0401
- from pylib import constants # pylint: disable=F0401
- from pylib import forwarder # pylint: disable=F0401
from pylib import ports # pylint: disable=F0401
- from pylib.utils import apk_helper # #pylint: disable=F0401
except Exception:
- android_commands = None
+ ports = None
+from pylib.device import device_utils # pylint: disable=F0401
def IsAndroidSupported():
- return android_commands != None
+ return device_utils != None
def GetAttachedDevices():
"""A thin wrapper around ADB"""
def __init__(self, device):
- self._adb = android_commands.AndroidCommands(device)
- self._device = device
+ self._device = device_utils.DeviceUtils(device)
+ self._device_serial = device
+
+ def device_serial(self):
+ return self._device_serial
def device(self):
return self._device
- def Adb(self):
- return self._adb
+ def __getattr__(self, name):
+ """Delegate all unknown calls to the underlying AndroidCommands object."""
+ return getattr(self._device.old_interface, name)
def Forward(self, local, remote):
- ret = self._adb.Adb().SendCommand('forward %s %s' % (local, remote))
+ ret = self._device.old_interface.Adb().SendCommand(
+ 'forward %s %s' % (local, remote))
assert ret == ''
- def RunShellCommand(self, command, timeout_time=20, log_result=False):
- """Send a command to the adb shell and return the result.
-
- Args:
- command: String containing the shell command to send. Must not include
- the single quotes as we use them to escape the whole command.
- timeout_time: Number of seconds to wait for command to respond before
- retrying, used by AdbInterface.SendShellCommand.
- log_result: Boolean to indicate whether we should log the result of the
- shell command.
-
- Returns:
- list containing the lines of output received from running the command
- """
- return self._adb.RunShellCommand(command, timeout_time, log_result)
-
- def CloseApplication(self, package):
- """Attempt to close down the application, using increasing violence.
-
- Args:
- package: Name of the process to kill off, e.g.
- com.google.android.apps.chrome
- """
- self._adb.CloseApplication(package)
-
- def KillAll(self, process):
- """Android version of killall, connected via adb.
-
- Args:
- process: name of the process to kill off
-
- Returns:
- the number of processess killed
- """
- return self._adb.KillAll(process)
-
- def ExtractPid(self, process_name):
- """Extracts Process Ids for a given process name from Android Shell.
-
- Args:
- process_name: name of the process on the device.
-
- Returns:
- List of all the process ids (as strings) that match the given name.
- If the name of a process exactly matches the given name, the pid of
- that process will be inserted to the front of the pid list.
- """
- return self._adb.ExtractPid(process_name)
-
- def Install(self, apk_path):
- """Installs specified package if necessary.
-
- Args:
- apk_path: Path to .apk file to install.
- """
-
- if (os.path.exists(os.path.join(
- constants.GetOutDirectory('Release'), 'md5sum_bin_host'))):
- constants.SetBuildType('Release')
- elif (os.path.exists(os.path.join(
- constants.GetOutDirectory('Debug'), 'md5sum_bin_host'))):
- constants.SetBuildType('Debug')
-
- apk_package_name = apk_helper.GetPackageName(apk_path)
- return self._adb.ManagedInstall(apk_path, package_name=apk_package_name)
-
- def StartActivity(self, package, activity, wait_for_completion=False,
- action='android.intent.action.VIEW',
- category=None, data=None,
- extras=None, trace_file_name=None,
- flags=None):
- """Starts |package|'s activity on the device.
-
- Args:
- package: Name of package to start (e.g. 'com.google.android.apps.chrome').
- activity: Name of activity (e.g. '.Main' or
- 'com.google.android.apps.chrome.Main').
- wait_for_completion: wait for the activity to finish launching (-W flag).
- action: string (e.g. 'android.intent.action.MAIN'). Default is VIEW.
- category: string (e.g. 'android.intent.category.HOME')
- data: Data string to pass to activity (e.g. 'http://www.example.com/').
- extras: Dict of extras to pass to activity. Values are significant.
- trace_file_name: If used, turns on and saves the trace to this file name.
- """
- return self._adb.StartActivity(package, activity, wait_for_completion,
- action,
- category, data,
- extras, trace_file_name,
- flags)
-
- def Push(self, local, remote):
- return self._adb.Adb().Push(local, remote)
-
- def Pull(self, remote, local):
- return self._adb.Adb().Pull(remote, local)
-
- def FileExistsOnDevice(self, file_name):
- return self._adb.FileExistsOnDevice(file_name)
-
- def IsRootEnabled(self):
- return self._adb.IsRootEnabled()
-
- def GoHome(self):
- return self._adb.GoHome()
-
-
-def SetupPrebuiltTools(device):
- # TODO(bulach): build the host tools for mac, and the targets for x86/mips.
- # Prebuilt tools from r226197.
- has_prebuilt = sys.platform.startswith('linux')
- if has_prebuilt:
- adb = AdbCommands(device)
- abi = adb.RunShellCommand('getprop ro.product.cpu.abi')
- has_prebuilt = abi and abi[0].startswith('armeabi')
- if not has_prebuilt:
- logging.error('Prebuilt tools only available for ARM.')
- return False
-
- prebuilt_tools = [
- 'forwarder_dist/device_forwarder',
- 'host_forwarder',
- 'md5sum_dist/md5sum_bin',
- 'md5sum_bin_host',
+ def IsUserBuild(self):
+ return self._device.GetProp('ro.build.type') == 'user'
+
+
+def GetBuildTypeOfPath(path):
+ if not path:
+ return None
+ for build_dir, build_type in util.GetBuildDirectories():
+ if os.path.join(build_dir, build_type) in path:
+ return build_type
+ return None
+
+
+def SetupPrebuiltTools(adb):
+ """Some of the android pylib scripts we depend on are lame and expect
+ binaries to be in the out/ directory. So we copy any prebuilt binaries there
+ as a prereq."""
+
+ # TODO(bulach): Build the targets for x86/mips.
+ device_tools = [
+ 'file_poller',
+ 'forwarder_dist/device_forwarder',
+ 'md5sum_dist/md5sum_bin',
+ 'purge_ashmem',
+ 'run_pie',
+ ]
+
+ host_tools = [
+ 'bitmaptools',
+ 'md5sum_bin_host',
]
- for t in prebuilt_tools:
- src = os.path.basename(t)
- android_prebuilt_profiler_helper.GetIfChanged(src)
+
+ if platform.GetHostPlatform().GetOSName() == 'linux':
+ host_tools.append('host_forwarder')
+
+ arch_name = adb.device().GetABI()
+ has_device_prebuilt = arch_name.startswith('armeabi')
+ if not has_device_prebuilt:
+ return all([support_binaries.FindLocallyBuiltPath(t) for t in device_tools])
+
+ build_type = None
+ for t in device_tools + host_tools:
+ executable = os.path.basename(t)
+ locally_built_path = support_binaries.FindLocallyBuiltPath(t)
+ if not build_type:
+ build_type = GetBuildTypeOfPath(locally_built_path) or 'Release'
+ constants.SetBuildType(build_type)
dest = os.path.join(constants.GetOutDirectory(), t)
- if not os.path.exists(dest):
- logging.warning('Setting up prebuilt %s', dest)
+ if not locally_built_path:
+ logging.info('Setting up prebuilt %s', dest)
if not os.path.exists(os.path.dirname(dest)):
os.makedirs(os.path.dirname(dest))
- shutil.copyfile(android_prebuilt_profiler_helper.GetHostPath(src), dest)
+ platform_name = ('android' if t in device_tools else
+ platform.GetHostPlatform().GetOSName())
+ bin_arch_name = (arch_name if t in device_tools else
+ platform.GetHostPlatform().GetArchName())
+ prebuilt_path = support_binaries.FindPath(
+ executable, bin_arch_name, platform_name)
+ if not prebuilt_path or not os.path.exists(prebuilt_path):
+ raise NotImplementedError("""
+%s must be checked into cloud storage.
+Instructions:
+http://www.chromium.org/developers/telemetry/upload_to_cloud_storage
+""" % t)
+ shutil.copyfile(prebuilt_path, dest)
os.chmod(dest, stat.S_IRUSR | stat.S_IWUSR | stat.S_IXUSR)
return True
-
-def HasForwarder(buildtype=None):
- if not buildtype:
- return (HasForwarder(buildtype='Release') or
- HasForwarder(buildtype='Debug'))
- device_forwarder = os.path.join(
- constants.GetOutDirectory(build_type=buildtype),
- 'forwarder_dist', 'device_forwarder')
- host_forwarder = os.path.join(
- constants.GetOutDirectory(build_type=buildtype), 'host_forwarder')
- return os.path.exists(device_forwarder) and os.path.exists(host_forwarder)
-
-
-class Forwarder(object):
- def __init__(self, adb, *port_pairs):
- self._adb = adb.Adb()
- self._host_port = port_pairs[0].local_port
-
- new_port_pairs = [(port_pair.local_port, port_pair.remote_port)
- for port_pair in port_pairs]
-
- self._port_pairs = new_port_pairs
- if HasForwarder('Release'):
- constants.SetBuildType('Release')
- elif HasForwarder('Debug'):
- constants.SetBuildType('Debug')
- else:
- raise Exception('Build forwarder2')
- forwarder.Forwarder.Map(new_port_pairs, self._adb)
-
- @property
- def url(self):
- return 'http://127.0.0.1:%i' % self._host_port
-
- def Close(self):
- for (device_port, _) in self._port_pairs:
- forwarder.Forwarder.UnmapDevicePort(device_port, self._adb)