blob: 5d63dc34f1f7d48974477b580a5025731d98bc25 [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
J. Richard Barnette1d78b012012-05-15 13:56:30 -07005import logging
Dan Shi0f466e82013-02-22 15:44:58 -08006import os
Simran Basid5e5e272012-09-24 15:23:59 -07007import re
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07008import time
9
mussa584b4462014-06-20 15:13:28 -070010import common
J. Richard Barnette45e93de2012-04-11 17:24:15 -070011from autotest_lib.client.bin import utils
Dan Shi9cb0eec2014-06-03 09:04:50 -070012from autotest_lib.client.common_lib import autotemp
Richard Barnette0c73ffc2012-11-19 15:21:18 -080013from autotest_lib.client.common_lib import error
14from autotest_lib.client.common_lib import global_config
J. Richard Barnette91137f02016-03-10 16:52:26 -080015from autotest_lib.client.common_lib import hosts
Dan Shi549fb822015-03-24 18:01:11 -070016from autotest_lib.client.common_lib import lsbrelease_utils
J. Richard Barnette45e93de2012-04-11 17:24:15 -070017from autotest_lib.client.common_lib.cros import autoupdater
Richard Barnette03a0c132012-11-05 12:40:35 -080018from autotest_lib.client.common_lib.cros import dev_server
Gabe Blackb72f4fb2015-01-20 16:47:13 -080019from autotest_lib.client.common_lib.cros.graphite import autotest_es
Gabe Black1e1c41b2015-02-04 23:55:15 -080020from autotest_lib.client.common_lib.cros.graphite import autotest_stats
Hsinyu Chaoe0b08e62015-08-11 10:50:37 +000021from autotest_lib.client.cros import constants as client_constants
J. Richard Barnette84890bd2014-02-21 11:05:47 -080022from autotest_lib.client.cros import cros_ui
Kevin Chenga328da62016-03-31 10:49:04 -070023from autotest_lib.client.cros.audio import cras_utils
24from autotest_lib.client.cros.input_playback import input_playback
25from autotest_lib.client.cros.video import constants as video_test_constants
Simran Basi5ace6f22016-01-06 17:30:44 -080026from autotest_lib.server import afe_utils
MK Ryu35d661e2014-09-25 17:44:10 -070027from autotest_lib.server import autoserv_parser
28from autotest_lib.server import autotest
29from autotest_lib.server import constants
Dan Shia1ecd5c2013-06-06 11:21:31 -070030from autotest_lib.server import utils as server_utils
Dan Shi9cb0eec2014-06-03 09:04:50 -070031from autotest_lib.server.cros import provision
Scott Zawalski89c44dd2013-02-26 09:28:02 -050032from autotest_lib.server.cros.dynamic_suite import constants as ds_constants
Simran Basi5e6339a2013-03-21 11:34:32 -070033from autotest_lib.server.cros.dynamic_suite import tools, frontend_wrappers
Dan Shi9cb0eec2014-06-03 09:04:50 -070034from autotest_lib.server.cros.faft.config.config import Config as FAFTConfig
Scottfe06ed82015-11-05 17:15:01 -080035from autotest_lib.server.cros.servo import plankton
Fang Deng96667ca2013-08-01 17:46:18 -070036from autotest_lib.server.hosts import abstract_ssh
Kevin Chenga2619dc2016-03-28 11:42:08 -070037from autotest_lib.server.hosts import base_label
38from autotest_lib.server.hosts import cros_label
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +080039from autotest_lib.server.hosts import chameleon_host
J. Richard Barnettea7e7fdc2016-02-12 12:35:36 -080040from autotest_lib.server.hosts import cros_repair
Scottfe06ed82015-11-05 17:15:01 -080041from autotest_lib.server.hosts import plankton_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
Dan Shib8540a52015-07-16 14:18:23 -070046CONFIG = global_config.global_config
47
Kevin Chenga328da62016-03-31 10:49:04 -070048LUCID_SLEEP_BOARDS = ['samus', 'lulu']
49
Dan Shid07ee2e2015-09-24 14:49:25 -070050
beepsc87ff602013-07-31 21:53:00 -070051class FactoryImageCheckerException(error.AutoservError):
52 """Exception raised when an image is a factory image."""
53 pass
54
55
Fang Deng0ca40e22013-08-27 17:47:44 -070056class CrosHost(abstract_ssh.AbstractSSHHost):
J. Richard Barnette45e93de2012-04-11 17:24:15 -070057 """Chromium OS specific subclass of Host."""
58
Simran Basi5ace6f22016-01-06 17:30:44 -080059 VERSION_PREFIX = provision.CROS_VERSION_PREFIX
60
J. Richard Barnette45e93de2012-04-11 17:24:15 -070061 _parser = autoserv_parser.autoserv_parser
Scott Zawalski62bacae2013-03-05 10:40:32 -050062 _AFE = frontend_wrappers.RetryingAFE(timeout_min=5, delay_sec=10)
J. Richard Barnette45e93de2012-04-11 17:24:15 -070063
Richard Barnette03a0c132012-11-05 12:40:35 -080064 # Timeout values (in seconds) associated with various Chrome OS
65 # state changes.
J. Richard Barnette134ec2c2012-04-25 12:59:37 -070066 #
Richard Barnette0c73ffc2012-11-19 15:21:18 -080067 # In general, a good rule of thumb is that the timeout can be up
68 # to twice the typical measured value on the slowest platform.
69 # The times here have not necessarily been empirically tested to
70 # meet this criterion.
J. Richard Barnetteeb69d722012-06-18 17:29:44 -070071 #
72 # SLEEP_TIMEOUT: Time to allow for suspend to memory.
Richard Barnette0c73ffc2012-11-19 15:21:18 -080073 # RESUME_TIMEOUT: Time to allow for resume after suspend, plus
74 # time to restart the netwowrk.
J. Richard Barnette84890bd2014-02-21 11:05:47 -080075 # SHUTDOWN_TIMEOUT: Time to allow for shut down.
J. Richard Barnetteeb69d722012-06-18 17:29:44 -070076 # BOOT_TIMEOUT: Time to allow for boot from power off. Among
Richard Barnette0c73ffc2012-11-19 15:21:18 -080077 # other things, this must account for the 30 second dev-mode
J. Richard Barnette417cc792015-10-01 09:56:36 -070078 # screen delay, time to start the network on the DUT, and the
79 # ssh timeout of 120 seconds.
J. Richard Barnetteeb69d722012-06-18 17:29:44 -070080 # USB_BOOT_TIMEOUT: Time to allow for boot from a USB device,
Richard Barnette0c73ffc2012-11-19 15:21:18 -080081 # including the 30 second dev-mode delay and time to start the
J. Richard Barnetted4649c62013-03-06 17:42:27 -080082 # network.
beepsf079cfb2013-09-18 17:49:51 -070083 # INSTALL_TIMEOUT: Time to allow for chromeos-install.
J. Richard Barnette84890bd2014-02-21 11:05:47 -080084 # POWERWASH_BOOT_TIMEOUT: Time to allow for a reboot that
85 # includes powerwash.
J. Richard Barnetteeb69d722012-06-18 17:29:44 -070086
87 SLEEP_TIMEOUT = 2
J. Richard Barnetted4649c62013-03-06 17:42:27 -080088 RESUME_TIMEOUT = 10
Tom Wai-Hong Tamfe005c22014-12-03 09:25:44 +080089 SHUTDOWN_TIMEOUT = 10
J. Richard Barnette417cc792015-10-01 09:56:36 -070090 BOOT_TIMEOUT = 150
J. Richard Barnette5bab5f52015-08-03 13:14:38 -070091 USB_BOOT_TIMEOUT = 300
J. Richard Barnette7817b052014-08-28 09:47:29 -070092 INSTALL_TIMEOUT = 480
Dan Shi2c88eed2013-11-12 10:18:38 -080093 POWERWASH_BOOT_TIMEOUT = 60
Chris Sosab76e0ee2013-05-22 16:55:41 -070094
Dan Shica503482015-03-30 17:23:25 -070095 # Minimum OS version that supports server side packaging. Older builds may
96 # not have server side package built or with Autotest code change to support
97 # server-side packaging.
Dan Shib8540a52015-07-16 14:18:23 -070098 MIN_VERSION_SUPPORT_SSP = CONFIG.get_config_value(
Dan Shiced09e42015-04-17 16:09:34 -070099 'AUTOSERV', 'min_version_support_ssp', type=int)
Dan Shica503482015-03-30 17:23:25 -0700100
J. Richard Barnette84890bd2014-02-21 11:05:47 -0800101 # REBOOT_TIMEOUT: How long to wait for a reboot.
102 #
Chris Sosab76e0ee2013-05-22 16:55:41 -0700103 # We have a long timeout to ensure we don't flakily fail due to other
104 # issues. Shorter timeouts are vetted in platform_RebootAfterUpdate.
Simran Basi1160e2c2013-10-04 16:00:24 -0700105 # TODO(sbasi - crbug.com/276094) Restore to 5 mins once the 'host did not
106 # return from reboot' bug is solved.
107 REBOOT_TIMEOUT = 480
Chris Sosab76e0ee2013-05-22 16:55:41 -0700108
Ismail Noorbasha07fdb612013-02-14 14:13:31 -0800109 # _USB_POWER_TIMEOUT: Time to allow for USB to power toggle ON and OFF.
110 # _POWER_CYCLE_TIMEOUT: Time to allow for manual power cycle.
111 _USB_POWER_TIMEOUT = 5
112 _POWER_CYCLE_TIMEOUT = 10
113
Dan Shib8540a52015-07-16 14:18:23 -0700114 _RPM_RECOVERY_BOARDS = CONFIG.get_config_value('CROS',
Richard Barnette82c35912012-11-20 10:09:10 -0800115 'rpm_recovery_boards', type=str).split(',')
116
Richard Barnette82c35912012-11-20 10:09:10 -0800117 _LAB_MACHINE_FILE = '/mnt/stateful_partition/.labmachine'
Fang Dengdeba14f2014-11-14 11:54:09 -0800118 _RPM_HOSTNAME_REGEX = ('chromeos(\d+)(-row(\d+))?-rack(\d+[a-z]*)'
119 '-host(\d+)')
Kevin Chenga328da62016-03-31 10:49:04 -0700120 _LIGHTSENSOR_FILES = [ "in_illuminance0_input",
121 "in_illuminance_input",
122 "in_illuminance0_raw",
123 "in_illuminance_raw",
124 "illuminance0_input"]
125 _LIGHTSENSOR_SEARCH_DIR = '/sys/bus/iio/devices'
J. Richard Barnette134ec2c2012-04-25 12:59:37 -0700126
J. Richard Barnetteb6de7e32013-02-14 13:28:04 -0800127 # Constants used in ping_wait_up() and ping_wait_down().
128 #
129 # _PING_WAIT_COUNT is the approximate number of polling
130 # cycles to use when waiting for a host state change.
131 #
132 # _PING_STATUS_DOWN and _PING_STATUS_UP are names used
133 # for arguments to the internal _ping_wait_for_status()
134 # method.
135 _PING_WAIT_COUNT = 40
136 _PING_STATUS_DOWN = False
137 _PING_STATUS_UP = True
138
Ismail Noorbasha07fdb612013-02-14 14:13:31 -0800139 # Allowed values for the power_method argument.
140
141 # POWER_CONTROL_RPM: Passed as default arg for power_off/on/cycle() methods.
142 # POWER_CONTROL_SERVO: Used in set_power() and power_cycle() methods.
143 # POWER_CONTROL_MANUAL: Used in set_power() and power_cycle() methods.
144 POWER_CONTROL_RPM = 'RPM'
145 POWER_CONTROL_SERVO = 'servoj10'
146 POWER_CONTROL_MANUAL = 'manual'
147
148 POWER_CONTROL_VALID_ARGS = (POWER_CONTROL_RPM,
149 POWER_CONTROL_SERVO,
150 POWER_CONTROL_MANUAL)
Richard Barnette0c73ffc2012-11-19 15:21:18 -0800151
Simran Basi5e6339a2013-03-21 11:34:32 -0700152 _RPM_OUTLET_CHANGED = 'outlet_changed'
153
Dan Shi9cb0eec2014-06-03 09:04:50 -0700154 # URL pattern to download firmware image.
Dan Shib8540a52015-07-16 14:18:23 -0700155 _FW_IMAGE_URL_PATTERN = CONFIG.get_config_value(
Dan Shi9cb0eec2014-06-03 09:04:50 -0700156 'CROS', 'firmware_url_pattern', type=str)
beeps687243d2013-07-18 15:29:27 -0700157
J. Richard Barnette91137f02016-03-10 16:52:26 -0800158
159 # A flag file to indicate provision failures. The file is created
160 # at the start of any AU procedure (see `machine_install()`). The
161 # file's location in stateful means that on successul update it will
162 # be removed. Thus, if this file exists, it indicates that we've
163 # tried and failed in a previous attempt to update.
164 PROVISION_FAILED = '/var/tmp/provision_failed'
MK Ryu35d661e2014-09-25 17:44:10 -0700165
MK Ryu35d661e2014-09-25 17:44:10 -0700166
J. Richard Barnette964fba02012-10-24 17:34:29 -0700167 @staticmethod
beeps46dadc92013-11-07 14:07:10 -0800168 def check_host(host, timeout=10):
169 """
170 Check if the given host is a chrome-os host.
171
172 @param host: An ssh host representing a device.
173 @param timeout: The timeout for the run command.
174
175 @return: True if the host device is chromeos.
176
beeps46dadc92013-11-07 14:07:10 -0800177 """
178 try:
Simran Basi933c8af2015-04-29 14:05:07 -0700179 result = host.run(
180 'grep -q CHROMEOS /etc/lsb-release && '
181 '! test -f /mnt/stateful_partition/.android_tester && '
182 '! grep -q moblab /etc/lsb-release',
183 ignore_status=True, timeout=timeout)
beeps46dadc92013-11-07 14:07:10 -0800184 except (error.AutoservRunError, error.AutoservSSHTimeout):
185 return False
186 return result.exit_status == 0
187
188
189 @staticmethod
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +0800190 def _extract_arguments(args_dict, key_subset):
191 """Extract options from `args_dict` and return a subset result.
J. Richard Barnette7214e0b2013-02-06 15:20:49 -0800192
193 Take the provided dictionary of argument options and return
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +0800194 a subset that represent standard arguments needed to construct
195 a test-assistant object (chameleon or servo) for a host. The
196 intent is to provide standard argument processing from
Christopher Wiley644ef3e2015-05-15 13:14:14 -0700197 CrosHost for tests that require a test-assistant board
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +0800198 to operate.
199
200 @param args_dict Dictionary from which to extract the arguments.
201 @param key_subset Tuple of keys to extract from the args_dict, e.g.
202 ('servo_host', 'servo_port').
203 """
204 result = {}
205 for arg in key_subset:
206 if arg in args_dict:
207 result[arg] = args_dict[arg]
208 return result
209
210
211 @staticmethod
212 def get_chameleon_arguments(args_dict):
213 """Extract chameleon options from `args_dict` and return the result.
214
215 Recommended usage:
216 ~~~~~~~~
217 args_dict = utils.args_to_dict(args)
218 chameleon_args = hosts.CrosHost.get_chameleon_arguments(args_dict)
219 host = hosts.create_host(machine, chameleon_args=chameleon_args)
220 ~~~~~~~~
221
222 @param args_dict Dictionary from which to extract the chameleon
223 arguments.
224 """
225 return CrosHost._extract_arguments(
226 args_dict, ('chameleon_host', 'chameleon_port'))
227
228
229 @staticmethod
Scottfe06ed82015-11-05 17:15:01 -0800230 def get_plankton_arguments(args_dict):
231 """Extract chameleon options from `args_dict` and return the result.
232
233 Recommended usage:
234 ~~~~~~~~
235 args_dict = utils.args_to_dict(args)
236 plankon_args = hosts.CrosHost.get_plankton_arguments(args_dict)
237 host = hosts.create_host(machine, plankton_args=polankton_args)
238 ~~~~~~~~
239
240 @param args_dict Dictionary from which to extract the plankton
241 arguments.
242 """
243 args = CrosHost._extract_arguments(
244 args_dict, ('plankton_host', 'plankton_port'))
245 return args
246
247
248 @staticmethod
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +0800249 def get_servo_arguments(args_dict):
250 """Extract servo options from `args_dict` and return the result.
J. Richard Barnette7214e0b2013-02-06 15:20:49 -0800251
252 Recommended usage:
253 ~~~~~~~~
254 args_dict = utils.args_to_dict(args)
Fang Deng0ca40e22013-08-27 17:47:44 -0700255 servo_args = hosts.CrosHost.get_servo_arguments(args_dict)
J. Richard Barnette7214e0b2013-02-06 15:20:49 -0800256 host = hosts.create_host(machine, servo_args=servo_args)
257 ~~~~~~~~
258
259 @param args_dict Dictionary from which to extract the servo
260 arguments.
261 """
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +0800262 return CrosHost._extract_arguments(
263 args_dict, ('servo_host', 'servo_port'))
J. Richard Barnette134ec2c2012-04-25 12:59:37 -0700264
J. Richard Barnette964fba02012-10-24 17:34:29 -0700265
J. Richard Barnette91137f02016-03-10 16:52:26 -0800266 def _initialize(self, hostname, chameleon_args=None, servo_args=None,
267 plankton_args=None, try_lab_servo=False,
268 ssh_verbosity_flag='', ssh_options='',
Fang Dengd1c2b732013-08-20 12:59:46 -0700269 *args, **dargs):
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +0800270 """Initialize superclasses, |self.chameleon|, and |self.servo|.
J. Richard Barnette67ccb872012-04-19 16:34:56 -0700271
Fang Denge545abb2014-12-30 18:43:47 -0800272 This method will attempt to create the test-assistant object
273 (chameleon/servo) when it is needed by the test. Check
274 the docstring of chameleon_host.create_chameleon_host and
275 servo_host.create_servo_host for how this is determined.
Fang Deng5d518f42013-08-02 14:04:32 -0700276
Fang Denge545abb2014-12-30 18:43:47 -0800277 @param hostname: Hostname of the dut.
278 @param chameleon_args: A dictionary that contains args for creating
279 a ChameleonHost. See chameleon_host for details.
280 @param servo_args: A dictionary that contains args for creating
281 a ServoHost object. See servo_host for details.
282 @param try_lab_servo: Boolean, False indicates that ServoHost should
283 not be created for a device in Cros test lab.
284 See servo_host for details.
285 @param ssh_verbosity_flag: String, to pass to the ssh command to control
286 verbosity.
287 @param ssh_options: String, other ssh options to pass to the ssh
288 command.
J. Richard Barnette67ccb872012-04-19 16:34:56 -0700289 """
Fang Deng0ca40e22013-08-27 17:47:44 -0700290 super(CrosHost, self)._initialize(hostname=hostname,
J. Richard Barnette45e93de2012-04-11 17:24:15 -0700291 *args, **dargs)
J. Richard Barnette91137f02016-03-10 16:52:26 -0800292 self._repair_strategy = cros_repair.create_cros_repair_strategy()
Kevin Chenga2619dc2016-03-28 11:42:08 -0700293 self.labels = base_label.LabelRetriever(cros_label.CROS_LABELS)
J. Richard Barnettef0859852012-08-20 14:55:50 -0700294 # self.env is a dictionary of environment variable settings
295 # to be exported for commands run on the host.
296 # LIBC_FATAL_STDERR_ can be useful for diagnosing certain
297 # errors that might happen.
298 self.env['LIBC_FATAL_STDERR_'] = '1'
Fang Dengd1c2b732013-08-20 12:59:46 -0700299 self._ssh_verbosity_flag = ssh_verbosity_flag
Aviv Keshetc5947fa2013-09-04 14:06:29 -0700300 self._ssh_options = ssh_options
Fang Deng5d518f42013-08-02 14:04:32 -0700301 # TODO(fdeng): We need to simplify the
302 # process of servo and servo_host initialization.
303 # crbug.com/298432
Fang Denge545abb2014-12-30 18:43:47 -0800304 self._servo_host = servo_host.create_servo_host(
305 dut=self.hostname, servo_args=servo_args,
306 try_lab_servo=try_lab_servo)
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +0800307 # TODO(waihong): Do the simplication on Chameleon too.
Tom Wai-Hong Tam3d6790d2014-04-14 16:15:47 +0800308 self._chameleon_host = chameleon_host.create_chameleon_host(
309 dut=self.hostname, chameleon_args=chameleon_args)
Scottfe06ed82015-11-05 17:15:01 -0800310 # Add plankton host if plankton args were added on command line
311 self._plankton_host = plankton_host.create_plankton_host(plankton_args)
Tom Wai-Hong Tam3d6790d2014-04-14 16:15:47 +0800312
Dan Shi4d478522014-02-14 13:46:32 -0800313 if self._servo_host is not None:
314 self.servo = self._servo_host.get_servo()
315 else:
316 self.servo = None
317
Tom Wai-Hong Tam3d6790d2014-04-14 16:15:47 +0800318 if self._chameleon_host:
Tom Wai-Hong Tameaee3402014-01-22 08:52:10 +0800319 self.chameleon = self._chameleon_host.create_chameleon_board()
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +0800320 else:
Tom Wai-Hong Tam3d6790d2014-04-14 16:15:47 +0800321 self.chameleon = None
Fang Deng5d518f42013-08-02 14:04:32 -0700322
Scottfe06ed82015-11-05 17:15:01 -0800323 if self._plankton_host:
324 self.plankton_servo = self._plankton_host.get_servo()
325 logging.info('plankton_servo: %r', self.plankton_servo)
326 # Create the plankton object used to access the ec uart
Scott07a848f2016-01-12 15:04:52 -0800327 self.plankton = plankton.Plankton(self.plankton_servo,
328 self._plankton_host.get_servod_server_proxy())
Scottfe06ed82015-11-05 17:15:01 -0800329 else:
Scott07a848f2016-01-12 15:04:52 -0800330 self.plankton = None
Scottfe06ed82015-11-05 17:15:01 -0800331
Fang Deng5d518f42013-08-02 14:04:32 -0700332
Dan Shi3d7a0e12015-10-12 11:55:45 -0700333 def get_repair_image_name(self, image_type='cros'):
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500334 """Generate a image_name from variables in the global config.
335
Dan Shi3d7a0e12015-10-12 11:55:45 -0700336 image_type is used to differentiate different images. Default is CrOS,
337 in which case, repair image's name follows the naming convention defined
338 in global setting CROS/stable_build_pattern.
339 If the image_type is not `cros`, the repair image will be looked up
340 using key `board_name/image_type`, e.g., daisy_spring/firmware.
341
342 @param image_type: Type of the image. Default is `cros`.
343
Dan Shi08173202015-11-12 13:08:45 -0800344 @returns a str of $board-version/$BUILD. Returns None if stable version
345 for the board and the default are both not set, e.g., stable
346 firmware version for a new board.
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500347
348 """
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500349 board = self._get_board_from_afe()
350 if board is None:
351 raise error.AutoservError('DUT has no board attribute, '
352 'cannot be repaired.')
Dan Shi3d7a0e12015-10-12 11:55:45 -0700353 if image_type != 'cros':
354 board = '%s/%s' % (board, image_type)
Simran Basibeb2bb22016-02-03 15:25:48 -0800355 stable_version = afe_utils.get_stable_version(board=board)
Dan Shi3d7a0e12015-10-12 11:55:45 -0700356 if image_type == 'cros':
357 build_pattern = CONFIG.get_config_value(
358 'CROS', 'stable_build_pattern')
359 stable_version = build_pattern % (board, stable_version)
Dan Shi08173202015-11-12 13:08:45 -0800360 elif image_type == 'firmware':
361 # If firmware stable version is not specified, `stable_version`
362 # from the RPC is the default stable version for CrOS image.
363 # firmware stable version must be from firmware branch, thus its
364 # value must be like board-firmware/R31-1234.0.0. Check if
365 # firmware exists in the stable version, if not, return None.
366 if 'firmware' not in stable_version:
367 return None
Dan Shi3d7a0e12015-10-12 11:55:45 -0700368 return stable_version
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500369
370
Chris Sosab76e0ee2013-05-22 16:55:41 -0700371 def lookup_job_repo_url(self):
372 """Looks up the job_repo_url for the host.
373
Dan Shi8190eb82016-02-11 17:15:58 -0800374 This is kept for backwards compatibility as AU test code in older
375 branch does not use server-side packaging and calls this method through
376 the host object.
377
378 TODO(dshi): Once R50 falls off the stable branch, we should remove this
379 method.
380
Chris Sosab76e0ee2013-05-22 16:55:41 -0700381 @returns job_repo_url from AFE or None if not found.
382
383 @raises KeyError if the host does not have a job_repo_url
384 """
Dan Shibe3636a2016-02-14 22:48:01 -0800385 return afe_utils.get_host_attribute(self, ds_constants.JOB_REPO_URL)
beepscb6f1e22013-06-28 19:14:10 -0700386
387
beepsdae65fd2013-07-26 16:24:41 -0700388 def verify_job_repo_url(self, tag=''):
beepscb6f1e22013-06-28 19:14:10 -0700389 """
390 Make sure job_repo_url of this host is valid.
391
joychen03eaad92013-06-26 09:55:21 -0700392 Eg: The job_repo_url "http://lmn.cd.ab.xyx:8080/static/\
beepscb6f1e22013-06-28 19:14:10 -0700393 lumpy-release/R29-4279.0.0/autotest/packages" claims to have the
394 autotest package for lumpy-release/R29-4279.0.0. If this isn't the case,
395 download and extract it. If the devserver embedded in the url is
396 unresponsive, update the job_repo_url of the host after staging it on
397 another devserver.
398
399 @param job_repo_url: A url pointing to the devserver where the autotest
400 package for this build should be staged.
beepsdae65fd2013-07-26 16:24:41 -0700401 @param tag: The tag from the server job, in the format
402 <job_id>-<user>/<hostname>, or <hostless> for a server job.
beepscb6f1e22013-06-28 19:14:10 -0700403
404 @raises DevServerException: If we could not resolve a devserver.
405 @raises AutoservError: If we're unable to save the new job_repo_url as
406 a result of choosing a new devserver because the old one failed to
407 respond to a health check.
beeps0c865032013-07-30 11:37:06 -0700408 @raises urllib2.URLError: If the devserver embedded in job_repo_url
409 doesn't respond within the timeout.
beepscb6f1e22013-06-28 19:14:10 -0700410 """
Dan Shibe3636a2016-02-14 22:48:01 -0800411 job_repo_url = afe_utils.get_host_attribute(self,
412 ds_constants.JOB_REPO_URL)
beepscb6f1e22013-06-28 19:14:10 -0700413 if not job_repo_url:
414 logging.warning('No job repo url set on host %s', self.hostname)
415 return
416
417 logging.info('Verifying job repo url %s', job_repo_url)
418 devserver_url, image_name = tools.get_devserver_build_from_package_url(
419 job_repo_url)
420
beeps0c865032013-07-30 11:37:06 -0700421 ds = dev_server.ImageServer(devserver_url)
beepscb6f1e22013-06-28 19:14:10 -0700422
423 logging.info('Staging autotest artifacts for %s on devserver %s',
424 image_name, ds.url())
beeps687243d2013-07-18 15:29:27 -0700425
426 start_time = time.time()
Simran Basi25e7a922014-10-31 11:56:10 -0700427 ds.stage_artifacts(image_name, ['autotest_packages'])
beeps687243d2013-07-18 15:29:27 -0700428 stage_time = time.time() - start_time
429
430 # Record how much of the verification time comes from a devserver
431 # restage. If we're doing things right we should not see multiple
432 # devservers for a given board/build/branch path.
433 try:
J. Richard Barnette3cbd76b2013-11-27 12:11:25 -0800434 board, build_type, branch = server_utils.ParseBuildName(
beeps687243d2013-07-18 15:29:27 -0700435 image_name)[:3]
J. Richard Barnette3cbd76b2013-11-27 12:11:25 -0800436 except server_utils.ParseBuildNameException:
beeps687243d2013-07-18 15:29:27 -0700437 pass
438 else:
beeps0c865032013-07-30 11:37:06 -0700439 devserver = devserver_url[
Chris Sosa65425082013-10-16 13:26:22 -0700440 devserver_url.find('/') + 2:devserver_url.rfind(':')]
beeps687243d2013-07-18 15:29:27 -0700441 stats_key = {
442 'board': board,
443 'build_type': build_type,
444 'branch': branch,
beeps0c865032013-07-30 11:37:06 -0700445 'devserver': devserver.replace('.', '_'),
beeps687243d2013-07-18 15:29:27 -0700446 }
Gabe Black1e1c41b2015-02-04 23:55:15 -0800447 autotest_stats.Gauge('verify_job_repo_url').send(
beeps687243d2013-07-18 15:29:27 -0700448 '%(board)s.%(build_type)s.%(branch)s.%(devserver)s' % stats_key,
449 stage_time)
beepscb6f1e22013-06-28 19:14:10 -0700450
Scott Zawalskieadbf702013-03-14 09:23:06 -0400451
Dan Shicf4d2032015-03-12 15:04:21 -0700452 def stage_server_side_package(self, image=None):
453 """Stage autotest server-side package on devserver.
454
455 @param image: Full path of an OS image to install or a build name.
456
457 @return: A url to the autotest server-side package.
458 """
459 if image:
460 image_name = tools.get_build_from_image(image)
461 if not image_name:
462 raise error.AutoservError(
463 'Failed to parse build name from %s' % image)
Dan Shi69b9b972016-04-15 23:19:57 -0700464 ds = dev_server.ImageServer.resolve(image_name, self.hostname)
Dan Shicf4d2032015-03-12 15:04:21 -0700465 else:
Dan Shibe3636a2016-02-14 22:48:01 -0800466 job_repo_url = afe_utils.get_host_attribute(
467 self, ds_constants.JOB_REPO_URL)
Dan Shicf4d2032015-03-12 15:04:21 -0700468 if job_repo_url:
469 devserver_url, image_name = (
470 tools.get_devserver_build_from_package_url(job_repo_url))
471 ds = dev_server.ImageServer(devserver_url)
472 else:
473 labels = self._AFE.get_labels(
474 name__startswith=ds_constants.VERSION_PREFIX,
475 host__hostname=self.hostname)
476 if not labels:
477 raise error.AutoservError(
478 'Failed to stage server-side package. The host has '
479 'no job_report_url attribute or version label.')
480 image_name = labels[0].name[len(ds_constants.VERSION_PREFIX):]
Dan Shi69b9b972016-04-15 23:19:57 -0700481 ds = dev_server.ImageServer.resolve(image_name, self.hostname)
Dan Shica503482015-03-30 17:23:25 -0700482
483 # Get the OS version of the build, for any build older than
484 # MIN_VERSION_SUPPORT_SSP, server side packaging is not supported.
485 match = re.match('.*/R\d+-(\d+)\.', image_name)
486 if match and int(match.group(1)) < self.MIN_VERSION_SUPPORT_SSP:
487 logging.warn('Build %s is older than %s. Server side packaging is '
488 'disabled.', image_name, self.MIN_VERSION_SUPPORT_SSP)
489 return None
490
Dan Shicf4d2032015-03-12 15:04:21 -0700491 ds.stage_artifacts(image_name, ['autotest_server_package'])
492 return '%s/static/%s/%s' % (ds.url(), image_name,
493 'autotest_server_package.tar.bz2')
494
495
Dan Shi0f466e82013-02-22 15:44:58 -0800496 def _try_stateful_update(self, update_url, force_update, updater):
497 """Try to use stateful update to initialize DUT.
498
499 When DUT is already running the same version that machine_install
500 tries to install, stateful update is a much faster way to clean up
501 the DUT for testing, compared to a full reimage. It is implemeted
502 by calling autoupdater.run_update, but skipping updating root, as
503 updating the kernel is time consuming and not necessary.
504
505 @param update_url: url of the image.
506 @param force_update: Set to True to update the image even if the DUT
507 is running the same version.
508 @param updater: ChromiumOSUpdater instance used to update the DUT.
509 @returns: True if the DUT was updated with stateful update.
510
511 """
Dan Shi10b98482016-02-02 14:38:50 -0800512 # Stop service ap-update-manager to prevent rebooting during autoupdate.
513 # The service is used in jetstream boards, but not other CrOS devices.
514 self.run('sudo stop ap-update-manager', ignore_status=True)
515
J. Richard Barnette3f731032014-04-07 17:42:59 -0700516 # TODO(jrbarnette): Yes, I hate this re.match() test case.
517 # It's better than the alternative: see crbug.com/360944.
518 image_name = autoupdater.url_to_image_name(update_url)
519 release_pattern = r'^.*-release/R[0-9]+-[0-9]+\.[0-9]+\.0$'
520 if not re.match(release_pattern, image_name):
521 return False
Dan Shi0f466e82013-02-22 15:44:58 -0800522 if not updater.check_version():
523 return False
524 if not force_update:
525 logging.info('Canceling stateful update because the new and '
526 'old versions are the same.')
527 return False
528 # Following folders should be rebuilt after stateful update.
529 # A test file is used to confirm each folder gets rebuilt after
530 # the stateful update.
531 folders_to_check = ['/var', '/home', '/mnt/stateful_partition']
532 test_file = '.test_file_to_be_deleted'
533 for folder in folders_to_check:
534 touch_path = os.path.join(folder, test_file)
535 self.run('touch %s' % touch_path)
536
Chris Sosae92399e2015-04-24 11:32:59 -0700537 updater.run_update(update_root=False)
Dan Shi0f466e82013-02-22 15:44:58 -0800538
539 # Reboot to complete stateful update.
Chris Sosab76e0ee2013-05-22 16:55:41 -0700540 self.reboot(timeout=self.REBOOT_TIMEOUT, wait=True)
Dan Shi0f466e82013-02-22 15:44:58 -0800541 check_file_cmd = 'test -f %s; echo $?'
542 for folder in folders_to_check:
543 test_file_path = os.path.join(folder, test_file)
544 result = self.run(check_file_cmd % test_file_path,
545 ignore_status=True)
546 if result.exit_status == 1:
547 return False
548 return True
549
550
J. Richard Barnette7275b612013-06-04 18:13:11 -0700551 def _post_update_processing(self, updater, expected_kernel=None):
Dan Shi0f466e82013-02-22 15:44:58 -0800552 """After the DUT is updated, confirm machine_install succeeded.
553
554 @param updater: ChromiumOSUpdater instance used to update the DUT.
J. Richard Barnette7275b612013-06-04 18:13:11 -0700555 @param expected_kernel: kernel expected to be active after reboot,
556 or `None` to skip rollback checking.
Dan Shi0f466e82013-02-22 15:44:58 -0800557
558 """
J. Richard Barnette7275b612013-06-04 18:13:11 -0700559 # Touch the lab machine file to leave a marker that
560 # distinguishes this image from other test images.
561 # Afterwards, we must re-run the autoreboot script because
562 # it depends on the _LAB_MACHINE_FILE.
J. Richard Barnette71cc1862015-12-02 10:32:38 -0800563 autoreboot_cmd = ('FILE="%s" ; [ -f "$FILE" ] || '
564 '( touch "$FILE" ; start autoreboot )')
565 self.run(autoreboot_cmd % self._LAB_MACHINE_FILE)
Chris Sosa65425082013-10-16 13:26:22 -0700566 updater.verify_boot_expectations(
567 expected_kernel, rollback_message=
Gilad Arnoldc26ae1f2015-10-22 16:09:41 -0700568 'Build %s failed to boot on %s; system rolled back to previous '
Chris Sosa65425082013-10-16 13:26:22 -0700569 'build' % (updater.update_version, self.hostname))
J. Richard Barnette7275b612013-06-04 18:13:11 -0700570 # Check that we've got the build we meant to install.
571 if not updater.check_version_to_confirm_install():
572 raise autoupdater.ChromiumOSError(
573 'Failed to update %s to build %s; found build '
574 '%s instead' % (self.hostname,
Chris Sosa65425082013-10-16 13:26:22 -0700575 updater.update_version,
Dan Shi0942b1d2015-03-31 11:07:00 -0700576 self.get_release_version()))
Dan Shi0f466e82013-02-22 15:44:58 -0800577
Chris Sosae92399e2015-04-24 11:32:59 -0700578 logging.debug('Cleaning up old autotest directories.')
579 try:
580 installed_autodir = autotest.Autotest.get_installed_autodir(self)
581 self.run('rm -rf ' + installed_autodir)
582 except autotest.AutodirNotFoundError:
583 logging.debug('No autotest installed directory found.')
584
Dan Shi0f466e82013-02-22 15:44:58 -0800585
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700586 def _stage_image_for_update(self, image_name=None):
Chris Sosae92399e2015-04-24 11:32:59 -0700587 """Stage a build on a devserver and return the update_url and devserver.
Scott Zawalskieadbf702013-03-14 09:23:06 -0400588
589 @param image_name: a name like lumpy-release/R27-3837.0.0
Chris Sosae92399e2015-04-24 11:32:59 -0700590 @returns a tuple with an update URL like:
Scott Zawalskieadbf702013-03-14 09:23:06 -0400591 http://172.22.50.205:8082/update/lumpy-release/R27-3837.0.0
Chris Sosae92399e2015-04-24 11:32:59 -0700592 and the devserver instance.
Scott Zawalskieadbf702013-03-14 09:23:06 -0400593 """
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700594 if not image_name:
595 image_name = self.get_repair_image_name()
Chris Sosae92399e2015-04-24 11:32:59 -0700596
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700597 logging.info('Staging build for AU: %s', image_name)
Dan Shi216389c2015-12-22 11:03:06 -0800598 devserver = dev_server.ImageServer.resolve(image_name, self.hostname)
Scott Zawalskieadbf702013-03-14 09:23:06 -0400599 devserver.trigger_download(image_name, synchronous=False)
Chris Sosae92399e2015-04-24 11:32:59 -0700600 return (tools.image_url_pattern() % (devserver.url(), image_name),
601 devserver)
Scott Zawalskieadbf702013-03-14 09:23:06 -0400602
603
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700604 def stage_image_for_servo(self, image_name=None):
605 """Stage a build on a devserver and return the update_url.
606
607 @param image_name: a name like lumpy-release/R27-3837.0.0
608 @returns an update URL like:
609 http://172.22.50.205:8082/update/lumpy-release/R27-3837.0.0
610 """
611 if not image_name:
612 image_name = self.get_repair_image_name()
613 logging.info('Staging build for servo install: %s', image_name)
Dan Shi216389c2015-12-22 11:03:06 -0800614 devserver = dev_server.ImageServer.resolve(image_name, self.hostname)
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700615 devserver.stage_artifacts(image_name, ['test_image'])
616 return devserver.get_test_image_url(image_name)
617
618
beepse539be02013-07-31 21:57:39 -0700619 def stage_factory_image_for_servo(self, image_name):
620 """Stage a build on a devserver and return the update_url.
621
622 @param image_name: a name like <baord>/4262.204.0
beeps12c0a3c2013-09-03 11:58:27 -0700623
beepse539be02013-07-31 21:57:39 -0700624 @return: An update URL, eg:
625 http://<devserver>/static/canary-channel/\
626 <board>/4262.204.0/factory_test/chromiumos_factory_image.bin
beeps12c0a3c2013-09-03 11:58:27 -0700627
628 @raises: ValueError if the factory artifact name is missing from
629 the config.
630
beepse539be02013-07-31 21:57:39 -0700631 """
632 if not image_name:
633 logging.error('Need an image_name to stage a factory image.')
634 return
635
Dan Shib8540a52015-07-16 14:18:23 -0700636 factory_artifact = CONFIG.get_config_value(
beeps12c0a3c2013-09-03 11:58:27 -0700637 'CROS', 'factory_artifact', type=str, default='')
638 if not factory_artifact:
639 raise ValueError('Cannot retrieve the factory artifact name from '
640 'autotest config, and hence cannot stage factory '
641 'artifacts.')
642
beepse539be02013-07-31 21:57:39 -0700643 logging.info('Staging build for servo install: %s', image_name)
Dan Shi216389c2015-12-22 11:03:06 -0800644 devserver = dev_server.ImageServer.resolve(image_name, self.hostname)
beepse539be02013-07-31 21:57:39 -0700645 devserver.stage_artifacts(
646 image_name,
beeps12c0a3c2013-09-03 11:58:27 -0700647 [factory_artifact],
648 archive_url=None)
beepse539be02013-07-31 21:57:39 -0700649
650 return tools.factory_image_url_pattern() % (devserver.url(), image_name)
651
652
Chris Sosaa3ac2152012-05-23 22:23:13 -0700653 def machine_install(self, update_url=None, force_update=False,
Richard Barnette0b023a72015-04-24 16:07:30 +0000654 local_devserver=False, repair=False,
655 force_full_update=False):
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500656 """Install the DUT.
657
Dan Shi0f466e82013-02-22 15:44:58 -0800658 Use stateful update if the DUT is already running the same build.
659 Stateful update does not update kernel and tends to run much faster
660 than a full reimage. If the DUT is running a different build, or it
661 failed to do a stateful update, full update, including kernel update,
662 will be applied to the DUT.
663
Simran Basi5ace6f22016-01-06 17:30:44 -0800664 Once a host enters machine_install its host attribute job_repo_url
665 (used for package install) will be removed and then updated.
Scott Zawalskieadbf702013-03-14 09:23:06 -0400666
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500667 @param update_url: The url to use for the update
668 pattern: http://$devserver:###/update/$build
669 If update_url is None and repair is True we will install the
Dan Shi6964fa52014-12-18 11:04:27 -0800670 stable image listed in afe_stable_versions table. If the table
671 is not setup, global_config value under CROS.stable_cros_version
672 will be used instead.
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500673 @param force_update: Force an update even if the version installed
674 is the same. Default:False
Christopher Wiley6a4ff932015-05-15 14:00:47 -0700675 @param local_devserver: Used by test_that to allow people to
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500676 use their local devserver. Default: False
Chris Sosae92399e2015-04-24 11:32:59 -0700677 @param repair: Forces update to repair image. Implies force_update.
Fang Deng3d3b9272014-12-22 12:20:28 -0800678 @param force_full_update: If True, do not attempt to run stateful
679 update, force a full reimage. If False, try stateful update
680 first when the dut is already installed with the same version.
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500681 @raises autoupdater.ChromiumOSError
682
Dan Shibe3636a2016-02-14 22:48:01 -0800683 @returns A tuple of (image_name, host_attributes).
684 image_name is the name of image installed, e.g.,
685 veyron_jerry-release/R50-7871.0.0
686 host_attributes is a dictionary of (attribute, value), which
687 can be saved to afe_host_attributes table in database. This
688 method returns a dictionary with a single entry of
689 `job_repo_url`: repo_url, where repo_url is a devserver url to
690 autotest packages.
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500691 """
Chris Sosae92399e2015-04-24 11:32:59 -0700692 devserver = None
Richard Barnette0b023a72015-04-24 16:07:30 +0000693 if repair:
Chris Sosae92399e2015-04-24 11:32:59 -0700694 update_url, devserver = self._stage_image_for_update()
Richard Barnette0b023a72015-04-24 16:07:30 +0000695 force_update = True
Dan Shi0f466e82013-02-22 15:44:58 -0800696
Chris Sosae92399e2015-04-24 11:32:59 -0700697 if not update_url and not self._parser.options.image:
698 raise error.AutoservError(
Dan Shid07ee2e2015-09-24 14:49:25 -0700699 'There is no update URL, nor a method to get one.')
Chris Sosae92399e2015-04-24 11:32:59 -0700700
701 if not update_url and self._parser.options.image:
702 # This is the base case where we have no given update URL i.e.
703 # dynamic suites logic etc. This is the most flexible case where we
704 # can serve an update from any of our fleet of devservers.
705 requested_build = self._parser.options.image
706 if not requested_build.startswith('http://'):
707 logging.debug('Update will be staged for this installation')
708 update_url, devserver = self._stage_image_for_update(
Dan Shid07ee2e2015-09-24 14:49:25 -0700709 requested_build)
Chris Sosae92399e2015-04-24 11:32:59 -0700710 else:
711 update_url = requested_build
712
713 logging.debug('Update URL is %s', update_url)
714
Dan Shif48f8132016-02-18 10:34:30 -0800715 # Report provision stats.
xixuan9e2c98d2016-02-26 19:04:53 -0800716 server_name = dev_server.ImageServer.get_server_name(update_url)
Dan Shif48f8132016-02-18 10:34:30 -0800717 server_name = server_name.replace('.', '_')
718 autotest_stats.Counter('cros_host_provision.' + server_name).increment()
719 autotest_stats.Counter('cros_host_provision.total').increment()
720
Dan Shid07ee2e2015-09-24 14:49:25 -0700721 # Create a file to indicate if provision fails. The file will be removed
722 # by stateful update or full install.
J. Richard Barnette91137f02016-03-10 16:52:26 -0800723 self.run('touch %s' % self.PROVISION_FAILED)
Dan Shid07ee2e2015-09-24 14:49:25 -0700724
Chris Sosae92399e2015-04-24 11:32:59 -0700725 update_complete = False
726 updater = autoupdater.ChromiumOSUpdater(
727 update_url, host=self, local_devserver=local_devserver)
Fang Deng3d3b9272014-12-22 12:20:28 -0800728 if not force_full_update:
729 try:
Chris Sosae92399e2015-04-24 11:32:59 -0700730 # If the DUT is already running the same build, try stateful
731 # update first as it's much quicker than a full re-image.
732 update_complete = self._try_stateful_update(
Dan Shid07ee2e2015-09-24 14:49:25 -0700733 update_url, force_update, updater)
Fang Deng3d3b9272014-12-22 12:20:28 -0800734 except Exception as e:
735 logging.exception(e)
J. Richard Barnette45e93de2012-04-11 17:24:15 -0700736
Dan Shi0f466e82013-02-22 15:44:58 -0800737 inactive_kernel = None
Chris Sosae92399e2015-04-24 11:32:59 -0700738 if update_complete or (not force_update and updater.check_version()):
739 logging.info('Install complete without full update')
740 else:
741 logging.info('DUT requires full update.')
742 self.reboot(timeout=self.REBOOT_TIMEOUT, wait=True)
Dan Shi10b98482016-02-02 14:38:50 -0800743 # Stop service ap-update-manager to prevent rebooting during
744 # autoupdate. The service is used in jetstream boards, but not other
745 # CrOS devices.
746 self.run('sudo stop ap-update-manager', ignore_status=True)
747
Chris Sosae92399e2015-04-24 11:32:59 -0700748 num_of_attempts = provision.FLAKY_DEVSERVER_ATTEMPTS
Chris Sosab7612bc2013-03-21 10:32:37 -0700749
Chris Sosae92399e2015-04-24 11:32:59 -0700750 while num_of_attempts > 0:
751 num_of_attempts -= 1
752 try:
753 updater.run_update()
754 except Exception:
755 logging.warn('Autoupdate did not complete.')
756 # Do additional check for the devserver health. Ideally,
757 # the autoupdater.py could raise an exception when it
758 # detected network flake but that would require
759 # instrumenting the update engine and parsing it log.
760 if (num_of_attempts <= 0 or
761 devserver is None or
xixuan9e2c98d2016-02-26 19:04:53 -0800762 dev_server.ImageServer.devserver_healthy(
Chris Sosae92399e2015-04-24 11:32:59 -0700763 devserver.url())):
Dan Shid07ee2e2015-09-24 14:49:25 -0700764 raise
J. Richard Barnette45e93de2012-04-11 17:24:15 -0700765
Chris Sosae92399e2015-04-24 11:32:59 -0700766 logging.warn('Devserver looks unhealthy. Trying another')
767 update_url, devserver = self._stage_image_for_update(
768 requested_build)
769 logging.debug('New Update URL is %s', update_url)
770 updater = autoupdater.ChromiumOSUpdater(
771 update_url, host=self,
772 local_devserver=local_devserver)
773 else:
774 break
J. Richard Barnette45e93de2012-04-11 17:24:15 -0700775
Chris Sosae92399e2015-04-24 11:32:59 -0700776 # Give it some time in case of IO issues.
777 time.sleep(10)
Dan Shi5699ac22014-12-19 10:55:49 -0800778
Chris Sosae92399e2015-04-24 11:32:59 -0700779 # Figure out active and inactive kernel.
780 active_kernel, inactive_kernel = updater.get_kernel_state()
Simran Basi13fa1ba2013-03-04 10:56:47 -0800781
Chris Sosae92399e2015-04-24 11:32:59 -0700782 # Ensure inactive kernel has higher priority than active.
783 if (updater.get_kernel_priority(inactive_kernel)
784 < updater.get_kernel_priority(active_kernel)):
785 raise autoupdater.ChromiumOSError(
786 'Update failed. The priority of the inactive kernel'
787 ' partition is less than that of the active kernel'
788 ' partition.')
789
790 # Updater has returned successfully; reboot the host.
791 self.reboot(timeout=self.REBOOT_TIMEOUT, wait=True)
792
793 self._post_update_processing(updater, inactive_kernel)
Simran Basi5ace6f22016-01-06 17:30:44 -0800794 image_name = autoupdater.url_to_image_name(update_url)
Dan Shibe3636a2016-02-14 22:48:01 -0800795 # update_url is different from devserver url needed to stage autotest
796 # packages, therefore, resolve a new devserver url here.
797 devserver_url = dev_server.ImageServer.resolve(image_name,
798 self.hostname).url()
799 repo_url = tools.get_package_url(devserver_url, image_name)
Simran Basi9cbdbc32016-02-11 18:15:46 -0800800 self.verify_software()
Dan Shibe3636a2016-02-14 22:48:01 -0800801 return image_name, {ds_constants.JOB_REPO_URL: repo_url}
J. Richard Barnette45e93de2012-04-11 17:24:15 -0700802
803
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +0800804 def _clear_fw_version_labels(self, rw_only):
805 """Clear firmware version labels from the machine.
806
807 @param rw_only: True to only clear fwrw_version; otherewise, clear
808 both fwro_version and fwrw_version.
809 """
Dan Shi9cb0eec2014-06-03 09:04:50 -0700810 labels = self._AFE.get_labels(
Dan Shi0723bf52015-06-24 10:52:38 -0700811 name__startswith=provision.FW_RW_VERSION_PREFIX,
Dan Shi9cb0eec2014-06-03 09:04:50 -0700812 host__hostname=self.hostname)
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +0800813 if not rw_only:
814 labels = labels + self._AFE.get_labels(
815 name__startswith=provision.FW_RO_VERSION_PREFIX,
816 host__hostname=self.hostname)
Dan Shi9cb0eec2014-06-03 09:04:50 -0700817 for label in labels:
818 label.remove_hosts(hosts=[self.hostname])
819
820
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +0800821 def _add_fw_version_label(self, build, rw_only):
Dan Shi9cb0eec2014-06-03 09:04:50 -0700822 """Add firmware version label to the machine.
823
824 @param build: Build of firmware.
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +0800825 @param rw_only: True to only add fwrw_version; otherwise, add both
826 fwro_version and fwrw_version.
Dan Shi9cb0eec2014-06-03 09:04:50 -0700827
828 """
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +0800829 fw_label = provision.fwrw_version_to_label(build)
MK Ryu73be9862015-07-06 12:25:00 -0700830 self._AFE.run('label_add_hosts', id=fw_label, hosts=[self.hostname])
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +0800831 if not rw_only:
832 fw_label = provision.fwro_version_to_label(build)
833 self._AFE.run('label_add_hosts', id=fw_label, hosts=[self.hostname])
Dan Shi9cb0eec2014-06-03 09:04:50 -0700834
835
Tom Wai-Hong Tam4ac78982016-01-08 02:34:37 +0800836 def firmware_install(self, build=None, rw_only=False):
Dan Shi9cb0eec2014-06-03 09:04:50 -0700837 """Install firmware to the DUT.
838
839 Use stateful update if the DUT is already running the same build.
840 Stateful update does not update kernel and tends to run much faster
841 than a full reimage. If the DUT is running a different build, or it
842 failed to do a stateful update, full update, including kernel update,
843 will be applied to the DUT.
844
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +0800845 Once a host enters firmware_install its fw[ro|rw]_version label will
846 be removed. After the firmware is updated successfully, a new
847 fw[ro|rw]_version label will be added to the host.
Dan Shi9cb0eec2014-06-03 09:04:50 -0700848
849 @param build: The build version to which we want to provision the
850 firmware of the machine,
851 e.g. 'link-firmware/R22-2695.1.144'.
Tom Wai-Hong Tam4ac78982016-01-08 02:34:37 +0800852 @param rw_only: True to only install firmware to its RW portions. Keep
853 the RO portions unchanged.
Dan Shi9cb0eec2014-06-03 09:04:50 -0700854
855 TODO(dshi): After bug 381718 is fixed, update here with corresponding
856 exceptions that could be raised.
857
858 """
859 if not self.servo:
860 raise error.TestError('Host %s does not have servo.' %
861 self.hostname)
862
863 # TODO(fdeng): use host.get_board() after
864 # crbug.com/271834 is fixed.
865 board = self._get_board_from_afe()
866
Chris Sosae92399e2015-04-24 11:32:59 -0700867 # If build is not set, try to install firmware from stable CrOS.
Dan Shi9cb0eec2014-06-03 09:04:50 -0700868 if not build:
Dan Shi3d7a0e12015-10-12 11:55:45 -0700869 build = self.get_repair_image_name(image_type='firmware')
870 if not build:
871 raise error.TestError(
872 'Failed to find stable firmware build for %s.',
873 self.hostname)
874 logging.info('Will install firmware from build %s.', build)
Dan Shi9cb0eec2014-06-03 09:04:50 -0700875
876 config = FAFTConfig(board)
877 if config.use_u_boot:
878 ap_image = 'image-%s.bin' % board
879 else: # Depthcharge platform
880 ap_image = 'image.bin'
881 ec_image = 'ec.bin'
Dan Shi216389c2015-12-22 11:03:06 -0800882 ds = dev_server.ImageServer.resolve(build, self.hostname)
Dan Shi9cb0eec2014-06-03 09:04:50 -0700883 ds.stage_artifacts(build, ['firmware'])
884
885 tmpd = autotemp.tempdir(unique_id='fwimage')
886 try:
887 fwurl = self._FW_IMAGE_URL_PATTERN % (ds.url(), build)
888 local_tarball = os.path.join(tmpd.name, os.path.basename(fwurl))
889 server_utils.system('wget -O %s %s' % (local_tarball, fwurl),
890 timeout=60)
Dan Shi9cb0eec2014-06-03 09:04:50 -0700891
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +0800892 self._clear_fw_version_labels(rw_only)
Tom Wai-Hong Tamf9db57a2016-03-17 05:32:22 +0800893 if self.get_ec():
894 logging.info('Will re-program EC %snow', 'RW ' if rw_only else '')
895 server_utils.system('tar xf %s -C %s %s' %
896 (local_tarball, tmpd.name, ec_image),
897 timeout=60)
898 self.servo.program_ec(os.path.join(tmpd.name, ec_image), rw_only)
899 else:
900 logging.info('Not a Chrome EC, ignore re-programing it')
Tom Wai-Hong Tam4ac78982016-01-08 02:34:37 +0800901 logging.info('Will re-program BIOS %snow', 'RW ' if rw_only else '')
Tom Wai-Hong Tamf9db57a2016-03-17 05:32:22 +0800902 server_utils.system('tar xf %s -C %s %s' %
903 (local_tarball, tmpd.name, ap_image),
904 timeout=60)
Tom Wai-Hong Tam4ac78982016-01-08 02:34:37 +0800905 self.servo.program_bios(os.path.join(tmpd.name, ap_image), rw_only)
Dan Shi9cb0eec2014-06-03 09:04:50 -0700906 self.servo.get_power_state_controller().reset()
907 time.sleep(self.servo.BOOT_DELAY)
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +0800908 self._add_fw_version_label(build, rw_only)
Dan Shi9cb0eec2014-06-03 09:04:50 -0700909 finally:
910 tmpd.clean()
911
912
Dan Shi10e992b2013-08-30 11:02:59 -0700913 def show_update_engine_log(self):
914 """Output update engine log."""
MK Ryu35d661e2014-09-25 17:44:10 -0700915 logging.debug('Dumping %s', client_constants.UPDATE_ENGINE_LOG)
916 self.run('cat %s' % client_constants.UPDATE_ENGINE_LOG)
Dan Shi10e992b2013-08-30 11:02:59 -0700917
918
Richard Barnette82c35912012-11-20 10:09:10 -0800919 def _get_board_from_afe(self):
920 """Retrieve this host's board from its labels in the AFE.
921
922 Looks for a host label of the form "board:<board>", and
923 returns the "<board>" part of the label. `None` is returned
924 if there is not a single, unique label matching the pattern.
925
926 @returns board from label, or `None`.
927 """
Dan Shia1ecd5c2013-06-06 11:21:31 -0700928 return server_utils.get_board_from_afe(self.hostname, self._AFE)
Simran Basi833814b2013-01-29 13:13:43 -0800929
930
Dan Shib3b6db32016-02-03 14:54:05 -0800931 def get_build(self):
932 """Retrieve the current build for this Host from the AFE.
933
934 Looks through this host's labels in the AFE to determine its build.
935 This method is replaced by afe_utils.get_build. It's kept here to
936 maintain backwards compatibility for test control files in older CrOS
937 builds (R48, R49 etc.) still call host.get_build, e.g.,
938 `provision_AutoUpdate.double`.
939 TODO(sbasi): Once R50 falls into release branch, this method can be
940 removed.
941
942 @returns The current build or None if it could not find it or if there
943 were multiple build labels assigned to this host.
944 """
945 return afe_utils.get_build(self)
946
947
beepsf079cfb2013-09-18 17:49:51 -0700948 def servo_install(self, image_url=None, usb_boot_timeout=USB_BOOT_TIMEOUT,
949 install_timeout=INSTALL_TIMEOUT):
Scott Zawalski62bacae2013-03-05 10:40:32 -0500950 """
951 Re-install the OS on the DUT by:
952 1) installing a test image on a USB storage device attached to the Servo
953 board,
Richard Barnette03a0c132012-11-05 12:40:35 -0800954 2) booting that image in recovery mode, and then
J. Richard Barnette31b2e312013-04-04 16:05:22 -0700955 3) installing the image with chromeos-install.
956
Scott Zawalski62bacae2013-03-05 10:40:32 -0500957 @param image_url: If specified use as the url to install on the DUT.
958 otherwise boot the currently staged image on the USB stick.
beepsf079cfb2013-09-18 17:49:51 -0700959 @param usb_boot_timeout: The usb_boot_timeout to use during reimage.
960 Factory images need a longer usb_boot_timeout than regular
961 cros images.
962 @param install_timeout: The timeout to use when installing the chromeos
963 image. Factory images need a longer install_timeout.
Richard Barnette03a0c132012-11-05 12:40:35 -0800964
Scott Zawalski62bacae2013-03-05 10:40:32 -0500965 @raises AutoservError if the image fails to boot.
beepsf079cfb2013-09-18 17:49:51 -0700966
J. Richard Barnette0199cc82014-12-05 17:08:40 -0800967 """
beepsf079cfb2013-09-18 17:49:51 -0700968 usb_boot_timer_key = ('servo_install.usb_boot_timeout_%s'
969 % usb_boot_timeout)
970 logging.info('Downloading image to USB, then booting from it. Usb boot '
971 'timeout = %s', usb_boot_timeout)
Gabe Black1e1c41b2015-02-04 23:55:15 -0800972 timer = autotest_stats.Timer(usb_boot_timer_key)
beepsf079cfb2013-09-18 17:49:51 -0700973 timer.start()
J. Richard Barnette31b2e312013-04-04 16:05:22 -0700974 self.servo.install_recovery_image(image_url)
beepsf079cfb2013-09-18 17:49:51 -0700975 if not self.wait_up(timeout=usb_boot_timeout):
J. Richard Barnette91137f02016-03-10 16:52:26 -0800976 raise hosts.AutoservRepairError(
Scott Zawalski62bacae2013-03-05 10:40:32 -0500977 'DUT failed to boot from USB after %d seconds' %
beepsf079cfb2013-09-18 17:49:51 -0700978 usb_boot_timeout)
979 timer.stop()
Scott Zawalski62bacae2013-03-05 10:40:32 -0500980
Tom Wai-Hong Tamf6b4f812015-08-08 04:14:59 +0800981 # The new chromeos-tpm-recovery has been merged since R44-7073.0.0.
982 # In old CrOS images, this command fails. Skip the error.
Tom Wai-Hong Tam27af7332015-07-25 06:09:39 +0800983 logging.info('Resetting the TPM status')
Tom Wai-Hong Tamf6b4f812015-08-08 04:14:59 +0800984 try:
985 self.run('chromeos-tpm-recovery')
986 except error.AutoservRunError:
987 logging.warn('chromeos-tpm-recovery is too old.')
988
Tom Wai-Hong Tam27af7332015-07-25 06:09:39 +0800989
beepsf079cfb2013-09-18 17:49:51 -0700990 install_timer_key = ('servo_install.install_timeout_%s'
991 % install_timeout)
Gabe Black1e1c41b2015-02-04 23:55:15 -0800992 timer = autotest_stats.Timer(install_timer_key)
beepsf079cfb2013-09-18 17:49:51 -0700993 timer.start()
994 logging.info('Installing image through chromeos-install.')
J. Richard Barnette9af19632015-09-25 12:18:03 -0700995 self.run('chromeos-install --yes', timeout=install_timeout)
996 self.halt()
beepsf079cfb2013-09-18 17:49:51 -0700997 timer.stop()
998
999 logging.info('Power cycling DUT through servo.')
J. Richard Barnette0199cc82014-12-05 17:08:40 -08001000 self.servo.get_power_state_controller().power_off()
Fang Dengafb88142013-05-30 17:44:31 -07001001 self.servo.switch_usbkey('off')
J. Richard Barnette0199cc82014-12-05 17:08:40 -08001002 # N.B. The Servo API requires that we use power_on() here
1003 # for two reasons:
1004 # 1) After turning on a DUT in recovery mode, you must turn
1005 # it off and then on with power_on() once more to
1006 # disable recovery mode (this is a Parrot specific
1007 # requirement).
1008 # 2) After power_off(), the only way to turn on is with
1009 # power_on() (this is a Storm specific requirement).
J. Richard Barnettefbcc7122013-07-24 18:24:59 -07001010 self.servo.get_power_state_controller().power_on()
beepsf079cfb2013-09-18 17:49:51 -07001011
1012 logging.info('Waiting for DUT to come back up.')
Richard Barnette03a0c132012-11-05 12:40:35 -08001013 if not self.wait_up(timeout=self.BOOT_TIMEOUT):
1014 raise error.AutoservError('DUT failed to reboot installed '
1015 'test image after %d seconds' %
Scott Zawalski62bacae2013-03-05 10:40:32 -05001016 self.BOOT_TIMEOUT)
1017
1018
Dan Shic1b8bdd2015-09-14 23:11:24 -07001019 def _setup_servo(self):
1020 """Try to force to create servo object if it's not set up yet.
1021 """
1022 if self.servo:
1023 return
1024
1025 try:
1026 # Setting servo_args to {} will force it to create the servo_host
1027 # object if possible.
1028 self._servo_host = servo_host.create_servo_host(
1029 dut=self.hostname, servo_args={})
1030 if self._servo_host:
1031 self.servo = self._servo_host.get_servo()
1032 else:
1033 logging.error('Failed to create servo_host object.')
1034 except Exception as e:
1035 logging.error('Failed to create servo object: %s', e)
1036
1037
Tom Wai-Hong Tambea741c2016-01-21 07:20:14 +08001038 def _is_firmware_repair_supported(self):
1039 """Check if the firmware repair is supported.
Dan Shi3d7a0e12015-10-12 11:55:45 -07001040
Tom Wai-Hong Tambea741c2016-01-21 07:20:14 +08001041 The firmware repair is only applicable to DUTs in pools listed in
1042 global config CROS/pools_support_firmware_repair.
Dan Shi3d7a0e12015-10-12 11:55:45 -07001043
Tom Wai-Hong Tambea741c2016-01-21 07:20:14 +08001044 @return: True if it is supported; otherwise False.
Dan Shi3d7a0e12015-10-12 11:55:45 -07001045 """
1046 logging.info('Checking if host %s can be repaired with firmware '
1047 'repair.', self.hostname)
1048 pools = server_utils.get_labels_from_afe(self.hostname, 'pool:',
1049 self._AFE)
1050 pools_support_firmware_repair = CONFIG.get_config_value('CROS',
1051 'pools_support_firmware_repair', type=str).split(',')
Tom Wai-Hong Tambea741c2016-01-21 07:20:14 +08001052
1053 return (pools and pools_support_firmware_repair and
1054 set(pools).intersection(set(pools_support_firmware_repair)))
1055
1056
Dan Shi90466352015-09-22 15:01:05 -07001057 def confirm_servo(self):
1058 """Confirm servo is initialized and verified.
1059
1060 @raise AutoservError: If servo is not initialized and verified.
1061 """
Tom Wai-Hong Tam244acc22016-01-09 07:04:21 +08001062 if self.servo and self._servo_host.required_by_test:
Dan Shi90466352015-09-22 15:01:05 -07001063 return
1064
1065 # Force to re-create the servo object to make sure servo is verified.
1066 logging.debug('Rebuilding the servo object.')
1067 self.servo = None
1068 self._servo_host = None
1069 self._setup_servo()
1070 if not self.servo:
1071 raise error.AutoservError('Failed to create servo object.')
1072
1073
J. Richard Barnettec2d99cf2015-11-18 12:46:15 -08001074 def repair(self):
1075 """Attempt to get the DUT to pass `self.verify()`.
Richard Barnette82c35912012-11-20 10:09:10 -08001076
1077 This overrides the base class function for repair; it does
J. Richard Barnette91137f02016-03-10 16:52:26 -08001078 not call back to the parent class, but instead relies on
1079 `self._repair_strategy` to coordinate the verification and
1080 repair steps needed to get the DUT working.
Richard Barnette82c35912012-11-20 10:09:10 -08001081
J. Richard Barnette91137f02016-03-10 16:52:26 -08001082 TODO(jrbarnette) Prior to invoking the repair strategy's
1083 `repair()` method, repair checks our servo, and if necessary,
1084 tries to repair it. This flow should instead be integrated
1085 into the main repair strategy.
Richard Barnette82c35912012-11-20 10:09:10 -08001086 """
J. Richard Barnette91137f02016-03-10 16:52:26 -08001087 # For a DUT connected to a moblab, the servo host creation flow
1088 # may not have created the servo object. Try again now that we
1089 # know we want the servo.
Nicolas Boichate30c7e12015-11-05 11:12:50 +08001090 if utils.is_moblab():
1091 self._setup_servo()
1092
Dan Shi4d478522014-02-14 13:46:32 -08001093 if self._servo_host and not self.servo:
Fang Deng03590af2013-10-07 17:34:20 -07001094 try:
J. Richard Barnette4fc59c42015-12-15 16:58:50 -08001095 self._servo_host.repair()
Fang Deng03590af2013-10-07 17:34:20 -07001096 except Exception as e:
Fang Deng03590af2013-10-07 17:34:20 -07001097 logging.error('Could not create a healthy servo: %s', e)
Dan Shi4d478522014-02-14 13:46:32 -08001098 self.servo = self._servo_host.get_servo()
Fang Deng5d518f42013-08-02 14:04:32 -07001099
J. Richard Barnette91137f02016-03-10 16:52:26 -08001100 self._repair_strategy.repair(self)
Scott Zawalski89c44dd2013-02-26 09:28:02 -05001101
Richard Barnette82c35912012-11-20 10:09:10 -08001102
J. Richard Barnette1d78b012012-05-15 13:56:30 -07001103 def close(self):
Fang Deng0ca40e22013-08-27 17:47:44 -07001104 super(CrosHost, self).close()
J. Richard Barnette1d78b012012-05-15 13:56:30 -07001105
1106
Dan Shi49ca0932014-11-14 11:22:27 -08001107 def get_power_supply_info(self):
1108 """Get the output of power_supply_info.
1109
1110 power_supply_info outputs the info of each power supply, e.g.,
1111 Device: Line Power
1112 online: no
1113 type: Mains
1114 voltage (V): 0
1115 current (A): 0
1116 Device: Battery
1117 state: Discharging
1118 percentage: 95.9276
1119 technology: Li-ion
1120
1121 Above output shows two devices, Line Power and Battery, with details of
1122 each device listed. This function parses the output into a dictionary,
1123 with key being the device name, and value being a dictionary of details
1124 of the device info.
1125
1126 @return: The dictionary of power_supply_info, e.g.,
1127 {'Line Power': {'online': 'yes', 'type': 'main'},
1128 'Battery': {'vendor': 'xyz', 'percentage': '100'}}
Dan Shie9b765d2014-12-29 16:59:49 -08001129 @raise error.AutoservRunError if power_supply_info tool is not found in
1130 the DUT. Caller should handle this error to avoid false failure
1131 on verification.
Dan Shi49ca0932014-11-14 11:22:27 -08001132 """
1133 result = self.run('power_supply_info').stdout.strip()
1134 info = {}
1135 device_name = None
1136 device_info = {}
1137 for line in result.split('\n'):
1138 pair = [v.strip() for v in line.split(':')]
1139 if len(pair) != 2:
1140 continue
1141 if pair[0] == 'Device':
1142 if device_name:
1143 info[device_name] = device_info
1144 device_name = pair[1]
1145 device_info = {}
1146 else:
1147 device_info[pair[0]] = pair[1]
1148 if device_name and not device_name in info:
1149 info[device_name] = device_info
1150 return info
1151
1152
1153 def get_battery_percentage(self):
1154 """Get the battery percentage.
1155
1156 @return: The percentage of battery level, value range from 0-100. Return
1157 None if the battery info cannot be retrieved.
1158 """
1159 try:
1160 info = self.get_power_supply_info()
1161 logging.info(info)
1162 return float(info['Battery']['percentage'])
Dan Shie9b765d2014-12-29 16:59:49 -08001163 except (KeyError, ValueError, error.AutoservRunError):
Dan Shi49ca0932014-11-14 11:22:27 -08001164 return None
1165
1166
1167 def is_ac_connected(self):
1168 """Check if the dut has power adapter connected and charging.
1169
1170 @return: True if power adapter is connected and charging.
1171 """
1172 try:
1173 info = self.get_power_supply_info()
1174 return info['Line Power']['online'] == 'yes'
Dan Shie9b765d2014-12-29 16:59:49 -08001175 except (KeyError, error.AutoservRunError):
1176 return None
Dan Shi49ca0932014-11-14 11:22:27 -08001177
1178
Simran Basi5e6339a2013-03-21 11:34:32 -07001179 def _cleanup_poweron(self):
1180 """Special cleanup method to make sure hosts always get power back."""
1181 afe = frontend_wrappers.RetryingAFE(timeout_min=5, delay_sec=10)
1182 hosts = afe.get_hosts(hostname=self.hostname)
1183 if not hosts or not (self._RPM_OUTLET_CHANGED in
1184 hosts[0].attributes):
1185 return
1186 logging.debug('This host has recently interacted with the RPM'
1187 ' Infrastructure. Ensuring power is on.')
1188 try:
1189 self.power_on()
Dan Shi7dca56e2014-11-11 17:07:56 -08001190 afe.set_host_attribute(self._RPM_OUTLET_CHANGED, None,
1191 hostname=self.hostname)
Simran Basi5e6339a2013-03-21 11:34:32 -07001192 except rpm_client.RemotePowerException:
Simran Basi5e6339a2013-03-21 11:34:32 -07001193 logging.error('Failed to turn Power On for this host after '
1194 'cleanup through the RPM Infrastructure.')
Gabe Blackb72f4fb2015-01-20 16:47:13 -08001195 autotest_es.post(
Dan Shi7dca56e2014-11-11 17:07:56 -08001196 type_str='RPM_poweron_failure',
1197 metadata={'hostname': self.hostname})
Dan Shi49ca0932014-11-14 11:22:27 -08001198
1199 battery_percentage = self.get_battery_percentage()
Dan Shif01ebe22014-12-05 13:10:57 -08001200 if battery_percentage and battery_percentage < 50:
Dan Shi49ca0932014-11-14 11:22:27 -08001201 raise
1202 elif self.is_ac_connected():
1203 logging.info('The device has power adapter connected and '
1204 'charging. No need to try to turn RPM on '
1205 'again.')
1206 afe.set_host_attribute(self._RPM_OUTLET_CHANGED, None,
1207 hostname=self.hostname)
1208 logging.info('Battery level is now at %s%%. The device may '
1209 'still have enough power to run test, so no '
1210 'exception will be raised.', battery_percentage)
1211
Simran Basi5e6339a2013-03-21 11:34:32 -07001212
beepsc87ff602013-07-31 21:53:00 -07001213 def _is_factory_image(self):
1214 """Checks if the image on the DUT is a factory image.
1215
1216 @return: True if the image on the DUT is a factory image.
1217 False otherwise.
1218 """
1219 result = self.run('[ -f /root/.factory_test ]', ignore_status=True)
1220 return result.exit_status == 0
1221
1222
1223 def _restart_ui(self):
J. Richard Barnette84890bd2014-02-21 11:05:47 -08001224 """Restart the Chrome UI.
beepsc87ff602013-07-31 21:53:00 -07001225
1226 @raises: FactoryImageCheckerException for factory images, since
1227 we cannot attempt to restart ui on them.
1228 error.AutoservRunError for any other type of error that
1229 occurs while restarting ui.
1230 """
1231 if self._is_factory_image():
Dan Shi549fb822015-03-24 18:01:11 -07001232 raise FactoryImageCheckerException('Cannot restart ui on factory '
1233 'images')
beepsc87ff602013-07-31 21:53:00 -07001234
J. Richard Barnette84890bd2014-02-21 11:05:47 -08001235 # TODO(jrbarnette): The command to stop/start the ui job
1236 # should live inside cros_ui, too. However that would seem
1237 # to imply interface changes to the existing start()/restart()
1238 # functions, which is a bridge too far (for now).
J. Richard Barnette6069aa12015-06-08 09:10:24 -07001239 prompt = cros_ui.get_chrome_session_ident(self)
J. Richard Barnette84890bd2014-02-21 11:05:47 -08001240 self.run('stop ui; start ui')
1241 cros_ui.wait_for_chrome_ready(prompt, self)
beepsc87ff602013-07-31 21:53:00 -07001242
1243
Dan Shi549fb822015-03-24 18:01:11 -07001244 def get_release_version(self):
1245 """Get the value of attribute CHROMEOS_RELEASE_VERSION from lsb-release.
1246
1247 @returns The version string in lsb-release, under attribute
1248 CHROMEOS_RELEASE_VERSION.
1249 """
1250 lsb_release_content = self.run(
1251 'cat "%s"' % client_constants.LSB_RELEASE).stdout.strip()
1252 return lsbrelease_utils.get_chromeos_release_version(
1253 lsb_release_content=lsb_release_content)
1254
1255
1256 def verify_cros_version_label(self):
1257 """ Make sure host's cros-version label match the actual image in dut.
1258
1259 Remove any cros-version: label that doesn't match that installed in
1260 the dut.
1261
1262 @param raise_error: Set to True to raise exception if any mismatch found
1263
1264 @raise error.AutoservError: If any mismatch between cros-version label
1265 and the build installed in dut is found.
1266 """
1267 labels = self._AFE.get_labels(
1268 name__startswith=ds_constants.VERSION_PREFIX,
1269 host__hostname=self.hostname)
1270 mismatch_found = False
1271 if labels:
1272 # Get CHROMEOS_RELEASE_VERSION from lsb-release, e.g., 6908.0.0.
1273 # Note that it's different from cros-version label, which has
1274 # builder and branch info, e.g.,
1275 # cros-version:peppy-release/R43-6908.0.0
1276 release_version = self.get_release_version()
1277 host_list = [self.hostname]
1278 for label in labels:
1279 # Remove any cros-version label that does not match
1280 # release_version.
1281 build_version = label.name[len(ds_constants.VERSION_PREFIX):]
1282 if not utils.version_match(build_version, release_version):
1283 logging.warn('cros-version label "%s" does not match '
1284 'release version %s. Removing the label.',
1285 label.name, release_version)
1286 label.remove_hosts(hosts=host_list)
1287 mismatch_found = True
1288 if mismatch_found:
Dan Shi1057bae2015-03-30 11:35:09 -07001289 autotest_es.post(use_http=True,
1290 type_str='cros_version_label_mismatch',
1291 metadata={'hostname': self.hostname})
Dan Shi549fb822015-03-24 18:01:11 -07001292 raise error.AutoservError('The host has wrong cros-version label.')
1293
1294
Tom Wai-Hong Tambea741c2016-01-21 07:20:14 +08001295 def verify_firmware_status(self):
1296 """Verify the host's firmware is in a good state.
1297
1298 @raise error.AutoservError: If state is not good.
1299 """
1300 if self._is_firmware_repair_supported():
Tom Wai-Hong Tam93a29562016-01-23 04:16:53 +08001301 try:
1302 # Read the AP firmware and dump the sections we are interested.
1303 cmd = ('mkdir /tmp/verify_firmware; '
1304 'cd /tmp/verify_firmware; '
1305 'for section in VBLOCK_A VBLOCK_B FW_MAIN_A FW_MAIN_B; '
1306 'do flashrom -r image.bin -i $section:$section; '
1307 'done')
1308 self.run(cmd)
1309
1310 # Verify the firmware blocks A and B.
1311 cmd = ('vbutil_firmware --verify /tmp/verify_firmware/VBLOCK_%c'
1312 ' --signpubkey /usr/share/vboot/devkeys/root_key.vbpubk'
1313 ' --fv /tmp/verify_firmware/FW_MAIN_%c')
1314 for c in ('A', 'B'):
1315 rv = self.run(cmd % (c, c), ignore_status=True)
1316 if rv.exit_status:
Dan Shic67f1332016-04-06 12:38:06 -07001317 raise error.AutoservError(
1318 'Firmware %c is in a bad state.' % c)
Tom Wai-Hong Tam93a29562016-01-23 04:16:53 +08001319 finally:
1320 # Remove the tempoary files.
1321 self.run('rm -rf /tmp/verify_firmware')
Tom Wai-Hong Tambea741c2016-01-21 07:20:14 +08001322 else:
1323 logging.info('Do not care about firmware status when the host '
1324 'is not in pools that support firmware repair.')
1325
1326
Shelley Chena11b9e72016-01-21 15:15:18 -08001327 def verify_filesystem_write_status(self):
1328 """Verify the DUT's filesystem is read/writable
1329
1330 @raise error.AutoservError: if filesystem is not writable.
1331 """
1332 # try to create & delete a file
1333 filename = "/mnt/stateful_partition/test.txt"
1334 cmd = 'touch %s && rm %s' % (filename, filename)
1335 rv = self.run(command=cmd, ignore_status=True)
1336
1337 if rv.exit_status == 1:
1338 raise error.AutoservError('DUT filesystem is read-only.')
1339
1340
beepsc87ff602013-07-31 21:53:00 -07001341 def cleanup(self):
MK Ryu35d661e2014-09-25 17:44:10 -07001342 self.run('rm -f %s' % client_constants.CLEANUP_LOGS_PAUSED_FILE)
Scott Zawalskiddbc31e2012-11-15 11:29:01 -05001343 try:
beepsc87ff602013-07-31 21:53:00 -07001344 self._restart_ui()
1345 except (error.AutotestRunError, error.AutoservRunError,
1346 FactoryImageCheckerException):
Ilja H. Friedel04be2bd2014-05-07 21:29:59 -07001347 logging.warning('Unable to restart ui, rebooting device.')
Scott Zawalskiddbc31e2012-11-15 11:29:01 -05001348 # Since restarting the UI fails fall back to normal Autotest
1349 # cleanup routines, i.e. reboot the machine.
Fang Deng0ca40e22013-08-27 17:47:44 -07001350 super(CrosHost, self).cleanup()
Simran Basi5e6339a2013-03-21 11:34:32 -07001351 # Check if the rpm outlet was manipulated.
Simran Basid5e5e272012-09-24 15:23:59 -07001352 if self.has_power():
Simran Basi5e6339a2013-03-21 11:34:32 -07001353 self._cleanup_poweron()
Dan Shi549fb822015-03-24 18:01:11 -07001354 self.verify_cros_version_label()
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001355
1356
Yu-Ju Honga2be94a2012-07-31 09:48:52 -07001357 def reboot(self, **dargs):
1358 """
1359 This function reboots the site host. The more generic
1360 RemoteHost.reboot() performs sync and sleeps for 5
1361 seconds. This is not necessary for Chrome OS devices as the
1362 sync should be finished in a short time during the reboot
1363 command.
1364 """
Tom Wai-Hong Tamf5cd1d42012-08-13 12:04:08 +08001365 if 'reboot_cmd' not in dargs:
Doug Anderson7d5aeb22014-02-27 15:12:17 -08001366 reboot_timeout = dargs.get('reboot_timeout', 10)
J. Richard Barnette9af19632015-09-25 12:18:03 -07001367 dargs['reboot_cmd'] = ('sleep 1; '
1368 'reboot & sleep %d; '
1369 'reboot -f' % reboot_timeout)
Yu-Ju Honga2be94a2012-07-31 09:48:52 -07001370 # Enable fastsync to avoid running extra sync commands before reboot.
Tom Wai-Hong Tamf5cd1d42012-08-13 12:04:08 +08001371 if 'fastsync' not in dargs:
1372 dargs['fastsync'] = True
Michael Liangda8c60a2014-06-03 13:24:51 -07001373
Charlie Mooneya8e6dab2014-05-29 14:37:55 -07001374 # For purposes of logging reboot times:
1375 # Get the board name i.e. 'daisy_spring'
Michael Liangca4f5a62014-07-10 15:45:13 -07001376 board_fullname = self.get_board()
1377
1378 # Strip the prefix and add it to dargs.
1379 dargs['board'] = board_fullname[board_fullname.find(':')+1:]
Fang Deng0ca40e22013-08-27 17:47:44 -07001380 super(CrosHost, self).reboot(**dargs)
Yu-Ju Honga2be94a2012-07-31 09:48:52 -07001381
1382
Gwendal Grignou7a61d2f2014-05-23 11:05:51 -07001383 def suspend(self, **dargs):
1384 """
1385 This function suspends the site host.
1386 """
1387 suspend_time = dargs.get('suspend_time', 60)
1388 dargs['timeout'] = suspend_time
1389 if 'suspend_cmd' not in dargs:
J. Richard Barnette9af19632015-09-25 12:18:03 -07001390 dargs['suspend_cmd'] = ' && '.join([
1391 'echo 0 > /sys/class/rtc/rtc0/wakealarm',
Gwendal Grignou7a61d2f2014-05-23 11:05:51 -07001392 'echo +%d > /sys/class/rtc/rtc0/wakealarm' % suspend_time,
J. Richard Barnette9af19632015-09-25 12:18:03 -07001393 'powerd_dbus_suspend --delay=0'])
Gwendal Grignou7a61d2f2014-05-23 11:05:51 -07001394 super(CrosHost, self).suspend(**dargs)
1395
1396
Simran Basiec564392014-08-25 16:48:09 -07001397 def upstart_status(self, service_name):
1398 """Check the status of an upstart init script.
1399
1400 @param service_name: Service to look up.
1401
1402 @returns True if the service is running, False otherwise.
1403 """
1404 return self.run('status %s | grep start/running' %
1405 service_name).stdout.strip() != ''
1406
1407
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001408 def verify_software(self):
Richard Barnetteb2bc13c2013-01-08 17:32:51 -08001409 """Verify working software on a Chrome OS system.
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001410
Richard Barnetteb2bc13c2013-01-08 17:32:51 -08001411 Tests for the following conditions:
1412 1. All conditions tested by the parent version of this
1413 function.
1414 2. Sufficient space in /mnt/stateful_partition.
Fang Deng6b05f5b2013-03-20 13:42:11 -07001415 3. Sufficient space in /mnt/stateful_partition/encrypted.
1416 4. update_engine answers a simple status request over DBus.
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001417
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001418 """
Fang Deng0ca40e22013-08-27 17:47:44 -07001419 super(CrosHost, self).verify_software()
Dan Shib8540a52015-07-16 14:18:23 -07001420 default_kilo_inodes_required = CONFIG.get_config_value(
1421 'SERVER', 'kilo_inodes_required', type=int, default=100)
1422 board = self.get_board().replace(ds_constants.BOARD_PREFIX, '')
1423 kilo_inodes_required = CONFIG.get_config_value(
1424 'SERVER', 'kilo_inodes_required_%s' % board,
1425 type=int, default=default_kilo_inodes_required)
1426 self.check_inodes('/mnt/stateful_partition', kilo_inodes_required)
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001427 self.check_diskspace(
1428 '/mnt/stateful_partition',
Dan Shib8540a52015-07-16 14:18:23 -07001429 CONFIG.get_config_value(
Fang Deng6b05f5b2013-03-20 13:42:11 -07001430 'SERVER', 'gb_diskspace_required', type=float,
1431 default=20.0))
Gaurav Shahe448af82014-06-19 15:18:59 -07001432 encrypted_stateful_path = '/mnt/stateful_partition/encrypted'
1433 # Not all targets build with encrypted stateful support.
1434 if self.path_exists(encrypted_stateful_path):
1435 self.check_diskspace(
1436 encrypted_stateful_path,
Dan Shib8540a52015-07-16 14:18:23 -07001437 CONFIG.get_config_value(
Gaurav Shahe448af82014-06-19 15:18:59 -07001438 'SERVER', 'gb_encrypted_diskspace_required', type=float,
1439 default=0.1))
beepsc87ff602013-07-31 21:53:00 -07001440
Simran Basiec564392014-08-25 16:48:09 -07001441 if not self.upstart_status('system-services'):
Prashanth B5d0a0512014-04-25 12:26:08 -07001442 raise error.AutoservError('Chrome failed to reach login. '
1443 'System services not running.')
1444
beepsc87ff602013-07-31 21:53:00 -07001445 # Factory images don't run update engine,
1446 # goofy controls dbus on these DUTs.
1447 if not self._is_factory_image():
1448 self.run('update_engine_client --status')
Scott Zawalskifbca4a92013-03-04 15:56:42 -05001449 # Makes sure python is present, loads and can use built in functions.
1450 # We have seen cases where importing cPickle fails with undefined
1451 # symbols in cPickle.so.
1452 self.run('python -c "import cPickle"')
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001453
Dan Shi549fb822015-03-24 18:01:11 -07001454 self.verify_cros_version_label()
1455
Tom Wai-Hong Tambea741c2016-01-21 07:20:14 +08001456 self.verify_firmware_status()
1457
Shelley Chena11b9e72016-01-21 15:15:18 -08001458 self.verify_filesystem_write_status()
1459
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001460
J. Richard Barnettea7e7fdc2016-02-12 12:35:36 -08001461 def verify(self):
1462 self._repair_strategy.verify(self)
1463
1464
Fang Deng96667ca2013-08-01 17:46:18 -07001465 def make_ssh_command(self, user='root', port=22, opts='', hosts_file=None,
1466 connect_timeout=None, alive_interval=None):
1467 """Override default make_ssh_command to use options tuned for Chrome OS.
1468
1469 Tuning changes:
1470 - ConnectTimeout=30; maximum of 30 seconds allowed for an SSH
1471 connection failure. Consistency with remote_access.sh.
1472
Samuel Tan2ce155b2015-06-23 18:24:38 -07001473 - ServerAliveInterval=900; which causes SSH to ping connection every
1474 900 seconds. In conjunction with ServerAliveCountMax ensures
1475 that if the connection dies, Autotest will bail out.
Fang Deng96667ca2013-08-01 17:46:18 -07001476 Originally tried 60 secs, but saw frequent job ABORTS where
Samuel Tan2ce155b2015-06-23 18:24:38 -07001477 the test completed successfully. Later increased from 180 seconds to
1478 900 seconds to account for tests where the DUT is suspended for
1479 longer periods of time.
Fang Deng96667ca2013-08-01 17:46:18 -07001480
1481 - ServerAliveCountMax=3; consistency with remote_access.sh.
1482
1483 - ConnectAttempts=4; reduce flakiness in connection errors;
1484 consistency with remote_access.sh.
1485
1486 - UserKnownHostsFile=/dev/null; we don't care about the keys.
1487 Host keys change with every new installation, don't waste
1488 memory/space saving them.
1489
1490 - SSH protocol forced to 2; needed for ServerAliveInterval.
1491
1492 @param user User name to use for the ssh connection.
1493 @param port Port on the target host to use for ssh connection.
1494 @param opts Additional options to the ssh command.
1495 @param hosts_file Ignored.
1496 @param connect_timeout Ignored.
1497 @param alive_interval Ignored.
1498 """
Aviv Keshetc5947fa2013-09-04 14:06:29 -07001499 base_command = ('/usr/bin/ssh -a -x %s %s %s'
1500 ' -o StrictHostKeyChecking=no'
Fang Deng96667ca2013-08-01 17:46:18 -07001501 ' -o UserKnownHostsFile=/dev/null -o BatchMode=yes'
Samuel Tan2ce155b2015-06-23 18:24:38 -07001502 ' -o ConnectTimeout=30 -o ServerAliveInterval=900'
Fang Deng96667ca2013-08-01 17:46:18 -07001503 ' -o ServerAliveCountMax=3 -o ConnectionAttempts=4'
1504 ' -o Protocol=2 -l %s -p %d')
Aviv Keshetc5947fa2013-09-04 14:06:29 -07001505 return base_command % (self._ssh_verbosity_flag, self._ssh_options,
1506 opts, user, port)
Jason Abeleb6f924f2013-11-13 16:01:54 -08001507 def syslog(self, message, tag='autotest'):
1508 """Logs a message to syslog on host.
1509
1510 @param message String message to log into syslog
1511 @param tag String tag prefix for syslog
1512
1513 """
1514 self.run('logger -t "%s" "%s"' % (tag, message))
1515
1516
J. Richard Barnetteb6de7e32013-02-14 13:28:04 -08001517 def _ping_check_status(self, status):
1518 """Ping the host once, and return whether it has a given status.
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001519
J. Richard Barnetteb6de7e32013-02-14 13:28:04 -08001520 @param status Check the ping status against this value.
1521 @return True iff `status` and the result of ping are the same
1522 (i.e. both True or both False).
1523
1524 """
1525 ping_val = utils.ping(self.hostname, tries=1, deadline=1)
1526 return not (status ^ (ping_val == 0))
1527
1528 def _ping_wait_for_status(self, status, timeout):
1529 """Wait for the host to have a given status (UP or DOWN).
1530
1531 Status is checked by polling. Polling will not last longer
1532 than the number of seconds in `timeout`. The polling
1533 interval will be long enough that only approximately
1534 _PING_WAIT_COUNT polling cycles will be executed, subject
1535 to a maximum interval of about one minute.
1536
1537 @param status Waiting will stop immediately if `ping` of the
1538 host returns this status.
1539 @param timeout Poll for at most this many seconds.
1540 @return True iff the host status from `ping` matched the
1541 requested status at the time of return.
1542
1543 """
1544 # _ping_check_status() takes about 1 second, hence the
1545 # "- 1" in the formula below.
1546 poll_interval = min(int(timeout / self._PING_WAIT_COUNT), 60) - 1
1547 end_time = time.time() + timeout
1548 while time.time() <= end_time:
1549 if self._ping_check_status(status):
1550 return True
1551 if poll_interval > 0:
1552 time.sleep(poll_interval)
1553
1554 # The last thing we did was sleep(poll_interval), so it may
1555 # have been too long since the last `ping`. Check one more
1556 # time, just to be sure.
1557 return self._ping_check_status(status)
1558
1559 def ping_wait_up(self, timeout):
1560 """Wait for the host to respond to `ping`.
1561
1562 N.B. This method is not a reliable substitute for
1563 `wait_up()`, because a host that responds to ping will not
1564 necessarily respond to ssh. This method should only be used
1565 if the target DUT can be considered functional even if it
1566 can't be reached via ssh.
1567
1568 @param timeout Minimum time to allow before declaring the
1569 host to be non-responsive.
1570 @return True iff the host answered to ping before the timeout.
1571
1572 """
1573 return self._ping_wait_for_status(self._PING_STATUS_UP, timeout)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001574
Andrew Bresticker678c0c72013-01-22 10:44:09 -08001575 def ping_wait_down(self, timeout):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001576 """Wait until the host no longer responds to `ping`.
1577
J. Richard Barnetteb6de7e32013-02-14 13:28:04 -08001578 This function can be used as a slightly faster version of
1579 `wait_down()`, by avoiding potentially long ssh timeouts.
1580
1581 @param timeout Minimum time to allow for the host to become
1582 non-responsive.
1583 @return True iff the host quit answering ping before the
1584 timeout.
1585
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001586 """
J. Richard Barnetteb6de7e32013-02-14 13:28:04 -08001587 return self._ping_wait_for_status(self._PING_STATUS_DOWN, timeout)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001588
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08001589 def test_wait_for_sleep(self, sleep_timeout=None):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001590 """Wait for the client to enter low-power sleep mode.
1591
1592 The test for "is asleep" can't distinguish a system that is
1593 powered off; to confirm that the unit was asleep, it is
1594 necessary to force resume, and then call
1595 `test_wait_for_resume()`.
1596
1597 This function is expected to be called from a test as part
1598 of a sequence like the following:
1599
1600 ~~~~~~~~
1601 boot_id = host.get_boot_id()
1602 # trigger sleep on the host
1603 host.test_wait_for_sleep()
1604 # trigger resume on the host
1605 host.test_wait_for_resume(boot_id)
1606 ~~~~~~~~
1607
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08001608 @param sleep_timeout time limit in seconds to allow the host sleep.
1609
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001610 @exception TestFail The host did not go to sleep within
1611 the allowed time.
1612 """
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08001613 if sleep_timeout is None:
1614 sleep_timeout = self.SLEEP_TIMEOUT
1615
1616 if not self.ping_wait_down(timeout=sleep_timeout):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001617 raise error.TestFail(
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08001618 'client failed to sleep after %d seconds' % sleep_timeout)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001619
1620
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08001621 def test_wait_for_resume(self, old_boot_id, resume_timeout=None):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001622 """Wait for the client to resume from low-power sleep mode.
1623
1624 The `old_boot_id` parameter should be the value from
1625 `get_boot_id()` obtained prior to entering sleep mode. A
1626 `TestFail` exception is raised if the boot id changes.
1627
1628 See @ref test_wait_for_sleep for more on this function's
1629 usage.
1630
J. Richard Barnette7214e0b2013-02-06 15:20:49 -08001631 @param old_boot_id A boot id value obtained before the
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001632 target host went to sleep.
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08001633 @param resume_timeout time limit in seconds to allow the host up.
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001634
1635 @exception TestFail The host did not respond within the
1636 allowed time.
1637 @exception TestFail The host responded, but the boot id test
1638 indicated a reboot rather than a sleep
1639 cycle.
1640 """
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08001641 if resume_timeout is None:
1642 resume_timeout = self.RESUME_TIMEOUT
1643
1644 if not self.wait_up(timeout=resume_timeout):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001645 raise error.TestFail(
1646 'client failed to resume from sleep after %d seconds' %
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08001647 resume_timeout)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001648 else:
1649 new_boot_id = self.get_boot_id()
1650 if new_boot_id != old_boot_id:
Tom Wai-Hong Tam01792682015-01-06 08:00:46 +08001651 logging.error('client rebooted (old boot %s, new boot %s)',
1652 old_boot_id, new_boot_id)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001653 raise error.TestFail(
Tom Wai-Hong Tam01792682015-01-06 08:00:46 +08001654 'client rebooted, but sleep was expected')
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001655
1656
Tom Wai-Hong Tamfe005c22014-12-03 09:25:44 +08001657 def test_wait_for_shutdown(self, shutdown_timeout=None):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001658 """Wait for the client to shut down.
1659
1660 The test for "has shut down" can't distinguish a system that
1661 is merely asleep; to confirm that the unit was down, it is
1662 necessary to force boot, and then call test_wait_for_boot().
1663
1664 This function is expected to be called from a test as part
1665 of a sequence like the following:
1666
1667 ~~~~~~~~
1668 boot_id = host.get_boot_id()
1669 # trigger shutdown on the host
1670 host.test_wait_for_shutdown()
1671 # trigger boot on the host
1672 host.test_wait_for_boot(boot_id)
1673 ~~~~~~~~
1674
Tom Wai-Hong Tamfe005c22014-12-03 09:25:44 +08001675 @param shutdown_timeout time limit in seconds to allow the host down.
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001676 @exception TestFail The host did not shut down within the
1677 allowed time.
1678 """
Tom Wai-Hong Tamfe005c22014-12-03 09:25:44 +08001679 if shutdown_timeout is None:
1680 shutdown_timeout = self.SHUTDOWN_TIMEOUT
1681
1682 if not self.ping_wait_down(timeout=shutdown_timeout):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001683 raise error.TestFail(
1684 'client failed to shut down after %d seconds' %
Tom Wai-Hong Tamfe005c22014-12-03 09:25:44 +08001685 shutdown_timeout)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001686
1687
1688 def test_wait_for_boot(self, old_boot_id=None):
1689 """Wait for the client to boot from cold power.
1690
1691 The `old_boot_id` parameter should be the value from
1692 `get_boot_id()` obtained prior to shutting down. A
1693 `TestFail` exception is raised if the boot id does not
1694 change. The boot id test is omitted if `old_boot_id` is not
1695 specified.
1696
1697 See @ref test_wait_for_shutdown for more on this function's
1698 usage.
1699
J. Richard Barnette7214e0b2013-02-06 15:20:49 -08001700 @param old_boot_id A boot id value obtained before the
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001701 shut down.
1702
1703 @exception TestFail The host did not respond within the
1704 allowed time.
1705 @exception TestFail The host responded, but the boot id test
1706 indicated that there was no reboot.
1707 """
J. Richard Barnetteeb69d722012-06-18 17:29:44 -07001708 if not self.wait_up(timeout=self.REBOOT_TIMEOUT):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001709 raise error.TestFail(
1710 'client failed to reboot after %d seconds' %
J. Richard Barnetteeb69d722012-06-18 17:29:44 -07001711 self.REBOOT_TIMEOUT)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001712 elif old_boot_id:
1713 if self.get_boot_id() == old_boot_id:
Tom Wai-Hong Tam01792682015-01-06 08:00:46 +08001714 logging.error('client not rebooted (boot %s)',
1715 old_boot_id)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001716 raise error.TestFail(
Tom Wai-Hong Tam01792682015-01-06 08:00:46 +08001717 'client is back up, but did not reboot')
Simran Basid5e5e272012-09-24 15:23:59 -07001718
1719
1720 @staticmethod
1721 def check_for_rpm_support(hostname):
1722 """For a given hostname, return whether or not it is powered by an RPM.
1723
Simran Basi1df55112013-09-06 11:25:09 -07001724 @param hostname: hostname to check for rpm support.
1725
Simran Basid5e5e272012-09-24 15:23:59 -07001726 @return None if this host does not follows the defined naming format
1727 for RPM powered DUT's in the lab. If it does follow the format,
1728 it returns a regular expression MatchObject instead.
1729 """
Fang Dengbaff9082015-01-06 13:46:15 -08001730 return re.match(CrosHost._RPM_HOSTNAME_REGEX, hostname)
Simran Basid5e5e272012-09-24 15:23:59 -07001731
1732
1733 def has_power(self):
1734 """For this host, return whether or not it is powered by an RPM.
1735
1736 @return True if this host is in the CROS lab and follows the defined
1737 naming format.
1738 """
Fang Deng0ca40e22013-08-27 17:47:44 -07001739 return CrosHost.check_for_rpm_support(self.hostname)
Simran Basid5e5e272012-09-24 15:23:59 -07001740
1741
Ismail Noorbasha07fdb612013-02-14 14:13:31 -08001742 def _set_power(self, state, power_method):
1743 """Sets the power to the host via RPM, Servo or manual.
1744
1745 @param state Specifies which power state to set to DUT
1746 @param power_method Specifies which method of power control to
1747 use. By default "RPM" will be used. Valid values
1748 are the strings "RPM", "manual", "servoj10".
1749
1750 """
1751 ACCEPTABLE_STATES = ['ON', 'OFF']
1752
1753 if state.upper() not in ACCEPTABLE_STATES:
1754 raise error.TestError('State must be one of: %s.'
1755 % (ACCEPTABLE_STATES,))
1756
1757 if power_method == self.POWER_CONTROL_SERVO:
1758 logging.info('Setting servo port J10 to %s', state)
1759 self.servo.set('prtctl3_pwren', state.lower())
1760 time.sleep(self._USB_POWER_TIMEOUT)
1761 elif power_method == self.POWER_CONTROL_MANUAL:
1762 logging.info('You have %d seconds to set the AC power to %s.',
1763 self._POWER_CYCLE_TIMEOUT, state)
1764 time.sleep(self._POWER_CYCLE_TIMEOUT)
1765 else:
1766 if not self.has_power():
1767 raise error.TestFail('DUT does not have RPM connected.')
Simran Basi5e6339a2013-03-21 11:34:32 -07001768 afe = frontend_wrappers.RetryingAFE(timeout_min=5, delay_sec=10)
1769 afe.set_host_attribute(self._RPM_OUTLET_CHANGED, True,
1770 hostname=self.hostname)
Simran Basi1df55112013-09-06 11:25:09 -07001771 rpm_client.set_power(self.hostname, state.upper(), timeout_mins=5)
Simran Basid5e5e272012-09-24 15:23:59 -07001772
1773
Ismail Noorbasha07fdb612013-02-14 14:13:31 -08001774 def power_off(self, power_method=POWER_CONTROL_RPM):
1775 """Turn off power to this host via RPM, Servo or manual.
1776
1777 @param power_method Specifies which method of power control to
1778 use. By default "RPM" will be used. Valid values
1779 are the strings "RPM", "manual", "servoj10".
1780
1781 """
1782 self._set_power('OFF', power_method)
Simran Basid5e5e272012-09-24 15:23:59 -07001783
1784
Ismail Noorbasha07fdb612013-02-14 14:13:31 -08001785 def power_on(self, power_method=POWER_CONTROL_RPM):
1786 """Turn on power to this host via RPM, Servo or manual.
1787
1788 @param power_method Specifies which method of power control to
1789 use. By default "RPM" will be used. Valid values
1790 are the strings "RPM", "manual", "servoj10".
1791
1792 """
1793 self._set_power('ON', power_method)
1794
1795
1796 def power_cycle(self, power_method=POWER_CONTROL_RPM):
1797 """Cycle power to this host by turning it OFF, then ON.
1798
1799 @param power_method Specifies which method of power control to
1800 use. By default "RPM" will be used. Valid values
1801 are the strings "RPM", "manual", "servoj10".
1802
1803 """
1804 if power_method in (self.POWER_CONTROL_SERVO,
1805 self.POWER_CONTROL_MANUAL):
1806 self.power_off(power_method=power_method)
1807 time.sleep(self._POWER_CYCLE_TIMEOUT)
1808 self.power_on(power_method=power_method)
1809 else:
1810 rpm_client.set_power(self.hostname, 'CYCLE')
Simran Basic6f1f7a2012-10-16 10:47:46 -07001811
1812
1813 def get_platform(self):
1814 """Determine the correct platform label for this host.
1815
1816 @returns a string representing this host's platform.
1817 """
1818 crossystem = utils.Crossystem(self)
1819 crossystem.init()
1820 # Extract fwid value and use the leading part as the platform id.
1821 # fwid generally follow the format of {platform}.{firmware version}
1822 # Example: Alex.X.YYY.Z or Google_Alex.X.YYY.Z
1823 platform = crossystem.fwid().split('.')[0].lower()
1824 # Newer platforms start with 'Google_' while the older ones do not.
1825 return platform.replace('google_', '')
1826
1827
Hung-ying Tyanb1328032014-04-01 14:18:54 +08001828 def get_architecture(self):
1829 """Determine the correct architecture label for this host.
1830
1831 @returns a string representing this host's architecture.
1832 """
1833 crossystem = utils.Crossystem(self)
1834 crossystem.init()
1835 return crossystem.arch()
1836
1837
Luis Lozano40b7d0d2014-01-17 15:12:06 -08001838 def get_chrome_version(self):
1839 """Gets the Chrome version number and milestone as strings.
1840
1841 Invokes "chrome --version" to get the version number and milestone.
1842
1843 @return A tuple (chrome_ver, milestone) where "chrome_ver" is the
1844 current Chrome version number as a string (in the form "W.X.Y.Z")
1845 and "milestone" is the first component of the version number
1846 (the "W" from "W.X.Y.Z"). If the version number cannot be parsed
1847 in the "W.X.Y.Z" format, the "chrome_ver" will be the full output
1848 of "chrome --version" and the milestone will be the empty string.
1849
1850 """
MK Ryu35d661e2014-09-25 17:44:10 -07001851 version_string = self.run(client_constants.CHROME_VERSION_COMMAND).stdout
Luis Lozano40b7d0d2014-01-17 15:12:06 -08001852 return utils.parse_chrome_version(version_string)
1853
J. Richard Barnetted2af5852016-02-05 15:03:10 -08001854
Kevin Chenga2619dc2016-03-28 11:42:08 -07001855 # TODO(kevcheng): change this to just return the board without the
1856 # 'board:' prefix and fix up all the callers. Also look into removing the
1857 # need for this method.
Simran Basic6f1f7a2012-10-16 10:47:46 -07001858 def get_board(self):
1859 """Determine the correct board label for this host.
1860
1861 @returns a string representing this host's board.
1862 """
1863 release_info = utils.parse_cmd_output('cat /etc/lsb-release',
1864 run_method=self.run)
J. Richard Barnetted2af5852016-02-05 15:03:10 -08001865 return (ds_constants.BOARD_PREFIX +
1866 release_info['CHROMEOS_RELEASE_BOARD'])
Simran Basic6f1f7a2012-10-16 10:47:46 -07001867
1868
Kevin Chenga328da62016-03-31 10:49:04 -07001869
1870 def has_lightsensor(self):
1871 """Determine the correct board label for this host.
1872
1873 @returns the string 'lightsensor' if this host has a lightsensor or
1874 None if it does not.
1875 """
1876 search_cmd = "find -L %s -maxdepth 4 | egrep '%s'" % (
1877 self._LIGHTSENSOR_SEARCH_DIR, '|'.join(self._LIGHTSENSOR_FILES))
1878 try:
1879 # Run the search cmd following the symlinks. Stderr_tee is set to
1880 # None as there can be a symlink loop, but the command will still
1881 # execute correctly with a few messages printed to stderr.
1882 self.run(search_cmd, stdout_tee=None, stderr_tee=None)
1883 return 'lightsensor'
1884 except error.AutoservRunError:
1885 # egrep exited with a return code of 1 meaning none of the possible
1886 # lightsensor files existed.
1887 return None
1888
1889
1890 def has_bluetooth(self):
1891 """Determine the correct board label for this host.
1892
1893 @returns the string 'bluetooth' if this host has bluetooth or
1894 None if it does not.
1895 """
1896 try:
1897 self.run('test -d /sys/class/bluetooth/hci0')
1898 # test exited with a return code of 0.
1899 return 'bluetooth'
1900 except error.AutoservRunError:
1901 # test exited with a return code 1 meaning the directory did not
1902 # exist.
1903 return None
1904
1905
1906 def get_ec(self):
1907 """
1908 Determine the type of EC on this host.
1909
1910 @returns a string representing this host's embedded controller type.
1911 At present, it only returns "ec:cros", for Chrome OS ECs. Other types
1912 of EC (or none) don't return any strings, since no tests depend on
1913 those.
1914 """
1915 cmd = 'mosys ec info'
1916 # The output should look like these, so that the last field should
1917 # match our EC version scheme:
1918 #
1919 # stm | stm32f100 | snow_v1.3.139-375eb9f
1920 # ti | Unknown-10de | peppy_v1.5.114-5d52788
1921 #
1922 # Non-Chrome OS ECs will look like these:
1923 #
1924 # ENE | KB932 | 00BE107A00
1925 # ite | it8518 | 3.08
1926 #
1927 # And some systems don't have ECs at all (Lumpy, for example).
1928 regexp = r'^.*\|\s*(\S+_v\d+\.\d+\.\d+-[0-9a-f]+)\s*$'
1929
1930 ecinfo = self.run(command=cmd, ignore_status=True)
1931 if ecinfo.exit_status == 0:
1932 res = re.search(regexp, ecinfo.stdout)
1933 if res:
1934 logging.info("EC version is %s", res.groups()[0])
1935 return 'ec:cros'
1936 logging.info("%s got: %s", cmd, ecinfo.stdout)
1937 # Has an EC, but it's not a Chrome OS EC
1938 return None
1939 logging.info("%s exited with status %d", cmd, ecinfo.exit_status)
1940 # No EC present
1941 return None
1942
1943
1944 def get_accels(self):
1945 """
1946 Determine the type of accelerometers on this host.
1947
1948 @returns a string representing this host's accelerometer type.
1949 At present, it only returns "accel:cros-ec", for accelerometers
1950 attached to a Chrome OS EC, or none, if no accelerometers.
1951 """
1952 # Check to make sure we have ectool
1953 rv = self.run('which ectool', ignore_status=True)
1954 if rv.exit_status:
1955 logging.info("No ectool cmd found, assuming no EC accelerometers")
1956 return None
1957
1958 # Check that the EC supports the motionsense command
1959 rv = self.run('ectool motionsense', ignore_status=True)
1960 if rv.exit_status:
1961 logging.info("EC does not support motionsense command "
1962 "assuming no EC accelerometers")
1963 return None
1964
1965 # Check that EC motion sensors are active
1966 active = self.run('ectool motionsense active').stdout.split('\n')
1967 if active[0] == "0":
1968 logging.info("Motion sense inactive, assuming no EC accelerometers")
1969 return None
1970
1971 logging.info("EC accelerometers found")
1972 return 'accel:cros-ec'
1973
1974
1975 def has_chameleon(self):
1976 """Determine if a Chameleon connected to this host.
1977
1978 @returns a list containing two strings ('chameleon' and
1979 'chameleon:' + label, e.g. 'chameleon:hdmi') if this host
1980 has a Chameleon or None if it has not.
1981 """
1982 if self._chameleon_host:
1983 return ['chameleon', 'chameleon:' + self.chameleon.get_label()]
1984 else:
1985 return None
1986
1987
1988 def has_loopback_dongle(self):
1989 """Determine if an audio loopback dongle is plugged to this host.
1990
1991 @returns 'audio_loopback_dongle' when there is an audio loopback dongle
1992 plugged to this host.
1993 None when there is no audio loopback dongle
1994 plugged to this host.
1995 """
1996 nodes_info = self.run(command=cras_utils.get_cras_nodes_cmd(),
1997 ignore_status=True).stdout
1998 if (cras_utils.node_type_is_plugged('HEADPHONE', nodes_info) and
1999 cras_utils.node_type_is_plugged('MIC', nodes_info)):
2000 return 'audio_loopback_dongle'
2001 else:
2002 return None
2003
2004
2005 def get_power_supply(self):
2006 """
2007 Determine what type of power supply the host has
2008
2009 @returns a string representing this host's power supply.
2010 'power:battery' when the device has a battery intended for
2011 extended use
2012 'power:AC_primary' when the device has a battery not intended
2013 for extended use (for moving the machine, etc)
2014 'power:AC_only' when the device has no battery at all.
2015 """
2016 psu = self.run(command='mosys psu type', ignore_status=True)
2017 if psu.exit_status:
2018 # The psu command for mosys is not included for all platforms. The
2019 # assumption is that the device will have a battery if the command
2020 # is not found.
2021 return 'power:battery'
2022
2023 psu_str = psu.stdout.strip()
2024 if psu_str == 'unknown':
2025 return None
2026
2027 return 'power:%s' % psu_str
2028
2029
2030 def get_storage(self):
2031 """
2032 Determine the type of boot device for this host.
2033
2034 Determine if the internal device is SCSI or dw_mmc device.
2035 Then check that it is SSD or HDD or eMMC or something else.
2036
2037 @returns a string representing this host's internal device type.
2038 'storage:ssd' when internal device is solid state drive
2039 'storage:hdd' when internal device is hard disk drive
2040 'storage:mmc' when internal device is mmc drive
2041 None When internal device is something else or
2042 when we are unable to determine the type
2043 """
2044 # The output should be /dev/mmcblk* for SD/eMMC or /dev/sd* for scsi
2045 rootdev_cmd = ' '.join(['. /usr/sbin/write_gpt.sh;',
2046 '. /usr/share/misc/chromeos-common.sh;',
2047 'load_base_vars;',
2048 'get_fixed_dst_drive'])
2049 rootdev = self.run(command=rootdev_cmd, ignore_status=True)
2050 if rootdev.exit_status:
2051 logging.info("Fail to run %s", rootdev_cmd)
2052 return None
2053 rootdev_str = rootdev.stdout.strip()
2054
2055 if not rootdev_str:
2056 return None
2057
2058 rootdev_base = os.path.basename(rootdev_str)
2059
2060 mmc_pattern = '/dev/mmcblk[0-9]'
2061 if re.match(mmc_pattern, rootdev_str):
2062 # Use type to determine if the internal device is eMMC or somthing
2063 # else. We can assume that MMC is always an internal device.
2064 type_cmd = 'cat /sys/block/%s/device/type' % rootdev_base
2065 type = self.run(command=type_cmd, ignore_status=True)
2066 if type.exit_status:
2067 logging.info("Fail to run %s", type_cmd)
2068 return None
2069 type_str = type.stdout.strip()
2070
2071 if type_str == 'MMC':
2072 return 'storage:mmc'
2073
2074 scsi_pattern = '/dev/sd[a-z]+'
2075 if re.match(scsi_pattern, rootdev.stdout):
2076 # Read symlink for /sys/block/sd* to determine if the internal
2077 # device is connected via ata or usb.
2078 link_cmd = 'readlink /sys/block/%s' % rootdev_base
2079 link = self.run(command=link_cmd, ignore_status=True)
2080 if link.exit_status:
2081 logging.info("Fail to run %s", link_cmd)
2082 return None
2083 link_str = link.stdout.strip()
2084 if 'usb' in link_str:
2085 return None
2086
2087 # Read rotation to determine if the internal device is ssd or hdd.
2088 rotate_cmd = str('cat /sys/block/%s/queue/rotational'
2089 % rootdev_base)
2090 rotate = self.run(command=rotate_cmd, ignore_status=True)
2091 if rotate.exit_status:
2092 logging.info("Fail to run %s", rotate_cmd)
2093 return None
2094 rotate_str = rotate.stdout.strip()
2095
2096 rotate_dict = {'0':'storage:ssd', '1':'storage:hdd'}
2097 return rotate_dict.get(rotate_str)
2098
2099 # All other internal device / error case will always fall here
2100 return None
2101
2102
2103 def get_servo(self):
2104 """Determine if the host has a servo attached.
2105
2106 If the host has a working servo attached, it should have a servo label.
2107
2108 @return: string 'servo' if the host has servo attached. Otherwise,
2109 returns None.
2110 """
2111 return 'servo' if self._servo_host else None
2112
2113
2114 def get_video_labels(self):
2115 """Run /usr/local/bin/avtest_label_detect to get a list of video labels.
2116
2117 Sample output of avtest_label_detect:
2118 Detected label: hw_video_acc_vp8
2119 Detected label: webcam
2120
2121 @return: A list of labels detected by tool avtest_label_detect.
2122 """
2123 try:
2124 result = self.run('/usr/local/bin/avtest_label_detect').stdout
2125 return re.findall('^Detected label: (\w+)$', result, re.M)
2126 except error.AutoservRunError:
2127 # The tool is not installed.
2128 return []
2129
2130
2131 def is_video_glitch_detection_supported(self):
2132 """ Determine if a board under test is supported for video glitch
2133 detection tests.
2134
2135 @return: 'video_glitch_detection' if board is supported, None otherwise.
2136 """
2137 board = self.get_board().replace(ds_constants.BOARD_PREFIX, '')
2138
2139 if board in video_test_constants.SUPPORTED_BOARDS:
2140 return 'video_glitch_detection'
2141
2142 return None
2143
2144
2145 def get_touch(self):
2146 """
2147 Determine whether board under test has a touchpad or touchscreen.
2148
2149 @return: A list of some combination of 'touchscreen' and 'touchpad',
2150 depending on what is present on the device.
2151
2152 """
2153 labels = []
2154 looking_for = ['touchpad', 'touchscreen']
2155 player = input_playback.InputPlayback()
2156 input_events = self.run('ls /dev/input/event*').stdout.strip().split()
2157 filename = '/tmp/touch_labels'
2158 for event in input_events:
2159 self.run('evtest %s > %s' % (event, filename), timeout=1,
2160 ignore_timeout=True)
2161 properties = self.run('cat %s' % filename).stdout
2162 input_type = player._determine_input_type(properties)
2163 if input_type in looking_for:
2164 labels.append(input_type)
2165 looking_for.remove(input_type)
2166 if len(looking_for) == 0:
2167 break
2168 self.run('rm %s' % filename)
2169
2170 return labels
2171
2172
2173 def has_internal_display(self):
2174 """Determine if the device under test is equipped with an internal
2175 display.
2176
2177 @return: 'internal_display' if one is present; None otherwise.
2178 """
2179 from autotest_lib.client.cros.graphics import graphics_utils
2180 from autotest_lib.client.common_lib import utils as common_utils
2181
2182 def __system_output(cmd):
2183 return self.run(cmd).stdout
2184
2185 def __read_file(remote_path):
2186 return self.run('cat %s' % remote_path).stdout
2187
2188 # Hijack the necessary client functions so that we can take advantage
2189 # of the client lib here.
2190 # FIXME: find a less hacky way than this
2191 original_system_output = utils.system_output
2192 original_read_file = common_utils.read_file
2193 utils.system_output = __system_output
2194 common_utils.read_file = __read_file
2195 try:
2196 return ('internal_display' if graphics_utils.has_internal_display()
2197 else None)
2198 finally:
2199 utils.system_output = original_system_output
2200 common_utils.read_file = original_read_file
2201
2202
2203 def has_lucid_sleep_support(self):
2204 """Determine if the device under test has support for lucid sleep.
2205
2206 @return 'lucidsleep' if this board supports lucid sleep; None otherwise
2207 """
2208 board = self.get_board().replace(ds_constants.BOARD_PREFIX, '')
2209 return 'lucidsleep' if board in LUCID_SLEEP_BOARDS else None
2210
2211
Dan Shi85276d42014-04-08 22:11:45 -07002212 def is_boot_from_usb(self):
2213 """Check if DUT is boot from USB.
2214
2215 @return: True if DUT is boot from usb.
2216 """
2217 device = self.run('rootdev -s -d').stdout.strip()
2218 removable = int(self.run('cat /sys/block/%s/removable' %
2219 os.path.basename(device)).stdout.strip())
2220 return removable == 1
Helen Zhang17dae2b2014-11-11 09:25:52 -08002221
2222
2223 def read_from_meminfo(self, key):
Dan Shi49ca0932014-11-14 11:22:27 -08002224 """Return the memory info from /proc/meminfo
Helen Zhang17dae2b2014-11-11 09:25:52 -08002225
2226 @param key: meminfo requested
2227
2228 @return the memory value as a string
2229
2230 """
Helen Zhang17dae2b2014-11-11 09:25:52 -08002231 meminfo = self.run('grep %s /proc/meminfo' % key).stdout.strip()
2232 logging.debug('%s', meminfo)
2233 return int(re.search(r'\d+', meminfo).group(0))
Rohit Makasana8a4923c2015-08-13 17:04:26 -07002234
2235
2236 def get_board_type(self):
2237 """
2238 Get the DUT's device type from /etc/lsb-release.
Danny Chan471a8d12015-08-18 14:57:41 -07002239 DEVICETYPE can be one of CHROMEBOX, CHROMEBASE, CHROMEBOOK or more.
2240
2241 @return value of DEVICETYPE param from lsb-release.
Rohit Makasana8a4923c2015-08-13 17:04:26 -07002242 """
Danny Chan471a8d12015-08-18 14:57:41 -07002243 device_type = self.run('grep DEVICETYPE /etc/lsb-release',
2244 ignore_status=True).stdout
2245 if device_type:
Kalin Stoyanov524310b2015-08-21 16:24:04 -07002246 return device_type.split('=')[-1].strip()
Danny Chan471a8d12015-08-18 14:57:41 -07002247 return ''
Gilad Arnolda76bef02015-09-29 13:55:15 -07002248
2249
2250 def get_os_type(self):
2251 return 'cros'
Simran Basia5522a32015-10-06 11:01:24 -07002252
2253
2254 def enable_adb_testing(self):
2255 """Mark this host as an adb tester."""
Dan Shia2872172015-10-31 01:16:51 -07002256 self.run('touch %s' % constants.ANDROID_TESTER_FILEFLAG)
Kevin Chenga2619dc2016-03-28 11:42:08 -07002257
2258
2259 def get_labels(self):
2260 return self.labels.get_labels(self)