blob: 4819112541cc1b2fc3237032a99632432f57c19d [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
Fang Denge545abb2014-12-30 18:43:47 -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):
Richard Barnette383ef9c2016-12-13 11:56:49 -0800334 return afe_utils.get_stable_cros_image_name(
335 self._get_board_from_afe())
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500336
337
beepsdae65fd2013-07-26 16:24:41 -0700338 def verify_job_repo_url(self, tag=''):
beepscb6f1e22013-06-28 19:14:10 -0700339 """
340 Make sure job_repo_url of this host is valid.
341
joychen03eaad92013-06-26 09:55:21 -0700342 Eg: The job_repo_url "http://lmn.cd.ab.xyx:8080/static/\
beepscb6f1e22013-06-28 19:14:10 -0700343 lumpy-release/R29-4279.0.0/autotest/packages" claims to have the
344 autotest package for lumpy-release/R29-4279.0.0. If this isn't the case,
345 download and extract it. If the devserver embedded in the url is
346 unresponsive, update the job_repo_url of the host after staging it on
347 another devserver.
348
349 @param job_repo_url: A url pointing to the devserver where the autotest
350 package for this build should be staged.
beepsdae65fd2013-07-26 16:24:41 -0700351 @param tag: The tag from the server job, in the format
352 <job_id>-<user>/<hostname>, or <hostless> for a server job.
beepscb6f1e22013-06-28 19:14:10 -0700353
354 @raises DevServerException: If we could not resolve a devserver.
355 @raises AutoservError: If we're unable to save the new job_repo_url as
356 a result of choosing a new devserver because the old one failed to
357 respond to a health check.
beeps0c865032013-07-30 11:37:06 -0700358 @raises urllib2.URLError: If the devserver embedded in job_repo_url
359 doesn't respond within the timeout.
beepscb6f1e22013-06-28 19:14:10 -0700360 """
Dan Shibe3636a2016-02-14 22:48:01 -0800361 job_repo_url = afe_utils.get_host_attribute(self,
362 ds_constants.JOB_REPO_URL)
beepscb6f1e22013-06-28 19:14:10 -0700363 if not job_repo_url:
364 logging.warning('No job repo url set on host %s', self.hostname)
365 return
366
367 logging.info('Verifying job repo url %s', job_repo_url)
368 devserver_url, image_name = tools.get_devserver_build_from_package_url(
369 job_repo_url)
370
beeps0c865032013-07-30 11:37:06 -0700371 ds = dev_server.ImageServer(devserver_url)
beepscb6f1e22013-06-28 19:14:10 -0700372
373 logging.info('Staging autotest artifacts for %s on devserver %s',
374 image_name, ds.url())
beeps687243d2013-07-18 15:29:27 -0700375
376 start_time = time.time()
Simran Basi25e7a922014-10-31 11:56:10 -0700377 ds.stage_artifacts(image_name, ['autotest_packages'])
beeps687243d2013-07-18 15:29:27 -0700378 stage_time = time.time() - start_time
379
380 # Record how much of the verification time comes from a devserver
381 # restage. If we're doing things right we should not see multiple
382 # devservers for a given board/build/branch path.
383 try:
J. Richard Barnette3cbd76b2013-11-27 12:11:25 -0800384 board, build_type, branch = server_utils.ParseBuildName(
beeps687243d2013-07-18 15:29:27 -0700385 image_name)[:3]
J. Richard Barnette3cbd76b2013-11-27 12:11:25 -0800386 except server_utils.ParseBuildNameException:
beeps687243d2013-07-18 15:29:27 -0700387 pass
388 else:
beeps0c865032013-07-30 11:37:06 -0700389 devserver = devserver_url[
Chris Sosa65425082013-10-16 13:26:22 -0700390 devserver_url.find('/') + 2:devserver_url.rfind(':')]
beeps687243d2013-07-18 15:29:27 -0700391 stats_key = {
392 'board': board,
393 'build_type': build_type,
394 'branch': branch,
beeps0c865032013-07-30 11:37:06 -0700395 'devserver': devserver.replace('.', '_'),
beeps687243d2013-07-18 15:29:27 -0700396 }
Dan Shi5e2efb72017-02-07 11:40:23 -0800397
398 monarch_fields = {
399 'board': board,
400 'build_type': build_type,
401 # TODO(akeshet): To be consistent with most other metrics,
402 # consider changing the following field to be named
403 # 'milestone'.
404 'branch': branch,
405 'dev_server': devserver,
406 }
407 metrics.Counter(
408 'chromeos/autotest/provision/verify_url'
409 ).increment(fields=monarch_fields)
410 metrics.SecondsDistribution(
411 'chromeos/autotest/provision/verify_url_duration'
412 ).add(stage_time, fields=monarch_fields)
413
414
Dan Shicf4d2032015-03-12 15:04:21 -0700415 def stage_server_side_package(self, image=None):
416 """Stage autotest server-side package on devserver.
417
418 @param image: Full path of an OS image to install or a build name.
419
420 @return: A url to the autotest server-side package.
Dan Shi14de7622016-08-22 11:09:06 -0700421
422 @raise: error.AutoservError if fail to locate the build to test with, or
423 fail to stage server-side package.
Dan Shicf4d2032015-03-12 15:04:21 -0700424 """
Dan Shid37736b2016-07-06 15:10:29 -0700425 # If enable_drone_in_restricted_subnet is False, do not set hostname
426 # in devserver.resolve call, so a devserver in non-restricted subnet
427 # is picked to stage autotest server package for drone to download.
428 hostname = self.hostname
429 if not server_utils.ENABLE_DRONE_IN_RESTRICTED_SUBNET:
430 hostname = None
Dan Shicf4d2032015-03-12 15:04:21 -0700431 if image:
432 image_name = tools.get_build_from_image(image)
433 if not image_name:
434 raise error.AutoservError(
435 'Failed to parse build name from %s' % image)
Dan Shid37736b2016-07-06 15:10:29 -0700436 ds = dev_server.ImageServer.resolve(image_name, hostname)
Dan Shicf4d2032015-03-12 15:04:21 -0700437 else:
Dan Shibe3636a2016-02-14 22:48:01 -0800438 job_repo_url = afe_utils.get_host_attribute(
439 self, ds_constants.JOB_REPO_URL)
Dan Shicf4d2032015-03-12 15:04:21 -0700440 if job_repo_url:
441 devserver_url, image_name = (
442 tools.get_devserver_build_from_package_url(job_repo_url))
Dan Shid37736b2016-07-06 15:10:29 -0700443 # If enable_drone_in_restricted_subnet is True, use the
444 # existing devserver. Otherwise, resolve a new one in
445 # non-restricted subnet.
446 if server_utils.ENABLE_DRONE_IN_RESTRICTED_SUBNET:
447 ds = dev_server.ImageServer(devserver_url)
448 else:
449 ds = dev_server.ImageServer.resolve(image_name)
Dan Shicf4d2032015-03-12 15:04:21 -0700450 else:
Kevin Cheng84a71ba2016-07-14 11:03:57 -0700451 labels = afe_utils.get_labels(self, self.VERSION_PREFIX)
Dan Shicf4d2032015-03-12 15:04:21 -0700452 if not labels:
453 raise error.AutoservError(
454 'Failed to stage server-side package. The host has '
455 'no job_report_url attribute or version label.')
Kevin Cheng84a71ba2016-07-14 11:03:57 -0700456 image_name = labels[0][len(self.VERSION_PREFIX + ':'):]
Dan Shid37736b2016-07-06 15:10:29 -0700457 ds = dev_server.ImageServer.resolve(image_name, hostname)
Dan Shica503482015-03-30 17:23:25 -0700458
459 # Get the OS version of the build, for any build older than
460 # MIN_VERSION_SUPPORT_SSP, server side packaging is not supported.
461 match = re.match('.*/R\d+-(\d+)\.', image_name)
462 if match and int(match.group(1)) < self.MIN_VERSION_SUPPORT_SSP:
Dan Shi14de7622016-08-22 11:09:06 -0700463 raise error.AutoservError(
464 'Build %s is older than %s. Server side packaging is '
465 'disabled.' % (image_name, self.MIN_VERSION_SUPPORT_SSP))
Dan Shica503482015-03-30 17:23:25 -0700466
Dan Shicf4d2032015-03-12 15:04:21 -0700467 ds.stage_artifacts(image_name, ['autotest_server_package'])
468 return '%s/static/%s/%s' % (ds.url(), image_name,
469 'autotest_server_package.tar.bz2')
470
471
Dan Shi0f466e82013-02-22 15:44:58 -0800472 def _try_stateful_update(self, update_url, force_update, updater):
473 """Try to use stateful update to initialize DUT.
474
475 When DUT is already running the same version that machine_install
476 tries to install, stateful update is a much faster way to clean up
477 the DUT for testing, compared to a full reimage. It is implemeted
478 by calling autoupdater.run_update, but skipping updating root, as
479 updating the kernel is time consuming and not necessary.
480
481 @param update_url: url of the image.
482 @param force_update: Set to True to update the image even if the DUT
483 is running the same version.
484 @param updater: ChromiumOSUpdater instance used to update the DUT.
485 @returns: True if the DUT was updated with stateful update.
486
487 """
Dan Shi10b98482016-02-02 14:38:50 -0800488 # Stop service ap-update-manager to prevent rebooting during autoupdate.
489 # The service is used in jetstream boards, but not other CrOS devices.
490 self.run('sudo stop ap-update-manager', ignore_status=True)
491
J. Richard Barnette3f731032014-04-07 17:42:59 -0700492 # TODO(jrbarnette): Yes, I hate this re.match() test case.
493 # It's better than the alternative: see crbug.com/360944.
494 image_name = autoupdater.url_to_image_name(update_url)
495 release_pattern = r'^.*-release/R[0-9]+-[0-9]+\.[0-9]+\.0$'
496 if not re.match(release_pattern, image_name):
497 return False
Dan Shi0f466e82013-02-22 15:44:58 -0800498 if not updater.check_version():
499 return False
500 if not force_update:
501 logging.info('Canceling stateful update because the new and '
502 'old versions are the same.')
503 return False
504 # Following folders should be rebuilt after stateful update.
505 # A test file is used to confirm each folder gets rebuilt after
506 # the stateful update.
507 folders_to_check = ['/var', '/home', '/mnt/stateful_partition']
508 test_file = '.test_file_to_be_deleted'
509 for folder in folders_to_check:
510 touch_path = os.path.join(folder, test_file)
511 self.run('touch %s' % touch_path)
512
Chris Sosae92399e2015-04-24 11:32:59 -0700513 updater.run_update(update_root=False)
Dan Shi0f466e82013-02-22 15:44:58 -0800514
515 # Reboot to complete stateful update.
Chris Sosab76e0ee2013-05-22 16:55:41 -0700516 self.reboot(timeout=self.REBOOT_TIMEOUT, wait=True)
xixuan908b0752016-12-08 18:54:23 -0800517
518 # After stateful update and a reboot, all of the test_files shouldn't
519 # exist any more. Otherwise the stateful update is failed.
520 return not any(
521 self.path_exists(os.path.join(folder, test_file))
522 for folder in folders_to_check)
Dan Shi0f466e82013-02-22 15:44:58 -0800523
524
J. Richard Barnette7275b612013-06-04 18:13:11 -0700525 def _post_update_processing(self, updater, expected_kernel=None):
Dan Shi0f466e82013-02-22 15:44:58 -0800526 """After the DUT is updated, confirm machine_install succeeded.
527
528 @param updater: ChromiumOSUpdater instance used to update the DUT.
J. Richard Barnette7275b612013-06-04 18:13:11 -0700529 @param expected_kernel: kernel expected to be active after reboot,
530 or `None` to skip rollback checking.
Dan Shi0f466e82013-02-22 15:44:58 -0800531
532 """
J. Richard Barnette7275b612013-06-04 18:13:11 -0700533 # Touch the lab machine file to leave a marker that
534 # distinguishes this image from other test images.
535 # Afterwards, we must re-run the autoreboot script because
536 # it depends on the _LAB_MACHINE_FILE.
J. Richard Barnette71cc1862015-12-02 10:32:38 -0800537 autoreboot_cmd = ('FILE="%s" ; [ -f "$FILE" ] || '
538 '( touch "$FILE" ; start autoreboot )')
539 self.run(autoreboot_cmd % self._LAB_MACHINE_FILE)
Chris Sosa65425082013-10-16 13:26:22 -0700540 updater.verify_boot_expectations(
541 expected_kernel, rollback_message=
Gilad Arnoldc26ae1f2015-10-22 16:09:41 -0700542 'Build %s failed to boot on %s; system rolled back to previous '
Chris Sosa65425082013-10-16 13:26:22 -0700543 'build' % (updater.update_version, self.hostname))
J. Richard Barnette7275b612013-06-04 18:13:11 -0700544 # Check that we've got the build we meant to install.
545 if not updater.check_version_to_confirm_install():
546 raise autoupdater.ChromiumOSError(
547 'Failed to update %s to build %s; found build '
548 '%s instead' % (self.hostname,
Chris Sosa65425082013-10-16 13:26:22 -0700549 updater.update_version,
Dan Shi0942b1d2015-03-31 11:07:00 -0700550 self.get_release_version()))
Dan Shi0f466e82013-02-22 15:44:58 -0800551
Chris Sosae92399e2015-04-24 11:32:59 -0700552 logging.debug('Cleaning up old autotest directories.')
553 try:
554 installed_autodir = autotest.Autotest.get_installed_autodir(self)
555 self.run('rm -rf ' + installed_autodir)
556 except autotest.AutodirNotFoundError:
557 logging.debug('No autotest installed directory found.')
558
Dan Shi0f466e82013-02-22 15:44:58 -0800559
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700560 def _stage_image_for_update(self, image_name=None):
Chris Sosae92399e2015-04-24 11:32:59 -0700561 """Stage a build on a devserver and return the update_url and devserver.
Scott Zawalskieadbf702013-03-14 09:23:06 -0400562
563 @param image_name: a name like lumpy-release/R27-3837.0.0
Chris Sosae92399e2015-04-24 11:32:59 -0700564 @returns a tuple with an update URL like:
Scott Zawalskieadbf702013-03-14 09:23:06 -0400565 http://172.22.50.205:8082/update/lumpy-release/R27-3837.0.0
Chris Sosae92399e2015-04-24 11:32:59 -0700566 and the devserver instance.
Scott Zawalskieadbf702013-03-14 09:23:06 -0400567 """
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700568 if not image_name:
Richard Barnette260cbd02016-10-06 12:23:28 -0700569 image_name = self._get_cros_repair_image_name()
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700570 logging.info('Staging build for AU: %s', image_name)
Dan Shi216389c2015-12-22 11:03:06 -0800571 devserver = dev_server.ImageServer.resolve(image_name, self.hostname)
Scott Zawalskieadbf702013-03-14 09:23:06 -0400572 devserver.trigger_download(image_name, synchronous=False)
Chris Sosae92399e2015-04-24 11:32:59 -0700573 return (tools.image_url_pattern() % (devserver.url(), image_name),
574 devserver)
Scott Zawalskieadbf702013-03-14 09:23:06 -0400575
576
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700577 def stage_image_for_servo(self, image_name=None):
578 """Stage a build on a devserver and return the update_url.
579
580 @param image_name: a name like lumpy-release/R27-3837.0.0
581 @returns an update URL like:
582 http://172.22.50.205:8082/update/lumpy-release/R27-3837.0.0
583 """
584 if not image_name:
Richard Barnette260cbd02016-10-06 12:23:28 -0700585 image_name = self._get_cros_repair_image_name()
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700586 logging.info('Staging build for servo install: %s', image_name)
Dan Shi216389c2015-12-22 11:03:06 -0800587 devserver = dev_server.ImageServer.resolve(image_name, self.hostname)
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700588 devserver.stage_artifacts(image_name, ['test_image'])
589 return devserver.get_test_image_url(image_name)
590
591
beepse539be02013-07-31 21:57:39 -0700592 def stage_factory_image_for_servo(self, image_name):
593 """Stage a build on a devserver and return the update_url.
594
595 @param image_name: a name like <baord>/4262.204.0
beeps12c0a3c2013-09-03 11:58:27 -0700596
beepse539be02013-07-31 21:57:39 -0700597 @return: An update URL, eg:
598 http://<devserver>/static/canary-channel/\
599 <board>/4262.204.0/factory_test/chromiumos_factory_image.bin
beeps12c0a3c2013-09-03 11:58:27 -0700600
601 @raises: ValueError if the factory artifact name is missing from
602 the config.
603
beepse539be02013-07-31 21:57:39 -0700604 """
605 if not image_name:
606 logging.error('Need an image_name to stage a factory image.')
607 return
608
Dan Shib8540a52015-07-16 14:18:23 -0700609 factory_artifact = CONFIG.get_config_value(
beeps12c0a3c2013-09-03 11:58:27 -0700610 'CROS', 'factory_artifact', type=str, default='')
611 if not factory_artifact:
612 raise ValueError('Cannot retrieve the factory artifact name from '
613 'autotest config, and hence cannot stage factory '
614 'artifacts.')
615
beepse539be02013-07-31 21:57:39 -0700616 logging.info('Staging build for servo install: %s', image_name)
Dan Shi216389c2015-12-22 11:03:06 -0800617 devserver = dev_server.ImageServer.resolve(image_name, self.hostname)
beepse539be02013-07-31 21:57:39 -0700618 devserver.stage_artifacts(
619 image_name,
beeps12c0a3c2013-09-03 11:58:27 -0700620 [factory_artifact],
621 archive_url=None)
beepse539be02013-07-31 21:57:39 -0700622
623 return tools.factory_image_url_pattern() % (devserver.url(), image_name)
624
625
xixuanec801e32016-08-25 10:20:22 -0700626 def machine_install_by_devserver(self, update_url=None, force_update=False,
xixuan5dc64ea2016-05-20 17:27:51 -0700627 local_devserver=False, repair=False,
628 force_full_update=False):
629 """Ultiize devserver to install the DUT.
630
631 (TODO) crbugs.com/627269: The logic in this function has some overlap
632 with those in function machine_install. The merge will be done later,
633 not in the same CL.
634
xixuanec801e32016-08-25 10:20:22 -0700635 @param update_url: The update_url or build for the host to update.
xixuan5dc64ea2016-05-20 17:27:51 -0700636 @param force_update: Force an update even if the version installed
637 is the same. Default:False
638 @param local_devserver: Used by test_that to allow people to
639 use their local devserver. Default: False
640 @param repair: Forces update to repair image. Implies force_update.
641 @param force_full_update: If True, do not attempt to run stateful
642 update, force a full reimage. If False, try stateful update
643 first when the dut is already installed with the same version.
644 @raises autoupdater.ChromiumOSError
645
646 @returns A tuple of (image_name, host_attributes).
647 image_name is the name of image installed, e.g.,
648 veyron_jerry-release/R50-7871.0.0
649 host_attributes is a dictionary of (attribute, value), which
650 can be saved to afe_host_attributes table in database. This
651 method returns a dictionary with a single entry of
652 `job_repo_url`: repo_url, where repo_url is a devserver url to
653 autotest packages.
654 """
xixuan5dc64ea2016-05-20 17:27:51 -0700655 if repair:
Richard Barnette260cbd02016-10-06 12:23:28 -0700656 update_url = self._get_cros_repair_image_name()
xixuan5dc64ea2016-05-20 17:27:51 -0700657 force_update = True
658
xixuanec801e32016-08-25 10:20:22 -0700659 if not update_url and not self._parser.options.image:
xixuan5dc64ea2016-05-20 17:27:51 -0700660 raise error.AutoservError(
661 'There is no update URL, nor a method to get one.')
662
xixuanec801e32016-08-25 10:20:22 -0700663 if not update_url and self._parser.options.image:
664 update_url = self._parser.options.image
xixuan5dc64ea2016-05-20 17:27:51 -0700665
xixuan5dc64ea2016-05-20 17:27:51 -0700666
667 # Get build from parameter or AFE.
668 # If the build is not a URL, let devserver to stage it first.
669 # Otherwise, choose a devserver to trigger auto-update.
xixuanec801e32016-08-25 10:20:22 -0700670 build = None
Aviv Keshetf9375482016-10-26 02:09:03 -0700671 devserver = None
672 logging.debug('Resolving a devserver for auto-update')
Aviv Keshet3a76f762016-10-27 17:41:51 -0700673 previously_resolved = False
xixuanec801e32016-08-25 10:20:22 -0700674 if update_url.startswith('http://'):
675 build = autoupdater.url_to_image_name(update_url)
Aviv Keshet3a76f762016-10-27 17:41:51 -0700676 previously_resolved = True
677 else:
678 build = update_url
679 devserver = dev_server.resolve(build, self.hostname)
Allen Lia5cfb972016-12-27 17:17:22 -0800680 server_name = devserver.hostname
Aviv Keshet3a76f762016-10-27 17:41:51 -0700681
682 try:
683 board, build_type, milestone, _ = server_utils.ParseBuildName(build)
684 except server_utils.ParseBuildNameException:
685 logging.warning('Unable to parse build name %s. Something is '
686 'likely broken, but will continue anyway.',
687 build)
688 board, build_type, milestone = ('', '', '')
689
690 monarch_fields = {
691 'dev_server': server_name,
692 'board': board,
693 'build_type': build_type,
694 'milestone': milestone
695 }
696
697 if previously_resolved:
xixuan2e62d162016-09-29 19:19:19 -0700698 # Make sure devserver for Auto-Update has staged the build.
xixuan2e62d162016-09-29 19:19:19 -0700699 if server_name not in update_url:
Aviv Keshet3a76f762016-10-27 17:41:51 -0700700 logging.debug('Resolved to devserver that does not match '
701 'update_url. The previously resolved devserver '
702 'must be unhealthy. Switching to use devserver %s,'
703 ' and re-staging.',
xixuan2e62d162016-09-29 19:19:19 -0700704 server_name)
Aviv Keshetf9375482016-10-26 02:09:03 -0700705 logging.info('Staging build for AU: %s', update_url)
xixuan2e62d162016-09-29 19:19:19 -0700706 devserver.trigger_download(build, synchronous=False)
Dan Shi5e2efb72017-02-07 11:40:23 -0800707 c = metrics.Counter(
708 'chromeos/autotest/provision/failover_download')
709 c.increment(fields=monarch_fields)
xixuan5dc64ea2016-05-20 17:27:51 -0700710 else:
Aviv Keshetf9375482016-10-26 02:09:03 -0700711 logging.info('Staging build for AU: %s', update_url)
xixuan5dc64ea2016-05-20 17:27:51 -0700712 devserver.trigger_download(build, synchronous=False)
Dan Shi5e2efb72017-02-07 11:40:23 -0800713 c = metrics.Counter('chromeos/autotest/provision/trigger_download')
714 c.increment(fields=monarch_fields)
xixuan5dc64ea2016-05-20 17:27:51 -0700715
716 # Report provision stats.
Allen Li48a13fe2016-11-22 14:10:40 -0800717 (metrics.Counter('chromeos/autotest/provision/install_with_devserver')
718 .increment(fields=monarch_fields))
xixuanec801e32016-08-25 10:20:22 -0700719 logging.debug('Resolved devserver for auto-update: %s', devserver.url())
xixuan5dc64ea2016-05-20 17:27:51 -0700720
Aviv Keshetf9375482016-10-26 02:09:03 -0700721 # and other metrics from this function.
Dan Shi5e2efb72017-02-07 11:40:23 -0800722 metrics.Counter('chromeos/autotest/provision/resolve'
723 ).increment(fields=monarch_fields)
Aviv Keshetf9375482016-10-26 02:09:03 -0700724
xixuan5dc64ea2016-05-20 17:27:51 -0700725 devserver.auto_update(self.hostname, build,
xixuan95c6cce2016-10-05 17:49:41 -0700726 log_dir=self.job.resultdir,
xixuan5dc64ea2016-05-20 17:27:51 -0700727 force_update=force_update,
728 full_update=force_full_update)
729
730 # The reason to resolve a new devserver in function machine_install
731 # is mostly because that the update_url there may has a strange format,
732 # and it's hard to parse the devserver url from it.
733 # Since we already resolve a devserver to trigger auto-update, the same
734 # devserver is used to form JOB_REPO_URL here. Verified in local test.
735 repo_url = tools.get_package_url(devserver.url(), build)
736 return build, {ds_constants.JOB_REPO_URL: repo_url}
737
738
Chris Sosaa3ac2152012-05-23 22:23:13 -0700739 def machine_install(self, update_url=None, force_update=False,
Richard Barnette0b023a72015-04-24 16:07:30 +0000740 local_devserver=False, repair=False,
741 force_full_update=False):
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500742 """Install the DUT.
743
Dan Shi0f466e82013-02-22 15:44:58 -0800744 Use stateful update if the DUT is already running the same build.
745 Stateful update does not update kernel and tends to run much faster
746 than a full reimage. If the DUT is running a different build, or it
747 failed to do a stateful update, full update, including kernel update,
748 will be applied to the DUT.
749
Simran Basi5ace6f22016-01-06 17:30:44 -0800750 Once a host enters machine_install its host attribute job_repo_url
751 (used for package install) will be removed and then updated.
Scott Zawalskieadbf702013-03-14 09:23:06 -0400752
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500753 @param update_url: The url to use for the update
754 pattern: http://$devserver:###/update/$build
755 If update_url is None and repair is True we will install the
Dan Shi6964fa52014-12-18 11:04:27 -0800756 stable image listed in afe_stable_versions table. If the table
757 is not setup, global_config value under CROS.stable_cros_version
758 will be used instead.
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500759 @param force_update: Force an update even if the version installed
760 is the same. Default:False
Christopher Wiley6a4ff932015-05-15 14:00:47 -0700761 @param local_devserver: Used by test_that to allow people to
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500762 use their local devserver. Default: False
Chris Sosae92399e2015-04-24 11:32:59 -0700763 @param repair: Forces update to repair image. Implies force_update.
Fang Deng3d3b9272014-12-22 12:20:28 -0800764 @param force_full_update: If True, do not attempt to run stateful
765 update, force a full reimage. If False, try stateful update
766 first when the dut is already installed with the same version.
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500767 @raises autoupdater.ChromiumOSError
768
Dan Shibe3636a2016-02-14 22:48:01 -0800769 @returns A tuple of (image_name, host_attributes).
770 image_name is the name of image installed, e.g.,
771 veyron_jerry-release/R50-7871.0.0
772 host_attributes is a dictionary of (attribute, value), which
773 can be saved to afe_host_attributes table in database. This
774 method returns a dictionary with a single entry of
775 `job_repo_url`: repo_url, where repo_url is a devserver url to
776 autotest packages.
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500777 """
Chris Sosae92399e2015-04-24 11:32:59 -0700778 devserver = None
Richard Barnette0b023a72015-04-24 16:07:30 +0000779 if repair:
Chris Sosae92399e2015-04-24 11:32:59 -0700780 update_url, devserver = self._stage_image_for_update()
Richard Barnette0b023a72015-04-24 16:07:30 +0000781 force_update = True
Dan Shi0f466e82013-02-22 15:44:58 -0800782
Chris Sosae92399e2015-04-24 11:32:59 -0700783 if not update_url and not self._parser.options.image:
784 raise error.AutoservError(
Dan Shid07ee2e2015-09-24 14:49:25 -0700785 'There is no update URL, nor a method to get one.')
Chris Sosae92399e2015-04-24 11:32:59 -0700786
787 if not update_url and self._parser.options.image:
788 # This is the base case where we have no given update URL i.e.
789 # dynamic suites logic etc. This is the most flexible case where we
790 # can serve an update from any of our fleet of devservers.
791 requested_build = self._parser.options.image
792 if not requested_build.startswith('http://'):
793 logging.debug('Update will be staged for this installation')
794 update_url, devserver = self._stage_image_for_update(
Dan Shid07ee2e2015-09-24 14:49:25 -0700795 requested_build)
Chris Sosae92399e2015-04-24 11:32:59 -0700796 else:
797 update_url = requested_build
798
799 logging.debug('Update URL is %s', update_url)
800
Dan Shif48f8132016-02-18 10:34:30 -0800801 # Report provision stats.
Allen Lia5cfb972016-12-27 17:17:22 -0800802 server_name = dev_server.get_hostname(update_url)
Allen Li48a13fe2016-11-22 14:10:40 -0800803 (metrics.Counter('chromeos/autotest/provision/install')
804 .increment(fields={'devserver': server_name}))
Dan Shif48f8132016-02-18 10:34:30 -0800805
Dan Shid07ee2e2015-09-24 14:49:25 -0700806 # Create a file to indicate if provision fails. The file will be removed
807 # by stateful update or full install.
J. Richard Barnette91137f02016-03-10 16:52:26 -0800808 self.run('touch %s' % self.PROVISION_FAILED)
Dan Shid07ee2e2015-09-24 14:49:25 -0700809
Chris Sosae92399e2015-04-24 11:32:59 -0700810 update_complete = False
811 updater = autoupdater.ChromiumOSUpdater(
812 update_url, host=self, local_devserver=local_devserver)
Fang Deng3d3b9272014-12-22 12:20:28 -0800813 if not force_full_update:
814 try:
Chris Sosae92399e2015-04-24 11:32:59 -0700815 # If the DUT is already running the same build, try stateful
816 # update first as it's much quicker than a full re-image.
817 update_complete = self._try_stateful_update(
Dan Shid07ee2e2015-09-24 14:49:25 -0700818 update_url, force_update, updater)
Fang Deng3d3b9272014-12-22 12:20:28 -0800819 except Exception as e:
820 logging.exception(e)
J. Richard Barnette45e93de2012-04-11 17:24:15 -0700821
Dan Shi0f466e82013-02-22 15:44:58 -0800822 inactive_kernel = None
Chris Sosae92399e2015-04-24 11:32:59 -0700823 if update_complete or (not force_update and updater.check_version()):
824 logging.info('Install complete without full update')
825 else:
826 logging.info('DUT requires full update.')
827 self.reboot(timeout=self.REBOOT_TIMEOUT, wait=True)
Dan Shi10b98482016-02-02 14:38:50 -0800828 # Stop service ap-update-manager to prevent rebooting during
829 # autoupdate. The service is used in jetstream boards, but not other
830 # CrOS devices.
831 self.run('sudo stop ap-update-manager', ignore_status=True)
832
Chris Sosae92399e2015-04-24 11:32:59 -0700833 num_of_attempts = provision.FLAKY_DEVSERVER_ATTEMPTS
Chris Sosab7612bc2013-03-21 10:32:37 -0700834
Chris Sosae92399e2015-04-24 11:32:59 -0700835 while num_of_attempts > 0:
836 num_of_attempts -= 1
837 try:
838 updater.run_update()
839 except Exception:
840 logging.warn('Autoupdate did not complete.')
841 # Do additional check for the devserver health. Ideally,
842 # the autoupdater.py could raise an exception when it
843 # detected network flake but that would require
844 # instrumenting the update engine and parsing it log.
845 if (num_of_attempts <= 0 or
846 devserver is None or
xixuan9e2c98d2016-02-26 19:04:53 -0800847 dev_server.ImageServer.devserver_healthy(
Chris Sosae92399e2015-04-24 11:32:59 -0700848 devserver.url())):
Dan Shid07ee2e2015-09-24 14:49:25 -0700849 raise
J. Richard Barnette45e93de2012-04-11 17:24:15 -0700850
Chris Sosae92399e2015-04-24 11:32:59 -0700851 logging.warn('Devserver looks unhealthy. Trying another')
852 update_url, devserver = self._stage_image_for_update(
853 requested_build)
854 logging.debug('New Update URL is %s', update_url)
855 updater = autoupdater.ChromiumOSUpdater(
856 update_url, host=self,
857 local_devserver=local_devserver)
858 else:
859 break
J. Richard Barnette45e93de2012-04-11 17:24:15 -0700860
Chris Sosae92399e2015-04-24 11:32:59 -0700861 # Give it some time in case of IO issues.
862 time.sleep(10)
Dan Shi5699ac22014-12-19 10:55:49 -0800863
Chris Sosae92399e2015-04-24 11:32:59 -0700864 # Figure out active and inactive kernel.
865 active_kernel, inactive_kernel = updater.get_kernel_state()
Simran Basi13fa1ba2013-03-04 10:56:47 -0800866
Chris Sosae92399e2015-04-24 11:32:59 -0700867 # Ensure inactive kernel has higher priority than active.
868 if (updater.get_kernel_priority(inactive_kernel)
869 < updater.get_kernel_priority(active_kernel)):
870 raise autoupdater.ChromiumOSError(
871 'Update failed. The priority of the inactive kernel'
872 ' partition is less than that of the active kernel'
873 ' partition.')
874
875 # Updater has returned successfully; reboot the host.
Richard Barnettefc3d7432016-07-07 10:21:31 -0700876 #
877 # Regarding the 'crossystem' command: In some cases, the
878 # TPM gets into a state such that it fails verification.
879 # We don't know why. However, this call papers over the
880 # problem by clearing the TPM during the reboot.
881 #
882 # We ignore failures from 'crossystem'. Although failure
883 # here is unexpected, and could signal a bug, the point
884 # of the exercise is to paper over problems; allowing
885 # this to fail would defeat the purpose.
886 self.run('crossystem clear_tpm_owner_request=1',
887 ignore_status=True)
Chris Sosae92399e2015-04-24 11:32:59 -0700888 self.reboot(timeout=self.REBOOT_TIMEOUT, wait=True)
889
890 self._post_update_processing(updater, inactive_kernel)
Simran Basi5ace6f22016-01-06 17:30:44 -0800891 image_name = autoupdater.url_to_image_name(update_url)
Dan Shibe3636a2016-02-14 22:48:01 -0800892 # update_url is different from devserver url needed to stage autotest
893 # packages, therefore, resolve a new devserver url here.
894 devserver_url = dev_server.ImageServer.resolve(image_name,
895 self.hostname).url()
896 repo_url = tools.get_package_url(devserver_url, image_name)
897 return image_name, {ds_constants.JOB_REPO_URL: repo_url}
J. Richard Barnette45e93de2012-04-11 17:24:15 -0700898
899
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +0800900 def _clear_fw_version_labels(self, rw_only):
901 """Clear firmware version labels from the machine.
902
903 @param rw_only: True to only clear fwrw_version; otherewise, clear
904 both fwro_version and fwrw_version.
905 """
Dan Shi9cb0eec2014-06-03 09:04:50 -0700906 labels = self._AFE.get_labels(
Dan Shi0723bf52015-06-24 10:52:38 -0700907 name__startswith=provision.FW_RW_VERSION_PREFIX,
Dan Shi9cb0eec2014-06-03 09:04:50 -0700908 host__hostname=self.hostname)
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +0800909 if not rw_only:
910 labels = labels + self._AFE.get_labels(
911 name__startswith=provision.FW_RO_VERSION_PREFIX,
912 host__hostname=self.hostname)
Dan Shi9cb0eec2014-06-03 09:04:50 -0700913 for label in labels:
914 label.remove_hosts(hosts=[self.hostname])
915
916
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +0800917 def _add_fw_version_label(self, build, rw_only):
Dan Shi9cb0eec2014-06-03 09:04:50 -0700918 """Add firmware version label to the machine.
919
920 @param build: Build of firmware.
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +0800921 @param rw_only: True to only add fwrw_version; otherwise, add both
922 fwro_version and fwrw_version.
Dan Shi9cb0eec2014-06-03 09:04:50 -0700923
924 """
Prathmesh Prabhu2c7471d2016-11-15 20:19:57 +0000925 fw_label = provision.fwrw_version_to_label(build)
MK Ryu73be9862015-07-06 12:25:00 -0700926 self._AFE.run('label_add_hosts', id=fw_label, hosts=[self.hostname])
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +0800927 if not rw_only:
Prathmesh Prabhu2c7471d2016-11-15 20:19:57 +0000928 fw_label = provision.fwro_version_to_label(build)
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +0800929 self._AFE.run('label_add_hosts', id=fw_label, hosts=[self.hostname])
Dan Shi9cb0eec2014-06-03 09:04:50 -0700930
931
Tom Wai-Hong Tam4ac78982016-01-08 02:34:37 +0800932 def firmware_install(self, build=None, rw_only=False):
Dan Shi9cb0eec2014-06-03 09:04:50 -0700933 """Install firmware to the DUT.
934
935 Use stateful update if the DUT is already running the same build.
936 Stateful update does not update kernel and tends to run much faster
937 than a full reimage. If the DUT is running a different build, or it
938 failed to do a stateful update, full update, including kernel update,
939 will be applied to the DUT.
940
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +0800941 Once a host enters firmware_install its fw[ro|rw]_version label will
942 be removed. After the firmware is updated successfully, a new
943 fw[ro|rw]_version label will be added to the host.
Dan Shi9cb0eec2014-06-03 09:04:50 -0700944
945 @param build: The build version to which we want to provision the
946 firmware of the machine,
947 e.g. 'link-firmware/R22-2695.1.144'.
Tom Wai-Hong Tam4ac78982016-01-08 02:34:37 +0800948 @param rw_only: True to only install firmware to its RW portions. Keep
949 the RO portions unchanged.
Dan Shi9cb0eec2014-06-03 09:04:50 -0700950
951 TODO(dshi): After bug 381718 is fixed, update here with corresponding
952 exceptions that could be raised.
953
954 """
955 if not self.servo:
956 raise error.TestError('Host %s does not have servo.' %
957 self.hostname)
958
Wai-Hong Tam7b424e92016-11-10 15:35:24 -0800959 board = self.get_board().replace(ds_constants.BOARD_PREFIX, '')
Dan Shi9cb0eec2014-06-03 09:04:50 -0700960
Chris Sosae92399e2015-04-24 11:32:59 -0700961 # If build is not set, try to install firmware from stable CrOS.
Dan Shi9cb0eec2014-06-03 09:04:50 -0700962 if not build:
Richard Barnette260cbd02016-10-06 12:23:28 -0700963 build = afe_utils.get_stable_faft_version(board)
Dan Shi3d7a0e12015-10-12 11:55:45 -0700964 if not build:
965 raise error.TestError(
966 'Failed to find stable firmware build for %s.',
967 self.hostname)
968 logging.info('Will install firmware from build %s.', build)
Dan Shi9cb0eec2014-06-03 09:04:50 -0700969
970 config = FAFTConfig(board)
971 if config.use_u_boot:
972 ap_image = 'image-%s.bin' % board
973 else: # Depthcharge platform
974 ap_image = 'image.bin'
975 ec_image = 'ec.bin'
Dan Shi216389c2015-12-22 11:03:06 -0800976 ds = dev_server.ImageServer.resolve(build, self.hostname)
Dan Shi9cb0eec2014-06-03 09:04:50 -0700977 ds.stage_artifacts(build, ['firmware'])
978
979 tmpd = autotemp.tempdir(unique_id='fwimage')
980 try:
981 fwurl = self._FW_IMAGE_URL_PATTERN % (ds.url(), build)
982 local_tarball = os.path.join(tmpd.name, os.path.basename(fwurl))
xixuan4e116822016-11-17 15:32:10 -0800983 ds.download_file(fwurl, local_tarball)
Dan Shi9cb0eec2014-06-03 09:04:50 -0700984
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +0800985 self._clear_fw_version_labels(rw_only)
Tom Wai-Hong Tam552fee12016-05-11 06:27:58 +0800986 if config.chrome_ec:
Tom Wai-Hong Tamf9db57a2016-03-17 05:32:22 +0800987 logging.info('Will re-program EC %snow', 'RW ' if rw_only else '')
988 server_utils.system('tar xf %s -C %s %s' %
989 (local_tarball, tmpd.name, ec_image),
990 timeout=60)
991 self.servo.program_ec(os.path.join(tmpd.name, ec_image), rw_only)
992 else:
993 logging.info('Not a Chrome EC, ignore re-programing it')
Tom Wai-Hong Tam4ac78982016-01-08 02:34:37 +0800994 logging.info('Will re-program BIOS %snow', 'RW ' if rw_only else '')
Tom Wai-Hong Tamf9db57a2016-03-17 05:32:22 +0800995 server_utils.system('tar xf %s -C %s %s' %
996 (local_tarball, tmpd.name, ap_image),
997 timeout=60)
Tom Wai-Hong Tam4ac78982016-01-08 02:34:37 +0800998 self.servo.program_bios(os.path.join(tmpd.name, ap_image), rw_only)
Dan Shi9cb0eec2014-06-03 09:04:50 -0700999 self.servo.get_power_state_controller().reset()
1000 time.sleep(self.servo.BOOT_DELAY)
Wai-Hong Tamb5f66ce2016-11-10 15:45:30 -08001001 if utils.host_is_in_lab_zone(self.hostname):
1002 self._add_fw_version_label(build, rw_only)
Dan Shi9cb0eec2014-06-03 09:04:50 -07001003 finally:
1004 tmpd.clean()
1005
1006
Dan Shi10e992b2013-08-30 11:02:59 -07001007 def show_update_engine_log(self):
1008 """Output update engine log."""
MK Ryu35d661e2014-09-25 17:44:10 -07001009 logging.debug('Dumping %s', client_constants.UPDATE_ENGINE_LOG)
1010 self.run('cat %s' % client_constants.UPDATE_ENGINE_LOG)
Dan Shi10e992b2013-08-30 11:02:59 -07001011
1012
Richard Barnette82c35912012-11-20 10:09:10 -08001013 def _get_board_from_afe(self):
Kevin Cheng84a71ba2016-07-14 11:03:57 -07001014 """Retrieve this host's board from its labels stored locally.
Richard Barnette82c35912012-11-20 10:09:10 -08001015
1016 Looks for a host label of the form "board:<board>", and
1017 returns the "<board>" part of the label. `None` is returned
1018 if there is not a single, unique label matching the pattern.
1019
1020 @returns board from label, or `None`.
1021 """
Richard Barnette260cbd02016-10-06 12:23:28 -07001022 board = afe_utils.get_board(self)
1023 if board is None:
1024 raise error.AutoservError('DUT cannot be repaired, '
1025 'there is no board attribute.')
1026 return board
Simran Basi833814b2013-01-29 13:13:43 -08001027
1028
beepsf079cfb2013-09-18 17:49:51 -07001029 def servo_install(self, image_url=None, usb_boot_timeout=USB_BOOT_TIMEOUT,
1030 install_timeout=INSTALL_TIMEOUT):
Scott Zawalski62bacae2013-03-05 10:40:32 -05001031 """
1032 Re-install the OS on the DUT by:
1033 1) installing a test image on a USB storage device attached to the Servo
1034 board,
Richard Barnette03a0c132012-11-05 12:40:35 -08001035 2) booting that image in recovery mode, and then
J. Richard Barnette31b2e312013-04-04 16:05:22 -07001036 3) installing the image with chromeos-install.
1037
Scott Zawalski62bacae2013-03-05 10:40:32 -05001038 @param image_url: If specified use as the url to install on the DUT.
1039 otherwise boot the currently staged image on the USB stick.
beepsf079cfb2013-09-18 17:49:51 -07001040 @param usb_boot_timeout: The usb_boot_timeout to use during reimage.
1041 Factory images need a longer usb_boot_timeout than regular
1042 cros images.
1043 @param install_timeout: The timeout to use when installing the chromeos
1044 image. Factory images need a longer install_timeout.
Richard Barnette03a0c132012-11-05 12:40:35 -08001045
Scott Zawalski62bacae2013-03-05 10:40:32 -05001046 @raises AutoservError if the image fails to boot.
beepsf079cfb2013-09-18 17:49:51 -07001047
J. Richard Barnette0199cc82014-12-05 17:08:40 -08001048 """
beepsf079cfb2013-09-18 17:49:51 -07001049 logging.info('Downloading image to USB, then booting from it. Usb boot '
1050 'timeout = %s', usb_boot_timeout)
Allen Li48a13fe2016-11-22 14:10:40 -08001051 with metrics.SecondsTimer(
1052 'chromeos/autotest/provision/servo_install/boot_duration'):
1053 self.servo.install_recovery_image(image_url)
1054 if not self.wait_up(timeout=usb_boot_timeout):
1055 raise hosts.AutoservRepairError(
1056 'DUT failed to boot from USB after %d seconds' %
1057 usb_boot_timeout)
Scott Zawalski62bacae2013-03-05 10:40:32 -05001058
Tom Wai-Hong Tamf6b4f812015-08-08 04:14:59 +08001059 # The new chromeos-tpm-recovery has been merged since R44-7073.0.0.
1060 # In old CrOS images, this command fails. Skip the error.
Tom Wai-Hong Tam27af7332015-07-25 06:09:39 +08001061 logging.info('Resetting the TPM status')
Tom Wai-Hong Tamf6b4f812015-08-08 04:14:59 +08001062 try:
1063 self.run('chromeos-tpm-recovery')
1064 except error.AutoservRunError:
1065 logging.warn('chromeos-tpm-recovery is too old.')
1066
Tom Wai-Hong Tam27af7332015-07-25 06:09:39 +08001067
Allen Li48a13fe2016-11-22 14:10:40 -08001068 with metrics.SecondsTimer(
1069 'chromeos/autotest/provision/servo_install/install_duration'):
1070 logging.info('Installing image through chromeos-install.')
1071 self.run('chromeos-install --yes', timeout=install_timeout)
1072 self.halt()
beepsf079cfb2013-09-18 17:49:51 -07001073
1074 logging.info('Power cycling DUT through servo.')
J. Richard Barnette0199cc82014-12-05 17:08:40 -08001075 self.servo.get_power_state_controller().power_off()
Fang Dengafb88142013-05-30 17:44:31 -07001076 self.servo.switch_usbkey('off')
J. Richard Barnette0199cc82014-12-05 17:08:40 -08001077 # N.B. The Servo API requires that we use power_on() here
1078 # for two reasons:
1079 # 1) After turning on a DUT in recovery mode, you must turn
1080 # it off and then on with power_on() once more to
1081 # disable recovery mode (this is a Parrot specific
1082 # requirement).
1083 # 2) After power_off(), the only way to turn on is with
1084 # power_on() (this is a Storm specific requirement).
J. Richard Barnettefbcc7122013-07-24 18:24:59 -07001085 self.servo.get_power_state_controller().power_on()
beepsf079cfb2013-09-18 17:49:51 -07001086
1087 logging.info('Waiting for DUT to come back up.')
Richard Barnette03a0c132012-11-05 12:40:35 -08001088 if not self.wait_up(timeout=self.BOOT_TIMEOUT):
1089 raise error.AutoservError('DUT failed to reboot installed '
1090 'test image after %d seconds' %
Scott Zawalski62bacae2013-03-05 10:40:32 -05001091 self.BOOT_TIMEOUT)
1092
1093
Richard Barnette9a26ad62016-06-10 12:03:08 -07001094 def repair_servo(self):
Dan Shi90466352015-09-22 15:01:05 -07001095 """
Richard Barnette9a26ad62016-06-10 12:03:08 -07001096 Confirm that servo is initialized and verified.
Dan Shi90466352015-09-22 15:01:05 -07001097
Richard Barnette9a26ad62016-06-10 12:03:08 -07001098 If the servo object is missing, attempt to repair the servo
1099 host. Repair failures are passed back to the caller.
1100
1101 @raise AutoservError: If there is no servo host for this CrOS
1102 host.
1103 """
1104 if self.servo:
1105 return
1106 if not self._servo_host:
1107 raise error.AutoservError('No servo host for %s.' %
1108 self.hostname)
1109 self._servo_host.repair()
1110 self.servo = self._servo_host.get_servo()
Dan Shi90466352015-09-22 15:01:05 -07001111
1112
J. Richard Barnettec2d99cf2015-11-18 12:46:15 -08001113 def repair(self):
1114 """Attempt to get the DUT to pass `self.verify()`.
Richard Barnette82c35912012-11-20 10:09:10 -08001115
1116 This overrides the base class function for repair; it does
J. Richard Barnette91137f02016-03-10 16:52:26 -08001117 not call back to the parent class, but instead relies on
1118 `self._repair_strategy` to coordinate the verification and
1119 repair steps needed to get the DUT working.
Richard Barnette82c35912012-11-20 10:09:10 -08001120 """
J. Richard Barnette91137f02016-03-10 16:52:26 -08001121 self._repair_strategy.repair(self)
Scott Zawalski89c44dd2013-02-26 09:28:02 -05001122
Richard Barnette82c35912012-11-20 10:09:10 -08001123
J. Richard Barnette1d78b012012-05-15 13:56:30 -07001124 def close(self):
Fang Deng0ca40e22013-08-27 17:47:44 -07001125 super(CrosHost, self).close()
xixuand6011f12016-12-08 15:01:58 -08001126 if self._chameleon_host:
1127 self._chameleon_host.close()
1128
1129 if self._servo_host:
1130 self._servo_host.close()
J. Richard Barnette1d78b012012-05-15 13:56:30 -07001131
1132
Dan Shi49ca0932014-11-14 11:22:27 -08001133 def get_power_supply_info(self):
1134 """Get the output of power_supply_info.
1135
1136 power_supply_info outputs the info of each power supply, e.g.,
1137 Device: Line Power
1138 online: no
1139 type: Mains
1140 voltage (V): 0
1141 current (A): 0
1142 Device: Battery
1143 state: Discharging
1144 percentage: 95.9276
1145 technology: Li-ion
1146
1147 Above output shows two devices, Line Power and Battery, with details of
1148 each device listed. This function parses the output into a dictionary,
1149 with key being the device name, and value being a dictionary of details
1150 of the device info.
1151
1152 @return: The dictionary of power_supply_info, e.g.,
1153 {'Line Power': {'online': 'yes', 'type': 'main'},
1154 'Battery': {'vendor': 'xyz', 'percentage': '100'}}
Dan Shie9b765d2014-12-29 16:59:49 -08001155 @raise error.AutoservRunError if power_supply_info tool is not found in
1156 the DUT. Caller should handle this error to avoid false failure
1157 on verification.
Dan Shi49ca0932014-11-14 11:22:27 -08001158 """
1159 result = self.run('power_supply_info').stdout.strip()
1160 info = {}
1161 device_name = None
1162 device_info = {}
1163 for line in result.split('\n'):
1164 pair = [v.strip() for v in line.split(':')]
1165 if len(pair) != 2:
1166 continue
1167 if pair[0] == 'Device':
1168 if device_name:
1169 info[device_name] = device_info
1170 device_name = pair[1]
1171 device_info = {}
1172 else:
1173 device_info[pair[0]] = pair[1]
1174 if device_name and not device_name in info:
1175 info[device_name] = device_info
1176 return info
1177
1178
1179 def get_battery_percentage(self):
1180 """Get the battery percentage.
1181
1182 @return: The percentage of battery level, value range from 0-100. Return
1183 None if the battery info cannot be retrieved.
1184 """
1185 try:
1186 info = self.get_power_supply_info()
1187 logging.info(info)
1188 return float(info['Battery']['percentage'])
Dan Shie9b765d2014-12-29 16:59:49 -08001189 except (KeyError, ValueError, error.AutoservRunError):
Dan Shi49ca0932014-11-14 11:22:27 -08001190 return None
1191
1192
1193 def is_ac_connected(self):
1194 """Check if the dut has power adapter connected and charging.
1195
1196 @return: True if power adapter is connected and charging.
1197 """
1198 try:
1199 info = self.get_power_supply_info()
1200 return info['Line Power']['online'] == 'yes'
Dan Shie9b765d2014-12-29 16:59:49 -08001201 except (KeyError, error.AutoservRunError):
1202 return None
Dan Shi49ca0932014-11-14 11:22:27 -08001203
1204
Simran Basi5e6339a2013-03-21 11:34:32 -07001205 def _cleanup_poweron(self):
1206 """Special cleanup method to make sure hosts always get power back."""
1207 afe = frontend_wrappers.RetryingAFE(timeout_min=5, delay_sec=10)
1208 hosts = afe.get_hosts(hostname=self.hostname)
1209 if not hosts or not (self._RPM_OUTLET_CHANGED in
1210 hosts[0].attributes):
1211 return
1212 logging.debug('This host has recently interacted with the RPM'
1213 ' Infrastructure. Ensuring power is on.')
1214 try:
1215 self.power_on()
Dan Shi7dca56e2014-11-11 17:07:56 -08001216 afe.set_host_attribute(self._RPM_OUTLET_CHANGED, None,
1217 hostname=self.hostname)
Simran Basi5e6339a2013-03-21 11:34:32 -07001218 except rpm_client.RemotePowerException:
Simran Basi5e6339a2013-03-21 11:34:32 -07001219 logging.error('Failed to turn Power On for this host after '
1220 'cleanup through the RPM Infrastructure.')
Gabe Blackb72f4fb2015-01-20 16:47:13 -08001221 autotest_es.post(
Dan Shi7dca56e2014-11-11 17:07:56 -08001222 type_str='RPM_poweron_failure',
1223 metadata={'hostname': self.hostname})
Dan Shi49ca0932014-11-14 11:22:27 -08001224
1225 battery_percentage = self.get_battery_percentage()
Dan Shif01ebe22014-12-05 13:10:57 -08001226 if battery_percentage and battery_percentage < 50:
Dan Shi49ca0932014-11-14 11:22:27 -08001227 raise
1228 elif self.is_ac_connected():
1229 logging.info('The device has power adapter connected and '
1230 'charging. No need to try to turn RPM on '
1231 'again.')
1232 afe.set_host_attribute(self._RPM_OUTLET_CHANGED, None,
1233 hostname=self.hostname)
1234 logging.info('Battery level is now at %s%%. The device may '
1235 'still have enough power to run test, so no '
1236 'exception will be raised.', battery_percentage)
1237
Simran Basi5e6339a2013-03-21 11:34:32 -07001238
beepsc87ff602013-07-31 21:53:00 -07001239 def _is_factory_image(self):
1240 """Checks if the image on the DUT is a factory image.
1241
1242 @return: True if the image on the DUT is a factory image.
1243 False otherwise.
1244 """
1245 result = self.run('[ -f /root/.factory_test ]', ignore_status=True)
1246 return result.exit_status == 0
1247
1248
1249 def _restart_ui(self):
J. Richard Barnette84890bd2014-02-21 11:05:47 -08001250 """Restart the Chrome UI.
beepsc87ff602013-07-31 21:53:00 -07001251
1252 @raises: FactoryImageCheckerException for factory images, since
1253 we cannot attempt to restart ui on them.
1254 error.AutoservRunError for any other type of error that
1255 occurs while restarting ui.
1256 """
1257 if self._is_factory_image():
Dan Shi549fb822015-03-24 18:01:11 -07001258 raise FactoryImageCheckerException('Cannot restart ui on factory '
1259 'images')
beepsc87ff602013-07-31 21:53:00 -07001260
J. Richard Barnette84890bd2014-02-21 11:05:47 -08001261 # TODO(jrbarnette): The command to stop/start the ui job
1262 # should live inside cros_ui, too. However that would seem
1263 # to imply interface changes to the existing start()/restart()
1264 # functions, which is a bridge too far (for now).
J. Richard Barnette6069aa12015-06-08 09:10:24 -07001265 prompt = cros_ui.get_chrome_session_ident(self)
J. Richard Barnette84890bd2014-02-21 11:05:47 -08001266 self.run('stop ui; start ui')
1267 cros_ui.wait_for_chrome_ready(prompt, self)
beepsc87ff602013-07-31 21:53:00 -07001268
1269
Dan Shi549fb822015-03-24 18:01:11 -07001270 def get_release_version(self):
1271 """Get the value of attribute CHROMEOS_RELEASE_VERSION from lsb-release.
1272
1273 @returns The version string in lsb-release, under attribute
1274 CHROMEOS_RELEASE_VERSION.
1275 """
1276 lsb_release_content = self.run(
1277 'cat "%s"' % client_constants.LSB_RELEASE).stdout.strip()
1278 return lsbrelease_utils.get_chromeos_release_version(
1279 lsb_release_content=lsb_release_content)
1280
1281
1282 def verify_cros_version_label(self):
1283 """ Make sure host's cros-version label match the actual image in dut.
1284
1285 Remove any cros-version: label that doesn't match that installed in
1286 the dut.
1287
1288 @param raise_error: Set to True to raise exception if any mismatch found
1289
1290 @raise error.AutoservError: If any mismatch between cros-version label
1291 and the build installed in dut is found.
1292 """
1293 labels = self._AFE.get_labels(
1294 name__startswith=ds_constants.VERSION_PREFIX,
1295 host__hostname=self.hostname)
1296 mismatch_found = False
1297 if labels:
1298 # Get CHROMEOS_RELEASE_VERSION from lsb-release, e.g., 6908.0.0.
1299 # Note that it's different from cros-version label, which has
1300 # builder and branch info, e.g.,
1301 # cros-version:peppy-release/R43-6908.0.0
1302 release_version = self.get_release_version()
1303 host_list = [self.hostname]
1304 for label in labels:
1305 # Remove any cros-version label that does not match
1306 # release_version.
1307 build_version = label.name[len(ds_constants.VERSION_PREFIX):]
1308 if not utils.version_match(build_version, release_version):
1309 logging.warn('cros-version label "%s" does not match '
1310 'release version %s. Removing the label.',
1311 label.name, release_version)
1312 label.remove_hosts(hosts=host_list)
1313 mismatch_found = True
1314 if mismatch_found:
Dan Shi1057bae2015-03-30 11:35:09 -07001315 autotest_es.post(use_http=True,
1316 type_str='cros_version_label_mismatch',
1317 metadata={'hostname': self.hostname})
Dan Shi549fb822015-03-24 18:01:11 -07001318 raise error.AutoservError('The host has wrong cros-version label.')
1319
1320
beepsc87ff602013-07-31 21:53:00 -07001321 def cleanup(self):
MK Ryu35d661e2014-09-25 17:44:10 -07001322 self.run('rm -f %s' % client_constants.CLEANUP_LOGS_PAUSED_FILE)
Scott Zawalskiddbc31e2012-11-15 11:29:01 -05001323 try:
beepsc87ff602013-07-31 21:53:00 -07001324 self._restart_ui()
1325 except (error.AutotestRunError, error.AutoservRunError,
1326 FactoryImageCheckerException):
Ilja H. Friedel04be2bd2014-05-07 21:29:59 -07001327 logging.warning('Unable to restart ui, rebooting device.')
Scott Zawalskiddbc31e2012-11-15 11:29:01 -05001328 # Since restarting the UI fails fall back to normal Autotest
1329 # cleanup routines, i.e. reboot the machine.
Fang Deng0ca40e22013-08-27 17:47:44 -07001330 super(CrosHost, self).cleanup()
Simran Basi5e6339a2013-03-21 11:34:32 -07001331 # Check if the rpm outlet was manipulated.
Simran Basid5e5e272012-09-24 15:23:59 -07001332 if self.has_power():
Simran Basi5e6339a2013-03-21 11:34:32 -07001333 self._cleanup_poweron()
Dan Shi549fb822015-03-24 18:01:11 -07001334 self.verify_cros_version_label()
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001335
1336
Yu-Ju Honga2be94a2012-07-31 09:48:52 -07001337 def reboot(self, **dargs):
1338 """
1339 This function reboots the site host. The more generic
1340 RemoteHost.reboot() performs sync and sleeps for 5
1341 seconds. This is not necessary for Chrome OS devices as the
1342 sync should be finished in a short time during the reboot
1343 command.
1344 """
Tom Wai-Hong Tamf5cd1d42012-08-13 12:04:08 +08001345 if 'reboot_cmd' not in dargs:
Doug Anderson7d5aeb22014-02-27 15:12:17 -08001346 reboot_timeout = dargs.get('reboot_timeout', 10)
J. Richard Barnette9af19632015-09-25 12:18:03 -07001347 dargs['reboot_cmd'] = ('sleep 1; '
1348 'reboot & sleep %d; '
1349 'reboot -f' % reboot_timeout)
Yu-Ju Honga2be94a2012-07-31 09:48:52 -07001350 # Enable fastsync to avoid running extra sync commands before reboot.
Tom Wai-Hong Tamf5cd1d42012-08-13 12:04:08 +08001351 if 'fastsync' not in dargs:
1352 dargs['fastsync'] = True
Michael Liangda8c60a2014-06-03 13:24:51 -07001353
Charlie Mooneya8e6dab2014-05-29 14:37:55 -07001354 # For purposes of logging reboot times:
1355 # Get the board name i.e. 'daisy_spring'
Michael Liangca4f5a62014-07-10 15:45:13 -07001356 board_fullname = self.get_board()
1357
1358 # Strip the prefix and add it to dargs.
1359 dargs['board'] = board_fullname[board_fullname.find(':')+1:]
Vincent Palatindf2372c2016-10-07 17:03:00 +02001360 # Record who called us
1361 orig = sys._getframe(1).f_code
Vincent Palatin80780b22016-07-27 16:02:37 +02001362 metric_fields = {'board' : dargs['board'],
Vincent Palatindf2372c2016-10-07 17:03:00 +02001363 'dut_host_name' : self.hostname,
1364 'success' : True}
1365 metric_debug_fields = {'board' : dargs['board'],
1366 'caller' : "%s:%s" % (orig.co_filename, orig.co_name),
1367 'success' : True,
1368 'error' : ''}
1369
Vincent Palatin80780b22016-07-27 16:02:37 +02001370 t0 = time.time()
1371 try:
1372 super(CrosHost, self).reboot(**dargs)
1373 except Exception as e:
1374 metric_fields['success'] = False
Vincent Palatindf2372c2016-10-07 17:03:00 +02001375 metric_debug_fields['success'] = False
1376 metric_debug_fields['error'] = type(e).__name__
Vincent Palatin80780b22016-07-27 16:02:37 +02001377 raise
1378 finally:
1379 duration = int(time.time() - t0)
Dan Shi5e2efb72017-02-07 11:40:23 -08001380 metrics.Counter(
1381 'chromeos/autotest/autoserv/reboot_count').increment(
1382 fields=metric_fields)
1383 metrics.Counter(
1384 'chromeos/autotest/autoserv/reboot_debug').increment(
1385 fields=metric_debug_fields)
1386 metrics.SecondsDistribution(
1387 'chromeos/autotest/autoserv/reboot_duration').add(
1388 duration, fields=metric_fields)
Yu-Ju Honga2be94a2012-07-31 09:48:52 -07001389
1390
Gwendal Grignou7a61d2f2014-05-23 11:05:51 -07001391 def suspend(self, **dargs):
1392 """
1393 This function suspends the site host.
1394 """
1395 suspend_time = dargs.get('suspend_time', 60)
1396 dargs['timeout'] = suspend_time
1397 if 'suspend_cmd' not in dargs:
J. Richard Barnette9af19632015-09-25 12:18:03 -07001398 dargs['suspend_cmd'] = ' && '.join([
1399 'echo 0 > /sys/class/rtc/rtc0/wakealarm',
Gwendal Grignou7a61d2f2014-05-23 11:05:51 -07001400 'echo +%d > /sys/class/rtc/rtc0/wakealarm' % suspend_time,
J. Richard Barnette9af19632015-09-25 12:18:03 -07001401 'powerd_dbus_suspend --delay=0'])
Gwendal Grignou7a61d2f2014-05-23 11:05:51 -07001402 super(CrosHost, self).suspend(**dargs)
1403
1404
Simran Basiec564392014-08-25 16:48:09 -07001405 def upstart_status(self, service_name):
1406 """Check the status of an upstart init script.
1407
1408 @param service_name: Service to look up.
1409
1410 @returns True if the service is running, False otherwise.
1411 """
1412 return self.run('status %s | grep start/running' %
1413 service_name).stdout.strip() != ''
1414
1415
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001416 def verify_software(self):
Richard Barnetteb2bc13c2013-01-08 17:32:51 -08001417 """Verify working software on a Chrome OS system.
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001418
Richard Barnetteb2bc13c2013-01-08 17:32:51 -08001419 Tests for the following conditions:
1420 1. All conditions tested by the parent version of this
1421 function.
1422 2. Sufficient space in /mnt/stateful_partition.
Fang Deng6b05f5b2013-03-20 13:42:11 -07001423 3. Sufficient space in /mnt/stateful_partition/encrypted.
1424 4. update_engine answers a simple status request over DBus.
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001425
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001426 """
Fang Deng0ca40e22013-08-27 17:47:44 -07001427 super(CrosHost, self).verify_software()
Dan Shib8540a52015-07-16 14:18:23 -07001428 default_kilo_inodes_required = CONFIG.get_config_value(
1429 'SERVER', 'kilo_inodes_required', type=int, default=100)
1430 board = self.get_board().replace(ds_constants.BOARD_PREFIX, '')
1431 kilo_inodes_required = CONFIG.get_config_value(
1432 'SERVER', 'kilo_inodes_required_%s' % board,
1433 type=int, default=default_kilo_inodes_required)
1434 self.check_inodes('/mnt/stateful_partition', kilo_inodes_required)
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001435 self.check_diskspace(
1436 '/mnt/stateful_partition',
Dan Shib8540a52015-07-16 14:18:23 -07001437 CONFIG.get_config_value(
Fang Deng6b05f5b2013-03-20 13:42:11 -07001438 'SERVER', 'gb_diskspace_required', type=float,
1439 default=20.0))
Gaurav Shahe448af82014-06-19 15:18:59 -07001440 encrypted_stateful_path = '/mnt/stateful_partition/encrypted'
1441 # Not all targets build with encrypted stateful support.
1442 if self.path_exists(encrypted_stateful_path):
1443 self.check_diskspace(
1444 encrypted_stateful_path,
Dan Shib8540a52015-07-16 14:18:23 -07001445 CONFIG.get_config_value(
Gaurav Shahe448af82014-06-19 15:18:59 -07001446 'SERVER', 'gb_encrypted_diskspace_required', type=float,
1447 default=0.1))
beepsc87ff602013-07-31 21:53:00 -07001448
Simran Basiec564392014-08-25 16:48:09 -07001449 if not self.upstart_status('system-services'):
Prashanth B5d0a0512014-04-25 12:26:08 -07001450 raise error.AutoservError('Chrome failed to reach login. '
1451 'System services not running.')
1452
beepsc87ff602013-07-31 21:53:00 -07001453 # Factory images don't run update engine,
1454 # goofy controls dbus on these DUTs.
1455 if not self._is_factory_image():
1456 self.run('update_engine_client --status')
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001457
Dan Shi549fb822015-03-24 18:01:11 -07001458 self.verify_cros_version_label()
1459
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001460
J. Richard Barnettea7e7fdc2016-02-12 12:35:36 -08001461 def verify(self):
1462 self._repair_strategy.verify(self)
1463
1464
Fang Deng96667ca2013-08-01 17:46:18 -07001465 def make_ssh_command(self, user='root', port=22, opts='', hosts_file=None,
1466 connect_timeout=None, alive_interval=None):
1467 """Override default make_ssh_command to use options tuned for Chrome OS.
1468
1469 Tuning changes:
1470 - ConnectTimeout=30; maximum of 30 seconds allowed for an SSH
1471 connection failure. Consistency with remote_access.sh.
1472
Samuel Tan2ce155b2015-06-23 18:24:38 -07001473 - ServerAliveInterval=900; which causes SSH to ping connection every
1474 900 seconds. In conjunction with ServerAliveCountMax ensures
1475 that if the connection dies, Autotest will bail out.
Fang Deng96667ca2013-08-01 17:46:18 -07001476 Originally tried 60 secs, but saw frequent job ABORTS where
Samuel Tan2ce155b2015-06-23 18:24:38 -07001477 the test completed successfully. Later increased from 180 seconds to
1478 900 seconds to account for tests where the DUT is suspended for
1479 longer periods of time.
Fang Deng96667ca2013-08-01 17:46:18 -07001480
1481 - ServerAliveCountMax=3; consistency with remote_access.sh.
1482
1483 - ConnectAttempts=4; reduce flakiness in connection errors;
1484 consistency with remote_access.sh.
1485
1486 - UserKnownHostsFile=/dev/null; we don't care about the keys.
1487 Host keys change with every new installation, don't waste
1488 memory/space saving them.
1489
1490 - SSH protocol forced to 2; needed for ServerAliveInterval.
1491
1492 @param user User name to use for the ssh connection.
1493 @param port Port on the target host to use for ssh connection.
1494 @param opts Additional options to the ssh command.
1495 @param hosts_file Ignored.
1496 @param connect_timeout Ignored.
1497 @param alive_interval Ignored.
1498 """
Aviv Keshetc5947fa2013-09-04 14:06:29 -07001499 base_command = ('/usr/bin/ssh -a -x %s %s %s'
1500 ' -o StrictHostKeyChecking=no'
Fang Deng96667ca2013-08-01 17:46:18 -07001501 ' -o UserKnownHostsFile=/dev/null -o BatchMode=yes'
Samuel Tan2ce155b2015-06-23 18:24:38 -07001502 ' -o ConnectTimeout=30 -o ServerAliveInterval=900'
Fang Deng96667ca2013-08-01 17:46:18 -07001503 ' -o ServerAliveCountMax=3 -o ConnectionAttempts=4'
1504 ' -o Protocol=2 -l %s -p %d')
Aviv Keshetc5947fa2013-09-04 14:06:29 -07001505 return base_command % (self._ssh_verbosity_flag, self._ssh_options,
1506 opts, user, port)
Jason Abeleb6f924f2013-11-13 16:01:54 -08001507 def syslog(self, message, tag='autotest'):
1508 """Logs a message to syslog on host.
1509
1510 @param message String message to log into syslog
1511 @param tag String tag prefix for syslog
1512
1513 """
1514 self.run('logger -t "%s" "%s"' % (tag, message))
1515
1516
J. Richard Barnetteb6de7e32013-02-14 13:28:04 -08001517 def _ping_check_status(self, status):
1518 """Ping the host once, and return whether it has a given status.
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001519
J. Richard Barnetteb6de7e32013-02-14 13:28:04 -08001520 @param status Check the ping status against this value.
1521 @return True iff `status` and the result of ping are the same
1522 (i.e. both True or both False).
1523
1524 """
1525 ping_val = utils.ping(self.hostname, tries=1, deadline=1)
1526 return not (status ^ (ping_val == 0))
1527
1528 def _ping_wait_for_status(self, status, timeout):
1529 """Wait for the host to have a given status (UP or DOWN).
1530
1531 Status is checked by polling. Polling will not last longer
1532 than the number of seconds in `timeout`. The polling
1533 interval will be long enough that only approximately
1534 _PING_WAIT_COUNT polling cycles will be executed, subject
1535 to a maximum interval of about one minute.
1536
1537 @param status Waiting will stop immediately if `ping` of the
1538 host returns this status.
1539 @param timeout Poll for at most this many seconds.
1540 @return True iff the host status from `ping` matched the
1541 requested status at the time of return.
1542
1543 """
1544 # _ping_check_status() takes about 1 second, hence the
1545 # "- 1" in the formula below.
Nathan Ciobanu38480a32016-10-25 15:26:45 -07001546 # FIXME: if the ping command errors then _ping_check_status()
1547 # returns instantly. If timeout is also smaller than twice
1548 # _PING_WAIT_COUNT then the while loop below forks many
1549 # thousands of ping commands (see /tmp/test_that_results_XXXXX/
1550 # /results-1-logging_YYY.ZZZ/debug/autoserv.DEBUG) and hogs one
1551 # CPU core for 60 seconds.
J. Richard Barnetteb6de7e32013-02-14 13:28:04 -08001552 poll_interval = min(int(timeout / self._PING_WAIT_COUNT), 60) - 1
1553 end_time = time.time() + timeout
1554 while time.time() <= end_time:
1555 if self._ping_check_status(status):
1556 return True
1557 if poll_interval > 0:
1558 time.sleep(poll_interval)
1559
1560 # The last thing we did was sleep(poll_interval), so it may
1561 # have been too long since the last `ping`. Check one more
1562 # time, just to be sure.
1563 return self._ping_check_status(status)
1564
1565 def ping_wait_up(self, timeout):
1566 """Wait for the host to respond to `ping`.
1567
1568 N.B. This method is not a reliable substitute for
1569 `wait_up()`, because a host that responds to ping will not
1570 necessarily respond to ssh. This method should only be used
1571 if the target DUT can be considered functional even if it
1572 can't be reached via ssh.
1573
1574 @param timeout Minimum time to allow before declaring the
1575 host to be non-responsive.
1576 @return True iff the host answered to ping before the timeout.
1577
1578 """
1579 return self._ping_wait_for_status(self._PING_STATUS_UP, timeout)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001580
Andrew Bresticker678c0c72013-01-22 10:44:09 -08001581 def ping_wait_down(self, timeout):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001582 """Wait until the host no longer responds to `ping`.
1583
J. Richard Barnetteb6de7e32013-02-14 13:28:04 -08001584 This function can be used as a slightly faster version of
1585 `wait_down()`, by avoiding potentially long ssh timeouts.
1586
1587 @param timeout Minimum time to allow for the host to become
1588 non-responsive.
1589 @return True iff the host quit answering ping before the
1590 timeout.
1591
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001592 """
J. Richard Barnetteb6de7e32013-02-14 13:28:04 -08001593 return self._ping_wait_for_status(self._PING_STATUS_DOWN, timeout)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001594
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08001595 def test_wait_for_sleep(self, sleep_timeout=None):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001596 """Wait for the client to enter low-power sleep mode.
1597
1598 The test for "is asleep" can't distinguish a system that is
1599 powered off; to confirm that the unit was asleep, it is
1600 necessary to force resume, and then call
1601 `test_wait_for_resume()`.
1602
1603 This function is expected to be called from a test as part
1604 of a sequence like the following:
1605
1606 ~~~~~~~~
1607 boot_id = host.get_boot_id()
1608 # trigger sleep on the host
1609 host.test_wait_for_sleep()
1610 # trigger resume on the host
1611 host.test_wait_for_resume(boot_id)
1612 ~~~~~~~~
1613
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08001614 @param sleep_timeout time limit in seconds to allow the host sleep.
1615
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001616 @exception TestFail The host did not go to sleep within
1617 the allowed time.
1618 """
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08001619 if sleep_timeout is None:
1620 sleep_timeout = self.SLEEP_TIMEOUT
1621
1622 if not self.ping_wait_down(timeout=sleep_timeout):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001623 raise error.TestFail(
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08001624 'client failed to sleep after %d seconds' % sleep_timeout)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001625
1626
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08001627 def test_wait_for_resume(self, old_boot_id, resume_timeout=None):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001628 """Wait for the client to resume from low-power sleep mode.
1629
1630 The `old_boot_id` parameter should be the value from
1631 `get_boot_id()` obtained prior to entering sleep mode. A
1632 `TestFail` exception is raised if the boot id changes.
1633
1634 See @ref test_wait_for_sleep for more on this function's
1635 usage.
1636
J. Richard Barnette7214e0b2013-02-06 15:20:49 -08001637 @param old_boot_id A boot id value obtained before the
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001638 target host went to sleep.
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08001639 @param resume_timeout time limit in seconds to allow the host up.
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001640
1641 @exception TestFail The host did not respond within the
1642 allowed time.
1643 @exception TestFail The host responded, but the boot id test
1644 indicated a reboot rather than a sleep
1645 cycle.
1646 """
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08001647 if resume_timeout is None:
1648 resume_timeout = self.RESUME_TIMEOUT
1649
1650 if not self.wait_up(timeout=resume_timeout):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001651 raise error.TestFail(
1652 'client failed to resume from sleep after %d seconds' %
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08001653 resume_timeout)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001654 else:
1655 new_boot_id = self.get_boot_id()
1656 if new_boot_id != old_boot_id:
Tom Wai-Hong Tam01792682015-01-06 08:00:46 +08001657 logging.error('client rebooted (old boot %s, new boot %s)',
1658 old_boot_id, new_boot_id)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001659 raise error.TestFail(
Tom Wai-Hong Tam01792682015-01-06 08:00:46 +08001660 'client rebooted, but sleep was expected')
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001661
1662
Tom Wai-Hong Tamfe005c22014-12-03 09:25:44 +08001663 def test_wait_for_shutdown(self, shutdown_timeout=None):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001664 """Wait for the client to shut down.
1665
1666 The test for "has shut down" can't distinguish a system that
1667 is merely asleep; to confirm that the unit was down, it is
1668 necessary to force boot, and then call test_wait_for_boot().
1669
1670 This function is expected to be called from a test as part
1671 of a sequence like the following:
1672
1673 ~~~~~~~~
1674 boot_id = host.get_boot_id()
1675 # trigger shutdown on the host
1676 host.test_wait_for_shutdown()
1677 # trigger boot on the host
1678 host.test_wait_for_boot(boot_id)
1679 ~~~~~~~~
1680
Tom Wai-Hong Tamfe005c22014-12-03 09:25:44 +08001681 @param shutdown_timeout time limit in seconds to allow the host down.
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001682 @exception TestFail The host did not shut down within the
1683 allowed time.
1684 """
Tom Wai-Hong Tamfe005c22014-12-03 09:25:44 +08001685 if shutdown_timeout is None:
1686 shutdown_timeout = self.SHUTDOWN_TIMEOUT
1687
1688 if not self.ping_wait_down(timeout=shutdown_timeout):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001689 raise error.TestFail(
1690 'client failed to shut down after %d seconds' %
Tom Wai-Hong Tamfe005c22014-12-03 09:25:44 +08001691 shutdown_timeout)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001692
1693
1694 def test_wait_for_boot(self, old_boot_id=None):
1695 """Wait for the client to boot from cold power.
1696
1697 The `old_boot_id` parameter should be the value from
1698 `get_boot_id()` obtained prior to shutting down. A
1699 `TestFail` exception is raised if the boot id does not
1700 change. The boot id test is omitted if `old_boot_id` is not
1701 specified.
1702
1703 See @ref test_wait_for_shutdown for more on this function's
1704 usage.
1705
J. Richard Barnette7214e0b2013-02-06 15:20:49 -08001706 @param old_boot_id A boot id value obtained before the
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001707 shut down.
1708
1709 @exception TestFail The host did not respond within the
1710 allowed time.
1711 @exception TestFail The host responded, but the boot id test
1712 indicated that there was no reboot.
1713 """
J. Richard Barnetteeb69d722012-06-18 17:29:44 -07001714 if not self.wait_up(timeout=self.REBOOT_TIMEOUT):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001715 raise error.TestFail(
1716 'client failed to reboot after %d seconds' %
J. Richard Barnetteeb69d722012-06-18 17:29:44 -07001717 self.REBOOT_TIMEOUT)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001718 elif old_boot_id:
1719 if self.get_boot_id() == old_boot_id:
Tom Wai-Hong Tam01792682015-01-06 08:00:46 +08001720 logging.error('client not rebooted (boot %s)',
1721 old_boot_id)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001722 raise error.TestFail(
Tom Wai-Hong Tam01792682015-01-06 08:00:46 +08001723 'client is back up, but did not reboot')
Simran Basid5e5e272012-09-24 15:23:59 -07001724
1725
1726 @staticmethod
1727 def check_for_rpm_support(hostname):
1728 """For a given hostname, return whether or not it is powered by an RPM.
1729
Simran Basi1df55112013-09-06 11:25:09 -07001730 @param hostname: hostname to check for rpm support.
1731
Simran Basid5e5e272012-09-24 15:23:59 -07001732 @return None if this host does not follows the defined naming format
1733 for RPM powered DUT's in the lab. If it does follow the format,
1734 it returns a regular expression MatchObject instead.
1735 """
Fang Dengbaff9082015-01-06 13:46:15 -08001736 return re.match(CrosHost._RPM_HOSTNAME_REGEX, hostname)
Simran Basid5e5e272012-09-24 15:23:59 -07001737
1738
1739 def has_power(self):
1740 """For this host, return whether or not it is powered by an RPM.
1741
1742 @return True if this host is in the CROS lab and follows the defined
1743 naming format.
1744 """
Fang Deng0ca40e22013-08-27 17:47:44 -07001745 return CrosHost.check_for_rpm_support(self.hostname)
Simran Basid5e5e272012-09-24 15:23:59 -07001746
1747
Ismail Noorbasha07fdb612013-02-14 14:13:31 -08001748 def _set_power(self, state, power_method):
1749 """Sets the power to the host via RPM, Servo or manual.
1750
1751 @param state Specifies which power state to set to DUT
1752 @param power_method Specifies which method of power control to
1753 use. By default "RPM" will be used. Valid values
1754 are the strings "RPM", "manual", "servoj10".
1755
1756 """
1757 ACCEPTABLE_STATES = ['ON', 'OFF']
1758
1759 if state.upper() not in ACCEPTABLE_STATES:
1760 raise error.TestError('State must be one of: %s.'
1761 % (ACCEPTABLE_STATES,))
1762
1763 if power_method == self.POWER_CONTROL_SERVO:
1764 logging.info('Setting servo port J10 to %s', state)
1765 self.servo.set('prtctl3_pwren', state.lower())
1766 time.sleep(self._USB_POWER_TIMEOUT)
1767 elif power_method == self.POWER_CONTROL_MANUAL:
1768 logging.info('You have %d seconds to set the AC power to %s.',
1769 self._POWER_CYCLE_TIMEOUT, state)
1770 time.sleep(self._POWER_CYCLE_TIMEOUT)
1771 else:
1772 if not self.has_power():
1773 raise error.TestFail('DUT does not have RPM connected.')
Simran Basi5e6339a2013-03-21 11:34:32 -07001774 afe = frontend_wrappers.RetryingAFE(timeout_min=5, delay_sec=10)
1775 afe.set_host_attribute(self._RPM_OUTLET_CHANGED, True,
1776 hostname=self.hostname)
Simran Basi1df55112013-09-06 11:25:09 -07001777 rpm_client.set_power(self.hostname, state.upper(), timeout_mins=5)
Simran Basid5e5e272012-09-24 15:23:59 -07001778
1779
Ismail Noorbasha07fdb612013-02-14 14:13:31 -08001780 def power_off(self, power_method=POWER_CONTROL_RPM):
1781 """Turn off power to this host via RPM, Servo or manual.
1782
1783 @param power_method Specifies which method of power control to
1784 use. By default "RPM" will be used. Valid values
1785 are the strings "RPM", "manual", "servoj10".
1786
1787 """
1788 self._set_power('OFF', power_method)
Simran Basid5e5e272012-09-24 15:23:59 -07001789
1790
Ismail Noorbasha07fdb612013-02-14 14:13:31 -08001791 def power_on(self, power_method=POWER_CONTROL_RPM):
1792 """Turn on power to this host via RPM, Servo or manual.
1793
1794 @param power_method Specifies which method of power control to
1795 use. By default "RPM" will be used. Valid values
1796 are the strings "RPM", "manual", "servoj10".
1797
1798 """
1799 self._set_power('ON', power_method)
1800
1801
1802 def power_cycle(self, power_method=POWER_CONTROL_RPM):
1803 """Cycle power to this host by turning it OFF, then ON.
1804
1805 @param power_method Specifies which method of power control to
1806 use. By default "RPM" will be used. Valid values
1807 are the strings "RPM", "manual", "servoj10".
1808
1809 """
1810 if power_method in (self.POWER_CONTROL_SERVO,
1811 self.POWER_CONTROL_MANUAL):
1812 self.power_off(power_method=power_method)
1813 time.sleep(self._POWER_CYCLE_TIMEOUT)
1814 self.power_on(power_method=power_method)
1815 else:
1816 rpm_client.set_power(self.hostname, 'CYCLE')
Simran Basic6f1f7a2012-10-16 10:47:46 -07001817
1818
1819 def get_platform(self):
1820 """Determine the correct platform label for this host.
1821
1822 @returns a string representing this host's platform.
1823 """
1824 crossystem = utils.Crossystem(self)
1825 crossystem.init()
1826 # Extract fwid value and use the leading part as the platform id.
1827 # fwid generally follow the format of {platform}.{firmware version}
1828 # Example: Alex.X.YYY.Z or Google_Alex.X.YYY.Z
1829 platform = crossystem.fwid().split('.')[0].lower()
1830 # Newer platforms start with 'Google_' while the older ones do not.
1831 return platform.replace('google_', '')
1832
1833
Hung-ying Tyanb1328032014-04-01 14:18:54 +08001834 def get_architecture(self):
1835 """Determine the correct architecture label for this host.
1836
1837 @returns a string representing this host's architecture.
1838 """
1839 crossystem = utils.Crossystem(self)
1840 crossystem.init()
1841 return crossystem.arch()
1842
1843
Luis Lozano40b7d0d2014-01-17 15:12:06 -08001844 def get_chrome_version(self):
1845 """Gets the Chrome version number and milestone as strings.
1846
1847 Invokes "chrome --version" to get the version number and milestone.
1848
1849 @return A tuple (chrome_ver, milestone) where "chrome_ver" is the
1850 current Chrome version number as a string (in the form "W.X.Y.Z")
1851 and "milestone" is the first component of the version number
1852 (the "W" from "W.X.Y.Z"). If the version number cannot be parsed
1853 in the "W.X.Y.Z" format, the "chrome_ver" will be the full output
1854 of "chrome --version" and the milestone will be the empty string.
1855
1856 """
MK Ryu35d661e2014-09-25 17:44:10 -07001857 version_string = self.run(client_constants.CHROME_VERSION_COMMAND).stdout
Luis Lozano40b7d0d2014-01-17 15:12:06 -08001858 return utils.parse_chrome_version(version_string)
1859
J. Richard Barnetted2af5852016-02-05 15:03:10 -08001860
Kevin Chenga2619dc2016-03-28 11:42:08 -07001861 # TODO(kevcheng): change this to just return the board without the
1862 # 'board:' prefix and fix up all the callers. Also look into removing the
1863 # need for this method.
Simran Basic6f1f7a2012-10-16 10:47:46 -07001864 def get_board(self):
1865 """Determine the correct board label for this host.
1866
1867 @returns a string representing this host's board.
1868 """
1869 release_info = utils.parse_cmd_output('cat /etc/lsb-release',
1870 run_method=self.run)
J. Richard Barnetted2af5852016-02-05 15:03:10 -08001871 return (ds_constants.BOARD_PREFIX +
1872 release_info['CHROMEOS_RELEASE_BOARD'])
Simran Basic6f1f7a2012-10-16 10:47:46 -07001873
1874
Kevin Chenga328da62016-03-31 10:49:04 -07001875
1876 def has_lightsensor(self):
1877 """Determine the correct board label for this host.
1878
1879 @returns the string 'lightsensor' if this host has a lightsensor or
1880 None if it does not.
1881 """
1882 search_cmd = "find -L %s -maxdepth 4 | egrep '%s'" % (
1883 self._LIGHTSENSOR_SEARCH_DIR, '|'.join(self._LIGHTSENSOR_FILES))
1884 try:
1885 # Run the search cmd following the symlinks. Stderr_tee is set to
1886 # None as there can be a symlink loop, but the command will still
1887 # execute correctly with a few messages printed to stderr.
1888 self.run(search_cmd, stdout_tee=None, stderr_tee=None)
1889 return 'lightsensor'
1890 except error.AutoservRunError:
1891 # egrep exited with a return code of 1 meaning none of the possible
1892 # lightsensor files existed.
1893 return None
1894
1895
1896 def has_bluetooth(self):
1897 """Determine the correct board label for this host.
1898
1899 @returns the string 'bluetooth' if this host has bluetooth or
1900 None if it does not.
1901 """
1902 try:
1903 self.run('test -d /sys/class/bluetooth/hci0')
1904 # test exited with a return code of 0.
1905 return 'bluetooth'
1906 except error.AutoservRunError:
1907 # test exited with a return code 1 meaning the directory did not
1908 # exist.
1909 return None
1910
1911
Kevin Chenga328da62016-03-31 10:49:04 -07001912 def get_accels(self):
1913 """
1914 Determine the type of accelerometers on this host.
1915
1916 @returns a string representing this host's accelerometer type.
1917 At present, it only returns "accel:cros-ec", for accelerometers
1918 attached to a Chrome OS EC, or none, if no accelerometers.
1919 """
1920 # Check to make sure we have ectool
1921 rv = self.run('which ectool', ignore_status=True)
1922 if rv.exit_status:
1923 logging.info("No ectool cmd found, assuming no EC accelerometers")
1924 return None
1925
1926 # Check that the EC supports the motionsense command
1927 rv = self.run('ectool motionsense', ignore_status=True)
1928 if rv.exit_status:
1929 logging.info("EC does not support motionsense command "
1930 "assuming no EC accelerometers")
1931 return None
1932
1933 # Check that EC motion sensors are active
1934 active = self.run('ectool motionsense active').stdout.split('\n')
1935 if active[0] == "0":
1936 logging.info("Motion sense inactive, assuming no EC accelerometers")
1937 return None
1938
1939 logging.info("EC accelerometers found")
1940 return 'accel:cros-ec'
1941
1942
1943 def has_chameleon(self):
1944 """Determine if a Chameleon connected to this host.
1945
1946 @returns a list containing two strings ('chameleon' and
1947 'chameleon:' + label, e.g. 'chameleon:hdmi') if this host
1948 has a Chameleon or None if it has not.
1949 """
1950 if self._chameleon_host:
1951 return ['chameleon', 'chameleon:' + self.chameleon.get_label()]
1952 else:
1953 return None
1954
1955
1956 def has_loopback_dongle(self):
1957 """Determine if an audio loopback dongle is plugged to this host.
1958
1959 @returns 'audio_loopback_dongle' when there is an audio loopback dongle
1960 plugged to this host.
1961 None when there is no audio loopback dongle
1962 plugged to this host.
1963 """
1964 nodes_info = self.run(command=cras_utils.get_cras_nodes_cmd(),
1965 ignore_status=True).stdout
1966 if (cras_utils.node_type_is_plugged('HEADPHONE', nodes_info) and
1967 cras_utils.node_type_is_plugged('MIC', nodes_info)):
1968 return 'audio_loopback_dongle'
1969 else:
1970 return None
1971
1972
1973 def get_power_supply(self):
1974 """
1975 Determine what type of power supply the host has
1976
1977 @returns a string representing this host's power supply.
1978 'power:battery' when the device has a battery intended for
1979 extended use
1980 'power:AC_primary' when the device has a battery not intended
1981 for extended use (for moving the machine, etc)
1982 'power:AC_only' when the device has no battery at all.
1983 """
1984 psu = self.run(command='mosys psu type', ignore_status=True)
1985 if psu.exit_status:
1986 # The psu command for mosys is not included for all platforms. The
1987 # assumption is that the device will have a battery if the command
1988 # is not found.
1989 return 'power:battery'
1990
1991 psu_str = psu.stdout.strip()
1992 if psu_str == 'unknown':
1993 return None
1994
1995 return 'power:%s' % psu_str
1996
1997
1998 def get_storage(self):
1999 """
2000 Determine the type of boot device for this host.
2001
2002 Determine if the internal device is SCSI or dw_mmc device.
2003 Then check that it is SSD or HDD or eMMC or something else.
2004
2005 @returns a string representing this host's internal device type.
2006 'storage:ssd' when internal device is solid state drive
2007 'storage:hdd' when internal device is hard disk drive
2008 'storage:mmc' when internal device is mmc drive
2009 None When internal device is something else or
2010 when we are unable to determine the type
2011 """
2012 # The output should be /dev/mmcblk* for SD/eMMC or /dev/sd* for scsi
2013 rootdev_cmd = ' '.join(['. /usr/sbin/write_gpt.sh;',
2014 '. /usr/share/misc/chromeos-common.sh;',
2015 'load_base_vars;',
2016 'get_fixed_dst_drive'])
2017 rootdev = self.run(command=rootdev_cmd, ignore_status=True)
2018 if rootdev.exit_status:
2019 logging.info("Fail to run %s", rootdev_cmd)
2020 return None
2021 rootdev_str = rootdev.stdout.strip()
2022
2023 if not rootdev_str:
2024 return None
2025
2026 rootdev_base = os.path.basename(rootdev_str)
2027
2028 mmc_pattern = '/dev/mmcblk[0-9]'
2029 if re.match(mmc_pattern, rootdev_str):
2030 # Use type to determine if the internal device is eMMC or somthing
2031 # else. We can assume that MMC is always an internal device.
2032 type_cmd = 'cat /sys/block/%s/device/type' % rootdev_base
2033 type = self.run(command=type_cmd, ignore_status=True)
2034 if type.exit_status:
2035 logging.info("Fail to run %s", type_cmd)
2036 return None
2037 type_str = type.stdout.strip()
2038
2039 if type_str == 'MMC':
2040 return 'storage:mmc'
2041
2042 scsi_pattern = '/dev/sd[a-z]+'
2043 if re.match(scsi_pattern, rootdev.stdout):
2044 # Read symlink for /sys/block/sd* to determine if the internal
2045 # device is connected via ata or usb.
2046 link_cmd = 'readlink /sys/block/%s' % rootdev_base
2047 link = self.run(command=link_cmd, ignore_status=True)
2048 if link.exit_status:
2049 logging.info("Fail to run %s", link_cmd)
2050 return None
2051 link_str = link.stdout.strip()
2052 if 'usb' in link_str:
2053 return None
2054
2055 # Read rotation to determine if the internal device is ssd or hdd.
2056 rotate_cmd = str('cat /sys/block/%s/queue/rotational'
2057 % rootdev_base)
2058 rotate = self.run(command=rotate_cmd, ignore_status=True)
2059 if rotate.exit_status:
2060 logging.info("Fail to run %s", rotate_cmd)
2061 return None
2062 rotate_str = rotate.stdout.strip()
2063
2064 rotate_dict = {'0':'storage:ssd', '1':'storage:hdd'}
2065 return rotate_dict.get(rotate_str)
2066
2067 # All other internal device / error case will always fall here
2068 return None
2069
2070
2071 def get_servo(self):
2072 """Determine if the host has a servo attached.
2073
2074 If the host has a working servo attached, it should have a servo label.
2075
2076 @return: string 'servo' if the host has servo attached. Otherwise,
2077 returns None.
2078 """
2079 return 'servo' if self._servo_host else None
2080
2081
2082 def get_video_labels(self):
2083 """Run /usr/local/bin/avtest_label_detect to get a list of video labels.
2084
2085 Sample output of avtest_label_detect:
2086 Detected label: hw_video_acc_vp8
2087 Detected label: webcam
2088
2089 @return: A list of labels detected by tool avtest_label_detect.
2090 """
2091 try:
2092 result = self.run('/usr/local/bin/avtest_label_detect').stdout
2093 return re.findall('^Detected label: (\w+)$', result, re.M)
2094 except error.AutoservRunError:
2095 # The tool is not installed.
2096 return []
2097
2098
2099 def is_video_glitch_detection_supported(self):
2100 """ Determine if a board under test is supported for video glitch
2101 detection tests.
2102
2103 @return: 'video_glitch_detection' if board is supported, None otherwise.
2104 """
2105 board = self.get_board().replace(ds_constants.BOARD_PREFIX, '')
2106
2107 if board in video_test_constants.SUPPORTED_BOARDS:
2108 return 'video_glitch_detection'
2109
2110 return None
2111
2112
2113 def get_touch(self):
2114 """
2115 Determine whether board under test has a touchpad or touchscreen.
2116
2117 @return: A list of some combination of 'touchscreen' and 'touchpad',
2118 depending on what is present on the device.
2119
2120 """
2121 labels = []
2122 looking_for = ['touchpad', 'touchscreen']
2123 player = input_playback.InputPlayback()
2124 input_events = self.run('ls /dev/input/event*').stdout.strip().split()
2125 filename = '/tmp/touch_labels'
2126 for event in input_events:
2127 self.run('evtest %s > %s' % (event, filename), timeout=1,
2128 ignore_timeout=True)
2129 properties = self.run('cat %s' % filename).stdout
2130 input_type = player._determine_input_type(properties)
2131 if input_type in looking_for:
2132 labels.append(input_type)
2133 looking_for.remove(input_type)
2134 if len(looking_for) == 0:
2135 break
2136 self.run('rm %s' % filename)
2137
2138 return labels
2139
2140
2141 def has_internal_display(self):
2142 """Determine if the device under test is equipped with an internal
2143 display.
2144
2145 @return: 'internal_display' if one is present; None otherwise.
2146 """
2147 from autotest_lib.client.cros.graphics import graphics_utils
2148 from autotest_lib.client.common_lib import utils as common_utils
2149
2150 def __system_output(cmd):
2151 return self.run(cmd).stdout
2152
2153 def __read_file(remote_path):
2154 return self.run('cat %s' % remote_path).stdout
2155
2156 # Hijack the necessary client functions so that we can take advantage
2157 # of the client lib here.
2158 # FIXME: find a less hacky way than this
2159 original_system_output = utils.system_output
2160 original_read_file = common_utils.read_file
2161 utils.system_output = __system_output
2162 common_utils.read_file = __read_file
2163 try:
2164 return ('internal_display' if graphics_utils.has_internal_display()
2165 else None)
2166 finally:
2167 utils.system_output = original_system_output
2168 common_utils.read_file = original_read_file
2169
2170
2171 def has_lucid_sleep_support(self):
2172 """Determine if the device under test has support for lucid sleep.
2173
2174 @return 'lucidsleep' if this board supports lucid sleep; None otherwise
2175 """
2176 board = self.get_board().replace(ds_constants.BOARD_PREFIX, '')
2177 return 'lucidsleep' if board in LUCID_SLEEP_BOARDS else None
2178
2179
Dan Shi85276d42014-04-08 22:11:45 -07002180 def is_boot_from_usb(self):
2181 """Check if DUT is boot from USB.
2182
2183 @return: True if DUT is boot from usb.
2184 """
2185 device = self.run('rootdev -s -d').stdout.strip()
2186 removable = int(self.run('cat /sys/block/%s/removable' %
2187 os.path.basename(device)).stdout.strip())
2188 return removable == 1
Helen Zhang17dae2b2014-11-11 09:25:52 -08002189
2190
2191 def read_from_meminfo(self, key):
Dan Shi49ca0932014-11-14 11:22:27 -08002192 """Return the memory info from /proc/meminfo
Helen Zhang17dae2b2014-11-11 09:25:52 -08002193
2194 @param key: meminfo requested
2195
2196 @return the memory value as a string
2197
2198 """
Helen Zhang17dae2b2014-11-11 09:25:52 -08002199 meminfo = self.run('grep %s /proc/meminfo' % key).stdout.strip()
2200 logging.debug('%s', meminfo)
2201 return int(re.search(r'\d+', meminfo).group(0))
Rohit Makasana8a4923c2015-08-13 17:04:26 -07002202
2203
Rohit Makasana98e696f2016-06-03 18:48:10 -07002204 def get_cpu_arch(self):
2205 """Returns CPU arch of the device.
2206
2207 @return CPU architecture of the DUT.
2208 """
2209 # Add CPUs by following logic in client/bin/base_utils.py.
2210 if self.run("grep '^flags.*:.* lm .*' /proc/cpuinfo",
2211 ignore_status=True).stdout:
2212 return 'x86_64'
2213 if self.run("grep -Ei 'ARM|CPU implementer' /proc/cpuinfo",
2214 ignore_status=True).stdout:
2215 return 'arm'
2216 return 'i386'
2217
2218
Rohit Makasana8a4923c2015-08-13 17:04:26 -07002219 def get_board_type(self):
2220 """
2221 Get the DUT's device type from /etc/lsb-release.
Danny Chan471a8d12015-08-18 14:57:41 -07002222 DEVICETYPE can be one of CHROMEBOX, CHROMEBASE, CHROMEBOOK or more.
2223
2224 @return value of DEVICETYPE param from lsb-release.
Rohit Makasana8a4923c2015-08-13 17:04:26 -07002225 """
Danny Chan471a8d12015-08-18 14:57:41 -07002226 device_type = self.run('grep DEVICETYPE /etc/lsb-release',
2227 ignore_status=True).stdout
2228 if device_type:
Kalin Stoyanov524310b2015-08-21 16:24:04 -07002229 return device_type.split('=')[-1].strip()
Danny Chan471a8d12015-08-18 14:57:41 -07002230 return ''
Gilad Arnolda76bef02015-09-29 13:55:15 -07002231
2232
2233 def get_os_type(self):
2234 return 'cros'
Simran Basia5522a32015-10-06 11:01:24 -07002235
2236
2237 def enable_adb_testing(self):
2238 """Mark this host as an adb tester."""
Dan Shia2872172015-10-31 01:16:51 -07002239 self.run('touch %s' % constants.ANDROID_TESTER_FILEFLAG)
Kevin Chenga2619dc2016-03-28 11:42:08 -07002240
2241
2242 def get_labels(self):
Kevin Chengf8660142016-08-12 10:17:41 -07002243 """Return the detected labels on the host."""
Kevin Chenga2619dc2016-03-28 11:42:08 -07002244 return self.labels.get_labels(self)
Kevin Chengf8660142016-08-12 10:17:41 -07002245
2246
2247 def update_labels(self):
2248 """Update the labels on the host."""
2249 self.labels.update_labels(self)