| # Copyright (c) 2011 The Chromium OS Authors. All rights reserved. |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| import logging |
| |
| from autotest_lib.client.common_lib import error |
| from autotest_lib.server.cros import vboot_constants as vboot |
| from autotest_lib.server.cros.faft.firmware_test import ConnectionError |
| from autotest_lib.server.cros.faft.firmware_test import FirmwareTest |
| |
| |
| class firmware_InvalidUSB(FirmwareTest): |
| """ |
| Servo based booting an invalid USB image test. |
| |
| This test requires a USB disk plugged-in, which contains a Chrome OS test |
| image (built by "build_image --test"). On runtime, this test corrupts the |
| USB image and tries to boot into it. A failure is expected. It then |
| restores the USB image and boots into it again. |
| """ |
| version = 1 |
| |
| def restore_usb(self): |
| """Restore the USB image. USB plugs/unplugs happen in this method.""" |
| self.servo.switch_usbkey('host') |
| usb_dev = self.servo.probe_host_usb_dev() |
| self.restore_usb_kernel(usb_dev) |
| |
| def initialize(self, host, cmdline_args): |
| """Initialize the test""" |
| super(firmware_InvalidUSB, self).initialize(host, cmdline_args) |
| self.servo.switch_usbkey('host') |
| usb_dev = self.servo.probe_host_usb_dev() |
| self.assert_test_image_in_usb_disk(usb_dev) |
| self.corrupt_usb_kernel(usb_dev) |
| self.switcher.setup_mode('normal') |
| self.servo.switch_usbkey('dut') |
| |
| def cleanup(self): |
| """Cleanup the test""" |
| try: |
| self.restore_usb() |
| except Exception as e: |
| logging.error("Caught exception: %s", str(e)) |
| super(firmware_InvalidUSB, self).cleanup() |
| |
| def run_once(self): |
| """Main test logic""" |
| logging.info("Turn on the recovery boot. Remove and insert the" |
| "corrupted USB stick, a boot failure is expected." |
| "Restore the USB image and boot it again.") |
| self.check_state((self.checkers.crossystem_checker, { |
| 'devsw_boot': '0', |
| 'mainfw_type': 'normal', |
| })) |
| |
| # Switch servo v4 (if present) as a SNK. Make sure USB key is bootable. |
| self.set_servo_v4_role_to_snk() |
| |
| self.switcher.reboot_to_mode(to_mode='rec', wait_for_dut_up=False) |
| logging.info('Wait to ensure the USB image is unable to boot...') |
| try: |
| self.switcher.wait_for_client( |
| timeout=self.faft_config.usb_image_boot_timeout) |
| raise error.TestFail('Should not boot from the invalid USB image.') |
| except ConnectionError: |
| logging.info( |
| 'The USB image is surely unable to boot. Restore it and try...') |
| |
| self.restore_usb() |
| self.servo.switch_usbkey('dut') |
| self.switcher.wait_for_client() |
| |
| logging.info("Expected to boot the restored USB image and reboot.") |
| self.check_state((self.checkers.crossystem_checker, { |
| 'mainfw_type': 'recovery', |
| 'recovery_reason': vboot.RECOVERY_REASON['RO_MANUAL'], |
| })) |
| self.switcher.mode_aware_reboot() |
| |
| logging.info("Expected to normal boot and done.") |
| self.check_state((self.checkers.crossystem_checker, { |
| 'devsw_boot': '0', |
| 'mainfw_type': 'normal', |
| })) |
