blob: 1229e76b5441b621b2b3429708f2f6008cf56878 [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
Hsinyu Chaoe0b08e62015-08-11 10:50:37 +000020from autotest_lib.client.cros import constants as client_constants
J. Richard Barnette84890bd2014-02-21 11:05:47 -080021from autotest_lib.client.cros import cros_ui
Kevin Chenga328da62016-03-31 10:49:04 -070022from autotest_lib.client.cros.audio import cras_utils
23from autotest_lib.client.cros.input_playback import input_playback
24from autotest_lib.client.cros.video import constants as video_test_constants
Simran Basi5ace6f22016-01-06 17:30:44 -080025from autotest_lib.server import afe_utils
MK Ryu35d661e2014-09-25 17:44:10 -070026from autotest_lib.server import autoserv_parser
27from autotest_lib.server import autotest
28from autotest_lib.server import constants
Dan Shia1ecd5c2013-06-06 11:21:31 -070029from autotest_lib.server import utils as server_utils
Dan Shi9cb0eec2014-06-03 09:04:50 -070030from autotest_lib.server.cros import provision
Scott Zawalski89c44dd2013-02-26 09:28:02 -050031from autotest_lib.server.cros.dynamic_suite import constants as ds_constants
Simran Basi5e6339a2013-03-21 11:34:32 -070032from autotest_lib.server.cros.dynamic_suite import tools, frontend_wrappers
Dan Shi9cb0eec2014-06-03 09:04:50 -070033from autotest_lib.server.cros.faft.config.config import Config as FAFTConfig
Scottfe06ed82015-11-05 17:15:01 -080034from autotest_lib.server.cros.servo import plankton
Fang Deng96667ca2013-08-01 17:46:18 -070035from autotest_lib.server.hosts import abstract_ssh
Kevin Chenga2619dc2016-03-28 11:42:08 -070036from autotest_lib.server.hosts import base_label
37from autotest_lib.server.hosts import cros_label
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +080038from autotest_lib.server.hosts import chameleon_host
J. Richard Barnettea7e7fdc2016-02-12 12:35:36 -080039from autotest_lib.server.hosts import cros_repair
Scottfe06ed82015-11-05 17:15:01 -080040from autotest_lib.server.hosts import plankton_host
Fang Deng5d518f42013-08-02 14:04:32 -070041from autotest_lib.server.hosts import servo_host
Simran Basidcff4252012-11-20 16:13:20 -080042from autotest_lib.site_utils.rpm_control_system import rpm_client
Simran Basid5e5e272012-09-24 15:23:59 -070043
Simran Basi382506b2016-09-13 14:58:15 -070044# In case cros_host is being ran via SSP on an older Moblab version with an
45# older chromite version.
46try:
47 from chromite.lib import metrics
Dan Shi5e2efb72017-02-07 11:40:23 -080048except ImportError:
49 metrics = utils.metrics_mock
50
Simran Basid5e5e272012-09-24 15:23:59 -070051
Dan Shib8540a52015-07-16 14:18:23 -070052CONFIG = global_config.global_config
xixuanec801e32016-08-25 10:20:22 -070053ENABLE_DEVSERVER_TRIGGER_AUTO_UPDATE = CONFIG.get_config_value(
54 'CROS', 'enable_devserver_trigger_auto_update', type=bool,
55 default=False)
Dan Shib8540a52015-07-16 14:18:23 -070056
Dan Shid07ee2e2015-09-24 14:49:25 -070057
beepsc87ff602013-07-31 21:53:00 -070058class FactoryImageCheckerException(error.AutoservError):
59 """Exception raised when an image is a factory image."""
60 pass
61
62
Fang Deng0ca40e22013-08-27 17:47:44 -070063class CrosHost(abstract_ssh.AbstractSSHHost):
J. Richard Barnette45e93de2012-04-11 17:24:15 -070064 """Chromium OS specific subclass of Host."""
65
Simran Basi5ace6f22016-01-06 17:30:44 -080066 VERSION_PREFIX = provision.CROS_VERSION_PREFIX
67
J. Richard Barnette45e93de2012-04-11 17:24:15 -070068 _parser = autoserv_parser.autoserv_parser
Scott Zawalski62bacae2013-03-05 10:40:32 -050069 _AFE = frontend_wrappers.RetryingAFE(timeout_min=5, delay_sec=10)
xixuanec801e32016-08-25 10:20:22 -070070 support_devserver_provision = ENABLE_DEVSERVER_TRIGGER_AUTO_UPDATE
J. Richard Barnette45e93de2012-04-11 17:24:15 -070071
Richard Barnette03a0c132012-11-05 12:40:35 -080072 # Timeout values (in seconds) associated with various Chrome OS
73 # state changes.
J. Richard Barnette134ec2c2012-04-25 12:59:37 -070074 #
Richard Barnette0c73ffc2012-11-19 15:21:18 -080075 # In general, a good rule of thumb is that the timeout can be up
76 # to twice the typical measured value on the slowest platform.
77 # The times here have not necessarily been empirically tested to
78 # meet this criterion.
J. Richard Barnetteeb69d722012-06-18 17:29:44 -070079 #
80 # SLEEP_TIMEOUT: Time to allow for suspend to memory.
Richard Barnette0c73ffc2012-11-19 15:21:18 -080081 # RESUME_TIMEOUT: Time to allow for resume after suspend, plus
82 # time to restart the netwowrk.
J. Richard Barnette84890bd2014-02-21 11:05:47 -080083 # SHUTDOWN_TIMEOUT: Time to allow for shut down.
J. Richard Barnetteeb69d722012-06-18 17:29:44 -070084 # BOOT_TIMEOUT: Time to allow for boot from power off. Among
Richard Barnette0c73ffc2012-11-19 15:21:18 -080085 # other things, this must account for the 30 second dev-mode
J. Richard Barnette417cc792015-10-01 09:56:36 -070086 # screen delay, time to start the network on the DUT, and the
87 # ssh timeout of 120 seconds.
J. Richard Barnetteeb69d722012-06-18 17:29:44 -070088 # USB_BOOT_TIMEOUT: Time to allow for boot from a USB device,
Richard Barnette0c73ffc2012-11-19 15:21:18 -080089 # including the 30 second dev-mode delay and time to start the
J. Richard Barnetted4649c62013-03-06 17:42:27 -080090 # network.
beepsf079cfb2013-09-18 17:49:51 -070091 # INSTALL_TIMEOUT: Time to allow for chromeos-install.
J. Richard Barnette84890bd2014-02-21 11:05:47 -080092 # POWERWASH_BOOT_TIMEOUT: Time to allow for a reboot that
93 # includes powerwash.
J. Richard Barnetteeb69d722012-06-18 17:29:44 -070094
95 SLEEP_TIMEOUT = 2
J. Richard Barnetted4649c62013-03-06 17:42:27 -080096 RESUME_TIMEOUT = 10
Tom Wai-Hong Tamfe005c22014-12-03 09:25:44 +080097 SHUTDOWN_TIMEOUT = 10
J. Richard Barnette417cc792015-10-01 09:56:36 -070098 BOOT_TIMEOUT = 150
J. Richard Barnette5bab5f52015-08-03 13:14:38 -070099 USB_BOOT_TIMEOUT = 300
J. Richard Barnette7817b052014-08-28 09:47:29 -0700100 INSTALL_TIMEOUT = 480
Dan Shi2c88eed2013-11-12 10:18:38 -0800101 POWERWASH_BOOT_TIMEOUT = 60
Chris Sosab76e0ee2013-05-22 16:55:41 -0700102
Dan Shica503482015-03-30 17:23:25 -0700103 # Minimum OS version that supports server side packaging. Older builds may
104 # not have server side package built or with Autotest code change to support
105 # server-side packaging.
Dan Shib8540a52015-07-16 14:18:23 -0700106 MIN_VERSION_SUPPORT_SSP = CONFIG.get_config_value(
Dan Shiced09e42015-04-17 16:09:34 -0700107 'AUTOSERV', 'min_version_support_ssp', type=int)
Dan Shica503482015-03-30 17:23:25 -0700108
J. Richard Barnette84890bd2014-02-21 11:05:47 -0800109 # REBOOT_TIMEOUT: How long to wait for a reboot.
110 #
Chris Sosab76e0ee2013-05-22 16:55:41 -0700111 # We have a long timeout to ensure we don't flakily fail due to other
112 # issues. Shorter timeouts are vetted in platform_RebootAfterUpdate.
Simran Basi1160e2c2013-10-04 16:00:24 -0700113 # TODO(sbasi - crbug.com/276094) Restore to 5 mins once the 'host did not
114 # return from reboot' bug is solved.
115 REBOOT_TIMEOUT = 480
Chris Sosab76e0ee2013-05-22 16:55:41 -0700116
Ismail Noorbasha07fdb612013-02-14 14:13:31 -0800117 # _USB_POWER_TIMEOUT: Time to allow for USB to power toggle ON and OFF.
118 # _POWER_CYCLE_TIMEOUT: Time to allow for manual power cycle.
119 _USB_POWER_TIMEOUT = 5
120 _POWER_CYCLE_TIMEOUT = 10
121
Richard Barnette82c35912012-11-20 10:09:10 -0800122 _LAB_MACHINE_FILE = '/mnt/stateful_partition/.labmachine'
Fang Dengdeba14f2014-11-14 11:54:09 -0800123 _RPM_HOSTNAME_REGEX = ('chromeos(\d+)(-row(\d+))?-rack(\d+[a-z]*)'
124 '-host(\d+)')
Kevin Chenga328da62016-03-31 10:49:04 -0700125 _LIGHTSENSOR_FILES = [ "in_illuminance0_input",
126 "in_illuminance_input",
127 "in_illuminance0_raw",
128 "in_illuminance_raw",
129 "illuminance0_input"]
130 _LIGHTSENSOR_SEARCH_DIR = '/sys/bus/iio/devices'
J. Richard Barnette134ec2c2012-04-25 12:59:37 -0700131
J. Richard Barnetteb6de7e32013-02-14 13:28:04 -0800132 # Constants used in ping_wait_up() and ping_wait_down().
133 #
134 # _PING_WAIT_COUNT is the approximate number of polling
135 # cycles to use when waiting for a host state change.
136 #
137 # _PING_STATUS_DOWN and _PING_STATUS_UP are names used
138 # for arguments to the internal _ping_wait_for_status()
139 # method.
140 _PING_WAIT_COUNT = 40
141 _PING_STATUS_DOWN = False
142 _PING_STATUS_UP = True
143
Ismail Noorbasha07fdb612013-02-14 14:13:31 -0800144 # Allowed values for the power_method argument.
145
146 # POWER_CONTROL_RPM: Passed as default arg for power_off/on/cycle() methods.
147 # POWER_CONTROL_SERVO: Used in set_power() and power_cycle() methods.
148 # POWER_CONTROL_MANUAL: Used in set_power() and power_cycle() methods.
149 POWER_CONTROL_RPM = 'RPM'
150 POWER_CONTROL_SERVO = 'servoj10'
151 POWER_CONTROL_MANUAL = 'manual'
152
153 POWER_CONTROL_VALID_ARGS = (POWER_CONTROL_RPM,
154 POWER_CONTROL_SERVO,
155 POWER_CONTROL_MANUAL)
Richard Barnette0c73ffc2012-11-19 15:21:18 -0800156
Simran Basi5e6339a2013-03-21 11:34:32 -0700157 _RPM_OUTLET_CHANGED = 'outlet_changed'
158
Dan Shi9cb0eec2014-06-03 09:04:50 -0700159 # URL pattern to download firmware image.
Dan Shib8540a52015-07-16 14:18:23 -0700160 _FW_IMAGE_URL_PATTERN = CONFIG.get_config_value(
Dan Shi9cb0eec2014-06-03 09:04:50 -0700161 'CROS', 'firmware_url_pattern', type=str)
beeps687243d2013-07-18 15:29:27 -0700162
J. Richard Barnette91137f02016-03-10 16:52:26 -0800163
164 # A flag file to indicate provision failures. The file is created
165 # at the start of any AU procedure (see `machine_install()`). The
166 # file's location in stateful means that on successul update it will
167 # be removed. Thus, if this file exists, it indicates that we've
168 # tried and failed in a previous attempt to update.
169 PROVISION_FAILED = '/var/tmp/provision_failed'
MK Ryu35d661e2014-09-25 17:44:10 -0700170
MK Ryu35d661e2014-09-25 17:44:10 -0700171
J. Richard Barnette964fba02012-10-24 17:34:29 -0700172 @staticmethod
beeps46dadc92013-11-07 14:07:10 -0800173 def check_host(host, timeout=10):
174 """
175 Check if the given host is a chrome-os host.
176
177 @param host: An ssh host representing a device.
178 @param timeout: The timeout for the run command.
179
180 @return: True if the host device is chromeos.
181
beeps46dadc92013-11-07 14:07:10 -0800182 """
183 try:
Allen Liad719c12017-06-27 23:48:04 +0000184 result = host.run(
Simran Basi933c8af2015-04-29 14:05:07 -0700185 'grep -q CHROMEOS /etc/lsb-release && '
186 '! test -f /mnt/stateful_partition/.android_tester && '
187 '! grep -q moblab /etc/lsb-release',
188 ignore_status=True, timeout=timeout)
Laurence Goodby468de252017-06-08 17:22:53 -0700189 if result.exit_status == 0:
Allen Liad719c12017-06-27 23:48:04 +0000190 lsb_release_content = host.run(
Laurence Goodby468de252017-06-08 17:22:53 -0700191 'grep CHROMEOS_RELEASE_BOARD /etc/lsb-release',
192 timeout=timeout).stdout
193 return not lsbrelease_utils.is_jetstream(
194 lsb_release_content=lsb_release_content)
beeps46dadc92013-11-07 14:07:10 -0800195 except (error.AutoservRunError, error.AutoservSSHTimeout):
196 return False
Laurence Goodby468de252017-06-08 17:22:53 -0700197
198 return False
beeps46dadc92013-11-07 14:07:10 -0800199
200
201 @staticmethod
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +0800202 def get_chameleon_arguments(args_dict):
203 """Extract chameleon options from `args_dict` and return the result.
204
205 Recommended usage:
206 ~~~~~~~~
207 args_dict = utils.args_to_dict(args)
208 chameleon_args = hosts.CrosHost.get_chameleon_arguments(args_dict)
209 host = hosts.create_host(machine, chameleon_args=chameleon_args)
210 ~~~~~~~~
211
212 @param args_dict Dictionary from which to extract the chameleon
213 arguments.
214 """
Allen Li083866b2016-08-18 10:07:10 -0700215 return {key: args_dict[key]
216 for key in ('chameleon_host', 'chameleon_port')
217 if key in args_dict}
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +0800218
219
220 @staticmethod
Scottfe06ed82015-11-05 17:15:01 -0800221 def get_plankton_arguments(args_dict):
222 """Extract chameleon options from `args_dict` and return the result.
223
224 Recommended usage:
225 ~~~~~~~~
226 args_dict = utils.args_to_dict(args)
Richard Barnettee519dcd2016-08-15 17:37:17 -0700227 plankton_args = hosts.CrosHost.get_plankton_arguments(args_dict)
228 host = hosts.create_host(machine, plankton_args=plankton_args)
Scottfe06ed82015-11-05 17:15:01 -0800229 ~~~~~~~~
230
231 @param args_dict Dictionary from which to extract the plankton
232 arguments.
233 """
Allen Li083866b2016-08-18 10:07:10 -0700234 return {key: args_dict[key]
235 for key in ('plankton_host', 'plankton_port')
236 if key in args_dict}
Scottfe06ed82015-11-05 17:15:01 -0800237
238
239 @staticmethod
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +0800240 def get_servo_arguments(args_dict):
241 """Extract servo options from `args_dict` and return the result.
J. Richard Barnette7214e0b2013-02-06 15:20:49 -0800242
243 Recommended usage:
244 ~~~~~~~~
245 args_dict = utils.args_to_dict(args)
Fang Deng0ca40e22013-08-27 17:47:44 -0700246 servo_args = hosts.CrosHost.get_servo_arguments(args_dict)
J. Richard Barnette7214e0b2013-02-06 15:20:49 -0800247 host = hosts.create_host(machine, servo_args=servo_args)
248 ~~~~~~~~
249
250 @param args_dict Dictionary from which to extract the servo
251 arguments.
252 """
Richard Barnettee519dcd2016-08-15 17:37:17 -0700253 servo_attrs = (servo_host.SERVO_HOST_ATTR,
254 servo_host.SERVO_PORT_ATTR,
255 servo_host.SERVO_BOARD_ATTR)
Armando Miraglia2ac802e2017-08-15 14:54:47 +0200256 servo_args = {key: args_dict[key]
257 for key in servo_attrs
258 if key in args_dict}
259 return (
260 None
261 if servo_host.SERVO_HOST_ATTR in servo_args
262 and not servo_args[servo_host.SERVO_HOST_ATTR]
263 else servo_args)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -0700264
J. Richard Barnette964fba02012-10-24 17:34:29 -0700265
J. Richard Barnette91137f02016-03-10 16:52:26 -0800266 def _initialize(self, hostname, chameleon_args=None, servo_args=None,
267 plankton_args=None, try_lab_servo=False,
Richard Barnette9a26ad62016-06-10 12:03:08 -0700268 try_servo_repair=False,
J. Richard Barnette91137f02016-03-10 16:52:26 -0800269 ssh_verbosity_flag='', ssh_options='',
Fang Dengd1c2b732013-08-20 12:59:46 -0700270 *args, **dargs):
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +0800271 """Initialize superclasses, |self.chameleon|, and |self.servo|.
J. Richard Barnette67ccb872012-04-19 16:34:56 -0700272
Fang Denge545abb2014-12-30 18:43:47 -0800273 This method will attempt to create the test-assistant object
274 (chameleon/servo) when it is needed by the test. Check
275 the docstring of chameleon_host.create_chameleon_host and
276 servo_host.create_servo_host for how this is determined.
Fang Deng5d518f42013-08-02 14:04:32 -0700277
Fang Denge545abb2014-12-30 18:43:47 -0800278 @param hostname: Hostname of the dut.
279 @param chameleon_args: A dictionary that contains args for creating
280 a ChameleonHost. See chameleon_host for details.
281 @param servo_args: A dictionary that contains args for creating
282 a ServoHost object. See servo_host for details.
Richard Barnette9a26ad62016-06-10 12:03:08 -0700283 @param try_lab_servo: When true, indicates that an attempt should
284 be made to create a ServoHost for a DUT in
285 the test lab, even if not required by
286 `servo_args`. See servo_host for details.
287 @param try_servo_repair: If a servo host is created, check it
288 with `repair()` rather than `verify()`.
Fang Denge545abb2014-12-30 18:43:47 -0800289 See servo_host for details.
290 @param ssh_verbosity_flag: String, to pass to the ssh command to control
291 verbosity.
292 @param ssh_options: String, other ssh options to pass to the ssh
293 command.
J. Richard Barnette67ccb872012-04-19 16:34:56 -0700294 """
Fang Deng0ca40e22013-08-27 17:47:44 -0700295 super(CrosHost, self)._initialize(hostname=hostname,
J. Richard Barnette45e93de2012-04-11 17:24:15 -0700296 *args, **dargs)
J. Richard Barnette91137f02016-03-10 16:52:26 -0800297 self._repair_strategy = cros_repair.create_cros_repair_strategy()
Kevin Chenga2619dc2016-03-28 11:42:08 -0700298 self.labels = base_label.LabelRetriever(cros_label.CROS_LABELS)
J. Richard Barnettef0859852012-08-20 14:55:50 -0700299 # self.env is a dictionary of environment variable settings
300 # to be exported for commands run on the host.
301 # LIBC_FATAL_STDERR_ can be useful for diagnosing certain
302 # errors that might happen.
303 self.env['LIBC_FATAL_STDERR_'] = '1'
Fang Dengd1c2b732013-08-20 12:59:46 -0700304 self._ssh_verbosity_flag = ssh_verbosity_flag
Aviv Keshetc5947fa2013-09-04 14:06:29 -0700305 self._ssh_options = ssh_options
Fang Deng5d518f42013-08-02 14:04:32 -0700306 # TODO(fdeng): We need to simplify the
307 # process of servo and servo_host initialization.
308 # crbug.com/298432
Moja Hsu57d93a12017-01-13 17:57:52 +0800309 self._servo_host = servo_host.create_servo_host(
Richard Barnetteea3e4602016-06-10 12:36:41 -0700310 dut=self, servo_args=servo_args,
Richard Barnette9a26ad62016-06-10 12:03:08 -0700311 try_lab_servo=try_lab_servo,
312 try_servo_repair=try_servo_repair)
Richard Barnettee519dcd2016-08-15 17:37:17 -0700313 if self._servo_host is not None:
314 self.servo = self._servo_host.get_servo()
315 else:
316 self.servo = None
317
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +0800318 # TODO(waihong): Do the simplication on Chameleon too.
Tom Wai-Hong Tam3d6790d2014-04-14 16:15:47 +0800319 self._chameleon_host = chameleon_host.create_chameleon_host(
320 dut=self.hostname, chameleon_args=chameleon_args)
Scottfe06ed82015-11-05 17:15:01 -0800321 # Add plankton host if plankton args were added on command line
322 self._plankton_host = plankton_host.create_plankton_host(plankton_args)
Tom Wai-Hong Tam3d6790d2014-04-14 16:15:47 +0800323
Tom Wai-Hong Tam3d6790d2014-04-14 16:15:47 +0800324 if self._chameleon_host:
Tom Wai-Hong Tameaee3402014-01-22 08:52:10 +0800325 self.chameleon = self._chameleon_host.create_chameleon_board()
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +0800326 else:
Tom Wai-Hong Tam3d6790d2014-04-14 16:15:47 +0800327 self.chameleon = None
Fang Deng5d518f42013-08-02 14:04:32 -0700328
Scottfe06ed82015-11-05 17:15:01 -0800329 if self._plankton_host:
330 self.plankton_servo = self._plankton_host.get_servo()
331 logging.info('plankton_servo: %r', self.plankton_servo)
332 # Create the plankton object used to access the ec uart
Scott07a848f2016-01-12 15:04:52 -0800333 self.plankton = plankton.Plankton(self.plankton_servo,
334 self._plankton_host.get_servod_server_proxy())
Scottfe06ed82015-11-05 17:15:01 -0800335 else:
Scott07a848f2016-01-12 15:04:52 -0800336 self.plankton = None
Scottfe06ed82015-11-05 17:15:01 -0800337
Fang Deng5d518f42013-08-02 14:04:32 -0700338
Richard Barnette260cbd02016-10-06 12:23:28 -0700339 def _get_cros_repair_image_name(self):
Prathmesh Prabhu075fc922017-02-13 11:50:25 -0800340 info = self.host_info_store.get()
341 if info.board is None:
342 raise error.AutoservError('Cannot obtain repair image name. '
343 'No board label value found')
344
345 return afe_utils.get_stable_cros_image_name(info.board)
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500346
347
Rohit Makasanadf0a3a32017-06-30 13:55:18 -0700348 def host_version_prefix(self, image):
349 """Return version label prefix.
350
351 In case the CrOS provisioning version is something other than the
352 standard CrOS version e.g. CrOS TH version, this function will
353 find the prefix from provision.py.
354
355 @param image: The image name to find its version prefix.
356 @returns: A prefix string for the image type.
357 """
358 return provision.get_version_label_prefix(image)
359
360
beepsdae65fd2013-07-26 16:24:41 -0700361 def verify_job_repo_url(self, tag=''):
beepscb6f1e22013-06-28 19:14:10 -0700362 """
363 Make sure job_repo_url of this host is valid.
364
joychen03eaad92013-06-26 09:55:21 -0700365 Eg: The job_repo_url "http://lmn.cd.ab.xyx:8080/static/\
beepscb6f1e22013-06-28 19:14:10 -0700366 lumpy-release/R29-4279.0.0/autotest/packages" claims to have the
367 autotest package for lumpy-release/R29-4279.0.0. If this isn't the case,
368 download and extract it. If the devserver embedded in the url is
369 unresponsive, update the job_repo_url of the host after staging it on
370 another devserver.
371
372 @param job_repo_url: A url pointing to the devserver where the autotest
373 package for this build should be staged.
beepsdae65fd2013-07-26 16:24:41 -0700374 @param tag: The tag from the server job, in the format
375 <job_id>-<user>/<hostname>, or <hostless> for a server job.
beepscb6f1e22013-06-28 19:14:10 -0700376
377 @raises DevServerException: If we could not resolve a devserver.
378 @raises AutoservError: If we're unable to save the new job_repo_url as
379 a result of choosing a new devserver because the old one failed to
380 respond to a health check.
beeps0c865032013-07-30 11:37:06 -0700381 @raises urllib2.URLError: If the devserver embedded in job_repo_url
382 doesn't respond within the timeout.
beepscb6f1e22013-06-28 19:14:10 -0700383 """
Prathmesh Prabhu368abdf2017-02-14 11:23:47 -0800384 info = self.host_info_store.get()
385 job_repo_url = info.attributes.get(ds_constants.JOB_REPO_URL, '')
beepscb6f1e22013-06-28 19:14:10 -0700386 if not job_repo_url:
387 logging.warning('No job repo url set on host %s', self.hostname)
388 return
389
390 logging.info('Verifying job repo url %s', job_repo_url)
391 devserver_url, image_name = tools.get_devserver_build_from_package_url(
392 job_repo_url)
393
beeps0c865032013-07-30 11:37:06 -0700394 ds = dev_server.ImageServer(devserver_url)
beepscb6f1e22013-06-28 19:14:10 -0700395
396 logging.info('Staging autotest artifacts for %s on devserver %s',
397 image_name, ds.url())
beeps687243d2013-07-18 15:29:27 -0700398
399 start_time = time.time()
Simran Basi25e7a922014-10-31 11:56:10 -0700400 ds.stage_artifacts(image_name, ['autotest_packages'])
beeps687243d2013-07-18 15:29:27 -0700401 stage_time = time.time() - start_time
402
403 # Record how much of the verification time comes from a devserver
404 # restage. If we're doing things right we should not see multiple
405 # devservers for a given board/build/branch path.
406 try:
J. Richard Barnette3cbd76b2013-11-27 12:11:25 -0800407 board, build_type, branch = server_utils.ParseBuildName(
beeps687243d2013-07-18 15:29:27 -0700408 image_name)[:3]
J. Richard Barnette3cbd76b2013-11-27 12:11:25 -0800409 except server_utils.ParseBuildNameException:
beeps687243d2013-07-18 15:29:27 -0700410 pass
411 else:
beeps0c865032013-07-30 11:37:06 -0700412 devserver = devserver_url[
Chris Sosa65425082013-10-16 13:26:22 -0700413 devserver_url.find('/') + 2:devserver_url.rfind(':')]
beeps687243d2013-07-18 15:29:27 -0700414 stats_key = {
415 'board': board,
416 'build_type': build_type,
417 'branch': branch,
beeps0c865032013-07-30 11:37:06 -0700418 'devserver': devserver.replace('.', '_'),
beeps687243d2013-07-18 15:29:27 -0700419 }
Dan Shi5e2efb72017-02-07 11:40:23 -0800420
421 monarch_fields = {
422 'board': board,
423 'build_type': build_type,
424 # TODO(akeshet): To be consistent with most other metrics,
425 # consider changing the following field to be named
426 # 'milestone'.
427 'branch': branch,
428 'dev_server': devserver,
429 }
430 metrics.Counter(
431 'chromeos/autotest/provision/verify_url'
432 ).increment(fields=monarch_fields)
433 metrics.SecondsDistribution(
434 'chromeos/autotest/provision/verify_url_duration'
435 ).add(stage_time, fields=monarch_fields)
436
437
Dan Shicf4d2032015-03-12 15:04:21 -0700438 def stage_server_side_package(self, image=None):
439 """Stage autotest server-side package on devserver.
440
441 @param image: Full path of an OS image to install or a build name.
442
443 @return: A url to the autotest server-side package.
Dan Shi14de7622016-08-22 11:09:06 -0700444
445 @raise: error.AutoservError if fail to locate the build to test with, or
446 fail to stage server-side package.
Dan Shicf4d2032015-03-12 15:04:21 -0700447 """
Dan Shid37736b2016-07-06 15:10:29 -0700448 # If enable_drone_in_restricted_subnet is False, do not set hostname
449 # in devserver.resolve call, so a devserver in non-restricted subnet
450 # is picked to stage autotest server package for drone to download.
451 hostname = self.hostname
452 if not server_utils.ENABLE_DRONE_IN_RESTRICTED_SUBNET:
453 hostname = None
Dan Shicf4d2032015-03-12 15:04:21 -0700454 if image:
455 image_name = tools.get_build_from_image(image)
456 if not image_name:
457 raise error.AutoservError(
458 'Failed to parse build name from %s' % image)
Dan Shid37736b2016-07-06 15:10:29 -0700459 ds = dev_server.ImageServer.resolve(image_name, hostname)
Dan Shicf4d2032015-03-12 15:04:21 -0700460 else:
Prathmesh Prabhu9235e4c2017-03-28 13:16:06 -0700461 info = self.host_info_store.get()
Prathmesh Prabhu368abdf2017-02-14 11:23:47 -0800462 job_repo_url = info.attributes.get(ds_constants.JOB_REPO_URL, '')
Dan Shicf4d2032015-03-12 15:04:21 -0700463 if job_repo_url:
464 devserver_url, image_name = (
465 tools.get_devserver_build_from_package_url(job_repo_url))
Dan Shid37736b2016-07-06 15:10:29 -0700466 # If enable_drone_in_restricted_subnet is True, use the
467 # existing devserver. Otherwise, resolve a new one in
468 # non-restricted subnet.
469 if server_utils.ENABLE_DRONE_IN_RESTRICTED_SUBNET:
470 ds = dev_server.ImageServer(devserver_url)
471 else:
472 ds = dev_server.ImageServer.resolve(image_name)
Prathmesh Prabhub6cea612017-02-09 15:41:19 -0800473 elif info.build is not None:
474 ds = dev_server.ImageServer.resolve(info.build, hostname)
Prathmesh Prabhu0c1dd4d2017-06-07 13:01:53 -0700475 image_name = info.build
Dan Shicf4d2032015-03-12 15:04:21 -0700476 else:
Prathmesh Prabhub6cea612017-02-09 15:41:19 -0800477 raise error.AutoservError(
478 'Failed to stage server-side package. The host has '
479 'no job_report_url attribute or version label.')
Dan Shica503482015-03-30 17:23:25 -0700480
481 # Get the OS version of the build, for any build older than
482 # MIN_VERSION_SUPPORT_SSP, server side packaging is not supported.
483 match = re.match('.*/R\d+-(\d+)\.', image_name)
484 if match and int(match.group(1)) < self.MIN_VERSION_SUPPORT_SSP:
Dan Shi14de7622016-08-22 11:09:06 -0700485 raise error.AutoservError(
486 'Build %s is older than %s. Server side packaging is '
487 'disabled.' % (image_name, self.MIN_VERSION_SUPPORT_SSP))
Dan Shica503482015-03-30 17:23:25 -0700488
Dan Shicf4d2032015-03-12 15:04:21 -0700489 ds.stage_artifacts(image_name, ['autotest_server_package'])
490 return '%s/static/%s/%s' % (ds.url(), image_name,
491 'autotest_server_package.tar.bz2')
492
493
Dan Shi0f466e82013-02-22 15:44:58 -0800494 def _try_stateful_update(self, update_url, force_update, updater):
495 """Try to use stateful update to initialize DUT.
496
497 When DUT is already running the same version that machine_install
498 tries to install, stateful update is a much faster way to clean up
499 the DUT for testing, compared to a full reimage. It is implemeted
500 by calling autoupdater.run_update, but skipping updating root, as
501 updating the kernel is time consuming and not necessary.
502
503 @param update_url: url of the image.
504 @param force_update: Set to True to update the image even if the DUT
505 is running the same version.
506 @param updater: ChromiumOSUpdater instance used to update the DUT.
507 @returns: True if the DUT was updated with stateful update.
508
509 """
Laurence Goodby778c9a42017-05-24 19:24:07 -0700510 self.prepare_for_update()
Dan Shi10b98482016-02-02 14:38:50 -0800511
J. Richard Barnette3f731032014-04-07 17:42:59 -0700512 # TODO(jrbarnette): Yes, I hate this re.match() test case.
513 # It's better than the alternative: see crbug.com/360944.
514 image_name = autoupdater.url_to_image_name(update_url)
515 release_pattern = r'^.*-release/R[0-9]+-[0-9]+\.[0-9]+\.0$'
516 if not re.match(release_pattern, image_name):
517 return False
Dan Shi0f466e82013-02-22 15:44:58 -0800518 if not updater.check_version():
519 return False
520 if not force_update:
521 logging.info('Canceling stateful update because the new and '
522 'old versions are the same.')
523 return False
524 # Following folders should be rebuilt after stateful update.
525 # A test file is used to confirm each folder gets rebuilt after
526 # the stateful update.
527 folders_to_check = ['/var', '/home', '/mnt/stateful_partition']
528 test_file = '.test_file_to_be_deleted'
Paul Hobbs38bcb702017-07-25 00:50:05 -0700529 paths = [os.path.join(folder, test_file) for folder in folders_to_check]
530 self.run('touch %s' % ' '.join(paths))
Dan Shi0f466e82013-02-22 15:44:58 -0800531
Chris Sosae92399e2015-04-24 11:32:59 -0700532 updater.run_update(update_root=False)
Dan Shi0f466e82013-02-22 15:44:58 -0800533
534 # Reboot to complete stateful update.
Chris Sosab76e0ee2013-05-22 16:55:41 -0700535 self.reboot(timeout=self.REBOOT_TIMEOUT, wait=True)
xixuan908b0752016-12-08 18:54:23 -0800536
537 # After stateful update and a reboot, all of the test_files shouldn't
538 # exist any more. Otherwise the stateful update is failed.
539 return not any(
540 self.path_exists(os.path.join(folder, test_file))
541 for folder in folders_to_check)
Dan Shi0f466e82013-02-22 15:44:58 -0800542
543
J. Richard Barnette7275b612013-06-04 18:13:11 -0700544 def _post_update_processing(self, updater, expected_kernel=None):
Dan Shi0f466e82013-02-22 15:44:58 -0800545 """After the DUT is updated, confirm machine_install succeeded.
546
547 @param updater: ChromiumOSUpdater instance used to update the DUT.
J. Richard Barnette7275b612013-06-04 18:13:11 -0700548 @param expected_kernel: kernel expected to be active after reboot,
549 or `None` to skip rollback checking.
Dan Shi0f466e82013-02-22 15:44:58 -0800550
551 """
J. Richard Barnette7275b612013-06-04 18:13:11 -0700552 # Touch the lab machine file to leave a marker that
553 # distinguishes this image from other test images.
554 # Afterwards, we must re-run the autoreboot script because
555 # it depends on the _LAB_MACHINE_FILE.
J. Richard Barnette71cc1862015-12-02 10:32:38 -0800556 autoreboot_cmd = ('FILE="%s" ; [ -f "$FILE" ] || '
557 '( touch "$FILE" ; start autoreboot )')
558 self.run(autoreboot_cmd % self._LAB_MACHINE_FILE)
Chris Sosa65425082013-10-16 13:26:22 -0700559 updater.verify_boot_expectations(
560 expected_kernel, rollback_message=
Gilad Arnoldc26ae1f2015-10-22 16:09:41 -0700561 'Build %s failed to boot on %s; system rolled back to previous '
Chris Sosa65425082013-10-16 13:26:22 -0700562 'build' % (updater.update_version, self.hostname))
J. Richard Barnette7275b612013-06-04 18:13:11 -0700563 # Check that we've got the build we meant to install.
564 if not updater.check_version_to_confirm_install():
565 raise autoupdater.ChromiumOSError(
566 'Failed to update %s to build %s; found build '
567 '%s instead' % (self.hostname,
Chris Sosa65425082013-10-16 13:26:22 -0700568 updater.update_version,
Dan Shi0942b1d2015-03-31 11:07:00 -0700569 self.get_release_version()))
Dan Shi0f466e82013-02-22 15:44:58 -0800570
Chris Sosae92399e2015-04-24 11:32:59 -0700571 logging.debug('Cleaning up old autotest directories.')
572 try:
573 installed_autodir = autotest.Autotest.get_installed_autodir(self)
574 self.run('rm -rf ' + installed_autodir)
575 except autotest.AutodirNotFoundError:
576 logging.debug('No autotest installed directory found.')
577
Dan Shi0f466e82013-02-22 15:44:58 -0800578
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700579 def _stage_image_for_update(self, image_name=None):
Chris Sosae92399e2015-04-24 11:32:59 -0700580 """Stage a build on a devserver and return the update_url and devserver.
Scott Zawalskieadbf702013-03-14 09:23:06 -0400581
582 @param image_name: a name like lumpy-release/R27-3837.0.0
Chris Sosae92399e2015-04-24 11:32:59 -0700583 @returns a tuple with an update URL like:
Scott Zawalskieadbf702013-03-14 09:23:06 -0400584 http://172.22.50.205:8082/update/lumpy-release/R27-3837.0.0
Chris Sosae92399e2015-04-24 11:32:59 -0700585 and the devserver instance.
Scott Zawalskieadbf702013-03-14 09:23:06 -0400586 """
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700587 if not image_name:
Richard Barnette260cbd02016-10-06 12:23:28 -0700588 image_name = self._get_cros_repair_image_name()
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700589 logging.info('Staging build for AU: %s', image_name)
Dan Shi216389c2015-12-22 11:03:06 -0800590 devserver = dev_server.ImageServer.resolve(image_name, self.hostname)
Scott Zawalskieadbf702013-03-14 09:23:06 -0400591 devserver.trigger_download(image_name, synchronous=False)
Chris Sosae92399e2015-04-24 11:32:59 -0700592 return (tools.image_url_pattern() % (devserver.url(), image_name),
593 devserver)
Scott Zawalskieadbf702013-03-14 09:23:06 -0400594
595
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700596 def stage_image_for_servo(self, image_name=None):
597 """Stage a build on a devserver and return the update_url.
598
599 @param image_name: a name like lumpy-release/R27-3837.0.0
600 @returns an update URL like:
601 http://172.22.50.205:8082/update/lumpy-release/R27-3837.0.0
602 """
603 if not image_name:
Richard Barnette260cbd02016-10-06 12:23:28 -0700604 image_name = self._get_cros_repair_image_name()
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700605 logging.info('Staging build for servo install: %s', image_name)
Dan Shi216389c2015-12-22 11:03:06 -0800606 devserver = dev_server.ImageServer.resolve(image_name, self.hostname)
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700607 devserver.stage_artifacts(image_name, ['test_image'])
608 return devserver.get_test_image_url(image_name)
609
610
beepse539be02013-07-31 21:57:39 -0700611 def stage_factory_image_for_servo(self, image_name):
612 """Stage a build on a devserver and return the update_url.
613
614 @param image_name: a name like <baord>/4262.204.0
beeps12c0a3c2013-09-03 11:58:27 -0700615
beepse539be02013-07-31 21:57:39 -0700616 @return: An update URL, eg:
617 http://<devserver>/static/canary-channel/\
618 <board>/4262.204.0/factory_test/chromiumos_factory_image.bin
beeps12c0a3c2013-09-03 11:58:27 -0700619
620 @raises: ValueError if the factory artifact name is missing from
621 the config.
622
beepse539be02013-07-31 21:57:39 -0700623 """
624 if not image_name:
625 logging.error('Need an image_name to stage a factory image.')
626 return
627
Dan Shib8540a52015-07-16 14:18:23 -0700628 factory_artifact = CONFIG.get_config_value(
beeps12c0a3c2013-09-03 11:58:27 -0700629 'CROS', 'factory_artifact', type=str, default='')
630 if not factory_artifact:
631 raise ValueError('Cannot retrieve the factory artifact name from '
632 'autotest config, and hence cannot stage factory '
633 'artifacts.')
634
beepse539be02013-07-31 21:57:39 -0700635 logging.info('Staging build for servo install: %s', image_name)
Dan Shi216389c2015-12-22 11:03:06 -0800636 devserver = dev_server.ImageServer.resolve(image_name, self.hostname)
beepse539be02013-07-31 21:57:39 -0700637 devserver.stage_artifacts(
638 image_name,
beeps12c0a3c2013-09-03 11:58:27 -0700639 [factory_artifact],
640 archive_url=None)
beepse539be02013-07-31 21:57:39 -0700641
642 return tools.factory_image_url_pattern() % (devserver.url(), image_name)
643
644
xixuan016d95b2017-02-27 14:37:47 -0800645 def _get_au_monarch_fields(self, devserver, build):
646 """Form monarch fields by given devserve & build for auto-update.
647
648 @param devserver: the devserver (ImageServer instance) for auto-update.
649 @param build: the build to be updated.
650
651 @return A dictionary of monach fields.
652 """
653 try:
654 board, build_type, milestone, _ = server_utils.ParseBuildName(build)
655 except server_utils.ParseBuildNameException:
656 logging.warning('Unable to parse build name %s. Something is '
657 'likely broken, but will continue anyway.',
658 build)
659 board, build_type, milestone = ('', '', '')
660
661 monarch_fields = {
662 'dev_server': devserver.hostname,
663 'board': board,
664 'build_type': build_type,
665 'milestone': milestone
666 }
667 return monarch_fields
668
669
xixuanc30d7e92017-05-11 12:06:39 -0700670 def _retry_auto_update_with_new_devserver(self, build, last_devserver,
xixuana3bbc422017-05-04 15:57:21 -0700671 force_update, force_full_update,
672 force_original):
xixuanc30d7e92017-05-11 12:06:39 -0700673 """Kick off auto-update by devserver and send metrics.
674
675 @param build: the build to update.
676 @param last_devserver: the last devserver that failed to provision.
677 @param force_update: see |machine_install_by_devserver|'s force_udpate
678 for details.
679 @param force_full_update: see |machine_install_by_devserver|'s
680 force_full_update for details.
xixuana3bbc422017-05-04 15:57:21 -0700681 @param force_original: Whether to force stateful update with the
682 original payload.
xixuanc30d7e92017-05-11 12:06:39 -0700683
684 @return the result of |auto_update| in dev_server.
685 """
686 devserver = dev_server.resolve(
687 build, self.hostname, ban_list=[last_devserver.url()])
688 devserver.trigger_download(build, synchronous=False)
689 monarch_fields = self._get_au_monarch_fields(devserver, build)
690 logging.debug('Retry auto_update: resolved devserver for '
691 'auto-update: %s', devserver.url())
692
693 # Add metrics
694 install_with_dev_counter = metrics.Counter(
695 'chromeos/autotest/provision/install_with_devserver')
696 install_with_dev_counter.increment(fields=monarch_fields)
697 c = metrics.Counter(
698 'chromeos/autotest/provision/retry_by_devserver')
699 monarch_fields['last_devserver'] = last_devserver.hostname
700 monarch_fields['host'] = self.hostname
701 c.increment(fields=monarch_fields)
702
xixuan9f455342017-06-08 18:21:44 -0700703 # Won't retry auto_update in a retry of auto-update.
704 # In other words, we only retry auto-update once with a different
705 # devservers.
706 devserver.auto_update(
xixuanc30d7e92017-05-11 12:06:39 -0700707 self.hostname, build,
708 original_board=self.get_board().replace(
709 ds_constants.BOARD_PREFIX, ''),
710 original_release_version=self.get_release_version(),
711 log_dir=self.job.resultdir,
712 force_update=force_update,
xixuana3bbc422017-05-04 15:57:21 -0700713 full_update=force_full_update,
714 force_original=force_original)
xixuanc30d7e92017-05-11 12:06:39 -0700715
716
xixuanec801e32016-08-25 10:20:22 -0700717 def machine_install_by_devserver(self, update_url=None, force_update=False,
xixuan5dc64ea2016-05-20 17:27:51 -0700718 local_devserver=False, repair=False,
719 force_full_update=False):
720 """Ultiize devserver to install the DUT.
721
722 (TODO) crbugs.com/627269: The logic in this function has some overlap
723 with those in function machine_install. The merge will be done later,
724 not in the same CL.
725
xixuanec801e32016-08-25 10:20:22 -0700726 @param update_url: The update_url or build for the host to update.
xixuan5dc64ea2016-05-20 17:27:51 -0700727 @param force_update: Force an update even if the version installed
728 is the same. Default:False
729 @param local_devserver: Used by test_that to allow people to
730 use their local devserver. Default: False
731 @param repair: Forces update to repair image. Implies force_update.
732 @param force_full_update: If True, do not attempt to run stateful
733 update, force a full reimage. If False, try stateful update
734 first when the dut is already installed with the same version.
735 @raises autoupdater.ChromiumOSError
736
737 @returns A tuple of (image_name, host_attributes).
738 image_name is the name of image installed, e.g.,
739 veyron_jerry-release/R50-7871.0.0
740 host_attributes is a dictionary of (attribute, value), which
741 can be saved to afe_host_attributes table in database. This
742 method returns a dictionary with a single entry of
743 `job_repo_url`: repo_url, where repo_url is a devserver url to
744 autotest packages.
745 """
xixuan5dc64ea2016-05-20 17:27:51 -0700746 if repair:
Richard Barnette260cbd02016-10-06 12:23:28 -0700747 update_url = self._get_cros_repair_image_name()
xixuan5dc64ea2016-05-20 17:27:51 -0700748 force_update = True
749
xixuanec801e32016-08-25 10:20:22 -0700750 if not update_url and not self._parser.options.image:
xixuan5dc64ea2016-05-20 17:27:51 -0700751 raise error.AutoservError(
752 'There is no update URL, nor a method to get one.')
753
xixuanec801e32016-08-25 10:20:22 -0700754 if not update_url and self._parser.options.image:
755 update_url = self._parser.options.image
xixuan5dc64ea2016-05-20 17:27:51 -0700756
xixuan5dc64ea2016-05-20 17:27:51 -0700757
758 # Get build from parameter or AFE.
759 # If the build is not a URL, let devserver to stage it first.
760 # Otherwise, choose a devserver to trigger auto-update.
xixuanec801e32016-08-25 10:20:22 -0700761 build = None
Aviv Keshetf9375482016-10-26 02:09:03 -0700762 devserver = None
763 logging.debug('Resolving a devserver for auto-update')
Aviv Keshet3a76f762016-10-27 17:41:51 -0700764 previously_resolved = False
xixuanec801e32016-08-25 10:20:22 -0700765 if update_url.startswith('http://'):
766 build = autoupdater.url_to_image_name(update_url)
Aviv Keshet3a76f762016-10-27 17:41:51 -0700767 previously_resolved = True
768 else:
769 build = update_url
770 devserver = dev_server.resolve(build, self.hostname)
Allen Lia5cfb972016-12-27 17:17:22 -0800771 server_name = devserver.hostname
Aviv Keshet3a76f762016-10-27 17:41:51 -0700772
xixuan016d95b2017-02-27 14:37:47 -0800773 monarch_fields = self._get_au_monarch_fields(devserver, build)
Aviv Keshet3a76f762016-10-27 17:41:51 -0700774
775 if previously_resolved:
xixuan2e62d162016-09-29 19:19:19 -0700776 # Make sure devserver for Auto-Update has staged the build.
xixuan2e62d162016-09-29 19:19:19 -0700777 if server_name not in update_url:
Aviv Keshet3a76f762016-10-27 17:41:51 -0700778 logging.debug('Resolved to devserver that does not match '
779 'update_url. The previously resolved devserver '
780 'must be unhealthy. Switching to use devserver %s,'
781 ' and re-staging.',
xixuan2e62d162016-09-29 19:19:19 -0700782 server_name)
Aviv Keshetf9375482016-10-26 02:09:03 -0700783 logging.info('Staging build for AU: %s', update_url)
xixuan2e62d162016-09-29 19:19:19 -0700784 devserver.trigger_download(build, synchronous=False)
Dan Shi5e2efb72017-02-07 11:40:23 -0800785 c = metrics.Counter(
786 'chromeos/autotest/provision/failover_download')
787 c.increment(fields=monarch_fields)
xixuan5dc64ea2016-05-20 17:27:51 -0700788 else:
Aviv Keshetf9375482016-10-26 02:09:03 -0700789 logging.info('Staging build for AU: %s', update_url)
xixuan5dc64ea2016-05-20 17:27:51 -0700790 devserver.trigger_download(build, synchronous=False)
Dan Shi5e2efb72017-02-07 11:40:23 -0800791 c = metrics.Counter('chromeos/autotest/provision/trigger_download')
792 c.increment(fields=monarch_fields)
xixuan5dc64ea2016-05-20 17:27:51 -0700793
794 # Report provision stats.
xixuan016d95b2017-02-27 14:37:47 -0800795 install_with_dev_counter = metrics.Counter(
796 'chromeos/autotest/provision/install_with_devserver')
797 install_with_dev_counter.increment(fields=monarch_fields)
xixuanec801e32016-08-25 10:20:22 -0700798 logging.debug('Resolved devserver for auto-update: %s', devserver.url())
xixuan5dc64ea2016-05-20 17:27:51 -0700799
Aviv Keshetf9375482016-10-26 02:09:03 -0700800 # and other metrics from this function.
Dan Shi5e2efb72017-02-07 11:40:23 -0800801 metrics.Counter('chromeos/autotest/provision/resolve'
802 ).increment(fields=monarch_fields)
Aviv Keshetf9375482016-10-26 02:09:03 -0700803
xixuana3bbc422017-05-04 15:57:21 -0700804 force_original = self.get_chromeos_release_milestone() is None
805
xixuan9f455342017-06-08 18:21:44 -0700806 try:
807 devserver.auto_update(
808 self.hostname, build,
809 original_board=self.get_board().replace(
810 ds_constants.BOARD_PREFIX, ''),
811 original_release_version=self.get_release_version(),
812 log_dir=self.job.resultdir,
813 force_update=force_update,
814 full_update=force_full_update,
815 force_original=force_original)
816 except dev_server.RetryableProvisionException:
817 # It indicates that last provision failed due to devserver load
818 # issue, so another devserver is resolved to kick off provision
819 # job once again and only once.
820 logging.debug('Provision failed due to devserver issue,'
821 'retry it with another devserver.')
xixuan016d95b2017-02-27 14:37:47 -0800822
xixuan9f455342017-06-08 18:21:44 -0700823 # Check first whether this DUT is completely offline. If so, skip
824 # the following provision tries.
825 logging.debug('Checking whether host %s is online.', self.hostname)
826 if utils.ping(self.hostname, tries=1, deadline=1) == 0:
827 self._retry_auto_update_with_new_devserver(
828 build, devserver, force_update, force_full_update,
829 force_original)
830 else:
831 raise error.AutoservError(
832 'No answer to ping from %s' % self.hostname)
xixuan5dc64ea2016-05-20 17:27:51 -0700833
834 # The reason to resolve a new devserver in function machine_install
835 # is mostly because that the update_url there may has a strange format,
836 # and it's hard to parse the devserver url from it.
837 # Since we already resolve a devserver to trigger auto-update, the same
838 # devserver is used to form JOB_REPO_URL here. Verified in local test.
839 repo_url = tools.get_package_url(devserver.url(), build)
840 return build, {ds_constants.JOB_REPO_URL: repo_url}
841
842
Laurence Goodby778c9a42017-05-24 19:24:07 -0700843 def prepare_for_update(self):
844 """Prepares the DUT for an update.
845
846 Subclasses may override this to perform any special actions
847 required before updating.
848 """
Laurence Goodby468de252017-06-08 17:22:53 -0700849 pass
Laurence Goodby778c9a42017-05-24 19:24:07 -0700850
851
Chris Sosaa3ac2152012-05-23 22:23:13 -0700852 def machine_install(self, update_url=None, force_update=False,
Richard Barnette0b023a72015-04-24 16:07:30 +0000853 local_devserver=False, repair=False,
854 force_full_update=False):
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500855 """Install the DUT.
856
Dan Shi0f466e82013-02-22 15:44:58 -0800857 Use stateful update if the DUT is already running the same build.
858 Stateful update does not update kernel and tends to run much faster
859 than a full reimage. If the DUT is running a different build, or it
860 failed to do a stateful update, full update, including kernel update,
861 will be applied to the DUT.
862
Simran Basi5ace6f22016-01-06 17:30:44 -0800863 Once a host enters machine_install its host attribute job_repo_url
864 (used for package install) will be removed and then updated.
Scott Zawalskieadbf702013-03-14 09:23:06 -0400865
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500866 @param update_url: The url to use for the update
867 pattern: http://$devserver:###/update/$build
868 If update_url is None and repair is True we will install the
Dan Shi6964fa52014-12-18 11:04:27 -0800869 stable image listed in afe_stable_versions table. If the table
870 is not setup, global_config value under CROS.stable_cros_version
871 will be used instead.
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500872 @param force_update: Force an update even if the version installed
873 is the same. Default:False
Christopher Wiley6a4ff932015-05-15 14:00:47 -0700874 @param local_devserver: Used by test_that to allow people to
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500875 use their local devserver. Default: False
Chris Sosae92399e2015-04-24 11:32:59 -0700876 @param repair: Forces update to repair image. Implies force_update.
Fang Deng3d3b9272014-12-22 12:20:28 -0800877 @param force_full_update: If True, do not attempt to run stateful
878 update, force a full reimage. If False, try stateful update
879 first when the dut is already installed with the same version.
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500880 @raises autoupdater.ChromiumOSError
881
Dan Shibe3636a2016-02-14 22:48:01 -0800882 @returns A tuple of (image_name, host_attributes).
883 image_name is the name of image installed, e.g.,
884 veyron_jerry-release/R50-7871.0.0
885 host_attributes is a dictionary of (attribute, value), which
886 can be saved to afe_host_attributes table in database. This
887 method returns a dictionary with a single entry of
888 `job_repo_url`: repo_url, where repo_url is a devserver url to
889 autotest packages.
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500890 """
Chris Sosae92399e2015-04-24 11:32:59 -0700891 devserver = None
Richard Barnette0b023a72015-04-24 16:07:30 +0000892 if repair:
Chris Sosae92399e2015-04-24 11:32:59 -0700893 update_url, devserver = self._stage_image_for_update()
Richard Barnette0b023a72015-04-24 16:07:30 +0000894 force_update = True
Dan Shi0f466e82013-02-22 15:44:58 -0800895
Chris Sosae92399e2015-04-24 11:32:59 -0700896 if not update_url and not self._parser.options.image:
897 raise error.AutoservError(
Dan Shid07ee2e2015-09-24 14:49:25 -0700898 'There is no update URL, nor a method to get one.')
Chris Sosae92399e2015-04-24 11:32:59 -0700899
900 if not update_url and self._parser.options.image:
901 # This is the base case where we have no given update URL i.e.
902 # dynamic suites logic etc. This is the most flexible case where we
903 # can serve an update from any of our fleet of devservers.
904 requested_build = self._parser.options.image
905 if not requested_build.startswith('http://'):
906 logging.debug('Update will be staged for this installation')
907 update_url, devserver = self._stage_image_for_update(
Dan Shid07ee2e2015-09-24 14:49:25 -0700908 requested_build)
Chris Sosae92399e2015-04-24 11:32:59 -0700909 else:
910 update_url = requested_build
911
912 logging.debug('Update URL is %s', update_url)
913
Dan Shif48f8132016-02-18 10:34:30 -0800914 # Report provision stats.
Allen Lia5cfb972016-12-27 17:17:22 -0800915 server_name = dev_server.get_hostname(update_url)
Allen Li48a13fe2016-11-22 14:10:40 -0800916 (metrics.Counter('chromeos/autotest/provision/install')
917 .increment(fields={'devserver': server_name}))
Dan Shif48f8132016-02-18 10:34:30 -0800918
Dan Shid07ee2e2015-09-24 14:49:25 -0700919 # Create a file to indicate if provision fails. The file will be removed
920 # by stateful update or full install.
J. Richard Barnette91137f02016-03-10 16:52:26 -0800921 self.run('touch %s' % self.PROVISION_FAILED)
Dan Shid07ee2e2015-09-24 14:49:25 -0700922
Chris Sosae92399e2015-04-24 11:32:59 -0700923 update_complete = False
924 updater = autoupdater.ChromiumOSUpdater(
925 update_url, host=self, local_devserver=local_devserver)
Fang Deng3d3b9272014-12-22 12:20:28 -0800926 if not force_full_update:
927 try:
Chris Sosae92399e2015-04-24 11:32:59 -0700928 # If the DUT is already running the same build, try stateful
929 # update first as it's much quicker than a full re-image.
930 update_complete = self._try_stateful_update(
Dan Shid07ee2e2015-09-24 14:49:25 -0700931 update_url, force_update, updater)
Fang Deng3d3b9272014-12-22 12:20:28 -0800932 except Exception as e:
933 logging.exception(e)
J. Richard Barnette45e93de2012-04-11 17:24:15 -0700934
Dan Shi0f466e82013-02-22 15:44:58 -0800935 inactive_kernel = None
Chris Sosae92399e2015-04-24 11:32:59 -0700936 if update_complete or (not force_update and updater.check_version()):
937 logging.info('Install complete without full update')
938 else:
939 logging.info('DUT requires full update.')
940 self.reboot(timeout=self.REBOOT_TIMEOUT, wait=True)
Laurence Goodby778c9a42017-05-24 19:24:07 -0700941 self.prepare_for_update()
Dan Shi10b98482016-02-02 14:38:50 -0800942
Chris Sosae92399e2015-04-24 11:32:59 -0700943 num_of_attempts = provision.FLAKY_DEVSERVER_ATTEMPTS
Chris Sosab7612bc2013-03-21 10:32:37 -0700944
Chris Sosae92399e2015-04-24 11:32:59 -0700945 while num_of_attempts > 0:
946 num_of_attempts -= 1
947 try:
948 updater.run_update()
949 except Exception:
950 logging.warn('Autoupdate did not complete.')
951 # Do additional check for the devserver health. Ideally,
952 # the autoupdater.py could raise an exception when it
953 # detected network flake but that would require
954 # instrumenting the update engine and parsing it log.
955 if (num_of_attempts <= 0 or
956 devserver is None or
xixuan9e2c98d2016-02-26 19:04:53 -0800957 dev_server.ImageServer.devserver_healthy(
Chris Sosae92399e2015-04-24 11:32:59 -0700958 devserver.url())):
Dan Shid07ee2e2015-09-24 14:49:25 -0700959 raise
J. Richard Barnette45e93de2012-04-11 17:24:15 -0700960
Chris Sosae92399e2015-04-24 11:32:59 -0700961 logging.warn('Devserver looks unhealthy. Trying another')
962 update_url, devserver = self._stage_image_for_update(
963 requested_build)
964 logging.debug('New Update URL is %s', update_url)
965 updater = autoupdater.ChromiumOSUpdater(
966 update_url, host=self,
967 local_devserver=local_devserver)
968 else:
969 break
J. Richard Barnette45e93de2012-04-11 17:24:15 -0700970
Chris Sosae92399e2015-04-24 11:32:59 -0700971 # Give it some time in case of IO issues.
972 time.sleep(10)
Dan Shi5699ac22014-12-19 10:55:49 -0800973
Chris Sosae92399e2015-04-24 11:32:59 -0700974 # Figure out active and inactive kernel.
975 active_kernel, inactive_kernel = updater.get_kernel_state()
Simran Basi13fa1ba2013-03-04 10:56:47 -0800976
Chris Sosae92399e2015-04-24 11:32:59 -0700977 # Ensure inactive kernel has higher priority than active.
978 if (updater.get_kernel_priority(inactive_kernel)
979 < updater.get_kernel_priority(active_kernel)):
980 raise autoupdater.ChromiumOSError(
981 'Update failed. The priority of the inactive kernel'
982 ' partition is less than that of the active kernel'
983 ' partition.')
984
985 # Updater has returned successfully; reboot the host.
Richard Barnettefc3d7432016-07-07 10:21:31 -0700986 #
987 # Regarding the 'crossystem' command: In some cases, the
988 # TPM gets into a state such that it fails verification.
989 # We don't know why. However, this call papers over the
990 # problem by clearing the TPM during the reboot.
991 #
992 # We ignore failures from 'crossystem'. Although failure
993 # here is unexpected, and could signal a bug, the point
994 # of the exercise is to paper over problems; allowing
995 # this to fail would defeat the purpose.
996 self.run('crossystem clear_tpm_owner_request=1',
997 ignore_status=True)
Chris Sosae92399e2015-04-24 11:32:59 -0700998 self.reboot(timeout=self.REBOOT_TIMEOUT, wait=True)
999
1000 self._post_update_processing(updater, inactive_kernel)
Simran Basi5ace6f22016-01-06 17:30:44 -08001001 image_name = autoupdater.url_to_image_name(update_url)
Dan Shibe3636a2016-02-14 22:48:01 -08001002 # update_url is different from devserver url needed to stage autotest
1003 # packages, therefore, resolve a new devserver url here.
1004 devserver_url = dev_server.ImageServer.resolve(image_name,
1005 self.hostname).url()
1006 repo_url = tools.get_package_url(devserver_url, image_name)
1007 return image_name, {ds_constants.JOB_REPO_URL: repo_url}
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001008
1009
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +08001010 def _clear_fw_version_labels(self, rw_only):
1011 """Clear firmware version labels from the machine.
1012
1013 @param rw_only: True to only clear fwrw_version; otherewise, clear
1014 both fwro_version and fwrw_version.
1015 """
Dan Shi9cb0eec2014-06-03 09:04:50 -07001016 labels = self._AFE.get_labels(
Dan Shi0723bf52015-06-24 10:52:38 -07001017 name__startswith=provision.FW_RW_VERSION_PREFIX,
Dan Shi9cb0eec2014-06-03 09:04:50 -07001018 host__hostname=self.hostname)
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +08001019 if not rw_only:
1020 labels = labels + self._AFE.get_labels(
1021 name__startswith=provision.FW_RO_VERSION_PREFIX,
1022 host__hostname=self.hostname)
Dan Shi9cb0eec2014-06-03 09:04:50 -07001023 for label in labels:
1024 label.remove_hosts(hosts=[self.hostname])
1025
1026
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +08001027 def _add_fw_version_label(self, build, rw_only):
Dan Shi9cb0eec2014-06-03 09:04:50 -07001028 """Add firmware version label to the machine.
1029
1030 @param build: Build of firmware.
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +08001031 @param rw_only: True to only add fwrw_version; otherwise, add both
1032 fwro_version and fwrw_version.
Dan Shi9cb0eec2014-06-03 09:04:50 -07001033
1034 """
Prathmesh Prabhu2c7471d2016-11-15 20:19:57 +00001035 fw_label = provision.fwrw_version_to_label(build)
MK Ryu73be9862015-07-06 12:25:00 -07001036 self._AFE.run('label_add_hosts', id=fw_label, hosts=[self.hostname])
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +08001037 if not rw_only:
Prathmesh Prabhu2c7471d2016-11-15 20:19:57 +00001038 fw_label = provision.fwro_version_to_label(build)
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +08001039 self._AFE.run('label_add_hosts', id=fw_label, hosts=[self.hostname])
Dan Shi9cb0eec2014-06-03 09:04:50 -07001040
1041
Tom Wai-Hong Tam4ac78982016-01-08 02:34:37 +08001042 def firmware_install(self, build=None, rw_only=False):
Dan Shi9cb0eec2014-06-03 09:04:50 -07001043 """Install firmware to the DUT.
1044
1045 Use stateful update if the DUT is already running the same build.
1046 Stateful update does not update kernel and tends to run much faster
1047 than a full reimage. If the DUT is running a different build, or it
1048 failed to do a stateful update, full update, including kernel update,
1049 will be applied to the DUT.
1050
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +08001051 Once a host enters firmware_install its fw[ro|rw]_version label will
1052 be removed. After the firmware is updated successfully, a new
1053 fw[ro|rw]_version label will be added to the host.
Dan Shi9cb0eec2014-06-03 09:04:50 -07001054
1055 @param build: The build version to which we want to provision the
1056 firmware of the machine,
1057 e.g. 'link-firmware/R22-2695.1.144'.
Tom Wai-Hong Tam4ac78982016-01-08 02:34:37 +08001058 @param rw_only: True to only install firmware to its RW portions. Keep
1059 the RO portions unchanged.
Dan Shi9cb0eec2014-06-03 09:04:50 -07001060
1061 TODO(dshi): After bug 381718 is fixed, update here with corresponding
1062 exceptions that could be raised.
1063
1064 """
1065 if not self.servo:
1066 raise error.TestError('Host %s does not have servo.' %
1067 self.hostname)
1068
Wai-Hong Tam64cf7912017-05-23 16:41:40 -07001069 # Get the DUT board name from servod.
1070 board = self.servo.get_board()
Dan Shi9cb0eec2014-06-03 09:04:50 -07001071
Chris Sosae92399e2015-04-24 11:32:59 -07001072 # If build is not set, try to install firmware from stable CrOS.
Dan Shi9cb0eec2014-06-03 09:04:50 -07001073 if not build:
Richard Barnette260cbd02016-10-06 12:23:28 -07001074 build = afe_utils.get_stable_faft_version(board)
Dan Shi3d7a0e12015-10-12 11:55:45 -07001075 if not build:
1076 raise error.TestError(
1077 'Failed to find stable firmware build for %s.',
1078 self.hostname)
1079 logging.info('Will install firmware from build %s.', build)
Dan Shi9cb0eec2014-06-03 09:04:50 -07001080
1081 config = FAFTConfig(board)
1082 if config.use_u_boot:
1083 ap_image = 'image-%s.bin' % board
1084 else: # Depthcharge platform
1085 ap_image = 'image.bin'
1086 ec_image = 'ec.bin'
Dan Shi216389c2015-12-22 11:03:06 -08001087 ds = dev_server.ImageServer.resolve(build, self.hostname)
Dan Shi9cb0eec2014-06-03 09:04:50 -07001088 ds.stage_artifacts(build, ['firmware'])
1089
1090 tmpd = autotemp.tempdir(unique_id='fwimage')
1091 try:
1092 fwurl = self._FW_IMAGE_URL_PATTERN % (ds.url(), build)
1093 local_tarball = os.path.join(tmpd.name, os.path.basename(fwurl))
xixuan4e116822016-11-17 15:32:10 -08001094 ds.download_file(fwurl, local_tarball)
Dan Shi9cb0eec2014-06-03 09:04:50 -07001095
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +08001096 self._clear_fw_version_labels(rw_only)
Tom Wai-Hong Tam552fee12016-05-11 06:27:58 +08001097 if config.chrome_ec:
Tom Wai-Hong Tamf9db57a2016-03-17 05:32:22 +08001098 logging.info('Will re-program EC %snow', 'RW ' if rw_only else '')
1099 server_utils.system('tar xf %s -C %s %s' %
1100 (local_tarball, tmpd.name, ec_image),
1101 timeout=60)
1102 self.servo.program_ec(os.path.join(tmpd.name, ec_image), rw_only)
1103 else:
1104 logging.info('Not a Chrome EC, ignore re-programing it')
Tom Wai-Hong Tam4ac78982016-01-08 02:34:37 +08001105 logging.info('Will re-program BIOS %snow', 'RW ' if rw_only else '')
Tom Wai-Hong Tamf9db57a2016-03-17 05:32:22 +08001106 server_utils.system('tar xf %s -C %s %s' %
1107 (local_tarball, tmpd.name, ap_image),
1108 timeout=60)
Tom Wai-Hong Tam4ac78982016-01-08 02:34:37 +08001109 self.servo.program_bios(os.path.join(tmpd.name, ap_image), rw_only)
Dan Shi9cb0eec2014-06-03 09:04:50 -07001110 self.servo.get_power_state_controller().reset()
1111 time.sleep(self.servo.BOOT_DELAY)
Wai-Hong Tamb5f66ce2016-11-10 15:45:30 -08001112 if utils.host_is_in_lab_zone(self.hostname):
1113 self._add_fw_version_label(build, rw_only)
Dan Shi9cb0eec2014-06-03 09:04:50 -07001114 finally:
1115 tmpd.clean()
1116
1117
Dan Shi10e992b2013-08-30 11:02:59 -07001118 def show_update_engine_log(self):
1119 """Output update engine log."""
MK Ryu35d661e2014-09-25 17:44:10 -07001120 logging.debug('Dumping %s', client_constants.UPDATE_ENGINE_LOG)
1121 self.run('cat %s' % client_constants.UPDATE_ENGINE_LOG)
Dan Shi10e992b2013-08-30 11:02:59 -07001122
1123
beepsf079cfb2013-09-18 17:49:51 -07001124 def servo_install(self, image_url=None, usb_boot_timeout=USB_BOOT_TIMEOUT,
1125 install_timeout=INSTALL_TIMEOUT):
Scott Zawalski62bacae2013-03-05 10:40:32 -05001126 """
1127 Re-install the OS on the DUT by:
1128 1) installing a test image on a USB storage device attached to the Servo
1129 board,
Richard Barnette03a0c132012-11-05 12:40:35 -08001130 2) booting that image in recovery mode, and then
J. Richard Barnette31b2e312013-04-04 16:05:22 -07001131 3) installing the image with chromeos-install.
1132
Scott Zawalski62bacae2013-03-05 10:40:32 -05001133 @param image_url: If specified use as the url to install on the DUT.
1134 otherwise boot the currently staged image on the USB stick.
beepsf079cfb2013-09-18 17:49:51 -07001135 @param usb_boot_timeout: The usb_boot_timeout to use during reimage.
1136 Factory images need a longer usb_boot_timeout than regular
1137 cros images.
1138 @param install_timeout: The timeout to use when installing the chromeos
1139 image. Factory images need a longer install_timeout.
Richard Barnette03a0c132012-11-05 12:40:35 -08001140
Scott Zawalski62bacae2013-03-05 10:40:32 -05001141 @raises AutoservError if the image fails to boot.
beepsf079cfb2013-09-18 17:49:51 -07001142
J. Richard Barnette0199cc82014-12-05 17:08:40 -08001143 """
beepsf079cfb2013-09-18 17:49:51 -07001144 logging.info('Downloading image to USB, then booting from it. Usb boot '
1145 'timeout = %s', usb_boot_timeout)
Allen Li48a13fe2016-11-22 14:10:40 -08001146 with metrics.SecondsTimer(
1147 'chromeos/autotest/provision/servo_install/boot_duration'):
1148 self.servo.install_recovery_image(image_url)
1149 if not self.wait_up(timeout=usb_boot_timeout):
1150 raise hosts.AutoservRepairError(
1151 'DUT failed to boot from USB after %d seconds' %
1152 usb_boot_timeout)
Scott Zawalski62bacae2013-03-05 10:40:32 -05001153
Tom Wai-Hong Tamf6b4f812015-08-08 04:14:59 +08001154 # The new chromeos-tpm-recovery has been merged since R44-7073.0.0.
1155 # In old CrOS images, this command fails. Skip the error.
Tom Wai-Hong Tam27af7332015-07-25 06:09:39 +08001156 logging.info('Resetting the TPM status')
Tom Wai-Hong Tamf6b4f812015-08-08 04:14:59 +08001157 try:
1158 self.run('chromeos-tpm-recovery')
1159 except error.AutoservRunError:
1160 logging.warn('chromeos-tpm-recovery is too old.')
1161
Tom Wai-Hong Tam27af7332015-07-25 06:09:39 +08001162
Allen Li48a13fe2016-11-22 14:10:40 -08001163 with metrics.SecondsTimer(
1164 'chromeos/autotest/provision/servo_install/install_duration'):
1165 logging.info('Installing image through chromeos-install.')
1166 self.run('chromeos-install --yes', timeout=install_timeout)
1167 self.halt()
beepsf079cfb2013-09-18 17:49:51 -07001168
1169 logging.info('Power cycling DUT through servo.')
J. Richard Barnette0199cc82014-12-05 17:08:40 -08001170 self.servo.get_power_state_controller().power_off()
Fang Dengafb88142013-05-30 17:44:31 -07001171 self.servo.switch_usbkey('off')
J. Richard Barnette0199cc82014-12-05 17:08:40 -08001172 # N.B. The Servo API requires that we use power_on() here
1173 # for two reasons:
1174 # 1) After turning on a DUT in recovery mode, you must turn
1175 # it off and then on with power_on() once more to
1176 # disable recovery mode (this is a Parrot specific
1177 # requirement).
1178 # 2) After power_off(), the only way to turn on is with
1179 # power_on() (this is a Storm specific requirement).
J. Richard Barnettefbcc7122013-07-24 18:24:59 -07001180 self.servo.get_power_state_controller().power_on()
beepsf079cfb2013-09-18 17:49:51 -07001181
1182 logging.info('Waiting for DUT to come back up.')
Richard Barnette03a0c132012-11-05 12:40:35 -08001183 if not self.wait_up(timeout=self.BOOT_TIMEOUT):
1184 raise error.AutoservError('DUT failed to reboot installed '
1185 'test image after %d seconds' %
Scott Zawalski62bacae2013-03-05 10:40:32 -05001186 self.BOOT_TIMEOUT)
1187
1188
Richard Barnette9a26ad62016-06-10 12:03:08 -07001189 def repair_servo(self):
Dan Shi90466352015-09-22 15:01:05 -07001190 """
Richard Barnette9a26ad62016-06-10 12:03:08 -07001191 Confirm that servo is initialized and verified.
Dan Shi90466352015-09-22 15:01:05 -07001192
Richard Barnette9a26ad62016-06-10 12:03:08 -07001193 If the servo object is missing, attempt to repair the servo
1194 host. Repair failures are passed back to the caller.
1195
1196 @raise AutoservError: If there is no servo host for this CrOS
1197 host.
1198 """
1199 if self.servo:
1200 return
1201 if not self._servo_host:
1202 raise error.AutoservError('No servo host for %s.' %
1203 self.hostname)
1204 self._servo_host.repair()
1205 self.servo = self._servo_host.get_servo()
Dan Shi90466352015-09-22 15:01:05 -07001206
1207
J. Richard Barnettec2d99cf2015-11-18 12:46:15 -08001208 def repair(self):
1209 """Attempt to get the DUT to pass `self.verify()`.
Richard Barnette82c35912012-11-20 10:09:10 -08001210
1211 This overrides the base class function for repair; it does
J. Richard Barnette91137f02016-03-10 16:52:26 -08001212 not call back to the parent class, but instead relies on
1213 `self._repair_strategy` to coordinate the verification and
1214 repair steps needed to get the DUT working.
Richard Barnette82c35912012-11-20 10:09:10 -08001215 """
J. Richard Barnette91137f02016-03-10 16:52:26 -08001216 self._repair_strategy.repair(self)
Richard Barnettebfab8672017-08-10 14:48:28 -07001217 # Sometimes, hosts with certain ethernet dongles get stuck in a
1218 # bad network state where they're reachable from this code, but
1219 # not from the devservers during provisioning. Rebooting the
1220 # DUT fixes it.
1221 #
1222 # TODO(jrbarnette): Ideally, we'd get rid of the problem
1223 # dongles, and drop this code. Failing that, we could be smart
1224 # enough not to reboot if repair rebooted the DUT (e.g. by
1225 # looking at DUT uptime after repair completes).
1226
1227 self.reboot()
Scott Zawalski89c44dd2013-02-26 09:28:02 -05001228
Richard Barnette82c35912012-11-20 10:09:10 -08001229
J. Richard Barnette1d78b012012-05-15 13:56:30 -07001230 def close(self):
Fang Deng0ca40e22013-08-27 17:47:44 -07001231 super(CrosHost, self).close()
xixuand6011f12016-12-08 15:01:58 -08001232 if self._chameleon_host:
1233 self._chameleon_host.close()
1234
1235 if self._servo_host:
1236 self._servo_host.close()
J. Richard Barnette1d78b012012-05-15 13:56:30 -07001237
1238
Dan Shi49ca0932014-11-14 11:22:27 -08001239 def get_power_supply_info(self):
1240 """Get the output of power_supply_info.
1241
1242 power_supply_info outputs the info of each power supply, e.g.,
1243 Device: Line Power
1244 online: no
1245 type: Mains
1246 voltage (V): 0
1247 current (A): 0
1248 Device: Battery
1249 state: Discharging
1250 percentage: 95.9276
1251 technology: Li-ion
1252
1253 Above output shows two devices, Line Power and Battery, with details of
1254 each device listed. This function parses the output into a dictionary,
1255 with key being the device name, and value being a dictionary of details
1256 of the device info.
1257
1258 @return: The dictionary of power_supply_info, e.g.,
1259 {'Line Power': {'online': 'yes', 'type': 'main'},
1260 'Battery': {'vendor': 'xyz', 'percentage': '100'}}
Dan Shie9b765d2014-12-29 16:59:49 -08001261 @raise error.AutoservRunError if power_supply_info tool is not found in
1262 the DUT. Caller should handle this error to avoid false failure
1263 on verification.
Dan Shi49ca0932014-11-14 11:22:27 -08001264 """
1265 result = self.run('power_supply_info').stdout.strip()
1266 info = {}
1267 device_name = None
1268 device_info = {}
1269 for line in result.split('\n'):
1270 pair = [v.strip() for v in line.split(':')]
1271 if len(pair) != 2:
1272 continue
1273 if pair[0] == 'Device':
1274 if device_name:
1275 info[device_name] = device_info
1276 device_name = pair[1]
1277 device_info = {}
1278 else:
1279 device_info[pair[0]] = pair[1]
1280 if device_name and not device_name in info:
1281 info[device_name] = device_info
1282 return info
1283
1284
1285 def get_battery_percentage(self):
1286 """Get the battery percentage.
1287
1288 @return: The percentage of battery level, value range from 0-100. Return
1289 None if the battery info cannot be retrieved.
1290 """
1291 try:
1292 info = self.get_power_supply_info()
1293 logging.info(info)
1294 return float(info['Battery']['percentage'])
Dan Shie9b765d2014-12-29 16:59:49 -08001295 except (KeyError, ValueError, error.AutoservRunError):
Dan Shi49ca0932014-11-14 11:22:27 -08001296 return None
1297
1298
1299 def is_ac_connected(self):
1300 """Check if the dut has power adapter connected and charging.
1301
1302 @return: True if power adapter is connected and charging.
1303 """
1304 try:
1305 info = self.get_power_supply_info()
1306 return info['Line Power']['online'] == 'yes'
Dan Shie9b765d2014-12-29 16:59:49 -08001307 except (KeyError, error.AutoservRunError):
1308 return None
Dan Shi49ca0932014-11-14 11:22:27 -08001309
1310
Simran Basi5e6339a2013-03-21 11:34:32 -07001311 def _cleanup_poweron(self):
1312 """Special cleanup method to make sure hosts always get power back."""
1313 afe = frontend_wrappers.RetryingAFE(timeout_min=5, delay_sec=10)
1314 hosts = afe.get_hosts(hostname=self.hostname)
1315 if not hosts or not (self._RPM_OUTLET_CHANGED in
1316 hosts[0].attributes):
1317 return
1318 logging.debug('This host has recently interacted with the RPM'
1319 ' Infrastructure. Ensuring power is on.')
1320 try:
1321 self.power_on()
Dan Shi7dca56e2014-11-11 17:07:56 -08001322 afe.set_host_attribute(self._RPM_OUTLET_CHANGED, None,
1323 hostname=self.hostname)
Simran Basi5e6339a2013-03-21 11:34:32 -07001324 except rpm_client.RemotePowerException:
Simran Basi5e6339a2013-03-21 11:34:32 -07001325 logging.error('Failed to turn Power On for this host after '
1326 'cleanup through the RPM Infrastructure.')
Dan Shi49ca0932014-11-14 11:22:27 -08001327
1328 battery_percentage = self.get_battery_percentage()
Dan Shif01ebe22014-12-05 13:10:57 -08001329 if battery_percentage and battery_percentage < 50:
Dan Shi49ca0932014-11-14 11:22:27 -08001330 raise
1331 elif self.is_ac_connected():
1332 logging.info('The device has power adapter connected and '
1333 'charging. No need to try to turn RPM on '
1334 'again.')
1335 afe.set_host_attribute(self._RPM_OUTLET_CHANGED, None,
1336 hostname=self.hostname)
1337 logging.info('Battery level is now at %s%%. The device may '
1338 'still have enough power to run test, so no '
1339 'exception will be raised.', battery_percentage)
1340
Simran Basi5e6339a2013-03-21 11:34:32 -07001341
beepsc87ff602013-07-31 21:53:00 -07001342 def _is_factory_image(self):
1343 """Checks if the image on the DUT is a factory image.
1344
1345 @return: True if the image on the DUT is a factory image.
1346 False otherwise.
1347 """
1348 result = self.run('[ -f /root/.factory_test ]', ignore_status=True)
1349 return result.exit_status == 0
1350
1351
1352 def _restart_ui(self):
J. Richard Barnette84890bd2014-02-21 11:05:47 -08001353 """Restart the Chrome UI.
beepsc87ff602013-07-31 21:53:00 -07001354
1355 @raises: FactoryImageCheckerException for factory images, since
1356 we cannot attempt to restart ui on them.
1357 error.AutoservRunError for any other type of error that
1358 occurs while restarting ui.
1359 """
1360 if self._is_factory_image():
Dan Shi549fb822015-03-24 18:01:11 -07001361 raise FactoryImageCheckerException('Cannot restart ui on factory '
1362 'images')
beepsc87ff602013-07-31 21:53:00 -07001363
J. Richard Barnette84890bd2014-02-21 11:05:47 -08001364 # TODO(jrbarnette): The command to stop/start the ui job
1365 # should live inside cros_ui, too. However that would seem
1366 # to imply interface changes to the existing start()/restart()
1367 # functions, which is a bridge too far (for now).
J. Richard Barnette6069aa12015-06-08 09:10:24 -07001368 prompt = cros_ui.get_chrome_session_ident(self)
J. Richard Barnette84890bd2014-02-21 11:05:47 -08001369 self.run('stop ui; start ui')
1370 cros_ui.wait_for_chrome_ready(prompt, self)
beepsc87ff602013-07-31 21:53:00 -07001371
1372
xixuana3bbc422017-05-04 15:57:21 -07001373 def _get_lsb_release_content(self):
1374 """Return the content of lsb-release file of host."""
1375 return self.run(
1376 'cat "%s"' % client_constants.LSB_RELEASE).stdout.strip()
1377
1378
Dan Shi549fb822015-03-24 18:01:11 -07001379 def get_release_version(self):
1380 """Get the value of attribute CHROMEOS_RELEASE_VERSION from lsb-release.
1381
1382 @returns The version string in lsb-release, under attribute
1383 CHROMEOS_RELEASE_VERSION.
1384 """
Dan Shi549fb822015-03-24 18:01:11 -07001385 return lsbrelease_utils.get_chromeos_release_version(
xixuana3bbc422017-05-04 15:57:21 -07001386 lsb_release_content=self._get_lsb_release_content())
1387
1388
1389 def get_chromeos_release_milestone(self):
1390 """Get the value of attribute CHROMEOS_RELEASE_BUILD_TYPE
1391 from lsb-release.
1392
1393 @returns The version string in lsb-release, under attribute
1394 CHROMEOS_RELEASE_BUILD_TYPE.
1395 """
1396 return lsbrelease_utils.get_chromeos_release_milestone(
1397 lsb_release_content=self._get_lsb_release_content())
Dan Shi549fb822015-03-24 18:01:11 -07001398
1399
1400 def verify_cros_version_label(self):
1401 """ Make sure host's cros-version label match the actual image in dut.
1402
1403 Remove any cros-version: label that doesn't match that installed in
1404 the dut.
1405
1406 @param raise_error: Set to True to raise exception if any mismatch found
1407
1408 @raise error.AutoservError: If any mismatch between cros-version label
1409 and the build installed in dut is found.
1410 """
1411 labels = self._AFE.get_labels(
1412 name__startswith=ds_constants.VERSION_PREFIX,
1413 host__hostname=self.hostname)
1414 mismatch_found = False
1415 if labels:
1416 # Get CHROMEOS_RELEASE_VERSION from lsb-release, e.g., 6908.0.0.
1417 # Note that it's different from cros-version label, which has
1418 # builder and branch info, e.g.,
1419 # cros-version:peppy-release/R43-6908.0.0
1420 release_version = self.get_release_version()
1421 host_list = [self.hostname]
1422 for label in labels:
1423 # Remove any cros-version label that does not match
1424 # release_version.
1425 build_version = label.name[len(ds_constants.VERSION_PREFIX):]
1426 if not utils.version_match(build_version, release_version):
1427 logging.warn('cros-version label "%s" does not match '
1428 'release version %s. Removing the label.',
1429 label.name, release_version)
1430 label.remove_hosts(hosts=host_list)
1431 mismatch_found = True
1432 if mismatch_found:
1433 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 """
Richard Barnettee204dc52017-09-26 11:02:25 -07001545 return 'start/running' in self.run('status %s' % service_name,
1546 ignore_status=True).stdout
Simran Basiec564392014-08-25 16:48:09 -07001547
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 """
C Shapiro218e8752017-09-22 11:10:57 -06001957 cmd = 'mosys platform model'
1958 result = self.run(command=cmd, ignore_status=True)
1959 if result.exit_status == 0:
1960 return result.stdout.strip()
1961 else:
1962 # $(mosys platform model) should support all platforms, but it
1963 # currently doesn't, so this reverts to parsing the fw
1964 # for any unsupported mosys platforms.
1965 crossystem = utils.Crossystem(self)
1966 crossystem.init()
1967 # Extract fwid value and use the leading part as the platform id.
1968 # fwid generally follow the format of {platform}.{firmware version}
1969 # Example: Alex.X.YYY.Z or Google_Alex.X.YYY.Z
1970 platform = crossystem.fwid().split('.')[0].lower()
1971 # Newer platforms start with 'Google_' while the older ones do not.
1972 return platform.replace('google_', '')
Simran Basic6f1f7a2012-10-16 10:47:46 -07001973
1974
Hung-ying Tyanb1328032014-04-01 14:18:54 +08001975 def get_architecture(self):
1976 """Determine the correct architecture label for this host.
1977
1978 @returns a string representing this host's architecture.
1979 """
1980 crossystem = utils.Crossystem(self)
1981 crossystem.init()
1982 return crossystem.arch()
1983
1984
Luis Lozano40b7d0d2014-01-17 15:12:06 -08001985 def get_chrome_version(self):
1986 """Gets the Chrome version number and milestone as strings.
1987
1988 Invokes "chrome --version" to get the version number and milestone.
1989
1990 @return A tuple (chrome_ver, milestone) where "chrome_ver" is the
1991 current Chrome version number as a string (in the form "W.X.Y.Z")
1992 and "milestone" is the first component of the version number
1993 (the "W" from "W.X.Y.Z"). If the version number cannot be parsed
1994 in the "W.X.Y.Z" format, the "chrome_ver" will be the full output
1995 of "chrome --version" and the milestone will be the empty string.
1996
1997 """
MK Ryu35d661e2014-09-25 17:44:10 -07001998 version_string = self.run(client_constants.CHROME_VERSION_COMMAND).stdout
Luis Lozano40b7d0d2014-01-17 15:12:06 -08001999 return utils.parse_chrome_version(version_string)
2000
J. Richard Barnetted2af5852016-02-05 15:03:10 -08002001
Niranjan Kumar34618872017-05-31 12:57:09 -07002002 def is_chrome_switch_present(self, switch):
David Haddock3ce538e2017-06-22 13:37:05 -07002003 """Returns True if the specified switch was provided to Chrome.
2004
2005 @param switch The chrome switch to search for.
2006 """
Niranjan Kumar34618872017-05-31 12:57:09 -07002007
Niranjan Kumar5f23fe92017-06-22 15:18:55 -07002008 command = 'pgrep -x -f -c "/opt/google/chrome/chrome.*%s.*"' % switch
2009 return self.run(command, ignore_status=True).exit_status == 0
Niranjan Kumar34618872017-05-31 12:57:09 -07002010
2011
2012 def oobe_triggers_update(self):
2013 """Returns True if this host has an OOBE flow during which
2014 it will perform an update check and perhaps an update.
2015 One example of such a flow is Hands-Off Zero-Touch Enrollment.
2016 As more such flows are developed, code handling them needs
2017 to be added here.
2018
2019 @return Boolean indicating whether this host's OOBE triggers an update.
2020 """
2021 return self.is_chrome_switch_present(
2022 '--enterprise-enable-zero-touch-enrollment=hands-off')
2023
2024
Kevin Chenga2619dc2016-03-28 11:42:08 -07002025 # TODO(kevcheng): change this to just return the board without the
2026 # 'board:' prefix and fix up all the callers. Also look into removing the
2027 # need for this method.
Simran Basic6f1f7a2012-10-16 10:47:46 -07002028 def get_board(self):
2029 """Determine the correct board label for this host.
2030
2031 @returns a string representing this host's board.
2032 """
2033 release_info = utils.parse_cmd_output('cat /etc/lsb-release',
2034 run_method=self.run)
J. Richard Barnetted2af5852016-02-05 15:03:10 -08002035 return (ds_constants.BOARD_PREFIX +
2036 release_info['CHROMEOS_RELEASE_BOARD'])
Simran Basic6f1f7a2012-10-16 10:47:46 -07002037
2038
Kevin Chenga328da62016-03-31 10:49:04 -07002039
2040 def has_lightsensor(self):
2041 """Determine the correct board label for this host.
2042
2043 @returns the string 'lightsensor' if this host has a lightsensor or
2044 None if it does not.
2045 """
2046 search_cmd = "find -L %s -maxdepth 4 | egrep '%s'" % (
2047 self._LIGHTSENSOR_SEARCH_DIR, '|'.join(self._LIGHTSENSOR_FILES))
2048 try:
2049 # Run the search cmd following the symlinks. Stderr_tee is set to
2050 # None as there can be a symlink loop, but the command will still
2051 # execute correctly with a few messages printed to stderr.
2052 self.run(search_cmd, stdout_tee=None, stderr_tee=None)
2053 return 'lightsensor'
2054 except error.AutoservRunError:
2055 # egrep exited with a return code of 1 meaning none of the possible
2056 # lightsensor files existed.
2057 return None
2058
2059
2060 def has_bluetooth(self):
2061 """Determine the correct board label for this host.
2062
2063 @returns the string 'bluetooth' if this host has bluetooth or
2064 None if it does not.
2065 """
2066 try:
2067 self.run('test -d /sys/class/bluetooth/hci0')
2068 # test exited with a return code of 0.
2069 return 'bluetooth'
2070 except error.AutoservRunError:
2071 # test exited with a return code 1 meaning the directory did not
2072 # exist.
2073 return None
2074
2075
Kevin Chenga328da62016-03-31 10:49:04 -07002076 def get_accels(self):
2077 """
2078 Determine the type of accelerometers on this host.
2079
2080 @returns a string representing this host's accelerometer type.
2081 At present, it only returns "accel:cros-ec", for accelerometers
2082 attached to a Chrome OS EC, or none, if no accelerometers.
2083 """
2084 # Check to make sure we have ectool
2085 rv = self.run('which ectool', ignore_status=True)
2086 if rv.exit_status:
2087 logging.info("No ectool cmd found, assuming no EC accelerometers")
2088 return None
2089
2090 # Check that the EC supports the motionsense command
2091 rv = self.run('ectool motionsense', ignore_status=True)
2092 if rv.exit_status:
2093 logging.info("EC does not support motionsense command "
2094 "assuming no EC accelerometers")
2095 return None
2096
2097 # Check that EC motion sensors are active
2098 active = self.run('ectool motionsense active').stdout.split('\n')
2099 if active[0] == "0":
2100 logging.info("Motion sense inactive, assuming no EC accelerometers")
2101 return None
2102
2103 logging.info("EC accelerometers found")
2104 return 'accel:cros-ec'
2105
2106
2107 def has_chameleon(self):
2108 """Determine if a Chameleon connected to this host.
2109
2110 @returns a list containing two strings ('chameleon' and
2111 'chameleon:' + label, e.g. 'chameleon:hdmi') if this host
2112 has a Chameleon or None if it has not.
2113 """
2114 if self._chameleon_host:
2115 return ['chameleon', 'chameleon:' + self.chameleon.get_label()]
2116 else:
2117 return None
2118
2119
2120 def has_loopback_dongle(self):
2121 """Determine if an audio loopback dongle is plugged to this host.
2122
2123 @returns 'audio_loopback_dongle' when there is an audio loopback dongle
2124 plugged to this host.
2125 None when there is no audio loopback dongle
2126 plugged to this host.
2127 """
2128 nodes_info = self.run(command=cras_utils.get_cras_nodes_cmd(),
2129 ignore_status=True).stdout
2130 if (cras_utils.node_type_is_plugged('HEADPHONE', nodes_info) and
2131 cras_utils.node_type_is_plugged('MIC', nodes_info)):
2132 return 'audio_loopback_dongle'
2133 else:
2134 return None
2135
2136
2137 def get_power_supply(self):
2138 """
2139 Determine what type of power supply the host has
2140
2141 @returns a string representing this host's power supply.
2142 'power:battery' when the device has a battery intended for
2143 extended use
2144 'power:AC_primary' when the device has a battery not intended
2145 for extended use (for moving the machine, etc)
2146 'power:AC_only' when the device has no battery at all.
2147 """
2148 psu = self.run(command='mosys psu type', ignore_status=True)
2149 if psu.exit_status:
2150 # The psu command for mosys is not included for all platforms. The
2151 # assumption is that the device will have a battery if the command
2152 # is not found.
2153 return 'power:battery'
2154
2155 psu_str = psu.stdout.strip()
2156 if psu_str == 'unknown':
2157 return None
2158
2159 return 'power:%s' % psu_str
2160
2161
2162 def get_storage(self):
2163 """
2164 Determine the type of boot device for this host.
2165
2166 Determine if the internal device is SCSI or dw_mmc device.
2167 Then check that it is SSD or HDD or eMMC or something else.
2168
2169 @returns a string representing this host's internal device type.
2170 'storage:ssd' when internal device is solid state drive
2171 'storage:hdd' when internal device is hard disk drive
2172 'storage:mmc' when internal device is mmc drive
2173 None When internal device is something else or
2174 when we are unable to determine the type
2175 """
2176 # The output should be /dev/mmcblk* for SD/eMMC or /dev/sd* for scsi
2177 rootdev_cmd = ' '.join(['. /usr/sbin/write_gpt.sh;',
2178 '. /usr/share/misc/chromeos-common.sh;',
2179 'load_base_vars;',
2180 'get_fixed_dst_drive'])
2181 rootdev = self.run(command=rootdev_cmd, ignore_status=True)
2182 if rootdev.exit_status:
2183 logging.info("Fail to run %s", rootdev_cmd)
2184 return None
2185 rootdev_str = rootdev.stdout.strip()
2186
2187 if not rootdev_str:
2188 return None
2189
2190 rootdev_base = os.path.basename(rootdev_str)
2191
2192 mmc_pattern = '/dev/mmcblk[0-9]'
2193 if re.match(mmc_pattern, rootdev_str):
2194 # Use type to determine if the internal device is eMMC or somthing
2195 # else. We can assume that MMC is always an internal device.
2196 type_cmd = 'cat /sys/block/%s/device/type' % rootdev_base
2197 type = self.run(command=type_cmd, ignore_status=True)
2198 if type.exit_status:
2199 logging.info("Fail to run %s", type_cmd)
2200 return None
2201 type_str = type.stdout.strip()
2202
2203 if type_str == 'MMC':
2204 return 'storage:mmc'
2205
2206 scsi_pattern = '/dev/sd[a-z]+'
2207 if re.match(scsi_pattern, rootdev.stdout):
2208 # Read symlink for /sys/block/sd* to determine if the internal
2209 # device is connected via ata or usb.
2210 link_cmd = 'readlink /sys/block/%s' % rootdev_base
2211 link = self.run(command=link_cmd, ignore_status=True)
2212 if link.exit_status:
2213 logging.info("Fail to run %s", link_cmd)
2214 return None
2215 link_str = link.stdout.strip()
2216 if 'usb' in link_str:
2217 return None
2218
2219 # Read rotation to determine if the internal device is ssd or hdd.
2220 rotate_cmd = str('cat /sys/block/%s/queue/rotational'
2221 % rootdev_base)
2222 rotate = self.run(command=rotate_cmd, ignore_status=True)
2223 if rotate.exit_status:
2224 logging.info("Fail to run %s", rotate_cmd)
2225 return None
2226 rotate_str = rotate.stdout.strip()
2227
2228 rotate_dict = {'0':'storage:ssd', '1':'storage:hdd'}
2229 return rotate_dict.get(rotate_str)
2230
2231 # All other internal device / error case will always fall here
2232 return None
2233
2234
2235 def get_servo(self):
2236 """Determine if the host has a servo attached.
2237
2238 If the host has a working servo attached, it should have a servo label.
2239
2240 @return: string 'servo' if the host has servo attached. Otherwise,
2241 returns None.
2242 """
2243 return 'servo' if self._servo_host else None
2244
2245
2246 def get_video_labels(self):
2247 """Run /usr/local/bin/avtest_label_detect to get a list of video labels.
2248
2249 Sample output of avtest_label_detect:
2250 Detected label: hw_video_acc_vp8
2251 Detected label: webcam
2252
2253 @return: A list of labels detected by tool avtest_label_detect.
2254 """
2255 try:
2256 result = self.run('/usr/local/bin/avtest_label_detect').stdout
2257 return re.findall('^Detected label: (\w+)$', result, re.M)
2258 except error.AutoservRunError:
2259 # The tool is not installed.
2260 return []
2261
2262
2263 def is_video_glitch_detection_supported(self):
2264 """ Determine if a board under test is supported for video glitch
2265 detection tests.
2266
2267 @return: 'video_glitch_detection' if board is supported, None otherwise.
2268 """
2269 board = self.get_board().replace(ds_constants.BOARD_PREFIX, '')
2270
2271 if board in video_test_constants.SUPPORTED_BOARDS:
2272 return 'video_glitch_detection'
2273
2274 return None
2275
2276
2277 def get_touch(self):
2278 """
2279 Determine whether board under test has a touchpad or touchscreen.
2280
2281 @return: A list of some combination of 'touchscreen' and 'touchpad',
2282 depending on what is present on the device.
2283
2284 """
2285 labels = []
2286 looking_for = ['touchpad', 'touchscreen']
2287 player = input_playback.InputPlayback()
2288 input_events = self.run('ls /dev/input/event*').stdout.strip().split()
2289 filename = '/tmp/touch_labels'
2290 for event in input_events:
2291 self.run('evtest %s > %s' % (event, filename), timeout=1,
2292 ignore_timeout=True)
2293 properties = self.run('cat %s' % filename).stdout
2294 input_type = player._determine_input_type(properties)
2295 if input_type in looking_for:
2296 labels.append(input_type)
2297 looking_for.remove(input_type)
2298 if len(looking_for) == 0:
2299 break
2300 self.run('rm %s' % filename)
2301
2302 return labels
2303
2304
2305 def has_internal_display(self):
2306 """Determine if the device under test is equipped with an internal
2307 display.
2308
2309 @return: 'internal_display' if one is present; None otherwise.
2310 """
2311 from autotest_lib.client.cros.graphics import graphics_utils
2312 from autotest_lib.client.common_lib import utils as common_utils
2313
2314 def __system_output(cmd):
2315 return self.run(cmd).stdout
2316
2317 def __read_file(remote_path):
2318 return self.run('cat %s' % remote_path).stdout
2319
2320 # Hijack the necessary client functions so that we can take advantage
2321 # of the client lib here.
2322 # FIXME: find a less hacky way than this
2323 original_system_output = utils.system_output
2324 original_read_file = common_utils.read_file
2325 utils.system_output = __system_output
2326 common_utils.read_file = __read_file
2327 try:
2328 return ('internal_display' if graphics_utils.has_internal_display()
2329 else None)
2330 finally:
2331 utils.system_output = original_system_output
2332 common_utils.read_file = original_read_file
2333
2334
Dan Shi85276d42014-04-08 22:11:45 -07002335 def is_boot_from_usb(self):
2336 """Check if DUT is boot from USB.
2337
2338 @return: True if DUT is boot from usb.
2339 """
2340 device = self.run('rootdev -s -d').stdout.strip()
2341 removable = int(self.run('cat /sys/block/%s/removable' %
2342 os.path.basename(device)).stdout.strip())
2343 return removable == 1
Helen Zhang17dae2b2014-11-11 09:25:52 -08002344
2345
2346 def read_from_meminfo(self, key):
Dan Shi49ca0932014-11-14 11:22:27 -08002347 """Return the memory info from /proc/meminfo
Helen Zhang17dae2b2014-11-11 09:25:52 -08002348
2349 @param key: meminfo requested
2350
2351 @return the memory value as a string
2352
2353 """
Helen Zhang17dae2b2014-11-11 09:25:52 -08002354 meminfo = self.run('grep %s /proc/meminfo' % key).stdout.strip()
2355 logging.debug('%s', meminfo)
2356 return int(re.search(r'\d+', meminfo).group(0))
Rohit Makasana8a4923c2015-08-13 17:04:26 -07002357
2358
Rohit Makasana98e696f2016-06-03 18:48:10 -07002359 def get_cpu_arch(self):
2360 """Returns CPU arch of the device.
2361
2362 @return CPU architecture of the DUT.
2363 """
Allen Li2c32d6b2017-02-03 15:28:10 -08002364 # Add CPUs by following logic in client/bin/utils.py.
Rohit Makasana98e696f2016-06-03 18:48:10 -07002365 if self.run("grep '^flags.*:.* lm .*' /proc/cpuinfo",
2366 ignore_status=True).stdout:
2367 return 'x86_64'
2368 if self.run("grep -Ei 'ARM|CPU implementer' /proc/cpuinfo",
2369 ignore_status=True).stdout:
2370 return 'arm'
2371 return 'i386'
2372
2373
Rohit Makasana8a4923c2015-08-13 17:04:26 -07002374 def get_board_type(self):
2375 """
2376 Get the DUT's device type from /etc/lsb-release.
Danny Chan471a8d12015-08-18 14:57:41 -07002377 DEVICETYPE can be one of CHROMEBOX, CHROMEBASE, CHROMEBOOK or more.
2378
2379 @return value of DEVICETYPE param from lsb-release.
Rohit Makasana8a4923c2015-08-13 17:04:26 -07002380 """
Danny Chan471a8d12015-08-18 14:57:41 -07002381 device_type = self.run('grep DEVICETYPE /etc/lsb-release',
2382 ignore_status=True).stdout
2383 if device_type:
Kalin Stoyanov524310b2015-08-21 16:24:04 -07002384 return device_type.split('=')[-1].strip()
Danny Chan471a8d12015-08-18 14:57:41 -07002385 return ''
Gilad Arnolda76bef02015-09-29 13:55:15 -07002386
2387
Rohit Makasanadf0a3a32017-06-30 13:55:18 -07002388 def get_arc_version(self):
2389 """Return ARC version installed on the DUT.
2390
2391 @returns ARC version as string if the CrOS build has ARC, else None.
2392 """
2393 arc_version = self.run('grep CHROMEOS_ARC_VERSION /etc/lsb-release',
2394 ignore_status=True).stdout
2395 if arc_version:
2396 return arc_version.split('=')[-1].strip()
2397 return None
2398
2399
Gilad Arnolda76bef02015-09-29 13:55:15 -07002400 def get_os_type(self):
2401 return 'cros'
Simran Basia5522a32015-10-06 11:01:24 -07002402
2403
2404 def enable_adb_testing(self):
2405 """Mark this host as an adb tester."""
Dan Shia2872172015-10-31 01:16:51 -07002406 self.run('touch %s' % constants.ANDROID_TESTER_FILEFLAG)
Kevin Chenga2619dc2016-03-28 11:42:08 -07002407
2408
2409 def get_labels(self):
Kevin Chengf8660142016-08-12 10:17:41 -07002410 """Return the detected labels on the host."""
Kevin Chenga2619dc2016-03-28 11:42:08 -07002411 return self.labels.get_labels(self)
Kevin Chengf8660142016-08-12 10:17:41 -07002412
2413
2414 def update_labels(self):
2415 """Update the labels on the host."""
2416 self.labels.update_labels(self)