blob: 2182932f0199b52b9ccb1e3d0975ab375068df5e [file] [log] [blame]
J. Richard Barnette24adbf42012-04-11 15:04:53 -07001# Copyright (c) 2012 The Chromium OS Authors. All rights reserved.
Dale Curtisaa5eedb2011-08-23 16:18:52 -07002# Use of this source code is governed by a BSD-style license that can be
3# found in the LICENSE file.
4
J. Richard Barnette1d78b012012-05-15 13:56:30 -07005import logging
Dan Shi0f466e82013-02-22 15:44:58 -08006import os
Simran Basid5e5e272012-09-24 15:23:59 -07007import re
Vincent Palatindf2372c2016-10-07 17:03:00 +02008import sys
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07009import time
10
mussa584b4462014-06-20 15:13:28 -070011import common
J. Richard Barnette45e93de2012-04-11 17:24:15 -070012from autotest_lib.client.bin import utils
Dan Shi9cb0eec2014-06-03 09:04:50 -070013from autotest_lib.client.common_lib import autotemp
Richard Barnette0c73ffc2012-11-19 15:21:18 -080014from autotest_lib.client.common_lib import error
15from autotest_lib.client.common_lib import global_config
J. Richard Barnette91137f02016-03-10 16:52:26 -080016from autotest_lib.client.common_lib import hosts
Dan Shi549fb822015-03-24 18:01:11 -070017from autotest_lib.client.common_lib import lsbrelease_utils
J. Richard Barnette45e93de2012-04-11 17:24:15 -070018from autotest_lib.client.common_lib.cros import autoupdater
Richard Barnette03a0c132012-11-05 12:40:35 -080019from autotest_lib.client.common_lib.cros import dev_server
Gabe Blackb72f4fb2015-01-20 16:47:13 -080020from autotest_lib.client.common_lib.cros.graphite import autotest_es
Hsinyu Chaoe0b08e62015-08-11 10:50:37 +000021from autotest_lib.client.cros import constants as client_constants
J. Richard Barnette84890bd2014-02-21 11:05:47 -080022from autotest_lib.client.cros import cros_ui
Kevin Chenga328da62016-03-31 10:49:04 -070023from autotest_lib.client.cros.audio import cras_utils
24from autotest_lib.client.cros.input_playback import input_playback
25from autotest_lib.client.cros.video import constants as video_test_constants
Simran Basi5ace6f22016-01-06 17:30:44 -080026from autotest_lib.server import afe_utils
MK Ryu35d661e2014-09-25 17:44:10 -070027from autotest_lib.server import autoserv_parser
28from autotest_lib.server import autotest
29from autotest_lib.server import constants
Dan Shia1ecd5c2013-06-06 11:21:31 -070030from autotest_lib.server import utils as server_utils
Dan Shi9cb0eec2014-06-03 09:04:50 -070031from autotest_lib.server.cros import provision
Scott Zawalski89c44dd2013-02-26 09:28:02 -050032from autotest_lib.server.cros.dynamic_suite import constants as ds_constants
Simran Basi5e6339a2013-03-21 11:34:32 -070033from autotest_lib.server.cros.dynamic_suite import tools, frontend_wrappers
Dan Shi9cb0eec2014-06-03 09:04:50 -070034from autotest_lib.server.cros.faft.config.config import Config as FAFTConfig
Scottfe06ed82015-11-05 17:15:01 -080035from autotest_lib.server.cros.servo import plankton
Fang Deng96667ca2013-08-01 17:46:18 -070036from autotest_lib.server.hosts import abstract_ssh
Kevin Chenga2619dc2016-03-28 11:42:08 -070037from autotest_lib.server.hosts import base_label
38from autotest_lib.server.hosts import cros_label
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +080039from autotest_lib.server.hosts import chameleon_host
J. Richard Barnettea7e7fdc2016-02-12 12:35:36 -080040from autotest_lib.server.hosts import cros_repair
Scottfe06ed82015-11-05 17:15:01 -080041from autotest_lib.server.hosts import plankton_host
Fang Deng5d518f42013-08-02 14:04:32 -070042from autotest_lib.server.hosts import servo_host
Simran Basidcff4252012-11-20 16:13:20 -080043from autotest_lib.site_utils.rpm_control_system import rpm_client
Simran Basid5e5e272012-09-24 15:23:59 -070044
Simran Basi382506b2016-09-13 14:58:15 -070045# In case cros_host is being ran via SSP on an older Moblab version with an
46# older chromite version.
47try:
48 from chromite.lib import metrics
Dan Shi5e2efb72017-02-07 11:40:23 -080049except ImportError:
50 metrics = utils.metrics_mock
51
Simran Basid5e5e272012-09-24 15:23:59 -070052
Dan Shib8540a52015-07-16 14:18:23 -070053CONFIG = global_config.global_config
xixuanec801e32016-08-25 10:20:22 -070054ENABLE_DEVSERVER_TRIGGER_AUTO_UPDATE = CONFIG.get_config_value(
55 'CROS', 'enable_devserver_trigger_auto_update', type=bool,
56 default=False)
Dan Shib8540a52015-07-16 14:18:23 -070057
Kevin Chenga328da62016-03-31 10:49:04 -070058LUCID_SLEEP_BOARDS = ['samus', 'lulu']
59
Dan Shid07ee2e2015-09-24 14:49:25 -070060
beepsc87ff602013-07-31 21:53:00 -070061class FactoryImageCheckerException(error.AutoservError):
62 """Exception raised when an image is a factory image."""
63 pass
64
65
Fang Deng0ca40e22013-08-27 17:47:44 -070066class CrosHost(abstract_ssh.AbstractSSHHost):
J. Richard Barnette45e93de2012-04-11 17:24:15 -070067 """Chromium OS specific subclass of Host."""
68
Simran Basi5ace6f22016-01-06 17:30:44 -080069 VERSION_PREFIX = provision.CROS_VERSION_PREFIX
70
J. Richard Barnette45e93de2012-04-11 17:24:15 -070071 _parser = autoserv_parser.autoserv_parser
Scott Zawalski62bacae2013-03-05 10:40:32 -050072 _AFE = frontend_wrappers.RetryingAFE(timeout_min=5, delay_sec=10)
xixuanec801e32016-08-25 10:20:22 -070073 support_devserver_provision = ENABLE_DEVSERVER_TRIGGER_AUTO_UPDATE
J. Richard Barnette45e93de2012-04-11 17:24:15 -070074
Richard Barnette03a0c132012-11-05 12:40:35 -080075 # Timeout values (in seconds) associated with various Chrome OS
76 # state changes.
J. Richard Barnette134ec2c2012-04-25 12:59:37 -070077 #
Richard Barnette0c73ffc2012-11-19 15:21:18 -080078 # In general, a good rule of thumb is that the timeout can be up
79 # to twice the typical measured value on the slowest platform.
80 # The times here have not necessarily been empirically tested to
81 # meet this criterion.
J. Richard Barnetteeb69d722012-06-18 17:29:44 -070082 #
83 # SLEEP_TIMEOUT: Time to allow for suspend to memory.
Richard Barnette0c73ffc2012-11-19 15:21:18 -080084 # RESUME_TIMEOUT: Time to allow for resume after suspend, plus
85 # time to restart the netwowrk.
J. Richard Barnette84890bd2014-02-21 11:05:47 -080086 # SHUTDOWN_TIMEOUT: Time to allow for shut down.
J. Richard Barnetteeb69d722012-06-18 17:29:44 -070087 # BOOT_TIMEOUT: Time to allow for boot from power off. Among
Richard Barnette0c73ffc2012-11-19 15:21:18 -080088 # other things, this must account for the 30 second dev-mode
J. Richard Barnette417cc792015-10-01 09:56:36 -070089 # screen delay, time to start the network on the DUT, and the
90 # ssh timeout of 120 seconds.
J. Richard Barnetteeb69d722012-06-18 17:29:44 -070091 # USB_BOOT_TIMEOUT: Time to allow for boot from a USB device,
Richard Barnette0c73ffc2012-11-19 15:21:18 -080092 # including the 30 second dev-mode delay and time to start the
J. Richard Barnetted4649c62013-03-06 17:42:27 -080093 # network.
beepsf079cfb2013-09-18 17:49:51 -070094 # INSTALL_TIMEOUT: Time to allow for chromeos-install.
J. Richard Barnette84890bd2014-02-21 11:05:47 -080095 # POWERWASH_BOOT_TIMEOUT: Time to allow for a reboot that
96 # includes powerwash.
J. Richard Barnetteeb69d722012-06-18 17:29:44 -070097
98 SLEEP_TIMEOUT = 2
J. Richard Barnetted4649c62013-03-06 17:42:27 -080099 RESUME_TIMEOUT = 10
Tom Wai-Hong Tamfe005c22014-12-03 09:25:44 +0800100 SHUTDOWN_TIMEOUT = 10
J. Richard Barnette417cc792015-10-01 09:56:36 -0700101 BOOT_TIMEOUT = 150
J. Richard Barnette5bab5f52015-08-03 13:14:38 -0700102 USB_BOOT_TIMEOUT = 300
J. Richard Barnette7817b052014-08-28 09:47:29 -0700103 INSTALL_TIMEOUT = 480
Dan Shi2c88eed2013-11-12 10:18:38 -0800104 POWERWASH_BOOT_TIMEOUT = 60
Chris Sosab76e0ee2013-05-22 16:55:41 -0700105
Dan Shica503482015-03-30 17:23:25 -0700106 # Minimum OS version that supports server side packaging. Older builds may
107 # not have server side package built or with Autotest code change to support
108 # server-side packaging.
Dan Shib8540a52015-07-16 14:18:23 -0700109 MIN_VERSION_SUPPORT_SSP = CONFIG.get_config_value(
Dan Shiced09e42015-04-17 16:09:34 -0700110 'AUTOSERV', 'min_version_support_ssp', type=int)
Dan Shica503482015-03-30 17:23:25 -0700111
J. Richard Barnette84890bd2014-02-21 11:05:47 -0800112 # REBOOT_TIMEOUT: How long to wait for a reboot.
113 #
Chris Sosab76e0ee2013-05-22 16:55:41 -0700114 # We have a long timeout to ensure we don't flakily fail due to other
115 # issues. Shorter timeouts are vetted in platform_RebootAfterUpdate.
Simran Basi1160e2c2013-10-04 16:00:24 -0700116 # TODO(sbasi - crbug.com/276094) Restore to 5 mins once the 'host did not
117 # return from reboot' bug is solved.
118 REBOOT_TIMEOUT = 480
Chris Sosab76e0ee2013-05-22 16:55:41 -0700119
Ismail Noorbasha07fdb612013-02-14 14:13:31 -0800120 # _USB_POWER_TIMEOUT: Time to allow for USB to power toggle ON and OFF.
121 # _POWER_CYCLE_TIMEOUT: Time to allow for manual power cycle.
122 _USB_POWER_TIMEOUT = 5
123 _POWER_CYCLE_TIMEOUT = 10
124
Dan Shib8540a52015-07-16 14:18:23 -0700125 _RPM_RECOVERY_BOARDS = CONFIG.get_config_value('CROS',
Richard Barnette82c35912012-11-20 10:09:10 -0800126 'rpm_recovery_boards', type=str).split(',')
127
Richard Barnette82c35912012-11-20 10:09:10 -0800128 _LAB_MACHINE_FILE = '/mnt/stateful_partition/.labmachine'
Fang Dengdeba14f2014-11-14 11:54:09 -0800129 _RPM_HOSTNAME_REGEX = ('chromeos(\d+)(-row(\d+))?-rack(\d+[a-z]*)'
130 '-host(\d+)')
Kevin Chenga328da62016-03-31 10:49:04 -0700131 _LIGHTSENSOR_FILES = [ "in_illuminance0_input",
132 "in_illuminance_input",
133 "in_illuminance0_raw",
134 "in_illuminance_raw",
135 "illuminance0_input"]
136 _LIGHTSENSOR_SEARCH_DIR = '/sys/bus/iio/devices'
J. Richard Barnette134ec2c2012-04-25 12:59:37 -0700137
J. Richard Barnetteb6de7e32013-02-14 13:28:04 -0800138 # Constants used in ping_wait_up() and ping_wait_down().
139 #
140 # _PING_WAIT_COUNT is the approximate number of polling
141 # cycles to use when waiting for a host state change.
142 #
143 # _PING_STATUS_DOWN and _PING_STATUS_UP are names used
144 # for arguments to the internal _ping_wait_for_status()
145 # method.
146 _PING_WAIT_COUNT = 40
147 _PING_STATUS_DOWN = False
148 _PING_STATUS_UP = True
149
Ismail Noorbasha07fdb612013-02-14 14:13:31 -0800150 # Allowed values for the power_method argument.
151
152 # POWER_CONTROL_RPM: Passed as default arg for power_off/on/cycle() methods.
153 # POWER_CONTROL_SERVO: Used in set_power() and power_cycle() methods.
154 # POWER_CONTROL_MANUAL: Used in set_power() and power_cycle() methods.
155 POWER_CONTROL_RPM = 'RPM'
156 POWER_CONTROL_SERVO = 'servoj10'
157 POWER_CONTROL_MANUAL = 'manual'
158
159 POWER_CONTROL_VALID_ARGS = (POWER_CONTROL_RPM,
160 POWER_CONTROL_SERVO,
161 POWER_CONTROL_MANUAL)
Richard Barnette0c73ffc2012-11-19 15:21:18 -0800162
Simran Basi5e6339a2013-03-21 11:34:32 -0700163 _RPM_OUTLET_CHANGED = 'outlet_changed'
164
Dan Shi9cb0eec2014-06-03 09:04:50 -0700165 # URL pattern to download firmware image.
Dan Shib8540a52015-07-16 14:18:23 -0700166 _FW_IMAGE_URL_PATTERN = CONFIG.get_config_value(
Dan Shi9cb0eec2014-06-03 09:04:50 -0700167 'CROS', 'firmware_url_pattern', type=str)
beeps687243d2013-07-18 15:29:27 -0700168
J. Richard Barnette91137f02016-03-10 16:52:26 -0800169
170 # A flag file to indicate provision failures. The file is created
171 # at the start of any AU procedure (see `machine_install()`). The
172 # file's location in stateful means that on successul update it will
173 # be removed. Thus, if this file exists, it indicates that we've
174 # tried and failed in a previous attempt to update.
175 PROVISION_FAILED = '/var/tmp/provision_failed'
MK Ryu35d661e2014-09-25 17:44:10 -0700176
MK Ryu35d661e2014-09-25 17:44:10 -0700177
J. Richard Barnette964fba02012-10-24 17:34:29 -0700178 @staticmethod
beeps46dadc92013-11-07 14:07:10 -0800179 def check_host(host, timeout=10):
180 """
181 Check if the given host is a chrome-os host.
182
183 @param host: An ssh host representing a device.
184 @param timeout: The timeout for the run command.
185
186 @return: True if the host device is chromeos.
187
beeps46dadc92013-11-07 14:07:10 -0800188 """
189 try:
Simran Basi933c8af2015-04-29 14:05:07 -0700190 result = host.run(
191 'grep -q CHROMEOS /etc/lsb-release && '
192 '! test -f /mnt/stateful_partition/.android_tester && '
193 '! grep -q moblab /etc/lsb-release',
194 ignore_status=True, timeout=timeout)
beeps46dadc92013-11-07 14:07:10 -0800195 except (error.AutoservRunError, error.AutoservSSHTimeout):
196 return False
197 return result.exit_status == 0
198
199
200 @staticmethod
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +0800201 def get_chameleon_arguments(args_dict):
202 """Extract chameleon options from `args_dict` and return the result.
203
204 Recommended usage:
205 ~~~~~~~~
206 args_dict = utils.args_to_dict(args)
207 chameleon_args = hosts.CrosHost.get_chameleon_arguments(args_dict)
208 host = hosts.create_host(machine, chameleon_args=chameleon_args)
209 ~~~~~~~~
210
211 @param args_dict Dictionary from which to extract the chameleon
212 arguments.
213 """
Allen Li083866b2016-08-18 10:07:10 -0700214 return {key: args_dict[key]
215 for key in ('chameleon_host', 'chameleon_port')
216 if key in args_dict}
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +0800217
218
219 @staticmethod
Scottfe06ed82015-11-05 17:15:01 -0800220 def get_plankton_arguments(args_dict):
221 """Extract chameleon options from `args_dict` and return the result.
222
223 Recommended usage:
224 ~~~~~~~~
225 args_dict = utils.args_to_dict(args)
Richard Barnettee519dcd2016-08-15 17:37:17 -0700226 plankton_args = hosts.CrosHost.get_plankton_arguments(args_dict)
227 host = hosts.create_host(machine, plankton_args=plankton_args)
Scottfe06ed82015-11-05 17:15:01 -0800228 ~~~~~~~~
229
230 @param args_dict Dictionary from which to extract the plankton
231 arguments.
232 """
Allen Li083866b2016-08-18 10:07:10 -0700233 return {key: args_dict[key]
234 for key in ('plankton_host', 'plankton_port')
235 if key in args_dict}
Scottfe06ed82015-11-05 17:15:01 -0800236
237
238 @staticmethod
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +0800239 def get_servo_arguments(args_dict):
240 """Extract servo options from `args_dict` and return the result.
J. Richard Barnette7214e0b2013-02-06 15:20:49 -0800241
242 Recommended usage:
243 ~~~~~~~~
244 args_dict = utils.args_to_dict(args)
Fang Deng0ca40e22013-08-27 17:47:44 -0700245 servo_args = hosts.CrosHost.get_servo_arguments(args_dict)
J. Richard Barnette7214e0b2013-02-06 15:20:49 -0800246 host = hosts.create_host(machine, servo_args=servo_args)
247 ~~~~~~~~
248
249 @param args_dict Dictionary from which to extract the servo
250 arguments.
251 """
Richard Barnettee519dcd2016-08-15 17:37:17 -0700252 servo_attrs = (servo_host.SERVO_HOST_ATTR,
253 servo_host.SERVO_PORT_ATTR,
254 servo_host.SERVO_BOARD_ATTR)
Allen Li083866b2016-08-18 10:07:10 -0700255 return {key: args_dict[key]
256 for key in servo_attrs
257 if key in args_dict}
J. Richard Barnette134ec2c2012-04-25 12:59:37 -0700258
J. Richard Barnette964fba02012-10-24 17:34:29 -0700259
J. Richard Barnette91137f02016-03-10 16:52:26 -0800260 def _initialize(self, hostname, chameleon_args=None, servo_args=None,
261 plankton_args=None, try_lab_servo=False,
Richard Barnette9a26ad62016-06-10 12:03:08 -0700262 try_servo_repair=False,
J. Richard Barnette91137f02016-03-10 16:52:26 -0800263 ssh_verbosity_flag='', ssh_options='',
Fang Dengd1c2b732013-08-20 12:59:46 -0700264 *args, **dargs):
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +0800265 """Initialize superclasses, |self.chameleon|, and |self.servo|.
J. Richard Barnette67ccb872012-04-19 16:34:56 -0700266
Fang Denge545abb2014-12-30 18:43:47 -0800267 This method will attempt to create the test-assistant object
268 (chameleon/servo) when it is needed by the test. Check
269 the docstring of chameleon_host.create_chameleon_host and
270 servo_host.create_servo_host for how this is determined.
Fang Deng5d518f42013-08-02 14:04:32 -0700271
Fang Denge545abb2014-12-30 18:43:47 -0800272 @param hostname: Hostname of the dut.
273 @param chameleon_args: A dictionary that contains args for creating
274 a ChameleonHost. See chameleon_host for details.
275 @param servo_args: A dictionary that contains args for creating
276 a ServoHost object. See servo_host for details.
Richard Barnette9a26ad62016-06-10 12:03:08 -0700277 @param try_lab_servo: When true, indicates that an attempt should
278 be made to create a ServoHost for a DUT in
279 the test lab, even if not required by
280 `servo_args`. See servo_host for details.
281 @param try_servo_repair: If a servo host is created, check it
282 with `repair()` rather than `verify()`.
Fang Denge545abb2014-12-30 18:43:47 -0800283 See servo_host for details.
284 @param ssh_verbosity_flag: String, to pass to the ssh command to control
285 verbosity.
286 @param ssh_options: String, other ssh options to pass to the ssh
287 command.
J. Richard Barnette67ccb872012-04-19 16:34:56 -0700288 """
Fang Deng0ca40e22013-08-27 17:47:44 -0700289 super(CrosHost, self)._initialize(hostname=hostname,
J. Richard Barnette45e93de2012-04-11 17:24:15 -0700290 *args, **dargs)
J. Richard Barnette91137f02016-03-10 16:52:26 -0800291 self._repair_strategy = cros_repair.create_cros_repair_strategy()
Kevin Chenga2619dc2016-03-28 11:42:08 -0700292 self.labels = base_label.LabelRetriever(cros_label.CROS_LABELS)
J. Richard Barnettef0859852012-08-20 14:55:50 -0700293 # self.env is a dictionary of environment variable settings
294 # to be exported for commands run on the host.
295 # LIBC_FATAL_STDERR_ can be useful for diagnosing certain
296 # errors that might happen.
297 self.env['LIBC_FATAL_STDERR_'] = '1'
Fang Dengd1c2b732013-08-20 12:59:46 -0700298 self._ssh_verbosity_flag = ssh_verbosity_flag
Aviv Keshetc5947fa2013-09-04 14:06:29 -0700299 self._ssh_options = ssh_options
Fang Deng5d518f42013-08-02 14:04:32 -0700300 # TODO(fdeng): We need to simplify the
301 # process of servo and servo_host initialization.
302 # crbug.com/298432
Moja Hsu57d93a12017-01-13 17:57:52 +0800303 self._servo_host = servo_host.create_servo_host(
Richard Barnetteea3e4602016-06-10 12:36:41 -0700304 dut=self, servo_args=servo_args,
Richard Barnette9a26ad62016-06-10 12:03:08 -0700305 try_lab_servo=try_lab_servo,
306 try_servo_repair=try_servo_repair)
Richard Barnettee519dcd2016-08-15 17:37:17 -0700307 if self._servo_host is not None:
308 self.servo = self._servo_host.get_servo()
309 else:
310 self.servo = None
311
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +0800312 # TODO(waihong): Do the simplication on Chameleon too.
Tom Wai-Hong Tam3d6790d2014-04-14 16:15:47 +0800313 self._chameleon_host = chameleon_host.create_chameleon_host(
314 dut=self.hostname, chameleon_args=chameleon_args)
Scottfe06ed82015-11-05 17:15:01 -0800315 # Add plankton host if plankton args were added on command line
316 self._plankton_host = plankton_host.create_plankton_host(plankton_args)
Tom Wai-Hong Tam3d6790d2014-04-14 16:15:47 +0800317
Tom Wai-Hong Tam3d6790d2014-04-14 16:15:47 +0800318 if self._chameleon_host:
Tom Wai-Hong Tameaee3402014-01-22 08:52:10 +0800319 self.chameleon = self._chameleon_host.create_chameleon_board()
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +0800320 else:
Tom Wai-Hong Tam3d6790d2014-04-14 16:15:47 +0800321 self.chameleon = None
Fang Deng5d518f42013-08-02 14:04:32 -0700322
Scottfe06ed82015-11-05 17:15:01 -0800323 if self._plankton_host:
324 self.plankton_servo = self._plankton_host.get_servo()
325 logging.info('plankton_servo: %r', self.plankton_servo)
326 # Create the plankton object used to access the ec uart
Scott07a848f2016-01-12 15:04:52 -0800327 self.plankton = plankton.Plankton(self.plankton_servo,
328 self._plankton_host.get_servod_server_proxy())
Scottfe06ed82015-11-05 17:15:01 -0800329 else:
Scott07a848f2016-01-12 15:04:52 -0800330 self.plankton = None
Scottfe06ed82015-11-05 17:15:01 -0800331
Fang Deng5d518f42013-08-02 14:04:32 -0700332
Richard Barnette260cbd02016-10-06 12:23:28 -0700333 def _get_cros_repair_image_name(self):
Prathmesh Prabhu075fc922017-02-13 11:50:25 -0800334 info = self.host_info_store.get()
335 if info.board is None:
336 raise error.AutoservError('Cannot obtain repair image name. '
337 'No board label value found')
338
339 return afe_utils.get_stable_cros_image_name(info.board)
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500340
341
beepsdae65fd2013-07-26 16:24:41 -0700342 def verify_job_repo_url(self, tag=''):
beepscb6f1e22013-06-28 19:14:10 -0700343 """
344 Make sure job_repo_url of this host is valid.
345
joychen03eaad92013-06-26 09:55:21 -0700346 Eg: The job_repo_url "http://lmn.cd.ab.xyx:8080/static/\
beepscb6f1e22013-06-28 19:14:10 -0700347 lumpy-release/R29-4279.0.0/autotest/packages" claims to have the
348 autotest package for lumpy-release/R29-4279.0.0. If this isn't the case,
349 download and extract it. If the devserver embedded in the url is
350 unresponsive, update the job_repo_url of the host after staging it on
351 another devserver.
352
353 @param job_repo_url: A url pointing to the devserver where the autotest
354 package for this build should be staged.
beepsdae65fd2013-07-26 16:24:41 -0700355 @param tag: The tag from the server job, in the format
356 <job_id>-<user>/<hostname>, or <hostless> for a server job.
beepscb6f1e22013-06-28 19:14:10 -0700357
358 @raises DevServerException: If we could not resolve a devserver.
359 @raises AutoservError: If we're unable to save the new job_repo_url as
360 a result of choosing a new devserver because the old one failed to
361 respond to a health check.
beeps0c865032013-07-30 11:37:06 -0700362 @raises urllib2.URLError: If the devserver embedded in job_repo_url
363 doesn't respond within the timeout.
beepscb6f1e22013-06-28 19:14:10 -0700364 """
Prathmesh Prabhub9d7adf2017-03-07 18:31:03 +0000365 job_repo_url = afe_utils.get_host_attribute(self,
366 ds_constants.JOB_REPO_URL)
beepscb6f1e22013-06-28 19:14:10 -0700367 if not job_repo_url:
368 logging.warning('No job repo url set on host %s', self.hostname)
369 return
370
371 logging.info('Verifying job repo url %s', job_repo_url)
372 devserver_url, image_name = tools.get_devserver_build_from_package_url(
373 job_repo_url)
374
beeps0c865032013-07-30 11:37:06 -0700375 ds = dev_server.ImageServer(devserver_url)
beepscb6f1e22013-06-28 19:14:10 -0700376
377 logging.info('Staging autotest artifacts for %s on devserver %s',
378 image_name, ds.url())
beeps687243d2013-07-18 15:29:27 -0700379
380 start_time = time.time()
Simran Basi25e7a922014-10-31 11:56:10 -0700381 ds.stage_artifacts(image_name, ['autotest_packages'])
beeps687243d2013-07-18 15:29:27 -0700382 stage_time = time.time() - start_time
383
384 # Record how much of the verification time comes from a devserver
385 # restage. If we're doing things right we should not see multiple
386 # devservers for a given board/build/branch path.
387 try:
J. Richard Barnette3cbd76b2013-11-27 12:11:25 -0800388 board, build_type, branch = server_utils.ParseBuildName(
beeps687243d2013-07-18 15:29:27 -0700389 image_name)[:3]
J. Richard Barnette3cbd76b2013-11-27 12:11:25 -0800390 except server_utils.ParseBuildNameException:
beeps687243d2013-07-18 15:29:27 -0700391 pass
392 else:
beeps0c865032013-07-30 11:37:06 -0700393 devserver = devserver_url[
Chris Sosa65425082013-10-16 13:26:22 -0700394 devserver_url.find('/') + 2:devserver_url.rfind(':')]
beeps687243d2013-07-18 15:29:27 -0700395 stats_key = {
396 'board': board,
397 'build_type': build_type,
398 'branch': branch,
beeps0c865032013-07-30 11:37:06 -0700399 'devserver': devserver.replace('.', '_'),
beeps687243d2013-07-18 15:29:27 -0700400 }
Dan Shi5e2efb72017-02-07 11:40:23 -0800401
402 monarch_fields = {
403 'board': board,
404 'build_type': build_type,
405 # TODO(akeshet): To be consistent with most other metrics,
406 # consider changing the following field to be named
407 # 'milestone'.
408 'branch': branch,
409 'dev_server': devserver,
410 }
411 metrics.Counter(
412 'chromeos/autotest/provision/verify_url'
413 ).increment(fields=monarch_fields)
414 metrics.SecondsDistribution(
415 'chromeos/autotest/provision/verify_url_duration'
416 ).add(stage_time, fields=monarch_fields)
417
418
Dan Shicf4d2032015-03-12 15:04:21 -0700419 def stage_server_side_package(self, image=None):
420 """Stage autotest server-side package on devserver.
421
422 @param image: Full path of an OS image to install or a build name.
423
424 @return: A url to the autotest server-side package.
Dan Shi14de7622016-08-22 11:09:06 -0700425
426 @raise: error.AutoservError if fail to locate the build to test with, or
427 fail to stage server-side package.
Dan Shicf4d2032015-03-12 15:04:21 -0700428 """
Dan Shid37736b2016-07-06 15:10:29 -0700429 # If enable_drone_in_restricted_subnet is False, do not set hostname
430 # in devserver.resolve call, so a devserver in non-restricted subnet
431 # is picked to stage autotest server package for drone to download.
432 hostname = self.hostname
433 if not server_utils.ENABLE_DRONE_IN_RESTRICTED_SUBNET:
434 hostname = None
Dan Shicf4d2032015-03-12 15:04:21 -0700435 if image:
436 image_name = tools.get_build_from_image(image)
437 if not image_name:
438 raise error.AutoservError(
439 'Failed to parse build name from %s' % image)
Dan Shid37736b2016-07-06 15:10:29 -0700440 ds = dev_server.ImageServer.resolve(image_name, hostname)
Dan Shicf4d2032015-03-12 15:04:21 -0700441 else:
Prathmesh Prabhu9235e4c2017-03-28 13:16:06 -0700442 info = self.host_info_store.get()
Prathmesh Prabhub9d7adf2017-03-07 18:31:03 +0000443 job_repo_url = afe_utils.get_host_attribute(
444 self, ds_constants.JOB_REPO_URL)
Dan Shicf4d2032015-03-12 15:04:21 -0700445 if job_repo_url:
446 devserver_url, image_name = (
447 tools.get_devserver_build_from_package_url(job_repo_url))
Dan Shid37736b2016-07-06 15:10:29 -0700448 # If enable_drone_in_restricted_subnet is True, use the
449 # existing devserver. Otherwise, resolve a new one in
450 # non-restricted subnet.
451 if server_utils.ENABLE_DRONE_IN_RESTRICTED_SUBNET:
452 ds = dev_server.ImageServer(devserver_url)
453 else:
454 ds = dev_server.ImageServer.resolve(image_name)
Prathmesh Prabhub6cea612017-02-09 15:41:19 -0800455 elif info.build is not None:
456 ds = dev_server.ImageServer.resolve(info.build, hostname)
Dan Shicf4d2032015-03-12 15:04:21 -0700457 else:
Prathmesh Prabhub6cea612017-02-09 15:41:19 -0800458 raise error.AutoservError(
459 'Failed to stage server-side package. The host has '
460 'no job_report_url attribute or version label.')
Dan Shica503482015-03-30 17:23:25 -0700461
462 # Get the OS version of the build, for any build older than
463 # MIN_VERSION_SUPPORT_SSP, server side packaging is not supported.
464 match = re.match('.*/R\d+-(\d+)\.', image_name)
465 if match and int(match.group(1)) < self.MIN_VERSION_SUPPORT_SSP:
Dan Shi14de7622016-08-22 11:09:06 -0700466 raise error.AutoservError(
467 'Build %s is older than %s. Server side packaging is '
468 'disabled.' % (image_name, self.MIN_VERSION_SUPPORT_SSP))
Dan Shica503482015-03-30 17:23:25 -0700469
Dan Shicf4d2032015-03-12 15:04:21 -0700470 ds.stage_artifacts(image_name, ['autotest_server_package'])
471 return '%s/static/%s/%s' % (ds.url(), image_name,
472 'autotest_server_package.tar.bz2')
473
474
Dan Shi0f466e82013-02-22 15:44:58 -0800475 def _try_stateful_update(self, update_url, force_update, updater):
476 """Try to use stateful update to initialize DUT.
477
478 When DUT is already running the same version that machine_install
479 tries to install, stateful update is a much faster way to clean up
480 the DUT for testing, compared to a full reimage. It is implemeted
481 by calling autoupdater.run_update, but skipping updating root, as
482 updating the kernel is time consuming and not necessary.
483
484 @param update_url: url of the image.
485 @param force_update: Set to True to update the image even if the DUT
486 is running the same version.
487 @param updater: ChromiumOSUpdater instance used to update the DUT.
488 @returns: True if the DUT was updated with stateful update.
489
490 """
Dan Shi10b98482016-02-02 14:38:50 -0800491 # Stop service ap-update-manager to prevent rebooting during autoupdate.
492 # The service is used in jetstream boards, but not other CrOS devices.
493 self.run('sudo stop ap-update-manager', ignore_status=True)
494
J. Richard Barnette3f731032014-04-07 17:42:59 -0700495 # TODO(jrbarnette): Yes, I hate this re.match() test case.
496 # It's better than the alternative: see crbug.com/360944.
497 image_name = autoupdater.url_to_image_name(update_url)
498 release_pattern = r'^.*-release/R[0-9]+-[0-9]+\.[0-9]+\.0$'
499 if not re.match(release_pattern, image_name):
500 return False
Dan Shi0f466e82013-02-22 15:44:58 -0800501 if not updater.check_version():
502 return False
503 if not force_update:
504 logging.info('Canceling stateful update because the new and '
505 'old versions are the same.')
506 return False
507 # Following folders should be rebuilt after stateful update.
508 # A test file is used to confirm each folder gets rebuilt after
509 # the stateful update.
510 folders_to_check = ['/var', '/home', '/mnt/stateful_partition']
511 test_file = '.test_file_to_be_deleted'
512 for folder in folders_to_check:
513 touch_path = os.path.join(folder, test_file)
514 self.run('touch %s' % touch_path)
515
Chris Sosae92399e2015-04-24 11:32:59 -0700516 updater.run_update(update_root=False)
Dan Shi0f466e82013-02-22 15:44:58 -0800517
518 # Reboot to complete stateful update.
Chris Sosab76e0ee2013-05-22 16:55:41 -0700519 self.reboot(timeout=self.REBOOT_TIMEOUT, wait=True)
xixuan908b0752016-12-08 18:54:23 -0800520
521 # After stateful update and a reboot, all of the test_files shouldn't
522 # exist any more. Otherwise the stateful update is failed.
523 return not any(
524 self.path_exists(os.path.join(folder, test_file))
525 for folder in folders_to_check)
Dan Shi0f466e82013-02-22 15:44:58 -0800526
527
J. Richard Barnette7275b612013-06-04 18:13:11 -0700528 def _post_update_processing(self, updater, expected_kernel=None):
Dan Shi0f466e82013-02-22 15:44:58 -0800529 """After the DUT is updated, confirm machine_install succeeded.
530
531 @param updater: ChromiumOSUpdater instance used to update the DUT.
J. Richard Barnette7275b612013-06-04 18:13:11 -0700532 @param expected_kernel: kernel expected to be active after reboot,
533 or `None` to skip rollback checking.
Dan Shi0f466e82013-02-22 15:44:58 -0800534
535 """
J. Richard Barnette7275b612013-06-04 18:13:11 -0700536 # Touch the lab machine file to leave a marker that
537 # distinguishes this image from other test images.
538 # Afterwards, we must re-run the autoreboot script because
539 # it depends on the _LAB_MACHINE_FILE.
J. Richard Barnette71cc1862015-12-02 10:32:38 -0800540 autoreboot_cmd = ('FILE="%s" ; [ -f "$FILE" ] || '
541 '( touch "$FILE" ; start autoreboot )')
542 self.run(autoreboot_cmd % self._LAB_MACHINE_FILE)
Chris Sosa65425082013-10-16 13:26:22 -0700543 updater.verify_boot_expectations(
544 expected_kernel, rollback_message=
Gilad Arnoldc26ae1f2015-10-22 16:09:41 -0700545 'Build %s failed to boot on %s; system rolled back to previous '
Chris Sosa65425082013-10-16 13:26:22 -0700546 'build' % (updater.update_version, self.hostname))
J. Richard Barnette7275b612013-06-04 18:13:11 -0700547 # Check that we've got the build we meant to install.
548 if not updater.check_version_to_confirm_install():
549 raise autoupdater.ChromiumOSError(
550 'Failed to update %s to build %s; found build '
551 '%s instead' % (self.hostname,
Chris Sosa65425082013-10-16 13:26:22 -0700552 updater.update_version,
Dan Shi0942b1d2015-03-31 11:07:00 -0700553 self.get_release_version()))
Dan Shi0f466e82013-02-22 15:44:58 -0800554
Chris Sosae92399e2015-04-24 11:32:59 -0700555 logging.debug('Cleaning up old autotest directories.')
556 try:
557 installed_autodir = autotest.Autotest.get_installed_autodir(self)
558 self.run('rm -rf ' + installed_autodir)
559 except autotest.AutodirNotFoundError:
560 logging.debug('No autotest installed directory found.')
561
Dan Shi0f466e82013-02-22 15:44:58 -0800562
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700563 def _stage_image_for_update(self, image_name=None):
Chris Sosae92399e2015-04-24 11:32:59 -0700564 """Stage a build on a devserver and return the update_url and devserver.
Scott Zawalskieadbf702013-03-14 09:23:06 -0400565
566 @param image_name: a name like lumpy-release/R27-3837.0.0
Chris Sosae92399e2015-04-24 11:32:59 -0700567 @returns a tuple with an update URL like:
Scott Zawalskieadbf702013-03-14 09:23:06 -0400568 http://172.22.50.205:8082/update/lumpy-release/R27-3837.0.0
Chris Sosae92399e2015-04-24 11:32:59 -0700569 and the devserver instance.
Scott Zawalskieadbf702013-03-14 09:23:06 -0400570 """
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700571 if not image_name:
Richard Barnette260cbd02016-10-06 12:23:28 -0700572 image_name = self._get_cros_repair_image_name()
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700573 logging.info('Staging build for AU: %s', image_name)
Dan Shi216389c2015-12-22 11:03:06 -0800574 devserver = dev_server.ImageServer.resolve(image_name, self.hostname)
Scott Zawalskieadbf702013-03-14 09:23:06 -0400575 devserver.trigger_download(image_name, synchronous=False)
Chris Sosae92399e2015-04-24 11:32:59 -0700576 return (tools.image_url_pattern() % (devserver.url(), image_name),
577 devserver)
Scott Zawalskieadbf702013-03-14 09:23:06 -0400578
579
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700580 def stage_image_for_servo(self, image_name=None):
581 """Stage a build on a devserver and return the update_url.
582
583 @param image_name: a name like lumpy-release/R27-3837.0.0
584 @returns an update URL like:
585 http://172.22.50.205:8082/update/lumpy-release/R27-3837.0.0
586 """
587 if not image_name:
Richard Barnette260cbd02016-10-06 12:23:28 -0700588 image_name = self._get_cros_repair_image_name()
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700589 logging.info('Staging build for servo install: %s', image_name)
Dan Shi216389c2015-12-22 11:03:06 -0800590 devserver = dev_server.ImageServer.resolve(image_name, self.hostname)
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700591 devserver.stage_artifacts(image_name, ['test_image'])
592 return devserver.get_test_image_url(image_name)
593
594
beepse539be02013-07-31 21:57:39 -0700595 def stage_factory_image_for_servo(self, image_name):
596 """Stage a build on a devserver and return the update_url.
597
598 @param image_name: a name like <baord>/4262.204.0
beeps12c0a3c2013-09-03 11:58:27 -0700599
beepse539be02013-07-31 21:57:39 -0700600 @return: An update URL, eg:
601 http://<devserver>/static/canary-channel/\
602 <board>/4262.204.0/factory_test/chromiumos_factory_image.bin
beeps12c0a3c2013-09-03 11:58:27 -0700603
604 @raises: ValueError if the factory artifact name is missing from
605 the config.
606
beepse539be02013-07-31 21:57:39 -0700607 """
608 if not image_name:
609 logging.error('Need an image_name to stage a factory image.')
610 return
611
Dan Shib8540a52015-07-16 14:18:23 -0700612 factory_artifact = CONFIG.get_config_value(
beeps12c0a3c2013-09-03 11:58:27 -0700613 'CROS', 'factory_artifact', type=str, default='')
614 if not factory_artifact:
615 raise ValueError('Cannot retrieve the factory artifact name from '
616 'autotest config, and hence cannot stage factory '
617 'artifacts.')
618
beepse539be02013-07-31 21:57:39 -0700619 logging.info('Staging build for servo install: %s', image_name)
Dan Shi216389c2015-12-22 11:03:06 -0800620 devserver = dev_server.ImageServer.resolve(image_name, self.hostname)
beepse539be02013-07-31 21:57:39 -0700621 devserver.stage_artifacts(
622 image_name,
beeps12c0a3c2013-09-03 11:58:27 -0700623 [factory_artifact],
624 archive_url=None)
beepse539be02013-07-31 21:57:39 -0700625
626 return tools.factory_image_url_pattern() % (devserver.url(), image_name)
627
628
xixuan016d95b2017-02-27 14:37:47 -0800629 def _get_au_monarch_fields(self, devserver, build):
630 """Form monarch fields by given devserve & build for auto-update.
631
632 @param devserver: the devserver (ImageServer instance) for auto-update.
633 @param build: the build to be updated.
634
635 @return A dictionary of monach fields.
636 """
637 try:
638 board, build_type, milestone, _ = server_utils.ParseBuildName(build)
639 except server_utils.ParseBuildNameException:
640 logging.warning('Unable to parse build name %s. Something is '
641 'likely broken, but will continue anyway.',
642 build)
643 board, build_type, milestone = ('', '', '')
644
645 monarch_fields = {
646 'dev_server': devserver.hostname,
647 'board': board,
648 'build_type': build_type,
649 'milestone': milestone
650 }
651 return monarch_fields
652
653
xixuanc30d7e92017-05-11 12:06:39 -0700654 def _retry_auto_update_with_new_devserver(self, build, last_devserver,
xixuana3bbc422017-05-04 15:57:21 -0700655 force_update, force_full_update,
656 force_original):
xixuanc30d7e92017-05-11 12:06:39 -0700657 """Kick off auto-update by devserver and send metrics.
658
659 @param build: the build to update.
660 @param last_devserver: the last devserver that failed to provision.
661 @param force_update: see |machine_install_by_devserver|'s force_udpate
662 for details.
663 @param force_full_update: see |machine_install_by_devserver|'s
664 force_full_update for details.
xixuana3bbc422017-05-04 15:57:21 -0700665 @param force_original: Whether to force stateful update with the
666 original payload.
xixuanc30d7e92017-05-11 12:06:39 -0700667
668 @return the result of |auto_update| in dev_server.
669 """
670 devserver = dev_server.resolve(
671 build, self.hostname, ban_list=[last_devserver.url()])
672 devserver.trigger_download(build, synchronous=False)
673 monarch_fields = self._get_au_monarch_fields(devserver, build)
674 logging.debug('Retry auto_update: resolved devserver for '
675 'auto-update: %s', devserver.url())
676
677 # Add metrics
678 install_with_dev_counter = metrics.Counter(
679 'chromeos/autotest/provision/install_with_devserver')
680 install_with_dev_counter.increment(fields=monarch_fields)
681 c = metrics.Counter(
682 'chromeos/autotest/provision/retry_by_devserver')
683 monarch_fields['last_devserver'] = last_devserver.hostname
684 monarch_fields['host'] = self.hostname
685 c.increment(fields=monarch_fields)
686
687 return devserver.auto_update(
688 self.hostname, build,
689 original_board=self.get_board().replace(
690 ds_constants.BOARD_PREFIX, ''),
691 original_release_version=self.get_release_version(),
692 log_dir=self.job.resultdir,
693 force_update=force_update,
xixuana3bbc422017-05-04 15:57:21 -0700694 full_update=force_full_update,
695 force_original=force_original)
xixuanc30d7e92017-05-11 12:06:39 -0700696
697
xixuanec801e32016-08-25 10:20:22 -0700698 def machine_install_by_devserver(self, update_url=None, force_update=False,
xixuan5dc64ea2016-05-20 17:27:51 -0700699 local_devserver=False, repair=False,
700 force_full_update=False):
701 """Ultiize devserver to install the DUT.
702
703 (TODO) crbugs.com/627269: The logic in this function has some overlap
704 with those in function machine_install. The merge will be done later,
705 not in the same CL.
706
xixuanec801e32016-08-25 10:20:22 -0700707 @param update_url: The update_url or build for the host to update.
xixuan5dc64ea2016-05-20 17:27:51 -0700708 @param force_update: Force an update even if the version installed
709 is the same. Default:False
710 @param local_devserver: Used by test_that to allow people to
711 use their local devserver. Default: False
712 @param repair: Forces update to repair image. Implies force_update.
713 @param force_full_update: If True, do not attempt to run stateful
714 update, force a full reimage. If False, try stateful update
715 first when the dut is already installed with the same version.
716 @raises autoupdater.ChromiumOSError
717
718 @returns A tuple of (image_name, host_attributes).
719 image_name is the name of image installed, e.g.,
720 veyron_jerry-release/R50-7871.0.0
721 host_attributes is a dictionary of (attribute, value), which
722 can be saved to afe_host_attributes table in database. This
723 method returns a dictionary with a single entry of
724 `job_repo_url`: repo_url, where repo_url is a devserver url to
725 autotest packages.
726 """
xixuan5dc64ea2016-05-20 17:27:51 -0700727 if repair:
Richard Barnette260cbd02016-10-06 12:23:28 -0700728 update_url = self._get_cros_repair_image_name()
xixuan5dc64ea2016-05-20 17:27:51 -0700729 force_update = True
730
xixuanec801e32016-08-25 10:20:22 -0700731 if not update_url and not self._parser.options.image:
xixuan5dc64ea2016-05-20 17:27:51 -0700732 raise error.AutoservError(
733 'There is no update URL, nor a method to get one.')
734
xixuanec801e32016-08-25 10:20:22 -0700735 if not update_url and self._parser.options.image:
736 update_url = self._parser.options.image
xixuan5dc64ea2016-05-20 17:27:51 -0700737
xixuan5dc64ea2016-05-20 17:27:51 -0700738
739 # Get build from parameter or AFE.
740 # If the build is not a URL, let devserver to stage it first.
741 # Otherwise, choose a devserver to trigger auto-update.
xixuanec801e32016-08-25 10:20:22 -0700742 build = None
Aviv Keshetf9375482016-10-26 02:09:03 -0700743 devserver = None
744 logging.debug('Resolving a devserver for auto-update')
Aviv Keshet3a76f762016-10-27 17:41:51 -0700745 previously_resolved = False
xixuanec801e32016-08-25 10:20:22 -0700746 if update_url.startswith('http://'):
747 build = autoupdater.url_to_image_name(update_url)
Aviv Keshet3a76f762016-10-27 17:41:51 -0700748 previously_resolved = True
749 else:
750 build = update_url
751 devserver = dev_server.resolve(build, self.hostname)
Allen Lia5cfb972016-12-27 17:17:22 -0800752 server_name = devserver.hostname
Aviv Keshet3a76f762016-10-27 17:41:51 -0700753
xixuan016d95b2017-02-27 14:37:47 -0800754 monarch_fields = self._get_au_monarch_fields(devserver, build)
Aviv Keshet3a76f762016-10-27 17:41:51 -0700755
756 if previously_resolved:
xixuan2e62d162016-09-29 19:19:19 -0700757 # Make sure devserver for Auto-Update has staged the build.
xixuan2e62d162016-09-29 19:19:19 -0700758 if server_name not in update_url:
Aviv Keshet3a76f762016-10-27 17:41:51 -0700759 logging.debug('Resolved to devserver that does not match '
760 'update_url. The previously resolved devserver '
761 'must be unhealthy. Switching to use devserver %s,'
762 ' and re-staging.',
xixuan2e62d162016-09-29 19:19:19 -0700763 server_name)
Aviv Keshetf9375482016-10-26 02:09:03 -0700764 logging.info('Staging build for AU: %s', update_url)
xixuan2e62d162016-09-29 19:19:19 -0700765 devserver.trigger_download(build, synchronous=False)
Dan Shi5e2efb72017-02-07 11:40:23 -0800766 c = metrics.Counter(
767 'chromeos/autotest/provision/failover_download')
768 c.increment(fields=monarch_fields)
xixuan5dc64ea2016-05-20 17:27:51 -0700769 else:
Aviv Keshetf9375482016-10-26 02:09:03 -0700770 logging.info('Staging build for AU: %s', update_url)
xixuan5dc64ea2016-05-20 17:27:51 -0700771 devserver.trigger_download(build, synchronous=False)
Dan Shi5e2efb72017-02-07 11:40:23 -0800772 c = metrics.Counter('chromeos/autotest/provision/trigger_download')
773 c.increment(fields=monarch_fields)
xixuan5dc64ea2016-05-20 17:27:51 -0700774
775 # Report provision stats.
xixuan016d95b2017-02-27 14:37:47 -0800776 install_with_dev_counter = metrics.Counter(
777 'chromeos/autotest/provision/install_with_devserver')
778 install_with_dev_counter.increment(fields=monarch_fields)
xixuanec801e32016-08-25 10:20:22 -0700779 logging.debug('Resolved devserver for auto-update: %s', devserver.url())
xixuan5dc64ea2016-05-20 17:27:51 -0700780
Aviv Keshetf9375482016-10-26 02:09:03 -0700781 # and other metrics from this function.
Dan Shi5e2efb72017-02-07 11:40:23 -0800782 metrics.Counter('chromeos/autotest/provision/resolve'
783 ).increment(fields=monarch_fields)
Aviv Keshetf9375482016-10-26 02:09:03 -0700784
xixuana3bbc422017-05-04 15:57:21 -0700785 force_original = self.get_chromeos_release_milestone() is None
786
xixuan016d95b2017-02-27 14:37:47 -0800787 success, retryable = devserver.auto_update(
xixuanf5ee7862016-11-30 16:53:05 -0800788 self.hostname, build,
789 original_board=self.get_board().replace(
790 ds_constants.BOARD_PREFIX, ''),
791 original_release_version=self.get_release_version(),
792 log_dir=self.job.resultdir,
793 force_update=force_update,
xixuana3bbc422017-05-04 15:57:21 -0700794 full_update=force_full_update,
795 force_original=force_original)
xixuan016d95b2017-02-27 14:37:47 -0800796 if not success and retryable:
797 # It indicates that last provision failed due to devserver load
798 # issue, so another devserver is resolved to kick off provision
799 # job once again and only once.
800 logging.debug('Provision failed due to devserver issue,'
801 'retry it with another devserver.')
802
803 # Check first whether this DUT is completely offline. If so, skip
804 # the following provision tries.
805 logging.debug('Checking whether host %s is online.', self.hostname)
806 if utils.ping(self.hostname, tries=1, deadline=1) == 0:
xixuanc30d7e92017-05-11 12:06:39 -0700807 self._retry_auto_update_with_new_devserver(
xixuana3bbc422017-05-04 15:57:21 -0700808 build, devserver, force_update, force_full_update,
809 force_original)
xixuanc30d7e92017-05-11 12:06:39 -0700810 else:
811 raise error.AutoservError(
812 'No answer to ping from %s' % self.hostname)
xixuan5dc64ea2016-05-20 17:27:51 -0700813
814 # The reason to resolve a new devserver in function machine_install
815 # is mostly because that the update_url there may has a strange format,
816 # and it's hard to parse the devserver url from it.
817 # Since we already resolve a devserver to trigger auto-update, the same
818 # devserver is used to form JOB_REPO_URL here. Verified in local test.
819 repo_url = tools.get_package_url(devserver.url(), build)
820 return build, {ds_constants.JOB_REPO_URL: repo_url}
821
822
Chris Sosaa3ac2152012-05-23 22:23:13 -0700823 def machine_install(self, update_url=None, force_update=False,
Richard Barnette0b023a72015-04-24 16:07:30 +0000824 local_devserver=False, repair=False,
825 force_full_update=False):
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500826 """Install the DUT.
827
Dan Shi0f466e82013-02-22 15:44:58 -0800828 Use stateful update if the DUT is already running the same build.
829 Stateful update does not update kernel and tends to run much faster
830 than a full reimage. If the DUT is running a different build, or it
831 failed to do a stateful update, full update, including kernel update,
832 will be applied to the DUT.
833
Simran Basi5ace6f22016-01-06 17:30:44 -0800834 Once a host enters machine_install its host attribute job_repo_url
835 (used for package install) will be removed and then updated.
Scott Zawalskieadbf702013-03-14 09:23:06 -0400836
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500837 @param update_url: The url to use for the update
838 pattern: http://$devserver:###/update/$build
839 If update_url is None and repair is True we will install the
Dan Shi6964fa52014-12-18 11:04:27 -0800840 stable image listed in afe_stable_versions table. If the table
841 is not setup, global_config value under CROS.stable_cros_version
842 will be used instead.
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500843 @param force_update: Force an update even if the version installed
844 is the same. Default:False
Christopher Wiley6a4ff932015-05-15 14:00:47 -0700845 @param local_devserver: Used by test_that to allow people to
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500846 use their local devserver. Default: False
Chris Sosae92399e2015-04-24 11:32:59 -0700847 @param repair: Forces update to repair image. Implies force_update.
Fang Deng3d3b9272014-12-22 12:20:28 -0800848 @param force_full_update: If True, do not attempt to run stateful
849 update, force a full reimage. If False, try stateful update
850 first when the dut is already installed with the same version.
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500851 @raises autoupdater.ChromiumOSError
852
Dan Shibe3636a2016-02-14 22:48:01 -0800853 @returns A tuple of (image_name, host_attributes).
854 image_name is the name of image installed, e.g.,
855 veyron_jerry-release/R50-7871.0.0
856 host_attributes is a dictionary of (attribute, value), which
857 can be saved to afe_host_attributes table in database. This
858 method returns a dictionary with a single entry of
859 `job_repo_url`: repo_url, where repo_url is a devserver url to
860 autotest packages.
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500861 """
Chris Sosae92399e2015-04-24 11:32:59 -0700862 devserver = None
Richard Barnette0b023a72015-04-24 16:07:30 +0000863 if repair:
Chris Sosae92399e2015-04-24 11:32:59 -0700864 update_url, devserver = self._stage_image_for_update()
Richard Barnette0b023a72015-04-24 16:07:30 +0000865 force_update = True
Dan Shi0f466e82013-02-22 15:44:58 -0800866
Chris Sosae92399e2015-04-24 11:32:59 -0700867 if not update_url and not self._parser.options.image:
868 raise error.AutoservError(
Dan Shid07ee2e2015-09-24 14:49:25 -0700869 'There is no update URL, nor a method to get one.')
Chris Sosae92399e2015-04-24 11:32:59 -0700870
871 if not update_url and self._parser.options.image:
872 # This is the base case where we have no given update URL i.e.
873 # dynamic suites logic etc. This is the most flexible case where we
874 # can serve an update from any of our fleet of devservers.
875 requested_build = self._parser.options.image
876 if not requested_build.startswith('http://'):
877 logging.debug('Update will be staged for this installation')
878 update_url, devserver = self._stage_image_for_update(
Dan Shid07ee2e2015-09-24 14:49:25 -0700879 requested_build)
Chris Sosae92399e2015-04-24 11:32:59 -0700880 else:
881 update_url = requested_build
882
883 logging.debug('Update URL is %s', update_url)
884
Dan Shif48f8132016-02-18 10:34:30 -0800885 # Report provision stats.
Allen Lia5cfb972016-12-27 17:17:22 -0800886 server_name = dev_server.get_hostname(update_url)
Allen Li48a13fe2016-11-22 14:10:40 -0800887 (metrics.Counter('chromeos/autotest/provision/install')
888 .increment(fields={'devserver': server_name}))
Dan Shif48f8132016-02-18 10:34:30 -0800889
Dan Shid07ee2e2015-09-24 14:49:25 -0700890 # Create a file to indicate if provision fails. The file will be removed
891 # by stateful update or full install.
J. Richard Barnette91137f02016-03-10 16:52:26 -0800892 self.run('touch %s' % self.PROVISION_FAILED)
Dan Shid07ee2e2015-09-24 14:49:25 -0700893
Chris Sosae92399e2015-04-24 11:32:59 -0700894 update_complete = False
895 updater = autoupdater.ChromiumOSUpdater(
896 update_url, host=self, local_devserver=local_devserver)
Fang Deng3d3b9272014-12-22 12:20:28 -0800897 if not force_full_update:
898 try:
Chris Sosae92399e2015-04-24 11:32:59 -0700899 # If the DUT is already running the same build, try stateful
900 # update first as it's much quicker than a full re-image.
901 update_complete = self._try_stateful_update(
Dan Shid07ee2e2015-09-24 14:49:25 -0700902 update_url, force_update, updater)
Fang Deng3d3b9272014-12-22 12:20:28 -0800903 except Exception as e:
904 logging.exception(e)
J. Richard Barnette45e93de2012-04-11 17:24:15 -0700905
Dan Shi0f466e82013-02-22 15:44:58 -0800906 inactive_kernel = None
Chris Sosae92399e2015-04-24 11:32:59 -0700907 if update_complete or (not force_update and updater.check_version()):
908 logging.info('Install complete without full update')
909 else:
910 logging.info('DUT requires full update.')
911 self.reboot(timeout=self.REBOOT_TIMEOUT, wait=True)
Dan Shi10b98482016-02-02 14:38:50 -0800912 # Stop service ap-update-manager to prevent rebooting during
913 # autoupdate. The service is used in jetstream boards, but not other
914 # CrOS devices.
915 self.run('sudo stop ap-update-manager', ignore_status=True)
916
Chris Sosae92399e2015-04-24 11:32:59 -0700917 num_of_attempts = provision.FLAKY_DEVSERVER_ATTEMPTS
Chris Sosab7612bc2013-03-21 10:32:37 -0700918
Chris Sosae92399e2015-04-24 11:32:59 -0700919 while num_of_attempts > 0:
920 num_of_attempts -= 1
921 try:
922 updater.run_update()
923 except Exception:
924 logging.warn('Autoupdate did not complete.')
925 # Do additional check for the devserver health. Ideally,
926 # the autoupdater.py could raise an exception when it
927 # detected network flake but that would require
928 # instrumenting the update engine and parsing it log.
929 if (num_of_attempts <= 0 or
930 devserver is None or
xixuan9e2c98d2016-02-26 19:04:53 -0800931 dev_server.ImageServer.devserver_healthy(
Chris Sosae92399e2015-04-24 11:32:59 -0700932 devserver.url())):
Dan Shid07ee2e2015-09-24 14:49:25 -0700933 raise
J. Richard Barnette45e93de2012-04-11 17:24:15 -0700934
Chris Sosae92399e2015-04-24 11:32:59 -0700935 logging.warn('Devserver looks unhealthy. Trying another')
936 update_url, devserver = self._stage_image_for_update(
937 requested_build)
938 logging.debug('New Update URL is %s', update_url)
939 updater = autoupdater.ChromiumOSUpdater(
940 update_url, host=self,
941 local_devserver=local_devserver)
942 else:
943 break
J. Richard Barnette45e93de2012-04-11 17:24:15 -0700944
Chris Sosae92399e2015-04-24 11:32:59 -0700945 # Give it some time in case of IO issues.
946 time.sleep(10)
Dan Shi5699ac22014-12-19 10:55:49 -0800947
Chris Sosae92399e2015-04-24 11:32:59 -0700948 # Figure out active and inactive kernel.
949 active_kernel, inactive_kernel = updater.get_kernel_state()
Simran Basi13fa1ba2013-03-04 10:56:47 -0800950
Chris Sosae92399e2015-04-24 11:32:59 -0700951 # Ensure inactive kernel has higher priority than active.
952 if (updater.get_kernel_priority(inactive_kernel)
953 < updater.get_kernel_priority(active_kernel)):
954 raise autoupdater.ChromiumOSError(
955 'Update failed. The priority of the inactive kernel'
956 ' partition is less than that of the active kernel'
957 ' partition.')
958
959 # Updater has returned successfully; reboot the host.
Richard Barnettefc3d7432016-07-07 10:21:31 -0700960 #
961 # Regarding the 'crossystem' command: In some cases, the
962 # TPM gets into a state such that it fails verification.
963 # We don't know why. However, this call papers over the
964 # problem by clearing the TPM during the reboot.
965 #
966 # We ignore failures from 'crossystem'. Although failure
967 # here is unexpected, and could signal a bug, the point
968 # of the exercise is to paper over problems; allowing
969 # this to fail would defeat the purpose.
970 self.run('crossystem clear_tpm_owner_request=1',
971 ignore_status=True)
Chris Sosae92399e2015-04-24 11:32:59 -0700972 self.reboot(timeout=self.REBOOT_TIMEOUT, wait=True)
973
974 self._post_update_processing(updater, inactive_kernel)
Simran Basi5ace6f22016-01-06 17:30:44 -0800975 image_name = autoupdater.url_to_image_name(update_url)
Dan Shibe3636a2016-02-14 22:48:01 -0800976 # update_url is different from devserver url needed to stage autotest
977 # packages, therefore, resolve a new devserver url here.
978 devserver_url = dev_server.ImageServer.resolve(image_name,
979 self.hostname).url()
980 repo_url = tools.get_package_url(devserver_url, image_name)
981 return image_name, {ds_constants.JOB_REPO_URL: repo_url}
J. Richard Barnette45e93de2012-04-11 17:24:15 -0700982
983
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +0800984 def _clear_fw_version_labels(self, rw_only):
985 """Clear firmware version labels from the machine.
986
987 @param rw_only: True to only clear fwrw_version; otherewise, clear
988 both fwro_version and fwrw_version.
989 """
Dan Shi9cb0eec2014-06-03 09:04:50 -0700990 labels = self._AFE.get_labels(
Dan Shi0723bf52015-06-24 10:52:38 -0700991 name__startswith=provision.FW_RW_VERSION_PREFIX,
Dan Shi9cb0eec2014-06-03 09:04:50 -0700992 host__hostname=self.hostname)
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +0800993 if not rw_only:
994 labels = labels + self._AFE.get_labels(
995 name__startswith=provision.FW_RO_VERSION_PREFIX,
996 host__hostname=self.hostname)
Dan Shi9cb0eec2014-06-03 09:04:50 -0700997 for label in labels:
998 label.remove_hosts(hosts=[self.hostname])
999
1000
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +08001001 def _add_fw_version_label(self, build, rw_only):
Dan Shi9cb0eec2014-06-03 09:04:50 -07001002 """Add firmware version label to the machine.
1003
1004 @param build: Build of firmware.
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +08001005 @param rw_only: True to only add fwrw_version; otherwise, add both
1006 fwro_version and fwrw_version.
Dan Shi9cb0eec2014-06-03 09:04:50 -07001007
1008 """
Prathmesh Prabhu2c7471d2016-11-15 20:19:57 +00001009 fw_label = provision.fwrw_version_to_label(build)
MK Ryu73be9862015-07-06 12:25:00 -07001010 self._AFE.run('label_add_hosts', id=fw_label, hosts=[self.hostname])
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +08001011 if not rw_only:
Prathmesh Prabhu2c7471d2016-11-15 20:19:57 +00001012 fw_label = provision.fwro_version_to_label(build)
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +08001013 self._AFE.run('label_add_hosts', id=fw_label, hosts=[self.hostname])
Dan Shi9cb0eec2014-06-03 09:04:50 -07001014
1015
Tom Wai-Hong Tam4ac78982016-01-08 02:34:37 +08001016 def firmware_install(self, build=None, rw_only=False):
Dan Shi9cb0eec2014-06-03 09:04:50 -07001017 """Install firmware to the DUT.
1018
1019 Use stateful update if the DUT is already running the same build.
1020 Stateful update does not update kernel and tends to run much faster
1021 than a full reimage. If the DUT is running a different build, or it
1022 failed to do a stateful update, full update, including kernel update,
1023 will be applied to the DUT.
1024
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +08001025 Once a host enters firmware_install its fw[ro|rw]_version label will
1026 be removed. After the firmware is updated successfully, a new
1027 fw[ro|rw]_version label will be added to the host.
Dan Shi9cb0eec2014-06-03 09:04:50 -07001028
1029 @param build: The build version to which we want to provision the
1030 firmware of the machine,
1031 e.g. 'link-firmware/R22-2695.1.144'.
Tom Wai-Hong Tam4ac78982016-01-08 02:34:37 +08001032 @param rw_only: True to only install firmware to its RW portions. Keep
1033 the RO portions unchanged.
Dan Shi9cb0eec2014-06-03 09:04:50 -07001034
1035 TODO(dshi): After bug 381718 is fixed, update here with corresponding
1036 exceptions that could be raised.
1037
1038 """
1039 if not self.servo:
1040 raise error.TestError('Host %s does not have servo.' %
1041 self.hostname)
1042
Wai-Hong Tam64cf7912017-05-23 16:41:40 -07001043 # Get the DUT board name from servod.
1044 board = self.servo.get_board()
Dan Shi9cb0eec2014-06-03 09:04:50 -07001045
Chris Sosae92399e2015-04-24 11:32:59 -07001046 # If build is not set, try to install firmware from stable CrOS.
Dan Shi9cb0eec2014-06-03 09:04:50 -07001047 if not build:
Richard Barnette260cbd02016-10-06 12:23:28 -07001048 build = afe_utils.get_stable_faft_version(board)
Dan Shi3d7a0e12015-10-12 11:55:45 -07001049 if not build:
1050 raise error.TestError(
1051 'Failed to find stable firmware build for %s.',
1052 self.hostname)
1053 logging.info('Will install firmware from build %s.', build)
Dan Shi9cb0eec2014-06-03 09:04:50 -07001054
1055 config = FAFTConfig(board)
1056 if config.use_u_boot:
1057 ap_image = 'image-%s.bin' % board
1058 else: # Depthcharge platform
1059 ap_image = 'image.bin'
1060 ec_image = 'ec.bin'
Dan Shi216389c2015-12-22 11:03:06 -08001061 ds = dev_server.ImageServer.resolve(build, self.hostname)
Dan Shi9cb0eec2014-06-03 09:04:50 -07001062 ds.stage_artifacts(build, ['firmware'])
1063
1064 tmpd = autotemp.tempdir(unique_id='fwimage')
1065 try:
1066 fwurl = self._FW_IMAGE_URL_PATTERN % (ds.url(), build)
1067 local_tarball = os.path.join(tmpd.name, os.path.basename(fwurl))
xixuan4e116822016-11-17 15:32:10 -08001068 ds.download_file(fwurl, local_tarball)
Dan Shi9cb0eec2014-06-03 09:04:50 -07001069
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +08001070 self._clear_fw_version_labels(rw_only)
Tom Wai-Hong Tam552fee12016-05-11 06:27:58 +08001071 if config.chrome_ec:
Tom Wai-Hong Tamf9db57a2016-03-17 05:32:22 +08001072 logging.info('Will re-program EC %snow', 'RW ' if rw_only else '')
1073 server_utils.system('tar xf %s -C %s %s' %
1074 (local_tarball, tmpd.name, ec_image),
1075 timeout=60)
1076 self.servo.program_ec(os.path.join(tmpd.name, ec_image), rw_only)
1077 else:
1078 logging.info('Not a Chrome EC, ignore re-programing it')
Tom Wai-Hong Tam4ac78982016-01-08 02:34:37 +08001079 logging.info('Will re-program BIOS %snow', 'RW ' if rw_only else '')
Tom Wai-Hong Tamf9db57a2016-03-17 05:32:22 +08001080 server_utils.system('tar xf %s -C %s %s' %
1081 (local_tarball, tmpd.name, ap_image),
1082 timeout=60)
Tom Wai-Hong Tam4ac78982016-01-08 02:34:37 +08001083 self.servo.program_bios(os.path.join(tmpd.name, ap_image), rw_only)
Dan Shi9cb0eec2014-06-03 09:04:50 -07001084 self.servo.get_power_state_controller().reset()
1085 time.sleep(self.servo.BOOT_DELAY)
Wai-Hong Tamb5f66ce2016-11-10 15:45:30 -08001086 if utils.host_is_in_lab_zone(self.hostname):
1087 self._add_fw_version_label(build, rw_only)
Dan Shi9cb0eec2014-06-03 09:04:50 -07001088 finally:
1089 tmpd.clean()
1090
1091
Dan Shi10e992b2013-08-30 11:02:59 -07001092 def show_update_engine_log(self):
1093 """Output update engine log."""
MK Ryu35d661e2014-09-25 17:44:10 -07001094 logging.debug('Dumping %s', client_constants.UPDATE_ENGINE_LOG)
1095 self.run('cat %s' % client_constants.UPDATE_ENGINE_LOG)
Dan Shi10e992b2013-08-30 11:02:59 -07001096
1097
Richard Barnette82c35912012-11-20 10:09:10 -08001098 def _get_board_from_afe(self):
Kevin Cheng84a71ba2016-07-14 11:03:57 -07001099 """Retrieve this host's board from its labels stored locally.
Richard Barnette82c35912012-11-20 10:09:10 -08001100
1101 Looks for a host label of the form "board:<board>", and
1102 returns the "<board>" part of the label. `None` is returned
1103 if there is not a single, unique label matching the pattern.
1104
1105 @returns board from label, or `None`.
1106 """
Richard Barnette260cbd02016-10-06 12:23:28 -07001107 board = afe_utils.get_board(self)
1108 if board is None:
1109 raise error.AutoservError('DUT cannot be repaired, '
1110 'there is no board attribute.')
1111 return board
Simran Basi833814b2013-01-29 13:13:43 -08001112
1113
beepsf079cfb2013-09-18 17:49:51 -07001114 def servo_install(self, image_url=None, usb_boot_timeout=USB_BOOT_TIMEOUT,
1115 install_timeout=INSTALL_TIMEOUT):
Scott Zawalski62bacae2013-03-05 10:40:32 -05001116 """
1117 Re-install the OS on the DUT by:
1118 1) installing a test image on a USB storage device attached to the Servo
1119 board,
Richard Barnette03a0c132012-11-05 12:40:35 -08001120 2) booting that image in recovery mode, and then
J. Richard Barnette31b2e312013-04-04 16:05:22 -07001121 3) installing the image with chromeos-install.
1122
Scott Zawalski62bacae2013-03-05 10:40:32 -05001123 @param image_url: If specified use as the url to install on the DUT.
1124 otherwise boot the currently staged image on the USB stick.
beepsf079cfb2013-09-18 17:49:51 -07001125 @param usb_boot_timeout: The usb_boot_timeout to use during reimage.
1126 Factory images need a longer usb_boot_timeout than regular
1127 cros images.
1128 @param install_timeout: The timeout to use when installing the chromeos
1129 image. Factory images need a longer install_timeout.
Richard Barnette03a0c132012-11-05 12:40:35 -08001130
Scott Zawalski62bacae2013-03-05 10:40:32 -05001131 @raises AutoservError if the image fails to boot.
beepsf079cfb2013-09-18 17:49:51 -07001132
J. Richard Barnette0199cc82014-12-05 17:08:40 -08001133 """
beepsf079cfb2013-09-18 17:49:51 -07001134 logging.info('Downloading image to USB, then booting from it. Usb boot '
1135 'timeout = %s', usb_boot_timeout)
Allen Li48a13fe2016-11-22 14:10:40 -08001136 with metrics.SecondsTimer(
1137 'chromeos/autotest/provision/servo_install/boot_duration'):
1138 self.servo.install_recovery_image(image_url)
1139 if not self.wait_up(timeout=usb_boot_timeout):
1140 raise hosts.AutoservRepairError(
1141 'DUT failed to boot from USB after %d seconds' %
1142 usb_boot_timeout)
Scott Zawalski62bacae2013-03-05 10:40:32 -05001143
Tom Wai-Hong Tamf6b4f812015-08-08 04:14:59 +08001144 # The new chromeos-tpm-recovery has been merged since R44-7073.0.0.
1145 # In old CrOS images, this command fails. Skip the error.
Tom Wai-Hong Tam27af7332015-07-25 06:09:39 +08001146 logging.info('Resetting the TPM status')
Tom Wai-Hong Tamf6b4f812015-08-08 04:14:59 +08001147 try:
1148 self.run('chromeos-tpm-recovery')
1149 except error.AutoservRunError:
1150 logging.warn('chromeos-tpm-recovery is too old.')
1151
Tom Wai-Hong Tam27af7332015-07-25 06:09:39 +08001152
Allen Li48a13fe2016-11-22 14:10:40 -08001153 with metrics.SecondsTimer(
1154 'chromeos/autotest/provision/servo_install/install_duration'):
1155 logging.info('Installing image through chromeos-install.')
1156 self.run('chromeos-install --yes', timeout=install_timeout)
1157 self.halt()
beepsf079cfb2013-09-18 17:49:51 -07001158
1159 logging.info('Power cycling DUT through servo.')
J. Richard Barnette0199cc82014-12-05 17:08:40 -08001160 self.servo.get_power_state_controller().power_off()
Fang Dengafb88142013-05-30 17:44:31 -07001161 self.servo.switch_usbkey('off')
J. Richard Barnette0199cc82014-12-05 17:08:40 -08001162 # N.B. The Servo API requires that we use power_on() here
1163 # for two reasons:
1164 # 1) After turning on a DUT in recovery mode, you must turn
1165 # it off and then on with power_on() once more to
1166 # disable recovery mode (this is a Parrot specific
1167 # requirement).
1168 # 2) After power_off(), the only way to turn on is with
1169 # power_on() (this is a Storm specific requirement).
J. Richard Barnettefbcc7122013-07-24 18:24:59 -07001170 self.servo.get_power_state_controller().power_on()
beepsf079cfb2013-09-18 17:49:51 -07001171
1172 logging.info('Waiting for DUT to come back up.')
Richard Barnette03a0c132012-11-05 12:40:35 -08001173 if not self.wait_up(timeout=self.BOOT_TIMEOUT):
1174 raise error.AutoservError('DUT failed to reboot installed '
1175 'test image after %d seconds' %
Scott Zawalski62bacae2013-03-05 10:40:32 -05001176 self.BOOT_TIMEOUT)
1177
1178
Richard Barnette9a26ad62016-06-10 12:03:08 -07001179 def repair_servo(self):
Dan Shi90466352015-09-22 15:01:05 -07001180 """
Richard Barnette9a26ad62016-06-10 12:03:08 -07001181 Confirm that servo is initialized and verified.
Dan Shi90466352015-09-22 15:01:05 -07001182
Richard Barnette9a26ad62016-06-10 12:03:08 -07001183 If the servo object is missing, attempt to repair the servo
1184 host. Repair failures are passed back to the caller.
1185
1186 @raise AutoservError: If there is no servo host for this CrOS
1187 host.
1188 """
1189 if self.servo:
1190 return
1191 if not self._servo_host:
1192 raise error.AutoservError('No servo host for %s.' %
1193 self.hostname)
1194 self._servo_host.repair()
1195 self.servo = self._servo_host.get_servo()
Dan Shi90466352015-09-22 15:01:05 -07001196
1197
J. Richard Barnettec2d99cf2015-11-18 12:46:15 -08001198 def repair(self):
1199 """Attempt to get the DUT to pass `self.verify()`.
Richard Barnette82c35912012-11-20 10:09:10 -08001200
1201 This overrides the base class function for repair; it does
J. Richard Barnette91137f02016-03-10 16:52:26 -08001202 not call back to the parent class, but instead relies on
1203 `self._repair_strategy` to coordinate the verification and
1204 repair steps needed to get the DUT working.
Richard Barnette82c35912012-11-20 10:09:10 -08001205 """
J. Richard Barnette91137f02016-03-10 16:52:26 -08001206 self._repair_strategy.repair(self)
Scott Zawalski89c44dd2013-02-26 09:28:02 -05001207
Richard Barnette82c35912012-11-20 10:09:10 -08001208
J. Richard Barnette1d78b012012-05-15 13:56:30 -07001209 def close(self):
Fang Deng0ca40e22013-08-27 17:47:44 -07001210 super(CrosHost, self).close()
xixuand6011f12016-12-08 15:01:58 -08001211 if self._chameleon_host:
1212 self._chameleon_host.close()
1213
1214 if self._servo_host:
1215 self._servo_host.close()
J. Richard Barnette1d78b012012-05-15 13:56:30 -07001216
1217
Dan Shi49ca0932014-11-14 11:22:27 -08001218 def get_power_supply_info(self):
1219 """Get the output of power_supply_info.
1220
1221 power_supply_info outputs the info of each power supply, e.g.,
1222 Device: Line Power
1223 online: no
1224 type: Mains
1225 voltage (V): 0
1226 current (A): 0
1227 Device: Battery
1228 state: Discharging
1229 percentage: 95.9276
1230 technology: Li-ion
1231
1232 Above output shows two devices, Line Power and Battery, with details of
1233 each device listed. This function parses the output into a dictionary,
1234 with key being the device name, and value being a dictionary of details
1235 of the device info.
1236
1237 @return: The dictionary of power_supply_info, e.g.,
1238 {'Line Power': {'online': 'yes', 'type': 'main'},
1239 'Battery': {'vendor': 'xyz', 'percentage': '100'}}
Dan Shie9b765d2014-12-29 16:59:49 -08001240 @raise error.AutoservRunError if power_supply_info tool is not found in
1241 the DUT. Caller should handle this error to avoid false failure
1242 on verification.
Dan Shi49ca0932014-11-14 11:22:27 -08001243 """
1244 result = self.run('power_supply_info').stdout.strip()
1245 info = {}
1246 device_name = None
1247 device_info = {}
1248 for line in result.split('\n'):
1249 pair = [v.strip() for v in line.split(':')]
1250 if len(pair) != 2:
1251 continue
1252 if pair[0] == 'Device':
1253 if device_name:
1254 info[device_name] = device_info
1255 device_name = pair[1]
1256 device_info = {}
1257 else:
1258 device_info[pair[0]] = pair[1]
1259 if device_name and not device_name in info:
1260 info[device_name] = device_info
1261 return info
1262
1263
1264 def get_battery_percentage(self):
1265 """Get the battery percentage.
1266
1267 @return: The percentage of battery level, value range from 0-100. Return
1268 None if the battery info cannot be retrieved.
1269 """
1270 try:
1271 info = self.get_power_supply_info()
1272 logging.info(info)
1273 return float(info['Battery']['percentage'])
Dan Shie9b765d2014-12-29 16:59:49 -08001274 except (KeyError, ValueError, error.AutoservRunError):
Dan Shi49ca0932014-11-14 11:22:27 -08001275 return None
1276
1277
1278 def is_ac_connected(self):
1279 """Check if the dut has power adapter connected and charging.
1280
1281 @return: True if power adapter is connected and charging.
1282 """
1283 try:
1284 info = self.get_power_supply_info()
1285 return info['Line Power']['online'] == 'yes'
Dan Shie9b765d2014-12-29 16:59:49 -08001286 except (KeyError, error.AutoservRunError):
1287 return None
Dan Shi49ca0932014-11-14 11:22:27 -08001288
1289
Simran Basi5e6339a2013-03-21 11:34:32 -07001290 def _cleanup_poweron(self):
1291 """Special cleanup method to make sure hosts always get power back."""
1292 afe = frontend_wrappers.RetryingAFE(timeout_min=5, delay_sec=10)
1293 hosts = afe.get_hosts(hostname=self.hostname)
1294 if not hosts or not (self._RPM_OUTLET_CHANGED in
1295 hosts[0].attributes):
1296 return
1297 logging.debug('This host has recently interacted with the RPM'
1298 ' Infrastructure. Ensuring power is on.')
1299 try:
1300 self.power_on()
Dan Shi7dca56e2014-11-11 17:07:56 -08001301 afe.set_host_attribute(self._RPM_OUTLET_CHANGED, None,
1302 hostname=self.hostname)
Simran Basi5e6339a2013-03-21 11:34:32 -07001303 except rpm_client.RemotePowerException:
Simran Basi5e6339a2013-03-21 11:34:32 -07001304 logging.error('Failed to turn Power On for this host after '
1305 'cleanup through the RPM Infrastructure.')
Gabe Blackb72f4fb2015-01-20 16:47:13 -08001306 autotest_es.post(
Dan Shi7dca56e2014-11-11 17:07:56 -08001307 type_str='RPM_poweron_failure',
1308 metadata={'hostname': self.hostname})
Dan Shi49ca0932014-11-14 11:22:27 -08001309
1310 battery_percentage = self.get_battery_percentage()
Dan Shif01ebe22014-12-05 13:10:57 -08001311 if battery_percentage and battery_percentage < 50:
Dan Shi49ca0932014-11-14 11:22:27 -08001312 raise
1313 elif self.is_ac_connected():
1314 logging.info('The device has power adapter connected and '
1315 'charging. No need to try to turn RPM on '
1316 'again.')
1317 afe.set_host_attribute(self._RPM_OUTLET_CHANGED, None,
1318 hostname=self.hostname)
1319 logging.info('Battery level is now at %s%%. The device may '
1320 'still have enough power to run test, so no '
1321 'exception will be raised.', battery_percentage)
1322
Simran Basi5e6339a2013-03-21 11:34:32 -07001323
beepsc87ff602013-07-31 21:53:00 -07001324 def _is_factory_image(self):
1325 """Checks if the image on the DUT is a factory image.
1326
1327 @return: True if the image on the DUT is a factory image.
1328 False otherwise.
1329 """
1330 result = self.run('[ -f /root/.factory_test ]', ignore_status=True)
1331 return result.exit_status == 0
1332
1333
1334 def _restart_ui(self):
J. Richard Barnette84890bd2014-02-21 11:05:47 -08001335 """Restart the Chrome UI.
beepsc87ff602013-07-31 21:53:00 -07001336
1337 @raises: FactoryImageCheckerException for factory images, since
1338 we cannot attempt to restart ui on them.
1339 error.AutoservRunError for any other type of error that
1340 occurs while restarting ui.
1341 """
1342 if self._is_factory_image():
Dan Shi549fb822015-03-24 18:01:11 -07001343 raise FactoryImageCheckerException('Cannot restart ui on factory '
1344 'images')
beepsc87ff602013-07-31 21:53:00 -07001345
J. Richard Barnette84890bd2014-02-21 11:05:47 -08001346 # TODO(jrbarnette): The command to stop/start the ui job
1347 # should live inside cros_ui, too. However that would seem
1348 # to imply interface changes to the existing start()/restart()
1349 # functions, which is a bridge too far (for now).
J. Richard Barnette6069aa12015-06-08 09:10:24 -07001350 prompt = cros_ui.get_chrome_session_ident(self)
J. Richard Barnette84890bd2014-02-21 11:05:47 -08001351 self.run('stop ui; start ui')
1352 cros_ui.wait_for_chrome_ready(prompt, self)
beepsc87ff602013-07-31 21:53:00 -07001353
1354
xixuana3bbc422017-05-04 15:57:21 -07001355 def _get_lsb_release_content(self):
1356 """Return the content of lsb-release file of host."""
1357 return self.run(
1358 'cat "%s"' % client_constants.LSB_RELEASE).stdout.strip()
1359
1360
Dan Shi549fb822015-03-24 18:01:11 -07001361 def get_release_version(self):
1362 """Get the value of attribute CHROMEOS_RELEASE_VERSION from lsb-release.
1363
1364 @returns The version string in lsb-release, under attribute
1365 CHROMEOS_RELEASE_VERSION.
1366 """
Dan Shi549fb822015-03-24 18:01:11 -07001367 return lsbrelease_utils.get_chromeos_release_version(
xixuana3bbc422017-05-04 15:57:21 -07001368 lsb_release_content=self._get_lsb_release_content())
1369
1370
1371 def get_chromeos_release_milestone(self):
1372 """Get the value of attribute CHROMEOS_RELEASE_BUILD_TYPE
1373 from lsb-release.
1374
1375 @returns The version string in lsb-release, under attribute
1376 CHROMEOS_RELEASE_BUILD_TYPE.
1377 """
1378 return lsbrelease_utils.get_chromeos_release_milestone(
1379 lsb_release_content=self._get_lsb_release_content())
Dan Shi549fb822015-03-24 18:01:11 -07001380
1381
1382 def verify_cros_version_label(self):
1383 """ Make sure host's cros-version label match the actual image in dut.
1384
1385 Remove any cros-version: label that doesn't match that installed in
1386 the dut.
1387
1388 @param raise_error: Set to True to raise exception if any mismatch found
1389
1390 @raise error.AutoservError: If any mismatch between cros-version label
1391 and the build installed in dut is found.
1392 """
1393 labels = self._AFE.get_labels(
1394 name__startswith=ds_constants.VERSION_PREFIX,
1395 host__hostname=self.hostname)
1396 mismatch_found = False
1397 if labels:
1398 # Get CHROMEOS_RELEASE_VERSION from lsb-release, e.g., 6908.0.0.
1399 # Note that it's different from cros-version label, which has
1400 # builder and branch info, e.g.,
1401 # cros-version:peppy-release/R43-6908.0.0
1402 release_version = self.get_release_version()
1403 host_list = [self.hostname]
1404 for label in labels:
1405 # Remove any cros-version label that does not match
1406 # release_version.
1407 build_version = label.name[len(ds_constants.VERSION_PREFIX):]
1408 if not utils.version_match(build_version, release_version):
1409 logging.warn('cros-version label "%s" does not match '
1410 'release version %s. Removing the label.',
1411 label.name, release_version)
1412 label.remove_hosts(hosts=host_list)
1413 mismatch_found = True
1414 if mismatch_found:
Dan Shi1057bae2015-03-30 11:35:09 -07001415 autotest_es.post(use_http=True,
1416 type_str='cros_version_label_mismatch',
1417 metadata={'hostname': self.hostname})
Dan Shi549fb822015-03-24 18:01:11 -07001418 raise error.AutoservError('The host has wrong cros-version label.')
1419
1420
beepsc87ff602013-07-31 21:53:00 -07001421 def cleanup(self):
MK Ryu35d661e2014-09-25 17:44:10 -07001422 self.run('rm -f %s' % client_constants.CLEANUP_LOGS_PAUSED_FILE)
Scott Zawalskiddbc31e2012-11-15 11:29:01 -05001423 try:
beepsc87ff602013-07-31 21:53:00 -07001424 self._restart_ui()
1425 except (error.AutotestRunError, error.AutoservRunError,
1426 FactoryImageCheckerException):
Ilja H. Friedel04be2bd2014-05-07 21:29:59 -07001427 logging.warning('Unable to restart ui, rebooting device.')
Scott Zawalskiddbc31e2012-11-15 11:29:01 -05001428 # Since restarting the UI fails fall back to normal Autotest
1429 # cleanup routines, i.e. reboot the machine.
Fang Deng0ca40e22013-08-27 17:47:44 -07001430 super(CrosHost, self).cleanup()
Simran Basi5e6339a2013-03-21 11:34:32 -07001431 # Check if the rpm outlet was manipulated.
Simran Basid5e5e272012-09-24 15:23:59 -07001432 if self.has_power():
Simran Basi5e6339a2013-03-21 11:34:32 -07001433 self._cleanup_poweron()
Dan Shi549fb822015-03-24 18:01:11 -07001434 self.verify_cros_version_label()
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001435
1436
Yu-Ju Honga2be94a2012-07-31 09:48:52 -07001437 def reboot(self, **dargs):
1438 """
1439 This function reboots the site host. The more generic
1440 RemoteHost.reboot() performs sync and sleeps for 5
1441 seconds. This is not necessary for Chrome OS devices as the
1442 sync should be finished in a short time during the reboot
1443 command.
1444 """
Tom Wai-Hong Tamf5cd1d42012-08-13 12:04:08 +08001445 if 'reboot_cmd' not in dargs:
Doug Anderson7d5aeb22014-02-27 15:12:17 -08001446 reboot_timeout = dargs.get('reboot_timeout', 10)
J. Richard Barnette9af19632015-09-25 12:18:03 -07001447 dargs['reboot_cmd'] = ('sleep 1; '
1448 'reboot & sleep %d; '
1449 'reboot -f' % reboot_timeout)
Yu-Ju Honga2be94a2012-07-31 09:48:52 -07001450 # Enable fastsync to avoid running extra sync commands before reboot.
Tom Wai-Hong Tamf5cd1d42012-08-13 12:04:08 +08001451 if 'fastsync' not in dargs:
1452 dargs['fastsync'] = True
Michael Liangda8c60a2014-06-03 13:24:51 -07001453
Charlie Mooneya8e6dab2014-05-29 14:37:55 -07001454 # For purposes of logging reboot times:
1455 # Get the board name i.e. 'daisy_spring'
Michael Liangca4f5a62014-07-10 15:45:13 -07001456 board_fullname = self.get_board()
1457
1458 # Strip the prefix and add it to dargs.
1459 dargs['board'] = board_fullname[board_fullname.find(':')+1:]
Vincent Palatindf2372c2016-10-07 17:03:00 +02001460 # Record who called us
1461 orig = sys._getframe(1).f_code
Vincent Palatin80780b22016-07-27 16:02:37 +02001462 metric_fields = {'board' : dargs['board'],
Vincent Palatindf2372c2016-10-07 17:03:00 +02001463 'dut_host_name' : self.hostname,
1464 'success' : True}
1465 metric_debug_fields = {'board' : dargs['board'],
1466 'caller' : "%s:%s" % (orig.co_filename, orig.co_name),
1467 'success' : True,
1468 'error' : ''}
1469
Vincent Palatin80780b22016-07-27 16:02:37 +02001470 t0 = time.time()
1471 try:
1472 super(CrosHost, self).reboot(**dargs)
1473 except Exception as e:
1474 metric_fields['success'] = False
Vincent Palatindf2372c2016-10-07 17:03:00 +02001475 metric_debug_fields['success'] = False
1476 metric_debug_fields['error'] = type(e).__name__
Vincent Palatin80780b22016-07-27 16:02:37 +02001477 raise
1478 finally:
1479 duration = int(time.time() - t0)
Dan Shi5e2efb72017-02-07 11:40:23 -08001480 metrics.Counter(
1481 'chromeos/autotest/autoserv/reboot_count').increment(
1482 fields=metric_fields)
1483 metrics.Counter(
1484 'chromeos/autotest/autoserv/reboot_debug').increment(
1485 fields=metric_debug_fields)
1486 metrics.SecondsDistribution(
1487 'chromeos/autotest/autoserv/reboot_duration').add(
1488 duration, fields=metric_fields)
Yu-Ju Honga2be94a2012-07-31 09:48:52 -07001489
1490
Gwendal Grignou7a61d2f2014-05-23 11:05:51 -07001491 def suspend(self, **dargs):
1492 """
1493 This function suspends the site host.
1494 """
1495 suspend_time = dargs.get('suspend_time', 60)
1496 dargs['timeout'] = suspend_time
1497 if 'suspend_cmd' not in dargs:
J. Richard Barnette9af19632015-09-25 12:18:03 -07001498 dargs['suspend_cmd'] = ' && '.join([
1499 'echo 0 > /sys/class/rtc/rtc0/wakealarm',
Gwendal Grignou7a61d2f2014-05-23 11:05:51 -07001500 'echo +%d > /sys/class/rtc/rtc0/wakealarm' % suspend_time,
J. Richard Barnette9af19632015-09-25 12:18:03 -07001501 'powerd_dbus_suspend --delay=0'])
Gwendal Grignou7a61d2f2014-05-23 11:05:51 -07001502 super(CrosHost, self).suspend(**dargs)
1503
1504
Simran Basiec564392014-08-25 16:48:09 -07001505 def upstart_status(self, service_name):
1506 """Check the status of an upstart init script.
1507
1508 @param service_name: Service to look up.
1509
1510 @returns True if the service is running, False otherwise.
1511 """
1512 return self.run('status %s | grep start/running' %
1513 service_name).stdout.strip() != ''
1514
1515
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001516 def verify_software(self):
Richard Barnetteb2bc13c2013-01-08 17:32:51 -08001517 """Verify working software on a Chrome OS system.
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001518
Richard Barnetteb2bc13c2013-01-08 17:32:51 -08001519 Tests for the following conditions:
1520 1. All conditions tested by the parent version of this
1521 function.
1522 2. Sufficient space in /mnt/stateful_partition.
Fang Deng6b05f5b2013-03-20 13:42:11 -07001523 3. Sufficient space in /mnt/stateful_partition/encrypted.
1524 4. update_engine answers a simple status request over DBus.
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001525
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001526 """
Fang Deng0ca40e22013-08-27 17:47:44 -07001527 super(CrosHost, self).verify_software()
Dan Shib8540a52015-07-16 14:18:23 -07001528 default_kilo_inodes_required = CONFIG.get_config_value(
1529 'SERVER', 'kilo_inodes_required', type=int, default=100)
1530 board = self.get_board().replace(ds_constants.BOARD_PREFIX, '')
1531 kilo_inodes_required = CONFIG.get_config_value(
1532 'SERVER', 'kilo_inodes_required_%s' % board,
1533 type=int, default=default_kilo_inodes_required)
1534 self.check_inodes('/mnt/stateful_partition', kilo_inodes_required)
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001535 self.check_diskspace(
1536 '/mnt/stateful_partition',
Dan Shib8540a52015-07-16 14:18:23 -07001537 CONFIG.get_config_value(
Fang Deng6b05f5b2013-03-20 13:42:11 -07001538 'SERVER', 'gb_diskspace_required', type=float,
1539 default=20.0))
Gaurav Shahe448af82014-06-19 15:18:59 -07001540 encrypted_stateful_path = '/mnt/stateful_partition/encrypted'
1541 # Not all targets build with encrypted stateful support.
1542 if self.path_exists(encrypted_stateful_path):
1543 self.check_diskspace(
1544 encrypted_stateful_path,
Dan Shib8540a52015-07-16 14:18:23 -07001545 CONFIG.get_config_value(
Gaurav Shahe448af82014-06-19 15:18:59 -07001546 'SERVER', 'gb_encrypted_diskspace_required', type=float,
1547 default=0.1))
beepsc87ff602013-07-31 21:53:00 -07001548
Simran Basiec564392014-08-25 16:48:09 -07001549 if not self.upstart_status('system-services'):
Prashanth B5d0a0512014-04-25 12:26:08 -07001550 raise error.AutoservError('Chrome failed to reach login. '
1551 'System services not running.')
1552
beepsc87ff602013-07-31 21:53:00 -07001553 # Factory images don't run update engine,
1554 # goofy controls dbus on these DUTs.
1555 if not self._is_factory_image():
1556 self.run('update_engine_client --status')
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001557
Dan Shi549fb822015-03-24 18:01:11 -07001558 self.verify_cros_version_label()
1559
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001560
J. Richard Barnettea7e7fdc2016-02-12 12:35:36 -08001561 def verify(self):
1562 self._repair_strategy.verify(self)
1563
1564
Fang Deng96667ca2013-08-01 17:46:18 -07001565 def make_ssh_command(self, user='root', port=22, opts='', hosts_file=None,
1566 connect_timeout=None, alive_interval=None):
1567 """Override default make_ssh_command to use options tuned for Chrome OS.
1568
1569 Tuning changes:
1570 - ConnectTimeout=30; maximum of 30 seconds allowed for an SSH
1571 connection failure. Consistency with remote_access.sh.
1572
Samuel Tan2ce155b2015-06-23 18:24:38 -07001573 - ServerAliveInterval=900; which causes SSH to ping connection every
1574 900 seconds. In conjunction with ServerAliveCountMax ensures
1575 that if the connection dies, Autotest will bail out.
Fang Deng96667ca2013-08-01 17:46:18 -07001576 Originally tried 60 secs, but saw frequent job ABORTS where
Samuel Tan2ce155b2015-06-23 18:24:38 -07001577 the test completed successfully. Later increased from 180 seconds to
1578 900 seconds to account for tests where the DUT is suspended for
1579 longer periods of time.
Fang Deng96667ca2013-08-01 17:46:18 -07001580
1581 - ServerAliveCountMax=3; consistency with remote_access.sh.
1582
1583 - ConnectAttempts=4; reduce flakiness in connection errors;
1584 consistency with remote_access.sh.
1585
1586 - UserKnownHostsFile=/dev/null; we don't care about the keys.
1587 Host keys change with every new installation, don't waste
1588 memory/space saving them.
1589
1590 - SSH protocol forced to 2; needed for ServerAliveInterval.
1591
1592 @param user User name to use for the ssh connection.
1593 @param port Port on the target host to use for ssh connection.
1594 @param opts Additional options to the ssh command.
1595 @param hosts_file Ignored.
1596 @param connect_timeout Ignored.
1597 @param alive_interval Ignored.
1598 """
Aviv Keshetc5947fa2013-09-04 14:06:29 -07001599 base_command = ('/usr/bin/ssh -a -x %s %s %s'
1600 ' -o StrictHostKeyChecking=no'
Fang Deng96667ca2013-08-01 17:46:18 -07001601 ' -o UserKnownHostsFile=/dev/null -o BatchMode=yes'
Samuel Tan2ce155b2015-06-23 18:24:38 -07001602 ' -o ConnectTimeout=30 -o ServerAliveInterval=900'
Fang Deng96667ca2013-08-01 17:46:18 -07001603 ' -o ServerAliveCountMax=3 -o ConnectionAttempts=4'
1604 ' -o Protocol=2 -l %s -p %d')
Aviv Keshetc5947fa2013-09-04 14:06:29 -07001605 return base_command % (self._ssh_verbosity_flag, self._ssh_options,
1606 opts, user, port)
Jason Abeleb6f924f2013-11-13 16:01:54 -08001607 def syslog(self, message, tag='autotest'):
1608 """Logs a message to syslog on host.
1609
1610 @param message String message to log into syslog
1611 @param tag String tag prefix for syslog
1612
1613 """
1614 self.run('logger -t "%s" "%s"' % (tag, message))
1615
1616
J. Richard Barnetteb6de7e32013-02-14 13:28:04 -08001617 def _ping_check_status(self, status):
1618 """Ping the host once, and return whether it has a given status.
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001619
J. Richard Barnetteb6de7e32013-02-14 13:28:04 -08001620 @param status Check the ping status against this value.
1621 @return True iff `status` and the result of ping are the same
1622 (i.e. both True or both False).
1623
1624 """
1625 ping_val = utils.ping(self.hostname, tries=1, deadline=1)
1626 return not (status ^ (ping_val == 0))
1627
1628 def _ping_wait_for_status(self, status, timeout):
1629 """Wait for the host to have a given status (UP or DOWN).
1630
1631 Status is checked by polling. Polling will not last longer
1632 than the number of seconds in `timeout`. The polling
1633 interval will be long enough that only approximately
1634 _PING_WAIT_COUNT polling cycles will be executed, subject
1635 to a maximum interval of about one minute.
1636
1637 @param status Waiting will stop immediately if `ping` of the
1638 host returns this status.
1639 @param timeout Poll for at most this many seconds.
1640 @return True iff the host status from `ping` matched the
1641 requested status at the time of return.
1642
1643 """
1644 # _ping_check_status() takes about 1 second, hence the
1645 # "- 1" in the formula below.
Nathan Ciobanu38480a32016-10-25 15:26:45 -07001646 # FIXME: if the ping command errors then _ping_check_status()
1647 # returns instantly. If timeout is also smaller than twice
1648 # _PING_WAIT_COUNT then the while loop below forks many
1649 # thousands of ping commands (see /tmp/test_that_results_XXXXX/
1650 # /results-1-logging_YYY.ZZZ/debug/autoserv.DEBUG) and hogs one
1651 # CPU core for 60 seconds.
J. Richard Barnetteb6de7e32013-02-14 13:28:04 -08001652 poll_interval = min(int(timeout / self._PING_WAIT_COUNT), 60) - 1
1653 end_time = time.time() + timeout
1654 while time.time() <= end_time:
1655 if self._ping_check_status(status):
1656 return True
1657 if poll_interval > 0:
1658 time.sleep(poll_interval)
1659
1660 # The last thing we did was sleep(poll_interval), so it may
1661 # have been too long since the last `ping`. Check one more
1662 # time, just to be sure.
1663 return self._ping_check_status(status)
1664
1665 def ping_wait_up(self, timeout):
1666 """Wait for the host to respond to `ping`.
1667
1668 N.B. This method is not a reliable substitute for
1669 `wait_up()`, because a host that responds to ping will not
1670 necessarily respond to ssh. This method should only be used
1671 if the target DUT can be considered functional even if it
1672 can't be reached via ssh.
1673
1674 @param timeout Minimum time to allow before declaring the
1675 host to be non-responsive.
1676 @return True iff the host answered to ping before the timeout.
1677
1678 """
1679 return self._ping_wait_for_status(self._PING_STATUS_UP, timeout)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001680
Andrew Bresticker678c0c72013-01-22 10:44:09 -08001681 def ping_wait_down(self, timeout):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001682 """Wait until the host no longer responds to `ping`.
1683
J. Richard Barnetteb6de7e32013-02-14 13:28:04 -08001684 This function can be used as a slightly faster version of
1685 `wait_down()`, by avoiding potentially long ssh timeouts.
1686
1687 @param timeout Minimum time to allow for the host to become
1688 non-responsive.
1689 @return True iff the host quit answering ping before the
1690 timeout.
1691
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001692 """
J. Richard Barnetteb6de7e32013-02-14 13:28:04 -08001693 return self._ping_wait_for_status(self._PING_STATUS_DOWN, timeout)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001694
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08001695 def test_wait_for_sleep(self, sleep_timeout=None):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001696 """Wait for the client to enter low-power sleep mode.
1697
1698 The test for "is asleep" can't distinguish a system that is
1699 powered off; to confirm that the unit was asleep, it is
1700 necessary to force resume, and then call
1701 `test_wait_for_resume()`.
1702
1703 This function is expected to be called from a test as part
1704 of a sequence like the following:
1705
1706 ~~~~~~~~
1707 boot_id = host.get_boot_id()
1708 # trigger sleep on the host
1709 host.test_wait_for_sleep()
1710 # trigger resume on the host
1711 host.test_wait_for_resume(boot_id)
1712 ~~~~~~~~
1713
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08001714 @param sleep_timeout time limit in seconds to allow the host sleep.
1715
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001716 @exception TestFail The host did not go to sleep within
1717 the allowed time.
1718 """
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08001719 if sleep_timeout is None:
1720 sleep_timeout = self.SLEEP_TIMEOUT
1721
1722 if not self.ping_wait_down(timeout=sleep_timeout):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001723 raise error.TestFail(
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08001724 'client failed to sleep after %d seconds' % sleep_timeout)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001725
1726
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08001727 def test_wait_for_resume(self, old_boot_id, resume_timeout=None):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001728 """Wait for the client to resume from low-power sleep mode.
1729
1730 The `old_boot_id` parameter should be the value from
1731 `get_boot_id()` obtained prior to entering sleep mode. A
1732 `TestFail` exception is raised if the boot id changes.
1733
1734 See @ref test_wait_for_sleep for more on this function's
1735 usage.
1736
J. Richard Barnette7214e0b2013-02-06 15:20:49 -08001737 @param old_boot_id A boot id value obtained before the
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001738 target host went to sleep.
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08001739 @param resume_timeout time limit in seconds to allow the host up.
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001740
1741 @exception TestFail The host did not respond within the
1742 allowed time.
1743 @exception TestFail The host responded, but the boot id test
1744 indicated a reboot rather than a sleep
1745 cycle.
1746 """
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08001747 if resume_timeout is None:
1748 resume_timeout = self.RESUME_TIMEOUT
1749
1750 if not self.wait_up(timeout=resume_timeout):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001751 raise error.TestFail(
1752 'client failed to resume from sleep after %d seconds' %
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08001753 resume_timeout)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001754 else:
1755 new_boot_id = self.get_boot_id()
1756 if new_boot_id != old_boot_id:
Tom Wai-Hong Tam01792682015-01-06 08:00:46 +08001757 logging.error('client rebooted (old boot %s, new boot %s)',
1758 old_boot_id, new_boot_id)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001759 raise error.TestFail(
Tom Wai-Hong Tam01792682015-01-06 08:00:46 +08001760 'client rebooted, but sleep was expected')
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001761
1762
Tom Wai-Hong Tamfe005c22014-12-03 09:25:44 +08001763 def test_wait_for_shutdown(self, shutdown_timeout=None):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001764 """Wait for the client to shut down.
1765
1766 The test for "has shut down" can't distinguish a system that
1767 is merely asleep; to confirm that the unit was down, it is
1768 necessary to force boot, and then call test_wait_for_boot().
1769
1770 This function is expected to be called from a test as part
1771 of a sequence like the following:
1772
1773 ~~~~~~~~
1774 boot_id = host.get_boot_id()
1775 # trigger shutdown on the host
1776 host.test_wait_for_shutdown()
1777 # trigger boot on the host
1778 host.test_wait_for_boot(boot_id)
1779 ~~~~~~~~
1780
Tom Wai-Hong Tamfe005c22014-12-03 09:25:44 +08001781 @param shutdown_timeout time limit in seconds to allow the host down.
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001782 @exception TestFail The host did not shut down within the
1783 allowed time.
1784 """
Tom Wai-Hong Tamfe005c22014-12-03 09:25:44 +08001785 if shutdown_timeout is None:
1786 shutdown_timeout = self.SHUTDOWN_TIMEOUT
1787
1788 if not self.ping_wait_down(timeout=shutdown_timeout):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001789 raise error.TestFail(
1790 'client failed to shut down after %d seconds' %
Tom Wai-Hong Tamfe005c22014-12-03 09:25:44 +08001791 shutdown_timeout)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001792
1793
1794 def test_wait_for_boot(self, old_boot_id=None):
1795 """Wait for the client to boot from cold power.
1796
1797 The `old_boot_id` parameter should be the value from
1798 `get_boot_id()` obtained prior to shutting down. A
1799 `TestFail` exception is raised if the boot id does not
1800 change. The boot id test is omitted if `old_boot_id` is not
1801 specified.
1802
1803 See @ref test_wait_for_shutdown for more on this function's
1804 usage.
1805
J. Richard Barnette7214e0b2013-02-06 15:20:49 -08001806 @param old_boot_id A boot id value obtained before the
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001807 shut down.
1808
1809 @exception TestFail The host did not respond within the
1810 allowed time.
1811 @exception TestFail The host responded, but the boot id test
1812 indicated that there was no reboot.
1813 """
J. Richard Barnetteeb69d722012-06-18 17:29:44 -07001814 if not self.wait_up(timeout=self.REBOOT_TIMEOUT):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001815 raise error.TestFail(
1816 'client failed to reboot after %d seconds' %
J. Richard Barnetteeb69d722012-06-18 17:29:44 -07001817 self.REBOOT_TIMEOUT)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001818 elif old_boot_id:
1819 if self.get_boot_id() == old_boot_id:
Tom Wai-Hong Tam01792682015-01-06 08:00:46 +08001820 logging.error('client not rebooted (boot %s)',
1821 old_boot_id)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001822 raise error.TestFail(
Tom Wai-Hong Tam01792682015-01-06 08:00:46 +08001823 'client is back up, but did not reboot')
Simran Basid5e5e272012-09-24 15:23:59 -07001824
1825
1826 @staticmethod
1827 def check_for_rpm_support(hostname):
1828 """For a given hostname, return whether or not it is powered by an RPM.
1829
Simran Basi1df55112013-09-06 11:25:09 -07001830 @param hostname: hostname to check for rpm support.
1831
Simran Basid5e5e272012-09-24 15:23:59 -07001832 @return None if this host does not follows the defined naming format
1833 for RPM powered DUT's in the lab. If it does follow the format,
1834 it returns a regular expression MatchObject instead.
1835 """
Fang Dengbaff9082015-01-06 13:46:15 -08001836 return re.match(CrosHost._RPM_HOSTNAME_REGEX, hostname)
Simran Basid5e5e272012-09-24 15:23:59 -07001837
1838
1839 def has_power(self):
1840 """For this host, return whether or not it is powered by an RPM.
1841
1842 @return True if this host is in the CROS lab and follows the defined
1843 naming format.
1844 """
Fang Deng0ca40e22013-08-27 17:47:44 -07001845 return CrosHost.check_for_rpm_support(self.hostname)
Simran Basid5e5e272012-09-24 15:23:59 -07001846
1847
Ismail Noorbasha07fdb612013-02-14 14:13:31 -08001848 def _set_power(self, state, power_method):
1849 """Sets the power to the host via RPM, Servo or manual.
1850
1851 @param state Specifies which power state to set to DUT
1852 @param power_method Specifies which method of power control to
1853 use. By default "RPM" will be used. Valid values
1854 are the strings "RPM", "manual", "servoj10".
1855
1856 """
1857 ACCEPTABLE_STATES = ['ON', 'OFF']
1858
1859 if state.upper() not in ACCEPTABLE_STATES:
1860 raise error.TestError('State must be one of: %s.'
1861 % (ACCEPTABLE_STATES,))
1862
1863 if power_method == self.POWER_CONTROL_SERVO:
1864 logging.info('Setting servo port J10 to %s', state)
1865 self.servo.set('prtctl3_pwren', state.lower())
1866 time.sleep(self._USB_POWER_TIMEOUT)
1867 elif power_method == self.POWER_CONTROL_MANUAL:
1868 logging.info('You have %d seconds to set the AC power to %s.',
1869 self._POWER_CYCLE_TIMEOUT, state)
1870 time.sleep(self._POWER_CYCLE_TIMEOUT)
1871 else:
1872 if not self.has_power():
1873 raise error.TestFail('DUT does not have RPM connected.')
Simran Basi5e6339a2013-03-21 11:34:32 -07001874 afe = frontend_wrappers.RetryingAFE(timeout_min=5, delay_sec=10)
1875 afe.set_host_attribute(self._RPM_OUTLET_CHANGED, True,
1876 hostname=self.hostname)
Simran Basi1df55112013-09-06 11:25:09 -07001877 rpm_client.set_power(self.hostname, state.upper(), timeout_mins=5)
Simran Basid5e5e272012-09-24 15:23:59 -07001878
1879
Ismail Noorbasha07fdb612013-02-14 14:13:31 -08001880 def power_off(self, power_method=POWER_CONTROL_RPM):
1881 """Turn off power to this host via RPM, Servo or manual.
1882
1883 @param power_method Specifies which method of power control to
1884 use. By default "RPM" will be used. Valid values
1885 are the strings "RPM", "manual", "servoj10".
1886
1887 """
1888 self._set_power('OFF', power_method)
Simran Basid5e5e272012-09-24 15:23:59 -07001889
1890
Ismail Noorbasha07fdb612013-02-14 14:13:31 -08001891 def power_on(self, power_method=POWER_CONTROL_RPM):
1892 """Turn on power to this host via RPM, Servo or manual.
1893
1894 @param power_method Specifies which method of power control to
1895 use. By default "RPM" will be used. Valid values
1896 are the strings "RPM", "manual", "servoj10".
1897
1898 """
1899 self._set_power('ON', power_method)
1900
1901
1902 def power_cycle(self, power_method=POWER_CONTROL_RPM):
1903 """Cycle power to this host by turning it OFF, then ON.
1904
1905 @param power_method Specifies which method of power control to
1906 use. By default "RPM" will be used. Valid values
1907 are the strings "RPM", "manual", "servoj10".
1908
1909 """
1910 if power_method in (self.POWER_CONTROL_SERVO,
1911 self.POWER_CONTROL_MANUAL):
1912 self.power_off(power_method=power_method)
1913 time.sleep(self._POWER_CYCLE_TIMEOUT)
1914 self.power_on(power_method=power_method)
1915 else:
1916 rpm_client.set_power(self.hostname, 'CYCLE')
Simran Basic6f1f7a2012-10-16 10:47:46 -07001917
1918
1919 def get_platform(self):
1920 """Determine the correct platform label for this host.
1921
1922 @returns a string representing this host's platform.
1923 """
1924 crossystem = utils.Crossystem(self)
1925 crossystem.init()
1926 # Extract fwid value and use the leading part as the platform id.
1927 # fwid generally follow the format of {platform}.{firmware version}
1928 # Example: Alex.X.YYY.Z or Google_Alex.X.YYY.Z
1929 platform = crossystem.fwid().split('.')[0].lower()
1930 # Newer platforms start with 'Google_' while the older ones do not.
1931 return platform.replace('google_', '')
1932
1933
Hung-ying Tyanb1328032014-04-01 14:18:54 +08001934 def get_architecture(self):
1935 """Determine the correct architecture label for this host.
1936
1937 @returns a string representing this host's architecture.
1938 """
1939 crossystem = utils.Crossystem(self)
1940 crossystem.init()
1941 return crossystem.arch()
1942
1943
Luis Lozano40b7d0d2014-01-17 15:12:06 -08001944 def get_chrome_version(self):
1945 """Gets the Chrome version number and milestone as strings.
1946
1947 Invokes "chrome --version" to get the version number and milestone.
1948
1949 @return A tuple (chrome_ver, milestone) where "chrome_ver" is the
1950 current Chrome version number as a string (in the form "W.X.Y.Z")
1951 and "milestone" is the first component of the version number
1952 (the "W" from "W.X.Y.Z"). If the version number cannot be parsed
1953 in the "W.X.Y.Z" format, the "chrome_ver" will be the full output
1954 of "chrome --version" and the milestone will be the empty string.
1955
1956 """
MK Ryu35d661e2014-09-25 17:44:10 -07001957 version_string = self.run(client_constants.CHROME_VERSION_COMMAND).stdout
Luis Lozano40b7d0d2014-01-17 15:12:06 -08001958 return utils.parse_chrome_version(version_string)
1959
J. Richard Barnetted2af5852016-02-05 15:03:10 -08001960
Kevin Chenga2619dc2016-03-28 11:42:08 -07001961 # TODO(kevcheng): change this to just return the board without the
1962 # 'board:' prefix and fix up all the callers. Also look into removing the
1963 # need for this method.
Simran Basic6f1f7a2012-10-16 10:47:46 -07001964 def get_board(self):
1965 """Determine the correct board label for this host.
1966
1967 @returns a string representing this host's board.
1968 """
1969 release_info = utils.parse_cmd_output('cat /etc/lsb-release',
1970 run_method=self.run)
J. Richard Barnetted2af5852016-02-05 15:03:10 -08001971 return (ds_constants.BOARD_PREFIX +
1972 release_info['CHROMEOS_RELEASE_BOARD'])
Simran Basic6f1f7a2012-10-16 10:47:46 -07001973
1974
Kevin Chenga328da62016-03-31 10:49:04 -07001975
1976 def has_lightsensor(self):
1977 """Determine the correct board label for this host.
1978
1979 @returns the string 'lightsensor' if this host has a lightsensor or
1980 None if it does not.
1981 """
1982 search_cmd = "find -L %s -maxdepth 4 | egrep '%s'" % (
1983 self._LIGHTSENSOR_SEARCH_DIR, '|'.join(self._LIGHTSENSOR_FILES))
1984 try:
1985 # Run the search cmd following the symlinks. Stderr_tee is set to
1986 # None as there can be a symlink loop, but the command will still
1987 # execute correctly with a few messages printed to stderr.
1988 self.run(search_cmd, stdout_tee=None, stderr_tee=None)
1989 return 'lightsensor'
1990 except error.AutoservRunError:
1991 # egrep exited with a return code of 1 meaning none of the possible
1992 # lightsensor files existed.
1993 return None
1994
1995
1996 def has_bluetooth(self):
1997 """Determine the correct board label for this host.
1998
1999 @returns the string 'bluetooth' if this host has bluetooth or
2000 None if it does not.
2001 """
2002 try:
2003 self.run('test -d /sys/class/bluetooth/hci0')
2004 # test exited with a return code of 0.
2005 return 'bluetooth'
2006 except error.AutoservRunError:
2007 # test exited with a return code 1 meaning the directory did not
2008 # exist.
2009 return None
2010
2011
Kevin Chenga328da62016-03-31 10:49:04 -07002012 def get_accels(self):
2013 """
2014 Determine the type of accelerometers on this host.
2015
2016 @returns a string representing this host's accelerometer type.
2017 At present, it only returns "accel:cros-ec", for accelerometers
2018 attached to a Chrome OS EC, or none, if no accelerometers.
2019 """
2020 # Check to make sure we have ectool
2021 rv = self.run('which ectool', ignore_status=True)
2022 if rv.exit_status:
2023 logging.info("No ectool cmd found, assuming no EC accelerometers")
2024 return None
2025
2026 # Check that the EC supports the motionsense command
2027 rv = self.run('ectool motionsense', ignore_status=True)
2028 if rv.exit_status:
2029 logging.info("EC does not support motionsense command "
2030 "assuming no EC accelerometers")
2031 return None
2032
2033 # Check that EC motion sensors are active
2034 active = self.run('ectool motionsense active').stdout.split('\n')
2035 if active[0] == "0":
2036 logging.info("Motion sense inactive, assuming no EC accelerometers")
2037 return None
2038
2039 logging.info("EC accelerometers found")
2040 return 'accel:cros-ec'
2041
2042
2043 def has_chameleon(self):
2044 """Determine if a Chameleon connected to this host.
2045
2046 @returns a list containing two strings ('chameleon' and
2047 'chameleon:' + label, e.g. 'chameleon:hdmi') if this host
2048 has a Chameleon or None if it has not.
2049 """
2050 if self._chameleon_host:
2051 return ['chameleon', 'chameleon:' + self.chameleon.get_label()]
2052 else:
2053 return None
2054
2055
2056 def has_loopback_dongle(self):
2057 """Determine if an audio loopback dongle is plugged to this host.
2058
2059 @returns 'audio_loopback_dongle' when there is an audio loopback dongle
2060 plugged to this host.
2061 None when there is no audio loopback dongle
2062 plugged to this host.
2063 """
2064 nodes_info = self.run(command=cras_utils.get_cras_nodes_cmd(),
2065 ignore_status=True).stdout
2066 if (cras_utils.node_type_is_plugged('HEADPHONE', nodes_info) and
2067 cras_utils.node_type_is_plugged('MIC', nodes_info)):
2068 return 'audio_loopback_dongle'
2069 else:
2070 return None
2071
2072
2073 def get_power_supply(self):
2074 """
2075 Determine what type of power supply the host has
2076
2077 @returns a string representing this host's power supply.
2078 'power:battery' when the device has a battery intended for
2079 extended use
2080 'power:AC_primary' when the device has a battery not intended
2081 for extended use (for moving the machine, etc)
2082 'power:AC_only' when the device has no battery at all.
2083 """
2084 psu = self.run(command='mosys psu type', ignore_status=True)
2085 if psu.exit_status:
2086 # The psu command for mosys is not included for all platforms. The
2087 # assumption is that the device will have a battery if the command
2088 # is not found.
2089 return 'power:battery'
2090
2091 psu_str = psu.stdout.strip()
2092 if psu_str == 'unknown':
2093 return None
2094
2095 return 'power:%s' % psu_str
2096
2097
2098 def get_storage(self):
2099 """
2100 Determine the type of boot device for this host.
2101
2102 Determine if the internal device is SCSI or dw_mmc device.
2103 Then check that it is SSD or HDD or eMMC or something else.
2104
2105 @returns a string representing this host's internal device type.
2106 'storage:ssd' when internal device is solid state drive
2107 'storage:hdd' when internal device is hard disk drive
2108 'storage:mmc' when internal device is mmc drive
2109 None When internal device is something else or
2110 when we are unable to determine the type
2111 """
2112 # The output should be /dev/mmcblk* for SD/eMMC or /dev/sd* for scsi
2113 rootdev_cmd = ' '.join(['. /usr/sbin/write_gpt.sh;',
2114 '. /usr/share/misc/chromeos-common.sh;',
2115 'load_base_vars;',
2116 'get_fixed_dst_drive'])
2117 rootdev = self.run(command=rootdev_cmd, ignore_status=True)
2118 if rootdev.exit_status:
2119 logging.info("Fail to run %s", rootdev_cmd)
2120 return None
2121 rootdev_str = rootdev.stdout.strip()
2122
2123 if not rootdev_str:
2124 return None
2125
2126 rootdev_base = os.path.basename(rootdev_str)
2127
2128 mmc_pattern = '/dev/mmcblk[0-9]'
2129 if re.match(mmc_pattern, rootdev_str):
2130 # Use type to determine if the internal device is eMMC or somthing
2131 # else. We can assume that MMC is always an internal device.
2132 type_cmd = 'cat /sys/block/%s/device/type' % rootdev_base
2133 type = self.run(command=type_cmd, ignore_status=True)
2134 if type.exit_status:
2135 logging.info("Fail to run %s", type_cmd)
2136 return None
2137 type_str = type.stdout.strip()
2138
2139 if type_str == 'MMC':
2140 return 'storage:mmc'
2141
2142 scsi_pattern = '/dev/sd[a-z]+'
2143 if re.match(scsi_pattern, rootdev.stdout):
2144 # Read symlink for /sys/block/sd* to determine if the internal
2145 # device is connected via ata or usb.
2146 link_cmd = 'readlink /sys/block/%s' % rootdev_base
2147 link = self.run(command=link_cmd, ignore_status=True)
2148 if link.exit_status:
2149 logging.info("Fail to run %s", link_cmd)
2150 return None
2151 link_str = link.stdout.strip()
2152 if 'usb' in link_str:
2153 return None
2154
2155 # Read rotation to determine if the internal device is ssd or hdd.
2156 rotate_cmd = str('cat /sys/block/%s/queue/rotational'
2157 % rootdev_base)
2158 rotate = self.run(command=rotate_cmd, ignore_status=True)
2159 if rotate.exit_status:
2160 logging.info("Fail to run %s", rotate_cmd)
2161 return None
2162 rotate_str = rotate.stdout.strip()
2163
2164 rotate_dict = {'0':'storage:ssd', '1':'storage:hdd'}
2165 return rotate_dict.get(rotate_str)
2166
2167 # All other internal device / error case will always fall here
2168 return None
2169
2170
2171 def get_servo(self):
2172 """Determine if the host has a servo attached.
2173
2174 If the host has a working servo attached, it should have a servo label.
2175
2176 @return: string 'servo' if the host has servo attached. Otherwise,
2177 returns None.
2178 """
2179 return 'servo' if self._servo_host else None
2180
2181
2182 def get_video_labels(self):
2183 """Run /usr/local/bin/avtest_label_detect to get a list of video labels.
2184
2185 Sample output of avtest_label_detect:
2186 Detected label: hw_video_acc_vp8
2187 Detected label: webcam
2188
2189 @return: A list of labels detected by tool avtest_label_detect.
2190 """
2191 try:
2192 result = self.run('/usr/local/bin/avtest_label_detect').stdout
2193 return re.findall('^Detected label: (\w+)$', result, re.M)
2194 except error.AutoservRunError:
2195 # The tool is not installed.
2196 return []
2197
2198
2199 def is_video_glitch_detection_supported(self):
2200 """ Determine if a board under test is supported for video glitch
2201 detection tests.
2202
2203 @return: 'video_glitch_detection' if board is supported, None otherwise.
2204 """
2205 board = self.get_board().replace(ds_constants.BOARD_PREFIX, '')
2206
2207 if board in video_test_constants.SUPPORTED_BOARDS:
2208 return 'video_glitch_detection'
2209
2210 return None
2211
2212
2213 def get_touch(self):
2214 """
2215 Determine whether board under test has a touchpad or touchscreen.
2216
2217 @return: A list of some combination of 'touchscreen' and 'touchpad',
2218 depending on what is present on the device.
2219
2220 """
2221 labels = []
2222 looking_for = ['touchpad', 'touchscreen']
2223 player = input_playback.InputPlayback()
2224 input_events = self.run('ls /dev/input/event*').stdout.strip().split()
2225 filename = '/tmp/touch_labels'
2226 for event in input_events:
2227 self.run('evtest %s > %s' % (event, filename), timeout=1,
2228 ignore_timeout=True)
2229 properties = self.run('cat %s' % filename).stdout
2230 input_type = player._determine_input_type(properties)
2231 if input_type in looking_for:
2232 labels.append(input_type)
2233 looking_for.remove(input_type)
2234 if len(looking_for) == 0:
2235 break
2236 self.run('rm %s' % filename)
2237
2238 return labels
2239
2240
2241 def has_internal_display(self):
2242 """Determine if the device under test is equipped with an internal
2243 display.
2244
2245 @return: 'internal_display' if one is present; None otherwise.
2246 """
2247 from autotest_lib.client.cros.graphics import graphics_utils
2248 from autotest_lib.client.common_lib import utils as common_utils
2249
2250 def __system_output(cmd):
2251 return self.run(cmd).stdout
2252
2253 def __read_file(remote_path):
2254 return self.run('cat %s' % remote_path).stdout
2255
2256 # Hijack the necessary client functions so that we can take advantage
2257 # of the client lib here.
2258 # FIXME: find a less hacky way than this
2259 original_system_output = utils.system_output
2260 original_read_file = common_utils.read_file
2261 utils.system_output = __system_output
2262 common_utils.read_file = __read_file
2263 try:
2264 return ('internal_display' if graphics_utils.has_internal_display()
2265 else None)
2266 finally:
2267 utils.system_output = original_system_output
2268 common_utils.read_file = original_read_file
2269
2270
2271 def has_lucid_sleep_support(self):
2272 """Determine if the device under test has support for lucid sleep.
2273
2274 @return 'lucidsleep' if this board supports lucid sleep; None otherwise
2275 """
2276 board = self.get_board().replace(ds_constants.BOARD_PREFIX, '')
2277 return 'lucidsleep' if board in LUCID_SLEEP_BOARDS else None
2278
2279
Dan Shi85276d42014-04-08 22:11:45 -07002280 def is_boot_from_usb(self):
2281 """Check if DUT is boot from USB.
2282
2283 @return: True if DUT is boot from usb.
2284 """
2285 device = self.run('rootdev -s -d').stdout.strip()
2286 removable = int(self.run('cat /sys/block/%s/removable' %
2287 os.path.basename(device)).stdout.strip())
2288 return removable == 1
Helen Zhang17dae2b2014-11-11 09:25:52 -08002289
2290
2291 def read_from_meminfo(self, key):
Dan Shi49ca0932014-11-14 11:22:27 -08002292 """Return the memory info from /proc/meminfo
Helen Zhang17dae2b2014-11-11 09:25:52 -08002293
2294 @param key: meminfo requested
2295
2296 @return the memory value as a string
2297
2298 """
Helen Zhang17dae2b2014-11-11 09:25:52 -08002299 meminfo = self.run('grep %s /proc/meminfo' % key).stdout.strip()
2300 logging.debug('%s', meminfo)
2301 return int(re.search(r'\d+', meminfo).group(0))
Rohit Makasana8a4923c2015-08-13 17:04:26 -07002302
2303
Rohit Makasana98e696f2016-06-03 18:48:10 -07002304 def get_cpu_arch(self):
2305 """Returns CPU arch of the device.
2306
2307 @return CPU architecture of the DUT.
2308 """
Allen Li2c32d6b2017-02-03 15:28:10 -08002309 # Add CPUs by following logic in client/bin/utils.py.
Rohit Makasana98e696f2016-06-03 18:48:10 -07002310 if self.run("grep '^flags.*:.* lm .*' /proc/cpuinfo",
2311 ignore_status=True).stdout:
2312 return 'x86_64'
2313 if self.run("grep -Ei 'ARM|CPU implementer' /proc/cpuinfo",
2314 ignore_status=True).stdout:
2315 return 'arm'
2316 return 'i386'
2317
2318
Rohit Makasana8a4923c2015-08-13 17:04:26 -07002319 def get_board_type(self):
2320 """
2321 Get the DUT's device type from /etc/lsb-release.
Danny Chan471a8d12015-08-18 14:57:41 -07002322 DEVICETYPE can be one of CHROMEBOX, CHROMEBASE, CHROMEBOOK or more.
2323
2324 @return value of DEVICETYPE param from lsb-release.
Rohit Makasana8a4923c2015-08-13 17:04:26 -07002325 """
Danny Chan471a8d12015-08-18 14:57:41 -07002326 device_type = self.run('grep DEVICETYPE /etc/lsb-release',
2327 ignore_status=True).stdout
2328 if device_type:
Kalin Stoyanov524310b2015-08-21 16:24:04 -07002329 return device_type.split('=')[-1].strip()
Danny Chan471a8d12015-08-18 14:57:41 -07002330 return ''
Gilad Arnolda76bef02015-09-29 13:55:15 -07002331
2332
2333 def get_os_type(self):
2334 return 'cros'
Simran Basia5522a32015-10-06 11:01:24 -07002335
2336
2337 def enable_adb_testing(self):
2338 """Mark this host as an adb tester."""
Dan Shia2872172015-10-31 01:16:51 -07002339 self.run('touch %s' % constants.ANDROID_TESTER_FILEFLAG)
Kevin Chenga2619dc2016-03-28 11:42:08 -07002340
2341
2342 def get_labels(self):
Kevin Chengf8660142016-08-12 10:17:41 -07002343 """Return the detected labels on the host."""
Kevin Chenga2619dc2016-03-28 11:42:08 -07002344 return self.labels.get_labels(self)
Kevin Chengf8660142016-08-12 10:17:41 -07002345
2346
2347 def update_labels(self):
2348 """Update the labels on the host."""
2349 self.labels.update_labels(self)