J. Richard Barnette | 24adbf4 | 2012-04-11 15:04:53 -0700 | [diff] [blame] | 1 | # Copyright (c) 2012 The Chromium OS Authors. All rights reserved. |
Dale Curtis | aa5eedb | 2011-08-23 16:18:52 -0700 | [diff] [blame] | 2 | # Use of this source code is governed by a BSD-style license that can be |
| 3 | # found in the LICENSE file. |
| 4 | |
J. Richard Barnette | 1d78b01 | 2012-05-15 13:56:30 -0700 | [diff] [blame] | 5 | import logging |
Dan Shi | 0f466e8 | 2013-02-22 15:44:58 -0800 | [diff] [blame] | 6 | import os |
Simran Basi | d5e5e27 | 2012-09-24 15:23:59 -0700 | [diff] [blame] | 7 | import re |
Vincent Palatin | df2372c | 2016-10-07 17:03:00 +0200 | [diff] [blame] | 8 | import sys |
J. Richard Barnette | 134ec2c | 2012-04-25 12:59:37 -0700 | [diff] [blame] | 9 | import time |
| 10 | |
mussa | 584b446 | 2014-06-20 15:13:28 -0700 | [diff] [blame] | 11 | import common |
J. Richard Barnette | 45e93de | 2012-04-11 17:24:15 -0700 | [diff] [blame] | 12 | from autotest_lib.client.bin import utils |
Dan Shi | 9cb0eec | 2014-06-03 09:04:50 -0700 | [diff] [blame] | 13 | from autotest_lib.client.common_lib import autotemp |
Richard Barnette | 0c73ffc | 2012-11-19 15:21:18 -0800 | [diff] [blame] | 14 | from autotest_lib.client.common_lib import error |
| 15 | from autotest_lib.client.common_lib import global_config |
J. Richard Barnette | 91137f0 | 2016-03-10 16:52:26 -0800 | [diff] [blame] | 16 | from autotest_lib.client.common_lib import hosts |
Dan Shi | 549fb82 | 2015-03-24 18:01:11 -0700 | [diff] [blame] | 17 | from autotest_lib.client.common_lib import lsbrelease_utils |
J. Richard Barnette | 45e93de | 2012-04-11 17:24:15 -0700 | [diff] [blame] | 18 | from autotest_lib.client.common_lib.cros import autoupdater |
Richard Barnette | 03a0c13 | 2012-11-05 12:40:35 -0800 | [diff] [blame] | 19 | from autotest_lib.client.common_lib.cros import dev_server |
Hsinyu Chao | e0b08e6 | 2015-08-11 10:50:37 +0000 | [diff] [blame] | 20 | from autotest_lib.client.cros import constants as client_constants |
J. Richard Barnette | 84890bd | 2014-02-21 11:05:47 -0800 | [diff] [blame] | 21 | from autotest_lib.client.cros import cros_ui |
Kevin Cheng | a328da6 | 2016-03-31 10:49:04 -0700 | [diff] [blame] | 22 | from autotest_lib.client.cros.audio import cras_utils |
| 23 | from autotest_lib.client.cros.input_playback import input_playback |
| 24 | from autotest_lib.client.cros.video import constants as video_test_constants |
Simran Basi | 5ace6f2 | 2016-01-06 17:30:44 -0800 | [diff] [blame] | 25 | from autotest_lib.server import afe_utils |
MK Ryu | 35d661e | 2014-09-25 17:44:10 -0700 | [diff] [blame] | 26 | from autotest_lib.server import autoserv_parser |
| 27 | from autotest_lib.server import autotest |
| 28 | from autotest_lib.server import constants |
Dan Shi | a1ecd5c | 2013-06-06 11:21:31 -0700 | [diff] [blame] | 29 | from autotest_lib.server import utils as server_utils |
Dan Shi | 9cb0eec | 2014-06-03 09:04:50 -0700 | [diff] [blame] | 30 | from autotest_lib.server.cros import provision |
Scott Zawalski | 89c44dd | 2013-02-26 09:28:02 -0500 | [diff] [blame] | 31 | from autotest_lib.server.cros.dynamic_suite import constants as ds_constants |
Simran Basi | 5e6339a | 2013-03-21 11:34:32 -0700 | [diff] [blame] | 32 | from autotest_lib.server.cros.dynamic_suite import tools, frontend_wrappers |
Dan Shi | 9cb0eec | 2014-06-03 09:04:50 -0700 | [diff] [blame] | 33 | from autotest_lib.server.cros.faft.config.config import Config as FAFTConfig |
Scott | fe06ed8 | 2015-11-05 17:15:01 -0800 | [diff] [blame] | 34 | from autotest_lib.server.cros.servo import plankton |
Fang Deng | 96667ca | 2013-08-01 17:46:18 -0700 | [diff] [blame] | 35 | from autotest_lib.server.hosts import abstract_ssh |
Kevin Cheng | a2619dc | 2016-03-28 11:42:08 -0700 | [diff] [blame] | 36 | from autotest_lib.server.hosts import base_label |
| 37 | from autotest_lib.server.hosts import cros_label |
Tom Wai-Hong Tam | efe1c7f | 2014-01-02 14:00:11 +0800 | [diff] [blame] | 38 | from autotest_lib.server.hosts import chameleon_host |
J. Richard Barnette | a7e7fdc | 2016-02-12 12:35:36 -0800 | [diff] [blame] | 39 | from autotest_lib.server.hosts import cros_repair |
Scott | fe06ed8 | 2015-11-05 17:15:01 -0800 | [diff] [blame] | 40 | from autotest_lib.server.hosts import plankton_host |
Fang Deng | 5d518f4 | 2013-08-02 14:04:32 -0700 | [diff] [blame] | 41 | from autotest_lib.server.hosts import servo_host |
Simran Basi | dcff425 | 2012-11-20 16:13:20 -0800 | [diff] [blame] | 42 | from autotest_lib.site_utils.rpm_control_system import rpm_client |
Simran Basi | d5e5e27 | 2012-09-24 15:23:59 -0700 | [diff] [blame] | 43 | |
Simran Basi | 382506b | 2016-09-13 14:58:15 -0700 | [diff] [blame] | 44 | # In case cros_host is being ran via SSP on an older Moblab version with an |
| 45 | # older chromite version. |
| 46 | try: |
| 47 | from chromite.lib import metrics |
Dan Shi | 5e2efb7 | 2017-02-07 11:40:23 -0800 | [diff] [blame] | 48 | except ImportError: |
| 49 | metrics = utils.metrics_mock |
| 50 | |
Simran Basi | d5e5e27 | 2012-09-24 15:23:59 -0700 | [diff] [blame] | 51 | |
Dan Shi | b8540a5 | 2015-07-16 14:18:23 -0700 | [diff] [blame] | 52 | CONFIG = global_config.global_config |
xixuan | ec801e3 | 2016-08-25 10:20:22 -0700 | [diff] [blame] | 53 | ENABLE_DEVSERVER_TRIGGER_AUTO_UPDATE = CONFIG.get_config_value( |
| 54 | 'CROS', 'enable_devserver_trigger_auto_update', type=bool, |
| 55 | default=False) |
Dan Shi | b8540a5 | 2015-07-16 14:18:23 -0700 | [diff] [blame] | 56 | |
Dan Shi | d07ee2e | 2015-09-24 14:49:25 -0700 | [diff] [blame] | 57 | |
beeps | c87ff60 | 2013-07-31 21:53:00 -0700 | [diff] [blame] | 58 | class FactoryImageCheckerException(error.AutoservError): |
| 59 | """Exception raised when an image is a factory image.""" |
| 60 | pass |
| 61 | |
| 62 | |
Fang Deng | 0ca40e2 | 2013-08-27 17:47:44 -0700 | [diff] [blame] | 63 | class CrosHost(abstract_ssh.AbstractSSHHost): |
J. Richard Barnette | 45e93de | 2012-04-11 17:24:15 -0700 | [diff] [blame] | 64 | """Chromium OS specific subclass of Host.""" |
| 65 | |
Simran Basi | 5ace6f2 | 2016-01-06 17:30:44 -0800 | [diff] [blame] | 66 | VERSION_PREFIX = provision.CROS_VERSION_PREFIX |
| 67 | |
J. Richard Barnette | 45e93de | 2012-04-11 17:24:15 -0700 | [diff] [blame] | 68 | _parser = autoserv_parser.autoserv_parser |
Scott Zawalski | 62bacae | 2013-03-05 10:40:32 -0500 | [diff] [blame] | 69 | _AFE = frontend_wrappers.RetryingAFE(timeout_min=5, delay_sec=10) |
xixuan | ec801e3 | 2016-08-25 10:20:22 -0700 | [diff] [blame] | 70 | support_devserver_provision = ENABLE_DEVSERVER_TRIGGER_AUTO_UPDATE |
J. Richard Barnette | 45e93de | 2012-04-11 17:24:15 -0700 | [diff] [blame] | 71 | |
Richard Barnette | 03a0c13 | 2012-11-05 12:40:35 -0800 | [diff] [blame] | 72 | # Timeout values (in seconds) associated with various Chrome OS |
| 73 | # state changes. |
J. Richard Barnette | 134ec2c | 2012-04-25 12:59:37 -0700 | [diff] [blame] | 74 | # |
Richard Barnette | 0c73ffc | 2012-11-19 15:21:18 -0800 | [diff] [blame] | 75 | # In general, a good rule of thumb is that the timeout can be up |
| 76 | # to twice the typical measured value on the slowest platform. |
| 77 | # The times here have not necessarily been empirically tested to |
| 78 | # meet this criterion. |
J. Richard Barnette | eb69d72 | 2012-06-18 17:29:44 -0700 | [diff] [blame] | 79 | # |
| 80 | # SLEEP_TIMEOUT: Time to allow for suspend to memory. |
Richard Barnette | 0c73ffc | 2012-11-19 15:21:18 -0800 | [diff] [blame] | 81 | # RESUME_TIMEOUT: Time to allow for resume after suspend, plus |
| 82 | # time to restart the netwowrk. |
J. Richard Barnette | 84890bd | 2014-02-21 11:05:47 -0800 | [diff] [blame] | 83 | # SHUTDOWN_TIMEOUT: Time to allow for shut down. |
J. Richard Barnette | eb69d72 | 2012-06-18 17:29:44 -0700 | [diff] [blame] | 84 | # BOOT_TIMEOUT: Time to allow for boot from power off. Among |
Richard Barnette | 0c73ffc | 2012-11-19 15:21:18 -0800 | [diff] [blame] | 85 | # other things, this must account for the 30 second dev-mode |
J. Richard Barnette | 417cc79 | 2015-10-01 09:56:36 -0700 | [diff] [blame] | 86 | # screen delay, time to start the network on the DUT, and the |
| 87 | # ssh timeout of 120 seconds. |
J. Richard Barnette | eb69d72 | 2012-06-18 17:29:44 -0700 | [diff] [blame] | 88 | # USB_BOOT_TIMEOUT: Time to allow for boot from a USB device, |
Richard Barnette | 0c73ffc | 2012-11-19 15:21:18 -0800 | [diff] [blame] | 89 | # including the 30 second dev-mode delay and time to start the |
J. Richard Barnette | d4649c6 | 2013-03-06 17:42:27 -0800 | [diff] [blame] | 90 | # network. |
beeps | f079cfb | 2013-09-18 17:49:51 -0700 | [diff] [blame] | 91 | # INSTALL_TIMEOUT: Time to allow for chromeos-install. |
J. Richard Barnette | 84890bd | 2014-02-21 11:05:47 -0800 | [diff] [blame] | 92 | # POWERWASH_BOOT_TIMEOUT: Time to allow for a reboot that |
| 93 | # includes powerwash. |
J. Richard Barnette | eb69d72 | 2012-06-18 17:29:44 -0700 | [diff] [blame] | 94 | |
| 95 | SLEEP_TIMEOUT = 2 |
J. Richard Barnette | d4649c6 | 2013-03-06 17:42:27 -0800 | [diff] [blame] | 96 | RESUME_TIMEOUT = 10 |
Tom Wai-Hong Tam | fe005c2 | 2014-12-03 09:25:44 +0800 | [diff] [blame] | 97 | SHUTDOWN_TIMEOUT = 10 |
J. Richard Barnette | 417cc79 | 2015-10-01 09:56:36 -0700 | [diff] [blame] | 98 | BOOT_TIMEOUT = 150 |
J. Richard Barnette | 5bab5f5 | 2015-08-03 13:14:38 -0700 | [diff] [blame] | 99 | USB_BOOT_TIMEOUT = 300 |
J. Richard Barnette | 7817b05 | 2014-08-28 09:47:29 -0700 | [diff] [blame] | 100 | INSTALL_TIMEOUT = 480 |
Dan Shi | 2c88eed | 2013-11-12 10:18:38 -0800 | [diff] [blame] | 101 | POWERWASH_BOOT_TIMEOUT = 60 |
Chris Sosa | b76e0ee | 2013-05-22 16:55:41 -0700 | [diff] [blame] | 102 | |
Dan Shi | ca50348 | 2015-03-30 17:23:25 -0700 | [diff] [blame] | 103 | # Minimum OS version that supports server side packaging. Older builds may |
| 104 | # not have server side package built or with Autotest code change to support |
| 105 | # server-side packaging. |
Dan Shi | b8540a5 | 2015-07-16 14:18:23 -0700 | [diff] [blame] | 106 | MIN_VERSION_SUPPORT_SSP = CONFIG.get_config_value( |
Dan Shi | ced09e4 | 2015-04-17 16:09:34 -0700 | [diff] [blame] | 107 | 'AUTOSERV', 'min_version_support_ssp', type=int) |
Dan Shi | ca50348 | 2015-03-30 17:23:25 -0700 | [diff] [blame] | 108 | |
J. Richard Barnette | 84890bd | 2014-02-21 11:05:47 -0800 | [diff] [blame] | 109 | # REBOOT_TIMEOUT: How long to wait for a reboot. |
| 110 | # |
Chris Sosa | b76e0ee | 2013-05-22 16:55:41 -0700 | [diff] [blame] | 111 | # We have a long timeout to ensure we don't flakily fail due to other |
| 112 | # issues. Shorter timeouts are vetted in platform_RebootAfterUpdate. |
Simran Basi | 1160e2c | 2013-10-04 16:00:24 -0700 | [diff] [blame] | 113 | # TODO(sbasi - crbug.com/276094) Restore to 5 mins once the 'host did not |
| 114 | # return from reboot' bug is solved. |
| 115 | REBOOT_TIMEOUT = 480 |
Chris Sosa | b76e0ee | 2013-05-22 16:55:41 -0700 | [diff] [blame] | 116 | |
Ismail Noorbasha | 07fdb61 | 2013-02-14 14:13:31 -0800 | [diff] [blame] | 117 | # _USB_POWER_TIMEOUT: Time to allow for USB to power toggle ON and OFF. |
| 118 | # _POWER_CYCLE_TIMEOUT: Time to allow for manual power cycle. |
| 119 | _USB_POWER_TIMEOUT = 5 |
| 120 | _POWER_CYCLE_TIMEOUT = 10 |
| 121 | |
Richard Barnette | 82c3591 | 2012-11-20 10:09:10 -0800 | [diff] [blame] | 122 | _LAB_MACHINE_FILE = '/mnt/stateful_partition/.labmachine' |
Fang Deng | deba14f | 2014-11-14 11:54:09 -0800 | [diff] [blame] | 123 | _RPM_HOSTNAME_REGEX = ('chromeos(\d+)(-row(\d+))?-rack(\d+[a-z]*)' |
| 124 | '-host(\d+)') |
Kevin Cheng | a328da6 | 2016-03-31 10:49:04 -0700 | [diff] [blame] | 125 | _LIGHTSENSOR_FILES = [ "in_illuminance0_input", |
| 126 | "in_illuminance_input", |
| 127 | "in_illuminance0_raw", |
| 128 | "in_illuminance_raw", |
| 129 | "illuminance0_input"] |
| 130 | _LIGHTSENSOR_SEARCH_DIR = '/sys/bus/iio/devices' |
J. Richard Barnette | 134ec2c | 2012-04-25 12:59:37 -0700 | [diff] [blame] | 131 | |
J. Richard Barnette | b6de7e3 | 2013-02-14 13:28:04 -0800 | [diff] [blame] | 132 | # Constants used in ping_wait_up() and ping_wait_down(). |
| 133 | # |
| 134 | # _PING_WAIT_COUNT is the approximate number of polling |
| 135 | # cycles to use when waiting for a host state change. |
| 136 | # |
| 137 | # _PING_STATUS_DOWN and _PING_STATUS_UP are names used |
| 138 | # for arguments to the internal _ping_wait_for_status() |
| 139 | # method. |
| 140 | _PING_WAIT_COUNT = 40 |
| 141 | _PING_STATUS_DOWN = False |
| 142 | _PING_STATUS_UP = True |
| 143 | |
Ismail Noorbasha | 07fdb61 | 2013-02-14 14:13:31 -0800 | [diff] [blame] | 144 | # Allowed values for the power_method argument. |
| 145 | |
| 146 | # POWER_CONTROL_RPM: Passed as default arg for power_off/on/cycle() methods. |
| 147 | # POWER_CONTROL_SERVO: Used in set_power() and power_cycle() methods. |
| 148 | # POWER_CONTROL_MANUAL: Used in set_power() and power_cycle() methods. |
| 149 | POWER_CONTROL_RPM = 'RPM' |
| 150 | POWER_CONTROL_SERVO = 'servoj10' |
| 151 | POWER_CONTROL_MANUAL = 'manual' |
| 152 | |
| 153 | POWER_CONTROL_VALID_ARGS = (POWER_CONTROL_RPM, |
| 154 | POWER_CONTROL_SERVO, |
| 155 | POWER_CONTROL_MANUAL) |
Richard Barnette | 0c73ffc | 2012-11-19 15:21:18 -0800 | [diff] [blame] | 156 | |
Simran Basi | 5e6339a | 2013-03-21 11:34:32 -0700 | [diff] [blame] | 157 | _RPM_OUTLET_CHANGED = 'outlet_changed' |
| 158 | |
Dan Shi | 9cb0eec | 2014-06-03 09:04:50 -0700 | [diff] [blame] | 159 | # URL pattern to download firmware image. |
Dan Shi | b8540a5 | 2015-07-16 14:18:23 -0700 | [diff] [blame] | 160 | _FW_IMAGE_URL_PATTERN = CONFIG.get_config_value( |
Dan Shi | 9cb0eec | 2014-06-03 09:04:50 -0700 | [diff] [blame] | 161 | 'CROS', 'firmware_url_pattern', type=str) |
beeps | 687243d | 2013-07-18 15:29:27 -0700 | [diff] [blame] | 162 | |
J. Richard Barnette | 91137f0 | 2016-03-10 16:52:26 -0800 | [diff] [blame] | 163 | |
| 164 | # A flag file to indicate provision failures. The file is created |
| 165 | # at the start of any AU procedure (see `machine_install()`). The |
| 166 | # file's location in stateful means that on successul update it will |
| 167 | # be removed. Thus, if this file exists, it indicates that we've |
| 168 | # tried and failed in a previous attempt to update. |
| 169 | PROVISION_FAILED = '/var/tmp/provision_failed' |
MK Ryu | 35d661e | 2014-09-25 17:44:10 -0700 | [diff] [blame] | 170 | |
MK Ryu | 35d661e | 2014-09-25 17:44:10 -0700 | [diff] [blame] | 171 | |
J. Richard Barnette | 964fba0 | 2012-10-24 17:34:29 -0700 | [diff] [blame] | 172 | @staticmethod |
beeps | 46dadc9 | 2013-11-07 14:07:10 -0800 | [diff] [blame] | 173 | def check_host(host, timeout=10): |
| 174 | """ |
| 175 | Check if the given host is a chrome-os host. |
| 176 | |
| 177 | @param host: An ssh host representing a device. |
| 178 | @param timeout: The timeout for the run command. |
| 179 | |
| 180 | @return: True if the host device is chromeos. |
| 181 | |
beeps | 46dadc9 | 2013-11-07 14:07:10 -0800 | [diff] [blame] | 182 | """ |
| 183 | try: |
Allen Li | ad719c1 | 2017-06-27 23:48:04 +0000 | [diff] [blame] | 184 | result = host.run( |
Simran Basi | 933c8af | 2015-04-29 14:05:07 -0700 | [diff] [blame] | 185 | 'grep -q CHROMEOS /etc/lsb-release && ' |
| 186 | '! test -f /mnt/stateful_partition/.android_tester && ' |
| 187 | '! grep -q moblab /etc/lsb-release', |
| 188 | ignore_status=True, timeout=timeout) |
Laurence Goodby | 468de25 | 2017-06-08 17:22:53 -0700 | [diff] [blame] | 189 | if result.exit_status == 0: |
Allen Li | ad719c1 | 2017-06-27 23:48:04 +0000 | [diff] [blame] | 190 | lsb_release_content = host.run( |
Laurence Goodby | 468de25 | 2017-06-08 17:22:53 -0700 | [diff] [blame] | 191 | 'grep CHROMEOS_RELEASE_BOARD /etc/lsb-release', |
| 192 | timeout=timeout).stdout |
| 193 | return not lsbrelease_utils.is_jetstream( |
| 194 | lsb_release_content=lsb_release_content) |
beeps | 46dadc9 | 2013-11-07 14:07:10 -0800 | [diff] [blame] | 195 | except (error.AutoservRunError, error.AutoservSSHTimeout): |
| 196 | return False |
Laurence Goodby | 468de25 | 2017-06-08 17:22:53 -0700 | [diff] [blame] | 197 | |
| 198 | return False |
beeps | 46dadc9 | 2013-11-07 14:07:10 -0800 | [diff] [blame] | 199 | |
| 200 | |
| 201 | @staticmethod |
Tom Wai-Hong Tam | efe1c7f | 2014-01-02 14:00:11 +0800 | [diff] [blame] | 202 | def get_chameleon_arguments(args_dict): |
| 203 | """Extract chameleon options from `args_dict` and return the result. |
| 204 | |
| 205 | Recommended usage: |
| 206 | ~~~~~~~~ |
| 207 | args_dict = utils.args_to_dict(args) |
| 208 | chameleon_args = hosts.CrosHost.get_chameleon_arguments(args_dict) |
| 209 | host = hosts.create_host(machine, chameleon_args=chameleon_args) |
| 210 | ~~~~~~~~ |
| 211 | |
| 212 | @param args_dict Dictionary from which to extract the chameleon |
| 213 | arguments. |
| 214 | """ |
Allen Li | 083866b | 2016-08-18 10:07:10 -0700 | [diff] [blame] | 215 | return {key: args_dict[key] |
| 216 | for key in ('chameleon_host', 'chameleon_port') |
| 217 | if key in args_dict} |
Tom Wai-Hong Tam | efe1c7f | 2014-01-02 14:00:11 +0800 | [diff] [blame] | 218 | |
| 219 | |
| 220 | @staticmethod |
Scott | fe06ed8 | 2015-11-05 17:15:01 -0800 | [diff] [blame] | 221 | def get_plankton_arguments(args_dict): |
| 222 | """Extract chameleon options from `args_dict` and return the result. |
| 223 | |
| 224 | Recommended usage: |
| 225 | ~~~~~~~~ |
| 226 | args_dict = utils.args_to_dict(args) |
Richard Barnette | e519dcd | 2016-08-15 17:37:17 -0700 | [diff] [blame] | 227 | plankton_args = hosts.CrosHost.get_plankton_arguments(args_dict) |
| 228 | host = hosts.create_host(machine, plankton_args=plankton_args) |
Scott | fe06ed8 | 2015-11-05 17:15:01 -0800 | [diff] [blame] | 229 | ~~~~~~~~ |
| 230 | |
| 231 | @param args_dict Dictionary from which to extract the plankton |
| 232 | arguments. |
| 233 | """ |
Allen Li | 083866b | 2016-08-18 10:07:10 -0700 | [diff] [blame] | 234 | return {key: args_dict[key] |
| 235 | for key in ('plankton_host', 'plankton_port') |
| 236 | if key in args_dict} |
Scott | fe06ed8 | 2015-11-05 17:15:01 -0800 | [diff] [blame] | 237 | |
| 238 | |
| 239 | @staticmethod |
Tom Wai-Hong Tam | efe1c7f | 2014-01-02 14:00:11 +0800 | [diff] [blame] | 240 | def get_servo_arguments(args_dict): |
| 241 | """Extract servo options from `args_dict` and return the result. |
J. Richard Barnette | 7214e0b | 2013-02-06 15:20:49 -0800 | [diff] [blame] | 242 | |
| 243 | Recommended usage: |
| 244 | ~~~~~~~~ |
| 245 | args_dict = utils.args_to_dict(args) |
Fang Deng | 0ca40e2 | 2013-08-27 17:47:44 -0700 | [diff] [blame] | 246 | servo_args = hosts.CrosHost.get_servo_arguments(args_dict) |
J. Richard Barnette | 7214e0b | 2013-02-06 15:20:49 -0800 | [diff] [blame] | 247 | host = hosts.create_host(machine, servo_args=servo_args) |
| 248 | ~~~~~~~~ |
| 249 | |
| 250 | @param args_dict Dictionary from which to extract the servo |
| 251 | arguments. |
| 252 | """ |
Richard Barnette | e519dcd | 2016-08-15 17:37:17 -0700 | [diff] [blame] | 253 | servo_attrs = (servo_host.SERVO_HOST_ATTR, |
| 254 | servo_host.SERVO_PORT_ATTR, |
| 255 | servo_host.SERVO_BOARD_ATTR) |
Armando Miraglia | 2ac802e | 2017-08-15 14:54:47 +0200 | [diff] [blame] | 256 | servo_args = {key: args_dict[key] |
| 257 | for key in servo_attrs |
| 258 | if key in args_dict} |
| 259 | return ( |
| 260 | None |
| 261 | if servo_host.SERVO_HOST_ATTR in servo_args |
| 262 | and not servo_args[servo_host.SERVO_HOST_ATTR] |
| 263 | else servo_args) |
J. Richard Barnette | 134ec2c | 2012-04-25 12:59:37 -0700 | [diff] [blame] | 264 | |
J. Richard Barnette | 964fba0 | 2012-10-24 17:34:29 -0700 | [diff] [blame] | 265 | |
J. Richard Barnette | 91137f0 | 2016-03-10 16:52:26 -0800 | [diff] [blame] | 266 | def _initialize(self, hostname, chameleon_args=None, servo_args=None, |
| 267 | plankton_args=None, try_lab_servo=False, |
Richard Barnette | 9a26ad6 | 2016-06-10 12:03:08 -0700 | [diff] [blame] | 268 | try_servo_repair=False, |
J. Richard Barnette | 91137f0 | 2016-03-10 16:52:26 -0800 | [diff] [blame] | 269 | ssh_verbosity_flag='', ssh_options='', |
Fang Deng | d1c2b73 | 2013-08-20 12:59:46 -0700 | [diff] [blame] | 270 | *args, **dargs): |
Tom Wai-Hong Tam | efe1c7f | 2014-01-02 14:00:11 +0800 | [diff] [blame] | 271 | """Initialize superclasses, |self.chameleon|, and |self.servo|. |
J. Richard Barnette | 67ccb87 | 2012-04-19 16:34:56 -0700 | [diff] [blame] | 272 | |
Fang Deng | e545abb | 2014-12-30 18:43:47 -0800 | [diff] [blame] | 273 | This method will attempt to create the test-assistant object |
| 274 | (chameleon/servo) when it is needed by the test. Check |
| 275 | the docstring of chameleon_host.create_chameleon_host and |
| 276 | servo_host.create_servo_host for how this is determined. |
Fang Deng | 5d518f4 | 2013-08-02 14:04:32 -0700 | [diff] [blame] | 277 | |
Fang Deng | e545abb | 2014-12-30 18:43:47 -0800 | [diff] [blame] | 278 | @param hostname: Hostname of the dut. |
| 279 | @param chameleon_args: A dictionary that contains args for creating |
| 280 | a ChameleonHost. See chameleon_host for details. |
| 281 | @param servo_args: A dictionary that contains args for creating |
| 282 | a ServoHost object. See servo_host for details. |
Richard Barnette | 9a26ad6 | 2016-06-10 12:03:08 -0700 | [diff] [blame] | 283 | @param try_lab_servo: When true, indicates that an attempt should |
| 284 | be made to create a ServoHost for a DUT in |
| 285 | the test lab, even if not required by |
| 286 | `servo_args`. See servo_host for details. |
| 287 | @param try_servo_repair: If a servo host is created, check it |
| 288 | with `repair()` rather than `verify()`. |
Fang Deng | e545abb | 2014-12-30 18:43:47 -0800 | [diff] [blame] | 289 | See servo_host for details. |
| 290 | @param ssh_verbosity_flag: String, to pass to the ssh command to control |
| 291 | verbosity. |
| 292 | @param ssh_options: String, other ssh options to pass to the ssh |
| 293 | command. |
J. Richard Barnette | 67ccb87 | 2012-04-19 16:34:56 -0700 | [diff] [blame] | 294 | """ |
Fang Deng | 0ca40e2 | 2013-08-27 17:47:44 -0700 | [diff] [blame] | 295 | super(CrosHost, self)._initialize(hostname=hostname, |
J. Richard Barnette | 45e93de | 2012-04-11 17:24:15 -0700 | [diff] [blame] | 296 | *args, **dargs) |
J. Richard Barnette | 91137f0 | 2016-03-10 16:52:26 -0800 | [diff] [blame] | 297 | self._repair_strategy = cros_repair.create_cros_repair_strategy() |
Kevin Cheng | a2619dc | 2016-03-28 11:42:08 -0700 | [diff] [blame] | 298 | self.labels = base_label.LabelRetriever(cros_label.CROS_LABELS) |
J. Richard Barnette | f085985 | 2012-08-20 14:55:50 -0700 | [diff] [blame] | 299 | # self.env is a dictionary of environment variable settings |
| 300 | # to be exported for commands run on the host. |
| 301 | # LIBC_FATAL_STDERR_ can be useful for diagnosing certain |
| 302 | # errors that might happen. |
| 303 | self.env['LIBC_FATAL_STDERR_'] = '1' |
Fang Deng | d1c2b73 | 2013-08-20 12:59:46 -0700 | [diff] [blame] | 304 | self._ssh_verbosity_flag = ssh_verbosity_flag |
Aviv Keshet | c5947fa | 2013-09-04 14:06:29 -0700 | [diff] [blame] | 305 | self._ssh_options = ssh_options |
Fang Deng | 5d518f4 | 2013-08-02 14:04:32 -0700 | [diff] [blame] | 306 | # TODO(fdeng): We need to simplify the |
| 307 | # process of servo and servo_host initialization. |
| 308 | # crbug.com/298432 |
Moja Hsu | 57d93a1 | 2017-01-13 17:57:52 +0800 | [diff] [blame] | 309 | self._servo_host = servo_host.create_servo_host( |
Richard Barnette | ea3e460 | 2016-06-10 12:36:41 -0700 | [diff] [blame] | 310 | dut=self, servo_args=servo_args, |
Richard Barnette | 9a26ad6 | 2016-06-10 12:03:08 -0700 | [diff] [blame] | 311 | try_lab_servo=try_lab_servo, |
| 312 | try_servo_repair=try_servo_repair) |
Richard Barnette | e519dcd | 2016-08-15 17:37:17 -0700 | [diff] [blame] | 313 | if self._servo_host is not None: |
| 314 | self.servo = self._servo_host.get_servo() |
| 315 | else: |
| 316 | self.servo = None |
| 317 | |
Tom Wai-Hong Tam | efe1c7f | 2014-01-02 14:00:11 +0800 | [diff] [blame] | 318 | # TODO(waihong): Do the simplication on Chameleon too. |
Tom Wai-Hong Tam | 3d6790d | 2014-04-14 16:15:47 +0800 | [diff] [blame] | 319 | self._chameleon_host = chameleon_host.create_chameleon_host( |
| 320 | dut=self.hostname, chameleon_args=chameleon_args) |
Scott | fe06ed8 | 2015-11-05 17:15:01 -0800 | [diff] [blame] | 321 | # Add plankton host if plankton args were added on command line |
| 322 | self._plankton_host = plankton_host.create_plankton_host(plankton_args) |
Tom Wai-Hong Tam | 3d6790d | 2014-04-14 16:15:47 +0800 | [diff] [blame] | 323 | |
Tom Wai-Hong Tam | 3d6790d | 2014-04-14 16:15:47 +0800 | [diff] [blame] | 324 | if self._chameleon_host: |
Tom Wai-Hong Tam | eaee340 | 2014-01-22 08:52:10 +0800 | [diff] [blame] | 325 | self.chameleon = self._chameleon_host.create_chameleon_board() |
Tom Wai-Hong Tam | efe1c7f | 2014-01-02 14:00:11 +0800 | [diff] [blame] | 326 | else: |
Tom Wai-Hong Tam | 3d6790d | 2014-04-14 16:15:47 +0800 | [diff] [blame] | 327 | self.chameleon = None |
Fang Deng | 5d518f4 | 2013-08-02 14:04:32 -0700 | [diff] [blame] | 328 | |
Scott | fe06ed8 | 2015-11-05 17:15:01 -0800 | [diff] [blame] | 329 | if self._plankton_host: |
| 330 | self.plankton_servo = self._plankton_host.get_servo() |
| 331 | logging.info('plankton_servo: %r', self.plankton_servo) |
| 332 | # Create the plankton object used to access the ec uart |
Scott | 07a848f | 2016-01-12 15:04:52 -0800 | [diff] [blame] | 333 | self.plankton = plankton.Plankton(self.plankton_servo, |
| 334 | self._plankton_host.get_servod_server_proxy()) |
Scott | fe06ed8 | 2015-11-05 17:15:01 -0800 | [diff] [blame] | 335 | else: |
Scott | 07a848f | 2016-01-12 15:04:52 -0800 | [diff] [blame] | 336 | self.plankton = None |
Scott | fe06ed8 | 2015-11-05 17:15:01 -0800 | [diff] [blame] | 337 | |
Fang Deng | 5d518f4 | 2013-08-02 14:04:32 -0700 | [diff] [blame] | 338 | |
Richard Barnette | 260cbd0 | 2016-10-06 12:23:28 -0700 | [diff] [blame] | 339 | def _get_cros_repair_image_name(self): |
Prathmesh Prabhu | 075fc92 | 2017-02-13 11:50:25 -0800 | [diff] [blame] | 340 | info = self.host_info_store.get() |
| 341 | if info.board is None: |
| 342 | raise error.AutoservError('Cannot obtain repair image name. ' |
| 343 | 'No board label value found') |
| 344 | |
| 345 | return afe_utils.get_stable_cros_image_name(info.board) |
Scott Zawalski | 89c44dd | 2013-02-26 09:28:02 -0500 | [diff] [blame] | 346 | |
| 347 | |
Rohit Makasana | df0a3a3 | 2017-06-30 13:55:18 -0700 | [diff] [blame] | 348 | def host_version_prefix(self, image): |
| 349 | """Return version label prefix. |
| 350 | |
| 351 | In case the CrOS provisioning version is something other than the |
| 352 | standard CrOS version e.g. CrOS TH version, this function will |
| 353 | find the prefix from provision.py. |
| 354 | |
| 355 | @param image: The image name to find its version prefix. |
| 356 | @returns: A prefix string for the image type. |
| 357 | """ |
| 358 | return provision.get_version_label_prefix(image) |
| 359 | |
| 360 | |
beeps | dae65fd | 2013-07-26 16:24:41 -0700 | [diff] [blame] | 361 | def verify_job_repo_url(self, tag=''): |
beeps | cb6f1e2 | 2013-06-28 19:14:10 -0700 | [diff] [blame] | 362 | """ |
| 363 | Make sure job_repo_url of this host is valid. |
| 364 | |
joychen | 03eaad9 | 2013-06-26 09:55:21 -0700 | [diff] [blame] | 365 | Eg: The job_repo_url "http://lmn.cd.ab.xyx:8080/static/\ |
beeps | cb6f1e2 | 2013-06-28 19:14:10 -0700 | [diff] [blame] | 366 | lumpy-release/R29-4279.0.0/autotest/packages" claims to have the |
| 367 | autotest package for lumpy-release/R29-4279.0.0. If this isn't the case, |
| 368 | download and extract it. If the devserver embedded in the url is |
| 369 | unresponsive, update the job_repo_url of the host after staging it on |
| 370 | another devserver. |
| 371 | |
| 372 | @param job_repo_url: A url pointing to the devserver where the autotest |
| 373 | package for this build should be staged. |
beeps | dae65fd | 2013-07-26 16:24:41 -0700 | [diff] [blame] | 374 | @param tag: The tag from the server job, in the format |
| 375 | <job_id>-<user>/<hostname>, or <hostless> for a server job. |
beeps | cb6f1e2 | 2013-06-28 19:14:10 -0700 | [diff] [blame] | 376 | |
| 377 | @raises DevServerException: If we could not resolve a devserver. |
| 378 | @raises AutoservError: If we're unable to save the new job_repo_url as |
| 379 | a result of choosing a new devserver because the old one failed to |
| 380 | respond to a health check. |
beeps | 0c86503 | 2013-07-30 11:37:06 -0700 | [diff] [blame] | 381 | @raises urllib2.URLError: If the devserver embedded in job_repo_url |
| 382 | doesn't respond within the timeout. |
beeps | cb6f1e2 | 2013-06-28 19:14:10 -0700 | [diff] [blame] | 383 | """ |
Prathmesh Prabhu | 368abdf | 2017-02-14 11:23:47 -0800 | [diff] [blame] | 384 | info = self.host_info_store.get() |
| 385 | job_repo_url = info.attributes.get(ds_constants.JOB_REPO_URL, '') |
beeps | cb6f1e2 | 2013-06-28 19:14:10 -0700 | [diff] [blame] | 386 | if not job_repo_url: |
| 387 | logging.warning('No job repo url set on host %s', self.hostname) |
| 388 | return |
| 389 | |
| 390 | logging.info('Verifying job repo url %s', job_repo_url) |
| 391 | devserver_url, image_name = tools.get_devserver_build_from_package_url( |
| 392 | job_repo_url) |
| 393 | |
beeps | 0c86503 | 2013-07-30 11:37:06 -0700 | [diff] [blame] | 394 | ds = dev_server.ImageServer(devserver_url) |
beeps | cb6f1e2 | 2013-06-28 19:14:10 -0700 | [diff] [blame] | 395 | |
| 396 | logging.info('Staging autotest artifacts for %s on devserver %s', |
| 397 | image_name, ds.url()) |
beeps | 687243d | 2013-07-18 15:29:27 -0700 | [diff] [blame] | 398 | |
| 399 | start_time = time.time() |
Simran Basi | 25e7a92 | 2014-10-31 11:56:10 -0700 | [diff] [blame] | 400 | ds.stage_artifacts(image_name, ['autotest_packages']) |
beeps | 687243d | 2013-07-18 15:29:27 -0700 | [diff] [blame] | 401 | stage_time = time.time() - start_time |
| 402 | |
| 403 | # Record how much of the verification time comes from a devserver |
| 404 | # restage. If we're doing things right we should not see multiple |
| 405 | # devservers for a given board/build/branch path. |
| 406 | try: |
J. Richard Barnette | 3cbd76b | 2013-11-27 12:11:25 -0800 | [diff] [blame] | 407 | board, build_type, branch = server_utils.ParseBuildName( |
beeps | 687243d | 2013-07-18 15:29:27 -0700 | [diff] [blame] | 408 | image_name)[:3] |
J. Richard Barnette | 3cbd76b | 2013-11-27 12:11:25 -0800 | [diff] [blame] | 409 | except server_utils.ParseBuildNameException: |
beeps | 687243d | 2013-07-18 15:29:27 -0700 | [diff] [blame] | 410 | pass |
| 411 | else: |
beeps | 0c86503 | 2013-07-30 11:37:06 -0700 | [diff] [blame] | 412 | devserver = devserver_url[ |
Chris Sosa | 6542508 | 2013-10-16 13:26:22 -0700 | [diff] [blame] | 413 | devserver_url.find('/') + 2:devserver_url.rfind(':')] |
beeps | 687243d | 2013-07-18 15:29:27 -0700 | [diff] [blame] | 414 | stats_key = { |
| 415 | 'board': board, |
| 416 | 'build_type': build_type, |
| 417 | 'branch': branch, |
beeps | 0c86503 | 2013-07-30 11:37:06 -0700 | [diff] [blame] | 418 | 'devserver': devserver.replace('.', '_'), |
beeps | 687243d | 2013-07-18 15:29:27 -0700 | [diff] [blame] | 419 | } |
Dan Shi | 5e2efb7 | 2017-02-07 11:40:23 -0800 | [diff] [blame] | 420 | |
| 421 | monarch_fields = { |
| 422 | 'board': board, |
| 423 | 'build_type': build_type, |
| 424 | # TODO(akeshet): To be consistent with most other metrics, |
| 425 | # consider changing the following field to be named |
| 426 | # 'milestone'. |
| 427 | 'branch': branch, |
| 428 | 'dev_server': devserver, |
| 429 | } |
| 430 | metrics.Counter( |
| 431 | 'chromeos/autotest/provision/verify_url' |
| 432 | ).increment(fields=monarch_fields) |
| 433 | metrics.SecondsDistribution( |
| 434 | 'chromeos/autotest/provision/verify_url_duration' |
| 435 | ).add(stage_time, fields=monarch_fields) |
| 436 | |
| 437 | |
Dan Shi | cf4d203 | 2015-03-12 15:04:21 -0700 | [diff] [blame] | 438 | def stage_server_side_package(self, image=None): |
| 439 | """Stage autotest server-side package on devserver. |
| 440 | |
| 441 | @param image: Full path of an OS image to install or a build name. |
| 442 | |
| 443 | @return: A url to the autotest server-side package. |
Dan Shi | 14de762 | 2016-08-22 11:09:06 -0700 | [diff] [blame] | 444 | |
| 445 | @raise: error.AutoservError if fail to locate the build to test with, or |
| 446 | fail to stage server-side package. |
Dan Shi | cf4d203 | 2015-03-12 15:04:21 -0700 | [diff] [blame] | 447 | """ |
Dan Shi | d37736b | 2016-07-06 15:10:29 -0700 | [diff] [blame] | 448 | # If enable_drone_in_restricted_subnet is False, do not set hostname |
| 449 | # in devserver.resolve call, so a devserver in non-restricted subnet |
| 450 | # is picked to stage autotest server package for drone to download. |
| 451 | hostname = self.hostname |
| 452 | if not server_utils.ENABLE_DRONE_IN_RESTRICTED_SUBNET: |
| 453 | hostname = None |
Dan Shi | cf4d203 | 2015-03-12 15:04:21 -0700 | [diff] [blame] | 454 | if image: |
| 455 | image_name = tools.get_build_from_image(image) |
| 456 | if not image_name: |
| 457 | raise error.AutoservError( |
| 458 | 'Failed to parse build name from %s' % image) |
Dan Shi | d37736b | 2016-07-06 15:10:29 -0700 | [diff] [blame] | 459 | ds = dev_server.ImageServer.resolve(image_name, hostname) |
Dan Shi | cf4d203 | 2015-03-12 15:04:21 -0700 | [diff] [blame] | 460 | else: |
Prathmesh Prabhu | 9235e4c | 2017-03-28 13:16:06 -0700 | [diff] [blame] | 461 | info = self.host_info_store.get() |
Prathmesh Prabhu | 368abdf | 2017-02-14 11:23:47 -0800 | [diff] [blame] | 462 | job_repo_url = info.attributes.get(ds_constants.JOB_REPO_URL, '') |
Dan Shi | cf4d203 | 2015-03-12 15:04:21 -0700 | [diff] [blame] | 463 | if job_repo_url: |
| 464 | devserver_url, image_name = ( |
| 465 | tools.get_devserver_build_from_package_url(job_repo_url)) |
Dan Shi | d37736b | 2016-07-06 15:10:29 -0700 | [diff] [blame] | 466 | # If enable_drone_in_restricted_subnet is True, use the |
| 467 | # existing devserver. Otherwise, resolve a new one in |
| 468 | # non-restricted subnet. |
| 469 | if server_utils.ENABLE_DRONE_IN_RESTRICTED_SUBNET: |
| 470 | ds = dev_server.ImageServer(devserver_url) |
| 471 | else: |
| 472 | ds = dev_server.ImageServer.resolve(image_name) |
Prathmesh Prabhu | b6cea61 | 2017-02-09 15:41:19 -0800 | [diff] [blame] | 473 | elif info.build is not None: |
| 474 | ds = dev_server.ImageServer.resolve(info.build, hostname) |
Prathmesh Prabhu | 0c1dd4d | 2017-06-07 13:01:53 -0700 | [diff] [blame] | 475 | image_name = info.build |
Dan Shi | cf4d203 | 2015-03-12 15:04:21 -0700 | [diff] [blame] | 476 | else: |
Prathmesh Prabhu | b6cea61 | 2017-02-09 15:41:19 -0800 | [diff] [blame] | 477 | raise error.AutoservError( |
| 478 | 'Failed to stage server-side package. The host has ' |
| 479 | 'no job_report_url attribute or version label.') |
Dan Shi | ca50348 | 2015-03-30 17:23:25 -0700 | [diff] [blame] | 480 | |
| 481 | # Get the OS version of the build, for any build older than |
| 482 | # MIN_VERSION_SUPPORT_SSP, server side packaging is not supported. |
| 483 | match = re.match('.*/R\d+-(\d+)\.', image_name) |
| 484 | if match and int(match.group(1)) < self.MIN_VERSION_SUPPORT_SSP: |
Dan Shi | 14de762 | 2016-08-22 11:09:06 -0700 | [diff] [blame] | 485 | raise error.AutoservError( |
| 486 | 'Build %s is older than %s. Server side packaging is ' |
| 487 | 'disabled.' % (image_name, self.MIN_VERSION_SUPPORT_SSP)) |
Dan Shi | ca50348 | 2015-03-30 17:23:25 -0700 | [diff] [blame] | 488 | |
Dan Shi | cf4d203 | 2015-03-12 15:04:21 -0700 | [diff] [blame] | 489 | ds.stage_artifacts(image_name, ['autotest_server_package']) |
| 490 | return '%s/static/%s/%s' % (ds.url(), image_name, |
| 491 | 'autotest_server_package.tar.bz2') |
| 492 | |
| 493 | |
Dan Shi | 0f466e8 | 2013-02-22 15:44:58 -0800 | [diff] [blame] | 494 | def _try_stateful_update(self, update_url, force_update, updater): |
| 495 | """Try to use stateful update to initialize DUT. |
| 496 | |
| 497 | When DUT is already running the same version that machine_install |
| 498 | tries to install, stateful update is a much faster way to clean up |
| 499 | the DUT for testing, compared to a full reimage. It is implemeted |
| 500 | by calling autoupdater.run_update, but skipping updating root, as |
| 501 | updating the kernel is time consuming and not necessary. |
| 502 | |
| 503 | @param update_url: url of the image. |
| 504 | @param force_update: Set to True to update the image even if the DUT |
| 505 | is running the same version. |
| 506 | @param updater: ChromiumOSUpdater instance used to update the DUT. |
| 507 | @returns: True if the DUT was updated with stateful update. |
| 508 | |
| 509 | """ |
Laurence Goodby | 778c9a4 | 2017-05-24 19:24:07 -0700 | [diff] [blame] | 510 | self.prepare_for_update() |
Dan Shi | 10b9848 | 2016-02-02 14:38:50 -0800 | [diff] [blame] | 511 | |
J. Richard Barnette | 3f73103 | 2014-04-07 17:42:59 -0700 | [diff] [blame] | 512 | # TODO(jrbarnette): Yes, I hate this re.match() test case. |
| 513 | # It's better than the alternative: see crbug.com/360944. |
| 514 | image_name = autoupdater.url_to_image_name(update_url) |
| 515 | release_pattern = r'^.*-release/R[0-9]+-[0-9]+\.[0-9]+\.0$' |
| 516 | if not re.match(release_pattern, image_name): |
| 517 | return False |
Dan Shi | 0f466e8 | 2013-02-22 15:44:58 -0800 | [diff] [blame] | 518 | if not updater.check_version(): |
| 519 | return False |
| 520 | if not force_update: |
| 521 | logging.info('Canceling stateful update because the new and ' |
| 522 | 'old versions are the same.') |
| 523 | return False |
| 524 | # Following folders should be rebuilt after stateful update. |
| 525 | # A test file is used to confirm each folder gets rebuilt after |
| 526 | # the stateful update. |
| 527 | folders_to_check = ['/var', '/home', '/mnt/stateful_partition'] |
| 528 | test_file = '.test_file_to_be_deleted' |
Paul Hobbs | 38bcb70 | 2017-07-25 00:50:05 -0700 | [diff] [blame] | 529 | paths = [os.path.join(folder, test_file) for folder in folders_to_check] |
| 530 | self.run('touch %s' % ' '.join(paths)) |
Dan Shi | 0f466e8 | 2013-02-22 15:44:58 -0800 | [diff] [blame] | 531 | |
Chris Sosa | e92399e | 2015-04-24 11:32:59 -0700 | [diff] [blame] | 532 | updater.run_update(update_root=False) |
Dan Shi | 0f466e8 | 2013-02-22 15:44:58 -0800 | [diff] [blame] | 533 | |
| 534 | # Reboot to complete stateful update. |
Chris Sosa | b76e0ee | 2013-05-22 16:55:41 -0700 | [diff] [blame] | 535 | self.reboot(timeout=self.REBOOT_TIMEOUT, wait=True) |
xixuan | 908b075 | 2016-12-08 18:54:23 -0800 | [diff] [blame] | 536 | |
| 537 | # After stateful update and a reboot, all of the test_files shouldn't |
| 538 | # exist any more. Otherwise the stateful update is failed. |
| 539 | return not any( |
| 540 | self.path_exists(os.path.join(folder, test_file)) |
| 541 | for folder in folders_to_check) |
Dan Shi | 0f466e8 | 2013-02-22 15:44:58 -0800 | [diff] [blame] | 542 | |
| 543 | |
J. Richard Barnette | 7275b61 | 2013-06-04 18:13:11 -0700 | [diff] [blame] | 544 | def _post_update_processing(self, updater, expected_kernel=None): |
Dan Shi | 0f466e8 | 2013-02-22 15:44:58 -0800 | [diff] [blame] | 545 | """After the DUT is updated, confirm machine_install succeeded. |
| 546 | |
| 547 | @param updater: ChromiumOSUpdater instance used to update the DUT. |
J. Richard Barnette | 7275b61 | 2013-06-04 18:13:11 -0700 | [diff] [blame] | 548 | @param expected_kernel: kernel expected to be active after reboot, |
| 549 | or `None` to skip rollback checking. |
Dan Shi | 0f466e8 | 2013-02-22 15:44:58 -0800 | [diff] [blame] | 550 | |
| 551 | """ |
J. Richard Barnette | 7275b61 | 2013-06-04 18:13:11 -0700 | [diff] [blame] | 552 | # 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 Barnette | 71cc186 | 2015-12-02 10:32:38 -0800 | [diff] [blame] | 556 | autoreboot_cmd = ('FILE="%s" ; [ -f "$FILE" ] || ' |
| 557 | '( touch "$FILE" ; start autoreboot )') |
| 558 | self.run(autoreboot_cmd % self._LAB_MACHINE_FILE) |
Chris Sosa | 6542508 | 2013-10-16 13:26:22 -0700 | [diff] [blame] | 559 | updater.verify_boot_expectations( |
| 560 | expected_kernel, rollback_message= |
Gilad Arnold | c26ae1f | 2015-10-22 16:09:41 -0700 | [diff] [blame] | 561 | 'Build %s failed to boot on %s; system rolled back to previous ' |
Chris Sosa | 6542508 | 2013-10-16 13:26:22 -0700 | [diff] [blame] | 562 | 'build' % (updater.update_version, self.hostname)) |
J. Richard Barnette | 7275b61 | 2013-06-04 18:13:11 -0700 | [diff] [blame] | 563 | # 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 Sosa | 6542508 | 2013-10-16 13:26:22 -0700 | [diff] [blame] | 568 | updater.update_version, |
Dan Shi | 0942b1d | 2015-03-31 11:07:00 -0700 | [diff] [blame] | 569 | self.get_release_version())) |
Dan Shi | 0f466e8 | 2013-02-22 15:44:58 -0800 | [diff] [blame] | 570 | |
Chris Sosa | e92399e | 2015-04-24 11:32:59 -0700 | [diff] [blame] | 571 | 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 Shi | 0f466e8 | 2013-02-22 15:44:58 -0800 | [diff] [blame] | 578 | |
J. Richard Barnette | e4af8b9 | 2013-05-01 13:16:12 -0700 | [diff] [blame] | 579 | def _stage_image_for_update(self, image_name=None): |
Chris Sosa | e92399e | 2015-04-24 11:32:59 -0700 | [diff] [blame] | 580 | """Stage a build on a devserver and return the update_url and devserver. |
Scott Zawalski | eadbf70 | 2013-03-14 09:23:06 -0400 | [diff] [blame] | 581 | |
| 582 | @param image_name: a name like lumpy-release/R27-3837.0.0 |
Chris Sosa | e92399e | 2015-04-24 11:32:59 -0700 | [diff] [blame] | 583 | @returns a tuple with an update URL like: |
Scott Zawalski | eadbf70 | 2013-03-14 09:23:06 -0400 | [diff] [blame] | 584 | http://172.22.50.205:8082/update/lumpy-release/R27-3837.0.0 |
Chris Sosa | e92399e | 2015-04-24 11:32:59 -0700 | [diff] [blame] | 585 | and the devserver instance. |
Scott Zawalski | eadbf70 | 2013-03-14 09:23:06 -0400 | [diff] [blame] | 586 | """ |
J. Richard Barnette | e4af8b9 | 2013-05-01 13:16:12 -0700 | [diff] [blame] | 587 | if not image_name: |
Richard Barnette | 260cbd0 | 2016-10-06 12:23:28 -0700 | [diff] [blame] | 588 | image_name = self._get_cros_repair_image_name() |
J. Richard Barnette | e4af8b9 | 2013-05-01 13:16:12 -0700 | [diff] [blame] | 589 | logging.info('Staging build for AU: %s', image_name) |
Dan Shi | 216389c | 2015-12-22 11:03:06 -0800 | [diff] [blame] | 590 | devserver = dev_server.ImageServer.resolve(image_name, self.hostname) |
Scott Zawalski | eadbf70 | 2013-03-14 09:23:06 -0400 | [diff] [blame] | 591 | devserver.trigger_download(image_name, synchronous=False) |
Chris Sosa | e92399e | 2015-04-24 11:32:59 -0700 | [diff] [blame] | 592 | return (tools.image_url_pattern() % (devserver.url(), image_name), |
| 593 | devserver) |
Scott Zawalski | eadbf70 | 2013-03-14 09:23:06 -0400 | [diff] [blame] | 594 | |
| 595 | |
J. Richard Barnette | e4af8b9 | 2013-05-01 13:16:12 -0700 | [diff] [blame] | 596 | 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 Barnette | 260cbd0 | 2016-10-06 12:23:28 -0700 | [diff] [blame] | 604 | image_name = self._get_cros_repair_image_name() |
J. Richard Barnette | e4af8b9 | 2013-05-01 13:16:12 -0700 | [diff] [blame] | 605 | logging.info('Staging build for servo install: %s', image_name) |
Dan Shi | 216389c | 2015-12-22 11:03:06 -0800 | [diff] [blame] | 606 | devserver = dev_server.ImageServer.resolve(image_name, self.hostname) |
J. Richard Barnette | e4af8b9 | 2013-05-01 13:16:12 -0700 | [diff] [blame] | 607 | devserver.stage_artifacts(image_name, ['test_image']) |
| 608 | return devserver.get_test_image_url(image_name) |
| 609 | |
| 610 | |
beeps | e539be0 | 2013-07-31 21:57:39 -0700 | [diff] [blame] | 611 | 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 |
beeps | 12c0a3c | 2013-09-03 11:58:27 -0700 | [diff] [blame] | 615 | |
beeps | e539be0 | 2013-07-31 21:57:39 -0700 | [diff] [blame] | 616 | @return: An update URL, eg: |
| 617 | http://<devserver>/static/canary-channel/\ |
| 618 | <board>/4262.204.0/factory_test/chromiumos_factory_image.bin |
beeps | 12c0a3c | 2013-09-03 11:58:27 -0700 | [diff] [blame] | 619 | |
| 620 | @raises: ValueError if the factory artifact name is missing from |
| 621 | the config. |
| 622 | |
beeps | e539be0 | 2013-07-31 21:57:39 -0700 | [diff] [blame] | 623 | """ |
| 624 | if not image_name: |
| 625 | logging.error('Need an image_name to stage a factory image.') |
| 626 | return |
| 627 | |
Dan Shi | b8540a5 | 2015-07-16 14:18:23 -0700 | [diff] [blame] | 628 | factory_artifact = CONFIG.get_config_value( |
beeps | 12c0a3c | 2013-09-03 11:58:27 -0700 | [diff] [blame] | 629 | '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 | |
beeps | e539be0 | 2013-07-31 21:57:39 -0700 | [diff] [blame] | 635 | logging.info('Staging build for servo install: %s', image_name) |
Dan Shi | 216389c | 2015-12-22 11:03:06 -0800 | [diff] [blame] | 636 | devserver = dev_server.ImageServer.resolve(image_name, self.hostname) |
beeps | e539be0 | 2013-07-31 21:57:39 -0700 | [diff] [blame] | 637 | devserver.stage_artifacts( |
| 638 | image_name, |
beeps | 12c0a3c | 2013-09-03 11:58:27 -0700 | [diff] [blame] | 639 | [factory_artifact], |
| 640 | archive_url=None) |
beeps | e539be0 | 2013-07-31 21:57:39 -0700 | [diff] [blame] | 641 | |
| 642 | return tools.factory_image_url_pattern() % (devserver.url(), image_name) |
| 643 | |
| 644 | |
xixuan | 016d95b | 2017-02-27 14:37:47 -0800 | [diff] [blame] | 645 | def _get_au_monarch_fields(self, devserver, build): |
| 646 | """Form monarch fields by given devserve & build for auto-update. |
| 647 | |
| 648 | @param devserver: the devserver (ImageServer instance) for auto-update. |
| 649 | @param build: the build to be updated. |
| 650 | |
| 651 | @return A dictionary of monach fields. |
| 652 | """ |
| 653 | try: |
| 654 | board, build_type, milestone, _ = server_utils.ParseBuildName(build) |
| 655 | except server_utils.ParseBuildNameException: |
| 656 | logging.warning('Unable to parse build name %s. Something is ' |
| 657 | 'likely broken, but will continue anyway.', |
| 658 | build) |
| 659 | board, build_type, milestone = ('', '', '') |
| 660 | |
| 661 | monarch_fields = { |
| 662 | 'dev_server': devserver.hostname, |
| 663 | 'board': board, |
| 664 | 'build_type': build_type, |
| 665 | 'milestone': milestone |
| 666 | } |
| 667 | return monarch_fields |
| 668 | |
| 669 | |
xixuan | c30d7e9 | 2017-05-11 12:06:39 -0700 | [diff] [blame] | 670 | def _retry_auto_update_with_new_devserver(self, build, last_devserver, |
xixuan | a3bbc42 | 2017-05-04 15:57:21 -0700 | [diff] [blame] | 671 | force_update, force_full_update, |
| 672 | force_original): |
xixuan | c30d7e9 | 2017-05-11 12:06:39 -0700 | [diff] [blame] | 673 | """Kick off auto-update by devserver and send metrics. |
| 674 | |
| 675 | @param build: the build to update. |
| 676 | @param last_devserver: the last devserver that failed to provision. |
| 677 | @param force_update: see |machine_install_by_devserver|'s force_udpate |
| 678 | for details. |
| 679 | @param force_full_update: see |machine_install_by_devserver|'s |
| 680 | force_full_update for details. |
xixuan | a3bbc42 | 2017-05-04 15:57:21 -0700 | [diff] [blame] | 681 | @param force_original: Whether to force stateful update with the |
| 682 | original payload. |
xixuan | c30d7e9 | 2017-05-11 12:06:39 -0700 | [diff] [blame] | 683 | |
| 684 | @return the result of |auto_update| in dev_server. |
| 685 | """ |
| 686 | devserver = dev_server.resolve( |
| 687 | build, self.hostname, ban_list=[last_devserver.url()]) |
| 688 | devserver.trigger_download(build, synchronous=False) |
| 689 | monarch_fields = self._get_au_monarch_fields(devserver, build) |
| 690 | logging.debug('Retry auto_update: resolved devserver for ' |
| 691 | 'auto-update: %s', devserver.url()) |
| 692 | |
| 693 | # Add metrics |
| 694 | install_with_dev_counter = metrics.Counter( |
| 695 | 'chromeos/autotest/provision/install_with_devserver') |
| 696 | install_with_dev_counter.increment(fields=monarch_fields) |
| 697 | c = metrics.Counter( |
| 698 | 'chromeos/autotest/provision/retry_by_devserver') |
| 699 | monarch_fields['last_devserver'] = last_devserver.hostname |
| 700 | monarch_fields['host'] = self.hostname |
| 701 | c.increment(fields=monarch_fields) |
| 702 | |
xixuan | 9f45534 | 2017-06-08 18:21:44 -0700 | [diff] [blame] | 703 | # Won't retry auto_update in a retry of auto-update. |
| 704 | # In other words, we only retry auto-update once with a different |
| 705 | # devservers. |
| 706 | devserver.auto_update( |
xixuan | c30d7e9 | 2017-05-11 12:06:39 -0700 | [diff] [blame] | 707 | self.hostname, build, |
| 708 | original_board=self.get_board().replace( |
| 709 | ds_constants.BOARD_PREFIX, ''), |
| 710 | original_release_version=self.get_release_version(), |
| 711 | log_dir=self.job.resultdir, |
| 712 | force_update=force_update, |
xixuan | a3bbc42 | 2017-05-04 15:57:21 -0700 | [diff] [blame] | 713 | full_update=force_full_update, |
| 714 | force_original=force_original) |
xixuan | c30d7e9 | 2017-05-11 12:06:39 -0700 | [diff] [blame] | 715 | |
| 716 | |
xixuan | ec801e3 | 2016-08-25 10:20:22 -0700 | [diff] [blame] | 717 | def machine_install_by_devserver(self, update_url=None, force_update=False, |
xixuan | 5dc64ea | 2016-05-20 17:27:51 -0700 | [diff] [blame] | 718 | local_devserver=False, repair=False, |
| 719 | force_full_update=False): |
| 720 | """Ultiize devserver to install the DUT. |
| 721 | |
| 722 | (TODO) crbugs.com/627269: The logic in this function has some overlap |
| 723 | with those in function machine_install. The merge will be done later, |
| 724 | not in the same CL. |
| 725 | |
xixuan | ec801e3 | 2016-08-25 10:20:22 -0700 | [diff] [blame] | 726 | @param update_url: The update_url or build for the host to update. |
xixuan | 5dc64ea | 2016-05-20 17:27:51 -0700 | [diff] [blame] | 727 | @param force_update: Force an update even if the version installed |
| 728 | is the same. Default:False |
| 729 | @param local_devserver: Used by test_that to allow people to |
| 730 | use their local devserver. Default: False |
| 731 | @param repair: Forces update to repair image. Implies force_update. |
| 732 | @param force_full_update: If True, do not attempt to run stateful |
| 733 | update, force a full reimage. If False, try stateful update |
| 734 | first when the dut is already installed with the same version. |
| 735 | @raises autoupdater.ChromiumOSError |
| 736 | |
| 737 | @returns A tuple of (image_name, host_attributes). |
| 738 | image_name is the name of image installed, e.g., |
| 739 | veyron_jerry-release/R50-7871.0.0 |
| 740 | host_attributes is a dictionary of (attribute, value), which |
| 741 | can be saved to afe_host_attributes table in database. This |
| 742 | method returns a dictionary with a single entry of |
| 743 | `job_repo_url`: repo_url, where repo_url is a devserver url to |
| 744 | autotest packages. |
| 745 | """ |
xixuan | 5dc64ea | 2016-05-20 17:27:51 -0700 | [diff] [blame] | 746 | if repair: |
Richard Barnette | 260cbd0 | 2016-10-06 12:23:28 -0700 | [diff] [blame] | 747 | update_url = self._get_cros_repair_image_name() |
xixuan | 5dc64ea | 2016-05-20 17:27:51 -0700 | [diff] [blame] | 748 | force_update = True |
| 749 | |
xixuan | ec801e3 | 2016-08-25 10:20:22 -0700 | [diff] [blame] | 750 | if not update_url and not self._parser.options.image: |
xixuan | 5dc64ea | 2016-05-20 17:27:51 -0700 | [diff] [blame] | 751 | raise error.AutoservError( |
| 752 | 'There is no update URL, nor a method to get one.') |
| 753 | |
xixuan | ec801e3 | 2016-08-25 10:20:22 -0700 | [diff] [blame] | 754 | if not update_url and self._parser.options.image: |
| 755 | update_url = self._parser.options.image |
xixuan | 5dc64ea | 2016-05-20 17:27:51 -0700 | [diff] [blame] | 756 | |
xixuan | 5dc64ea | 2016-05-20 17:27:51 -0700 | [diff] [blame] | 757 | |
| 758 | # Get build from parameter or AFE. |
| 759 | # If the build is not a URL, let devserver to stage it first. |
| 760 | # Otherwise, choose a devserver to trigger auto-update. |
xixuan | ec801e3 | 2016-08-25 10:20:22 -0700 | [diff] [blame] | 761 | build = None |
Aviv Keshet | f937548 | 2016-10-26 02:09:03 -0700 | [diff] [blame] | 762 | devserver = None |
| 763 | logging.debug('Resolving a devserver for auto-update') |
Aviv Keshet | 3a76f76 | 2016-10-27 17:41:51 -0700 | [diff] [blame] | 764 | previously_resolved = False |
xixuan | ec801e3 | 2016-08-25 10:20:22 -0700 | [diff] [blame] | 765 | if update_url.startswith('http://'): |
| 766 | build = autoupdater.url_to_image_name(update_url) |
Aviv Keshet | 3a76f76 | 2016-10-27 17:41:51 -0700 | [diff] [blame] | 767 | previously_resolved = True |
| 768 | else: |
| 769 | build = update_url |
| 770 | devserver = dev_server.resolve(build, self.hostname) |
Allen Li | a5cfb97 | 2016-12-27 17:17:22 -0800 | [diff] [blame] | 771 | server_name = devserver.hostname |
Aviv Keshet | 3a76f76 | 2016-10-27 17:41:51 -0700 | [diff] [blame] | 772 | |
xixuan | 016d95b | 2017-02-27 14:37:47 -0800 | [diff] [blame] | 773 | monarch_fields = self._get_au_monarch_fields(devserver, build) |
Aviv Keshet | 3a76f76 | 2016-10-27 17:41:51 -0700 | [diff] [blame] | 774 | |
| 775 | if previously_resolved: |
xixuan | 2e62d16 | 2016-09-29 19:19:19 -0700 | [diff] [blame] | 776 | # Make sure devserver for Auto-Update has staged the build. |
xixuan | 2e62d16 | 2016-09-29 19:19:19 -0700 | [diff] [blame] | 777 | if server_name not in update_url: |
Aviv Keshet | 3a76f76 | 2016-10-27 17:41:51 -0700 | [diff] [blame] | 778 | logging.debug('Resolved to devserver that does not match ' |
| 779 | 'update_url. The previously resolved devserver ' |
| 780 | 'must be unhealthy. Switching to use devserver %s,' |
| 781 | ' and re-staging.', |
xixuan | 2e62d16 | 2016-09-29 19:19:19 -0700 | [diff] [blame] | 782 | server_name) |
Aviv Keshet | f937548 | 2016-10-26 02:09:03 -0700 | [diff] [blame] | 783 | logging.info('Staging build for AU: %s', update_url) |
xixuan | 2e62d16 | 2016-09-29 19:19:19 -0700 | [diff] [blame] | 784 | devserver.trigger_download(build, synchronous=False) |
Dan Shi | 5e2efb7 | 2017-02-07 11:40:23 -0800 | [diff] [blame] | 785 | c = metrics.Counter( |
| 786 | 'chromeos/autotest/provision/failover_download') |
| 787 | c.increment(fields=monarch_fields) |
xixuan | 5dc64ea | 2016-05-20 17:27:51 -0700 | [diff] [blame] | 788 | else: |
Aviv Keshet | f937548 | 2016-10-26 02:09:03 -0700 | [diff] [blame] | 789 | logging.info('Staging build for AU: %s', update_url) |
xixuan | 5dc64ea | 2016-05-20 17:27:51 -0700 | [diff] [blame] | 790 | devserver.trigger_download(build, synchronous=False) |
Dan Shi | 5e2efb7 | 2017-02-07 11:40:23 -0800 | [diff] [blame] | 791 | c = metrics.Counter('chromeos/autotest/provision/trigger_download') |
| 792 | c.increment(fields=monarch_fields) |
xixuan | 5dc64ea | 2016-05-20 17:27:51 -0700 | [diff] [blame] | 793 | |
| 794 | # Report provision stats. |
xixuan | 016d95b | 2017-02-27 14:37:47 -0800 | [diff] [blame] | 795 | install_with_dev_counter = metrics.Counter( |
| 796 | 'chromeos/autotest/provision/install_with_devserver') |
| 797 | install_with_dev_counter.increment(fields=monarch_fields) |
xixuan | ec801e3 | 2016-08-25 10:20:22 -0700 | [diff] [blame] | 798 | logging.debug('Resolved devserver for auto-update: %s', devserver.url()) |
xixuan | 5dc64ea | 2016-05-20 17:27:51 -0700 | [diff] [blame] | 799 | |
Aviv Keshet | f937548 | 2016-10-26 02:09:03 -0700 | [diff] [blame] | 800 | # and other metrics from this function. |
Dan Shi | 5e2efb7 | 2017-02-07 11:40:23 -0800 | [diff] [blame] | 801 | metrics.Counter('chromeos/autotest/provision/resolve' |
| 802 | ).increment(fields=monarch_fields) |
Aviv Keshet | f937548 | 2016-10-26 02:09:03 -0700 | [diff] [blame] | 803 | |
xixuan | a3bbc42 | 2017-05-04 15:57:21 -0700 | [diff] [blame] | 804 | force_original = self.get_chromeos_release_milestone() is None |
| 805 | |
xixuan | 9f45534 | 2017-06-08 18:21:44 -0700 | [diff] [blame] | 806 | try: |
| 807 | devserver.auto_update( |
| 808 | self.hostname, build, |
| 809 | original_board=self.get_board().replace( |
| 810 | ds_constants.BOARD_PREFIX, ''), |
| 811 | original_release_version=self.get_release_version(), |
| 812 | log_dir=self.job.resultdir, |
| 813 | force_update=force_update, |
| 814 | full_update=force_full_update, |
| 815 | force_original=force_original) |
| 816 | except dev_server.RetryableProvisionException: |
| 817 | # It indicates that last provision failed due to devserver load |
| 818 | # issue, so another devserver is resolved to kick off provision |
| 819 | # job once again and only once. |
| 820 | logging.debug('Provision failed due to devserver issue,' |
| 821 | 'retry it with another devserver.') |
xixuan | 016d95b | 2017-02-27 14:37:47 -0800 | [diff] [blame] | 822 | |
xixuan | 9f45534 | 2017-06-08 18:21:44 -0700 | [diff] [blame] | 823 | # Check first whether this DUT is completely offline. If so, skip |
| 824 | # the following provision tries. |
| 825 | logging.debug('Checking whether host %s is online.', self.hostname) |
| 826 | if utils.ping(self.hostname, tries=1, deadline=1) == 0: |
| 827 | self._retry_auto_update_with_new_devserver( |
| 828 | build, devserver, force_update, force_full_update, |
| 829 | force_original) |
| 830 | else: |
| 831 | raise error.AutoservError( |
| 832 | 'No answer to ping from %s' % self.hostname) |
xixuan | 5dc64ea | 2016-05-20 17:27:51 -0700 | [diff] [blame] | 833 | |
| 834 | # The reason to resolve a new devserver in function machine_install |
| 835 | # is mostly because that the update_url there may has a strange format, |
| 836 | # and it's hard to parse the devserver url from it. |
| 837 | # Since we already resolve a devserver to trigger auto-update, the same |
| 838 | # devserver is used to form JOB_REPO_URL here. Verified in local test. |
| 839 | repo_url = tools.get_package_url(devserver.url(), build) |
| 840 | return build, {ds_constants.JOB_REPO_URL: repo_url} |
| 841 | |
| 842 | |
Laurence Goodby | 778c9a4 | 2017-05-24 19:24:07 -0700 | [diff] [blame] | 843 | def prepare_for_update(self): |
| 844 | """Prepares the DUT for an update. |
| 845 | |
| 846 | Subclasses may override this to perform any special actions |
| 847 | required before updating. |
| 848 | """ |
Laurence Goodby | 468de25 | 2017-06-08 17:22:53 -0700 | [diff] [blame] | 849 | pass |
Laurence Goodby | 778c9a4 | 2017-05-24 19:24:07 -0700 | [diff] [blame] | 850 | |
| 851 | |
Chris Sosa | a3ac215 | 2012-05-23 22:23:13 -0700 | [diff] [blame] | 852 | def machine_install(self, update_url=None, force_update=False, |
Richard Barnette | 0b023a7 | 2015-04-24 16:07:30 +0000 | [diff] [blame] | 853 | local_devserver=False, repair=False, |
| 854 | force_full_update=False): |
Scott Zawalski | 89c44dd | 2013-02-26 09:28:02 -0500 | [diff] [blame] | 855 | """Install the DUT. |
| 856 | |
Dan Shi | 0f466e8 | 2013-02-22 15:44:58 -0800 | [diff] [blame] | 857 | Use stateful update if the DUT is already running the same build. |
| 858 | Stateful update does not update kernel and tends to run much faster |
| 859 | than a full reimage. If the DUT is running a different build, or it |
| 860 | failed to do a stateful update, full update, including kernel update, |
| 861 | will be applied to the DUT. |
| 862 | |
Simran Basi | 5ace6f2 | 2016-01-06 17:30:44 -0800 | [diff] [blame] | 863 | Once a host enters machine_install its host attribute job_repo_url |
| 864 | (used for package install) will be removed and then updated. |
Scott Zawalski | eadbf70 | 2013-03-14 09:23:06 -0400 | [diff] [blame] | 865 | |
Scott Zawalski | 89c44dd | 2013-02-26 09:28:02 -0500 | [diff] [blame] | 866 | @param update_url: The url to use for the update |
| 867 | pattern: http://$devserver:###/update/$build |
| 868 | If update_url is None and repair is True we will install the |
Dan Shi | 6964fa5 | 2014-12-18 11:04:27 -0800 | [diff] [blame] | 869 | stable image listed in afe_stable_versions table. If the table |
| 870 | is not setup, global_config value under CROS.stable_cros_version |
| 871 | will be used instead. |
Scott Zawalski | 89c44dd | 2013-02-26 09:28:02 -0500 | [diff] [blame] | 872 | @param force_update: Force an update even if the version installed |
| 873 | is the same. Default:False |
Christopher Wiley | 6a4ff93 | 2015-05-15 14:00:47 -0700 | [diff] [blame] | 874 | @param local_devserver: Used by test_that to allow people to |
Scott Zawalski | 89c44dd | 2013-02-26 09:28:02 -0500 | [diff] [blame] | 875 | use their local devserver. Default: False |
Chris Sosa | e92399e | 2015-04-24 11:32:59 -0700 | [diff] [blame] | 876 | @param repair: Forces update to repair image. Implies force_update. |
Fang Deng | 3d3b927 | 2014-12-22 12:20:28 -0800 | [diff] [blame] | 877 | @param force_full_update: If True, do not attempt to run stateful |
| 878 | update, force a full reimage. If False, try stateful update |
| 879 | first when the dut is already installed with the same version. |
Scott Zawalski | 89c44dd | 2013-02-26 09:28:02 -0500 | [diff] [blame] | 880 | @raises autoupdater.ChromiumOSError |
| 881 | |
Dan Shi | be3636a | 2016-02-14 22:48:01 -0800 | [diff] [blame] | 882 | @returns A tuple of (image_name, host_attributes). |
| 883 | image_name is the name of image installed, e.g., |
| 884 | veyron_jerry-release/R50-7871.0.0 |
| 885 | host_attributes is a dictionary of (attribute, value), which |
| 886 | can be saved to afe_host_attributes table in database. This |
| 887 | method returns a dictionary with a single entry of |
| 888 | `job_repo_url`: repo_url, where repo_url is a devserver url to |
| 889 | autotest packages. |
Scott Zawalski | 89c44dd | 2013-02-26 09:28:02 -0500 | [diff] [blame] | 890 | """ |
Chris Sosa | e92399e | 2015-04-24 11:32:59 -0700 | [diff] [blame] | 891 | devserver = None |
Richard Barnette | 0b023a7 | 2015-04-24 16:07:30 +0000 | [diff] [blame] | 892 | if repair: |
Chris Sosa | e92399e | 2015-04-24 11:32:59 -0700 | [diff] [blame] | 893 | update_url, devserver = self._stage_image_for_update() |
Richard Barnette | 0b023a7 | 2015-04-24 16:07:30 +0000 | [diff] [blame] | 894 | force_update = True |
Dan Shi | 0f466e8 | 2013-02-22 15:44:58 -0800 | [diff] [blame] | 895 | |
Chris Sosa | e92399e | 2015-04-24 11:32:59 -0700 | [diff] [blame] | 896 | if not update_url and not self._parser.options.image: |
| 897 | raise error.AutoservError( |
Dan Shi | d07ee2e | 2015-09-24 14:49:25 -0700 | [diff] [blame] | 898 | 'There is no update URL, nor a method to get one.') |
Chris Sosa | e92399e | 2015-04-24 11:32:59 -0700 | [diff] [blame] | 899 | |
| 900 | if not update_url and self._parser.options.image: |
| 901 | # This is the base case where we have no given update URL i.e. |
| 902 | # dynamic suites logic etc. This is the most flexible case where we |
| 903 | # can serve an update from any of our fleet of devservers. |
| 904 | requested_build = self._parser.options.image |
| 905 | if not requested_build.startswith('http://'): |
| 906 | logging.debug('Update will be staged for this installation') |
| 907 | update_url, devserver = self._stage_image_for_update( |
Dan Shi | d07ee2e | 2015-09-24 14:49:25 -0700 | [diff] [blame] | 908 | requested_build) |
Chris Sosa | e92399e | 2015-04-24 11:32:59 -0700 | [diff] [blame] | 909 | else: |
| 910 | update_url = requested_build |
| 911 | |
| 912 | logging.debug('Update URL is %s', update_url) |
| 913 | |
Dan Shi | f48f813 | 2016-02-18 10:34:30 -0800 | [diff] [blame] | 914 | # Report provision stats. |
Allen Li | a5cfb97 | 2016-12-27 17:17:22 -0800 | [diff] [blame] | 915 | server_name = dev_server.get_hostname(update_url) |
Allen Li | 48a13fe | 2016-11-22 14:10:40 -0800 | [diff] [blame] | 916 | (metrics.Counter('chromeos/autotest/provision/install') |
| 917 | .increment(fields={'devserver': server_name})) |
Dan Shi | f48f813 | 2016-02-18 10:34:30 -0800 | [diff] [blame] | 918 | |
Dan Shi | d07ee2e | 2015-09-24 14:49:25 -0700 | [diff] [blame] | 919 | # Create a file to indicate if provision fails. The file will be removed |
| 920 | # by stateful update or full install. |
J. Richard Barnette | 91137f0 | 2016-03-10 16:52:26 -0800 | [diff] [blame] | 921 | self.run('touch %s' % self.PROVISION_FAILED) |
Dan Shi | d07ee2e | 2015-09-24 14:49:25 -0700 | [diff] [blame] | 922 | |
Chris Sosa | e92399e | 2015-04-24 11:32:59 -0700 | [diff] [blame] | 923 | update_complete = False |
| 924 | updater = autoupdater.ChromiumOSUpdater( |
| 925 | update_url, host=self, local_devserver=local_devserver) |
Fang Deng | 3d3b927 | 2014-12-22 12:20:28 -0800 | [diff] [blame] | 926 | if not force_full_update: |
| 927 | try: |
Chris Sosa | e92399e | 2015-04-24 11:32:59 -0700 | [diff] [blame] | 928 | # If the DUT is already running the same build, try stateful |
| 929 | # update first as it's much quicker than a full re-image. |
| 930 | update_complete = self._try_stateful_update( |
Dan Shi | d07ee2e | 2015-09-24 14:49:25 -0700 | [diff] [blame] | 931 | update_url, force_update, updater) |
Fang Deng | 3d3b927 | 2014-12-22 12:20:28 -0800 | [diff] [blame] | 932 | except Exception as e: |
| 933 | logging.exception(e) |
J. Richard Barnette | 45e93de | 2012-04-11 17:24:15 -0700 | [diff] [blame] | 934 | |
Dan Shi | 0f466e8 | 2013-02-22 15:44:58 -0800 | [diff] [blame] | 935 | inactive_kernel = None |
Chris Sosa | e92399e | 2015-04-24 11:32:59 -0700 | [diff] [blame] | 936 | if update_complete or (not force_update and updater.check_version()): |
| 937 | logging.info('Install complete without full update') |
| 938 | else: |
| 939 | logging.info('DUT requires full update.') |
| 940 | self.reboot(timeout=self.REBOOT_TIMEOUT, wait=True) |
Laurence Goodby | 778c9a4 | 2017-05-24 19:24:07 -0700 | [diff] [blame] | 941 | self.prepare_for_update() |
Dan Shi | 10b9848 | 2016-02-02 14:38:50 -0800 | [diff] [blame] | 942 | |
Chris Sosa | e92399e | 2015-04-24 11:32:59 -0700 | [diff] [blame] | 943 | num_of_attempts = provision.FLAKY_DEVSERVER_ATTEMPTS |
Chris Sosa | b7612bc | 2013-03-21 10:32:37 -0700 | [diff] [blame] | 944 | |
Chris Sosa | e92399e | 2015-04-24 11:32:59 -0700 | [diff] [blame] | 945 | while num_of_attempts > 0: |
| 946 | num_of_attempts -= 1 |
| 947 | try: |
| 948 | updater.run_update() |
| 949 | except Exception: |
| 950 | logging.warn('Autoupdate did not complete.') |
| 951 | # Do additional check for the devserver health. Ideally, |
| 952 | # the autoupdater.py could raise an exception when it |
| 953 | # detected network flake but that would require |
| 954 | # instrumenting the update engine and parsing it log. |
| 955 | if (num_of_attempts <= 0 or |
| 956 | devserver is None or |
xixuan | 9e2c98d | 2016-02-26 19:04:53 -0800 | [diff] [blame] | 957 | dev_server.ImageServer.devserver_healthy( |
Chris Sosa | e92399e | 2015-04-24 11:32:59 -0700 | [diff] [blame] | 958 | devserver.url())): |
Dan Shi | d07ee2e | 2015-09-24 14:49:25 -0700 | [diff] [blame] | 959 | raise |
J. Richard Barnette | 45e93de | 2012-04-11 17:24:15 -0700 | [diff] [blame] | 960 | |
Chris Sosa | e92399e | 2015-04-24 11:32:59 -0700 | [diff] [blame] | 961 | logging.warn('Devserver looks unhealthy. Trying another') |
| 962 | update_url, devserver = self._stage_image_for_update( |
| 963 | requested_build) |
| 964 | logging.debug('New Update URL is %s', update_url) |
| 965 | updater = autoupdater.ChromiumOSUpdater( |
| 966 | update_url, host=self, |
| 967 | local_devserver=local_devserver) |
| 968 | else: |
| 969 | break |
J. Richard Barnette | 45e93de | 2012-04-11 17:24:15 -0700 | [diff] [blame] | 970 | |
Chris Sosa | e92399e | 2015-04-24 11:32:59 -0700 | [diff] [blame] | 971 | # Give it some time in case of IO issues. |
| 972 | time.sleep(10) |
Dan Shi | 5699ac2 | 2014-12-19 10:55:49 -0800 | [diff] [blame] | 973 | |
Chris Sosa | e92399e | 2015-04-24 11:32:59 -0700 | [diff] [blame] | 974 | # Figure out active and inactive kernel. |
| 975 | active_kernel, inactive_kernel = updater.get_kernel_state() |
Simran Basi | 13fa1ba | 2013-03-04 10:56:47 -0800 | [diff] [blame] | 976 | |
Chris Sosa | e92399e | 2015-04-24 11:32:59 -0700 | [diff] [blame] | 977 | # Ensure inactive kernel has higher priority than active. |
| 978 | if (updater.get_kernel_priority(inactive_kernel) |
| 979 | < updater.get_kernel_priority(active_kernel)): |
| 980 | raise autoupdater.ChromiumOSError( |
| 981 | 'Update failed. The priority of the inactive kernel' |
| 982 | ' partition is less than that of the active kernel' |
| 983 | ' partition.') |
| 984 | |
| 985 | # Updater has returned successfully; reboot the host. |
Richard Barnette | fc3d743 | 2016-07-07 10:21:31 -0700 | [diff] [blame] | 986 | # |
| 987 | # Regarding the 'crossystem' command: In some cases, the |
| 988 | # TPM gets into a state such that it fails verification. |
| 989 | # We don't know why. However, this call papers over the |
| 990 | # problem by clearing the TPM during the reboot. |
| 991 | # |
| 992 | # We ignore failures from 'crossystem'. Although failure |
| 993 | # here is unexpected, and could signal a bug, the point |
| 994 | # of the exercise is to paper over problems; allowing |
| 995 | # this to fail would defeat the purpose. |
| 996 | self.run('crossystem clear_tpm_owner_request=1', |
| 997 | ignore_status=True) |
Chris Sosa | e92399e | 2015-04-24 11:32:59 -0700 | [diff] [blame] | 998 | self.reboot(timeout=self.REBOOT_TIMEOUT, wait=True) |
| 999 | |
| 1000 | self._post_update_processing(updater, inactive_kernel) |
Simran Basi | 5ace6f2 | 2016-01-06 17:30:44 -0800 | [diff] [blame] | 1001 | image_name = autoupdater.url_to_image_name(update_url) |
Dan Shi | be3636a | 2016-02-14 22:48:01 -0800 | [diff] [blame] | 1002 | # update_url is different from devserver url needed to stage autotest |
| 1003 | # packages, therefore, resolve a new devserver url here. |
| 1004 | devserver_url = dev_server.ImageServer.resolve(image_name, |
| 1005 | self.hostname).url() |
| 1006 | repo_url = tools.get_package_url(devserver_url, image_name) |
| 1007 | return image_name, {ds_constants.JOB_REPO_URL: repo_url} |
J. Richard Barnette | 45e93de | 2012-04-11 17:24:15 -0700 | [diff] [blame] | 1008 | |
| 1009 | |
Tom Wai-Hong Tam | 2d00cb2 | 2016-01-08 06:40:50 +0800 | [diff] [blame] | 1010 | def _clear_fw_version_labels(self, rw_only): |
| 1011 | """Clear firmware version labels from the machine. |
| 1012 | |
| 1013 | @param rw_only: True to only clear fwrw_version; otherewise, clear |
| 1014 | both fwro_version and fwrw_version. |
| 1015 | """ |
Dan Shi | 9cb0eec | 2014-06-03 09:04:50 -0700 | [diff] [blame] | 1016 | labels = self._AFE.get_labels( |
Dan Shi | 0723bf5 | 2015-06-24 10:52:38 -0700 | [diff] [blame] | 1017 | name__startswith=provision.FW_RW_VERSION_PREFIX, |
Dan Shi | 9cb0eec | 2014-06-03 09:04:50 -0700 | [diff] [blame] | 1018 | host__hostname=self.hostname) |
Tom Wai-Hong Tam | 2d00cb2 | 2016-01-08 06:40:50 +0800 | [diff] [blame] | 1019 | if not rw_only: |
| 1020 | labels = labels + self._AFE.get_labels( |
| 1021 | name__startswith=provision.FW_RO_VERSION_PREFIX, |
| 1022 | host__hostname=self.hostname) |
Dan Shi | 9cb0eec | 2014-06-03 09:04:50 -0700 | [diff] [blame] | 1023 | for label in labels: |
| 1024 | label.remove_hosts(hosts=[self.hostname]) |
| 1025 | |
| 1026 | |
Tom Wai-Hong Tam | 2d00cb2 | 2016-01-08 06:40:50 +0800 | [diff] [blame] | 1027 | def _add_fw_version_label(self, build, rw_only): |
Dan Shi | 9cb0eec | 2014-06-03 09:04:50 -0700 | [diff] [blame] | 1028 | """Add firmware version label to the machine. |
| 1029 | |
| 1030 | @param build: Build of firmware. |
Tom Wai-Hong Tam | 2d00cb2 | 2016-01-08 06:40:50 +0800 | [diff] [blame] | 1031 | @param rw_only: True to only add fwrw_version; otherwise, add both |
| 1032 | fwro_version and fwrw_version. |
Dan Shi | 9cb0eec | 2014-06-03 09:04:50 -0700 | [diff] [blame] | 1033 | |
| 1034 | """ |
Prathmesh Prabhu | 2c7471d | 2016-11-15 20:19:57 +0000 | [diff] [blame] | 1035 | fw_label = provision.fwrw_version_to_label(build) |
MK Ryu | 73be986 | 2015-07-06 12:25:00 -0700 | [diff] [blame] | 1036 | self._AFE.run('label_add_hosts', id=fw_label, hosts=[self.hostname]) |
Tom Wai-Hong Tam | 2d00cb2 | 2016-01-08 06:40:50 +0800 | [diff] [blame] | 1037 | if not rw_only: |
Prathmesh Prabhu | 2c7471d | 2016-11-15 20:19:57 +0000 | [diff] [blame] | 1038 | fw_label = provision.fwro_version_to_label(build) |
Tom Wai-Hong Tam | 2d00cb2 | 2016-01-08 06:40:50 +0800 | [diff] [blame] | 1039 | self._AFE.run('label_add_hosts', id=fw_label, hosts=[self.hostname]) |
Dan Shi | 9cb0eec | 2014-06-03 09:04:50 -0700 | [diff] [blame] | 1040 | |
| 1041 | |
Tom Wai-Hong Tam | 4ac7898 | 2016-01-08 02:34:37 +0800 | [diff] [blame] | 1042 | def firmware_install(self, build=None, rw_only=False): |
Dan Shi | 9cb0eec | 2014-06-03 09:04:50 -0700 | [diff] [blame] | 1043 | """Install firmware to the DUT. |
| 1044 | |
| 1045 | Use stateful update if the DUT is already running the same build. |
| 1046 | Stateful update does not update kernel and tends to run much faster |
| 1047 | than a full reimage. If the DUT is running a different build, or it |
| 1048 | failed to do a stateful update, full update, including kernel update, |
| 1049 | will be applied to the DUT. |
| 1050 | |
Tom Wai-Hong Tam | 2d00cb2 | 2016-01-08 06:40:50 +0800 | [diff] [blame] | 1051 | Once a host enters firmware_install its fw[ro|rw]_version label will |
| 1052 | be removed. After the firmware is updated successfully, a new |
| 1053 | fw[ro|rw]_version label will be added to the host. |
Dan Shi | 9cb0eec | 2014-06-03 09:04:50 -0700 | [diff] [blame] | 1054 | |
| 1055 | @param build: The build version to which we want to provision the |
| 1056 | firmware of the machine, |
| 1057 | e.g. 'link-firmware/R22-2695.1.144'. |
Tom Wai-Hong Tam | 4ac7898 | 2016-01-08 02:34:37 +0800 | [diff] [blame] | 1058 | @param rw_only: True to only install firmware to its RW portions. Keep |
| 1059 | the RO portions unchanged. |
Dan Shi | 9cb0eec | 2014-06-03 09:04:50 -0700 | [diff] [blame] | 1060 | |
| 1061 | TODO(dshi): After bug 381718 is fixed, update here with corresponding |
| 1062 | exceptions that could be raised. |
| 1063 | |
| 1064 | """ |
| 1065 | if not self.servo: |
| 1066 | raise error.TestError('Host %s does not have servo.' % |
| 1067 | self.hostname) |
| 1068 | |
Wai-Hong Tam | 64cf791 | 2017-05-23 16:41:40 -0700 | [diff] [blame] | 1069 | # Get the DUT board name from servod. |
| 1070 | board = self.servo.get_board() |
Dan Shi | 9cb0eec | 2014-06-03 09:04:50 -0700 | [diff] [blame] | 1071 | |
Chris Sosa | e92399e | 2015-04-24 11:32:59 -0700 | [diff] [blame] | 1072 | # If build is not set, try to install firmware from stable CrOS. |
Dan Shi | 9cb0eec | 2014-06-03 09:04:50 -0700 | [diff] [blame] | 1073 | if not build: |
Richard Barnette | 260cbd0 | 2016-10-06 12:23:28 -0700 | [diff] [blame] | 1074 | build = afe_utils.get_stable_faft_version(board) |
Dan Shi | 3d7a0e1 | 2015-10-12 11:55:45 -0700 | [diff] [blame] | 1075 | if not build: |
| 1076 | raise error.TestError( |
| 1077 | 'Failed to find stable firmware build for %s.', |
| 1078 | self.hostname) |
| 1079 | logging.info('Will install firmware from build %s.', build) |
Dan Shi | 9cb0eec | 2014-06-03 09:04:50 -0700 | [diff] [blame] | 1080 | |
| 1081 | config = FAFTConfig(board) |
| 1082 | if config.use_u_boot: |
| 1083 | ap_image = 'image-%s.bin' % board |
| 1084 | else: # Depthcharge platform |
| 1085 | ap_image = 'image.bin' |
| 1086 | ec_image = 'ec.bin' |
Dan Shi | 216389c | 2015-12-22 11:03:06 -0800 | [diff] [blame] | 1087 | ds = dev_server.ImageServer.resolve(build, self.hostname) |
Dan Shi | 9cb0eec | 2014-06-03 09:04:50 -0700 | [diff] [blame] | 1088 | ds.stage_artifacts(build, ['firmware']) |
| 1089 | |
| 1090 | tmpd = autotemp.tempdir(unique_id='fwimage') |
| 1091 | try: |
| 1092 | fwurl = self._FW_IMAGE_URL_PATTERN % (ds.url(), build) |
| 1093 | local_tarball = os.path.join(tmpd.name, os.path.basename(fwurl)) |
xixuan | 4e11682 | 2016-11-17 15:32:10 -0800 | [diff] [blame] | 1094 | ds.download_file(fwurl, local_tarball) |
Dan Shi | 9cb0eec | 2014-06-03 09:04:50 -0700 | [diff] [blame] | 1095 | |
Tom Wai-Hong Tam | 2d00cb2 | 2016-01-08 06:40:50 +0800 | [diff] [blame] | 1096 | self._clear_fw_version_labels(rw_only) |
Tom Wai-Hong Tam | 552fee1 | 2016-05-11 06:27:58 +0800 | [diff] [blame] | 1097 | if config.chrome_ec: |
Tom Wai-Hong Tam | f9db57a | 2016-03-17 05:32:22 +0800 | [diff] [blame] | 1098 | logging.info('Will re-program EC %snow', 'RW ' if rw_only else '') |
| 1099 | server_utils.system('tar xf %s -C %s %s' % |
| 1100 | (local_tarball, tmpd.name, ec_image), |
| 1101 | timeout=60) |
| 1102 | self.servo.program_ec(os.path.join(tmpd.name, ec_image), rw_only) |
| 1103 | else: |
| 1104 | logging.info('Not a Chrome EC, ignore re-programing it') |
Tom Wai-Hong Tam | 4ac7898 | 2016-01-08 02:34:37 +0800 | [diff] [blame] | 1105 | logging.info('Will re-program BIOS %snow', 'RW ' if rw_only else '') |
Tom Wai-Hong Tam | f9db57a | 2016-03-17 05:32:22 +0800 | [diff] [blame] | 1106 | server_utils.system('tar xf %s -C %s %s' % |
| 1107 | (local_tarball, tmpd.name, ap_image), |
| 1108 | timeout=60) |
Tom Wai-Hong Tam | 4ac7898 | 2016-01-08 02:34:37 +0800 | [diff] [blame] | 1109 | self.servo.program_bios(os.path.join(tmpd.name, ap_image), rw_only) |
Dan Shi | 9cb0eec | 2014-06-03 09:04:50 -0700 | [diff] [blame] | 1110 | self.servo.get_power_state_controller().reset() |
| 1111 | time.sleep(self.servo.BOOT_DELAY) |
Wai-Hong Tam | b5f66ce | 2016-11-10 15:45:30 -0800 | [diff] [blame] | 1112 | if utils.host_is_in_lab_zone(self.hostname): |
| 1113 | self._add_fw_version_label(build, rw_only) |
Dan Shi | 9cb0eec | 2014-06-03 09:04:50 -0700 | [diff] [blame] | 1114 | finally: |
| 1115 | tmpd.clean() |
| 1116 | |
| 1117 | |
Dan Shi | 10e992b | 2013-08-30 11:02:59 -0700 | [diff] [blame] | 1118 | def show_update_engine_log(self): |
| 1119 | """Output update engine log.""" |
MK Ryu | 35d661e | 2014-09-25 17:44:10 -0700 | [diff] [blame] | 1120 | logging.debug('Dumping %s', client_constants.UPDATE_ENGINE_LOG) |
| 1121 | self.run('cat %s' % client_constants.UPDATE_ENGINE_LOG) |
Dan Shi | 10e992b | 2013-08-30 11:02:59 -0700 | [diff] [blame] | 1122 | |
| 1123 | |
beeps | f079cfb | 2013-09-18 17:49:51 -0700 | [diff] [blame] | 1124 | def servo_install(self, image_url=None, usb_boot_timeout=USB_BOOT_TIMEOUT, |
| 1125 | install_timeout=INSTALL_TIMEOUT): |
Scott Zawalski | 62bacae | 2013-03-05 10:40:32 -0500 | [diff] [blame] | 1126 | """ |
| 1127 | Re-install the OS on the DUT by: |
| 1128 | 1) installing a test image on a USB storage device attached to the Servo |
| 1129 | board, |
Richard Barnette | 03a0c13 | 2012-11-05 12:40:35 -0800 | [diff] [blame] | 1130 | 2) booting that image in recovery mode, and then |
J. Richard Barnette | 31b2e31 | 2013-04-04 16:05:22 -0700 | [diff] [blame] | 1131 | 3) installing the image with chromeos-install. |
| 1132 | |
Scott Zawalski | 62bacae | 2013-03-05 10:40:32 -0500 | [diff] [blame] | 1133 | @param image_url: If specified use as the url to install on the DUT. |
| 1134 | otherwise boot the currently staged image on the USB stick. |
beeps | f079cfb | 2013-09-18 17:49:51 -0700 | [diff] [blame] | 1135 | @param usb_boot_timeout: The usb_boot_timeout to use during reimage. |
| 1136 | Factory images need a longer usb_boot_timeout than regular |
| 1137 | cros images. |
| 1138 | @param install_timeout: The timeout to use when installing the chromeos |
| 1139 | image. Factory images need a longer install_timeout. |
Richard Barnette | 03a0c13 | 2012-11-05 12:40:35 -0800 | [diff] [blame] | 1140 | |
Scott Zawalski | 62bacae | 2013-03-05 10:40:32 -0500 | [diff] [blame] | 1141 | @raises AutoservError if the image fails to boot. |
beeps | f079cfb | 2013-09-18 17:49:51 -0700 | [diff] [blame] | 1142 | |
J. Richard Barnette | 0199cc8 | 2014-12-05 17:08:40 -0800 | [diff] [blame] | 1143 | """ |
beeps | f079cfb | 2013-09-18 17:49:51 -0700 | [diff] [blame] | 1144 | logging.info('Downloading image to USB, then booting from it. Usb boot ' |
| 1145 | 'timeout = %s', usb_boot_timeout) |
Allen Li | 48a13fe | 2016-11-22 14:10:40 -0800 | [diff] [blame] | 1146 | with metrics.SecondsTimer( |
| 1147 | 'chromeos/autotest/provision/servo_install/boot_duration'): |
| 1148 | self.servo.install_recovery_image(image_url) |
| 1149 | if not self.wait_up(timeout=usb_boot_timeout): |
| 1150 | raise hosts.AutoservRepairError( |
| 1151 | 'DUT failed to boot from USB after %d seconds' % |
| 1152 | usb_boot_timeout) |
Scott Zawalski | 62bacae | 2013-03-05 10:40:32 -0500 | [diff] [blame] | 1153 | |
Tom Wai-Hong Tam | f6b4f81 | 2015-08-08 04:14:59 +0800 | [diff] [blame] | 1154 | # The new chromeos-tpm-recovery has been merged since R44-7073.0.0. |
| 1155 | # In old CrOS images, this command fails. Skip the error. |
Tom Wai-Hong Tam | 27af733 | 2015-07-25 06:09:39 +0800 | [diff] [blame] | 1156 | logging.info('Resetting the TPM status') |
Tom Wai-Hong Tam | f6b4f81 | 2015-08-08 04:14:59 +0800 | [diff] [blame] | 1157 | try: |
| 1158 | self.run('chromeos-tpm-recovery') |
| 1159 | except error.AutoservRunError: |
| 1160 | logging.warn('chromeos-tpm-recovery is too old.') |
| 1161 | |
Tom Wai-Hong Tam | 27af733 | 2015-07-25 06:09:39 +0800 | [diff] [blame] | 1162 | |
Allen Li | 48a13fe | 2016-11-22 14:10:40 -0800 | [diff] [blame] | 1163 | with metrics.SecondsTimer( |
| 1164 | 'chromeos/autotest/provision/servo_install/install_duration'): |
| 1165 | logging.info('Installing image through chromeos-install.') |
| 1166 | self.run('chromeos-install --yes', timeout=install_timeout) |
| 1167 | self.halt() |
beeps | f079cfb | 2013-09-18 17:49:51 -0700 | [diff] [blame] | 1168 | |
| 1169 | logging.info('Power cycling DUT through servo.') |
J. Richard Barnette | 0199cc8 | 2014-12-05 17:08:40 -0800 | [diff] [blame] | 1170 | self.servo.get_power_state_controller().power_off() |
Fang Deng | afb8814 | 2013-05-30 17:44:31 -0700 | [diff] [blame] | 1171 | self.servo.switch_usbkey('off') |
J. Richard Barnette | 0199cc8 | 2014-12-05 17:08:40 -0800 | [diff] [blame] | 1172 | # N.B. The Servo API requires that we use power_on() here |
| 1173 | # for two reasons: |
| 1174 | # 1) After turning on a DUT in recovery mode, you must turn |
| 1175 | # it off and then on with power_on() once more to |
| 1176 | # disable recovery mode (this is a Parrot specific |
| 1177 | # requirement). |
| 1178 | # 2) After power_off(), the only way to turn on is with |
| 1179 | # power_on() (this is a Storm specific requirement). |
J. Richard Barnette | fbcc712 | 2013-07-24 18:24:59 -0700 | [diff] [blame] | 1180 | self.servo.get_power_state_controller().power_on() |
beeps | f079cfb | 2013-09-18 17:49:51 -0700 | [diff] [blame] | 1181 | |
| 1182 | logging.info('Waiting for DUT to come back up.') |
Richard Barnette | 03a0c13 | 2012-11-05 12:40:35 -0800 | [diff] [blame] | 1183 | if not self.wait_up(timeout=self.BOOT_TIMEOUT): |
| 1184 | raise error.AutoservError('DUT failed to reboot installed ' |
| 1185 | 'test image after %d seconds' % |
Scott Zawalski | 62bacae | 2013-03-05 10:40:32 -0500 | [diff] [blame] | 1186 | self.BOOT_TIMEOUT) |
| 1187 | |
| 1188 | |
Richard Barnette | 9a26ad6 | 2016-06-10 12:03:08 -0700 | [diff] [blame] | 1189 | def repair_servo(self): |
Dan Shi | 9046635 | 2015-09-22 15:01:05 -0700 | [diff] [blame] | 1190 | """ |
Richard Barnette | 9a26ad6 | 2016-06-10 12:03:08 -0700 | [diff] [blame] | 1191 | Confirm that servo is initialized and verified. |
Dan Shi | 9046635 | 2015-09-22 15:01:05 -0700 | [diff] [blame] | 1192 | |
Richard Barnette | 9a26ad6 | 2016-06-10 12:03:08 -0700 | [diff] [blame] | 1193 | If the servo object is missing, attempt to repair the servo |
| 1194 | host. Repair failures are passed back to the caller. |
| 1195 | |
| 1196 | @raise AutoservError: If there is no servo host for this CrOS |
| 1197 | host. |
| 1198 | """ |
| 1199 | if self.servo: |
| 1200 | return |
| 1201 | if not self._servo_host: |
| 1202 | raise error.AutoservError('No servo host for %s.' % |
| 1203 | self.hostname) |
| 1204 | self._servo_host.repair() |
| 1205 | self.servo = self._servo_host.get_servo() |
Dan Shi | 9046635 | 2015-09-22 15:01:05 -0700 | [diff] [blame] | 1206 | |
| 1207 | |
J. Richard Barnette | c2d99cf | 2015-11-18 12:46:15 -0800 | [diff] [blame] | 1208 | def repair(self): |
| 1209 | """Attempt to get the DUT to pass `self.verify()`. |
Richard Barnette | 82c3591 | 2012-11-20 10:09:10 -0800 | [diff] [blame] | 1210 | |
| 1211 | This overrides the base class function for repair; it does |
J. Richard Barnette | 91137f0 | 2016-03-10 16:52:26 -0800 | [diff] [blame] | 1212 | not call back to the parent class, but instead relies on |
| 1213 | `self._repair_strategy` to coordinate the verification and |
| 1214 | repair steps needed to get the DUT working. |
Richard Barnette | 82c3591 | 2012-11-20 10:09:10 -0800 | [diff] [blame] | 1215 | """ |
J. Richard Barnette | 91137f0 | 2016-03-10 16:52:26 -0800 | [diff] [blame] | 1216 | self._repair_strategy.repair(self) |
Richard Barnette | bfab867 | 2017-08-10 14:48:28 -0700 | [diff] [blame] | 1217 | # Sometimes, hosts with certain ethernet dongles get stuck in a |
| 1218 | # bad network state where they're reachable from this code, but |
| 1219 | # not from the devservers during provisioning. Rebooting the |
| 1220 | # DUT fixes it. |
| 1221 | # |
| 1222 | # TODO(jrbarnette): Ideally, we'd get rid of the problem |
| 1223 | # dongles, and drop this code. Failing that, we could be smart |
| 1224 | # enough not to reboot if repair rebooted the DUT (e.g. by |
| 1225 | # looking at DUT uptime after repair completes). |
| 1226 | |
| 1227 | self.reboot() |
Scott Zawalski | 89c44dd | 2013-02-26 09:28:02 -0500 | [diff] [blame] | 1228 | |
Richard Barnette | 82c3591 | 2012-11-20 10:09:10 -0800 | [diff] [blame] | 1229 | |
J. Richard Barnette | 1d78b01 | 2012-05-15 13:56:30 -0700 | [diff] [blame] | 1230 | def close(self): |
Fang Deng | 0ca40e2 | 2013-08-27 17:47:44 -0700 | [diff] [blame] | 1231 | super(CrosHost, self).close() |
xixuan | d6011f1 | 2016-12-08 15:01:58 -0800 | [diff] [blame] | 1232 | if self._chameleon_host: |
| 1233 | self._chameleon_host.close() |
| 1234 | |
| 1235 | if self._servo_host: |
| 1236 | self._servo_host.close() |
J. Richard Barnette | 1d78b01 | 2012-05-15 13:56:30 -0700 | [diff] [blame] | 1237 | |
| 1238 | |
Dan Shi | 49ca093 | 2014-11-14 11:22:27 -0800 | [diff] [blame] | 1239 | def get_power_supply_info(self): |
| 1240 | """Get the output of power_supply_info. |
| 1241 | |
| 1242 | power_supply_info outputs the info of each power supply, e.g., |
| 1243 | Device: Line Power |
| 1244 | online: no |
| 1245 | type: Mains |
| 1246 | voltage (V): 0 |
| 1247 | current (A): 0 |
| 1248 | Device: Battery |
| 1249 | state: Discharging |
| 1250 | percentage: 95.9276 |
| 1251 | technology: Li-ion |
| 1252 | |
| 1253 | Above output shows two devices, Line Power and Battery, with details of |
| 1254 | each device listed. This function parses the output into a dictionary, |
| 1255 | with key being the device name, and value being a dictionary of details |
| 1256 | of the device info. |
| 1257 | |
| 1258 | @return: The dictionary of power_supply_info, e.g., |
| 1259 | {'Line Power': {'online': 'yes', 'type': 'main'}, |
| 1260 | 'Battery': {'vendor': 'xyz', 'percentage': '100'}} |
Dan Shi | e9b765d | 2014-12-29 16:59:49 -0800 | [diff] [blame] | 1261 | @raise error.AutoservRunError if power_supply_info tool is not found in |
| 1262 | the DUT. Caller should handle this error to avoid false failure |
| 1263 | on verification. |
Dan Shi | 49ca093 | 2014-11-14 11:22:27 -0800 | [diff] [blame] | 1264 | """ |
| 1265 | result = self.run('power_supply_info').stdout.strip() |
| 1266 | info = {} |
| 1267 | device_name = None |
| 1268 | device_info = {} |
| 1269 | for line in result.split('\n'): |
| 1270 | pair = [v.strip() for v in line.split(':')] |
| 1271 | if len(pair) != 2: |
| 1272 | continue |
| 1273 | if pair[0] == 'Device': |
| 1274 | if device_name: |
| 1275 | info[device_name] = device_info |
| 1276 | device_name = pair[1] |
| 1277 | device_info = {} |
| 1278 | else: |
| 1279 | device_info[pair[0]] = pair[1] |
| 1280 | if device_name and not device_name in info: |
| 1281 | info[device_name] = device_info |
| 1282 | return info |
| 1283 | |
| 1284 | |
| 1285 | def get_battery_percentage(self): |
| 1286 | """Get the battery percentage. |
| 1287 | |
| 1288 | @return: The percentage of battery level, value range from 0-100. Return |
| 1289 | None if the battery info cannot be retrieved. |
| 1290 | """ |
| 1291 | try: |
| 1292 | info = self.get_power_supply_info() |
| 1293 | logging.info(info) |
| 1294 | return float(info['Battery']['percentage']) |
Dan Shi | e9b765d | 2014-12-29 16:59:49 -0800 | [diff] [blame] | 1295 | except (KeyError, ValueError, error.AutoservRunError): |
Dan Shi | 49ca093 | 2014-11-14 11:22:27 -0800 | [diff] [blame] | 1296 | return None |
| 1297 | |
| 1298 | |
| 1299 | def is_ac_connected(self): |
| 1300 | """Check if the dut has power adapter connected and charging. |
| 1301 | |
| 1302 | @return: True if power adapter is connected and charging. |
| 1303 | """ |
| 1304 | try: |
| 1305 | info = self.get_power_supply_info() |
| 1306 | return info['Line Power']['online'] == 'yes' |
Dan Shi | e9b765d | 2014-12-29 16:59:49 -0800 | [diff] [blame] | 1307 | except (KeyError, error.AutoservRunError): |
| 1308 | return None |
Dan Shi | 49ca093 | 2014-11-14 11:22:27 -0800 | [diff] [blame] | 1309 | |
| 1310 | |
Simran Basi | 5e6339a | 2013-03-21 11:34:32 -0700 | [diff] [blame] | 1311 | def _cleanup_poweron(self): |
| 1312 | """Special cleanup method to make sure hosts always get power back.""" |
| 1313 | afe = frontend_wrappers.RetryingAFE(timeout_min=5, delay_sec=10) |
| 1314 | hosts = afe.get_hosts(hostname=self.hostname) |
| 1315 | if not hosts or not (self._RPM_OUTLET_CHANGED in |
| 1316 | hosts[0].attributes): |
| 1317 | return |
| 1318 | logging.debug('This host has recently interacted with the RPM' |
| 1319 | ' Infrastructure. Ensuring power is on.') |
| 1320 | try: |
| 1321 | self.power_on() |
Dan Shi | 7dca56e | 2014-11-11 17:07:56 -0800 | [diff] [blame] | 1322 | afe.set_host_attribute(self._RPM_OUTLET_CHANGED, None, |
| 1323 | hostname=self.hostname) |
Simran Basi | 5e6339a | 2013-03-21 11:34:32 -0700 | [diff] [blame] | 1324 | except rpm_client.RemotePowerException: |
Simran Basi | 5e6339a | 2013-03-21 11:34:32 -0700 | [diff] [blame] | 1325 | logging.error('Failed to turn Power On for this host after ' |
| 1326 | 'cleanup through the RPM Infrastructure.') |
Dan Shi | 49ca093 | 2014-11-14 11:22:27 -0800 | [diff] [blame] | 1327 | |
| 1328 | battery_percentage = self.get_battery_percentage() |
Dan Shi | f01ebe2 | 2014-12-05 13:10:57 -0800 | [diff] [blame] | 1329 | if battery_percentage and battery_percentage < 50: |
Dan Shi | 49ca093 | 2014-11-14 11:22:27 -0800 | [diff] [blame] | 1330 | raise |
| 1331 | elif self.is_ac_connected(): |
| 1332 | logging.info('The device has power adapter connected and ' |
| 1333 | 'charging. No need to try to turn RPM on ' |
| 1334 | 'again.') |
| 1335 | afe.set_host_attribute(self._RPM_OUTLET_CHANGED, None, |
| 1336 | hostname=self.hostname) |
| 1337 | logging.info('Battery level is now at %s%%. The device may ' |
| 1338 | 'still have enough power to run test, so no ' |
| 1339 | 'exception will be raised.', battery_percentage) |
| 1340 | |
Simran Basi | 5e6339a | 2013-03-21 11:34:32 -0700 | [diff] [blame] | 1341 | |
beeps | c87ff60 | 2013-07-31 21:53:00 -0700 | [diff] [blame] | 1342 | def _is_factory_image(self): |
| 1343 | """Checks if the image on the DUT is a factory image. |
| 1344 | |
| 1345 | @return: True if the image on the DUT is a factory image. |
| 1346 | False otherwise. |
| 1347 | """ |
| 1348 | result = self.run('[ -f /root/.factory_test ]', ignore_status=True) |
| 1349 | return result.exit_status == 0 |
| 1350 | |
| 1351 | |
| 1352 | def _restart_ui(self): |
J. Richard Barnette | 84890bd | 2014-02-21 11:05:47 -0800 | [diff] [blame] | 1353 | """Restart the Chrome UI. |
beeps | c87ff60 | 2013-07-31 21:53:00 -0700 | [diff] [blame] | 1354 | |
| 1355 | @raises: FactoryImageCheckerException for factory images, since |
| 1356 | we cannot attempt to restart ui on them. |
| 1357 | error.AutoservRunError for any other type of error that |
| 1358 | occurs while restarting ui. |
| 1359 | """ |
| 1360 | if self._is_factory_image(): |
Dan Shi | 549fb82 | 2015-03-24 18:01:11 -0700 | [diff] [blame] | 1361 | raise FactoryImageCheckerException('Cannot restart ui on factory ' |
| 1362 | 'images') |
beeps | c87ff60 | 2013-07-31 21:53:00 -0700 | [diff] [blame] | 1363 | |
J. Richard Barnette | 84890bd | 2014-02-21 11:05:47 -0800 | [diff] [blame] | 1364 | # TODO(jrbarnette): The command to stop/start the ui job |
| 1365 | # should live inside cros_ui, too. However that would seem |
| 1366 | # to imply interface changes to the existing start()/restart() |
| 1367 | # functions, which is a bridge too far (for now). |
J. Richard Barnette | 6069aa1 | 2015-06-08 09:10:24 -0700 | [diff] [blame] | 1368 | prompt = cros_ui.get_chrome_session_ident(self) |
J. Richard Barnette | 84890bd | 2014-02-21 11:05:47 -0800 | [diff] [blame] | 1369 | self.run('stop ui; start ui') |
| 1370 | cros_ui.wait_for_chrome_ready(prompt, self) |
beeps | c87ff60 | 2013-07-31 21:53:00 -0700 | [diff] [blame] | 1371 | |
| 1372 | |
xixuan | a3bbc42 | 2017-05-04 15:57:21 -0700 | [diff] [blame] | 1373 | def _get_lsb_release_content(self): |
| 1374 | """Return the content of lsb-release file of host.""" |
| 1375 | return self.run( |
| 1376 | 'cat "%s"' % client_constants.LSB_RELEASE).stdout.strip() |
| 1377 | |
| 1378 | |
Dan Shi | 549fb82 | 2015-03-24 18:01:11 -0700 | [diff] [blame] | 1379 | def get_release_version(self): |
| 1380 | """Get the value of attribute CHROMEOS_RELEASE_VERSION from lsb-release. |
| 1381 | |
| 1382 | @returns The version string in lsb-release, under attribute |
| 1383 | CHROMEOS_RELEASE_VERSION. |
| 1384 | """ |
Dan Shi | 549fb82 | 2015-03-24 18:01:11 -0700 | [diff] [blame] | 1385 | return lsbrelease_utils.get_chromeos_release_version( |
xixuan | a3bbc42 | 2017-05-04 15:57:21 -0700 | [diff] [blame] | 1386 | lsb_release_content=self._get_lsb_release_content()) |
| 1387 | |
| 1388 | |
| 1389 | def get_chromeos_release_milestone(self): |
| 1390 | """Get the value of attribute CHROMEOS_RELEASE_BUILD_TYPE |
| 1391 | from lsb-release. |
| 1392 | |
| 1393 | @returns The version string in lsb-release, under attribute |
| 1394 | CHROMEOS_RELEASE_BUILD_TYPE. |
| 1395 | """ |
| 1396 | return lsbrelease_utils.get_chromeos_release_milestone( |
| 1397 | lsb_release_content=self._get_lsb_release_content()) |
Dan Shi | 549fb82 | 2015-03-24 18:01:11 -0700 | [diff] [blame] | 1398 | |
| 1399 | |
| 1400 | def verify_cros_version_label(self): |
| 1401 | """ Make sure host's cros-version label match the actual image in dut. |
| 1402 | |
| 1403 | Remove any cros-version: label that doesn't match that installed in |
| 1404 | the dut. |
| 1405 | |
| 1406 | @param raise_error: Set to True to raise exception if any mismatch found |
| 1407 | |
| 1408 | @raise error.AutoservError: If any mismatch between cros-version label |
| 1409 | and the build installed in dut is found. |
| 1410 | """ |
| 1411 | labels = self._AFE.get_labels( |
| 1412 | name__startswith=ds_constants.VERSION_PREFIX, |
| 1413 | host__hostname=self.hostname) |
| 1414 | mismatch_found = False |
| 1415 | if labels: |
| 1416 | # Get CHROMEOS_RELEASE_VERSION from lsb-release, e.g., 6908.0.0. |
| 1417 | # Note that it's different from cros-version label, which has |
| 1418 | # builder and branch info, e.g., |
| 1419 | # cros-version:peppy-release/R43-6908.0.0 |
| 1420 | release_version = self.get_release_version() |
| 1421 | host_list = [self.hostname] |
| 1422 | for label in labels: |
| 1423 | # Remove any cros-version label that does not match |
| 1424 | # release_version. |
| 1425 | build_version = label.name[len(ds_constants.VERSION_PREFIX):] |
| 1426 | if not utils.version_match(build_version, release_version): |
| 1427 | logging.warn('cros-version label "%s" does not match ' |
| 1428 | 'release version %s. Removing the label.', |
| 1429 | label.name, release_version) |
| 1430 | label.remove_hosts(hosts=host_list) |
| 1431 | mismatch_found = True |
| 1432 | if mismatch_found: |
| 1433 | raise error.AutoservError('The host has wrong cros-version label.') |
| 1434 | |
| 1435 | |
Laurence Goodby | 778c9a4 | 2017-05-24 19:24:07 -0700 | [diff] [blame] | 1436 | def cleanup_services(self): |
| 1437 | """Reinitializes the device for cleanup. |
| 1438 | |
| 1439 | Subclasses may override this to customize the cleanup method. |
| 1440 | |
| 1441 | To indicate failure of the reset, the implementation may raise |
| 1442 | any of: |
| 1443 | error.AutoservRunError |
| 1444 | error.AutotestRunError |
| 1445 | FactoryImageCheckerException |
| 1446 | |
| 1447 | @raises error.AutoservRunError |
| 1448 | @raises error.AutotestRunError |
| 1449 | @raises error.FactoryImageCheckerException |
| 1450 | """ |
| 1451 | self._restart_ui() |
| 1452 | |
| 1453 | |
beeps | c87ff60 | 2013-07-31 21:53:00 -0700 | [diff] [blame] | 1454 | def cleanup(self): |
MK Ryu | 35d661e | 2014-09-25 17:44:10 -0700 | [diff] [blame] | 1455 | self.run('rm -f %s' % client_constants.CLEANUP_LOGS_PAUSED_FILE) |
Scott Zawalski | ddbc31e | 2012-11-15 11:29:01 -0500 | [diff] [blame] | 1456 | try: |
Laurence Goodby | 778c9a4 | 2017-05-24 19:24:07 -0700 | [diff] [blame] | 1457 | self.cleanup_services() |
beeps | c87ff60 | 2013-07-31 21:53:00 -0700 | [diff] [blame] | 1458 | except (error.AutotestRunError, error.AutoservRunError, |
| 1459 | FactoryImageCheckerException): |
Ilja H. Friedel | 04be2bd | 2014-05-07 21:29:59 -0700 | [diff] [blame] | 1460 | logging.warning('Unable to restart ui, rebooting device.') |
Scott Zawalski | ddbc31e | 2012-11-15 11:29:01 -0500 | [diff] [blame] | 1461 | # Since restarting the UI fails fall back to normal Autotest |
| 1462 | # cleanup routines, i.e. reboot the machine. |
Fang Deng | 0ca40e2 | 2013-08-27 17:47:44 -0700 | [diff] [blame] | 1463 | super(CrosHost, self).cleanup() |
Simran Basi | 5e6339a | 2013-03-21 11:34:32 -0700 | [diff] [blame] | 1464 | # Check if the rpm outlet was manipulated. |
Simran Basi | d5e5e27 | 2012-09-24 15:23:59 -0700 | [diff] [blame] | 1465 | if self.has_power(): |
Simran Basi | 5e6339a | 2013-03-21 11:34:32 -0700 | [diff] [blame] | 1466 | self._cleanup_poweron() |
Dan Shi | 549fb82 | 2015-03-24 18:01:11 -0700 | [diff] [blame] | 1467 | self.verify_cros_version_label() |
J. Richard Barnette | 45e93de | 2012-04-11 17:24:15 -0700 | [diff] [blame] | 1468 | |
| 1469 | |
Yu-Ju Hong | a2be94a | 2012-07-31 09:48:52 -0700 | [diff] [blame] | 1470 | def reboot(self, **dargs): |
| 1471 | """ |
| 1472 | This function reboots the site host. The more generic |
| 1473 | RemoteHost.reboot() performs sync and sleeps for 5 |
| 1474 | seconds. This is not necessary for Chrome OS devices as the |
| 1475 | sync should be finished in a short time during the reboot |
| 1476 | command. |
| 1477 | """ |
Tom Wai-Hong Tam | f5cd1d4 | 2012-08-13 12:04:08 +0800 | [diff] [blame] | 1478 | if 'reboot_cmd' not in dargs: |
Doug Anderson | 7d5aeb2 | 2014-02-27 15:12:17 -0800 | [diff] [blame] | 1479 | reboot_timeout = dargs.get('reboot_timeout', 10) |
J. Richard Barnette | 9af1963 | 2015-09-25 12:18:03 -0700 | [diff] [blame] | 1480 | dargs['reboot_cmd'] = ('sleep 1; ' |
| 1481 | 'reboot & sleep %d; ' |
| 1482 | 'reboot -f' % reboot_timeout) |
Yu-Ju Hong | a2be94a | 2012-07-31 09:48:52 -0700 | [diff] [blame] | 1483 | # Enable fastsync to avoid running extra sync commands before reboot. |
Tom Wai-Hong Tam | f5cd1d4 | 2012-08-13 12:04:08 +0800 | [diff] [blame] | 1484 | if 'fastsync' not in dargs: |
| 1485 | dargs['fastsync'] = True |
Michael Liang | da8c60a | 2014-06-03 13:24:51 -0700 | [diff] [blame] | 1486 | |
Charlie Mooney | a8e6dab | 2014-05-29 14:37:55 -0700 | [diff] [blame] | 1487 | # For purposes of logging reboot times: |
| 1488 | # Get the board name i.e. 'daisy_spring' |
Michael Liang | ca4f5a6 | 2014-07-10 15:45:13 -0700 | [diff] [blame] | 1489 | board_fullname = self.get_board() |
| 1490 | |
| 1491 | # Strip the prefix and add it to dargs. |
| 1492 | dargs['board'] = board_fullname[board_fullname.find(':')+1:] |
Vincent Palatin | df2372c | 2016-10-07 17:03:00 +0200 | [diff] [blame] | 1493 | # Record who called us |
| 1494 | orig = sys._getframe(1).f_code |
Vincent Palatin | 80780b2 | 2016-07-27 16:02:37 +0200 | [diff] [blame] | 1495 | metric_fields = {'board' : dargs['board'], |
Vincent Palatin | df2372c | 2016-10-07 17:03:00 +0200 | [diff] [blame] | 1496 | 'dut_host_name' : self.hostname, |
| 1497 | 'success' : True} |
| 1498 | metric_debug_fields = {'board' : dargs['board'], |
| 1499 | 'caller' : "%s:%s" % (orig.co_filename, orig.co_name), |
| 1500 | 'success' : True, |
| 1501 | 'error' : ''} |
| 1502 | |
Vincent Palatin | 80780b2 | 2016-07-27 16:02:37 +0200 | [diff] [blame] | 1503 | t0 = time.time() |
| 1504 | try: |
| 1505 | super(CrosHost, self).reboot(**dargs) |
| 1506 | except Exception as e: |
| 1507 | metric_fields['success'] = False |
Vincent Palatin | df2372c | 2016-10-07 17:03:00 +0200 | [diff] [blame] | 1508 | metric_debug_fields['success'] = False |
| 1509 | metric_debug_fields['error'] = type(e).__name__ |
Vincent Palatin | 80780b2 | 2016-07-27 16:02:37 +0200 | [diff] [blame] | 1510 | raise |
| 1511 | finally: |
| 1512 | duration = int(time.time() - t0) |
Dan Shi | 5e2efb7 | 2017-02-07 11:40:23 -0800 | [diff] [blame] | 1513 | metrics.Counter( |
| 1514 | 'chromeos/autotest/autoserv/reboot_count').increment( |
| 1515 | fields=metric_fields) |
| 1516 | metrics.Counter( |
| 1517 | 'chromeos/autotest/autoserv/reboot_debug').increment( |
| 1518 | fields=metric_debug_fields) |
| 1519 | metrics.SecondsDistribution( |
| 1520 | 'chromeos/autotest/autoserv/reboot_duration').add( |
| 1521 | duration, fields=metric_fields) |
Yu-Ju Hong | a2be94a | 2012-07-31 09:48:52 -0700 | [diff] [blame] | 1522 | |
| 1523 | |
Gwendal Grignou | 7a61d2f | 2014-05-23 11:05:51 -0700 | [diff] [blame] | 1524 | def suspend(self, **dargs): |
| 1525 | """ |
| 1526 | This function suspends the site host. |
| 1527 | """ |
| 1528 | suspend_time = dargs.get('suspend_time', 60) |
| 1529 | dargs['timeout'] = suspend_time |
| 1530 | if 'suspend_cmd' not in dargs: |
J. Richard Barnette | 9af1963 | 2015-09-25 12:18:03 -0700 | [diff] [blame] | 1531 | dargs['suspend_cmd'] = ' && '.join([ |
| 1532 | 'echo 0 > /sys/class/rtc/rtc0/wakealarm', |
Gwendal Grignou | 7a61d2f | 2014-05-23 11:05:51 -0700 | [diff] [blame] | 1533 | 'echo +%d > /sys/class/rtc/rtc0/wakealarm' % suspend_time, |
J. Richard Barnette | 9af1963 | 2015-09-25 12:18:03 -0700 | [diff] [blame] | 1534 | 'powerd_dbus_suspend --delay=0']) |
Gwendal Grignou | 7a61d2f | 2014-05-23 11:05:51 -0700 | [diff] [blame] | 1535 | super(CrosHost, self).suspend(**dargs) |
| 1536 | |
| 1537 | |
Simran Basi | ec56439 | 2014-08-25 16:48:09 -0700 | [diff] [blame] | 1538 | def upstart_status(self, service_name): |
| 1539 | """Check the status of an upstart init script. |
| 1540 | |
| 1541 | @param service_name: Service to look up. |
| 1542 | |
| 1543 | @returns True if the service is running, False otherwise. |
| 1544 | """ |
Richard Barnette | e204dc5 | 2017-09-26 11:02:25 -0700 | [diff] [blame] | 1545 | return 'start/running' in self.run('status %s' % service_name, |
| 1546 | ignore_status=True).stdout |
Simran Basi | ec56439 | 2014-08-25 16:48:09 -0700 | [diff] [blame] | 1547 | |
| 1548 | |
J. Richard Barnette | 45e93de | 2012-04-11 17:24:15 -0700 | [diff] [blame] | 1549 | def verify_software(self): |
Richard Barnette | b2bc13c | 2013-01-08 17:32:51 -0800 | [diff] [blame] | 1550 | """Verify working software on a Chrome OS system. |
J. Richard Barnette | 45e93de | 2012-04-11 17:24:15 -0700 | [diff] [blame] | 1551 | |
Richard Barnette | b2bc13c | 2013-01-08 17:32:51 -0800 | [diff] [blame] | 1552 | Tests for the following conditions: |
| 1553 | 1. All conditions tested by the parent version of this |
| 1554 | function. |
| 1555 | 2. Sufficient space in /mnt/stateful_partition. |
Fang Deng | 6b05f5b | 2013-03-20 13:42:11 -0700 | [diff] [blame] | 1556 | 3. Sufficient space in /mnt/stateful_partition/encrypted. |
| 1557 | 4. update_engine answers a simple status request over DBus. |
J. Richard Barnette | 45e93de | 2012-04-11 17:24:15 -0700 | [diff] [blame] | 1558 | |
J. Richard Barnette | 45e93de | 2012-04-11 17:24:15 -0700 | [diff] [blame] | 1559 | """ |
Fang Deng | 0ca40e2 | 2013-08-27 17:47:44 -0700 | [diff] [blame] | 1560 | super(CrosHost, self).verify_software() |
Dan Shi | b8540a5 | 2015-07-16 14:18:23 -0700 | [diff] [blame] | 1561 | default_kilo_inodes_required = CONFIG.get_config_value( |
| 1562 | 'SERVER', 'kilo_inodes_required', type=int, default=100) |
| 1563 | board = self.get_board().replace(ds_constants.BOARD_PREFIX, '') |
| 1564 | kilo_inodes_required = CONFIG.get_config_value( |
| 1565 | 'SERVER', 'kilo_inodes_required_%s' % board, |
| 1566 | type=int, default=default_kilo_inodes_required) |
| 1567 | self.check_inodes('/mnt/stateful_partition', kilo_inodes_required) |
J. Richard Barnette | 45e93de | 2012-04-11 17:24:15 -0700 | [diff] [blame] | 1568 | self.check_diskspace( |
| 1569 | '/mnt/stateful_partition', |
Dan Shi | b8540a5 | 2015-07-16 14:18:23 -0700 | [diff] [blame] | 1570 | CONFIG.get_config_value( |
Fang Deng | 6b05f5b | 2013-03-20 13:42:11 -0700 | [diff] [blame] | 1571 | 'SERVER', 'gb_diskspace_required', type=float, |
| 1572 | default=20.0)) |
Gaurav Shah | e448af8 | 2014-06-19 15:18:59 -0700 | [diff] [blame] | 1573 | encrypted_stateful_path = '/mnt/stateful_partition/encrypted' |
| 1574 | # Not all targets build with encrypted stateful support. |
| 1575 | if self.path_exists(encrypted_stateful_path): |
| 1576 | self.check_diskspace( |
| 1577 | encrypted_stateful_path, |
Dan Shi | b8540a5 | 2015-07-16 14:18:23 -0700 | [diff] [blame] | 1578 | CONFIG.get_config_value( |
Gaurav Shah | e448af8 | 2014-06-19 15:18:59 -0700 | [diff] [blame] | 1579 | 'SERVER', 'gb_encrypted_diskspace_required', type=float, |
| 1580 | default=0.1)) |
beeps | c87ff60 | 2013-07-31 21:53:00 -0700 | [diff] [blame] | 1581 | |
Simran Basi | ec56439 | 2014-08-25 16:48:09 -0700 | [diff] [blame] | 1582 | if not self.upstart_status('system-services'): |
Prashanth B | 5d0a051 | 2014-04-25 12:26:08 -0700 | [diff] [blame] | 1583 | raise error.AutoservError('Chrome failed to reach login. ' |
| 1584 | 'System services not running.') |
| 1585 | |
beeps | c87ff60 | 2013-07-31 21:53:00 -0700 | [diff] [blame] | 1586 | # Factory images don't run update engine, |
| 1587 | # goofy controls dbus on these DUTs. |
| 1588 | if not self._is_factory_image(): |
| 1589 | self.run('update_engine_client --status') |
J. Richard Barnette | 134ec2c | 2012-04-25 12:59:37 -0700 | [diff] [blame] | 1590 | |
Dan Shi | 549fb82 | 2015-03-24 18:01:11 -0700 | [diff] [blame] | 1591 | self.verify_cros_version_label() |
| 1592 | |
J. Richard Barnette | 134ec2c | 2012-04-25 12:59:37 -0700 | [diff] [blame] | 1593 | |
J. Richard Barnette | a7e7fdc | 2016-02-12 12:35:36 -0800 | [diff] [blame] | 1594 | def verify(self): |
| 1595 | self._repair_strategy.verify(self) |
| 1596 | |
| 1597 | |
Fang Deng | 96667ca | 2013-08-01 17:46:18 -0700 | [diff] [blame] | 1598 | def make_ssh_command(self, user='root', port=22, opts='', hosts_file=None, |
| 1599 | connect_timeout=None, alive_interval=None): |
| 1600 | """Override default make_ssh_command to use options tuned for Chrome OS. |
| 1601 | |
| 1602 | Tuning changes: |
| 1603 | - ConnectTimeout=30; maximum of 30 seconds allowed for an SSH |
| 1604 | connection failure. Consistency with remote_access.sh. |
| 1605 | |
Samuel Tan | 2ce155b | 2015-06-23 18:24:38 -0700 | [diff] [blame] | 1606 | - ServerAliveInterval=900; which causes SSH to ping connection every |
| 1607 | 900 seconds. In conjunction with ServerAliveCountMax ensures |
| 1608 | that if the connection dies, Autotest will bail out. |
Fang Deng | 96667ca | 2013-08-01 17:46:18 -0700 | [diff] [blame] | 1609 | Originally tried 60 secs, but saw frequent job ABORTS where |
Samuel Tan | 2ce155b | 2015-06-23 18:24:38 -0700 | [diff] [blame] | 1610 | the test completed successfully. Later increased from 180 seconds to |
| 1611 | 900 seconds to account for tests where the DUT is suspended for |
| 1612 | longer periods of time. |
Fang Deng | 96667ca | 2013-08-01 17:46:18 -0700 | [diff] [blame] | 1613 | |
| 1614 | - ServerAliveCountMax=3; consistency with remote_access.sh. |
| 1615 | |
| 1616 | - ConnectAttempts=4; reduce flakiness in connection errors; |
| 1617 | consistency with remote_access.sh. |
| 1618 | |
| 1619 | - UserKnownHostsFile=/dev/null; we don't care about the keys. |
| 1620 | Host keys change with every new installation, don't waste |
| 1621 | memory/space saving them. |
| 1622 | |
| 1623 | - SSH protocol forced to 2; needed for ServerAliveInterval. |
| 1624 | |
| 1625 | @param user User name to use for the ssh connection. |
| 1626 | @param port Port on the target host to use for ssh connection. |
| 1627 | @param opts Additional options to the ssh command. |
| 1628 | @param hosts_file Ignored. |
| 1629 | @param connect_timeout Ignored. |
| 1630 | @param alive_interval Ignored. |
| 1631 | """ |
Aviv Keshet | c5947fa | 2013-09-04 14:06:29 -0700 | [diff] [blame] | 1632 | base_command = ('/usr/bin/ssh -a -x %s %s %s' |
| 1633 | ' -o StrictHostKeyChecking=no' |
Fang Deng | 96667ca | 2013-08-01 17:46:18 -0700 | [diff] [blame] | 1634 | ' -o UserKnownHostsFile=/dev/null -o BatchMode=yes' |
Samuel Tan | 2ce155b | 2015-06-23 18:24:38 -0700 | [diff] [blame] | 1635 | ' -o ConnectTimeout=30 -o ServerAliveInterval=900' |
Fang Deng | 96667ca | 2013-08-01 17:46:18 -0700 | [diff] [blame] | 1636 | ' -o ServerAliveCountMax=3 -o ConnectionAttempts=4' |
| 1637 | ' -o Protocol=2 -l %s -p %d') |
Aviv Keshet | c5947fa | 2013-09-04 14:06:29 -0700 | [diff] [blame] | 1638 | return base_command % (self._ssh_verbosity_flag, self._ssh_options, |
| 1639 | opts, user, port) |
Jason Abele | b6f924f | 2013-11-13 16:01:54 -0800 | [diff] [blame] | 1640 | def syslog(self, message, tag='autotest'): |
| 1641 | """Logs a message to syslog on host. |
| 1642 | |
| 1643 | @param message String message to log into syslog |
| 1644 | @param tag String tag prefix for syslog |
| 1645 | |
| 1646 | """ |
| 1647 | self.run('logger -t "%s" "%s"' % (tag, message)) |
| 1648 | |
| 1649 | |
J. Richard Barnette | b6de7e3 | 2013-02-14 13:28:04 -0800 | [diff] [blame] | 1650 | def _ping_check_status(self, status): |
| 1651 | """Ping the host once, and return whether it has a given status. |
J. Richard Barnette | 134ec2c | 2012-04-25 12:59:37 -0700 | [diff] [blame] | 1652 | |
J. Richard Barnette | b6de7e3 | 2013-02-14 13:28:04 -0800 | [diff] [blame] | 1653 | @param status Check the ping status against this value. |
| 1654 | @return True iff `status` and the result of ping are the same |
| 1655 | (i.e. both True or both False). |
| 1656 | |
| 1657 | """ |
| 1658 | ping_val = utils.ping(self.hostname, tries=1, deadline=1) |
| 1659 | return not (status ^ (ping_val == 0)) |
| 1660 | |
| 1661 | def _ping_wait_for_status(self, status, timeout): |
| 1662 | """Wait for the host to have a given status (UP or DOWN). |
| 1663 | |
| 1664 | Status is checked by polling. Polling will not last longer |
| 1665 | than the number of seconds in `timeout`. The polling |
| 1666 | interval will be long enough that only approximately |
| 1667 | _PING_WAIT_COUNT polling cycles will be executed, subject |
| 1668 | to a maximum interval of about one minute. |
| 1669 | |
| 1670 | @param status Waiting will stop immediately if `ping` of the |
| 1671 | host returns this status. |
| 1672 | @param timeout Poll for at most this many seconds. |
| 1673 | @return True iff the host status from `ping` matched the |
| 1674 | requested status at the time of return. |
| 1675 | |
| 1676 | """ |
| 1677 | # _ping_check_status() takes about 1 second, hence the |
| 1678 | # "- 1" in the formula below. |
Nathan Ciobanu | 38480a3 | 2016-10-25 15:26:45 -0700 | [diff] [blame] | 1679 | # FIXME: if the ping command errors then _ping_check_status() |
| 1680 | # returns instantly. If timeout is also smaller than twice |
| 1681 | # _PING_WAIT_COUNT then the while loop below forks many |
| 1682 | # thousands of ping commands (see /tmp/test_that_results_XXXXX/ |
| 1683 | # /results-1-logging_YYY.ZZZ/debug/autoserv.DEBUG) and hogs one |
| 1684 | # CPU core for 60 seconds. |
J. Richard Barnette | b6de7e3 | 2013-02-14 13:28:04 -0800 | [diff] [blame] | 1685 | poll_interval = min(int(timeout / self._PING_WAIT_COUNT), 60) - 1 |
| 1686 | end_time = time.time() + timeout |
| 1687 | while time.time() <= end_time: |
| 1688 | if self._ping_check_status(status): |
| 1689 | return True |
| 1690 | if poll_interval > 0: |
| 1691 | time.sleep(poll_interval) |
| 1692 | |
| 1693 | # The last thing we did was sleep(poll_interval), so it may |
| 1694 | # have been too long since the last `ping`. Check one more |
| 1695 | # time, just to be sure. |
| 1696 | return self._ping_check_status(status) |
| 1697 | |
| 1698 | def ping_wait_up(self, timeout): |
| 1699 | """Wait for the host to respond to `ping`. |
| 1700 | |
| 1701 | N.B. This method is not a reliable substitute for |
| 1702 | `wait_up()`, because a host that responds to ping will not |
| 1703 | necessarily respond to ssh. This method should only be used |
| 1704 | if the target DUT can be considered functional even if it |
| 1705 | can't be reached via ssh. |
| 1706 | |
| 1707 | @param timeout Minimum time to allow before declaring the |
| 1708 | host to be non-responsive. |
| 1709 | @return True iff the host answered to ping before the timeout. |
| 1710 | |
| 1711 | """ |
| 1712 | return self._ping_wait_for_status(self._PING_STATUS_UP, timeout) |
J. Richard Barnette | 134ec2c | 2012-04-25 12:59:37 -0700 | [diff] [blame] | 1713 | |
Andrew Bresticker | 678c0c7 | 2013-01-22 10:44:09 -0800 | [diff] [blame] | 1714 | def ping_wait_down(self, timeout): |
J. Richard Barnette | 134ec2c | 2012-04-25 12:59:37 -0700 | [diff] [blame] | 1715 | """Wait until the host no longer responds to `ping`. |
| 1716 | |
J. Richard Barnette | b6de7e3 | 2013-02-14 13:28:04 -0800 | [diff] [blame] | 1717 | This function can be used as a slightly faster version of |
| 1718 | `wait_down()`, by avoiding potentially long ssh timeouts. |
| 1719 | |
| 1720 | @param timeout Minimum time to allow for the host to become |
| 1721 | non-responsive. |
| 1722 | @return True iff the host quit answering ping before the |
| 1723 | timeout. |
| 1724 | |
J. Richard Barnette | 134ec2c | 2012-04-25 12:59:37 -0700 | [diff] [blame] | 1725 | """ |
J. Richard Barnette | b6de7e3 | 2013-02-14 13:28:04 -0800 | [diff] [blame] | 1726 | return self._ping_wait_for_status(self._PING_STATUS_DOWN, timeout) |
J. Richard Barnette | 134ec2c | 2012-04-25 12:59:37 -0700 | [diff] [blame] | 1727 | |
Tom Wai-Hong Tam | fced4f6 | 2014-04-17 10:56:30 +0800 | [diff] [blame] | 1728 | def test_wait_for_sleep(self, sleep_timeout=None): |
J. Richard Barnette | 134ec2c | 2012-04-25 12:59:37 -0700 | [diff] [blame] | 1729 | """Wait for the client to enter low-power sleep mode. |
| 1730 | |
| 1731 | The test for "is asleep" can't distinguish a system that is |
| 1732 | powered off; to confirm that the unit was asleep, it is |
| 1733 | necessary to force resume, and then call |
| 1734 | `test_wait_for_resume()`. |
| 1735 | |
| 1736 | This function is expected to be called from a test as part |
| 1737 | of a sequence like the following: |
| 1738 | |
| 1739 | ~~~~~~~~ |
| 1740 | boot_id = host.get_boot_id() |
| 1741 | # trigger sleep on the host |
| 1742 | host.test_wait_for_sleep() |
| 1743 | # trigger resume on the host |
| 1744 | host.test_wait_for_resume(boot_id) |
| 1745 | ~~~~~~~~ |
| 1746 | |
Tom Wai-Hong Tam | fced4f6 | 2014-04-17 10:56:30 +0800 | [diff] [blame] | 1747 | @param sleep_timeout time limit in seconds to allow the host sleep. |
| 1748 | |
J. Richard Barnette | 134ec2c | 2012-04-25 12:59:37 -0700 | [diff] [blame] | 1749 | @exception TestFail The host did not go to sleep within |
| 1750 | the allowed time. |
| 1751 | """ |
Tom Wai-Hong Tam | fced4f6 | 2014-04-17 10:56:30 +0800 | [diff] [blame] | 1752 | if sleep_timeout is None: |
| 1753 | sleep_timeout = self.SLEEP_TIMEOUT |
| 1754 | |
| 1755 | if not self.ping_wait_down(timeout=sleep_timeout): |
J. Richard Barnette | 134ec2c | 2012-04-25 12:59:37 -0700 | [diff] [blame] | 1756 | raise error.TestFail( |
Tom Wai-Hong Tam | fced4f6 | 2014-04-17 10:56:30 +0800 | [diff] [blame] | 1757 | 'client failed to sleep after %d seconds' % sleep_timeout) |
J. Richard Barnette | 134ec2c | 2012-04-25 12:59:37 -0700 | [diff] [blame] | 1758 | |
| 1759 | |
Tom Wai-Hong Tam | fced4f6 | 2014-04-17 10:56:30 +0800 | [diff] [blame] | 1760 | def test_wait_for_resume(self, old_boot_id, resume_timeout=None): |
J. Richard Barnette | 134ec2c | 2012-04-25 12:59:37 -0700 | [diff] [blame] | 1761 | """Wait for the client to resume from low-power sleep mode. |
| 1762 | |
| 1763 | The `old_boot_id` parameter should be the value from |
| 1764 | `get_boot_id()` obtained prior to entering sleep mode. A |
| 1765 | `TestFail` exception is raised if the boot id changes. |
| 1766 | |
| 1767 | See @ref test_wait_for_sleep for more on this function's |
| 1768 | usage. |
| 1769 | |
J. Richard Barnette | 7214e0b | 2013-02-06 15:20:49 -0800 | [diff] [blame] | 1770 | @param old_boot_id A boot id value obtained before the |
J. Richard Barnette | 134ec2c | 2012-04-25 12:59:37 -0700 | [diff] [blame] | 1771 | target host went to sleep. |
Tom Wai-Hong Tam | fced4f6 | 2014-04-17 10:56:30 +0800 | [diff] [blame] | 1772 | @param resume_timeout time limit in seconds to allow the host up. |
J. Richard Barnette | 134ec2c | 2012-04-25 12:59:37 -0700 | [diff] [blame] | 1773 | |
| 1774 | @exception TestFail The host did not respond within the |
| 1775 | allowed time. |
| 1776 | @exception TestFail The host responded, but the boot id test |
| 1777 | indicated a reboot rather than a sleep |
| 1778 | cycle. |
| 1779 | """ |
Tom Wai-Hong Tam | fced4f6 | 2014-04-17 10:56:30 +0800 | [diff] [blame] | 1780 | if resume_timeout is None: |
| 1781 | resume_timeout = self.RESUME_TIMEOUT |
| 1782 | |
| 1783 | if not self.wait_up(timeout=resume_timeout): |
J. Richard Barnette | 134ec2c | 2012-04-25 12:59:37 -0700 | [diff] [blame] | 1784 | raise error.TestFail( |
| 1785 | 'client failed to resume from sleep after %d seconds' % |
Tom Wai-Hong Tam | fced4f6 | 2014-04-17 10:56:30 +0800 | [diff] [blame] | 1786 | resume_timeout) |
J. Richard Barnette | 134ec2c | 2012-04-25 12:59:37 -0700 | [diff] [blame] | 1787 | else: |
| 1788 | new_boot_id = self.get_boot_id() |
| 1789 | if new_boot_id != old_boot_id: |
Tom Wai-Hong Tam | 0179268 | 2015-01-06 08:00:46 +0800 | [diff] [blame] | 1790 | logging.error('client rebooted (old boot %s, new boot %s)', |
| 1791 | old_boot_id, new_boot_id) |
J. Richard Barnette | 134ec2c | 2012-04-25 12:59:37 -0700 | [diff] [blame] | 1792 | raise error.TestFail( |
Tom Wai-Hong Tam | 0179268 | 2015-01-06 08:00:46 +0800 | [diff] [blame] | 1793 | 'client rebooted, but sleep was expected') |
J. Richard Barnette | 134ec2c | 2012-04-25 12:59:37 -0700 | [diff] [blame] | 1794 | |
| 1795 | |
Tom Wai-Hong Tam | fe005c2 | 2014-12-03 09:25:44 +0800 | [diff] [blame] | 1796 | def test_wait_for_shutdown(self, shutdown_timeout=None): |
J. Richard Barnette | 134ec2c | 2012-04-25 12:59:37 -0700 | [diff] [blame] | 1797 | """Wait for the client to shut down. |
| 1798 | |
| 1799 | The test for "has shut down" can't distinguish a system that |
| 1800 | is merely asleep; to confirm that the unit was down, it is |
| 1801 | necessary to force boot, and then call test_wait_for_boot(). |
| 1802 | |
| 1803 | This function is expected to be called from a test as part |
| 1804 | of a sequence like the following: |
| 1805 | |
| 1806 | ~~~~~~~~ |
| 1807 | boot_id = host.get_boot_id() |
| 1808 | # trigger shutdown on the host |
| 1809 | host.test_wait_for_shutdown() |
| 1810 | # trigger boot on the host |
| 1811 | host.test_wait_for_boot(boot_id) |
| 1812 | ~~~~~~~~ |
| 1813 | |
Tom Wai-Hong Tam | fe005c2 | 2014-12-03 09:25:44 +0800 | [diff] [blame] | 1814 | @param shutdown_timeout time limit in seconds to allow the host down. |
J. Richard Barnette | 134ec2c | 2012-04-25 12:59:37 -0700 | [diff] [blame] | 1815 | @exception TestFail The host did not shut down within the |
| 1816 | allowed time. |
| 1817 | """ |
Tom Wai-Hong Tam | fe005c2 | 2014-12-03 09:25:44 +0800 | [diff] [blame] | 1818 | if shutdown_timeout is None: |
| 1819 | shutdown_timeout = self.SHUTDOWN_TIMEOUT |
| 1820 | |
| 1821 | if not self.ping_wait_down(timeout=shutdown_timeout): |
J. Richard Barnette | 134ec2c | 2012-04-25 12:59:37 -0700 | [diff] [blame] | 1822 | raise error.TestFail( |
| 1823 | 'client failed to shut down after %d seconds' % |
Tom Wai-Hong Tam | fe005c2 | 2014-12-03 09:25:44 +0800 | [diff] [blame] | 1824 | shutdown_timeout) |
J. Richard Barnette | 134ec2c | 2012-04-25 12:59:37 -0700 | [diff] [blame] | 1825 | |
| 1826 | |
| 1827 | def test_wait_for_boot(self, old_boot_id=None): |
| 1828 | """Wait for the client to boot from cold power. |
| 1829 | |
| 1830 | The `old_boot_id` parameter should be the value from |
| 1831 | `get_boot_id()` obtained prior to shutting down. A |
| 1832 | `TestFail` exception is raised if the boot id does not |
| 1833 | change. The boot id test is omitted if `old_boot_id` is not |
| 1834 | specified. |
| 1835 | |
| 1836 | See @ref test_wait_for_shutdown for more on this function's |
| 1837 | usage. |
| 1838 | |
J. Richard Barnette | 7214e0b | 2013-02-06 15:20:49 -0800 | [diff] [blame] | 1839 | @param old_boot_id A boot id value obtained before the |
J. Richard Barnette | 134ec2c | 2012-04-25 12:59:37 -0700 | [diff] [blame] | 1840 | shut down. |
| 1841 | |
| 1842 | @exception TestFail The host did not respond within the |
| 1843 | allowed time. |
| 1844 | @exception TestFail The host responded, but the boot id test |
| 1845 | indicated that there was no reboot. |
| 1846 | """ |
J. Richard Barnette | eb69d72 | 2012-06-18 17:29:44 -0700 | [diff] [blame] | 1847 | if not self.wait_up(timeout=self.REBOOT_TIMEOUT): |
J. Richard Barnette | 134ec2c | 2012-04-25 12:59:37 -0700 | [diff] [blame] | 1848 | raise error.TestFail( |
| 1849 | 'client failed to reboot after %d seconds' % |
J. Richard Barnette | eb69d72 | 2012-06-18 17:29:44 -0700 | [diff] [blame] | 1850 | self.REBOOT_TIMEOUT) |
J. Richard Barnette | 134ec2c | 2012-04-25 12:59:37 -0700 | [diff] [blame] | 1851 | elif old_boot_id: |
| 1852 | if self.get_boot_id() == old_boot_id: |
Tom Wai-Hong Tam | 0179268 | 2015-01-06 08:00:46 +0800 | [diff] [blame] | 1853 | logging.error('client not rebooted (boot %s)', |
| 1854 | old_boot_id) |
J. Richard Barnette | 134ec2c | 2012-04-25 12:59:37 -0700 | [diff] [blame] | 1855 | raise error.TestFail( |
Tom Wai-Hong Tam | 0179268 | 2015-01-06 08:00:46 +0800 | [diff] [blame] | 1856 | 'client is back up, but did not reboot') |
Simran Basi | d5e5e27 | 2012-09-24 15:23:59 -0700 | [diff] [blame] | 1857 | |
| 1858 | |
| 1859 | @staticmethod |
| 1860 | def check_for_rpm_support(hostname): |
| 1861 | """For a given hostname, return whether or not it is powered by an RPM. |
| 1862 | |
Simran Basi | 1df5511 | 2013-09-06 11:25:09 -0700 | [diff] [blame] | 1863 | @param hostname: hostname to check for rpm support. |
| 1864 | |
Simran Basi | d5e5e27 | 2012-09-24 15:23:59 -0700 | [diff] [blame] | 1865 | @return None if this host does not follows the defined naming format |
| 1866 | for RPM powered DUT's in the lab. If it does follow the format, |
| 1867 | it returns a regular expression MatchObject instead. |
| 1868 | """ |
Fang Deng | baff908 | 2015-01-06 13:46:15 -0800 | [diff] [blame] | 1869 | return re.match(CrosHost._RPM_HOSTNAME_REGEX, hostname) |
Simran Basi | d5e5e27 | 2012-09-24 15:23:59 -0700 | [diff] [blame] | 1870 | |
| 1871 | |
| 1872 | def has_power(self): |
| 1873 | """For this host, return whether or not it is powered by an RPM. |
| 1874 | |
| 1875 | @return True if this host is in the CROS lab and follows the defined |
| 1876 | naming format. |
| 1877 | """ |
Fang Deng | 0ca40e2 | 2013-08-27 17:47:44 -0700 | [diff] [blame] | 1878 | return CrosHost.check_for_rpm_support(self.hostname) |
Simran Basi | d5e5e27 | 2012-09-24 15:23:59 -0700 | [diff] [blame] | 1879 | |
| 1880 | |
Ismail Noorbasha | 07fdb61 | 2013-02-14 14:13:31 -0800 | [diff] [blame] | 1881 | def _set_power(self, state, power_method): |
| 1882 | """Sets the power to the host via RPM, Servo or manual. |
| 1883 | |
| 1884 | @param state Specifies which power state to set to DUT |
| 1885 | @param power_method Specifies which method of power control to |
| 1886 | use. By default "RPM" will be used. Valid values |
| 1887 | are the strings "RPM", "manual", "servoj10". |
| 1888 | |
| 1889 | """ |
| 1890 | ACCEPTABLE_STATES = ['ON', 'OFF'] |
| 1891 | |
| 1892 | if state.upper() not in ACCEPTABLE_STATES: |
| 1893 | raise error.TestError('State must be one of: %s.' |
| 1894 | % (ACCEPTABLE_STATES,)) |
| 1895 | |
| 1896 | if power_method == self.POWER_CONTROL_SERVO: |
| 1897 | logging.info('Setting servo port J10 to %s', state) |
| 1898 | self.servo.set('prtctl3_pwren', state.lower()) |
| 1899 | time.sleep(self._USB_POWER_TIMEOUT) |
| 1900 | elif power_method == self.POWER_CONTROL_MANUAL: |
| 1901 | logging.info('You have %d seconds to set the AC power to %s.', |
| 1902 | self._POWER_CYCLE_TIMEOUT, state) |
| 1903 | time.sleep(self._POWER_CYCLE_TIMEOUT) |
| 1904 | else: |
| 1905 | if not self.has_power(): |
| 1906 | raise error.TestFail('DUT does not have RPM connected.') |
Simran Basi | 5e6339a | 2013-03-21 11:34:32 -0700 | [diff] [blame] | 1907 | afe = frontend_wrappers.RetryingAFE(timeout_min=5, delay_sec=10) |
| 1908 | afe.set_host_attribute(self._RPM_OUTLET_CHANGED, True, |
| 1909 | hostname=self.hostname) |
Simran Basi | 1df5511 | 2013-09-06 11:25:09 -0700 | [diff] [blame] | 1910 | rpm_client.set_power(self.hostname, state.upper(), timeout_mins=5) |
Simran Basi | d5e5e27 | 2012-09-24 15:23:59 -0700 | [diff] [blame] | 1911 | |
| 1912 | |
Ismail Noorbasha | 07fdb61 | 2013-02-14 14:13:31 -0800 | [diff] [blame] | 1913 | def power_off(self, power_method=POWER_CONTROL_RPM): |
| 1914 | """Turn off power to this host via RPM, Servo or manual. |
| 1915 | |
| 1916 | @param power_method Specifies which method of power control to |
| 1917 | use. By default "RPM" will be used. Valid values |
| 1918 | are the strings "RPM", "manual", "servoj10". |
| 1919 | |
| 1920 | """ |
| 1921 | self._set_power('OFF', power_method) |
Simran Basi | d5e5e27 | 2012-09-24 15:23:59 -0700 | [diff] [blame] | 1922 | |
| 1923 | |
Ismail Noorbasha | 07fdb61 | 2013-02-14 14:13:31 -0800 | [diff] [blame] | 1924 | def power_on(self, power_method=POWER_CONTROL_RPM): |
| 1925 | """Turn on power to this host via RPM, Servo or manual. |
| 1926 | |
| 1927 | @param power_method Specifies which method of power control to |
| 1928 | use. By default "RPM" will be used. Valid values |
| 1929 | are the strings "RPM", "manual", "servoj10". |
| 1930 | |
| 1931 | """ |
| 1932 | self._set_power('ON', power_method) |
| 1933 | |
| 1934 | |
| 1935 | def power_cycle(self, power_method=POWER_CONTROL_RPM): |
| 1936 | """Cycle power to this host by turning it OFF, then ON. |
| 1937 | |
| 1938 | @param power_method Specifies which method of power control to |
| 1939 | use. By default "RPM" will be used. Valid values |
| 1940 | are the strings "RPM", "manual", "servoj10". |
| 1941 | |
| 1942 | """ |
| 1943 | if power_method in (self.POWER_CONTROL_SERVO, |
| 1944 | self.POWER_CONTROL_MANUAL): |
| 1945 | self.power_off(power_method=power_method) |
| 1946 | time.sleep(self._POWER_CYCLE_TIMEOUT) |
| 1947 | self.power_on(power_method=power_method) |
| 1948 | else: |
| 1949 | rpm_client.set_power(self.hostname, 'CYCLE') |
Simran Basi | c6f1f7a | 2012-10-16 10:47:46 -0700 | [diff] [blame] | 1950 | |
| 1951 | |
| 1952 | def get_platform(self): |
| 1953 | """Determine the correct platform label for this host. |
| 1954 | |
| 1955 | @returns a string representing this host's platform. |
| 1956 | """ |
C Shapiro | 218e875 | 2017-09-22 11:10:57 -0600 | [diff] [blame^] | 1957 | cmd = 'mosys platform model' |
| 1958 | result = self.run(command=cmd, ignore_status=True) |
| 1959 | if result.exit_status == 0: |
| 1960 | return result.stdout.strip() |
| 1961 | else: |
| 1962 | # $(mosys platform model) should support all platforms, but it |
| 1963 | # currently doesn't, so this reverts to parsing the fw |
| 1964 | # for any unsupported mosys platforms. |
| 1965 | crossystem = utils.Crossystem(self) |
| 1966 | crossystem.init() |
| 1967 | # Extract fwid value and use the leading part as the platform id. |
| 1968 | # fwid generally follow the format of {platform}.{firmware version} |
| 1969 | # Example: Alex.X.YYY.Z or Google_Alex.X.YYY.Z |
| 1970 | platform = crossystem.fwid().split('.')[0].lower() |
| 1971 | # Newer platforms start with 'Google_' while the older ones do not. |
| 1972 | return platform.replace('google_', '') |
Simran Basi | c6f1f7a | 2012-10-16 10:47:46 -0700 | [diff] [blame] | 1973 | |
| 1974 | |
Hung-ying Tyan | b132803 | 2014-04-01 14:18:54 +0800 | [diff] [blame] | 1975 | def get_architecture(self): |
| 1976 | """Determine the correct architecture label for this host. |
| 1977 | |
| 1978 | @returns a string representing this host's architecture. |
| 1979 | """ |
| 1980 | crossystem = utils.Crossystem(self) |
| 1981 | crossystem.init() |
| 1982 | return crossystem.arch() |
| 1983 | |
| 1984 | |
Luis Lozano | 40b7d0d | 2014-01-17 15:12:06 -0800 | [diff] [blame] | 1985 | def get_chrome_version(self): |
| 1986 | """Gets the Chrome version number and milestone as strings. |
| 1987 | |
| 1988 | Invokes "chrome --version" to get the version number and milestone. |
| 1989 | |
| 1990 | @return A tuple (chrome_ver, milestone) where "chrome_ver" is the |
| 1991 | current Chrome version number as a string (in the form "W.X.Y.Z") |
| 1992 | and "milestone" is the first component of the version number |
| 1993 | (the "W" from "W.X.Y.Z"). If the version number cannot be parsed |
| 1994 | in the "W.X.Y.Z" format, the "chrome_ver" will be the full output |
| 1995 | of "chrome --version" and the milestone will be the empty string. |
| 1996 | |
| 1997 | """ |
MK Ryu | 35d661e | 2014-09-25 17:44:10 -0700 | [diff] [blame] | 1998 | version_string = self.run(client_constants.CHROME_VERSION_COMMAND).stdout |
Luis Lozano | 40b7d0d | 2014-01-17 15:12:06 -0800 | [diff] [blame] | 1999 | return utils.parse_chrome_version(version_string) |
| 2000 | |
J. Richard Barnette | d2af585 | 2016-02-05 15:03:10 -0800 | [diff] [blame] | 2001 | |
Niranjan Kumar | 3461887 | 2017-05-31 12:57:09 -0700 | [diff] [blame] | 2002 | def is_chrome_switch_present(self, switch): |
David Haddock | 3ce538e | 2017-06-22 13:37:05 -0700 | [diff] [blame] | 2003 | """Returns True if the specified switch was provided to Chrome. |
| 2004 | |
| 2005 | @param switch The chrome switch to search for. |
| 2006 | """ |
Niranjan Kumar | 3461887 | 2017-05-31 12:57:09 -0700 | [diff] [blame] | 2007 | |
Niranjan Kumar | 5f23fe9 | 2017-06-22 15:18:55 -0700 | [diff] [blame] | 2008 | command = 'pgrep -x -f -c "/opt/google/chrome/chrome.*%s.*"' % switch |
| 2009 | return self.run(command, ignore_status=True).exit_status == 0 |
Niranjan Kumar | 3461887 | 2017-05-31 12:57:09 -0700 | [diff] [blame] | 2010 | |
| 2011 | |
| 2012 | def oobe_triggers_update(self): |
| 2013 | """Returns True if this host has an OOBE flow during which |
| 2014 | it will perform an update check and perhaps an update. |
| 2015 | One example of such a flow is Hands-Off Zero-Touch Enrollment. |
| 2016 | As more such flows are developed, code handling them needs |
| 2017 | to be added here. |
| 2018 | |
| 2019 | @return Boolean indicating whether this host's OOBE triggers an update. |
| 2020 | """ |
| 2021 | return self.is_chrome_switch_present( |
| 2022 | '--enterprise-enable-zero-touch-enrollment=hands-off') |
| 2023 | |
| 2024 | |
Kevin Cheng | a2619dc | 2016-03-28 11:42:08 -0700 | [diff] [blame] | 2025 | # TODO(kevcheng): change this to just return the board without the |
| 2026 | # 'board:' prefix and fix up all the callers. Also look into removing the |
| 2027 | # need for this method. |
Simran Basi | c6f1f7a | 2012-10-16 10:47:46 -0700 | [diff] [blame] | 2028 | def get_board(self): |
| 2029 | """Determine the correct board label for this host. |
| 2030 | |
| 2031 | @returns a string representing this host's board. |
| 2032 | """ |
| 2033 | release_info = utils.parse_cmd_output('cat /etc/lsb-release', |
| 2034 | run_method=self.run) |
J. Richard Barnette | d2af585 | 2016-02-05 15:03:10 -0800 | [diff] [blame] | 2035 | return (ds_constants.BOARD_PREFIX + |
| 2036 | release_info['CHROMEOS_RELEASE_BOARD']) |
Simran Basi | c6f1f7a | 2012-10-16 10:47:46 -0700 | [diff] [blame] | 2037 | |
| 2038 | |
Kevin Cheng | a328da6 | 2016-03-31 10:49:04 -0700 | [diff] [blame] | 2039 | |
| 2040 | def has_lightsensor(self): |
| 2041 | """Determine the correct board label for this host. |
| 2042 | |
| 2043 | @returns the string 'lightsensor' if this host has a lightsensor or |
| 2044 | None if it does not. |
| 2045 | """ |
| 2046 | search_cmd = "find -L %s -maxdepth 4 | egrep '%s'" % ( |
| 2047 | self._LIGHTSENSOR_SEARCH_DIR, '|'.join(self._LIGHTSENSOR_FILES)) |
| 2048 | try: |
| 2049 | # Run the search cmd following the symlinks. Stderr_tee is set to |
| 2050 | # None as there can be a symlink loop, but the command will still |
| 2051 | # execute correctly with a few messages printed to stderr. |
| 2052 | self.run(search_cmd, stdout_tee=None, stderr_tee=None) |
| 2053 | return 'lightsensor' |
| 2054 | except error.AutoservRunError: |
| 2055 | # egrep exited with a return code of 1 meaning none of the possible |
| 2056 | # lightsensor files existed. |
| 2057 | return None |
| 2058 | |
| 2059 | |
| 2060 | def has_bluetooth(self): |
| 2061 | """Determine the correct board label for this host. |
| 2062 | |
| 2063 | @returns the string 'bluetooth' if this host has bluetooth or |
| 2064 | None if it does not. |
| 2065 | """ |
| 2066 | try: |
| 2067 | self.run('test -d /sys/class/bluetooth/hci0') |
| 2068 | # test exited with a return code of 0. |
| 2069 | return 'bluetooth' |
| 2070 | except error.AutoservRunError: |
| 2071 | # test exited with a return code 1 meaning the directory did not |
| 2072 | # exist. |
| 2073 | return None |
| 2074 | |
| 2075 | |
Kevin Cheng | a328da6 | 2016-03-31 10:49:04 -0700 | [diff] [blame] | 2076 | def get_accels(self): |
| 2077 | """ |
| 2078 | Determine the type of accelerometers on this host. |
| 2079 | |
| 2080 | @returns a string representing this host's accelerometer type. |
| 2081 | At present, it only returns "accel:cros-ec", for accelerometers |
| 2082 | attached to a Chrome OS EC, or none, if no accelerometers. |
| 2083 | """ |
| 2084 | # Check to make sure we have ectool |
| 2085 | rv = self.run('which ectool', ignore_status=True) |
| 2086 | if rv.exit_status: |
| 2087 | logging.info("No ectool cmd found, assuming no EC accelerometers") |
| 2088 | return None |
| 2089 | |
| 2090 | # Check that the EC supports the motionsense command |
| 2091 | rv = self.run('ectool motionsense', ignore_status=True) |
| 2092 | if rv.exit_status: |
| 2093 | logging.info("EC does not support motionsense command " |
| 2094 | "assuming no EC accelerometers") |
| 2095 | return None |
| 2096 | |
| 2097 | # Check that EC motion sensors are active |
| 2098 | active = self.run('ectool motionsense active').stdout.split('\n') |
| 2099 | if active[0] == "0": |
| 2100 | logging.info("Motion sense inactive, assuming no EC accelerometers") |
| 2101 | return None |
| 2102 | |
| 2103 | logging.info("EC accelerometers found") |
| 2104 | return 'accel:cros-ec' |
| 2105 | |
| 2106 | |
| 2107 | def has_chameleon(self): |
| 2108 | """Determine if a Chameleon connected to this host. |
| 2109 | |
| 2110 | @returns a list containing two strings ('chameleon' and |
| 2111 | 'chameleon:' + label, e.g. 'chameleon:hdmi') if this host |
| 2112 | has a Chameleon or None if it has not. |
| 2113 | """ |
| 2114 | if self._chameleon_host: |
| 2115 | return ['chameleon', 'chameleon:' + self.chameleon.get_label()] |
| 2116 | else: |
| 2117 | return None |
| 2118 | |
| 2119 | |
| 2120 | def has_loopback_dongle(self): |
| 2121 | """Determine if an audio loopback dongle is plugged to this host. |
| 2122 | |
| 2123 | @returns 'audio_loopback_dongle' when there is an audio loopback dongle |
| 2124 | plugged to this host. |
| 2125 | None when there is no audio loopback dongle |
| 2126 | plugged to this host. |
| 2127 | """ |
| 2128 | nodes_info = self.run(command=cras_utils.get_cras_nodes_cmd(), |
| 2129 | ignore_status=True).stdout |
| 2130 | if (cras_utils.node_type_is_plugged('HEADPHONE', nodes_info) and |
| 2131 | cras_utils.node_type_is_plugged('MIC', nodes_info)): |
| 2132 | return 'audio_loopback_dongle' |
| 2133 | else: |
| 2134 | return None |
| 2135 | |
| 2136 | |
| 2137 | def get_power_supply(self): |
| 2138 | """ |
| 2139 | Determine what type of power supply the host has |
| 2140 | |
| 2141 | @returns a string representing this host's power supply. |
| 2142 | 'power:battery' when the device has a battery intended for |
| 2143 | extended use |
| 2144 | 'power:AC_primary' when the device has a battery not intended |
| 2145 | for extended use (for moving the machine, etc) |
| 2146 | 'power:AC_only' when the device has no battery at all. |
| 2147 | """ |
| 2148 | psu = self.run(command='mosys psu type', ignore_status=True) |
| 2149 | if psu.exit_status: |
| 2150 | # The psu command for mosys is not included for all platforms. The |
| 2151 | # assumption is that the device will have a battery if the command |
| 2152 | # is not found. |
| 2153 | return 'power:battery' |
| 2154 | |
| 2155 | psu_str = psu.stdout.strip() |
| 2156 | if psu_str == 'unknown': |
| 2157 | return None |
| 2158 | |
| 2159 | return 'power:%s' % psu_str |
| 2160 | |
| 2161 | |
| 2162 | def get_storage(self): |
| 2163 | """ |
| 2164 | Determine the type of boot device for this host. |
| 2165 | |
| 2166 | Determine if the internal device is SCSI or dw_mmc device. |
| 2167 | Then check that it is SSD or HDD or eMMC or something else. |
| 2168 | |
| 2169 | @returns a string representing this host's internal device type. |
| 2170 | 'storage:ssd' when internal device is solid state drive |
| 2171 | 'storage:hdd' when internal device is hard disk drive |
| 2172 | 'storage:mmc' when internal device is mmc drive |
| 2173 | None When internal device is something else or |
| 2174 | when we are unable to determine the type |
| 2175 | """ |
| 2176 | # The output should be /dev/mmcblk* for SD/eMMC or /dev/sd* for scsi |
| 2177 | rootdev_cmd = ' '.join(['. /usr/sbin/write_gpt.sh;', |
| 2178 | '. /usr/share/misc/chromeos-common.sh;', |
| 2179 | 'load_base_vars;', |
| 2180 | 'get_fixed_dst_drive']) |
| 2181 | rootdev = self.run(command=rootdev_cmd, ignore_status=True) |
| 2182 | if rootdev.exit_status: |
| 2183 | logging.info("Fail to run %s", rootdev_cmd) |
| 2184 | return None |
| 2185 | rootdev_str = rootdev.stdout.strip() |
| 2186 | |
| 2187 | if not rootdev_str: |
| 2188 | return None |
| 2189 | |
| 2190 | rootdev_base = os.path.basename(rootdev_str) |
| 2191 | |
| 2192 | mmc_pattern = '/dev/mmcblk[0-9]' |
| 2193 | if re.match(mmc_pattern, rootdev_str): |
| 2194 | # Use type to determine if the internal device is eMMC or somthing |
| 2195 | # else. We can assume that MMC is always an internal device. |
| 2196 | type_cmd = 'cat /sys/block/%s/device/type' % rootdev_base |
| 2197 | type = self.run(command=type_cmd, ignore_status=True) |
| 2198 | if type.exit_status: |
| 2199 | logging.info("Fail to run %s", type_cmd) |
| 2200 | return None |
| 2201 | type_str = type.stdout.strip() |
| 2202 | |
| 2203 | if type_str == 'MMC': |
| 2204 | return 'storage:mmc' |
| 2205 | |
| 2206 | scsi_pattern = '/dev/sd[a-z]+' |
| 2207 | if re.match(scsi_pattern, rootdev.stdout): |
| 2208 | # Read symlink for /sys/block/sd* to determine if the internal |
| 2209 | # device is connected via ata or usb. |
| 2210 | link_cmd = 'readlink /sys/block/%s' % rootdev_base |
| 2211 | link = self.run(command=link_cmd, ignore_status=True) |
| 2212 | if link.exit_status: |
| 2213 | logging.info("Fail to run %s", link_cmd) |
| 2214 | return None |
| 2215 | link_str = link.stdout.strip() |
| 2216 | if 'usb' in link_str: |
| 2217 | return None |
| 2218 | |
| 2219 | # Read rotation to determine if the internal device is ssd or hdd. |
| 2220 | rotate_cmd = str('cat /sys/block/%s/queue/rotational' |
| 2221 | % rootdev_base) |
| 2222 | rotate = self.run(command=rotate_cmd, ignore_status=True) |
| 2223 | if rotate.exit_status: |
| 2224 | logging.info("Fail to run %s", rotate_cmd) |
| 2225 | return None |
| 2226 | rotate_str = rotate.stdout.strip() |
| 2227 | |
| 2228 | rotate_dict = {'0':'storage:ssd', '1':'storage:hdd'} |
| 2229 | return rotate_dict.get(rotate_str) |
| 2230 | |
| 2231 | # All other internal device / error case will always fall here |
| 2232 | return None |
| 2233 | |
| 2234 | |
| 2235 | def get_servo(self): |
| 2236 | """Determine if the host has a servo attached. |
| 2237 | |
| 2238 | If the host has a working servo attached, it should have a servo label. |
| 2239 | |
| 2240 | @return: string 'servo' if the host has servo attached. Otherwise, |
| 2241 | returns None. |
| 2242 | """ |
| 2243 | return 'servo' if self._servo_host else None |
| 2244 | |
| 2245 | |
| 2246 | def get_video_labels(self): |
| 2247 | """Run /usr/local/bin/avtest_label_detect to get a list of video labels. |
| 2248 | |
| 2249 | Sample output of avtest_label_detect: |
| 2250 | Detected label: hw_video_acc_vp8 |
| 2251 | Detected label: webcam |
| 2252 | |
| 2253 | @return: A list of labels detected by tool avtest_label_detect. |
| 2254 | """ |
| 2255 | try: |
| 2256 | result = self.run('/usr/local/bin/avtest_label_detect').stdout |
| 2257 | return re.findall('^Detected label: (\w+)$', result, re.M) |
| 2258 | except error.AutoservRunError: |
| 2259 | # The tool is not installed. |
| 2260 | return [] |
| 2261 | |
| 2262 | |
| 2263 | def is_video_glitch_detection_supported(self): |
| 2264 | """ Determine if a board under test is supported for video glitch |
| 2265 | detection tests. |
| 2266 | |
| 2267 | @return: 'video_glitch_detection' if board is supported, None otherwise. |
| 2268 | """ |
| 2269 | board = self.get_board().replace(ds_constants.BOARD_PREFIX, '') |
| 2270 | |
| 2271 | if board in video_test_constants.SUPPORTED_BOARDS: |
| 2272 | return 'video_glitch_detection' |
| 2273 | |
| 2274 | return None |
| 2275 | |
| 2276 | |
| 2277 | def get_touch(self): |
| 2278 | """ |
| 2279 | Determine whether board under test has a touchpad or touchscreen. |
| 2280 | |
| 2281 | @return: A list of some combination of 'touchscreen' and 'touchpad', |
| 2282 | depending on what is present on the device. |
| 2283 | |
| 2284 | """ |
| 2285 | labels = [] |
| 2286 | looking_for = ['touchpad', 'touchscreen'] |
| 2287 | player = input_playback.InputPlayback() |
| 2288 | input_events = self.run('ls /dev/input/event*').stdout.strip().split() |
| 2289 | filename = '/tmp/touch_labels' |
| 2290 | for event in input_events: |
| 2291 | self.run('evtest %s > %s' % (event, filename), timeout=1, |
| 2292 | ignore_timeout=True) |
| 2293 | properties = self.run('cat %s' % filename).stdout |
| 2294 | input_type = player._determine_input_type(properties) |
| 2295 | if input_type in looking_for: |
| 2296 | labels.append(input_type) |
| 2297 | looking_for.remove(input_type) |
| 2298 | if len(looking_for) == 0: |
| 2299 | break |
| 2300 | self.run('rm %s' % filename) |
| 2301 | |
| 2302 | return labels |
| 2303 | |
| 2304 | |
| 2305 | def has_internal_display(self): |
| 2306 | """Determine if the device under test is equipped with an internal |
| 2307 | display. |
| 2308 | |
| 2309 | @return: 'internal_display' if one is present; None otherwise. |
| 2310 | """ |
| 2311 | from autotest_lib.client.cros.graphics import graphics_utils |
| 2312 | from autotest_lib.client.common_lib import utils as common_utils |
| 2313 | |
| 2314 | def __system_output(cmd): |
| 2315 | return self.run(cmd).stdout |
| 2316 | |
| 2317 | def __read_file(remote_path): |
| 2318 | return self.run('cat %s' % remote_path).stdout |
| 2319 | |
| 2320 | # Hijack the necessary client functions so that we can take advantage |
| 2321 | # of the client lib here. |
| 2322 | # FIXME: find a less hacky way than this |
| 2323 | original_system_output = utils.system_output |
| 2324 | original_read_file = common_utils.read_file |
| 2325 | utils.system_output = __system_output |
| 2326 | common_utils.read_file = __read_file |
| 2327 | try: |
| 2328 | return ('internal_display' if graphics_utils.has_internal_display() |
| 2329 | else None) |
| 2330 | finally: |
| 2331 | utils.system_output = original_system_output |
| 2332 | common_utils.read_file = original_read_file |
| 2333 | |
| 2334 | |
Dan Shi | 85276d4 | 2014-04-08 22:11:45 -0700 | [diff] [blame] | 2335 | def is_boot_from_usb(self): |
| 2336 | """Check if DUT is boot from USB. |
| 2337 | |
| 2338 | @return: True if DUT is boot from usb. |
| 2339 | """ |
| 2340 | device = self.run('rootdev -s -d').stdout.strip() |
| 2341 | removable = int(self.run('cat /sys/block/%s/removable' % |
| 2342 | os.path.basename(device)).stdout.strip()) |
| 2343 | return removable == 1 |
Helen Zhang | 17dae2b | 2014-11-11 09:25:52 -0800 | [diff] [blame] | 2344 | |
| 2345 | |
| 2346 | def read_from_meminfo(self, key): |
Dan Shi | 49ca093 | 2014-11-14 11:22:27 -0800 | [diff] [blame] | 2347 | """Return the memory info from /proc/meminfo |
Helen Zhang | 17dae2b | 2014-11-11 09:25:52 -0800 | [diff] [blame] | 2348 | |
| 2349 | @param key: meminfo requested |
| 2350 | |
| 2351 | @return the memory value as a string |
| 2352 | |
| 2353 | """ |
Helen Zhang | 17dae2b | 2014-11-11 09:25:52 -0800 | [diff] [blame] | 2354 | meminfo = self.run('grep %s /proc/meminfo' % key).stdout.strip() |
| 2355 | logging.debug('%s', meminfo) |
| 2356 | return int(re.search(r'\d+', meminfo).group(0)) |
Rohit Makasana | 8a4923c | 2015-08-13 17:04:26 -0700 | [diff] [blame] | 2357 | |
| 2358 | |
Rohit Makasana | 98e696f | 2016-06-03 18:48:10 -0700 | [diff] [blame] | 2359 | def get_cpu_arch(self): |
| 2360 | """Returns CPU arch of the device. |
| 2361 | |
| 2362 | @return CPU architecture of the DUT. |
| 2363 | """ |
Allen Li | 2c32d6b | 2017-02-03 15:28:10 -0800 | [diff] [blame] | 2364 | # Add CPUs by following logic in client/bin/utils.py. |
Rohit Makasana | 98e696f | 2016-06-03 18:48:10 -0700 | [diff] [blame] | 2365 | if self.run("grep '^flags.*:.* lm .*' /proc/cpuinfo", |
| 2366 | ignore_status=True).stdout: |
| 2367 | return 'x86_64' |
| 2368 | if self.run("grep -Ei 'ARM|CPU implementer' /proc/cpuinfo", |
| 2369 | ignore_status=True).stdout: |
| 2370 | return 'arm' |
| 2371 | return 'i386' |
| 2372 | |
| 2373 | |
Rohit Makasana | 8a4923c | 2015-08-13 17:04:26 -0700 | [diff] [blame] | 2374 | def get_board_type(self): |
| 2375 | """ |
| 2376 | Get the DUT's device type from /etc/lsb-release. |
Danny Chan | 471a8d1 | 2015-08-18 14:57:41 -0700 | [diff] [blame] | 2377 | DEVICETYPE can be one of CHROMEBOX, CHROMEBASE, CHROMEBOOK or more. |
| 2378 | |
| 2379 | @return value of DEVICETYPE param from lsb-release. |
Rohit Makasana | 8a4923c | 2015-08-13 17:04:26 -0700 | [diff] [blame] | 2380 | """ |
Danny Chan | 471a8d1 | 2015-08-18 14:57:41 -0700 | [diff] [blame] | 2381 | device_type = self.run('grep DEVICETYPE /etc/lsb-release', |
| 2382 | ignore_status=True).stdout |
| 2383 | if device_type: |
Kalin Stoyanov | 524310b | 2015-08-21 16:24:04 -0700 | [diff] [blame] | 2384 | return device_type.split('=')[-1].strip() |
Danny Chan | 471a8d1 | 2015-08-18 14:57:41 -0700 | [diff] [blame] | 2385 | return '' |
Gilad Arnold | a76bef0 | 2015-09-29 13:55:15 -0700 | [diff] [blame] | 2386 | |
| 2387 | |
Rohit Makasana | df0a3a3 | 2017-06-30 13:55:18 -0700 | [diff] [blame] | 2388 | def get_arc_version(self): |
| 2389 | """Return ARC version installed on the DUT. |
| 2390 | |
| 2391 | @returns ARC version as string if the CrOS build has ARC, else None. |
| 2392 | """ |
| 2393 | arc_version = self.run('grep CHROMEOS_ARC_VERSION /etc/lsb-release', |
| 2394 | ignore_status=True).stdout |
| 2395 | if arc_version: |
| 2396 | return arc_version.split('=')[-1].strip() |
| 2397 | return None |
| 2398 | |
| 2399 | |
Gilad Arnold | a76bef0 | 2015-09-29 13:55:15 -0700 | [diff] [blame] | 2400 | def get_os_type(self): |
| 2401 | return 'cros' |
Simran Basi | a5522a3 | 2015-10-06 11:01:24 -0700 | [diff] [blame] | 2402 | |
| 2403 | |
| 2404 | def enable_adb_testing(self): |
| 2405 | """Mark this host as an adb tester.""" |
Dan Shi | a287217 | 2015-10-31 01:16:51 -0700 | [diff] [blame] | 2406 | self.run('touch %s' % constants.ANDROID_TESTER_FILEFLAG) |
Kevin Cheng | a2619dc | 2016-03-28 11:42:08 -0700 | [diff] [blame] | 2407 | |
| 2408 | |
| 2409 | def get_labels(self): |
Kevin Cheng | f866014 | 2016-08-12 10:17:41 -0700 | [diff] [blame] | 2410 | """Return the detected labels on the host.""" |
Kevin Cheng | a2619dc | 2016-03-28 11:42:08 -0700 | [diff] [blame] | 2411 | return self.labels.get_labels(self) |
Kevin Cheng | f866014 | 2016-08-12 10:17:41 -0700 | [diff] [blame] | 2412 | |
| 2413 | |
| 2414 | def update_labels(self): |
| 2415 | """Update the labels on the host.""" |
| 2416 | self.labels.update_labels(self) |