blob: edbb5eaf5d088b68b1d66ed1147147c279b59db1 [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
Gabe Blackb72f4fb2015-01-20 16:47:13 -080020from autotest_lib.client.common_lib.cros.graphite import autotest_es
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
Kevin Chenga328da62016-03-31 10:49:04 -070058LUCID_SLEEP_BOARDS = ['samus', 'lulu']
59
Dan Shid07ee2e2015-09-24 14:49:25 -070060
beepsc87ff602013-07-31 21:53:00 -070061class FactoryImageCheckerException(error.AutoservError):
62 """Exception raised when an image is a factory image."""
63 pass
64
65
Fang Deng0ca40e22013-08-27 17:47:44 -070066class CrosHost(abstract_ssh.AbstractSSHHost):
J. Richard Barnette45e93de2012-04-11 17:24:15 -070067 """Chromium OS specific subclass of Host."""
68
Simran Basi5ace6f22016-01-06 17:30:44 -080069 VERSION_PREFIX = provision.CROS_VERSION_PREFIX
70
J. Richard Barnette45e93de2012-04-11 17:24:15 -070071 _parser = autoserv_parser.autoserv_parser
Scott Zawalski62bacae2013-03-05 10:40:32 -050072 _AFE = frontend_wrappers.RetryingAFE(timeout_min=5, delay_sec=10)
xixuanec801e32016-08-25 10:20:22 -070073 support_devserver_provision = ENABLE_DEVSERVER_TRIGGER_AUTO_UPDATE
J. Richard Barnette45e93de2012-04-11 17:24:15 -070074
Richard Barnette03a0c132012-11-05 12:40:35 -080075 # Timeout values (in seconds) associated with various Chrome OS
76 # state changes.
J. Richard Barnette134ec2c2012-04-25 12:59:37 -070077 #
Richard Barnette0c73ffc2012-11-19 15:21:18 -080078 # In general, a good rule of thumb is that the timeout can be up
79 # to twice the typical measured value on the slowest platform.
80 # The times here have not necessarily been empirically tested to
81 # meet this criterion.
J. Richard Barnetteeb69d722012-06-18 17:29:44 -070082 #
83 # SLEEP_TIMEOUT: Time to allow for suspend to memory.
Richard Barnette0c73ffc2012-11-19 15:21:18 -080084 # RESUME_TIMEOUT: Time to allow for resume after suspend, plus
85 # time to restart the netwowrk.
J. Richard Barnette84890bd2014-02-21 11:05:47 -080086 # SHUTDOWN_TIMEOUT: Time to allow for shut down.
J. Richard Barnetteeb69d722012-06-18 17:29:44 -070087 # BOOT_TIMEOUT: Time to allow for boot from power off. Among
Richard Barnette0c73ffc2012-11-19 15:21:18 -080088 # other things, this must account for the 30 second dev-mode
J. Richard Barnette417cc792015-10-01 09:56:36 -070089 # screen delay, time to start the network on the DUT, and the
90 # ssh timeout of 120 seconds.
J. Richard Barnetteeb69d722012-06-18 17:29:44 -070091 # USB_BOOT_TIMEOUT: Time to allow for boot from a USB device,
Richard Barnette0c73ffc2012-11-19 15:21:18 -080092 # including the 30 second dev-mode delay and time to start the
J. Richard Barnetted4649c62013-03-06 17:42:27 -080093 # network.
beepsf079cfb2013-09-18 17:49:51 -070094 # INSTALL_TIMEOUT: Time to allow for chromeos-install.
J. Richard Barnette84890bd2014-02-21 11:05:47 -080095 # POWERWASH_BOOT_TIMEOUT: Time to allow for a reboot that
96 # includes powerwash.
J. Richard Barnetteeb69d722012-06-18 17:29:44 -070097
98 SLEEP_TIMEOUT = 2
J. Richard Barnetted4649c62013-03-06 17:42:27 -080099 RESUME_TIMEOUT = 10
Tom Wai-Hong Tamfe005c22014-12-03 09:25:44 +0800100 SHUTDOWN_TIMEOUT = 10
J. Richard Barnette417cc792015-10-01 09:56:36 -0700101 BOOT_TIMEOUT = 150
J. Richard Barnette5bab5f52015-08-03 13:14:38 -0700102 USB_BOOT_TIMEOUT = 300
J. Richard Barnette7817b052014-08-28 09:47:29 -0700103 INSTALL_TIMEOUT = 480
Dan Shi2c88eed2013-11-12 10:18:38 -0800104 POWERWASH_BOOT_TIMEOUT = 60
Chris Sosab76e0ee2013-05-22 16:55:41 -0700105
Dan Shica503482015-03-30 17:23:25 -0700106 # Minimum OS version that supports server side packaging. Older builds may
107 # not have server side package built or with Autotest code change to support
108 # server-side packaging.
Dan Shib8540a52015-07-16 14:18:23 -0700109 MIN_VERSION_SUPPORT_SSP = CONFIG.get_config_value(
Dan Shiced09e42015-04-17 16:09:34 -0700110 'AUTOSERV', 'min_version_support_ssp', type=int)
Dan Shica503482015-03-30 17:23:25 -0700111
J. Richard Barnette84890bd2014-02-21 11:05:47 -0800112 # REBOOT_TIMEOUT: How long to wait for a reboot.
113 #
Chris Sosab76e0ee2013-05-22 16:55:41 -0700114 # We have a long timeout to ensure we don't flakily fail due to other
115 # issues. Shorter timeouts are vetted in platform_RebootAfterUpdate.
Simran Basi1160e2c2013-10-04 16:00:24 -0700116 # TODO(sbasi - crbug.com/276094) Restore to 5 mins once the 'host did not
117 # return from reboot' bug is solved.
118 REBOOT_TIMEOUT = 480
Chris Sosab76e0ee2013-05-22 16:55:41 -0700119
Ismail Noorbasha07fdb612013-02-14 14:13:31 -0800120 # _USB_POWER_TIMEOUT: Time to allow for USB to power toggle ON and OFF.
121 # _POWER_CYCLE_TIMEOUT: Time to allow for manual power cycle.
122 _USB_POWER_TIMEOUT = 5
123 _POWER_CYCLE_TIMEOUT = 10
124
Dan Shib8540a52015-07-16 14:18:23 -0700125 _RPM_RECOVERY_BOARDS = CONFIG.get_config_value('CROS',
Richard Barnette82c35912012-11-20 10:09:10 -0800126 'rpm_recovery_boards', type=str).split(',')
127
Richard Barnette82c35912012-11-20 10:09:10 -0800128 _LAB_MACHINE_FILE = '/mnt/stateful_partition/.labmachine'
Fang Dengdeba14f2014-11-14 11:54:09 -0800129 _RPM_HOSTNAME_REGEX = ('chromeos(\d+)(-row(\d+))?-rack(\d+[a-z]*)'
130 '-host(\d+)')
Kevin Chenga328da62016-03-31 10:49:04 -0700131 _LIGHTSENSOR_FILES = [ "in_illuminance0_input",
132 "in_illuminance_input",
133 "in_illuminance0_raw",
134 "in_illuminance_raw",
135 "illuminance0_input"]
136 _LIGHTSENSOR_SEARCH_DIR = '/sys/bus/iio/devices'
J. Richard Barnette134ec2c2012-04-25 12:59:37 -0700137
J. Richard Barnetteb6de7e32013-02-14 13:28:04 -0800138 # Constants used in ping_wait_up() and ping_wait_down().
139 #
140 # _PING_WAIT_COUNT is the approximate number of polling
141 # cycles to use when waiting for a host state change.
142 #
143 # _PING_STATUS_DOWN and _PING_STATUS_UP are names used
144 # for arguments to the internal _ping_wait_for_status()
145 # method.
146 _PING_WAIT_COUNT = 40
147 _PING_STATUS_DOWN = False
148 _PING_STATUS_UP = True
149
Ismail Noorbasha07fdb612013-02-14 14:13:31 -0800150 # Allowed values for the power_method argument.
151
152 # POWER_CONTROL_RPM: Passed as default arg for power_off/on/cycle() methods.
153 # POWER_CONTROL_SERVO: Used in set_power() and power_cycle() methods.
154 # POWER_CONTROL_MANUAL: Used in set_power() and power_cycle() methods.
155 POWER_CONTROL_RPM = 'RPM'
156 POWER_CONTROL_SERVO = 'servoj10'
157 POWER_CONTROL_MANUAL = 'manual'
158
159 POWER_CONTROL_VALID_ARGS = (POWER_CONTROL_RPM,
160 POWER_CONTROL_SERVO,
161 POWER_CONTROL_MANUAL)
Richard Barnette0c73ffc2012-11-19 15:21:18 -0800162
Simran Basi5e6339a2013-03-21 11:34:32 -0700163 _RPM_OUTLET_CHANGED = 'outlet_changed'
164
Dan Shi9cb0eec2014-06-03 09:04:50 -0700165 # URL pattern to download firmware image.
Dan Shib8540a52015-07-16 14:18:23 -0700166 _FW_IMAGE_URL_PATTERN = CONFIG.get_config_value(
Dan Shi9cb0eec2014-06-03 09:04:50 -0700167 'CROS', 'firmware_url_pattern', type=str)
beeps687243d2013-07-18 15:29:27 -0700168
J. Richard Barnette91137f02016-03-10 16:52:26 -0800169
170 # A flag file to indicate provision failures. The file is created
171 # at the start of any AU procedure (see `machine_install()`). The
172 # file's location in stateful means that on successul update it will
173 # be removed. Thus, if this file exists, it indicates that we've
174 # tried and failed in a previous attempt to update.
175 PROVISION_FAILED = '/var/tmp/provision_failed'
MK Ryu35d661e2014-09-25 17:44:10 -0700176
MK Ryu35d661e2014-09-25 17:44:10 -0700177
J. Richard Barnette964fba02012-10-24 17:34:29 -0700178 @staticmethod
beeps46dadc92013-11-07 14:07:10 -0800179 def check_host(host, timeout=10):
180 """
181 Check if the given host is a chrome-os host.
182
183 @param host: An ssh host representing a device.
184 @param timeout: The timeout for the run command.
185
186 @return: True if the host device is chromeos.
187
beeps46dadc92013-11-07 14:07:10 -0800188 """
189 try:
Allen Liad719c12017-06-27 23:48:04 +0000190 result = host.run(
Simran Basi933c8af2015-04-29 14:05:07 -0700191 'grep -q CHROMEOS /etc/lsb-release && '
192 '! test -f /mnt/stateful_partition/.android_tester && '
193 '! grep -q moblab /etc/lsb-release',
194 ignore_status=True, timeout=timeout)
Laurence Goodby468de252017-06-08 17:22:53 -0700195 if result.exit_status == 0:
Allen Liad719c12017-06-27 23:48:04 +0000196 lsb_release_content = host.run(
Laurence Goodby468de252017-06-08 17:22:53 -0700197 'grep CHROMEOS_RELEASE_BOARD /etc/lsb-release',
198 timeout=timeout).stdout
199 return not lsbrelease_utils.is_jetstream(
200 lsb_release_content=lsb_release_content)
beeps46dadc92013-11-07 14:07:10 -0800201 except (error.AutoservRunError, error.AutoservSSHTimeout):
202 return False
Laurence Goodby468de252017-06-08 17:22:53 -0700203
204 return False
beeps46dadc92013-11-07 14:07:10 -0800205
206
207 @staticmethod
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +0800208 def get_chameleon_arguments(args_dict):
209 """Extract chameleon options from `args_dict` and return the result.
210
211 Recommended usage:
212 ~~~~~~~~
213 args_dict = utils.args_to_dict(args)
214 chameleon_args = hosts.CrosHost.get_chameleon_arguments(args_dict)
215 host = hosts.create_host(machine, chameleon_args=chameleon_args)
216 ~~~~~~~~
217
218 @param args_dict Dictionary from which to extract the chameleon
219 arguments.
220 """
Allen Li083866b2016-08-18 10:07:10 -0700221 return {key: args_dict[key]
222 for key in ('chameleon_host', 'chameleon_port')
223 if key in args_dict}
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +0800224
225
226 @staticmethod
Scottfe06ed82015-11-05 17:15:01 -0800227 def get_plankton_arguments(args_dict):
228 """Extract chameleon options from `args_dict` and return the result.
229
230 Recommended usage:
231 ~~~~~~~~
232 args_dict = utils.args_to_dict(args)
Richard Barnettee519dcd2016-08-15 17:37:17 -0700233 plankton_args = hosts.CrosHost.get_plankton_arguments(args_dict)
234 host = hosts.create_host(machine, plankton_args=plankton_args)
Scottfe06ed82015-11-05 17:15:01 -0800235 ~~~~~~~~
236
237 @param args_dict Dictionary from which to extract the plankton
238 arguments.
239 """
Allen Li083866b2016-08-18 10:07:10 -0700240 return {key: args_dict[key]
241 for key in ('plankton_host', 'plankton_port')
242 if key in args_dict}
Scottfe06ed82015-11-05 17:15:01 -0800243
244
245 @staticmethod
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +0800246 def get_servo_arguments(args_dict):
247 """Extract servo options from `args_dict` and return the result.
J. Richard Barnette7214e0b2013-02-06 15:20:49 -0800248
249 Recommended usage:
250 ~~~~~~~~
251 args_dict = utils.args_to_dict(args)
Fang Deng0ca40e22013-08-27 17:47:44 -0700252 servo_args = hosts.CrosHost.get_servo_arguments(args_dict)
J. Richard Barnette7214e0b2013-02-06 15:20:49 -0800253 host = hosts.create_host(machine, servo_args=servo_args)
254 ~~~~~~~~
255
256 @param args_dict Dictionary from which to extract the servo
257 arguments.
258 """
Richard Barnettee519dcd2016-08-15 17:37:17 -0700259 servo_attrs = (servo_host.SERVO_HOST_ATTR,
260 servo_host.SERVO_PORT_ATTR,
261 servo_host.SERVO_BOARD_ATTR)
Allen Li083866b2016-08-18 10:07:10 -0700262 return {key: args_dict[key]
263 for key in servo_attrs
264 if key in args_dict}
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
beepsdae65fd2013-07-26 16:24:41 -0700349 def verify_job_repo_url(self, tag=''):
beepscb6f1e22013-06-28 19:14:10 -0700350 """
351 Make sure job_repo_url of this host is valid.
352
joychen03eaad92013-06-26 09:55:21 -0700353 Eg: The job_repo_url "http://lmn.cd.ab.xyx:8080/static/\
beepscb6f1e22013-06-28 19:14:10 -0700354 lumpy-release/R29-4279.0.0/autotest/packages" claims to have the
355 autotest package for lumpy-release/R29-4279.0.0. If this isn't the case,
356 download and extract it. If the devserver embedded in the url is
357 unresponsive, update the job_repo_url of the host after staging it on
358 another devserver.
359
360 @param job_repo_url: A url pointing to the devserver where the autotest
361 package for this build should be staged.
beepsdae65fd2013-07-26 16:24:41 -0700362 @param tag: The tag from the server job, in the format
363 <job_id>-<user>/<hostname>, or <hostless> for a server job.
beepscb6f1e22013-06-28 19:14:10 -0700364
365 @raises DevServerException: If we could not resolve a devserver.
366 @raises AutoservError: If we're unable to save the new job_repo_url as
367 a result of choosing a new devserver because the old one failed to
368 respond to a health check.
beeps0c865032013-07-30 11:37:06 -0700369 @raises urllib2.URLError: If the devserver embedded in job_repo_url
370 doesn't respond within the timeout.
beepscb6f1e22013-06-28 19:14:10 -0700371 """
Prathmesh Prabhu368abdf2017-02-14 11:23:47 -0800372 info = self.host_info_store.get()
373 job_repo_url = info.attributes.get(ds_constants.JOB_REPO_URL, '')
beepscb6f1e22013-06-28 19:14:10 -0700374 if not job_repo_url:
375 logging.warning('No job repo url set on host %s', self.hostname)
376 return
377
378 logging.info('Verifying job repo url %s', job_repo_url)
379 devserver_url, image_name = tools.get_devserver_build_from_package_url(
380 job_repo_url)
381
beeps0c865032013-07-30 11:37:06 -0700382 ds = dev_server.ImageServer(devserver_url)
beepscb6f1e22013-06-28 19:14:10 -0700383
384 logging.info('Staging autotest artifacts for %s on devserver %s',
385 image_name, ds.url())
beeps687243d2013-07-18 15:29:27 -0700386
387 start_time = time.time()
Simran Basi25e7a922014-10-31 11:56:10 -0700388 ds.stage_artifacts(image_name, ['autotest_packages'])
beeps687243d2013-07-18 15:29:27 -0700389 stage_time = time.time() - start_time
390
391 # Record how much of the verification time comes from a devserver
392 # restage. If we're doing things right we should not see multiple
393 # devservers for a given board/build/branch path.
394 try:
J. Richard Barnette3cbd76b2013-11-27 12:11:25 -0800395 board, build_type, branch = server_utils.ParseBuildName(
beeps687243d2013-07-18 15:29:27 -0700396 image_name)[:3]
J. Richard Barnette3cbd76b2013-11-27 12:11:25 -0800397 except server_utils.ParseBuildNameException:
beeps687243d2013-07-18 15:29:27 -0700398 pass
399 else:
beeps0c865032013-07-30 11:37:06 -0700400 devserver = devserver_url[
Chris Sosa65425082013-10-16 13:26:22 -0700401 devserver_url.find('/') + 2:devserver_url.rfind(':')]
beeps687243d2013-07-18 15:29:27 -0700402 stats_key = {
403 'board': board,
404 'build_type': build_type,
405 'branch': branch,
beeps0c865032013-07-30 11:37:06 -0700406 'devserver': devserver.replace('.', '_'),
beeps687243d2013-07-18 15:29:27 -0700407 }
Dan Shi5e2efb72017-02-07 11:40:23 -0800408
409 monarch_fields = {
410 'board': board,
411 'build_type': build_type,
412 # TODO(akeshet): To be consistent with most other metrics,
413 # consider changing the following field to be named
414 # 'milestone'.
415 'branch': branch,
416 'dev_server': devserver,
417 }
418 metrics.Counter(
419 'chromeos/autotest/provision/verify_url'
420 ).increment(fields=monarch_fields)
421 metrics.SecondsDistribution(
422 'chromeos/autotest/provision/verify_url_duration'
423 ).add(stage_time, fields=monarch_fields)
424
425
Dan Shicf4d2032015-03-12 15:04:21 -0700426 def stage_server_side_package(self, image=None):
427 """Stage autotest server-side package on devserver.
428
429 @param image: Full path of an OS image to install or a build name.
430
431 @return: A url to the autotest server-side package.
Dan Shi14de7622016-08-22 11:09:06 -0700432
433 @raise: error.AutoservError if fail to locate the build to test with, or
434 fail to stage server-side package.
Dan Shicf4d2032015-03-12 15:04:21 -0700435 """
Dan Shid37736b2016-07-06 15:10:29 -0700436 # If enable_drone_in_restricted_subnet is False, do not set hostname
437 # in devserver.resolve call, so a devserver in non-restricted subnet
438 # is picked to stage autotest server package for drone to download.
439 hostname = self.hostname
440 if not server_utils.ENABLE_DRONE_IN_RESTRICTED_SUBNET:
441 hostname = None
Dan Shicf4d2032015-03-12 15:04:21 -0700442 if image:
443 image_name = tools.get_build_from_image(image)
444 if not image_name:
445 raise error.AutoservError(
446 'Failed to parse build name from %s' % image)
Dan Shid37736b2016-07-06 15:10:29 -0700447 ds = dev_server.ImageServer.resolve(image_name, hostname)
Dan Shicf4d2032015-03-12 15:04:21 -0700448 else:
Prathmesh Prabhu9235e4c2017-03-28 13:16:06 -0700449 info = self.host_info_store.get()
Prathmesh Prabhu368abdf2017-02-14 11:23:47 -0800450 job_repo_url = info.attributes.get(ds_constants.JOB_REPO_URL, '')
Dan Shicf4d2032015-03-12 15:04:21 -0700451 if job_repo_url:
452 devserver_url, image_name = (
453 tools.get_devserver_build_from_package_url(job_repo_url))
Dan Shid37736b2016-07-06 15:10:29 -0700454 # If enable_drone_in_restricted_subnet is True, use the
455 # existing devserver. Otherwise, resolve a new one in
456 # non-restricted subnet.
457 if server_utils.ENABLE_DRONE_IN_RESTRICTED_SUBNET:
458 ds = dev_server.ImageServer(devserver_url)
459 else:
460 ds = dev_server.ImageServer.resolve(image_name)
Prathmesh Prabhub6cea612017-02-09 15:41:19 -0800461 elif info.build is not None:
462 ds = dev_server.ImageServer.resolve(info.build, hostname)
Prathmesh Prabhu0c1dd4d2017-06-07 13:01:53 -0700463 image_name = info.build
Dan Shicf4d2032015-03-12 15:04:21 -0700464 else:
Prathmesh Prabhub6cea612017-02-09 15:41:19 -0800465 raise error.AutoservError(
466 'Failed to stage server-side package. The host has '
467 'no job_report_url attribute or version label.')
Dan Shica503482015-03-30 17:23:25 -0700468
469 # Get the OS version of the build, for any build older than
470 # MIN_VERSION_SUPPORT_SSP, server side packaging is not supported.
471 match = re.match('.*/R\d+-(\d+)\.', image_name)
472 if match and int(match.group(1)) < self.MIN_VERSION_SUPPORT_SSP:
Dan Shi14de7622016-08-22 11:09:06 -0700473 raise error.AutoservError(
474 'Build %s is older than %s. Server side packaging is '
475 'disabled.' % (image_name, self.MIN_VERSION_SUPPORT_SSP))
Dan Shica503482015-03-30 17:23:25 -0700476
Dan Shicf4d2032015-03-12 15:04:21 -0700477 ds.stage_artifacts(image_name, ['autotest_server_package'])
478 return '%s/static/%s/%s' % (ds.url(), image_name,
479 'autotest_server_package.tar.bz2')
480
481
Dan Shi0f466e82013-02-22 15:44:58 -0800482 def _try_stateful_update(self, update_url, force_update, updater):
483 """Try to use stateful update to initialize DUT.
484
485 When DUT is already running the same version that machine_install
486 tries to install, stateful update is a much faster way to clean up
487 the DUT for testing, compared to a full reimage. It is implemeted
488 by calling autoupdater.run_update, but skipping updating root, as
489 updating the kernel is time consuming and not necessary.
490
491 @param update_url: url of the image.
492 @param force_update: Set to True to update the image even if the DUT
493 is running the same version.
494 @param updater: ChromiumOSUpdater instance used to update the DUT.
495 @returns: True if the DUT was updated with stateful update.
496
497 """
Laurence Goodby778c9a42017-05-24 19:24:07 -0700498 self.prepare_for_update()
Dan Shi10b98482016-02-02 14:38:50 -0800499
J. Richard Barnette3f731032014-04-07 17:42:59 -0700500 # TODO(jrbarnette): Yes, I hate this re.match() test case.
501 # It's better than the alternative: see crbug.com/360944.
502 image_name = autoupdater.url_to_image_name(update_url)
503 release_pattern = r'^.*-release/R[0-9]+-[0-9]+\.[0-9]+\.0$'
504 if not re.match(release_pattern, image_name):
505 return False
Dan Shi0f466e82013-02-22 15:44:58 -0800506 if not updater.check_version():
507 return False
508 if not force_update:
509 logging.info('Canceling stateful update because the new and '
510 'old versions are the same.')
511 return False
512 # Following folders should be rebuilt after stateful update.
513 # A test file is used to confirm each folder gets rebuilt after
514 # the stateful update.
515 folders_to_check = ['/var', '/home', '/mnt/stateful_partition']
516 test_file = '.test_file_to_be_deleted'
517 for folder in folders_to_check:
518 touch_path = os.path.join(folder, test_file)
519 self.run('touch %s' % touch_path)
520
Chris Sosae92399e2015-04-24 11:32:59 -0700521 updater.run_update(update_root=False)
Dan Shi0f466e82013-02-22 15:44:58 -0800522
523 # Reboot to complete stateful update.
Chris Sosab76e0ee2013-05-22 16:55:41 -0700524 self.reboot(timeout=self.REBOOT_TIMEOUT, wait=True)
xixuan908b0752016-12-08 18:54:23 -0800525
526 # After stateful update and a reboot, all of the test_files shouldn't
527 # exist any more. Otherwise the stateful update is failed.
528 return not any(
529 self.path_exists(os.path.join(folder, test_file))
530 for folder in folders_to_check)
Dan Shi0f466e82013-02-22 15:44:58 -0800531
532
J. Richard Barnette7275b612013-06-04 18:13:11 -0700533 def _post_update_processing(self, updater, expected_kernel=None):
Dan Shi0f466e82013-02-22 15:44:58 -0800534 """After the DUT is updated, confirm machine_install succeeded.
535
536 @param updater: ChromiumOSUpdater instance used to update the DUT.
J. Richard Barnette7275b612013-06-04 18:13:11 -0700537 @param expected_kernel: kernel expected to be active after reboot,
538 or `None` to skip rollback checking.
Dan Shi0f466e82013-02-22 15:44:58 -0800539
540 """
J. Richard Barnette7275b612013-06-04 18:13:11 -0700541 # Touch the lab machine file to leave a marker that
542 # distinguishes this image from other test images.
543 # Afterwards, we must re-run the autoreboot script because
544 # it depends on the _LAB_MACHINE_FILE.
J. Richard Barnette71cc1862015-12-02 10:32:38 -0800545 autoreboot_cmd = ('FILE="%s" ; [ -f "$FILE" ] || '
546 '( touch "$FILE" ; start autoreboot )')
547 self.run(autoreboot_cmd % self._LAB_MACHINE_FILE)
Chris Sosa65425082013-10-16 13:26:22 -0700548 updater.verify_boot_expectations(
549 expected_kernel, rollback_message=
Gilad Arnoldc26ae1f2015-10-22 16:09:41 -0700550 'Build %s failed to boot on %s; system rolled back to previous '
Chris Sosa65425082013-10-16 13:26:22 -0700551 'build' % (updater.update_version, self.hostname))
J. Richard Barnette7275b612013-06-04 18:13:11 -0700552 # Check that we've got the build we meant to install.
553 if not updater.check_version_to_confirm_install():
554 raise autoupdater.ChromiumOSError(
555 'Failed to update %s to build %s; found build '
556 '%s instead' % (self.hostname,
Chris Sosa65425082013-10-16 13:26:22 -0700557 updater.update_version,
Dan Shi0942b1d2015-03-31 11:07:00 -0700558 self.get_release_version()))
Dan Shi0f466e82013-02-22 15:44:58 -0800559
Chris Sosae92399e2015-04-24 11:32:59 -0700560 logging.debug('Cleaning up old autotest directories.')
561 try:
562 installed_autodir = autotest.Autotest.get_installed_autodir(self)
563 self.run('rm -rf ' + installed_autodir)
564 except autotest.AutodirNotFoundError:
565 logging.debug('No autotest installed directory found.')
566
Dan Shi0f466e82013-02-22 15:44:58 -0800567
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700568 def _stage_image_for_update(self, image_name=None):
Chris Sosae92399e2015-04-24 11:32:59 -0700569 """Stage a build on a devserver and return the update_url and devserver.
Scott Zawalskieadbf702013-03-14 09:23:06 -0400570
571 @param image_name: a name like lumpy-release/R27-3837.0.0
Chris Sosae92399e2015-04-24 11:32:59 -0700572 @returns a tuple with an update URL like:
Scott Zawalskieadbf702013-03-14 09:23:06 -0400573 http://172.22.50.205:8082/update/lumpy-release/R27-3837.0.0
Chris Sosae92399e2015-04-24 11:32:59 -0700574 and the devserver instance.
Scott Zawalskieadbf702013-03-14 09:23:06 -0400575 """
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700576 if not image_name:
Richard Barnette260cbd02016-10-06 12:23:28 -0700577 image_name = self._get_cros_repair_image_name()
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700578 logging.info('Staging build for AU: %s', image_name)
Dan Shi216389c2015-12-22 11:03:06 -0800579 devserver = dev_server.ImageServer.resolve(image_name, self.hostname)
Scott Zawalskieadbf702013-03-14 09:23:06 -0400580 devserver.trigger_download(image_name, synchronous=False)
Chris Sosae92399e2015-04-24 11:32:59 -0700581 return (tools.image_url_pattern() % (devserver.url(), image_name),
582 devserver)
Scott Zawalskieadbf702013-03-14 09:23:06 -0400583
584
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700585 def stage_image_for_servo(self, image_name=None):
586 """Stage a build on a devserver and return the update_url.
587
588 @param image_name: a name like lumpy-release/R27-3837.0.0
589 @returns an update URL like:
590 http://172.22.50.205:8082/update/lumpy-release/R27-3837.0.0
591 """
592 if not image_name:
Richard Barnette260cbd02016-10-06 12:23:28 -0700593 image_name = self._get_cros_repair_image_name()
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700594 logging.info('Staging build for servo install: %s', image_name)
Dan Shi216389c2015-12-22 11:03:06 -0800595 devserver = dev_server.ImageServer.resolve(image_name, self.hostname)
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700596 devserver.stage_artifacts(image_name, ['test_image'])
597 return devserver.get_test_image_url(image_name)
598
599
beepse539be02013-07-31 21:57:39 -0700600 def stage_factory_image_for_servo(self, image_name):
601 """Stage a build on a devserver and return the update_url.
602
603 @param image_name: a name like <baord>/4262.204.0
beeps12c0a3c2013-09-03 11:58:27 -0700604
beepse539be02013-07-31 21:57:39 -0700605 @return: An update URL, eg:
606 http://<devserver>/static/canary-channel/\
607 <board>/4262.204.0/factory_test/chromiumos_factory_image.bin
beeps12c0a3c2013-09-03 11:58:27 -0700608
609 @raises: ValueError if the factory artifact name is missing from
610 the config.
611
beepse539be02013-07-31 21:57:39 -0700612 """
613 if not image_name:
614 logging.error('Need an image_name to stage a factory image.')
615 return
616
Dan Shib8540a52015-07-16 14:18:23 -0700617 factory_artifact = CONFIG.get_config_value(
beeps12c0a3c2013-09-03 11:58:27 -0700618 'CROS', 'factory_artifact', type=str, default='')
619 if not factory_artifact:
620 raise ValueError('Cannot retrieve the factory artifact name from '
621 'autotest config, and hence cannot stage factory '
622 'artifacts.')
623
beepse539be02013-07-31 21:57:39 -0700624 logging.info('Staging build for servo install: %s', image_name)
Dan Shi216389c2015-12-22 11:03:06 -0800625 devserver = dev_server.ImageServer.resolve(image_name, self.hostname)
beepse539be02013-07-31 21:57:39 -0700626 devserver.stage_artifacts(
627 image_name,
beeps12c0a3c2013-09-03 11:58:27 -0700628 [factory_artifact],
629 archive_url=None)
beepse539be02013-07-31 21:57:39 -0700630
631 return tools.factory_image_url_pattern() % (devserver.url(), image_name)
632
633
xixuan016d95b2017-02-27 14:37:47 -0800634 def _get_au_monarch_fields(self, devserver, build):
635 """Form monarch fields by given devserve & build for auto-update.
636
637 @param devserver: the devserver (ImageServer instance) for auto-update.
638 @param build: the build to be updated.
639
640 @return A dictionary of monach fields.
641 """
642 try:
643 board, build_type, milestone, _ = server_utils.ParseBuildName(build)
644 except server_utils.ParseBuildNameException:
645 logging.warning('Unable to parse build name %s. Something is '
646 'likely broken, but will continue anyway.',
647 build)
648 board, build_type, milestone = ('', '', '')
649
650 monarch_fields = {
651 'dev_server': devserver.hostname,
652 'board': board,
653 'build_type': build_type,
654 'milestone': milestone
655 }
656 return monarch_fields
657
658
xixuanc30d7e92017-05-11 12:06:39 -0700659 def _retry_auto_update_with_new_devserver(self, build, last_devserver,
xixuana3bbc422017-05-04 15:57:21 -0700660 force_update, force_full_update,
661 force_original):
xixuanc30d7e92017-05-11 12:06:39 -0700662 """Kick off auto-update by devserver and send metrics.
663
664 @param build: the build to update.
665 @param last_devserver: the last devserver that failed to provision.
666 @param force_update: see |machine_install_by_devserver|'s force_udpate
667 for details.
668 @param force_full_update: see |machine_install_by_devserver|'s
669 force_full_update for details.
xixuana3bbc422017-05-04 15:57:21 -0700670 @param force_original: Whether to force stateful update with the
671 original payload.
xixuanc30d7e92017-05-11 12:06:39 -0700672
673 @return the result of |auto_update| in dev_server.
674 """
675 devserver = dev_server.resolve(
676 build, self.hostname, ban_list=[last_devserver.url()])
677 devserver.trigger_download(build, synchronous=False)
678 monarch_fields = self._get_au_monarch_fields(devserver, build)
679 logging.debug('Retry auto_update: resolved devserver for '
680 'auto-update: %s', devserver.url())
681
682 # Add metrics
683 install_with_dev_counter = metrics.Counter(
684 'chromeos/autotest/provision/install_with_devserver')
685 install_with_dev_counter.increment(fields=monarch_fields)
686 c = metrics.Counter(
687 'chromeos/autotest/provision/retry_by_devserver')
688 monarch_fields['last_devserver'] = last_devserver.hostname
689 monarch_fields['host'] = self.hostname
690 c.increment(fields=monarch_fields)
691
xixuan9f455342017-06-08 18:21:44 -0700692 # Won't retry auto_update in a retry of auto-update.
693 # In other words, we only retry auto-update once with a different
694 # devservers.
695 devserver.auto_update(
xixuanc30d7e92017-05-11 12:06:39 -0700696 self.hostname, build,
697 original_board=self.get_board().replace(
698 ds_constants.BOARD_PREFIX, ''),
699 original_release_version=self.get_release_version(),
700 log_dir=self.job.resultdir,
701 force_update=force_update,
xixuana3bbc422017-05-04 15:57:21 -0700702 full_update=force_full_update,
703 force_original=force_original)
xixuanc30d7e92017-05-11 12:06:39 -0700704
705
xixuanec801e32016-08-25 10:20:22 -0700706 def machine_install_by_devserver(self, update_url=None, force_update=False,
xixuan5dc64ea2016-05-20 17:27:51 -0700707 local_devserver=False, repair=False,
708 force_full_update=False):
709 """Ultiize devserver to install the DUT.
710
711 (TODO) crbugs.com/627269: The logic in this function has some overlap
712 with those in function machine_install. The merge will be done later,
713 not in the same CL.
714
xixuanec801e32016-08-25 10:20:22 -0700715 @param update_url: The update_url or build for the host to update.
xixuan5dc64ea2016-05-20 17:27:51 -0700716 @param force_update: Force an update even if the version installed
717 is the same. Default:False
718 @param local_devserver: Used by test_that to allow people to
719 use their local devserver. Default: False
720 @param repair: Forces update to repair image. Implies force_update.
721 @param force_full_update: If True, do not attempt to run stateful
722 update, force a full reimage. If False, try stateful update
723 first when the dut is already installed with the same version.
724 @raises autoupdater.ChromiumOSError
725
726 @returns A tuple of (image_name, host_attributes).
727 image_name is the name of image installed, e.g.,
728 veyron_jerry-release/R50-7871.0.0
729 host_attributes is a dictionary of (attribute, value), which
730 can be saved to afe_host_attributes table in database. This
731 method returns a dictionary with a single entry of
732 `job_repo_url`: repo_url, where repo_url is a devserver url to
733 autotest packages.
734 """
xixuan5dc64ea2016-05-20 17:27:51 -0700735 if repair:
Richard Barnette260cbd02016-10-06 12:23:28 -0700736 update_url = self._get_cros_repair_image_name()
xixuan5dc64ea2016-05-20 17:27:51 -0700737 force_update = True
738
xixuanec801e32016-08-25 10:20:22 -0700739 if not update_url and not self._parser.options.image:
xixuan5dc64ea2016-05-20 17:27:51 -0700740 raise error.AutoservError(
741 'There is no update URL, nor a method to get one.')
742
xixuanec801e32016-08-25 10:20:22 -0700743 if not update_url and self._parser.options.image:
744 update_url = self._parser.options.image
xixuan5dc64ea2016-05-20 17:27:51 -0700745
xixuan5dc64ea2016-05-20 17:27:51 -0700746
747 # Get build from parameter or AFE.
748 # If the build is not a URL, let devserver to stage it first.
749 # Otherwise, choose a devserver to trigger auto-update.
xixuanec801e32016-08-25 10:20:22 -0700750 build = None
Aviv Keshetf9375482016-10-26 02:09:03 -0700751 devserver = None
752 logging.debug('Resolving a devserver for auto-update')
Aviv Keshet3a76f762016-10-27 17:41:51 -0700753 previously_resolved = False
xixuanec801e32016-08-25 10:20:22 -0700754 if update_url.startswith('http://'):
755 build = autoupdater.url_to_image_name(update_url)
Aviv Keshet3a76f762016-10-27 17:41:51 -0700756 previously_resolved = True
757 else:
758 build = update_url
759 devserver = dev_server.resolve(build, self.hostname)
Allen Lia5cfb972016-12-27 17:17:22 -0800760 server_name = devserver.hostname
Aviv Keshet3a76f762016-10-27 17:41:51 -0700761
xixuan016d95b2017-02-27 14:37:47 -0800762 monarch_fields = self._get_au_monarch_fields(devserver, build)
Aviv Keshet3a76f762016-10-27 17:41:51 -0700763
764 if previously_resolved:
xixuan2e62d162016-09-29 19:19:19 -0700765 # Make sure devserver for Auto-Update has staged the build.
xixuan2e62d162016-09-29 19:19:19 -0700766 if server_name not in update_url:
Aviv Keshet3a76f762016-10-27 17:41:51 -0700767 logging.debug('Resolved to devserver that does not match '
768 'update_url. The previously resolved devserver '
769 'must be unhealthy. Switching to use devserver %s,'
770 ' and re-staging.',
xixuan2e62d162016-09-29 19:19:19 -0700771 server_name)
Aviv Keshetf9375482016-10-26 02:09:03 -0700772 logging.info('Staging build for AU: %s', update_url)
xixuan2e62d162016-09-29 19:19:19 -0700773 devserver.trigger_download(build, synchronous=False)
Dan Shi5e2efb72017-02-07 11:40:23 -0800774 c = metrics.Counter(
775 'chromeos/autotest/provision/failover_download')
776 c.increment(fields=monarch_fields)
xixuan5dc64ea2016-05-20 17:27:51 -0700777 else:
Aviv Keshetf9375482016-10-26 02:09:03 -0700778 logging.info('Staging build for AU: %s', update_url)
xixuan5dc64ea2016-05-20 17:27:51 -0700779 devserver.trigger_download(build, synchronous=False)
Dan Shi5e2efb72017-02-07 11:40:23 -0800780 c = metrics.Counter('chromeos/autotest/provision/trigger_download')
781 c.increment(fields=monarch_fields)
xixuan5dc64ea2016-05-20 17:27:51 -0700782
783 # Report provision stats.
xixuan016d95b2017-02-27 14:37:47 -0800784 install_with_dev_counter = metrics.Counter(
785 'chromeos/autotest/provision/install_with_devserver')
786 install_with_dev_counter.increment(fields=monarch_fields)
xixuanec801e32016-08-25 10:20:22 -0700787 logging.debug('Resolved devserver for auto-update: %s', devserver.url())
xixuan5dc64ea2016-05-20 17:27:51 -0700788
Aviv Keshetf9375482016-10-26 02:09:03 -0700789 # and other metrics from this function.
Dan Shi5e2efb72017-02-07 11:40:23 -0800790 metrics.Counter('chromeos/autotest/provision/resolve'
791 ).increment(fields=monarch_fields)
Aviv Keshetf9375482016-10-26 02:09:03 -0700792
xixuana3bbc422017-05-04 15:57:21 -0700793 force_original = self.get_chromeos_release_milestone() is None
794
xixuan9f455342017-06-08 18:21:44 -0700795 try:
796 devserver.auto_update(
797 self.hostname, build,
798 original_board=self.get_board().replace(
799 ds_constants.BOARD_PREFIX, ''),
800 original_release_version=self.get_release_version(),
801 log_dir=self.job.resultdir,
802 force_update=force_update,
803 full_update=force_full_update,
804 force_original=force_original)
805 except dev_server.RetryableProvisionException:
806 # It indicates that last provision failed due to devserver load
807 # issue, so another devserver is resolved to kick off provision
808 # job once again and only once.
809 logging.debug('Provision failed due to devserver issue,'
810 'retry it with another devserver.')
xixuan016d95b2017-02-27 14:37:47 -0800811
xixuan9f455342017-06-08 18:21:44 -0700812 # Check first whether this DUT is completely offline. If so, skip
813 # the following provision tries.
814 logging.debug('Checking whether host %s is online.', self.hostname)
815 if utils.ping(self.hostname, tries=1, deadline=1) == 0:
816 self._retry_auto_update_with_new_devserver(
817 build, devserver, force_update, force_full_update,
818 force_original)
819 else:
820 raise error.AutoservError(
821 'No answer to ping from %s' % self.hostname)
xixuan5dc64ea2016-05-20 17:27:51 -0700822
823 # The reason to resolve a new devserver in function machine_install
824 # is mostly because that the update_url there may has a strange format,
825 # and it's hard to parse the devserver url from it.
826 # Since we already resolve a devserver to trigger auto-update, the same
827 # devserver is used to form JOB_REPO_URL here. Verified in local test.
828 repo_url = tools.get_package_url(devserver.url(), build)
829 return build, {ds_constants.JOB_REPO_URL: repo_url}
830
831
Laurence Goodby778c9a42017-05-24 19:24:07 -0700832 def prepare_for_update(self):
833 """Prepares the DUT for an update.
834
835 Subclasses may override this to perform any special actions
836 required before updating.
837 """
Laurence Goodby468de252017-06-08 17:22:53 -0700838 pass
Laurence Goodby778c9a42017-05-24 19:24:07 -0700839
840
Chris Sosaa3ac2152012-05-23 22:23:13 -0700841 def machine_install(self, update_url=None, force_update=False,
Richard Barnette0b023a72015-04-24 16:07:30 +0000842 local_devserver=False, repair=False,
843 force_full_update=False):
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500844 """Install the DUT.
845
Dan Shi0f466e82013-02-22 15:44:58 -0800846 Use stateful update if the DUT is already running the same build.
847 Stateful update does not update kernel and tends to run much faster
848 than a full reimage. If the DUT is running a different build, or it
849 failed to do a stateful update, full update, including kernel update,
850 will be applied to the DUT.
851
Simran Basi5ace6f22016-01-06 17:30:44 -0800852 Once a host enters machine_install its host attribute job_repo_url
853 (used for package install) will be removed and then updated.
Scott Zawalskieadbf702013-03-14 09:23:06 -0400854
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500855 @param update_url: The url to use for the update
856 pattern: http://$devserver:###/update/$build
857 If update_url is None and repair is True we will install the
Dan Shi6964fa52014-12-18 11:04:27 -0800858 stable image listed in afe_stable_versions table. If the table
859 is not setup, global_config value under CROS.stable_cros_version
860 will be used instead.
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500861 @param force_update: Force an update even if the version installed
862 is the same. Default:False
Christopher Wiley6a4ff932015-05-15 14:00:47 -0700863 @param local_devserver: Used by test_that to allow people to
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500864 use their local devserver. Default: False
Chris Sosae92399e2015-04-24 11:32:59 -0700865 @param repair: Forces update to repair image. Implies force_update.
Fang Deng3d3b9272014-12-22 12:20:28 -0800866 @param force_full_update: If True, do not attempt to run stateful
867 update, force a full reimage. If False, try stateful update
868 first when the dut is already installed with the same version.
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500869 @raises autoupdater.ChromiumOSError
870
Dan Shibe3636a2016-02-14 22:48:01 -0800871 @returns A tuple of (image_name, host_attributes).
872 image_name is the name of image installed, e.g.,
873 veyron_jerry-release/R50-7871.0.0
874 host_attributes is a dictionary of (attribute, value), which
875 can be saved to afe_host_attributes table in database. This
876 method returns a dictionary with a single entry of
877 `job_repo_url`: repo_url, where repo_url is a devserver url to
878 autotest packages.
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500879 """
Chris Sosae92399e2015-04-24 11:32:59 -0700880 devserver = None
Richard Barnette0b023a72015-04-24 16:07:30 +0000881 if repair:
Chris Sosae92399e2015-04-24 11:32:59 -0700882 update_url, devserver = self._stage_image_for_update()
Richard Barnette0b023a72015-04-24 16:07:30 +0000883 force_update = True
Dan Shi0f466e82013-02-22 15:44:58 -0800884
Chris Sosae92399e2015-04-24 11:32:59 -0700885 if not update_url and not self._parser.options.image:
886 raise error.AutoservError(
Dan Shid07ee2e2015-09-24 14:49:25 -0700887 'There is no update URL, nor a method to get one.')
Chris Sosae92399e2015-04-24 11:32:59 -0700888
889 if not update_url and self._parser.options.image:
890 # This is the base case where we have no given update URL i.e.
891 # dynamic suites logic etc. This is the most flexible case where we
892 # can serve an update from any of our fleet of devservers.
893 requested_build = self._parser.options.image
894 if not requested_build.startswith('http://'):
895 logging.debug('Update will be staged for this installation')
896 update_url, devserver = self._stage_image_for_update(
Dan Shid07ee2e2015-09-24 14:49:25 -0700897 requested_build)
Chris Sosae92399e2015-04-24 11:32:59 -0700898 else:
899 update_url = requested_build
900
901 logging.debug('Update URL is %s', update_url)
902
Dan Shif48f8132016-02-18 10:34:30 -0800903 # Report provision stats.
Allen Lia5cfb972016-12-27 17:17:22 -0800904 server_name = dev_server.get_hostname(update_url)
Allen Li48a13fe2016-11-22 14:10:40 -0800905 (metrics.Counter('chromeos/autotest/provision/install')
906 .increment(fields={'devserver': server_name}))
Dan Shif48f8132016-02-18 10:34:30 -0800907
Dan Shid07ee2e2015-09-24 14:49:25 -0700908 # Create a file to indicate if provision fails. The file will be removed
909 # by stateful update or full install.
J. Richard Barnette91137f02016-03-10 16:52:26 -0800910 self.run('touch %s' % self.PROVISION_FAILED)
Dan Shid07ee2e2015-09-24 14:49:25 -0700911
Chris Sosae92399e2015-04-24 11:32:59 -0700912 update_complete = False
913 updater = autoupdater.ChromiumOSUpdater(
914 update_url, host=self, local_devserver=local_devserver)
Fang Deng3d3b9272014-12-22 12:20:28 -0800915 if not force_full_update:
916 try:
Chris Sosae92399e2015-04-24 11:32:59 -0700917 # If the DUT is already running the same build, try stateful
918 # update first as it's much quicker than a full re-image.
919 update_complete = self._try_stateful_update(
Dan Shid07ee2e2015-09-24 14:49:25 -0700920 update_url, force_update, updater)
Fang Deng3d3b9272014-12-22 12:20:28 -0800921 except Exception as e:
922 logging.exception(e)
J. Richard Barnette45e93de2012-04-11 17:24:15 -0700923
Dan Shi0f466e82013-02-22 15:44:58 -0800924 inactive_kernel = None
Chris Sosae92399e2015-04-24 11:32:59 -0700925 if update_complete or (not force_update and updater.check_version()):
926 logging.info('Install complete without full update')
927 else:
928 logging.info('DUT requires full update.')
929 self.reboot(timeout=self.REBOOT_TIMEOUT, wait=True)
Laurence Goodby778c9a42017-05-24 19:24:07 -0700930 self.prepare_for_update()
Dan Shi10b98482016-02-02 14:38:50 -0800931
Chris Sosae92399e2015-04-24 11:32:59 -0700932 num_of_attempts = provision.FLAKY_DEVSERVER_ATTEMPTS
Chris Sosab7612bc2013-03-21 10:32:37 -0700933
Chris Sosae92399e2015-04-24 11:32:59 -0700934 while num_of_attempts > 0:
935 num_of_attempts -= 1
936 try:
937 updater.run_update()
938 except Exception:
939 logging.warn('Autoupdate did not complete.')
940 # Do additional check for the devserver health. Ideally,
941 # the autoupdater.py could raise an exception when it
942 # detected network flake but that would require
943 # instrumenting the update engine and parsing it log.
944 if (num_of_attempts <= 0 or
945 devserver is None or
xixuan9e2c98d2016-02-26 19:04:53 -0800946 dev_server.ImageServer.devserver_healthy(
Chris Sosae92399e2015-04-24 11:32:59 -0700947 devserver.url())):
Dan Shid07ee2e2015-09-24 14:49:25 -0700948 raise
J. Richard Barnette45e93de2012-04-11 17:24:15 -0700949
Chris Sosae92399e2015-04-24 11:32:59 -0700950 logging.warn('Devserver looks unhealthy. Trying another')
951 update_url, devserver = self._stage_image_for_update(
952 requested_build)
953 logging.debug('New Update URL is %s', update_url)
954 updater = autoupdater.ChromiumOSUpdater(
955 update_url, host=self,
956 local_devserver=local_devserver)
957 else:
958 break
J. Richard Barnette45e93de2012-04-11 17:24:15 -0700959
Chris Sosae92399e2015-04-24 11:32:59 -0700960 # Give it some time in case of IO issues.
961 time.sleep(10)
Dan Shi5699ac22014-12-19 10:55:49 -0800962
Chris Sosae92399e2015-04-24 11:32:59 -0700963 # Figure out active and inactive kernel.
964 active_kernel, inactive_kernel = updater.get_kernel_state()
Simran Basi13fa1ba2013-03-04 10:56:47 -0800965
Chris Sosae92399e2015-04-24 11:32:59 -0700966 # Ensure inactive kernel has higher priority than active.
967 if (updater.get_kernel_priority(inactive_kernel)
968 < updater.get_kernel_priority(active_kernel)):
969 raise autoupdater.ChromiumOSError(
970 'Update failed. The priority of the inactive kernel'
971 ' partition is less than that of the active kernel'
972 ' partition.')
973
974 # Updater has returned successfully; reboot the host.
Richard Barnettefc3d7432016-07-07 10:21:31 -0700975 #
976 # Regarding the 'crossystem' command: In some cases, the
977 # TPM gets into a state such that it fails verification.
978 # We don't know why. However, this call papers over the
979 # problem by clearing the TPM during the reboot.
980 #
981 # We ignore failures from 'crossystem'. Although failure
982 # here is unexpected, and could signal a bug, the point
983 # of the exercise is to paper over problems; allowing
984 # this to fail would defeat the purpose.
985 self.run('crossystem clear_tpm_owner_request=1',
986 ignore_status=True)
Chris Sosae92399e2015-04-24 11:32:59 -0700987 self.reboot(timeout=self.REBOOT_TIMEOUT, wait=True)
988
989 self._post_update_processing(updater, inactive_kernel)
Simran Basi5ace6f22016-01-06 17:30:44 -0800990 image_name = autoupdater.url_to_image_name(update_url)
Dan Shibe3636a2016-02-14 22:48:01 -0800991 # update_url is different from devserver url needed to stage autotest
992 # packages, therefore, resolve a new devserver url here.
993 devserver_url = dev_server.ImageServer.resolve(image_name,
994 self.hostname).url()
995 repo_url = tools.get_package_url(devserver_url, image_name)
996 return image_name, {ds_constants.JOB_REPO_URL: repo_url}
J. Richard Barnette45e93de2012-04-11 17:24:15 -0700997
998
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +0800999 def _clear_fw_version_labels(self, rw_only):
1000 """Clear firmware version labels from the machine.
1001
1002 @param rw_only: True to only clear fwrw_version; otherewise, clear
1003 both fwro_version and fwrw_version.
1004 """
Dan Shi9cb0eec2014-06-03 09:04:50 -07001005 labels = self._AFE.get_labels(
Dan Shi0723bf52015-06-24 10:52:38 -07001006 name__startswith=provision.FW_RW_VERSION_PREFIX,
Dan Shi9cb0eec2014-06-03 09:04:50 -07001007 host__hostname=self.hostname)
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +08001008 if not rw_only:
1009 labels = labels + self._AFE.get_labels(
1010 name__startswith=provision.FW_RO_VERSION_PREFIX,
1011 host__hostname=self.hostname)
Dan Shi9cb0eec2014-06-03 09:04:50 -07001012 for label in labels:
1013 label.remove_hosts(hosts=[self.hostname])
1014
1015
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +08001016 def _add_fw_version_label(self, build, rw_only):
Dan Shi9cb0eec2014-06-03 09:04:50 -07001017 """Add firmware version label to the machine.
1018
1019 @param build: Build of firmware.
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +08001020 @param rw_only: True to only add fwrw_version; otherwise, add both
1021 fwro_version and fwrw_version.
Dan Shi9cb0eec2014-06-03 09:04:50 -07001022
1023 """
Prathmesh Prabhu2c7471d2016-11-15 20:19:57 +00001024 fw_label = provision.fwrw_version_to_label(build)
MK Ryu73be9862015-07-06 12:25:00 -07001025 self._AFE.run('label_add_hosts', id=fw_label, hosts=[self.hostname])
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +08001026 if not rw_only:
Prathmesh Prabhu2c7471d2016-11-15 20:19:57 +00001027 fw_label = provision.fwro_version_to_label(build)
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +08001028 self._AFE.run('label_add_hosts', id=fw_label, hosts=[self.hostname])
Dan Shi9cb0eec2014-06-03 09:04:50 -07001029
1030
Tom Wai-Hong Tam4ac78982016-01-08 02:34:37 +08001031 def firmware_install(self, build=None, rw_only=False):
Dan Shi9cb0eec2014-06-03 09:04:50 -07001032 """Install firmware to the DUT.
1033
1034 Use stateful update if the DUT is already running the same build.
1035 Stateful update does not update kernel and tends to run much faster
1036 than a full reimage. If the DUT is running a different build, or it
1037 failed to do a stateful update, full update, including kernel update,
1038 will be applied to the DUT.
1039
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +08001040 Once a host enters firmware_install its fw[ro|rw]_version label will
1041 be removed. After the firmware is updated successfully, a new
1042 fw[ro|rw]_version label will be added to the host.
Dan Shi9cb0eec2014-06-03 09:04:50 -07001043
1044 @param build: The build version to which we want to provision the
1045 firmware of the machine,
1046 e.g. 'link-firmware/R22-2695.1.144'.
Tom Wai-Hong Tam4ac78982016-01-08 02:34:37 +08001047 @param rw_only: True to only install firmware to its RW portions. Keep
1048 the RO portions unchanged.
Dan Shi9cb0eec2014-06-03 09:04:50 -07001049
1050 TODO(dshi): After bug 381718 is fixed, update here with corresponding
1051 exceptions that could be raised.
1052
1053 """
1054 if not self.servo:
1055 raise error.TestError('Host %s does not have servo.' %
1056 self.hostname)
1057
Wai-Hong Tam64cf7912017-05-23 16:41:40 -07001058 # Get the DUT board name from servod.
1059 board = self.servo.get_board()
Dan Shi9cb0eec2014-06-03 09:04:50 -07001060
Chris Sosae92399e2015-04-24 11:32:59 -07001061 # If build is not set, try to install firmware from stable CrOS.
Dan Shi9cb0eec2014-06-03 09:04:50 -07001062 if not build:
Richard Barnette260cbd02016-10-06 12:23:28 -07001063 build = afe_utils.get_stable_faft_version(board)
Dan Shi3d7a0e12015-10-12 11:55:45 -07001064 if not build:
1065 raise error.TestError(
1066 'Failed to find stable firmware build for %s.',
1067 self.hostname)
1068 logging.info('Will install firmware from build %s.', build)
Dan Shi9cb0eec2014-06-03 09:04:50 -07001069
1070 config = FAFTConfig(board)
1071 if config.use_u_boot:
1072 ap_image = 'image-%s.bin' % board
1073 else: # Depthcharge platform
1074 ap_image = 'image.bin'
1075 ec_image = 'ec.bin'
Dan Shi216389c2015-12-22 11:03:06 -08001076 ds = dev_server.ImageServer.resolve(build, self.hostname)
Dan Shi9cb0eec2014-06-03 09:04:50 -07001077 ds.stage_artifacts(build, ['firmware'])
1078
1079 tmpd = autotemp.tempdir(unique_id='fwimage')
1080 try:
1081 fwurl = self._FW_IMAGE_URL_PATTERN % (ds.url(), build)
1082 local_tarball = os.path.join(tmpd.name, os.path.basename(fwurl))
xixuan4e116822016-11-17 15:32:10 -08001083 ds.download_file(fwurl, local_tarball)
Dan Shi9cb0eec2014-06-03 09:04:50 -07001084
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +08001085 self._clear_fw_version_labels(rw_only)
Tom Wai-Hong Tam552fee12016-05-11 06:27:58 +08001086 if config.chrome_ec:
Tom Wai-Hong Tamf9db57a2016-03-17 05:32:22 +08001087 logging.info('Will re-program EC %snow', 'RW ' if rw_only else '')
1088 server_utils.system('tar xf %s -C %s %s' %
1089 (local_tarball, tmpd.name, ec_image),
1090 timeout=60)
1091 self.servo.program_ec(os.path.join(tmpd.name, ec_image), rw_only)
1092 else:
1093 logging.info('Not a Chrome EC, ignore re-programing it')
Tom Wai-Hong Tam4ac78982016-01-08 02:34:37 +08001094 logging.info('Will re-program BIOS %snow', 'RW ' if rw_only else '')
Tom Wai-Hong Tamf9db57a2016-03-17 05:32:22 +08001095 server_utils.system('tar xf %s -C %s %s' %
1096 (local_tarball, tmpd.name, ap_image),
1097 timeout=60)
Tom Wai-Hong Tam4ac78982016-01-08 02:34:37 +08001098 self.servo.program_bios(os.path.join(tmpd.name, ap_image), rw_only)
Dan Shi9cb0eec2014-06-03 09:04:50 -07001099 self.servo.get_power_state_controller().reset()
1100 time.sleep(self.servo.BOOT_DELAY)
Wai-Hong Tamb5f66ce2016-11-10 15:45:30 -08001101 if utils.host_is_in_lab_zone(self.hostname):
1102 self._add_fw_version_label(build, rw_only)
Dan Shi9cb0eec2014-06-03 09:04:50 -07001103 finally:
1104 tmpd.clean()
1105
1106
Dan Shi10e992b2013-08-30 11:02:59 -07001107 def show_update_engine_log(self):
1108 """Output update engine log."""
MK Ryu35d661e2014-09-25 17:44:10 -07001109 logging.debug('Dumping %s', client_constants.UPDATE_ENGINE_LOG)
1110 self.run('cat %s' % client_constants.UPDATE_ENGINE_LOG)
Dan Shi10e992b2013-08-30 11:02:59 -07001111
1112
Richard Barnette82c35912012-11-20 10:09:10 -08001113 def _get_board_from_afe(self):
Kevin Cheng84a71ba2016-07-14 11:03:57 -07001114 """Retrieve this host's board from its labels stored locally.
Richard Barnette82c35912012-11-20 10:09:10 -08001115
1116 Looks for a host label of the form "board:<board>", and
1117 returns the "<board>" part of the label. `None` is returned
1118 if there is not a single, unique label matching the pattern.
1119
1120 @returns board from label, or `None`.
1121 """
Richard Barnette260cbd02016-10-06 12:23:28 -07001122 board = afe_utils.get_board(self)
1123 if board is None:
1124 raise error.AutoservError('DUT cannot be repaired, '
1125 'there is no board attribute.')
1126 return board
Simran Basi833814b2013-01-29 13:13:43 -08001127
1128
beepsf079cfb2013-09-18 17:49:51 -07001129 def servo_install(self, image_url=None, usb_boot_timeout=USB_BOOT_TIMEOUT,
1130 install_timeout=INSTALL_TIMEOUT):
Scott Zawalski62bacae2013-03-05 10:40:32 -05001131 """
1132 Re-install the OS on the DUT by:
1133 1) installing a test image on a USB storage device attached to the Servo
1134 board,
Richard Barnette03a0c132012-11-05 12:40:35 -08001135 2) booting that image in recovery mode, and then
J. Richard Barnette31b2e312013-04-04 16:05:22 -07001136 3) installing the image with chromeos-install.
1137
Scott Zawalski62bacae2013-03-05 10:40:32 -05001138 @param image_url: If specified use as the url to install on the DUT.
1139 otherwise boot the currently staged image on the USB stick.
beepsf079cfb2013-09-18 17:49:51 -07001140 @param usb_boot_timeout: The usb_boot_timeout to use during reimage.
1141 Factory images need a longer usb_boot_timeout than regular
1142 cros images.
1143 @param install_timeout: The timeout to use when installing the chromeos
1144 image. Factory images need a longer install_timeout.
Richard Barnette03a0c132012-11-05 12:40:35 -08001145
Scott Zawalski62bacae2013-03-05 10:40:32 -05001146 @raises AutoservError if the image fails to boot.
beepsf079cfb2013-09-18 17:49:51 -07001147
J. Richard Barnette0199cc82014-12-05 17:08:40 -08001148 """
beepsf079cfb2013-09-18 17:49:51 -07001149 logging.info('Downloading image to USB, then booting from it. Usb boot '
1150 'timeout = %s', usb_boot_timeout)
Allen Li48a13fe2016-11-22 14:10:40 -08001151 with metrics.SecondsTimer(
1152 'chromeos/autotest/provision/servo_install/boot_duration'):
1153 self.servo.install_recovery_image(image_url)
1154 if not self.wait_up(timeout=usb_boot_timeout):
1155 raise hosts.AutoservRepairError(
1156 'DUT failed to boot from USB after %d seconds' %
1157 usb_boot_timeout)
Scott Zawalski62bacae2013-03-05 10:40:32 -05001158
Tom Wai-Hong Tamf6b4f812015-08-08 04:14:59 +08001159 # The new chromeos-tpm-recovery has been merged since R44-7073.0.0.
1160 # In old CrOS images, this command fails. Skip the error.
Tom Wai-Hong Tam27af7332015-07-25 06:09:39 +08001161 logging.info('Resetting the TPM status')
Tom Wai-Hong Tamf6b4f812015-08-08 04:14:59 +08001162 try:
1163 self.run('chromeos-tpm-recovery')
1164 except error.AutoservRunError:
1165 logging.warn('chromeos-tpm-recovery is too old.')
1166
Tom Wai-Hong Tam27af7332015-07-25 06:09:39 +08001167
Allen Li48a13fe2016-11-22 14:10:40 -08001168 with metrics.SecondsTimer(
1169 'chromeos/autotest/provision/servo_install/install_duration'):
1170 logging.info('Installing image through chromeos-install.')
1171 self.run('chromeos-install --yes', timeout=install_timeout)
1172 self.halt()
beepsf079cfb2013-09-18 17:49:51 -07001173
1174 logging.info('Power cycling DUT through servo.')
J. Richard Barnette0199cc82014-12-05 17:08:40 -08001175 self.servo.get_power_state_controller().power_off()
Fang Dengafb88142013-05-30 17:44:31 -07001176 self.servo.switch_usbkey('off')
J. Richard Barnette0199cc82014-12-05 17:08:40 -08001177 # N.B. The Servo API requires that we use power_on() here
1178 # for two reasons:
1179 # 1) After turning on a DUT in recovery mode, you must turn
1180 # it off and then on with power_on() once more to
1181 # disable recovery mode (this is a Parrot specific
1182 # requirement).
1183 # 2) After power_off(), the only way to turn on is with
1184 # power_on() (this is a Storm specific requirement).
J. Richard Barnettefbcc7122013-07-24 18:24:59 -07001185 self.servo.get_power_state_controller().power_on()
beepsf079cfb2013-09-18 17:49:51 -07001186
1187 logging.info('Waiting for DUT to come back up.')
Richard Barnette03a0c132012-11-05 12:40:35 -08001188 if not self.wait_up(timeout=self.BOOT_TIMEOUT):
1189 raise error.AutoservError('DUT failed to reboot installed '
1190 'test image after %d seconds' %
Scott Zawalski62bacae2013-03-05 10:40:32 -05001191 self.BOOT_TIMEOUT)
1192
1193
Richard Barnette9a26ad62016-06-10 12:03:08 -07001194 def repair_servo(self):
Dan Shi90466352015-09-22 15:01:05 -07001195 """
Richard Barnette9a26ad62016-06-10 12:03:08 -07001196 Confirm that servo is initialized and verified.
Dan Shi90466352015-09-22 15:01:05 -07001197
Richard Barnette9a26ad62016-06-10 12:03:08 -07001198 If the servo object is missing, attempt to repair the servo
1199 host. Repair failures are passed back to the caller.
1200
1201 @raise AutoservError: If there is no servo host for this CrOS
1202 host.
1203 """
1204 if self.servo:
1205 return
1206 if not self._servo_host:
1207 raise error.AutoservError('No servo host for %s.' %
1208 self.hostname)
1209 self._servo_host.repair()
1210 self.servo = self._servo_host.get_servo()
Dan Shi90466352015-09-22 15:01:05 -07001211
1212
J. Richard Barnettec2d99cf2015-11-18 12:46:15 -08001213 def repair(self):
1214 """Attempt to get the DUT to pass `self.verify()`.
Richard Barnette82c35912012-11-20 10:09:10 -08001215
1216 This overrides the base class function for repair; it does
J. Richard Barnette91137f02016-03-10 16:52:26 -08001217 not call back to the parent class, but instead relies on
1218 `self._repair_strategy` to coordinate the verification and
1219 repair steps needed to get the DUT working.
Richard Barnette82c35912012-11-20 10:09:10 -08001220 """
J. Richard Barnette91137f02016-03-10 16:52:26 -08001221 self._repair_strategy.repair(self)
Scott Zawalski89c44dd2013-02-26 09:28:02 -05001222
Richard Barnette82c35912012-11-20 10:09:10 -08001223
J. Richard Barnette1d78b012012-05-15 13:56:30 -07001224 def close(self):
Fang Deng0ca40e22013-08-27 17:47:44 -07001225 super(CrosHost, self).close()
xixuand6011f12016-12-08 15:01:58 -08001226 if self._chameleon_host:
1227 self._chameleon_host.close()
1228
1229 if self._servo_host:
1230 self._servo_host.close()
J. Richard Barnette1d78b012012-05-15 13:56:30 -07001231
1232
Dan Shi49ca0932014-11-14 11:22:27 -08001233 def get_power_supply_info(self):
1234 """Get the output of power_supply_info.
1235
1236 power_supply_info outputs the info of each power supply, e.g.,
1237 Device: Line Power
1238 online: no
1239 type: Mains
1240 voltage (V): 0
1241 current (A): 0
1242 Device: Battery
1243 state: Discharging
1244 percentage: 95.9276
1245 technology: Li-ion
1246
1247 Above output shows two devices, Line Power and Battery, with details of
1248 each device listed. This function parses the output into a dictionary,
1249 with key being the device name, and value being a dictionary of details
1250 of the device info.
1251
1252 @return: The dictionary of power_supply_info, e.g.,
1253 {'Line Power': {'online': 'yes', 'type': 'main'},
1254 'Battery': {'vendor': 'xyz', 'percentage': '100'}}
Dan Shie9b765d2014-12-29 16:59:49 -08001255 @raise error.AutoservRunError if power_supply_info tool is not found in
1256 the DUT. Caller should handle this error to avoid false failure
1257 on verification.
Dan Shi49ca0932014-11-14 11:22:27 -08001258 """
1259 result = self.run('power_supply_info').stdout.strip()
1260 info = {}
1261 device_name = None
1262 device_info = {}
1263 for line in result.split('\n'):
1264 pair = [v.strip() for v in line.split(':')]
1265 if len(pair) != 2:
1266 continue
1267 if pair[0] == 'Device':
1268 if device_name:
1269 info[device_name] = device_info
1270 device_name = pair[1]
1271 device_info = {}
1272 else:
1273 device_info[pair[0]] = pair[1]
1274 if device_name and not device_name in info:
1275 info[device_name] = device_info
1276 return info
1277
1278
1279 def get_battery_percentage(self):
1280 """Get the battery percentage.
1281
1282 @return: The percentage of battery level, value range from 0-100. Return
1283 None if the battery info cannot be retrieved.
1284 """
1285 try:
1286 info = self.get_power_supply_info()
1287 logging.info(info)
1288 return float(info['Battery']['percentage'])
Dan Shie9b765d2014-12-29 16:59:49 -08001289 except (KeyError, ValueError, error.AutoservRunError):
Dan Shi49ca0932014-11-14 11:22:27 -08001290 return None
1291
1292
1293 def is_ac_connected(self):
1294 """Check if the dut has power adapter connected and charging.
1295
1296 @return: True if power adapter is connected and charging.
1297 """
1298 try:
1299 info = self.get_power_supply_info()
1300 return info['Line Power']['online'] == 'yes'
Dan Shie9b765d2014-12-29 16:59:49 -08001301 except (KeyError, error.AutoservRunError):
1302 return None
Dan Shi49ca0932014-11-14 11:22:27 -08001303
1304
Simran Basi5e6339a2013-03-21 11:34:32 -07001305 def _cleanup_poweron(self):
1306 """Special cleanup method to make sure hosts always get power back."""
1307 afe = frontend_wrappers.RetryingAFE(timeout_min=5, delay_sec=10)
1308 hosts = afe.get_hosts(hostname=self.hostname)
1309 if not hosts or not (self._RPM_OUTLET_CHANGED in
1310 hosts[0].attributes):
1311 return
1312 logging.debug('This host has recently interacted with the RPM'
1313 ' Infrastructure. Ensuring power is on.')
1314 try:
1315 self.power_on()
Dan Shi7dca56e2014-11-11 17:07:56 -08001316 afe.set_host_attribute(self._RPM_OUTLET_CHANGED, None,
1317 hostname=self.hostname)
Simran Basi5e6339a2013-03-21 11:34:32 -07001318 except rpm_client.RemotePowerException:
Simran Basi5e6339a2013-03-21 11:34:32 -07001319 logging.error('Failed to turn Power On for this host after '
1320 'cleanup through the RPM Infrastructure.')
Gabe Blackb72f4fb2015-01-20 16:47:13 -08001321 autotest_es.post(
Dan Shi7dca56e2014-11-11 17:07:56 -08001322 type_str='RPM_poweron_failure',
1323 metadata={'hostname': self.hostname})
Dan Shi49ca0932014-11-14 11:22:27 -08001324
1325 battery_percentage = self.get_battery_percentage()
Dan Shif01ebe22014-12-05 13:10:57 -08001326 if battery_percentage and battery_percentage < 50:
Dan Shi49ca0932014-11-14 11:22:27 -08001327 raise
1328 elif self.is_ac_connected():
1329 logging.info('The device has power adapter connected and '
1330 'charging. No need to try to turn RPM on '
1331 'again.')
1332 afe.set_host_attribute(self._RPM_OUTLET_CHANGED, None,
1333 hostname=self.hostname)
1334 logging.info('Battery level is now at %s%%. The device may '
1335 'still have enough power to run test, so no '
1336 'exception will be raised.', battery_percentage)
1337
Simran Basi5e6339a2013-03-21 11:34:32 -07001338
beepsc87ff602013-07-31 21:53:00 -07001339 def _is_factory_image(self):
1340 """Checks if the image on the DUT is a factory image.
1341
1342 @return: True if the image on the DUT is a factory image.
1343 False otherwise.
1344 """
1345 result = self.run('[ -f /root/.factory_test ]', ignore_status=True)
1346 return result.exit_status == 0
1347
1348
1349 def _restart_ui(self):
J. Richard Barnette84890bd2014-02-21 11:05:47 -08001350 """Restart the Chrome UI.
beepsc87ff602013-07-31 21:53:00 -07001351
1352 @raises: FactoryImageCheckerException for factory images, since
1353 we cannot attempt to restart ui on them.
1354 error.AutoservRunError for any other type of error that
1355 occurs while restarting ui.
1356 """
1357 if self._is_factory_image():
Dan Shi549fb822015-03-24 18:01:11 -07001358 raise FactoryImageCheckerException('Cannot restart ui on factory '
1359 'images')
beepsc87ff602013-07-31 21:53:00 -07001360
J. Richard Barnette84890bd2014-02-21 11:05:47 -08001361 # TODO(jrbarnette): The command to stop/start the ui job
1362 # should live inside cros_ui, too. However that would seem
1363 # to imply interface changes to the existing start()/restart()
1364 # functions, which is a bridge too far (for now).
J. Richard Barnette6069aa12015-06-08 09:10:24 -07001365 prompt = cros_ui.get_chrome_session_ident(self)
J. Richard Barnette84890bd2014-02-21 11:05:47 -08001366 self.run('stop ui; start ui')
1367 cros_ui.wait_for_chrome_ready(prompt, self)
beepsc87ff602013-07-31 21:53:00 -07001368
1369
xixuana3bbc422017-05-04 15:57:21 -07001370 def _get_lsb_release_content(self):
1371 """Return the content of lsb-release file of host."""
1372 return self.run(
1373 'cat "%s"' % client_constants.LSB_RELEASE).stdout.strip()
1374
1375
Dan Shi549fb822015-03-24 18:01:11 -07001376 def get_release_version(self):
1377 """Get the value of attribute CHROMEOS_RELEASE_VERSION from lsb-release.
1378
1379 @returns The version string in lsb-release, under attribute
1380 CHROMEOS_RELEASE_VERSION.
1381 """
Dan Shi549fb822015-03-24 18:01:11 -07001382 return lsbrelease_utils.get_chromeos_release_version(
xixuana3bbc422017-05-04 15:57:21 -07001383 lsb_release_content=self._get_lsb_release_content())
1384
1385
1386 def get_chromeos_release_milestone(self):
1387 """Get the value of attribute CHROMEOS_RELEASE_BUILD_TYPE
1388 from lsb-release.
1389
1390 @returns The version string in lsb-release, under attribute
1391 CHROMEOS_RELEASE_BUILD_TYPE.
1392 """
1393 return lsbrelease_utils.get_chromeos_release_milestone(
1394 lsb_release_content=self._get_lsb_release_content())
Dan Shi549fb822015-03-24 18:01:11 -07001395
1396
1397 def verify_cros_version_label(self):
1398 """ Make sure host's cros-version label match the actual image in dut.
1399
1400 Remove any cros-version: label that doesn't match that installed in
1401 the dut.
1402
1403 @param raise_error: Set to True to raise exception if any mismatch found
1404
1405 @raise error.AutoservError: If any mismatch between cros-version label
1406 and the build installed in dut is found.
1407 """
1408 labels = self._AFE.get_labels(
1409 name__startswith=ds_constants.VERSION_PREFIX,
1410 host__hostname=self.hostname)
1411 mismatch_found = False
1412 if labels:
1413 # Get CHROMEOS_RELEASE_VERSION from lsb-release, e.g., 6908.0.0.
1414 # Note that it's different from cros-version label, which has
1415 # builder and branch info, e.g.,
1416 # cros-version:peppy-release/R43-6908.0.0
1417 release_version = self.get_release_version()
1418 host_list = [self.hostname]
1419 for label in labels:
1420 # Remove any cros-version label that does not match
1421 # release_version.
1422 build_version = label.name[len(ds_constants.VERSION_PREFIX):]
1423 if not utils.version_match(build_version, release_version):
1424 logging.warn('cros-version label "%s" does not match '
1425 'release version %s. Removing the label.',
1426 label.name, release_version)
1427 label.remove_hosts(hosts=host_list)
1428 mismatch_found = True
1429 if mismatch_found:
Dan Shi1057bae2015-03-30 11:35:09 -07001430 autotest_es.post(use_http=True,
1431 type_str='cros_version_label_mismatch',
1432 metadata={'hostname': self.hostname})
Dan Shi549fb822015-03-24 18:01:11 -07001433 raise error.AutoservError('The host has wrong cros-version label.')
1434
1435
Laurence Goodby778c9a42017-05-24 19:24:07 -07001436 def cleanup_services(self):
1437 """Reinitializes the device for cleanup.
1438
1439 Subclasses may override this to customize the cleanup method.
1440
1441 To indicate failure of the reset, the implementation may raise
1442 any of:
1443 error.AutoservRunError
1444 error.AutotestRunError
1445 FactoryImageCheckerException
1446
1447 @raises error.AutoservRunError
1448 @raises error.AutotestRunError
1449 @raises error.FactoryImageCheckerException
1450 """
1451 self._restart_ui()
1452
1453
beepsc87ff602013-07-31 21:53:00 -07001454 def cleanup(self):
MK Ryu35d661e2014-09-25 17:44:10 -07001455 self.run('rm -f %s' % client_constants.CLEANUP_LOGS_PAUSED_FILE)
Scott Zawalskiddbc31e2012-11-15 11:29:01 -05001456 try:
Laurence Goodby778c9a42017-05-24 19:24:07 -07001457 self.cleanup_services()
beepsc87ff602013-07-31 21:53:00 -07001458 except (error.AutotestRunError, error.AutoservRunError,
1459 FactoryImageCheckerException):
Ilja H. Friedel04be2bd2014-05-07 21:29:59 -07001460 logging.warning('Unable to restart ui, rebooting device.')
Scott Zawalskiddbc31e2012-11-15 11:29:01 -05001461 # Since restarting the UI fails fall back to normal Autotest
1462 # cleanup routines, i.e. reboot the machine.
Fang Deng0ca40e22013-08-27 17:47:44 -07001463 super(CrosHost, self).cleanup()
Simran Basi5e6339a2013-03-21 11:34:32 -07001464 # Check if the rpm outlet was manipulated.
Simran Basid5e5e272012-09-24 15:23:59 -07001465 if self.has_power():
Simran Basi5e6339a2013-03-21 11:34:32 -07001466 self._cleanup_poweron()
Dan Shi549fb822015-03-24 18:01:11 -07001467 self.verify_cros_version_label()
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001468
1469
Yu-Ju Honga2be94a2012-07-31 09:48:52 -07001470 def reboot(self, **dargs):
1471 """
1472 This function reboots the site host. The more generic
1473 RemoteHost.reboot() performs sync and sleeps for 5
1474 seconds. This is not necessary for Chrome OS devices as the
1475 sync should be finished in a short time during the reboot
1476 command.
1477 """
Tom Wai-Hong Tamf5cd1d42012-08-13 12:04:08 +08001478 if 'reboot_cmd' not in dargs:
Doug Anderson7d5aeb22014-02-27 15:12:17 -08001479 reboot_timeout = dargs.get('reboot_timeout', 10)
J. Richard Barnette9af19632015-09-25 12:18:03 -07001480 dargs['reboot_cmd'] = ('sleep 1; '
1481 'reboot & sleep %d; '
1482 'reboot -f' % reboot_timeout)
Yu-Ju Honga2be94a2012-07-31 09:48:52 -07001483 # Enable fastsync to avoid running extra sync commands before reboot.
Tom Wai-Hong Tamf5cd1d42012-08-13 12:04:08 +08001484 if 'fastsync' not in dargs:
1485 dargs['fastsync'] = True
Michael Liangda8c60a2014-06-03 13:24:51 -07001486
Charlie Mooneya8e6dab2014-05-29 14:37:55 -07001487 # For purposes of logging reboot times:
1488 # Get the board name i.e. 'daisy_spring'
Michael Liangca4f5a62014-07-10 15:45:13 -07001489 board_fullname = self.get_board()
1490
1491 # Strip the prefix and add it to dargs.
1492 dargs['board'] = board_fullname[board_fullname.find(':')+1:]
Vincent Palatindf2372c2016-10-07 17:03:00 +02001493 # Record who called us
1494 orig = sys._getframe(1).f_code
Vincent Palatin80780b22016-07-27 16:02:37 +02001495 metric_fields = {'board' : dargs['board'],
Vincent Palatindf2372c2016-10-07 17:03:00 +02001496 'dut_host_name' : self.hostname,
1497 'success' : True}
1498 metric_debug_fields = {'board' : dargs['board'],
1499 'caller' : "%s:%s" % (orig.co_filename, orig.co_name),
1500 'success' : True,
1501 'error' : ''}
1502
Vincent Palatin80780b22016-07-27 16:02:37 +02001503 t0 = time.time()
1504 try:
1505 super(CrosHost, self).reboot(**dargs)
1506 except Exception as e:
1507 metric_fields['success'] = False
Vincent Palatindf2372c2016-10-07 17:03:00 +02001508 metric_debug_fields['success'] = False
1509 metric_debug_fields['error'] = type(e).__name__
Vincent Palatin80780b22016-07-27 16:02:37 +02001510 raise
1511 finally:
1512 duration = int(time.time() - t0)
Dan Shi5e2efb72017-02-07 11:40:23 -08001513 metrics.Counter(
1514 'chromeos/autotest/autoserv/reboot_count').increment(
1515 fields=metric_fields)
1516 metrics.Counter(
1517 'chromeos/autotest/autoserv/reboot_debug').increment(
1518 fields=metric_debug_fields)
1519 metrics.SecondsDistribution(
1520 'chromeos/autotest/autoserv/reboot_duration').add(
1521 duration, fields=metric_fields)
Yu-Ju Honga2be94a2012-07-31 09:48:52 -07001522
1523
Gwendal Grignou7a61d2f2014-05-23 11:05:51 -07001524 def suspend(self, **dargs):
1525 """
1526 This function suspends the site host.
1527 """
1528 suspend_time = dargs.get('suspend_time', 60)
1529 dargs['timeout'] = suspend_time
1530 if 'suspend_cmd' not in dargs:
J. Richard Barnette9af19632015-09-25 12:18:03 -07001531 dargs['suspend_cmd'] = ' && '.join([
1532 'echo 0 > /sys/class/rtc/rtc0/wakealarm',
Gwendal Grignou7a61d2f2014-05-23 11:05:51 -07001533 'echo +%d > /sys/class/rtc/rtc0/wakealarm' % suspend_time,
J. Richard Barnette9af19632015-09-25 12:18:03 -07001534 'powerd_dbus_suspend --delay=0'])
Gwendal Grignou7a61d2f2014-05-23 11:05:51 -07001535 super(CrosHost, self).suspend(**dargs)
1536
1537
Simran Basiec564392014-08-25 16:48:09 -07001538 def upstart_status(self, service_name):
1539 """Check the status of an upstart init script.
1540
1541 @param service_name: Service to look up.
1542
1543 @returns True if the service is running, False otherwise.
1544 """
1545 return self.run('status %s | grep start/running' %
1546 service_name).stdout.strip() != ''
1547
1548
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001549 def verify_software(self):
Richard Barnetteb2bc13c2013-01-08 17:32:51 -08001550 """Verify working software on a Chrome OS system.
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001551
Richard Barnetteb2bc13c2013-01-08 17:32:51 -08001552 Tests for the following conditions:
1553 1. All conditions tested by the parent version of this
1554 function.
1555 2. Sufficient space in /mnt/stateful_partition.
Fang Deng6b05f5b2013-03-20 13:42:11 -07001556 3. Sufficient space in /mnt/stateful_partition/encrypted.
1557 4. update_engine answers a simple status request over DBus.
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001558
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001559 """
Fang Deng0ca40e22013-08-27 17:47:44 -07001560 super(CrosHost, self).verify_software()
Dan Shib8540a52015-07-16 14:18:23 -07001561 default_kilo_inodes_required = CONFIG.get_config_value(
1562 'SERVER', 'kilo_inodes_required', type=int, default=100)
1563 board = self.get_board().replace(ds_constants.BOARD_PREFIX, '')
1564 kilo_inodes_required = CONFIG.get_config_value(
1565 'SERVER', 'kilo_inodes_required_%s' % board,
1566 type=int, default=default_kilo_inodes_required)
1567 self.check_inodes('/mnt/stateful_partition', kilo_inodes_required)
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001568 self.check_diskspace(
1569 '/mnt/stateful_partition',
Dan Shib8540a52015-07-16 14:18:23 -07001570 CONFIG.get_config_value(
Fang Deng6b05f5b2013-03-20 13:42:11 -07001571 'SERVER', 'gb_diskspace_required', type=float,
1572 default=20.0))
Gaurav Shahe448af82014-06-19 15:18:59 -07001573 encrypted_stateful_path = '/mnt/stateful_partition/encrypted'
1574 # Not all targets build with encrypted stateful support.
1575 if self.path_exists(encrypted_stateful_path):
1576 self.check_diskspace(
1577 encrypted_stateful_path,
Dan Shib8540a52015-07-16 14:18:23 -07001578 CONFIG.get_config_value(
Gaurav Shahe448af82014-06-19 15:18:59 -07001579 'SERVER', 'gb_encrypted_diskspace_required', type=float,
1580 default=0.1))
beepsc87ff602013-07-31 21:53:00 -07001581
Simran Basiec564392014-08-25 16:48:09 -07001582 if not self.upstart_status('system-services'):
Prashanth B5d0a0512014-04-25 12:26:08 -07001583 raise error.AutoservError('Chrome failed to reach login. '
1584 'System services not running.')
1585
beepsc87ff602013-07-31 21:53:00 -07001586 # Factory images don't run update engine,
1587 # goofy controls dbus on these DUTs.
1588 if not self._is_factory_image():
1589 self.run('update_engine_client --status')
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001590
Dan Shi549fb822015-03-24 18:01:11 -07001591 self.verify_cros_version_label()
1592
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001593
J. Richard Barnettea7e7fdc2016-02-12 12:35:36 -08001594 def verify(self):
1595 self._repair_strategy.verify(self)
1596
1597
Fang Deng96667ca2013-08-01 17:46:18 -07001598 def make_ssh_command(self, user='root', port=22, opts='', hosts_file=None,
1599 connect_timeout=None, alive_interval=None):
1600 """Override default make_ssh_command to use options tuned for Chrome OS.
1601
1602 Tuning changes:
1603 - ConnectTimeout=30; maximum of 30 seconds allowed for an SSH
1604 connection failure. Consistency with remote_access.sh.
1605
Samuel Tan2ce155b2015-06-23 18:24:38 -07001606 - ServerAliveInterval=900; which causes SSH to ping connection every
1607 900 seconds. In conjunction with ServerAliveCountMax ensures
1608 that if the connection dies, Autotest will bail out.
Fang Deng96667ca2013-08-01 17:46:18 -07001609 Originally tried 60 secs, but saw frequent job ABORTS where
Samuel Tan2ce155b2015-06-23 18:24:38 -07001610 the test completed successfully. Later increased from 180 seconds to
1611 900 seconds to account for tests where the DUT is suspended for
1612 longer periods of time.
Fang Deng96667ca2013-08-01 17:46:18 -07001613
1614 - ServerAliveCountMax=3; consistency with remote_access.sh.
1615
1616 - ConnectAttempts=4; reduce flakiness in connection errors;
1617 consistency with remote_access.sh.
1618
1619 - UserKnownHostsFile=/dev/null; we don't care about the keys.
1620 Host keys change with every new installation, don't waste
1621 memory/space saving them.
1622
1623 - SSH protocol forced to 2; needed for ServerAliveInterval.
1624
1625 @param user User name to use for the ssh connection.
1626 @param port Port on the target host to use for ssh connection.
1627 @param opts Additional options to the ssh command.
1628 @param hosts_file Ignored.
1629 @param connect_timeout Ignored.
1630 @param alive_interval Ignored.
1631 """
Aviv Keshetc5947fa2013-09-04 14:06:29 -07001632 base_command = ('/usr/bin/ssh -a -x %s %s %s'
1633 ' -o StrictHostKeyChecking=no'
Fang Deng96667ca2013-08-01 17:46:18 -07001634 ' -o UserKnownHostsFile=/dev/null -o BatchMode=yes'
Samuel Tan2ce155b2015-06-23 18:24:38 -07001635 ' -o ConnectTimeout=30 -o ServerAliveInterval=900'
Fang Deng96667ca2013-08-01 17:46:18 -07001636 ' -o ServerAliveCountMax=3 -o ConnectionAttempts=4'
1637 ' -o Protocol=2 -l %s -p %d')
Aviv Keshetc5947fa2013-09-04 14:06:29 -07001638 return base_command % (self._ssh_verbosity_flag, self._ssh_options,
1639 opts, user, port)
Jason Abeleb6f924f2013-11-13 16:01:54 -08001640 def syslog(self, message, tag='autotest'):
1641 """Logs a message to syslog on host.
1642
1643 @param message String message to log into syslog
1644 @param tag String tag prefix for syslog
1645
1646 """
1647 self.run('logger -t "%s" "%s"' % (tag, message))
1648
1649
J. Richard Barnetteb6de7e32013-02-14 13:28:04 -08001650 def _ping_check_status(self, status):
1651 """Ping the host once, and return whether it has a given status.
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001652
J. Richard Barnetteb6de7e32013-02-14 13:28:04 -08001653 @param status Check the ping status against this value.
1654 @return True iff `status` and the result of ping are the same
1655 (i.e. both True or both False).
1656
1657 """
1658 ping_val = utils.ping(self.hostname, tries=1, deadline=1)
1659 return not (status ^ (ping_val == 0))
1660
1661 def _ping_wait_for_status(self, status, timeout):
1662 """Wait for the host to have a given status (UP or DOWN).
1663
1664 Status is checked by polling. Polling will not last longer
1665 than the number of seconds in `timeout`. The polling
1666 interval will be long enough that only approximately
1667 _PING_WAIT_COUNT polling cycles will be executed, subject
1668 to a maximum interval of about one minute.
1669
1670 @param status Waiting will stop immediately if `ping` of the
1671 host returns this status.
1672 @param timeout Poll for at most this many seconds.
1673 @return True iff the host status from `ping` matched the
1674 requested status at the time of return.
1675
1676 """
1677 # _ping_check_status() takes about 1 second, hence the
1678 # "- 1" in the formula below.
Nathan Ciobanu38480a32016-10-25 15:26:45 -07001679 # FIXME: if the ping command errors then _ping_check_status()
1680 # returns instantly. If timeout is also smaller than twice
1681 # _PING_WAIT_COUNT then the while loop below forks many
1682 # thousands of ping commands (see /tmp/test_that_results_XXXXX/
1683 # /results-1-logging_YYY.ZZZ/debug/autoserv.DEBUG) and hogs one
1684 # CPU core for 60 seconds.
J. Richard Barnetteb6de7e32013-02-14 13:28:04 -08001685 poll_interval = min(int(timeout / self._PING_WAIT_COUNT), 60) - 1
1686 end_time = time.time() + timeout
1687 while time.time() <= end_time:
1688 if self._ping_check_status(status):
1689 return True
1690 if poll_interval > 0:
1691 time.sleep(poll_interval)
1692
1693 # The last thing we did was sleep(poll_interval), so it may
1694 # have been too long since the last `ping`. Check one more
1695 # time, just to be sure.
1696 return self._ping_check_status(status)
1697
1698 def ping_wait_up(self, timeout):
1699 """Wait for the host to respond to `ping`.
1700
1701 N.B. This method is not a reliable substitute for
1702 `wait_up()`, because a host that responds to ping will not
1703 necessarily respond to ssh. This method should only be used
1704 if the target DUT can be considered functional even if it
1705 can't be reached via ssh.
1706
1707 @param timeout Minimum time to allow before declaring the
1708 host to be non-responsive.
1709 @return True iff the host answered to ping before the timeout.
1710
1711 """
1712 return self._ping_wait_for_status(self._PING_STATUS_UP, timeout)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001713
Andrew Bresticker678c0c72013-01-22 10:44:09 -08001714 def ping_wait_down(self, timeout):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001715 """Wait until the host no longer responds to `ping`.
1716
J. Richard Barnetteb6de7e32013-02-14 13:28:04 -08001717 This function can be used as a slightly faster version of
1718 `wait_down()`, by avoiding potentially long ssh timeouts.
1719
1720 @param timeout Minimum time to allow for the host to become
1721 non-responsive.
1722 @return True iff the host quit answering ping before the
1723 timeout.
1724
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001725 """
J. Richard Barnetteb6de7e32013-02-14 13:28:04 -08001726 return self._ping_wait_for_status(self._PING_STATUS_DOWN, timeout)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001727
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08001728 def test_wait_for_sleep(self, sleep_timeout=None):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001729 """Wait for the client to enter low-power sleep mode.
1730
1731 The test for "is asleep" can't distinguish a system that is
1732 powered off; to confirm that the unit was asleep, it is
1733 necessary to force resume, and then call
1734 `test_wait_for_resume()`.
1735
1736 This function is expected to be called from a test as part
1737 of a sequence like the following:
1738
1739 ~~~~~~~~
1740 boot_id = host.get_boot_id()
1741 # trigger sleep on the host
1742 host.test_wait_for_sleep()
1743 # trigger resume on the host
1744 host.test_wait_for_resume(boot_id)
1745 ~~~~~~~~
1746
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08001747 @param sleep_timeout time limit in seconds to allow the host sleep.
1748
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001749 @exception TestFail The host did not go to sleep within
1750 the allowed time.
1751 """
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08001752 if sleep_timeout is None:
1753 sleep_timeout = self.SLEEP_TIMEOUT
1754
1755 if not self.ping_wait_down(timeout=sleep_timeout):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001756 raise error.TestFail(
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08001757 'client failed to sleep after %d seconds' % sleep_timeout)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001758
1759
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08001760 def test_wait_for_resume(self, old_boot_id, resume_timeout=None):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001761 """Wait for the client to resume from low-power sleep mode.
1762
1763 The `old_boot_id` parameter should be the value from
1764 `get_boot_id()` obtained prior to entering sleep mode. A
1765 `TestFail` exception is raised if the boot id changes.
1766
1767 See @ref test_wait_for_sleep for more on this function's
1768 usage.
1769
J. Richard Barnette7214e0b2013-02-06 15:20:49 -08001770 @param old_boot_id A boot id value obtained before the
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001771 target host went to sleep.
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08001772 @param resume_timeout time limit in seconds to allow the host up.
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001773
1774 @exception TestFail The host did not respond within the
1775 allowed time.
1776 @exception TestFail The host responded, but the boot id test
1777 indicated a reboot rather than a sleep
1778 cycle.
1779 """
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08001780 if resume_timeout is None:
1781 resume_timeout = self.RESUME_TIMEOUT
1782
1783 if not self.wait_up(timeout=resume_timeout):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001784 raise error.TestFail(
1785 'client failed to resume from sleep after %d seconds' %
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08001786 resume_timeout)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001787 else:
1788 new_boot_id = self.get_boot_id()
1789 if new_boot_id != old_boot_id:
Tom Wai-Hong Tam01792682015-01-06 08:00:46 +08001790 logging.error('client rebooted (old boot %s, new boot %s)',
1791 old_boot_id, new_boot_id)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001792 raise error.TestFail(
Tom Wai-Hong Tam01792682015-01-06 08:00:46 +08001793 'client rebooted, but sleep was expected')
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001794
1795
Tom Wai-Hong Tamfe005c22014-12-03 09:25:44 +08001796 def test_wait_for_shutdown(self, shutdown_timeout=None):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001797 """Wait for the client to shut down.
1798
1799 The test for "has shut down" can't distinguish a system that
1800 is merely asleep; to confirm that the unit was down, it is
1801 necessary to force boot, and then call test_wait_for_boot().
1802
1803 This function is expected to be called from a test as part
1804 of a sequence like the following:
1805
1806 ~~~~~~~~
1807 boot_id = host.get_boot_id()
1808 # trigger shutdown on the host
1809 host.test_wait_for_shutdown()
1810 # trigger boot on the host
1811 host.test_wait_for_boot(boot_id)
1812 ~~~~~~~~
1813
Tom Wai-Hong Tamfe005c22014-12-03 09:25:44 +08001814 @param shutdown_timeout time limit in seconds to allow the host down.
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001815 @exception TestFail The host did not shut down within the
1816 allowed time.
1817 """
Tom Wai-Hong Tamfe005c22014-12-03 09:25:44 +08001818 if shutdown_timeout is None:
1819 shutdown_timeout = self.SHUTDOWN_TIMEOUT
1820
1821 if not self.ping_wait_down(timeout=shutdown_timeout):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001822 raise error.TestFail(
1823 'client failed to shut down after %d seconds' %
Tom Wai-Hong Tamfe005c22014-12-03 09:25:44 +08001824 shutdown_timeout)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001825
1826
1827 def test_wait_for_boot(self, old_boot_id=None):
1828 """Wait for the client to boot from cold power.
1829
1830 The `old_boot_id` parameter should be the value from
1831 `get_boot_id()` obtained prior to shutting down. A
1832 `TestFail` exception is raised if the boot id does not
1833 change. The boot id test is omitted if `old_boot_id` is not
1834 specified.
1835
1836 See @ref test_wait_for_shutdown for more on this function's
1837 usage.
1838
J. Richard Barnette7214e0b2013-02-06 15:20:49 -08001839 @param old_boot_id A boot id value obtained before the
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001840 shut down.
1841
1842 @exception TestFail The host did not respond within the
1843 allowed time.
1844 @exception TestFail The host responded, but the boot id test
1845 indicated that there was no reboot.
1846 """
J. Richard Barnetteeb69d722012-06-18 17:29:44 -07001847 if not self.wait_up(timeout=self.REBOOT_TIMEOUT):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001848 raise error.TestFail(
1849 'client failed to reboot after %d seconds' %
J. Richard Barnetteeb69d722012-06-18 17:29:44 -07001850 self.REBOOT_TIMEOUT)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001851 elif old_boot_id:
1852 if self.get_boot_id() == old_boot_id:
Tom Wai-Hong Tam01792682015-01-06 08:00:46 +08001853 logging.error('client not rebooted (boot %s)',
1854 old_boot_id)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001855 raise error.TestFail(
Tom Wai-Hong Tam01792682015-01-06 08:00:46 +08001856 'client is back up, but did not reboot')
Simran Basid5e5e272012-09-24 15:23:59 -07001857
1858
1859 @staticmethod
1860 def check_for_rpm_support(hostname):
1861 """For a given hostname, return whether or not it is powered by an RPM.
1862
Simran Basi1df55112013-09-06 11:25:09 -07001863 @param hostname: hostname to check for rpm support.
1864
Simran Basid5e5e272012-09-24 15:23:59 -07001865 @return None if this host does not follows the defined naming format
1866 for RPM powered DUT's in the lab. If it does follow the format,
1867 it returns a regular expression MatchObject instead.
1868 """
Fang Dengbaff9082015-01-06 13:46:15 -08001869 return re.match(CrosHost._RPM_HOSTNAME_REGEX, hostname)
Simran Basid5e5e272012-09-24 15:23:59 -07001870
1871
1872 def has_power(self):
1873 """For this host, return whether or not it is powered by an RPM.
1874
1875 @return True if this host is in the CROS lab and follows the defined
1876 naming format.
1877 """
Fang Deng0ca40e22013-08-27 17:47:44 -07001878 return CrosHost.check_for_rpm_support(self.hostname)
Simran Basid5e5e272012-09-24 15:23:59 -07001879
1880
Ismail Noorbasha07fdb612013-02-14 14:13:31 -08001881 def _set_power(self, state, power_method):
1882 """Sets the power to the host via RPM, Servo or manual.
1883
1884 @param state Specifies which power state to set to DUT
1885 @param power_method Specifies which method of power control to
1886 use. By default "RPM" will be used. Valid values
1887 are the strings "RPM", "manual", "servoj10".
1888
1889 """
1890 ACCEPTABLE_STATES = ['ON', 'OFF']
1891
1892 if state.upper() not in ACCEPTABLE_STATES:
1893 raise error.TestError('State must be one of: %s.'
1894 % (ACCEPTABLE_STATES,))
1895
1896 if power_method == self.POWER_CONTROL_SERVO:
1897 logging.info('Setting servo port J10 to %s', state)
1898 self.servo.set('prtctl3_pwren', state.lower())
1899 time.sleep(self._USB_POWER_TIMEOUT)
1900 elif power_method == self.POWER_CONTROL_MANUAL:
1901 logging.info('You have %d seconds to set the AC power to %s.',
1902 self._POWER_CYCLE_TIMEOUT, state)
1903 time.sleep(self._POWER_CYCLE_TIMEOUT)
1904 else:
1905 if not self.has_power():
1906 raise error.TestFail('DUT does not have RPM connected.')
Simran Basi5e6339a2013-03-21 11:34:32 -07001907 afe = frontend_wrappers.RetryingAFE(timeout_min=5, delay_sec=10)
1908 afe.set_host_attribute(self._RPM_OUTLET_CHANGED, True,
1909 hostname=self.hostname)
Simran Basi1df55112013-09-06 11:25:09 -07001910 rpm_client.set_power(self.hostname, state.upper(), timeout_mins=5)
Simran Basid5e5e272012-09-24 15:23:59 -07001911
1912
Ismail Noorbasha07fdb612013-02-14 14:13:31 -08001913 def power_off(self, power_method=POWER_CONTROL_RPM):
1914 """Turn off power to this host via RPM, Servo or manual.
1915
1916 @param power_method Specifies which method of power control to
1917 use. By default "RPM" will be used. Valid values
1918 are the strings "RPM", "manual", "servoj10".
1919
1920 """
1921 self._set_power('OFF', power_method)
Simran Basid5e5e272012-09-24 15:23:59 -07001922
1923
Ismail Noorbasha07fdb612013-02-14 14:13:31 -08001924 def power_on(self, power_method=POWER_CONTROL_RPM):
1925 """Turn on power to this host via RPM, Servo or manual.
1926
1927 @param power_method Specifies which method of power control to
1928 use. By default "RPM" will be used. Valid values
1929 are the strings "RPM", "manual", "servoj10".
1930
1931 """
1932 self._set_power('ON', power_method)
1933
1934
1935 def power_cycle(self, power_method=POWER_CONTROL_RPM):
1936 """Cycle power to this host by turning it OFF, then ON.
1937
1938 @param power_method Specifies which method of power control to
1939 use. By default "RPM" will be used. Valid values
1940 are the strings "RPM", "manual", "servoj10".
1941
1942 """
1943 if power_method in (self.POWER_CONTROL_SERVO,
1944 self.POWER_CONTROL_MANUAL):
1945 self.power_off(power_method=power_method)
1946 time.sleep(self._POWER_CYCLE_TIMEOUT)
1947 self.power_on(power_method=power_method)
1948 else:
1949 rpm_client.set_power(self.hostname, 'CYCLE')
Simran Basic6f1f7a2012-10-16 10:47:46 -07001950
1951
1952 def get_platform(self):
1953 """Determine the correct platform label for this host.
1954
1955 @returns a string representing this host's platform.
1956 """
1957 crossystem = utils.Crossystem(self)
1958 crossystem.init()
1959 # Extract fwid value and use the leading part as the platform id.
1960 # fwid generally follow the format of {platform}.{firmware version}
1961 # Example: Alex.X.YYY.Z or Google_Alex.X.YYY.Z
1962 platform = crossystem.fwid().split('.')[0].lower()
1963 # Newer platforms start with 'Google_' while the older ones do not.
1964 return platform.replace('google_', '')
1965
1966
Hung-ying Tyanb1328032014-04-01 14:18:54 +08001967 def get_architecture(self):
1968 """Determine the correct architecture label for this host.
1969
1970 @returns a string representing this host's architecture.
1971 """
1972 crossystem = utils.Crossystem(self)
1973 crossystem.init()
1974 return crossystem.arch()
1975
1976
Luis Lozano40b7d0d2014-01-17 15:12:06 -08001977 def get_chrome_version(self):
1978 """Gets the Chrome version number and milestone as strings.
1979
1980 Invokes "chrome --version" to get the version number and milestone.
1981
1982 @return A tuple (chrome_ver, milestone) where "chrome_ver" is the
1983 current Chrome version number as a string (in the form "W.X.Y.Z")
1984 and "milestone" is the first component of the version number
1985 (the "W" from "W.X.Y.Z"). If the version number cannot be parsed
1986 in the "W.X.Y.Z" format, the "chrome_ver" will be the full output
1987 of "chrome --version" and the milestone will be the empty string.
1988
1989 """
MK Ryu35d661e2014-09-25 17:44:10 -07001990 version_string = self.run(client_constants.CHROME_VERSION_COMMAND).stdout
Luis Lozano40b7d0d2014-01-17 15:12:06 -08001991 return utils.parse_chrome_version(version_string)
1992
J. Richard Barnetted2af5852016-02-05 15:03:10 -08001993
Niranjan Kumar34618872017-05-31 12:57:09 -07001994 def is_chrome_switch_present(self, switch):
David Haddock3ce538e2017-06-22 13:37:05 -07001995 """Returns True if the specified switch was provided to Chrome.
1996
1997 @param switch The chrome switch to search for.
1998 """
Niranjan Kumar34618872017-05-31 12:57:09 -07001999
Niranjan Kumar5f23fe92017-06-22 15:18:55 -07002000 command = 'pgrep -x -f -c "/opt/google/chrome/chrome.*%s.*"' % switch
2001 return self.run(command, ignore_status=True).exit_status == 0
Niranjan Kumar34618872017-05-31 12:57:09 -07002002
2003
2004 def oobe_triggers_update(self):
2005 """Returns True if this host has an OOBE flow during which
2006 it will perform an update check and perhaps an update.
2007 One example of such a flow is Hands-Off Zero-Touch Enrollment.
2008 As more such flows are developed, code handling them needs
2009 to be added here.
2010
2011 @return Boolean indicating whether this host's OOBE triggers an update.
2012 """
2013 return self.is_chrome_switch_present(
2014 '--enterprise-enable-zero-touch-enrollment=hands-off')
2015
2016
Kevin Chenga2619dc2016-03-28 11:42:08 -07002017 # TODO(kevcheng): change this to just return the board without the
2018 # 'board:' prefix and fix up all the callers. Also look into removing the
2019 # need for this method.
Simran Basic6f1f7a2012-10-16 10:47:46 -07002020 def get_board(self):
2021 """Determine the correct board label for this host.
2022
2023 @returns a string representing this host's board.
2024 """
2025 release_info = utils.parse_cmd_output('cat /etc/lsb-release',
2026 run_method=self.run)
J. Richard Barnetted2af5852016-02-05 15:03:10 -08002027 return (ds_constants.BOARD_PREFIX +
2028 release_info['CHROMEOS_RELEASE_BOARD'])
Simran Basic6f1f7a2012-10-16 10:47:46 -07002029
2030
Kevin Chenga328da62016-03-31 10:49:04 -07002031
2032 def has_lightsensor(self):
2033 """Determine the correct board label for this host.
2034
2035 @returns the string 'lightsensor' if this host has a lightsensor or
2036 None if it does not.
2037 """
2038 search_cmd = "find -L %s -maxdepth 4 | egrep '%s'" % (
2039 self._LIGHTSENSOR_SEARCH_DIR, '|'.join(self._LIGHTSENSOR_FILES))
2040 try:
2041 # Run the search cmd following the symlinks. Stderr_tee is set to
2042 # None as there can be a symlink loop, but the command will still
2043 # execute correctly with a few messages printed to stderr.
2044 self.run(search_cmd, stdout_tee=None, stderr_tee=None)
2045 return 'lightsensor'
2046 except error.AutoservRunError:
2047 # egrep exited with a return code of 1 meaning none of the possible
2048 # lightsensor files existed.
2049 return None
2050
2051
2052 def has_bluetooth(self):
2053 """Determine the correct board label for this host.
2054
2055 @returns the string 'bluetooth' if this host has bluetooth or
2056 None if it does not.
2057 """
2058 try:
2059 self.run('test -d /sys/class/bluetooth/hci0')
2060 # test exited with a return code of 0.
2061 return 'bluetooth'
2062 except error.AutoservRunError:
2063 # test exited with a return code 1 meaning the directory did not
2064 # exist.
2065 return None
2066
2067
Kevin Chenga328da62016-03-31 10:49:04 -07002068 def get_accels(self):
2069 """
2070 Determine the type of accelerometers on this host.
2071
2072 @returns a string representing this host's accelerometer type.
2073 At present, it only returns "accel:cros-ec", for accelerometers
2074 attached to a Chrome OS EC, or none, if no accelerometers.
2075 """
2076 # Check to make sure we have ectool
2077 rv = self.run('which ectool', ignore_status=True)
2078 if rv.exit_status:
2079 logging.info("No ectool cmd found, assuming no EC accelerometers")
2080 return None
2081
2082 # Check that the EC supports the motionsense command
2083 rv = self.run('ectool motionsense', ignore_status=True)
2084 if rv.exit_status:
2085 logging.info("EC does not support motionsense command "
2086 "assuming no EC accelerometers")
2087 return None
2088
2089 # Check that EC motion sensors are active
2090 active = self.run('ectool motionsense active').stdout.split('\n')
2091 if active[0] == "0":
2092 logging.info("Motion sense inactive, assuming no EC accelerometers")
2093 return None
2094
2095 logging.info("EC accelerometers found")
2096 return 'accel:cros-ec'
2097
2098
2099 def has_chameleon(self):
2100 """Determine if a Chameleon connected to this host.
2101
2102 @returns a list containing two strings ('chameleon' and
2103 'chameleon:' + label, e.g. 'chameleon:hdmi') if this host
2104 has a Chameleon or None if it has not.
2105 """
2106 if self._chameleon_host:
2107 return ['chameleon', 'chameleon:' + self.chameleon.get_label()]
2108 else:
2109 return None
2110
2111
2112 def has_loopback_dongle(self):
2113 """Determine if an audio loopback dongle is plugged to this host.
2114
2115 @returns 'audio_loopback_dongle' when there is an audio loopback dongle
2116 plugged to this host.
2117 None when there is no audio loopback dongle
2118 plugged to this host.
2119 """
2120 nodes_info = self.run(command=cras_utils.get_cras_nodes_cmd(),
2121 ignore_status=True).stdout
2122 if (cras_utils.node_type_is_plugged('HEADPHONE', nodes_info) and
2123 cras_utils.node_type_is_plugged('MIC', nodes_info)):
2124 return 'audio_loopback_dongle'
2125 else:
2126 return None
2127
2128
2129 def get_power_supply(self):
2130 """
2131 Determine what type of power supply the host has
2132
2133 @returns a string representing this host's power supply.
2134 'power:battery' when the device has a battery intended for
2135 extended use
2136 'power:AC_primary' when the device has a battery not intended
2137 for extended use (for moving the machine, etc)
2138 'power:AC_only' when the device has no battery at all.
2139 """
2140 psu = self.run(command='mosys psu type', ignore_status=True)
2141 if psu.exit_status:
2142 # The psu command for mosys is not included for all platforms. The
2143 # assumption is that the device will have a battery if the command
2144 # is not found.
2145 return 'power:battery'
2146
2147 psu_str = psu.stdout.strip()
2148 if psu_str == 'unknown':
2149 return None
2150
2151 return 'power:%s' % psu_str
2152
2153
2154 def get_storage(self):
2155 """
2156 Determine the type of boot device for this host.
2157
2158 Determine if the internal device is SCSI or dw_mmc device.
2159 Then check that it is SSD or HDD or eMMC or something else.
2160
2161 @returns a string representing this host's internal device type.
2162 'storage:ssd' when internal device is solid state drive
2163 'storage:hdd' when internal device is hard disk drive
2164 'storage:mmc' when internal device is mmc drive
2165 None When internal device is something else or
2166 when we are unable to determine the type
2167 """
2168 # The output should be /dev/mmcblk* for SD/eMMC or /dev/sd* for scsi
2169 rootdev_cmd = ' '.join(['. /usr/sbin/write_gpt.sh;',
2170 '. /usr/share/misc/chromeos-common.sh;',
2171 'load_base_vars;',
2172 'get_fixed_dst_drive'])
2173 rootdev = self.run(command=rootdev_cmd, ignore_status=True)
2174 if rootdev.exit_status:
2175 logging.info("Fail to run %s", rootdev_cmd)
2176 return None
2177 rootdev_str = rootdev.stdout.strip()
2178
2179 if not rootdev_str:
2180 return None
2181
2182 rootdev_base = os.path.basename(rootdev_str)
2183
2184 mmc_pattern = '/dev/mmcblk[0-9]'
2185 if re.match(mmc_pattern, rootdev_str):
2186 # Use type to determine if the internal device is eMMC or somthing
2187 # else. We can assume that MMC is always an internal device.
2188 type_cmd = 'cat /sys/block/%s/device/type' % rootdev_base
2189 type = self.run(command=type_cmd, ignore_status=True)
2190 if type.exit_status:
2191 logging.info("Fail to run %s", type_cmd)
2192 return None
2193 type_str = type.stdout.strip()
2194
2195 if type_str == 'MMC':
2196 return 'storage:mmc'
2197
2198 scsi_pattern = '/dev/sd[a-z]+'
2199 if re.match(scsi_pattern, rootdev.stdout):
2200 # Read symlink for /sys/block/sd* to determine if the internal
2201 # device is connected via ata or usb.
2202 link_cmd = 'readlink /sys/block/%s' % rootdev_base
2203 link = self.run(command=link_cmd, ignore_status=True)
2204 if link.exit_status:
2205 logging.info("Fail to run %s", link_cmd)
2206 return None
2207 link_str = link.stdout.strip()
2208 if 'usb' in link_str:
2209 return None
2210
2211 # Read rotation to determine if the internal device is ssd or hdd.
2212 rotate_cmd = str('cat /sys/block/%s/queue/rotational'
2213 % rootdev_base)
2214 rotate = self.run(command=rotate_cmd, ignore_status=True)
2215 if rotate.exit_status:
2216 logging.info("Fail to run %s", rotate_cmd)
2217 return None
2218 rotate_str = rotate.stdout.strip()
2219
2220 rotate_dict = {'0':'storage:ssd', '1':'storage:hdd'}
2221 return rotate_dict.get(rotate_str)
2222
2223 # All other internal device / error case will always fall here
2224 return None
2225
2226
2227 def get_servo(self):
2228 """Determine if the host has a servo attached.
2229
2230 If the host has a working servo attached, it should have a servo label.
2231
2232 @return: string 'servo' if the host has servo attached. Otherwise,
2233 returns None.
2234 """
2235 return 'servo' if self._servo_host else None
2236
2237
2238 def get_video_labels(self):
2239 """Run /usr/local/bin/avtest_label_detect to get a list of video labels.
2240
2241 Sample output of avtest_label_detect:
2242 Detected label: hw_video_acc_vp8
2243 Detected label: webcam
2244
2245 @return: A list of labels detected by tool avtest_label_detect.
2246 """
2247 try:
2248 result = self.run('/usr/local/bin/avtest_label_detect').stdout
2249 return re.findall('^Detected label: (\w+)$', result, re.M)
2250 except error.AutoservRunError:
2251 # The tool is not installed.
2252 return []
2253
2254
2255 def is_video_glitch_detection_supported(self):
2256 """ Determine if a board under test is supported for video glitch
2257 detection tests.
2258
2259 @return: 'video_glitch_detection' if board is supported, None otherwise.
2260 """
2261 board = self.get_board().replace(ds_constants.BOARD_PREFIX, '')
2262
2263 if board in video_test_constants.SUPPORTED_BOARDS:
2264 return 'video_glitch_detection'
2265
2266 return None
2267
2268
2269 def get_touch(self):
2270 """
2271 Determine whether board under test has a touchpad or touchscreen.
2272
2273 @return: A list of some combination of 'touchscreen' and 'touchpad',
2274 depending on what is present on the device.
2275
2276 """
2277 labels = []
2278 looking_for = ['touchpad', 'touchscreen']
2279 player = input_playback.InputPlayback()
2280 input_events = self.run('ls /dev/input/event*').stdout.strip().split()
2281 filename = '/tmp/touch_labels'
2282 for event in input_events:
2283 self.run('evtest %s > %s' % (event, filename), timeout=1,
2284 ignore_timeout=True)
2285 properties = self.run('cat %s' % filename).stdout
2286 input_type = player._determine_input_type(properties)
2287 if input_type in looking_for:
2288 labels.append(input_type)
2289 looking_for.remove(input_type)
2290 if len(looking_for) == 0:
2291 break
2292 self.run('rm %s' % filename)
2293
2294 return labels
2295
2296
2297 def has_internal_display(self):
2298 """Determine if the device under test is equipped with an internal
2299 display.
2300
2301 @return: 'internal_display' if one is present; None otherwise.
2302 """
2303 from autotest_lib.client.cros.graphics import graphics_utils
2304 from autotest_lib.client.common_lib import utils as common_utils
2305
2306 def __system_output(cmd):
2307 return self.run(cmd).stdout
2308
2309 def __read_file(remote_path):
2310 return self.run('cat %s' % remote_path).stdout
2311
2312 # Hijack the necessary client functions so that we can take advantage
2313 # of the client lib here.
2314 # FIXME: find a less hacky way than this
2315 original_system_output = utils.system_output
2316 original_read_file = common_utils.read_file
2317 utils.system_output = __system_output
2318 common_utils.read_file = __read_file
2319 try:
2320 return ('internal_display' if graphics_utils.has_internal_display()
2321 else None)
2322 finally:
2323 utils.system_output = original_system_output
2324 common_utils.read_file = original_read_file
2325
2326
2327 def has_lucid_sleep_support(self):
2328 """Determine if the device under test has support for lucid sleep.
2329
2330 @return 'lucidsleep' if this board supports lucid sleep; None otherwise
2331 """
2332 board = self.get_board().replace(ds_constants.BOARD_PREFIX, '')
2333 return 'lucidsleep' if board in LUCID_SLEEP_BOARDS else None
2334
2335
Dan Shi85276d42014-04-08 22:11:45 -07002336 def is_boot_from_usb(self):
2337 """Check if DUT is boot from USB.
2338
2339 @return: True if DUT is boot from usb.
2340 """
2341 device = self.run('rootdev -s -d').stdout.strip()
2342 removable = int(self.run('cat /sys/block/%s/removable' %
2343 os.path.basename(device)).stdout.strip())
2344 return removable == 1
Helen Zhang17dae2b2014-11-11 09:25:52 -08002345
2346
2347 def read_from_meminfo(self, key):
Dan Shi49ca0932014-11-14 11:22:27 -08002348 """Return the memory info from /proc/meminfo
Helen Zhang17dae2b2014-11-11 09:25:52 -08002349
2350 @param key: meminfo requested
2351
2352 @return the memory value as a string
2353
2354 """
Helen Zhang17dae2b2014-11-11 09:25:52 -08002355 meminfo = self.run('grep %s /proc/meminfo' % key).stdout.strip()
2356 logging.debug('%s', meminfo)
2357 return int(re.search(r'\d+', meminfo).group(0))
Rohit Makasana8a4923c2015-08-13 17:04:26 -07002358
2359
Rohit Makasana98e696f2016-06-03 18:48:10 -07002360 def get_cpu_arch(self):
2361 """Returns CPU arch of the device.
2362
2363 @return CPU architecture of the DUT.
2364 """
Allen Li2c32d6b2017-02-03 15:28:10 -08002365 # Add CPUs by following logic in client/bin/utils.py.
Rohit Makasana98e696f2016-06-03 18:48:10 -07002366 if self.run("grep '^flags.*:.* lm .*' /proc/cpuinfo",
2367 ignore_status=True).stdout:
2368 return 'x86_64'
2369 if self.run("grep -Ei 'ARM|CPU implementer' /proc/cpuinfo",
2370 ignore_status=True).stdout:
2371 return 'arm'
2372 return 'i386'
2373
2374
Rohit Makasana8a4923c2015-08-13 17:04:26 -07002375 def get_board_type(self):
2376 """
2377 Get the DUT's device type from /etc/lsb-release.
Danny Chan471a8d12015-08-18 14:57:41 -07002378 DEVICETYPE can be one of CHROMEBOX, CHROMEBASE, CHROMEBOOK or more.
2379
2380 @return value of DEVICETYPE param from lsb-release.
Rohit Makasana8a4923c2015-08-13 17:04:26 -07002381 """
Danny Chan471a8d12015-08-18 14:57:41 -07002382 device_type = self.run('grep DEVICETYPE /etc/lsb-release',
2383 ignore_status=True).stdout
2384 if device_type:
Kalin Stoyanov524310b2015-08-21 16:24:04 -07002385 return device_type.split('=')[-1].strip()
Danny Chan471a8d12015-08-18 14:57:41 -07002386 return ''
Gilad Arnolda76bef02015-09-29 13:55:15 -07002387
2388
2389 def get_os_type(self):
2390 return 'cros'
Simran Basia5522a32015-10-06 11:01:24 -07002391
2392
2393 def enable_adb_testing(self):
2394 """Mark this host as an adb tester."""
Dan Shia2872172015-10-31 01:16:51 -07002395 self.run('touch %s' % constants.ANDROID_TESTER_FILEFLAG)
Kevin Chenga2619dc2016-03-28 11:42:08 -07002396
2397
2398 def get_labels(self):
Kevin Chengf8660142016-08-12 10:17:41 -07002399 """Return the detected labels on the host."""
Kevin Chenga2619dc2016-03-28 11:42:08 -07002400 return self.labels.get_labels(self)
Kevin Chengf8660142016-08-12 10:17:41 -07002401
2402
2403 def update_labels(self):
2404 """Update the labels on the host."""
2405 self.labels.update_labels(self)