blob: 86ab5bab54830dcf08a850e7cb32a88598c11a79 [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
Dan Shia1ecd5c2013-06-06 11:21:31 -070030from autotest_lib.server import utils as server_utils
Dan Shi9cb0eec2014-06-03 09:04:50 -070031from autotest_lib.server.cros import provision
Scott Zawalski89c44dd2013-02-26 09:28:02 -050032from autotest_lib.server.cros.dynamic_suite import constants as ds_constants
Derek Beckettcc0c8302021-07-14 09:53:48 -070033from autotest_lib.server.cros.dynamic_suite import tools
Garry Wang1a493d82020-08-31 21:01:19 -070034from autotest_lib.server.cros.device_health_profile import device_health_profile
Garry Wanga2e78172020-09-09 23:49:07 -070035from autotest_lib.server.cros.device_health_profile import profile_constants
Wai-Hong Tam0ed9fe12019-05-23 11:00:58 -070036from autotest_lib.server.cros.servo import pdtester
Fang Deng96667ca2013-08-01 17:46:18 -070037from autotest_lib.server.hosts import abstract_ssh
Kevin Chenga2619dc2016-03-28 11:42:08 -070038from autotest_lib.server.hosts import base_label
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +080039from autotest_lib.server.hosts import chameleon_host
Otabek Kasimov832d9162020-07-27 19:24:57 -070040from autotest_lib.server.hosts import cros_constants
Richard Barnetted31580e2018-05-14 19:58:00 +000041from autotest_lib.server.hosts import cros_label
J. Richard Barnettea7e7fdc2016-02-12 12:35:36 -080042from autotest_lib.server.hosts import cros_repair
Wai-Hong Tam0ed9fe12019-05-23 11:00:58 -070043from autotest_lib.server.hosts import pdtester_host
Fang Deng5d518f42013-08-02 14:04:32 -070044from autotest_lib.server.hosts import servo_host
Garry Wang11b5e872020-03-11 15:14:08 -070045from autotest_lib.server.hosts import servo_constants
Simran Basidcff4252012-11-20 16:13:20 -080046from autotest_lib.site_utils.rpm_control_system import rpm_client
Otabek Kasimov808cd832020-05-28 18:27:46 -070047from autotest_lib.site_utils.admin_audit import constants as audit_const
Otabek Kasimov27bb2862020-08-10 14:40:45 -070048from autotest_lib.site_utils.admin_audit import verifiers as audit_verify
Derek Beckettf73baca2020-08-19 15:08:47 -070049from six.moves import zip
Simran Basid5e5e272012-09-24 15:23:59 -070050
Andrew Luo4be621d2020-03-21 07:01:13 -070051
Dan Shib8540a52015-07-16 14:18:23 -070052CONFIG = global_config.global_config
53
beepsc87ff602013-07-31 21:53:00 -070054class FactoryImageCheckerException(error.AutoservError):
55 """Exception raised when an image is a factory image."""
56 pass
57
58
Fang Deng0ca40e22013-08-27 17:47:44 -070059class CrosHost(abstract_ssh.AbstractSSHHost):
J. Richard Barnette45e93de2012-04-11 17:24:15 -070060 """Chromium OS specific subclass of Host."""
61
Simran Basi5ace6f22016-01-06 17:30:44 -080062 VERSION_PREFIX = provision.CROS_VERSION_PREFIX
63
J. Richard Barnette45e93de2012-04-11 17:24:15 -070064
Richard Barnette03a0c132012-11-05 12:40:35 -080065 # Timeout values (in seconds) associated with various Chrome OS
66 # state changes.
J. Richard Barnette134ec2c2012-04-25 12:59:37 -070067 #
Richard Barnette0c73ffc2012-11-19 15:21:18 -080068 # In general, a good rule of thumb is that the timeout can be up
69 # to twice the typical measured value on the slowest platform.
70 # The times here have not necessarily been empirically tested to
71 # meet this criterion.
J. Richard Barnetteeb69d722012-06-18 17:29:44 -070072 #
73 # SLEEP_TIMEOUT: Time to allow for suspend to memory.
Richard Barnette0c73ffc2012-11-19 15:21:18 -080074 # RESUME_TIMEOUT: Time to allow for resume after suspend, plus
75 # time to restart the netwowrk.
J. Richard Barnette84890bd2014-02-21 11:05:47 -080076 # SHUTDOWN_TIMEOUT: Time to allow for shut down.
J. Richard Barnetteeb69d722012-06-18 17:29:44 -070077 # BOOT_TIMEOUT: Time to allow for boot from power off. Among
Richard Barnette0c73ffc2012-11-19 15:21:18 -080078 # other things, this must account for the 30 second dev-mode
J. Richard Barnette417cc792015-10-01 09:56:36 -070079 # screen delay, time to start the network on the DUT, and the
80 # ssh timeout of 120 seconds.
J. Richard Barnetteeb69d722012-06-18 17:29:44 -070081 # USB_BOOT_TIMEOUT: Time to allow for boot from a USB device,
Richard Barnette0c73ffc2012-11-19 15:21:18 -080082 # including the 30 second dev-mode delay and time to start the
J. Richard Barnetted4649c62013-03-06 17:42:27 -080083 # network.
beepsf079cfb2013-09-18 17:49:51 -070084 # INSTALL_TIMEOUT: Time to allow for chromeos-install.
Otabek Kasimovaeb47fe2021-01-26 20:53:55 -080085 # ADMIN_INSTALL_TIMEOUT: Time to allow for chromeos-install
86 # used by admin tasks.
J. Richard Barnette84890bd2014-02-21 11:05:47 -080087 # POWERWASH_BOOT_TIMEOUT: Time to allow for a reboot that
88 # includes powerwash.
J. Richard Barnetteeb69d722012-06-18 17:29:44 -070089
90 SLEEP_TIMEOUT = 2
J. Richard Barnetted4649c62013-03-06 17:42:27 -080091 RESUME_TIMEOUT = 10
Tom Wai-Hong Tamfe005c22014-12-03 09:25:44 +080092 SHUTDOWN_TIMEOUT = 10
J. Richard Barnette417cc792015-10-01 09:56:36 -070093 BOOT_TIMEOUT = 150
J. Richard Barnette5bab5f52015-08-03 13:14:38 -070094 USB_BOOT_TIMEOUT = 300
J. Richard Barnette7817b052014-08-28 09:47:29 -070095 INSTALL_TIMEOUT = 480
Otabek Kasimovaeb47fe2021-01-26 20:53:55 -080096 ADMIN_INSTALL_TIMEOUT = 600
Dan Shi2c88eed2013-11-12 10:18:38 -080097 POWERWASH_BOOT_TIMEOUT = 60
Chris Sosab76e0ee2013-05-22 16:55:41 -070098
Dan Shica503482015-03-30 17:23:25 -070099 # Minimum OS version that supports server side packaging. Older builds may
100 # not have server side package built or with Autotest code change to support
101 # server-side packaging.
Dan Shib8540a52015-07-16 14:18:23 -0700102 MIN_VERSION_SUPPORT_SSP = CONFIG.get_config_value(
Dan Shiced09e42015-04-17 16:09:34 -0700103 'AUTOSERV', 'min_version_support_ssp', type=int)
Dan Shica503482015-03-30 17:23:25 -0700104
Dana Goyettec172b172020-07-29 16:26:15 -0700105 USE_FSFREEZE = CONFIG.get_config_value(
Dana Goyette6242cb32020-09-23 11:02:57 -0700106 'CROS', 'enable_fs_freeze', type=bool, default=False)
Dana Goyettec172b172020-07-29 16:26:15 -0700107
J. Richard Barnette84890bd2014-02-21 11:05:47 -0800108 # REBOOT_TIMEOUT: How long to wait for a reboot.
109 #
Chris Sosab76e0ee2013-05-22 16:55:41 -0700110 # We have a long timeout to ensure we don't flakily fail due to other
111 # issues. Shorter timeouts are vetted in platform_RebootAfterUpdate.
Simran Basi1160e2c2013-10-04 16:00:24 -0700112 # TODO(sbasi - crbug.com/276094) Restore to 5 mins once the 'host did not
113 # return from reboot' bug is solved.
114 REBOOT_TIMEOUT = 480
Chris Sosab76e0ee2013-05-22 16:55:41 -0700115
Ismail Noorbasha07fdb612013-02-14 14:13:31 -0800116 # _USB_POWER_TIMEOUT: Time to allow for USB to power toggle ON and OFF.
117 # _POWER_CYCLE_TIMEOUT: Time to allow for manual power cycle.
Garry Wang5e5538a2019-04-08 15:36:18 -0700118 # _CHANGE_SERVO_ROLE_TIMEOUT: Time to allow DUT regain network connection
119 # since changing servo role will reset USB state
120 # and causes temporary ethernet drop.
Ismail Noorbasha07fdb612013-02-14 14:13:31 -0800121 _USB_POWER_TIMEOUT = 5
122 _POWER_CYCLE_TIMEOUT = 10
Garry Wang5e5538a2019-04-08 15:36:18 -0700123 _CHANGE_SERVO_ROLE_TIMEOUT = 180
Ismail Noorbasha07fdb612013-02-14 14:13:31 -0800124
Fang Dengdeba14f2014-11-14 11:54:09 -0800125 _RPM_HOSTNAME_REGEX = ('chromeos(\d+)(-row(\d+))?-rack(\d+[a-z]*)'
126 '-host(\d+)')
J. Richard Barnette134ec2c2012-04-25 12:59:37 -0700127
J. Richard Barnetteb6de7e32013-02-14 13:28:04 -0800128 # Constants used in ping_wait_up() and ping_wait_down().
129 #
130 # _PING_WAIT_COUNT is the approximate number of polling
131 # cycles to use when waiting for a host state change.
132 #
133 # _PING_STATUS_DOWN and _PING_STATUS_UP are names used
134 # for arguments to the internal _ping_wait_for_status()
135 # method.
136 _PING_WAIT_COUNT = 40
137 _PING_STATUS_DOWN = False
138 _PING_STATUS_UP = True
139
Ismail Noorbasha07fdb612013-02-14 14:13:31 -0800140 # Allowed values for the power_method argument.
141
Garry Wang5e5538a2019-04-08 15:36:18 -0700142 # POWER_CONTROL_RPM: Used in power_off/on/cycle() methods, default for all
143 # DUTs except those with servo_v4 CCD.
144 # POWER_CONTROL_CCD: Used in power_off/on/cycle() methods, default for all
145 # DUTs with servo_v4 CCD.
Ismail Noorbasha07fdb612013-02-14 14:13:31 -0800146 # POWER_CONTROL_SERVO: Used in set_power() and power_cycle() methods.
147 # POWER_CONTROL_MANUAL: Used in set_power() and power_cycle() methods.
148 POWER_CONTROL_RPM = 'RPM'
Garry Wang5e5538a2019-04-08 15:36:18 -0700149 POWER_CONTROL_CCD = 'CCD'
Ismail Noorbasha07fdb612013-02-14 14:13:31 -0800150 POWER_CONTROL_SERVO = 'servoj10'
151 POWER_CONTROL_MANUAL = 'manual'
152
153 POWER_CONTROL_VALID_ARGS = (POWER_CONTROL_RPM,
Garry Wang5e5538a2019-04-08 15:36:18 -0700154 POWER_CONTROL_CCD,
Ismail Noorbasha07fdb612013-02-14 14:13:31 -0800155 POWER_CONTROL_SERVO,
156 POWER_CONTROL_MANUAL)
Richard Barnette0c73ffc2012-11-19 15:21:18 -0800157
Simran Basi5e6339a2013-03-21 11:34:32 -0700158 _RPM_OUTLET_CHANGED = 'outlet_changed'
159
Dan Shi9cb0eec2014-06-03 09:04:50 -0700160 # URL pattern to download firmware image.
Dan Shib8540a52015-07-16 14:18:23 -0700161 _FW_IMAGE_URL_PATTERN = CONFIG.get_config_value(
Dan Shi9cb0eec2014-06-03 09:04:50 -0700162 'CROS', 'firmware_url_pattern', type=str)
beeps687243d2013-07-18 15:29:27 -0700163
Brent Peterson1cb623a2020-01-09 13:14:28 -0800164 # Regular expression for extracting EC version string
165 _EC_REGEX = '(%s_\w*[-\.]\w*[-\.]\w*[-\.]\w*)'
166
167 # Regular expression for extracting BIOS version string
168 _BIOS_REGEX = '(%s\.\w*\.\w*\.\w*)'
169
Brent Petersonc70a1832020-01-24 15:54:35 -0800170 # Command to update firmware located on DUT
Namyoon Woo382e5892020-05-20 16:48:40 -0700171 _FW_UPDATE_CMD = 'chromeos-firmwareupdate --mode=recovery %s'
Brent Petersonc70a1832020-01-24 15:54:35 -0800172
J. Richard Barnette964fba02012-10-24 17:34:29 -0700173 @staticmethod
beeps46dadc92013-11-07 14:07:10 -0800174 def check_host(host, timeout=10):
175 """
176 Check if the given host is a chrome-os host.
177
178 @param host: An ssh host representing a device.
179 @param timeout: The timeout for the run command.
180
181 @return: True if the host device is chromeos.
182
beeps46dadc92013-11-07 14:07:10 -0800183 """
184 try:
Allen Liad719c12017-06-27 23:48:04 +0000185 result = host.run(
Simran Basi933c8af2015-04-29 14:05:07 -0700186 'grep -q CHROMEOS /etc/lsb-release && '
Garry Wange4b6d6e2019-06-17 17:08:46 -0700187 '! grep -q moblab /etc/lsb-release && '
Derek Beckett342e3e62021-01-05 17:17:23 -0800188 '! grep -q labstation /etc/lsb-release &&'
189 ' grep CHROMEOS_RELEASE_BOARD /etc/lsb-release',
190 ignore_status=True,
Laurence Goodby468de252017-06-08 17:22:53 -0700191 timeout=timeout).stdout
Derek Beckett342e3e62021-01-05 17:17:23 -0800192 if result:
Pradeep Sawlaniaa013fa2017-11-09 06:34:18 -0800193 return not (
194 lsbrelease_utils.is_jetstream(
Derek Beckett342e3e62021-01-05 17:17:23 -0800195 lsb_release_content=result) or
Pradeep Sawlaniaa013fa2017-11-09 06:34:18 -0800196 lsbrelease_utils.is_gce_board(
Derek Beckett342e3e62021-01-05 17:17:23 -0800197 lsb_release_content=result))
Pradeep Sawlaniaa013fa2017-11-09 06:34:18 -0800198
beeps46dadc92013-11-07 14:07:10 -0800199 except (error.AutoservRunError, error.AutoservSSHTimeout):
200 return False
Laurence Goodby468de252017-06-08 17:22:53 -0700201
202 return False
beeps46dadc92013-11-07 14:07:10 -0800203
204
205 @staticmethod
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +0800206 def get_chameleon_arguments(args_dict):
207 """Extract chameleon options from `args_dict` and return the result.
208
209 Recommended usage:
210 ~~~~~~~~
211 args_dict = utils.args_to_dict(args)
212 chameleon_args = hosts.CrosHost.get_chameleon_arguments(args_dict)
213 host = hosts.create_host(machine, chameleon_args=chameleon_args)
214 ~~~~~~~~
215
216 @param args_dict Dictionary from which to extract the chameleon
217 arguments.
218 """
Sam McNally66594ca2019-12-09 12:45:44 +1100219 chameleon_args = {key: args_dict[key]
220 for key in ('chameleon_host', 'chameleon_port')
221 if key in args_dict}
222 if 'chameleon_ssh_port' in args_dict:
223 chameleon_args['port'] = int(args_dict['chameleon_ssh_port'])
224 return chameleon_args
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +0800225
Shijin Abrahamc09587d2020-02-14 20:46:55 -0800226 @staticmethod
227 def get_btpeer_arguments(args_dict):
228 """Extract btpeer options from `args_dict` and return the result.
229
230 This is used to parse details of Bluetooth peer.
231 Recommended usage:
232 ~~~~~~~~
233 args_dict = utils.args_to_dict(args)
234 btpeer_args = hosts.CrosHost.get_btpeer_arguments(args_dict)
Shijin Abraham22f24cb2020-03-26 09:07:41 -0700235 host = hosts.create_host(machine, btpeer_args=btpeer_args)
Shijin Abrahamc09587d2020-02-14 20:46:55 -0800236 ~~~~~~~~
237
238 @param args_dict: Dictionary from which to extract the btpeer
239 arguments.
240 """
241 if 'btpeer_host_list' in args_dict:
242 result = []
243 for btpeer in args_dict['btpeer_host_list'].split(','):
Claire Changd0b19842020-11-04 22:28:45 +0800244 # IPv6 addresses including a port number should be enclosed in
245 # square brackets.
246 delimiter = ']:' if re.search(r':.*:', btpeer) else ':'
Shijin Abrahamc09587d2020-02-14 20:46:55 -0800247 result.append({key: value for key,value in
248 zip(('btpeer_host','btpeer_port'),
Claire Changd0b19842020-11-04 22:28:45 +0800249 btpeer.strip('[]').split(delimiter))})
Shijin Abrahamc09587d2020-02-14 20:46:55 -0800250 return result
251 else:
Anand K Mistrye8933092020-08-05 14:49:41 +1000252 return {key: args_dict[key]
253 for key in ('btpeer_host', 'btpeer_port', 'btpeer_ssh_port')
Shijin Abrahamc09587d2020-02-14 20:46:55 -0800254 if key in args_dict}
255
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +0800256
257 @staticmethod
Wai-Hong Tam0ed9fe12019-05-23 11:00:58 -0700258 def get_pdtester_arguments(args_dict):
Scottfe06ed82015-11-05 17:15:01 -0800259 """Extract chameleon options from `args_dict` and return the result.
260
261 Recommended usage:
262 ~~~~~~~~
263 args_dict = utils.args_to_dict(args)
Wai-Hong Tam0ed9fe12019-05-23 11:00:58 -0700264 pdtester_args = hosts.CrosHost.get_pdtester_arguments(args_dict)
265 host = hosts.create_host(machine, pdtester_args=pdtester_args)
Scottfe06ed82015-11-05 17:15:01 -0800266 ~~~~~~~~
267
Wai-Hong Tam0ed9fe12019-05-23 11:00:58 -0700268 @param args_dict Dictionary from which to extract the pdtester
Scottfe06ed82015-11-05 17:15:01 -0800269 arguments.
270 """
Allen Li083866b2016-08-18 10:07:10 -0700271 return {key: args_dict[key]
Wai-Hong Tam0ed9fe12019-05-23 11:00:58 -0700272 for key in ('pdtester_host', 'pdtester_port')
Allen Li083866b2016-08-18 10:07:10 -0700273 if key in args_dict}
Scottfe06ed82015-11-05 17:15:01 -0800274
275
276 @staticmethod
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +0800277 def get_servo_arguments(args_dict):
278 """Extract servo options from `args_dict` and return the result.
J. Richard Barnette7214e0b2013-02-06 15:20:49 -0800279
280 Recommended usage:
281 ~~~~~~~~
282 args_dict = utils.args_to_dict(args)
Fang Deng0ca40e22013-08-27 17:47:44 -0700283 servo_args = hosts.CrosHost.get_servo_arguments(args_dict)
J. Richard Barnette7214e0b2013-02-06 15:20:49 -0800284 host = hosts.create_host(machine, servo_args=servo_args)
285 ~~~~~~~~
286
287 @param args_dict Dictionary from which to extract the servo
288 arguments.
289 """
Garry Wang11b5e872020-03-11 15:14:08 -0700290 servo_attrs = (servo_constants.SERVO_HOST_ATTR,
Andrew Luo4be621d2020-03-21 07:01:13 -0700291 servo_constants.SERVO_HOST_SSH_PORT_ATTR,
Garry Wang11b5e872020-03-11 15:14:08 -0700292 servo_constants.SERVO_PORT_ATTR,
Otabek Kasimov382c3bb2020-10-28 13:22:45 -0700293 servo_constants.SERVO_SERIAL_ATTR,
Garry Wang11b5e872020-03-11 15:14:08 -0700294 servo_constants.SERVO_BOARD_ATTR,
295 servo_constants.SERVO_MODEL_ATTR)
Armando Miraglia2ac802e2017-08-15 14:54:47 +0200296 servo_args = {key: args_dict[key]
297 for key in servo_attrs
298 if key in args_dict}
299 return (
300 None
Garry Wang11b5e872020-03-11 15:14:08 -0700301 if servo_constants.SERVO_HOST_ATTR in servo_args
302 and not servo_args[servo_constants.SERVO_HOST_ATTR]
Armando Miraglia2ac802e2017-08-15 14:54:47 +0200303 else servo_args)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -0700304
J. Richard Barnette964fba02012-10-24 17:34:29 -0700305
Otabek Kasimov1b70e8d2020-12-30 13:51:00 -0800306 def _initialize(self,
307 hostname,
308 chameleon_args=None,
309 servo_args=None,
310 pdtester_args=None,
311 try_lab_servo=False,
312 try_servo_repair=False,
313 ssh_verbosity_flag='',
314 ssh_options='',
315 try_servo_recovery=False,
316 *args,
317 **dargs):
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +0800318 """Initialize superclasses, |self.chameleon|, and |self.servo|.
J. Richard Barnette67ccb872012-04-19 16:34:56 -0700319
Fang Denge545abb2014-12-30 18:43:47 -0800320 This method will attempt to create the test-assistant object
321 (chameleon/servo) when it is needed by the test. Check
322 the docstring of chameleon_host.create_chameleon_host and
323 servo_host.create_servo_host for how this is determined.
Fang Deng5d518f42013-08-02 14:04:32 -0700324
Fang Denge545abb2014-12-30 18:43:47 -0800325 @param hostname: Hostname of the dut.
326 @param chameleon_args: A dictionary that contains args for creating
327 a ChameleonHost. See chameleon_host for details.
328 @param servo_args: A dictionary that contains args for creating
329 a ServoHost object. See servo_host for details.
Richard Barnette9a26ad62016-06-10 12:03:08 -0700330 @param try_lab_servo: When true, indicates that an attempt should
331 be made to create a ServoHost for a DUT in
332 the test lab, even if not required by
333 `servo_args`. See servo_host for details.
334 @param try_servo_repair: If a servo host is created, check it
335 with `repair()` rather than `verify()`.
Fang Denge545abb2014-12-30 18:43:47 -0800336 See servo_host for details.
337 @param ssh_verbosity_flag: String, to pass to the ssh command to control
338 verbosity.
339 @param ssh_options: String, other ssh options to pass to the ssh
340 command.
Otabek Kasimov1b70e8d2020-12-30 13:51:00 -0800341 @param try_servo_recovery: When True, start servod in recovery mode.
342 See servo_host for details.
J. Richard Barnette67ccb872012-04-19 16:34:56 -0700343 """
Andrew Luo4be621d2020-03-21 07:01:13 -0700344 super(CrosHost, self)._initialize(hostname=hostname, *args, **dargs)
J. Richard Barnette91137f02016-03-10 16:52:26 -0800345 self._repair_strategy = cros_repair.create_cros_repair_strategy()
Otabek Kasimov6825b762020-06-23 23:42:44 -0700346 # hold special dut_state for repair process
347 self._device_repair_state = None
Kevin Chenga2619dc2016-03-28 11:42:08 -0700348 self.labels = base_label.LabelRetriever(cros_label.CROS_LABELS)
J. Richard Barnettef0859852012-08-20 14:55:50 -0700349 # self.env is a dictionary of environment variable settings
350 # to be exported for commands run on the host.
351 # LIBC_FATAL_STDERR_ can be useful for diagnosing certain
352 # errors that might happen.
353 self.env['LIBC_FATAL_STDERR_'] = '1'
Fang Dengd1c2b732013-08-20 12:59:46 -0700354 self._ssh_verbosity_flag = ssh_verbosity_flag
Aviv Keshetc5947fa2013-09-04 14:06:29 -0700355 self._ssh_options = ssh_options
Garry Wang1a493d82020-08-31 21:01:19 -0700356 self.health_profile = None
Garry Wang5e5538a2019-04-08 15:36:18 -0700357 self._default_power_method = None
Otabek Kasimov39637412020-11-23 19:09:27 -0800358 dut_health_profile = device_health_profile.DeviceHealthProfile(
359 hostname=self.hostname,
360 host_info=self.host_info_store.get(),
361 result_dir=self.get_result_dir())
Otabek Kasimovb61729d2020-11-24 00:42:43 -0800362
363 # TODO(otabek@): remove when b/171414073 closed
Andrew Luo4be621d2020-03-21 07:01:13 -0700364 if self.use_icmp:
Derek Beckettc7677812021-02-12 14:41:11 -0800365 pingable_before_servo = self.is_up_fast(count=1)
Andrew Luo4be621d2020-03-21 07:01:13 -0700366 if pingable_before_servo:
367 logging.info('DUT is pingable before init Servo.')
368 else:
369 logging.info('Skipping ping to DUT before init Servo.')
Otabek Kasimov39637412020-11-23 19:09:27 -0800370 _servo_host, servo_state = servo_host.create_servo_host(
371 dut=self,
372 servo_args=servo_args,
373 try_lab_servo=try_lab_servo,
374 try_servo_repair=try_servo_repair,
Otabek Kasimov1b70e8d2020-12-30 13:51:00 -0800375 try_servo_recovery=try_servo_recovery,
Otabek Kasimov39637412020-11-23 19:09:27 -0800376 dut_host_info=self.host_info_store.get(),
377 dut_health_profile=dut_health_profile)
378 if dut_health_profile.is_loaded():
379 logging.info('Device health profile loaded.')
380 # The device profile is located in the servo_host which make it
381 # dependency. If profile is not loaded yet then we do not have it
382 # TODO(otabek@) persist device provide out of servo-host.
383 self.health_profile = dut_health_profile
384 self.set_servo_host(_servo_host, servo_state)
Richard Barnettee519dcd2016-08-15 17:37:17 -0700385
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +0800386 # TODO(waihong): Do the simplication on Chameleon too.
Shijin Abraham783a7dd2020-02-14 15:36:11 -0800387 self._chameleon_host = chameleon_host.create_chameleon_host(
Otabek Kasimova7ba91a2020-03-09 08:31:01 -0700388 dut=self.hostname,
389 chameleon_args=chameleon_args)
Shijin Abraham783a7dd2020-02-14 15:36:11 -0800390 if self._chameleon_host:
391 self.chameleon = self._chameleon_host.create_chameleon_board()
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +0800392 else:
Tom Wai-Hong Tam3d6790d2014-04-14 16:15:47 +0800393 self.chameleon = None
Fang Deng5d518f42013-08-02 14:04:32 -0700394
Shijin Abraham78ce4402020-09-08 22:04:27 -0700395 # Bluetooth peers will be populated by the test if needed
396 self._btpeer_host_list = []
397 self.btpeer_list = []
398 self.btpeer = None
Shijin Abrahamc09587d2020-02-14 20:46:55 -0800399
howardchung83e55272019-08-08 14:08:05 +0800400 # Add pdtester host if pdtester args were added on command line
Wai-Hong Tam16e5edb2019-09-17 16:10:07 -0700401 self._pdtester_host = pdtester_host.create_pdtester_host(
Wai-Hong Tam90b164d2019-10-25 13:15:39 -0700402 pdtester_args, self._servo_host)
howardchung83e55272019-08-08 14:08:05 +0800403
Wai-Hong Tam0ed9fe12019-05-23 11:00:58 -0700404 if self._pdtester_host:
405 self.pdtester_servo = self._pdtester_host.get_servo()
406 logging.info('pdtester_servo: %r', self.pdtester_servo)
407 # Create the pdtester object used to access the ec uart
408 self.pdtester = pdtester.PDTester(self.pdtester_servo,
409 self._pdtester_host.get_servod_server_proxy())
Scottfe06ed82015-11-05 17:15:01 -0800410 else:
Wai-Hong Tam0ed9fe12019-05-23 11:00:58 -0700411 self.pdtester = None
Scottfe06ed82015-11-05 17:15:01 -0800412
Fang Deng5d518f42013-08-02 14:04:32 -0700413
Shijin Abraham78ce4402020-09-08 22:04:27 -0700414 def initialize_btpeer(self, btpeer_args=[]):
Shijin Abrahamc09587d2020-02-14 20:46:55 -0800415 """ Initialize the Bluetooth peers
416
417 Initialize Bluetooth peer devices given in the arguments. Bluetooth peer
418 is chameleon host on Raspberry Pi.
419 @param btpeer_args: A dictionary that contains args for creating
420 a ChameleonHost. See chameleon_host for details.
421
422 """
Shijin Abraham78ce4402020-09-08 22:04:27 -0700423 logging.debug('Attempting to initialize bluetooth peers if available')
Shijin Abraham22f24cb2020-03-26 09:07:41 -0700424 try:
Shijin Abraham22f24cb2020-03-26 09:07:41 -0700425 if type(btpeer_args) is list:
426 btpeer_args_list = btpeer_args
427 else:
428 btpeer_args_list = [btpeer_args]
429
430 self._btpeer_host_list = chameleon_host.create_btpeer_host(
431 dut=self.hostname, btpeer_args_list=btpeer_args_list)
432 logging.debug('Bluetooth peer hosts are %s',
433 self._btpeer_host_list)
434 self.btpeer_list = [_host.create_chameleon_board() for _host in
435 self._btpeer_host_list if _host is not None]
436
437 if len(self.btpeer_list) > 0:
438 self.btpeer = self.btpeer_list[0]
439
440 logging.debug('After initialize_btpeer btpeer_list %s '
441 'btpeer_host_list is %s and btpeer is %s',
442 self.btpeer_list, self._btpeer_host_list,
443 self.btpeer)
444 except Exception as e:
445 logging.error('Exception %s in initialize_btpeer', str(e))
446
Shijin Abrahamc09587d2020-02-14 20:46:55 -0800447
Rohit Makasanadf0a3a32017-06-30 13:55:18 -0700448 def host_version_prefix(self, image):
449 """Return version label prefix.
450
451 In case the CrOS provisioning version is something other than the
452 standard CrOS version e.g. CrOS TH version, this function will
453 find the prefix from provision.py.
454
455 @param image: The image name to find its version prefix.
456 @returns: A prefix string for the image type.
457 """
458 return provision.get_version_label_prefix(image)
459
Andrew Luo3332ab22020-04-28 16:42:03 -0700460 def stage_build_to_usb(self, build):
461 """Stage the current ChromeOS image on the USB stick connected to the
462 servo.
463
464 @param build: The build to download and send to USB.
465 """
466 if not self.servo:
467 raise error.TestError('Host %s does not have servo.' %
468 self.hostname)
469
470 _, update_url = self.stage_image_for_servo(build)
Andrew Luob0355ea2020-06-24 16:12:57 -0700471
472 try:
473 self.servo.image_to_servo_usb(update_url)
474 finally:
475 # servo.image_to_servo_usb turned the DUT off, so turn it back on
476 logging.debug('Turn DUT power back on.')
477 self.servo.get_power_state_controller().power_on()
478
Andrew Luo3332ab22020-04-28 16:42:03 -0700479 logging.debug('ChromeOS image %s is staged on the USB stick.',
480 build)
Rohit Makasanadf0a3a32017-06-30 13:55:18 -0700481
beepsdae65fd2013-07-26 16:24:41 -0700482 def verify_job_repo_url(self, tag=''):
beepscb6f1e22013-06-28 19:14:10 -0700483 """
484 Make sure job_repo_url of this host is valid.
485
joychen03eaad92013-06-26 09:55:21 -0700486 Eg: The job_repo_url "http://lmn.cd.ab.xyx:8080/static/\
beepscb6f1e22013-06-28 19:14:10 -0700487 lumpy-release/R29-4279.0.0/autotest/packages" claims to have the
488 autotest package for lumpy-release/R29-4279.0.0. If this isn't the case,
489 download and extract it. If the devserver embedded in the url is
490 unresponsive, update the job_repo_url of the host after staging it on
491 another devserver.
492
493 @param job_repo_url: A url pointing to the devserver where the autotest
494 package for this build should be staged.
beepsdae65fd2013-07-26 16:24:41 -0700495 @param tag: The tag from the server job, in the format
496 <job_id>-<user>/<hostname>, or <hostless> for a server job.
beepscb6f1e22013-06-28 19:14:10 -0700497
498 @raises DevServerException: If we could not resolve a devserver.
499 @raises AutoservError: If we're unable to save the new job_repo_url as
500 a result of choosing a new devserver because the old one failed to
501 respond to a health check.
beeps0c865032013-07-30 11:37:06 -0700502 @raises urllib2.URLError: If the devserver embedded in job_repo_url
503 doesn't respond within the timeout.
beepscb6f1e22013-06-28 19:14:10 -0700504 """
Prathmesh Prabhu368abdf2017-02-14 11:23:47 -0800505 info = self.host_info_store.get()
506 job_repo_url = info.attributes.get(ds_constants.JOB_REPO_URL, '')
beepscb6f1e22013-06-28 19:14:10 -0700507 if not job_repo_url:
508 logging.warning('No job repo url set on host %s', self.hostname)
509 return
510
511 logging.info('Verifying job repo url %s', job_repo_url)
512 devserver_url, image_name = tools.get_devserver_build_from_package_url(
513 job_repo_url)
514
beeps0c865032013-07-30 11:37:06 -0700515 ds = dev_server.ImageServer(devserver_url)
beepscb6f1e22013-06-28 19:14:10 -0700516
517 logging.info('Staging autotest artifacts for %s on devserver %s',
518 image_name, ds.url())
beeps687243d2013-07-18 15:29:27 -0700519
520 start_time = time.time()
Simran Basi25e7a922014-10-31 11:56:10 -0700521 ds.stage_artifacts(image_name, ['autotest_packages'])
beeps687243d2013-07-18 15:29:27 -0700522 stage_time = time.time() - start_time
523
524 # Record how much of the verification time comes from a devserver
525 # restage. If we're doing things right we should not see multiple
526 # devservers for a given board/build/branch path.
527 try:
J. Richard Barnette3cbd76b2013-11-27 12:11:25 -0800528 board, build_type, branch = server_utils.ParseBuildName(
beeps687243d2013-07-18 15:29:27 -0700529 image_name)[:3]
J. Richard Barnette3cbd76b2013-11-27 12:11:25 -0800530 except server_utils.ParseBuildNameException:
beeps687243d2013-07-18 15:29:27 -0700531 pass
532 else:
beeps0c865032013-07-30 11:37:06 -0700533 devserver = devserver_url[
Chris Sosa65425082013-10-16 13:26:22 -0700534 devserver_url.find('/') + 2:devserver_url.rfind(':')]
beeps687243d2013-07-18 15:29:27 -0700535 stats_key = {
536 'board': board,
537 'build_type': build_type,
538 'branch': branch,
beeps0c865032013-07-30 11:37:06 -0700539 'devserver': devserver.replace('.', '_'),
beeps687243d2013-07-18 15:29:27 -0700540 }
Dan Shi5e2efb72017-02-07 11:40:23 -0800541
542 monarch_fields = {
543 'board': board,
544 'build_type': build_type,
Dan Shi5e2efb72017-02-07 11:40:23 -0800545 'branch': branch,
546 'dev_server': devserver,
547 }
Dan Shi5e2efb72017-02-07 11:40:23 -0800548
Dan Shicf4d2032015-03-12 15:04:21 -0700549 def stage_server_side_package(self, image=None):
550 """Stage autotest server-side package on devserver.
551
552 @param image: Full path of an OS image to install or a build name.
553
554 @return: A url to the autotest server-side package.
Dan Shi14de7622016-08-22 11:09:06 -0700555
556 @raise: error.AutoservError if fail to locate the build to test with, or
557 fail to stage server-side package.
Dan Shicf4d2032015-03-12 15:04:21 -0700558 """
Dan Shid37736b2016-07-06 15:10:29 -0700559 # If enable_drone_in_restricted_subnet is False, do not set hostname
560 # in devserver.resolve call, so a devserver in non-restricted subnet
561 # is picked to stage autotest server package for drone to download.
562 hostname = self.hostname
563 if not server_utils.ENABLE_DRONE_IN_RESTRICTED_SUBNET:
564 hostname = None
Dan Shicf4d2032015-03-12 15:04:21 -0700565 if image:
566 image_name = tools.get_build_from_image(image)
567 if not image_name:
568 raise error.AutoservError(
569 'Failed to parse build name from %s' % image)
Dan Shid37736b2016-07-06 15:10:29 -0700570 ds = dev_server.ImageServer.resolve(image_name, hostname)
Dan Shicf4d2032015-03-12 15:04:21 -0700571 else:
Prathmesh Prabhu9235e4c2017-03-28 13:16:06 -0700572 info = self.host_info_store.get()
Prathmesh Prabhu368abdf2017-02-14 11:23:47 -0800573 job_repo_url = info.attributes.get(ds_constants.JOB_REPO_URL, '')
Dan Shicf4d2032015-03-12 15:04:21 -0700574 if job_repo_url:
575 devserver_url, image_name = (
576 tools.get_devserver_build_from_package_url(job_repo_url))
Dan Shid37736b2016-07-06 15:10:29 -0700577 # If enable_drone_in_restricted_subnet is True, use the
578 # existing devserver. Otherwise, resolve a new one in
579 # non-restricted subnet.
580 if server_utils.ENABLE_DRONE_IN_RESTRICTED_SUBNET:
581 ds = dev_server.ImageServer(devserver_url)
582 else:
583 ds = dev_server.ImageServer.resolve(image_name)
Prathmesh Prabhub6cea612017-02-09 15:41:19 -0800584 elif info.build is not None:
585 ds = dev_server.ImageServer.resolve(info.build, hostname)
Prathmesh Prabhu0c1dd4d2017-06-07 13:01:53 -0700586 image_name = info.build
Dan Shicf4d2032015-03-12 15:04:21 -0700587 else:
Prathmesh Prabhub6cea612017-02-09 15:41:19 -0800588 raise error.AutoservError(
589 'Failed to stage server-side package. The host has '
Garry Wang12b9baf2019-06-24 18:58:54 -0700590 'no job_repo_url attribute or cros-version label.')
Dan Shica503482015-03-30 17:23:25 -0700591
592 # Get the OS version of the build, for any build older than
593 # MIN_VERSION_SUPPORT_SSP, server side packaging is not supported.
594 match = re.match('.*/R\d+-(\d+)\.', image_name)
595 if match and int(match.group(1)) < self.MIN_VERSION_SUPPORT_SSP:
Dan Shi14de7622016-08-22 11:09:06 -0700596 raise error.AutoservError(
597 'Build %s is older than %s. Server side packaging is '
598 'disabled.' % (image_name, self.MIN_VERSION_SUPPORT_SSP))
Dan Shica503482015-03-30 17:23:25 -0700599
Dan Shicf4d2032015-03-12 15:04:21 -0700600 ds.stage_artifacts(image_name, ['autotest_server_package'])
601 return '%s/static/%s/%s' % (ds.url(), image_name,
602 'autotest_server_package.tar.bz2')
603
604
Kalin Stoyanovb3c11f32018-05-11 09:02:00 -0700605 def stage_image_for_servo(self, image_name=None, artifact='test_image'):
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700606 """Stage a build on a devserver and return the update_url.
607
608 @param image_name: a name like lumpy-release/R27-3837.0.0
Kalin Stoyanovb3c11f32018-05-11 09:02:00 -0700609 @param artifact: a string like 'test_image'. Requests
610 appropriate image to be staged.
Xixuan Wufee57542019-10-15 11:50:27 -0700611 @returns a tuple of (image_name, URL) like
612 (lumpy-release/R27-3837.0.0,
613 http://172.22.50.205:8082/update/lumpy-release/R27-3837.0.0)
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700614 """
615 if not image_name:
Richard Barnetteb4b4d6f2018-05-05 01:47:41 +0000616 image_name = self.get_cros_repair_image_name()
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700617 logging.info('Staging build for servo install: %s', image_name)
Dan Shi216389c2015-12-22 11:03:06 -0800618 devserver = dev_server.ImageServer.resolve(image_name, self.hostname)
Kalin Stoyanovb3c11f32018-05-11 09:02:00 -0700619 devserver.stage_artifacts(image_name, [artifact])
620 if artifact == 'test_image':
Xixuan Wufee57542019-10-15 11:50:27 -0700621 return image_name, devserver.get_test_image_url(image_name)
Kalin Stoyanovb3c11f32018-05-11 09:02:00 -0700622 elif artifact == 'recovery_image':
Xixuan Wufee57542019-10-15 11:50:27 -0700623 return image_name, devserver.get_recovery_image_url(image_name)
Kalin Stoyanovb3c11f32018-05-11 09:02:00 -0700624 else:
625 raise error.AutoservError("Bad artifact!")
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700626
627
beepse539be02013-07-31 21:57:39 -0700628 def stage_factory_image_for_servo(self, image_name):
629 """Stage a build on a devserver and return the update_url.
630
631 @param image_name: a name like <baord>/4262.204.0
beeps12c0a3c2013-09-03 11:58:27 -0700632
beepse539be02013-07-31 21:57:39 -0700633 @return: An update URL, eg:
634 http://<devserver>/static/canary-channel/\
635 <board>/4262.204.0/factory_test/chromiumos_factory_image.bin
beeps12c0a3c2013-09-03 11:58:27 -0700636
637 @raises: ValueError if the factory artifact name is missing from
638 the config.
639
beepse539be02013-07-31 21:57:39 -0700640 """
641 if not image_name:
642 logging.error('Need an image_name to stage a factory image.')
643 return
644
Dan Shib8540a52015-07-16 14:18:23 -0700645 factory_artifact = CONFIG.get_config_value(
beeps12c0a3c2013-09-03 11:58:27 -0700646 'CROS', 'factory_artifact', type=str, default='')
647 if not factory_artifact:
648 raise ValueError('Cannot retrieve the factory artifact name from '
649 'autotest config, and hence cannot stage factory '
650 'artifacts.')
651
beepse539be02013-07-31 21:57:39 -0700652 logging.info('Staging build for servo install: %s', image_name)
Dan Shi216389c2015-12-22 11:03:06 -0800653 devserver = dev_server.ImageServer.resolve(image_name, self.hostname)
beepse539be02013-07-31 21:57:39 -0700654 devserver.stage_artifacts(
655 image_name,
beeps12c0a3c2013-09-03 11:58:27 -0700656 [factory_artifact],
657 archive_url=None)
beepse539be02013-07-31 21:57:39 -0700658
659 return tools.factory_image_url_pattern() % (devserver.url(), image_name)
660
661
Laurence Goodby778c9a42017-05-24 19:24:07 -0700662 def prepare_for_update(self):
663 """Prepares the DUT for an update.
664
665 Subclasses may override this to perform any special actions
666 required before updating.
667 """
Laurence Goodby468de252017-06-08 17:22:53 -0700668 pass
Laurence Goodby778c9a42017-05-24 19:24:07 -0700669
670
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +0800671 def _clear_fw_version_labels(self, rw_only):
672 """Clear firmware version labels from the machine.
673
674 @param rw_only: True to only clear fwrw_version; otherewise, clear
675 both fwro_version and fwrw_version.
676 """
Prathmesh Prabhuf8ae9a82019-10-04 15:34:13 -0700677 info = self.host_info_store.get()
678 info.clear_version_labels(provision.FW_RW_VERSION_PREFIX)
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +0800679 if not rw_only:
Prathmesh Prabhuf8ae9a82019-10-04 15:34:13 -0700680 info.clear_version_labels(provision.FW_RO_VERSION_PREFIX)
681 self.host_info_store.commit(info)
Dan Shi9cb0eec2014-06-03 09:04:50 -0700682
683
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +0800684 def _add_fw_version_label(self, build, rw_only):
Dan Shi9cb0eec2014-06-03 09:04:50 -0700685 """Add firmware version label to the machine.
686
687 @param build: Build of firmware.
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +0800688 @param rw_only: True to only add fwrw_version; otherwise, add both
689 fwro_version and fwrw_version.
Dan Shi9cb0eec2014-06-03 09:04:50 -0700690
691 """
Prathmesh Prabhuf8ae9a82019-10-04 15:34:13 -0700692 info = self.host_info_store.get()
693 info.set_version_label(provision.FW_RW_VERSION_PREFIX, build)
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +0800694 if not rw_only:
Prathmesh Prabhuf8ae9a82019-10-04 15:34:13 -0700695 info.set_version_label(provision.FW_RO_VERSION_PREFIX, build)
696 self.host_info_store.commit(info)
Dan Shi9cb0eec2014-06-03 09:04:50 -0700697
698
Namyoon Woo33f38852020-04-13 17:26:58 -0700699 def get_latest_release_version(self, platform, ref_board=None):
Namyoon Woo5f894662019-11-15 15:23:23 -0800700 """Search for the latest package release version from the image archive,
701 and return it.
702
Namyoon Woo33f38852020-04-13 17:26:58 -0700703 @param platform: platform name, a.k.a. board or model
704 @param ref_board: reference board name, a.k.a. baseboard, parent
Namyoon Woo5f894662019-11-15 15:23:23 -0800705
Namyoon Woo33f38852020-04-13 17:26:58 -0700706 @return 'firmware-{platform}-{branch}-firmwarebranch/{release-version}/'
707 '{platform}'
Namyoon Woo5f894662019-11-15 15:23:23 -0800708 or None if LATEST release file does not exist.
709 """
710
Namyoon Woo33f38852020-04-13 17:26:58 -0700711 platforms = [ platform ]
Namyoon Woo5f894662019-11-15 15:23:23 -0800712
Namyoon Woo33f38852020-04-13 17:26:58 -0700713 # Search the image path in reference board archive as well.
714 # For example, bob has its binary image under its reference board (gru)
715 # image archive.
716 if ref_board:
717 platforms.append(ref_board)
Namyoon Woo5f894662019-11-15 15:23:23 -0800718
Namyoon Woo33f38852020-04-13 17:26:58 -0700719 for board in platforms:
720 # Read 'LATEST-1.0.0' file
721 branch_dir = provision.FW_BRANCH_GLOB % board
722 latest_file = os.path.join(provision.CROS_IMAGE_ARCHIVE, branch_dir,
723 'LATEST-1.0.0')
Namyoon Woo406c7d42020-01-24 15:57:11 -0800724
Namyoon Woo33f38852020-04-13 17:26:58 -0700725 try:
726 # The result could be one or more.
727 result = utils.system_output('gsutil ls -d ' + latest_file)
728
729 candidates = re.findall('gs://.*', result)
730
731 # Found the directory candidates. No need to check the other
732 # board name cadidates. Let's break the loop.
733 break
734 except error.CmdError:
735 # It doesn't exist. Let's move on to the next item.
736 pass
737 else:
Namyoon Woo5f894662019-11-15 15:23:23 -0800738 logging.error('No LATEST release info is available.')
739 return None
740
Namyoon Woo406c7d42020-01-24 15:57:11 -0800741 for cand_dir in candidates:
742 result = utils.system_output('gsutil cat ' + cand_dir)
Namyoon Woo5f894662019-11-15 15:23:23 -0800743
Namyoon Woo406c7d42020-01-24 15:57:11 -0800744 release_path = cand_dir.replace('LATEST-1.0.0', result)
Namyoon Woo33f38852020-04-13 17:26:58 -0700745 release_path = os.path.join(release_path, platform)
Namyoon Woo406c7d42020-01-24 15:57:11 -0800746 try:
747 # Check if release_path does exist.
748 release = utils.system_output('gsutil ls -d ' + release_path)
749 # Now 'release' has a full directory path: e.g.
750 # gs://chromeos-image-archive/firmware-octopus-11297.B-
751 # firmwarebranch/RNone-1.0.0-b4395530/octopus/
752
753 # Remove "gs://chromeos-image-archive".
754 release = release.replace(provision.CROS_IMAGE_ARCHIVE, '')
755
756 # Remove CROS_IMAGE_ARCHIVE and any surrounding '/'s.
757 return release.strip('/')
758 except error.CmdError:
759 # The directory might not exist. Let's try next candidate.
760 pass
761 else:
762 raise error.AutoservError('Cannot find the latest firmware')
Namyoon Woo5f894662019-11-15 15:23:23 -0800763
Brent Peterson1cb623a2020-01-09 13:14:28 -0800764 @staticmethod
765 def get_version_from_image(image, version_regex):
Brent Peterson8039b472020-02-14 10:51:23 -0800766 """Get version string from binary image using regular expression.
767
768 @param image: Binary image to search
769 @param version_regex: Regular expression to search for
770
771 @return Version string
772
773 @raises TestFail if no version string is found in image
774 """
Brent Peterson1cb623a2020-01-09 13:14:28 -0800775 with open(image, 'rb') as f:
776 image_data = f.read()
Derek Beckett98345552020-08-31 16:07:22 -0700777 match = re.findall(version_regex,
778 image_data.decode('ISO-8859-1', errors='ignore'))
Brent Peterson1cb623a2020-01-09 13:14:28 -0800779 if match:
780 return match[0]
781 else:
782 raise error.TestFail('Failed to read version from %s.' % image)
783
784
Garry Wangad2a1712020-03-26 15:06:43 -0700785 def firmware_install(self, build, rw_only=False, dest=None,
Brent Petersonc70a1832020-01-24 15:54:35 -0800786 local_tarball=None, verify_version=False,
Namyoon Woo382e5892020-05-20 16:48:40 -0700787 try_scp=False, install_ec=True, install_bios=True,
788 board_as=None):
Dan Shi9cb0eec2014-06-03 09:04:50 -0700789 """Install firmware to the DUT.
790
791 Use stateful update if the DUT is already running the same build.
792 Stateful update does not update kernel and tends to run much faster
793 than a full reimage. If the DUT is running a different build, or it
794 failed to do a stateful update, full update, including kernel update,
795 will be applied to the DUT.
796
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +0800797 Once a host enters firmware_install its fw[ro|rw]_version label will
798 be removed. After the firmware is updated successfully, a new
799 fw[ro|rw]_version label will be added to the host.
Dan Shi9cb0eec2014-06-03 09:04:50 -0700800
801 @param build: The build version to which we want to provision the
802 firmware of the machine,
803 e.g. 'link-firmware/R22-2695.1.144'.
Tom Wai-Hong Tam4ac78982016-01-08 02:34:37 +0800804 @param rw_only: True to only install firmware to its RW portions. Keep
805 the RO portions unchanged.
Mary Ruthven6481a9f2019-08-23 12:46:05 -0700806 @param dest: Directory to store the firmware in.
Brent Peterson5b7c5b12020-01-09 13:14:28 -0800807 @param local_tarball: Path to local firmware image for installing
808 without devserver.
Brent Peterson1cb623a2020-01-09 13:14:28 -0800809 @param verify_version: True to verify EC and BIOS versions after
810 programming firmware, default is False.
Brent Petersonc70a1832020-01-24 15:54:35 -0800811 @param try_scp: False to always program using servo, true to try copying
812 the firmware and programming from the DUT.
Namyoon Woo382e5892020-05-20 16:48:40 -0700813 @param install_ec: True to install EC FW, and False to skip it.
814 @param install_bios: True to install BIOS, and False to skip it.
815 @param board_as: A board name to force to use.
Dan Shi9cb0eec2014-06-03 09:04:50 -0700816
817 TODO(dshi): After bug 381718 is fixed, update here with corresponding
818 exceptions that could be raised.
819
820 """
821 if not self.servo:
822 raise error.TestError('Host %s does not have servo.' %
823 self.hostname)
824
Wai-Hong Tam3fa455a2018-07-18 14:40:43 -0700825 info = self.host_info_store.get()
826 board = info.board
Shelley Chenac61d5a2019-06-24 15:35:46 -0700827 model = info.model
Namyoon Woo8dbfcf92019-01-15 18:37:12 -0800828
829 if board is None or board == '':
830 board = self.servo.get_board()
Dan Shi9cb0eec2014-06-03 09:04:50 -0700831
Namyoon Woo382e5892020-05-20 16:48:40 -0700832 # if board_as argument is passed, then use it instead of the original
833 # board name.
834 if board_as:
835 board = board_as
836
Mary Ruthvende14a8b2019-08-23 12:43:52 -0700837 if model is None or model == '':
Namyoon Woofb16eae2020-08-14 10:02:39 -0700838 try:
Mary Ruthven9c15a4e2020-10-23 10:10:53 -0700839 model = self.get_platform()
Namyoon Woofb16eae2020-08-14 10:02:39 -0700840 except Exception as e:
841 logging.warn('Dut is unresponsive: %s', str(e))
Mary Ruthvende14a8b2019-08-23 12:43:52 -0700842
Brent Peterson5b7c5b12020-01-09 13:14:28 -0800843 # If local firmware path not provided fetch it from the dev server
Mary Ruthven6481a9f2019-08-23 12:46:05 -0700844 tmpd = None
Brent Peterson5b7c5b12020-01-09 13:14:28 -0800845 if not local_tarball:
Garry Wangad2a1712020-03-26 15:06:43 -0700846 logging.info('Will install firmware from build %s.', build)
Brent Peterson5b7c5b12020-01-09 13:14:28 -0800847
Namyoon Woo43c1cb22020-05-28 18:58:34 -0700848 try:
849 ds = dev_server.ImageServer.resolve(build, self.hostname)
850 ds.stage_artifacts(build, ['firmware'])
Brent Peterson5b7c5b12020-01-09 13:14:28 -0800851
Namyoon Woo43c1cb22020-05-28 18:58:34 -0700852 if not dest:
853 tmpd = autotemp.tempdir(unique_id='fwimage')
854 dest = tmpd.name
Brent Peterson5b7c5b12020-01-09 13:14:28 -0800855
Namyoon Woo43c1cb22020-05-28 18:58:34 -0700856 # Download firmware image
857 fwurl = self._FW_IMAGE_URL_PATTERN % (ds.url(), build)
858 local_tarball = os.path.join(dest, os.path.basename(fwurl))
859 ds.download_file(fwurl, local_tarball)
860 except Exception as e:
861 raise error.TestError('Failed to download firmware package: %s'
862 % str(e))
Dan Shi9cb0eec2014-06-03 09:04:50 -0700863
Namyoon Woo382e5892020-05-20 16:48:40 -0700864 ec_image = None
865 if install_ec:
866 # Extract EC image from tarball
867 logging.info('Extracting EC image.')
868 ec_image = self.servo.extract_ec_image(board, model, local_tarball)
Mary Ruthven9c15a4e2020-10-23 10:10:53 -0700869 logging.info('Extracted: %s', ec_image)
Brent Peterson1cb623a2020-01-09 13:14:28 -0800870
Namyoon Woo382e5892020-05-20 16:48:40 -0700871 bios_image = None
872 if install_bios:
873 # Extract BIOS image from tarball
874 logging.info('Extracting BIOS image.')
875 bios_image = self.servo.extract_bios_image(board, model,
876 local_tarball)
Mary Ruthven9c15a4e2020-10-23 10:10:53 -0700877 logging.info('Extracted: %s', bios_image)
Namyoon Woo382e5892020-05-20 16:48:40 -0700878
879 if not bios_image and not ec_image:
880 raise error.TestError('No firmware installation was processed.')
Brent Peterson1cb623a2020-01-09 13:14:28 -0800881
Brent Petersonc70a1832020-01-24 15:54:35 -0800882 # Clear firmware version labels
883 self._clear_fw_version_labels(rw_only)
884
885 # Install firmware from local tarball
Brent Peterson5b7c5b12020-01-09 13:14:28 -0800886 try:
Garry Wang50e4a492020-08-05 12:29:57 -0700887 # Check if copying to DUT is enabled and DUT is available
888 if try_scp and self.is_up():
Brent Petersonc70a1832020-01-24 15:54:35 -0800889 # DUT is available, make temp firmware directory to store images
890 logging.info('Making temp folder.')
891 dest_folder = '/tmp/firmware'
892 self.run('mkdir -p ' + dest_folder)
893
Namyoon Woo68b68082020-06-02 13:13:14 -0700894 fw_cmd = self._FW_UPDATE_CMD % ('--wp=1' if rw_only else '')
Brent Petersonc70a1832020-01-24 15:54:35 -0800895
Namyoon Woo382e5892020-05-20 16:48:40 -0700896 if bios_image:
897 # Send BIOS firmware image to DUT
898 logging.info('Sending BIOS firmware.')
899 dest_bios_path = os.path.join(dest_folder,
900 os.path.basename(bios_image))
901 self.send_file(bios_image, dest_bios_path)
902
903 # Initialize firmware update command for BIOS image
904 fw_cmd += ' -i %s' % dest_bios_path
Brent Peterson669edf42020-02-07 15:07:54 -0800905
906 # Send EC firmware image to DUT when EC image was found
907 if ec_image:
908 logging.info('Sending EC firmware.')
909 dest_ec_path = os.path.join(dest_folder,
910 os.path.basename(ec_image))
911 self.send_file(ec_image, dest_ec_path)
912
913 # Add EC image to firmware update command
914 fw_cmd += ' -e %s' % dest_ec_path
915
Dana Goyettee84ef8d2020-07-09 11:55:09 -0700916 # Make sure command is allowed to finish even if ssh fails.
917 fw_cmd = "trap '' SIGHUP; %s" % fw_cmd
918
Brent Peterson669edf42020-02-07 15:07:54 -0800919 # Update firmware on DUT
920 logging.info('Updating firmware.')
Dana Goyettee84ef8d2020-07-09 11:55:09 -0700921 try:
Dana Goyette935b3fe2020-07-23 14:19:39 -0700922 self.run(fw_cmd, options="-o LogLevel=verbose")
Dana Goyettee84ef8d2020-07-09 11:55:09 -0700923 except error.AutoservRunError as e:
924 if e.result_obj.exit_status != 255:
925 raise
926 elif ec_image:
927 logging.warn("DUT network dropped during update"
928 " (often caused by EC resetting USB)")
929 else:
930 logging.error("DUT network dropped during update"
931 " (unexpected, since no EC image)")
932 raise
Brent Petersonc70a1832020-01-24 15:54:35 -0800933 else:
934 # Host is not available, program firmware using servo
Brent Peterson669edf42020-02-07 15:07:54 -0800935 if ec_image:
936 self.servo.program_ec(ec_image, rw_only)
Namyoon Woo382e5892020-05-20 16:48:40 -0700937 if bios_image:
938 self.servo.program_bios(bios_image, rw_only)
Brent Petersonc70a1832020-01-24 15:54:35 -0800939 if utils.host_is_in_lab_zone(self.hostname):
940 self._add_fw_version_label(build, rw_only)
Brent Peterson1cb623a2020-01-09 13:14:28 -0800941
942 # Reboot and wait for DUT after installing firmware
943 logging.info('Rebooting DUT.')
944 self.servo.get_power_state_controller().reset()
945 time.sleep(self.servo.BOOT_DELAY)
946 self.test_wait_for_boot()
947
948 # When enabled verify EC and BIOS firmware version after programming
949 if verify_version:
Brent Peterson669edf42020-02-07 15:07:54 -0800950 # Check programmed EC firmware when EC image was found
951 if ec_image:
952 logging.info('Checking EC firmware version.')
953 dest_ec_version = self.get_ec_version()
Brent Peterson8039b472020-02-14 10:51:23 -0800954 ec_version_prefix = dest_ec_version.split('_', 1)[0]
955 ec_regex = self._EC_REGEX % ec_version_prefix
Brent Peterson669edf42020-02-07 15:07:54 -0800956 image_ec_version = self.get_version_from_image(ec_image,
Brent Peterson8039b472020-02-14 10:51:23 -0800957 ec_regex)
Brent Peterson669edf42020-02-07 15:07:54 -0800958 if dest_ec_version != image_ec_version:
959 raise error.TestFail(
Namyoon Wooe2c009a2020-07-22 10:09:03 -0700960 'Failed to update EC firmware, version %s '
961 '(expected %s)' % (dest_ec_version,
962 image_ec_version))
Brent Peterson1cb623a2020-01-09 13:14:28 -0800963
Namyoon Woo382e5892020-05-20 16:48:40 -0700964 if bios_image:
965 # Check programmed BIOS firmware against expected version
966 logging.info('Checking BIOS firmware version.')
967 dest_bios_version = self.get_firmware_version()
968 bios_version_prefix = dest_bios_version.split('.', 1)[0]
969 bios_regex = self._BIOS_REGEX % bios_version_prefix
970 image_bios_version = self.get_version_from_image(bios_image,
971 bios_regex)
972 if dest_bios_version != image_bios_version:
973 raise error.TestFail(
Namyoon Wooe2c009a2020-07-22 10:09:03 -0700974 'Failed to update BIOS, version %s '
Namyoon Woo382e5892020-05-20 16:48:40 -0700975 '(expected %s)' % (dest_bios_version,
976 image_bios_version))
Dan Shi9cb0eec2014-06-03 09:04:50 -0700977 finally:
Mary Ruthven6481a9f2019-08-23 12:46:05 -0700978 if tmpd:
979 tmpd.clean()
Dan Shi9cb0eec2014-06-03 09:04:50 -0700980
981
Otabek Kasimov2cb72cf2021-02-25 10:59:22 -0800982 def install_image_to_servo_usb(self, image_url=None):
983 """Installing a test image on a USB storage device.
J. Richard Barnette31b2e312013-04-04 16:05:22 -0700984
Otabek Kasimov2cb72cf2021-02-25 10:59:22 -0800985 Download image to USB-storage attached to the Servo board.
Richard Barnette03a0c132012-11-05 12:40:35 -0800986
Otabek Kasimov2cb72cf2021-02-25 10:59:22 -0800987 @param image_url: If specified use as the url to download to
988 USB-storage.
989
990 @raises AutoservError if the image fails to download.
beepsf079cfb2013-09-18 17:49:51 -0700991
J. Richard Barnette0199cc82014-12-05 17:08:40 -0800992 """
Otabek Kasimov2cb72cf2021-02-25 10:59:22 -0800993 if not image_url:
994 logging.debug('Skip download as image_url not provided!')
995 return
Garry Wang7b0e1b72020-03-25 19:08:59 -0700996
Otabek Kasimov2cb72cf2021-02-25 10:59:22 -0800997 logging.info('Downloading image to USB')
Derek Beckett5f4edf02021-07-27 14:56:44 -0700998 try:
999 self.servo.image_to_servo_usb(image_path=image_url,
1000 power_off_dut=False)
1001 except error.AutotestError as e:
1002 six.reraise(error.AutotestError, str(e), sys.exc_info()[2])
Otabek Kasimov2cb72cf2021-02-25 10:59:22 -08001003
1004 def boot_in_recovery_mode(self,
1005 usb_boot_timeout=USB_BOOT_TIMEOUT,
1006 need_snk=False):
1007 """Booting host in recovery mode.
1008
1009 Boot device in recovery mode and verify that device booted from
1010 external storage as expected.
1011
1012 @param usb_boot_timeout: The usb_boot_timeout to use wait the host
1013 to boot. Factory images need a longer
1014 usb_boot_timeout than regular cros images.
1015 @param snk_mode: If True, switch servo_v4 role to 'snk'
1016 mode before boot DUT into recovery mode.
1017
1018 @raises AutoservError if the image fails to boot.
1019
1020 """
1021 logging.info('Booting from USB directly. Usb boot timeout: %s',
1022 usb_boot_timeout)
Derek Beckett5f4edf02021-07-27 14:56:44 -07001023 self.servo.boot_in_recovery_mode(snk_mode=need_snk)
1024 if not self.wait_up(timeout=usb_boot_timeout):
1025 if need_snk:
1026 # Attempt to restore servo_v4 role to 'src' mode.
1027 self.servo.set_servo_v4_role('src')
1028 raise hosts.AutoservRepairError(
1029 'DUT failed to boot from USB after %d seconds' %
1030 usb_boot_timeout, 'failed_to_boot_pre_install')
Scott Zawalski62bacae2013-03-05 10:40:32 -05001031
Garry Wang2e347df2020-10-30 14:04:26 -07001032 # Make sure the DUT is boot from an external device.
1033 if not self.is_boot_from_external_device():
1034 raise hosts.AutoservRepairError(
1035 'DUT is expected to boot from an external device(e.g. '
1036 'a usb stick), however it seems still boot from an'
1037 ' internal storage.', 'boot_from_internal_storage')
1038
Otabek Kasimov2cb72cf2021-02-25 10:59:22 -08001039 def run_install_image(self,
1040 install_timeout=INSTALL_TIMEOUT,
1041 need_snk=False,
1042 is_repair=False):
1043 """Installing the image with chromeos-install.
1044
1045 Steps included:
1046 1) Recover TPM on the device
1047 2) Run chromeos-install
1048 2.a) if success: power off/on the device
1049 2.b) if fail:
1050 2.b.1) Mark for replacement if fail with hardware issue
1051 2.b.2) Run internal storage check. (Only if is_repair=True)
1052 3) Wait the device to boot as verifier of success install
1053
1054 Device has to booted from external storage.
1055
1056 @param install_timeout: The timeout to use when installing the
1057 chromeos image. Factory images need a
1058 longer install_timeout.
1059 @param snk_mode: If True, switch servo_v4 role to 'snk'
1060 mode before boot DUT into recovery mode.
1061 @param is_repair: Indicates if the method is called from a
1062 repair task.
1063
1064 @raises AutoservError if the fail in process of install image.
1065 @raises AutoservRepairError if fail to boot after install image.
1066
1067 """
Tom Wai-Hong Tamf6b4f812015-08-08 04:14:59 +08001068 # The new chromeos-tpm-recovery has been merged since R44-7073.0.0.
1069 # In old CrOS images, this command fails. Skip the error.
Tom Wai-Hong Tam27af7332015-07-25 06:09:39 +08001070 logging.info('Resetting the TPM status')
Tom Wai-Hong Tamf6b4f812015-08-08 04:14:59 +08001071 try:
1072 self.run('chromeos-tpm-recovery')
1073 except error.AutoservRunError:
1074 logging.warn('chromeos-tpm-recovery is too old.')
1075
Derek Beckett5f4edf02021-07-27 14:56:44 -07001076 logging.info('Installing image through chromeos-install.')
1077 try:
1078 self.run('chromeos-install --yes', timeout=install_timeout)
1079 self.halt()
1080 except Exception as e:
1081 storage_errors = [
1082 'No space left on device',
1083 'I/O error when trying to write primary GPT',
1084 'Input/output error while writing out',
1085 'cannot read GPT header',
1086 'can not determine destination device',
1087 'wrong fs type',
1088 'bad superblock on',
1089 ]
1090 has_error = [msg for msg in storage_errors if (msg in str(e))]
1091 if has_error:
1092 info = self.host_info_store.get()
1093 info.set_version_label(audit_const.DUT_STORAGE_STATE_PREFIX,
1094 audit_const.HW_STATE_NEED_REPLACEMENT)
1095 self.host_info_store.commit(info)
1096 self.set_device_repair_state(
Otabek Kasimov832d9162020-07-27 19:24:57 -07001097 cros_constants.DEVICE_STATE_NEEDS_REPLACEMENT)
Derek Beckett5f4edf02021-07-27 14:56:44 -07001098 logging.debug('Fail install image from USB; Storage error; %s',
1099 e)
1100 raise error.AutoservError(
Otabek Kasimov808cd832020-05-28 18:27:46 -07001101 'Failed to install image from USB due to a suspect '
1102 'disk failure, DUT storage state changed to '
1103 'need_replacement, please check debug log '
1104 'for details.')
Derek Beckett5f4edf02021-07-27 14:56:44 -07001105 else:
1106 if is_repair:
1107 # DUT will be marked for replacement if storage is bad.
1108 audit_verify.VerifyDutStorage(self).verify()
Otabek Kasimov27bb2862020-08-10 14:40:45 -07001109
Derek Beckett5f4edf02021-07-27 14:56:44 -07001110 logging.debug('Fail install image from USB; %s', e)
1111 raise error.AutoservError(
Otabek Kasimov808cd832020-05-28 18:27:46 -07001112 'Failed to install image from USB due to unexpected '
1113 'error, please check debug log for details.')
Derek Beckett5f4edf02021-07-27 14:56:44 -07001114 finally:
1115 # We need reset the DUT no matter re-install success or not,
1116 # as we don't want leave the DUT in boot from usb state.
1117 logging.info('Power cycling DUT through servo.')
1118 self.servo.get_power_state_controller().power_off()
1119 self.servo.switch_usbkey('off')
1120 if need_snk:
1121 # Attempt to restore servo_v4 role to 'src' mode.
1122 self.servo.set_servo_v4_role('src')
1123 # N.B. The Servo API requires that we use power_on() here
1124 # for two reasons:
1125 # 1) After turning on a DUT in recovery mode, you must turn
1126 # it off and then on with power_on() once more to
1127 # disable recovery mode (this is a Parrot specific
1128 # requirement).
1129 # 2) After power_off(), the only way to turn on is with
1130 # power_on() (this is a Storm specific requirement).
1131 self.servo.get_power_state_controller().power_on()
beepsf079cfb2013-09-18 17:49:51 -07001132
1133 logging.info('Waiting for DUT to come back up.')
Richard Barnette03a0c132012-11-05 12:40:35 -08001134 if not self.wait_up(timeout=self.BOOT_TIMEOUT):
Garry Wang9ced7aa2020-04-10 17:26:35 -07001135 raise hosts.AutoservRepairError('DUT failed to reboot installed '
1136 'test image after %d seconds' %
1137 self.BOOT_TIMEOUT,
1138 'failed_to_boot_post_install')
Scott Zawalski62bacae2013-03-05 10:40:32 -05001139
Otabek Kasimov2cb72cf2021-02-25 10:59:22 -08001140 def servo_install(self,
1141 image_url=None,
1142 usb_boot_timeout=USB_BOOT_TIMEOUT,
1143 install_timeout=INSTALL_TIMEOUT,
1144 is_repair=False):
1145 """Re-install the OS on the DUT by:
1146
1147 Steps:
1148 1) Power off the host
1149 2) Installing an image on a USB-storage attached to the Servo board
1150 3) Booting that image in recovery mode
1151 4) Installing the image with chromeos-install.
1152
1153 @param image_url: If specified use as the url to install on
1154 the DUT otherwise boot the currently
1155 staged image on the USB stick.
1156 @param usb_boot_timeout: The usb_boot_timeout to use during
1157 re-image. Factory images need a longer
1158 usb_boot_timeout than regular cros images.
1159 @param install_timeout: The timeout to use when installing the
1160 chromeos image. Factory images need a
1161 longer install_timeout.
1162 @param is_repair: Indicates if the method is called from a
1163 repair task.
1164
1165 @raises AutoservError if the image fails to boot.
1166
1167 """
1168 self.servo.get_power_state_controller().power_off()
1169 if image_url:
1170 self.install_image_to_servo_usb(image_url=image_url)
1171 else:
1172 # Give the DUT some time to power_off if we skip
1173 # download image to usb. (crbug.com/982993)
1174 time.sleep(10)
1175
1176 need_snk = self.require_snk_mode_in_recovery()
1177
1178 self.boot_in_recovery_mode(usb_boot_timeout=usb_boot_timeout,
1179 need_snk=need_snk)
1180
1181 self.run_install_image(install_timeout=install_timeout,
1182 need_snk=need_snk,
1183 is_repair=is_repair)
Scott Zawalski62bacae2013-03-05 10:40:32 -05001184
Garry Wanga2e78172020-09-09 23:49:07 -07001185 def set_servo_host(self, host, servo_state=None):
Richard Barnette4aeb01c2018-09-20 09:36:12 -07001186 """Set our servo host member, and associated servo.
1187
1188 @param host Our new `ServoHost`.
1189 """
1190 self._servo_host = host
Derek Beckettb66e5c82020-08-12 15:31:02 -07001191 self.servo_pwr_supported = None
Richard Barnette4aeb01c2018-09-20 09:36:12 -07001192 if self._servo_host is not None:
1193 self.servo = self._servo_host.get_servo()
Otabek Kasimova7ba91a2020-03-09 08:31:01 -07001194 servo_state = self._servo_host.get_servo_state()
Garry Wang000c6c02020-05-11 21:27:23 -07001195 self._set_smart_usbhub_label(self._servo_host.smart_usbhub)
Derek Beckettb66e5c82020-08-12 15:31:02 -07001196 try:
1197 self.servo_pwr_supported = self.servo.has_control('power_state')
1198 except Exception as e:
1199 logging.debug(
1200 "Could not get servo power state due to {}".format(e))
Gregory Nisbet93b23e22020-10-02 20:42:16 +00001201 else:
1202 self.servo = None
Derek Beckettb66e5c82020-08-12 15:31:02 -07001203 self.servo_pwr_supported = False
Otabek Kasimov41301a22020-05-10 15:28:21 -07001204 self.set_servo_type()
Otabek Kasimova7ba91a2020-03-09 08:31:01 -07001205 self.set_servo_state(servo_state)
Otabek Kasimov382c3bb2020-10-28 13:22:45 -07001206 self._set_servo_topology()
Otabek Kasimova7ba91a2020-03-09 08:31:01 -07001207
Richard Barnette4aeb01c2018-09-20 09:36:12 -07001208
Richard Barnette9a26ad62016-06-10 12:03:08 -07001209 def repair_servo(self):
Dan Shi90466352015-09-22 15:01:05 -07001210 """
Richard Barnette9a26ad62016-06-10 12:03:08 -07001211 Confirm that servo is initialized and verified.
Dan Shi90466352015-09-22 15:01:05 -07001212
Richard Barnette9a26ad62016-06-10 12:03:08 -07001213 If the servo object is missing, attempt to repair the servo
1214 host. Repair failures are passed back to the caller.
1215
1216 @raise AutoservError: If there is no servo host for this CrOS
1217 host.
1218 """
1219 if self.servo:
1220 return
1221 if not self._servo_host:
1222 raise error.AutoservError('No servo host for %s.' %
1223 self.hostname)
Otabek Kasimovcc9738e2020-02-14 16:17:15 -08001224 try:
1225 self._servo_host.repair()
1226 except:
1227 raise
1228 finally:
1229 self.set_servo_host(self._servo_host)
1230
1231
Otabek Kasimov41301a22020-05-10 15:28:21 -07001232 def set_servo_type(self):
1233 """Set servo info labels to dut host_info"""
1234 if not self.servo:
Otabek Kasimovbea89e52020-05-12 15:40:00 -07001235 logging.debug('Servo is not initialized to get servo_type.')
Otabek Kasimov41301a22020-05-10 15:28:21 -07001236 return
Otabek Kasimov1b70e8d2020-12-30 13:51:00 -08001237 if not self.is_servo_in_working_state():
1238 logging.debug('Servo is not good, skip update servo_type.')
1239 return
Otabek Kasimov41301a22020-05-10 15:28:21 -07001240 servo_type = self.servo.get_servo_type()
1241 if not servo_type:
Otabek Kasimovbea89e52020-05-12 15:40:00 -07001242 logging.debug('Cannot collect servo_type from servo'
Otabek Kasimov41301a22020-05-10 15:28:21 -07001243 ' by `dut-control servo_type`! Please file a bug'
1244 ' and inform infra team as we are not expected '
1245 ' to reach this point.')
1246 return
1247 host_info = self.host_info_store.get()
1248 prefix = servo_constants.SERVO_TYPE_LABEL_PREFIX
1249 old_type = host_info.get_label_value(prefix)
1250 if old_type == servo_type:
1251 # do not need update
1252 return
1253 host_info.set_version_label(prefix, servo_type)
1254 self.host_info_store.commit(host_info)
1255 logging.info('ServoHost: servo_type updated to %s '
1256 '(previous: %s)', servo_type, old_type)
1257
1258
Otabek Kasimova7ba91a2020-03-09 08:31:01 -07001259 def set_servo_state(self, servo_state):
Otabek Kasimovcc9738e2020-02-14 16:17:15 -08001260 """Set servo info labels to dut host_info"""
Otabek Kasimova7ba91a2020-03-09 08:31:01 -07001261 if servo_state is not None:
Otabek Kasimovcc9738e2020-02-14 16:17:15 -08001262 host_info = self.host_info_store.get()
Garry Wang11b5e872020-03-11 15:14:08 -07001263 servo_state_prefix = servo_constants.SERVO_STATE_LABEL_PREFIX
Otabek Kasimova7ba91a2020-03-09 08:31:01 -07001264 old_state = host_info.get_label_value(servo_state_prefix)
1265 if old_state == servo_state:
1266 # do not need update
1267 return
1268 host_info.set_version_label(servo_state_prefix, servo_state)
Otabek Kasimovcc9738e2020-02-14 16:17:15 -08001269 self.host_info_store.commit(host_info)
Otabek Kasimova7ba91a2020-03-09 08:31:01 -07001270 logging.info('ServoHost: servo_state updated to %s (previous: %s)',
1271 servo_state, old_state)
Dan Shi90466352015-09-22 15:01:05 -07001272
1273
Otabek Kasimova7ba91a2020-03-09 08:31:01 -07001274 def get_servo_state(self):
1275 host_info = self.host_info_store.get()
Garry Wang11b5e872020-03-11 15:14:08 -07001276 servo_state_prefix = servo_constants.SERVO_STATE_LABEL_PREFIX
Otabek Kasimova7ba91a2020-03-09 08:31:01 -07001277 return host_info.get_label_value(servo_state_prefix)
1278
Otabek Kasimov5f039202020-10-28 15:45:29 -07001279 def is_servo_in_working_state(self):
1280 """Validate servo is in WORKING state."""
1281 servo_state = self.get_servo_state()
1282 return servo_state == servo_constants.SERVO_STATE_WORKING
1283
Dana Goyette655af512020-09-03 10:48:23 -07001284 def get_servo_usb_state(self):
1285 """Get the label value indicating the health of the USB drive.
1286
1287 @return: The label value if defined, otherwise '' (empty string).
1288 @rtype: str
1289 """
1290 host_info = self.host_info_store.get()
1291 servo_usb_state_prefix = audit_const.SERVO_USB_STATE_PREFIX
1292 return host_info.get_label_value(servo_usb_state_prefix)
1293
1294 def is_servo_usb_usable(self):
1295 """Check if the servo USB storage device is usable for FAFT.
1296
1297 @return: False if the label indicates a state that will break FAFT.
1298 True if state is okay, or if state is not defined.
1299 @rtype: bool
1300 """
1301 usb_state = self.get_servo_usb_state()
1302 return usb_state in ('', audit_const.HW_STATE_ACCEPTABLE,
1303 audit_const.HW_STATE_NORMAL,
1304 audit_const.HW_STATE_UNKNOWN)
Otabek Kasimov41301a22020-05-10 15:28:21 -07001305
Garry Wang000c6c02020-05-11 21:27:23 -07001306 def _set_smart_usbhub_label(self, smart_usbhub_detected):
1307 if smart_usbhub_detected is None:
1308 # skip the label update here as this indicate we wasn't able
1309 # to confirm usbhub type.
1310 return
1311 host_info = self.host_info_store.get()
1312 if (smart_usbhub_detected ==
1313 (servo_constants.SMART_USBHUB_LABEL in host_info.labels)):
1314 # skip label update if current label match the truth.
1315 return
1316 if smart_usbhub_detected:
1317 logging.info('Adding %s label to host %s',
1318 servo_constants.SMART_USBHUB_LABEL,
1319 self.hostname)
1320 host_info.labels.append(servo_constants.SMART_USBHUB_LABEL)
1321 else:
1322 logging.info('Removing %s label from host %s',
1323 servo_constants.SMART_USBHUB_LABEL,
1324 self.hostname)
1325 host_info.labels.remove(servo_constants.SMART_USBHUB_LABEL)
1326 self.host_info_store.commit(host_info)
1327
J. Richard Barnettec2d99cf2015-11-18 12:46:15 -08001328 def repair(self):
1329 """Attempt to get the DUT to pass `self.verify()`.
Richard Barnette82c35912012-11-20 10:09:10 -08001330
1331 This overrides the base class function for repair; it does
J. Richard Barnette91137f02016-03-10 16:52:26 -08001332 not call back to the parent class, but instead relies on
1333 `self._repair_strategy` to coordinate the verification and
1334 repair steps needed to get the DUT working.
Richard Barnette82c35912012-11-20 10:09:10 -08001335 """
Richard Barnetteabbdc252018-07-26 16:57:42 -07001336 message = 'Beginning repair for host %s board %s model %s'
1337 info = self.host_info_store.get()
1338 message %= (self.hostname, info.board, info.model)
1339 self.record('INFO', None, None, message)
Garry Wanga2e78172020-09-09 23:49:07 -07001340 profile_state = profile_constants.DUT_STATE_READY
Shijin Abraham78ce4402020-09-08 22:04:27 -07001341 # Initialize bluetooth peers
1342 self.initialize_btpeer()
Garry Wang87af1d02020-05-26 17:55:54 -07001343 try:
1344 self._repair_strategy.repair(self)
1345 except hosts.AutoservVerifyDependencyError as e:
Otabek Kasimovd48389b2020-12-07 02:38:34 -08001346 # TODO(otabek): remove when finish b/174191325
1347 self._stat_if_pingable_but_not_sshable()
Garry Wang87af1d02020-05-26 17:55:54 -07001348 # We don't want flag a DUT as failed if only non-critical
1349 # verifier(s) failed during the repair.
1350 if e.is_critical():
Garry Wanga2e78172020-09-09 23:49:07 -07001351 profile_state = profile_constants.DUT_STATE_REPAIR_FAILED
Otabek Kasimov86062d02020-11-17 13:30:22 -08001352 self._reboot_labstation_if_needed()
Otabek Kasimov7cad9ce2020-07-28 14:58:48 -07001353 self.try_set_device_needs_manual_repair()
Garry Wang87af1d02020-05-26 17:55:54 -07001354 raise
Garry Wanga2e78172020-09-09 23:49:07 -07001355 finally:
1356 self.set_health_profile_dut_state(profile_state)
Scott Zawalski89c44dd2013-02-26 09:28:02 -05001357
Otabek Kasimov8bb09912020-10-01 14:44:57 -07001358 def get_verifier_state(self, tag):
Otabek Kasimov44273d22021-02-26 17:13:24 -08001359 """Return the state of host verifier by tag.
Otabek Kasimov8bb09912020-10-01 14:44:57 -07001360
1361 @returns: bool or None
1362 """
1363 return self._repair_strategy.verifier_is_good(tag)
Richard Barnette82c35912012-11-20 10:09:10 -08001364
Otabek Kasimov44273d22021-02-26 17:13:24 -08001365 def get_repair_strategy_node(self, tag):
1366 """Return the instance of verifier/repair node for host by tag.
1367
1368 @returns: _DependencyNode or None
1369 """
1370 return self._repair_strategy.node_by_tag(tag)
1371
J. Richard Barnette1d78b012012-05-15 13:56:30 -07001372 def close(self):
David Rileye2c6be12017-12-11 10:20:57 -08001373 """Close connection."""
Fang Deng0ca40e22013-08-27 17:47:44 -07001374 super(CrosHost, self).close()
howardchung83e55272019-08-08 14:08:05 +08001375
Shijin Abraham783a7dd2020-02-14 15:36:11 -08001376 if self._chameleon_host:
1377 self._chameleon_host.close()
xixuand6011f12016-12-08 15:01:58 -08001378
Garry Wang1a493d82020-08-31 21:01:19 -07001379 if self.health_profile:
1380 try:
1381 self.health_profile.close()
1382 except Exception as e:
1383 logging.warning(
1384 'Failed to finalize device health profile; %s', e)
1385
xixuand6011f12016-12-08 15:01:58 -08001386 if self._servo_host:
1387 self._servo_host.close()
J. Richard Barnette1d78b012012-05-15 13:56:30 -07001388
Dan Shi49ca0932014-11-14 11:22:27 -08001389 def get_power_supply_info(self):
1390 """Get the output of power_supply_info.
1391
1392 power_supply_info outputs the info of each power supply, e.g.,
1393 Device: Line Power
1394 online: no
1395 type: Mains
1396 voltage (V): 0
1397 current (A): 0
1398 Device: Battery
1399 state: Discharging
1400 percentage: 95.9276
1401 technology: Li-ion
1402
1403 Above output shows two devices, Line Power and Battery, with details of
1404 each device listed. This function parses the output into a dictionary,
1405 with key being the device name, and value being a dictionary of details
1406 of the device info.
1407
1408 @return: The dictionary of power_supply_info, e.g.,
1409 {'Line Power': {'online': 'yes', 'type': 'main'},
1410 'Battery': {'vendor': 'xyz', 'percentage': '100'}}
Dan Shie9b765d2014-12-29 16:59:49 -08001411 @raise error.AutoservRunError if power_supply_info tool is not found in
1412 the DUT. Caller should handle this error to avoid false failure
1413 on verification.
Dan Shi49ca0932014-11-14 11:22:27 -08001414 """
1415 result = self.run('power_supply_info').stdout.strip()
1416 info = {}
1417 device_name = None
1418 device_info = {}
1419 for line in result.split('\n'):
1420 pair = [v.strip() for v in line.split(':')]
1421 if len(pair) != 2:
1422 continue
1423 if pair[0] == 'Device':
1424 if device_name:
1425 info[device_name] = device_info
1426 device_name = pair[1]
1427 device_info = {}
1428 else:
1429 device_info[pair[0]] = pair[1]
1430 if device_name and not device_name in info:
1431 info[device_name] = device_info
1432 return info
1433
1434
1435 def get_battery_percentage(self):
1436 """Get the battery percentage.
1437
1438 @return: The percentage of battery level, value range from 0-100. Return
1439 None if the battery info cannot be retrieved.
1440 """
1441 try:
1442 info = self.get_power_supply_info()
1443 logging.info(info)
1444 return float(info['Battery']['percentage'])
Dan Shie9b765d2014-12-29 16:59:49 -08001445 except (KeyError, ValueError, error.AutoservRunError):
Dan Shi49ca0932014-11-14 11:22:27 -08001446 return None
1447
1448
Philip Chenaf69ead2020-03-27 13:06:42 -07001449 def get_battery_state(self):
1450 """Get the battery charging state.
1451
1452 @return: A string representing the battery charging state. It can be
1453 'Charging', 'Fully charged', or 'Discharging'.
1454 """
1455 try:
1456 info = self.get_power_supply_info()
1457 logging.info(info)
1458 return info['Battery']['state']
1459 except (KeyError, ValueError, error.AutoservRunError):
1460 return None
1461
1462
Daniel Campello8ca25c22019-12-13 16:48:26 -07001463 def get_battery_display_percentage(self):
1464 """Get the battery display percentage.
1465
1466 @return: The display percentage of battery level, value range from
1467 0-100. Return None if the battery info cannot be retrieved.
1468 """
1469 try:
1470 info = self.get_power_supply_info()
1471 logging.info(info)
1472 return float(info['Battery']['display percentage'])
1473 except (KeyError, ValueError, error.AutoservRunError):
1474 return None
1475
1476
Dan Shi49ca0932014-11-14 11:22:27 -08001477 def is_ac_connected(self):
1478 """Check if the dut has power adapter connected and charging.
1479
1480 @return: True if power adapter is connected and charging.
1481 """
1482 try:
1483 info = self.get_power_supply_info()
1484 return info['Line Power']['online'] == 'yes'
Dan Shie9b765d2014-12-29 16:59:49 -08001485 except (KeyError, error.AutoservRunError):
1486 return None
Dan Shi49ca0932014-11-14 11:22:27 -08001487
1488
Simran Basi5e6339a2013-03-21 11:34:32 -07001489 def _cleanup_poweron(self):
1490 """Special cleanup method to make sure hosts always get power back."""
Garry Wangad4d4fd2019-01-30 17:00:38 -08001491 info = self.host_info_store.get()
1492 if self._RPM_OUTLET_CHANGED not in info.attributes:
Simran Basi5e6339a2013-03-21 11:34:32 -07001493 return
1494 logging.debug('This host has recently interacted with the RPM'
1495 ' Infrastructure. Ensuring power is on.')
1496 try:
1497 self.power_on()
Garry Wangad4d4fd2019-01-30 17:00:38 -08001498 self._remove_rpm_changed_tag()
Simran Basi5e6339a2013-03-21 11:34:32 -07001499 except rpm_client.RemotePowerException:
Simran Basi5e6339a2013-03-21 11:34:32 -07001500 logging.error('Failed to turn Power On for this host after '
1501 'cleanup through the RPM Infrastructure.')
Dan Shi49ca0932014-11-14 11:22:27 -08001502
1503 battery_percentage = self.get_battery_percentage()
Otabek Kasimov58e22562020-11-03 17:17:41 -08001504 if (
1505 battery_percentage
1506 and battery_percentage < cros_constants.MIN_BATTERY_LEVEL):
Dan Shi49ca0932014-11-14 11:22:27 -08001507 raise
1508 elif self.is_ac_connected():
1509 logging.info('The device has power adapter connected and '
1510 'charging. No need to try to turn RPM on '
1511 'again.')
Garry Wangad4d4fd2019-01-30 17:00:38 -08001512 self._remove_rpm_changed_tag()
Dan Shi49ca0932014-11-14 11:22:27 -08001513 logging.info('Battery level is now at %s%%. The device may '
1514 'still have enough power to run test, so no '
1515 'exception will be raised.', battery_percentage)
1516
Simran Basi5e6339a2013-03-21 11:34:32 -07001517
Garry Wangad4d4fd2019-01-30 17:00:38 -08001518 def _remove_rpm_changed_tag(self):
1519 info = self.host_info_store.get()
1520 del info.attributes[self._RPM_OUTLET_CHANGED]
1521 self.host_info_store.commit(info)
1522
1523
1524 def _add_rpm_changed_tag(self):
1525 info = self.host_info_store.get()
Garry Wang518831d2019-02-21 15:15:36 -08001526 info.attributes[self._RPM_OUTLET_CHANGED] = 'true'
Garry Wangad4d4fd2019-01-30 17:00:38 -08001527 self.host_info_store.commit(info)
1528
1529
1530
beepsc87ff602013-07-31 21:53:00 -07001531 def _is_factory_image(self):
1532 """Checks if the image on the DUT is a factory image.
1533
1534 @return: True if the image on the DUT is a factory image.
1535 False otherwise.
1536 """
1537 result = self.run('[ -f /root/.factory_test ]', ignore_status=True)
1538 return result.exit_status == 0
1539
1540
1541 def _restart_ui(self):
J. Richard Barnette84890bd2014-02-21 11:05:47 -08001542 """Restart the Chrome UI.
beepsc87ff602013-07-31 21:53:00 -07001543
1544 @raises: FactoryImageCheckerException for factory images, since
1545 we cannot attempt to restart ui on them.
1546 error.AutoservRunError for any other type of error that
1547 occurs while restarting ui.
1548 """
1549 if self._is_factory_image():
Dan Shi549fb822015-03-24 18:01:11 -07001550 raise FactoryImageCheckerException('Cannot restart ui on factory '
1551 'images')
beepsc87ff602013-07-31 21:53:00 -07001552
J. Richard Barnette84890bd2014-02-21 11:05:47 -08001553 # TODO(jrbarnette): The command to stop/start the ui job
1554 # should live inside cros_ui, too. However that would seem
1555 # to imply interface changes to the existing start()/restart()
1556 # functions, which is a bridge too far (for now).
J. Richard Barnette6069aa12015-06-08 09:10:24 -07001557 prompt = cros_ui.get_chrome_session_ident(self)
J. Richard Barnette84890bd2014-02-21 11:05:47 -08001558 self.run('stop ui; start ui')
1559 cros_ui.wait_for_chrome_ready(prompt, self)
beepsc87ff602013-07-31 21:53:00 -07001560
1561
Daniel Erat4b5f7e02018-07-04 22:05:30 -07001562 def _start_powerd_if_needed(self):
1563 """Start powerd if it isn't already running."""
1564 self.run('start powerd', ignore_status=True)
1565
Kazuhiro Inabaa7a00492020-12-17 16:05:12 +09001566 def _read_arc_prop_file(self, filename):
1567 for path in [
1568 '/usr/share/arcvm/properties/', '/usr/share/arc/properties/'
1569 ]:
1570 if self.path_exists(path + filename):
1571 return utils.parse_cmd_output('cat ' + path + filename,
1572 run_method=self.run)
1573 return None
Daniel Erat4b5f7e02018-07-04 22:05:30 -07001574
Jiyoun Hac172ee72020-12-15 08:57:29 +09001575 def _get_arc_build_info(self):
1576 """Returns a dictionary mapping build properties to their values."""
1577 build_info = None
Kazuhiro Inabaa7a00492020-12-17 16:05:12 +09001578 for filename in ['build.prop', 'vendor_build.prop']:
1579 properties = self._read_arc_prop_file(filename)
1580 if properties:
1581 if build_info:
1582 build_info.update(properties)
1583 else:
1584 build_info = properties
1585 else:
1586 logging.error('Failed to find %s in device.', filename)
Jiyoun Hac172ee72020-12-15 08:57:29 +09001587 return build_info
1588
Jiyoun Haba37f312021-01-13 09:44:16 +09001589 def get_arc_primary_abi(self):
Jiyoun Hac172ee72020-12-15 08:57:29 +09001590 """Returns the primary abi of the host."""
1591 return self._get_arc_build_info().get('ro.product.cpu.abi')
1592
Jiyoun Haba37f312021-01-13 09:44:16 +09001593 def get_arc_security_patch(self):
Jiyoun Hac172ee72020-12-15 08:57:29 +09001594 """Returns the security patch of the host."""
1595 return self._get_arc_build_info().get('ro.build.version.security_patch')
1596
Kazuhiro Inabaa7a00492020-12-17 16:05:12 +09001597 def get_arc_first_api_level(self):
1598 """Returns the security patch of the host."""
1599 return self._get_arc_build_info().get('ro.product.first_api_level')
1600
xixuana3bbc422017-05-04 15:57:21 -07001601 def _get_lsb_release_content(self):
1602 """Return the content of lsb-release file of host."""
1603 return self.run(
1604 'cat "%s"' % client_constants.LSB_RELEASE).stdout.strip()
1605
1606
Dan Shi549fb822015-03-24 18:01:11 -07001607 def get_release_version(self):
1608 """Get the value of attribute CHROMEOS_RELEASE_VERSION from lsb-release.
1609
1610 @returns The version string in lsb-release, under attribute
1611 CHROMEOS_RELEASE_VERSION.
1612 """
Dan Shi549fb822015-03-24 18:01:11 -07001613 return lsbrelease_utils.get_chromeos_release_version(
xixuana3bbc422017-05-04 15:57:21 -07001614 lsb_release_content=self._get_lsb_release_content())
1615
1616
Don Garrettb9f35802018-01-22 18:25:40 -08001617 def get_release_builder_path(self):
1618 """Get the value of CHROMEOS_RELEASE_BUILDER_PATH from lsb-release.
1619
1620 @returns The version string in lsb-release, under attribute
1621 CHROMEOS_RELEASE_BUILDER_PATH.
1622 """
1623 return lsbrelease_utils.get_chromeos_release_builder_path(
1624 lsb_release_content=self._get_lsb_release_content())
1625
1626
xixuana3bbc422017-05-04 15:57:21 -07001627 def get_chromeos_release_milestone(self):
1628 """Get the value of attribute CHROMEOS_RELEASE_BUILD_TYPE
1629 from lsb-release.
1630
1631 @returns The version string in lsb-release, under attribute
1632 CHROMEOS_RELEASE_BUILD_TYPE.
1633 """
1634 return lsbrelease_utils.get_chromeos_release_milestone(
1635 lsb_release_content=self._get_lsb_release_content())
Dan Shi549fb822015-03-24 18:01:11 -07001636
1637
1638 def verify_cros_version_label(self):
Garry Wangd18e7b32020-08-07 18:31:44 -07001639 """Verify if host's cros-version label match the actual image in dut.
Dan Shi549fb822015-03-24 18:01:11 -07001640
Garry Wangd18e7b32020-08-07 18:31:44 -07001641 @returns True if the label match with image in dut, otherwise False
Dan Shi549fb822015-03-24 18:01:11 -07001642 """
Garry Wangd18e7b32020-08-07 18:31:44 -07001643 os_from_host = self.get_release_builder_path()
1644 info = self.host_info_store.get()
1645 os_from_label = info.get_label_value(self.VERSION_PREFIX)
1646 if not os_from_label:
1647 logging.debug('No existing %s label detected', self.VERSION_PREFIX)
1648 return True
1649
1650 # known cases where the version label will not match the
1651 # original CHROMEOS_RELEASE_BUILDER_PATH setting:
1652 # * Tests for the `arc-presubmit` append "-cheetsth" to the label.
1653 if os_from_label.endswith(provision.CHEETS_SUFFIX):
1654 logging.debug('%s label with %s suffix detected, this suffix will'
1655 ' be ignored when comparing label.',
1656 self.VERSION_PREFIX, provision.CHEETS_SUFFIX)
1657 os_from_label = os_from_label[:-len(provision.CHEETS_SUFFIX)]
1658 logging.debug('OS version from host: %s; OS verision cached in '
1659 'label: %s', os_from_host, os_from_label)
1660 return os_from_label == os_from_host
Dan Shi549fb822015-03-24 18:01:11 -07001661
1662
Laurence Goodby778c9a42017-05-24 19:24:07 -07001663 def cleanup_services(self):
1664 """Reinitializes the device for cleanup.
1665
1666 Subclasses may override this to customize the cleanup method.
1667
1668 To indicate failure of the reset, the implementation may raise
1669 any of:
1670 error.AutoservRunError
1671 error.AutotestRunError
1672 FactoryImageCheckerException
1673
1674 @raises error.AutoservRunError
1675 @raises error.AutotestRunError
1676 @raises error.FactoryImageCheckerException
1677 """
1678 self._restart_ui()
Daniel Erat4b5f7e02018-07-04 22:05:30 -07001679 self._start_powerd_if_needed()
Laurence Goodby778c9a42017-05-24 19:24:07 -07001680
1681
Gregory Nisbetec615d62020-12-11 17:59:20 +00001682 def cleanup(self):
1683 """Cleanup state on device."""
MK Ryu35d661e2014-09-25 17:44:10 -07001684 self.run('rm -f %s' % client_constants.CLEANUP_LOGS_PAUSED_FILE)
Scott Zawalskiddbc31e2012-11-15 11:29:01 -05001685 try:
Laurence Goodby778c9a42017-05-24 19:24:07 -07001686 self.cleanup_services()
beepsc87ff602013-07-31 21:53:00 -07001687 except (error.AutotestRunError, error.AutoservRunError,
1688 FactoryImageCheckerException):
Otabek Kasimovc0d77e82020-03-27 15:01:57 -07001689 logging.warning('Unable to restart ui.')
Namyoon Woo33f38852020-04-13 17:26:58 -07001690
Otabek Kasimovc0d77e82020-03-27 15:01:57 -07001691 # cleanup routines, i.e. reboot the machine.
Gregory Nisbetec615d62020-12-11 17:59:20 +00001692 super(CrosHost, self).cleanup()
Otabek Kasimovc0d77e82020-03-27 15:01:57 -07001693
Simran Basi5e6339a2013-03-21 11:34:32 -07001694 # Check if the rpm outlet was manipulated.
Simran Basid5e5e272012-09-24 15:23:59 -07001695 if self.has_power():
Simran Basi5e6339a2013-03-21 11:34:32 -07001696 self._cleanup_poweron()
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001697
Gregory Nisbetec615d62020-12-11 17:59:20 +00001698
Yu-Ju Honga2be94a2012-07-31 09:48:52 -07001699 def reboot(self, **dargs):
1700 """
1701 This function reboots the site host. The more generic
1702 RemoteHost.reboot() performs sync and sleeps for 5
1703 seconds. This is not necessary for Chrome OS devices as the
1704 sync should be finished in a short time during the reboot
1705 command.
1706 """
Gregory Nisbetec615d62020-12-11 17:59:20 +00001707 if 'reboot_cmd' not in dargs:
Doug Anderson7d5aeb22014-02-27 15:12:17 -08001708 reboot_timeout = dargs.get('reboot_timeout', 10)
J. Richard Barnette9af19632015-09-25 12:18:03 -07001709 dargs['reboot_cmd'] = ('sleep 1; '
1710 'reboot & sleep %d; '
1711 'reboot -f' % reboot_timeout)
Yu-Ju Honga2be94a2012-07-31 09:48:52 -07001712 # Enable fastsync to avoid running extra sync commands before reboot.
Tom Wai-Hong Tamf5cd1d42012-08-13 12:04:08 +08001713 if 'fastsync' not in dargs:
1714 dargs['fastsync'] = True
Michael Liangda8c60a2014-06-03 13:24:51 -07001715
Prathmesh Prabhu53a4c1c2018-09-14 16:36:27 -07001716 dargs['board'] = self.host_info_store.get().board
Vincent Palatindf2372c2016-10-07 17:03:00 +02001717 # Record who called us
1718 orig = sys._getframe(1).f_code
Vincent Palatin80780b22016-07-27 16:02:37 +02001719 metric_fields = {'board' : dargs['board'],
Vincent Palatindf2372c2016-10-07 17:03:00 +02001720 'dut_host_name' : self.hostname,
1721 'success' : True}
1722 metric_debug_fields = {'board' : dargs['board'],
Prathmesh Prabhu53a4c1c2018-09-14 16:36:27 -07001723 'caller' : "%s:%s" % (orig.co_filename,
1724 orig.co_name),
Vincent Palatindf2372c2016-10-07 17:03:00 +02001725 'success' : True,
1726 'error' : ''}
1727
Vincent Palatin80780b22016-07-27 16:02:37 +02001728 try:
1729 super(CrosHost, self).reboot(**dargs)
1730 except Exception as e:
1731 metric_fields['success'] = False
Vincent Palatindf2372c2016-10-07 17:03:00 +02001732 metric_debug_fields['success'] = False
1733 metric_debug_fields['error'] = type(e).__name__
Vincent Palatin80780b22016-07-27 16:02:37 +02001734 raise
Yu-Ju Honga2be94a2012-07-31 09:48:52 -07001735
Kalin Stoyanov83d9caa2020-01-31 16:58:27 -08001736 def suspend(self, suspend_time=60, delay_seconds=0,
Ravi Chandra Sadineni812e61b2018-07-09 11:16:50 -07001737 suspend_cmd=None, allow_early_resume=False):
Gwendal Grignou7a61d2f2014-05-23 11:05:51 -07001738 """
1739 This function suspends the site host.
Ravi Chandra Sadineni812e61b2018-07-09 11:16:50 -07001740
1741 @param suspend_time: How long to suspend as integer seconds.
1742 @param suspend_cmd: Suspend command to execute.
1743 @param allow_early_resume: If False and if device resumes before
1744 |suspend_time|, throw an error.
1745
1746 @exception AutoservSuspendError Host resumed earlier than
1747 |suspend_time|.
Gwendal Grignou7a61d2f2014-05-23 11:05:51 -07001748 """
Ravi Chandra Sadineni812e61b2018-07-09 11:16:50 -07001749
1750 if suspend_cmd is None:
1751 suspend_cmd = ' && '.join([
J. Richard Barnette9af19632015-09-25 12:18:03 -07001752 'echo 0 > /sys/class/rtc/rtc0/wakealarm',
Gwendal Grignou7a61d2f2014-05-23 11:05:51 -07001753 'echo +%d > /sys/class/rtc/rtc0/wakealarm' % suspend_time,
Kalin Stoyanov83d9caa2020-01-31 16:58:27 -08001754 'powerd_dbus_suspend --delay=%d' % delay_seconds])
Ravi Chandra Sadineni812e61b2018-07-09 11:16:50 -07001755 super(CrosHost, self).suspend(suspend_time, suspend_cmd,
1756 allow_early_resume);
Gwendal Grignou7a61d2f2014-05-23 11:05:51 -07001757
1758
Simran Basiec564392014-08-25 16:48:09 -07001759 def upstart_status(self, service_name):
1760 """Check the status of an upstart init script.
1761
1762 @param service_name: Service to look up.
1763
1764 @returns True if the service is running, False otherwise.
1765 """
Richard Barnettee204dc52017-09-26 11:02:25 -07001766 return 'start/running' in self.run('status %s' % service_name,
1767 ignore_status=True).stdout
Simran Basiec564392014-08-25 16:48:09 -07001768
Tom Hughese9552342018-12-18 14:29:25 -08001769 def upstart_stop(self, service_name):
1770 """Stops an upstart job if it's running.
1771
1772 @param service_name: Service to stop
1773
1774 @returns True if service has been stopped or was already stopped
1775 False otherwise.
1776 """
1777 if not self.upstart_status(service_name):
1778 return True
1779
1780 result = self.run('stop %s' % service_name, ignore_status=True)
1781 if result.exit_status != 0:
1782 return False
1783 return True
1784
1785 def upstart_restart(self, service_name):
1786 """Restarts (or starts) an upstart job.
1787
1788 @param service_name: Service to start/restart
1789
1790 @returns True if service has been started/restarted, False otherwise.
1791 """
1792 cmd = 'start'
1793 if self.upstart_status(service_name):
1794 cmd = 'restart'
1795 cmd = cmd + ' %s' % service_name
1796 result = self.run(cmd)
1797 if result.exit_status != 0:
1798 return False
1799 return True
Simran Basiec564392014-08-25 16:48:09 -07001800
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001801 def verify_software(self):
Richard Barnetteb2bc13c2013-01-08 17:32:51 -08001802 """Verify working software on a Chrome OS system.
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001803
Richard Barnetteb2bc13c2013-01-08 17:32:51 -08001804 Tests for the following conditions:
1805 1. All conditions tested by the parent version of this
1806 function.
1807 2. Sufficient space in /mnt/stateful_partition.
Fang Deng6b05f5b2013-03-20 13:42:11 -07001808 3. Sufficient space in /mnt/stateful_partition/encrypted.
1809 4. update_engine answers a simple status request over DBus.
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001810
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001811 """
Fang Deng0ca40e22013-08-27 17:47:44 -07001812 super(CrosHost, self).verify_software()
Dan Shib8540a52015-07-16 14:18:23 -07001813 default_kilo_inodes_required = CONFIG.get_config_value(
1814 'SERVER', 'kilo_inodes_required', type=int, default=100)
1815 board = self.get_board().replace(ds_constants.BOARD_PREFIX, '')
1816 kilo_inodes_required = CONFIG.get_config_value(
1817 'SERVER', 'kilo_inodes_required_%s' % board,
1818 type=int, default=default_kilo_inodes_required)
1819 self.check_inodes('/mnt/stateful_partition', kilo_inodes_required)
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001820 self.check_diskspace(
1821 '/mnt/stateful_partition',
Dan Shib8540a52015-07-16 14:18:23 -07001822 CONFIG.get_config_value(
Fang Deng6b05f5b2013-03-20 13:42:11 -07001823 'SERVER', 'gb_diskspace_required', type=float,
1824 default=20.0))
Gaurav Shahe448af82014-06-19 15:18:59 -07001825 encrypted_stateful_path = '/mnt/stateful_partition/encrypted'
1826 # Not all targets build with encrypted stateful support.
1827 if self.path_exists(encrypted_stateful_path):
1828 self.check_diskspace(
1829 encrypted_stateful_path,
Dan Shib8540a52015-07-16 14:18:23 -07001830 CONFIG.get_config_value(
Gaurav Shahe448af82014-06-19 15:18:59 -07001831 'SERVER', 'gb_encrypted_diskspace_required', type=float,
1832 default=0.1))
beepsc87ff602013-07-31 21:53:00 -07001833
Justin TerAvest3b0c5ba2017-11-07 09:59:11 -07001834 self.wait_for_system_services()
Prashanth B5d0a0512014-04-25 12:26:08 -07001835
beepsc87ff602013-07-31 21:53:00 -07001836 # Factory images don't run update engine,
1837 # goofy controls dbus on these DUTs.
1838 if not self._is_factory_image():
1839 self.run('update_engine_client --status')
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001840
1841
Justin TerAvest3b0c5ba2017-11-07 09:59:11 -07001842 @retry.retry(error.AutoservError, timeout_min=5, delay_sec=10)
Kuo Jen Wei50ef6e82021-03-29 12:04:47 +08001843 def wait_for_service(self, service_name):
1844 """Wait for target status of an upstart init script.
1845
1846 @param service_name: Service to wait for.
1847 """
1848 if not self.upstart_status(service_name):
1849 raise error.AutoservError('Service %s not running.' % service_name)
1850
Justin TerAvest3b0c5ba2017-11-07 09:59:11 -07001851 def wait_for_system_services(self):
1852 """Waits for system-services to be running.
1853
1854 Sometimes, update_engine will take a while to update firmware, so we
1855 should give this some time to finish. See crbug.com/765686#c38 for
1856 details.
1857 """
Kuo Jen Wei50ef6e82021-03-29 12:04:47 +08001858 self.wait_for_service('system-services')
Justin TerAvest3b0c5ba2017-11-07 09:59:11 -07001859
1860
J. Richard Barnettea7e7fdc2016-02-12 12:35:36 -08001861 def verify(self):
Pradeep Sawlaniaa013fa2017-11-09 06:34:18 -08001862 """Verify Chrome OS system is in good state."""
Richard Barnetteabbdc252018-07-26 16:57:42 -07001863 message = 'Beginning verify for host %s board %s model %s'
1864 info = self.host_info_store.get()
1865 message %= (self.hostname, info.board, info.model)
1866 self.record('INFO', None, None, message)
Garry Wang87af1d02020-05-26 17:55:54 -07001867 try:
1868 self._repair_strategy.verify(self)
1869 except hosts.AutoservVerifyDependencyError as e:
1870 # We don't want flag a DUT as failed if only non-critical
1871 # verifier(s) failed during the repair.
1872 if e.is_critical():
1873 raise
J. Richard Barnettea7e7fdc2016-02-12 12:35:36 -08001874
1875
Fang Deng96667ca2013-08-01 17:46:18 -07001876 def make_ssh_command(self, user='root', port=22, opts='', hosts_file=None,
Dean Liaoe3e75f62017-11-14 10:36:43 +08001877 connect_timeout=None, alive_interval=None,
1878 alive_count_max=None, connection_attempts=None):
Fang Deng96667ca2013-08-01 17:46:18 -07001879 """Override default make_ssh_command to use options tuned for Chrome OS.
1880
1881 Tuning changes:
1882 - ConnectTimeout=30; maximum of 30 seconds allowed for an SSH
1883 connection failure. Consistency with remote_access.sh.
1884
Samuel Tan2ce155b2015-06-23 18:24:38 -07001885 - ServerAliveInterval=900; which causes SSH to ping connection every
1886 900 seconds. In conjunction with ServerAliveCountMax ensures
1887 that if the connection dies, Autotest will bail out.
Fang Deng96667ca2013-08-01 17:46:18 -07001888 Originally tried 60 secs, but saw frequent job ABORTS where
Samuel Tan2ce155b2015-06-23 18:24:38 -07001889 the test completed successfully. Later increased from 180 seconds to
1890 900 seconds to account for tests where the DUT is suspended for
1891 longer periods of time.
Fang Deng96667ca2013-08-01 17:46:18 -07001892
1893 - ServerAliveCountMax=3; consistency with remote_access.sh.
1894
1895 - ConnectAttempts=4; reduce flakiness in connection errors;
1896 consistency with remote_access.sh.
1897
1898 - UserKnownHostsFile=/dev/null; we don't care about the keys.
1899 Host keys change with every new installation, don't waste
1900 memory/space saving them.
1901
1902 - SSH protocol forced to 2; needed for ServerAliveInterval.
1903
1904 @param user User name to use for the ssh connection.
1905 @param port Port on the target host to use for ssh connection.
1906 @param opts Additional options to the ssh command.
1907 @param hosts_file Ignored.
1908 @param connect_timeout Ignored.
1909 @param alive_interval Ignored.
Dean Liaoe3e75f62017-11-14 10:36:43 +08001910 @param alive_count_max Ignored.
1911 @param connection_attempts Ignored.
Fang Deng96667ca2013-08-01 17:46:18 -07001912 """
Dean Liaoe3e75f62017-11-14 10:36:43 +08001913 options = ' '.join([opts, '-o Protocol=2'])
1914 return super(CrosHost, self).make_ssh_command(
1915 user=user, port=port, opts=options, hosts_file='/dev/null',
1916 connect_timeout=30, alive_interval=900, alive_count_max=3,
1917 connection_attempts=4)
1918
1919
Jason Abeleb6f924f2013-11-13 16:01:54 -08001920 def syslog(self, message, tag='autotest'):
1921 """Logs a message to syslog on host.
1922
1923 @param message String message to log into syslog
1924 @param tag String tag prefix for syslog
1925
1926 """
1927 self.run('logger -t "%s" "%s"' % (tag, message))
1928
1929
J. Richard Barnetteb6de7e32013-02-14 13:28:04 -08001930 def _ping_check_status(self, status):
1931 """Ping the host once, and return whether it has a given status.
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001932
J. Richard Barnetteb6de7e32013-02-14 13:28:04 -08001933 @param status Check the ping status against this value.
1934 @return True iff `status` and the result of ping are the same
1935 (i.e. both True or both False).
1936
1937 """
Abhishek Pandit-Subedi038df162020-09-14 16:37:43 -07001938 ping_val = utils.ping(self.hostname,
1939 tries=1,
1940 deadline=1,
1941 timeout=2,
1942 ignore_timeout=True)
J. Richard Barnetteb6de7e32013-02-14 13:28:04 -08001943 return not (status ^ (ping_val == 0))
1944
1945 def _ping_wait_for_status(self, status, timeout):
1946 """Wait for the host to have a given status (UP or DOWN).
1947
1948 Status is checked by polling. Polling will not last longer
1949 than the number of seconds in `timeout`. The polling
1950 interval will be long enough that only approximately
1951 _PING_WAIT_COUNT polling cycles will be executed, subject
1952 to a maximum interval of about one minute.
1953
1954 @param status Waiting will stop immediately if `ping` of the
1955 host returns this status.
1956 @param timeout Poll for at most this many seconds.
1957 @return True iff the host status from `ping` matched the
1958 requested status at the time of return.
1959
1960 """
1961 # _ping_check_status() takes about 1 second, hence the
1962 # "- 1" in the formula below.
Nathan Ciobanu38480a32016-10-25 15:26:45 -07001963 # FIXME: if the ping command errors then _ping_check_status()
1964 # returns instantly. If timeout is also smaller than twice
1965 # _PING_WAIT_COUNT then the while loop below forks many
1966 # thousands of ping commands (see /tmp/test_that_results_XXXXX/
1967 # /results-1-logging_YYY.ZZZ/debug/autoserv.DEBUG) and hogs one
1968 # CPU core for 60 seconds.
J. Richard Barnetteb6de7e32013-02-14 13:28:04 -08001969 poll_interval = min(int(timeout / self._PING_WAIT_COUNT), 60) - 1
1970 end_time = time.time() + timeout
1971 while time.time() <= end_time:
1972 if self._ping_check_status(status):
1973 return True
1974 if poll_interval > 0:
1975 time.sleep(poll_interval)
1976
1977 # The last thing we did was sleep(poll_interval), so it may
1978 # have been too long since the last `ping`. Check one more
1979 # time, just to be sure.
1980 return self._ping_check_status(status)
1981
1982 def ping_wait_up(self, timeout):
1983 """Wait for the host to respond to `ping`.
1984
1985 N.B. This method is not a reliable substitute for
1986 `wait_up()`, because a host that responds to ping will not
1987 necessarily respond to ssh. This method should only be used
1988 if the target DUT can be considered functional even if it
1989 can't be reached via ssh.
1990
1991 @param timeout Minimum time to allow before declaring the
1992 host to be non-responsive.
1993 @return True iff the host answered to ping before the timeout.
1994
1995 """
Andrew Luo4be621d2020-03-21 07:01:13 -07001996 if self.use_icmp:
1997 return self._ping_wait_for_status(self._PING_STATUS_UP, timeout)
1998 else:
1999 logging.debug('Using SSH instead of ICMP for ping_wait_up.')
2000 return self.wait_up(timeout)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002001
Andrew Bresticker678c0c72013-01-22 10:44:09 -08002002 def ping_wait_down(self, timeout):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002003 """Wait until the host no longer responds to `ping`.
2004
J. Richard Barnetteb6de7e32013-02-14 13:28:04 -08002005 This function can be used as a slightly faster version of
2006 `wait_down()`, by avoiding potentially long ssh timeouts.
2007
2008 @param timeout Minimum time to allow for the host to become
2009 non-responsive.
2010 @return True iff the host quit answering ping before the
2011 timeout.
2012
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002013 """
Andrew Luo4be621d2020-03-21 07:01:13 -07002014 if self.use_icmp:
2015 return self._ping_wait_for_status(self._PING_STATUS_DOWN, timeout)
2016 else:
2017 logging.debug('Using SSH instead of ICMP for ping_wait_down.')
2018 return self.wait_down(timeout)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002019
Anand K Mistry50f218e2020-07-31 14:50:15 +10002020 def _is_host_port_forwarded(self):
Garry Wanga2e78172020-09-09 23:49:07 -07002021 """Checks if the dut is connected over port forwarding.
Anand K Mistry50f218e2020-07-31 14:50:15 +10002022
2023 N.B. This method does not detect all situations where port forwarding is
2024 occurring. Namely, running autotest on the dut may result in a
2025 false-positive, and port forwarding using a different machine on the
2026 same network will be a false-negative.
2027
2028 @return True if the dut is connected over port forwarding
2029 False otherwise
2030 """
Garry Wanga2e78172020-09-09 23:49:07 -07002031 is_localhost = self.hostname in ['localhost', '127.0.0.1']
2032 is_forwarded = is_localhost and not self.is_default_port
2033 if is_forwarded:
2034 logging.info('Detected DUT connected by port forwarding')
2035 return is_forwarded
Anand K Mistry50f218e2020-07-31 14:50:15 +10002036
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08002037 def test_wait_for_sleep(self, sleep_timeout=None):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002038 """Wait for the client to enter low-power sleep mode.
2039
2040 The test for "is asleep" can't distinguish a system that is
2041 powered off; to confirm that the unit was asleep, it is
2042 necessary to force resume, and then call
2043 `test_wait_for_resume()`.
2044
2045 This function is expected to be called from a test as part
2046 of a sequence like the following:
2047
2048 ~~~~~~~~
2049 boot_id = host.get_boot_id()
2050 # trigger sleep on the host
2051 host.test_wait_for_sleep()
2052 # trigger resume on the host
2053 host.test_wait_for_resume(boot_id)
2054 ~~~~~~~~
2055
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08002056 @param sleep_timeout time limit in seconds to allow the host sleep.
2057
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002058 @exception TestFail The host did not go to sleep within
2059 the allowed time.
2060 """
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08002061 if sleep_timeout is None:
2062 sleep_timeout = self.SLEEP_TIMEOUT
2063
Anand K Mistry50f218e2020-07-31 14:50:15 +10002064 # If the dut is accessed over SSH port-forwarding, `ping` is not useful
2065 # for detecting the dut is down since a ping to localhost will always
2066 # succeed. In this case, fall back to wait_down() which uses SSH.
2067 if self._is_host_port_forwarded():
Garry Wanga2e78172020-09-09 23:49:07 -07002068 success = self.wait_down(timeout=sleep_timeout)
Anand K Mistry50f218e2020-07-31 14:50:15 +10002069 else:
Garry Wanga2e78172020-09-09 23:49:07 -07002070 success = self.ping_wait_down(timeout=sleep_timeout)
Anand K Mistry50f218e2020-07-31 14:50:15 +10002071
2072 if not success:
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002073 raise error.TestFail(
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08002074 'client failed to sleep after %d seconds' % sleep_timeout)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002075
2076
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08002077 def test_wait_for_resume(self, old_boot_id, resume_timeout=None):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002078 """Wait for the client to resume from low-power sleep mode.
2079
2080 The `old_boot_id` parameter should be the value from
2081 `get_boot_id()` obtained prior to entering sleep mode. A
2082 `TestFail` exception is raised if the boot id changes.
2083
2084 See @ref test_wait_for_sleep for more on this function's
2085 usage.
2086
J. Richard Barnette7214e0b2013-02-06 15:20:49 -08002087 @param old_boot_id A boot id value obtained before the
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002088 target host went to sleep.
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08002089 @param resume_timeout time limit in seconds to allow the host up.
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002090
2091 @exception TestFail The host did not respond within the
2092 allowed time.
2093 @exception TestFail The host responded, but the boot id test
2094 indicated a reboot rather than a sleep
2095 cycle.
2096 """
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08002097 if resume_timeout is None:
2098 resume_timeout = self.RESUME_TIMEOUT
2099
2100 if not self.wait_up(timeout=resume_timeout):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002101 raise error.TestFail(
2102 'client failed to resume from sleep after %d seconds' %
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08002103 resume_timeout)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002104 else:
2105 new_boot_id = self.get_boot_id()
2106 if new_boot_id != old_boot_id:
Tom Wai-Hong Tam01792682015-01-06 08:00:46 +08002107 logging.error('client rebooted (old boot %s, new boot %s)',
2108 old_boot_id, new_boot_id)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002109 raise error.TestFail(
Tom Wai-Hong Tam01792682015-01-06 08:00:46 +08002110 'client rebooted, but sleep was expected')
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002111
2112
Tom Wai-Hong Tamfe005c22014-12-03 09:25:44 +08002113 def test_wait_for_shutdown(self, shutdown_timeout=None):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002114 """Wait for the client to shut down.
2115
2116 The test for "has shut down" can't distinguish a system that
2117 is merely asleep; to confirm that the unit was down, it is
2118 necessary to force boot, and then call test_wait_for_boot().
2119
2120 This function is expected to be called from a test as part
2121 of a sequence like the following:
2122
2123 ~~~~~~~~
2124 boot_id = host.get_boot_id()
2125 # trigger shutdown on the host
2126 host.test_wait_for_shutdown()
2127 # trigger boot on the host
2128 host.test_wait_for_boot(boot_id)
2129 ~~~~~~~~
2130
Tom Wai-Hong Tamfe005c22014-12-03 09:25:44 +08002131 @param shutdown_timeout time limit in seconds to allow the host down.
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002132 @exception TestFail The host did not shut down within the
2133 allowed time.
2134 """
Tom Wai-Hong Tamfe005c22014-12-03 09:25:44 +08002135 if shutdown_timeout is None:
2136 shutdown_timeout = self.SHUTDOWN_TIMEOUT
2137
Anand K Mistry50f218e2020-07-31 14:50:15 +10002138 if self._is_host_port_forwarded():
Garry Wanga2e78172020-09-09 23:49:07 -07002139 success = self.wait_down(timeout=shutdown_timeout)
Anand K Mistry50f218e2020-07-31 14:50:15 +10002140 else:
Garry Wanga2e78172020-09-09 23:49:07 -07002141 success = self.ping_wait_down(timeout=shutdown_timeout)
Anand K Mistry50f218e2020-07-31 14:50:15 +10002142
2143 if not success:
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002144 raise error.TestFail(
2145 'client failed to shut down after %d seconds' %
Tom Wai-Hong Tamfe005c22014-12-03 09:25:44 +08002146 shutdown_timeout)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002147
2148
2149 def test_wait_for_boot(self, old_boot_id=None):
2150 """Wait for the client to boot from cold power.
2151
2152 The `old_boot_id` parameter should be the value from
2153 `get_boot_id()` obtained prior to shutting down. A
2154 `TestFail` exception is raised if the boot id does not
2155 change. The boot id test is omitted if `old_boot_id` is not
2156 specified.
2157
2158 See @ref test_wait_for_shutdown for more on this function's
2159 usage.
2160
J. Richard Barnette7214e0b2013-02-06 15:20:49 -08002161 @param old_boot_id A boot id value obtained before the
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002162 shut down.
2163
2164 @exception TestFail The host did not respond within the
2165 allowed time.
2166 @exception TestFail The host responded, but the boot id test
2167 indicated that there was no reboot.
2168 """
J. Richard Barnetteeb69d722012-06-18 17:29:44 -07002169 if not self.wait_up(timeout=self.REBOOT_TIMEOUT):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002170 raise error.TestFail(
2171 'client failed to reboot after %d seconds' %
J. Richard Barnetteeb69d722012-06-18 17:29:44 -07002172 self.REBOOT_TIMEOUT)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002173 elif old_boot_id:
2174 if self.get_boot_id() == old_boot_id:
Tom Wai-Hong Tam01792682015-01-06 08:00:46 +08002175 logging.error('client not rebooted (boot %s)',
2176 old_boot_id)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07002177 raise error.TestFail(
Tom Wai-Hong Tam01792682015-01-06 08:00:46 +08002178 'client is back up, but did not reboot')
Simran Basid5e5e272012-09-24 15:23:59 -07002179
2180
2181 @staticmethod
2182 def check_for_rpm_support(hostname):
2183 """For a given hostname, return whether or not it is powered by an RPM.
2184
Simran Basi1df55112013-09-06 11:25:09 -07002185 @param hostname: hostname to check for rpm support.
2186
Simran Basid5e5e272012-09-24 15:23:59 -07002187 @return None if this host does not follows the defined naming format
2188 for RPM powered DUT's in the lab. If it does follow the format,
2189 it returns a regular expression MatchObject instead.
2190 """
Fang Dengbaff9082015-01-06 13:46:15 -08002191 return re.match(CrosHost._RPM_HOSTNAME_REGEX, hostname)
Simran Basid5e5e272012-09-24 15:23:59 -07002192
2193
2194 def has_power(self):
2195 """For this host, return whether or not it is powered by an RPM.
2196
2197 @return True if this host is in the CROS lab and follows the defined
2198 naming format.
2199 """
Fang Deng0ca40e22013-08-27 17:47:44 -07002200 return CrosHost.check_for_rpm_support(self.hostname)
Simran Basid5e5e272012-09-24 15:23:59 -07002201
2202
Ismail Noorbasha07fdb612013-02-14 14:13:31 -08002203 def _set_power(self, state, power_method):
Garry Wang5e5538a2019-04-08 15:36:18 -07002204 """Sets the power to the host via RPM, CCD, Servo or manual.
Ismail Noorbasha07fdb612013-02-14 14:13:31 -08002205
2206 @param state Specifies which power state to set to DUT
2207 @param power_method Specifies which method of power control to
Garry Wang5e5538a2019-04-08 15:36:18 -07002208 use. By default "RPM" or "CCD" will be used based
2209 on servo type. Valid values from
2210 POWER_CONTROL_VALID_ARGS, or None to use default.
Ismail Noorbasha07fdb612013-02-14 14:13:31 -08002211
2212 """
2213 ACCEPTABLE_STATES = ['ON', 'OFF']
2214
Garry Wang5e5538a2019-04-08 15:36:18 -07002215 if not power_method:
2216 power_method = self.get_default_power_method()
2217
2218 state = state.upper()
2219 if state not in ACCEPTABLE_STATES:
Ismail Noorbasha07fdb612013-02-14 14:13:31 -08002220 raise error.TestError('State must be one of: %s.'
2221 % (ACCEPTABLE_STATES,))
2222
2223 if power_method == self.POWER_CONTROL_SERVO:
2224 logging.info('Setting servo port J10 to %s', state)
2225 self.servo.set('prtctl3_pwren', state.lower())
2226 time.sleep(self._USB_POWER_TIMEOUT)
2227 elif power_method == self.POWER_CONTROL_MANUAL:
2228 logging.info('You have %d seconds to set the AC power to %s.',
2229 self._POWER_CYCLE_TIMEOUT, state)
2230 time.sleep(self._POWER_CYCLE_TIMEOUT)
Garry Wang5e5538a2019-04-08 15:36:18 -07002231 elif power_method == self.POWER_CONTROL_CCD:
2232 servo_role = 'src' if state == 'ON' else 'snk'
2233 logging.info('servo ccd power pass through detected,'
2234 ' changing servo_role to %s.', servo_role)
2235 self.servo.set_servo_v4_role(servo_role)
2236 if not self.ping_wait_up(timeout=self._CHANGE_SERVO_ROLE_TIMEOUT):
Garry Wang94bf9de2019-06-10 17:23:37 -07002237 # Make sure we don't leave DUT with no power(servo_role=snk)
2238 # when DUT is not pingable, as we raise a exception here
2239 # that may break a power cycle in the middle.
2240 self.servo.set_servo_v4_role('src')
Garry Wang5e5538a2019-04-08 15:36:18 -07002241 raise error.AutoservError(
2242 'DUT failed to regain network connection after %d seconds.'
2243 % self._CHANGE_SERVO_ROLE_TIMEOUT)
Ismail Noorbasha07fdb612013-02-14 14:13:31 -08002244 else:
2245 if not self.has_power():
2246 raise error.TestFail('DUT does not have RPM connected.')
Garry Wangad4d4fd2019-01-30 17:00:38 -08002247 self._add_rpm_changed_tag()
Garry Wang5e5538a2019-04-08 15:36:18 -07002248 rpm_client.set_power(self, state, timeout_mins=5)
Simran Basid5e5e272012-09-24 15:23:59 -07002249
2250
Garry Wang5e5538a2019-04-08 15:36:18 -07002251 def power_off(self, power_method=None):
2252 """Turn off power to this host via RPM, CCD, Servo or manual.
Ismail Noorbasha07fdb612013-02-14 14:13:31 -08002253
2254 @param power_method Specifies which method of power control to
Garry Wang5e5538a2019-04-08 15:36:18 -07002255 use. By default "RPM" or "CCD" will be used based
2256 on servo type. Valid values from
2257 POWER_CONTROL_VALID_ARGS, or None to use default.
Ismail Noorbasha07fdb612013-02-14 14:13:31 -08002258
2259 """
Derek Beckettb66e5c82020-08-12 15:31:02 -07002260 self._sync_if_up()
Ismail Noorbasha07fdb612013-02-14 14:13:31 -08002261 self._set_power('OFF', power_method)
Simran Basid5e5e272012-09-24 15:23:59 -07002262
Derek Beckettb66e5c82020-08-12 15:31:02 -07002263 def _check_supported(self):
2264 """Throw an error if dts mode control is not supported."""
2265 if not self.servo_pwr_supported:
2266 raise error.TestFail('power_state controls not supported')
2267
2268 def _sync_if_up(self):
2269 """Run sync on the DUT and wait for completion if the DUT is up.
2270
2271 Additionally, try to sync and ignore status if its not up.
2272
2273 Useful prior to reboots to ensure files are written to disc.
2274
2275 """
2276 if self.is_up_fast():
2277 self.run("sync")
2278 return
2279 # If it is not up, attempt to sync in the rare event the DUT is up but
2280 # doesn't respond to a ping. Ignore any errors.
2281 try:
2282 self.run("sync", ignore_status=True, timeout=1)
2283 except Exception:
2284 pass
2285
2286 def power_off_via_servo(self):
2287 """Force the DUT to power off.
2288
2289 The DUT is guaranteed to be off at the end of this call,
2290 regardless of its previous state, provided that there is
2291 working EC and boot firmware. There is no requirement for
2292 working OS software.
2293
2294 """
2295 self._check_supported()
2296 self._sync_if_up()
2297 self.servo.set_nocheck('power_state', 'off')
2298
2299 def power_on_via_servo(self, rec_mode='on'):
2300 """Force the DUT to power on.
2301
2302 Prior to calling this function, the DUT must be powered off,
2303 e.g. with a call to `power_off()`.
2304
2305 At power on, recovery mode is set as specified by the
2306 corresponding argument. When booting with recovery mode on, it
2307 is the caller's responsibility to unplug/plug in a bootable
2308 external storage device.
2309
2310 If the DUT requires a delay after powering on but before
2311 processing inputs such as USB stick insertion, the delay is
2312 handled by this method; the caller is not responsible for such
2313 delays.
2314
2315 @param rec_mode Setting of recovery mode to be applied at
2316 power on. default: REC_OFF aka 'off'
2317
2318 """
2319 self._check_supported()
2320 self.servo.set_nocheck('power_state', rec_mode)
2321
2322 def reset_via_servo(self):
2323 """Force the DUT to reset.
2324
2325 The DUT is guaranteed to be on at the end of this call,
2326 regardless of its previous state, provided that there is
2327 working OS software. This also guarantees that the EC has
2328 been restarted.
2329
2330 """
2331 self._check_supported()
2332 self._sync_if_up()
2333 self.servo.set_nocheck('power_state', 'reset')
2334
Simran Basid5e5e272012-09-24 15:23:59 -07002335
Garry Wang5e5538a2019-04-08 15:36:18 -07002336 def power_on(self, power_method=None):
2337 """Turn on power to this host via RPM, CCD, Servo or manual.
Ismail Noorbasha07fdb612013-02-14 14:13:31 -08002338
2339 @param power_method Specifies which method of power control to
Garry Wang5e5538a2019-04-08 15:36:18 -07002340 use. By default "RPM" or "CCD" will be used based
2341 on servo type. Valid values from
2342 POWER_CONTROL_VALID_ARGS, or None to use default.
Ismail Noorbasha07fdb612013-02-14 14:13:31 -08002343
2344 """
2345 self._set_power('ON', power_method)
2346
2347
Garry Wang5e5538a2019-04-08 15:36:18 -07002348 def power_cycle(self, power_method=None):
Ismail Noorbasha07fdb612013-02-14 14:13:31 -08002349 """Cycle power to this host by turning it OFF, then ON.
2350
2351 @param power_method Specifies which method of power control to
Garry Wang5e5538a2019-04-08 15:36:18 -07002352 use. By default "RPM" or "CCD" will be used based
2353 on servo type. Valid values from
2354 POWER_CONTROL_VALID_ARGS, or None to use default.
Ismail Noorbasha07fdb612013-02-14 14:13:31 -08002355
2356 """
Garry Wang5e5538a2019-04-08 15:36:18 -07002357 if not power_method:
2358 power_method = self.get_default_power_method()
2359
Ismail Noorbasha07fdb612013-02-14 14:13:31 -08002360 if power_method in (self.POWER_CONTROL_SERVO,
Garry Wang5e5538a2019-04-08 15:36:18 -07002361 self.POWER_CONTROL_MANUAL,
2362 self.POWER_CONTROL_CCD):
Ismail Noorbasha07fdb612013-02-14 14:13:31 -08002363 self.power_off(power_method=power_method)
2364 time.sleep(self._POWER_CYCLE_TIMEOUT)
2365 self.power_on(power_method=power_method)
2366 else:
Garry Wangad4d4fd2019-01-30 17:00:38 -08002367 self._add_rpm_changed_tag()
2368 rpm_client.set_power(self, 'CYCLE')
Simran Basic6f1f7a2012-10-16 10:47:46 -07002369
2370
Mary Ruthvende14a8b2019-08-23 12:43:52 -07002371 def get_platform_from_fwid(self):
2372 """Determine the platform from the crossystem fwid.
2373
2374 @returns a string representing this host's platform.
2375 """
Greg Edelstona7b05d12020-04-01 16:00:51 -06002376 # Look at the firmware for non-unibuild cases or if cros_config fails.
Mary Ruthvende14a8b2019-08-23 12:43:52 -07002377 crossystem = utils.Crossystem(self)
2378 crossystem.init()
2379 # Extract fwid value and use the leading part as the platform id.
2380 # fwid generally follow the format of {platform}.{firmware version}
2381 # Example: Alex.X.YYY.Z or Google_Alex.X.YYY.Z
2382 platform = crossystem.fwid().split('.')[0].lower()
2383 # Newer platforms start with 'Google_' while the older ones do not.
2384 return platform.replace('google_', '')
2385
Puthikorn Voravootivat0f7c3d82019-11-27 13:48:39 -08002386
Simran Basic6f1f7a2012-10-16 10:47:46 -07002387 def get_platform(self):
2388 """Determine the correct platform label for this host.
2389
2390 @returns a string representing this host's platform.
2391 """
C Shapiroed87c6f2018-04-19 09:13:58 -06002392 release_info = utils.parse_cmd_output('cat /etc/lsb-release',
2393 run_method=self.run)
C Shapiroed87c6f2018-04-19 09:13:58 -06002394 platform = ''
Puthikorn Voravootivat0f7c3d82019-11-27 13:48:39 -08002395 if release_info.get('CHROMEOS_RELEASE_UNIBUILD') == '1':
Greg Edelstona7b05d12020-04-01 16:00:51 -06002396 platform = self.get_model_from_cros_config()
Mary Ruthvende14a8b2019-08-23 12:43:52 -07002397 return platform if platform else self.get_platform_from_fwid()
Simran Basic6f1f7a2012-10-16 10:47:46 -07002398
2399
Greg Edelstona7b05d12020-04-01 16:00:51 -06002400 def get_model_from_cros_config(self):
2401 """Get the host model from cros_config command.
Puthikorn Voravootivat0f7c3d82019-11-27 13:48:39 -08002402
Greg Edelstona7b05d12020-04-01 16:00:51 -06002403 @returns a string representing this host's model.
Puthikorn Voravootivat0f7c3d82019-11-27 13:48:39 -08002404 """
Greg Edelstona7b05d12020-04-01 16:00:51 -06002405 return cros_config.call_cros_config_get_output('/ name',
2406 self.run, ignore_status=True)
Puthikorn Voravootivat0f7c3d82019-11-27 13:48:39 -08002407
2408
Hung-ying Tyanb1328032014-04-01 14:18:54 +08002409 def get_architecture(self):
2410 """Determine the correct architecture label for this host.
2411
2412 @returns a string representing this host's architecture.
2413 """
2414 crossystem = utils.Crossystem(self)
2415 crossystem.init()
2416 return crossystem.arch()
2417
2418
Luis Lozano40b7d0d2014-01-17 15:12:06 -08002419 def get_chrome_version(self):
2420 """Gets the Chrome version number and milestone as strings.
2421
2422 Invokes "chrome --version" to get the version number and milestone.
2423
2424 @return A tuple (chrome_ver, milestone) where "chrome_ver" is the
2425 current Chrome version number as a string (in the form "W.X.Y.Z")
2426 and "milestone" is the first component of the version number
2427 (the "W" from "W.X.Y.Z"). If the version number cannot be parsed
2428 in the "W.X.Y.Z" format, the "chrome_ver" will be the full output
2429 of "chrome --version" and the milestone will be the empty string.
2430
2431 """
MK Ryu35d661e2014-09-25 17:44:10 -07002432 version_string = self.run(client_constants.CHROME_VERSION_COMMAND).stdout
Luis Lozano40b7d0d2014-01-17 15:12:06 -08002433 return utils.parse_chrome_version(version_string)
2434
J. Richard Barnetted2af5852016-02-05 15:03:10 -08002435
Puthikorn Voravootivatfd132172017-11-16 18:24:38 -08002436 def get_ec_version(self):
2437 """Get the ec version as strings.
2438
2439 @returns a string representing this host's ec version.
2440 """
Puthikorn Voravootivat65ad7672018-01-30 14:00:01 -08002441 command = 'mosys ec info -s fw_version'
Puthikorn Voravootivat720640d2018-02-16 17:32:38 -08002442 result = self.run(command, ignore_status=True)
2443 if result.exit_status != 0:
2444 return ''
2445 return result.stdout.strip()
Puthikorn Voravootivatfd132172017-11-16 18:24:38 -08002446
2447
2448 def get_firmware_version(self):
2449 """Get the firmware version as strings.
2450
2451 @returns a string representing this host's firmware version.
2452 """
2453 crossystem = utils.Crossystem(self)
2454 crossystem.init()
2455 return crossystem.fwid()
2456
2457
Puthikorn Voravootivat58f739b2021-01-07 08:28:36 -08002458 def get_hardware_id(self):
2459 """Get hardware id as strings.
2460
2461 @returns a string representing this host's hardware id.
2462 """
2463 crossystem = utils.Crossystem(self)
2464 crossystem.init()
2465 return crossystem.hwid()
2466
Puthikorn Voravootivatfd132172017-11-16 18:24:38 -08002467 def get_hardware_revision(self):
2468 """Get the hardware revision as strings.
2469
2470 @returns a string representing this host's hardware revision.
2471 """
Puthikorn Voravootivat65ad7672018-01-30 14:00:01 -08002472 command = 'mosys platform version'
Puthikorn Voravootivat720640d2018-02-16 17:32:38 -08002473 result = self.run(command, ignore_status=True)
2474 if result.exit_status != 0:
2475 return ''
2476 return result.stdout.strip()
Puthikorn Voravootivatfd132172017-11-16 18:24:38 -08002477
2478
2479 def get_kernel_version(self):
2480 """Get the kernel version as strings.
2481
2482 @returns a string representing this host's kernel version.
2483 """
2484 return self.run('uname -r').stdout.strip()
2485
2486
Puthikorn Voravootivat54aa53c2018-03-01 19:00:25 -08002487 def get_cpu_name(self):
2488 """Get the cpu name as strings.
2489
2490 @returns a string representing this host's cpu name.
2491 """
2492
2493 # Try get cpu name from device tree first
2494 if self.path_exists('/proc/device-tree/compatible'):
2495 command = ' | '.join(
2496 ["sed -e 's/\\x0/\\n/g' /proc/device-tree/compatible",
2497 'tail -1'])
2498 return self.run(command).stdout.strip().replace(',', ' ')
2499
2500 # Get cpu name from uname -p
2501 command = 'uname -p'
2502 ret = self.run(command).stdout.strip()
2503
2504 # 'uname -p' return variant of unknown or amd64 or x86_64 or i686
2505 # Try get cpu name from /proc/cpuinfo instead
2506 if re.match("unknown|amd64|[ix][0-9]?86(_64)?", ret, re.IGNORECASE):
2507 command = "grep model.name /proc/cpuinfo | cut -f 2 -d: | head -1"
2508 self = self.run(command).stdout.strip()
2509
2510 # Remove bloat from CPU name, for example
2511 # Intel(R) Core(TM) i5-7Y57 CPU @ 1.20GHz -> Intel Core i5-7Y57
2512 # Intel(R) Xeon(R) CPU E5-2690 v4 @ 2.60GHz -> Intel Xeon E5-2690 v4
2513 # AMD A10-7850K APU with Radeon(TM) R7 Graphics -> AMD A10-7850K
2514 # AMD GX-212JC SOC with Radeon(TM) R2E Graphics -> AMD GX-212JC
2515 trim_re = r' (@|processor|apu|soc|radeon).*|\(.*?\)| cpu'
2516 return re.sub(trim_re, '', ret, flags=re.IGNORECASE)
2517
2518
2519 def get_screen_resolution(self):
2520 """Get the screen(s) resolution as strings.
2521 In case of more than 1 monitor, return resolution for each monitor
2522 separate with plus sign.
2523
2524 @returns a string representing this host's screen(s) resolution.
2525 """
2526 command = 'for f in /sys/class/drm/*/*/modes; do head -1 $f; done'
2527 ret = self.run(command, ignore_status=True)
2528 # We might have Chromebox without a screen
2529 if ret.exit_status != 0:
2530 return ''
2531 return ret.stdout.strip().replace('\n', '+')
2532
2533
2534 def get_mem_total_gb(self):
2535 """Get total memory available in the system in GiB (2^20).
2536
2537 @returns an integer representing total memory
2538 """
2539 mem_total_kb = self.read_from_meminfo('MemTotal')
2540 kb_in_gb = float(2 ** 20)
2541 return int(round(mem_total_kb / kb_in_gb))
2542
2543
2544 def get_disk_size_gb(self):
2545 """Get size of disk in GB (10^9)
2546
2547 @returns an integer representing size of disk, 0 in Error Case
2548 """
2549 command = 'grep $(rootdev -s -d | cut -f3 -d/)$ /proc/partitions'
2550 result = self.run(command, ignore_status=True)
2551 if result.exit_status != 0:
2552 return 0
2553 _, _, block, _ = re.split(r' +', result.stdout.strip())
2554 byte_per_block = 1024.0
2555 disk_kb_in_gb = 1e9
2556 return int(int(block) * byte_per_block / disk_kb_in_gb + 0.5)
2557
2558
2559 def get_battery_size(self):
2560 """Get size of battery in Watt-hour via sysfs
2561
2562 This method assumes that battery support voltage_min_design and
2563 charge_full_design sysfs.
2564
2565 @returns a float representing Battery size, 0 if error.
2566 """
2567 # sysfs report data in micro scale
2568 battery_scale = 1e6
2569
2570 command = 'cat /sys/class/power_supply/*/voltage_min_design'
2571 result = self.run(command, ignore_status=True)
2572 if result.exit_status != 0:
2573 return 0
2574 voltage = float(result.stdout.strip()) / battery_scale
2575
2576 command = 'cat /sys/class/power_supply/*/charge_full_design'
2577 result = self.run(command, ignore_status=True)
2578 if result.exit_status != 0:
2579 return 0
2580 amphereHour = float(result.stdout.strip()) / battery_scale
2581
2582 return voltage * amphereHour
2583
2584
2585 def get_low_battery_shutdown_percent(self):
2586 """Get the percent-based low-battery shutdown threshold.
2587
2588 @returns a float representing low-battery shutdown percent, 0 if error.
2589 """
2590 ret = 0.0
2591 try:
2592 command = 'check_powerd_config --low_battery_shutdown_percent'
2593 ret = float(self.run(command).stdout)
2594 except error.CmdError:
2595 logging.debug("Can't run %s", command)
2596 except ValueError:
2597 logging.debug("Didn't get number from %s", command)
2598
2599 return ret
2600
2601
Puthikorn Voravootivat09c83d72018-08-10 15:58:32 -07002602 def has_hammer(self):
2603 """Check whether DUT has hammer device or not.
2604
2605 @returns boolean whether device has hammer or not
2606 """
2607 command = 'grep Hammer /sys/bus/usb/devices/*/product'
2608 return self.run(command, ignore_status=True).exit_status == 0
2609
2610
Niranjan Kumar34618872017-05-31 12:57:09 -07002611 def is_chrome_switch_present(self, switch):
David Haddock3ce538e2017-06-22 13:37:05 -07002612 """Returns True if the specified switch was provided to Chrome.
2613
2614 @param switch The chrome switch to search for.
2615 """
Niranjan Kumar34618872017-05-31 12:57:09 -07002616
Niranjan Kumar5f23fe92017-06-22 15:18:55 -07002617 command = 'pgrep -x -f -c "/opt/google/chrome/chrome.*%s.*"' % switch
2618 return self.run(command, ignore_status=True).exit_status == 0
Niranjan Kumar34618872017-05-31 12:57:09 -07002619
2620
2621 def oobe_triggers_update(self):
2622 """Returns True if this host has an OOBE flow during which
2623 it will perform an update check and perhaps an update.
2624 One example of such a flow is Hands-Off Zero-Touch Enrollment.
2625 As more such flows are developed, code handling them needs
2626 to be added here.
2627
2628 @return Boolean indicating whether this host's OOBE triggers an update.
2629 """
2630 return self.is_chrome_switch_present(
2631 '--enterprise-enable-zero-touch-enrollment=hands-off')
2632
2633
Kevin Chenga2619dc2016-03-28 11:42:08 -07002634 # TODO(kevcheng): change this to just return the board without the
2635 # 'board:' prefix and fix up all the callers. Also look into removing the
2636 # need for this method.
Simran Basic6f1f7a2012-10-16 10:47:46 -07002637 def get_board(self):
2638 """Determine the correct board label for this host.
2639
2640 @returns a string representing this host's board.
2641 """
2642 release_info = utils.parse_cmd_output('cat /etc/lsb-release',
2643 run_method=self.run)
J. Richard Barnetted2af5852016-02-05 15:03:10 -08002644 return (ds_constants.BOARD_PREFIX +
2645 release_info['CHROMEOS_RELEASE_BOARD'])
Simran Basic6f1f7a2012-10-16 10:47:46 -07002646
Po-Hsien Wang4618adf2017-10-10 14:40:21 -07002647 def get_channel(self):
2648 """Determine the correct channel label for this host.
Simran Basic6f1f7a2012-10-16 10:47:46 -07002649
Po-Hsien Wang4618adf2017-10-10 14:40:21 -07002650 @returns: a string represeting this host's build channel.
2651 (stable, dev, beta). None on fail.
2652 """
2653 return lsbrelease_utils.get_chromeos_channel(
2654 lsb_release_content=self._get_lsb_release_content())
Kevin Chenga328da62016-03-31 10:49:04 -07002655
Kevin Chenga328da62016-03-31 10:49:04 -07002656 def get_power_supply(self):
2657 """
2658 Determine what type of power supply the host has
2659
2660 @returns a string representing this host's power supply.
2661 'power:battery' when the device has a battery intended for
2662 extended use
2663 'power:AC_primary' when the device has a battery not intended
2664 for extended use (for moving the machine, etc)
2665 'power:AC_only' when the device has no battery at all.
2666 """
Jack Rosenthal01ee2cf2021-03-30 21:01:32 -06002667 psu = self.run(command='cros_config /hardware-properties psu-type',
2668 ignore_status=True)
Kevin Chenga328da62016-03-31 10:49:04 -07002669 if psu.exit_status:
Jack Rosenthal01ee2cf2021-03-30 21:01:32 -06002670 # Assume battery if unspecified in cros_config.
Kevin Chenga328da62016-03-31 10:49:04 -07002671 return 'power:battery'
2672
2673 psu_str = psu.stdout.strip()
2674 if psu_str == 'unknown':
2675 return None
2676
2677 return 'power:%s' % psu_str
2678
2679
Puthikorn Voravootivat54aa53c2018-03-01 19:00:25 -08002680 def has_battery(self):
2681 """Determine if DUT has a battery.
2682
2683 Returns:
2684 Boolean, False if known not to have battery, True otherwise.
2685 """
Jack Rosenthal01ee2cf2021-03-30 21:01:32 -06002686 return self.get_power_supply() == 'power:battery'
Puthikorn Voravootivat54aa53c2018-03-01 19:00:25 -08002687
2688
Kevin Chenga328da62016-03-31 10:49:04 -07002689 def get_servo(self):
2690 """Determine if the host has a servo attached.
2691
2692 If the host has a working servo attached, it should have a servo label.
2693
2694 @return: string 'servo' if the host has servo attached. Otherwise,
2695 returns None.
2696 """
2697 return 'servo' if self._servo_host else None
2698
2699
Kevin Chenga328da62016-03-31 10:49:04 -07002700 def has_internal_display(self):
2701 """Determine if the device under test is equipped with an internal
2702 display.
2703
2704 @return: 'internal_display' if one is present; None otherwise.
2705 """
2706 from autotest_lib.client.cros.graphics import graphics_utils
2707 from autotest_lib.client.common_lib import utils as common_utils
2708
2709 def __system_output(cmd):
2710 return self.run(cmd).stdout
2711
2712 def __read_file(remote_path):
2713 return self.run('cat %s' % remote_path).stdout
2714
2715 # Hijack the necessary client functions so that we can take advantage
2716 # of the client lib here.
2717 # FIXME: find a less hacky way than this
2718 original_system_output = utils.system_output
2719 original_read_file = common_utils.read_file
2720 utils.system_output = __system_output
2721 common_utils.read_file = __read_file
2722 try:
2723 return ('internal_display' if graphics_utils.has_internal_display()
2724 else None)
2725 finally:
2726 utils.system_output = original_system_output
2727 common_utils.read_file = original_read_file
2728
2729
Dan Shi85276d42014-04-08 22:11:45 -07002730 def is_boot_from_usb(self):
2731 """Check if DUT is boot from USB.
2732
2733 @return: True if DUT is boot from usb.
2734 """
2735 device = self.run('rootdev -s -d').stdout.strip()
2736 removable = int(self.run('cat /sys/block/%s/removable' %
2737 os.path.basename(device)).stdout.strip())
2738 return removable == 1
Helen Zhang17dae2b2014-11-11 09:25:52 -08002739
Otabek Kasimov77a40332020-10-20 15:40:03 -07002740 def is_boot_from_external_device(self):
2741 """Check if DUT is boot from external storage.
2742
2743 @return: True if DUT is boot from external storage.
2744 """
2745 boot_device = self.run('rootdev -s -d', ignore_status=True,
2746 timeout=60).stdout.strip()
2747 if not boot_device:
2748 logging.debug('Boot storage not detected on the host.')
2749 return False
2750 main_storage_cmd = ('. /usr/sbin/write_gpt.sh;'
2751 ' . /usr/share/misc/chromeos-common.sh;'
2752 ' load_base_vars; get_fixed_dst_drive')
2753 main_storage = self.run(main_storage_cmd,
2754 ignore_status=True,
2755 timeout=60).stdout.strip()
Otabek Kasimov723e8562020-12-08 13:29:34 -08002756 if not main_storage or boot_device != main_storage:
2757 logging.debug('Device booted from external storage storage.')
2758 return True
2759 logging.debug('Device booted from main storage.')
2760 return False
Helen Zhang17dae2b2014-11-11 09:25:52 -08002761
2762 def read_from_meminfo(self, key):
Dan Shi49ca0932014-11-14 11:22:27 -08002763 """Return the memory info from /proc/meminfo
Helen Zhang17dae2b2014-11-11 09:25:52 -08002764
2765 @param key: meminfo requested
2766
2767 @return the memory value as a string
2768
2769 """
Helen Zhang17dae2b2014-11-11 09:25:52 -08002770 meminfo = self.run('grep %s /proc/meminfo' % key).stdout.strip()
2771 logging.debug('%s', meminfo)
2772 return int(re.search(r'\d+', meminfo).group(0))
Rohit Makasana8a4923c2015-08-13 17:04:26 -07002773
2774
Rohit Makasana98e696f2016-06-03 18:48:10 -07002775 def get_cpu_arch(self):
2776 """Returns CPU arch of the device.
2777
2778 @return CPU architecture of the DUT.
2779 """
Allen Li2c32d6b2017-02-03 15:28:10 -08002780 # Add CPUs by following logic in client/bin/utils.py.
Rohit Makasana98e696f2016-06-03 18:48:10 -07002781 if self.run("grep '^flags.*:.* lm .*' /proc/cpuinfo",
2782 ignore_status=True).stdout:
2783 return 'x86_64'
2784 if self.run("grep -Ei 'ARM|CPU implementer' /proc/cpuinfo",
2785 ignore_status=True).stdout:
2786 return 'arm'
2787 return 'i386'
2788
2789
Rohit Makasana8a4923c2015-08-13 17:04:26 -07002790 def get_board_type(self):
2791 """
Nikolai Artemiev5b5b6802021-05-12 12:56:43 +10002792 Get the DUT's device type / form factor from cros_config. It can be one
2793 of CHROMEBOX, CHROMEBASE, CHROMEBOOK, or CHROMEBIT.
Danny Chan471a8d12015-08-18 14:57:41 -07002794
Nikolai Artemiev5b5b6802021-05-12 12:56:43 +10002795 @return form factor value from cros_config.
Rohit Makasana8a4923c2015-08-13 17:04:26 -07002796 """
Nikolai Artemiev5b5b6802021-05-12 12:56:43 +10002797
2798 device_type = self.run('cros_config /hardware-properties form-factor',
2799 ignore_status=True).stdout
2800 if device_type:
2801 return device_type
2802
2803 # TODO: remove lsb-release fallback once cros_config works everywhere
Danny Chan471a8d12015-08-18 14:57:41 -07002804 device_type = self.run('grep DEVICETYPE /etc/lsb-release',
2805 ignore_status=True).stdout
2806 if device_type:
Kalin Stoyanov524310b2015-08-21 16:24:04 -07002807 return device_type.split('=')[-1].strip()
Danny Chan471a8d12015-08-18 14:57:41 -07002808 return ''
Gilad Arnolda76bef02015-09-29 13:55:15 -07002809
2810
Rohit Makasanadf0a3a32017-06-30 13:55:18 -07002811 def get_arc_version(self):
2812 """Return ARC version installed on the DUT.
2813
2814 @returns ARC version as string if the CrOS build has ARC, else None.
2815 """
2816 arc_version = self.run('grep CHROMEOS_ARC_VERSION /etc/lsb-release',
2817 ignore_status=True).stdout
2818 if arc_version:
2819 return arc_version.split('=')[-1].strip()
2820 return None
2821
2822
Gilad Arnolda76bef02015-09-29 13:55:15 -07002823 def get_os_type(self):
2824 return 'cros'
Simran Basia5522a32015-10-06 11:01:24 -07002825
2826
Kevin Chenga2619dc2016-03-28 11:42:08 -07002827 def get_labels(self):
Kevin Chengf8660142016-08-12 10:17:41 -07002828 """Return the detected labels on the host."""
Kevin Chenga2619dc2016-03-28 11:42:08 -07002829 return self.labels.get_labels(self)
Garry Wang5e5538a2019-04-08 15:36:18 -07002830
2831
2832 def get_default_power_method(self):
2833 """
2834 Get the default power method for power_on/off/cycle() methods.
2835 @return POWER_CONTROL_RPM or POWER_CONTROL_CCD
2836 """
2837 if not self._default_power_method:
Garry Wang1a004aa2019-05-16 22:56:51 -07002838 self._default_power_method = self.POWER_CONTROL_RPM
Ruben Rodriguez Buchillon3eeeab32019-10-02 15:29:58 -07002839 if self.servo and self.servo.supports_built_in_pd_control():
2840 self._default_power_method = self.POWER_CONTROL_CCD
2841 else:
2842 logging.debug('Either servo is unitialized or the servo '
2843 'setup does not support pd controls. Falling '
2844 'back to default RPM method.')
Garry Wang5e5538a2019-04-08 15:36:18 -07002845 return self._default_power_method
Puthikorn Voravootivat4a054792019-12-13 16:44:17 -08002846
2847
2848 def find_usb_devices(self, idVendor, idProduct):
2849 """
2850 Get usb device sysfs name for specific device.
2851
2852 @param idVendor Vendor ID to search in sysfs directory.
2853 @param idProduct Product ID to search in sysfs directory.
2854
2855 @return Usb node names in /sys/bus/usb/drivers/usb/ that match.
2856 """
2857 # Look for matching file and cut at position 7 to get dir name.
2858 grep_cmd = 'grep {} /sys/bus/usb/drivers/usb/*/{} | cut -f 7 -d /'
2859
2860 vendor_cmd = grep_cmd.format(idVendor, 'idVendor')
2861 product_cmd = grep_cmd.format(idProduct, 'idProduct')
2862
2863 # Use uniq -d to print duplicate line from both command
2864 cmd = 'sort <({}) <({}) | uniq -d'.format(vendor_cmd, product_cmd)
2865
2866 return self.run(cmd, ignore_status=True).stdout.strip().split('\n')
2867
2868
2869 def bind_usb_device(self, usb_node):
2870 """
2871 Bind usb device
2872
2873 @param usb_node Node name in /sys/bus/usb/drivers/usb/
2874 """
2875 cmd = 'echo {} > /sys/bus/usb/drivers/usb/bind'.format(usb_node)
2876 self.run(cmd, ignore_status=True)
2877
2878
2879 def unbind_usb_device(self, usb_node):
2880 """
2881 Unbind usb device
2882
2883 @param usb_node Node name in /sys/bus/usb/drivers/usb/
2884 """
2885 cmd = 'echo {} > /sys/bus/usb/drivers/usb/unbind'.format(usb_node)
2886 self.run(cmd, ignore_status=True)
2887
2888
2889 def get_wlan_ip(self):
2890 """
2891 Get ip address of wlan interface.
2892
2893 @return ip address of wlan or empty string if wlan is not connected.
2894 """
2895 cmds = [
2896 'iw dev', # List wlan physical device
2897 'grep Interface', # Grep only interface name
2898 'cut -f 2 -d" "', # Cut the name part
2899 'xargs ifconfig', # Feed it to ifconfig to get ip
2900 'grep -oE "inet [0-9.]+"', # Grep only ipv4
2901 'cut -f 2 -d " "' # Cut the ip part
2902 ]
2903 return self.run(' | '.join(cmds), ignore_status=True).stdout.strip()
Puthikorn Voravootivatcd0dc9e2020-01-22 14:22:22 -08002904
2905 def connect_to_wifi(self, ssid, passphrase=None, security=None):
2906 """
2907 Connect to wifi network
2908
2909 @param ssid SSID of the wifi network.
2910 @param passphrase Passphrase of the wifi network. None if not existed.
2911 @param security Security of the wifi network. Default to "psk" if
2912 passphase is given without security. Possible values
2913 are "none", "psk", "802_1x".
2914
2915 @return True if succeed, False if not.
2916 """
2917 cmd = '/usr/local/autotest/cros/scripts/wifi connect ' + ssid
2918 if passphrase:
2919 cmd += ' ' + passphrase
2920 if security:
2921 cmd += ' ' + security
2922 return self.run(cmd, ignore_status=True).exit_status == 0
Otabek Kasimov6825b762020-06-23 23:42:44 -07002923
2924 def get_device_repair_state(self):
2925 """Get device repair state"""
2926 return self._device_repair_state
2927
Otabek Kasimov44273d22021-02-26 17:13:24 -08002928 def is_marked_for_replacement(self):
2929 """Verify if device was marked for replacemnet during admin task."""
2930 expected_state = cros_constants.DEVICE_STATE_NEEDS_REPLACEMENT
2931 return self.get_device_repair_state() == expected_state
2932
Otabek Kasimov7cad9ce2020-07-28 14:58:48 -07002933 def set_device_repair_state(self, state, resultdir=None):
Otabek Kasimov6825b762020-06-23 23:42:44 -07002934 """Set device repair state.
2935
2936 The special device state will be written to the 'dut_state.repair'
Otabek Kasimov7cad9ce2020-07-28 14:58:48 -07002937 file in result directory. The file will be read by Lucifer. The
2938 file will not be created if result directory not specified.
2939
2940 @params state: The new state for the device.
2941 @params resultdir: The path to result directory. If path not provided
2942 will be attempt to get retrieve it from job
2943 if present.
Otabek Kasimov6825b762020-06-23 23:42:44 -07002944 """
Otabek Kasimov7cad9ce2020-07-28 14:58:48 -07002945 resultdir = resultdir or getattr(self.job, 'resultdir', '')
2946 if resultdir:
2947 target = os.path.join(resultdir, 'dut_state.repair')
Otabek Kasimov6825b762020-06-23 23:42:44 -07002948 common_utils.open_write_close(target, state)
Otabek Kasimov7cad9ce2020-07-28 14:58:48 -07002949 logging.info('Set device state as %s. '
2950 'Created dut_state.repair file.', state)
Otabek Kasimov6825b762020-06-23 23:42:44 -07002951 else:
2952 logging.debug('Cannot write the device state due missing info '
2953 'about result dir.')
2954 self._device_repair_state = state
2955
Otabek Kasimov7cad9ce2020-07-28 14:58:48 -07002956 def set_device_needs_replacement(self, resultdir=None):
2957 """Set device as required replacement.
2958
2959 @params resultdir: The path to result directory. If path not provided
2960 will be attempt to get retrieve it from job
2961 if present.
2962 """
2963 self.set_device_repair_state(
2964 cros_constants.DEVICE_STATE_NEEDS_REPLACEMENT,
2965 resultdir=resultdir)
2966
Otabek Kasimovc7fa5072021-01-20 21:24:18 -08002967 def _dut_is_accessible_by_verifier(self):
2968 """Check if DUT accessible by SSH or PING verifier.
Otabek Kasimov86062d02020-11-17 13:30:22 -08002969
Otabek Kasimovc7fa5072021-01-20 21:24:18 -08002970 @returns: bool, True - verifier marked as success.
2971 False - result not reachable, verifier did not success.
Otabek Kasimov86062d02020-11-17 13:30:22 -08002972 """
2973 if not self._repair_strategy:
2974 return False
Otabek Kasimovc7fa5072021-01-20 21:24:18 -08002975 dut_ssh = self._repair_strategy.verifier_is_good('ssh')
2976 dut_ping = self._repair_strategy.verifier_is_good('ping')
2977 return dut_ssh == hosts.VERIFY_SUCCESS or dut_ssh == hosts.VERIFY_SUCCESS
Otabek Kasimov86062d02020-11-17 13:30:22 -08002978
Otabek Kasimovd48389b2020-12-07 02:38:34 -08002979 def _stat_if_pingable_but_not_sshable(self):
2980 """Check if DUT pingable but failed SSH verifier."""
2981 if not self._repair_strategy:
2982 return
Derek Beckett5f4edf02021-07-27 14:56:44 -07002983 self._repair_strategy.verifier_is_good('ssh')
2984 self._repair_strategy.verifier_is_good('ping')
Otabek Kasimovd48389b2020-12-07 02:38:34 -08002985
Otabek Kasimov7cad9ce2020-07-28 14:58:48 -07002986 def try_set_device_needs_manual_repair(self):
Otabek Kasimov6825b762020-06-23 23:42:44 -07002987 """Check if device require manual attention to be fixed.
2988
2989 The state 'needs_manual_repair' can be set when auto repair cannot
2990 fix the device due hardware or cable issues.
2991 """
2992 # ignore the logic if state present
2993 # state can be set by any cros repair actions
Otabek Kasimov86062d02020-11-17 13:30:22 -08002994 if self.get_device_repair_state():
Otabek Kasimov6825b762020-06-23 23:42:44 -07002995 return
Otabek Kasimovc7fa5072021-01-20 21:24:18 -08002996 if self._dut_is_accessible_by_verifier():
2997 # DUT is accessible and we still have many options to repair it.
Otabek Kasimovc9812582020-10-08 18:52:52 -07002998 return
Otabek Kasimov9189ede2020-11-09 14:08:58 -08002999 needs_manual_repair = False
3000 dhp = self.health_profile
Otabek Kasimov69253822020-11-24 10:52:27 -08003001 if dhp and dhp.get_repair_fail_count() > 49:
3002 # 42 = 6 times during 7 days. (every 4 hour repair)
3003 # round up to 50 in case somebody will run some attempt on it.
Otabek Kasimovc9812582020-10-08 18:52:52 -07003004 logging.info(
Otabek Kasimov9189ede2020-11-09 14:08:58 -08003005 'DUT is not sshable and fail %s times.'
Otabek Kasimov69253822020-11-24 10:52:27 -08003006 ' Limit to try repair is 50 times',
Otabek Kasimov9189ede2020-11-09 14:08:58 -08003007 dhp.get_repair_fail_count())
3008 needs_manual_repair = True
3009
3010 if not needs_manual_repair:
3011 # We cannot ssh to the DUT and we have hardware or set-up issues
3012 # with servo then we need request manual repair for the DUT.
3013 servo_state_required_manual_fix = [
3014 servo_constants.SERVO_STATE_DUT_NOT_CONNECTED,
3015 servo_constants.SERVO_STATE_NEED_REPLACEMENT,
3016 ]
3017 if self.get_servo_state() in servo_state_required_manual_fix:
3018 logging.info(
3019 'DUT required manual repair because it is not sshable'
3020 ' and possible have setup issue with Servo. Please'
3021 ' verify all connections and present of devices.')
3022 needs_manual_repair = True
3023
3024 if needs_manual_repair:
Otabek Kasimovc9812582020-10-08 18:52:52 -07003025 self.set_device_repair_state(
3026 cros_constants.DEVICE_STATE_NEEDS_MANUAL_REPAIR)
Otabek Kasimov42506d02020-07-29 14:44:57 -07003027
Otabek Kasimov86062d02020-11-17 13:30:22 -08003028 def _reboot_labstation_if_needed(self):
3029 """Place request to reboot the labstation if DUT is not sshable.
3030
3031 @returns: None
3032 """
Puthikorn Voravootivat58f739b2021-01-07 08:28:36 -08003033 message_prefix = "Don't need to request servo-host reboot"
Otabek Kasimovc7fa5072021-01-20 21:24:18 -08003034 if self._dut_is_accessible_by_verifier():
Otabek Kasimov86062d02020-11-17 13:30:22 -08003035 return
3036 if not self._servo_host:
Puthikorn Voravootivat58f739b2021-01-07 08:28:36 -08003037 logging.debug('%s as it not initialized', message_prefix)
Otabek Kasimov86062d02020-11-17 13:30:22 -08003038 return
3039 if not self._servo_host.is_up_fast():
Puthikorn Voravootivat58f739b2021-01-07 08:28:36 -08003040 logging.debug('%s as servo-host is not sshable', message_prefix)
Otabek Kasimov86062d02020-11-17 13:30:22 -08003041 return
3042 if not self._servo_host.is_labstation():
3043 logging.debug('Servo_v3 is not requested to reboot for the DUT')
3044 return
3045 usb_path = self._servo_host.get_main_servo_usb_path()
3046 if usb_path:
3047 connected_port = os.path.basename(os.path.normpath(usb_path))
3048 # Directly connected servo to the labstation looks like '1-5.3'
3049 # and when connected by hub - '1-5.2.3' or '1-5.2.1.3'. Where:
3050 # - '1-5' - port on labstation
3051 # - '2' or '2.1' - port on the hub or smart-hub
3052 # - '3' - port on servo hub
3053 if len(connected_port.split('.')) > 2:
Puthikorn Voravootivat58f739b2021-01-07 08:28:36 -08003054 logging.debug('%s as servo connected by hub', message_prefix)
Otabek Kasimov86062d02020-11-17 13:30:22 -08003055 return
3056 self._servo_host.request_reboot()
3057 logging.info('Requested labstation reboot because DUT is not sshable')
3058
Otabek Kasimov42506d02020-07-29 14:44:57 -07003059 def is_file_system_writable(self, testdirs=None):
3060 """Check is the file systems are writable.
3061
3062 The standard linux response to certain unexpected file system errors
3063 (including hardware errors in block devices) is to change the file
3064 system status to read-only. This checks that that hasn't happened.
3065
3066 @param testdirs: List of directories to check. If no data provided
3067 then '/mnt/stateful_partition' and '/var/tmp'
3068 directories will be checked.
3069
3070 @returns boolean whether file-system writable.
3071 """
3072 def _check_dir(testdir):
3073 # check if we can create a file
3074 filename = os.path.join(testdir, 'writable_my_test_file')
3075 command = 'touch %s && rm %s' % (filename, filename)
3076 rv = self.run(command=command,
3077 timeout=30,
3078 ignore_status=True)
3079 is_writable = rv.exit_status == 0
3080 if not is_writable:
3081 logging.info('Cannot create a file in "%s"!'
3082 ' Probably the FS is read-only', testdir)
3083 logging.info("FileSystem is not writable!")
3084 return False
3085 return True
3086
3087 if not testdirs or len(testdirs) == 0:
3088 # N.B. Order matters here: Encrypted stateful is loop-mounted
3089 # from a file in unencrypted stateful, so we don't test for
3090 # errors in encrypted stateful if unencrypted fails.
3091 testdirs = ['/mnt/stateful_partition', '/var/tmp']
3092
3093 for dir in testdirs:
3094 # loop will be stopped if any directory fill fail the check
3095 try:
3096 if not _check_dir(dir):
3097 return False
3098 except Exception as e:
3099 # here expected only timeout error, all other will
3100 # be catch by 'ignore_status=True'
3101 logging.debug('Fail to check %s to write in it', dir)
3102 return False
3103 return True
Garry Wang1a493d82020-08-31 21:01:19 -07003104
Dana Goyettec172b172020-07-29 16:26:15 -07003105 def blocking_sync(self, freeze_for_reset=False):
3106 """Sync root device and internal device, via script.
3107
3108 The actual calls end up logged by the run() call, since they're printed
3109 to stdout/stderr in the script.
3110
3111 @param freeze_for_reset: if True, prepare for reset by blocking writes
3112 (only if enable_fs_sync_fsfreeze=True)
3113 """
3114
3115 if freeze_for_reset and self.USE_FSFREEZE:
3116 logging.info('Blocking sync and freeze')
3117 elif freeze_for_reset:
3118 logging.info('Blocking sync for reset')
3119 else:
3120 logging.info('Blocking sync')
3121
3122 # client/bin is installed on the DUT as /usr/local/autotest/bin
3123 sync_cmd = '/usr/local/autotest/bin/fs_sync.py'
3124 if freeze_for_reset and self.USE_FSFREEZE:
3125 sync_cmd += ' --freeze'
3126 return self.run(sync_cmd)
3127
Garry Wanga2e78172020-09-09 23:49:07 -07003128 def set_health_profile_dut_state(self, state):
3129 if not self.health_profile:
3130 logging.debug('Device health profile is not initialized, skip'
3131 ' set dut state.')
3132 return
3133 reset_counters = state in profile_constants.STATES_NEED_RESET_COUNTER
3134 self.health_profile.update_dut_state(state, reset_counters)
Garry Wang53fc8f32020-09-18 13:30:08 -07003135
3136 def require_snk_mode_in_recovery(self):
3137 """Check whether we need to switch servo_v4 role to snk when
3138 booting into recovery mode. (See crbug.com/1129165)
3139 """
Garry Wanga8739cc2020-10-30 00:49:23 -07003140 has_battery = True
3141 # Determine if the host has battery based on host_info first.
3142 power_info = self.host_info_store.get().get_label_value('power')
3143 if power_info:
3144 has_battery = power_info == 'battery'
3145 elif self.is_up_fast():
3146 # when running local tests host_info is not available, so we
3147 # need to determine whether the host has battery by checking
3148 # from host side.
3149 logging.debug('Label `power` is not found in host_info, checking'
3150 ' if the host has battery from host side.')
3151 has_battery = self.has_battery()
3152
3153 if not has_battery:
Garry Wang53fc8f32020-09-18 13:30:08 -07003154 logging.info(
3155 '%s does not has battery, snk mode is not needed'
3156 ' for recovery.', self.hostname)
3157 return False
Garry Wanga8739cc2020-10-30 00:49:23 -07003158
Garry Wang53fc8f32020-09-18 13:30:08 -07003159 if not self.servo.supports_built_in_pd_control():
3160 logging.info('Power delivery is not supported on this servo, snk'
3161 ' mode is not needed for recovery.')
3162 return False
3163 try:
Garry Wang53fc8f32020-09-18 13:30:08 -07003164 battery_percent = self.servo.get('battery_charge_percent')
Otabek Kasimov58e22562020-11-03 17:17:41 -08003165 if battery_percent < cros_constants.MIN_BATTERY_LEVEL:
Garry Wang53fc8f32020-09-18 13:30:08 -07003166 logging.info(
3167 'Current battery level %s%% below %s%% threshold, we'
3168 ' will attempt to boot host in recovery mode without'
3169 ' changing servo to snk mode. Please note the host may'
3170 ' not able to see usb drive in recovery mode later due'
3171 ' to servo not in snk mode.', battery_percent,
Otabek Kasimov58e22562020-11-03 17:17:41 -08003172 cros_constants.MIN_BATTERY_LEVEL)
Garry Wang53fc8f32020-09-18 13:30:08 -07003173 return False
3174 except Exception as e:
3175 logging.info(
3176 'Unexpected error occurred when getting'
3177 ' battery_charge_percent from servo; %s', str(e))
3178 return False
3179 return True
Otabek Kasimov382c3bb2020-10-28 13:22:45 -07003180
3181 def _set_servo_topology(self):
3182 """Set servo-topology info to the host-info."""
3183 logging.debug('Try to save servo topology to host-info.')
3184 if not self._servo_host:
Greg Edelstonff2665d2021-04-21 14:32:27 -06003185 logging.debug('Servo host is not initialized.')
Otabek Kasimovfe41e2d2021-02-14 20:48:52 -08003186 return
3187 if not self.is_servo_in_working_state():
3188 logging.debug('Is servo is not in working state then'
3189 ' update topology is not allowed.')
Otabek Kasimov382c3bb2020-10-28 13:22:45 -07003190 return
3191 if not self._servo_host.is_servo_topology_supported():
Otabek Kasimovfe41e2d2021-02-14 20:48:52 -08003192 logging.debug('Servo-topology is not supported.')
Otabek Kasimov382c3bb2020-10-28 13:22:45 -07003193 return
3194 servo_topology = self._servo_host.get_topology()
3195 if not servo_topology or servo_topology.is_empty():
Otabek Kasimovfe41e2d2021-02-14 20:48:52 -08003196 logging.debug('Servo topology is empty')
Otabek Kasimov382c3bb2020-10-28 13:22:45 -07003197 return
3198 servo_topology.save(self.host_info_store)