blob: 3e12a77538c918e8fa306994a575206e03416ceb [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
Gabe Black1e1c41b2015-02-04 23:55:15 -080021from autotest_lib.client.common_lib.cros.graphite import autotest_stats
Hsinyu Chaoe0b08e62015-08-11 10:50:37 +000022from autotest_lib.client.cros import constants as client_constants
J. Richard Barnette84890bd2014-02-21 11:05:47 -080023from autotest_lib.client.cros import cros_ui
Kevin Chenga328da62016-03-31 10:49:04 -070024from autotest_lib.client.cros.audio import cras_utils
25from autotest_lib.client.cros.input_playback import input_playback
26from autotest_lib.client.cros.video import constants as video_test_constants
Simran Basi5ace6f22016-01-06 17:30:44 -080027from autotest_lib.server import afe_utils
MK Ryu35d661e2014-09-25 17:44:10 -070028from autotest_lib.server import autoserv_parser
29from autotest_lib.server import autotest
30from autotest_lib.server import constants
Dan Shia1ecd5c2013-06-06 11:21:31 -070031from autotest_lib.server import utils as server_utils
Dan Shi9cb0eec2014-06-03 09:04:50 -070032from autotest_lib.server.cros import provision
Scott Zawalski89c44dd2013-02-26 09:28:02 -050033from autotest_lib.server.cros.dynamic_suite import constants as ds_constants
Simran Basi5e6339a2013-03-21 11:34:32 -070034from autotest_lib.server.cros.dynamic_suite import tools, frontend_wrappers
Dan Shi9cb0eec2014-06-03 09:04:50 -070035from autotest_lib.server.cros.faft.config.config import Config as FAFTConfig
Scottfe06ed82015-11-05 17:15:01 -080036from autotest_lib.server.cros.servo import plankton
Fang Deng96667ca2013-08-01 17:46:18 -070037from autotest_lib.server.hosts import abstract_ssh
Kevin Chenga2619dc2016-03-28 11:42:08 -070038from autotest_lib.server.hosts import base_label
39from autotest_lib.server.hosts import cros_label
Tom Wai-Hong Tamefe1c7f2014-01-02 14:00:11 +080040from autotest_lib.server.hosts import chameleon_host
J. Richard Barnettea7e7fdc2016-02-12 12:35:36 -080041from autotest_lib.server.hosts import cros_repair
Scottfe06ed82015-11-05 17:15:01 -080042from autotest_lib.server.hosts import plankton_host
Fang Deng5d518f42013-08-02 14:04:32 -070043from autotest_lib.server.hosts import servo_host
Simran Basidcff4252012-11-20 16:13:20 -080044from autotest_lib.site_utils.rpm_control_system import rpm_client
Simran Basid5e5e272012-09-24 15:23:59 -070045
Simran Basi382506b2016-09-13 14:58:15 -070046# In case cros_host is being ran via SSP on an older Moblab version with an
47# older chromite version.
48try:
49 from chromite.lib import metrics
50except:
51 metrics = None
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):
334 return afe_utils.get_stable_cros_version(self._get_board_from_afe())
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500335
336
Chris Sosab76e0ee2013-05-22 16:55:41 -0700337 def lookup_job_repo_url(self):
338 """Looks up the job_repo_url for the host.
339
Dan Shi8190eb82016-02-11 17:15:58 -0800340 This is kept for backwards compatibility as AU test code in older
341 branch does not use server-side packaging and calls this method through
342 the host object.
343
344 TODO(dshi): Once R50 falls off the stable branch, we should remove this
345 method.
346
Chris Sosab76e0ee2013-05-22 16:55:41 -0700347 @returns job_repo_url from AFE or None if not found.
348
349 @raises KeyError if the host does not have a job_repo_url
350 """
Dan Shibe3636a2016-02-14 22:48:01 -0800351 return afe_utils.get_host_attribute(self, ds_constants.JOB_REPO_URL)
beepscb6f1e22013-06-28 19:14:10 -0700352
353
beepsdae65fd2013-07-26 16:24:41 -0700354 def verify_job_repo_url(self, tag=''):
beepscb6f1e22013-06-28 19:14:10 -0700355 """
356 Make sure job_repo_url of this host is valid.
357
joychen03eaad92013-06-26 09:55:21 -0700358 Eg: The job_repo_url "http://lmn.cd.ab.xyx:8080/static/\
beepscb6f1e22013-06-28 19:14:10 -0700359 lumpy-release/R29-4279.0.0/autotest/packages" claims to have the
360 autotest package for lumpy-release/R29-4279.0.0. If this isn't the case,
361 download and extract it. If the devserver embedded in the url is
362 unresponsive, update the job_repo_url of the host after staging it on
363 another devserver.
364
365 @param job_repo_url: A url pointing to the devserver where the autotest
366 package for this build should be staged.
beepsdae65fd2013-07-26 16:24:41 -0700367 @param tag: The tag from the server job, in the format
368 <job_id>-<user>/<hostname>, or <hostless> for a server job.
beepscb6f1e22013-06-28 19:14:10 -0700369
370 @raises DevServerException: If we could not resolve a devserver.
371 @raises AutoservError: If we're unable to save the new job_repo_url as
372 a result of choosing a new devserver because the old one failed to
373 respond to a health check.
beeps0c865032013-07-30 11:37:06 -0700374 @raises urllib2.URLError: If the devserver embedded in job_repo_url
375 doesn't respond within the timeout.
beepscb6f1e22013-06-28 19:14:10 -0700376 """
Dan Shibe3636a2016-02-14 22:48:01 -0800377 job_repo_url = afe_utils.get_host_attribute(self,
378 ds_constants.JOB_REPO_URL)
beepscb6f1e22013-06-28 19:14:10 -0700379 if not job_repo_url:
380 logging.warning('No job repo url set on host %s', self.hostname)
381 return
382
383 logging.info('Verifying job repo url %s', job_repo_url)
384 devserver_url, image_name = tools.get_devserver_build_from_package_url(
385 job_repo_url)
386
beeps0c865032013-07-30 11:37:06 -0700387 ds = dev_server.ImageServer(devserver_url)
beepscb6f1e22013-06-28 19:14:10 -0700388
389 logging.info('Staging autotest artifacts for %s on devserver %s',
390 image_name, ds.url())
beeps687243d2013-07-18 15:29:27 -0700391
392 start_time = time.time()
Simran Basi25e7a922014-10-31 11:56:10 -0700393 ds.stage_artifacts(image_name, ['autotest_packages'])
beeps687243d2013-07-18 15:29:27 -0700394 stage_time = time.time() - start_time
395
396 # Record how much of the verification time comes from a devserver
397 # restage. If we're doing things right we should not see multiple
398 # devservers for a given board/build/branch path.
399 try:
J. Richard Barnette3cbd76b2013-11-27 12:11:25 -0800400 board, build_type, branch = server_utils.ParseBuildName(
beeps687243d2013-07-18 15:29:27 -0700401 image_name)[:3]
J. Richard Barnette3cbd76b2013-11-27 12:11:25 -0800402 except server_utils.ParseBuildNameException:
beeps687243d2013-07-18 15:29:27 -0700403 pass
404 else:
beeps0c865032013-07-30 11:37:06 -0700405 devserver = devserver_url[
Chris Sosa65425082013-10-16 13:26:22 -0700406 devserver_url.find('/') + 2:devserver_url.rfind(':')]
beeps687243d2013-07-18 15:29:27 -0700407 stats_key = {
408 'board': board,
409 'build_type': build_type,
410 'branch': branch,
beeps0c865032013-07-30 11:37:06 -0700411 'devserver': devserver.replace('.', '_'),
beeps687243d2013-07-18 15:29:27 -0700412 }
Gabe Black1e1c41b2015-02-04 23:55:15 -0800413 autotest_stats.Gauge('verify_job_repo_url').send(
beeps687243d2013-07-18 15:29:27 -0700414 '%(board)s.%(build_type)s.%(branch)s.%(devserver)s' % stats_key,
415 stage_time)
Aviv Keshetf9375482016-10-26 02:09:03 -0700416 if metrics:
417 monarch_fields = {
418 'board': board,
419 'build_type': build_type,
Aviv Keshet3a76f762016-10-27 17:41:51 -0700420 # TODO(akeshet): To be consistent with most other metrics,
421 # consider changing the following field to be named
422 # 'milestone'.
Aviv Keshetf9375482016-10-26 02:09:03 -0700423 'branch': branch,
424 'dev_server': devserver,
425 }
426 metrics.Counter(
427 'chromeos/autotest/provision/verify_url'
428 ).increment(fields=monarch_fields)
Aviv Keshet3a76f762016-10-27 17:41:51 -0700429 metrics.SecondsDistribution(
430 'chromeos/autotest/provision/verify_url_duration'
431 ).add(stage_time, fields=monarch_fields)
Dan Shicf4d2032015-03-12 15:04:21 -0700432 def stage_server_side_package(self, image=None):
433 """Stage autotest server-side package on devserver.
434
435 @param image: Full path of an OS image to install or a build name.
436
437 @return: A url to the autotest server-side package.
Dan Shi14de7622016-08-22 11:09:06 -0700438
439 @raise: error.AutoservError if fail to locate the build to test with, or
440 fail to stage server-side package.
Dan Shicf4d2032015-03-12 15:04:21 -0700441 """
Dan Shid37736b2016-07-06 15:10:29 -0700442 # If enable_drone_in_restricted_subnet is False, do not set hostname
443 # in devserver.resolve call, so a devserver in non-restricted subnet
444 # is picked to stage autotest server package for drone to download.
445 hostname = self.hostname
446 if not server_utils.ENABLE_DRONE_IN_RESTRICTED_SUBNET:
447 hostname = None
Dan Shicf4d2032015-03-12 15:04:21 -0700448 if image:
449 image_name = tools.get_build_from_image(image)
450 if not image_name:
451 raise error.AutoservError(
452 'Failed to parse build name from %s' % image)
Dan Shid37736b2016-07-06 15:10:29 -0700453 ds = dev_server.ImageServer.resolve(image_name, hostname)
Dan Shicf4d2032015-03-12 15:04:21 -0700454 else:
Dan Shibe3636a2016-02-14 22:48:01 -0800455 job_repo_url = afe_utils.get_host_attribute(
456 self, ds_constants.JOB_REPO_URL)
Dan Shicf4d2032015-03-12 15:04:21 -0700457 if job_repo_url:
458 devserver_url, image_name = (
459 tools.get_devserver_build_from_package_url(job_repo_url))
Dan Shid37736b2016-07-06 15:10:29 -0700460 # If enable_drone_in_restricted_subnet is True, use the
461 # existing devserver. Otherwise, resolve a new one in
462 # non-restricted subnet.
463 if server_utils.ENABLE_DRONE_IN_RESTRICTED_SUBNET:
464 ds = dev_server.ImageServer(devserver_url)
465 else:
466 ds = dev_server.ImageServer.resolve(image_name)
Dan Shicf4d2032015-03-12 15:04:21 -0700467 else:
Kevin Cheng84a71ba2016-07-14 11:03:57 -0700468 labels = afe_utils.get_labels(self, self.VERSION_PREFIX)
Dan Shicf4d2032015-03-12 15:04:21 -0700469 if not labels:
470 raise error.AutoservError(
471 'Failed to stage server-side package. The host has '
472 'no job_report_url attribute or version label.')
Kevin Cheng84a71ba2016-07-14 11:03:57 -0700473 image_name = labels[0][len(self.VERSION_PREFIX + ':'):]
Dan Shid37736b2016-07-06 15:10:29 -0700474 ds = dev_server.ImageServer.resolve(image_name, hostname)
Dan Shica503482015-03-30 17:23:25 -0700475
476 # Get the OS version of the build, for any build older than
477 # MIN_VERSION_SUPPORT_SSP, server side packaging is not supported.
478 match = re.match('.*/R\d+-(\d+)\.', image_name)
479 if match and int(match.group(1)) < self.MIN_VERSION_SUPPORT_SSP:
Dan Shi14de7622016-08-22 11:09:06 -0700480 raise error.AutoservError(
481 'Build %s is older than %s. Server side packaging is '
482 'disabled.' % (image_name, self.MIN_VERSION_SUPPORT_SSP))
Dan Shica503482015-03-30 17:23:25 -0700483
Dan Shicf4d2032015-03-12 15:04:21 -0700484 ds.stage_artifacts(image_name, ['autotest_server_package'])
485 return '%s/static/%s/%s' % (ds.url(), image_name,
486 'autotest_server_package.tar.bz2')
487
488
Dan Shi0f466e82013-02-22 15:44:58 -0800489 def _try_stateful_update(self, update_url, force_update, updater):
490 """Try to use stateful update to initialize DUT.
491
492 When DUT is already running the same version that machine_install
493 tries to install, stateful update is a much faster way to clean up
494 the DUT for testing, compared to a full reimage. It is implemeted
495 by calling autoupdater.run_update, but skipping updating root, as
496 updating the kernel is time consuming and not necessary.
497
498 @param update_url: url of the image.
499 @param force_update: Set to True to update the image even if the DUT
500 is running the same version.
501 @param updater: ChromiumOSUpdater instance used to update the DUT.
502 @returns: True if the DUT was updated with stateful update.
503
504 """
Dan Shi10b98482016-02-02 14:38:50 -0800505 # Stop service ap-update-manager to prevent rebooting during autoupdate.
506 # The service is used in jetstream boards, but not other CrOS devices.
507 self.run('sudo stop ap-update-manager', ignore_status=True)
508
J. Richard Barnette3f731032014-04-07 17:42:59 -0700509 # TODO(jrbarnette): Yes, I hate this re.match() test case.
510 # It's better than the alternative: see crbug.com/360944.
511 image_name = autoupdater.url_to_image_name(update_url)
512 release_pattern = r'^.*-release/R[0-9]+-[0-9]+\.[0-9]+\.0$'
513 if not re.match(release_pattern, image_name):
514 return False
Dan Shi0f466e82013-02-22 15:44:58 -0800515 if not updater.check_version():
516 return False
517 if not force_update:
518 logging.info('Canceling stateful update because the new and '
519 'old versions are the same.')
520 return False
521 # Following folders should be rebuilt after stateful update.
522 # A test file is used to confirm each folder gets rebuilt after
523 # the stateful update.
524 folders_to_check = ['/var', '/home', '/mnt/stateful_partition']
525 test_file = '.test_file_to_be_deleted'
526 for folder in folders_to_check:
527 touch_path = os.path.join(folder, test_file)
528 self.run('touch %s' % touch_path)
529
Chris Sosae92399e2015-04-24 11:32:59 -0700530 updater.run_update(update_root=False)
Dan Shi0f466e82013-02-22 15:44:58 -0800531
532 # Reboot to complete stateful update.
Chris Sosab76e0ee2013-05-22 16:55:41 -0700533 self.reboot(timeout=self.REBOOT_TIMEOUT, wait=True)
Dan Shi0f466e82013-02-22 15:44:58 -0800534 check_file_cmd = 'test -f %s; echo $?'
535 for folder in folders_to_check:
536 test_file_path = os.path.join(folder, test_file)
537 result = self.run(check_file_cmd % test_file_path,
538 ignore_status=True)
539 if result.exit_status == 1:
540 return False
541 return True
542
543
J. Richard Barnette7275b612013-06-04 18:13:11 -0700544 def _post_update_processing(self, updater, expected_kernel=None):
Dan Shi0f466e82013-02-22 15:44:58 -0800545 """After the DUT is updated, confirm machine_install succeeded.
546
547 @param updater: ChromiumOSUpdater instance used to update the DUT.
J. Richard Barnette7275b612013-06-04 18:13:11 -0700548 @param expected_kernel: kernel expected to be active after reboot,
549 or `None` to skip rollback checking.
Dan Shi0f466e82013-02-22 15:44:58 -0800550
551 """
J. Richard Barnette7275b612013-06-04 18:13:11 -0700552 # Touch the lab machine file to leave a marker that
553 # distinguishes this image from other test images.
554 # Afterwards, we must re-run the autoreboot script because
555 # it depends on the _LAB_MACHINE_FILE.
J. Richard Barnette71cc1862015-12-02 10:32:38 -0800556 autoreboot_cmd = ('FILE="%s" ; [ -f "$FILE" ] || '
557 '( touch "$FILE" ; start autoreboot )')
558 self.run(autoreboot_cmd % self._LAB_MACHINE_FILE)
Chris Sosa65425082013-10-16 13:26:22 -0700559 updater.verify_boot_expectations(
560 expected_kernel, rollback_message=
Gilad Arnoldc26ae1f2015-10-22 16:09:41 -0700561 'Build %s failed to boot on %s; system rolled back to previous '
Chris Sosa65425082013-10-16 13:26:22 -0700562 'build' % (updater.update_version, self.hostname))
J. Richard Barnette7275b612013-06-04 18:13:11 -0700563 # Check that we've got the build we meant to install.
564 if not updater.check_version_to_confirm_install():
565 raise autoupdater.ChromiumOSError(
566 'Failed to update %s to build %s; found build '
567 '%s instead' % (self.hostname,
Chris Sosa65425082013-10-16 13:26:22 -0700568 updater.update_version,
Dan Shi0942b1d2015-03-31 11:07:00 -0700569 self.get_release_version()))
Dan Shi0f466e82013-02-22 15:44:58 -0800570
Chris Sosae92399e2015-04-24 11:32:59 -0700571 logging.debug('Cleaning up old autotest directories.')
572 try:
573 installed_autodir = autotest.Autotest.get_installed_autodir(self)
574 self.run('rm -rf ' + installed_autodir)
575 except autotest.AutodirNotFoundError:
576 logging.debug('No autotest installed directory found.')
577
Dan Shi0f466e82013-02-22 15:44:58 -0800578
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700579 def _stage_image_for_update(self, image_name=None):
Chris Sosae92399e2015-04-24 11:32:59 -0700580 """Stage a build on a devserver and return the update_url and devserver.
Scott Zawalskieadbf702013-03-14 09:23:06 -0400581
582 @param image_name: a name like lumpy-release/R27-3837.0.0
Chris Sosae92399e2015-04-24 11:32:59 -0700583 @returns a tuple with an update URL like:
Scott Zawalskieadbf702013-03-14 09:23:06 -0400584 http://172.22.50.205:8082/update/lumpy-release/R27-3837.0.0
Chris Sosae92399e2015-04-24 11:32:59 -0700585 and the devserver instance.
Scott Zawalskieadbf702013-03-14 09:23:06 -0400586 """
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700587 if not image_name:
Richard Barnette260cbd02016-10-06 12:23:28 -0700588 image_name = self._get_cros_repair_image_name()
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700589 logging.info('Staging build for AU: %s', image_name)
Dan Shi216389c2015-12-22 11:03:06 -0800590 devserver = dev_server.ImageServer.resolve(image_name, self.hostname)
Scott Zawalskieadbf702013-03-14 09:23:06 -0400591 devserver.trigger_download(image_name, synchronous=False)
Chris Sosae92399e2015-04-24 11:32:59 -0700592 return (tools.image_url_pattern() % (devserver.url(), image_name),
593 devserver)
Scott Zawalskieadbf702013-03-14 09:23:06 -0400594
595
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700596 def stage_image_for_servo(self, image_name=None):
597 """Stage a build on a devserver and return the update_url.
598
599 @param image_name: a name like lumpy-release/R27-3837.0.0
600 @returns an update URL like:
601 http://172.22.50.205:8082/update/lumpy-release/R27-3837.0.0
602 """
603 if not image_name:
Richard Barnette260cbd02016-10-06 12:23:28 -0700604 image_name = self._get_cros_repair_image_name()
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700605 logging.info('Staging build for servo install: %s', image_name)
Dan Shi216389c2015-12-22 11:03:06 -0800606 devserver = dev_server.ImageServer.resolve(image_name, self.hostname)
J. Richard Barnettee4af8b92013-05-01 13:16:12 -0700607 devserver.stage_artifacts(image_name, ['test_image'])
608 return devserver.get_test_image_url(image_name)
609
610
beepse539be02013-07-31 21:57:39 -0700611 def stage_factory_image_for_servo(self, image_name):
612 """Stage a build on a devserver and return the update_url.
613
614 @param image_name: a name like <baord>/4262.204.0
beeps12c0a3c2013-09-03 11:58:27 -0700615
beepse539be02013-07-31 21:57:39 -0700616 @return: An update URL, eg:
617 http://<devserver>/static/canary-channel/\
618 <board>/4262.204.0/factory_test/chromiumos_factory_image.bin
beeps12c0a3c2013-09-03 11:58:27 -0700619
620 @raises: ValueError if the factory artifact name is missing from
621 the config.
622
beepse539be02013-07-31 21:57:39 -0700623 """
624 if not image_name:
625 logging.error('Need an image_name to stage a factory image.')
626 return
627
Dan Shib8540a52015-07-16 14:18:23 -0700628 factory_artifact = CONFIG.get_config_value(
beeps12c0a3c2013-09-03 11:58:27 -0700629 'CROS', 'factory_artifact', type=str, default='')
630 if not factory_artifact:
631 raise ValueError('Cannot retrieve the factory artifact name from '
632 'autotest config, and hence cannot stage factory '
633 'artifacts.')
634
beepse539be02013-07-31 21:57:39 -0700635 logging.info('Staging build for servo install: %s', image_name)
Dan Shi216389c2015-12-22 11:03:06 -0800636 devserver = dev_server.ImageServer.resolve(image_name, self.hostname)
beepse539be02013-07-31 21:57:39 -0700637 devserver.stage_artifacts(
638 image_name,
beeps12c0a3c2013-09-03 11:58:27 -0700639 [factory_artifact],
640 archive_url=None)
beepse539be02013-07-31 21:57:39 -0700641
642 return tools.factory_image_url_pattern() % (devserver.url(), image_name)
643
644
xixuanec801e32016-08-25 10:20:22 -0700645 def machine_install_by_devserver(self, update_url=None, force_update=False,
xixuan5dc64ea2016-05-20 17:27:51 -0700646 local_devserver=False, repair=False,
647 force_full_update=False):
648 """Ultiize devserver to install the DUT.
649
650 (TODO) crbugs.com/627269: The logic in this function has some overlap
651 with those in function machine_install. The merge will be done later,
652 not in the same CL.
653
xixuanec801e32016-08-25 10:20:22 -0700654 @param update_url: The update_url or build for the host to update.
xixuan5dc64ea2016-05-20 17:27:51 -0700655 @param force_update: Force an update even if the version installed
656 is the same. Default:False
657 @param local_devserver: Used by test_that to allow people to
658 use their local devserver. Default: False
659 @param repair: Forces update to repair image. Implies force_update.
660 @param force_full_update: If True, do not attempt to run stateful
661 update, force a full reimage. If False, try stateful update
662 first when the dut is already installed with the same version.
663 @raises autoupdater.ChromiumOSError
664
665 @returns A tuple of (image_name, host_attributes).
666 image_name is the name of image installed, e.g.,
667 veyron_jerry-release/R50-7871.0.0
668 host_attributes is a dictionary of (attribute, value), which
669 can be saved to afe_host_attributes table in database. This
670 method returns a dictionary with a single entry of
671 `job_repo_url`: repo_url, where repo_url is a devserver url to
672 autotest packages.
673 """
xixuan5dc64ea2016-05-20 17:27:51 -0700674 if repair:
Richard Barnette260cbd02016-10-06 12:23:28 -0700675 update_url = self._get_cros_repair_image_name()
xixuan5dc64ea2016-05-20 17:27:51 -0700676 force_update = True
677
xixuanec801e32016-08-25 10:20:22 -0700678 if not update_url and not self._parser.options.image:
xixuan5dc64ea2016-05-20 17:27:51 -0700679 raise error.AutoservError(
680 'There is no update URL, nor a method to get one.')
681
xixuanec801e32016-08-25 10:20:22 -0700682 if not update_url and self._parser.options.image:
683 update_url = self._parser.options.image
xixuan5dc64ea2016-05-20 17:27:51 -0700684
xixuan5dc64ea2016-05-20 17:27:51 -0700685
686 # Get build from parameter or AFE.
687 # If the build is not a URL, let devserver to stage it first.
688 # Otherwise, choose a devserver to trigger auto-update.
xixuanec801e32016-08-25 10:20:22 -0700689 build = None
Aviv Keshetf9375482016-10-26 02:09:03 -0700690 devserver = None
691 logging.debug('Resolving a devserver for auto-update')
Aviv Keshet3a76f762016-10-27 17:41:51 -0700692 previously_resolved = False
xixuanec801e32016-08-25 10:20:22 -0700693 if update_url.startswith('http://'):
694 build = autoupdater.url_to_image_name(update_url)
Aviv Keshet3a76f762016-10-27 17:41:51 -0700695 previously_resolved = True
696 else:
697 build = update_url
698 devserver = dev_server.resolve(build, self.hostname)
699 server_name = dev_server.ImageServer.get_server_name(devserver.url())
700
701 try:
702 board, build_type, milestone, _ = server_utils.ParseBuildName(build)
703 except server_utils.ParseBuildNameException:
704 logging.warning('Unable to parse build name %s. Something is '
705 'likely broken, but will continue anyway.',
706 build)
707 board, build_type, milestone = ('', '', '')
708
709 monarch_fields = {
710 'dev_server': server_name,
711 'board': board,
712 'build_type': build_type,
713 'milestone': milestone
714 }
715
716 if previously_resolved:
xixuan2e62d162016-09-29 19:19:19 -0700717 # Make sure devserver for Auto-Update has staged the build.
xixuan2e62d162016-09-29 19:19:19 -0700718 if server_name not in update_url:
Aviv Keshet3a76f762016-10-27 17:41:51 -0700719 logging.debug('Resolved to devserver that does not match '
720 'update_url. The previously resolved devserver '
721 'must be unhealthy. Switching to use devserver %s,'
722 ' and re-staging.',
xixuan2e62d162016-09-29 19:19:19 -0700723 server_name)
Aviv Keshetf9375482016-10-26 02:09:03 -0700724 logging.info('Staging build for AU: %s', update_url)
xixuan2e62d162016-09-29 19:19:19 -0700725 devserver.trigger_download(build, synchronous=False)
Aviv Keshetf9375482016-10-26 02:09:03 -0700726 if metrics:
727 c = metrics.Counter(
728 'chromeos/autotest/provision/failover_download')
Aviv Keshet3a76f762016-10-27 17:41:51 -0700729 c.increment(fields=monarch_fields)
xixuan5dc64ea2016-05-20 17:27:51 -0700730 else:
Aviv Keshetf9375482016-10-26 02:09:03 -0700731 logging.info('Staging build for AU: %s', update_url)
xixuan5dc64ea2016-05-20 17:27:51 -0700732 devserver.trigger_download(build, synchronous=False)
Aviv Keshetf9375482016-10-26 02:09:03 -0700733 if metrics:
734 c = metrics.Counter(
735 'chromeos/autotest/provision/trigger_download')
Aviv Keshet3a76f762016-10-27 17:41:51 -0700736 c.increment(fields=monarch_fields)
xixuan5dc64ea2016-05-20 17:27:51 -0700737
738 # Report provision stats.
Aviv Keshetf9375482016-10-26 02:09:03 -0700739 statsd_server_name = server_name.replace('.', '_')
740 autotest_stats.Counter('cros_host_provision.' + statsd_server_name).increment()
xixuan5dc64ea2016-05-20 17:27:51 -0700741 autotest_stats.Counter('cros_host_provision.total').increment()
xixuanec801e32016-08-25 10:20:22 -0700742 logging.debug('Resolved devserver for auto-update: %s', devserver.url())
xixuan5dc64ea2016-05-20 17:27:51 -0700743
Aviv Keshetf9375482016-10-26 02:09:03 -0700744 # and other metrics from this function.
745 if metrics:
746 metrics.Counter('chromeos/autotest/provision/resolve',
Aviv Keshet3a76f762016-10-27 17:41:51 -0700747 ).increment(fields=monarch_fields)
Aviv Keshetf9375482016-10-26 02:09:03 -0700748
xixuan5dc64ea2016-05-20 17:27:51 -0700749 devserver.auto_update(self.hostname, build,
xixuan95c6cce2016-10-05 17:49:41 -0700750 log_dir=self.job.resultdir,
xixuan5dc64ea2016-05-20 17:27:51 -0700751 force_update=force_update,
752 full_update=force_full_update)
753
754 # The reason to resolve a new devserver in function machine_install
755 # is mostly because that the update_url there may has a strange format,
756 # and it's hard to parse the devserver url from it.
757 # Since we already resolve a devserver to trigger auto-update, the same
758 # devserver is used to form JOB_REPO_URL here. Verified in local test.
759 repo_url = tools.get_package_url(devserver.url(), build)
760 return build, {ds_constants.JOB_REPO_URL: repo_url}
761
762
Chris Sosaa3ac2152012-05-23 22:23:13 -0700763 def machine_install(self, update_url=None, force_update=False,
Richard Barnette0b023a72015-04-24 16:07:30 +0000764 local_devserver=False, repair=False,
765 force_full_update=False):
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500766 """Install the DUT.
767
Dan Shi0f466e82013-02-22 15:44:58 -0800768 Use stateful update if the DUT is already running the same build.
769 Stateful update does not update kernel and tends to run much faster
770 than a full reimage. If the DUT is running a different build, or it
771 failed to do a stateful update, full update, including kernel update,
772 will be applied to the DUT.
773
Simran Basi5ace6f22016-01-06 17:30:44 -0800774 Once a host enters machine_install its host attribute job_repo_url
775 (used for package install) will be removed and then updated.
Scott Zawalskieadbf702013-03-14 09:23:06 -0400776
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500777 @param update_url: The url to use for the update
778 pattern: http://$devserver:###/update/$build
779 If update_url is None and repair is True we will install the
Dan Shi6964fa52014-12-18 11:04:27 -0800780 stable image listed in afe_stable_versions table. If the table
781 is not setup, global_config value under CROS.stable_cros_version
782 will be used instead.
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500783 @param force_update: Force an update even if the version installed
784 is the same. Default:False
Christopher Wiley6a4ff932015-05-15 14:00:47 -0700785 @param local_devserver: Used by test_that to allow people to
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500786 use their local devserver. Default: False
Chris Sosae92399e2015-04-24 11:32:59 -0700787 @param repair: Forces update to repair image. Implies force_update.
Fang Deng3d3b9272014-12-22 12:20:28 -0800788 @param force_full_update: If True, do not attempt to run stateful
789 update, force a full reimage. If False, try stateful update
790 first when the dut is already installed with the same version.
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500791 @raises autoupdater.ChromiumOSError
792
Dan Shibe3636a2016-02-14 22:48:01 -0800793 @returns A tuple of (image_name, host_attributes).
794 image_name is the name of image installed, e.g.,
795 veyron_jerry-release/R50-7871.0.0
796 host_attributes is a dictionary of (attribute, value), which
797 can be saved to afe_host_attributes table in database. This
798 method returns a dictionary with a single entry of
799 `job_repo_url`: repo_url, where repo_url is a devserver url to
800 autotest packages.
Scott Zawalski89c44dd2013-02-26 09:28:02 -0500801 """
Chris Sosae92399e2015-04-24 11:32:59 -0700802 devserver = None
Richard Barnette0b023a72015-04-24 16:07:30 +0000803 if repair:
Chris Sosae92399e2015-04-24 11:32:59 -0700804 update_url, devserver = self._stage_image_for_update()
Richard Barnette0b023a72015-04-24 16:07:30 +0000805 force_update = True
Dan Shi0f466e82013-02-22 15:44:58 -0800806
Chris Sosae92399e2015-04-24 11:32:59 -0700807 if not update_url and not self._parser.options.image:
808 raise error.AutoservError(
Dan Shid07ee2e2015-09-24 14:49:25 -0700809 'There is no update URL, nor a method to get one.')
Chris Sosae92399e2015-04-24 11:32:59 -0700810
811 if not update_url and self._parser.options.image:
812 # This is the base case where we have no given update URL i.e.
813 # dynamic suites logic etc. This is the most flexible case where we
814 # can serve an update from any of our fleet of devservers.
815 requested_build = self._parser.options.image
816 if not requested_build.startswith('http://'):
817 logging.debug('Update will be staged for this installation')
818 update_url, devserver = self._stage_image_for_update(
Dan Shid07ee2e2015-09-24 14:49:25 -0700819 requested_build)
Chris Sosae92399e2015-04-24 11:32:59 -0700820 else:
821 update_url = requested_build
822
823 logging.debug('Update URL is %s', update_url)
824
Dan Shif48f8132016-02-18 10:34:30 -0800825 # Report provision stats.
xixuan9e2c98d2016-02-26 19:04:53 -0800826 server_name = dev_server.ImageServer.get_server_name(update_url)
Aviv Keshetf9375482016-10-26 02:09:03 -0700827 statsd_server_name = server_name.replace('.', '_')
828 autotest_stats.Counter('cros_host_provision.' + statsd_server_name).increment()
Dan Shif48f8132016-02-18 10:34:30 -0800829 autotest_stats.Counter('cros_host_provision.total').increment()
830
Dan Shid07ee2e2015-09-24 14:49:25 -0700831 # Create a file to indicate if provision fails. The file will be removed
832 # by stateful update or full install.
J. Richard Barnette91137f02016-03-10 16:52:26 -0800833 self.run('touch %s' % self.PROVISION_FAILED)
Dan Shid07ee2e2015-09-24 14:49:25 -0700834
Chris Sosae92399e2015-04-24 11:32:59 -0700835 update_complete = False
836 updater = autoupdater.ChromiumOSUpdater(
837 update_url, host=self, local_devserver=local_devserver)
Fang Deng3d3b9272014-12-22 12:20:28 -0800838 if not force_full_update:
839 try:
Chris Sosae92399e2015-04-24 11:32:59 -0700840 # If the DUT is already running the same build, try stateful
841 # update first as it's much quicker than a full re-image.
842 update_complete = self._try_stateful_update(
Dan Shid07ee2e2015-09-24 14:49:25 -0700843 update_url, force_update, updater)
Fang Deng3d3b9272014-12-22 12:20:28 -0800844 except Exception as e:
845 logging.exception(e)
J. Richard Barnette45e93de2012-04-11 17:24:15 -0700846
Dan Shi0f466e82013-02-22 15:44:58 -0800847 inactive_kernel = None
Chris Sosae92399e2015-04-24 11:32:59 -0700848 if update_complete or (not force_update and updater.check_version()):
849 logging.info('Install complete without full update')
850 else:
851 logging.info('DUT requires full update.')
852 self.reboot(timeout=self.REBOOT_TIMEOUT, wait=True)
Dan Shi10b98482016-02-02 14:38:50 -0800853 # Stop service ap-update-manager to prevent rebooting during
854 # autoupdate. The service is used in jetstream boards, but not other
855 # CrOS devices.
856 self.run('sudo stop ap-update-manager', ignore_status=True)
857
Chris Sosae92399e2015-04-24 11:32:59 -0700858 num_of_attempts = provision.FLAKY_DEVSERVER_ATTEMPTS
Chris Sosab7612bc2013-03-21 10:32:37 -0700859
Chris Sosae92399e2015-04-24 11:32:59 -0700860 while num_of_attempts > 0:
861 num_of_attempts -= 1
862 try:
863 updater.run_update()
864 except Exception:
865 logging.warn('Autoupdate did not complete.')
866 # Do additional check for the devserver health. Ideally,
867 # the autoupdater.py could raise an exception when it
868 # detected network flake but that would require
869 # instrumenting the update engine and parsing it log.
870 if (num_of_attempts <= 0 or
871 devserver is None or
xixuan9e2c98d2016-02-26 19:04:53 -0800872 dev_server.ImageServer.devserver_healthy(
Chris Sosae92399e2015-04-24 11:32:59 -0700873 devserver.url())):
Dan Shid07ee2e2015-09-24 14:49:25 -0700874 raise
J. Richard Barnette45e93de2012-04-11 17:24:15 -0700875
Chris Sosae92399e2015-04-24 11:32:59 -0700876 logging.warn('Devserver looks unhealthy. Trying another')
877 update_url, devserver = self._stage_image_for_update(
878 requested_build)
879 logging.debug('New Update URL is %s', update_url)
880 updater = autoupdater.ChromiumOSUpdater(
881 update_url, host=self,
882 local_devserver=local_devserver)
883 else:
884 break
J. Richard Barnette45e93de2012-04-11 17:24:15 -0700885
Chris Sosae92399e2015-04-24 11:32:59 -0700886 # Give it some time in case of IO issues.
887 time.sleep(10)
Dan Shi5699ac22014-12-19 10:55:49 -0800888
Chris Sosae92399e2015-04-24 11:32:59 -0700889 # Figure out active and inactive kernel.
890 active_kernel, inactive_kernel = updater.get_kernel_state()
Simran Basi13fa1ba2013-03-04 10:56:47 -0800891
Chris Sosae92399e2015-04-24 11:32:59 -0700892 # Ensure inactive kernel has higher priority than active.
893 if (updater.get_kernel_priority(inactive_kernel)
894 < updater.get_kernel_priority(active_kernel)):
895 raise autoupdater.ChromiumOSError(
896 'Update failed. The priority of the inactive kernel'
897 ' partition is less than that of the active kernel'
898 ' partition.')
899
900 # Updater has returned successfully; reboot the host.
Richard Barnettefc3d7432016-07-07 10:21:31 -0700901 #
902 # Regarding the 'crossystem' command: In some cases, the
903 # TPM gets into a state such that it fails verification.
904 # We don't know why. However, this call papers over the
905 # problem by clearing the TPM during the reboot.
906 #
907 # We ignore failures from 'crossystem'. Although failure
908 # here is unexpected, and could signal a bug, the point
909 # of the exercise is to paper over problems; allowing
910 # this to fail would defeat the purpose.
911 self.run('crossystem clear_tpm_owner_request=1',
912 ignore_status=True)
Chris Sosae92399e2015-04-24 11:32:59 -0700913 self.reboot(timeout=self.REBOOT_TIMEOUT, wait=True)
914
915 self._post_update_processing(updater, inactive_kernel)
Simran Basi5ace6f22016-01-06 17:30:44 -0800916 image_name = autoupdater.url_to_image_name(update_url)
Dan Shibe3636a2016-02-14 22:48:01 -0800917 # update_url is different from devserver url needed to stage autotest
918 # packages, therefore, resolve a new devserver url here.
919 devserver_url = dev_server.ImageServer.resolve(image_name,
920 self.hostname).url()
921 repo_url = tools.get_package_url(devserver_url, image_name)
922 return image_name, {ds_constants.JOB_REPO_URL: repo_url}
J. Richard Barnette45e93de2012-04-11 17:24:15 -0700923
924
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +0800925 def _clear_fw_version_labels(self, rw_only):
926 """Clear firmware version labels from the machine.
927
928 @param rw_only: True to only clear fwrw_version; otherewise, clear
929 both fwro_version and fwrw_version.
930 """
Dan Shi9cb0eec2014-06-03 09:04:50 -0700931 labels = self._AFE.get_labels(
Dan Shi0723bf52015-06-24 10:52:38 -0700932 name__startswith=provision.FW_RW_VERSION_PREFIX,
Dan Shi9cb0eec2014-06-03 09:04:50 -0700933 host__hostname=self.hostname)
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +0800934 if not rw_only:
935 labels = labels + self._AFE.get_labels(
936 name__startswith=provision.FW_RO_VERSION_PREFIX,
937 host__hostname=self.hostname)
Dan Shi9cb0eec2014-06-03 09:04:50 -0700938 for label in labels:
939 label.remove_hosts(hosts=[self.hostname])
940
941
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +0800942 def _add_fw_version_label(self, build, rw_only):
Dan Shi9cb0eec2014-06-03 09:04:50 -0700943 """Add firmware version label to the machine.
944
945 @param build: Build of firmware.
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +0800946 @param rw_only: True to only add fwrw_version; otherwise, add both
947 fwro_version and fwrw_version.
Dan Shi9cb0eec2014-06-03 09:04:50 -0700948
949 """
Allen Lifda3e232016-10-17 14:54:12 -0700950 fw_label = provision.FWRWVersionLabel(build)
MK Ryu73be9862015-07-06 12:25:00 -0700951 self._AFE.run('label_add_hosts', id=fw_label, hosts=[self.hostname])
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +0800952 if not rw_only:
Allen Lifda3e232016-10-17 14:54:12 -0700953 fw_label = provision.FWROVersionLabel(build)
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +0800954 self._AFE.run('label_add_hosts', id=fw_label, hosts=[self.hostname])
Dan Shi9cb0eec2014-06-03 09:04:50 -0700955
956
Tom Wai-Hong Tam4ac78982016-01-08 02:34:37 +0800957 def firmware_install(self, build=None, rw_only=False):
Dan Shi9cb0eec2014-06-03 09:04:50 -0700958 """Install firmware to the DUT.
959
960 Use stateful update if the DUT is already running the same build.
961 Stateful update does not update kernel and tends to run much faster
962 than a full reimage. If the DUT is running a different build, or it
963 failed to do a stateful update, full update, including kernel update,
964 will be applied to the DUT.
965
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +0800966 Once a host enters firmware_install its fw[ro|rw]_version label will
967 be removed. After the firmware is updated successfully, a new
968 fw[ro|rw]_version label will be added to the host.
Dan Shi9cb0eec2014-06-03 09:04:50 -0700969
970 @param build: The build version to which we want to provision the
971 firmware of the machine,
972 e.g. 'link-firmware/R22-2695.1.144'.
Tom Wai-Hong Tam4ac78982016-01-08 02:34:37 +0800973 @param rw_only: True to only install firmware to its RW portions. Keep
974 the RO portions unchanged.
Dan Shi9cb0eec2014-06-03 09:04:50 -0700975
976 TODO(dshi): After bug 381718 is fixed, update here with corresponding
977 exceptions that could be raised.
978
979 """
980 if not self.servo:
981 raise error.TestError('Host %s does not have servo.' %
982 self.hostname)
983
Wai-Hong Tam7b424e92016-11-10 15:35:24 -0800984 board = self.get_board().replace(ds_constants.BOARD_PREFIX, '')
Dan Shi9cb0eec2014-06-03 09:04:50 -0700985
Chris Sosae92399e2015-04-24 11:32:59 -0700986 # If build is not set, try to install firmware from stable CrOS.
Dan Shi9cb0eec2014-06-03 09:04:50 -0700987 if not build:
Richard Barnette260cbd02016-10-06 12:23:28 -0700988 build = afe_utils.get_stable_faft_version(board)
Dan Shi3d7a0e12015-10-12 11:55:45 -0700989 if not build:
990 raise error.TestError(
991 'Failed to find stable firmware build for %s.',
992 self.hostname)
993 logging.info('Will install firmware from build %s.', build)
Dan Shi9cb0eec2014-06-03 09:04:50 -0700994
995 config = FAFTConfig(board)
996 if config.use_u_boot:
997 ap_image = 'image-%s.bin' % board
998 else: # Depthcharge platform
999 ap_image = 'image.bin'
1000 ec_image = 'ec.bin'
Dan Shi216389c2015-12-22 11:03:06 -08001001 ds = dev_server.ImageServer.resolve(build, self.hostname)
Dan Shi9cb0eec2014-06-03 09:04:50 -07001002 ds.stage_artifacts(build, ['firmware'])
1003
1004 tmpd = autotemp.tempdir(unique_id='fwimage')
1005 try:
1006 fwurl = self._FW_IMAGE_URL_PATTERN % (ds.url(), build)
1007 local_tarball = os.path.join(tmpd.name, os.path.basename(fwurl))
1008 server_utils.system('wget -O %s %s' % (local_tarball, fwurl),
1009 timeout=60)
Dan Shi9cb0eec2014-06-03 09:04:50 -07001010
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +08001011 self._clear_fw_version_labels(rw_only)
Tom Wai-Hong Tam552fee12016-05-11 06:27:58 +08001012 if config.chrome_ec:
Tom Wai-Hong Tamf9db57a2016-03-17 05:32:22 +08001013 logging.info('Will re-program EC %snow', 'RW ' if rw_only else '')
1014 server_utils.system('tar xf %s -C %s %s' %
1015 (local_tarball, tmpd.name, ec_image),
1016 timeout=60)
1017 self.servo.program_ec(os.path.join(tmpd.name, ec_image), rw_only)
1018 else:
1019 logging.info('Not a Chrome EC, ignore re-programing it')
Tom Wai-Hong Tam4ac78982016-01-08 02:34:37 +08001020 logging.info('Will re-program BIOS %snow', 'RW ' if rw_only else '')
Tom Wai-Hong Tamf9db57a2016-03-17 05:32:22 +08001021 server_utils.system('tar xf %s -C %s %s' %
1022 (local_tarball, tmpd.name, ap_image),
1023 timeout=60)
Tom Wai-Hong Tam4ac78982016-01-08 02:34:37 +08001024 self.servo.program_bios(os.path.join(tmpd.name, ap_image), rw_only)
Dan Shi9cb0eec2014-06-03 09:04:50 -07001025 self.servo.get_power_state_controller().reset()
1026 time.sleep(self.servo.BOOT_DELAY)
Tom Wai-Hong Tam2d00cb22016-01-08 06:40:50 +08001027 self._add_fw_version_label(build, rw_only)
Dan Shi9cb0eec2014-06-03 09:04:50 -07001028 finally:
1029 tmpd.clean()
1030
1031
Dan Shi10e992b2013-08-30 11:02:59 -07001032 def show_update_engine_log(self):
1033 """Output update engine log."""
MK Ryu35d661e2014-09-25 17:44:10 -07001034 logging.debug('Dumping %s', client_constants.UPDATE_ENGINE_LOG)
1035 self.run('cat %s' % client_constants.UPDATE_ENGINE_LOG)
Dan Shi10e992b2013-08-30 11:02:59 -07001036
1037
Richard Barnette82c35912012-11-20 10:09:10 -08001038 def _get_board_from_afe(self):
Kevin Cheng84a71ba2016-07-14 11:03:57 -07001039 """Retrieve this host's board from its labels stored locally.
Richard Barnette82c35912012-11-20 10:09:10 -08001040
1041 Looks for a host label of the form "board:<board>", and
1042 returns the "<board>" part of the label. `None` is returned
1043 if there is not a single, unique label matching the pattern.
1044
1045 @returns board from label, or `None`.
1046 """
Richard Barnette260cbd02016-10-06 12:23:28 -07001047 board = afe_utils.get_board(self)
1048 if board is None:
1049 raise error.AutoservError('DUT cannot be repaired, '
1050 'there is no board attribute.')
1051 return board
Simran Basi833814b2013-01-29 13:13:43 -08001052
1053
Dan Shib3b6db32016-02-03 14:54:05 -08001054 def get_build(self):
1055 """Retrieve the current build for this Host from the AFE.
1056
1057 Looks through this host's labels in the AFE to determine its build.
1058 This method is replaced by afe_utils.get_build. It's kept here to
1059 maintain backwards compatibility for test control files in older CrOS
1060 builds (R48, R49 etc.) still call host.get_build, e.g.,
1061 `provision_AutoUpdate.double`.
1062 TODO(sbasi): Once R50 falls into release branch, this method can be
1063 removed.
1064
1065 @returns The current build or None if it could not find it or if there
1066 were multiple build labels assigned to this host.
1067 """
1068 return afe_utils.get_build(self)
1069
1070
beepsf079cfb2013-09-18 17:49:51 -07001071 def servo_install(self, image_url=None, usb_boot_timeout=USB_BOOT_TIMEOUT,
1072 install_timeout=INSTALL_TIMEOUT):
Scott Zawalski62bacae2013-03-05 10:40:32 -05001073 """
1074 Re-install the OS on the DUT by:
1075 1) installing a test image on a USB storage device attached to the Servo
1076 board,
Richard Barnette03a0c132012-11-05 12:40:35 -08001077 2) booting that image in recovery mode, and then
J. Richard Barnette31b2e312013-04-04 16:05:22 -07001078 3) installing the image with chromeos-install.
1079
Scott Zawalski62bacae2013-03-05 10:40:32 -05001080 @param image_url: If specified use as the url to install on the DUT.
1081 otherwise boot the currently staged image on the USB stick.
beepsf079cfb2013-09-18 17:49:51 -07001082 @param usb_boot_timeout: The usb_boot_timeout to use during reimage.
1083 Factory images need a longer usb_boot_timeout than regular
1084 cros images.
1085 @param install_timeout: The timeout to use when installing the chromeos
1086 image. Factory images need a longer install_timeout.
Richard Barnette03a0c132012-11-05 12:40:35 -08001087
Scott Zawalski62bacae2013-03-05 10:40:32 -05001088 @raises AutoservError if the image fails to boot.
beepsf079cfb2013-09-18 17:49:51 -07001089
J. Richard Barnette0199cc82014-12-05 17:08:40 -08001090 """
beepsf079cfb2013-09-18 17:49:51 -07001091 usb_boot_timer_key = ('servo_install.usb_boot_timeout_%s'
1092 % usb_boot_timeout)
1093 logging.info('Downloading image to USB, then booting from it. Usb boot '
1094 'timeout = %s', usb_boot_timeout)
Gabe Black1e1c41b2015-02-04 23:55:15 -08001095 timer = autotest_stats.Timer(usb_boot_timer_key)
beepsf079cfb2013-09-18 17:49:51 -07001096 timer.start()
J. Richard Barnette31b2e312013-04-04 16:05:22 -07001097 self.servo.install_recovery_image(image_url)
beepsf079cfb2013-09-18 17:49:51 -07001098 if not self.wait_up(timeout=usb_boot_timeout):
J. Richard Barnette91137f02016-03-10 16:52:26 -08001099 raise hosts.AutoservRepairError(
Scott Zawalski62bacae2013-03-05 10:40:32 -05001100 'DUT failed to boot from USB after %d seconds' %
beepsf079cfb2013-09-18 17:49:51 -07001101 usb_boot_timeout)
1102 timer.stop()
Scott Zawalski62bacae2013-03-05 10:40:32 -05001103
Tom Wai-Hong Tamf6b4f812015-08-08 04:14:59 +08001104 # The new chromeos-tpm-recovery has been merged since R44-7073.0.0.
1105 # In old CrOS images, this command fails. Skip the error.
Tom Wai-Hong Tam27af7332015-07-25 06:09:39 +08001106 logging.info('Resetting the TPM status')
Tom Wai-Hong Tamf6b4f812015-08-08 04:14:59 +08001107 try:
1108 self.run('chromeos-tpm-recovery')
1109 except error.AutoservRunError:
1110 logging.warn('chromeos-tpm-recovery is too old.')
1111
Tom Wai-Hong Tam27af7332015-07-25 06:09:39 +08001112
beepsf079cfb2013-09-18 17:49:51 -07001113 install_timer_key = ('servo_install.install_timeout_%s'
1114 % install_timeout)
Gabe Black1e1c41b2015-02-04 23:55:15 -08001115 timer = autotest_stats.Timer(install_timer_key)
beepsf079cfb2013-09-18 17:49:51 -07001116 timer.start()
1117 logging.info('Installing image through chromeos-install.')
J. Richard Barnette9af19632015-09-25 12:18:03 -07001118 self.run('chromeos-install --yes', timeout=install_timeout)
1119 self.halt()
beepsf079cfb2013-09-18 17:49:51 -07001120 timer.stop()
1121
1122 logging.info('Power cycling DUT through servo.')
J. Richard Barnette0199cc82014-12-05 17:08:40 -08001123 self.servo.get_power_state_controller().power_off()
Fang Dengafb88142013-05-30 17:44:31 -07001124 self.servo.switch_usbkey('off')
J. Richard Barnette0199cc82014-12-05 17:08:40 -08001125 # N.B. The Servo API requires that we use power_on() here
1126 # for two reasons:
1127 # 1) After turning on a DUT in recovery mode, you must turn
1128 # it off and then on with power_on() once more to
1129 # disable recovery mode (this is a Parrot specific
1130 # requirement).
1131 # 2) After power_off(), the only way to turn on is with
1132 # power_on() (this is a Storm specific requirement).
J. Richard Barnettefbcc7122013-07-24 18:24:59 -07001133 self.servo.get_power_state_controller().power_on()
beepsf079cfb2013-09-18 17:49:51 -07001134
1135 logging.info('Waiting for DUT to come back up.')
Richard Barnette03a0c132012-11-05 12:40:35 -08001136 if not self.wait_up(timeout=self.BOOT_TIMEOUT):
1137 raise error.AutoservError('DUT failed to reboot installed '
1138 'test image after %d seconds' %
Scott Zawalski62bacae2013-03-05 10:40:32 -05001139 self.BOOT_TIMEOUT)
1140
1141
Tom Wai-Hong Tambea741c2016-01-21 07:20:14 +08001142 def _is_firmware_repair_supported(self):
1143 """Check if the firmware repair is supported.
Dan Shi3d7a0e12015-10-12 11:55:45 -07001144
Tom Wai-Hong Tambea741c2016-01-21 07:20:14 +08001145 The firmware repair is only applicable to DUTs in pools listed in
1146 global config CROS/pools_support_firmware_repair.
Dan Shi3d7a0e12015-10-12 11:55:45 -07001147
Tom Wai-Hong Tambea741c2016-01-21 07:20:14 +08001148 @return: True if it is supported; otherwise False.
Dan Shi3d7a0e12015-10-12 11:55:45 -07001149 """
1150 logging.info('Checking if host %s can be repaired with firmware '
1151 'repair.', self.hostname)
1152 pools = server_utils.get_labels_from_afe(self.hostname, 'pool:',
1153 self._AFE)
1154 pools_support_firmware_repair = CONFIG.get_config_value('CROS',
1155 'pools_support_firmware_repair', type=str).split(',')
Tom Wai-Hong Tambea741c2016-01-21 07:20:14 +08001156
1157 return (pools and pools_support_firmware_repair and
1158 set(pools).intersection(set(pools_support_firmware_repair)))
1159
1160
Richard Barnette9a26ad62016-06-10 12:03:08 -07001161 def repair_servo(self):
Dan Shi90466352015-09-22 15:01:05 -07001162 """
Richard Barnette9a26ad62016-06-10 12:03:08 -07001163 Confirm that servo is initialized and verified.
Dan Shi90466352015-09-22 15:01:05 -07001164
Richard Barnette9a26ad62016-06-10 12:03:08 -07001165 If the servo object is missing, attempt to repair the servo
1166 host. Repair failures are passed back to the caller.
1167
1168 @raise AutoservError: If there is no servo host for this CrOS
1169 host.
1170 """
1171 if self.servo:
1172 return
1173 if not self._servo_host:
1174 raise error.AutoservError('No servo host for %s.' %
1175 self.hostname)
1176 self._servo_host.repair()
1177 self.servo = self._servo_host.get_servo()
Dan Shi90466352015-09-22 15:01:05 -07001178
1179
J. Richard Barnettec2d99cf2015-11-18 12:46:15 -08001180 def repair(self):
1181 """Attempt to get the DUT to pass `self.verify()`.
Richard Barnette82c35912012-11-20 10:09:10 -08001182
1183 This overrides the base class function for repair; it does
J. Richard Barnette91137f02016-03-10 16:52:26 -08001184 not call back to the parent class, but instead relies on
1185 `self._repair_strategy` to coordinate the verification and
1186 repair steps needed to get the DUT working.
Richard Barnette82c35912012-11-20 10:09:10 -08001187 """
J. Richard Barnette91137f02016-03-10 16:52:26 -08001188 self._repair_strategy.repair(self)
Scott Zawalski89c44dd2013-02-26 09:28:02 -05001189
Richard Barnette82c35912012-11-20 10:09:10 -08001190
J. Richard Barnette1d78b012012-05-15 13:56:30 -07001191 def close(self):
Fang Deng0ca40e22013-08-27 17:47:44 -07001192 super(CrosHost, self).close()
J. Richard Barnette1d78b012012-05-15 13:56:30 -07001193
1194
Dan Shi49ca0932014-11-14 11:22:27 -08001195 def get_power_supply_info(self):
1196 """Get the output of power_supply_info.
1197
1198 power_supply_info outputs the info of each power supply, e.g.,
1199 Device: Line Power
1200 online: no
1201 type: Mains
1202 voltage (V): 0
1203 current (A): 0
1204 Device: Battery
1205 state: Discharging
1206 percentage: 95.9276
1207 technology: Li-ion
1208
1209 Above output shows two devices, Line Power and Battery, with details of
1210 each device listed. This function parses the output into a dictionary,
1211 with key being the device name, and value being a dictionary of details
1212 of the device info.
1213
1214 @return: The dictionary of power_supply_info, e.g.,
1215 {'Line Power': {'online': 'yes', 'type': 'main'},
1216 'Battery': {'vendor': 'xyz', 'percentage': '100'}}
Dan Shie9b765d2014-12-29 16:59:49 -08001217 @raise error.AutoservRunError if power_supply_info tool is not found in
1218 the DUT. Caller should handle this error to avoid false failure
1219 on verification.
Dan Shi49ca0932014-11-14 11:22:27 -08001220 """
1221 result = self.run('power_supply_info').stdout.strip()
1222 info = {}
1223 device_name = None
1224 device_info = {}
1225 for line in result.split('\n'):
1226 pair = [v.strip() for v in line.split(':')]
1227 if len(pair) != 2:
1228 continue
1229 if pair[0] == 'Device':
1230 if device_name:
1231 info[device_name] = device_info
1232 device_name = pair[1]
1233 device_info = {}
1234 else:
1235 device_info[pair[0]] = pair[1]
1236 if device_name and not device_name in info:
1237 info[device_name] = device_info
1238 return info
1239
1240
1241 def get_battery_percentage(self):
1242 """Get the battery percentage.
1243
1244 @return: The percentage of battery level, value range from 0-100. Return
1245 None if the battery info cannot be retrieved.
1246 """
1247 try:
1248 info = self.get_power_supply_info()
1249 logging.info(info)
1250 return float(info['Battery']['percentage'])
Dan Shie9b765d2014-12-29 16:59:49 -08001251 except (KeyError, ValueError, error.AutoservRunError):
Dan Shi49ca0932014-11-14 11:22:27 -08001252 return None
1253
1254
1255 def is_ac_connected(self):
1256 """Check if the dut has power adapter connected and charging.
1257
1258 @return: True if power adapter is connected and charging.
1259 """
1260 try:
1261 info = self.get_power_supply_info()
1262 return info['Line Power']['online'] == 'yes'
Dan Shie9b765d2014-12-29 16:59:49 -08001263 except (KeyError, error.AutoservRunError):
1264 return None
Dan Shi49ca0932014-11-14 11:22:27 -08001265
1266
Simran Basi5e6339a2013-03-21 11:34:32 -07001267 def _cleanup_poweron(self):
1268 """Special cleanup method to make sure hosts always get power back."""
1269 afe = frontend_wrappers.RetryingAFE(timeout_min=5, delay_sec=10)
1270 hosts = afe.get_hosts(hostname=self.hostname)
1271 if not hosts or not (self._RPM_OUTLET_CHANGED in
1272 hosts[0].attributes):
1273 return
1274 logging.debug('This host has recently interacted with the RPM'
1275 ' Infrastructure. Ensuring power is on.')
1276 try:
1277 self.power_on()
Dan Shi7dca56e2014-11-11 17:07:56 -08001278 afe.set_host_attribute(self._RPM_OUTLET_CHANGED, None,
1279 hostname=self.hostname)
Simran Basi5e6339a2013-03-21 11:34:32 -07001280 except rpm_client.RemotePowerException:
Simran Basi5e6339a2013-03-21 11:34:32 -07001281 logging.error('Failed to turn Power On for this host after '
1282 'cleanup through the RPM Infrastructure.')
Gabe Blackb72f4fb2015-01-20 16:47:13 -08001283 autotest_es.post(
Dan Shi7dca56e2014-11-11 17:07:56 -08001284 type_str='RPM_poweron_failure',
1285 metadata={'hostname': self.hostname})
Dan Shi49ca0932014-11-14 11:22:27 -08001286
1287 battery_percentage = self.get_battery_percentage()
Dan Shif01ebe22014-12-05 13:10:57 -08001288 if battery_percentage and battery_percentage < 50:
Dan Shi49ca0932014-11-14 11:22:27 -08001289 raise
1290 elif self.is_ac_connected():
1291 logging.info('The device has power adapter connected and '
1292 'charging. No need to try to turn RPM on '
1293 'again.')
1294 afe.set_host_attribute(self._RPM_OUTLET_CHANGED, None,
1295 hostname=self.hostname)
1296 logging.info('Battery level is now at %s%%. The device may '
1297 'still have enough power to run test, so no '
1298 'exception will be raised.', battery_percentage)
1299
Simran Basi5e6339a2013-03-21 11:34:32 -07001300
beepsc87ff602013-07-31 21:53:00 -07001301 def _is_factory_image(self):
1302 """Checks if the image on the DUT is a factory image.
1303
1304 @return: True if the image on the DUT is a factory image.
1305 False otherwise.
1306 """
1307 result = self.run('[ -f /root/.factory_test ]', ignore_status=True)
1308 return result.exit_status == 0
1309
1310
1311 def _restart_ui(self):
J. Richard Barnette84890bd2014-02-21 11:05:47 -08001312 """Restart the Chrome UI.
beepsc87ff602013-07-31 21:53:00 -07001313
1314 @raises: FactoryImageCheckerException for factory images, since
1315 we cannot attempt to restart ui on them.
1316 error.AutoservRunError for any other type of error that
1317 occurs while restarting ui.
1318 """
1319 if self._is_factory_image():
Dan Shi549fb822015-03-24 18:01:11 -07001320 raise FactoryImageCheckerException('Cannot restart ui on factory '
1321 'images')
beepsc87ff602013-07-31 21:53:00 -07001322
J. Richard Barnette84890bd2014-02-21 11:05:47 -08001323 # TODO(jrbarnette): The command to stop/start the ui job
1324 # should live inside cros_ui, too. However that would seem
1325 # to imply interface changes to the existing start()/restart()
1326 # functions, which is a bridge too far (for now).
J. Richard Barnette6069aa12015-06-08 09:10:24 -07001327 prompt = cros_ui.get_chrome_session_ident(self)
J. Richard Barnette84890bd2014-02-21 11:05:47 -08001328 self.run('stop ui; start ui')
1329 cros_ui.wait_for_chrome_ready(prompt, self)
beepsc87ff602013-07-31 21:53:00 -07001330
1331
Dan Shi549fb822015-03-24 18:01:11 -07001332 def get_release_version(self):
1333 """Get the value of attribute CHROMEOS_RELEASE_VERSION from lsb-release.
1334
1335 @returns The version string in lsb-release, under attribute
1336 CHROMEOS_RELEASE_VERSION.
1337 """
1338 lsb_release_content = self.run(
1339 'cat "%s"' % client_constants.LSB_RELEASE).stdout.strip()
1340 return lsbrelease_utils.get_chromeos_release_version(
1341 lsb_release_content=lsb_release_content)
1342
1343
1344 def verify_cros_version_label(self):
1345 """ Make sure host's cros-version label match the actual image in dut.
1346
1347 Remove any cros-version: label that doesn't match that installed in
1348 the dut.
1349
1350 @param raise_error: Set to True to raise exception if any mismatch found
1351
1352 @raise error.AutoservError: If any mismatch between cros-version label
1353 and the build installed in dut is found.
1354 """
1355 labels = self._AFE.get_labels(
1356 name__startswith=ds_constants.VERSION_PREFIX,
1357 host__hostname=self.hostname)
1358 mismatch_found = False
1359 if labels:
1360 # Get CHROMEOS_RELEASE_VERSION from lsb-release, e.g., 6908.0.0.
1361 # Note that it's different from cros-version label, which has
1362 # builder and branch info, e.g.,
1363 # cros-version:peppy-release/R43-6908.0.0
1364 release_version = self.get_release_version()
1365 host_list = [self.hostname]
1366 for label in labels:
1367 # Remove any cros-version label that does not match
1368 # release_version.
1369 build_version = label.name[len(ds_constants.VERSION_PREFIX):]
1370 if not utils.version_match(build_version, release_version):
1371 logging.warn('cros-version label "%s" does not match '
1372 'release version %s. Removing the label.',
1373 label.name, release_version)
1374 label.remove_hosts(hosts=host_list)
1375 mismatch_found = True
1376 if mismatch_found:
Dan Shi1057bae2015-03-30 11:35:09 -07001377 autotest_es.post(use_http=True,
1378 type_str='cros_version_label_mismatch',
1379 metadata={'hostname': self.hostname})
Dan Shi549fb822015-03-24 18:01:11 -07001380 raise error.AutoservError('The host has wrong cros-version label.')
1381
1382
Tom Wai-Hong Tambea741c2016-01-21 07:20:14 +08001383 def verify_firmware_status(self):
1384 """Verify the host's firmware is in a good state.
1385
1386 @raise error.AutoservError: If state is not good.
1387 """
1388 if self._is_firmware_repair_supported():
Tom Wai-Hong Tam93a29562016-01-23 04:16:53 +08001389 try:
1390 # Read the AP firmware and dump the sections we are interested.
1391 cmd = ('mkdir /tmp/verify_firmware; '
1392 'cd /tmp/verify_firmware; '
1393 'for section in VBLOCK_A VBLOCK_B FW_MAIN_A FW_MAIN_B; '
1394 'do flashrom -r image.bin -i $section:$section; '
1395 'done')
1396 self.run(cmd)
1397
1398 # Verify the firmware blocks A and B.
1399 cmd = ('vbutil_firmware --verify /tmp/verify_firmware/VBLOCK_%c'
1400 ' --signpubkey /usr/share/vboot/devkeys/root_key.vbpubk'
1401 ' --fv /tmp/verify_firmware/FW_MAIN_%c')
1402 for c in ('A', 'B'):
1403 rv = self.run(cmd % (c, c), ignore_status=True)
1404 if rv.exit_status:
Dan Shic67f1332016-04-06 12:38:06 -07001405 raise error.AutoservError(
1406 'Firmware %c is in a bad state.' % c)
Tom Wai-Hong Tam93a29562016-01-23 04:16:53 +08001407 finally:
1408 # Remove the tempoary files.
1409 self.run('rm -rf /tmp/verify_firmware')
Tom Wai-Hong Tambea741c2016-01-21 07:20:14 +08001410 else:
1411 logging.info('Do not care about firmware status when the host '
1412 'is not in pools that support firmware repair.')
1413
1414
beepsc87ff602013-07-31 21:53:00 -07001415 def cleanup(self):
MK Ryu35d661e2014-09-25 17:44:10 -07001416 self.run('rm -f %s' % client_constants.CLEANUP_LOGS_PAUSED_FILE)
Scott Zawalskiddbc31e2012-11-15 11:29:01 -05001417 try:
beepsc87ff602013-07-31 21:53:00 -07001418 self._restart_ui()
1419 except (error.AutotestRunError, error.AutoservRunError,
1420 FactoryImageCheckerException):
Ilja H. Friedel04be2bd2014-05-07 21:29:59 -07001421 logging.warning('Unable to restart ui, rebooting device.')
Scott Zawalskiddbc31e2012-11-15 11:29:01 -05001422 # Since restarting the UI fails fall back to normal Autotest
1423 # cleanup routines, i.e. reboot the machine.
Fang Deng0ca40e22013-08-27 17:47:44 -07001424 super(CrosHost, self).cleanup()
Simran Basi5e6339a2013-03-21 11:34:32 -07001425 # Check if the rpm outlet was manipulated.
Simran Basid5e5e272012-09-24 15:23:59 -07001426 if self.has_power():
Simran Basi5e6339a2013-03-21 11:34:32 -07001427 self._cleanup_poweron()
Dan Shi549fb822015-03-24 18:01:11 -07001428 self.verify_cros_version_label()
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001429
1430
Yu-Ju Honga2be94a2012-07-31 09:48:52 -07001431 def reboot(self, **dargs):
1432 """
1433 This function reboots the site host. The more generic
1434 RemoteHost.reboot() performs sync and sleeps for 5
1435 seconds. This is not necessary for Chrome OS devices as the
1436 sync should be finished in a short time during the reboot
1437 command.
1438 """
Tom Wai-Hong Tamf5cd1d42012-08-13 12:04:08 +08001439 if 'reboot_cmd' not in dargs:
Doug Anderson7d5aeb22014-02-27 15:12:17 -08001440 reboot_timeout = dargs.get('reboot_timeout', 10)
J. Richard Barnette9af19632015-09-25 12:18:03 -07001441 dargs['reboot_cmd'] = ('sleep 1; '
1442 'reboot & sleep %d; '
1443 'reboot -f' % reboot_timeout)
Yu-Ju Honga2be94a2012-07-31 09:48:52 -07001444 # Enable fastsync to avoid running extra sync commands before reboot.
Tom Wai-Hong Tamf5cd1d42012-08-13 12:04:08 +08001445 if 'fastsync' not in dargs:
1446 dargs['fastsync'] = True
Michael Liangda8c60a2014-06-03 13:24:51 -07001447
Charlie Mooneya8e6dab2014-05-29 14:37:55 -07001448 # For purposes of logging reboot times:
1449 # Get the board name i.e. 'daisy_spring'
Michael Liangca4f5a62014-07-10 15:45:13 -07001450 board_fullname = self.get_board()
1451
1452 # Strip the prefix and add it to dargs.
1453 dargs['board'] = board_fullname[board_fullname.find(':')+1:]
Vincent Palatindf2372c2016-10-07 17:03:00 +02001454 # Record who called us
1455 orig = sys._getframe(1).f_code
Vincent Palatin80780b22016-07-27 16:02:37 +02001456 metric_fields = {'board' : dargs['board'],
Vincent Palatindf2372c2016-10-07 17:03:00 +02001457 'dut_host_name' : self.hostname,
1458 'success' : True}
1459 metric_debug_fields = {'board' : dargs['board'],
1460 'caller' : "%s:%s" % (orig.co_filename, orig.co_name),
1461 'success' : True,
1462 'error' : ''}
1463
Vincent Palatin80780b22016-07-27 16:02:37 +02001464 t0 = time.time()
1465 try:
1466 super(CrosHost, self).reboot(**dargs)
1467 except Exception as e:
1468 metric_fields['success'] = False
Vincent Palatindf2372c2016-10-07 17:03:00 +02001469 metric_debug_fields['success'] = False
1470 metric_debug_fields['error'] = type(e).__name__
Vincent Palatin80780b22016-07-27 16:02:37 +02001471 raise
1472 finally:
1473 duration = int(time.time() - t0)
Simran Basi382506b2016-09-13 14:58:15 -07001474 if metrics:
1475 metrics.Counter(
1476 'chromeos/autotest/autoserv/reboot_count').increment(
1477 fields=metric_fields)
Vincent Palatindf2372c2016-10-07 17:03:00 +02001478 metrics.Counter(
1479 'chromeos/autotest/autoserv/reboot_debug').increment(
1480 fields=metric_debug_fields)
Simran Basi382506b2016-09-13 14:58:15 -07001481 metrics.SecondsDistribution(
1482 'chromeos/autotest/autoserv/reboot_duration').add(
1483 duration, fields=metric_fields)
Yu-Ju Honga2be94a2012-07-31 09:48:52 -07001484
1485
Gwendal Grignou7a61d2f2014-05-23 11:05:51 -07001486 def suspend(self, **dargs):
1487 """
1488 This function suspends the site host.
1489 """
1490 suspend_time = dargs.get('suspend_time', 60)
1491 dargs['timeout'] = suspend_time
1492 if 'suspend_cmd' not in dargs:
J. Richard Barnette9af19632015-09-25 12:18:03 -07001493 dargs['suspend_cmd'] = ' && '.join([
1494 'echo 0 > /sys/class/rtc/rtc0/wakealarm',
Gwendal Grignou7a61d2f2014-05-23 11:05:51 -07001495 'echo +%d > /sys/class/rtc/rtc0/wakealarm' % suspend_time,
J. Richard Barnette9af19632015-09-25 12:18:03 -07001496 'powerd_dbus_suspend --delay=0'])
Gwendal Grignou7a61d2f2014-05-23 11:05:51 -07001497 super(CrosHost, self).suspend(**dargs)
1498
1499
Simran Basiec564392014-08-25 16:48:09 -07001500 def upstart_status(self, service_name):
1501 """Check the status of an upstart init script.
1502
1503 @param service_name: Service to look up.
1504
1505 @returns True if the service is running, False otherwise.
1506 """
1507 return self.run('status %s | grep start/running' %
1508 service_name).stdout.strip() != ''
1509
1510
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001511 def verify_software(self):
Richard Barnetteb2bc13c2013-01-08 17:32:51 -08001512 """Verify working software on a Chrome OS system.
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001513
Richard Barnetteb2bc13c2013-01-08 17:32:51 -08001514 Tests for the following conditions:
1515 1. All conditions tested by the parent version of this
1516 function.
1517 2. Sufficient space in /mnt/stateful_partition.
Fang Deng6b05f5b2013-03-20 13:42:11 -07001518 3. Sufficient space in /mnt/stateful_partition/encrypted.
1519 4. update_engine answers a simple status request over DBus.
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001520
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001521 """
Fang Deng0ca40e22013-08-27 17:47:44 -07001522 super(CrosHost, self).verify_software()
Dan Shib8540a52015-07-16 14:18:23 -07001523 default_kilo_inodes_required = CONFIG.get_config_value(
1524 'SERVER', 'kilo_inodes_required', type=int, default=100)
1525 board = self.get_board().replace(ds_constants.BOARD_PREFIX, '')
1526 kilo_inodes_required = CONFIG.get_config_value(
1527 'SERVER', 'kilo_inodes_required_%s' % board,
1528 type=int, default=default_kilo_inodes_required)
1529 self.check_inodes('/mnt/stateful_partition', kilo_inodes_required)
J. Richard Barnette45e93de2012-04-11 17:24:15 -07001530 self.check_diskspace(
1531 '/mnt/stateful_partition',
Dan Shib8540a52015-07-16 14:18:23 -07001532 CONFIG.get_config_value(
Fang Deng6b05f5b2013-03-20 13:42:11 -07001533 'SERVER', 'gb_diskspace_required', type=float,
1534 default=20.0))
Gaurav Shahe448af82014-06-19 15:18:59 -07001535 encrypted_stateful_path = '/mnt/stateful_partition/encrypted'
1536 # Not all targets build with encrypted stateful support.
1537 if self.path_exists(encrypted_stateful_path):
1538 self.check_diskspace(
1539 encrypted_stateful_path,
Dan Shib8540a52015-07-16 14:18:23 -07001540 CONFIG.get_config_value(
Gaurav Shahe448af82014-06-19 15:18:59 -07001541 'SERVER', 'gb_encrypted_diskspace_required', type=float,
1542 default=0.1))
beepsc87ff602013-07-31 21:53:00 -07001543
Simran Basiec564392014-08-25 16:48:09 -07001544 if not self.upstart_status('system-services'):
Prashanth B5d0a0512014-04-25 12:26:08 -07001545 raise error.AutoservError('Chrome failed to reach login. '
1546 'System services not running.')
1547
beepsc87ff602013-07-31 21:53:00 -07001548 # Factory images don't run update engine,
1549 # goofy controls dbus on these DUTs.
1550 if not self._is_factory_image():
1551 self.run('update_engine_client --status')
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001552
Dan Shi549fb822015-03-24 18:01:11 -07001553 self.verify_cros_version_label()
1554
Tom Wai-Hong Tambea741c2016-01-21 07:20:14 +08001555 self.verify_firmware_status()
1556
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001557
J. Richard Barnettea7e7fdc2016-02-12 12:35:36 -08001558 def verify(self):
1559 self._repair_strategy.verify(self)
1560
1561
Fang Deng96667ca2013-08-01 17:46:18 -07001562 def make_ssh_command(self, user='root', port=22, opts='', hosts_file=None,
1563 connect_timeout=None, alive_interval=None):
1564 """Override default make_ssh_command to use options tuned for Chrome OS.
1565
1566 Tuning changes:
1567 - ConnectTimeout=30; maximum of 30 seconds allowed for an SSH
1568 connection failure. Consistency with remote_access.sh.
1569
Samuel Tan2ce155b2015-06-23 18:24:38 -07001570 - ServerAliveInterval=900; which causes SSH to ping connection every
1571 900 seconds. In conjunction with ServerAliveCountMax ensures
1572 that if the connection dies, Autotest will bail out.
Fang Deng96667ca2013-08-01 17:46:18 -07001573 Originally tried 60 secs, but saw frequent job ABORTS where
Samuel Tan2ce155b2015-06-23 18:24:38 -07001574 the test completed successfully. Later increased from 180 seconds to
1575 900 seconds to account for tests where the DUT is suspended for
1576 longer periods of time.
Fang Deng96667ca2013-08-01 17:46:18 -07001577
1578 - ServerAliveCountMax=3; consistency with remote_access.sh.
1579
1580 - ConnectAttempts=4; reduce flakiness in connection errors;
1581 consistency with remote_access.sh.
1582
1583 - UserKnownHostsFile=/dev/null; we don't care about the keys.
1584 Host keys change with every new installation, don't waste
1585 memory/space saving them.
1586
1587 - SSH protocol forced to 2; needed for ServerAliveInterval.
1588
1589 @param user User name to use for the ssh connection.
1590 @param port Port on the target host to use for ssh connection.
1591 @param opts Additional options to the ssh command.
1592 @param hosts_file Ignored.
1593 @param connect_timeout Ignored.
1594 @param alive_interval Ignored.
1595 """
Aviv Keshetc5947fa2013-09-04 14:06:29 -07001596 base_command = ('/usr/bin/ssh -a -x %s %s %s'
1597 ' -o StrictHostKeyChecking=no'
Fang Deng96667ca2013-08-01 17:46:18 -07001598 ' -o UserKnownHostsFile=/dev/null -o BatchMode=yes'
Samuel Tan2ce155b2015-06-23 18:24:38 -07001599 ' -o ConnectTimeout=30 -o ServerAliveInterval=900'
Fang Deng96667ca2013-08-01 17:46:18 -07001600 ' -o ServerAliveCountMax=3 -o ConnectionAttempts=4'
1601 ' -o Protocol=2 -l %s -p %d')
Aviv Keshetc5947fa2013-09-04 14:06:29 -07001602 return base_command % (self._ssh_verbosity_flag, self._ssh_options,
1603 opts, user, port)
Jason Abeleb6f924f2013-11-13 16:01:54 -08001604 def syslog(self, message, tag='autotest'):
1605 """Logs a message to syslog on host.
1606
1607 @param message String message to log into syslog
1608 @param tag String tag prefix for syslog
1609
1610 """
1611 self.run('logger -t "%s" "%s"' % (tag, message))
1612
1613
J. Richard Barnetteb6de7e32013-02-14 13:28:04 -08001614 def _ping_check_status(self, status):
1615 """Ping the host once, and return whether it has a given status.
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001616
J. Richard Barnetteb6de7e32013-02-14 13:28:04 -08001617 @param status Check the ping status against this value.
1618 @return True iff `status` and the result of ping are the same
1619 (i.e. both True or both False).
1620
1621 """
1622 ping_val = utils.ping(self.hostname, tries=1, deadline=1)
1623 return not (status ^ (ping_val == 0))
1624
1625 def _ping_wait_for_status(self, status, timeout):
1626 """Wait for the host to have a given status (UP or DOWN).
1627
1628 Status is checked by polling. Polling will not last longer
1629 than the number of seconds in `timeout`. The polling
1630 interval will be long enough that only approximately
1631 _PING_WAIT_COUNT polling cycles will be executed, subject
1632 to a maximum interval of about one minute.
1633
1634 @param status Waiting will stop immediately if `ping` of the
1635 host returns this status.
1636 @param timeout Poll for at most this many seconds.
1637 @return True iff the host status from `ping` matched the
1638 requested status at the time of return.
1639
1640 """
1641 # _ping_check_status() takes about 1 second, hence the
1642 # "- 1" in the formula below.
Nathan Ciobanu38480a32016-10-25 15:26:45 -07001643 # FIXME: if the ping command errors then _ping_check_status()
1644 # returns instantly. If timeout is also smaller than twice
1645 # _PING_WAIT_COUNT then the while loop below forks many
1646 # thousands of ping commands (see /tmp/test_that_results_XXXXX/
1647 # /results-1-logging_YYY.ZZZ/debug/autoserv.DEBUG) and hogs one
1648 # CPU core for 60 seconds.
J. Richard Barnetteb6de7e32013-02-14 13:28:04 -08001649 poll_interval = min(int(timeout / self._PING_WAIT_COUNT), 60) - 1
1650 end_time = time.time() + timeout
1651 while time.time() <= end_time:
1652 if self._ping_check_status(status):
1653 return True
1654 if poll_interval > 0:
1655 time.sleep(poll_interval)
1656
1657 # The last thing we did was sleep(poll_interval), so it may
1658 # have been too long since the last `ping`. Check one more
1659 # time, just to be sure.
1660 return self._ping_check_status(status)
1661
1662 def ping_wait_up(self, timeout):
1663 """Wait for the host to respond to `ping`.
1664
1665 N.B. This method is not a reliable substitute for
1666 `wait_up()`, because a host that responds to ping will not
1667 necessarily respond to ssh. This method should only be used
1668 if the target DUT can be considered functional even if it
1669 can't be reached via ssh.
1670
1671 @param timeout Minimum time to allow before declaring the
1672 host to be non-responsive.
1673 @return True iff the host answered to ping before the timeout.
1674
1675 """
1676 return self._ping_wait_for_status(self._PING_STATUS_UP, timeout)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001677
Andrew Bresticker678c0c72013-01-22 10:44:09 -08001678 def ping_wait_down(self, timeout):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001679 """Wait until the host no longer responds to `ping`.
1680
J. Richard Barnetteb6de7e32013-02-14 13:28:04 -08001681 This function can be used as a slightly faster version of
1682 `wait_down()`, by avoiding potentially long ssh timeouts.
1683
1684 @param timeout Minimum time to allow for the host to become
1685 non-responsive.
1686 @return True iff the host quit answering ping before the
1687 timeout.
1688
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001689 """
J. Richard Barnetteb6de7e32013-02-14 13:28:04 -08001690 return self._ping_wait_for_status(self._PING_STATUS_DOWN, timeout)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001691
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08001692 def test_wait_for_sleep(self, sleep_timeout=None):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001693 """Wait for the client to enter low-power sleep mode.
1694
1695 The test for "is asleep" can't distinguish a system that is
1696 powered off; to confirm that the unit was asleep, it is
1697 necessary to force resume, and then call
1698 `test_wait_for_resume()`.
1699
1700 This function is expected to be called from a test as part
1701 of a sequence like the following:
1702
1703 ~~~~~~~~
1704 boot_id = host.get_boot_id()
1705 # trigger sleep on the host
1706 host.test_wait_for_sleep()
1707 # trigger resume on the host
1708 host.test_wait_for_resume(boot_id)
1709 ~~~~~~~~
1710
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08001711 @param sleep_timeout time limit in seconds to allow the host sleep.
1712
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001713 @exception TestFail The host did not go to sleep within
1714 the allowed time.
1715 """
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08001716 if sleep_timeout is None:
1717 sleep_timeout = self.SLEEP_TIMEOUT
1718
1719 if not self.ping_wait_down(timeout=sleep_timeout):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001720 raise error.TestFail(
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08001721 'client failed to sleep after %d seconds' % sleep_timeout)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001722
1723
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08001724 def test_wait_for_resume(self, old_boot_id, resume_timeout=None):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001725 """Wait for the client to resume from low-power sleep mode.
1726
1727 The `old_boot_id` parameter should be the value from
1728 `get_boot_id()` obtained prior to entering sleep mode. A
1729 `TestFail` exception is raised if the boot id changes.
1730
1731 See @ref test_wait_for_sleep for more on this function's
1732 usage.
1733
J. Richard Barnette7214e0b2013-02-06 15:20:49 -08001734 @param old_boot_id A boot id value obtained before the
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001735 target host went to sleep.
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08001736 @param resume_timeout time limit in seconds to allow the host up.
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001737
1738 @exception TestFail The host did not respond within the
1739 allowed time.
1740 @exception TestFail The host responded, but the boot id test
1741 indicated a reboot rather than a sleep
1742 cycle.
1743 """
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08001744 if resume_timeout is None:
1745 resume_timeout = self.RESUME_TIMEOUT
1746
1747 if not self.wait_up(timeout=resume_timeout):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001748 raise error.TestFail(
1749 'client failed to resume from sleep after %d seconds' %
Tom Wai-Hong Tamfced4f62014-04-17 10:56:30 +08001750 resume_timeout)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001751 else:
1752 new_boot_id = self.get_boot_id()
1753 if new_boot_id != old_boot_id:
Tom Wai-Hong Tam01792682015-01-06 08:00:46 +08001754 logging.error('client rebooted (old boot %s, new boot %s)',
1755 old_boot_id, new_boot_id)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001756 raise error.TestFail(
Tom Wai-Hong Tam01792682015-01-06 08:00:46 +08001757 'client rebooted, but sleep was expected')
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001758
1759
Tom Wai-Hong Tamfe005c22014-12-03 09:25:44 +08001760 def test_wait_for_shutdown(self, shutdown_timeout=None):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001761 """Wait for the client to shut down.
1762
1763 The test for "has shut down" can't distinguish a system that
1764 is merely asleep; to confirm that the unit was down, it is
1765 necessary to force boot, and then call test_wait_for_boot().
1766
1767 This function is expected to be called from a test as part
1768 of a sequence like the following:
1769
1770 ~~~~~~~~
1771 boot_id = host.get_boot_id()
1772 # trigger shutdown on the host
1773 host.test_wait_for_shutdown()
1774 # trigger boot on the host
1775 host.test_wait_for_boot(boot_id)
1776 ~~~~~~~~
1777
Tom Wai-Hong Tamfe005c22014-12-03 09:25:44 +08001778 @param shutdown_timeout time limit in seconds to allow the host down.
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001779 @exception TestFail The host did not shut down within the
1780 allowed time.
1781 """
Tom Wai-Hong Tamfe005c22014-12-03 09:25:44 +08001782 if shutdown_timeout is None:
1783 shutdown_timeout = self.SHUTDOWN_TIMEOUT
1784
1785 if not self.ping_wait_down(timeout=shutdown_timeout):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001786 raise error.TestFail(
1787 'client failed to shut down after %d seconds' %
Tom Wai-Hong Tamfe005c22014-12-03 09:25:44 +08001788 shutdown_timeout)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001789
1790
1791 def test_wait_for_boot(self, old_boot_id=None):
1792 """Wait for the client to boot from cold power.
1793
1794 The `old_boot_id` parameter should be the value from
1795 `get_boot_id()` obtained prior to shutting down. A
1796 `TestFail` exception is raised if the boot id does not
1797 change. The boot id test is omitted if `old_boot_id` is not
1798 specified.
1799
1800 See @ref test_wait_for_shutdown for more on this function's
1801 usage.
1802
J. Richard Barnette7214e0b2013-02-06 15:20:49 -08001803 @param old_boot_id A boot id value obtained before the
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001804 shut down.
1805
1806 @exception TestFail The host did not respond within the
1807 allowed time.
1808 @exception TestFail The host responded, but the boot id test
1809 indicated that there was no reboot.
1810 """
J. Richard Barnetteeb69d722012-06-18 17:29:44 -07001811 if not self.wait_up(timeout=self.REBOOT_TIMEOUT):
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001812 raise error.TestFail(
1813 'client failed to reboot after %d seconds' %
J. Richard Barnetteeb69d722012-06-18 17:29:44 -07001814 self.REBOOT_TIMEOUT)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001815 elif old_boot_id:
1816 if self.get_boot_id() == old_boot_id:
Tom Wai-Hong Tam01792682015-01-06 08:00:46 +08001817 logging.error('client not rebooted (boot %s)',
1818 old_boot_id)
J. Richard Barnette134ec2c2012-04-25 12:59:37 -07001819 raise error.TestFail(
Tom Wai-Hong Tam01792682015-01-06 08:00:46 +08001820 'client is back up, but did not reboot')
Simran Basid5e5e272012-09-24 15:23:59 -07001821
1822
1823 @staticmethod
1824 def check_for_rpm_support(hostname):
1825 """For a given hostname, return whether or not it is powered by an RPM.
1826
Simran Basi1df55112013-09-06 11:25:09 -07001827 @param hostname: hostname to check for rpm support.
1828
Simran Basid5e5e272012-09-24 15:23:59 -07001829 @return None if this host does not follows the defined naming format
1830 for RPM powered DUT's in the lab. If it does follow the format,
1831 it returns a regular expression MatchObject instead.
1832 """
Fang Dengbaff9082015-01-06 13:46:15 -08001833 return re.match(CrosHost._RPM_HOSTNAME_REGEX, hostname)
Simran Basid5e5e272012-09-24 15:23:59 -07001834
1835
1836 def has_power(self):
1837 """For this host, return whether or not it is powered by an RPM.
1838
1839 @return True if this host is in the CROS lab and follows the defined
1840 naming format.
1841 """
Fang Deng0ca40e22013-08-27 17:47:44 -07001842 return CrosHost.check_for_rpm_support(self.hostname)
Simran Basid5e5e272012-09-24 15:23:59 -07001843
1844
Ismail Noorbasha07fdb612013-02-14 14:13:31 -08001845 def _set_power(self, state, power_method):
1846 """Sets the power to the host via RPM, Servo or manual.
1847
1848 @param state Specifies which power state to set to DUT
1849 @param power_method Specifies which method of power control to
1850 use. By default "RPM" will be used. Valid values
1851 are the strings "RPM", "manual", "servoj10".
1852
1853 """
1854 ACCEPTABLE_STATES = ['ON', 'OFF']
1855
1856 if state.upper() not in ACCEPTABLE_STATES:
1857 raise error.TestError('State must be one of: %s.'
1858 % (ACCEPTABLE_STATES,))
1859
1860 if power_method == self.POWER_CONTROL_SERVO:
1861 logging.info('Setting servo port J10 to %s', state)
1862 self.servo.set('prtctl3_pwren', state.lower())
1863 time.sleep(self._USB_POWER_TIMEOUT)
1864 elif power_method == self.POWER_CONTROL_MANUAL:
1865 logging.info('You have %d seconds to set the AC power to %s.',
1866 self._POWER_CYCLE_TIMEOUT, state)
1867 time.sleep(self._POWER_CYCLE_TIMEOUT)
1868 else:
1869 if not self.has_power():
1870 raise error.TestFail('DUT does not have RPM connected.')
Simran Basi5e6339a2013-03-21 11:34:32 -07001871 afe = frontend_wrappers.RetryingAFE(timeout_min=5, delay_sec=10)
1872 afe.set_host_attribute(self._RPM_OUTLET_CHANGED, True,
1873 hostname=self.hostname)
Simran Basi1df55112013-09-06 11:25:09 -07001874 rpm_client.set_power(self.hostname, state.upper(), timeout_mins=5)
Simran Basid5e5e272012-09-24 15:23:59 -07001875
1876
Ismail Noorbasha07fdb612013-02-14 14:13:31 -08001877 def power_off(self, power_method=POWER_CONTROL_RPM):
1878 """Turn off power to this host via RPM, Servo or manual.
1879
1880 @param power_method Specifies which method of power control to
1881 use. By default "RPM" will be used. Valid values
1882 are the strings "RPM", "manual", "servoj10".
1883
1884 """
1885 self._set_power('OFF', power_method)
Simran Basid5e5e272012-09-24 15:23:59 -07001886
1887
Ismail Noorbasha07fdb612013-02-14 14:13:31 -08001888 def power_on(self, power_method=POWER_CONTROL_RPM):
1889 """Turn on power to this host via RPM, Servo or manual.
1890
1891 @param power_method Specifies which method of power control to
1892 use. By default "RPM" will be used. Valid values
1893 are the strings "RPM", "manual", "servoj10".
1894
1895 """
1896 self._set_power('ON', power_method)
1897
1898
1899 def power_cycle(self, power_method=POWER_CONTROL_RPM):
1900 """Cycle power to this host by turning it OFF, then ON.
1901
1902 @param power_method Specifies which method of power control to
1903 use. By default "RPM" will be used. Valid values
1904 are the strings "RPM", "manual", "servoj10".
1905
1906 """
1907 if power_method in (self.POWER_CONTROL_SERVO,
1908 self.POWER_CONTROL_MANUAL):
1909 self.power_off(power_method=power_method)
1910 time.sleep(self._POWER_CYCLE_TIMEOUT)
1911 self.power_on(power_method=power_method)
1912 else:
1913 rpm_client.set_power(self.hostname, 'CYCLE')
Simran Basic6f1f7a2012-10-16 10:47:46 -07001914
1915
1916 def get_platform(self):
1917 """Determine the correct platform label for this host.
1918
1919 @returns a string representing this host's platform.
1920 """
1921 crossystem = utils.Crossystem(self)
1922 crossystem.init()
1923 # Extract fwid value and use the leading part as the platform id.
1924 # fwid generally follow the format of {platform}.{firmware version}
1925 # Example: Alex.X.YYY.Z or Google_Alex.X.YYY.Z
1926 platform = crossystem.fwid().split('.')[0].lower()
1927 # Newer platforms start with 'Google_' while the older ones do not.
1928 return platform.replace('google_', '')
1929
1930
Hung-ying Tyanb1328032014-04-01 14:18:54 +08001931 def get_architecture(self):
1932 """Determine the correct architecture label for this host.
1933
1934 @returns a string representing this host's architecture.
1935 """
1936 crossystem = utils.Crossystem(self)
1937 crossystem.init()
1938 return crossystem.arch()
1939
1940
Luis Lozano40b7d0d2014-01-17 15:12:06 -08001941 def get_chrome_version(self):
1942 """Gets the Chrome version number and milestone as strings.
1943
1944 Invokes "chrome --version" to get the version number and milestone.
1945
1946 @return A tuple (chrome_ver, milestone) where "chrome_ver" is the
1947 current Chrome version number as a string (in the form "W.X.Y.Z")
1948 and "milestone" is the first component of the version number
1949 (the "W" from "W.X.Y.Z"). If the version number cannot be parsed
1950 in the "W.X.Y.Z" format, the "chrome_ver" will be the full output
1951 of "chrome --version" and the milestone will be the empty string.
1952
1953 """
MK Ryu35d661e2014-09-25 17:44:10 -07001954 version_string = self.run(client_constants.CHROME_VERSION_COMMAND).stdout
Luis Lozano40b7d0d2014-01-17 15:12:06 -08001955 return utils.parse_chrome_version(version_string)
1956
J. Richard Barnetted2af5852016-02-05 15:03:10 -08001957
Kevin Chenga2619dc2016-03-28 11:42:08 -07001958 # TODO(kevcheng): change this to just return the board without the
1959 # 'board:' prefix and fix up all the callers. Also look into removing the
1960 # need for this method.
Simran Basic6f1f7a2012-10-16 10:47:46 -07001961 def get_board(self):
1962 """Determine the correct board label for this host.
1963
1964 @returns a string representing this host's board.
1965 """
1966 release_info = utils.parse_cmd_output('cat /etc/lsb-release',
1967 run_method=self.run)
J. Richard Barnetted2af5852016-02-05 15:03:10 -08001968 return (ds_constants.BOARD_PREFIX +
1969 release_info['CHROMEOS_RELEASE_BOARD'])
Simran Basic6f1f7a2012-10-16 10:47:46 -07001970
1971
Kevin Chenga328da62016-03-31 10:49:04 -07001972
1973 def has_lightsensor(self):
1974 """Determine the correct board label for this host.
1975
1976 @returns the string 'lightsensor' if this host has a lightsensor or
1977 None if it does not.
1978 """
1979 search_cmd = "find -L %s -maxdepth 4 | egrep '%s'" % (
1980 self._LIGHTSENSOR_SEARCH_DIR, '|'.join(self._LIGHTSENSOR_FILES))
1981 try:
1982 # Run the search cmd following the symlinks. Stderr_tee is set to
1983 # None as there can be a symlink loop, but the command will still
1984 # execute correctly with a few messages printed to stderr.
1985 self.run(search_cmd, stdout_tee=None, stderr_tee=None)
1986 return 'lightsensor'
1987 except error.AutoservRunError:
1988 # egrep exited with a return code of 1 meaning none of the possible
1989 # lightsensor files existed.
1990 return None
1991
1992
1993 def has_bluetooth(self):
1994 """Determine the correct board label for this host.
1995
1996 @returns the string 'bluetooth' if this host has bluetooth or
1997 None if it does not.
1998 """
1999 try:
2000 self.run('test -d /sys/class/bluetooth/hci0')
2001 # test exited with a return code of 0.
2002 return 'bluetooth'
2003 except error.AutoservRunError:
2004 # test exited with a return code 1 meaning the directory did not
2005 # exist.
2006 return None
2007
2008
Kevin Chenga328da62016-03-31 10:49:04 -07002009 def get_accels(self):
2010 """
2011 Determine the type of accelerometers on this host.
2012
2013 @returns a string representing this host's accelerometer type.
2014 At present, it only returns "accel:cros-ec", for accelerometers
2015 attached to a Chrome OS EC, or none, if no accelerometers.
2016 """
2017 # Check to make sure we have ectool
2018 rv = self.run('which ectool', ignore_status=True)
2019 if rv.exit_status:
2020 logging.info("No ectool cmd found, assuming no EC accelerometers")
2021 return None
2022
2023 # Check that the EC supports the motionsense command
2024 rv = self.run('ectool motionsense', ignore_status=True)
2025 if rv.exit_status:
2026 logging.info("EC does not support motionsense command "
2027 "assuming no EC accelerometers")
2028 return None
2029
2030 # Check that EC motion sensors are active
2031 active = self.run('ectool motionsense active').stdout.split('\n')
2032 if active[0] == "0":
2033 logging.info("Motion sense inactive, assuming no EC accelerometers")
2034 return None
2035
2036 logging.info("EC accelerometers found")
2037 return 'accel:cros-ec'
2038
2039
2040 def has_chameleon(self):
2041 """Determine if a Chameleon connected to this host.
2042
2043 @returns a list containing two strings ('chameleon' and
2044 'chameleon:' + label, e.g. 'chameleon:hdmi') if this host
2045 has a Chameleon or None if it has not.
2046 """
2047 if self._chameleon_host:
2048 return ['chameleon', 'chameleon:' + self.chameleon.get_label()]
2049 else:
2050 return None
2051
2052
2053 def has_loopback_dongle(self):
2054 """Determine if an audio loopback dongle is plugged to this host.
2055
2056 @returns 'audio_loopback_dongle' when there is an audio loopback dongle
2057 plugged to this host.
2058 None when there is no audio loopback dongle
2059 plugged to this host.
2060 """
2061 nodes_info = self.run(command=cras_utils.get_cras_nodes_cmd(),
2062 ignore_status=True).stdout
2063 if (cras_utils.node_type_is_plugged('HEADPHONE', nodes_info) and
2064 cras_utils.node_type_is_plugged('MIC', nodes_info)):
2065 return 'audio_loopback_dongle'
2066 else:
2067 return None
2068
2069
2070 def get_power_supply(self):
2071 """
2072 Determine what type of power supply the host has
2073
2074 @returns a string representing this host's power supply.
2075 'power:battery' when the device has a battery intended for
2076 extended use
2077 'power:AC_primary' when the device has a battery not intended
2078 for extended use (for moving the machine, etc)
2079 'power:AC_only' when the device has no battery at all.
2080 """
2081 psu = self.run(command='mosys psu type', ignore_status=True)
2082 if psu.exit_status:
2083 # The psu command for mosys is not included for all platforms. The
2084 # assumption is that the device will have a battery if the command
2085 # is not found.
2086 return 'power:battery'
2087
2088 psu_str = psu.stdout.strip()
2089 if psu_str == 'unknown':
2090 return None
2091
2092 return 'power:%s' % psu_str
2093
2094
2095 def get_storage(self):
2096 """
2097 Determine the type of boot device for this host.
2098
2099 Determine if the internal device is SCSI or dw_mmc device.
2100 Then check that it is SSD or HDD or eMMC or something else.
2101
2102 @returns a string representing this host's internal device type.
2103 'storage:ssd' when internal device is solid state drive
2104 'storage:hdd' when internal device is hard disk drive
2105 'storage:mmc' when internal device is mmc drive
2106 None When internal device is something else or
2107 when we are unable to determine the type
2108 """
2109 # The output should be /dev/mmcblk* for SD/eMMC or /dev/sd* for scsi
2110 rootdev_cmd = ' '.join(['. /usr/sbin/write_gpt.sh;',
2111 '. /usr/share/misc/chromeos-common.sh;',
2112 'load_base_vars;',
2113 'get_fixed_dst_drive'])
2114 rootdev = self.run(command=rootdev_cmd, ignore_status=True)
2115 if rootdev.exit_status:
2116 logging.info("Fail to run %s", rootdev_cmd)
2117 return None
2118 rootdev_str = rootdev.stdout.strip()
2119
2120 if not rootdev_str:
2121 return None
2122
2123 rootdev_base = os.path.basename(rootdev_str)
2124
2125 mmc_pattern = '/dev/mmcblk[0-9]'
2126 if re.match(mmc_pattern, rootdev_str):
2127 # Use type to determine if the internal device is eMMC or somthing
2128 # else. We can assume that MMC is always an internal device.
2129 type_cmd = 'cat /sys/block/%s/device/type' % rootdev_base
2130 type = self.run(command=type_cmd, ignore_status=True)
2131 if type.exit_status:
2132 logging.info("Fail to run %s", type_cmd)
2133 return None
2134 type_str = type.stdout.strip()
2135
2136 if type_str == 'MMC':
2137 return 'storage:mmc'
2138
2139 scsi_pattern = '/dev/sd[a-z]+'
2140 if re.match(scsi_pattern, rootdev.stdout):
2141 # Read symlink for /sys/block/sd* to determine if the internal
2142 # device is connected via ata or usb.
2143 link_cmd = 'readlink /sys/block/%s' % rootdev_base
2144 link = self.run(command=link_cmd, ignore_status=True)
2145 if link.exit_status:
2146 logging.info("Fail to run %s", link_cmd)
2147 return None
2148 link_str = link.stdout.strip()
2149 if 'usb' in link_str:
2150 return None
2151
2152 # Read rotation to determine if the internal device is ssd or hdd.
2153 rotate_cmd = str('cat /sys/block/%s/queue/rotational'
2154 % rootdev_base)
2155 rotate = self.run(command=rotate_cmd, ignore_status=True)
2156 if rotate.exit_status:
2157 logging.info("Fail to run %s", rotate_cmd)
2158 return None
2159 rotate_str = rotate.stdout.strip()
2160
2161 rotate_dict = {'0':'storage:ssd', '1':'storage:hdd'}
2162 return rotate_dict.get(rotate_str)
2163
2164 # All other internal device / error case will always fall here
2165 return None
2166
2167
2168 def get_servo(self):
2169 """Determine if the host has a servo attached.
2170
2171 If the host has a working servo attached, it should have a servo label.
2172
2173 @return: string 'servo' if the host has servo attached. Otherwise,
2174 returns None.
2175 """
2176 return 'servo' if self._servo_host else None
2177
2178
2179 def get_video_labels(self):
2180 """Run /usr/local/bin/avtest_label_detect to get a list of video labels.
2181
2182 Sample output of avtest_label_detect:
2183 Detected label: hw_video_acc_vp8
2184 Detected label: webcam
2185
2186 @return: A list of labels detected by tool avtest_label_detect.
2187 """
2188 try:
2189 result = self.run('/usr/local/bin/avtest_label_detect').stdout
2190 return re.findall('^Detected label: (\w+)$', result, re.M)
2191 except error.AutoservRunError:
2192 # The tool is not installed.
2193 return []
2194
2195
2196 def is_video_glitch_detection_supported(self):
2197 """ Determine if a board under test is supported for video glitch
2198 detection tests.
2199
2200 @return: 'video_glitch_detection' if board is supported, None otherwise.
2201 """
2202 board = self.get_board().replace(ds_constants.BOARD_PREFIX, '')
2203
2204 if board in video_test_constants.SUPPORTED_BOARDS:
2205 return 'video_glitch_detection'
2206
2207 return None
2208
2209
2210 def get_touch(self):
2211 """
2212 Determine whether board under test has a touchpad or touchscreen.
2213
2214 @return: A list of some combination of 'touchscreen' and 'touchpad',
2215 depending on what is present on the device.
2216
2217 """
2218 labels = []
2219 looking_for = ['touchpad', 'touchscreen']
2220 player = input_playback.InputPlayback()
2221 input_events = self.run('ls /dev/input/event*').stdout.strip().split()
2222 filename = '/tmp/touch_labels'
2223 for event in input_events:
2224 self.run('evtest %s > %s' % (event, filename), timeout=1,
2225 ignore_timeout=True)
2226 properties = self.run('cat %s' % filename).stdout
2227 input_type = player._determine_input_type(properties)
2228 if input_type in looking_for:
2229 labels.append(input_type)
2230 looking_for.remove(input_type)
2231 if len(looking_for) == 0:
2232 break
2233 self.run('rm %s' % filename)
2234
2235 return labels
2236
2237
2238 def has_internal_display(self):
2239 """Determine if the device under test is equipped with an internal
2240 display.
2241
2242 @return: 'internal_display' if one is present; None otherwise.
2243 """
2244 from autotest_lib.client.cros.graphics import graphics_utils
2245 from autotest_lib.client.common_lib import utils as common_utils
2246
2247 def __system_output(cmd):
2248 return self.run(cmd).stdout
2249
2250 def __read_file(remote_path):
2251 return self.run('cat %s' % remote_path).stdout
2252
2253 # Hijack the necessary client functions so that we can take advantage
2254 # of the client lib here.
2255 # FIXME: find a less hacky way than this
2256 original_system_output = utils.system_output
2257 original_read_file = common_utils.read_file
2258 utils.system_output = __system_output
2259 common_utils.read_file = __read_file
2260 try:
2261 return ('internal_display' if graphics_utils.has_internal_display()
2262 else None)
2263 finally:
2264 utils.system_output = original_system_output
2265 common_utils.read_file = original_read_file
2266
2267
2268 def has_lucid_sleep_support(self):
2269 """Determine if the device under test has support for lucid sleep.
2270
2271 @return 'lucidsleep' if this board supports lucid sleep; None otherwise
2272 """
2273 board = self.get_board().replace(ds_constants.BOARD_PREFIX, '')
2274 return 'lucidsleep' if board in LUCID_SLEEP_BOARDS else None
2275
2276
Dan Shi85276d42014-04-08 22:11:45 -07002277 def is_boot_from_usb(self):
2278 """Check if DUT is boot from USB.
2279
2280 @return: True if DUT is boot from usb.
2281 """
2282 device = self.run('rootdev -s -d').stdout.strip()
2283 removable = int(self.run('cat /sys/block/%s/removable' %
2284 os.path.basename(device)).stdout.strip())
2285 return removable == 1
Helen Zhang17dae2b2014-11-11 09:25:52 -08002286
2287
2288 def read_from_meminfo(self, key):
Dan Shi49ca0932014-11-14 11:22:27 -08002289 """Return the memory info from /proc/meminfo
Helen Zhang17dae2b2014-11-11 09:25:52 -08002290
2291 @param key: meminfo requested
2292
2293 @return the memory value as a string
2294
2295 """
Helen Zhang17dae2b2014-11-11 09:25:52 -08002296 meminfo = self.run('grep %s /proc/meminfo' % key).stdout.strip()
2297 logging.debug('%s', meminfo)
2298 return int(re.search(r'\d+', meminfo).group(0))
Rohit Makasana8a4923c2015-08-13 17:04:26 -07002299
2300
Rohit Makasana98e696f2016-06-03 18:48:10 -07002301 def get_cpu_arch(self):
2302 """Returns CPU arch of the device.
2303
2304 @return CPU architecture of the DUT.
2305 """
2306 # Add CPUs by following logic in client/bin/base_utils.py.
2307 if self.run("grep '^flags.*:.* lm .*' /proc/cpuinfo",
2308 ignore_status=True).stdout:
2309 return 'x86_64'
2310 if self.run("grep -Ei 'ARM|CPU implementer' /proc/cpuinfo",
2311 ignore_status=True).stdout:
2312 return 'arm'
2313 return 'i386'
2314
2315
Rohit Makasana8a4923c2015-08-13 17:04:26 -07002316 def get_board_type(self):
2317 """
2318 Get the DUT's device type from /etc/lsb-release.
Danny Chan471a8d12015-08-18 14:57:41 -07002319 DEVICETYPE can be one of CHROMEBOX, CHROMEBASE, CHROMEBOOK or more.
2320
2321 @return value of DEVICETYPE param from lsb-release.
Rohit Makasana8a4923c2015-08-13 17:04:26 -07002322 """
Danny Chan471a8d12015-08-18 14:57:41 -07002323 device_type = self.run('grep DEVICETYPE /etc/lsb-release',
2324 ignore_status=True).stdout
2325 if device_type:
Kalin Stoyanov524310b2015-08-21 16:24:04 -07002326 return device_type.split('=')[-1].strip()
Danny Chan471a8d12015-08-18 14:57:41 -07002327 return ''
Gilad Arnolda76bef02015-09-29 13:55:15 -07002328
2329
2330 def get_os_type(self):
2331 return 'cros'
Simran Basia5522a32015-10-06 11:01:24 -07002332
2333
2334 def enable_adb_testing(self):
2335 """Mark this host as an adb tester."""
Dan Shia2872172015-10-31 01:16:51 -07002336 self.run('touch %s' % constants.ANDROID_TESTER_FILEFLAG)
Kevin Chenga2619dc2016-03-28 11:42:08 -07002337
2338
2339 def get_labels(self):
Kevin Chengf8660142016-08-12 10:17:41 -07002340 """Return the detected labels on the host."""
Kevin Chenga2619dc2016-03-28 11:42:08 -07002341 return self.labels.get_labels(self)
Kevin Chengf8660142016-08-12 10:17:41 -07002342
2343
2344 def update_labels(self):
2345 """Update the labels on the host."""
2346 self.labels.update_labels(self)