111 lines
4.6 KiB
Python
111 lines
4.6 KiB
Python
# Copyright (c) 2012 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.server import utils
|
|
from autotest_lib.server.cros.faft.firmware_test import FirmwareTest
|
|
from autotest_lib.client.common_lib import error
|
|
|
|
|
|
class firmware_UpdateFirmwareVersion(FirmwareTest):
|
|
"""
|
|
Servo based firmware update test which checks the firmware version.
|
|
|
|
This test requires a USB test image plugged in. The firmware id of
|
|
the current running firmware must matches the system shellball's, or user
|
|
can provide a shellball to do this test. In this way, the client will be
|
|
update with the given shellball first. On runtime, this test modifies the
|
|
firmware version of the shellball and runs autoupdate. Check firmware
|
|
version after boot with firmware B, and then recover firmware A and B to
|
|
original shellball.
|
|
"""
|
|
version = 1
|
|
NEEDS_SERVO_USB = True
|
|
|
|
def check_firmware_version(self, expected_ver):
|
|
"""Checks the firmware version."""
|
|
actual_ver = self.faft_client.bios.get_version(
|
|
'b' if self.fw_vboot2 else 'a')
|
|
actual_tpm_fwver = self.faft_client.tpm.get_firmware_version()
|
|
if actual_ver != expected_ver or actual_tpm_fwver != expected_ver:
|
|
raise error.TestFail(
|
|
'Firmware version should be %s,'
|
|
'but got (fwver, tpm_fwver) = (%s, %s).' %
|
|
(expected_ver, actual_ver, actual_tpm_fwver))
|
|
else:
|
|
logging.info(
|
|
'Update success, now version is %s',
|
|
actual_ver)
|
|
|
|
def initialize(self, host, cmdline_args):
|
|
"""Setup the test"""
|
|
dict_args = utils.args_to_dict(cmdline_args)
|
|
shellball_path = dict_args.get('shellball', None)
|
|
super(firmware_UpdateFirmwareVersion, self).initialize(
|
|
host, cmdline_args)
|
|
self.switcher.setup_mode('normal')
|
|
self.backup_firmware()
|
|
self.setup_firmwareupdate_shellball(shellball_path)
|
|
|
|
# Update firmware if needed
|
|
if shellball_path:
|
|
self.set_ap_write_protect_and_reboot(enable=False)
|
|
self.faft_client.updater.run_factory_install()
|
|
self.switcher.mode_aware_reboot()
|
|
|
|
self.setup_usbkey(usbkey=True)
|
|
|
|
self._fwid = self.faft_client.updater.get_section_fwid()
|
|
|
|
self.fw_ver_tpm = self.faft_client.tpm.get_firmware_version()
|
|
actual_ver = self.faft_client.bios.get_version('a')
|
|
logging.info('Origin version is %s', actual_ver)
|
|
self._update_version = actual_ver + 1
|
|
logging.info('Firmware version will update to version %s',
|
|
self._update_version)
|
|
|
|
self.faft_client.updater.resign_firmware(self._update_version)
|
|
self.faft_client.updater.repack_shellball('test')
|
|
|
|
def cleanup(self):
|
|
"""Cleanup after the test"""
|
|
try:
|
|
if self.faft_client.tpm.get_firmware_version() != self.fw_ver_tpm:
|
|
self.reboot_and_reset_tpm()
|
|
self.restore_firmware()
|
|
self.invalidate_firmware_setup()
|
|
except Exception as e:
|
|
logging.error("Caught exception: %s", str(e))
|
|
super(firmware_UpdateFirmwareVersion, self).cleanup()
|
|
|
|
def run_once(self):
|
|
"""Runs a single iteration of the test."""
|
|
logging.info("Update firmware with new version.")
|
|
self.check_state((self.checkers.crossystem_checker, {
|
|
'fwid': self._fwid
|
|
}))
|
|
self.check_state((self.checkers.fw_tries_checker, 'A'))
|
|
self.faft_client.updater.run_autoupdate('test')
|
|
self.switcher.mode_aware_reboot()
|
|
|
|
logging.info("Copy firmware form B to A.")
|
|
self.faft_client.updater.run_bootok('test')
|
|
self.check_state((self.checkers.fw_tries_checker, 'B'))
|
|
self.switcher.mode_aware_reboot()
|
|
|
|
logging.info("Check firmware and TPM version, then recovery.")
|
|
self.check_state((self.checkers.fw_tries_checker,
|
|
'B' if self.fw_vboot2 else 'A'))
|
|
self.check_firmware_version(self._update_version)
|
|
self.faft_client.updater.run_recovery()
|
|
self.reboot_and_reset_tpm()
|
|
|
|
logging.info("Check Rollback version.")
|
|
self.check_state((self.checkers.crossystem_checker, {
|
|
'fwid': self._fwid
|
|
}))
|
|
self.check_state((self.checkers.fw_tries_checker,
|
|
'B' if self.fw_vboot2 else 'A'))
|
|
self.check_firmware_version(self._update_version - 1)
|