blob: 34fcc60a050683a0ba123e9bb2271605b5c279f5 [file] [log] [blame]
Derek Beckettf73baca2020-08-19 15:08:47 -07001# Lint as: python2, python3
J. Richard Barnette24adbf42012-04-11 15:04:53 -07002# Copyright (c) 2012 The Chromium OS Authors. All rights reserved.
Dale Curtisaa5eedb2011-08-23 16:18:52 -07003# Use of this source code is governed by a BSD-style license that can be
4# found in the LICENSE file.
5
Derek Beckettf73baca2020-08-19 15:08:47 -07006from __future__ import absolute_import
7from __future__ import division
8from __future__ import print_function
9
J. Richard Barnette1d78b012012-05-15 13:56:30 -070010import logging
Dan Shi0f466e82013-02-22 15:44:58 -080011import os
Simran Basid5e5e272012-09-24 15:23:59 -070012import re
Vincent Palatindf2372c2016-10-07 17:03:00 +020013import sys
Otabek Kasimovb7cb8422020-12-23 02:38:32 -080014import six
J. Richard Barnette134ec2c2012-04-25 12:59:37 -070015import time
16
mussa584b4462014-06-20 15:13:28 -070017import common
J. Richard Barnette45e93de2012-04-11 17:24:15 -070018from autotest_lib.client.bin import utils
Dan Shi9cb0eec2014-06-03 09:04:50 -070019from autotest_lib.client.common_lib import autotemp
Richard Barnette0c73ffc2012-11-19 15:21:18 -080020from autotest_lib.client.common_lib import error
21from autotest_lib.client.common_lib import global_config
J. Richard Barnette91137f02016-03-10 16:52:26 -080022from autotest_lib.client.common_lib import hosts
Dan Shi549fb822015-03-24 18:01:11 -070023from autotest_lib.client.common_lib import lsbrelease_utils
Otabek Kasimov6825b762020-06-23 23:42:44 -070024from autotest_lib.client.common_lib import utils as common_utils
Greg Edelstona7b05d12020-04-01 16:00:51 -060025from autotest_lib.client.common_lib.cros import cros_config
Richard Barnette03a0c132012-11-05 12:40:35 -080026from autotest_lib.client.common_lib.cros import dev_server
Justin TerAvest3b0c5ba2017-11-07 09:59:11 -070027from autotest_lib.client.common_lib.cros import retry
Hsinyu Chaoe0b08e62015-08-11 10:50:37 +000028from autotest_lib.client.cros import constants as client_constants
J. Richard Barnette84890bd2014-02-21 11:05:47 -080029from autotest_lib.client.cros import cros_ui
Simran Basi5ace6f22016-01-06 17:30:44 -080030from autotest_lib.server import afe_utils
Dan Shia1ecd5c2013-06-06 11:21:31 -070031from autotest_lib.server import utils as server_utils
Dan Shi9cb0eec2014-06-03 09:04:50 -070032from autotest_lib.server.cros import provision
Scott Zawalski89c44dd2013-02-26 09:28:02 -050033from autotest_lib.server.cros.dynamic_suite import constants as ds_constants
Simran Basi5e6339a2013-03-21 11:34:32 -070034from autotest_lib.server.cros.dynamic_suite import tools, frontend_wrappers
Garry Wang1a493d82020-08-31 21:01:19 -070035from autotest_lib.server.cros.device_health_profile import device_health_profile
Garry Wanga2e78172020-09-09 23:49:07 -070036from autotest_lib.server.cros.device_health_profile import profile_constants
Wai-Hong Tam0ed9fe12019-05-23 11:00:58 -070037from autotest_lib.server.cros.servo import pdtester
Fang Deng96667ca2013-08-01 17:46:18 -070038from autotest_lib.server.hosts import abstract_ssh
Kevin Chenga2619dc2016-03-28 11:42:08 -070039from autotest_lib.server.hosts import base_label
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +080040from autotest_lib.server.hosts import chameleon_host
Otabek Kasimov832d9162020-07-27 19:24:57 -070041from autotest_lib.server.hosts import cros_constants
Richard Barnetted31580e2018-05-14 19:58:00 +000042from autotest_lib.server.hosts import cros_label
J. Richard Barnettea7e7fdc2016-02-12 12:35:36 -080043from autotest_lib.server.hosts import cros_repair
Wai-Hong Tam0ed9fe12019-05-23 11:00:58 -070044from autotest_lib.server.hosts import pdtester_host
Fang Deng5d518f42013-08-02 14:04:32 -070045from autotest_lib.server.hosts import servo_host
Garry Wang11b5e872020-03-11 15:14:08 -070046from autotest_lib.server.hosts import servo_constants
Simran Basidcff4252012-11-20 16:13:20 -080047from autotest_lib.site_utils.rpm_control_system import rpm_client
Otabek Kasimov808cd832020-05-28 18:27:46 -070048from autotest_lib.site_utils.admin_audit import constants as audit_const
Otabek Kasimov27bb2862020-08-10 14:40:45 -070049from autotest_lib.site_utils.admin_audit import verifiers as audit_verify
Derek Beckettf73baca2020-08-19 15:08:47 -070050from six.moves import zip
Simran Basid5e5e272012-09-24 15:23:59 -070051
Andrew Luo4be621d2020-03-21 07:01:13 -070052
Simran Basi382506b2016-09-13 14:58:15 -070053# In case cros_host is being ran via SSP on an older Moblab version with an
54# older chromite version.
55try:
Mike Frysinger714c5b02020-09-04 23:22:54 -040056 from autotest_lib.utils.frozen_chromite.lib import metrics
Dan Shi5e2efb72017-02-07 11:40:23 -080057except ImportError:
Congbin Guo42427612019-02-12 10:22:06 -080058 metrics = utils.metrics_mock
Dan Shi5e2efb72017-02-07 11:40:23 -080059
Simran Basid5e5e272012-09-24 15:23:59 -070060
Dan Shib8540a52015-07-16 14:18:23 -070061CONFIG = global_config.global_config
62
beepsc87ff602013-07-31 21:53:00 -070063class FactoryImageCheckerException(error.AutoservError):
64 """Exception raised when an image is a factory image."""
65 pass
66
67
Fang Deng0ca40e22013-08-27 17:47:44 -070068class CrosHost(abstract_ssh.AbstractSSHHost):
J. Richard Barnette45e93de2012-04-11 17:24:15 -070069 """Chromium OS specific subclass of Host."""
70
Simran Basi5ace6f22016-01-06 17:30:44 -080071 VERSION_PREFIX = provision.CROS_VERSION_PREFIX
72
Scott Zawalski62bacae2013-03-05 10:40:32 -050073 _AFE = frontend_wrappers.RetryingAFE(timeout_min=5, delay_sec=10)
J. Richard Barnette45e93de2012-04-11 17:24:15 -070074
Richard Barnette03a0c132012-11-05 12:40:35 -080075 # Timeout values (in seconds) associated with various Chrome OS
76 # state changes.
J. Richard Barnette134ec2c2012-04-25 12:59:37 -070077 #
Richard Barnette0c73ffc2012-11-19 15:21:18 -080078 # In general, a good rule of thumb is that the timeout can be up
79 # to twice the typical measured value on the slowest platform.
80 # The times here have not necessarily been empirically tested to
81 # meet this criterion.
J. Richard Barnetteeb69d722012-06-18 17:29:44 -070082 #
83 # SLEEP_TIMEOUT: Time to allow for suspend to memory.
Richard Barnette0c73ffc2012-11-19 15:21:18 -080084 # RESUME_TIMEOUT: Time to allow for resume after suspend, plus
85 # time to restart the netwowrk.
J. Richard Barnette84890bd2014-02-21 11:05:47 -080086 # SHUTDOWN_TIMEOUT: Time to allow for shut down.
J. Richard Barnetteeb69d722012-06-18 17:29:44 -070087 # BOOT_TIMEOUT: Time to allow for boot from power off. Among
Richard Barnette0c73ffc2012-11-19 15:21:18 -080088 # other things, this must account for the 30 second dev-mode
J. Richard Barnette417cc792015-10-01 09:56:36 -070089 # screen delay, time to start the network on the DUT, and the
90 # ssh timeout of 120 seconds.
J. Richard Barnetteeb69d722012-06-18 17:29:44 -070091 # USB_BOOT_TIMEOUT: Time to allow for boot from a USB device,
Richard Barnette0c73ffc2012-11-19 15:21:18 -080092 # including the 30 second dev-mode delay and time to start the
J. Richard Barnetted4649c62013-03-06 17:42:27 -080093 # network.
beepsf079cfb2013-09-18 17:49:51 -070094 # INSTALL_TIMEOUT: Time to allow for chromeos-install.
Otabek Kasimovaeb47fe2021-01-26 20:53:55 -080095 # ADMIN_INSTALL_TIMEOUT: Time to allow for chromeos-install
96 # used by admin tasks.
J. Richard Barnette84890bd2014-02-21 11:05:47 -080097 # POWERWASH_BOOT_TIMEOUT: Time to allow for a reboot that
98 # includes powerwash.
J. Richard Barnetteeb69d722012-06-18 17:29:44 -070099
100 SLEEP_TIMEOUT = 2
J. Richard Barnetted4649c62013-03-06 17:42:27 -0800101 RESUME_TIMEOUT = 10
Tom Wai-Hong Tamfe005c22014-12-03 09:25:44 +0800102 SHUTDOWN_TIMEOUT = 10
J. Richard Barnette417cc792015-10-01 09:56:36 -0700103 BOOT_TIMEOUT = 150
J. Richard Barnette5bab5f52015-08-03 13:14:38 -0700104 USB_BOOT_TIMEOUT = 300
J. Richard Barnette7817b052014-08-28 09:47:29 -0700105 INSTALL_TIMEOUT = 480
Otabek Kasimovaeb47fe2021-01-26 20:53:55 -0800106 ADMIN_INSTALL_TIMEOUT = 600
Dan Shi2c88eed2013-11-12 10:18:38 -0800107 POWERWASH_BOOT_TIMEOUT = 60
Chris Sosab76e0ee2013-05-22 16:55:41 -0700108
Dan Shica503482015-03-30 17:23:25 -0700109 # Minimum OS version that supports server side packaging. Older builds may
110 # not have server side package built or with Autotest code change to support
111 # server-side packaging.
Dan Shib8540a52015-07-16 14:18:23 -0700112 MIN_VERSION_SUPPORT_SSP = CONFIG.get_config_value(
Dan Shiced09e42015-04-17 16:09:34 -0700113 'AUTOSERV', 'min_version_support_ssp', type=int)
Dan Shica503482015-03-30 17:23:25 -0700114
Dana Goyettec172b172020-07-29 16:26:15 -0700115 USE_FSFREEZE = CONFIG.get_config_value(
Dana Goyette6242cb32020-09-23 11:02:57 -0700116 'CROS', 'enable_fs_freeze', type=bool, default=False)
Dana Goyettec172b172020-07-29 16:26:15 -0700117
J. Richard Barnette84890bd2014-02-21 11:05:47 -0800118 # REBOOT_TIMEOUT: How long to wait for a reboot.
119 #
Chris Sosab76e0ee2013-05-22 16:55:41 -0700120 # We have a long timeout to ensure we don't flakily fail due to other
121 # issues. Shorter timeouts are vetted in platform_RebootAfterUpdate.
Simran Basi1160e2c2013-10-04 16:00:24 -0700122 # TODO(sbasi - crbug.com/276094) Restore to 5 mins once the 'host did not
123 # return from reboot' bug is solved.
124 REBOOT_TIMEOUT = 480
Chris Sosab76e0ee2013-05-22 16:55:41 -0700125
Ismail Noorbasha07fdb612013-02-14 14:13:31 -0800126 # _USB_POWER_TIMEOUT: Time to allow for USB to power toggle ON and OFF.
127 # _POWER_CYCLE_TIMEOUT: Time to allow for manual power cycle.
Garry Wang5e5538a2019-04-08 15:36:18 -0700128 # _CHANGE_SERVO_ROLE_TIMEOUT: Time to allow DUT regain network connection
129 # since changing servo role will reset USB state
130 # and causes temporary ethernet drop.
Ismail Noorbasha07fdb612013-02-14 14:13:31 -0800131 _USB_POWER_TIMEOUT = 5
132 _POWER_CYCLE_TIMEOUT = 10
Garry Wang5e5538a2019-04-08 15:36:18 -0700133 _CHANGE_SERVO_ROLE_TIMEOUT = 180
Ismail Noorbasha07fdb612013-02-14 14:13:31 -0800134
Fang Dengdeba14f2014-11-14 11:54:09 -0800135 _RPM_HOSTNAME_REGEX = ('chromeos(\d+)(-row(\d+))?-rack(\d+[a-z]*)'
136 '-host(\d+)')
J. Richard Barnette134ec2c2012-04-25 12:59:37 -0700137
J. Richard Barnetteb6de7e32013-02-14 13:28:04 -0800138 # Constants used in ping_wait_up() and ping_wait_down().
139 #
140 # _PING_WAIT_COUNT is the approximate number of polling
141 # cycles to use when waiting for a host state change.
142 #
143 # _PING_STATUS_DOWN and _PING_STATUS_UP are names used
144 # for arguments to the internal _ping_wait_for_status()
145 # method.
146 _PING_WAIT_COUNT = 40
147 _PING_STATUS_DOWN = False
148 _PING_STATUS_UP = True
149
Ismail Noorbasha07fdb612013-02-14 14:13:31 -0800150 # Allowed values for the power_method argument.
151
Garry Wang5e5538a2019-04-08 15:36:18 -0700152 # POWER_CONTROL_RPM: Used in power_off/on/cycle() methods, default for all
153 # DUTs except those with servo_v4 CCD.
154 # POWER_CONTROL_CCD: Used in power_off/on/cycle() methods, default for all
155 # DUTs with servo_v4 CCD.
Ismail Noorbasha07fdb612013-02-14 14:13:31 -0800156 # POWER_CONTROL_SERVO: Used in set_power() and power_cycle() methods.
157 # POWER_CONTROL_MANUAL: Used in set_power() and power_cycle() methods.
158 POWER_CONTROL_RPM = 'RPM'
Garry Wang5e5538a2019-04-08 15:36:18 -0700159 POWER_CONTROL_CCD = 'CCD'
Ismail Noorbasha07fdb612013-02-14 14:13:31 -0800160 POWER_CONTROL_SERVO = 'servoj10'
161 POWER_CONTROL_MANUAL = 'manual'
162
163 POWER_CONTROL_VALID_ARGS = (POWER_CONTROL_RPM,
Garry Wang5e5538a2019-04-08 15:36:18 -0700164 POWER_CONTROL_CCD,
Ismail Noorbasha07fdb612013-02-14 14:13:31 -0800165 POWER_CONTROL_SERVO,
166 POWER_CONTROL_MANUAL)
Richard Barnette0c73ffc2012-11-19 15:21:18 -0800167
Simran Basi5e6339a2013-03-21 11:34:32 -0700168 _RPM_OUTLET_CHANGED = 'outlet_changed'
169
Dan Shi9cb0eec2014-06-03 09:04:50 -0700170 # URL pattern to download firmware image.
Dan Shib8540a52015-07-16 14:18:23 -0700171 _FW_IMAGE_URL_PATTERN = CONFIG.get_config_value(
Dan Shi9cb0eec2014-06-03 09:04:50 -0700172 'CROS', 'firmware_url_pattern', type=str)
beeps687243d2013-07-18 15:29:27 -0700173
Brent Peterson1cb623a2020-01-09 13:14:28 -0800174 # Regular expression for extracting EC version string
175 _EC_REGEX = '(%s_\w*[-\.]\w*[-\.]\w*[-\.]\w*)'
176
177 # Regular expression for extracting BIOS version string
178 _BIOS_REGEX = '(%s\.\w*\.\w*\.\w*)'
179
Brent Petersonc70a1832020-01-24 15:54:35 -0800180 # Command to update firmware located on DUT
Namyoon Woo382e5892020-05-20 16:48:40 -0700181 _FW_UPDATE_CMD = 'chromeos-firmwareupdate --mode=recovery %s'
Brent Petersonc70a1832020-01-24 15:54:35 -0800182
J. Richard Barnette964fba02012-10-24 17:34:29 -0700183 @staticmethod
beeps46dadc92013-11-07 14:07:10 -0800184 def check_host(host, timeout=10):
185 """
186 Check if the given host is a chrome-os host.
187
188 @param host: An ssh host representing a device.
189 @param timeout: The timeout for the run command.
190
191 @return: True if the host device is chromeos.
192
beeps46dadc92013-11-07 14:07:10 -0800193 """
194 try:
Allen Liad719c12017-06-27 23:48:04 +0000195 result = host.run(
Simran Basi933c8af2015-04-29 14:05:07 -0700196 'grep -q CHROMEOS /etc/lsb-release && '
Garry Wange4b6d6e2019-06-17 17:08:46 -0700197 '! grep -q moblab /etc/lsb-release && '
Derek Beckett342e3e62021-01-05 17:17:23 -0800198 '! grep -q labstation /etc/lsb-release &&'
199 ' grep CHROMEOS_RELEASE_BOARD /etc/lsb-release',
200 ignore_status=True,
Laurence Goodby468de252017-06-08 17:22:53 -0700201 timeout=timeout).stdout
Derek Beckett342e3e62021-01-05 17:17:23 -0800202 if result:
Pradeep Sawlaniaa013fa2017-11-09 06:34:18 -0800203 return not (
204 lsbrelease_utils.is_jetstream(
Derek Beckett342e3e62021-01-05 17:17:23 -0800205 lsb_release_content=result) or
Pradeep Sawlaniaa013fa2017-11-09 06:34:18 -0800206 lsbrelease_utils.is_gce_board(
Derek Beckett342e3e62021-01-05 17:17:23 -0800207 lsb_release_content=result))
Pradeep Sawlaniaa013fa2017-11-09 06:34:18 -0800208
beeps46dadc92013-11-07 14:07:10 -0800209 except (error.AutoservRunError, error.AutoservSSHTimeout):
210 return False
Laurence Goodby468de252017-06-08 17:22:53 -0700211
212 return False
beeps46dadc92013-11-07 14:07:10 -0800213
214
215 @staticmethod
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +0800216 def get_chameleon_arguments(args_dict):
217 """Extract chameleon options from `args_dict` and return the result.
218
219 Recommended usage:
220 ~~~~~~~~
221 args_dict = utils.args_to_dict(args)
222 chameleon_args = hosts.CrosHost.get_chameleon_arguments(args_dict)
223 host = hosts.create_host(machine, chameleon_args=chameleon_args)
224 ~~~~~~~~
225
226 @param args_dict Dictionary from which to extract the chameleon
227 arguments.
228 """
Sam McNally66594ca2019-12-09 12:45:44 +1100229 chameleon_args = {key: args_dict[key]
230 for key in ('chameleon_host', 'chameleon_port')
231 if key in args_dict}
232 if 'chameleon_ssh_port' in args_dict:
233 chameleon_args['port'] = int(args_dict['chameleon_ssh_port'])
234 return chameleon_args
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +0800235
Shijin Abrahamc09587d2020-02-14 20:46:55 -0800236 @staticmethod
237 def get_btpeer_arguments(args_dict):
238 """Extract btpeer options from `args_dict` and return the result.
239
240 This is used to parse details of Bluetooth peer.
241 Recommended usage:
242 ~~~~~~~~
243 args_dict = utils.args_to_dict(args)
244 btpeer_args = hosts.CrosHost.get_btpeer_arguments(args_dict)
Shijin Abraham22f24cb2020-03-26 09:07:41 -0700245 host = hosts.create_host(machine, btpeer_args=btpeer_args)
Shijin Abrahamc09587d2020-02-14 20:46:55 -0800246 ~~~~~~~~
247
248 @param args_dict: Dictionary from which to extract the btpeer
249 arguments.
250 """
251 if 'btpeer_host_list' in args_dict:
252 result = []
253 for btpeer in args_dict['btpeer_host_list'].split(','):
Claire Changd0b19842020-11-04 22:28:45 +0800254 # IPv6 addresses including a port number should be enclosed in
255 # square brackets.
256 delimiter = ']:' if re.search(r':.*:', btpeer) else ':'
Shijin Abrahamc09587d2020-02-14 20:46:55 -0800257 result.append({key: value for key,value in
258 zip(('btpeer_host','btpeer_port'),
Claire Changd0b19842020-11-04 22:28:45 +0800259 btpeer.strip('[]').split(delimiter))})
Shijin Abrahamc09587d2020-02-14 20:46:55 -0800260 return result
261 else:
Anand K Mistrye8933092020-08-05 14:49:41 +1000262 return {key: args_dict[key]
263 for key in ('btpeer_host', 'btpeer_port', 'btpeer_ssh_port')
Shijin Abrahamc09587d2020-02-14 20:46:55 -0800264 if key in args_dict}
265
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +0800266
267 @staticmethod
Wai-Hong Tam0ed9fe12019-05-23 11:00:58 -0700268 def get_pdtester_arguments(args_dict):
Scottfe06ed82015-11-05 17:15:01 -0800269 """Extract chameleon options from `args_dict` and return the result.
270
271 Recommended usage:
272 ~~~~~~~~
273 args_dict = utils.args_to_dict(args)
Wai-Hong Tam0ed9fe12019-05-23 11:00:58 -0700274 pdtester_args = hosts.CrosHost.get_pdtester_arguments(args_dict)
275 host = hosts.create_host(machine, pdtester_args=pdtester_args)
Scottfe06ed82015-11-05 17:15:01 -0800276 ~~~~~~~~
277
Wai-Hong Tam0ed9fe12019-05-23 11:00:58 -0700278 @param args_dict Dictionary from which to extract the pdtester
Scottfe06ed82015-11-05 17:15:01 -0800279 arguments.
280 """
Allen Li083866b2016-08-18 10:07:10 -0700281 return {key: args_dict[key]
Wai-Hong Tam0ed9fe12019-05-23 11:00:58 -0700282 for key in ('pdtester_host', 'pdtester_port')
Allen Li083866b2016-08-18 10:07:10 -0700283 if key in args_dict}
Scottfe06ed82015-11-05 17:15:01 -0800284
285
286 @staticmethod
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +0800287 def get_servo_arguments(args_dict):
288 """Extract servo options from `args_dict` and return the result.
J. Richard Barnette7214e0b2013-02-06 15:20:49 -0800289
290 Recommended usage:
291 ~~~~~~~~
292 args_dict = utils.args_to_dict(args)
Fang Deng0ca40e22013-08-27 17:47:44 -0700293 servo_args = hosts.CrosHost.get_servo_arguments(args_dict)
J. Richard Barnette7214e0b2013-02-06 15:20:49 -0800294 host = hosts.create_host(machine, servo_args=servo_args)
295 ~~~~~~~~
296
297 @param args_dict Dictionary from which to extract the servo
298 arguments.
299 """
Garry Wang11b5e872020-03-11 15:14:08 -0700300 servo_attrs = (servo_constants.SERVO_HOST_ATTR,
Andrew Luo4be621d2020-03-21 07:01:13 -0700301 servo_constants.SERVO_HOST_SSH_PORT_ATTR,
Garry Wang11b5e872020-03-11 15:14:08 -0700302 servo_constants.SERVO_PORT_ATTR,
Otabek Kasimov382c3bb2020-10-28 13:22:45 -0700303 servo_constants.SERVO_SERIAL_ATTR,
Garry Wang11b5e872020-03-11 15:14:08 -0700304 servo_constants.SERVO_BOARD_ATTR,
305 servo_constants.SERVO_MODEL_ATTR)
Armando Miraglia2ac802e2017-08-15 14:54:47 +0200306 servo_args = {key: args_dict[key]
307 for key in servo_attrs
308 if key in args_dict}
309 return (
310 None
Garry Wang11b5e872020-03-11 15:14:08 -0700311 if servo_constants.SERVO_HOST_ATTR in servo_args
312 and not servo_args[servo_constants.SERVO_HOST_ATTR]
Armando Miraglia2ac802e2017-08-15 14:54:47 +0200313 else servo_args)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -0700314
J. Richard Barnette964fba02012-10-24 17:34:29 -0700315
Otabek Kasimov1b70e8d2020-12-30 13:51:00 -0800316 def _initialize(self,
317 hostname,
318 chameleon_args=None,
319 servo_args=None,
320 pdtester_args=None,
321 try_lab_servo=False,
322 try_servo_repair=False,
323 ssh_verbosity_flag='',
324 ssh_options='',
325 try_servo_recovery=False,
326 *args,
327 **dargs):
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +0800328 """Initialize superclasses, |self.chameleon|, and |self.servo|.
J. Richard Barnette67ccb872012-04-19 16:34:56 -0700329
Fang Denge545abb2014-12-30 18:43:47 -0800330 This method will attempt to create the test-assistant object
331 (chameleon/servo) when it is needed by the test. Check
332 the docstring of chameleon_host.create_chameleon_host and
333 servo_host.create_servo_host for how this is determined.
Fang Deng5d518f42013-08-02 14:04:32 -0700334
Fang Denge545abb2014-12-30 18:43:47 -0800335 @param hostname: Hostname of the dut.
336 @param chameleon_args: A dictionary that contains args for creating
337 a ChameleonHost. See chameleon_host for details.
338 @param servo_args: A dictionary that contains args for creating
339 a ServoHost object. See servo_host for details.
Richard Barnette9a26ad62016-06-10 12:03:08 -0700340 @param try_lab_servo: When true, indicates that an attempt should
341 be made to create a ServoHost for a DUT in
342 the test lab, even if not required by
343 `servo_args`. See servo_host for details.
344 @param try_servo_repair: If a servo host is created, check it
345 with `repair()` rather than `verify()`.
Fang Denge545abb2014-12-30 18:43:47 -0800346 See servo_host for details.
347 @param ssh_verbosity_flag: String, to pass to the ssh command to control
348 verbosity.
349 @param ssh_options: String, other ssh options to pass to the ssh
350 command.
Otabek Kasimov1b70e8d2020-12-30 13:51:00 -0800351 @param try_servo_recovery: When True, start servod in recovery mode.
352 See servo_host for details.
J. Richard Barnette67ccb872012-04-19 16:34:56 -0700353 """
Andrew Luo4be621d2020-03-21 07:01:13 -0700354 super(CrosHost, self)._initialize(hostname=hostname, *args, **dargs)
J. Richard Barnette91137f02016-03-10 16:52:26 -0800355 self._repair_strategy = cros_repair.create_cros_repair_strategy()
Otabek Kasimov6825b762020-06-23 23:42:44 -0700356 # hold special dut_state for repair process
357 self._device_repair_state = None
Kevin Chenga2619dc2016-03-28 11:42:08 -0700358 self.labels = base_label.LabelRetriever(cros_label.CROS_LABELS)
J. Richard Barnettef0859852012-08-20 14:55:50 -0700359 # self.env is a dictionary of environment variable settings
360 # to be exported for commands run on the host.
361 # LIBC_FATAL_STDERR_ can be useful for diagnosing certain
362 # errors that might happen.
363 self.env['LIBC_FATAL_STDERR_'] = '1'
Fang Dengd1c2b732013-08-20 12:59:46 -0700364 self._ssh_verbosity_flag = ssh_verbosity_flag
Aviv Keshetc5947fa2013-09-04 14:06:29 -0700365 self._ssh_options = ssh_options
Garry Wang1a493d82020-08-31 21:01:19 -0700366 self.health_profile = None
Garry Wang5e5538a2019-04-08 15:36:18 -0700367 self._default_power_method = None
Otabek Kasimov39637412020-11-23 19:09:27 -0800368 dut_health_profile = device_health_profile.DeviceHealthProfile(
369 hostname=self.hostname,
370 host_info=self.host_info_store.get(),
371 result_dir=self.get_result_dir())
Otabek Kasimovb61729d2020-11-24 00:42:43 -0800372
373 # TODO(otabek@): remove when b/171414073 closed
Andrew Luo4be621d2020-03-21 07:01:13 -0700374 if self.use_icmp:
Derek Beckettc7677812021-02-12 14:41:11 -0800375 pingable_before_servo = self.is_up_fast(count=1)
Andrew Luo4be621d2020-03-21 07:01:13 -0700376 if pingable_before_servo:
377 logging.info('DUT is pingable before init Servo.')
378 else:
379 logging.info('Skipping ping to DUT before init Servo.')
Otabek Kasimov39637412020-11-23 19:09:27 -0800380 _servo_host, servo_state = servo_host.create_servo_host(
381 dut=self,
382 servo_args=servo_args,
383 try_lab_servo=try_lab_servo,
384 try_servo_repair=try_servo_repair,
Otabek Kasimov1b70e8d2020-12-30 13:51:00 -0800385 try_servo_recovery=try_servo_recovery,
Otabek Kasimov39637412020-11-23 19:09:27 -0800386 dut_host_info=self.host_info_store.get(),
387 dut_health_profile=dut_health_profile)
388 if dut_health_profile.is_loaded():
389 logging.info('Device health profile loaded.')
390 # The device profile is located in the servo_host which make it
391 # dependency. If profile is not loaded yet then we do not have it
392 # TODO(otabek@) persist device provide out of servo-host.
393 self.health_profile = dut_health_profile
394 self.set_servo_host(_servo_host, servo_state)
Richard Barnettee519dcd2016-08-15 17:37:17 -0700395
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +0800396 # TODO(waihong): Do the simplication on Chameleon too.
Shijin Abraham783a7dd2020-02-14 15:36:11 -0800397 self._chameleon_host = chameleon_host.create_chameleon_host(
Otabek Kasimova7ba91a2020-03-09 08:31:01 -0700398 dut=self.hostname,
399 chameleon_args=chameleon_args)
Shijin Abraham783a7dd2020-02-14 15:36:11 -0800400 if self._chameleon_host:
401 self.chameleon = self._chameleon_host.create_chameleon_board()
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +0800402 else:
Tom Wai-Hong Tam3d6790d2014-04-14 16:15:47 +0800403 self.chameleon = None
Fang Deng5d518f42013-08-02 14:04:32 -0700404
Shijin Abraham78ce4402020-09-08 22:04:27 -0700405 # Bluetooth peers will be populated by the test if needed
406 self._btpeer_host_list = []
407 self.btpeer_list = []
408 self.btpeer = None
Shijin Abrahamc09587d2020-02-14 20:46:55 -0800409
howardchung83e55272019-08-08 14:08:05 +0800410 # Add pdtester host if pdtester args were added on command line
Wai-Hong Tam16e5edb2019-09-17 16:10:07 -0700411 self._pdtester_host = pdtester_host.create_pdtester_host(
Wai-Hong Tam90b164d2019-10-25 13:15:39 -0700412 pdtester_args, self._servo_host)
howardchung83e55272019-08-08 14:08:05 +0800413
Wai-Hong Tam0ed9fe12019-05-23 11:00:58 -0700414 if self._pdtester_host:
415 self.pdtester_servo = self._pdtester_host.get_servo()
416 logging.info('pdtester_servo: %r', self.pdtester_servo)
417 # Create the pdtester object used to access the ec uart
418 self.pdtester = pdtester.PDTester(self.pdtester_servo,
419 self._pdtester_host.get_servod_server_proxy())
Scottfe06ed82015-11-05 17:15:01 -0800420 else:
Wai-Hong Tam0ed9fe12019-05-23 11:00:58 -0700421 self.pdtester = None
Scottfe06ed82015-11-05 17:15:01 -0800422
Fang Deng5d518f42013-08-02 14:04:32 -0700423
Shijin Abraham78ce4402020-09-08 22:04:27 -0700424 def initialize_btpeer(self, btpeer_args=[]):
Shijin Abrahamc09587d2020-02-14 20:46:55 -0800425 """ Initialize the Bluetooth peers
426
427 Initialize Bluetooth peer devices given in the arguments. Bluetooth peer
428 is chameleon host on Raspberry Pi.
429 @param btpeer_args: A dictionary that contains args for creating
430 a ChameleonHost. See chameleon_host for details.
431
432 """
Shijin Abraham78ce4402020-09-08 22:04:27 -0700433 logging.debug('Attempting to initialize bluetooth peers if available')
Shijin Abraham22f24cb2020-03-26 09:07:41 -0700434 try:
Shijin Abraham22f24cb2020-03-26 09:07:41 -0700435 if type(btpeer_args) is list:
436 btpeer_args_list = btpeer_args
437 else:
438 btpeer_args_list = [btpeer_args]
439
440 self._btpeer_host_list = chameleon_host.create_btpeer_host(
441 dut=self.hostname, btpeer_args_list=btpeer_args_list)
442 logging.debug('Bluetooth peer hosts are %s',
443 self._btpeer_host_list)
444 self.btpeer_list = [_host.create_chameleon_board() for _host in
445 self._btpeer_host_list if _host is not None]
446
447 if len(self.btpeer_list) > 0:
448 self.btpeer = self.btpeer_list[0]
449
450 logging.debug('After initialize_btpeer btpeer_list %s '
451 'btpeer_host_list is %s and btpeer is %s',
452 self.btpeer_list, self._btpeer_host_list,
453 self.btpeer)
454 except Exception as e:
455 logging.error('Exception %s in initialize_btpeer', str(e))
456
Shijin Abrahamc09587d2020-02-14 20:46:55 -0800457
Richard Barnetteb4b4d6f2018-05-05 01:47:41 +0000458 def get_cros_repair_image_name(self):
Ruben Rodriguez Buchilloncee72f72019-05-01 13:20:58 -0700459 """Get latest stable cros image name from AFE.
460
461 Use the board name from the info store. Should that fail, try to
462 retrieve the board name from the host's installed image itself.
463
464 @returns: current stable cros image name for this host.
465 """
Garry Wange8a8fc22020-04-13 15:04:53 -0700466 info = self.host_info_store.get()
467 if not info.board:
Ruben Rodriguez Buchilloncee72f72019-05-01 13:20:58 -0700468 logging.warn('No board label value found. Trying to infer '
469 'from the host itself.')
470 try:
Garry Wange8a8fc22020-04-13 15:04:53 -0700471 info.labels.append(self.get_board())
Ruben Rodriguez Buchilloncee72f72019-05-01 13:20:58 -0700472 except (error.AutoservRunError, error.AutoservSSHTimeout) as e:
473 logging.error('Also failed to get the board name from the DUT '
474 'itself. %s.', str(e))
Garry Wange8a8fc22020-04-13 15:04:53 -0700475 raise error.AutoservError('Cannot determine board of the DUT'
476 ' while getting repair image name.')
477 return afe_utils.get_stable_cros_image_name_v2(info)
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500478
479
Rohit Makasanadf0a3a32017-06-30 13:55:18 -0700480 def host_version_prefix(self, image):
481 """Return version label prefix.
482
483 In case the CrOS provisioning version is something other than the
484 standard CrOS version e.g. CrOS TH version, this function will
485 find the prefix from provision.py.
486
487 @param image: The image name to find its version prefix.
488 @returns: A prefix string for the image type.
489 """
490 return provision.get_version_label_prefix(image)
491
Andrew Luo3332ab22020-04-28 16:42:03 -0700492 def stage_build_to_usb(self, build):
493 """Stage the current ChromeOS image on the USB stick connected to the
494 servo.
495
496 @param build: The build to download and send to USB.
497 """
498 if not self.servo:
499 raise error.TestError('Host %s does not have servo.' %
500 self.hostname)
501
502 _, update_url = self.stage_image_for_servo(build)
Andrew Luob0355ea2020-06-24 16:12:57 -0700503
504 try:
505 self.servo.image_to_servo_usb(update_url)
506 finally:
507 # servo.image_to_servo_usb turned the DUT off, so turn it back on
508 logging.debug('Turn DUT power back on.')
509 self.servo.get_power_state_controller().power_on()
510
Andrew Luo3332ab22020-04-28 16:42:03 -0700511 logging.debug('ChromeOS image %s is staged on the USB stick.',
512 build)
Rohit Makasanadf0a3a32017-06-30 13:55:18 -0700513
beepsdae65fd2013-07-26 16:24:41 -0700514 def verify_job_repo_url(self, tag=''):
beepscb6f1e22013-06-28 19:14:10 -0700515 """
516 Make sure job_repo_url of this host is valid.
517
joychen03eaad92013-06-26 09:55:21 -0700518 Eg: The job_repo_url "http://lmn.cd.ab.xyx:8080/static/\
beepscb6f1e22013-06-28 19:14:10 -0700519 lumpy-release/R29-4279.0.0/autotest/packages" claims to have the
520 autotest package for lumpy-release/R29-4279.0.0. If this isn't the case,
521 download and extract it. If the devserver embedded in the url is
522 unresponsive, update the job_repo_url of the host after staging it on
523 another devserver.
524
525 @param job_repo_url: A url pointing to the devserver where the autotest
526 package for this build should be staged.
beepsdae65fd2013-07-26 16:24:41 -0700527 @param tag: The tag from the server job, in the format
528 <job_id>-<user>/<hostname>, or <hostless> for a server job.
beepscb6f1e22013-06-28 19:14:10 -0700529
530 @raises DevServerException: If we could not resolve a devserver.
531 @raises AutoservError: If we're unable to save the new job_repo_url as
532 a result of choosing a new devserver because the old one failed to
533 respond to a health check.
beeps0c865032013-07-30 11:37:06 -0700534 @raises urllib2.URLError: If the devserver embedded in job_repo_url
535 doesn't respond within the timeout.
beepscb6f1e22013-06-28 19:14:10 -0700536 """
Prathmesh Prabhu368abdf2017-02-14 11:23:47 -0800537 info = self.host_info_store.get()
538 job_repo_url = info.attributes.get(ds_constants.JOB_REPO_URL, '')
beepscb6f1e22013-06-28 19:14:10 -0700539 if not job_repo_url:
540 logging.warning('No job repo url set on host %s', self.hostname)
541 return
542
543 logging.info('Verifying job repo url %s', job_repo_url)
544 devserver_url, image_name = tools.get_devserver_build_from_package_url(
545 job_repo_url)
546
beeps0c865032013-07-30 11:37:06 -0700547 ds = dev_server.ImageServer(devserver_url)
beepscb6f1e22013-06-28 19:14:10 -0700548
549 logging.info('Staging autotest artifacts for %s on devserver %s',
550 image_name, ds.url())
beeps687243d2013-07-18 15:29:27 -0700551
552 start_time = time.time()
Simran Basi25e7a922014-10-31 11:56:10 -0700553 ds.stage_artifacts(image_name, ['autotest_packages'])
beeps687243d2013-07-18 15:29:27 -0700554 stage_time = time.time() - start_time
555
556 # Record how much of the verification time comes from a devserver
557 # restage. If we're doing things right we should not see multiple
558 # devservers for a given board/build/branch path.
559 try:
J. Richard Barnette3cbd76b2013-11-27 12:11:25 -0800560 board, build_type, branch = server_utils.ParseBuildName(
beeps687243d2013-07-18 15:29:27 -0700561 image_name)[:3]
J. Richard Barnette3cbd76b2013-11-27 12:11:25 -0800562 except server_utils.ParseBuildNameException:
beeps687243d2013-07-18 15:29:27 -0700563 pass
564 else:
beeps0c865032013-07-30 11:37:06 -0700565 devserver = devserver_url[
Chris Sosa65425082013-10-16 13:26:22 -0700566 devserver_url.find('/') + 2:devserver_url.rfind(':')]
beeps687243d2013-07-18 15:29:27 -0700567 stats_key = {
568 'board': board,
569 'build_type': build_type,
570 'branch': branch,
beeps0c865032013-07-30 11:37:06 -0700571 'devserver': devserver.replace('.', '_'),
beeps687243d2013-07-18 15:29:27 -0700572 }
Dan Shi5e2efb72017-02-07 11:40:23 -0800573
574 monarch_fields = {
575 'board': board,
576 'build_type': build_type,
Dan Shi5e2efb72017-02-07 11:40:23 -0800577 'branch': branch,
578 'dev_server': devserver,
579 }
580 metrics.Counter(
581 'chromeos/autotest/provision/verify_url'
582 ).increment(fields=monarch_fields)
583 metrics.SecondsDistribution(
584 'chromeos/autotest/provision/verify_url_duration'
585 ).add(stage_time, fields=monarch_fields)
586
587
Dan Shicf4d2032015-03-12 15:04:21 -0700588 def stage_server_side_package(self, image=None):
589 """Stage autotest server-side package on devserver.
590
591 @param image: Full path of an OS image to install or a build name.
592
593 @return: A url to the autotest server-side package.
Dan Shi14de7622016-08-22 11:09:06 -0700594
595 @raise: error.AutoservError if fail to locate the build to test with, or
596 fail to stage server-side package.
Dan Shicf4d2032015-03-12 15:04:21 -0700597 """
Dan Shid37736b2016-07-06 15:10:29 -0700598 # If enable_drone_in_restricted_subnet is False, do not set hostname
599 # in devserver.resolve call, so a devserver in non-restricted subnet
600 # is picked to stage autotest server package for drone to download.
601 hostname = self.hostname
602 if not server_utils.ENABLE_DRONE_IN_RESTRICTED_SUBNET:
603 hostname = None
Dan Shicf4d2032015-03-12 15:04:21 -0700604 if image:
605 image_name = tools.get_build_from_image(image)
606 if not image_name:
607 raise error.AutoservError(
608 'Failed to parse build name from %s' % image)
Dan Shid37736b2016-07-06 15:10:29 -0700609 ds = dev_server.ImageServer.resolve(image_name, hostname)
Dan Shicf4d2032015-03-12 15:04:21 -0700610 else:
Prathmesh Prabhu9235e4c2017-03-28 13:16:06 -0700611 info = self.host_info_store.get()
Prathmesh Prabhu368abdf2017-02-14 11:23:47 -0800612 job_repo_url = info.attributes.get(ds_constants.JOB_REPO_URL, '')
Dan Shicf4d2032015-03-12 15:04:21 -0700613 if job_repo_url:
614 devserver_url, image_name = (
615 tools.get_devserver_build_from_package_url(job_repo_url))
Dan Shid37736b2016-07-06 15:10:29 -0700616 # If enable_drone_in_restricted_subnet is True, use the
617 # existing devserver. Otherwise, resolve a new one in
618 # non-restricted subnet.
619 if server_utils.ENABLE_DRONE_IN_RESTRICTED_SUBNET:
620 ds = dev_server.ImageServer(devserver_url)
621 else:
622 ds = dev_server.ImageServer.resolve(image_name)
Prathmesh Prabhub6cea612017-02-09 15:41:19 -0800623 elif info.build is not None:
624 ds = dev_server.ImageServer.resolve(info.build, hostname)
Prathmesh Prabhu0c1dd4d2017-06-07 13:01:53 -0700625 image_name = info.build
Dan Shicf4d2032015-03-12 15:04:21 -0700626 else:
Prathmesh Prabhub6cea612017-02-09 15:41:19 -0800627 raise error.AutoservError(
628 'Failed to stage server-side package. The host has '
Garry Wang12b9baf2019-06-24 18:58:54 -0700629 'no job_repo_url attribute or cros-version label.')
Dan Shica503482015-03-30 17:23:25 -0700630
631 # Get the OS version of the build, for any build older than
632 # MIN_VERSION_SUPPORT_SSP, server side packaging is not supported.
633 match = re.match('.*/R\d+-(\d+)\.', image_name)
634 if match and int(match.group(1)) < self.MIN_VERSION_SUPPORT_SSP:
Dan Shi14de7622016-08-22 11:09:06 -0700635 raise error.AutoservError(
636 'Build %s is older than %s. Server side packaging is '
637 'disabled.' % (image_name, self.MIN_VERSION_SUPPORT_SSP))
Dan Shica503482015-03-30 17:23:25 -0700638
Dan Shicf4d2032015-03-12 15:04:21 -0700639 ds.stage_artifacts(image_name, ['autotest_server_package'])
640 return '%s/static/%s/%s' % (ds.url(), image_name,
641 'autotest_server_package.tar.bz2')
642
643
Kalin Stoyanovb3c11f32018-05-11 09:02:00 -0700644 def stage_image_for_servo(self, image_name=None, artifact='test_image'):
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700645 """Stage a build on a devserver and return the update_url.
646
647 @param image_name: a name like lumpy-release/R27-3837.0.0
Kalin Stoyanovb3c11f32018-05-11 09:02:00 -0700648 @param artifact: a string like 'test_image'. Requests
649 appropriate image to be staged.
Xixuan Wufee57542019-10-15 11:50:27 -0700650 @returns a tuple of (image_name, URL) like
651 (lumpy-release/R27-3837.0.0,
652 http://172.22.50.205:8082/update/lumpy-release/R27-3837.0.0)
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700653 """
654 if not image_name:
Richard Barnetteb4b4d6f2018-05-05 01:47:41 +0000655 image_name = self.get_cros_repair_image_name()
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700656 logging.info('Staging build for servo install: %s', image_name)
Dan Shi216389c2015-12-22 11:03:06 -0800657 devserver = dev_server.ImageServer.resolve(image_name, self.hostname)
Kalin Stoyanovb3c11f32018-05-11 09:02:00 -0700658 devserver.stage_artifacts(image_name, [artifact])
659 if artifact == 'test_image':
Xixuan Wufee57542019-10-15 11:50:27 -0700660 return image_name, devserver.get_test_image_url(image_name)
Kalin Stoyanovb3c11f32018-05-11 09:02:00 -0700661 elif artifact == 'recovery_image':
Xixuan Wufee57542019-10-15 11:50:27 -0700662 return image_name, devserver.get_recovery_image_url(image_name)
Kalin Stoyanovb3c11f32018-05-11 09:02:00 -0700663 else:
664 raise error.AutoservError("Bad artifact!")
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700665
666
beepse539be02013-07-31 21:57:39 -0700667 def stage_factory_image_for_servo(self, image_name):
668 """Stage a build on a devserver and return the update_url.
669
670 @param image_name: a name like <baord>/4262.204.0
beeps12c0a3c2013-09-03 11:58:27 -0700671
beepse539be02013-07-31 21:57:39 -0700672 @return: An update URL, eg:
673 http://<devserver>/static/canary-channel/\
674 <board>/4262.204.0/factory_test/chromiumos_factory_image.bin
beeps12c0a3c2013-09-03 11:58:27 -0700675
676 @raises: ValueError if the factory artifact name is missing from
677 the config.
678
beepse539be02013-07-31 21:57:39 -0700679 """
680 if not image_name:
681 logging.error('Need an image_name to stage a factory image.')
682 return
683
Dan Shib8540a52015-07-16 14:18:23 -0700684 factory_artifact = CONFIG.get_config_value(
beeps12c0a3c2013-09-03 11:58:27 -0700685 'CROS', 'factory_artifact', type=str, default='')
686 if not factory_artifact:
687 raise ValueError('Cannot retrieve the factory artifact name from '
688 'autotest config, and hence cannot stage factory '
689 'artifacts.')
690
beepse539be02013-07-31 21:57:39 -0700691 logging.info('Staging build for servo install: %s', image_name)
Dan Shi216389c2015-12-22 11:03:06 -0800692 devserver = dev_server.ImageServer.resolve(image_name, self.hostname)
beepse539be02013-07-31 21:57:39 -0700693 devserver.stage_artifacts(
694 image_name,
beeps12c0a3c2013-09-03 11:58:27 -0700695 [factory_artifact],
696 archive_url=None)
beepse539be02013-07-31 21:57:39 -0700697
698 return tools.factory_image_url_pattern() % (devserver.url(), image_name)
699
700
Laurence Goodby778c9a42017-05-24 19:24:07 -0700701 def prepare_for_update(self):
702 """Prepares the DUT for an update.
703
704 Subclasses may override this to perform any special actions
705 required before updating.
706 """
Laurence Goodby468de252017-06-08 17:22:53 -0700707 pass
Laurence Goodby778c9a42017-05-24 19:24:07 -0700708
709
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +0800710 def _clear_fw_version_labels(self, rw_only):
711 """Clear firmware version labels from the machine.
712
713 @param rw_only: True to only clear fwrw_version; otherewise, clear
714 both fwro_version and fwrw_version.
715 """
Prathmesh Prabhuf8ae9a82019-10-04 15:34:13 -0700716 info = self.host_info_store.get()
717 info.clear_version_labels(provision.FW_RW_VERSION_PREFIX)
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +0800718 if not rw_only:
Prathmesh Prabhuf8ae9a82019-10-04 15:34:13 -0700719 info.clear_version_labels(provision.FW_RO_VERSION_PREFIX)
720 self.host_info_store.commit(info)
Dan Shi9cb0eec2014-06-03 09:04:50 -0700721
722
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +0800723 def _add_fw_version_label(self, build, rw_only):
Dan Shi9cb0eec2014-06-03 09:04:50 -0700724 """Add firmware version label to the machine.
725
726 @param build: Build of firmware.
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +0800727 @param rw_only: True to only add fwrw_version; otherwise, add both
728 fwro_version and fwrw_version.
Dan Shi9cb0eec2014-06-03 09:04:50 -0700729
730 """
Prathmesh Prabhuf8ae9a82019-10-04 15:34:13 -0700731 info = self.host_info_store.get()
732 info.set_version_label(provision.FW_RW_VERSION_PREFIX, build)
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +0800733 if not rw_only:
Prathmesh Prabhuf8ae9a82019-10-04 15:34:13 -0700734 info.set_version_label(provision.FW_RO_VERSION_PREFIX, build)
735 self.host_info_store.commit(info)
Dan Shi9cb0eec2014-06-03 09:04:50 -0700736
737
Namyoon Woo33f38852020-04-13 17:26:58 -0700738 def get_latest_release_version(self, platform, ref_board=None):
Namyoon Woo5f894662019-11-15 15:23:23 -0800739 """Search for the latest package release version from the image archive,
740 and return it.
741
Namyoon Woo33f38852020-04-13 17:26:58 -0700742 @param platform: platform name, a.k.a. board or model
743 @param ref_board: reference board name, a.k.a. baseboard, parent
Namyoon Woo5f894662019-11-15 15:23:23 -0800744
Namyoon Woo33f38852020-04-13 17:26:58 -0700745 @return 'firmware-{platform}-{branch}-firmwarebranch/{release-version}/'
746 '{platform}'
Namyoon Woo5f894662019-11-15 15:23:23 -0800747 or None if LATEST release file does not exist.
748 """
749
Namyoon Woo33f38852020-04-13 17:26:58 -0700750 platforms = [ platform ]
Namyoon Woo5f894662019-11-15 15:23:23 -0800751
Namyoon Woo33f38852020-04-13 17:26:58 -0700752 # Search the image path in reference board archive as well.
753 # For example, bob has its binary image under its reference board (gru)
754 # image archive.
755 if ref_board:
756 platforms.append(ref_board)
Namyoon Woo5f894662019-11-15 15:23:23 -0800757
Namyoon Woo33f38852020-04-13 17:26:58 -0700758 for board in platforms:
759 # Read 'LATEST-1.0.0' file
760 branch_dir = provision.FW_BRANCH_GLOB % board
761 latest_file = os.path.join(provision.CROS_IMAGE_ARCHIVE, branch_dir,
762 'LATEST-1.0.0')
Namyoon Woo406c7d42020-01-24 15:57:11 -0800763
Namyoon Woo33f38852020-04-13 17:26:58 -0700764 try:
765 # The result could be one or more.
766 result = utils.system_output('gsutil ls -d ' + latest_file)
767
768 candidates = re.findall('gs://.*', result)
769
770 # Found the directory candidates. No need to check the other
771 # board name cadidates. Let's break the loop.
772 break
773 except error.CmdError:
774 # It doesn't exist. Let's move on to the next item.
775 pass
776 else:
Namyoon Woo5f894662019-11-15 15:23:23 -0800777 logging.error('No LATEST release info is available.')
778 return None
779
Namyoon Woo406c7d42020-01-24 15:57:11 -0800780 for cand_dir in candidates:
781 result = utils.system_output('gsutil cat ' + cand_dir)
Namyoon Woo5f894662019-11-15 15:23:23 -0800782
Namyoon Woo406c7d42020-01-24 15:57:11 -0800783 release_path = cand_dir.replace('LATEST-1.0.0', result)
Namyoon Woo33f38852020-04-13 17:26:58 -0700784 release_path = os.path.join(release_path, platform)
Namyoon Woo406c7d42020-01-24 15:57:11 -0800785 try:
786 # Check if release_path does exist.
787 release = utils.system_output('gsutil ls -d ' + release_path)
788 # Now 'release' has a full directory path: e.g.
789 # gs://chromeos-image-archive/firmware-octopus-11297.B-
790 # firmwarebranch/RNone-1.0.0-b4395530/octopus/
791
792 # Remove "gs://chromeos-image-archive".
793 release = release.replace(provision.CROS_IMAGE_ARCHIVE, '')
794
795 # Remove CROS_IMAGE_ARCHIVE and any surrounding '/'s.
796 return release.strip('/')
797 except error.CmdError:
798 # The directory might not exist. Let's try next candidate.
799 pass
800 else:
801 raise error.AutoservError('Cannot find the latest firmware')
Namyoon Woo5f894662019-11-15 15:23:23 -0800802
Brent Peterson1cb623a2020-01-09 13:14:28 -0800803 @staticmethod
804 def get_version_from_image(image, version_regex):
Brent Peterson8039b472020-02-14 10:51:23 -0800805 """Get version string from binary image using regular expression.
806
807 @param image: Binary image to search
808 @param version_regex: Regular expression to search for
809
810 @return Version string
811
812 @raises TestFail if no version string is found in image
813 """
Brent Peterson1cb623a2020-01-09 13:14:28 -0800814 with open(image, 'rb') as f:
815 image_data = f.read()
Derek Beckett98345552020-08-31 16:07:22 -0700816 match = re.findall(version_regex,
817 image_data.decode('ISO-8859-1', errors='ignore'))
Brent Peterson1cb623a2020-01-09 13:14:28 -0800818 if match:
819 return match[0]
820 else:
821 raise error.TestFail('Failed to read version from %s.' % image)
822
823
Garry Wangad2a1712020-03-26 15:06:43 -0700824 def firmware_install(self, build, rw_only=False, dest=None,
Brent Petersonc70a1832020-01-24 15:54:35 -0800825 local_tarball=None, verify_version=False,
Namyoon Woo382e5892020-05-20 16:48:40 -0700826 try_scp=False, install_ec=True, install_bios=True,
827 board_as=None):
Dan Shi9cb0eec2014-06-03 09:04:50 -0700828 """Install firmware to the DUT.
829
830 Use stateful update if the DUT is already running the same build.
831 Stateful update does not update kernel and tends to run much faster
832 than a full reimage. If the DUT is running a different build, or it
833 failed to do a stateful update, full update, including kernel update,
834 will be applied to the DUT.
835
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +0800836 Once a host enters firmware_install its fw[ro|rw]_version label will
837 be removed. After the firmware is updated successfully, a new
838 fw[ro|rw]_version label will be added to the host.
Dan Shi9cb0eec2014-06-03 09:04:50 -0700839
840 @param build: The build version to which we want to provision the
841 firmware of the machine,
842 e.g. 'link-firmware/R22-2695.1.144'.
Tom Wai-Hong Tam4ac78982016-01-08 02:34:37 +0800843 @param rw_only: True to only install firmware to its RW portions. Keep
844 the RO portions unchanged.
Mary Ruthven6481a9f2019-08-23 12:46:05 -0700845 @param dest: Directory to store the firmware in.
Brent Peterson5b7c5b12020-01-09 13:14:28 -0800846 @param local_tarball: Path to local firmware image for installing
847 without devserver.
Brent Peterson1cb623a2020-01-09 13:14:28 -0800848 @param verify_version: True to verify EC and BIOS versions after
849 programming firmware, default is False.
Brent Petersonc70a1832020-01-24 15:54:35 -0800850 @param try_scp: False to always program using servo, true to try copying
851 the firmware and programming from the DUT.
Namyoon Woo382e5892020-05-20 16:48:40 -0700852 @param install_ec: True to install EC FW, and False to skip it.
853 @param install_bios: True to install BIOS, and False to skip it.
854 @param board_as: A board name to force to use.
Dan Shi9cb0eec2014-06-03 09:04:50 -0700855
856 TODO(dshi): After bug 381718 is fixed, update here with corresponding
857 exceptions that could be raised.
858
859 """
860 if not self.servo:
861 raise error.TestError('Host %s does not have servo.' %
862 self.hostname)
863
Wai-Hong Tam3fa455a2018-07-18 14:40:43 -0700864 # Get the DUT board name from AFE.
865 info = self.host_info_store.get()
866 board = info.board
Shelley Chenac61d5a2019-06-24 15:35:46 -0700867 model = info.model
Namyoon Woo8dbfcf92019-01-15 18:37:12 -0800868
869 if board is None or board == '':
870 board = self.servo.get_board()
Dan Shi9cb0eec2014-06-03 09:04:50 -0700871
Namyoon Woo382e5892020-05-20 16:48:40 -0700872 # if board_as argument is passed, then use it instead of the original
873 # board name.
874 if board_as:
875 board = board_as
876
Mary Ruthvende14a8b2019-08-23 12:43:52 -0700877 if model is None or model == '':
Namyoon Woofb16eae2020-08-14 10:02:39 -0700878 try:
Mary Ruthven9c15a4e2020-10-23 10:10:53 -0700879 model = self.get_platform()
Namyoon Woofb16eae2020-08-14 10:02:39 -0700880 except Exception as e:
881 logging.warn('Dut is unresponsive: %s', str(e))
Mary Ruthvende14a8b2019-08-23 12:43:52 -0700882
Brent Peterson5b7c5b12020-01-09 13:14:28 -0800883 # If local firmware path not provided fetch it from the dev server
Mary Ruthven6481a9f2019-08-23 12:46:05 -0700884 tmpd = None
Brent Peterson5b7c5b12020-01-09 13:14:28 -0800885 if not local_tarball:
Garry Wangad2a1712020-03-26 15:06:43 -0700886 logging.info('Will install firmware from build %s.', build)
Brent Peterson5b7c5b12020-01-09 13:14:28 -0800887
Namyoon Woo43c1cb22020-05-28 18:58:34 -0700888 try:
889 ds = dev_server.ImageServer.resolve(build, self.hostname)
890 ds.stage_artifacts(build, ['firmware'])
Brent Peterson5b7c5b12020-01-09 13:14:28 -0800891
Namyoon Woo43c1cb22020-05-28 18:58:34 -0700892 if not dest:
893 tmpd = autotemp.tempdir(unique_id='fwimage')
894 dest = tmpd.name
Brent Peterson5b7c5b12020-01-09 13:14:28 -0800895
Namyoon Woo43c1cb22020-05-28 18:58:34 -0700896 # Download firmware image
897 fwurl = self._FW_IMAGE_URL_PATTERN % (ds.url(), build)
898 local_tarball = os.path.join(dest, os.path.basename(fwurl))
899 ds.download_file(fwurl, local_tarball)
900 except Exception as e:
901 raise error.TestError('Failed to download firmware package: %s'
902 % str(e))
Dan Shi9cb0eec2014-06-03 09:04:50 -0700903
Namyoon Woo382e5892020-05-20 16:48:40 -0700904 ec_image = None
905 if install_ec:
906 # Extract EC image from tarball
907 logging.info('Extracting EC image.')
908 ec_image = self.servo.extract_ec_image(board, model, local_tarball)
Mary Ruthven9c15a4e2020-10-23 10:10:53 -0700909 logging.info('Extracted: %s', ec_image)
Brent Peterson1cb623a2020-01-09 13:14:28 -0800910
Namyoon Woo382e5892020-05-20 16:48:40 -0700911 bios_image = None
912 if install_bios:
913 # Extract BIOS image from tarball
914 logging.info('Extracting BIOS image.')
915 bios_image = self.servo.extract_bios_image(board, model,
916 local_tarball)
Mary Ruthven9c15a4e2020-10-23 10:10:53 -0700917 logging.info('Extracted: %s', bios_image)
Namyoon Woo382e5892020-05-20 16:48:40 -0700918
919 if not bios_image and not ec_image:
920 raise error.TestError('No firmware installation was processed.')
Brent Peterson1cb623a2020-01-09 13:14:28 -0800921
Brent Petersonc70a1832020-01-24 15:54:35 -0800922 # Clear firmware version labels
923 self._clear_fw_version_labels(rw_only)
924
925 # Install firmware from local tarball
Brent Peterson5b7c5b12020-01-09 13:14:28 -0800926 try:
Garry Wang50e4a492020-08-05 12:29:57 -0700927 # Check if copying to DUT is enabled and DUT is available
928 if try_scp and self.is_up():
Brent Petersonc70a1832020-01-24 15:54:35 -0800929 # DUT is available, make temp firmware directory to store images
930 logging.info('Making temp folder.')
931 dest_folder = '/tmp/firmware'
932 self.run('mkdir -p ' + dest_folder)
933
Namyoon Woo68b68082020-06-02 13:13:14 -0700934 fw_cmd = self._FW_UPDATE_CMD % ('--wp=1' if rw_only else '')
Brent Petersonc70a1832020-01-24 15:54:35 -0800935
Namyoon Woo382e5892020-05-20 16:48:40 -0700936 if bios_image:
937 # Send BIOS firmware image to DUT
938 logging.info('Sending BIOS firmware.')
939 dest_bios_path = os.path.join(dest_folder,
940 os.path.basename(bios_image))
941 self.send_file(bios_image, dest_bios_path)
942
943 # Initialize firmware update command for BIOS image
944 fw_cmd += ' -i %s' % dest_bios_path
Brent Peterson669edf42020-02-07 15:07:54 -0800945
946 # Send EC firmware image to DUT when EC image was found
947 if ec_image:
948 logging.info('Sending EC firmware.')
949 dest_ec_path = os.path.join(dest_folder,
950 os.path.basename(ec_image))
951 self.send_file(ec_image, dest_ec_path)
952
953 # Add EC image to firmware update command
954 fw_cmd += ' -e %s' % dest_ec_path
955
Dana Goyettee84ef8d2020-07-09 11:55:09 -0700956 # Make sure command is allowed to finish even if ssh fails.
957 fw_cmd = "trap '' SIGHUP; %s" % fw_cmd
958
Brent Peterson669edf42020-02-07 15:07:54 -0800959 # Update firmware on DUT
960 logging.info('Updating firmware.')
Dana Goyettee84ef8d2020-07-09 11:55:09 -0700961 try:
Dana Goyette935b3fe2020-07-23 14:19:39 -0700962 self.run(fw_cmd, options="-o LogLevel=verbose")
Dana Goyettee84ef8d2020-07-09 11:55:09 -0700963 except error.AutoservRunError as e:
964 if e.result_obj.exit_status != 255:
965 raise
966 elif ec_image:
967 logging.warn("DUT network dropped during update"
968 " (often caused by EC resetting USB)")
969 else:
970 logging.error("DUT network dropped during update"
971 " (unexpected, since no EC image)")
972 raise
Brent Petersonc70a1832020-01-24 15:54:35 -0800973 else:
974 # Host is not available, program firmware using servo
Brent Peterson669edf42020-02-07 15:07:54 -0800975 if ec_image:
976 self.servo.program_ec(ec_image, rw_only)
Namyoon Woo382e5892020-05-20 16:48:40 -0700977 if bios_image:
978 self.servo.program_bios(bios_image, rw_only)
Brent Petersonc70a1832020-01-24 15:54:35 -0800979 if utils.host_is_in_lab_zone(self.hostname):
980 self._add_fw_version_label(build, rw_only)
Brent Peterson1cb623a2020-01-09 13:14:28 -0800981
982 # Reboot and wait for DUT after installing firmware
983 logging.info('Rebooting DUT.')
984 self.servo.get_power_state_controller().reset()
985 time.sleep(self.servo.BOOT_DELAY)
986 self.test_wait_for_boot()
987
988 # When enabled verify EC and BIOS firmware version after programming
989 if verify_version:
Brent Peterson669edf42020-02-07 15:07:54 -0800990 # Check programmed EC firmware when EC image was found
991 if ec_image:
992 logging.info('Checking EC firmware version.')
993 dest_ec_version = self.get_ec_version()
Brent Peterson8039b472020-02-14 10:51:23 -0800994 ec_version_prefix = dest_ec_version.split('_', 1)[0]
995 ec_regex = self._EC_REGEX % ec_version_prefix
Brent Peterson669edf42020-02-07 15:07:54 -0800996 image_ec_version = self.get_version_from_image(ec_image,
Brent Peterson8039b472020-02-14 10:51:23 -0800997 ec_regex)
Brent Peterson669edf42020-02-07 15:07:54 -0800998 if dest_ec_version != image_ec_version:
999 raise error.TestFail(
Namyoon Wooe2c009a2020-07-22 10:09:03 -07001000 'Failed to update EC firmware, version %s '
1001 '(expected %s)' % (dest_ec_version,
1002 image_ec_version))
Brent Peterson1cb623a2020-01-09 13:14:28 -08001003
Namyoon Woo382e5892020-05-20 16:48:40 -07001004 if bios_image:
1005 # Check programmed BIOS firmware against expected version
1006 logging.info('Checking BIOS firmware version.')
1007 dest_bios_version = self.get_firmware_version()
1008 bios_version_prefix = dest_bios_version.split('.', 1)[0]
1009 bios_regex = self._BIOS_REGEX % bios_version_prefix
1010 image_bios_version = self.get_version_from_image(bios_image,
1011 bios_regex)
1012 if dest_bios_version != image_bios_version:
1013 raise error.TestFail(
Namyoon Wooe2c009a2020-07-22 10:09:03 -07001014 'Failed to update BIOS, version %s '
Namyoon Woo382e5892020-05-20 16:48:40 -07001015 '(expected %s)' % (dest_bios_version,
1016 image_bios_version))
Dan Shi9cb0eec2014-06-03 09:04:50 -07001017 finally:
Mary Ruthven6481a9f2019-08-23 12:46:05 -07001018 if tmpd:
1019 tmpd.clean()
Dan Shi9cb0eec2014-06-03 09:04:50 -07001020
1021
Garry Wang790953f2020-10-29 21:11:57 -07001022 def servo_install(self,
1023 image_url=None,
1024 usb_boot_timeout=USB_BOOT_TIMEOUT,
1025 install_timeout=INSTALL_TIMEOUT,
1026 is_repair=False):
Scott Zawalski62bacae2013-03-05 10:40:32 -05001027 """
1028 Re-install the OS on the DUT by:
1029 1) installing a test image on a USB storage device attached to the Servo
1030 board,
Richard Barnette03a0c132012-11-05 12:40:35 -08001031 2) booting that image in recovery mode, and then
J. Richard Barnette31b2e312013-04-04 16:05:22 -07001032 3) installing the image with chromeos-install.
1033
Scott Zawalski62bacae2013-03-05 10:40:32 -05001034 @param image_url: If specified use as the url to install on the DUT.
1035 otherwise boot the currently staged image on the USB stick.
beepsf079cfb2013-09-18 17:49:51 -07001036 @param usb_boot_timeout: The usb_boot_timeout to use during reimage.
1037 Factory images need a longer usb_boot_timeout than regular
1038 cros images.
1039 @param install_timeout: The timeout to use when installing the chromeos
1040 image. Factory images need a longer install_timeout.
Garry Wang790953f2020-10-29 21:11:57 -07001041 @param is_repair: Indicates if the method is called from a repair task.
Richard Barnette03a0c132012-11-05 12:40:35 -08001042
Scott Zawalski62bacae2013-03-05 10:40:32 -05001043 @raises AutoservError if the image fails to boot.
beepsf079cfb2013-09-18 17:49:51 -07001044
J. Richard Barnette0199cc82014-12-05 17:08:40 -08001045 """
Garry Wang7b0e1b72020-03-25 19:08:59 -07001046 if image_url:
1047 logging.info('Downloading image to USB, then booting from it.'
1048 ' Usb boot timeout = %s', usb_boot_timeout)
1049 else:
1050 logging.info('Booting from USB directly. Usb boot timeout = %s',
1051 usb_boot_timeout)
1052
1053 metrics_field = {'download': bool(image_url)}
1054 metrics.Counter(
1055 'chromeos/autotest/provision/servo_install/download_image'
1056 ).increment(fields=metrics_field)
1057
Allen Li48a13fe2016-11-22 14:10:40 -08001058 with metrics.SecondsTimer(
1059 'chromeos/autotest/provision/servo_install/boot_duration'):
Otabek Kasimovb887a002020-12-29 02:54:38 -08001060 self.servo.get_power_state_controller().power_off()
Otabek Kasimovb7cb8422020-12-23 02:38:32 -08001061 try:
1062 self.servo.image_to_servo_usb(image_path=image_url,
1063 power_off_dut=False)
1064 except error.AutotestError as e:
1065 metrics.Counter('chromeos/autotest/repair/image_to_usb_error'
1066 ).increment(
1067 fields={'host': self.hostname or ''})
1068 six.reraise(error.AutotestError, str(e), sys.exc_info()[2])
1069 # Give the DUT some time to power_off if we skip
1070 # download image to usb. (crbug.com/982993)
1071 if not image_url:
1072 time.sleep(10)
Garry Wang53fc8f32020-09-18 13:30:08 -07001073 need_snk = self.require_snk_mode_in_recovery()
Otabek Kasimovb7cb8422020-12-23 02:38:32 -08001074 self.servo.boot_in_recovery_mode(snk_mode=need_snk)
Allen Li48a13fe2016-11-22 14:10:40 -08001075 if not self.wait_up(timeout=usb_boot_timeout):
Garry Wang53fc8f32020-09-18 13:30:08 -07001076 if need_snk:
1077 # Attempt to restore servo_v4 role to 'src' mode.
1078 self.servo.set_servo_v4_role('src')
Allen Li48a13fe2016-11-22 14:10:40 -08001079 raise hosts.AutoservRepairError(
1080 'DUT failed to boot from USB after %d seconds' %
Garry Wang9ced7aa2020-04-10 17:26:35 -07001081 usb_boot_timeout, 'failed_to_boot_pre_install')
Scott Zawalski62bacae2013-03-05 10:40:32 -05001082
Garry Wang2e347df2020-10-30 14:04:26 -07001083 # Make sure the DUT is boot from an external device.
1084 if not self.is_boot_from_external_device():
1085 raise hosts.AutoservRepairError(
1086 'DUT is expected to boot from an external device(e.g. '
1087 'a usb stick), however it seems still boot from an'
1088 ' internal storage.', 'boot_from_internal_storage')
1089
Tom Wai-Hong Tamf6b4f812015-08-08 04:14:59 +08001090 # The new chromeos-tpm-recovery has been merged since R44-7073.0.0.
1091 # In old CrOS images, this command fails. Skip the error.
Tom Wai-Hong Tam27af7332015-07-25 06:09:39 +08001092 logging.info('Resetting the TPM status')
Tom Wai-Hong Tamf6b4f812015-08-08 04:14:59 +08001093 try:
1094 self.run('chromeos-tpm-recovery')
1095 except error.AutoservRunError:
1096 logging.warn('chromeos-tpm-recovery is too old.')
1097
Tom Wai-Hong Tam27af7332015-07-25 06:09:39 +08001098
Allen Li48a13fe2016-11-22 14:10:40 -08001099 with metrics.SecondsTimer(
1100 'chromeos/autotest/provision/servo_install/install_duration'):
1101 logging.info('Installing image through chromeos-install.')
Garry Wang033a31e2020-04-10 17:20:49 -07001102 try:
1103 self.run('chromeos-install --yes',timeout=install_timeout)
1104 self.halt()
Otabek Kasimov808cd832020-05-28 18:27:46 -07001105 except Exception as e:
1106 storage_errors = [
1107 'No space left on device',
1108 'I/O error when trying to write primary GPT',
1109 'Input/output error while writing out',
1110 'cannot read GPT header',
Otabek Kasimov2b7e8302020-08-21 09:23:31 -07001111 'can not determine destination device',
1112 'wrong fs type',
1113 'bad superblock on',
Otabek Kasimov808cd832020-05-28 18:27:46 -07001114 ]
1115 has_error = [msg for msg in storage_errors if(msg in str(e))]
1116 if has_error:
1117 info = self.host_info_store.get()
1118 info.set_version_label(
1119 audit_const.DUT_STORAGE_STATE_PREFIX,
1120 audit_const.HW_STATE_NEED_REPLACEMENT)
1121 self.host_info_store.commit(info)
Otabek Kasimov6825b762020-06-23 23:42:44 -07001122 self.set_device_repair_state(
Otabek Kasimov832d9162020-07-27 19:24:57 -07001123 cros_constants.DEVICE_STATE_NEEDS_REPLACEMENT)
Otabek Kasimov808cd832020-05-28 18:27:46 -07001124 logging.debug(
1125 'Fail install image from USB; Storage error; %s', e)
1126 raise error.AutoservError(
1127 'Failed to install image from USB due to a suspect '
1128 'disk failure, DUT storage state changed to '
1129 'need_replacement, please check debug log '
1130 'for details.')
1131 else:
Garry Wang790953f2020-10-29 21:11:57 -07001132 if is_repair:
1133 # DUT will be marked for replacement if storage is bad.
1134 audit_verify.VerifyDutStorage(self).verify()
Otabek Kasimov27bb2862020-08-10 14:40:45 -07001135
Otabek Kasimov808cd832020-05-28 18:27:46 -07001136 logging.debug('Fail install image from USB; %s', e)
1137 raise error.AutoservError(
1138 'Failed to install image from USB due to unexpected '
1139 'error, please check debug log for details.')
Garry Wang033a31e2020-04-10 17:20:49 -07001140 finally:
1141 # We need reset the DUT no matter re-install success or not,
1142 # as we don't want leave the DUT in boot from usb state.
1143 logging.info('Power cycling DUT through servo.')
1144 self.servo.get_power_state_controller().power_off()
1145 self.servo.switch_usbkey('off')
Garry Wang53fc8f32020-09-18 13:30:08 -07001146 if need_snk:
1147 # Attempt to restore servo_v4 role to 'src' mode.
1148 self.servo.set_servo_v4_role('src')
Garry Wang033a31e2020-04-10 17:20:49 -07001149 # N.B. The Servo API requires that we use power_on() here
1150 # for two reasons:
1151 # 1) After turning on a DUT in recovery mode, you must turn
1152 # it off and then on with power_on() once more to
1153 # disable recovery mode (this is a Parrot specific
1154 # requirement).
1155 # 2) After power_off(), the only way to turn on is with
1156 # power_on() (this is a Storm specific requirement).
1157 self.servo.get_power_state_controller().power_on()
beepsf079cfb2013-09-18 17:49:51 -07001158
1159 logging.info('Waiting for DUT to come back up.')
Richard Barnette03a0c132012-11-05 12:40:35 -08001160 if not self.wait_up(timeout=self.BOOT_TIMEOUT):
Garry Wang9ced7aa2020-04-10 17:26:35 -07001161 raise hosts.AutoservRepairError('DUT failed to reboot installed '
1162 'test image after %d seconds' %
1163 self.BOOT_TIMEOUT,
1164 'failed_to_boot_post_install')
Scott Zawalski62bacae2013-03-05 10:40:32 -05001165
1166
Garry Wanga2e78172020-09-09 23:49:07 -07001167 def set_servo_host(self, host, servo_state=None):
Richard Barnette4aeb01c2018-09-20 09:36:12 -07001168 """Set our servo host member, and associated servo.
1169
1170 @param host Our new `ServoHost`.
1171 """
1172 self._servo_host = host
Derek Beckettb66e5c82020-08-12 15:31:02 -07001173 self.servo_pwr_supported = None
Richard Barnette4aeb01c2018-09-20 09:36:12 -07001174 if self._servo_host is not None:
1175 self.servo = self._servo_host.get_servo()
Otabek Kasimova7ba91a2020-03-09 08:31:01 -07001176 servo_state = self._servo_host.get_servo_state()
Garry Wang000c6c02020-05-11 21:27:23 -07001177 self._set_smart_usbhub_label(self._servo_host.smart_usbhub)
Derek Beckettb66e5c82020-08-12 15:31:02 -07001178 try:
1179 self.servo_pwr_supported = self.servo.has_control('power_state')
1180 except Exception as e:
1181 logging.debug(
1182 "Could not get servo power state due to {}".format(e))
Gregory Nisbet93b23e22020-10-02 20:42:16 +00001183 else:
1184 self.servo = None
Derek Beckettb66e5c82020-08-12 15:31:02 -07001185 self.servo_pwr_supported = False
Otabek Kasimov41301a22020-05-10 15:28:21 -07001186 self.set_servo_type()
Otabek Kasimova7ba91a2020-03-09 08:31:01 -07001187 self.set_servo_state(servo_state)
Otabek Kasimov382c3bb2020-10-28 13:22:45 -07001188 self._set_servo_topology()
Otabek Kasimova7ba91a2020-03-09 08:31:01 -07001189
Richard Barnette4aeb01c2018-09-20 09:36:12 -07001190
Richard Barnette9a26ad62016-06-10 12:03:08 -07001191 def repair_servo(self):
Dan Shi90466352015-09-22 15:01:05 -07001192 """
Richard Barnette9a26ad62016-06-10 12:03:08 -07001193 Confirm that servo is initialized and verified.
Dan Shi90466352015-09-22 15:01:05 -07001194
Richard Barnette9a26ad62016-06-10 12:03:08 -07001195 If the servo object is missing, attempt to repair the servo
1196 host. Repair failures are passed back to the caller.
1197
1198 @raise AutoservError: If there is no servo host for this CrOS
1199 host.
1200 """
1201 if self.servo:
1202 return
1203 if not self._servo_host:
1204 raise error.AutoservError('No servo host for %s.' %
1205 self.hostname)
Otabek Kasimovcc9738e2020-02-14 16:17:15 -08001206 try:
1207 self._servo_host.repair()
1208 except:
1209 raise
1210 finally:
1211 self.set_servo_host(self._servo_host)
1212
1213
Otabek Kasimov41301a22020-05-10 15:28:21 -07001214 def set_servo_type(self):
1215 """Set servo info labels to dut host_info"""
1216 if not self.servo:
Otabek Kasimovbea89e52020-05-12 15:40:00 -07001217 logging.debug('Servo is not initialized to get servo_type.')
Otabek Kasimov41301a22020-05-10 15:28:21 -07001218 return
Otabek Kasimov1b70e8d2020-12-30 13:51:00 -08001219 if not self.is_servo_in_working_state():
1220 logging.debug('Servo is not good, skip update servo_type.')
1221 return
Otabek Kasimov41301a22020-05-10 15:28:21 -07001222 servo_type = self.servo.get_servo_type()
1223 if not servo_type:
Otabek Kasimovbea89e52020-05-12 15:40:00 -07001224 logging.debug('Cannot collect servo_type from servo'
Otabek Kasimov41301a22020-05-10 15:28:21 -07001225 ' by `dut-control servo_type`! Please file a bug'
1226 ' and inform infra team as we are not expected '
1227 ' to reach this point.')
1228 return
1229 host_info = self.host_info_store.get()
1230 prefix = servo_constants.SERVO_TYPE_LABEL_PREFIX
1231 old_type = host_info.get_label_value(prefix)
1232 if old_type == servo_type:
1233 # do not need update
1234 return
1235 host_info.set_version_label(prefix, servo_type)
1236 self.host_info_store.commit(host_info)
1237 logging.info('ServoHost: servo_type updated to %s '
1238 '(previous: %s)', servo_type, old_type)
1239
1240
Otabek Kasimova7ba91a2020-03-09 08:31:01 -07001241 def set_servo_state(self, servo_state):
Otabek Kasimovcc9738e2020-02-14 16:17:15 -08001242 """Set servo info labels to dut host_info"""
Otabek Kasimova7ba91a2020-03-09 08:31:01 -07001243 if servo_state is not None:
Otabek Kasimovcc9738e2020-02-14 16:17:15 -08001244 host_info = self.host_info_store.get()
Garry Wang11b5e872020-03-11 15:14:08 -07001245 servo_state_prefix = servo_constants.SERVO_STATE_LABEL_PREFIX
Otabek Kasimova7ba91a2020-03-09 08:31:01 -07001246 old_state = host_info.get_label_value(servo_state_prefix)
1247 if old_state == servo_state:
1248 # do not need update
1249 return
1250 host_info.set_version_label(servo_state_prefix, servo_state)
Otabek Kasimovcc9738e2020-02-14 16:17:15 -08001251 self.host_info_store.commit(host_info)
Otabek Kasimova7ba91a2020-03-09 08:31:01 -07001252 logging.info('ServoHost: servo_state updated to %s (previous: %s)',
1253 servo_state, old_state)
Dan Shi90466352015-09-22 15:01:05 -07001254
1255
Otabek Kasimova7ba91a2020-03-09 08:31:01 -07001256 def get_servo_state(self):
1257 host_info = self.host_info_store.get()
Garry Wang11b5e872020-03-11 15:14:08 -07001258 servo_state_prefix = servo_constants.SERVO_STATE_LABEL_PREFIX
Otabek Kasimova7ba91a2020-03-09 08:31:01 -07001259 return host_info.get_label_value(servo_state_prefix)
1260
Otabek Kasimov5f039202020-10-28 15:45:29 -07001261 def is_servo_in_working_state(self):
1262 """Validate servo is in WORKING state."""
1263 servo_state = self.get_servo_state()
1264 return servo_state == servo_constants.SERVO_STATE_WORKING
1265
Dana Goyette655af512020-09-03 10:48:23 -07001266 def get_servo_usb_state(self):
1267 """Get the label value indicating the health of the USB drive.
1268
1269 @return: The label value if defined, otherwise '' (empty string).
1270 @rtype: str
1271 """
1272 host_info = self.host_info_store.get()
1273 servo_usb_state_prefix = audit_const.SERVO_USB_STATE_PREFIX
1274 return host_info.get_label_value(servo_usb_state_prefix)
1275
1276 def is_servo_usb_usable(self):
1277 """Check if the servo USB storage device is usable for FAFT.
1278
1279 @return: False if the label indicates a state that will break FAFT.
1280 True if state is okay, or if state is not defined.
1281 @rtype: bool
1282 """
1283 usb_state = self.get_servo_usb_state()
1284 return usb_state in ('', audit_const.HW_STATE_ACCEPTABLE,
1285 audit_const.HW_STATE_NORMAL,
1286 audit_const.HW_STATE_UNKNOWN)
Otabek Kasimov41301a22020-05-10 15:28:21 -07001287
Garry Wang000c6c02020-05-11 21:27:23 -07001288 def _set_smart_usbhub_label(self, smart_usbhub_detected):
1289 if smart_usbhub_detected is None:
1290 # skip the label update here as this indicate we wasn't able
1291 # to confirm usbhub type.
1292 return
1293 host_info = self.host_info_store.get()
1294 if (smart_usbhub_detected ==
1295 (servo_constants.SMART_USBHUB_LABEL in host_info.labels)):
1296 # skip label update if current label match the truth.
1297 return
1298 if smart_usbhub_detected:
1299 logging.info('Adding %s label to host %s',
1300 servo_constants.SMART_USBHUB_LABEL,
1301 self.hostname)
1302 host_info.labels.append(servo_constants.SMART_USBHUB_LABEL)
1303 else:
1304 logging.info('Removing %s label from host %s',
1305 servo_constants.SMART_USBHUB_LABEL,
1306 self.hostname)
1307 host_info.labels.remove(servo_constants.SMART_USBHUB_LABEL)
1308 self.host_info_store.commit(host_info)
1309
J. Richard Barnettec2d99cf2015-11-18 12:46:15 -08001310 def repair(self):
1311 """Attempt to get the DUT to pass `self.verify()`.
Richard Barnette82c35912012-11-20 10:09:10 -08001312
1313 This overrides the base class function for repair; it does
J. Richard Barnette91137f02016-03-10 16:52:26 -08001314 not call back to the parent class, but instead relies on
1315 `self._repair_strategy` to coordinate the verification and
1316 repair steps needed to get the DUT working.
Richard Barnette82c35912012-11-20 10:09:10 -08001317 """
Richard Barnetteabbdc252018-07-26 16:57:42 -07001318 message = 'Beginning repair for host %s board %s model %s'
1319 info = self.host_info_store.get()
1320 message %= (self.hostname, info.board, info.model)
1321 self.record('INFO', None, None, message)
Garry Wanga2e78172020-09-09 23:49:07 -07001322 profile_state = profile_constants.DUT_STATE_READY
Shijin Abraham78ce4402020-09-08 22:04:27 -07001323 # Initialize bluetooth peers
1324 self.initialize_btpeer()
Garry Wang87af1d02020-05-26 17:55:54 -07001325 try:
1326 self._repair_strategy.repair(self)
1327 except hosts.AutoservVerifyDependencyError as e:
Otabek Kasimovd48389b2020-12-07 02:38:34 -08001328 # TODO(otabek): remove when finish b/174191325
1329 self._stat_if_pingable_but_not_sshable()
Garry Wang87af1d02020-05-26 17:55:54 -07001330 # We don't want flag a DUT as failed if only non-critical
1331 # verifier(s) failed during the repair.
1332 if e.is_critical():
Garry Wanga2e78172020-09-09 23:49:07 -07001333 profile_state = profile_constants.DUT_STATE_REPAIR_FAILED
Otabek Kasimov86062d02020-11-17 13:30:22 -08001334 self._reboot_labstation_if_needed()
Otabek Kasimov7cad9ce2020-07-28 14:58:48 -07001335 self.try_set_device_needs_manual_repair()
Garry Wang87af1d02020-05-26 17:55:54 -07001336 raise
Garry Wanga2e78172020-09-09 23:49:07 -07001337 finally:
1338 self.set_health_profile_dut_state(profile_state)
Scott Zawalski89c44dd2013-02-26 09:28:02 -05001339
Otabek Kasimov8bb09912020-10-01 14:44:57 -07001340 def get_verifier_state(self, tag):
1341 """Return the state of servo verifier.
1342
1343 @returns: bool or None
1344 """
1345 return self._repair_strategy.verifier_is_good(tag)
Richard Barnette82c35912012-11-20 10:09:10 -08001346
J. Richard Barnette1d78b012012-05-15 13:56:30 -07001347 def close(self):
David Rileye2c6be12017-12-11 10:20:57 -08001348 """Close connection."""
Fang Deng0ca40e22013-08-27 17:47:44 -07001349 super(CrosHost, self).close()
howardchung83e55272019-08-08 14:08:05 +08001350
Shijin Abraham783a7dd2020-02-14 15:36:11 -08001351 if self._chameleon_host:
1352 self._chameleon_host.close()
xixuand6011f12016-12-08 15:01:58 -08001353
Garry Wang1a493d82020-08-31 21:01:19 -07001354 if self.health_profile:
1355 try:
1356 self.health_profile.close()
1357 except Exception as e:
1358 logging.warning(
1359 'Failed to finalize device health profile; %s', e)
1360
xixuand6011f12016-12-08 15:01:58 -08001361 if self._servo_host:
1362 self._servo_host.close()
J. Richard Barnette1d78b012012-05-15 13:56:30 -07001363
Dan Shi49ca0932014-11-14 11:22:27 -08001364 def get_power_supply_info(self):
1365 """Get the output of power_supply_info.
1366
1367 power_supply_info outputs the info of each power supply, e.g.,
1368 Device: Line Power
1369 online: no
1370 type: Mains
1371 voltage (V): 0
1372 current (A): 0
1373 Device: Battery
1374 state: Discharging
1375 percentage: 95.9276
1376 technology: Li-ion
1377
1378 Above output shows two devices, Line Power and Battery, with details of
1379 each device listed. This function parses the output into a dictionary,
1380 with key being the device name, and value being a dictionary of details
1381 of the device info.
1382
1383 @return: The dictionary of power_supply_info, e.g.,
1384 {'Line Power': {'online': 'yes', 'type': 'main'},
1385 'Battery': {'vendor': 'xyz', 'percentage': '100'}}
Dan Shie9b765d2014-12-29 16:59:49 -08001386 @raise error.AutoservRunError if power_supply_info tool is not found in
1387 the DUT. Caller should handle this error to avoid false failure
1388 on verification.
Dan Shi49ca0932014-11-14 11:22:27 -08001389 """
1390 result = self.run('power_supply_info').stdout.strip()
1391 info = {}
1392 device_name = None
1393 device_info = {}
1394 for line in result.split('\n'):
1395 pair = [v.strip() for v in line.split(':')]
1396 if len(pair) != 2:
1397 continue
1398 if pair[0] == 'Device':
1399 if device_name:
1400 info[device_name] = device_info
1401 device_name = pair[1]
1402 device_info = {}
1403 else:
1404 device_info[pair[0]] = pair[1]
1405 if device_name and not device_name in info:
1406 info[device_name] = device_info
1407 return info
1408
1409
1410 def get_battery_percentage(self):
1411 """Get the battery percentage.
1412
1413 @return: The percentage of battery level, value range from 0-100. Return
1414 None if the battery info cannot be retrieved.
1415 """
1416 try:
1417 info = self.get_power_supply_info()
1418 logging.info(info)
1419 return float(info['Battery']['percentage'])
Dan Shie9b765d2014-12-29 16:59:49 -08001420 except (KeyError, ValueError, error.AutoservRunError):
Dan Shi49ca0932014-11-14 11:22:27 -08001421 return None
1422
1423
Philip Chenaf69ead2020-03-27 13:06:42 -07001424 def get_battery_state(self):
1425 """Get the battery charging state.
1426
1427 @return: A string representing the battery charging state. It can be
1428 'Charging', 'Fully charged', or 'Discharging'.
1429 """
1430 try:
1431 info = self.get_power_supply_info()
1432 logging.info(info)
1433 return info['Battery']['state']
1434 except (KeyError, ValueError, error.AutoservRunError):
1435 return None
1436
1437
Daniel Campello8ca25c22019-12-13 16:48:26 -07001438 def get_battery_display_percentage(self):
1439 """Get the battery display percentage.
1440
1441 @return: The display percentage of battery level, value range from
1442 0-100. Return None if the battery info cannot be retrieved.
1443 """
1444 try:
1445 info = self.get_power_supply_info()
1446 logging.info(info)
1447 return float(info['Battery']['display percentage'])
1448 except (KeyError, ValueError, error.AutoservRunError):
1449 return None
1450
1451
Dan Shi49ca0932014-11-14 11:22:27 -08001452 def is_ac_connected(self):
1453 """Check if the dut has power adapter connected and charging.
1454
1455 @return: True if power adapter is connected and charging.
1456 """
1457 try:
1458 info = self.get_power_supply_info()
1459 return info['Line Power']['online'] == 'yes'
Dan Shie9b765d2014-12-29 16:59:49 -08001460 except (KeyError, error.AutoservRunError):
1461 return None
Dan Shi49ca0932014-11-14 11:22:27 -08001462
1463
Simran Basi5e6339a2013-03-21 11:34:32 -07001464 def _cleanup_poweron(self):
1465 """Special cleanup method to make sure hosts always get power back."""
Garry Wangad4d4fd2019-01-30 17:00:38 -08001466 info = self.host_info_store.get()
1467 if self._RPM_OUTLET_CHANGED not in info.attributes:
Simran Basi5e6339a2013-03-21 11:34:32 -07001468 return
1469 logging.debug('This host has recently interacted with the RPM'
1470 ' Infrastructure. Ensuring power is on.')
1471 try:
1472 self.power_on()
Garry Wangad4d4fd2019-01-30 17:00:38 -08001473 self._remove_rpm_changed_tag()
Simran Basi5e6339a2013-03-21 11:34:32 -07001474 except rpm_client.RemotePowerException:
Simran Basi5e6339a2013-03-21 11:34:32 -07001475 logging.error('Failed to turn Power On for this host after '
1476 'cleanup through the RPM Infrastructure.')
Dan Shi49ca0932014-11-14 11:22:27 -08001477
1478 battery_percentage = self.get_battery_percentage()
Otabek Kasimov58e22562020-11-03 17:17:41 -08001479 if (
1480 battery_percentage
1481 and battery_percentage < cros_constants.MIN_BATTERY_LEVEL):
Dan Shi49ca0932014-11-14 11:22:27 -08001482 raise
1483 elif self.is_ac_connected():
1484 logging.info('The device has power adapter connected and '
1485 'charging. No need to try to turn RPM on '
1486 'again.')
Garry Wangad4d4fd2019-01-30 17:00:38 -08001487 self._remove_rpm_changed_tag()
Dan Shi49ca0932014-11-14 11:22:27 -08001488 logging.info('Battery level is now at %s%%. The device may '
1489 'still have enough power to run test, so no '
1490 'exception will be raised.', battery_percentage)
1491
Simran Basi5e6339a2013-03-21 11:34:32 -07001492
Garry Wangad4d4fd2019-01-30 17:00:38 -08001493 def _remove_rpm_changed_tag(self):
1494 info = self.host_info_store.get()
1495 del info.attributes[self._RPM_OUTLET_CHANGED]
1496 self.host_info_store.commit(info)
1497
1498
1499 def _add_rpm_changed_tag(self):
1500 info = self.host_info_store.get()
Garry Wang518831d2019-02-21 15:15:36 -08001501 info.attributes[self._RPM_OUTLET_CHANGED] = 'true'
Garry Wangad4d4fd2019-01-30 17:00:38 -08001502 self.host_info_store.commit(info)
1503
1504
1505
beepsc87ff602013-07-31 21:53:00 -07001506 def _is_factory_image(self):
1507 """Checks if the image on the DUT is a factory image.
1508
1509 @return: True if the image on the DUT is a factory image.
1510 False otherwise.
1511 """
1512 result = self.run('[ -f /root/.factory_test ]', ignore_status=True)
1513 return result.exit_status == 0
1514
1515
1516 def _restart_ui(self):
J. Richard Barnette84890bd2014-02-21 11:05:47 -08001517 """Restart the Chrome UI.
beepsc87ff602013-07-31 21:53:00 -07001518
1519 @raises: FactoryImageCheckerException for factory images, since
1520 we cannot attempt to restart ui on them.
1521 error.AutoservRunError for any other type of error that
1522 occurs while restarting ui.
1523 """
1524 if self._is_factory_image():
Dan Shi549fb822015-03-24 18:01:11 -07001525 raise FactoryImageCheckerException('Cannot restart ui on factory '
1526 'images')
beepsc87ff602013-07-31 21:53:00 -07001527
J. Richard Barnette84890bd2014-02-21 11:05:47 -08001528 # TODO(jrbarnette): The command to stop/start the ui job
1529 # should live inside cros_ui, too. However that would seem
1530 # to imply interface changes to the existing start()/restart()
1531 # functions, which is a bridge too far (for now).
J. Richard Barnette6069aa12015-06-08 09:10:24 -07001532 prompt = cros_ui.get_chrome_session_ident(self)
J. Richard Barnette84890bd2014-02-21 11:05:47 -08001533 self.run('stop ui; start ui')
1534 cros_ui.wait_for_chrome_ready(prompt, self)
beepsc87ff602013-07-31 21:53:00 -07001535
1536
Daniel Erat4b5f7e02018-07-04 22:05:30 -07001537 def _start_powerd_if_needed(self):
1538 """Start powerd if it isn't already running."""
1539 self.run('start powerd', ignore_status=True)
1540
Kazuhiro Inabaa7a00492020-12-17 16:05:12 +09001541 def _read_arc_prop_file(self, filename):
1542 for path in [
1543 '/usr/share/arcvm/properties/', '/usr/share/arc/properties/'
1544 ]:
1545 if self.path_exists(path + filename):
1546 return utils.parse_cmd_output('cat ' + path + filename,
1547 run_method=self.run)
1548 return None
Daniel Erat4b5f7e02018-07-04 22:05:30 -07001549
Jiyoun Hac172ee72020-12-15 08:57:29 +09001550 def _get_arc_build_info(self):
1551 """Returns a dictionary mapping build properties to their values."""
1552 build_info = None
Kazuhiro Inabaa7a00492020-12-17 16:05:12 +09001553 for filename in ['build.prop', 'vendor_build.prop']:
1554 properties = self._read_arc_prop_file(filename)
1555 if properties:
1556 if build_info:
1557 build_info.update(properties)
1558 else:
1559 build_info = properties
1560 else:
1561 logging.error('Failed to find %s in device.', filename)
Jiyoun Hac172ee72020-12-15 08:57:29 +09001562 return build_info
1563
Jiyoun Haba37f312021-01-13 09:44:16 +09001564 def get_arc_primary_abi(self):
Jiyoun Hac172ee72020-12-15 08:57:29 +09001565 """Returns the primary abi of the host."""
1566 return self._get_arc_build_info().get('ro.product.cpu.abi')
1567
Jiyoun Haba37f312021-01-13 09:44:16 +09001568 def get_arc_security_patch(self):
Jiyoun Hac172ee72020-12-15 08:57:29 +09001569 """Returns the security patch of the host."""
1570 return self._get_arc_build_info().get('ro.build.version.security_patch')
1571
Kazuhiro Inabaa7a00492020-12-17 16:05:12 +09001572 def get_arc_first_api_level(self):
1573 """Returns the security patch of the host."""
1574 return self._get_arc_build_info().get('ro.product.first_api_level')
1575
xixuana3bbc422017-05-04 15:57:21 -07001576 def _get_lsb_release_content(self):
1577 """Return the content of lsb-release file of host."""
1578 return self.run(
1579 'cat "%s"' % client_constants.LSB_RELEASE).stdout.strip()
1580
1581
Dan Shi549fb822015-03-24 18:01:11 -07001582 def get_release_version(self):
1583 """Get the value of attribute CHROMEOS_RELEASE_VERSION from lsb-release.
1584
1585 @returns The version string in lsb-release, under attribute
1586 CHROMEOS_RELEASE_VERSION.
1587 """
Dan Shi549fb822015-03-24 18:01:11 -07001588 return lsbrelease_utils.get_chromeos_release_version(
xixuana3bbc422017-05-04 15:57:21 -07001589 lsb_release_content=self._get_lsb_release_content())
1590
1591
Don Garrettb9f35802018-01-22 18:25:40 -08001592 def get_release_builder_path(self):
1593 """Get the value of CHROMEOS_RELEASE_BUILDER_PATH from lsb-release.
1594
1595 @returns The version string in lsb-release, under attribute
1596 CHROMEOS_RELEASE_BUILDER_PATH.
1597 """
1598 return lsbrelease_utils.get_chromeos_release_builder_path(
1599 lsb_release_content=self._get_lsb_release_content())
1600
1601
xixuana3bbc422017-05-04 15:57:21 -07001602 def get_chromeos_release_milestone(self):
1603 """Get the value of attribute CHROMEOS_RELEASE_BUILD_TYPE
1604 from lsb-release.
1605
1606 @returns The version string in lsb-release, under attribute
1607 CHROMEOS_RELEASE_BUILD_TYPE.
1608 """
1609 return lsbrelease_utils.get_chromeos_release_milestone(
1610 lsb_release_content=self._get_lsb_release_content())
Dan Shi549fb822015-03-24 18:01:11 -07001611
1612
1613 def verify_cros_version_label(self):
Garry Wangd18e7b32020-08-07 18:31:44 -07001614 """Verify if host's cros-version label match the actual image in dut.
Dan Shi549fb822015-03-24 18:01:11 -07001615
Garry Wangd18e7b32020-08-07 18:31:44 -07001616 @returns True if the label match with image in dut, otherwise False
Dan Shi549fb822015-03-24 18:01:11 -07001617 """
Garry Wangd18e7b32020-08-07 18:31:44 -07001618 os_from_host = self.get_release_builder_path()
1619 info = self.host_info_store.get()
1620 os_from_label = info.get_label_value(self.VERSION_PREFIX)
1621 if not os_from_label:
1622 logging.debug('No existing %s label detected', self.VERSION_PREFIX)
1623 return True
1624
1625 # known cases where the version label will not match the
1626 # original CHROMEOS_RELEASE_BUILDER_PATH setting:
1627 # * Tests for the `arc-presubmit` append "-cheetsth" to the label.
1628 if os_from_label.endswith(provision.CHEETS_SUFFIX):
1629 logging.debug('%s label with %s suffix detected, this suffix will'
1630 ' be ignored when comparing label.',
1631 self.VERSION_PREFIX, provision.CHEETS_SUFFIX)
1632 os_from_label = os_from_label[:-len(provision.CHEETS_SUFFIX)]
1633 logging.debug('OS version from host: %s; OS verision cached in '
1634 'label: %s', os_from_host, os_from_label)
1635 return os_from_label == os_from_host
Dan Shi549fb822015-03-24 18:01:11 -07001636
1637
Laurence Goodby778c9a42017-05-24 19:24:07 -07001638 def cleanup_services(self):
1639 """Reinitializes the device for cleanup.
1640
1641 Subclasses may override this to customize the cleanup method.
1642
1643 To indicate failure of the reset, the implementation may raise
1644 any of:
1645 error.AutoservRunError
1646 error.AutotestRunError
1647 FactoryImageCheckerException
1648
1649 @raises error.AutoservRunError
1650 @raises error.AutotestRunError
1651 @raises error.FactoryImageCheckerException
1652 """
1653 self._restart_ui()
Daniel Erat4b5f7e02018-07-04 22:05:30 -07001654 self._start_powerd_if_needed()
Laurence Goodby778c9a42017-05-24 19:24:07 -07001655
1656
Gregory Nisbetec615d62020-12-11 17:59:20 +00001657 def cleanup(self):
1658 """Cleanup state on device."""
MK Ryu35d661e2014-09-25 17:44:10 -07001659 self.run('rm -f %s' % client_constants.CLEANUP_LOGS_PAUSED_FILE)
Scott Zawalskiddbc31e2012-11-15 11:29:01 -05001660 try:
Laurence Goodby778c9a42017-05-24 19:24:07 -07001661 self.cleanup_services()
beepsc87ff602013-07-31 21:53:00 -07001662 except (error.AutotestRunError, error.AutoservRunError,
1663 FactoryImageCheckerException):
Otabek Kasimovc0d77e82020-03-27 15:01:57 -07001664 logging.warning('Unable to restart ui.')
Namyoon Woo33f38852020-04-13 17:26:58 -07001665
Otabek Kasimovc0d77e82020-03-27 15:01:57 -07001666 # cleanup routines, i.e. reboot the machine.
Gregory Nisbetec615d62020-12-11 17:59:20 +00001667 super(CrosHost, self).cleanup()
Otabek Kasimovc0d77e82020-03-27 15:01:57 -07001668
Simran Basi5e6339a2013-03-21 11:34:32 -07001669 # Check if the rpm outlet was manipulated.
Simran Basid5e5e272012-09-24 15:23:59 -07001670 if self.has_power():
Simran Basi5e6339a2013-03-21 11:34:32 -07001671 self._cleanup_poweron()
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001672
Gregory Nisbetec615d62020-12-11 17:59:20 +00001673
Yu-Ju Honga2be94a2012-07-31 09:48:52 -07001674 def reboot(self, **dargs):
1675 """
1676 This function reboots the site host. The more generic
1677 RemoteHost.reboot() performs sync and sleeps for 5
1678 seconds. This is not necessary for Chrome OS devices as the
1679 sync should be finished in a short time during the reboot
1680 command.
1681 """
Gregory Nisbetec615d62020-12-11 17:59:20 +00001682 if 'reboot_cmd' not in dargs:
Doug Anderson7d5aeb22014-02-27 15:12:17 -08001683 reboot_timeout = dargs.get('reboot_timeout', 10)
J. Richard Barnette9af19632015-09-25 12:18:03 -07001684 dargs['reboot_cmd'] = ('sleep 1; '
1685 'reboot & sleep %d; '
1686 'reboot -f' % reboot_timeout)
Yu-Ju Honga2be94a2012-07-31 09:48:52 -07001687 # Enable fastsync to avoid running extra sync commands before reboot.
Tom Wai-Hong Tamf5cd1d42012-08-13 12:04:08 +08001688 if 'fastsync' not in dargs:
1689 dargs['fastsync'] = True
Michael Liangda8c60a2014-06-03 13:24:51 -07001690
Prathmesh Prabhu53a4c1c2018-09-14 16:36:27 -07001691 dargs['board'] = self.host_info_store.get().board
Vincent Palatindf2372c2016-10-07 17:03:00 +02001692 # Record who called us
1693 orig = sys._getframe(1).f_code
Vincent Palatin80780b22016-07-27 16:02:37 +02001694 metric_fields = {'board' : dargs['board'],
Vincent Palatindf2372c2016-10-07 17:03:00 +02001695 'dut_host_name' : self.hostname,
1696 'success' : True}
1697 metric_debug_fields = {'board' : dargs['board'],
Prathmesh Prabhu53a4c1c2018-09-14 16:36:27 -07001698 'caller' : "%s:%s" % (orig.co_filename,
1699 orig.co_name),
Vincent Palatindf2372c2016-10-07 17:03:00 +02001700 'success' : True,
1701 'error' : ''}
1702
Vincent Palatin80780b22016-07-27 16:02:37 +02001703 t0 = time.time()
1704 try:
1705 super(CrosHost, self).reboot(**dargs)
1706 except Exception as e:
1707 metric_fields['success'] = False
Vincent Palatindf2372c2016-10-07 17:03:00 +02001708 metric_debug_fields['success'] = False
1709 metric_debug_fields['error'] = type(e).__name__
Vincent Palatin80780b22016-07-27 16:02:37 +02001710 raise
1711 finally:
1712 duration = int(time.time() - t0)
Dan Shi5e2efb72017-02-07 11:40:23 -08001713 metrics.Counter(
1714 'chromeos/autotest/autoserv/reboot_count').increment(
1715 fields=metric_fields)
1716 metrics.Counter(
1717 'chromeos/autotest/autoserv/reboot_debug').increment(
1718 fields=metric_debug_fields)
1719 metrics.SecondsDistribution(
1720 'chromeos/autotest/autoserv/reboot_duration').add(
1721 duration, fields=metric_fields)
Yu-Ju Honga2be94a2012-07-31 09:48:52 -07001722
1723
Kalin Stoyanov83d9caa2020-01-31 16:58:27 -08001724 def suspend(self, suspend_time=60, delay_seconds=0,
Ravi Chandra Sadineni812e61b2018-07-09 11:16:50 -07001725 suspend_cmd=None, allow_early_resume=False):
Gwendal Grignou7a61d2f2014-05-23 11:05:51 -07001726 """
1727 This function suspends the site host.
Ravi Chandra Sadineni812e61b2018-07-09 11:16:50 -07001728
1729 @param suspend_time: How long to suspend as integer seconds.
1730 @param suspend_cmd: Suspend command to execute.
1731 @param allow_early_resume: If False and if device resumes before
1732 |suspend_time|, throw an error.
1733
1734 @exception AutoservSuspendError Host resumed earlier than
1735 |suspend_time|.
Gwendal Grignou7a61d2f2014-05-23 11:05:51 -07001736 """
Ravi Chandra Sadineni812e61b2018-07-09 11:16:50 -07001737
1738 if suspend_cmd is None:
1739 suspend_cmd = ' && '.join([
J. Richard Barnette9af19632015-09-25 12:18:03 -07001740 'echo 0 > /sys/class/rtc/rtc0/wakealarm',
Gwendal Grignou7a61d2f2014-05-23 11:05:51 -07001741 'echo +%d > /sys/class/rtc/rtc0/wakealarm' % suspend_time,
Kalin Stoyanov83d9caa2020-01-31 16:58:27 -08001742 'powerd_dbus_suspend --delay=%d' % delay_seconds])
Ravi Chandra Sadineni812e61b2018-07-09 11:16:50 -07001743 super(CrosHost, self).suspend(suspend_time, suspend_cmd,
1744 allow_early_resume);
Gwendal Grignou7a61d2f2014-05-23 11:05:51 -07001745
1746
Simran Basiec564392014-08-25 16:48:09 -07001747 def upstart_status(self, service_name):
1748 """Check the status of an upstart init script.
1749
1750 @param service_name: Service to look up.
1751
1752 @returns True if the service is running, False otherwise.
1753 """
Richard Barnettee204dc52017-09-26 11:02:25 -07001754 return 'start/running' in self.run('status %s' % service_name,
1755 ignore_status=True).stdout
Simran Basiec564392014-08-25 16:48:09 -07001756
Tom Hughese9552342018-12-18 14:29:25 -08001757 def upstart_stop(self, service_name):
1758 """Stops an upstart job if it's running.
1759
1760 @param service_name: Service to stop
1761
1762 @returns True if service has been stopped or was already stopped
1763 False otherwise.
1764 """
1765 if not self.upstart_status(service_name):
1766 return True
1767
1768 result = self.run('stop %s' % service_name, ignore_status=True)
1769 if result.exit_status != 0:
1770 return False
1771 return True
1772
1773 def upstart_restart(self, service_name):
1774 """Restarts (or starts) an upstart job.
1775
1776 @param service_name: Service to start/restart
1777
1778 @returns True if service has been started/restarted, False otherwise.
1779 """
1780 cmd = 'start'
1781 if self.upstart_status(service_name):
1782 cmd = 'restart'
1783 cmd = cmd + ' %s' % service_name
1784 result = self.run(cmd)
1785 if result.exit_status != 0:
1786 return False
1787 return True
Simran Basiec564392014-08-25 16:48:09 -07001788
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001789 def verify_software(self):
Richard Barnetteb2bc13c2013-01-08 17:32:51 -08001790 """Verify working software on a Chrome OS system.
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001791
Richard Barnetteb2bc13c2013-01-08 17:32:51 -08001792 Tests for the following conditions:
1793 1. All conditions tested by the parent version of this
1794 function.
1795 2. Sufficient space in /mnt/stateful_partition.
Fang Deng6b05f5b2013-03-20 13:42:11 -07001796 3. Sufficient space in /mnt/stateful_partition/encrypted.
1797 4. update_engine answers a simple status request over DBus.
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001798
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001799 """
Fang Deng0ca40e22013-08-27 17:47:44 -07001800 super(CrosHost, self).verify_software()
Dan Shib8540a52015-07-16 14:18:23 -07001801 default_kilo_inodes_required = CONFIG.get_config_value(
1802 'SERVER', 'kilo_inodes_required', type=int, default=100)
1803 board = self.get_board().replace(ds_constants.BOARD_PREFIX, '')
1804 kilo_inodes_required = CONFIG.get_config_value(
1805 'SERVER', 'kilo_inodes_required_%s' % board,
1806 type=int, default=default_kilo_inodes_required)
1807 self.check_inodes('/mnt/stateful_partition', kilo_inodes_required)
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001808 self.check_diskspace(
1809 '/mnt/stateful_partition',
Dan Shib8540a52015-07-16 14:18:23 -07001810 CONFIG.get_config_value(
Fang Deng6b05f5b2013-03-20 13:42:11 -07001811 'SERVER', 'gb_diskspace_required', type=float,
1812 default=20.0))
Gaurav Shahe448af82014-06-19 15:18:59 -07001813 encrypted_stateful_path = '/mnt/stateful_partition/encrypted'
1814 # Not all targets build with encrypted stateful support.
1815 if self.path_exists(encrypted_stateful_path):
1816 self.check_diskspace(
1817 encrypted_stateful_path,
Dan Shib8540a52015-07-16 14:18:23 -07001818 CONFIG.get_config_value(
Gaurav Shahe448af82014-06-19 15:18:59 -07001819 'SERVER', 'gb_encrypted_diskspace_required', type=float,
1820 default=0.1))
beepsc87ff602013-07-31 21:53:00 -07001821
Justin TerAvest3b0c5ba2017-11-07 09:59:11 -07001822 self.wait_for_system_services()
Prashanth B5d0a0512014-04-25 12:26:08 -07001823
beepsc87ff602013-07-31 21:53:00 -07001824 # Factory images don't run update engine,
1825 # goofy controls dbus on these DUTs.
1826 if not self._is_factory_image():
1827 self.run('update_engine_client --status')
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001828
1829
Justin TerAvest3b0c5ba2017-11-07 09:59:11 -07001830 @retry.retry(error.AutoservError, timeout_min=5, delay_sec=10)
1831 def wait_for_system_services(self):
1832 """Waits for system-services to be running.
1833
1834 Sometimes, update_engine will take a while to update firmware, so we
1835 should give this some time to finish. See crbug.com/765686#c38 for
1836 details.
1837 """
1838 if not self.upstart_status('system-services'):
1839 raise error.AutoservError('Chrome failed to reach login. '
1840 'System services not running.')
1841
1842
J. Richard Barnettea7e7fdc2016-02-12 12:35:36 -08001843 def verify(self):
Pradeep Sawlaniaa013fa2017-11-09 06:34:18 -08001844 """Verify Chrome OS system is in good state."""
Richard Barnetteabbdc252018-07-26 16:57:42 -07001845 message = 'Beginning verify for host %s board %s model %s'
1846 info = self.host_info_store.get()
1847 message %= (self.hostname, info.board, info.model)
1848 self.record('INFO', None, None, message)
Garry Wang87af1d02020-05-26 17:55:54 -07001849 try:
1850 self._repair_strategy.verify(self)
1851 except hosts.AutoservVerifyDependencyError as e:
1852 # We don't want flag a DUT as failed if only non-critical
1853 # verifier(s) failed during the repair.
1854 if e.is_critical():
1855 raise
J. Richard Barnettea7e7fdc2016-02-12 12:35:36 -08001856
1857
Fang Deng96667ca2013-08-01 17:46:18 -07001858 def make_ssh_command(self, user='root', port=22, opts='', hosts_file=None,
Dean Liaoe3e75f62017-11-14 10:36:43 +08001859 connect_timeout=None, alive_interval=None,
1860 alive_count_max=None, connection_attempts=None):
Fang Deng96667ca2013-08-01 17:46:18 -07001861 """Override default make_ssh_command to use options tuned for Chrome OS.
1862
1863 Tuning changes:
1864 - ConnectTimeout=30; maximum of 30 seconds allowed for an SSH
1865 connection failure. Consistency with remote_access.sh.
1866
Samuel Tan2ce155b2015-06-23 18:24:38 -07001867 - ServerAliveInterval=900; which causes SSH to ping connection every
1868 900 seconds. In conjunction with ServerAliveCountMax ensures
1869 that if the connection dies, Autotest will bail out.
Fang Deng96667ca2013-08-01 17:46:18 -07001870 Originally tried 60 secs, but saw frequent job ABORTS where
Samuel Tan2ce155b2015-06-23 18:24:38 -07001871 the test completed successfully. Later increased from 180 seconds to
1872 900 seconds to account for tests where the DUT is suspended for
1873 longer periods of time.
Fang Deng96667ca2013-08-01 17:46:18 -07001874
1875 - ServerAliveCountMax=3; consistency with remote_access.sh.
1876
1877 - ConnectAttempts=4; reduce flakiness in connection errors;
1878 consistency with remote_access.sh.
1879
1880 - UserKnownHostsFile=/dev/null; we don't care about the keys.
1881 Host keys change with every new installation, don't waste
1882 memory/space saving them.
1883
1884 - SSH protocol forced to 2; needed for ServerAliveInterval.
1885
1886 @param user User name to use for the ssh connection.
1887 @param port Port on the target host to use for ssh connection.
1888 @param opts Additional options to the ssh command.
1889 @param hosts_file Ignored.
1890 @param connect_timeout Ignored.
1891 @param alive_interval Ignored.
Dean Liaoe3e75f62017-11-14 10:36:43 +08001892 @param alive_count_max Ignored.
1893 @param connection_attempts Ignored.
Fang Deng96667ca2013-08-01 17:46:18 -07001894 """
Dean Liaoe3e75f62017-11-14 10:36:43 +08001895 options = ' '.join([opts, '-o Protocol=2'])
1896 return super(CrosHost, self).make_ssh_command(
1897 user=user, port=port, opts=options, hosts_file='/dev/null',
1898 connect_timeout=30, alive_interval=900, alive_count_max=3,
1899 connection_attempts=4)
1900
1901
Jason Abeleb6f924f2013-11-13 16:01:54 -08001902 def syslog(self, message, tag='autotest'):
1903 """Logs a message to syslog on host.
1904
1905 @param message String message to log into syslog
1906 @param tag String tag prefix for syslog
1907
1908 """
1909 self.run('logger -t "%s" "%s"' % (tag, message))
1910
1911
J. Richard Barnetteb6de7e32013-02-14 13:28:04 -08001912 def _ping_check_status(self, status):
1913 """Ping the host once, and return whether it has a given status.
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001914
J. Richard Barnetteb6de7e32013-02-14 13:28:04 -08001915 @param status Check the ping status against this value.
1916 @return True iff `status` and the result of ping are the same
1917 (i.e. both True or both False).
1918
1919 """
Abhishek Pandit-Subedi038df162020-09-14 16:37:43 -07001920 ping_val = utils.ping(self.hostname,
1921 tries=1,
1922 deadline=1,
1923 timeout=2,
1924 ignore_timeout=True)
J. Richard Barnetteb6de7e32013-02-14 13:28:04 -08001925 return not (status ^ (ping_val == 0))
1926
1927 def _ping_wait_for_status(self, status, timeout):
1928 """Wait for the host to have a given status (UP or DOWN).
1929
1930 Status is checked by polling. Polling will not last longer
1931 than the number of seconds in `timeout`. The polling
1932 interval will be long enough that only approximately
1933 _PING_WAIT_COUNT polling cycles will be executed, subject
1934 to a maximum interval of about one minute.
1935
1936 @param status Waiting will stop immediately if `ping` of the
1937 host returns this status.
1938 @param timeout Poll for at most this many seconds.
1939 @return True iff the host status from `ping` matched the
1940 requested status at the time of return.
1941
1942 """
1943 # _ping_check_status() takes about 1 second, hence the
1944 # "- 1" in the formula below.
Nathan Ciobanu38480a32016-10-25 15:26:45 -07001945 # FIXME: if the ping command errors then _ping_check_status()
1946 # returns instantly. If timeout is also smaller than twice
1947 # _PING_WAIT_COUNT then the while loop below forks many
1948 # thousands of ping commands (see /tmp/test_that_results_XXXXX/
1949 # /results-1-logging_YYY.ZZZ/debug/autoserv.DEBUG) and hogs one
1950 # CPU core for 60 seconds.
J. Richard Barnetteb6de7e32013-02-14 13:28:04 -08001951 poll_interval = min(int(timeout / self._PING_WAIT_COUNT), 60) - 1
1952 end_time = time.time() + timeout
1953 while time.time() <= end_time:
1954 if self._ping_check_status(status):
1955 return True
1956 if poll_interval > 0:
1957 time.sleep(poll_interval)
1958
1959 # The last thing we did was sleep(poll_interval), so it may
1960 # have been too long since the last `ping`. Check one more
1961 # time, just to be sure.
1962 return self._ping_check_status(status)
1963
1964 def ping_wait_up(self, timeout):
1965 """Wait for the host to respond to `ping`.
1966
1967 N.B. This method is not a reliable substitute for
1968 `wait_up()`, because a host that responds to ping will not
1969 necessarily respond to ssh. This method should only be used
1970 if the target DUT can be considered functional even if it
1971 can't be reached via ssh.
1972
1973 @param timeout Minimum time to allow before declaring the
1974 host to be non-responsive.
1975 @return True iff the host answered to ping before the timeout.
1976
1977 """
Andrew Luo4be621d2020-03-21 07:01:13 -07001978 if self.use_icmp:
1979 return self._ping_wait_for_status(self._PING_STATUS_UP, timeout)
1980 else:
1981 logging.debug('Using SSH instead of ICMP for ping_wait_up.')
1982 return self.wait_up(timeout)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001983
Andrew Bresticker678c0c72013-01-22 10:44:09 -08001984 def ping_wait_down(self, timeout):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001985 """Wait until the host no longer responds to `ping`.
1986
J. Richard Barnetteb6de7e32013-02-14 13:28:04 -08001987 This function can be used as a slightly faster version of
1988 `wait_down()`, by avoiding potentially long ssh timeouts.
1989
1990 @param timeout Minimum time to allow for the host to become
1991 non-responsive.
1992 @return True iff the host quit answering ping before the
1993 timeout.
1994
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001995 """
Andrew Luo4be621d2020-03-21 07:01:13 -07001996 if self.use_icmp:
1997 return self._ping_wait_for_status(self._PING_STATUS_DOWN, timeout)
1998 else:
1999 logging.debug('Using SSH instead of ICMP for ping_wait_down.')
2000 return self.wait_down(timeout)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002001
Anand K Mistry50f218e2020-07-31 14:50:15 +10002002 def _is_host_port_forwarded(self):
Garry Wanga2e78172020-09-09 23:49:07 -07002003 """Checks if the dut is connected over port forwarding.
Anand K Mistry50f218e2020-07-31 14:50:15 +10002004
2005 N.B. This method does not detect all situations where port forwarding is
2006 occurring. Namely, running autotest on the dut may result in a
2007 false-positive, and port forwarding using a different machine on the
2008 same network will be a false-negative.
2009
2010 @return True if the dut is connected over port forwarding
2011 False otherwise
2012 """
Garry Wanga2e78172020-09-09 23:49:07 -07002013 is_localhost = self.hostname in ['localhost', '127.0.0.1']
2014 is_forwarded = is_localhost and not self.is_default_port
2015 if is_forwarded:
2016 logging.info('Detected DUT connected by port forwarding')
2017 return is_forwarded
Anand K Mistry50f218e2020-07-31 14:50:15 +10002018
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08002019 def test_wait_for_sleep(self, sleep_timeout=None):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002020 """Wait for the client to enter low-power sleep mode.
2021
2022 The test for "is asleep" can't distinguish a system that is
2023 powered off; to confirm that the unit was asleep, it is
2024 necessary to force resume, and then call
2025 `test_wait_for_resume()`.
2026
2027 This function is expected to be called from a test as part
2028 of a sequence like the following:
2029
2030 ~~~~~~~~
2031 boot_id = host.get_boot_id()
2032 # trigger sleep on the host
2033 host.test_wait_for_sleep()
2034 # trigger resume on the host
2035 host.test_wait_for_resume(boot_id)
2036 ~~~~~~~~
2037
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08002038 @param sleep_timeout time limit in seconds to allow the host sleep.
2039
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002040 @exception TestFail The host did not go to sleep within
2041 the allowed time.
2042 """
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08002043 if sleep_timeout is None:
2044 sleep_timeout = self.SLEEP_TIMEOUT
2045
Anand K Mistry50f218e2020-07-31 14:50:15 +10002046 # If the dut is accessed over SSH port-forwarding, `ping` is not useful
2047 # for detecting the dut is down since a ping to localhost will always
2048 # succeed. In this case, fall back to wait_down() which uses SSH.
2049 if self._is_host_port_forwarded():
Garry Wanga2e78172020-09-09 23:49:07 -07002050 success = self.wait_down(timeout=sleep_timeout)
Anand K Mistry50f218e2020-07-31 14:50:15 +10002051 else:
Garry Wanga2e78172020-09-09 23:49:07 -07002052 success = self.ping_wait_down(timeout=sleep_timeout)
Anand K Mistry50f218e2020-07-31 14:50:15 +10002053
2054 if not success:
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002055 raise error.TestFail(
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08002056 'client failed to sleep after %d seconds' % sleep_timeout)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002057
2058
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08002059 def test_wait_for_resume(self, old_boot_id, resume_timeout=None):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002060 """Wait for the client to resume from low-power sleep mode.
2061
2062 The `old_boot_id` parameter should be the value from
2063 `get_boot_id()` obtained prior to entering sleep mode. A
2064 `TestFail` exception is raised if the boot id changes.
2065
2066 See @ref test_wait_for_sleep for more on this function's
2067 usage.
2068
J. Richard Barnette7214e0b2013-02-06 15:20:49 -08002069 @param old_boot_id A boot id value obtained before the
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002070 target host went to sleep.
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08002071 @param resume_timeout time limit in seconds to allow the host up.
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002072
2073 @exception TestFail The host did not respond within the
2074 allowed time.
2075 @exception TestFail The host responded, but the boot id test
2076 indicated a reboot rather than a sleep
2077 cycle.
2078 """
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08002079 if resume_timeout is None:
2080 resume_timeout = self.RESUME_TIMEOUT
2081
2082 if not self.wait_up(timeout=resume_timeout):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002083 raise error.TestFail(
2084 'client failed to resume from sleep after %d seconds' %
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08002085 resume_timeout)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002086 else:
2087 new_boot_id = self.get_boot_id()
2088 if new_boot_id != old_boot_id:
Tom Wai-Hong Tam01792682015-01-06 08:00:46 +08002089 logging.error('client rebooted (old boot %s, new boot %s)',
2090 old_boot_id, new_boot_id)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002091 raise error.TestFail(
Tom Wai-Hong Tam01792682015-01-06 08:00:46 +08002092 'client rebooted, but sleep was expected')
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002093
2094
Tom Wai-Hong Tamfe005c22014-12-03 09:25:44 +08002095 def test_wait_for_shutdown(self, shutdown_timeout=None):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002096 """Wait for the client to shut down.
2097
2098 The test for "has shut down" can't distinguish a system that
2099 is merely asleep; to confirm that the unit was down, it is
2100 necessary to force boot, and then call test_wait_for_boot().
2101
2102 This function is expected to be called from a test as part
2103 of a sequence like the following:
2104
2105 ~~~~~~~~
2106 boot_id = host.get_boot_id()
2107 # trigger shutdown on the host
2108 host.test_wait_for_shutdown()
2109 # trigger boot on the host
2110 host.test_wait_for_boot(boot_id)
2111 ~~~~~~~~
2112
Tom Wai-Hong Tamfe005c22014-12-03 09:25:44 +08002113 @param shutdown_timeout time limit in seconds to allow the host down.
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002114 @exception TestFail The host did not shut down within the
2115 allowed time.
2116 """
Tom Wai-Hong Tamfe005c22014-12-03 09:25:44 +08002117 if shutdown_timeout is None:
2118 shutdown_timeout = self.SHUTDOWN_TIMEOUT
2119
Anand K Mistry50f218e2020-07-31 14:50:15 +10002120 if self._is_host_port_forwarded():
Garry Wanga2e78172020-09-09 23:49:07 -07002121 success = self.wait_down(timeout=shutdown_timeout)
Anand K Mistry50f218e2020-07-31 14:50:15 +10002122 else:
Garry Wanga2e78172020-09-09 23:49:07 -07002123 success = self.ping_wait_down(timeout=shutdown_timeout)
Anand K Mistry50f218e2020-07-31 14:50:15 +10002124
2125 if not success:
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002126 raise error.TestFail(
2127 'client failed to shut down after %d seconds' %
Tom Wai-Hong Tamfe005c22014-12-03 09:25:44 +08002128 shutdown_timeout)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002129
2130
2131 def test_wait_for_boot(self, old_boot_id=None):
2132 """Wait for the client to boot from cold power.
2133
2134 The `old_boot_id` parameter should be the value from
2135 `get_boot_id()` obtained prior to shutting down. A
2136 `TestFail` exception is raised if the boot id does not
2137 change. The boot id test is omitted if `old_boot_id` is not
2138 specified.
2139
2140 See @ref test_wait_for_shutdown for more on this function's
2141 usage.
2142
J. Richard Barnette7214e0b2013-02-06 15:20:49 -08002143 @param old_boot_id A boot id value obtained before the
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002144 shut down.
2145
2146 @exception TestFail The host did not respond within the
2147 allowed time.
2148 @exception TestFail The host responded, but the boot id test
2149 indicated that there was no reboot.
2150 """
J. Richard Barnetteeb69d722012-06-18 17:29:44 -07002151 if not self.wait_up(timeout=self.REBOOT_TIMEOUT):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002152 raise error.TestFail(
2153 'client failed to reboot after %d seconds' %
J. Richard Barnetteeb69d722012-06-18 17:29:44 -07002154 self.REBOOT_TIMEOUT)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002155 elif old_boot_id:
2156 if self.get_boot_id() == old_boot_id:
Tom Wai-Hong Tam01792682015-01-06 08:00:46 +08002157 logging.error('client not rebooted (boot %s)',
2158 old_boot_id)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002159 raise error.TestFail(
Tom Wai-Hong Tam01792682015-01-06 08:00:46 +08002160 'client is back up, but did not reboot')
Simran Basid5e5e272012-09-24 15:23:59 -07002161
2162
2163 @staticmethod
2164 def check_for_rpm_support(hostname):
2165 """For a given hostname, return whether or not it is powered by an RPM.
2166
Simran Basi1df55112013-09-06 11:25:09 -07002167 @param hostname: hostname to check for rpm support.
2168
Simran Basid5e5e272012-09-24 15:23:59 -07002169 @return None if this host does not follows the defined naming format
2170 for RPM powered DUT's in the lab. If it does follow the format,
2171 it returns a regular expression MatchObject instead.
2172 """
Fang Dengbaff9082015-01-06 13:46:15 -08002173 return re.match(CrosHost._RPM_HOSTNAME_REGEX, hostname)
Simran Basid5e5e272012-09-24 15:23:59 -07002174
2175
2176 def has_power(self):
2177 """For this host, return whether or not it is powered by an RPM.
2178
2179 @return True if this host is in the CROS lab and follows the defined
2180 naming format.
2181 """
Fang Deng0ca40e22013-08-27 17:47:44 -07002182 return CrosHost.check_for_rpm_support(self.hostname)
Simran Basid5e5e272012-09-24 15:23:59 -07002183
2184
Ismail Noorbasha07fdb612013-02-14 14:13:31 -08002185 def _set_power(self, state, power_method):
Garry Wang5e5538a2019-04-08 15:36:18 -07002186 """Sets the power to the host via RPM, CCD, Servo or manual.
Ismail Noorbasha07fdb612013-02-14 14:13:31 -08002187
2188 @param state Specifies which power state to set to DUT
2189 @param power_method Specifies which method of power control to
Garry Wang5e5538a2019-04-08 15:36:18 -07002190 use. By default "RPM" or "CCD" will be used based
2191 on servo type. Valid values from
2192 POWER_CONTROL_VALID_ARGS, or None to use default.
Ismail Noorbasha07fdb612013-02-14 14:13:31 -08002193
2194 """
2195 ACCEPTABLE_STATES = ['ON', 'OFF']
2196
Garry Wang5e5538a2019-04-08 15:36:18 -07002197 if not power_method:
2198 power_method = self.get_default_power_method()
2199
2200 state = state.upper()
2201 if state not in ACCEPTABLE_STATES:
Ismail Noorbasha07fdb612013-02-14 14:13:31 -08002202 raise error.TestError('State must be one of: %s.'
2203 % (ACCEPTABLE_STATES,))
2204
2205 if power_method == self.POWER_CONTROL_SERVO:
2206 logging.info('Setting servo port J10 to %s', state)
2207 self.servo.set('prtctl3_pwren', state.lower())
2208 time.sleep(self._USB_POWER_TIMEOUT)
2209 elif power_method == self.POWER_CONTROL_MANUAL:
2210 logging.info('You have %d seconds to set the AC power to %s.',
2211 self._POWER_CYCLE_TIMEOUT, state)
2212 time.sleep(self._POWER_CYCLE_TIMEOUT)
Garry Wang5e5538a2019-04-08 15:36:18 -07002213 elif power_method == self.POWER_CONTROL_CCD:
2214 servo_role = 'src' if state == 'ON' else 'snk'
2215 logging.info('servo ccd power pass through detected,'
2216 ' changing servo_role to %s.', servo_role)
2217 self.servo.set_servo_v4_role(servo_role)
2218 if not self.ping_wait_up(timeout=self._CHANGE_SERVO_ROLE_TIMEOUT):
Garry Wang94bf9de2019-06-10 17:23:37 -07002219 # Make sure we don't leave DUT with no power(servo_role=snk)
2220 # when DUT is not pingable, as we raise a exception here
2221 # that may break a power cycle in the middle.
2222 self.servo.set_servo_v4_role('src')
Garry Wang5e5538a2019-04-08 15:36:18 -07002223 raise error.AutoservError(
2224 'DUT failed to regain network connection after %d seconds.'
2225 % self._CHANGE_SERVO_ROLE_TIMEOUT)
Ismail Noorbasha07fdb612013-02-14 14:13:31 -08002226 else:
2227 if not self.has_power():
2228 raise error.TestFail('DUT does not have RPM connected.')
Garry Wangad4d4fd2019-01-30 17:00:38 -08002229 self._add_rpm_changed_tag()
Garry Wang5e5538a2019-04-08 15:36:18 -07002230 rpm_client.set_power(self, state, timeout_mins=5)
Simran Basid5e5e272012-09-24 15:23:59 -07002231
2232
Garry Wang5e5538a2019-04-08 15:36:18 -07002233 def power_off(self, power_method=None):
2234 """Turn off power to this host via RPM, CCD, Servo or manual.
Ismail Noorbasha07fdb612013-02-14 14:13:31 -08002235
2236 @param power_method Specifies which method of power control to
Garry Wang5e5538a2019-04-08 15:36:18 -07002237 use. By default "RPM" or "CCD" will be used based
2238 on servo type. Valid values from
2239 POWER_CONTROL_VALID_ARGS, or None to use default.
Ismail Noorbasha07fdb612013-02-14 14:13:31 -08002240
2241 """
Derek Beckettb66e5c82020-08-12 15:31:02 -07002242 self._sync_if_up()
Ismail Noorbasha07fdb612013-02-14 14:13:31 -08002243 self._set_power('OFF', power_method)
Simran Basid5e5e272012-09-24 15:23:59 -07002244
Derek Beckettb66e5c82020-08-12 15:31:02 -07002245 def _check_supported(self):
2246 """Throw an error if dts mode control is not supported."""
2247 if not self.servo_pwr_supported:
2248 raise error.TestFail('power_state controls not supported')
2249
2250 def _sync_if_up(self):
2251 """Run sync on the DUT and wait for completion if the DUT is up.
2252
2253 Additionally, try to sync and ignore status if its not up.
2254
2255 Useful prior to reboots to ensure files are written to disc.
2256
2257 """
2258 if self.is_up_fast():
2259 self.run("sync")
2260 return
2261 # If it is not up, attempt to sync in the rare event the DUT is up but
2262 # doesn't respond to a ping. Ignore any errors.
2263 try:
2264 self.run("sync", ignore_status=True, timeout=1)
2265 except Exception:
2266 pass
2267
2268 def power_off_via_servo(self):
2269 """Force the DUT to power off.
2270
2271 The DUT is guaranteed to be off at the end of this call,
2272 regardless of its previous state, provided that there is
2273 working EC and boot firmware. There is no requirement for
2274 working OS software.
2275
2276 """
2277 self._check_supported()
2278 self._sync_if_up()
2279 self.servo.set_nocheck('power_state', 'off')
2280
2281 def power_on_via_servo(self, rec_mode='on'):
2282 """Force the DUT to power on.
2283
2284 Prior to calling this function, the DUT must be powered off,
2285 e.g. with a call to `power_off()`.
2286
2287 At power on, recovery mode is set as specified by the
2288 corresponding argument. When booting with recovery mode on, it
2289 is the caller's responsibility to unplug/plug in a bootable
2290 external storage device.
2291
2292 If the DUT requires a delay after powering on but before
2293 processing inputs such as USB stick insertion, the delay is
2294 handled by this method; the caller is not responsible for such
2295 delays.
2296
2297 @param rec_mode Setting of recovery mode to be applied at
2298 power on. default: REC_OFF aka 'off'
2299
2300 """
2301 self._check_supported()
2302 self.servo.set_nocheck('power_state', rec_mode)
2303
2304 def reset_via_servo(self):
2305 """Force the DUT to reset.
2306
2307 The DUT is guaranteed to be on at the end of this call,
2308 regardless of its previous state, provided that there is
2309 working OS software. This also guarantees that the EC has
2310 been restarted.
2311
2312 """
2313 self._check_supported()
2314 self._sync_if_up()
2315 self.servo.set_nocheck('power_state', 'reset')
2316
Simran Basid5e5e272012-09-24 15:23:59 -07002317
Garry Wang5e5538a2019-04-08 15:36:18 -07002318 def power_on(self, power_method=None):
2319 """Turn on power to this host via RPM, CCD, Servo or manual.
Ismail Noorbasha07fdb612013-02-14 14:13:31 -08002320
2321 @param power_method Specifies which method of power control to
Garry Wang5e5538a2019-04-08 15:36:18 -07002322 use. By default "RPM" or "CCD" will be used based
2323 on servo type. Valid values from
2324 POWER_CONTROL_VALID_ARGS, or None to use default.
Ismail Noorbasha07fdb612013-02-14 14:13:31 -08002325
2326 """
2327 self._set_power('ON', power_method)
2328
2329
Garry Wang5e5538a2019-04-08 15:36:18 -07002330 def power_cycle(self, power_method=None):
Ismail Noorbasha07fdb612013-02-14 14:13:31 -08002331 """Cycle power to this host by turning it OFF, then ON.
2332
2333 @param power_method Specifies which method of power control to
Garry Wang5e5538a2019-04-08 15:36:18 -07002334 use. By default "RPM" or "CCD" will be used based
2335 on servo type. Valid values from
2336 POWER_CONTROL_VALID_ARGS, or None to use default.
Ismail Noorbasha07fdb612013-02-14 14:13:31 -08002337
2338 """
Garry Wang5e5538a2019-04-08 15:36:18 -07002339 if not power_method:
2340 power_method = self.get_default_power_method()
2341
Ismail Noorbasha07fdb612013-02-14 14:13:31 -08002342 if power_method in (self.POWER_CONTROL_SERVO,
Garry Wang5e5538a2019-04-08 15:36:18 -07002343 self.POWER_CONTROL_MANUAL,
2344 self.POWER_CONTROL_CCD):
Ismail Noorbasha07fdb612013-02-14 14:13:31 -08002345 self.power_off(power_method=power_method)
2346 time.sleep(self._POWER_CYCLE_TIMEOUT)
2347 self.power_on(power_method=power_method)
2348 else:
Garry Wangad4d4fd2019-01-30 17:00:38 -08002349 self._add_rpm_changed_tag()
2350 rpm_client.set_power(self, 'CYCLE')
Simran Basic6f1f7a2012-10-16 10:47:46 -07002351
2352
Mary Ruthvende14a8b2019-08-23 12:43:52 -07002353 def get_platform_from_fwid(self):
2354 """Determine the platform from the crossystem fwid.
2355
2356 @returns a string representing this host's platform.
2357 """
Greg Edelstona7b05d12020-04-01 16:00:51 -06002358 # Look at the firmware for non-unibuild cases or if cros_config fails.
Mary Ruthvende14a8b2019-08-23 12:43:52 -07002359 crossystem = utils.Crossystem(self)
2360 crossystem.init()
2361 # Extract fwid value and use the leading part as the platform id.
2362 # fwid generally follow the format of {platform}.{firmware version}
2363 # Example: Alex.X.YYY.Z or Google_Alex.X.YYY.Z
2364 platform = crossystem.fwid().split('.')[0].lower()
2365 # Newer platforms start with 'Google_' while the older ones do not.
2366 return platform.replace('google_', '')
2367
Puthikorn Voravootivat0f7c3d82019-11-27 13:48:39 -08002368
Simran Basic6f1f7a2012-10-16 10:47:46 -07002369 def get_platform(self):
2370 """Determine the correct platform label for this host.
2371
2372 @returns a string representing this host's platform.
2373 """
C Shapiroed87c6f2018-04-19 09:13:58 -06002374 release_info = utils.parse_cmd_output('cat /etc/lsb-release',
2375 run_method=self.run)
C Shapiroed87c6f2018-04-19 09:13:58 -06002376 platform = ''
Puthikorn Voravootivat0f7c3d82019-11-27 13:48:39 -08002377 if release_info.get('CHROMEOS_RELEASE_UNIBUILD') == '1':
Greg Edelstona7b05d12020-04-01 16:00:51 -06002378 platform = self.get_model_from_cros_config()
Mary Ruthvende14a8b2019-08-23 12:43:52 -07002379 return platform if platform else self.get_platform_from_fwid()
Simran Basic6f1f7a2012-10-16 10:47:46 -07002380
2381
Greg Edelstona7b05d12020-04-01 16:00:51 -06002382 def get_model_from_cros_config(self):
2383 """Get the host model from cros_config command.
Puthikorn Voravootivat0f7c3d82019-11-27 13:48:39 -08002384
Greg Edelstona7b05d12020-04-01 16:00:51 -06002385 @returns a string representing this host's model.
Puthikorn Voravootivat0f7c3d82019-11-27 13:48:39 -08002386 """
Greg Edelstona7b05d12020-04-01 16:00:51 -06002387 return cros_config.call_cros_config_get_output('/ name',
2388 self.run, ignore_status=True)
Puthikorn Voravootivat0f7c3d82019-11-27 13:48:39 -08002389
2390
Hung-ying Tyanb1328032014-04-01 14:18:54 +08002391 def get_architecture(self):
2392 """Determine the correct architecture label for this host.
2393
2394 @returns a string representing this host's architecture.
2395 """
2396 crossystem = utils.Crossystem(self)
2397 crossystem.init()
2398 return crossystem.arch()
2399
2400
Luis Lozano40b7d0d2014-01-17 15:12:06 -08002401 def get_chrome_version(self):
2402 """Gets the Chrome version number and milestone as strings.
2403
2404 Invokes "chrome --version" to get the version number and milestone.
2405
2406 @return A tuple (chrome_ver, milestone) where "chrome_ver" is the
2407 current Chrome version number as a string (in the form "W.X.Y.Z")
2408 and "milestone" is the first component of the version number
2409 (the "W" from "W.X.Y.Z"). If the version number cannot be parsed
2410 in the "W.X.Y.Z" format, the "chrome_ver" will be the full output
2411 of "chrome --version" and the milestone will be the empty string.
2412
2413 """
MK Ryu35d661e2014-09-25 17:44:10 -07002414 version_string = self.run(client_constants.CHROME_VERSION_COMMAND).stdout
Luis Lozano40b7d0d2014-01-17 15:12:06 -08002415 return utils.parse_chrome_version(version_string)
2416
J. Richard Barnetted2af5852016-02-05 15:03:10 -08002417
Puthikorn Voravootivatfd132172017-11-16 18:24:38 -08002418 def get_ec_version(self):
2419 """Get the ec version as strings.
2420
2421 @returns a string representing this host's ec version.
2422 """
Puthikorn Voravootivat65ad7672018-01-30 14:00:01 -08002423 command = 'mosys ec info -s fw_version'
Puthikorn Voravootivat720640d2018-02-16 17:32:38 -08002424 result = self.run(command, ignore_status=True)
2425 if result.exit_status != 0:
2426 return ''
2427 return result.stdout.strip()
Puthikorn Voravootivatfd132172017-11-16 18:24:38 -08002428
2429
2430 def get_firmware_version(self):
2431 """Get the firmware version as strings.
2432
2433 @returns a string representing this host's firmware version.
2434 """
2435 crossystem = utils.Crossystem(self)
2436 crossystem.init()
2437 return crossystem.fwid()
2438
2439
Puthikorn Voravootivat58f739b2021-01-07 08:28:36 -08002440 def get_hardware_id(self):
2441 """Get hardware id as strings.
2442
2443 @returns a string representing this host's hardware id.
2444 """
2445 crossystem = utils.Crossystem(self)
2446 crossystem.init()
2447 return crossystem.hwid()
2448
Puthikorn Voravootivatfd132172017-11-16 18:24:38 -08002449 def get_hardware_revision(self):
2450 """Get the hardware revision as strings.
2451
2452 @returns a string representing this host's hardware revision.
2453 """
Puthikorn Voravootivat65ad7672018-01-30 14:00:01 -08002454 command = 'mosys platform version'
Puthikorn Voravootivat720640d2018-02-16 17:32:38 -08002455 result = self.run(command, ignore_status=True)
2456 if result.exit_status != 0:
2457 return ''
2458 return result.stdout.strip()
Puthikorn Voravootivatfd132172017-11-16 18:24:38 -08002459
2460
2461 def get_kernel_version(self):
2462 """Get the kernel version as strings.
2463
2464 @returns a string representing this host's kernel version.
2465 """
2466 return self.run('uname -r').stdout.strip()
2467
2468
Puthikorn Voravootivat54aa53c2018-03-01 19:00:25 -08002469 def get_cpu_name(self):
2470 """Get the cpu name as strings.
2471
2472 @returns a string representing this host's cpu name.
2473 """
2474
2475 # Try get cpu name from device tree first
2476 if self.path_exists('/proc/device-tree/compatible'):
2477 command = ' | '.join(
2478 ["sed -e 's/\\x0/\\n/g' /proc/device-tree/compatible",
2479 'tail -1'])
2480 return self.run(command).stdout.strip().replace(',', ' ')
2481
2482 # Get cpu name from uname -p
2483 command = 'uname -p'
2484 ret = self.run(command).stdout.strip()
2485
2486 # 'uname -p' return variant of unknown or amd64 or x86_64 or i686
2487 # Try get cpu name from /proc/cpuinfo instead
2488 if re.match("unknown|amd64|[ix][0-9]?86(_64)?", ret, re.IGNORECASE):
2489 command = "grep model.name /proc/cpuinfo | cut -f 2 -d: | head -1"
2490 self = self.run(command).stdout.strip()
2491
2492 # Remove bloat from CPU name, for example
2493 # Intel(R) Core(TM) i5-7Y57 CPU @ 1.20GHz -> Intel Core i5-7Y57
2494 # Intel(R) Xeon(R) CPU E5-2690 v4 @ 2.60GHz -> Intel Xeon E5-2690 v4
2495 # AMD A10-7850K APU with Radeon(TM) R7 Graphics -> AMD A10-7850K
2496 # AMD GX-212JC SOC with Radeon(TM) R2E Graphics -> AMD GX-212JC
2497 trim_re = r' (@|processor|apu|soc|radeon).*|\(.*?\)| cpu'
2498 return re.sub(trim_re, '', ret, flags=re.IGNORECASE)
2499
2500
2501 def get_screen_resolution(self):
2502 """Get the screen(s) resolution as strings.
2503 In case of more than 1 monitor, return resolution for each monitor
2504 separate with plus sign.
2505
2506 @returns a string representing this host's screen(s) resolution.
2507 """
2508 command = 'for f in /sys/class/drm/*/*/modes; do head -1 $f; done'
2509 ret = self.run(command, ignore_status=True)
2510 # We might have Chromebox without a screen
2511 if ret.exit_status != 0:
2512 return ''
2513 return ret.stdout.strip().replace('\n', '+')
2514
2515
2516 def get_mem_total_gb(self):
2517 """Get total memory available in the system in GiB (2^20).
2518
2519 @returns an integer representing total memory
2520 """
2521 mem_total_kb = self.read_from_meminfo('MemTotal')
2522 kb_in_gb = float(2 ** 20)
2523 return int(round(mem_total_kb / kb_in_gb))
2524
2525
2526 def get_disk_size_gb(self):
2527 """Get size of disk in GB (10^9)
2528
2529 @returns an integer representing size of disk, 0 in Error Case
2530 """
2531 command = 'grep $(rootdev -s -d | cut -f3 -d/)$ /proc/partitions'
2532 result = self.run(command, ignore_status=True)
2533 if result.exit_status != 0:
2534 return 0
2535 _, _, block, _ = re.split(r' +', result.stdout.strip())
2536 byte_per_block = 1024.0
2537 disk_kb_in_gb = 1e9
2538 return int(int(block) * byte_per_block / disk_kb_in_gb + 0.5)
2539
2540
2541 def get_battery_size(self):
2542 """Get size of battery in Watt-hour via sysfs
2543
2544 This method assumes that battery support voltage_min_design and
2545 charge_full_design sysfs.
2546
2547 @returns a float representing Battery size, 0 if error.
2548 """
2549 # sysfs report data in micro scale
2550 battery_scale = 1e6
2551
2552 command = 'cat /sys/class/power_supply/*/voltage_min_design'
2553 result = self.run(command, ignore_status=True)
2554 if result.exit_status != 0:
2555 return 0
2556 voltage = float(result.stdout.strip()) / battery_scale
2557
2558 command = 'cat /sys/class/power_supply/*/charge_full_design'
2559 result = self.run(command, ignore_status=True)
2560 if result.exit_status != 0:
2561 return 0
2562 amphereHour = float(result.stdout.strip()) / battery_scale
2563
2564 return voltage * amphereHour
2565
2566
2567 def get_low_battery_shutdown_percent(self):
2568 """Get the percent-based low-battery shutdown threshold.
2569
2570 @returns a float representing low-battery shutdown percent, 0 if error.
2571 """
2572 ret = 0.0
2573 try:
2574 command = 'check_powerd_config --low_battery_shutdown_percent'
2575 ret = float(self.run(command).stdout)
2576 except error.CmdError:
2577 logging.debug("Can't run %s", command)
2578 except ValueError:
2579 logging.debug("Didn't get number from %s", command)
2580
2581 return ret
2582
2583
Puthikorn Voravootivat09c83d72018-08-10 15:58:32 -07002584 def has_hammer(self):
2585 """Check whether DUT has hammer device or not.
2586
2587 @returns boolean whether device has hammer or not
2588 """
2589 command = 'grep Hammer /sys/bus/usb/devices/*/product'
2590 return self.run(command, ignore_status=True).exit_status == 0
2591
2592
Niranjan Kumar34618872017-05-31 12:57:09 -07002593 def is_chrome_switch_present(self, switch):
David Haddock3ce538e2017-06-22 13:37:05 -07002594 """Returns True if the specified switch was provided to Chrome.
2595
2596 @param switch The chrome switch to search for.
2597 """
Niranjan Kumar34618872017-05-31 12:57:09 -07002598
Niranjan Kumar5f23fe92017-06-22 15:18:55 -07002599 command = 'pgrep -x -f -c "/opt/google/chrome/chrome.*%s.*"' % switch
2600 return self.run(command, ignore_status=True).exit_status == 0
Niranjan Kumar34618872017-05-31 12:57:09 -07002601
2602
2603 def oobe_triggers_update(self):
2604 """Returns True if this host has an OOBE flow during which
2605 it will perform an update check and perhaps an update.
2606 One example of such a flow is Hands-Off Zero-Touch Enrollment.
2607 As more such flows are developed, code handling them needs
2608 to be added here.
2609
2610 @return Boolean indicating whether this host's OOBE triggers an update.
2611 """
2612 return self.is_chrome_switch_present(
2613 '--enterprise-enable-zero-touch-enrollment=hands-off')
2614
2615
Kevin Chenga2619dc2016-03-28 11:42:08 -07002616 # TODO(kevcheng): change this to just return the board without the
2617 # 'board:' prefix and fix up all the callers. Also look into removing the
2618 # need for this method.
Simran Basic6f1f7a2012-10-16 10:47:46 -07002619 def get_board(self):
2620 """Determine the correct board label for this host.
2621
2622 @returns a string representing this host's board.
2623 """
2624 release_info = utils.parse_cmd_output('cat /etc/lsb-release',
2625 run_method=self.run)
J. Richard Barnetted2af5852016-02-05 15:03:10 -08002626 return (ds_constants.BOARD_PREFIX +
2627 release_info['CHROMEOS_RELEASE_BOARD'])
Simran Basic6f1f7a2012-10-16 10:47:46 -07002628
Po-Hsien Wang4618adf2017-10-10 14:40:21 -07002629 def get_channel(self):
2630 """Determine the correct channel label for this host.
Simran Basic6f1f7a2012-10-16 10:47:46 -07002631
Po-Hsien Wang4618adf2017-10-10 14:40:21 -07002632 @returns: a string represeting this host's build channel.
2633 (stable, dev, beta). None on fail.
2634 """
2635 return lsbrelease_utils.get_chromeos_channel(
2636 lsb_release_content=self._get_lsb_release_content())
Kevin Chenga328da62016-03-31 10:49:04 -07002637
Kevin Chenga328da62016-03-31 10:49:04 -07002638 def get_power_supply(self):
2639 """
2640 Determine what type of power supply the host has
2641
2642 @returns a string representing this host's power supply.
2643 'power:battery' when the device has a battery intended for
2644 extended use
2645 'power:AC_primary' when the device has a battery not intended
2646 for extended use (for moving the machine, etc)
2647 'power:AC_only' when the device has no battery at all.
2648 """
2649 psu = self.run(command='mosys psu type', ignore_status=True)
2650 if psu.exit_status:
2651 # The psu command for mosys is not included for all platforms. The
2652 # assumption is that the device will have a battery if the command
2653 # is not found.
2654 return 'power:battery'
2655
2656 psu_str = psu.stdout.strip()
2657 if psu_str == 'unknown':
2658 return None
2659
2660 return 'power:%s' % psu_str
2661
2662
Puthikorn Voravootivat54aa53c2018-03-01 19:00:25 -08002663 def has_battery(self):
2664 """Determine if DUT has a battery.
2665
2666 Returns:
2667 Boolean, False if known not to have battery, True otherwise.
2668 """
2669 rv = True
2670 power_supply = self.get_power_supply()
2671 if power_supply == 'power:battery':
2672 _NO_BATTERY_BOARD_TYPE = ['CHROMEBOX', 'CHROMEBIT', 'CHROMEBASE']
2673 board_type = self.get_board_type()
2674 if board_type in _NO_BATTERY_BOARD_TYPE:
2675 logging.warn('Do NOT believe type %s has battery. '
2676 'See debug for mosys details', board_type)
Sam Hurst57fa60a2020-05-08 08:55:47 -07002677 psu = utils.system_output('mosys -vvvv psu type',
Puthikorn Voravootivat54aa53c2018-03-01 19:00:25 -08002678 ignore_status=True)
2679 logging.debug(psu)
2680 rv = False
2681 elif power_supply == 'power:AC_only':
2682 rv = False
2683
2684 return rv
2685
2686
Kevin Chenga328da62016-03-31 10:49:04 -07002687 def get_servo(self):
2688 """Determine if the host has a servo attached.
2689
2690 If the host has a working servo attached, it should have a servo label.
2691
2692 @return: string 'servo' if the host has servo attached. Otherwise,
2693 returns None.
2694 """
2695 return 'servo' if self._servo_host else None
2696
2697
Kevin Chenga328da62016-03-31 10:49:04 -07002698 def has_internal_display(self):
2699 """Determine if the device under test is equipped with an internal
2700 display.
2701
2702 @return: 'internal_display' if one is present; None otherwise.
2703 """
2704 from autotest_lib.client.cros.graphics import graphics_utils
2705 from autotest_lib.client.common_lib import utils as common_utils
2706
2707 def __system_output(cmd):
2708 return self.run(cmd).stdout
2709
2710 def __read_file(remote_path):
2711 return self.run('cat %s' % remote_path).stdout
2712
2713 # Hijack the necessary client functions so that we can take advantage
2714 # of the client lib here.
2715 # FIXME: find a less hacky way than this
2716 original_system_output = utils.system_output
2717 original_read_file = common_utils.read_file
2718 utils.system_output = __system_output
2719 common_utils.read_file = __read_file
2720 try:
2721 return ('internal_display' if graphics_utils.has_internal_display()
2722 else None)
2723 finally:
2724 utils.system_output = original_system_output
2725 common_utils.read_file = original_read_file
2726
2727
Dan Shi85276d42014-04-08 22:11:45 -07002728 def is_boot_from_usb(self):
2729 """Check if DUT is boot from USB.
2730
2731 @return: True if DUT is boot from usb.
2732 """
2733 device = self.run('rootdev -s -d').stdout.strip()
2734 removable = int(self.run('cat /sys/block/%s/removable' %
2735 os.path.basename(device)).stdout.strip())
2736 return removable == 1
Helen Zhang17dae2b2014-11-11 09:25:52 -08002737
Otabek Kasimov77a40332020-10-20 15:40:03 -07002738 def is_boot_from_external_device(self):
2739 """Check if DUT is boot from external storage.
2740
2741 @return: True if DUT is boot from external storage.
2742 """
2743 boot_device = self.run('rootdev -s -d', ignore_status=True,
2744 timeout=60).stdout.strip()
2745 if not boot_device:
2746 logging.debug('Boot storage not detected on the host.')
2747 return False
2748 main_storage_cmd = ('. /usr/sbin/write_gpt.sh;'
2749 ' . /usr/share/misc/chromeos-common.sh;'
2750 ' load_base_vars; get_fixed_dst_drive')
2751 main_storage = self.run(main_storage_cmd,
2752 ignore_status=True,
2753 timeout=60).stdout.strip()
Otabek Kasimov723e8562020-12-08 13:29:34 -08002754 if not main_storage or boot_device != main_storage:
2755 logging.debug('Device booted from external storage storage.')
2756 return True
2757 logging.debug('Device booted from main storage.')
2758 return False
Helen Zhang17dae2b2014-11-11 09:25:52 -08002759
2760 def read_from_meminfo(self, key):
Dan Shi49ca0932014-11-14 11:22:27 -08002761 """Return the memory info from /proc/meminfo
Helen Zhang17dae2b2014-11-11 09:25:52 -08002762
2763 @param key: meminfo requested
2764
2765 @return the memory value as a string
2766
2767 """
Helen Zhang17dae2b2014-11-11 09:25:52 -08002768 meminfo = self.run('grep %s /proc/meminfo' % key).stdout.strip()
2769 logging.debug('%s', meminfo)
2770 return int(re.search(r'\d+', meminfo).group(0))
Rohit Makasana8a4923c2015-08-13 17:04:26 -07002771
2772
Rohit Makasana98e696f2016-06-03 18:48:10 -07002773 def get_cpu_arch(self):
2774 """Returns CPU arch of the device.
2775
2776 @return CPU architecture of the DUT.
2777 """
Allen Li2c32d6b2017-02-03 15:28:10 -08002778 # Add CPUs by following logic in client/bin/utils.py.
Rohit Makasana98e696f2016-06-03 18:48:10 -07002779 if self.run("grep '^flags.*:.* lm .*' /proc/cpuinfo",
2780 ignore_status=True).stdout:
2781 return 'x86_64'
2782 if self.run("grep -Ei 'ARM|CPU implementer' /proc/cpuinfo",
2783 ignore_status=True).stdout:
2784 return 'arm'
2785 return 'i386'
2786
2787
Rohit Makasana8a4923c2015-08-13 17:04:26 -07002788 def get_board_type(self):
2789 """
2790 Get the DUT's device type from /etc/lsb-release.
Danny Chan471a8d12015-08-18 14:57:41 -07002791 DEVICETYPE can be one of CHROMEBOX, CHROMEBASE, CHROMEBOOK or more.
2792
2793 @return value of DEVICETYPE param from lsb-release.
Rohit Makasana8a4923c2015-08-13 17:04:26 -07002794 """
Danny Chan471a8d12015-08-18 14:57:41 -07002795 device_type = self.run('grep DEVICETYPE /etc/lsb-release',
2796 ignore_status=True).stdout
2797 if device_type:
Kalin Stoyanov524310b2015-08-21 16:24:04 -07002798 return device_type.split('=')[-1].strip()
Danny Chan471a8d12015-08-18 14:57:41 -07002799 return ''
Gilad Arnolda76bef02015-09-29 13:55:15 -07002800
2801
Rohit Makasanadf0a3a32017-06-30 13:55:18 -07002802 def get_arc_version(self):
2803 """Return ARC version installed on the DUT.
2804
2805 @returns ARC version as string if the CrOS build has ARC, else None.
2806 """
2807 arc_version = self.run('grep CHROMEOS_ARC_VERSION /etc/lsb-release',
2808 ignore_status=True).stdout
2809 if arc_version:
2810 return arc_version.split('=')[-1].strip()
2811 return None
2812
2813
Gilad Arnolda76bef02015-09-29 13:55:15 -07002814 def get_os_type(self):
2815 return 'cros'
Simran Basia5522a32015-10-06 11:01:24 -07002816
2817
Kevin Chenga2619dc2016-03-28 11:42:08 -07002818 def get_labels(self):
Kevin Chengf8660142016-08-12 10:17:41 -07002819 """Return the detected labels on the host."""
Kevin Chenga2619dc2016-03-28 11:42:08 -07002820 return self.labels.get_labels(self)
Garry Wang5e5538a2019-04-08 15:36:18 -07002821
2822
2823 def get_default_power_method(self):
2824 """
2825 Get the default power method for power_on/off/cycle() methods.
2826 @return POWER_CONTROL_RPM or POWER_CONTROL_CCD
2827 """
2828 if not self._default_power_method:
Garry Wang1a004aa2019-05-16 22:56:51 -07002829 self._default_power_method = self.POWER_CONTROL_RPM
Ruben Rodriguez Buchillon3eeeab32019-10-02 15:29:58 -07002830 if self.servo and self.servo.supports_built_in_pd_control():
2831 self._default_power_method = self.POWER_CONTROL_CCD
2832 else:
2833 logging.debug('Either servo is unitialized or the servo '
2834 'setup does not support pd controls. Falling '
2835 'back to default RPM method.')
Garry Wang5e5538a2019-04-08 15:36:18 -07002836 return self._default_power_method
Puthikorn Voravootivat4a054792019-12-13 16:44:17 -08002837
2838
2839 def find_usb_devices(self, idVendor, idProduct):
2840 """
2841 Get usb device sysfs name for specific device.
2842
2843 @param idVendor Vendor ID to search in sysfs directory.
2844 @param idProduct Product ID to search in sysfs directory.
2845
2846 @return Usb node names in /sys/bus/usb/drivers/usb/ that match.
2847 """
2848 # Look for matching file and cut at position 7 to get dir name.
2849 grep_cmd = 'grep {} /sys/bus/usb/drivers/usb/*/{} | cut -f 7 -d /'
2850
2851 vendor_cmd = grep_cmd.format(idVendor, 'idVendor')
2852 product_cmd = grep_cmd.format(idProduct, 'idProduct')
2853
2854 # Use uniq -d to print duplicate line from both command
2855 cmd = 'sort <({}) <({}) | uniq -d'.format(vendor_cmd, product_cmd)
2856
2857 return self.run(cmd, ignore_status=True).stdout.strip().split('\n')
2858
2859
2860 def bind_usb_device(self, usb_node):
2861 """
2862 Bind usb device
2863
2864 @param usb_node Node name in /sys/bus/usb/drivers/usb/
2865 """
2866 cmd = 'echo {} > /sys/bus/usb/drivers/usb/bind'.format(usb_node)
2867 self.run(cmd, ignore_status=True)
2868
2869
2870 def unbind_usb_device(self, usb_node):
2871 """
2872 Unbind usb device
2873
2874 @param usb_node Node name in /sys/bus/usb/drivers/usb/
2875 """
2876 cmd = 'echo {} > /sys/bus/usb/drivers/usb/unbind'.format(usb_node)
2877 self.run(cmd, ignore_status=True)
2878
2879
2880 def get_wlan_ip(self):
2881 """
2882 Get ip address of wlan interface.
2883
2884 @return ip address of wlan or empty string if wlan is not connected.
2885 """
2886 cmds = [
2887 'iw dev', # List wlan physical device
2888 'grep Interface', # Grep only interface name
2889 'cut -f 2 -d" "', # Cut the name part
2890 'xargs ifconfig', # Feed it to ifconfig to get ip
2891 'grep -oE "inet [0-9.]+"', # Grep only ipv4
2892 'cut -f 2 -d " "' # Cut the ip part
2893 ]
2894 return self.run(' | '.join(cmds), ignore_status=True).stdout.strip()
Puthikorn Voravootivatcd0dc9e2020-01-22 14:22:22 -08002895
2896 def connect_to_wifi(self, ssid, passphrase=None, security=None):
2897 """
2898 Connect to wifi network
2899
2900 @param ssid SSID of the wifi network.
2901 @param passphrase Passphrase of the wifi network. None if not existed.
2902 @param security Security of the wifi network. Default to "psk" if
2903 passphase is given without security. Possible values
2904 are "none", "psk", "802_1x".
2905
2906 @return True if succeed, False if not.
2907 """
2908 cmd = '/usr/local/autotest/cros/scripts/wifi connect ' + ssid
2909 if passphrase:
2910 cmd += ' ' + passphrase
2911 if security:
2912 cmd += ' ' + security
2913 return self.run(cmd, ignore_status=True).exit_status == 0
Otabek Kasimov6825b762020-06-23 23:42:44 -07002914
2915 def get_device_repair_state(self):
2916 """Get device repair state"""
2917 return self._device_repair_state
2918
Otabek Kasimov7cad9ce2020-07-28 14:58:48 -07002919 def set_device_repair_state(self, state, resultdir=None):
Otabek Kasimov6825b762020-06-23 23:42:44 -07002920 """Set device repair state.
2921
2922 The special device state will be written to the 'dut_state.repair'
Otabek Kasimov7cad9ce2020-07-28 14:58:48 -07002923 file in result directory. The file will be read by Lucifer. The
2924 file will not be created if result directory not specified.
2925
2926 @params state: The new state for the device.
2927 @params resultdir: The path to result directory. If path not provided
2928 will be attempt to get retrieve it from job
2929 if present.
Otabek Kasimov6825b762020-06-23 23:42:44 -07002930 """
Otabek Kasimov7cad9ce2020-07-28 14:58:48 -07002931 resultdir = resultdir or getattr(self.job, 'resultdir', '')
2932 if resultdir:
2933 target = os.path.join(resultdir, 'dut_state.repair')
Otabek Kasimov6825b762020-06-23 23:42:44 -07002934 common_utils.open_write_close(target, state)
Otabek Kasimov7cad9ce2020-07-28 14:58:48 -07002935 logging.info('Set device state as %s. '
2936 'Created dut_state.repair file.', state)
Otabek Kasimov6825b762020-06-23 23:42:44 -07002937 else:
2938 logging.debug('Cannot write the device state due missing info '
2939 'about result dir.')
2940 self._device_repair_state = state
2941
Otabek Kasimov7cad9ce2020-07-28 14:58:48 -07002942 def set_device_needs_replacement(self, resultdir=None):
2943 """Set device as required replacement.
2944
2945 @params resultdir: The path to result directory. If path not provided
2946 will be attempt to get retrieve it from job
2947 if present.
2948 """
2949 self.set_device_repair_state(
2950 cros_constants.DEVICE_STATE_NEEDS_REPLACEMENT,
2951 resultdir=resultdir)
2952
Otabek Kasimovc7fa5072021-01-20 21:24:18 -08002953 def _dut_is_accessible_by_verifier(self):
2954 """Check if DUT accessible by SSH or PING verifier.
Otabek Kasimov86062d02020-11-17 13:30:22 -08002955
Otabek Kasimovc7fa5072021-01-20 21:24:18 -08002956 @returns: bool, True - verifier marked as success.
2957 False - result not reachable, verifier did not success.
Otabek Kasimov86062d02020-11-17 13:30:22 -08002958 """
2959 if not self._repair_strategy:
2960 return False
Otabek Kasimovc7fa5072021-01-20 21:24:18 -08002961 dut_ssh = self._repair_strategy.verifier_is_good('ssh')
2962 dut_ping = self._repair_strategy.verifier_is_good('ping')
2963 return dut_ssh == hosts.VERIFY_SUCCESS or dut_ssh == hosts.VERIFY_SUCCESS
Otabek Kasimov86062d02020-11-17 13:30:22 -08002964
Otabek Kasimovd48389b2020-12-07 02:38:34 -08002965 def _stat_if_pingable_but_not_sshable(self):
2966 """Check if DUT pingable but failed SSH verifier."""
2967 if not self._repair_strategy:
2968 return
2969 dut_ssh = self._repair_strategy.verifier_is_good('ssh')
2970 dut_ping = self._repair_strategy.verifier_is_good('ping')
2971 if (dut_ping == hosts.VERIFY_FAILED
2972 and dut_ssh == hosts.VERIFY_FAILED):
2973 metrics.Counter('chromeos/autotest/dut_pingable_no_ssh').increment(
2974 fields={'host': self.hostname})
2975
Otabek Kasimov7cad9ce2020-07-28 14:58:48 -07002976 def try_set_device_needs_manual_repair(self):
Otabek Kasimov6825b762020-06-23 23:42:44 -07002977 """Check if device require manual attention to be fixed.
2978
2979 The state 'needs_manual_repair' can be set when auto repair cannot
2980 fix the device due hardware or cable issues.
2981 """
2982 # ignore the logic if state present
2983 # state can be set by any cros repair actions
Otabek Kasimov86062d02020-11-17 13:30:22 -08002984 if self.get_device_repair_state():
Otabek Kasimov6825b762020-06-23 23:42:44 -07002985 return
Otabek Kasimovc7fa5072021-01-20 21:24:18 -08002986 if self._dut_is_accessible_by_verifier():
2987 # DUT is accessible and we still have many options to repair it.
Otabek Kasimovc9812582020-10-08 18:52:52 -07002988 return
Otabek Kasimov9189ede2020-11-09 14:08:58 -08002989 needs_manual_repair = False
2990 dhp = self.health_profile
Otabek Kasimov69253822020-11-24 10:52:27 -08002991 if dhp and dhp.get_repair_fail_count() > 49:
2992 # 42 = 6 times during 7 days. (every 4 hour repair)
2993 # round up to 50 in case somebody will run some attempt on it.
Otabek Kasimovc9812582020-10-08 18:52:52 -07002994 logging.info(
Otabek Kasimov9189ede2020-11-09 14:08:58 -08002995 'DUT is not sshable and fail %s times.'
Otabek Kasimov69253822020-11-24 10:52:27 -08002996 ' Limit to try repair is 50 times',
Otabek Kasimov9189ede2020-11-09 14:08:58 -08002997 dhp.get_repair_fail_count())
2998 needs_manual_repair = True
2999
3000 if not needs_manual_repair:
3001 # We cannot ssh to the DUT and we have hardware or set-up issues
3002 # with servo then we need request manual repair for the DUT.
3003 servo_state_required_manual_fix = [
3004 servo_constants.SERVO_STATE_DUT_NOT_CONNECTED,
3005 servo_constants.SERVO_STATE_NEED_REPLACEMENT,
3006 ]
3007 if self.get_servo_state() in servo_state_required_manual_fix:
3008 logging.info(
3009 'DUT required manual repair because it is not sshable'
3010 ' and possible have setup issue with Servo. Please'
3011 ' verify all connections and present of devices.')
3012 needs_manual_repair = True
3013
3014 if needs_manual_repair:
Otabek Kasimovc9812582020-10-08 18:52:52 -07003015 self.set_device_repair_state(
3016 cros_constants.DEVICE_STATE_NEEDS_MANUAL_REPAIR)
Otabek Kasimov42506d02020-07-29 14:44:57 -07003017
Otabek Kasimov86062d02020-11-17 13:30:22 -08003018 def _reboot_labstation_if_needed(self):
3019 """Place request to reboot the labstation if DUT is not sshable.
3020
3021 @returns: None
3022 """
Puthikorn Voravootivat58f739b2021-01-07 08:28:36 -08003023 message_prefix = "Don't need to request servo-host reboot"
Otabek Kasimovc7fa5072021-01-20 21:24:18 -08003024 if self._dut_is_accessible_by_verifier():
Otabek Kasimov86062d02020-11-17 13:30:22 -08003025 return
3026 if not self._servo_host:
Puthikorn Voravootivat58f739b2021-01-07 08:28:36 -08003027 logging.debug('%s as it not initialized', message_prefix)
Otabek Kasimov86062d02020-11-17 13:30:22 -08003028 return
3029 if not self._servo_host.is_up_fast():
Puthikorn Voravootivat58f739b2021-01-07 08:28:36 -08003030 logging.debug('%s as servo-host is not sshable', message_prefix)
Otabek Kasimov86062d02020-11-17 13:30:22 -08003031 return
3032 if not self._servo_host.is_labstation():
3033 logging.debug('Servo_v3 is not requested to reboot for the DUT')
3034 return
3035 usb_path = self._servo_host.get_main_servo_usb_path()
3036 if usb_path:
3037 connected_port = os.path.basename(os.path.normpath(usb_path))
3038 # Directly connected servo to the labstation looks like '1-5.3'
3039 # and when connected by hub - '1-5.2.3' or '1-5.2.1.3'. Where:
3040 # - '1-5' - port on labstation
3041 # - '2' or '2.1' - port on the hub or smart-hub
3042 # - '3' - port on servo hub
3043 if len(connected_port.split('.')) > 2:
Puthikorn Voravootivat58f739b2021-01-07 08:28:36 -08003044 logging.debug('%s as servo connected by hub', message_prefix)
Otabek Kasimov86062d02020-11-17 13:30:22 -08003045 return
3046 self._servo_host.request_reboot()
3047 logging.info('Requested labstation reboot because DUT is not sshable')
3048
Otabek Kasimov42506d02020-07-29 14:44:57 -07003049 def is_file_system_writable(self, testdirs=None):
3050 """Check is the file systems are writable.
3051
3052 The standard linux response to certain unexpected file system errors
3053 (including hardware errors in block devices) is to change the file
3054 system status to read-only. This checks that that hasn't happened.
3055
3056 @param testdirs: List of directories to check. If no data provided
3057 then '/mnt/stateful_partition' and '/var/tmp'
3058 directories will be checked.
3059
3060 @returns boolean whether file-system writable.
3061 """
3062 def _check_dir(testdir):
3063 # check if we can create a file
3064 filename = os.path.join(testdir, 'writable_my_test_file')
3065 command = 'touch %s && rm %s' % (filename, filename)
3066 rv = self.run(command=command,
3067 timeout=30,
3068 ignore_status=True)
3069 is_writable = rv.exit_status == 0
3070 if not is_writable:
3071 logging.info('Cannot create a file in "%s"!'
3072 ' Probably the FS is read-only', testdir)
3073 logging.info("FileSystem is not writable!")
3074 return False
3075 return True
3076
3077 if not testdirs or len(testdirs) == 0:
3078 # N.B. Order matters here: Encrypted stateful is loop-mounted
3079 # from a file in unencrypted stateful, so we don't test for
3080 # errors in encrypted stateful if unencrypted fails.
3081 testdirs = ['/mnt/stateful_partition', '/var/tmp']
3082
3083 for dir in testdirs:
3084 # loop will be stopped if any directory fill fail the check
3085 try:
3086 if not _check_dir(dir):
3087 return False
3088 except Exception as e:
3089 # here expected only timeout error, all other will
3090 # be catch by 'ignore_status=True'
3091 logging.debug('Fail to check %s to write in it', dir)
3092 return False
3093 return True
Garry Wang1a493d82020-08-31 21:01:19 -07003094
Dana Goyettec172b172020-07-29 16:26:15 -07003095 def blocking_sync(self, freeze_for_reset=False):
3096 """Sync root device and internal device, via script.
3097
3098 The actual calls end up logged by the run() call, since they're printed
3099 to stdout/stderr in the script.
3100
3101 @param freeze_for_reset: if True, prepare for reset by blocking writes
3102 (only if enable_fs_sync_fsfreeze=True)
3103 """
3104
3105 if freeze_for_reset and self.USE_FSFREEZE:
3106 logging.info('Blocking sync and freeze')
3107 elif freeze_for_reset:
3108 logging.info('Blocking sync for reset')
3109 else:
3110 logging.info('Blocking sync')
3111
3112 # client/bin is installed on the DUT as /usr/local/autotest/bin
3113 sync_cmd = '/usr/local/autotest/bin/fs_sync.py'
3114 if freeze_for_reset and self.USE_FSFREEZE:
3115 sync_cmd += ' --freeze'
3116 return self.run(sync_cmd)
3117
Garry Wanga2e78172020-09-09 23:49:07 -07003118 def set_health_profile_dut_state(self, state):
3119 if not self.health_profile:
3120 logging.debug('Device health profile is not initialized, skip'
3121 ' set dut state.')
3122 return
3123 reset_counters = state in profile_constants.STATES_NEED_RESET_COUNTER
3124 self.health_profile.update_dut_state(state, reset_counters)
Garry Wang53fc8f32020-09-18 13:30:08 -07003125
3126 def require_snk_mode_in_recovery(self):
3127 """Check whether we need to switch servo_v4 role to snk when
3128 booting into recovery mode. (See crbug.com/1129165)
3129 """
Garry Wanga8739cc2020-10-30 00:49:23 -07003130 has_battery = True
3131 # Determine if the host has battery based on host_info first.
3132 power_info = self.host_info_store.get().get_label_value('power')
3133 if power_info:
3134 has_battery = power_info == 'battery'
3135 elif self.is_up_fast():
3136 # when running local tests host_info is not available, so we
3137 # need to determine whether the host has battery by checking
3138 # from host side.
3139 logging.debug('Label `power` is not found in host_info, checking'
3140 ' if the host has battery from host side.')
3141 has_battery = self.has_battery()
3142
3143 if not has_battery:
Garry Wang53fc8f32020-09-18 13:30:08 -07003144 logging.info(
3145 '%s does not has battery, snk mode is not needed'
3146 ' for recovery.', self.hostname)
3147 return False
Garry Wanga8739cc2020-10-30 00:49:23 -07003148
Garry Wang53fc8f32020-09-18 13:30:08 -07003149 if not self.servo.supports_built_in_pd_control():
3150 logging.info('Power delivery is not supported on this servo, snk'
3151 ' mode is not needed for recovery.')
3152 return False
3153 try:
Garry Wang53fc8f32020-09-18 13:30:08 -07003154 battery_percent = self.servo.get('battery_charge_percent')
Otabek Kasimov58e22562020-11-03 17:17:41 -08003155 if battery_percent < cros_constants.MIN_BATTERY_LEVEL:
Garry Wang53fc8f32020-09-18 13:30:08 -07003156 logging.info(
3157 'Current battery level %s%% below %s%% threshold, we'
3158 ' will attempt to boot host in recovery mode without'
3159 ' changing servo to snk mode. Please note the host may'
3160 ' not able to see usb drive in recovery mode later due'
3161 ' to servo not in snk mode.', battery_percent,
Otabek Kasimov58e22562020-11-03 17:17:41 -08003162 cros_constants.MIN_BATTERY_LEVEL)
Garry Wang53fc8f32020-09-18 13:30:08 -07003163 return False
3164 except Exception as e:
3165 logging.info(
3166 'Unexpected error occurred when getting'
3167 ' battery_charge_percent from servo; %s', str(e))
3168 return False
3169 return True
Otabek Kasimov382c3bb2020-10-28 13:22:45 -07003170
3171 def _set_servo_topology(self):
3172 """Set servo-topology info to the host-info."""
3173 logging.debug('Try to save servo topology to host-info.')
3174 if not self._servo_host:
3175 logging.info('Servo host is not initilized.')
3176 return
3177 if not self._servo_host.is_servo_topology_supported():
3178 logging.info('Servo-topology is not supported.')
3179 return
3180 servo_topology = self._servo_host.get_topology()
3181 if not servo_topology or servo_topology.is_empty():
3182 logging.info('Servo topology is empty')
3183 return
3184 servo_topology.save(self.host_info_store)