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 3# found in the LICENSE file. 4 5""" A simple device interface for build steps. 6 7""" 8 9import logging 10import os 11import re 12import sys 13 14from util import build_utils 15 16BUILD_ANDROID_DIR = os.path.join(os.path.dirname(__file__), '..', '..') 17sys.path.append(BUILD_ANDROID_DIR) 18 19from pylib import android_commands 20from pylib.device import device_errors 21from pylib.device import device_utils 22 23GetAttachedDevices = android_commands.GetAttachedDevices 24 25 26class BuildDevice(object): 27 def __init__(self, configuration): 28 self.id = configuration['id'] 29 self.description = configuration['description'] 30 self.install_metadata = configuration['install_metadata'] 31 self.device = device_utils.DeviceUtils(self.id) 32 33 def RunShellCommand(self, *args, **kwargs): 34 return self.device.RunShellCommand(*args, **kwargs) 35 36 def PushIfNeeded(self, *args, **kwargs): 37 return self.device.old_interface.PushIfNeeded(*args, **kwargs) 38 39 def GetSerialNumber(self): 40 return self.id 41 42 def Install(self, *args, **kwargs): 43 return self.device.old_interface.Install(*args, **kwargs) 44 45 def GetInstallMetadata(self, apk_package): 46 """Gets the metadata on the device for the apk_package apk.""" 47 # Matches lines like: 48 # -rw-r--r-- system system 7376582 2013-04-19 16:34 \ 49 # org.chromium.chrome.shell.apk 50 # -rw-r--r-- system system 7376582 2013-04-19 16:34 \ 51 # org.chromium.chrome.shell-1.apk 52 apk_matcher = lambda s: re.match('.*%s(-[0-9]*)?.apk$' % apk_package, s) 53 matches = filter(apk_matcher, self.install_metadata) 54 return matches[0] if matches else None 55 56 57def GetConfigurationForDevice(device_id): 58 device = device_utils.DeviceUtils(device_id) 59 configuration = None 60 has_root = False 61 is_online = device.IsOnline() 62 if is_online: 63 cmd = 'ls -l /data/app; getprop ro.build.description' 64 cmd_output = device.RunShellCommand(cmd) 65 has_root = not 'Permission denied' in cmd_output[0] 66 if not has_root: 67 # Disable warning log messages from EnableRoot() 68 logging.getLogger().disabled = True 69 try: 70 device.EnableRoot() 71 has_root = True 72 except device_errors.CommandFailedError: 73 has_root = False 74 finally: 75 logging.getLogger().disabled = False 76 cmd_output = device.RunShellCommand(cmd) 77 78 configuration = { 79 'id': device_id, 80 'description': cmd_output[-1], 81 'install_metadata': cmd_output[:-1], 82 } 83 return configuration, is_online, has_root 84 85 86def WriteConfigurations(configurations, path): 87 # Currently we only support installing to the first device. 88 build_utils.WriteJson(configurations[:1], path, only_if_changed=True) 89 90 91def ReadConfigurations(path): 92 return build_utils.ReadJson(path) 93 94 95def GetBuildDevice(configurations): 96 assert len(configurations) == 1 97 return BuildDevice(configurations[0]) 98 99 100def GetBuildDeviceFromPath(path): 101 configurations = ReadConfigurations(path) 102 if len(configurations) > 0: 103 return GetBuildDevice(ReadConfigurations(path)) 104 return None 105 106