blob: aee9b804052b293a317d501945a908cb451ef31b [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
Vincent Palatindf2372c2016-10-07 17:03:00 +02008import sys
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07009import time
10
mussa584b4462014-06-20 15:13:28 -070011import common
J. Richard Barnette45e93de2012-04-11 17:24:15 -070012from autotest_lib.client.bin import utils
Dan Shi9cb0eec2014-06-03 09:04:50 -070013from autotest_lib.client.common_lib import autotemp
Richard Barnette0c73ffc2012-11-19 15:21:18 -080014from autotest_lib.client.common_lib import error
15from autotest_lib.client.common_lib import global_config
J. Richard Barnette91137f02016-03-10 16:52:26 -080016from autotest_lib.client.common_lib import hosts
Dan Shi549fb822015-03-24 18:01:11 -070017from autotest_lib.client.common_lib import lsbrelease_utils
J. Richard Barnette45e93de2012-04-11 17:24:15 -070018from autotest_lib.client.common_lib.cros import autoupdater
Richard Barnette03a0c132012-11-05 12:40:35 -080019from autotest_lib.client.common_lib.cros import dev_server
Justin TerAvest3b0c5ba2017-11-07 09:59:11 -070020from autotest_lib.client.common_lib.cros import retry
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
Simran Basi382506b2016-09-13 14:58:15 -070045# In case cros_host is being ran via SSP on an older Moblab version with an
46# older chromite version.
47try:
48 from chromite.lib import metrics
Dan Shi5e2efb72017-02-07 11:40:23 -080049except ImportError:
50 metrics = utils.metrics_mock
51
Simran Basid5e5e272012-09-24 15:23:59 -070052
Dan Shib8540a52015-07-16 14:18:23 -070053CONFIG = global_config.global_config
xixuanec801e32016-08-25 10:20:22 -070054ENABLE_DEVSERVER_TRIGGER_AUTO_UPDATE = CONFIG.get_config_value(
55 'CROS', 'enable_devserver_trigger_auto_update', type=bool,
56 default=False)
Dan Shib8540a52015-07-16 14:18:23 -070057
Dan Shid07ee2e2015-09-24 14:49:25 -070058
beepsc87ff602013-07-31 21:53:00 -070059class FactoryImageCheckerException(error.AutoservError):
60 """Exception raised when an image is a factory image."""
61 pass
62
63
Fang Deng0ca40e22013-08-27 17:47:44 -070064class CrosHost(abstract_ssh.AbstractSSHHost):
J. Richard Barnette45e93de2012-04-11 17:24:15 -070065 """Chromium OS specific subclass of Host."""
66
Simran Basi5ace6f22016-01-06 17:30:44 -080067 VERSION_PREFIX = provision.CROS_VERSION_PREFIX
68
J. Richard Barnette45e93de2012-04-11 17:24:15 -070069 _parser = autoserv_parser.autoserv_parser
Scott Zawalski62bacae2013-03-05 10:40:32 -050070 _AFE = frontend_wrappers.RetryingAFE(timeout_min=5, delay_sec=10)
xixuanec801e32016-08-25 10:20:22 -070071 support_devserver_provision = ENABLE_DEVSERVER_TRIGGER_AUTO_UPDATE
J. Richard Barnette45e93de2012-04-11 17:24:15 -070072
Richard Barnette03a0c132012-11-05 12:40:35 -080073 # Timeout values (in seconds) associated with various Chrome OS
74 # state changes.
J. Richard Barnette134ec2c2012-04-25 12:59:37 -070075 #
Richard Barnette0c73ffc2012-11-19 15:21:18 -080076 # In general, a good rule of thumb is that the timeout can be up
77 # to twice the typical measured value on the slowest platform.
78 # The times here have not necessarily been empirically tested to
79 # meet this criterion.
J. Richard Barnetteeb69d722012-06-18 17:29:44 -070080 #
81 # SLEEP_TIMEOUT: Time to allow for suspend to memory.
Richard Barnette0c73ffc2012-11-19 15:21:18 -080082 # RESUME_TIMEOUT: Time to allow for resume after suspend, plus
83 # time to restart the netwowrk.
J. Richard Barnette84890bd2014-02-21 11:05:47 -080084 # SHUTDOWN_TIMEOUT: Time to allow for shut down.
J. Richard Barnetteeb69d722012-06-18 17:29:44 -070085 # BOOT_TIMEOUT: Time to allow for boot from power off. Among
Richard Barnette0c73ffc2012-11-19 15:21:18 -080086 # other things, this must account for the 30 second dev-mode
J. Richard Barnette417cc792015-10-01 09:56:36 -070087 # screen delay, time to start the network on the DUT, and the
88 # ssh timeout of 120 seconds.
J. Richard Barnetteeb69d722012-06-18 17:29:44 -070089 # USB_BOOT_TIMEOUT: Time to allow for boot from a USB device,
Richard Barnette0c73ffc2012-11-19 15:21:18 -080090 # including the 30 second dev-mode delay and time to start the
J. Richard Barnetted4649c62013-03-06 17:42:27 -080091 # network.
beepsf079cfb2013-09-18 17:49:51 -070092 # INSTALL_TIMEOUT: Time to allow for chromeos-install.
J. Richard Barnette84890bd2014-02-21 11:05:47 -080093 # POWERWASH_BOOT_TIMEOUT: Time to allow for a reboot that
94 # includes powerwash.
J. Richard Barnetteeb69d722012-06-18 17:29:44 -070095
96 SLEEP_TIMEOUT = 2
J. Richard Barnetted4649c62013-03-06 17:42:27 -080097 RESUME_TIMEOUT = 10
Tom Wai-Hong Tamfe005c22014-12-03 09:25:44 +080098 SHUTDOWN_TIMEOUT = 10
J. Richard Barnette417cc792015-10-01 09:56:36 -070099 BOOT_TIMEOUT = 150
J. Richard Barnette5bab5f52015-08-03 13:14:38 -0700100 USB_BOOT_TIMEOUT = 300
J. Richard Barnette7817b052014-08-28 09:47:29 -0700101 INSTALL_TIMEOUT = 480
Dan Shi2c88eed2013-11-12 10:18:38 -0800102 POWERWASH_BOOT_TIMEOUT = 60
Chris Sosab76e0ee2013-05-22 16:55:41 -0700103
Dan Shica503482015-03-30 17:23:25 -0700104 # Minimum OS version that supports server side packaging. Older builds may
105 # not have server side package built or with Autotest code change to support
106 # server-side packaging.
Dan Shib8540a52015-07-16 14:18:23 -0700107 MIN_VERSION_SUPPORT_SSP = CONFIG.get_config_value(
Dan Shiced09e42015-04-17 16:09:34 -0700108 'AUTOSERV', 'min_version_support_ssp', type=int)
Dan Shica503482015-03-30 17:23:25 -0700109
J. Richard Barnette84890bd2014-02-21 11:05:47 -0800110 # REBOOT_TIMEOUT: How long to wait for a reboot.
111 #
Chris Sosab76e0ee2013-05-22 16:55:41 -0700112 # We have a long timeout to ensure we don't flakily fail due to other
113 # issues. Shorter timeouts are vetted in platform_RebootAfterUpdate.
Simran Basi1160e2c2013-10-04 16:00:24 -0700114 # TODO(sbasi - crbug.com/276094) Restore to 5 mins once the 'host did not
115 # return from reboot' bug is solved.
116 REBOOT_TIMEOUT = 480
Chris Sosab76e0ee2013-05-22 16:55:41 -0700117
Ismail Noorbasha07fdb612013-02-14 14:13:31 -0800118 # _USB_POWER_TIMEOUT: Time to allow for USB to power toggle ON and OFF.
119 # _POWER_CYCLE_TIMEOUT: Time to allow for manual power cycle.
120 _USB_POWER_TIMEOUT = 5
121 _POWER_CYCLE_TIMEOUT = 10
122
Richard Barnette82c35912012-11-20 10:09:10 -0800123 _LAB_MACHINE_FILE = '/mnt/stateful_partition/.labmachine'
Fang Dengdeba14f2014-11-14 11:54:09 -0800124 _RPM_HOSTNAME_REGEX = ('chromeos(\d+)(-row(\d+))?-rack(\d+[a-z]*)'
125 '-host(\d+)')
Kevin Chenga328da62016-03-31 10:49:04 -0700126 _LIGHTSENSOR_FILES = [ "in_illuminance0_input",
127 "in_illuminance_input",
128 "in_illuminance0_raw",
129 "in_illuminance_raw",
130 "illuminance0_input"]
131 _LIGHTSENSOR_SEARCH_DIR = '/sys/bus/iio/devices'
J. Richard Barnette134ec2c2012-04-25 12:59:37 -0700132
J. Richard Barnetteb6de7e32013-02-14 13:28:04 -0800133 # Constants used in ping_wait_up() and ping_wait_down().
134 #
135 # _PING_WAIT_COUNT is the approximate number of polling
136 # cycles to use when waiting for a host state change.
137 #
138 # _PING_STATUS_DOWN and _PING_STATUS_UP are names used
139 # for arguments to the internal _ping_wait_for_status()
140 # method.
141 _PING_WAIT_COUNT = 40
142 _PING_STATUS_DOWN = False
143 _PING_STATUS_UP = True
144
Ismail Noorbasha07fdb612013-02-14 14:13:31 -0800145 # Allowed values for the power_method argument.
146
147 # POWER_CONTROL_RPM: Passed as default arg for power_off/on/cycle() methods.
148 # POWER_CONTROL_SERVO: Used in set_power() and power_cycle() methods.
149 # POWER_CONTROL_MANUAL: Used in set_power() and power_cycle() methods.
150 POWER_CONTROL_RPM = 'RPM'
151 POWER_CONTROL_SERVO = 'servoj10'
152 POWER_CONTROL_MANUAL = 'manual'
153
154 POWER_CONTROL_VALID_ARGS = (POWER_CONTROL_RPM,
155 POWER_CONTROL_SERVO,
156 POWER_CONTROL_MANUAL)
Richard Barnette0c73ffc2012-11-19 15:21:18 -0800157
Simran Basi5e6339a2013-03-21 11:34:32 -0700158 _RPM_OUTLET_CHANGED = 'outlet_changed'
159
Dan Shi9cb0eec2014-06-03 09:04:50 -0700160 # URL pattern to download firmware image.
Dan Shib8540a52015-07-16 14:18:23 -0700161 _FW_IMAGE_URL_PATTERN = CONFIG.get_config_value(
Dan Shi9cb0eec2014-06-03 09:04:50 -0700162 'CROS', 'firmware_url_pattern', type=str)
beeps687243d2013-07-18 15:29:27 -0700163
J. Richard Barnette91137f02016-03-10 16:52:26 -0800164
165 # A flag file to indicate provision failures. The file is created
166 # at the start of any AU procedure (see `machine_install()`). The
167 # file's location in stateful means that on successul update it will
168 # be removed. Thus, if this file exists, it indicates that we've
169 # tried and failed in a previous attempt to update.
170 PROVISION_FAILED = '/var/tmp/provision_failed'
MK Ryu35d661e2014-09-25 17:44:10 -0700171
MK Ryu35d661e2014-09-25 17:44:10 -0700172
J. Richard Barnette964fba02012-10-24 17:34:29 -0700173 @staticmethod
beeps46dadc92013-11-07 14:07:10 -0800174 def check_host(host, timeout=10):
175 """
176 Check if the given host is a chrome-os host.
177
178 @param host: An ssh host representing a device.
179 @param timeout: The timeout for the run command.
180
181 @return: True if the host device is chromeos.
182
beeps46dadc92013-11-07 14:07:10 -0800183 """
184 try:
Allen Liad719c12017-06-27 23:48:04 +0000185 result = host.run(
Simran Basi933c8af2015-04-29 14:05:07 -0700186 'grep -q CHROMEOS /etc/lsb-release && '
187 '! test -f /mnt/stateful_partition/.android_tester && '
188 '! grep -q moblab /etc/lsb-release',
189 ignore_status=True, timeout=timeout)
Laurence Goodby468de252017-06-08 17:22:53 -0700190 if result.exit_status == 0:
Allen Liad719c12017-06-27 23:48:04 +0000191 lsb_release_content = host.run(
Laurence Goodby468de252017-06-08 17:22:53 -0700192 'grep CHROMEOS_RELEASE_BOARD /etc/lsb-release',
193 timeout=timeout).stdout
194 return not lsbrelease_utils.is_jetstream(
195 lsb_release_content=lsb_release_content)
beeps46dadc92013-11-07 14:07:10 -0800196 except (error.AutoservRunError, error.AutoservSSHTimeout):
197 return False
Laurence Goodby468de252017-06-08 17:22:53 -0700198
199 return False
beeps46dadc92013-11-07 14:07:10 -0800200
201
202 @staticmethod
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +0800203 def get_chameleon_arguments(args_dict):
204 """Extract chameleon options from `args_dict` and return the result.
205
206 Recommended usage:
207 ~~~~~~~~
208 args_dict = utils.args_to_dict(args)
209 chameleon_args = hosts.CrosHost.get_chameleon_arguments(args_dict)
210 host = hosts.create_host(machine, chameleon_args=chameleon_args)
211 ~~~~~~~~
212
213 @param args_dict Dictionary from which to extract the chameleon
214 arguments.
215 """
Allen Li083866b2016-08-18 10:07:10 -0700216 return {key: args_dict[key]
217 for key in ('chameleon_host', 'chameleon_port')
218 if key in args_dict}
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +0800219
220
221 @staticmethod
Scottfe06ed82015-11-05 17:15:01 -0800222 def get_plankton_arguments(args_dict):
223 """Extract chameleon options from `args_dict` and return the result.
224
225 Recommended usage:
226 ~~~~~~~~
227 args_dict = utils.args_to_dict(args)
Richard Barnettee519dcd2016-08-15 17:37:17 -0700228 plankton_args = hosts.CrosHost.get_plankton_arguments(args_dict)
229 host = hosts.create_host(machine, plankton_args=plankton_args)
Scottfe06ed82015-11-05 17:15:01 -0800230 ~~~~~~~~
231
232 @param args_dict Dictionary from which to extract the plankton
233 arguments.
234 """
Allen Li083866b2016-08-18 10:07:10 -0700235 return {key: args_dict[key]
236 for key in ('plankton_host', 'plankton_port')
237 if key in args_dict}
Scottfe06ed82015-11-05 17:15:01 -0800238
239
240 @staticmethod
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +0800241 def get_servo_arguments(args_dict):
242 """Extract servo options from `args_dict` and return the result.
J. Richard Barnette7214e0b2013-02-06 15:20:49 -0800243
244 Recommended usage:
245 ~~~~~~~~
246 args_dict = utils.args_to_dict(args)
Fang Deng0ca40e22013-08-27 17:47:44 -0700247 servo_args = hosts.CrosHost.get_servo_arguments(args_dict)
J. Richard Barnette7214e0b2013-02-06 15:20:49 -0800248 host = hosts.create_host(machine, servo_args=servo_args)
249 ~~~~~~~~
250
251 @param args_dict Dictionary from which to extract the servo
252 arguments.
253 """
Richard Barnettee519dcd2016-08-15 17:37:17 -0700254 servo_attrs = (servo_host.SERVO_HOST_ATTR,
255 servo_host.SERVO_PORT_ATTR,
256 servo_host.SERVO_BOARD_ATTR)
Armando Miraglia2ac802e2017-08-15 14:54:47 +0200257 servo_args = {key: args_dict[key]
258 for key in servo_attrs
259 if key in args_dict}
260 return (
261 None
262 if servo_host.SERVO_HOST_ATTR in servo_args
263 and not servo_args[servo_host.SERVO_HOST_ATTR]
264 else servo_args)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -0700265
J. Richard Barnette964fba02012-10-24 17:34:29 -0700266
J. Richard Barnette91137f02016-03-10 16:52:26 -0800267 def _initialize(self, hostname, chameleon_args=None, servo_args=None,
268 plankton_args=None, try_lab_servo=False,
Richard Barnette9a26ad62016-06-10 12:03:08 -0700269 try_servo_repair=False,
J. Richard Barnette91137f02016-03-10 16:52:26 -0800270 ssh_verbosity_flag='', ssh_options='',
Fang Dengd1c2b732013-08-20 12:59:46 -0700271 *args, **dargs):
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +0800272 """Initialize superclasses, |self.chameleon|, and |self.servo|.
J. Richard Barnette67ccb872012-04-19 16:34:56 -0700273
Fang Denge545abb2014-12-30 18:43:47 -0800274 This method will attempt to create the test-assistant object
275 (chameleon/servo) when it is needed by the test. Check
276 the docstring of chameleon_host.create_chameleon_host and
277 servo_host.create_servo_host for how this is determined.
Fang Deng5d518f42013-08-02 14:04:32 -0700278
Fang Denge545abb2014-12-30 18:43:47 -0800279 @param hostname: Hostname of the dut.
280 @param chameleon_args: A dictionary that contains args for creating
281 a ChameleonHost. See chameleon_host for details.
282 @param servo_args: A dictionary that contains args for creating
283 a ServoHost object. See servo_host for details.
Richard Barnette9a26ad62016-06-10 12:03:08 -0700284 @param try_lab_servo: When true, indicates that an attempt should
285 be made to create a ServoHost for a DUT in
286 the test lab, even if not required by
287 `servo_args`. See servo_host for details.
288 @param try_servo_repair: If a servo host is created, check it
289 with `repair()` rather than `verify()`.
Fang Denge545abb2014-12-30 18:43:47 -0800290 See servo_host for details.
291 @param ssh_verbosity_flag: String, to pass to the ssh command to control
292 verbosity.
293 @param ssh_options: String, other ssh options to pass to the ssh
294 command.
J. Richard Barnette67ccb872012-04-19 16:34:56 -0700295 """
Fang Deng0ca40e22013-08-27 17:47:44 -0700296 super(CrosHost, self)._initialize(hostname=hostname,
J. Richard Barnette45e93de2012-04-11 17:24:15 -0700297 *args, **dargs)
J. Richard Barnette91137f02016-03-10 16:52:26 -0800298 self._repair_strategy = cros_repair.create_cros_repair_strategy()
Kevin Chenga2619dc2016-03-28 11:42:08 -0700299 self.labels = base_label.LabelRetriever(cros_label.CROS_LABELS)
J. Richard Barnettef0859852012-08-20 14:55:50 -0700300 # self.env is a dictionary of environment variable settings
301 # to be exported for commands run on the host.
302 # LIBC_FATAL_STDERR_ can be useful for diagnosing certain
303 # errors that might happen.
304 self.env['LIBC_FATAL_STDERR_'] = '1'
Fang Dengd1c2b732013-08-20 12:59:46 -0700305 self._ssh_verbosity_flag = ssh_verbosity_flag
Aviv Keshetc5947fa2013-09-04 14:06:29 -0700306 self._ssh_options = ssh_options
Fang Deng5d518f42013-08-02 14:04:32 -0700307 # TODO(fdeng): We need to simplify the
308 # process of servo and servo_host initialization.
309 # crbug.com/298432
Moja Hsu57d93a12017-01-13 17:57:52 +0800310 self._servo_host = servo_host.create_servo_host(
Richard Barnetteea3e4602016-06-10 12:36:41 -0700311 dut=self, servo_args=servo_args,
Richard Barnette9a26ad62016-06-10 12:03:08 -0700312 try_lab_servo=try_lab_servo,
313 try_servo_repair=try_servo_repair)
Richard Barnettee519dcd2016-08-15 17:37:17 -0700314 if self._servo_host is not None:
315 self.servo = self._servo_host.get_servo()
316 else:
317 self.servo = None
318
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +0800319 # TODO(waihong): Do the simplication on Chameleon too.
Tom Wai-Hong Tam3d6790d2014-04-14 16:15:47 +0800320 self._chameleon_host = chameleon_host.create_chameleon_host(
321 dut=self.hostname, chameleon_args=chameleon_args)
Scottfe06ed82015-11-05 17:15:01 -0800322 # Add plankton host if plankton args were added on command line
323 self._plankton_host = plankton_host.create_plankton_host(plankton_args)
Tom Wai-Hong Tam3d6790d2014-04-14 16:15:47 +0800324
Tom Wai-Hong Tam3d6790d2014-04-14 16:15:47 +0800325 if self._chameleon_host:
Tom Wai-Hong Tameaee3402014-01-22 08:52:10 +0800326 self.chameleon = self._chameleon_host.create_chameleon_board()
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +0800327 else:
Tom Wai-Hong Tam3d6790d2014-04-14 16:15:47 +0800328 self.chameleon = None
Fang Deng5d518f42013-08-02 14:04:32 -0700329
Scottfe06ed82015-11-05 17:15:01 -0800330 if self._plankton_host:
331 self.plankton_servo = self._plankton_host.get_servo()
332 logging.info('plankton_servo: %r', self.plankton_servo)
333 # Create the plankton object used to access the ec uart
Scott07a848f2016-01-12 15:04:52 -0800334 self.plankton = plankton.Plankton(self.plankton_servo,
335 self._plankton_host.get_servod_server_proxy())
Scottfe06ed82015-11-05 17:15:01 -0800336 else:
Scott07a848f2016-01-12 15:04:52 -0800337 self.plankton = None
Scottfe06ed82015-11-05 17:15:01 -0800338
Fang Deng5d518f42013-08-02 14:04:32 -0700339
Richard Barnette260cbd02016-10-06 12:23:28 -0700340 def _get_cros_repair_image_name(self):
Prathmesh Prabhu075fc922017-02-13 11:50:25 -0800341 info = self.host_info_store.get()
342 if info.board is None:
343 raise error.AutoservError('Cannot obtain repair image name. '
344 'No board label value found')
345
346 return afe_utils.get_stable_cros_image_name(info.board)
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500347
348
Rohit Makasanadf0a3a32017-06-30 13:55:18 -0700349 def host_version_prefix(self, image):
350 """Return version label prefix.
351
352 In case the CrOS provisioning version is something other than the
353 standard CrOS version e.g. CrOS TH version, this function will
354 find the prefix from provision.py.
355
356 @param image: The image name to find its version prefix.
357 @returns: A prefix string for the image type.
358 """
359 return provision.get_version_label_prefix(image)
360
361
beepsdae65fd2013-07-26 16:24:41 -0700362 def verify_job_repo_url(self, tag=''):
beepscb6f1e22013-06-28 19:14:10 -0700363 """
364 Make sure job_repo_url of this host is valid.
365
joychen03eaad92013-06-26 09:55:21 -0700366 Eg: The job_repo_url "http://lmn.cd.ab.xyx:8080/static/\
beepscb6f1e22013-06-28 19:14:10 -0700367 lumpy-release/R29-4279.0.0/autotest/packages" claims to have the
368 autotest package for lumpy-release/R29-4279.0.0. If this isn't the case,
369 download and extract it. If the devserver embedded in the url is
370 unresponsive, update the job_repo_url of the host after staging it on
371 another devserver.
372
373 @param job_repo_url: A url pointing to the devserver where the autotest
374 package for this build should be staged.
beepsdae65fd2013-07-26 16:24:41 -0700375 @param tag: The tag from the server job, in the format
376 <job_id>-<user>/<hostname>, or <hostless> for a server job.
beepscb6f1e22013-06-28 19:14:10 -0700377
378 @raises DevServerException: If we could not resolve a devserver.
379 @raises AutoservError: If we're unable to save the new job_repo_url as
380 a result of choosing a new devserver because the old one failed to
381 respond to a health check.
beeps0c865032013-07-30 11:37:06 -0700382 @raises urllib2.URLError: If the devserver embedded in job_repo_url
383 doesn't respond within the timeout.
beepscb6f1e22013-06-28 19:14:10 -0700384 """
Prathmesh Prabhu368abdf2017-02-14 11:23:47 -0800385 info = self.host_info_store.get()
386 job_repo_url = info.attributes.get(ds_constants.JOB_REPO_URL, '')
beepscb6f1e22013-06-28 19:14:10 -0700387 if not job_repo_url:
388 logging.warning('No job repo url set on host %s', self.hostname)
389 return
390
391 logging.info('Verifying job repo url %s', job_repo_url)
392 devserver_url, image_name = tools.get_devserver_build_from_package_url(
393 job_repo_url)
394
beeps0c865032013-07-30 11:37:06 -0700395 ds = dev_server.ImageServer(devserver_url)
beepscb6f1e22013-06-28 19:14:10 -0700396
397 logging.info('Staging autotest artifacts for %s on devserver %s',
398 image_name, ds.url())
beeps687243d2013-07-18 15:29:27 -0700399
400 start_time = time.time()
Simran Basi25e7a922014-10-31 11:56:10 -0700401 ds.stage_artifacts(image_name, ['autotest_packages'])
beeps687243d2013-07-18 15:29:27 -0700402 stage_time = time.time() - start_time
403
404 # Record how much of the verification time comes from a devserver
405 # restage. If we're doing things right we should not see multiple
406 # devservers for a given board/build/branch path.
407 try:
J. Richard Barnette3cbd76b2013-11-27 12:11:25 -0800408 board, build_type, branch = server_utils.ParseBuildName(
beeps687243d2013-07-18 15:29:27 -0700409 image_name)[:3]
J. Richard Barnette3cbd76b2013-11-27 12:11:25 -0800410 except server_utils.ParseBuildNameException:
beeps687243d2013-07-18 15:29:27 -0700411 pass
412 else:
beeps0c865032013-07-30 11:37:06 -0700413 devserver = devserver_url[
Chris Sosa65425082013-10-16 13:26:22 -0700414 devserver_url.find('/') + 2:devserver_url.rfind(':')]
beeps687243d2013-07-18 15:29:27 -0700415 stats_key = {
416 'board': board,
417 'build_type': build_type,
418 'branch': branch,
beeps0c865032013-07-30 11:37:06 -0700419 'devserver': devserver.replace('.', '_'),
beeps687243d2013-07-18 15:29:27 -0700420 }
Dan Shi5e2efb72017-02-07 11:40:23 -0800421
422 monarch_fields = {
423 'board': board,
424 'build_type': build_type,
425 # TODO(akeshet): To be consistent with most other metrics,
426 # consider changing the following field to be named
427 # 'milestone'.
428 'branch': branch,
429 'dev_server': devserver,
430 }
431 metrics.Counter(
432 'chromeos/autotest/provision/verify_url'
433 ).increment(fields=monarch_fields)
434 metrics.SecondsDistribution(
435 'chromeos/autotest/provision/verify_url_duration'
436 ).add(stage_time, fields=monarch_fields)
437
438
Dan Shicf4d2032015-03-12 15:04:21 -0700439 def stage_server_side_package(self, image=None):
440 """Stage autotest server-side package on devserver.
441
442 @param image: Full path of an OS image to install or a build name.
443
444 @return: A url to the autotest server-side package.
Dan Shi14de7622016-08-22 11:09:06 -0700445
446 @raise: error.AutoservError if fail to locate the build to test with, or
447 fail to stage server-side package.
Dan Shicf4d2032015-03-12 15:04:21 -0700448 """
Dan Shid37736b2016-07-06 15:10:29 -0700449 # If enable_drone_in_restricted_subnet is False, do not set hostname
450 # in devserver.resolve call, so a devserver in non-restricted subnet
451 # is picked to stage autotest server package for drone to download.
452 hostname = self.hostname
453 if not server_utils.ENABLE_DRONE_IN_RESTRICTED_SUBNET:
454 hostname = None
Dan Shicf4d2032015-03-12 15:04:21 -0700455 if image:
456 image_name = tools.get_build_from_image(image)
457 if not image_name:
458 raise error.AutoservError(
459 'Failed to parse build name from %s' % image)
Dan Shid37736b2016-07-06 15:10:29 -0700460 ds = dev_server.ImageServer.resolve(image_name, hostname)
Dan Shicf4d2032015-03-12 15:04:21 -0700461 else:
Prathmesh Prabhu9235e4c2017-03-28 13:16:06 -0700462 info = self.host_info_store.get()
Prathmesh Prabhu368abdf2017-02-14 11:23:47 -0800463 job_repo_url = info.attributes.get(ds_constants.JOB_REPO_URL, '')
Dan Shicf4d2032015-03-12 15:04:21 -0700464 if job_repo_url:
465 devserver_url, image_name = (
466 tools.get_devserver_build_from_package_url(job_repo_url))
Dan Shid37736b2016-07-06 15:10:29 -0700467 # If enable_drone_in_restricted_subnet is True, use the
468 # existing devserver. Otherwise, resolve a new one in
469 # non-restricted subnet.
470 if server_utils.ENABLE_DRONE_IN_RESTRICTED_SUBNET:
471 ds = dev_server.ImageServer(devserver_url)
472 else:
473 ds = dev_server.ImageServer.resolve(image_name)
Prathmesh Prabhub6cea612017-02-09 15:41:19 -0800474 elif info.build is not None:
475 ds = dev_server.ImageServer.resolve(info.build, hostname)
Prathmesh Prabhu0c1dd4d2017-06-07 13:01:53 -0700476 image_name = info.build
Dan Shicf4d2032015-03-12 15:04:21 -0700477 else:
Prathmesh Prabhub6cea612017-02-09 15:41:19 -0800478 raise error.AutoservError(
479 'Failed to stage server-side package. The host has '
480 'no job_report_url attribute or version label.')
Dan Shica503482015-03-30 17:23:25 -0700481
482 # Get the OS version of the build, for any build older than
483 # MIN_VERSION_SUPPORT_SSP, server side packaging is not supported.
484 match = re.match('.*/R\d+-(\d+)\.', image_name)
485 if match and int(match.group(1)) < self.MIN_VERSION_SUPPORT_SSP:
Dan Shi14de7622016-08-22 11:09:06 -0700486 raise error.AutoservError(
487 'Build %s is older than %s. Server side packaging is '
488 'disabled.' % (image_name, self.MIN_VERSION_SUPPORT_SSP))
Dan Shica503482015-03-30 17:23:25 -0700489
Dan Shicf4d2032015-03-12 15:04:21 -0700490 ds.stage_artifacts(image_name, ['autotest_server_package'])
491 return '%s/static/%s/%s' % (ds.url(), image_name,
492 'autotest_server_package.tar.bz2')
493
494
Dan Shi0f466e82013-02-22 15:44:58 -0800495 def _try_stateful_update(self, update_url, force_update, updater):
496 """Try to use stateful update to initialize DUT.
497
498 When DUT is already running the same version that machine_install
499 tries to install, stateful update is a much faster way to clean up
500 the DUT for testing, compared to a full reimage. It is implemeted
501 by calling autoupdater.run_update, but skipping updating root, as
502 updating the kernel is time consuming and not necessary.
503
504 @param update_url: url of the image.
505 @param force_update: Set to True to update the image even if the DUT
506 is running the same version.
507 @param updater: ChromiumOSUpdater instance used to update the DUT.
508 @returns: True if the DUT was updated with stateful update.
509
510 """
Laurence Goodby778c9a42017-05-24 19:24:07 -0700511 self.prepare_for_update()
Dan Shi10b98482016-02-02 14:38:50 -0800512
J. Richard Barnette3f731032014-04-07 17:42:59 -0700513 # TODO(jrbarnette): Yes, I hate this re.match() test case.
514 # It's better than the alternative: see crbug.com/360944.
515 image_name = autoupdater.url_to_image_name(update_url)
516 release_pattern = r'^.*-release/R[0-9]+-[0-9]+\.[0-9]+\.0$'
517 if not re.match(release_pattern, image_name):
518 return False
Dan Shi0f466e82013-02-22 15:44:58 -0800519 if not updater.check_version():
520 return False
521 if not force_update:
522 logging.info('Canceling stateful update because the new and '
523 'old versions are the same.')
524 return False
525 # Following folders should be rebuilt after stateful update.
526 # A test file is used to confirm each folder gets rebuilt after
527 # the stateful update.
528 folders_to_check = ['/var', '/home', '/mnt/stateful_partition']
529 test_file = '.test_file_to_be_deleted'
Paul Hobbs38bcb702017-07-25 00:50:05 -0700530 paths = [os.path.join(folder, test_file) for folder in folders_to_check]
531 self.run('touch %s' % ' '.join(paths))
Dan Shi0f466e82013-02-22 15:44:58 -0800532
Chris Sosae92399e2015-04-24 11:32:59 -0700533 updater.run_update(update_root=False)
Dan Shi0f466e82013-02-22 15:44:58 -0800534
535 # Reboot to complete stateful update.
Chris Sosab76e0ee2013-05-22 16:55:41 -0700536 self.reboot(timeout=self.REBOOT_TIMEOUT, wait=True)
xixuan908b0752016-12-08 18:54:23 -0800537
538 # After stateful update and a reboot, all of the test_files shouldn't
539 # exist any more. Otherwise the stateful update is failed.
540 return not any(
541 self.path_exists(os.path.join(folder, test_file))
542 for folder in folders_to_check)
Dan Shi0f466e82013-02-22 15:44:58 -0800543
544
J. Richard Barnette7275b612013-06-04 18:13:11 -0700545 def _post_update_processing(self, updater, expected_kernel=None):
Dan Shi0f466e82013-02-22 15:44:58 -0800546 """After the DUT is updated, confirm machine_install succeeded.
547
548 @param updater: ChromiumOSUpdater instance used to update the DUT.
J. Richard Barnette7275b612013-06-04 18:13:11 -0700549 @param expected_kernel: kernel expected to be active after reboot,
550 or `None` to skip rollback checking.
Dan Shi0f466e82013-02-22 15:44:58 -0800551
552 """
J. Richard Barnette7275b612013-06-04 18:13:11 -0700553 # Touch the lab machine file to leave a marker that
554 # distinguishes this image from other test images.
555 # Afterwards, we must re-run the autoreboot script because
556 # it depends on the _LAB_MACHINE_FILE.
J. Richard Barnette71cc1862015-12-02 10:32:38 -0800557 autoreboot_cmd = ('FILE="%s" ; [ -f "$FILE" ] || '
558 '( touch "$FILE" ; start autoreboot )')
559 self.run(autoreboot_cmd % self._LAB_MACHINE_FILE)
Chris Sosa65425082013-10-16 13:26:22 -0700560 updater.verify_boot_expectations(
561 expected_kernel, rollback_message=
Gilad Arnoldc26ae1f2015-10-22 16:09:41 -0700562 'Build %s failed to boot on %s; system rolled back to previous '
Chris Sosa65425082013-10-16 13:26:22 -0700563 'build' % (updater.update_version, self.hostname))
J. Richard Barnette7275b612013-06-04 18:13:11 -0700564 # Check that we've got the build we meant to install.
565 if not updater.check_version_to_confirm_install():
566 raise autoupdater.ChromiumOSError(
567 'Failed to update %s to build %s; found build '
568 '%s instead' % (self.hostname,
Chris Sosa65425082013-10-16 13:26:22 -0700569 updater.update_version,
Dan Shi0942b1d2015-03-31 11:07:00 -0700570 self.get_release_version()))
Dan Shi0f466e82013-02-22 15:44:58 -0800571
Chris Sosae92399e2015-04-24 11:32:59 -0700572 logging.debug('Cleaning up old autotest directories.')
573 try:
574 installed_autodir = autotest.Autotest.get_installed_autodir(self)
575 self.run('rm -rf ' + installed_autodir)
576 except autotest.AutodirNotFoundError:
577 logging.debug('No autotest installed directory found.')
578
Dan Shi0f466e82013-02-22 15:44:58 -0800579
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700580 def _stage_image_for_update(self, image_name=None):
Chris Sosae92399e2015-04-24 11:32:59 -0700581 """Stage a build on a devserver and return the update_url and devserver.
Scott Zawalskieadbf702013-03-14 09:23:06 -0400582
583 @param image_name: a name like lumpy-release/R27-3837.0.0
Chris Sosae92399e2015-04-24 11:32:59 -0700584 @returns a tuple with an update URL like:
Scott Zawalskieadbf702013-03-14 09:23:06 -0400585 http://172.22.50.205:8082/update/lumpy-release/R27-3837.0.0
Chris Sosae92399e2015-04-24 11:32:59 -0700586 and the devserver instance.
Scott Zawalskieadbf702013-03-14 09:23:06 -0400587 """
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700588 if not image_name:
Richard Barnette260cbd02016-10-06 12:23:28 -0700589 image_name = self._get_cros_repair_image_name()
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700590 logging.info('Staging build for AU: %s', image_name)
Dan Shi216389c2015-12-22 11:03:06 -0800591 devserver = dev_server.ImageServer.resolve(image_name, self.hostname)
Scott Zawalskieadbf702013-03-14 09:23:06 -0400592 devserver.trigger_download(image_name, synchronous=False)
Chris Sosae92399e2015-04-24 11:32:59 -0700593 return (tools.image_url_pattern() % (devserver.url(), image_name),
594 devserver)
Scott Zawalskieadbf702013-03-14 09:23:06 -0400595
596
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700597 def stage_image_for_servo(self, image_name=None):
598 """Stage a build on a devserver and return the update_url.
599
600 @param image_name: a name like lumpy-release/R27-3837.0.0
601 @returns an update URL like:
602 http://172.22.50.205:8082/update/lumpy-release/R27-3837.0.0
603 """
604 if not image_name:
Richard Barnette260cbd02016-10-06 12:23:28 -0700605 image_name = self._get_cros_repair_image_name()
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700606 logging.info('Staging build for servo install: %s', image_name)
Dan Shi216389c2015-12-22 11:03:06 -0800607 devserver = dev_server.ImageServer.resolve(image_name, self.hostname)
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700608 devserver.stage_artifacts(image_name, ['test_image'])
609 return devserver.get_test_image_url(image_name)
610
611
beepse539be02013-07-31 21:57:39 -0700612 def stage_factory_image_for_servo(self, image_name):
613 """Stage a build on a devserver and return the update_url.
614
615 @param image_name: a name like <baord>/4262.204.0
beeps12c0a3c2013-09-03 11:58:27 -0700616
beepse539be02013-07-31 21:57:39 -0700617 @return: An update URL, eg:
618 http://<devserver>/static/canary-channel/\
619 <board>/4262.204.0/factory_test/chromiumos_factory_image.bin
beeps12c0a3c2013-09-03 11:58:27 -0700620
621 @raises: ValueError if the factory artifact name is missing from
622 the config.
623
beepse539be02013-07-31 21:57:39 -0700624 """
625 if not image_name:
626 logging.error('Need an image_name to stage a factory image.')
627 return
628
Dan Shib8540a52015-07-16 14:18:23 -0700629 factory_artifact = CONFIG.get_config_value(
beeps12c0a3c2013-09-03 11:58:27 -0700630 'CROS', 'factory_artifact', type=str, default='')
631 if not factory_artifact:
632 raise ValueError('Cannot retrieve the factory artifact name from '
633 'autotest config, and hence cannot stage factory '
634 'artifacts.')
635
beepse539be02013-07-31 21:57:39 -0700636 logging.info('Staging build for servo install: %s', image_name)
Dan Shi216389c2015-12-22 11:03:06 -0800637 devserver = dev_server.ImageServer.resolve(image_name, self.hostname)
beepse539be02013-07-31 21:57:39 -0700638 devserver.stage_artifacts(
639 image_name,
beeps12c0a3c2013-09-03 11:58:27 -0700640 [factory_artifact],
641 archive_url=None)
beepse539be02013-07-31 21:57:39 -0700642
643 return tools.factory_image_url_pattern() % (devserver.url(), image_name)
644
645
xixuan016d95b2017-02-27 14:37:47 -0800646 def _get_au_monarch_fields(self, devserver, build):
647 """Form monarch fields by given devserve & build for auto-update.
648
649 @param devserver: the devserver (ImageServer instance) for auto-update.
650 @param build: the build to be updated.
651
652 @return A dictionary of monach fields.
653 """
654 try:
655 board, build_type, milestone, _ = server_utils.ParseBuildName(build)
656 except server_utils.ParseBuildNameException:
657 logging.warning('Unable to parse build name %s. Something is '
658 'likely broken, but will continue anyway.',
659 build)
660 board, build_type, milestone = ('', '', '')
661
662 monarch_fields = {
663 'dev_server': devserver.hostname,
664 'board': board,
665 'build_type': build_type,
666 'milestone': milestone
667 }
668 return monarch_fields
669
670
xixuanc30d7e92017-05-11 12:06:39 -0700671 def _retry_auto_update_with_new_devserver(self, build, last_devserver,
xixuana3bbc422017-05-04 15:57:21 -0700672 force_update, force_full_update,
673 force_original):
xixuanc30d7e92017-05-11 12:06:39 -0700674 """Kick off auto-update by devserver and send metrics.
675
676 @param build: the build to update.
677 @param last_devserver: the last devserver that failed to provision.
678 @param force_update: see |machine_install_by_devserver|'s force_udpate
679 for details.
680 @param force_full_update: see |machine_install_by_devserver|'s
681 force_full_update for details.
xixuana3bbc422017-05-04 15:57:21 -0700682 @param force_original: Whether to force stateful update with the
683 original payload.
xixuanc30d7e92017-05-11 12:06:39 -0700684
685 @return the result of |auto_update| in dev_server.
686 """
687 devserver = dev_server.resolve(
688 build, self.hostname, ban_list=[last_devserver.url()])
689 devserver.trigger_download(build, synchronous=False)
690 monarch_fields = self._get_au_monarch_fields(devserver, build)
691 logging.debug('Retry auto_update: resolved devserver for '
692 'auto-update: %s', devserver.url())
693
694 # Add metrics
695 install_with_dev_counter = metrics.Counter(
696 'chromeos/autotest/provision/install_with_devserver')
697 install_with_dev_counter.increment(fields=monarch_fields)
698 c = metrics.Counter(
699 'chromeos/autotest/provision/retry_by_devserver')
700 monarch_fields['last_devserver'] = last_devserver.hostname
701 monarch_fields['host'] = self.hostname
702 c.increment(fields=monarch_fields)
703
xixuan9f455342017-06-08 18:21:44 -0700704 # Won't retry auto_update in a retry of auto-update.
705 # In other words, we only retry auto-update once with a different
706 # devservers.
707 devserver.auto_update(
xixuanc30d7e92017-05-11 12:06:39 -0700708 self.hostname, build,
709 original_board=self.get_board().replace(
710 ds_constants.BOARD_PREFIX, ''),
711 original_release_version=self.get_release_version(),
712 log_dir=self.job.resultdir,
713 force_update=force_update,
xixuana3bbc422017-05-04 15:57:21 -0700714 full_update=force_full_update,
715 force_original=force_original)
xixuanc30d7e92017-05-11 12:06:39 -0700716
717
xixuanec801e32016-08-25 10:20:22 -0700718 def machine_install_by_devserver(self, update_url=None, force_update=False,
xixuan5dc64ea2016-05-20 17:27:51 -0700719 local_devserver=False, repair=False,
720 force_full_update=False):
721 """Ultiize devserver to install the DUT.
722
723 (TODO) crbugs.com/627269: The logic in this function has some overlap
724 with those in function machine_install. The merge will be done later,
725 not in the same CL.
726
xixuanec801e32016-08-25 10:20:22 -0700727 @param update_url: The update_url or build for the host to update.
xixuan5dc64ea2016-05-20 17:27:51 -0700728 @param force_update: Force an update even if the version installed
729 is the same. Default:False
730 @param local_devserver: Used by test_that to allow people to
731 use their local devserver. Default: False
732 @param repair: Forces update to repair image. Implies force_update.
733 @param force_full_update: If True, do not attempt to run stateful
734 update, force a full reimage. If False, try stateful update
735 first when the dut is already installed with the same version.
736 @raises autoupdater.ChromiumOSError
737
738 @returns A tuple of (image_name, host_attributes).
739 image_name is the name of image installed, e.g.,
740 veyron_jerry-release/R50-7871.0.0
741 host_attributes is a dictionary of (attribute, value), which
742 can be saved to afe_host_attributes table in database. This
743 method returns a dictionary with a single entry of
744 `job_repo_url`: repo_url, where repo_url is a devserver url to
745 autotest packages.
746 """
xixuan5dc64ea2016-05-20 17:27:51 -0700747 if repair:
Richard Barnette260cbd02016-10-06 12:23:28 -0700748 update_url = self._get_cros_repair_image_name()
xixuan5dc64ea2016-05-20 17:27:51 -0700749 force_update = True
750
xixuanec801e32016-08-25 10:20:22 -0700751 if not update_url and not self._parser.options.image:
xixuan5dc64ea2016-05-20 17:27:51 -0700752 raise error.AutoservError(
753 'There is no update URL, nor a method to get one.')
754
xixuanec801e32016-08-25 10:20:22 -0700755 if not update_url and self._parser.options.image:
756 update_url = self._parser.options.image
xixuan5dc64ea2016-05-20 17:27:51 -0700757
xixuan5dc64ea2016-05-20 17:27:51 -0700758
759 # Get build from parameter or AFE.
760 # If the build is not a URL, let devserver to stage it first.
761 # Otherwise, choose a devserver to trigger auto-update.
xixuanec801e32016-08-25 10:20:22 -0700762 build = None
Aviv Keshetf9375482016-10-26 02:09:03 -0700763 devserver = None
764 logging.debug('Resolving a devserver for auto-update')
Aviv Keshet3a76f762016-10-27 17:41:51 -0700765 previously_resolved = False
xixuanec801e32016-08-25 10:20:22 -0700766 if update_url.startswith('http://'):
767 build = autoupdater.url_to_image_name(update_url)
Aviv Keshet3a76f762016-10-27 17:41:51 -0700768 previously_resolved = True
769 else:
770 build = update_url
771 devserver = dev_server.resolve(build, self.hostname)
Allen Lia5cfb972016-12-27 17:17:22 -0800772 server_name = devserver.hostname
Aviv Keshet3a76f762016-10-27 17:41:51 -0700773
xixuan016d95b2017-02-27 14:37:47 -0800774 monarch_fields = self._get_au_monarch_fields(devserver, build)
Aviv Keshet3a76f762016-10-27 17:41:51 -0700775
776 if previously_resolved:
xixuan2e62d162016-09-29 19:19:19 -0700777 # Make sure devserver for Auto-Update has staged the build.
xixuan2e62d162016-09-29 19:19:19 -0700778 if server_name not in update_url:
Aviv Keshet3a76f762016-10-27 17:41:51 -0700779 logging.debug('Resolved to devserver that does not match '
780 'update_url. The previously resolved devserver '
781 'must be unhealthy. Switching to use devserver %s,'
782 ' and re-staging.',
xixuan2e62d162016-09-29 19:19:19 -0700783 server_name)
Aviv Keshetf9375482016-10-26 02:09:03 -0700784 logging.info('Staging build for AU: %s', update_url)
xixuan2e62d162016-09-29 19:19:19 -0700785 devserver.trigger_download(build, synchronous=False)
Dan Shi5e2efb72017-02-07 11:40:23 -0800786 c = metrics.Counter(
787 'chromeos/autotest/provision/failover_download')
788 c.increment(fields=monarch_fields)
xixuan5dc64ea2016-05-20 17:27:51 -0700789 else:
Aviv Keshetf9375482016-10-26 02:09:03 -0700790 logging.info('Staging build for AU: %s', update_url)
xixuan5dc64ea2016-05-20 17:27:51 -0700791 devserver.trigger_download(build, synchronous=False)
Dan Shi5e2efb72017-02-07 11:40:23 -0800792 c = metrics.Counter('chromeos/autotest/provision/trigger_download')
793 c.increment(fields=monarch_fields)
xixuan5dc64ea2016-05-20 17:27:51 -0700794
795 # Report provision stats.
xixuan016d95b2017-02-27 14:37:47 -0800796 install_with_dev_counter = metrics.Counter(
797 'chromeos/autotest/provision/install_with_devserver')
798 install_with_dev_counter.increment(fields=monarch_fields)
xixuanec801e32016-08-25 10:20:22 -0700799 logging.debug('Resolved devserver for auto-update: %s', devserver.url())
xixuan5dc64ea2016-05-20 17:27:51 -0700800
Aviv Keshetf9375482016-10-26 02:09:03 -0700801 # and other metrics from this function.
Dan Shi5e2efb72017-02-07 11:40:23 -0800802 metrics.Counter('chromeos/autotest/provision/resolve'
803 ).increment(fields=monarch_fields)
Aviv Keshetf9375482016-10-26 02:09:03 -0700804
xixuana3bbc422017-05-04 15:57:21 -0700805 force_original = self.get_chromeos_release_milestone() is None
806
xixuan9f455342017-06-08 18:21:44 -0700807 try:
808 devserver.auto_update(
809 self.hostname, build,
810 original_board=self.get_board().replace(
811 ds_constants.BOARD_PREFIX, ''),
812 original_release_version=self.get_release_version(),
813 log_dir=self.job.resultdir,
814 force_update=force_update,
815 full_update=force_full_update,
816 force_original=force_original)
817 except dev_server.RetryableProvisionException:
818 # It indicates that last provision failed due to devserver load
819 # issue, so another devserver is resolved to kick off provision
820 # job once again and only once.
821 logging.debug('Provision failed due to devserver issue,'
822 'retry it with another devserver.')
xixuan016d95b2017-02-27 14:37:47 -0800823
xixuan9f455342017-06-08 18:21:44 -0700824 # Check first whether this DUT is completely offline. If so, skip
825 # the following provision tries.
826 logging.debug('Checking whether host %s is online.', self.hostname)
827 if utils.ping(self.hostname, tries=1, deadline=1) == 0:
828 self._retry_auto_update_with_new_devserver(
829 build, devserver, force_update, force_full_update,
830 force_original)
831 else:
832 raise error.AutoservError(
833 'No answer to ping from %s' % self.hostname)
xixuan5dc64ea2016-05-20 17:27:51 -0700834
835 # The reason to resolve a new devserver in function machine_install
836 # is mostly because that the update_url there may has a strange format,
837 # and it's hard to parse the devserver url from it.
838 # Since we already resolve a devserver to trigger auto-update, the same
839 # devserver is used to form JOB_REPO_URL here. Verified in local test.
840 repo_url = tools.get_package_url(devserver.url(), build)
841 return build, {ds_constants.JOB_REPO_URL: repo_url}
842
843
Laurence Goodby778c9a42017-05-24 19:24:07 -0700844 def prepare_for_update(self):
845 """Prepares the DUT for an update.
846
847 Subclasses may override this to perform any special actions
848 required before updating.
849 """
Laurence Goodby468de252017-06-08 17:22:53 -0700850 pass
Laurence Goodby778c9a42017-05-24 19:24:07 -0700851
852
Chris Sosaa3ac2152012-05-23 22:23:13 -0700853 def machine_install(self, update_url=None, force_update=False,
Richard Barnette0b023a72015-04-24 16:07:30 +0000854 local_devserver=False, repair=False,
855 force_full_update=False):
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500856 """Install the DUT.
857
Dan Shi0f466e82013-02-22 15:44:58 -0800858 Use stateful update if the DUT is already running the same build.
859 Stateful update does not update kernel and tends to run much faster
860 than a full reimage. If the DUT is running a different build, or it
861 failed to do a stateful update, full update, including kernel update,
862 will be applied to the DUT.
863
Simran Basi5ace6f22016-01-06 17:30:44 -0800864 Once a host enters machine_install its host attribute job_repo_url
865 (used for package install) will be removed and then updated.
Scott Zawalskieadbf702013-03-14 09:23:06 -0400866
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500867 @param update_url: The url to use for the update
868 pattern: http://$devserver:###/update/$build
869 If update_url is None and repair is True we will install the
Dan Shi6964fa52014-12-18 11:04:27 -0800870 stable image listed in afe_stable_versions table. If the table
871 is not setup, global_config value under CROS.stable_cros_version
872 will be used instead.
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500873 @param force_update: Force an update even if the version installed
874 is the same. Default:False
Christopher Wiley6a4ff932015-05-15 14:00:47 -0700875 @param local_devserver: Used by test_that to allow people to
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500876 use their local devserver. Default: False
Chris Sosae92399e2015-04-24 11:32:59 -0700877 @param repair: Forces update to repair image. Implies force_update.
Fang Deng3d3b9272014-12-22 12:20:28 -0800878 @param force_full_update: If True, do not attempt to run stateful
879 update, force a full reimage. If False, try stateful update
880 first when the dut is already installed with the same version.
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500881 @raises autoupdater.ChromiumOSError
882
Dan Shibe3636a2016-02-14 22:48:01 -0800883 @returns A tuple of (image_name, host_attributes).
884 image_name is the name of image installed, e.g.,
885 veyron_jerry-release/R50-7871.0.0
886 host_attributes is a dictionary of (attribute, value), which
887 can be saved to afe_host_attributes table in database. This
888 method returns a dictionary with a single entry of
889 `job_repo_url`: repo_url, where repo_url is a devserver url to
890 autotest packages.
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500891 """
Chris Sosae92399e2015-04-24 11:32:59 -0700892 devserver = None
Richard Barnette0b023a72015-04-24 16:07:30 +0000893 if repair:
Chris Sosae92399e2015-04-24 11:32:59 -0700894 update_url, devserver = self._stage_image_for_update()
Richard Barnette0b023a72015-04-24 16:07:30 +0000895 force_update = True
Dan Shi0f466e82013-02-22 15:44:58 -0800896
Chris Sosae92399e2015-04-24 11:32:59 -0700897 if not update_url and not self._parser.options.image:
898 raise error.AutoservError(
Dan Shid07ee2e2015-09-24 14:49:25 -0700899 'There is no update URL, nor a method to get one.')
Chris Sosae92399e2015-04-24 11:32:59 -0700900
901 if not update_url and self._parser.options.image:
902 # This is the base case where we have no given update URL i.e.
903 # dynamic suites logic etc. This is the most flexible case where we
904 # can serve an update from any of our fleet of devservers.
905 requested_build = self._parser.options.image
906 if not requested_build.startswith('http://'):
907 logging.debug('Update will be staged for this installation')
908 update_url, devserver = self._stage_image_for_update(
Dan Shid07ee2e2015-09-24 14:49:25 -0700909 requested_build)
Chris Sosae92399e2015-04-24 11:32:59 -0700910 else:
911 update_url = requested_build
912
913 logging.debug('Update URL is %s', update_url)
914
Dan Shif48f8132016-02-18 10:34:30 -0800915 # Report provision stats.
Allen Lia5cfb972016-12-27 17:17:22 -0800916 server_name = dev_server.get_hostname(update_url)
Allen Li48a13fe2016-11-22 14:10:40 -0800917 (metrics.Counter('chromeos/autotest/provision/install')
918 .increment(fields={'devserver': server_name}))
Dan Shif48f8132016-02-18 10:34:30 -0800919
Dan Shid07ee2e2015-09-24 14:49:25 -0700920 # Create a file to indicate if provision fails. The file will be removed
921 # by stateful update or full install.
J. Richard Barnette91137f02016-03-10 16:52:26 -0800922 self.run('touch %s' % self.PROVISION_FAILED)
Dan Shid07ee2e2015-09-24 14:49:25 -0700923
Chris Sosae92399e2015-04-24 11:32:59 -0700924 update_complete = False
925 updater = autoupdater.ChromiumOSUpdater(
926 update_url, host=self, local_devserver=local_devserver)
Fang Deng3d3b9272014-12-22 12:20:28 -0800927 if not force_full_update:
928 try:
Chris Sosae92399e2015-04-24 11:32:59 -0700929 # If the DUT is already running the same build, try stateful
930 # update first as it's much quicker than a full re-image.
931 update_complete = self._try_stateful_update(
Dan Shid07ee2e2015-09-24 14:49:25 -0700932 update_url, force_update, updater)
Fang Deng3d3b9272014-12-22 12:20:28 -0800933 except Exception as e:
934 logging.exception(e)
J. Richard Barnette45e93de2012-04-11 17:24:15 -0700935
Dan Shi0f466e82013-02-22 15:44:58 -0800936 inactive_kernel = None
Chris Sosae92399e2015-04-24 11:32:59 -0700937 if update_complete or (not force_update and updater.check_version()):
938 logging.info('Install complete without full update')
939 else:
940 logging.info('DUT requires full update.')
941 self.reboot(timeout=self.REBOOT_TIMEOUT, wait=True)
Laurence Goodby778c9a42017-05-24 19:24:07 -0700942 self.prepare_for_update()
Dan Shi10b98482016-02-02 14:38:50 -0800943
Chris Sosae92399e2015-04-24 11:32:59 -0700944 num_of_attempts = provision.FLAKY_DEVSERVER_ATTEMPTS
Chris Sosab7612bc2013-03-21 10:32:37 -0700945
Chris Sosae92399e2015-04-24 11:32:59 -0700946 while num_of_attempts > 0:
947 num_of_attempts -= 1
948 try:
949 updater.run_update()
950 except Exception:
951 logging.warn('Autoupdate did not complete.')
952 # Do additional check for the devserver health. Ideally,
953 # the autoupdater.py could raise an exception when it
954 # detected network flake but that would require
955 # instrumenting the update engine and parsing it log.
956 if (num_of_attempts <= 0 or
957 devserver is None or
xixuan9e2c98d2016-02-26 19:04:53 -0800958 dev_server.ImageServer.devserver_healthy(
Chris Sosae92399e2015-04-24 11:32:59 -0700959 devserver.url())):
Dan Shid07ee2e2015-09-24 14:49:25 -0700960 raise
J. Richard Barnette45e93de2012-04-11 17:24:15 -0700961
Chris Sosae92399e2015-04-24 11:32:59 -0700962 logging.warn('Devserver looks unhealthy. Trying another')
963 update_url, devserver = self._stage_image_for_update(
964 requested_build)
965 logging.debug('New Update URL is %s', update_url)
966 updater = autoupdater.ChromiumOSUpdater(
967 update_url, host=self,
968 local_devserver=local_devserver)
969 else:
970 break
J. Richard Barnette45e93de2012-04-11 17:24:15 -0700971
Chris Sosae92399e2015-04-24 11:32:59 -0700972 # Give it some time in case of IO issues.
973 time.sleep(10)
Dan Shi5699ac22014-12-19 10:55:49 -0800974
Chris Sosae92399e2015-04-24 11:32:59 -0700975 # Figure out active and inactive kernel.
976 active_kernel, inactive_kernel = updater.get_kernel_state()
Simran Basi13fa1ba2013-03-04 10:56:47 -0800977
Chris Sosae92399e2015-04-24 11:32:59 -0700978 # Ensure inactive kernel has higher priority than active.
979 if (updater.get_kernel_priority(inactive_kernel)
980 < updater.get_kernel_priority(active_kernel)):
981 raise autoupdater.ChromiumOSError(
982 'Update failed. The priority of the inactive kernel'
983 ' partition is less than that of the active kernel'
984 ' partition.')
985
986 # Updater has returned successfully; reboot the host.
Richard Barnettefc3d7432016-07-07 10:21:31 -0700987 #
988 # Regarding the 'crossystem' command: In some cases, the
989 # TPM gets into a state such that it fails verification.
990 # We don't know why. However, this call papers over the
991 # problem by clearing the TPM during the reboot.
992 #
993 # We ignore failures from 'crossystem'. Although failure
994 # here is unexpected, and could signal a bug, the point
995 # of the exercise is to paper over problems; allowing
996 # this to fail would defeat the purpose.
997 self.run('crossystem clear_tpm_owner_request=1',
998 ignore_status=True)
Chris Sosae92399e2015-04-24 11:32:59 -0700999 self.reboot(timeout=self.REBOOT_TIMEOUT, wait=True)
1000
1001 self._post_update_processing(updater, inactive_kernel)
Simran Basi5ace6f22016-01-06 17:30:44 -08001002 image_name = autoupdater.url_to_image_name(update_url)
Dan Shibe3636a2016-02-14 22:48:01 -08001003 # update_url is different from devserver url needed to stage autotest
1004 # packages, therefore, resolve a new devserver url here.
1005 devserver_url = dev_server.ImageServer.resolve(image_name,
1006 self.hostname).url()
1007 repo_url = tools.get_package_url(devserver_url, image_name)
1008 return image_name, {ds_constants.JOB_REPO_URL: repo_url}
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001009
1010
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +08001011 def _clear_fw_version_labels(self, rw_only):
1012 """Clear firmware version labels from the machine.
1013
1014 @param rw_only: True to only clear fwrw_version; otherewise, clear
1015 both fwro_version and fwrw_version.
1016 """
Dan Shi9cb0eec2014-06-03 09:04:50 -07001017 labels = self._AFE.get_labels(
Dan Shi0723bf52015-06-24 10:52:38 -07001018 name__startswith=provision.FW_RW_VERSION_PREFIX,
Dan Shi9cb0eec2014-06-03 09:04:50 -07001019 host__hostname=self.hostname)
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +08001020 if not rw_only:
1021 labels = labels + self._AFE.get_labels(
1022 name__startswith=provision.FW_RO_VERSION_PREFIX,
1023 host__hostname=self.hostname)
Dan Shi9cb0eec2014-06-03 09:04:50 -07001024 for label in labels:
1025 label.remove_hosts(hosts=[self.hostname])
1026
1027
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +08001028 def _add_fw_version_label(self, build, rw_only):
Dan Shi9cb0eec2014-06-03 09:04:50 -07001029 """Add firmware version label to the machine.
1030
1031 @param build: Build of firmware.
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +08001032 @param rw_only: True to only add fwrw_version; otherwise, add both
1033 fwro_version and fwrw_version.
Dan Shi9cb0eec2014-06-03 09:04:50 -07001034
1035 """
Prathmesh Prabhu2c7471d2016-11-15 20:19:57 +00001036 fw_label = provision.fwrw_version_to_label(build)
MK Ryu73be9862015-07-06 12:25:00 -07001037 self._AFE.run('label_add_hosts', id=fw_label, hosts=[self.hostname])
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +08001038 if not rw_only:
Prathmesh Prabhu2c7471d2016-11-15 20:19:57 +00001039 fw_label = provision.fwro_version_to_label(build)
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +08001040 self._AFE.run('label_add_hosts', id=fw_label, hosts=[self.hostname])
Dan Shi9cb0eec2014-06-03 09:04:50 -07001041
1042
Tom Wai-Hong Tam4ac78982016-01-08 02:34:37 +08001043 def firmware_install(self, build=None, rw_only=False):
Dan Shi9cb0eec2014-06-03 09:04:50 -07001044 """Install firmware to the DUT.
1045
1046 Use stateful update if the DUT is already running the same build.
1047 Stateful update does not update kernel and tends to run much faster
1048 than a full reimage. If the DUT is running a different build, or it
1049 failed to do a stateful update, full update, including kernel update,
1050 will be applied to the DUT.
1051
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +08001052 Once a host enters firmware_install its fw[ro|rw]_version label will
1053 be removed. After the firmware is updated successfully, a new
1054 fw[ro|rw]_version label will be added to the host.
Dan Shi9cb0eec2014-06-03 09:04:50 -07001055
1056 @param build: The build version to which we want to provision the
1057 firmware of the machine,
1058 e.g. 'link-firmware/R22-2695.1.144'.
Tom Wai-Hong Tam4ac78982016-01-08 02:34:37 +08001059 @param rw_only: True to only install firmware to its RW portions. Keep
1060 the RO portions unchanged.
Dan Shi9cb0eec2014-06-03 09:04:50 -07001061
1062 TODO(dshi): After bug 381718 is fixed, update here with corresponding
1063 exceptions that could be raised.
1064
1065 """
1066 if not self.servo:
1067 raise error.TestError('Host %s does not have servo.' %
1068 self.hostname)
1069
Wai-Hong Tam64cf7912017-05-23 16:41:40 -07001070 # Get the DUT board name from servod.
1071 board = self.servo.get_board()
Dan Shi9cb0eec2014-06-03 09:04:50 -07001072
Chris Sosae92399e2015-04-24 11:32:59 -07001073 # If build is not set, try to install firmware from stable CrOS.
Dan Shi9cb0eec2014-06-03 09:04:50 -07001074 if not build:
Richard Barnette260cbd02016-10-06 12:23:28 -07001075 build = afe_utils.get_stable_faft_version(board)
Dan Shi3d7a0e12015-10-12 11:55:45 -07001076 if not build:
1077 raise error.TestError(
1078 'Failed to find stable firmware build for %s.',
1079 self.hostname)
1080 logging.info('Will install firmware from build %s.', build)
Dan Shi9cb0eec2014-06-03 09:04:50 -07001081
1082 config = FAFTConfig(board)
1083 if config.use_u_boot:
1084 ap_image = 'image-%s.bin' % board
1085 else: # Depthcharge platform
1086 ap_image = 'image.bin'
1087 ec_image = 'ec.bin'
Dan Shi216389c2015-12-22 11:03:06 -08001088 ds = dev_server.ImageServer.resolve(build, self.hostname)
Dan Shi9cb0eec2014-06-03 09:04:50 -07001089 ds.stage_artifacts(build, ['firmware'])
1090
1091 tmpd = autotemp.tempdir(unique_id='fwimage')
1092 try:
1093 fwurl = self._FW_IMAGE_URL_PATTERN % (ds.url(), build)
1094 local_tarball = os.path.join(tmpd.name, os.path.basename(fwurl))
xixuan4e116822016-11-17 15:32:10 -08001095 ds.download_file(fwurl, local_tarball)
Dan Shi9cb0eec2014-06-03 09:04:50 -07001096
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +08001097 self._clear_fw_version_labels(rw_only)
Tom Wai-Hong Tam552fee12016-05-11 06:27:58 +08001098 if config.chrome_ec:
Tom Wai-Hong Tamf9db57a2016-03-17 05:32:22 +08001099 logging.info('Will re-program EC %snow', 'RW ' if rw_only else '')
1100 server_utils.system('tar xf %s -C %s %s' %
1101 (local_tarball, tmpd.name, ec_image),
1102 timeout=60)
1103 self.servo.program_ec(os.path.join(tmpd.name, ec_image), rw_only)
1104 else:
1105 logging.info('Not a Chrome EC, ignore re-programing it')
Tom Wai-Hong Tam4ac78982016-01-08 02:34:37 +08001106 logging.info('Will re-program BIOS %snow', 'RW ' if rw_only else '')
Tom Wai-Hong Tamf9db57a2016-03-17 05:32:22 +08001107 server_utils.system('tar xf %s -C %s %s' %
1108 (local_tarball, tmpd.name, ap_image),
1109 timeout=60)
Tom Wai-Hong Tam4ac78982016-01-08 02:34:37 +08001110 self.servo.program_bios(os.path.join(tmpd.name, ap_image), rw_only)
Dan Shi9cb0eec2014-06-03 09:04:50 -07001111 self.servo.get_power_state_controller().reset()
1112 time.sleep(self.servo.BOOT_DELAY)
Wai-Hong Tamb5f66ce2016-11-10 15:45:30 -08001113 if utils.host_is_in_lab_zone(self.hostname):
1114 self._add_fw_version_label(build, rw_only)
Dan Shi9cb0eec2014-06-03 09:04:50 -07001115 finally:
1116 tmpd.clean()
1117
1118
Dan Shi10e992b2013-08-30 11:02:59 -07001119 def show_update_engine_log(self):
1120 """Output update engine log."""
MK Ryu35d661e2014-09-25 17:44:10 -07001121 logging.debug('Dumping %s', client_constants.UPDATE_ENGINE_LOG)
1122 self.run('cat %s' % client_constants.UPDATE_ENGINE_LOG)
Dan Shi10e992b2013-08-30 11:02:59 -07001123
1124
beepsf079cfb2013-09-18 17:49:51 -07001125 def servo_install(self, image_url=None, usb_boot_timeout=USB_BOOT_TIMEOUT,
1126 install_timeout=INSTALL_TIMEOUT):
Scott Zawalski62bacae2013-03-05 10:40:32 -05001127 """
1128 Re-install the OS on the DUT by:
1129 1) installing a test image on a USB storage device attached to the Servo
1130 board,
Richard Barnette03a0c132012-11-05 12:40:35 -08001131 2) booting that image in recovery mode, and then
J. Richard Barnette31b2e312013-04-04 16:05:22 -07001132 3) installing the image with chromeos-install.
1133
Scott Zawalski62bacae2013-03-05 10:40:32 -05001134 @param image_url: If specified use as the url to install on the DUT.
1135 otherwise boot the currently staged image on the USB stick.
beepsf079cfb2013-09-18 17:49:51 -07001136 @param usb_boot_timeout: The usb_boot_timeout to use during reimage.
1137 Factory images need a longer usb_boot_timeout than regular
1138 cros images.
1139 @param install_timeout: The timeout to use when installing the chromeos
1140 image. Factory images need a longer install_timeout.
Richard Barnette03a0c132012-11-05 12:40:35 -08001141
Scott Zawalski62bacae2013-03-05 10:40:32 -05001142 @raises AutoservError if the image fails to boot.
beepsf079cfb2013-09-18 17:49:51 -07001143
J. Richard Barnette0199cc82014-12-05 17:08:40 -08001144 """
beepsf079cfb2013-09-18 17:49:51 -07001145 logging.info('Downloading image to USB, then booting from it. Usb boot '
1146 'timeout = %s', usb_boot_timeout)
Allen Li48a13fe2016-11-22 14:10:40 -08001147 with metrics.SecondsTimer(
1148 'chromeos/autotest/provision/servo_install/boot_duration'):
1149 self.servo.install_recovery_image(image_url)
1150 if not self.wait_up(timeout=usb_boot_timeout):
1151 raise hosts.AutoservRepairError(
1152 'DUT failed to boot from USB after %d seconds' %
1153 usb_boot_timeout)
Scott Zawalski62bacae2013-03-05 10:40:32 -05001154
Tom Wai-Hong Tamf6b4f812015-08-08 04:14:59 +08001155 # The new chromeos-tpm-recovery has been merged since R44-7073.0.0.
1156 # In old CrOS images, this command fails. Skip the error.
Tom Wai-Hong Tam27af7332015-07-25 06:09:39 +08001157 logging.info('Resetting the TPM status')
Tom Wai-Hong Tamf6b4f812015-08-08 04:14:59 +08001158 try:
1159 self.run('chromeos-tpm-recovery')
1160 except error.AutoservRunError:
1161 logging.warn('chromeos-tpm-recovery is too old.')
1162
Tom Wai-Hong Tam27af7332015-07-25 06:09:39 +08001163
Allen Li48a13fe2016-11-22 14:10:40 -08001164 with metrics.SecondsTimer(
1165 'chromeos/autotest/provision/servo_install/install_duration'):
1166 logging.info('Installing image through chromeos-install.')
1167 self.run('chromeos-install --yes', timeout=install_timeout)
1168 self.halt()
beepsf079cfb2013-09-18 17:49:51 -07001169
1170 logging.info('Power cycling DUT through servo.')
J. Richard Barnette0199cc82014-12-05 17:08:40 -08001171 self.servo.get_power_state_controller().power_off()
Fang Dengafb88142013-05-30 17:44:31 -07001172 self.servo.switch_usbkey('off')
J. Richard Barnette0199cc82014-12-05 17:08:40 -08001173 # N.B. The Servo API requires that we use power_on() here
1174 # for two reasons:
1175 # 1) After turning on a DUT in recovery mode, you must turn
1176 # it off and then on with power_on() once more to
1177 # disable recovery mode (this is a Parrot specific
1178 # requirement).
1179 # 2) After power_off(), the only way to turn on is with
1180 # power_on() (this is a Storm specific requirement).
J. Richard Barnettefbcc7122013-07-24 18:24:59 -07001181 self.servo.get_power_state_controller().power_on()
beepsf079cfb2013-09-18 17:49:51 -07001182
1183 logging.info('Waiting for DUT to come back up.')
Richard Barnette03a0c132012-11-05 12:40:35 -08001184 if not self.wait_up(timeout=self.BOOT_TIMEOUT):
1185 raise error.AutoservError('DUT failed to reboot installed '
1186 'test image after %d seconds' %
Scott Zawalski62bacae2013-03-05 10:40:32 -05001187 self.BOOT_TIMEOUT)
1188
1189
Richard Barnette9a26ad62016-06-10 12:03:08 -07001190 def repair_servo(self):
Dan Shi90466352015-09-22 15:01:05 -07001191 """
Richard Barnette9a26ad62016-06-10 12:03:08 -07001192 Confirm that servo is initialized and verified.
Dan Shi90466352015-09-22 15:01:05 -07001193
Richard Barnette9a26ad62016-06-10 12:03:08 -07001194 If the servo object is missing, attempt to repair the servo
1195 host. Repair failures are passed back to the caller.
1196
1197 @raise AutoservError: If there is no servo host for this CrOS
1198 host.
1199 """
1200 if self.servo:
1201 return
1202 if not self._servo_host:
1203 raise error.AutoservError('No servo host for %s.' %
1204 self.hostname)
1205 self._servo_host.repair()
1206 self.servo = self._servo_host.get_servo()
Dan Shi90466352015-09-22 15:01:05 -07001207
1208
J. Richard Barnettec2d99cf2015-11-18 12:46:15 -08001209 def repair(self):
1210 """Attempt to get the DUT to pass `self.verify()`.
Richard Barnette82c35912012-11-20 10:09:10 -08001211
1212 This overrides the base class function for repair; it does
J. Richard Barnette91137f02016-03-10 16:52:26 -08001213 not call back to the parent class, but instead relies on
1214 `self._repair_strategy` to coordinate the verification and
1215 repair steps needed to get the DUT working.
Richard Barnette82c35912012-11-20 10:09:10 -08001216 """
J. Richard Barnette91137f02016-03-10 16:52:26 -08001217 self._repair_strategy.repair(self)
Richard Barnettebfab8672017-08-10 14:48:28 -07001218 # Sometimes, hosts with certain ethernet dongles get stuck in a
1219 # bad network state where they're reachable from this code, but
1220 # not from the devservers during provisioning. Rebooting the
1221 # DUT fixes it.
1222 #
1223 # TODO(jrbarnette): Ideally, we'd get rid of the problem
1224 # dongles, and drop this code. Failing that, we could be smart
1225 # enough not to reboot if repair rebooted the DUT (e.g. by
1226 # looking at DUT uptime after repair completes).
1227
1228 self.reboot()
Scott Zawalski89c44dd2013-02-26 09:28:02 -05001229
Richard Barnette82c35912012-11-20 10:09:10 -08001230
J. Richard Barnette1d78b012012-05-15 13:56:30 -07001231 def close(self):
Fang Deng0ca40e22013-08-27 17:47:44 -07001232 super(CrosHost, self).close()
xixuand6011f12016-12-08 15:01:58 -08001233 if self._chameleon_host:
1234 self._chameleon_host.close()
1235
1236 if self._servo_host:
1237 self._servo_host.close()
J. Richard Barnette1d78b012012-05-15 13:56:30 -07001238
1239
Dan Shi49ca0932014-11-14 11:22:27 -08001240 def get_power_supply_info(self):
1241 """Get the output of power_supply_info.
1242
1243 power_supply_info outputs the info of each power supply, e.g.,
1244 Device: Line Power
1245 online: no
1246 type: Mains
1247 voltage (V): 0
1248 current (A): 0
1249 Device: Battery
1250 state: Discharging
1251 percentage: 95.9276
1252 technology: Li-ion
1253
1254 Above output shows two devices, Line Power and Battery, with details of
1255 each device listed. This function parses the output into a dictionary,
1256 with key being the device name, and value being a dictionary of details
1257 of the device info.
1258
1259 @return: The dictionary of power_supply_info, e.g.,
1260 {'Line Power': {'online': 'yes', 'type': 'main'},
1261 'Battery': {'vendor': 'xyz', 'percentage': '100'}}
Dan Shie9b765d2014-12-29 16:59:49 -08001262 @raise error.AutoservRunError if power_supply_info tool is not found in
1263 the DUT. Caller should handle this error to avoid false failure
1264 on verification.
Dan Shi49ca0932014-11-14 11:22:27 -08001265 """
1266 result = self.run('power_supply_info').stdout.strip()
1267 info = {}
1268 device_name = None
1269 device_info = {}
1270 for line in result.split('\n'):
1271 pair = [v.strip() for v in line.split(':')]
1272 if len(pair) != 2:
1273 continue
1274 if pair[0] == 'Device':
1275 if device_name:
1276 info[device_name] = device_info
1277 device_name = pair[1]
1278 device_info = {}
1279 else:
1280 device_info[pair[0]] = pair[1]
1281 if device_name and not device_name in info:
1282 info[device_name] = device_info
1283 return info
1284
1285
1286 def get_battery_percentage(self):
1287 """Get the battery percentage.
1288
1289 @return: The percentage of battery level, value range from 0-100. Return
1290 None if the battery info cannot be retrieved.
1291 """
1292 try:
1293 info = self.get_power_supply_info()
1294 logging.info(info)
1295 return float(info['Battery']['percentage'])
Dan Shie9b765d2014-12-29 16:59:49 -08001296 except (KeyError, ValueError, error.AutoservRunError):
Dan Shi49ca0932014-11-14 11:22:27 -08001297 return None
1298
1299
1300 def is_ac_connected(self):
1301 """Check if the dut has power adapter connected and charging.
1302
1303 @return: True if power adapter is connected and charging.
1304 """
1305 try:
1306 info = self.get_power_supply_info()
1307 return info['Line Power']['online'] == 'yes'
Dan Shie9b765d2014-12-29 16:59:49 -08001308 except (KeyError, error.AutoservRunError):
1309 return None
Dan Shi49ca0932014-11-14 11:22:27 -08001310
1311
Simran Basi5e6339a2013-03-21 11:34:32 -07001312 def _cleanup_poweron(self):
1313 """Special cleanup method to make sure hosts always get power back."""
1314 afe = frontend_wrappers.RetryingAFE(timeout_min=5, delay_sec=10)
1315 hosts = afe.get_hosts(hostname=self.hostname)
1316 if not hosts or not (self._RPM_OUTLET_CHANGED in
1317 hosts[0].attributes):
1318 return
1319 logging.debug('This host has recently interacted with the RPM'
1320 ' Infrastructure. Ensuring power is on.')
1321 try:
1322 self.power_on()
Dan Shi7dca56e2014-11-11 17:07:56 -08001323 afe.set_host_attribute(self._RPM_OUTLET_CHANGED, None,
1324 hostname=self.hostname)
Simran Basi5e6339a2013-03-21 11:34:32 -07001325 except rpm_client.RemotePowerException:
Simran Basi5e6339a2013-03-21 11:34:32 -07001326 logging.error('Failed to turn Power On for this host after '
1327 'cleanup through the RPM Infrastructure.')
Dan Shi49ca0932014-11-14 11:22:27 -08001328
1329 battery_percentage = self.get_battery_percentage()
Dan Shif01ebe22014-12-05 13:10:57 -08001330 if battery_percentage and battery_percentage < 50:
Dan Shi49ca0932014-11-14 11:22:27 -08001331 raise
1332 elif self.is_ac_connected():
1333 logging.info('The device has power adapter connected and '
1334 'charging. No need to try to turn RPM on '
1335 'again.')
1336 afe.set_host_attribute(self._RPM_OUTLET_CHANGED, None,
1337 hostname=self.hostname)
1338 logging.info('Battery level is now at %s%%. The device may '
1339 'still have enough power to run test, so no '
1340 'exception will be raised.', battery_percentage)
1341
Simran Basi5e6339a2013-03-21 11:34:32 -07001342
beepsc87ff602013-07-31 21:53:00 -07001343 def _is_factory_image(self):
1344 """Checks if the image on the DUT is a factory image.
1345
1346 @return: True if the image on the DUT is a factory image.
1347 False otherwise.
1348 """
1349 result = self.run('[ -f /root/.factory_test ]', ignore_status=True)
1350 return result.exit_status == 0
1351
1352
1353 def _restart_ui(self):
J. Richard Barnette84890bd2014-02-21 11:05:47 -08001354 """Restart the Chrome UI.
beepsc87ff602013-07-31 21:53:00 -07001355
1356 @raises: FactoryImageCheckerException for factory images, since
1357 we cannot attempt to restart ui on them.
1358 error.AutoservRunError for any other type of error that
1359 occurs while restarting ui.
1360 """
1361 if self._is_factory_image():
Dan Shi549fb822015-03-24 18:01:11 -07001362 raise FactoryImageCheckerException('Cannot restart ui on factory '
1363 'images')
beepsc87ff602013-07-31 21:53:00 -07001364
J. Richard Barnette84890bd2014-02-21 11:05:47 -08001365 # TODO(jrbarnette): The command to stop/start the ui job
1366 # should live inside cros_ui, too. However that would seem
1367 # to imply interface changes to the existing start()/restart()
1368 # functions, which is a bridge too far (for now).
J. Richard Barnette6069aa12015-06-08 09:10:24 -07001369 prompt = cros_ui.get_chrome_session_ident(self)
J. Richard Barnette84890bd2014-02-21 11:05:47 -08001370 self.run('stop ui; start ui')
1371 cros_ui.wait_for_chrome_ready(prompt, self)
beepsc87ff602013-07-31 21:53:00 -07001372
1373
xixuana3bbc422017-05-04 15:57:21 -07001374 def _get_lsb_release_content(self):
1375 """Return the content of lsb-release file of host."""
1376 return self.run(
1377 'cat "%s"' % client_constants.LSB_RELEASE).stdout.strip()
1378
1379
Dan Shi549fb822015-03-24 18:01:11 -07001380 def get_release_version(self):
1381 """Get the value of attribute CHROMEOS_RELEASE_VERSION from lsb-release.
1382
1383 @returns The version string in lsb-release, under attribute
1384 CHROMEOS_RELEASE_VERSION.
1385 """
Dan Shi549fb822015-03-24 18:01:11 -07001386 return lsbrelease_utils.get_chromeos_release_version(
xixuana3bbc422017-05-04 15:57:21 -07001387 lsb_release_content=self._get_lsb_release_content())
1388
1389
1390 def get_chromeos_release_milestone(self):
1391 """Get the value of attribute CHROMEOS_RELEASE_BUILD_TYPE
1392 from lsb-release.
1393
1394 @returns The version string in lsb-release, under attribute
1395 CHROMEOS_RELEASE_BUILD_TYPE.
1396 """
1397 return lsbrelease_utils.get_chromeos_release_milestone(
1398 lsb_release_content=self._get_lsb_release_content())
Dan Shi549fb822015-03-24 18:01:11 -07001399
1400
1401 def verify_cros_version_label(self):
1402 """ Make sure host's cros-version label match the actual image in dut.
1403
1404 Remove any cros-version: label that doesn't match that installed in
1405 the dut.
1406
1407 @param raise_error: Set to True to raise exception if any mismatch found
1408
1409 @raise error.AutoservError: If any mismatch between cros-version label
1410 and the build installed in dut is found.
1411 """
1412 labels = self._AFE.get_labels(
1413 name__startswith=ds_constants.VERSION_PREFIX,
1414 host__hostname=self.hostname)
1415 mismatch_found = False
1416 if labels:
1417 # Get CHROMEOS_RELEASE_VERSION from lsb-release, e.g., 6908.0.0.
1418 # Note that it's different from cros-version label, which has
1419 # builder and branch info, e.g.,
1420 # cros-version:peppy-release/R43-6908.0.0
1421 release_version = self.get_release_version()
1422 host_list = [self.hostname]
1423 for label in labels:
1424 # Remove any cros-version label that does not match
1425 # release_version.
1426 build_version = label.name[len(ds_constants.VERSION_PREFIX):]
1427 if not utils.version_match(build_version, release_version):
1428 logging.warn('cros-version label "%s" does not match '
1429 'release version %s. Removing the label.',
1430 label.name, release_version)
1431 label.remove_hosts(hosts=host_list)
1432 mismatch_found = True
1433 if mismatch_found:
1434 raise error.AutoservError('The host has wrong cros-version label.')
1435
1436
Laurence Goodby778c9a42017-05-24 19:24:07 -07001437 def cleanup_services(self):
1438 """Reinitializes the device for cleanup.
1439
1440 Subclasses may override this to customize the cleanup method.
1441
1442 To indicate failure of the reset, the implementation may raise
1443 any of:
1444 error.AutoservRunError
1445 error.AutotestRunError
1446 FactoryImageCheckerException
1447
1448 @raises error.AutoservRunError
1449 @raises error.AutotestRunError
1450 @raises error.FactoryImageCheckerException
1451 """
1452 self._restart_ui()
1453
1454
beepsc87ff602013-07-31 21:53:00 -07001455 def cleanup(self):
MK Ryu35d661e2014-09-25 17:44:10 -07001456 self.run('rm -f %s' % client_constants.CLEANUP_LOGS_PAUSED_FILE)
Scott Zawalskiddbc31e2012-11-15 11:29:01 -05001457 try:
Laurence Goodby778c9a42017-05-24 19:24:07 -07001458 self.cleanup_services()
beepsc87ff602013-07-31 21:53:00 -07001459 except (error.AutotestRunError, error.AutoservRunError,
1460 FactoryImageCheckerException):
Ilja H. Friedel04be2bd2014-05-07 21:29:59 -07001461 logging.warning('Unable to restart ui, rebooting device.')
Scott Zawalskiddbc31e2012-11-15 11:29:01 -05001462 # Since restarting the UI fails fall back to normal Autotest
1463 # cleanup routines, i.e. reboot the machine.
Fang Deng0ca40e22013-08-27 17:47:44 -07001464 super(CrosHost, self).cleanup()
Simran Basi5e6339a2013-03-21 11:34:32 -07001465 # Check if the rpm outlet was manipulated.
Simran Basid5e5e272012-09-24 15:23:59 -07001466 if self.has_power():
Simran Basi5e6339a2013-03-21 11:34:32 -07001467 self._cleanup_poweron()
Dan Shi549fb822015-03-24 18:01:11 -07001468 self.verify_cros_version_label()
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001469
1470
Yu-Ju Honga2be94a2012-07-31 09:48:52 -07001471 def reboot(self, **dargs):
1472 """
1473 This function reboots the site host. The more generic
1474 RemoteHost.reboot() performs sync and sleeps for 5
1475 seconds. This is not necessary for Chrome OS devices as the
1476 sync should be finished in a short time during the reboot
1477 command.
1478 """
Tom Wai-Hong Tamf5cd1d42012-08-13 12:04:08 +08001479 if 'reboot_cmd' not in dargs:
Doug Anderson7d5aeb22014-02-27 15:12:17 -08001480 reboot_timeout = dargs.get('reboot_timeout', 10)
J. Richard Barnette9af19632015-09-25 12:18:03 -07001481 dargs['reboot_cmd'] = ('sleep 1; '
1482 'reboot & sleep %d; '
1483 'reboot -f' % reboot_timeout)
Yu-Ju Honga2be94a2012-07-31 09:48:52 -07001484 # Enable fastsync to avoid running extra sync commands before reboot.
Tom Wai-Hong Tamf5cd1d42012-08-13 12:04:08 +08001485 if 'fastsync' not in dargs:
1486 dargs['fastsync'] = True
Michael Liangda8c60a2014-06-03 13:24:51 -07001487
Charlie Mooneya8e6dab2014-05-29 14:37:55 -07001488 # For purposes of logging reboot times:
1489 # Get the board name i.e. 'daisy_spring'
Michael Liangca4f5a62014-07-10 15:45:13 -07001490 board_fullname = self.get_board()
1491
1492 # Strip the prefix and add it to dargs.
1493 dargs['board'] = board_fullname[board_fullname.find(':')+1:]
Vincent Palatindf2372c2016-10-07 17:03:00 +02001494 # Record who called us
1495 orig = sys._getframe(1).f_code
Vincent Palatin80780b22016-07-27 16:02:37 +02001496 metric_fields = {'board' : dargs['board'],
Vincent Palatindf2372c2016-10-07 17:03:00 +02001497 'dut_host_name' : self.hostname,
1498 'success' : True}
1499 metric_debug_fields = {'board' : dargs['board'],
1500 'caller' : "%s:%s" % (orig.co_filename, orig.co_name),
1501 'success' : True,
1502 'error' : ''}
1503
Vincent Palatin80780b22016-07-27 16:02:37 +02001504 t0 = time.time()
1505 try:
1506 super(CrosHost, self).reboot(**dargs)
1507 except Exception as e:
1508 metric_fields['success'] = False
Vincent Palatindf2372c2016-10-07 17:03:00 +02001509 metric_debug_fields['success'] = False
1510 metric_debug_fields['error'] = type(e).__name__
Vincent Palatin80780b22016-07-27 16:02:37 +02001511 raise
1512 finally:
1513 duration = int(time.time() - t0)
Dan Shi5e2efb72017-02-07 11:40:23 -08001514 metrics.Counter(
1515 'chromeos/autotest/autoserv/reboot_count').increment(
1516 fields=metric_fields)
1517 metrics.Counter(
1518 'chromeos/autotest/autoserv/reboot_debug').increment(
1519 fields=metric_debug_fields)
1520 metrics.SecondsDistribution(
1521 'chromeos/autotest/autoserv/reboot_duration').add(
1522 duration, fields=metric_fields)
Yu-Ju Honga2be94a2012-07-31 09:48:52 -07001523
1524
Gwendal Grignou7a61d2f2014-05-23 11:05:51 -07001525 def suspend(self, **dargs):
1526 """
1527 This function suspends the site host.
1528 """
1529 suspend_time = dargs.get('suspend_time', 60)
1530 dargs['timeout'] = suspend_time
1531 if 'suspend_cmd' not in dargs:
J. Richard Barnette9af19632015-09-25 12:18:03 -07001532 dargs['suspend_cmd'] = ' && '.join([
1533 'echo 0 > /sys/class/rtc/rtc0/wakealarm',
Gwendal Grignou7a61d2f2014-05-23 11:05:51 -07001534 'echo +%d > /sys/class/rtc/rtc0/wakealarm' % suspend_time,
J. Richard Barnette9af19632015-09-25 12:18:03 -07001535 'powerd_dbus_suspend --delay=0'])
Gwendal Grignou7a61d2f2014-05-23 11:05:51 -07001536 super(CrosHost, self).suspend(**dargs)
1537
1538
Simran Basiec564392014-08-25 16:48:09 -07001539 def upstart_status(self, service_name):
1540 """Check the status of an upstart init script.
1541
1542 @param service_name: Service to look up.
1543
1544 @returns True if the service is running, False otherwise.
1545 """
Richard Barnettee204dc52017-09-26 11:02:25 -07001546 return 'start/running' in self.run('status %s' % service_name,
1547 ignore_status=True).stdout
Simran Basiec564392014-08-25 16:48:09 -07001548
1549
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001550 def verify_software(self):
Richard Barnetteb2bc13c2013-01-08 17:32:51 -08001551 """Verify working software on a Chrome OS system.
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001552
Richard Barnetteb2bc13c2013-01-08 17:32:51 -08001553 Tests for the following conditions:
1554 1. All conditions tested by the parent version of this
1555 function.
1556 2. Sufficient space in /mnt/stateful_partition.
Fang Deng6b05f5b2013-03-20 13:42:11 -07001557 3. Sufficient space in /mnt/stateful_partition/encrypted.
1558 4. update_engine answers a simple status request over DBus.
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001559
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001560 """
Fang Deng0ca40e22013-08-27 17:47:44 -07001561 super(CrosHost, self).verify_software()
Dan Shib8540a52015-07-16 14:18:23 -07001562 default_kilo_inodes_required = CONFIG.get_config_value(
1563 'SERVER', 'kilo_inodes_required', type=int, default=100)
1564 board = self.get_board().replace(ds_constants.BOARD_PREFIX, '')
1565 kilo_inodes_required = CONFIG.get_config_value(
1566 'SERVER', 'kilo_inodes_required_%s' % board,
1567 type=int, default=default_kilo_inodes_required)
1568 self.check_inodes('/mnt/stateful_partition', kilo_inodes_required)
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001569 self.check_diskspace(
1570 '/mnt/stateful_partition',
Dan Shib8540a52015-07-16 14:18:23 -07001571 CONFIG.get_config_value(
Fang Deng6b05f5b2013-03-20 13:42:11 -07001572 'SERVER', 'gb_diskspace_required', type=float,
1573 default=20.0))
Gaurav Shahe448af82014-06-19 15:18:59 -07001574 encrypted_stateful_path = '/mnt/stateful_partition/encrypted'
1575 # Not all targets build with encrypted stateful support.
1576 if self.path_exists(encrypted_stateful_path):
1577 self.check_diskspace(
1578 encrypted_stateful_path,
Dan Shib8540a52015-07-16 14:18:23 -07001579 CONFIG.get_config_value(
Gaurav Shahe448af82014-06-19 15:18:59 -07001580 'SERVER', 'gb_encrypted_diskspace_required', type=float,
1581 default=0.1))
beepsc87ff602013-07-31 21:53:00 -07001582
Justin TerAvest3b0c5ba2017-11-07 09:59:11 -07001583 self.wait_for_system_services()
Prashanth B5d0a0512014-04-25 12:26:08 -07001584
beepsc87ff602013-07-31 21:53:00 -07001585 # Factory images don't run update engine,
1586 # goofy controls dbus on these DUTs.
1587 if not self._is_factory_image():
1588 self.run('update_engine_client --status')
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001589
Dan Shi549fb822015-03-24 18:01:11 -07001590 self.verify_cros_version_label()
1591
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001592
Justin TerAvest3b0c5ba2017-11-07 09:59:11 -07001593 @retry.retry(error.AutoservError, timeout_min=5, delay_sec=10)
1594 def wait_for_system_services(self):
1595 """Waits for system-services to be running.
1596
1597 Sometimes, update_engine will take a while to update firmware, so we
1598 should give this some time to finish. See crbug.com/765686#c38 for
1599 details.
1600 """
1601 if not self.upstart_status('system-services'):
1602 raise error.AutoservError('Chrome failed to reach login. '
1603 'System services not running.')
1604
1605
J. Richard Barnettea7e7fdc2016-02-12 12:35:36 -08001606 def verify(self):
1607 self._repair_strategy.verify(self)
1608
1609
Fang Deng96667ca2013-08-01 17:46:18 -07001610 def make_ssh_command(self, user='root', port=22, opts='', hosts_file=None,
Dean Liao4bd3e312017-11-13 17:32:17 +08001611 connect_timeout=None, alive_interval=None):
Fang Deng96667ca2013-08-01 17:46:18 -07001612 """Override default make_ssh_command to use options tuned for Chrome OS.
1613
1614 Tuning changes:
1615 - ConnectTimeout=30; maximum of 30 seconds allowed for an SSH
1616 connection failure. Consistency with remote_access.sh.
1617
Samuel Tan2ce155b2015-06-23 18:24:38 -07001618 - ServerAliveInterval=900; which causes SSH to ping connection every
1619 900 seconds. In conjunction with ServerAliveCountMax ensures
1620 that if the connection dies, Autotest will bail out.
Fang Deng96667ca2013-08-01 17:46:18 -07001621 Originally tried 60 secs, but saw frequent job ABORTS where
Samuel Tan2ce155b2015-06-23 18:24:38 -07001622 the test completed successfully. Later increased from 180 seconds to
1623 900 seconds to account for tests where the DUT is suspended for
1624 longer periods of time.
Fang Deng96667ca2013-08-01 17:46:18 -07001625
1626 - ServerAliveCountMax=3; consistency with remote_access.sh.
1627
1628 - ConnectAttempts=4; reduce flakiness in connection errors;
1629 consistency with remote_access.sh.
1630
1631 - UserKnownHostsFile=/dev/null; we don't care about the keys.
1632 Host keys change with every new installation, don't waste
1633 memory/space saving them.
1634
1635 - SSH protocol forced to 2; needed for ServerAliveInterval.
1636
1637 @param user User name to use for the ssh connection.
1638 @param port Port on the target host to use for ssh connection.
1639 @param opts Additional options to the ssh command.
1640 @param hosts_file Ignored.
1641 @param connect_timeout Ignored.
1642 @param alive_interval Ignored.
1643 """
Dean Liao4bd3e312017-11-13 17:32:17 +08001644 base_command = ('/usr/bin/ssh -a -x %s %s %s'
1645 ' -o StrictHostKeyChecking=no'
1646 ' -o UserKnownHostsFile=/dev/null -o BatchMode=yes'
1647 ' -o ConnectTimeout=30 -o ServerAliveInterval=900'
1648 ' -o ServerAliveCountMax=3 -o ConnectionAttempts=4'
1649 ' -o Protocol=2 -l %s -p %d')
1650 return base_command % (self._ssh_verbosity_flag, self._ssh_options,
1651 opts, user, port)
Jason Abeleb6f924f2013-11-13 16:01:54 -08001652 def syslog(self, message, tag='autotest'):
1653 """Logs a message to syslog on host.
1654
1655 @param message String message to log into syslog
1656 @param tag String tag prefix for syslog
1657
1658 """
1659 self.run('logger -t "%s" "%s"' % (tag, message))
1660
1661
J. Richard Barnetteb6de7e32013-02-14 13:28:04 -08001662 def _ping_check_status(self, status):
1663 """Ping the host once, and return whether it has a given status.
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001664
J. Richard Barnetteb6de7e32013-02-14 13:28:04 -08001665 @param status Check the ping status against this value.
1666 @return True iff `status` and the result of ping are the same
1667 (i.e. both True or both False).
1668
1669 """
1670 ping_val = utils.ping(self.hostname, tries=1, deadline=1)
1671 return not (status ^ (ping_val == 0))
1672
1673 def _ping_wait_for_status(self, status, timeout):
1674 """Wait for the host to have a given status (UP or DOWN).
1675
1676 Status is checked by polling. Polling will not last longer
1677 than the number of seconds in `timeout`. The polling
1678 interval will be long enough that only approximately
1679 _PING_WAIT_COUNT polling cycles will be executed, subject
1680 to a maximum interval of about one minute.
1681
1682 @param status Waiting will stop immediately if `ping` of the
1683 host returns this status.
1684 @param timeout Poll for at most this many seconds.
1685 @return True iff the host status from `ping` matched the
1686 requested status at the time of return.
1687
1688 """
1689 # _ping_check_status() takes about 1 second, hence the
1690 # "- 1" in the formula below.
Nathan Ciobanu38480a32016-10-25 15:26:45 -07001691 # FIXME: if the ping command errors then _ping_check_status()
1692 # returns instantly. If timeout is also smaller than twice
1693 # _PING_WAIT_COUNT then the while loop below forks many
1694 # thousands of ping commands (see /tmp/test_that_results_XXXXX/
1695 # /results-1-logging_YYY.ZZZ/debug/autoserv.DEBUG) and hogs one
1696 # CPU core for 60 seconds.
J. Richard Barnetteb6de7e32013-02-14 13:28:04 -08001697 poll_interval = min(int(timeout / self._PING_WAIT_COUNT), 60) - 1
1698 end_time = time.time() + timeout
1699 while time.time() <= end_time:
1700 if self._ping_check_status(status):
1701 return True
1702 if poll_interval > 0:
1703 time.sleep(poll_interval)
1704
1705 # The last thing we did was sleep(poll_interval), so it may
1706 # have been too long since the last `ping`. Check one more
1707 # time, just to be sure.
1708 return self._ping_check_status(status)
1709
1710 def ping_wait_up(self, timeout):
1711 """Wait for the host to respond to `ping`.
1712
1713 N.B. This method is not a reliable substitute for
1714 `wait_up()`, because a host that responds to ping will not
1715 necessarily respond to ssh. This method should only be used
1716 if the target DUT can be considered functional even if it
1717 can't be reached via ssh.
1718
1719 @param timeout Minimum time to allow before declaring the
1720 host to be non-responsive.
1721 @return True iff the host answered to ping before the timeout.
1722
1723 """
1724 return self._ping_wait_for_status(self._PING_STATUS_UP, timeout)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001725
Andrew Bresticker678c0c72013-01-22 10:44:09 -08001726 def ping_wait_down(self, timeout):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001727 """Wait until the host no longer responds to `ping`.
1728
J. Richard Barnetteb6de7e32013-02-14 13:28:04 -08001729 This function can be used as a slightly faster version of
1730 `wait_down()`, by avoiding potentially long ssh timeouts.
1731
1732 @param timeout Minimum time to allow for the host to become
1733 non-responsive.
1734 @return True iff the host quit answering ping before the
1735 timeout.
1736
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001737 """
J. Richard Barnetteb6de7e32013-02-14 13:28:04 -08001738 return self._ping_wait_for_status(self._PING_STATUS_DOWN, timeout)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001739
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08001740 def test_wait_for_sleep(self, sleep_timeout=None):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001741 """Wait for the client to enter low-power sleep mode.
1742
1743 The test for "is asleep" can't distinguish a system that is
1744 powered off; to confirm that the unit was asleep, it is
1745 necessary to force resume, and then call
1746 `test_wait_for_resume()`.
1747
1748 This function is expected to be called from a test as part
1749 of a sequence like the following:
1750
1751 ~~~~~~~~
1752 boot_id = host.get_boot_id()
1753 # trigger sleep on the host
1754 host.test_wait_for_sleep()
1755 # trigger resume on the host
1756 host.test_wait_for_resume(boot_id)
1757 ~~~~~~~~
1758
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08001759 @param sleep_timeout time limit in seconds to allow the host sleep.
1760
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001761 @exception TestFail The host did not go to sleep within
1762 the allowed time.
1763 """
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08001764 if sleep_timeout is None:
1765 sleep_timeout = self.SLEEP_TIMEOUT
1766
1767 if not self.ping_wait_down(timeout=sleep_timeout):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001768 raise error.TestFail(
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08001769 'client failed to sleep after %d seconds' % sleep_timeout)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001770
1771
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08001772 def test_wait_for_resume(self, old_boot_id, resume_timeout=None):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001773 """Wait for the client to resume from low-power sleep mode.
1774
1775 The `old_boot_id` parameter should be the value from
1776 `get_boot_id()` obtained prior to entering sleep mode. A
1777 `TestFail` exception is raised if the boot id changes.
1778
1779 See @ref test_wait_for_sleep for more on this function's
1780 usage.
1781
J. Richard Barnette7214e0b2013-02-06 15:20:49 -08001782 @param old_boot_id A boot id value obtained before the
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001783 target host went to sleep.
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08001784 @param resume_timeout time limit in seconds to allow the host up.
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001785
1786 @exception TestFail The host did not respond within the
1787 allowed time.
1788 @exception TestFail The host responded, but the boot id test
1789 indicated a reboot rather than a sleep
1790 cycle.
1791 """
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08001792 if resume_timeout is None:
1793 resume_timeout = self.RESUME_TIMEOUT
1794
1795 if not self.wait_up(timeout=resume_timeout):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001796 raise error.TestFail(
1797 'client failed to resume from sleep after %d seconds' %
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08001798 resume_timeout)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001799 else:
1800 new_boot_id = self.get_boot_id()
1801 if new_boot_id != old_boot_id:
Tom Wai-Hong Tam01792682015-01-06 08:00:46 +08001802 logging.error('client rebooted (old boot %s, new boot %s)',
1803 old_boot_id, new_boot_id)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001804 raise error.TestFail(
Tom Wai-Hong Tam01792682015-01-06 08:00:46 +08001805 'client rebooted, but sleep was expected')
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001806
1807
Tom Wai-Hong Tamfe005c22014-12-03 09:25:44 +08001808 def test_wait_for_shutdown(self, shutdown_timeout=None):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001809 """Wait for the client to shut down.
1810
1811 The test for "has shut down" can't distinguish a system that
1812 is merely asleep; to confirm that the unit was down, it is
1813 necessary to force boot, and then call test_wait_for_boot().
1814
1815 This function is expected to be called from a test as part
1816 of a sequence like the following:
1817
1818 ~~~~~~~~
1819 boot_id = host.get_boot_id()
1820 # trigger shutdown on the host
1821 host.test_wait_for_shutdown()
1822 # trigger boot on the host
1823 host.test_wait_for_boot(boot_id)
1824 ~~~~~~~~
1825
Tom Wai-Hong Tamfe005c22014-12-03 09:25:44 +08001826 @param shutdown_timeout time limit in seconds to allow the host down.
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001827 @exception TestFail The host did not shut down within the
1828 allowed time.
1829 """
Tom Wai-Hong Tamfe005c22014-12-03 09:25:44 +08001830 if shutdown_timeout is None:
1831 shutdown_timeout = self.SHUTDOWN_TIMEOUT
1832
1833 if not self.ping_wait_down(timeout=shutdown_timeout):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001834 raise error.TestFail(
1835 'client failed to shut down after %d seconds' %
Tom Wai-Hong Tamfe005c22014-12-03 09:25:44 +08001836 shutdown_timeout)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001837
1838
1839 def test_wait_for_boot(self, old_boot_id=None):
1840 """Wait for the client to boot from cold power.
1841
1842 The `old_boot_id` parameter should be the value from
1843 `get_boot_id()` obtained prior to shutting down. A
1844 `TestFail` exception is raised if the boot id does not
1845 change. The boot id test is omitted if `old_boot_id` is not
1846 specified.
1847
1848 See @ref test_wait_for_shutdown for more on this function's
1849 usage.
1850
J. Richard Barnette7214e0b2013-02-06 15:20:49 -08001851 @param old_boot_id A boot id value obtained before the
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001852 shut down.
1853
1854 @exception TestFail The host did not respond within the
1855 allowed time.
1856 @exception TestFail The host responded, but the boot id test
1857 indicated that there was no reboot.
1858 """
J. Richard Barnetteeb69d722012-06-18 17:29:44 -07001859 if not self.wait_up(timeout=self.REBOOT_TIMEOUT):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001860 raise error.TestFail(
1861 'client failed to reboot after %d seconds' %
J. Richard Barnetteeb69d722012-06-18 17:29:44 -07001862 self.REBOOT_TIMEOUT)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001863 elif old_boot_id:
1864 if self.get_boot_id() == old_boot_id:
Tom Wai-Hong Tam01792682015-01-06 08:00:46 +08001865 logging.error('client not rebooted (boot %s)',
1866 old_boot_id)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001867 raise error.TestFail(
Tom Wai-Hong Tam01792682015-01-06 08:00:46 +08001868 'client is back up, but did not reboot')
Simran Basid5e5e272012-09-24 15:23:59 -07001869
1870
1871 @staticmethod
1872 def check_for_rpm_support(hostname):
1873 """For a given hostname, return whether or not it is powered by an RPM.
1874
Simran Basi1df55112013-09-06 11:25:09 -07001875 @param hostname: hostname to check for rpm support.
1876
Simran Basid5e5e272012-09-24 15:23:59 -07001877 @return None if this host does not follows the defined naming format
1878 for RPM powered DUT's in the lab. If it does follow the format,
1879 it returns a regular expression MatchObject instead.
1880 """
Fang Dengbaff9082015-01-06 13:46:15 -08001881 return re.match(CrosHost._RPM_HOSTNAME_REGEX, hostname)
Simran Basid5e5e272012-09-24 15:23:59 -07001882
1883
1884 def has_power(self):
1885 """For this host, return whether or not it is powered by an RPM.
1886
1887 @return True if this host is in the CROS lab and follows the defined
1888 naming format.
1889 """
Fang Deng0ca40e22013-08-27 17:47:44 -07001890 return CrosHost.check_for_rpm_support(self.hostname)
Simran Basid5e5e272012-09-24 15:23:59 -07001891
1892
Ismail Noorbasha07fdb612013-02-14 14:13:31 -08001893 def _set_power(self, state, power_method):
1894 """Sets the power to the host via RPM, Servo or manual.
1895
1896 @param state Specifies which power state to set to DUT
1897 @param power_method Specifies which method of power control to
1898 use. By default "RPM" will be used. Valid values
1899 are the strings "RPM", "manual", "servoj10".
1900
1901 """
1902 ACCEPTABLE_STATES = ['ON', 'OFF']
1903
1904 if state.upper() not in ACCEPTABLE_STATES:
1905 raise error.TestError('State must be one of: %s.'
1906 % (ACCEPTABLE_STATES,))
1907
1908 if power_method == self.POWER_CONTROL_SERVO:
1909 logging.info('Setting servo port J10 to %s', state)
1910 self.servo.set('prtctl3_pwren', state.lower())
1911 time.sleep(self._USB_POWER_TIMEOUT)
1912 elif power_method == self.POWER_CONTROL_MANUAL:
1913 logging.info('You have %d seconds to set the AC power to %s.',
1914 self._POWER_CYCLE_TIMEOUT, state)
1915 time.sleep(self._POWER_CYCLE_TIMEOUT)
1916 else:
1917 if not self.has_power():
1918 raise error.TestFail('DUT does not have RPM connected.')
Simran Basi5e6339a2013-03-21 11:34:32 -07001919 afe = frontend_wrappers.RetryingAFE(timeout_min=5, delay_sec=10)
1920 afe.set_host_attribute(self._RPM_OUTLET_CHANGED, True,
1921 hostname=self.hostname)
Simran Basi1df55112013-09-06 11:25:09 -07001922 rpm_client.set_power(self.hostname, state.upper(), timeout_mins=5)
Simran Basid5e5e272012-09-24 15:23:59 -07001923
1924
Ismail Noorbasha07fdb612013-02-14 14:13:31 -08001925 def power_off(self, power_method=POWER_CONTROL_RPM):
1926 """Turn off power to this host via RPM, Servo or manual.
1927
1928 @param power_method Specifies which method of power control to
1929 use. By default "RPM" will be used. Valid values
1930 are the strings "RPM", "manual", "servoj10".
1931
1932 """
1933 self._set_power('OFF', power_method)
Simran Basid5e5e272012-09-24 15:23:59 -07001934
1935
Ismail Noorbasha07fdb612013-02-14 14:13:31 -08001936 def power_on(self, power_method=POWER_CONTROL_RPM):
1937 """Turn on power to this host via RPM, Servo or manual.
1938
1939 @param power_method Specifies which method of power control to
1940 use. By default "RPM" will be used. Valid values
1941 are the strings "RPM", "manual", "servoj10".
1942
1943 """
1944 self._set_power('ON', power_method)
1945
1946
1947 def power_cycle(self, power_method=POWER_CONTROL_RPM):
1948 """Cycle power to this host by turning it OFF, then ON.
1949
1950 @param power_method Specifies which method of power control to
1951 use. By default "RPM" will be used. Valid values
1952 are the strings "RPM", "manual", "servoj10".
1953
1954 """
1955 if power_method in (self.POWER_CONTROL_SERVO,
1956 self.POWER_CONTROL_MANUAL):
1957 self.power_off(power_method=power_method)
1958 time.sleep(self._POWER_CYCLE_TIMEOUT)
1959 self.power_on(power_method=power_method)
1960 else:
1961 rpm_client.set_power(self.hostname, 'CYCLE')
Simran Basic6f1f7a2012-10-16 10:47:46 -07001962
1963
1964 def get_platform(self):
1965 """Determine the correct platform label for this host.
1966
1967 @returns a string representing this host's platform.
1968 """
C Shapiro218e8752017-09-22 11:10:57 -06001969 cmd = 'mosys platform model'
1970 result = self.run(command=cmd, ignore_status=True)
1971 if result.exit_status == 0:
1972 return result.stdout.strip()
1973 else:
1974 # $(mosys platform model) should support all platforms, but it
1975 # currently doesn't, so this reverts to parsing the fw
1976 # for any unsupported mosys platforms.
1977 crossystem = utils.Crossystem(self)
1978 crossystem.init()
1979 # Extract fwid value and use the leading part as the platform id.
1980 # fwid generally follow the format of {platform}.{firmware version}
1981 # Example: Alex.X.YYY.Z or Google_Alex.X.YYY.Z
1982 platform = crossystem.fwid().split('.')[0].lower()
1983 # Newer platforms start with 'Google_' while the older ones do not.
1984 return platform.replace('google_', '')
Simran Basic6f1f7a2012-10-16 10:47:46 -07001985
1986
Hung-ying Tyanb1328032014-04-01 14:18:54 +08001987 def get_architecture(self):
1988 """Determine the correct architecture label for this host.
1989
1990 @returns a string representing this host's architecture.
1991 """
1992 crossystem = utils.Crossystem(self)
1993 crossystem.init()
1994 return crossystem.arch()
1995
1996
Luis Lozano40b7d0d2014-01-17 15:12:06 -08001997 def get_chrome_version(self):
1998 """Gets the Chrome version number and milestone as strings.
1999
2000 Invokes "chrome --version" to get the version number and milestone.
2001
2002 @return A tuple (chrome_ver, milestone) where "chrome_ver" is the
2003 current Chrome version number as a string (in the form "W.X.Y.Z")
2004 and "milestone" is the first component of the version number
2005 (the "W" from "W.X.Y.Z"). If the version number cannot be parsed
2006 in the "W.X.Y.Z" format, the "chrome_ver" will be the full output
2007 of "chrome --version" and the milestone will be the empty string.
2008
2009 """
MK Ryu35d661e2014-09-25 17:44:10 -07002010 version_string = self.run(client_constants.CHROME_VERSION_COMMAND).stdout
Luis Lozano40b7d0d2014-01-17 15:12:06 -08002011 return utils.parse_chrome_version(version_string)
2012
J. Richard Barnetted2af5852016-02-05 15:03:10 -08002013
Niranjan Kumar34618872017-05-31 12:57:09 -07002014 def is_chrome_switch_present(self, switch):
David Haddock3ce538e2017-06-22 13:37:05 -07002015 """Returns True if the specified switch was provided to Chrome.
2016
2017 @param switch The chrome switch to search for.
2018 """
Niranjan Kumar34618872017-05-31 12:57:09 -07002019
Niranjan Kumar5f23fe92017-06-22 15:18:55 -07002020 command = 'pgrep -x -f -c "/opt/google/chrome/chrome.*%s.*"' % switch
2021 return self.run(command, ignore_status=True).exit_status == 0
Niranjan Kumar34618872017-05-31 12:57:09 -07002022
2023
2024 def oobe_triggers_update(self):
2025 """Returns True if this host has an OOBE flow during which
2026 it will perform an update check and perhaps an update.
2027 One example of such a flow is Hands-Off Zero-Touch Enrollment.
2028 As more such flows are developed, code handling them needs
2029 to be added here.
2030
2031 @return Boolean indicating whether this host's OOBE triggers an update.
2032 """
2033 return self.is_chrome_switch_present(
2034 '--enterprise-enable-zero-touch-enrollment=hands-off')
2035
2036
Kevin Chenga2619dc2016-03-28 11:42:08 -07002037 # TODO(kevcheng): change this to just return the board without the
2038 # 'board:' prefix and fix up all the callers. Also look into removing the
2039 # need for this method.
Simran Basic6f1f7a2012-10-16 10:47:46 -07002040 def get_board(self):
2041 """Determine the correct board label for this host.
2042
2043 @returns a string representing this host's board.
2044 """
2045 release_info = utils.parse_cmd_output('cat /etc/lsb-release',
2046 run_method=self.run)
J. Richard Barnetted2af5852016-02-05 15:03:10 -08002047 return (ds_constants.BOARD_PREFIX +
2048 release_info['CHROMEOS_RELEASE_BOARD'])
Simran Basic6f1f7a2012-10-16 10:47:46 -07002049
Po-Hsien Wang4618adf2017-10-10 14:40:21 -07002050 def get_channel(self):
2051 """Determine the correct channel label for this host.
Simran Basic6f1f7a2012-10-16 10:47:46 -07002052
Po-Hsien Wang4618adf2017-10-10 14:40:21 -07002053 @returns: a string represeting this host's build channel.
2054 (stable, dev, beta). None on fail.
2055 """
2056 return lsbrelease_utils.get_chromeos_channel(
2057 lsb_release_content=self._get_lsb_release_content())
Kevin Chenga328da62016-03-31 10:49:04 -07002058
2059 def has_lightsensor(self):
2060 """Determine the correct board label for this host.
2061
2062 @returns the string 'lightsensor' if this host has a lightsensor or
2063 None if it does not.
2064 """
2065 search_cmd = "find -L %s -maxdepth 4 | egrep '%s'" % (
2066 self._LIGHTSENSOR_SEARCH_DIR, '|'.join(self._LIGHTSENSOR_FILES))
2067 try:
2068 # Run the search cmd following the symlinks. Stderr_tee is set to
2069 # None as there can be a symlink loop, but the command will still
2070 # execute correctly with a few messages printed to stderr.
2071 self.run(search_cmd, stdout_tee=None, stderr_tee=None)
2072 return 'lightsensor'
2073 except error.AutoservRunError:
2074 # egrep exited with a return code of 1 meaning none of the possible
2075 # lightsensor files existed.
2076 return None
2077
2078
2079 def has_bluetooth(self):
2080 """Determine the correct board label for this host.
2081
2082 @returns the string 'bluetooth' if this host has bluetooth or
2083 None if it does not.
2084 """
2085 try:
2086 self.run('test -d /sys/class/bluetooth/hci0')
2087 # test exited with a return code of 0.
2088 return 'bluetooth'
2089 except error.AutoservRunError:
2090 # test exited with a return code 1 meaning the directory did not
2091 # exist.
2092 return None
2093
2094
Kevin Chenga328da62016-03-31 10:49:04 -07002095 def get_accels(self):
2096 """
2097 Determine the type of accelerometers on this host.
2098
2099 @returns a string representing this host's accelerometer type.
2100 At present, it only returns "accel:cros-ec", for accelerometers
2101 attached to a Chrome OS EC, or none, if no accelerometers.
2102 """
2103 # Check to make sure we have ectool
2104 rv = self.run('which ectool', ignore_status=True)
2105 if rv.exit_status:
2106 logging.info("No ectool cmd found, assuming no EC accelerometers")
2107 return None
2108
2109 # Check that the EC supports the motionsense command
2110 rv = self.run('ectool motionsense', ignore_status=True)
2111 if rv.exit_status:
2112 logging.info("EC does not support motionsense command "
2113 "assuming no EC accelerometers")
2114 return None
2115
2116 # Check that EC motion sensors are active
2117 active = self.run('ectool motionsense active').stdout.split('\n')
2118 if active[0] == "0":
2119 logging.info("Motion sense inactive, assuming no EC accelerometers")
2120 return None
2121
2122 logging.info("EC accelerometers found")
2123 return 'accel:cros-ec'
2124
2125
2126 def has_chameleon(self):
2127 """Determine if a Chameleon connected to this host.
2128
2129 @returns a list containing two strings ('chameleon' and
2130 'chameleon:' + label, e.g. 'chameleon:hdmi') if this host
2131 has a Chameleon or None if it has not.
2132 """
2133 if self._chameleon_host:
2134 return ['chameleon', 'chameleon:' + self.chameleon.get_label()]
2135 else:
2136 return None
2137
2138
2139 def has_loopback_dongle(self):
2140 """Determine if an audio loopback dongle is plugged to this host.
2141
2142 @returns 'audio_loopback_dongle' when there is an audio loopback dongle
2143 plugged to this host.
2144 None when there is no audio loopback dongle
2145 plugged to this host.
2146 """
2147 nodes_info = self.run(command=cras_utils.get_cras_nodes_cmd(),
2148 ignore_status=True).stdout
2149 if (cras_utils.node_type_is_plugged('HEADPHONE', nodes_info) and
2150 cras_utils.node_type_is_plugged('MIC', nodes_info)):
2151 return 'audio_loopback_dongle'
2152 else:
2153 return None
2154
2155
2156 def get_power_supply(self):
2157 """
2158 Determine what type of power supply the host has
2159
2160 @returns a string representing this host's power supply.
2161 'power:battery' when the device has a battery intended for
2162 extended use
2163 'power:AC_primary' when the device has a battery not intended
2164 for extended use (for moving the machine, etc)
2165 'power:AC_only' when the device has no battery at all.
2166 """
2167 psu = self.run(command='mosys psu type', ignore_status=True)
2168 if psu.exit_status:
2169 # The psu command for mosys is not included for all platforms. The
2170 # assumption is that the device will have a battery if the command
2171 # is not found.
2172 return 'power:battery'
2173
2174 psu_str = psu.stdout.strip()
2175 if psu_str == 'unknown':
2176 return None
2177
2178 return 'power:%s' % psu_str
2179
2180
2181 def get_storage(self):
2182 """
2183 Determine the type of boot device for this host.
2184
2185 Determine if the internal device is SCSI or dw_mmc device.
2186 Then check that it is SSD or HDD or eMMC or something else.
2187
2188 @returns a string representing this host's internal device type.
2189 'storage:ssd' when internal device is solid state drive
2190 'storage:hdd' when internal device is hard disk drive
2191 'storage:mmc' when internal device is mmc drive
2192 None When internal device is something else or
2193 when we are unable to determine the type
2194 """
2195 # The output should be /dev/mmcblk* for SD/eMMC or /dev/sd* for scsi
2196 rootdev_cmd = ' '.join(['. /usr/sbin/write_gpt.sh;',
2197 '. /usr/share/misc/chromeos-common.sh;',
2198 'load_base_vars;',
2199 'get_fixed_dst_drive'])
2200 rootdev = self.run(command=rootdev_cmd, ignore_status=True)
2201 if rootdev.exit_status:
2202 logging.info("Fail to run %s", rootdev_cmd)
2203 return None
2204 rootdev_str = rootdev.stdout.strip()
2205
2206 if not rootdev_str:
2207 return None
2208
2209 rootdev_base = os.path.basename(rootdev_str)
2210
2211 mmc_pattern = '/dev/mmcblk[0-9]'
2212 if re.match(mmc_pattern, rootdev_str):
2213 # Use type to determine if the internal device is eMMC or somthing
2214 # else. We can assume that MMC is always an internal device.
2215 type_cmd = 'cat /sys/block/%s/device/type' % rootdev_base
2216 type = self.run(command=type_cmd, ignore_status=True)
2217 if type.exit_status:
2218 logging.info("Fail to run %s", type_cmd)
2219 return None
2220 type_str = type.stdout.strip()
2221
2222 if type_str == 'MMC':
2223 return 'storage:mmc'
2224
2225 scsi_pattern = '/dev/sd[a-z]+'
2226 if re.match(scsi_pattern, rootdev.stdout):
2227 # Read symlink for /sys/block/sd* to determine if the internal
2228 # device is connected via ata or usb.
2229 link_cmd = 'readlink /sys/block/%s' % rootdev_base
2230 link = self.run(command=link_cmd, ignore_status=True)
2231 if link.exit_status:
2232 logging.info("Fail to run %s", link_cmd)
2233 return None
2234 link_str = link.stdout.strip()
2235 if 'usb' in link_str:
2236 return None
2237
2238 # Read rotation to determine if the internal device is ssd or hdd.
2239 rotate_cmd = str('cat /sys/block/%s/queue/rotational'
2240 % rootdev_base)
2241 rotate = self.run(command=rotate_cmd, ignore_status=True)
2242 if rotate.exit_status:
2243 logging.info("Fail to run %s", rotate_cmd)
2244 return None
2245 rotate_str = rotate.stdout.strip()
2246
2247 rotate_dict = {'0':'storage:ssd', '1':'storage:hdd'}
2248 return rotate_dict.get(rotate_str)
2249
2250 # All other internal device / error case will always fall here
2251 return None
2252
2253
2254 def get_servo(self):
2255 """Determine if the host has a servo attached.
2256
2257 If the host has a working servo attached, it should have a servo label.
2258
2259 @return: string 'servo' if the host has servo attached. Otherwise,
2260 returns None.
2261 """
2262 return 'servo' if self._servo_host else None
2263
2264
2265 def get_video_labels(self):
2266 """Run /usr/local/bin/avtest_label_detect to get a list of video labels.
2267
2268 Sample output of avtest_label_detect:
2269 Detected label: hw_video_acc_vp8
2270 Detected label: webcam
2271
2272 @return: A list of labels detected by tool avtest_label_detect.
2273 """
2274 try:
2275 result = self.run('/usr/local/bin/avtest_label_detect').stdout
2276 return re.findall('^Detected label: (\w+)$', result, re.M)
2277 except error.AutoservRunError:
2278 # The tool is not installed.
2279 return []
2280
2281
2282 def is_video_glitch_detection_supported(self):
2283 """ Determine if a board under test is supported for video glitch
2284 detection tests.
2285
2286 @return: 'video_glitch_detection' if board is supported, None otherwise.
2287 """
2288 board = self.get_board().replace(ds_constants.BOARD_PREFIX, '')
2289
2290 if board in video_test_constants.SUPPORTED_BOARDS:
2291 return 'video_glitch_detection'
2292
2293 return None
2294
2295
2296 def get_touch(self):
2297 """
2298 Determine whether board under test has a touchpad or touchscreen.
2299
2300 @return: A list of some combination of 'touchscreen' and 'touchpad',
2301 depending on what is present on the device.
2302
2303 """
2304 labels = []
2305 looking_for = ['touchpad', 'touchscreen']
2306 player = input_playback.InputPlayback()
2307 input_events = self.run('ls /dev/input/event*').stdout.strip().split()
2308 filename = '/tmp/touch_labels'
2309 for event in input_events:
2310 self.run('evtest %s > %s' % (event, filename), timeout=1,
2311 ignore_timeout=True)
2312 properties = self.run('cat %s' % filename).stdout
2313 input_type = player._determine_input_type(properties)
2314 if input_type in looking_for:
2315 labels.append(input_type)
2316 looking_for.remove(input_type)
2317 if len(looking_for) == 0:
2318 break
2319 self.run('rm %s' % filename)
2320
2321 return labels
2322
2323
2324 def has_internal_display(self):
2325 """Determine if the device under test is equipped with an internal
2326 display.
2327
2328 @return: 'internal_display' if one is present; None otherwise.
2329 """
2330 from autotest_lib.client.cros.graphics import graphics_utils
2331 from autotest_lib.client.common_lib import utils as common_utils
2332
2333 def __system_output(cmd):
2334 return self.run(cmd).stdout
2335
2336 def __read_file(remote_path):
2337 return self.run('cat %s' % remote_path).stdout
2338
2339 # Hijack the necessary client functions so that we can take advantage
2340 # of the client lib here.
2341 # FIXME: find a less hacky way than this
2342 original_system_output = utils.system_output
2343 original_read_file = common_utils.read_file
2344 utils.system_output = __system_output
2345 common_utils.read_file = __read_file
2346 try:
2347 return ('internal_display' if graphics_utils.has_internal_display()
2348 else None)
2349 finally:
2350 utils.system_output = original_system_output
2351 common_utils.read_file = original_read_file
2352
2353
Dan Shi85276d42014-04-08 22:11:45 -07002354 def is_boot_from_usb(self):
2355 """Check if DUT is boot from USB.
2356
2357 @return: True if DUT is boot from usb.
2358 """
2359 device = self.run('rootdev -s -d').stdout.strip()
2360 removable = int(self.run('cat /sys/block/%s/removable' %
2361 os.path.basename(device)).stdout.strip())
2362 return removable == 1
Helen Zhang17dae2b2014-11-11 09:25:52 -08002363
2364
2365 def read_from_meminfo(self, key):
Dan Shi49ca0932014-11-14 11:22:27 -08002366 """Return the memory info from /proc/meminfo
Helen Zhang17dae2b2014-11-11 09:25:52 -08002367
2368 @param key: meminfo requested
2369
2370 @return the memory value as a string
2371
2372 """
Helen Zhang17dae2b2014-11-11 09:25:52 -08002373 meminfo = self.run('grep %s /proc/meminfo' % key).stdout.strip()
2374 logging.debug('%s', meminfo)
2375 return int(re.search(r'\d+', meminfo).group(0))
Rohit Makasana8a4923c2015-08-13 17:04:26 -07002376
2377
Rohit Makasana98e696f2016-06-03 18:48:10 -07002378 def get_cpu_arch(self):
2379 """Returns CPU arch of the device.
2380
2381 @return CPU architecture of the DUT.
2382 """
Allen Li2c32d6b2017-02-03 15:28:10 -08002383 # Add CPUs by following logic in client/bin/utils.py.
Rohit Makasana98e696f2016-06-03 18:48:10 -07002384 if self.run("grep '^flags.*:.* lm .*' /proc/cpuinfo",
2385 ignore_status=True).stdout:
2386 return 'x86_64'
2387 if self.run("grep -Ei 'ARM|CPU implementer' /proc/cpuinfo",
2388 ignore_status=True).stdout:
2389 return 'arm'
2390 return 'i386'
2391
2392
Rohit Makasana8a4923c2015-08-13 17:04:26 -07002393 def get_board_type(self):
2394 """
2395 Get the DUT's device type from /etc/lsb-release.
Danny Chan471a8d12015-08-18 14:57:41 -07002396 DEVICETYPE can be one of CHROMEBOX, CHROMEBASE, CHROMEBOOK or more.
2397
2398 @return value of DEVICETYPE param from lsb-release.
Rohit Makasana8a4923c2015-08-13 17:04:26 -07002399 """
Danny Chan471a8d12015-08-18 14:57:41 -07002400 device_type = self.run('grep DEVICETYPE /etc/lsb-release',
2401 ignore_status=True).stdout
2402 if device_type:
Kalin Stoyanov524310b2015-08-21 16:24:04 -07002403 return device_type.split('=')[-1].strip()
Danny Chan471a8d12015-08-18 14:57:41 -07002404 return ''
Gilad Arnolda76bef02015-09-29 13:55:15 -07002405
2406
Rohit Makasanadf0a3a32017-06-30 13:55:18 -07002407 def get_arc_version(self):
2408 """Return ARC version installed on the DUT.
2409
2410 @returns ARC version as string if the CrOS build has ARC, else None.
2411 """
2412 arc_version = self.run('grep CHROMEOS_ARC_VERSION /etc/lsb-release',
2413 ignore_status=True).stdout
2414 if arc_version:
2415 return arc_version.split('=')[-1].strip()
2416 return None
2417
2418
Gilad Arnolda76bef02015-09-29 13:55:15 -07002419 def get_os_type(self):
2420 return 'cros'
Simran Basia5522a32015-10-06 11:01:24 -07002421
2422
2423 def enable_adb_testing(self):
2424 """Mark this host as an adb tester."""
Dan Shia2872172015-10-31 01:16:51 -07002425 self.run('touch %s' % constants.ANDROID_TESTER_FILEFLAG)
Kevin Chenga2619dc2016-03-28 11:42:08 -07002426
2427
2428 def get_labels(self):
Kevin Chengf8660142016-08-12 10:17:41 -07002429 """Return the detected labels on the host."""
Kevin Chenga2619dc2016-03-28 11:42:08 -07002430 return self.labels.get_labels(self)
Kevin Chengf8660142016-08-12 10:17:41 -07002431
2432
2433 def update_labels(self):
2434 """Update the labels on the host."""
2435 self.labels.update_labels(self)