blob: 6c5662c3b76ad6881fe82c7f3e449e12c341152e [file] [log] [blame]
J. Richard Barnette24adbf42012-04-11 15:04:53 -07001# Copyright (c) 2012 The Chromium OS Authors. All rights reserved.
Dale Curtisaa5eedb2011-08-23 16:18:52 -07002# Use of this source code is governed by a BSD-style license that can be
3# found in the LICENSE file.
4
mussa584b4462014-06-20 15:13:28 -07005import ConfigParser
Aviv Keshet74c89a92013-02-04 15:18:30 -08006import functools
Christopher Wiley0ed712b2013-04-09 15:25:12 -07007import httplib
J. Richard Barnette1d78b012012-05-15 13:56:30 -07008import logging
Dan Shi0f466e82013-02-22 15:44:58 -08009import os
Simran Basid5e5e272012-09-24 15:23:59 -070010import re
Christopher Wileyd78249a2013-03-01 13:05:31 -080011import socket
J. Richard Barnette134ec2c2012-04-25 12:59:37 -070012import time
J. Richard Barnette1d78b012012-05-15 13:56:30 -070013import xmlrpclib
J. Richard Barnette134ec2c2012-04-25 12:59:37 -070014
mussa584b4462014-06-20 15:13:28 -070015import common
J. Richard Barnette45e93de2012-04-11 17:24:15 -070016from autotest_lib.client.bin import utils
Dan Shi9cb0eec2014-06-03 09:04:50 -070017from autotest_lib.client.common_lib import autotemp
Richard Barnette0c73ffc2012-11-19 15:21:18 -080018from autotest_lib.client.common_lib import error
19from autotest_lib.client.common_lib import global_config
Dan Shi549fb822015-03-24 18:01:11 -070020from autotest_lib.client.common_lib import lsbrelease_utils
J. Richard Barnette45e93de2012-04-11 17:24:15 -070021from autotest_lib.client.common_lib.cros import autoupdater
Richard Barnette03a0c132012-11-05 12:40:35 -080022from autotest_lib.client.common_lib.cros import dev_server
Christopher Wileyd78249a2013-03-01 13:05:31 -080023from autotest_lib.client.common_lib.cros import retry
Gabe Blackb72f4fb2015-01-20 16:47:13 -080024from autotest_lib.client.common_lib.cros.graphite import autotest_es
Gabe Black1e1c41b2015-02-04 23:55:15 -080025from autotest_lib.client.common_lib.cros.graphite import autotest_stats
Hsinyu Chaoe0b08e62015-08-11 10:50:37 +000026from autotest_lib.client.cros import constants as client_constants
J. Richard Barnette84890bd2014-02-21 11:05:47 -080027from autotest_lib.client.cros import cros_ui
Cheng-Yi Chiangf4104ff2014-12-23 19:39:01 +080028from autotest_lib.client.cros.audio import cras_utils
Katherine Threlkeldab83d392015-06-18 16:45:57 -070029from autotest_lib.client.cros.input_playback import input_playback
MK Ryu35d661e2014-09-25 17:44:10 -070030from autotest_lib.server import autoserv_parser
31from autotest_lib.server import autotest
32from autotest_lib.server import constants
33from autotest_lib.server import crashcollect
Dan Shia1ecd5c2013-06-06 11:21:31 -070034from autotest_lib.server import utils as server_utils
Dan Shi9cb0eec2014-06-03 09:04:50 -070035from autotest_lib.server.cros import provision
Scott Zawalski89c44dd2013-02-26 09:28:02 -050036from autotest_lib.server.cros.dynamic_suite import constants as ds_constants
Simran Basi5e6339a2013-03-21 11:34:32 -070037from autotest_lib.server.cros.dynamic_suite import tools, frontend_wrappers
Dan Shi9cb0eec2014-06-03 09:04:50 -070038from autotest_lib.server.cros.faft.config.config import Config as FAFTConfig
Fang Deng96667ca2013-08-01 17:46:18 -070039from autotest_lib.server.hosts import abstract_ssh
Simran Basia5522a32015-10-06 11:01:24 -070040from autotest_lib.server.hosts import adb_host
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +080041from autotest_lib.server.hosts import chameleon_host
Fang Deng5d518f42013-08-02 14:04:32 -070042from autotest_lib.server.hosts import servo_host
Simran Basidcff4252012-11-20 16:13:20 -080043from autotest_lib.site_utils.rpm_control_system import rpm_client
Simran Basid5e5e272012-09-24 15:23:59 -070044
45
beeps32a63082013-08-22 14:02:29 -070046try:
47 import jsonrpclib
48except ImportError:
49 jsonrpclib = None
Fang Deng96667ca2013-08-01 17:46:18 -070050
Fang Dengd1c2b732013-08-20 12:59:46 -070051
Dan Shib8540a52015-07-16 14:18:23 -070052CONFIG = global_config.global_config
53
Eric Carusoee673ac2015-08-05 17:03:04 -070054LUCID_SLEEP_BOARDS = ['samus', 'lulu']
55
Dan Shid07ee2e2015-09-24 14:49:25 -070056# A file to indicate provision failure and require Repair job to powerwash the
57# dut.
58PROVISION_FAILED = '/var/tmp/provision_failed'
59
beepsc87ff602013-07-31 21:53:00 -070060class FactoryImageCheckerException(error.AutoservError):
61 """Exception raised when an image is a factory image."""
62 pass
63
64
Fang Deng0ca40e22013-08-27 17:47:44 -070065class CrosHost(abstract_ssh.AbstractSSHHost):
J. Richard Barnette45e93de2012-04-11 17:24:15 -070066 """Chromium OS specific subclass of Host."""
67
68 _parser = autoserv_parser.autoserv_parser
Scott Zawalski62bacae2013-03-05 10:40:32 -050069 _AFE = frontend_wrappers.RetryingAFE(timeout_min=5, delay_sec=10)
J. Richard Barnette45e93de2012-04-11 17:24:15 -070070
Richard Barnette03a0c132012-11-05 12:40:35 -080071 # Timeout values (in seconds) associated with various Chrome OS
72 # state changes.
J. Richard Barnette134ec2c2012-04-25 12:59:37 -070073 #
Richard Barnette0c73ffc2012-11-19 15:21:18 -080074 # In general, a good rule of thumb is that the timeout can be up
75 # to twice the typical measured value on the slowest platform.
76 # The times here have not necessarily been empirically tested to
77 # meet this criterion.
J. Richard Barnetteeb69d722012-06-18 17:29:44 -070078 #
79 # SLEEP_TIMEOUT: Time to allow for suspend to memory.
Richard Barnette0c73ffc2012-11-19 15:21:18 -080080 # RESUME_TIMEOUT: Time to allow for resume after suspend, plus
81 # time to restart the netwowrk.
J. Richard Barnette84890bd2014-02-21 11:05:47 -080082 # SHUTDOWN_TIMEOUT: Time to allow for shut down.
J. Richard Barnetteeb69d722012-06-18 17:29:44 -070083 # BOOT_TIMEOUT: Time to allow for boot from power off. Among
Richard Barnette0c73ffc2012-11-19 15:21:18 -080084 # other things, this must account for the 30 second dev-mode
J. Richard Barnette417cc792015-10-01 09:56:36 -070085 # screen delay, time to start the network on the DUT, and the
86 # ssh timeout of 120 seconds.
J. Richard Barnetteeb69d722012-06-18 17:29:44 -070087 # USB_BOOT_TIMEOUT: Time to allow for boot from a USB device,
Richard Barnette0c73ffc2012-11-19 15:21:18 -080088 # including the 30 second dev-mode delay and time to start the
J. Richard Barnetted4649c62013-03-06 17:42:27 -080089 # network.
beepsf079cfb2013-09-18 17:49:51 -070090 # INSTALL_TIMEOUT: Time to allow for chromeos-install.
J. Richard Barnette84890bd2014-02-21 11:05:47 -080091 # POWERWASH_BOOT_TIMEOUT: Time to allow for a reboot that
92 # includes powerwash.
J. Richard Barnetteeb69d722012-06-18 17:29:44 -070093
94 SLEEP_TIMEOUT = 2
J. Richard Barnetted4649c62013-03-06 17:42:27 -080095 RESUME_TIMEOUT = 10
Tom Wai-Hong Tamfe005c22014-12-03 09:25:44 +080096 SHUTDOWN_TIMEOUT = 10
J. Richard Barnette417cc792015-10-01 09:56:36 -070097 BOOT_TIMEOUT = 150
J. Richard Barnette5bab5f52015-08-03 13:14:38 -070098 USB_BOOT_TIMEOUT = 300
J. Richard Barnette7817b052014-08-28 09:47:29 -070099 INSTALL_TIMEOUT = 480
Dan Shi2c88eed2013-11-12 10:18:38 -0800100 POWERWASH_BOOT_TIMEOUT = 60
Chris Sosab76e0ee2013-05-22 16:55:41 -0700101
Dan Shica503482015-03-30 17:23:25 -0700102 # Minimum OS version that supports server side packaging. Older builds may
103 # not have server side package built or with Autotest code change to support
104 # server-side packaging.
Dan Shib8540a52015-07-16 14:18:23 -0700105 MIN_VERSION_SUPPORT_SSP = CONFIG.get_config_value(
Dan Shiced09e42015-04-17 16:09:34 -0700106 'AUTOSERV', 'min_version_support_ssp', type=int)
Dan Shica503482015-03-30 17:23:25 -0700107
J. Richard Barnette84890bd2014-02-21 11:05:47 -0800108 # REBOOT_TIMEOUT: How long to wait for a reboot.
109 #
Chris Sosab76e0ee2013-05-22 16:55:41 -0700110 # We have a long timeout to ensure we don't flakily fail due to other
111 # issues. Shorter timeouts are vetted in platform_RebootAfterUpdate.
Simran Basi1160e2c2013-10-04 16:00:24 -0700112 # TODO(sbasi - crbug.com/276094) Restore to 5 mins once the 'host did not
113 # return from reboot' bug is solved.
114 REBOOT_TIMEOUT = 480
Chris Sosab76e0ee2013-05-22 16:55:41 -0700115
Ismail Noorbasha07fdb612013-02-14 14:13:31 -0800116 # _USB_POWER_TIMEOUT: Time to allow for USB to power toggle ON and OFF.
117 # _POWER_CYCLE_TIMEOUT: Time to allow for manual power cycle.
118 _USB_POWER_TIMEOUT = 5
119 _POWER_CYCLE_TIMEOUT = 10
120
beeps32a63082013-08-22 14:02:29 -0700121 _RPC_PROXY_URL = 'http://localhost:%d'
Christopher Wileydd181852013-10-10 19:56:58 -0700122 _RPC_SHUTDOWN_POLLING_PERIOD_SECONDS = 2
Peter Qiu4410db72014-06-05 10:32:41 -0700123 # Set shutdown timeout to account for the time for restarting the UI.
124 _RPC_SHUTDOWN_TIMEOUT_SECONDS = cros_ui.RESTART_UI_TIMEOUT
Richard Barnette0c73ffc2012-11-19 15:21:18 -0800125
Dan Shib8540a52015-07-16 14:18:23 -0700126 _RPM_RECOVERY_BOARDS = CONFIG.get_config_value('CROS',
Richard Barnette82c35912012-11-20 10:09:10 -0800127 'rpm_recovery_boards', type=str).split(',')
128
129 _MAX_POWER_CYCLE_ATTEMPTS = 6
130 _LAB_MACHINE_FILE = '/mnt/stateful_partition/.labmachine'
Fang Dengdeba14f2014-11-14 11:54:09 -0800131 _RPM_HOSTNAME_REGEX = ('chromeos(\d+)(-row(\d+))?-rack(\d+[a-z]*)'
132 '-host(\d+)')
Katherine Threlkeldab83d392015-06-18 16:45:57 -0700133 _LIGHTSENSOR_FILES = [ "in_illuminance0_input",
134 "in_illuminance_input",
135 "in_illuminance0_raw",
136 "in_illuminance_raw",
137 "illuminance0_input"]
Richard Barnette82c35912012-11-20 10:09:10 -0800138 _LIGHTSENSOR_SEARCH_DIR = '/sys/bus/iio/devices'
139 _LABEL_FUNCTIONS = []
Aviv Keshet74c89a92013-02-04 15:18:30 -0800140 _DETECTABLE_LABELS = []
Kevin Cheng3a4a57a2015-09-30 12:09:50 -0700141 label_decorator = functools.partial(server_utils.add_label_detector,
142 _LABEL_FUNCTIONS,
Aviv Keshet74c89a92013-02-04 15:18:30 -0800143 _DETECTABLE_LABELS)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -0700144
J. Richard Barnetteb6de7e32013-02-14 13:28:04 -0800145 # Constants used in ping_wait_up() and ping_wait_down().
146 #
147 # _PING_WAIT_COUNT is the approximate number of polling
148 # cycles to use when waiting for a host state change.
149 #
150 # _PING_STATUS_DOWN and _PING_STATUS_UP are names used
151 # for arguments to the internal _ping_wait_for_status()
152 # method.
153 _PING_WAIT_COUNT = 40
154 _PING_STATUS_DOWN = False
155 _PING_STATUS_UP = True
156
Ismail Noorbasha07fdb612013-02-14 14:13:31 -0800157 # Allowed values for the power_method argument.
158
159 # POWER_CONTROL_RPM: Passed as default arg for power_off/on/cycle() methods.
160 # POWER_CONTROL_SERVO: Used in set_power() and power_cycle() methods.
161 # POWER_CONTROL_MANUAL: Used in set_power() and power_cycle() methods.
162 POWER_CONTROL_RPM = 'RPM'
163 POWER_CONTROL_SERVO = 'servoj10'
164 POWER_CONTROL_MANUAL = 'manual'
165
166 POWER_CONTROL_VALID_ARGS = (POWER_CONTROL_RPM,
167 POWER_CONTROL_SERVO,
168 POWER_CONTROL_MANUAL)
Richard Barnette0c73ffc2012-11-19 15:21:18 -0800169
Simran Basi5e6339a2013-03-21 11:34:32 -0700170 _RPM_OUTLET_CHANGED = 'outlet_changed'
171
Dan Shi9cb0eec2014-06-03 09:04:50 -0700172 # URL pattern to download firmware image.
Dan Shib8540a52015-07-16 14:18:23 -0700173 _FW_IMAGE_URL_PATTERN = CONFIG.get_config_value(
Dan Shi9cb0eec2014-06-03 09:04:50 -0700174 'CROS', 'firmware_url_pattern', type=str)
beeps687243d2013-07-18 15:29:27 -0700175
MK Ryu35d661e2014-09-25 17:44:10 -0700176 # File that has a list of directories to be collected
177 _LOGS_TO_COLLECT_FILE = os.path.join(
178 common.client_dir, 'common_lib', 'logs_to_collect')
179
180 # Prefix of logging message w.r.t. crash collection
181 _CRASHLOGS_PREFIX = 'collect_crashlogs'
182
183 # Time duration waiting for host up/down check
184 _CHECK_HOST_UP_TIMEOUT_SECS = 15
185
186 # A command that interacts with kernel and hardware (e.g., rm, mkdir, etc)
187 # might not be completely done deep through the hardware when the machine
188 # is powered down right after the command returns.
189 # We should wait for a few seconds to make them done. Finger crossed.
190 _SAFE_WAIT_SECS = 10
191
192
J. Richard Barnette964fba02012-10-24 17:34:29 -0700193 @staticmethod
beeps46dadc92013-11-07 14:07:10 -0800194 def check_host(host, timeout=10):
195 """
196 Check if the given host is a chrome-os host.
197
198 @param host: An ssh host representing a device.
199 @param timeout: The timeout for the run command.
200
201 @return: True if the host device is chromeos.
202
beeps46dadc92013-11-07 14:07:10 -0800203 """
204 try:
Simran Basi933c8af2015-04-29 14:05:07 -0700205 result = host.run(
206 'grep -q CHROMEOS /etc/lsb-release && '
207 '! test -f /mnt/stateful_partition/.android_tester && '
208 '! grep -q moblab /etc/lsb-release',
209 ignore_status=True, timeout=timeout)
beeps46dadc92013-11-07 14:07:10 -0800210 except (error.AutoservRunError, error.AutoservSSHTimeout):
211 return False
212 return result.exit_status == 0
213
214
215 @staticmethod
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +0800216 def _extract_arguments(args_dict, key_subset):
217 """Extract options from `args_dict` and return a subset result.
J. Richard Barnette7214e0b2013-02-06 15:20:49 -0800218
219 Take the provided dictionary of argument options and return
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +0800220 a subset that represent standard arguments needed to construct
221 a test-assistant object (chameleon or servo) for a host. The
222 intent is to provide standard argument processing from
Christopher Wiley644ef3e2015-05-15 13:14:14 -0700223 CrosHost for tests that require a test-assistant board
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +0800224 to operate.
225
226 @param args_dict Dictionary from which to extract the arguments.
227 @param key_subset Tuple of keys to extract from the args_dict, e.g.
228 ('servo_host', 'servo_port').
229 """
230 result = {}
231 for arg in key_subset:
232 if arg in args_dict:
233 result[arg] = args_dict[arg]
234 return result
235
236
237 @staticmethod
238 def get_chameleon_arguments(args_dict):
239 """Extract chameleon options from `args_dict` and return the result.
240
241 Recommended usage:
242 ~~~~~~~~
243 args_dict = utils.args_to_dict(args)
244 chameleon_args = hosts.CrosHost.get_chameleon_arguments(args_dict)
245 host = hosts.create_host(machine, chameleon_args=chameleon_args)
246 ~~~~~~~~
247
248 @param args_dict Dictionary from which to extract the chameleon
249 arguments.
250 """
251 return CrosHost._extract_arguments(
252 args_dict, ('chameleon_host', 'chameleon_port'))
253
254
255 @staticmethod
256 def get_servo_arguments(args_dict):
257 """Extract servo options from `args_dict` and return the result.
J. Richard Barnette7214e0b2013-02-06 15:20:49 -0800258
259 Recommended usage:
260 ~~~~~~~~
261 args_dict = utils.args_to_dict(args)
Fang Deng0ca40e22013-08-27 17:47:44 -0700262 servo_args = hosts.CrosHost.get_servo_arguments(args_dict)
J. Richard Barnette7214e0b2013-02-06 15:20:49 -0800263 host = hosts.create_host(machine, servo_args=servo_args)
264 ~~~~~~~~
265
266 @param args_dict Dictionary from which to extract the servo
267 arguments.
268 """
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +0800269 return CrosHost._extract_arguments(
270 args_dict, ('servo_host', 'servo_port'))
J. Richard Barnette134ec2c2012-04-25 12:59:37 -0700271
J. Richard Barnette964fba02012-10-24 17:34:29 -0700272
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +0800273 def _initialize(self, hostname, chameleon_args=None, servo_args=None,
Fang Denge545abb2014-12-30 18:43:47 -0800274 try_lab_servo=False, ssh_verbosity_flag='', ssh_options='',
Fang Dengd1c2b732013-08-20 12:59:46 -0700275 *args, **dargs):
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +0800276 """Initialize superclasses, |self.chameleon|, and |self.servo|.
J. Richard Barnette67ccb872012-04-19 16:34:56 -0700277
Fang Denge545abb2014-12-30 18:43:47 -0800278 This method will attempt to create the test-assistant object
279 (chameleon/servo) when it is needed by the test. Check
280 the docstring of chameleon_host.create_chameleon_host and
281 servo_host.create_servo_host for how this is determined.
Fang Deng5d518f42013-08-02 14:04:32 -0700282
Fang Denge545abb2014-12-30 18:43:47 -0800283 @param hostname: Hostname of the dut.
284 @param chameleon_args: A dictionary that contains args for creating
285 a ChameleonHost. See chameleon_host for details.
286 @param servo_args: A dictionary that contains args for creating
287 a ServoHost object. See servo_host for details.
288 @param try_lab_servo: Boolean, False indicates that ServoHost should
289 not be created for a device in Cros test lab.
290 See servo_host for details.
291 @param ssh_verbosity_flag: String, to pass to the ssh command to control
292 verbosity.
293 @param ssh_options: String, other ssh options to pass to the ssh
294 command.
J. Richard Barnette67ccb872012-04-19 16:34:56 -0700295 """
Fang Deng0ca40e22013-08-27 17:47:44 -0700296 super(CrosHost, self)._initialize(hostname=hostname,
J. Richard Barnette45e93de2012-04-11 17:24:15 -0700297 *args, **dargs)
J. Richard Barnettef0859852012-08-20 14:55:50 -0700298 # self.env is a dictionary of environment variable settings
299 # to be exported for commands run on the host.
300 # LIBC_FATAL_STDERR_ can be useful for diagnosing certain
301 # errors that might happen.
302 self.env['LIBC_FATAL_STDERR_'] = '1'
beeps32a63082013-08-22 14:02:29 -0700303 self._rpc_proxy_map = {}
Fang Dengd1c2b732013-08-20 12:59:46 -0700304 self._ssh_verbosity_flag = ssh_verbosity_flag
Aviv Keshetc5947fa2013-09-04 14:06:29 -0700305 self._ssh_options = ssh_options
Fang Deng5d518f42013-08-02 14:04:32 -0700306 # TODO(fdeng): We need to simplify the
307 # process of servo and servo_host initialization.
308 # crbug.com/298432
Fang Denge545abb2014-12-30 18:43:47 -0800309 self._servo_host = servo_host.create_servo_host(
310 dut=self.hostname, servo_args=servo_args,
311 try_lab_servo=try_lab_servo)
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +0800312 # TODO(waihong): Do the simplication on Chameleon too.
Tom Wai-Hong Tam3d6790d2014-04-14 16:15:47 +0800313 self._chameleon_host = chameleon_host.create_chameleon_host(
314 dut=self.hostname, chameleon_args=chameleon_args)
315
Dan Shi4d478522014-02-14 13:46:32 -0800316 if self._servo_host is not None:
317 self.servo = self._servo_host.get_servo()
318 else:
319 self.servo = None
320
Tom Wai-Hong Tam3d6790d2014-04-14 16:15:47 +0800321 if self._chameleon_host:
Tom Wai-Hong Tameaee3402014-01-22 08:52:10 +0800322 self.chameleon = self._chameleon_host.create_chameleon_board()
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +0800323 else:
Tom Wai-Hong Tam3d6790d2014-04-14 16:15:47 +0800324 self.chameleon = None
Fang Deng5d518f42013-08-02 14:04:32 -0700325
326
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500327 def get_repair_image_name(self):
328 """Generate a image_name from variables in the global config.
329
330 @returns a str of $board-version/$BUILD.
331
332 """
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500333 board = self._get_board_from_afe()
334 if board is None:
335 raise error.AutoservError('DUT has no board attribute, '
336 'cannot be repaired.')
Dan Shi6964fa52014-12-18 11:04:27 -0800337 stable_version = self._AFE.run('get_stable_version', board=board)
Dan Shib8540a52015-07-16 14:18:23 -0700338 build_pattern = CONFIG.get_config_value(
Dan Shi6964fa52014-12-18 11:04:27 -0800339 'CROS', 'stable_build_pattern')
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500340 return build_pattern % (board, stable_version)
341
342
Scott Zawalski62bacae2013-03-05 10:40:32 -0500343 def _host_in_AFE(self):
344 """Check if the host is an object the AFE knows.
345
346 @returns the host object.
347 """
348 return self._AFE.get_hosts(hostname=self.hostname)
349
350
Chris Sosab76e0ee2013-05-22 16:55:41 -0700351 def lookup_job_repo_url(self):
352 """Looks up the job_repo_url for the host.
353
354 @returns job_repo_url from AFE or None if not found.
355
356 @raises KeyError if the host does not have a job_repo_url
357 """
Chris Sosab76e0ee2013-05-22 16:55:41 -0700358 hosts = self._AFE.get_hosts(hostname=self.hostname)
beepsb5efc532013-06-04 11:29:34 -0700359 if hosts and ds_constants.JOB_REPO_URL in hosts[0].attributes:
360 return hosts[0].attributes[ds_constants.JOB_REPO_URL]
J. Richard Barnette85d0aac2015-08-20 10:34:39 -0700361 else:
362 return None
Chris Sosab76e0ee2013-05-22 16:55:41 -0700363
364
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500365 def clear_cros_version_labels_and_job_repo_url(self):
366 """Clear cros_version labels and host attribute job_repo_url."""
Scott Zawalski62bacae2013-03-05 10:40:32 -0500367 if not self._host_in_AFE():
Scott Zawalskieadbf702013-03-14 09:23:06 -0400368 return
369
Scott Zawalski62bacae2013-03-05 10:40:32 -0500370 host_list = [self.hostname]
371 labels = self._AFE.get_labels(
372 name__startswith=ds_constants.VERSION_PREFIX,
373 host__hostname=self.hostname)
Dan Shi0f466e82013-02-22 15:44:58 -0800374
Scott Zawalski62bacae2013-03-05 10:40:32 -0500375 for label in labels:
376 label.remove_hosts(hosts=host_list)
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500377
beepscb6f1e22013-06-28 19:14:10 -0700378 self.update_job_repo_url(None, None)
379
380
381 def update_job_repo_url(self, devserver_url, image_name):
382 """
383 Updates the job_repo_url host attribute and asserts it's value.
384
385 @param devserver_url: The devserver to use in the job_repo_url.
386 @param image_name: The name of the image to use in the job_repo_url.
387
388 @raises AutoservError: If we failed to update the job_repo_url.
389 """
390 repo_url = None
391 if devserver_url and image_name:
392 repo_url = tools.get_package_url(devserver_url, image_name)
393 self._AFE.set_host_attribute(ds_constants.JOB_REPO_URL, repo_url,
Scott Zawalski62bacae2013-03-05 10:40:32 -0500394 hostname=self.hostname)
beepscb6f1e22013-06-28 19:14:10 -0700395 if self.lookup_job_repo_url() != repo_url:
396 raise error.AutoservError('Failed to update job_repo_url with %s, '
397 'host %s' % (repo_url, self.hostname))
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500398
399
Dan Shie9309262013-06-19 22:50:21 -0700400 def add_cros_version_labels_and_job_repo_url(self, image_name):
Scott Zawalskieadbf702013-03-14 09:23:06 -0400401 """Add cros_version labels and host attribute job_repo_url.
402
403 @param image_name: The name of the image e.g.
404 lumpy-release/R27-3837.0.0
Dan Shi7458bf62013-06-10 12:50:16 -0700405
Scott Zawalskieadbf702013-03-14 09:23:06 -0400406 """
Scott Zawalski62bacae2013-03-05 10:40:32 -0500407 if not self._host_in_AFE():
Scott Zawalskieadbf702013-03-14 09:23:06 -0400408 return
Scott Zawalski62bacae2013-03-05 10:40:32 -0500409
Scott Zawalskieadbf702013-03-14 09:23:06 -0400410 cros_label = '%s%s' % (ds_constants.VERSION_PREFIX, image_name)
Dan Shie9309262013-06-19 22:50:21 -0700411 devserver_url = dev_server.ImageServer.resolve(image_name).url()
Scott Zawalski62bacae2013-03-05 10:40:32 -0500412
MK Ryufb5e3a82015-07-01 12:21:20 -0700413 self._AFE.run('label_add_hosts', id=cros_label, hosts=[self.hostname])
beepscb6f1e22013-06-28 19:14:10 -0700414 self.update_job_repo_url(devserver_url, image_name)
415
416
beepsdae65fd2013-07-26 16:24:41 -0700417 def verify_job_repo_url(self, tag=''):
beepscb6f1e22013-06-28 19:14:10 -0700418 """
419 Make sure job_repo_url of this host is valid.
420
joychen03eaad92013-06-26 09:55:21 -0700421 Eg: The job_repo_url "http://lmn.cd.ab.xyx:8080/static/\
beepscb6f1e22013-06-28 19:14:10 -0700422 lumpy-release/R29-4279.0.0/autotest/packages" claims to have the
423 autotest package for lumpy-release/R29-4279.0.0. If this isn't the case,
424 download and extract it. If the devserver embedded in the url is
425 unresponsive, update the job_repo_url of the host after staging it on
426 another devserver.
427
428 @param job_repo_url: A url pointing to the devserver where the autotest
429 package for this build should be staged.
beepsdae65fd2013-07-26 16:24:41 -0700430 @param tag: The tag from the server job, in the format
431 <job_id>-<user>/<hostname>, or <hostless> for a server job.
beepscb6f1e22013-06-28 19:14:10 -0700432
433 @raises DevServerException: If we could not resolve a devserver.
434 @raises AutoservError: If we're unable to save the new job_repo_url as
435 a result of choosing a new devserver because the old one failed to
436 respond to a health check.
beeps0c865032013-07-30 11:37:06 -0700437 @raises urllib2.URLError: If the devserver embedded in job_repo_url
438 doesn't respond within the timeout.
beepscb6f1e22013-06-28 19:14:10 -0700439 """
440 job_repo_url = self.lookup_job_repo_url()
441 if not job_repo_url:
442 logging.warning('No job repo url set on host %s', self.hostname)
443 return
444
445 logging.info('Verifying job repo url %s', job_repo_url)
446 devserver_url, image_name = tools.get_devserver_build_from_package_url(
447 job_repo_url)
448
beeps0c865032013-07-30 11:37:06 -0700449 ds = dev_server.ImageServer(devserver_url)
beepscb6f1e22013-06-28 19:14:10 -0700450
451 logging.info('Staging autotest artifacts for %s on devserver %s',
452 image_name, ds.url())
beeps687243d2013-07-18 15:29:27 -0700453
454 start_time = time.time()
Simran Basi25e7a922014-10-31 11:56:10 -0700455 ds.stage_artifacts(image_name, ['autotest_packages'])
beeps687243d2013-07-18 15:29:27 -0700456 stage_time = time.time() - start_time
457
458 # Record how much of the verification time comes from a devserver
459 # restage. If we're doing things right we should not see multiple
460 # devservers for a given board/build/branch path.
461 try:
J. Richard Barnette3cbd76b2013-11-27 12:11:25 -0800462 board, build_type, branch = server_utils.ParseBuildName(
beeps687243d2013-07-18 15:29:27 -0700463 image_name)[:3]
J. Richard Barnette3cbd76b2013-11-27 12:11:25 -0800464 except server_utils.ParseBuildNameException:
beeps687243d2013-07-18 15:29:27 -0700465 pass
466 else:
beeps0c865032013-07-30 11:37:06 -0700467 devserver = devserver_url[
Chris Sosa65425082013-10-16 13:26:22 -0700468 devserver_url.find('/') + 2:devserver_url.rfind(':')]
beeps687243d2013-07-18 15:29:27 -0700469 stats_key = {
470 'board': board,
471 'build_type': build_type,
472 'branch': branch,
beeps0c865032013-07-30 11:37:06 -0700473 'devserver': devserver.replace('.', '_'),
beeps687243d2013-07-18 15:29:27 -0700474 }
Gabe Black1e1c41b2015-02-04 23:55:15 -0800475 autotest_stats.Gauge('verify_job_repo_url').send(
beeps687243d2013-07-18 15:29:27 -0700476 '%(board)s.%(build_type)s.%(branch)s.%(devserver)s' % stats_key,
477 stage_time)
beepscb6f1e22013-06-28 19:14:10 -0700478
Scott Zawalskieadbf702013-03-14 09:23:06 -0400479
Dan Shicf4d2032015-03-12 15:04:21 -0700480 def stage_server_side_package(self, image=None):
481 """Stage autotest server-side package on devserver.
482
483 @param image: Full path of an OS image to install or a build name.
484
485 @return: A url to the autotest server-side package.
486 """
487 if image:
488 image_name = tools.get_build_from_image(image)
489 if not image_name:
490 raise error.AutoservError(
491 'Failed to parse build name from %s' % image)
492 ds = dev_server.ImageServer.resolve(image_name)
493 else:
494 job_repo_url = self.lookup_job_repo_url()
495 if job_repo_url:
496 devserver_url, image_name = (
497 tools.get_devserver_build_from_package_url(job_repo_url))
498 ds = dev_server.ImageServer(devserver_url)
499 else:
500 labels = self._AFE.get_labels(
501 name__startswith=ds_constants.VERSION_PREFIX,
502 host__hostname=self.hostname)
503 if not labels:
504 raise error.AutoservError(
505 'Failed to stage server-side package. The host has '
506 'no job_report_url attribute or version label.')
507 image_name = labels[0].name[len(ds_constants.VERSION_PREFIX):]
508 ds = dev_server.ImageServer.resolve(image_name)
Dan Shica503482015-03-30 17:23:25 -0700509
510 # Get the OS version of the build, for any build older than
511 # MIN_VERSION_SUPPORT_SSP, server side packaging is not supported.
512 match = re.match('.*/R\d+-(\d+)\.', image_name)
513 if match and int(match.group(1)) < self.MIN_VERSION_SUPPORT_SSP:
514 logging.warn('Build %s is older than %s. Server side packaging is '
515 'disabled.', image_name, self.MIN_VERSION_SUPPORT_SSP)
516 return None
517
Dan Shicf4d2032015-03-12 15:04:21 -0700518 ds.stage_artifacts(image_name, ['autotest_server_package'])
519 return '%s/static/%s/%s' % (ds.url(), image_name,
520 'autotest_server_package.tar.bz2')
521
522
Dan Shi0f466e82013-02-22 15:44:58 -0800523 def _try_stateful_update(self, update_url, force_update, updater):
524 """Try to use stateful update to initialize DUT.
525
526 When DUT is already running the same version that machine_install
527 tries to install, stateful update is a much faster way to clean up
528 the DUT for testing, compared to a full reimage. It is implemeted
529 by calling autoupdater.run_update, but skipping updating root, as
530 updating the kernel is time consuming and not necessary.
531
532 @param update_url: url of the image.
533 @param force_update: Set to True to update the image even if the DUT
534 is running the same version.
535 @param updater: ChromiumOSUpdater instance used to update the DUT.
536 @returns: True if the DUT was updated with stateful update.
537
538 """
J. Richard Barnette3f731032014-04-07 17:42:59 -0700539 # TODO(jrbarnette): Yes, I hate this re.match() test case.
540 # It's better than the alternative: see crbug.com/360944.
541 image_name = autoupdater.url_to_image_name(update_url)
542 release_pattern = r'^.*-release/R[0-9]+-[0-9]+\.[0-9]+\.0$'
543 if not re.match(release_pattern, image_name):
544 return False
Dan Shi0f466e82013-02-22 15:44:58 -0800545 if not updater.check_version():
546 return False
547 if not force_update:
548 logging.info('Canceling stateful update because the new and '
549 'old versions are the same.')
550 return False
551 # Following folders should be rebuilt after stateful update.
552 # A test file is used to confirm each folder gets rebuilt after
553 # the stateful update.
554 folders_to_check = ['/var', '/home', '/mnt/stateful_partition']
555 test_file = '.test_file_to_be_deleted'
556 for folder in folders_to_check:
557 touch_path = os.path.join(folder, test_file)
558 self.run('touch %s' % touch_path)
559
Chris Sosae92399e2015-04-24 11:32:59 -0700560 updater.run_update(update_root=False)
Dan Shi0f466e82013-02-22 15:44:58 -0800561
562 # Reboot to complete stateful update.
Chris Sosab76e0ee2013-05-22 16:55:41 -0700563 self.reboot(timeout=self.REBOOT_TIMEOUT, wait=True)
Dan Shi0f466e82013-02-22 15:44:58 -0800564 check_file_cmd = 'test -f %s; echo $?'
565 for folder in folders_to_check:
566 test_file_path = os.path.join(folder, test_file)
567 result = self.run(check_file_cmd % test_file_path,
568 ignore_status=True)
569 if result.exit_status == 1:
570 return False
571 return True
572
573
J. Richard Barnette7275b612013-06-04 18:13:11 -0700574 def _post_update_processing(self, updater, expected_kernel=None):
Dan Shi0f466e82013-02-22 15:44:58 -0800575 """After the DUT is updated, confirm machine_install succeeded.
576
577 @param updater: ChromiumOSUpdater instance used to update the DUT.
J. Richard Barnette7275b612013-06-04 18:13:11 -0700578 @param expected_kernel: kernel expected to be active after reboot,
579 or `None` to skip rollback checking.
Dan Shi0f466e82013-02-22 15:44:58 -0800580
581 """
J. Richard Barnette7275b612013-06-04 18:13:11 -0700582 # Touch the lab machine file to leave a marker that
583 # distinguishes this image from other test images.
584 # Afterwards, we must re-run the autoreboot script because
585 # it depends on the _LAB_MACHINE_FILE.
Dan Shi0f466e82013-02-22 15:44:58 -0800586 self.run('touch %s' % self._LAB_MACHINE_FILE)
Dan Shi0f466e82013-02-22 15:44:58 -0800587 self.run('start autoreboot')
Chris Sosa65425082013-10-16 13:26:22 -0700588 updater.verify_boot_expectations(
589 expected_kernel, rollback_message=
590 'Build %s failed to boot on %s; system rolled back to previous'
591 'build' % (updater.update_version, self.hostname))
J. Richard Barnette7275b612013-06-04 18:13:11 -0700592 # Check that we've got the build we meant to install.
593 if not updater.check_version_to_confirm_install():
594 raise autoupdater.ChromiumOSError(
595 'Failed to update %s to build %s; found build '
596 '%s instead' % (self.hostname,
Chris Sosa65425082013-10-16 13:26:22 -0700597 updater.update_version,
Dan Shi0942b1d2015-03-31 11:07:00 -0700598 self.get_release_version()))
Dan Shi0f466e82013-02-22 15:44:58 -0800599
Chris Sosae92399e2015-04-24 11:32:59 -0700600 logging.debug('Cleaning up old autotest directories.')
601 try:
602 installed_autodir = autotest.Autotest.get_installed_autodir(self)
603 self.run('rm -rf ' + installed_autodir)
604 except autotest.AutodirNotFoundError:
605 logging.debug('No autotest installed directory found.')
606
Dan Shi0f466e82013-02-22 15:44:58 -0800607
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700608 def _stage_image_for_update(self, image_name=None):
Chris Sosae92399e2015-04-24 11:32:59 -0700609 """Stage a build on a devserver and return the update_url and devserver.
Scott Zawalskieadbf702013-03-14 09:23:06 -0400610
611 @param image_name: a name like lumpy-release/R27-3837.0.0
Chris Sosae92399e2015-04-24 11:32:59 -0700612 @returns a tuple with an update URL like:
Scott Zawalskieadbf702013-03-14 09:23:06 -0400613 http://172.22.50.205:8082/update/lumpy-release/R27-3837.0.0
Chris Sosae92399e2015-04-24 11:32:59 -0700614 and the devserver instance.
Scott Zawalskieadbf702013-03-14 09:23:06 -0400615 """
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700616 if not image_name:
617 image_name = self.get_repair_image_name()
Chris Sosae92399e2015-04-24 11:32:59 -0700618
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700619 logging.info('Staging build for AU: %s', image_name)
Scott Zawalskieadbf702013-03-14 09:23:06 -0400620 devserver = dev_server.ImageServer.resolve(image_name)
621 devserver.trigger_download(image_name, synchronous=False)
Chris Sosae92399e2015-04-24 11:32:59 -0700622 return (tools.image_url_pattern() % (devserver.url(), image_name),
623 devserver)
Scott Zawalskieadbf702013-03-14 09:23:06 -0400624
625
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700626 def stage_image_for_servo(self, image_name=None):
627 """Stage a build on a devserver and return the update_url.
628
629 @param image_name: a name like lumpy-release/R27-3837.0.0
630 @returns an update URL like:
631 http://172.22.50.205:8082/update/lumpy-release/R27-3837.0.0
632 """
633 if not image_name:
634 image_name = self.get_repair_image_name()
635 logging.info('Staging build for servo install: %s', image_name)
636 devserver = dev_server.ImageServer.resolve(image_name)
637 devserver.stage_artifacts(image_name, ['test_image'])
638 return devserver.get_test_image_url(image_name)
639
640
beepse539be02013-07-31 21:57:39 -0700641 def stage_factory_image_for_servo(self, image_name):
642 """Stage a build on a devserver and return the update_url.
643
644 @param image_name: a name like <baord>/4262.204.0
beeps12c0a3c2013-09-03 11:58:27 -0700645
beepse539be02013-07-31 21:57:39 -0700646 @return: An update URL, eg:
647 http://<devserver>/static/canary-channel/\
648 <board>/4262.204.0/factory_test/chromiumos_factory_image.bin
beeps12c0a3c2013-09-03 11:58:27 -0700649
650 @raises: ValueError if the factory artifact name is missing from
651 the config.
652
beepse539be02013-07-31 21:57:39 -0700653 """
654 if not image_name:
655 logging.error('Need an image_name to stage a factory image.')
656 return
657
Dan Shib8540a52015-07-16 14:18:23 -0700658 factory_artifact = CONFIG.get_config_value(
beeps12c0a3c2013-09-03 11:58:27 -0700659 'CROS', 'factory_artifact', type=str, default='')
660 if not factory_artifact:
661 raise ValueError('Cannot retrieve the factory artifact name from '
662 'autotest config, and hence cannot stage factory '
663 'artifacts.')
664
beepse539be02013-07-31 21:57:39 -0700665 logging.info('Staging build for servo install: %s', image_name)
666 devserver = dev_server.ImageServer.resolve(image_name)
667 devserver.stage_artifacts(
668 image_name,
beeps12c0a3c2013-09-03 11:58:27 -0700669 [factory_artifact],
670 archive_url=None)
beepse539be02013-07-31 21:57:39 -0700671
672 return tools.factory_image_url_pattern() % (devserver.url(), image_name)
673
674
Chris Sosaa3ac2152012-05-23 22:23:13 -0700675 def machine_install(self, update_url=None, force_update=False,
Richard Barnette0b023a72015-04-24 16:07:30 +0000676 local_devserver=False, repair=False,
677 force_full_update=False):
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500678 """Install the DUT.
679
Dan Shi0f466e82013-02-22 15:44:58 -0800680 Use stateful update if the DUT is already running the same build.
681 Stateful update does not update kernel and tends to run much faster
682 than a full reimage. If the DUT is running a different build, or it
683 failed to do a stateful update, full update, including kernel update,
684 will be applied to the DUT.
685
Scott Zawalskieadbf702013-03-14 09:23:06 -0400686 Once a host enters machine_install its cros_version label will be
687 removed as well as its host attribute job_repo_url (used for
688 package install).
689
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500690 @param update_url: The url to use for the update
691 pattern: http://$devserver:###/update/$build
692 If update_url is None and repair is True we will install the
Dan Shi6964fa52014-12-18 11:04:27 -0800693 stable image listed in afe_stable_versions table. If the table
694 is not setup, global_config value under CROS.stable_cros_version
695 will be used instead.
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500696 @param force_update: Force an update even if the version installed
697 is the same. Default:False
Christopher Wiley6a4ff932015-05-15 14:00:47 -0700698 @param local_devserver: Used by test_that to allow people to
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500699 use their local devserver. Default: False
Chris Sosae92399e2015-04-24 11:32:59 -0700700 @param repair: Forces update to repair image. Implies force_update.
Fang Deng3d3b9272014-12-22 12:20:28 -0800701 @param force_full_update: If True, do not attempt to run stateful
702 update, force a full reimage. If False, try stateful update
703 first when the dut is already installed with the same version.
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500704 @raises autoupdater.ChromiumOSError
705
706 """
Chris Sosae92399e2015-04-24 11:32:59 -0700707 devserver = None
Richard Barnette0b023a72015-04-24 16:07:30 +0000708 if repair:
Chris Sosae92399e2015-04-24 11:32:59 -0700709 update_url, devserver = self._stage_image_for_update()
Richard Barnette0b023a72015-04-24 16:07:30 +0000710 force_update = True
Dan Shi0f466e82013-02-22 15:44:58 -0800711
Chris Sosae92399e2015-04-24 11:32:59 -0700712 if not update_url and not self._parser.options.image:
713 raise error.AutoservError(
Dan Shid07ee2e2015-09-24 14:49:25 -0700714 'There is no update URL, nor a method to get one.')
Chris Sosae92399e2015-04-24 11:32:59 -0700715
716 if not update_url and self._parser.options.image:
717 # This is the base case where we have no given update URL i.e.
718 # dynamic suites logic etc. This is the most flexible case where we
719 # can serve an update from any of our fleet of devservers.
720 requested_build = self._parser.options.image
721 if not requested_build.startswith('http://'):
722 logging.debug('Update will be staged for this installation')
723 update_url, devserver = self._stage_image_for_update(
Dan Shid07ee2e2015-09-24 14:49:25 -0700724 requested_build)
Chris Sosae92399e2015-04-24 11:32:59 -0700725 else:
726 update_url = requested_build
727
728 logging.debug('Update URL is %s', update_url)
729
Scott Zawalskieadbf702013-03-14 09:23:06 -0400730 # Remove cros-version and job_repo_url host attribute from host.
731 self.clear_cros_version_labels_and_job_repo_url()
Chris Sosae92399e2015-04-24 11:32:59 -0700732
Dan Shid07ee2e2015-09-24 14:49:25 -0700733 # Create a file to indicate if provision fails. The file will be removed
734 # by stateful update or full install.
735 self.run('touch %s' % PROVISION_FAILED)
736
Chris Sosae92399e2015-04-24 11:32:59 -0700737 update_complete = False
738 updater = autoupdater.ChromiumOSUpdater(
739 update_url, host=self, local_devserver=local_devserver)
Fang Deng3d3b9272014-12-22 12:20:28 -0800740 if not force_full_update:
741 try:
Chris Sosae92399e2015-04-24 11:32:59 -0700742 # If the DUT is already running the same build, try stateful
743 # update first as it's much quicker than a full re-image.
744 update_complete = self._try_stateful_update(
Dan Shid07ee2e2015-09-24 14:49:25 -0700745 update_url, force_update, updater)
Fang Deng3d3b9272014-12-22 12:20:28 -0800746 except Exception as e:
747 logging.exception(e)
J. Richard Barnette45e93de2012-04-11 17:24:15 -0700748
Dan Shi0f466e82013-02-22 15:44:58 -0800749 inactive_kernel = None
Chris Sosae92399e2015-04-24 11:32:59 -0700750 if update_complete or (not force_update and updater.check_version()):
751 logging.info('Install complete without full update')
752 else:
753 logging.info('DUT requires full update.')
754 self.reboot(timeout=self.REBOOT_TIMEOUT, wait=True)
755 num_of_attempts = provision.FLAKY_DEVSERVER_ATTEMPTS
Chris Sosab7612bc2013-03-21 10:32:37 -0700756
Chris Sosae92399e2015-04-24 11:32:59 -0700757 while num_of_attempts > 0:
758 num_of_attempts -= 1
759 try:
760 updater.run_update()
761 except Exception:
762 logging.warn('Autoupdate did not complete.')
763 # Do additional check for the devserver health. Ideally,
764 # the autoupdater.py could raise an exception when it
765 # detected network flake but that would require
766 # instrumenting the update engine and parsing it log.
767 if (num_of_attempts <= 0 or
768 devserver is None or
769 dev_server.DevServer.devserver_healthy(
770 devserver.url())):
Dan Shid07ee2e2015-09-24 14:49:25 -0700771 raise
J. Richard Barnette45e93de2012-04-11 17:24:15 -0700772
Chris Sosae92399e2015-04-24 11:32:59 -0700773 logging.warn('Devserver looks unhealthy. Trying another')
774 update_url, devserver = self._stage_image_for_update(
775 requested_build)
776 logging.debug('New Update URL is %s', update_url)
777 updater = autoupdater.ChromiumOSUpdater(
778 update_url, host=self,
779 local_devserver=local_devserver)
780 else:
781 break
J. Richard Barnette45e93de2012-04-11 17:24:15 -0700782
Chris Sosae92399e2015-04-24 11:32:59 -0700783 # Give it some time in case of IO issues.
784 time.sleep(10)
Dan Shi5699ac22014-12-19 10:55:49 -0800785
Chris Sosae92399e2015-04-24 11:32:59 -0700786 # Figure out active and inactive kernel.
787 active_kernel, inactive_kernel = updater.get_kernel_state()
Simran Basi13fa1ba2013-03-04 10:56:47 -0800788
Chris Sosae92399e2015-04-24 11:32:59 -0700789 # Ensure inactive kernel has higher priority than active.
790 if (updater.get_kernel_priority(inactive_kernel)
791 < updater.get_kernel_priority(active_kernel)):
792 raise autoupdater.ChromiumOSError(
793 'Update failed. The priority of the inactive kernel'
794 ' partition is less than that of the active kernel'
795 ' partition.')
796
797 # Updater has returned successfully; reboot the host.
798 self.reboot(timeout=self.REBOOT_TIMEOUT, wait=True)
799
800 self._post_update_processing(updater, inactive_kernel)
801 self.add_cros_version_labels_and_job_repo_url(
802 autoupdater.url_to_image_name(update_url))
J. Richard Barnette45e93de2012-04-11 17:24:15 -0700803
804
Dan Shi9cb0eec2014-06-03 09:04:50 -0700805 def _clear_fw_version_labels(self):
806 """Clear firmware version labels from the machine."""
807 labels = self._AFE.get_labels(
Dan Shi0723bf52015-06-24 10:52:38 -0700808 name__startswith=provision.FW_RW_VERSION_PREFIX,
Dan Shi9cb0eec2014-06-03 09:04:50 -0700809 host__hostname=self.hostname)
810 for label in labels:
811 label.remove_hosts(hosts=[self.hostname])
812
813
814 def _add_fw_version_label(self, build):
815 """Add firmware version label to the machine.
816
817 @param build: Build of firmware.
818
819 """
820 fw_label = provision.fw_version_to_label(build)
MK Ryu73be9862015-07-06 12:25:00 -0700821 self._AFE.run('label_add_hosts', id=fw_label, hosts=[self.hostname])
Dan Shi9cb0eec2014-06-03 09:04:50 -0700822
823
824 def firmware_install(self, build=None):
825 """Install firmware to the DUT.
826
827 Use stateful update if the DUT is already running the same build.
828 Stateful update does not update kernel and tends to run much faster
829 than a full reimage. If the DUT is running a different build, or it
830 failed to do a stateful update, full update, including kernel update,
831 will be applied to the DUT.
832
833 Once a host enters firmware_install its fw_version label will be
834 removed. After the firmware is updated successfully, a new fw_version
835 label will be added to the host.
836
837 @param build: The build version to which we want to provision the
838 firmware of the machine,
839 e.g. 'link-firmware/R22-2695.1.144'.
840
841 TODO(dshi): After bug 381718 is fixed, update here with corresponding
842 exceptions that could be raised.
843
844 """
845 if not self.servo:
846 raise error.TestError('Host %s does not have servo.' %
847 self.hostname)
848
849 # TODO(fdeng): use host.get_board() after
850 # crbug.com/271834 is fixed.
851 board = self._get_board_from_afe()
852
Chris Sosae92399e2015-04-24 11:32:59 -0700853 # If build is not set, try to install firmware from stable CrOS.
Dan Shi9cb0eec2014-06-03 09:04:50 -0700854 if not build:
855 build = self.get_repair_image_name()
856
857 config = FAFTConfig(board)
858 if config.use_u_boot:
859 ap_image = 'image-%s.bin' % board
860 else: # Depthcharge platform
861 ap_image = 'image.bin'
862 ec_image = 'ec.bin'
863 ds = dev_server.ImageServer.resolve(build)
864 ds.stage_artifacts(build, ['firmware'])
865
866 tmpd = autotemp.tempdir(unique_id='fwimage')
867 try:
868 fwurl = self._FW_IMAGE_URL_PATTERN % (ds.url(), build)
869 local_tarball = os.path.join(tmpd.name, os.path.basename(fwurl))
870 server_utils.system('wget -O %s %s' % (local_tarball, fwurl),
871 timeout=60)
872 server_utils.system('tar xf %s -C %s %s %s' %
873 (local_tarball, tmpd.name, ap_image, ec_image),
874 timeout=60)
875 server_utils.system('tar xf %s --wildcards -C %s "dts/*"' %
876 (local_tarball, tmpd.name),
877 timeout=60, ignore_status=True)
878
879 self._clear_fw_version_labels()
880 logging.info('Will re-program EC now')
881 self.servo.program_ec(os.path.join(tmpd.name, ec_image))
882 logging.info('Will re-program BIOS now')
883 self.servo.program_bios(os.path.join(tmpd.name, ap_image))
884 self.servo.get_power_state_controller().reset()
885 time.sleep(self.servo.BOOT_DELAY)
Dan Shia5fef052015-05-18 23:28:47 -0700886 self._add_fw_version_label(build)
Dan Shi9cb0eec2014-06-03 09:04:50 -0700887 finally:
888 tmpd.clean()
889
890
Dan Shi10e992b2013-08-30 11:02:59 -0700891 def show_update_engine_log(self):
892 """Output update engine log."""
MK Ryu35d661e2014-09-25 17:44:10 -0700893 logging.debug('Dumping %s', client_constants.UPDATE_ENGINE_LOG)
894 self.run('cat %s' % client_constants.UPDATE_ENGINE_LOG)
Dan Shi10e992b2013-08-30 11:02:59 -0700895
896
Richard Barnette82c35912012-11-20 10:09:10 -0800897 def _get_board_from_afe(self):
898 """Retrieve this host's board from its labels in the AFE.
899
900 Looks for a host label of the form "board:<board>", and
901 returns the "<board>" part of the label. `None` is returned
902 if there is not a single, unique label matching the pattern.
903
904 @returns board from label, or `None`.
905 """
Dan Shia1ecd5c2013-06-06 11:21:31 -0700906 return server_utils.get_board_from_afe(self.hostname, self._AFE)
Simran Basi833814b2013-01-29 13:13:43 -0800907
908
909 def get_build(self):
910 """Retrieve the current build for this Host from the AFE.
911
912 Looks through this host's labels in the AFE to determine its build.
913
914 @returns The current build or None if it could not find it or if there
915 were multiple build labels assigned to this host.
916 """
Dan Shia1ecd5c2013-06-06 11:21:31 -0700917 return server_utils.get_build_from_afe(self.hostname, self._AFE)
Richard Barnette82c35912012-11-20 10:09:10 -0800918
919
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500920 def _install_repair(self):
Chris Sosae92399e2015-04-24 11:32:59 -0700921 """Attempt to repair this host using the update-engine.
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500922
923 If the host is up, try installing the DUT with a stable
Dan Shi6964fa52014-12-18 11:04:27 -0800924 "repair" version of Chrome OS as defined in afe_stable_versions table.
925 If the table is not setup, global_config value under
926 CROS.stable_cros_version will be used instead.
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500927
Scott Zawalski62bacae2013-03-05 10:40:32 -0500928 @raises AutoservRepairMethodNA if the DUT is not reachable.
929 @raises ChromiumOSError if the install failed for some reason.
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500930
931 """
932 if not self.is_up():
Scott Zawalski62bacae2013-03-05 10:40:32 -0500933 raise error.AutoservRepairMethodNA('DUT unreachable for install.')
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500934 logging.info('Attempting to reimage machine to repair image.')
935 try:
Richard Barnette0b023a72015-04-24 16:07:30 +0000936 self.machine_install(repair=True)
Fang Dengd0672f32013-03-18 17:18:09 -0700937 except autoupdater.ChromiumOSError as e:
938 logging.exception(e)
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500939 logging.info('Repair via install failed.')
Scott Zawalski62bacae2013-03-05 10:40:32 -0500940 raise
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500941
942
Dan Shi2c88eed2013-11-12 10:18:38 -0800943 def _install_repair_with_powerwash(self):
Dan Shi9cc48452013-11-12 12:39:26 -0800944 """Attempt to powerwash first then repair this host using update-engine.
Dan Shi2c88eed2013-11-12 10:18:38 -0800945
Dan Shi9cc48452013-11-12 12:39:26 -0800946 update-engine may fail due to a bad image. In such case, powerwash
947 may help to cleanup the DUT for update-engine to work again.
Dan Shi2c88eed2013-11-12 10:18:38 -0800948
949 @raises AutoservRepairMethodNA if the DUT is not reachable.
950 @raises ChromiumOSError if the install failed for some reason.
951
952 """
953 if not self.is_up():
954 raise error.AutoservRepairMethodNA('DUT unreachable for install.')
955
956 logging.info('Attempting to powerwash the DUT.')
957 self.run('echo "fast safe" > '
958 '/mnt/stateful_partition/factory_install_reset')
959 self.reboot(timeout=self.POWERWASH_BOOT_TIMEOUT, wait=True)
960 if not self.is_up():
Dan Shi9cc48452013-11-12 12:39:26 -0800961 logging.error('Powerwash failed. DUT did not come back after '
Dan Shi2c88eed2013-11-12 10:18:38 -0800962 'reboot.')
963 raise error.AutoservRepairFailure(
964 'DUT failed to boot from powerwash after %d seconds' %
965 self.POWERWASH_BOOT_TIMEOUT)
966
967 logging.info('Powerwash succeeded.')
968 self._install_repair()
969
970
beepsf079cfb2013-09-18 17:49:51 -0700971 def servo_install(self, image_url=None, usb_boot_timeout=USB_BOOT_TIMEOUT,
972 install_timeout=INSTALL_TIMEOUT):
Scott Zawalski62bacae2013-03-05 10:40:32 -0500973 """
974 Re-install the OS on the DUT by:
975 1) installing a test image on a USB storage device attached to the Servo
976 board,
Richard Barnette03a0c132012-11-05 12:40:35 -0800977 2) booting that image in recovery mode, and then
J. Richard Barnette31b2e312013-04-04 16:05:22 -0700978 3) installing the image with chromeos-install.
979
Scott Zawalski62bacae2013-03-05 10:40:32 -0500980 @param image_url: If specified use as the url to install on the DUT.
981 otherwise boot the currently staged image on the USB stick.
beepsf079cfb2013-09-18 17:49:51 -0700982 @param usb_boot_timeout: The usb_boot_timeout to use during reimage.
983 Factory images need a longer usb_boot_timeout than regular
984 cros images.
985 @param install_timeout: The timeout to use when installing the chromeos
986 image. Factory images need a longer install_timeout.
Richard Barnette03a0c132012-11-05 12:40:35 -0800987
Scott Zawalski62bacae2013-03-05 10:40:32 -0500988 @raises AutoservError if the image fails to boot.
beepsf079cfb2013-09-18 17:49:51 -0700989
J. Richard Barnette0199cc82014-12-05 17:08:40 -0800990 """
beepsf079cfb2013-09-18 17:49:51 -0700991 usb_boot_timer_key = ('servo_install.usb_boot_timeout_%s'
992 % usb_boot_timeout)
993 logging.info('Downloading image to USB, then booting from it. Usb boot '
994 'timeout = %s', usb_boot_timeout)
Gabe Black1e1c41b2015-02-04 23:55:15 -0800995 timer = autotest_stats.Timer(usb_boot_timer_key)
beepsf079cfb2013-09-18 17:49:51 -0700996 timer.start()
J. Richard Barnette31b2e312013-04-04 16:05:22 -0700997 self.servo.install_recovery_image(image_url)
beepsf079cfb2013-09-18 17:49:51 -0700998 if not self.wait_up(timeout=usb_boot_timeout):
Scott Zawalski62bacae2013-03-05 10:40:32 -0500999 raise error.AutoservRepairFailure(
1000 'DUT failed to boot from USB after %d seconds' %
beepsf079cfb2013-09-18 17:49:51 -07001001 usb_boot_timeout)
1002 timer.stop()
Scott Zawalski62bacae2013-03-05 10:40:32 -05001003
Tom Wai-Hong Tamf6b4f812015-08-08 04:14:59 +08001004 # The new chromeos-tpm-recovery has been merged since R44-7073.0.0.
1005 # In old CrOS images, this command fails. Skip the error.
Tom Wai-Hong Tam27af7332015-07-25 06:09:39 +08001006 logging.info('Resetting the TPM status')
Tom Wai-Hong Tamf6b4f812015-08-08 04:14:59 +08001007 try:
1008 self.run('chromeos-tpm-recovery')
1009 except error.AutoservRunError:
1010 logging.warn('chromeos-tpm-recovery is too old.')
1011
Tom Wai-Hong Tam27af7332015-07-25 06:09:39 +08001012
beepsf079cfb2013-09-18 17:49:51 -07001013 install_timer_key = ('servo_install.install_timeout_%s'
1014 % install_timeout)
Gabe Black1e1c41b2015-02-04 23:55:15 -08001015 timer = autotest_stats.Timer(install_timer_key)
beepsf079cfb2013-09-18 17:49:51 -07001016 timer.start()
1017 logging.info('Installing image through chromeos-install.')
J. Richard Barnette9af19632015-09-25 12:18:03 -07001018 self.run('chromeos-install --yes', timeout=install_timeout)
1019 self.halt()
beepsf079cfb2013-09-18 17:49:51 -07001020 timer.stop()
1021
1022 logging.info('Power cycling DUT through servo.')
J. Richard Barnette0199cc82014-12-05 17:08:40 -08001023 self.servo.get_power_state_controller().power_off()
Fang Dengafb88142013-05-30 17:44:31 -07001024 self.servo.switch_usbkey('off')
J. Richard Barnette0199cc82014-12-05 17:08:40 -08001025 # N.B. The Servo API requires that we use power_on() here
1026 # for two reasons:
1027 # 1) After turning on a DUT in recovery mode, you must turn
1028 # it off and then on with power_on() once more to
1029 # disable recovery mode (this is a Parrot specific
1030 # requirement).
1031 # 2) After power_off(), the only way to turn on is with
1032 # power_on() (this is a Storm specific requirement).
J. Richard Barnettefbcc7122013-07-24 18:24:59 -07001033 self.servo.get_power_state_controller().power_on()
beepsf079cfb2013-09-18 17:49:51 -07001034
1035 logging.info('Waiting for DUT to come back up.')
Richard Barnette03a0c132012-11-05 12:40:35 -08001036 if not self.wait_up(timeout=self.BOOT_TIMEOUT):
1037 raise error.AutoservError('DUT failed to reboot installed '
1038 'test image after %d seconds' %
Scott Zawalski62bacae2013-03-05 10:40:32 -05001039 self.BOOT_TIMEOUT)
1040
1041
Dan Shic1b8bdd2015-09-14 23:11:24 -07001042 def _setup_servo(self):
1043 """Try to force to create servo object if it's not set up yet.
1044 """
1045 if self.servo:
1046 return
1047
1048 try:
1049 # Setting servo_args to {} will force it to create the servo_host
1050 # object if possible.
1051 self._servo_host = servo_host.create_servo_host(
1052 dut=self.hostname, servo_args={})
1053 if self._servo_host:
1054 self.servo = self._servo_host.get_servo()
1055 else:
1056 logging.error('Failed to create servo_host object.')
1057 except Exception as e:
1058 logging.error('Failed to create servo object: %s', e)
1059
1060
J. Richard Barnettee4af8b92013-05-01 13:16:12 -07001061 def _servo_repair_reinstall(self):
Scott Zawalski62bacae2013-03-05 10:40:32 -05001062 """Reinstall the DUT utilizing servo and a test image.
1063
1064 Re-install the OS on the DUT by:
1065 1) installing a test image on a USB storage device attached to the Servo
1066 board,
Tom Wai-Hong Tam27af7332015-07-25 06:09:39 +08001067 2) booting that image in recovery mode,
1068 3) resetting the TPM status, and then
1069 4) installing the image with chromeos-install.
Scott Zawalski62bacae2013-03-05 10:40:32 -05001070
Scott Zawalski62bacae2013-03-05 10:40:32 -05001071 @raises AutoservRepairMethodNA if the device does not have servo
1072 support.
1073
1074 """
Dan Shic1b8bdd2015-09-14 23:11:24 -07001075 # To repair a DUT connected to a moblab, try to create a servo object if
1076 # it was failed to be created earlier as there may be a servo_host host
1077 # attribute for this host.
1078 if utils.is_moblab():
1079 self._setup_servo()
1080
Scott Zawalski62bacae2013-03-05 10:40:32 -05001081 if not self.servo:
1082 raise error.AutoservRepairMethodNA('Repair Reinstall NA: '
1083 'DUT has no servo support.')
1084
1085 logging.info('Attempting to recovery servo enabled device with '
1086 'servo_repair_reinstall')
1087
J. Richard Barnettee4af8b92013-05-01 13:16:12 -07001088 image_url = self.stage_image_for_servo()
Scott Zawalski62bacae2013-03-05 10:40:32 -05001089 self.servo_install(image_url)
1090
1091
1092 def _servo_repair_power(self):
1093 """Attempt to repair DUT using an attached Servo.
1094
1095 Attempt to power on the DUT via power_long_press.
1096
1097 @raises AutoservRepairMethodNA if the device does not have servo
1098 support.
1099 @raises AutoservRepairFailure if the repair fails for any reason.
1100 """
1101 if not self.servo:
1102 raise error.AutoservRepairMethodNA('Repair Power NA: '
1103 'DUT has no servo support.')
1104
1105 logging.info('Attempting to recover servo enabled device by '
1106 'powering it off and on.')
1107 self.servo.get_power_state_controller().power_off()
1108 self.servo.get_power_state_controller().power_on()
1109 if self.wait_up(self.BOOT_TIMEOUT):
1110 return
1111
1112 raise error.AutoservRepairFailure('DUT did not boot after long_press.')
Richard Barnette03a0c132012-11-05 12:40:35 -08001113
1114
Richard Barnette82c35912012-11-20 10:09:10 -08001115 def _powercycle_to_repair(self):
1116 """Utilize the RPM Infrastructure to bring the host back up.
1117
1118 If the host is not up/repaired after the first powercycle we utilize
1119 auto fallback to the last good install by powercycling and rebooting the
1120 host 6 times.
Scott Zawalski62bacae2013-03-05 10:40:32 -05001121
1122 @raises AutoservRepairMethodNA if the device does not support remote
1123 power.
1124 @raises AutoservRepairFailure if the repair fails for any reason.
1125
Richard Barnette82c35912012-11-20 10:09:10 -08001126 """
Scott Zawalski62bacae2013-03-05 10:40:32 -05001127 if not self.has_power():
1128 raise error.AutoservRepairMethodNA('Device does not support power.')
1129
Richard Barnette82c35912012-11-20 10:09:10 -08001130 logging.info('Attempting repair via RPM powercycle.')
1131 failed_cycles = 0
1132 self.power_cycle()
1133 while not self.wait_up(timeout=self.BOOT_TIMEOUT):
1134 failed_cycles += 1
1135 if failed_cycles >= self._MAX_POWER_CYCLE_ATTEMPTS:
Scott Zawalski62bacae2013-03-05 10:40:32 -05001136 raise error.AutoservRepairFailure(
1137 'Powercycled host %s %d times; device did not come back'
1138 ' online.' % (self.hostname, failed_cycles))
Richard Barnette82c35912012-11-20 10:09:10 -08001139 self.power_cycle()
1140 if failed_cycles == 0:
1141 logging.info('Powercycling was successful first time.')
1142 else:
1143 logging.info('Powercycling was successful after %d failures.',
1144 failed_cycles)
1145
1146
MK Ryu35d661e2014-09-25 17:44:10 -07001147 def _reboot_repair(self):
1148 """SSH to this host and reboot."""
1149 if not self.is_up(self._CHECK_HOST_UP_TIMEOUT_SECS):
1150 raise error.AutoservRepairMethodNA('DUT unreachable for reboot.')
1151 logging.info('Attempting repair via SSH reboot.')
1152 self.reboot(timeout=self.BOOT_TIMEOUT, wait=True)
1153
1154
Prashanth B4d8184f2014-05-05 12:22:02 -07001155 def check_device(self):
1156 """Check if a device is ssh-able, and if so, clean and verify it.
1157
1158 @raise AutoservSSHTimeout: If the ssh ping times out.
1159 @raise AutoservSshPermissionDeniedError: If ssh ping fails due to
1160 permissions.
1161 @raise AutoservSshPingHostError: For other AutoservRunErrors during
1162 ssh_ping.
1163 @raises AutoservError: As appropriate, during cleanup and verify.
1164 """
1165 self.ssh_ping()
1166 self.cleanup()
1167 self.verify()
1168
1169
Dan Shi90466352015-09-22 15:01:05 -07001170 def confirm_servo(self):
1171 """Confirm servo is initialized and verified.
1172
1173 @raise AutoservError: If servo is not initialized and verified.
1174 """
1175 if self._servo_host.required_by_test and self.servo:
1176 return
1177
1178 # Force to re-create the servo object to make sure servo is verified.
1179 logging.debug('Rebuilding the servo object.')
1180 self.servo = None
1181 self._servo_host = None
1182 self._setup_servo()
1183 if not self.servo:
1184 raise error.AutoservError('Failed to create servo object.')
1185
1186
Dan Shid07ee2e2015-09-24 14:49:25 -07001187 def _is_last_provision_failed(self):
1188 """Checks if the last provision job failed.
1189
1190 @return: True if there exists file /var/tmp/provision_failed, which
1191 indicates the last provision job failed.
1192 False if the file does not exist or the dut can't be reached.
1193 """
1194 try:
1195 result = self.run('test -f %s' % PROVISION_FAILED,
1196 ignore_status=True, timeout=5)
1197 return result.exit_status == 0
1198 except (error.AutoservRunError, error.AutoservSSHTimeout):
1199 # Default to False, for repair to try all repair method if the dut
1200 # can't be reached.
1201 return False
1202
1203
Richard Barnette82c35912012-11-20 10:09:10 -08001204 def repair_full(self):
1205 """Repair a host for repair level NO_PROTECTION.
1206
1207 This overrides the base class function for repair; it does
1208 not call back to the parent class, but instead offers a
1209 simplified implementation based on the capabilities in the
1210 Chrome OS test lab.
1211
Fang Deng5d518f42013-08-02 14:04:32 -07001212 It first verifies and repairs servo if it is a DUT in CrOS
Fang Deng03590af2013-10-07 17:34:20 -07001213 lab and a servo is attached.
Fang Deng5d518f42013-08-02 14:04:32 -07001214
Jakob Juelich82b7d1c2014-09-15 16:10:57 -07001215 This escalates in order through the following procedures and verifies
1216 the status using `self.check_device()` after each of them. This is done
1217 until both the repair and the veryfing step succeed.
1218
MK Ryu35d661e2014-09-25 17:44:10 -07001219 Escalation order of repair procedures from less intrusive to
1220 more intrusive repairs:
1221 1. SSH to the DUT and reboot.
Scott Zawalski62bacae2013-03-05 10:40:32 -05001222 2. If there's a servo for the DUT, try to power the DUT off and
1223 on.
MK Ryu35d661e2014-09-25 17:44:10 -07001224 3. If the DUT can be power-cycled via RPM, try to repair
Richard Barnette82c35912012-11-20 10:09:10 -08001225 by power-cycling.
MK Ryu35d661e2014-09-25 17:44:10 -07001226 4. Try to re-install to a known stable image using
1227 auto-update.
1228 5. If there's a servo for the DUT, try to re-install via
1229 the servo.
Richard Barnette82c35912012-11-20 10:09:10 -08001230
1231 As with the parent method, the last operation performed on
Prashanth B4d8184f2014-05-05 12:22:02 -07001232 the DUT must be to call `self.check_device()`; If that call fails the
1233 exception it raises is passed back to the caller.
J. Richard Barnettefde55fc2013-03-15 17:47:01 -07001234
Scott Zawalski62bacae2013-03-05 10:40:32 -05001235 @raises AutoservRepairTotalFailure if the repair process fails to
1236 fix the DUT.
Fang Deng5d518f42013-08-02 14:04:32 -07001237 @raises ServoHostRepairTotalFailure if the repair process fails to
1238 fix the servo host if one is attached to the DUT.
1239 @raises AutoservSshPermissionDeniedError if it is unable
1240 to ssh to the servo host due to permission error.
1241
Richard Barnette82c35912012-11-20 10:09:10 -08001242 """
Jakob Juelich82b7d1c2014-09-15 16:10:57 -07001243 # Caution: Deleting shards relies on repair to always reboot the DUT.
1244
Dan Shi4d478522014-02-14 13:46:32 -08001245 if self._servo_host and not self.servo:
Fang Deng03590af2013-10-07 17:34:20 -07001246 try:
Dan Shi4d478522014-02-14 13:46:32 -08001247 self._servo_host.repair_full()
Fang Deng03590af2013-10-07 17:34:20 -07001248 except Exception as e:
Fang Deng03590af2013-10-07 17:34:20 -07001249 logging.error('Could not create a healthy servo: %s', e)
Dan Shi4d478522014-02-14 13:46:32 -08001250 self.servo = self._servo_host.get_servo()
Fang Deng5d518f42013-08-02 14:04:32 -07001251
MK Ryu35d661e2014-09-25 17:44:10 -07001252 self.try_collect_crashlogs()
1253
Scott Zawalski62bacae2013-03-05 10:40:32 -05001254 # TODO(scottz): This should use something similar to label_decorator,
1255 # but needs to be populated in order so DUTs are repaired with the
1256 # least amount of effort.
Dan Shid07ee2e2015-09-24 14:49:25 -07001257 force_powerwash = self._is_last_provision_failed()
1258 if force_powerwash:
1259 logging.info('Last provision failed, try powerwash first.')
1260 autotest_stats.Counter(
1261 'repair_force_powerwash.TOTAL').increment()
1262 repair_funcs = [self._install_repair_with_powerwash,
1263 self._servo_repair_reinstall]
1264 else:
1265 repair_funcs = [self._reboot_repair,
1266 self._servo_repair_power,
1267 self._powercycle_to_repair,
1268 self._install_repair,
1269 self._install_repair_with_powerwash,
1270 self._servo_repair_reinstall]
Scott Zawalski62bacae2013-03-05 10:40:32 -05001271 errors = []
Simran Basie6130932013-10-01 14:07:52 -07001272 board = self._get_board_from_afe()
Scott Zawalski62bacae2013-03-05 10:40:32 -05001273 for repair_func in repair_funcs:
1274 try:
1275 repair_func()
MK Ryu35d661e2014-09-25 17:44:10 -07001276 self.try_collect_crashlogs()
Prashanth B4d8184f2014-05-05 12:22:02 -07001277 self.check_device()
Gabe Black1e1c41b2015-02-04 23:55:15 -08001278 autotest_stats.Counter(
Simran Basie6130932013-10-01 14:07:52 -07001279 '%s.SUCCEEDED' % repair_func.__name__).increment()
1280 if board:
Gabe Black1e1c41b2015-02-04 23:55:15 -08001281 autotest_stats.Counter(
Dan Shid07ee2e2015-09-24 14:49:25 -07001282 '%s.%s.SUCCEEDED' % (repair_func.__name__,
Simran Basie6130932013-10-01 14:07:52 -07001283 board)).increment()
Dan Shid07ee2e2015-09-24 14:49:25 -07001284 if force_powerwash:
1285 autotest_stats.Counter(
1286 'repair_force_powerwash.SUCCEEDED').increment()
Scott Zawalski62bacae2013-03-05 10:40:32 -05001287 return
Simran Basie6130932013-10-01 14:07:52 -07001288 except error.AutoservRepairMethodNA as e:
Gabe Black1e1c41b2015-02-04 23:55:15 -08001289 autotest_stats.Counter(
Simran Basie6130932013-10-01 14:07:52 -07001290 '%s.RepairNA' % repair_func.__name__).increment()
1291 if board:
Gabe Black1e1c41b2015-02-04 23:55:15 -08001292 autotest_stats.Counter(
Dan Shid07ee2e2015-09-24 14:49:25 -07001293 '%s.%s.RepairNA' % (repair_func.__name__,
1294 board)).increment()
Ilja H. Friedel04be2bd2014-05-07 21:29:59 -07001295 logging.warning('Repair function NA: %s', e)
Simran Basie6130932013-10-01 14:07:52 -07001296 errors.append(str(e))
Scott Zawalski62bacae2013-03-05 10:40:32 -05001297 except Exception as e:
Gabe Black1e1c41b2015-02-04 23:55:15 -08001298 autotest_stats.Counter(
Simran Basie6130932013-10-01 14:07:52 -07001299 '%s.FAILED' % repair_func.__name__).increment()
1300 if board:
Gabe Black1e1c41b2015-02-04 23:55:15 -08001301 autotest_stats.Counter(
Dan Shid07ee2e2015-09-24 14:49:25 -07001302 '%s.%s.FAILED' % (repair_func.__name__,
1303 board)).increment()
Ilja H. Friedel04be2bd2014-05-07 21:29:59 -07001304 logging.warning('Failed to repair device: %s', e)
Scott Zawalski62bacae2013-03-05 10:40:32 -05001305 errors.append(str(e))
Scott Zawalski89c44dd2013-02-26 09:28:02 -05001306
Dan Shid07ee2e2015-09-24 14:49:25 -07001307 if force_powerwash:
1308 autotest_stats.Counter(
1309 'repair_force_powerwash.FAILED').increment()
Gabe Black1e1c41b2015-02-04 23:55:15 -08001310 autotest_stats.Counter('Full_Repair_Failed').increment()
Simran Basie6130932013-10-01 14:07:52 -07001311 if board:
Gabe Black1e1c41b2015-02-04 23:55:15 -08001312 autotest_stats.Counter(
Dan Shid07ee2e2015-09-24 14:49:25 -07001313 'Full_Repair_Failed.%s' % board).increment()
Scott Zawalski62bacae2013-03-05 10:40:32 -05001314 raise error.AutoservRepairTotalFailure(
1315 'All attempts at repairing the device failed:\n%s' %
1316 '\n'.join(errors))
Richard Barnette82c35912012-11-20 10:09:10 -08001317
1318
MK Ryu35d661e2014-09-25 17:44:10 -07001319 def try_collect_crashlogs(self, check_host_up=True):
1320 """
1321 Check if a host is up and logs need to be collected from the host,
1322 if yes, collect them.
1323
1324 @param check_host_up: Flag for checking host is up. Default is True.
1325 """
1326 try:
1327 crash_job = self._need_crash_logs()
1328 if crash_job:
1329 logging.debug('%s: Job %s was crashed', self._CRASHLOGS_PREFIX,
1330 crash_job)
1331 if not check_host_up or self.is_up(
1332 self._CHECK_HOST_UP_TIMEOUT_SECS):
1333 self._collect_crashlogs(crash_job)
1334 logging.debug('%s: Completed collecting logs for the '
1335 'crashed job %s', self._CRASHLOGS_PREFIX,
1336 crash_job)
1337 except Exception as e:
1338 # Exception should not result in repair failure.
1339 # Therefore, suppress all exceptions here.
1340 logging.error('%s: Failed while trying to collect crash-logs: %s',
1341 self._CRASHLOGS_PREFIX, e)
1342
1343
1344 def _need_crash_logs(self):
1345 """Get the value of need_crash_logs attribute of this host.
1346
1347 @return: Value string of need_crash_logs attribute
1348 None if there is no need_crash_logs attribute
1349 """
1350 attrs = self._AFE.get_host_attribute(constants.CRASHLOGS_HOST_ATTRIBUTE,
1351 hostname=self.hostname)
1352 assert len(attrs) < 2
1353 return attrs[0].value if attrs else None
1354
1355
1356 def _collect_crashlogs(self, job_id):
1357 """Grab logs from the host where a job was crashed.
1358
1359 First, check if PRIOR_LOGS_DIR exists in the host.
1360 If yes, collect them.
1361 Otherwise, check if a lab-machine marker (_LAB_MACHINE_FILE) exists
1362 in the host.
1363 If yes, the host was repaired automatically, and we collect normal
1364 system logs.
1365
1366 @param job_id: Id of the job that was crashed.
1367 """
1368 crashlogs_dir = crashcollect.get_crashinfo_dir(self,
1369 constants.CRASHLOGS_DEST_DIR_PREFIX)
1370 flag_prior_logs = False
1371
1372 if self.path_exists(client_constants.PRIOR_LOGS_DIR):
1373 flag_prior_logs = True
1374 self._collect_prior_logs(crashlogs_dir)
1375 elif self.path_exists(self._LAB_MACHINE_FILE):
1376 self._collect_system_logs(crashlogs_dir)
1377 else:
1378 logging.warning('%s: Host was manually re-installed without '
1379 '--lab_preserve_log option. Skip collecting '
1380 'crash-logs.', self._CRASHLOGS_PREFIX)
1381
1382 # We make crash collection be one-time effort.
1383 # _collect_prior_logs() and _collect_system_logs() will not throw
1384 # any exception, and following codes will be executed even when
1385 # those methods fail.
1386 # _collect_crashlogs() is called only when the host is up (refer
1387 # to try_collect_crashlogs()). We assume _collect_prior_logs() and
1388 # _collect_system_logs() fail rarely when the host is up.
1389 # In addition, it is not clear how many times we should try crash
1390 # collection again while not triggering next repair unnecessarily.
1391 # Threfore, we try crash collection one time.
1392
1393 # Create a marker file as soon as log collection is done.
1394 # Leave the job id to this marker for gs_offloader to consume.
1395 marker_file = os.path.join(crashlogs_dir, constants.CRASHLOGS_MARKER)
1396 with open(marker_file, 'a') as f:
1397 f.write('%s\n' % job_id)
1398
1399 # Remove need_crash_logs attribute
1400 logging.debug('%s: Remove attribute need_crash_logs from host %s',
1401 self._CRASHLOGS_PREFIX, self.hostname)
1402 self._AFE.set_host_attribute(constants.CRASHLOGS_HOST_ATTRIBUTE,
1403 None, hostname=self.hostname)
1404
1405 if flag_prior_logs:
1406 logging.debug('%s: Remove %s from host %s', self._CRASHLOGS_PREFIX,
1407 client_constants.PRIOR_LOGS_DIR, self.hostname)
1408 self.run('rm -rf %s; sync' % client_constants.PRIOR_LOGS_DIR)
1409 # Wait for a few seconds to make sure the prior command is
1410 # done deep through storage.
1411 time.sleep(self._SAFE_WAIT_SECS)
1412
1413
1414 def _collect_prior_logs(self, crashlogs_dir):
1415 """Grab prior logs that were stashed before re-installing a host.
1416
1417 @param crashlogs_dir: Directory path where crash-logs are stored.
1418 """
1419 logging.debug('%s: Found %s, collecting them...',
1420 self._CRASHLOGS_PREFIX, client_constants.PRIOR_LOGS_DIR)
1421 try:
1422 self.collect_logs(client_constants.PRIOR_LOGS_DIR,
1423 crashlogs_dir, False)
1424 logging.debug('%s: %s is collected',
1425 self._CRASHLOGS_PREFIX, client_constants.PRIOR_LOGS_DIR)
1426 except Exception as e:
1427 logging.error('%s: Failed to collect %s: %s',
1428 self._CRASHLOGS_PREFIX, client_constants.PRIOR_LOGS_DIR,
1429 e)
1430
1431
1432 def _collect_system_logs(self, crashlogs_dir):
1433 """Grab normal system logs from a host.
1434
1435 @param crashlogs_dir: Directory path where crash-logs are stored.
1436 """
1437 logging.debug('%s: Found %s, collecting system logs...',
1438 self._CRASHLOGS_PREFIX, self._LAB_MACHINE_FILE)
1439 sources = server_utils.parse_simple_config(self._LOGS_TO_COLLECT_FILE)
1440 for src in sources:
1441 try:
1442 if self.path_exists(src):
1443 logging.debug('%s: Collecting %s...',
1444 self._CRASHLOGS_PREFIX, src)
1445 dest = server_utils.concat_path_except_last(
1446 crashlogs_dir, src)
1447 self.collect_logs(src, dest, False)
1448 logging.debug('%s: %s is collected',
1449 self._CRASHLOGS_PREFIX, src)
1450 except Exception as e:
1451 logging.error('%s: Failed to collect %s: %s',
1452 self._CRASHLOGS_PREFIX, src, e)
1453
1454
J. Richard Barnette1d78b012012-05-15 13:56:30 -07001455 def close(self):
beeps32a63082013-08-22 14:02:29 -07001456 self.rpc_disconnect_all()
Fang Deng0ca40e22013-08-27 17:47:44 -07001457 super(CrosHost, self).close()
J. Richard Barnette1d78b012012-05-15 13:56:30 -07001458
1459
Dan Shi49ca0932014-11-14 11:22:27 -08001460 def get_power_supply_info(self):
1461 """Get the output of power_supply_info.
1462
1463 power_supply_info outputs the info of each power supply, e.g.,
1464 Device: Line Power
1465 online: no
1466 type: Mains
1467 voltage (V): 0
1468 current (A): 0
1469 Device: Battery
1470 state: Discharging
1471 percentage: 95.9276
1472 technology: Li-ion
1473
1474 Above output shows two devices, Line Power and Battery, with details of
1475 each device listed. This function parses the output into a dictionary,
1476 with key being the device name, and value being a dictionary of details
1477 of the device info.
1478
1479 @return: The dictionary of power_supply_info, e.g.,
1480 {'Line Power': {'online': 'yes', 'type': 'main'},
1481 'Battery': {'vendor': 'xyz', 'percentage': '100'}}
Dan Shie9b765d2014-12-29 16:59:49 -08001482 @raise error.AutoservRunError if power_supply_info tool is not found in
1483 the DUT. Caller should handle this error to avoid false failure
1484 on verification.
Dan Shi49ca0932014-11-14 11:22:27 -08001485 """
1486 result = self.run('power_supply_info').stdout.strip()
1487 info = {}
1488 device_name = None
1489 device_info = {}
1490 for line in result.split('\n'):
1491 pair = [v.strip() for v in line.split(':')]
1492 if len(pair) != 2:
1493 continue
1494 if pair[0] == 'Device':
1495 if device_name:
1496 info[device_name] = device_info
1497 device_name = pair[1]
1498 device_info = {}
1499 else:
1500 device_info[pair[0]] = pair[1]
1501 if device_name and not device_name in info:
1502 info[device_name] = device_info
1503 return info
1504
1505
1506 def get_battery_percentage(self):
1507 """Get the battery percentage.
1508
1509 @return: The percentage of battery level, value range from 0-100. Return
1510 None if the battery info cannot be retrieved.
1511 """
1512 try:
1513 info = self.get_power_supply_info()
1514 logging.info(info)
1515 return float(info['Battery']['percentage'])
Dan Shie9b765d2014-12-29 16:59:49 -08001516 except (KeyError, ValueError, error.AutoservRunError):
Dan Shi49ca0932014-11-14 11:22:27 -08001517 return None
1518
1519
1520 def is_ac_connected(self):
1521 """Check if the dut has power adapter connected and charging.
1522
1523 @return: True if power adapter is connected and charging.
1524 """
1525 try:
1526 info = self.get_power_supply_info()
1527 return info['Line Power']['online'] == 'yes'
Dan Shie9b765d2014-12-29 16:59:49 -08001528 except (KeyError, error.AutoservRunError):
1529 return None
Dan Shi49ca0932014-11-14 11:22:27 -08001530
1531
Simran Basi5e6339a2013-03-21 11:34:32 -07001532 def _cleanup_poweron(self):
1533 """Special cleanup method to make sure hosts always get power back."""
1534 afe = frontend_wrappers.RetryingAFE(timeout_min=5, delay_sec=10)
1535 hosts = afe.get_hosts(hostname=self.hostname)
1536 if not hosts or not (self._RPM_OUTLET_CHANGED in
1537 hosts[0].attributes):
1538 return
1539 logging.debug('This host has recently interacted with the RPM'
1540 ' Infrastructure. Ensuring power is on.')
1541 try:
1542 self.power_on()
Dan Shi7dca56e2014-11-11 17:07:56 -08001543 afe.set_host_attribute(self._RPM_OUTLET_CHANGED, None,
1544 hostname=self.hostname)
Simran Basi5e6339a2013-03-21 11:34:32 -07001545 except rpm_client.RemotePowerException:
Simran Basi5e6339a2013-03-21 11:34:32 -07001546 logging.error('Failed to turn Power On for this host after '
1547 'cleanup through the RPM Infrastructure.')
Gabe Blackb72f4fb2015-01-20 16:47:13 -08001548 autotest_es.post(
Dan Shi7dca56e2014-11-11 17:07:56 -08001549 type_str='RPM_poweron_failure',
1550 metadata={'hostname': self.hostname})
Dan Shi49ca0932014-11-14 11:22:27 -08001551
1552 battery_percentage = self.get_battery_percentage()
Dan Shif01ebe22014-12-05 13:10:57 -08001553 if battery_percentage and battery_percentage < 50:
Dan Shi49ca0932014-11-14 11:22:27 -08001554 raise
1555 elif self.is_ac_connected():
1556 logging.info('The device has power adapter connected and '
1557 'charging. No need to try to turn RPM on '
1558 'again.')
1559 afe.set_host_attribute(self._RPM_OUTLET_CHANGED, None,
1560 hostname=self.hostname)
1561 logging.info('Battery level is now at %s%%. The device may '
1562 'still have enough power to run test, so no '
1563 'exception will be raised.', battery_percentage)
1564
Simran Basi5e6339a2013-03-21 11:34:32 -07001565
beepsc87ff602013-07-31 21:53:00 -07001566 def _is_factory_image(self):
1567 """Checks if the image on the DUT is a factory image.
1568
1569 @return: True if the image on the DUT is a factory image.
1570 False otherwise.
1571 """
1572 result = self.run('[ -f /root/.factory_test ]', ignore_status=True)
1573 return result.exit_status == 0
1574
1575
1576 def _restart_ui(self):
J. Richard Barnette84890bd2014-02-21 11:05:47 -08001577 """Restart the Chrome UI.
beepsc87ff602013-07-31 21:53:00 -07001578
1579 @raises: FactoryImageCheckerException for factory images, since
1580 we cannot attempt to restart ui on them.
1581 error.AutoservRunError for any other type of error that
1582 occurs while restarting ui.
1583 """
1584 if self._is_factory_image():
Dan Shi549fb822015-03-24 18:01:11 -07001585 raise FactoryImageCheckerException('Cannot restart ui on factory '
1586 'images')
beepsc87ff602013-07-31 21:53:00 -07001587
J. Richard Barnette84890bd2014-02-21 11:05:47 -08001588 # TODO(jrbarnette): The command to stop/start the ui job
1589 # should live inside cros_ui, too. However that would seem
1590 # to imply interface changes to the existing start()/restart()
1591 # functions, which is a bridge too far (for now).
J. Richard Barnette6069aa12015-06-08 09:10:24 -07001592 prompt = cros_ui.get_chrome_session_ident(self)
J. Richard Barnette84890bd2014-02-21 11:05:47 -08001593 self.run('stop ui; start ui')
1594 cros_ui.wait_for_chrome_ready(prompt, self)
beepsc87ff602013-07-31 21:53:00 -07001595
1596
Dan Shi549fb822015-03-24 18:01:11 -07001597 def get_release_version(self):
1598 """Get the value of attribute CHROMEOS_RELEASE_VERSION from lsb-release.
1599
1600 @returns The version string in lsb-release, under attribute
1601 CHROMEOS_RELEASE_VERSION.
1602 """
1603 lsb_release_content = self.run(
1604 'cat "%s"' % client_constants.LSB_RELEASE).stdout.strip()
1605 return lsbrelease_utils.get_chromeos_release_version(
1606 lsb_release_content=lsb_release_content)
1607
1608
1609 def verify_cros_version_label(self):
1610 """ Make sure host's cros-version label match the actual image in dut.
1611
1612 Remove any cros-version: label that doesn't match that installed in
1613 the dut.
1614
1615 @param raise_error: Set to True to raise exception if any mismatch found
1616
1617 @raise error.AutoservError: If any mismatch between cros-version label
1618 and the build installed in dut is found.
1619 """
1620 labels = self._AFE.get_labels(
1621 name__startswith=ds_constants.VERSION_PREFIX,
1622 host__hostname=self.hostname)
1623 mismatch_found = False
1624 if labels:
1625 # Get CHROMEOS_RELEASE_VERSION from lsb-release, e.g., 6908.0.0.
1626 # Note that it's different from cros-version label, which has
1627 # builder and branch info, e.g.,
1628 # cros-version:peppy-release/R43-6908.0.0
1629 release_version = self.get_release_version()
1630 host_list = [self.hostname]
1631 for label in labels:
1632 # Remove any cros-version label that does not match
1633 # release_version.
1634 build_version = label.name[len(ds_constants.VERSION_PREFIX):]
1635 if not utils.version_match(build_version, release_version):
1636 logging.warn('cros-version label "%s" does not match '
1637 'release version %s. Removing the label.',
1638 label.name, release_version)
1639 label.remove_hosts(hosts=host_list)
1640 mismatch_found = True
1641 if mismatch_found:
Dan Shi1057bae2015-03-30 11:35:09 -07001642 autotest_es.post(use_http=True,
1643 type_str='cros_version_label_mismatch',
1644 metadata={'hostname': self.hostname})
Dan Shi549fb822015-03-24 18:01:11 -07001645 raise error.AutoservError('The host has wrong cros-version label.')
1646
1647
beepsc87ff602013-07-31 21:53:00 -07001648 def cleanup(self):
MK Ryu35d661e2014-09-25 17:44:10 -07001649 self.run('rm -f %s' % client_constants.CLEANUP_LOGS_PAUSED_FILE)
Scott Zawalskiddbc31e2012-11-15 11:29:01 -05001650 try:
beepsc87ff602013-07-31 21:53:00 -07001651 self._restart_ui()
1652 except (error.AutotestRunError, error.AutoservRunError,
1653 FactoryImageCheckerException):
Ilja H. Friedel04be2bd2014-05-07 21:29:59 -07001654 logging.warning('Unable to restart ui, rebooting device.')
Scott Zawalskiddbc31e2012-11-15 11:29:01 -05001655 # Since restarting the UI fails fall back to normal Autotest
1656 # cleanup routines, i.e. reboot the machine.
Fang Deng0ca40e22013-08-27 17:47:44 -07001657 super(CrosHost, self).cleanup()
Simran Basi5e6339a2013-03-21 11:34:32 -07001658 # Check if the rpm outlet was manipulated.
Simran Basid5e5e272012-09-24 15:23:59 -07001659 if self.has_power():
Simran Basi5e6339a2013-03-21 11:34:32 -07001660 self._cleanup_poweron()
Dan Shi549fb822015-03-24 18:01:11 -07001661 self.verify_cros_version_label()
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001662
1663
Yu-Ju Honga2be94a2012-07-31 09:48:52 -07001664 def reboot(self, **dargs):
1665 """
1666 This function reboots the site host. The more generic
1667 RemoteHost.reboot() performs sync and sleeps for 5
1668 seconds. This is not necessary for Chrome OS devices as the
1669 sync should be finished in a short time during the reboot
1670 command.
1671 """
Tom Wai-Hong Tamf5cd1d42012-08-13 12:04:08 +08001672 if 'reboot_cmd' not in dargs:
Doug Anderson7d5aeb22014-02-27 15:12:17 -08001673 reboot_timeout = dargs.get('reboot_timeout', 10)
J. Richard Barnette9af19632015-09-25 12:18:03 -07001674 dargs['reboot_cmd'] = ('sleep 1; '
1675 'reboot & sleep %d; '
1676 'reboot -f' % reboot_timeout)
Yu-Ju Honga2be94a2012-07-31 09:48:52 -07001677 # Enable fastsync to avoid running extra sync commands before reboot.
Tom Wai-Hong Tamf5cd1d42012-08-13 12:04:08 +08001678 if 'fastsync' not in dargs:
1679 dargs['fastsync'] = True
Michael Liangda8c60a2014-06-03 13:24:51 -07001680
Charlie Mooneya8e6dab2014-05-29 14:37:55 -07001681 # For purposes of logging reboot times:
1682 # Get the board name i.e. 'daisy_spring'
Michael Liangca4f5a62014-07-10 15:45:13 -07001683 board_fullname = self.get_board()
1684
1685 # Strip the prefix and add it to dargs.
1686 dargs['board'] = board_fullname[board_fullname.find(':')+1:]
Fang Deng0ca40e22013-08-27 17:47:44 -07001687 super(CrosHost, self).reboot(**dargs)
Yu-Ju Honga2be94a2012-07-31 09:48:52 -07001688
1689
Gwendal Grignou7a61d2f2014-05-23 11:05:51 -07001690 def suspend(self, **dargs):
1691 """
1692 This function suspends the site host.
1693 """
1694 suspend_time = dargs.get('suspend_time', 60)
1695 dargs['timeout'] = suspend_time
1696 if 'suspend_cmd' not in dargs:
J. Richard Barnette9af19632015-09-25 12:18:03 -07001697 dargs['suspend_cmd'] = ' && '.join([
1698 'echo 0 > /sys/class/rtc/rtc0/wakealarm',
Gwendal Grignou7a61d2f2014-05-23 11:05:51 -07001699 'echo +%d > /sys/class/rtc/rtc0/wakealarm' % suspend_time,
J. Richard Barnette9af19632015-09-25 12:18:03 -07001700 'powerd_dbus_suspend --delay=0'])
Gwendal Grignou7a61d2f2014-05-23 11:05:51 -07001701 super(CrosHost, self).suspend(**dargs)
1702
1703
Simran Basiec564392014-08-25 16:48:09 -07001704 def upstart_status(self, service_name):
1705 """Check the status of an upstart init script.
1706
1707 @param service_name: Service to look up.
1708
1709 @returns True if the service is running, False otherwise.
1710 """
1711 return self.run('status %s | grep start/running' %
1712 service_name).stdout.strip() != ''
1713
1714
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001715 def verify_software(self):
Richard Barnetteb2bc13c2013-01-08 17:32:51 -08001716 """Verify working software on a Chrome OS system.
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001717
Richard Barnetteb2bc13c2013-01-08 17:32:51 -08001718 Tests for the following conditions:
1719 1. All conditions tested by the parent version of this
1720 function.
1721 2. Sufficient space in /mnt/stateful_partition.
Fang Deng6b05f5b2013-03-20 13:42:11 -07001722 3. Sufficient space in /mnt/stateful_partition/encrypted.
1723 4. update_engine answers a simple status request over DBus.
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001724
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001725 """
MK Ryu35d661e2014-09-25 17:44:10 -07001726 # Check if a job was crashed on this host.
1727 # If yes, avoid verification until crash-logs are collected.
1728 if self._need_crash_logs():
1729 raise error.AutoservCrashLogCollectRequired(
1730 'Need to collect crash-logs before verification')
1731
Fang Deng0ca40e22013-08-27 17:47:44 -07001732 super(CrosHost, self).verify_software()
Dan Shib8540a52015-07-16 14:18:23 -07001733 default_kilo_inodes_required = CONFIG.get_config_value(
1734 'SERVER', 'kilo_inodes_required', type=int, default=100)
1735 board = self.get_board().replace(ds_constants.BOARD_PREFIX, '')
1736 kilo_inodes_required = CONFIG.get_config_value(
1737 'SERVER', 'kilo_inodes_required_%s' % board,
1738 type=int, default=default_kilo_inodes_required)
1739 self.check_inodes('/mnt/stateful_partition', kilo_inodes_required)
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001740 self.check_diskspace(
1741 '/mnt/stateful_partition',
Dan Shib8540a52015-07-16 14:18:23 -07001742 CONFIG.get_config_value(
Fang Deng6b05f5b2013-03-20 13:42:11 -07001743 'SERVER', 'gb_diskspace_required', type=float,
1744 default=20.0))
Gaurav Shahe448af82014-06-19 15:18:59 -07001745 encrypted_stateful_path = '/mnt/stateful_partition/encrypted'
1746 # Not all targets build with encrypted stateful support.
1747 if self.path_exists(encrypted_stateful_path):
1748 self.check_diskspace(
1749 encrypted_stateful_path,
Dan Shib8540a52015-07-16 14:18:23 -07001750 CONFIG.get_config_value(
Gaurav Shahe448af82014-06-19 15:18:59 -07001751 'SERVER', 'gb_encrypted_diskspace_required', type=float,
1752 default=0.1))
beepsc87ff602013-07-31 21:53:00 -07001753
Simran Basiec564392014-08-25 16:48:09 -07001754 if not self.upstart_status('system-services'):
Prashanth B5d0a0512014-04-25 12:26:08 -07001755 raise error.AutoservError('Chrome failed to reach login. '
1756 'System services not running.')
1757
beepsc87ff602013-07-31 21:53:00 -07001758 # Factory images don't run update engine,
1759 # goofy controls dbus on these DUTs.
1760 if not self._is_factory_image():
1761 self.run('update_engine_client --status')
Scott Zawalskifbca4a92013-03-04 15:56:42 -05001762 # Makes sure python is present, loads and can use built in functions.
1763 # We have seen cases where importing cPickle fails with undefined
1764 # symbols in cPickle.so.
1765 self.run('python -c "import cPickle"')
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001766
Dan Shi549fb822015-03-24 18:01:11 -07001767 self.verify_cros_version_label()
1768
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001769
Dan Shi49ca0932014-11-14 11:22:27 -08001770 def verify_hardware(self):
1771 """Verify hardware system of a Chrome OS system.
1772
1773 Check following hardware conditions:
1774 1. Battery level.
1775 2. Is power adapter connected.
1776 """
1777 logging.info('Battery percentage: %s', self.get_battery_percentage())
Dan Shie9b765d2014-12-29 16:59:49 -08001778 if self.is_ac_connected() is None:
1779 logging.info('Can not determine if the device has power adapter '
1780 'connected.')
1781 else:
1782 logging.info('Device %s power adapter connected and charging.',
1783 'has' if self.is_ac_connected() else 'does not have')
Dan Shi49ca0932014-11-14 11:22:27 -08001784
1785
Fang Deng96667ca2013-08-01 17:46:18 -07001786 def make_ssh_command(self, user='root', port=22, opts='', hosts_file=None,
1787 connect_timeout=None, alive_interval=None):
1788 """Override default make_ssh_command to use options tuned for Chrome OS.
1789
1790 Tuning changes:
1791 - ConnectTimeout=30; maximum of 30 seconds allowed for an SSH
1792 connection failure. Consistency with remote_access.sh.
1793
Samuel Tan2ce155b2015-06-23 18:24:38 -07001794 - ServerAliveInterval=900; which causes SSH to ping connection every
1795 900 seconds. In conjunction with ServerAliveCountMax ensures
1796 that if the connection dies, Autotest will bail out.
Fang Deng96667ca2013-08-01 17:46:18 -07001797 Originally tried 60 secs, but saw frequent job ABORTS where
Samuel Tan2ce155b2015-06-23 18:24:38 -07001798 the test completed successfully. Later increased from 180 seconds to
1799 900 seconds to account for tests where the DUT is suspended for
1800 longer periods of time.
Fang Deng96667ca2013-08-01 17:46:18 -07001801
1802 - ServerAliveCountMax=3; consistency with remote_access.sh.
1803
1804 - ConnectAttempts=4; reduce flakiness in connection errors;
1805 consistency with remote_access.sh.
1806
1807 - UserKnownHostsFile=/dev/null; we don't care about the keys.
1808 Host keys change with every new installation, don't waste
1809 memory/space saving them.
1810
1811 - SSH protocol forced to 2; needed for ServerAliveInterval.
1812
1813 @param user User name to use for the ssh connection.
1814 @param port Port on the target host to use for ssh connection.
1815 @param opts Additional options to the ssh command.
1816 @param hosts_file Ignored.
1817 @param connect_timeout Ignored.
1818 @param alive_interval Ignored.
1819 """
Aviv Keshetc5947fa2013-09-04 14:06:29 -07001820 base_command = ('/usr/bin/ssh -a -x %s %s %s'
1821 ' -o StrictHostKeyChecking=no'
Fang Deng96667ca2013-08-01 17:46:18 -07001822 ' -o UserKnownHostsFile=/dev/null -o BatchMode=yes'
Samuel Tan2ce155b2015-06-23 18:24:38 -07001823 ' -o ConnectTimeout=30 -o ServerAliveInterval=900'
Fang Deng96667ca2013-08-01 17:46:18 -07001824 ' -o ServerAliveCountMax=3 -o ConnectionAttempts=4'
1825 ' -o Protocol=2 -l %s -p %d')
Aviv Keshetc5947fa2013-09-04 14:06:29 -07001826 return base_command % (self._ssh_verbosity_flag, self._ssh_options,
1827 opts, user, port)
Fang Deng96667ca2013-08-01 17:46:18 -07001828
1829
Christopher Wileydd181852013-10-10 19:56:58 -07001830 def _setup_rpc(self, port, command_name, remote_pid=None):
beeps32a63082013-08-22 14:02:29 -07001831 """Sets up a tunnel process and performs rpc connection book keeping.
1832
Cheng-Yi Chianga155e7e2015-08-20 20:42:04 +08001833 Chrome OS on the target closes down most external ports for security.
1834 We could open the port, but doing that would conflict with security
1835 tests that check that only expected ports are open. So, to get to
1836 the port on the target we use an ssh tunnel.
1837
beeps32a63082013-08-22 14:02:29 -07001838 This method assumes that xmlrpc and jsonrpc never conflict, since
1839 we can only either have an xmlrpc or a jsonrpc server listening on
1840 a remote port. As such, it enforces a single proxy->remote port
1841 policy, i.e if one starts a jsonrpc proxy/server from port A->B,
1842 and then tries to start an xmlrpc proxy forwarded to the same port,
1843 the xmlrpc proxy will override the jsonrpc tunnel process, however:
1844
1845 1. None of the methods on the xmlrpc proxy will work because
1846 the server listening on B is jsonrpc.
1847
1848 2. The xmlrpc client cannot initiate a termination of the JsonRPC
1849 server, as the only use case currently is goofy, which is tied to
1850 the factory image. It is much easier to handle a failed xmlrpc
1851 call on the client than it is to terminate goofy in this scenario,
1852 as doing the latter might leave the DUT in a hard to recover state.
1853
1854 With the current implementation newer rpc proxy connections will
1855 terminate the tunnel processes of older rpc connections tunneling
1856 to the same remote port. If methods are invoked on the client
1857 after this has happened they will fail with connection closed errors.
1858
1859 @param port: The remote forwarding port.
1860 @param command_name: The name of the remote process, to terminate
1861 using pkill.
1862
1863 @return A url that we can use to initiate the rpc connection.
1864 """
1865 self.rpc_disconnect(port)
1866 local_port = utils.get_unused_port()
1867 tunnel_proc = self._create_ssh_tunnel(port, local_port)
Christopher Wileydd181852013-10-10 19:56:58 -07001868 self._rpc_proxy_map[port] = (command_name, tunnel_proc, remote_pid)
beeps32a63082013-08-22 14:02:29 -07001869 return self._RPC_PROXY_URL % local_port
1870
1871
Christopher Wileyd78249a2013-03-01 13:05:31 -08001872 def xmlrpc_connect(self, command, port, command_name=None,
Yusuf Mohsinallyfff89d62013-11-18 16:34:07 -08001873 ready_test_name=None, timeout_seconds=10,
1874 logfile='/dev/null'):
J. Richard Barnette1d78b012012-05-15 13:56:30 -07001875 """Connect to an XMLRPC server on the host.
1876
1877 The `command` argument should be a simple shell command that
1878 starts an XMLRPC server on the given `port`. The command
1879 must not daemonize, and must terminate cleanly on SIGTERM.
1880 The command is started in the background on the host, and a
1881 local XMLRPC client for the server is created and returned
1882 to the caller.
1883
1884 Note that the process of creating an XMLRPC client makes no
1885 attempt to connect to the remote server; the caller is
1886 responsible for determining whether the server is running
1887 correctly, and is ready to serve requests.
1888
Christopher Wileyd78249a2013-03-01 13:05:31 -08001889 Optionally, the caller can pass ready_test_name, a string
1890 containing the name of a method to call on the proxy. This
1891 method should take no parameters and return successfully only
1892 when the server is ready to process client requests. When
1893 ready_test_name is set, xmlrpc_connect will block until the
1894 proxy is ready, and throw a TestError if the server isn't
1895 ready by timeout_seconds.
1896
beeps32a63082013-08-22 14:02:29 -07001897 If a server is already running on the remote port, this
1898 method will kill it and disconnect the tunnel process
1899 associated with the connection before establishing a new one,
1900 by consulting the rpc_proxy_map in rpc_disconnect.
1901
J. Richard Barnette1d78b012012-05-15 13:56:30 -07001902 @param command Shell command to start the server.
1903 @param port Port number on which the server is expected to
1904 be serving.
J. Richard Barnette7214e0b2013-02-06 15:20:49 -08001905 @param command_name String to use as input to `pkill` to
1906 terminate the XMLRPC server on the host.
Christopher Wileyd78249a2013-03-01 13:05:31 -08001907 @param ready_test_name String containing the name of a
1908 method defined on the XMLRPC server.
1909 @param timeout_seconds Number of seconds to wait
1910 for the server to become 'ready.' Will throw a
1911 TestFail error if server is not ready in time.
Yusuf Mohsinallyfff89d62013-11-18 16:34:07 -08001912 @param logfile Logfile to send output when running
1913 'command' argument.
Yusuf Mohsinally8d19e3c2013-11-21 14:25:45 -08001914
J. Richard Barnette1d78b012012-05-15 13:56:30 -07001915 """
Christopher Wileyc14f06a2013-10-16 13:55:39 -07001916 # Clean up any existing state. If the caller is willing
1917 # to believe their server is down, we ought to clean up
1918 # any tunnels we might have sitting around.
1919 self.rpc_disconnect(port)
J. Richard Barnette1d78b012012-05-15 13:56:30 -07001920 # Start the server on the host. Redirection in the command
1921 # below is necessary, because 'ssh' won't terminate until
1922 # background child processes close stdin, stdout, and
1923 # stderr.
J. Richard Barnette9af19632015-09-25 12:18:03 -07001924 remote_cmd = '%s >%s 2>&1' % (command, logfile)
1925 remote_pid = self.run_background(remote_cmd)
J. Richard Barnette1d78b012012-05-15 13:56:30 -07001926 logging.debug('Started XMLRPC server on host %s, pid = %s',
1927 self.hostname, remote_pid)
1928
Christopher Wileydd181852013-10-10 19:56:58 -07001929 # Tunnel through SSH to be able to reach that remote port.
1930 rpc_url = self._setup_rpc(port, command_name, remote_pid=remote_pid)
Christopher Wileyd78249a2013-03-01 13:05:31 -08001931 proxy = xmlrpclib.ServerProxy(rpc_url, allow_none=True)
Christopher Wileydd181852013-10-10 19:56:58 -07001932
Christopher Wileyd78249a2013-03-01 13:05:31 -08001933 if ready_test_name is not None:
J. Richard Barnette13eb7c02013-03-07 12:06:29 -08001934 # retry.retry logs each attempt; calculate delay_sec to
1935 # keep log spam to a dull roar.
Christopher Wiley0ed712b2013-04-09 15:25:12 -07001936 @retry.retry((socket.error,
1937 xmlrpclib.ProtocolError,
1938 httplib.BadStatusLine),
Chris Sosa65425082013-10-16 13:26:22 -07001939 timeout_min=timeout_seconds / 60.0,
1940 delay_sec=min(max(timeout_seconds / 20.0, 0.1), 1))
Christopher Wileyd78249a2013-03-01 13:05:31 -08001941 def ready_test():
1942 """ Call proxy.ready_test_name(). """
1943 getattr(proxy, ready_test_name)()
1944 successful = False
1945 try:
1946 logging.info('Waiting %d seconds for XMLRPC server '
1947 'to start.', timeout_seconds)
1948 ready_test()
1949 successful = True
Christopher Wileyd78249a2013-03-01 13:05:31 -08001950 finally:
1951 if not successful:
1952 logging.error('Failed to start XMLRPC server.')
beeps32a63082013-08-22 14:02:29 -07001953 self.rpc_disconnect(port)
Christopher Wileyd78249a2013-03-01 13:05:31 -08001954 logging.info('XMLRPC server started successfully.')
1955 return proxy
J. Richard Barnette1d78b012012-05-15 13:56:30 -07001956
J. Richard Barnette1d78b012012-05-15 13:56:30 -07001957
Jason Abeleb6f924f2013-11-13 16:01:54 -08001958 def syslog(self, message, tag='autotest'):
1959 """Logs a message to syslog on host.
1960
1961 @param message String message to log into syslog
1962 @param tag String tag prefix for syslog
1963
1964 """
1965 self.run('logger -t "%s" "%s"' % (tag, message))
1966
1967
beeps32a63082013-08-22 14:02:29 -07001968 def jsonrpc_connect(self, port):
1969 """Creates a jsonrpc proxy connection through an ssh tunnel.
1970
1971 This method exists to facilitate communication with goofy (which is
1972 the default system manager on all factory images) and as such, leaves
1973 most of the rpc server sanity checking to the caller. Unlike
1974 xmlrpc_connect, this method does not facilitate the creation of a remote
1975 jsonrpc server, as the only clients of this code are factory tests,
1976 for which the goofy system manager is built in to the image and starts
1977 when the target boots.
1978
1979 One can theoretically create multiple jsonrpc proxies all forwarded
1980 to the same remote port, provided the remote port has an rpc server
1981 listening. However, in doing so we stand the risk of leaking an
1982 existing tunnel process, so we always disconnect any older tunnels
1983 we might have through rpc_disconnect.
1984
1985 @param port: port on the remote host that is serving this proxy.
1986
1987 @return: The client proxy.
1988 """
1989 if not jsonrpclib:
1990 logging.warning('Jsonrpclib could not be imported. Check that '
1991 'site-packages contains jsonrpclib.')
1992 return None
1993
1994 proxy = jsonrpclib.jsonrpc.ServerProxy(self._setup_rpc(port, None))
1995
1996 logging.info('Established a jsonrpc connection through port %s.', port)
1997 return proxy
1998
1999
2000 def rpc_disconnect(self, port):
2001 """Disconnect from an RPC server on the host.
2002
2003 Terminates the remote RPC server previously started for
J. Richard Barnette1d78b012012-05-15 13:56:30 -07002004 the given `port`. Also closes the local ssh tunnel created
2005 for the connection to the host. This function does not
beeps32a63082013-08-22 14:02:29 -07002006 directly alter the state of a previously returned RPC
J. Richard Barnette1d78b012012-05-15 13:56:30 -07002007 client object; however disconnection will cause all
2008 subsequent calls to methods on the object to fail.
2009
2010 This function does nothing if requested to disconnect a port
beeps32a63082013-08-22 14:02:29 -07002011 that was not previously connected via _setup_rpc.
J. Richard Barnette1d78b012012-05-15 13:56:30 -07002012
2013 @param port Port number passed to a previous call to
beeps32a63082013-08-22 14:02:29 -07002014 `_setup_rpc()`.
J. Richard Barnette1d78b012012-05-15 13:56:30 -07002015 """
beeps32a63082013-08-22 14:02:29 -07002016 if port not in self._rpc_proxy_map:
J. Richard Barnette1d78b012012-05-15 13:56:30 -07002017 return
Christopher Wileydd181852013-10-10 19:56:58 -07002018 remote_name, tunnel_proc, remote_pid = self._rpc_proxy_map[port]
J. Richard Barnette1d78b012012-05-15 13:56:30 -07002019 if remote_name:
2020 # We use 'pkill' to find our target process rather than
2021 # a PID, because the host may have rebooted since
2022 # connecting, and we don't want to kill an innocent
2023 # process with the same PID.
2024 #
2025 # 'pkill' helpfully exits with status 1 if no target
2026 # process is found, for which run() will throw an
Simran Basid5e5e272012-09-24 15:23:59 -07002027 # exception. We don't want that, so we the ignore
J. Richard Barnette1d78b012012-05-15 13:56:30 -07002028 # status.
2029 self.run("pkill -f '%s'" % remote_name, ignore_status=True)
Christopher Wileydd181852013-10-10 19:56:58 -07002030 if remote_pid:
2031 logging.info('Waiting for RPC server "%s" shutdown',
2032 remote_name)
2033 start_time = time.time()
2034 while (time.time() - start_time <
2035 self._RPC_SHUTDOWN_TIMEOUT_SECONDS):
2036 running_processes = self.run(
2037 "pgrep -f '%s'" % remote_name,
2038 ignore_status=True).stdout.split()
2039 if not remote_pid in running_processes:
2040 logging.info('Shut down RPC server.')
2041 break
2042 time.sleep(self._RPC_SHUTDOWN_POLLING_PERIOD_SECONDS)
2043 else:
2044 raise error.TestError('Failed to shutdown RPC server %s' %
2045 remote_name)
J. Richard Barnette1d78b012012-05-15 13:56:30 -07002046
2047 if tunnel_proc.poll() is None:
2048 tunnel_proc.terminate()
2049 logging.debug('Terminated tunnel, pid %d', tunnel_proc.pid)
2050 else:
2051 logging.debug('Tunnel pid %d terminated early, status %d',
2052 tunnel_proc.pid, tunnel_proc.returncode)
beeps32a63082013-08-22 14:02:29 -07002053 del self._rpc_proxy_map[port]
J. Richard Barnette1d78b012012-05-15 13:56:30 -07002054
2055
beeps32a63082013-08-22 14:02:29 -07002056 def rpc_disconnect_all(self):
2057 """Disconnect all known RPC proxy ports."""
2058 for port in self._rpc_proxy_map.keys():
2059 self.rpc_disconnect(port)
J. Richard Barnette1d78b012012-05-15 13:56:30 -07002060
2061
Ilja H. Friedel1232e8a2014-06-17 21:30:48 -07002062 def poor_mans_rpc(self, fun):
2063 """
2064 Calls a function from client utils on the host and returns a string.
2065
2066 @param fun function in client utils namespace.
2067 @return output string from calling fun.
2068 """
Simran Basi263a9d32014-08-19 11:16:51 -07002069 script = 'cd %s/bin; ' % autotest.Autotest.get_installed_autodir(self)
Ilja H. Friedel1232e8a2014-06-17 21:30:48 -07002070 script += 'python -c "import common; import utils;'
2071 script += 'print utils.%s"' % fun
2072 return script
2073
2074
J. Richard Barnetteb6de7e32013-02-14 13:28:04 -08002075 def _ping_check_status(self, status):
2076 """Ping the host once, and return whether it has a given status.
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002077
J. Richard Barnetteb6de7e32013-02-14 13:28:04 -08002078 @param status Check the ping status against this value.
2079 @return True iff `status` and the result of ping are the same
2080 (i.e. both True or both False).
2081
2082 """
2083 ping_val = utils.ping(self.hostname, tries=1, deadline=1)
2084 return not (status ^ (ping_val == 0))
2085
2086 def _ping_wait_for_status(self, status, timeout):
2087 """Wait for the host to have a given status (UP or DOWN).
2088
2089 Status is checked by polling. Polling will not last longer
2090 than the number of seconds in `timeout`. The polling
2091 interval will be long enough that only approximately
2092 _PING_WAIT_COUNT polling cycles will be executed, subject
2093 to a maximum interval of about one minute.
2094
2095 @param status Waiting will stop immediately if `ping` of the
2096 host returns this status.
2097 @param timeout Poll for at most this many seconds.
2098 @return True iff the host status from `ping` matched the
2099 requested status at the time of return.
2100
2101 """
2102 # _ping_check_status() takes about 1 second, hence the
2103 # "- 1" in the formula below.
2104 poll_interval = min(int(timeout / self._PING_WAIT_COUNT), 60) - 1
2105 end_time = time.time() + timeout
2106 while time.time() <= end_time:
2107 if self._ping_check_status(status):
2108 return True
2109 if poll_interval > 0:
2110 time.sleep(poll_interval)
2111
2112 # The last thing we did was sleep(poll_interval), so it may
2113 # have been too long since the last `ping`. Check one more
2114 # time, just to be sure.
2115 return self._ping_check_status(status)
2116
2117 def ping_wait_up(self, timeout):
2118 """Wait for the host to respond to `ping`.
2119
2120 N.B. This method is not a reliable substitute for
2121 `wait_up()`, because a host that responds to ping will not
2122 necessarily respond to ssh. This method should only be used
2123 if the target DUT can be considered functional even if it
2124 can't be reached via ssh.
2125
2126 @param timeout Minimum time to allow before declaring the
2127 host to be non-responsive.
2128 @return True iff the host answered to ping before the timeout.
2129
2130 """
2131 return self._ping_wait_for_status(self._PING_STATUS_UP, timeout)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002132
Andrew Bresticker678c0c72013-01-22 10:44:09 -08002133 def ping_wait_down(self, timeout):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002134 """Wait until the host no longer responds to `ping`.
2135
J. Richard Barnetteb6de7e32013-02-14 13:28:04 -08002136 This function can be used as a slightly faster version of
2137 `wait_down()`, by avoiding potentially long ssh timeouts.
2138
2139 @param timeout Minimum time to allow for the host to become
2140 non-responsive.
2141 @return True iff the host quit answering ping before the
2142 timeout.
2143
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002144 """
J. Richard Barnetteb6de7e32013-02-14 13:28:04 -08002145 return self._ping_wait_for_status(self._PING_STATUS_DOWN, timeout)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002146
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08002147 def test_wait_for_sleep(self, sleep_timeout=None):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002148 """Wait for the client to enter low-power sleep mode.
2149
2150 The test for "is asleep" can't distinguish a system that is
2151 powered off; to confirm that the unit was asleep, it is
2152 necessary to force resume, and then call
2153 `test_wait_for_resume()`.
2154
2155 This function is expected to be called from a test as part
2156 of a sequence like the following:
2157
2158 ~~~~~~~~
2159 boot_id = host.get_boot_id()
2160 # trigger sleep on the host
2161 host.test_wait_for_sleep()
2162 # trigger resume on the host
2163 host.test_wait_for_resume(boot_id)
2164 ~~~~~~~~
2165
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08002166 @param sleep_timeout time limit in seconds to allow the host sleep.
2167
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002168 @exception TestFail The host did not go to sleep within
2169 the allowed time.
2170 """
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08002171 if sleep_timeout is None:
2172 sleep_timeout = self.SLEEP_TIMEOUT
2173
2174 if not self.ping_wait_down(timeout=sleep_timeout):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002175 raise error.TestFail(
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08002176 'client failed to sleep after %d seconds' % sleep_timeout)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002177
2178
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08002179 def test_wait_for_resume(self, old_boot_id, resume_timeout=None):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002180 """Wait for the client to resume from low-power sleep mode.
2181
2182 The `old_boot_id` parameter should be the value from
2183 `get_boot_id()` obtained prior to entering sleep mode. A
2184 `TestFail` exception is raised if the boot id changes.
2185
2186 See @ref test_wait_for_sleep for more on this function's
2187 usage.
2188
J. Richard Barnette7214e0b2013-02-06 15:20:49 -08002189 @param old_boot_id A boot id value obtained before the
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002190 target host went to sleep.
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08002191 @param resume_timeout time limit in seconds to allow the host up.
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002192
2193 @exception TestFail The host did not respond within the
2194 allowed time.
2195 @exception TestFail The host responded, but the boot id test
2196 indicated a reboot rather than a sleep
2197 cycle.
2198 """
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08002199 if resume_timeout is None:
2200 resume_timeout = self.RESUME_TIMEOUT
2201
2202 if not self.wait_up(timeout=resume_timeout):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002203 raise error.TestFail(
2204 'client failed to resume from sleep after %d seconds' %
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08002205 resume_timeout)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002206 else:
2207 new_boot_id = self.get_boot_id()
2208 if new_boot_id != old_boot_id:
Tom Wai-Hong Tam01792682015-01-06 08:00:46 +08002209 logging.error('client rebooted (old boot %s, new boot %s)',
2210 old_boot_id, new_boot_id)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002211 raise error.TestFail(
Tom Wai-Hong Tam01792682015-01-06 08:00:46 +08002212 'client rebooted, but sleep was expected')
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002213
2214
Tom Wai-Hong Tamfe005c22014-12-03 09:25:44 +08002215 def test_wait_for_shutdown(self, shutdown_timeout=None):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002216 """Wait for the client to shut down.
2217
2218 The test for "has shut down" can't distinguish a system that
2219 is merely asleep; to confirm that the unit was down, it is
2220 necessary to force boot, and then call test_wait_for_boot().
2221
2222 This function is expected to be called from a test as part
2223 of a sequence like the following:
2224
2225 ~~~~~~~~
2226 boot_id = host.get_boot_id()
2227 # trigger shutdown on the host
2228 host.test_wait_for_shutdown()
2229 # trigger boot on the host
2230 host.test_wait_for_boot(boot_id)
2231 ~~~~~~~~
2232
Tom Wai-Hong Tamfe005c22014-12-03 09:25:44 +08002233 @param shutdown_timeout time limit in seconds to allow the host down.
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002234 @exception TestFail The host did not shut down within the
2235 allowed time.
2236 """
Tom Wai-Hong Tamfe005c22014-12-03 09:25:44 +08002237 if shutdown_timeout is None:
2238 shutdown_timeout = self.SHUTDOWN_TIMEOUT
2239
2240 if not self.ping_wait_down(timeout=shutdown_timeout):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002241 raise error.TestFail(
2242 'client failed to shut down after %d seconds' %
Tom Wai-Hong Tamfe005c22014-12-03 09:25:44 +08002243 shutdown_timeout)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002244
2245
2246 def test_wait_for_boot(self, old_boot_id=None):
2247 """Wait for the client to boot from cold power.
2248
2249 The `old_boot_id` parameter should be the value from
2250 `get_boot_id()` obtained prior to shutting down. A
2251 `TestFail` exception is raised if the boot id does not
2252 change. The boot id test is omitted if `old_boot_id` is not
2253 specified.
2254
2255 See @ref test_wait_for_shutdown for more on this function's
2256 usage.
2257
J. Richard Barnette7214e0b2013-02-06 15:20:49 -08002258 @param old_boot_id A boot id value obtained before the
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002259 shut down.
2260
2261 @exception TestFail The host did not respond within the
2262 allowed time.
2263 @exception TestFail The host responded, but the boot id test
2264 indicated that there was no reboot.
2265 """
J. Richard Barnetteeb69d722012-06-18 17:29:44 -07002266 if not self.wait_up(timeout=self.REBOOT_TIMEOUT):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002267 raise error.TestFail(
2268 'client failed to reboot after %d seconds' %
J. Richard Barnetteeb69d722012-06-18 17:29:44 -07002269 self.REBOOT_TIMEOUT)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002270 elif old_boot_id:
2271 if self.get_boot_id() == old_boot_id:
Tom Wai-Hong Tam01792682015-01-06 08:00:46 +08002272 logging.error('client not rebooted (boot %s)',
2273 old_boot_id)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002274 raise error.TestFail(
Tom Wai-Hong Tam01792682015-01-06 08:00:46 +08002275 'client is back up, but did not reboot')
Simran Basid5e5e272012-09-24 15:23:59 -07002276
2277
2278 @staticmethod
2279 def check_for_rpm_support(hostname):
2280 """For a given hostname, return whether or not it is powered by an RPM.
2281
Simran Basi1df55112013-09-06 11:25:09 -07002282 @param hostname: hostname to check for rpm support.
2283
Simran Basid5e5e272012-09-24 15:23:59 -07002284 @return None if this host does not follows the defined naming format
2285 for RPM powered DUT's in the lab. If it does follow the format,
2286 it returns a regular expression MatchObject instead.
2287 """
Fang Dengbaff9082015-01-06 13:46:15 -08002288 return re.match(CrosHost._RPM_HOSTNAME_REGEX, hostname)
Simran Basid5e5e272012-09-24 15:23:59 -07002289
2290
2291 def has_power(self):
2292 """For this host, return whether or not it is powered by an RPM.
2293
2294 @return True if this host is in the CROS lab and follows the defined
2295 naming format.
2296 """
Fang Deng0ca40e22013-08-27 17:47:44 -07002297 return CrosHost.check_for_rpm_support(self.hostname)
Simran Basid5e5e272012-09-24 15:23:59 -07002298
2299
Ismail Noorbasha07fdb612013-02-14 14:13:31 -08002300 def _set_power(self, state, power_method):
2301 """Sets the power to the host via RPM, Servo or manual.
2302
2303 @param state Specifies which power state to set to DUT
2304 @param power_method Specifies which method of power control to
2305 use. By default "RPM" will be used. Valid values
2306 are the strings "RPM", "manual", "servoj10".
2307
2308 """
2309 ACCEPTABLE_STATES = ['ON', 'OFF']
2310
2311 if state.upper() not in ACCEPTABLE_STATES:
2312 raise error.TestError('State must be one of: %s.'
2313 % (ACCEPTABLE_STATES,))
2314
2315 if power_method == self.POWER_CONTROL_SERVO:
2316 logging.info('Setting servo port J10 to %s', state)
2317 self.servo.set('prtctl3_pwren', state.lower())
2318 time.sleep(self._USB_POWER_TIMEOUT)
2319 elif power_method == self.POWER_CONTROL_MANUAL:
2320 logging.info('You have %d seconds to set the AC power to %s.',
2321 self._POWER_CYCLE_TIMEOUT, state)
2322 time.sleep(self._POWER_CYCLE_TIMEOUT)
2323 else:
2324 if not self.has_power():
2325 raise error.TestFail('DUT does not have RPM connected.')
Simran Basi5e6339a2013-03-21 11:34:32 -07002326 afe = frontend_wrappers.RetryingAFE(timeout_min=5, delay_sec=10)
2327 afe.set_host_attribute(self._RPM_OUTLET_CHANGED, True,
2328 hostname=self.hostname)
Simran Basi1df55112013-09-06 11:25:09 -07002329 rpm_client.set_power(self.hostname, state.upper(), timeout_mins=5)
Simran Basid5e5e272012-09-24 15:23:59 -07002330
2331
Ismail Noorbasha07fdb612013-02-14 14:13:31 -08002332 def power_off(self, power_method=POWER_CONTROL_RPM):
2333 """Turn off power to this host via RPM, Servo or manual.
2334
2335 @param power_method Specifies which method of power control to
2336 use. By default "RPM" will be used. Valid values
2337 are the strings "RPM", "manual", "servoj10".
2338
2339 """
2340 self._set_power('OFF', power_method)
Simran Basid5e5e272012-09-24 15:23:59 -07002341
2342
Ismail Noorbasha07fdb612013-02-14 14:13:31 -08002343 def power_on(self, power_method=POWER_CONTROL_RPM):
2344 """Turn on power to this host via RPM, Servo or manual.
2345
2346 @param power_method Specifies which method of power control to
2347 use. By default "RPM" will be used. Valid values
2348 are the strings "RPM", "manual", "servoj10".
2349
2350 """
2351 self._set_power('ON', power_method)
2352
2353
2354 def power_cycle(self, power_method=POWER_CONTROL_RPM):
2355 """Cycle power to this host by turning it OFF, then ON.
2356
2357 @param power_method Specifies which method of power control to
2358 use. By default "RPM" will be used. Valid values
2359 are the strings "RPM", "manual", "servoj10".
2360
2361 """
2362 if power_method in (self.POWER_CONTROL_SERVO,
2363 self.POWER_CONTROL_MANUAL):
2364 self.power_off(power_method=power_method)
2365 time.sleep(self._POWER_CYCLE_TIMEOUT)
2366 self.power_on(power_method=power_method)
2367 else:
2368 rpm_client.set_power(self.hostname, 'CYCLE')
Simran Basic6f1f7a2012-10-16 10:47:46 -07002369
2370
2371 def get_platform(self):
2372 """Determine the correct platform label for this host.
2373
2374 @returns a string representing this host's platform.
2375 """
2376 crossystem = utils.Crossystem(self)
2377 crossystem.init()
2378 # Extract fwid value and use the leading part as the platform id.
2379 # fwid generally follow the format of {platform}.{firmware version}
2380 # Example: Alex.X.YYY.Z or Google_Alex.X.YYY.Z
2381 platform = crossystem.fwid().split('.')[0].lower()
2382 # Newer platforms start with 'Google_' while the older ones do not.
2383 return platform.replace('google_', '')
2384
2385
Hung-ying Tyanb1328032014-04-01 14:18:54 +08002386 def get_architecture(self):
2387 """Determine the correct architecture label for this host.
2388
2389 @returns a string representing this host's architecture.
2390 """
2391 crossystem = utils.Crossystem(self)
2392 crossystem.init()
2393 return crossystem.arch()
2394
2395
Luis Lozano40b7d0d2014-01-17 15:12:06 -08002396 def get_chrome_version(self):
2397 """Gets the Chrome version number and milestone as strings.
2398
2399 Invokes "chrome --version" to get the version number and milestone.
2400
2401 @return A tuple (chrome_ver, milestone) where "chrome_ver" is the
2402 current Chrome version number as a string (in the form "W.X.Y.Z")
2403 and "milestone" is the first component of the version number
2404 (the "W" from "W.X.Y.Z"). If the version number cannot be parsed
2405 in the "W.X.Y.Z" format, the "chrome_ver" will be the full output
2406 of "chrome --version" and the milestone will be the empty string.
2407
2408 """
MK Ryu35d661e2014-09-25 17:44:10 -07002409 version_string = self.run(client_constants.CHROME_VERSION_COMMAND).stdout
Luis Lozano40b7d0d2014-01-17 15:12:06 -08002410 return utils.parse_chrome_version(version_string)
2411
Aviv Keshet74c89a92013-02-04 15:18:30 -08002412 @label_decorator()
Simran Basic6f1f7a2012-10-16 10:47:46 -07002413 def get_board(self):
2414 """Determine the correct board label for this host.
2415
2416 @returns a string representing this host's board.
2417 """
2418 release_info = utils.parse_cmd_output('cat /etc/lsb-release',
2419 run_method=self.run)
2420 board = release_info['CHROMEOS_RELEASE_BOARD']
2421 # Devices in the lab generally have the correct board name but our own
2422 # development devices have {board_name}-signed-{key_type}. The board
2423 # name may also begin with 'x86-' which we need to keep.
Simran Basi833814b2013-01-29 13:13:43 -08002424 board_format_string = ds_constants.BOARD_PREFIX + '%s'
Simran Basic6f1f7a2012-10-16 10:47:46 -07002425 if 'x86' not in board:
Simran Basi833814b2013-01-29 13:13:43 -08002426 return board_format_string % board.split('-')[0]
2427 return board_format_string % '-'.join(board.split('-')[0:2])
Simran Basic6f1f7a2012-10-16 10:47:46 -07002428
2429
Ilja H. Friedel1232e8a2014-06-17 21:30:48 -07002430 @label_decorator('board_freq_mem')
2431 def get_board_with_frequency_and_memory(self):
2432 """
2433 Determines the board name with frequency and memory.
2434
2435 @returns a more detailed string representing the board. Examples are
2436 butterfly_1.1GHz_2GB, link_1.8GHz_4GB, x86-zgb_1.7GHz_2GB
2437 """
2438 board = self.run(self.poor_mans_rpc(
2439 'get_board_with_frequency_and_memory()')).stdout
2440 return 'board_freq_mem:%s' % str.strip(board)
2441
2442
Aviv Keshet74c89a92013-02-04 15:18:30 -08002443 @label_decorator('lightsensor')
Simran Basic6f1f7a2012-10-16 10:47:46 -07002444 def has_lightsensor(self):
2445 """Determine the correct board label for this host.
2446
2447 @returns the string 'lightsensor' if this host has a lightsensor or
2448 None if it does not.
2449 """
2450 search_cmd = "find -L %s -maxdepth 4 | egrep '%s'" % (
Richard Barnette82c35912012-11-20 10:09:10 -08002451 self._LIGHTSENSOR_SEARCH_DIR, '|'.join(self._LIGHTSENSOR_FILES))
Simran Basic6f1f7a2012-10-16 10:47:46 -07002452 try:
2453 # Run the search cmd following the symlinks. Stderr_tee is set to
2454 # None as there can be a symlink loop, but the command will still
2455 # execute correctly with a few messages printed to stderr.
2456 self.run(search_cmd, stdout_tee=None, stderr_tee=None)
2457 return 'lightsensor'
2458 except error.AutoservRunError:
2459 # egrep exited with a return code of 1 meaning none of the possible
2460 # lightsensor files existed.
2461 return None
2462
2463
Aviv Keshet74c89a92013-02-04 15:18:30 -08002464 @label_decorator('bluetooth')
Simran Basic6f1f7a2012-10-16 10:47:46 -07002465 def has_bluetooth(self):
2466 """Determine the correct board label for this host.
2467
2468 @returns the string 'bluetooth' if this host has bluetooth or
2469 None if it does not.
2470 """
2471 try:
2472 self.run('test -d /sys/class/bluetooth/hci0')
2473 # test exited with a return code of 0.
2474 return 'bluetooth'
2475 except error.AutoservRunError:
2476 # test exited with a return code 1 meaning the directory did not
2477 # exist.
2478 return None
2479
2480
Ilja H. Friedel1232e8a2014-06-17 21:30:48 -07002481 @label_decorator('gpu_family')
2482 def get_gpu_family(self):
2483 """
2484 Determine GPU family.
2485
2486 @returns a string representing the gpu family. Examples are mali, tegra,
2487 pinetrail, sandybridge, ivybridge, haswell and baytrail.
2488 """
2489 gpu_family = self.run(self.poor_mans_rpc('get_gpu_family()')).stdout
2490 return 'gpu_family:%s' % str.strip(gpu_family)
2491
2492
Ilja Friedel0ce0b602013-08-15 18:45:27 -07002493 @label_decorator('graphics')
2494 def get_graphics(self):
2495 """
2496 Determine the correct board label for this host.
2497
2498 @returns a string representing this host's graphics. For now ARM boards
2499 return graphics:gles while all other boards return graphics:gl. This
2500 may change over time, but for robustness reasons this should avoid
2501 executing code in actual graphics libraries (which may not be ready and
2502 is tested by graphics_GLAPICheck).
2503 """
2504 uname = self.run('uname -a').stdout.lower()
2505 if 'arm' in uname:
2506 return 'graphics:gles'
2507 return 'graphics:gl'
2508
2509
Bill Richardson4f595f52014-02-13 16:20:26 -08002510 @label_decorator('ec')
2511 def get_ec(self):
2512 """
2513 Determine the type of EC on this host.
2514
2515 @returns a string representing this host's embedded controller type.
2516 At present, it only returns "ec:cros", for Chrome OS ECs. Other types
2517 of EC (or none) don't return any strings, since no tests depend on
2518 those.
2519 """
2520 cmd = 'mosys ec info'
2521 # The output should look like these, so that the last field should
2522 # match our EC version scheme:
2523 #
2524 # stm | stm32f100 | snow_v1.3.139-375eb9f
2525 # ti | Unknown-10de | peppy_v1.5.114-5d52788
2526 #
2527 # Non-Chrome OS ECs will look like these:
2528 #
2529 # ENE | KB932 | 00BE107A00
2530 # ite | it8518 | 3.08
2531 #
2532 # And some systems don't have ECs at all (Lumpy, for example).
2533 regexp = r'^.*\|\s*(\S+_v\d+\.\d+\.\d+-[0-9a-f]+)\s*$'
2534
2535 ecinfo = self.run(command=cmd, ignore_status=True)
2536 if ecinfo.exit_status == 0:
2537 res = re.search(regexp, ecinfo.stdout)
2538 if res:
2539 logging.info("EC version is %s", res.groups()[0])
2540 return 'ec:cros'
2541 logging.info("%s got: %s", cmd, ecinfo.stdout)
2542 # Has an EC, but it's not a Chrome OS EC
2543 return None
2544 logging.info("%s exited with status %d", cmd, ecinfo.exit_status)
2545 # No EC present
2546 return None
2547
2548
Alec Berg31b932b2014-04-04 16:09:11 -07002549 @label_decorator('accels')
2550 def get_accels(self):
2551 """
2552 Determine the type of accelerometers on this host.
2553
2554 @returns a string representing this host's accelerometer type.
2555 At present, it only returns "accel:cros-ec", for accelerometers
2556 attached to a Chrome OS EC, or none, if no accelerometers.
2557 """
2558 # Check to make sure we have ectool
2559 rv = self.run('which ectool', ignore_status=True)
2560 if rv.exit_status:
2561 logging.info("No ectool cmd found, assuming no EC accelerometers")
2562 return None
2563
2564 # Check that the EC supports the motionsense command
2565 rv = self.run('ectool motionsense', ignore_status=True)
2566 if rv.exit_status:
2567 logging.info("EC does not support motionsense command "
2568 "assuming no EC accelerometers")
2569 return None
2570
2571 # Check that EC motion sensors are active
2572 active = self.run('ectool motionsense active').stdout.split('\n')
2573 if active[0] == "0":
2574 logging.info("Motion sense inactive, assuming no EC accelerometers")
2575 return None
2576
2577 logging.info("EC accelerometers found")
2578 return 'accel:cros-ec'
2579
2580
Tom Wai-Hong Tam3d6790d2014-04-14 16:15:47 +08002581 @label_decorator('chameleon')
2582 def has_chameleon(self):
2583 """Determine if a Chameleon connected to this host.
2584
Tom Wai-Hong Tambadbb332014-10-10 02:59:41 +08002585 @returns a list containing two strings ('chameleon' and
2586 'chameleon:' + label, e.g. 'chameleon:hdmi') if this host
2587 has a Chameleon or None if it has not.
Tom Wai-Hong Tam3d6790d2014-04-14 16:15:47 +08002588 """
2589 if self._chameleon_host:
Tom Wai-Hong Tambadbb332014-10-10 02:59:41 +08002590 return ['chameleon', 'chameleon:' + self.chameleon.get_label()]
Tom Wai-Hong Tam3d6790d2014-04-14 16:15:47 +08002591 else:
2592 return None
2593
2594
Cheng-Yi Chiangf4104ff2014-12-23 19:39:01 +08002595 @label_decorator('audio_loopback_dongle')
2596 def has_loopback_dongle(self):
2597 """Determine if an audio loopback dongle is plugged to this host.
2598
2599 @returns 'audio_loopback_dongle' when there is an audio loopback dongle
2600 plugged to this host.
2601 None when there is no audio loopback dongle
2602 plugged to this host.
2603 """
Cheng-Yi Chiang8de78112015-05-27 14:47:08 +08002604 nodes_info = self.run(command=cras_utils.get_cras_nodes_cmd(),
2605 ignore_status=True).stdout
2606 if (cras_utils.node_type_is_plugged('HEADPHONE', nodes_info) and
2607 cras_utils.node_type_is_plugged('MIC', nodes_info)):
Cheng-Yi Chiangf4104ff2014-12-23 19:39:01 +08002608 return 'audio_loopback_dongle'
2609 else:
2610 return None
2611
2612
Derek Basehorec71ff622014-07-07 15:18:40 -07002613 @label_decorator('power_supply')
2614 def get_power_supply(self):
2615 """
2616 Determine what type of power supply the host has
2617
2618 @returns a string representing this host's power supply.
2619 'power:battery' when the device has a battery intended for
2620 extended use
2621 'power:AC_primary' when the device has a battery not intended
2622 for extended use (for moving the machine, etc)
2623 'power:AC_only' when the device has no battery at all.
2624 """
2625 psu = self.run(command='mosys psu type', ignore_status=True)
2626 if psu.exit_status:
2627 # The psu command for mosys is not included for all platforms. The
2628 # assumption is that the device will have a battery if the command
2629 # is not found.
2630 return 'power:battery'
2631
2632 psu_str = psu.stdout.strip()
2633 if psu_str == 'unknown':
2634 return None
2635
2636 return 'power:%s' % psu_str
2637
2638
Puthikorn Voravootivatfa011242014-03-14 18:45:11 -07002639 @label_decorator('storage')
2640 def get_storage(self):
2641 """
2642 Determine the type of boot device for this host.
2643
2644 Determine if the internal device is SCSI or dw_mmc device.
2645 Then check that it is SSD or HDD or eMMC or something else.
2646
2647 @returns a string representing this host's internal device type.
2648 'storage:ssd' when internal device is solid state drive
2649 'storage:hdd' when internal device is hard disk drive
2650 'storage:mmc' when internal device is mmc drive
2651 None When internal device is something else or
2652 when we are unable to determine the type
2653 """
2654 # The output should be /dev/mmcblk* for SD/eMMC or /dev/sd* for scsi
2655 rootdev_cmd = ' '.join(['. /usr/sbin/write_gpt.sh;',
2656 '. /usr/share/misc/chromeos-common.sh;',
2657 'load_base_vars;',
2658 'get_fixed_dst_drive'])
Puthikorn Voravootivat03c51682014-04-24 13:52:12 -07002659 rootdev = self.run(command=rootdev_cmd, ignore_status=True)
2660 if rootdev.exit_status:
2661 logging.info("Fail to run %s", rootdev_cmd)
2662 return None
Puthikorn Voravootivatfa011242014-03-14 18:45:11 -07002663 rootdev_str = rootdev.stdout.strip()
2664
2665 if not rootdev_str:
2666 return None
2667
2668 rootdev_base = os.path.basename(rootdev_str)
2669
2670 mmc_pattern = '/dev/mmcblk[0-9]'
2671 if re.match(mmc_pattern, rootdev_str):
2672 # Use type to determine if the internal device is eMMC or somthing
2673 # else. We can assume that MMC is always an internal device.
2674 type_cmd = 'cat /sys/block/%s/device/type' % rootdev_base
Puthikorn Voravootivat03c51682014-04-24 13:52:12 -07002675 type = self.run(command=type_cmd, ignore_status=True)
2676 if type.exit_status:
2677 logging.info("Fail to run %s", type_cmd)
2678 return None
Puthikorn Voravootivatfa011242014-03-14 18:45:11 -07002679 type_str = type.stdout.strip()
2680
2681 if type_str == 'MMC':
2682 return 'storage:mmc'
2683
2684 scsi_pattern = '/dev/sd[a-z]+'
2685 if re.match(scsi_pattern, rootdev.stdout):
2686 # Read symlink for /sys/block/sd* to determine if the internal
2687 # device is connected via ata or usb.
2688 link_cmd = 'readlink /sys/block/%s' % rootdev_base
Puthikorn Voravootivat03c51682014-04-24 13:52:12 -07002689 link = self.run(command=link_cmd, ignore_status=True)
2690 if link.exit_status:
2691 logging.info("Fail to run %s", link_cmd)
2692 return None
Puthikorn Voravootivatfa011242014-03-14 18:45:11 -07002693 link_str = link.stdout.strip()
2694 if 'usb' in link_str:
2695 return None
2696
2697 # Read rotation to determine if the internal device is ssd or hdd.
2698 rotate_cmd = str('cat /sys/block/%s/queue/rotational'
2699 % rootdev_base)
Puthikorn Voravootivat03c51682014-04-24 13:52:12 -07002700 rotate = self.run(command=rotate_cmd, ignore_status=True)
2701 if rotate.exit_status:
2702 logging.info("Fail to run %s", rotate_cmd)
2703 return None
Puthikorn Voravootivatfa011242014-03-14 18:45:11 -07002704 rotate_str = rotate.stdout.strip()
2705
2706 rotate_dict = {'0':'storage:ssd', '1':'storage:hdd'}
2707 return rotate_dict.get(rotate_str)
2708
2709 # All other internal device / error case will always fall here
2710 return None
2711
2712
Dan Shi4e9a2aa2014-03-24 14:28:42 -07002713 @label_decorator('servo')
2714 def get_servo(self):
2715 """Determine if the host has a servo attached.
2716
2717 If the host has a working servo attached, it should have a servo label.
2718
2719 @return: string 'servo' if the host has servo attached. Otherwise,
2720 returns None.
2721 """
2722 return 'servo' if self._servo_host else None
2723
2724
Dan Shi5beba472014-05-28 22:46:07 -07002725 @label_decorator('video_labels')
2726 def get_video_labels(self):
2727 """Run /usr/local/bin/avtest_label_detect to get a list of video labels.
2728
2729 Sample output of avtest_label_detect:
2730 Detected label: hw_video_acc_vp8
2731 Detected label: webcam
2732
2733 @return: A list of labels detected by tool avtest_label_detect.
2734 """
2735 try:
2736 result = self.run('/usr/local/bin/avtest_label_detect').stdout
2737 return re.findall('^Detected label: (\w+)$', result, re.M)
2738 except error.AutoservRunError:
2739 # The tool is not installed.
2740 return []
2741
2742
mussa584b4462014-06-20 15:13:28 -07002743 @label_decorator('video_glitch_detection')
2744 def is_video_glitch_detection_supported(self):
2745 """ Determine if a board under test is supported for video glitch
2746 detection tests.
2747
2748 @return: 'video_glitch_detection' if board is supported, None otherwise.
2749 """
2750 parser = ConfigParser.SafeConfigParser()
2751 filename = os.path.join(
2752 common.autotest_dir, 'client/cros/video/device_spec.conf')
2753
2754 dut = self.get_board().replace(ds_constants.BOARD_PREFIX, '')
2755
2756 try:
2757 parser.read(filename)
mussa584b4462014-06-20 15:13:28 -07002758 supported_boards = parser.sections()
2759
Mussa83c84d62014-10-02 12:11:28 -07002760 return 'video_glitch_detection' if dut in supported_boards else None
mussa584b4462014-06-20 15:13:28 -07002761
2762 except ConfigParser.error:
2763 # something went wrong while parsing the conf file
2764 return None
2765
Katherine Threlkeld7b97a9f2014-06-24 13:47:14 -07002766 @label_decorator('touch_labels')
2767 def get_touch(self):
2768 """
2769 Determine whether board under test has a touchpad or touchscreen.
2770
2771 @return: A list of some combination of 'touchscreen' and 'touchpad',
2772 depending on what is present on the device.
Katherine Threlkeldab83d392015-06-18 16:45:57 -07002773
Katherine Threlkeld7b97a9f2014-06-24 13:47:14 -07002774 """
2775 labels = []
Katherine Threlkeldab83d392015-06-18 16:45:57 -07002776 looking_for = ['touchpad', 'touchscreen']
2777 player = input_playback.InputPlayback()
2778 input_events = self.run('ls /dev/input/event*').stdout.strip().split()
2779 filename = '/tmp/touch_labels'
2780 for event in input_events:
2781 self.run('evtest %s > %s' % (event, filename), timeout=1,
2782 ignore_timeout=True)
2783 properties = self.run('cat %s' % filename).stdout
2784 input_type = player._determine_input_type(properties)
2785 if input_type in looking_for:
2786 labels.append(input_type)
2787 looking_for.remove(input_type)
2788 if len(looking_for) == 0:
2789 break
2790 self.run('rm %s' % filename)
2791
Katherine Threlkeld7b97a9f2014-06-24 13:47:14 -07002792 return labels
2793
Hung-ying Tyana39b0542015-06-30 10:36:42 +08002794
2795 @label_decorator('internal_display')
2796 def has_internal_display(self):
2797 """Determine if the device under test is equipped with an internal
2798 display.
2799
2800 @return: 'internal_display' if one is present; None otherwise.
2801 """
2802 from autotest_lib.client.cros.graphics import graphics_utils
2803 from autotest_lib.client.common_lib import utils as common_utils
2804
2805 def __system_output(cmd):
2806 return self.run(cmd).stdout
2807
2808 def __read_file(remote_path):
2809 return self.run('cat %s' % remote_path).stdout
2810
2811 # Hijack the necessary client functions so that we can take advantage
2812 # of the client lib here.
2813 # FIXME: find a less hacky way than this
2814 original_system_output = utils.system_output
2815 original_read_file = common_utils.read_file
2816 utils.system_output = __system_output
2817 common_utils.read_file = __read_file
2818 try:
2819 return ('internal_display' if graphics_utils.has_internal_display()
2820 else None)
2821 finally:
2822 utils.system_output = original_system_output
2823 common_utils.read_file = original_read_file
2824
2825
Eric Carusoee673ac2015-08-05 17:03:04 -07002826 @label_decorator('lucidsleep')
2827 def has_lucid_sleep_support(self):
2828 """Determine if the device under test has support for lucid sleep.
2829
2830 @return 'lucidsleep' if this board supports lucid sleep; None otherwise
2831 """
2832 board = self.get_board().replace(ds_constants.BOARD_PREFIX, '')
2833 return 'lucidsleep' if board in LUCID_SLEEP_BOARDS else None
2834
2835
Dan Shi85276d42014-04-08 22:11:45 -07002836 def is_boot_from_usb(self):
2837 """Check if DUT is boot from USB.
2838
2839 @return: True if DUT is boot from usb.
2840 """
2841 device = self.run('rootdev -s -d').stdout.strip()
2842 removable = int(self.run('cat /sys/block/%s/removable' %
2843 os.path.basename(device)).stdout.strip())
2844 return removable == 1
Helen Zhang17dae2b2014-11-11 09:25:52 -08002845
2846
2847 def read_from_meminfo(self, key):
Dan Shi49ca0932014-11-14 11:22:27 -08002848 """Return the memory info from /proc/meminfo
Helen Zhang17dae2b2014-11-11 09:25:52 -08002849
2850 @param key: meminfo requested
2851
2852 @return the memory value as a string
2853
2854 """
Helen Zhang17dae2b2014-11-11 09:25:52 -08002855 meminfo = self.run('grep %s /proc/meminfo' % key).stdout.strip()
2856 logging.debug('%s', meminfo)
2857 return int(re.search(r'\d+', meminfo).group(0))
Rohit Makasana8a4923c2015-08-13 17:04:26 -07002858
2859
2860 def get_board_type(self):
2861 """
2862 Get the DUT's device type from /etc/lsb-release.
Danny Chan471a8d12015-08-18 14:57:41 -07002863 DEVICETYPE can be one of CHROMEBOX, CHROMEBASE, CHROMEBOOK or more.
2864
2865 @return value of DEVICETYPE param from lsb-release.
Rohit Makasana8a4923c2015-08-13 17:04:26 -07002866 """
Danny Chan471a8d12015-08-18 14:57:41 -07002867 device_type = self.run('grep DEVICETYPE /etc/lsb-release',
2868 ignore_status=True).stdout
2869 if device_type:
Kalin Stoyanov524310b2015-08-21 16:24:04 -07002870 return device_type.split('=')[-1].strip()
Danny Chan471a8d12015-08-18 14:57:41 -07002871 return ''
Gilad Arnolda76bef02015-09-29 13:55:15 -07002872
2873
2874 def get_os_type(self):
2875 return 'cros'
Simran Basia5522a32015-10-06 11:01:24 -07002876
2877
2878 def enable_adb_testing(self):
2879 """Mark this host as an adb tester."""
2880 self.run('touch %s' % adb_host.ANDROID_TESTER_FILEFLAG)