146d03b126f07e8ad45c04e3d7d562b03778feec0Tom Wai-Hong Tam# Copyright (c) 2012 The Chromium OS Authors. All rights reserved. 246d03b126f07e8ad45c04e3d7d562b03778feec0Tom Wai-Hong Tam# Use of this source code is governed by a BSD-style license that can be 346d03b126f07e8ad45c04e3d7d562b03778feec0Tom Wai-Hong Tam# found in the LICENSE file. 446d03b126f07e8ad45c04e3d7d562b03778feec0Tom Wai-Hong Tam 5d74e1bbcb1b7ecabea3c9ea536d0123e71d33917Yusuf Mohsinallyimport logging 6d74e1bbcb1b7ecabea3c9ea536d0123e71d33917Yusuf Mohsinally 76ec46e3ef11a31c23917e1c1b8a661bb39949cefTom Wai-Hong Tamfrom autotest_lib.server.cros import vboot_constants as vboot 8d74e1bbcb1b7ecabea3c9ea536d0123e71d33917Yusuf Mohsinallyfrom autotest_lib.server.cros.faft.firmware_test import FirmwareTest 946d03b126f07e8ad45c04e3d7d562b03778feec0Tom Wai-Hong Tam 1046d03b126f07e8ad45c04e3d7d562b03778feec0Tom Wai-Hong Tam 11d74e1bbcb1b7ecabea3c9ea536d0123e71d33917Yusuf Mohsinallyclass firmware_RollbackFirmware(FirmwareTest): 1246d03b126f07e8ad45c04e3d7d562b03778feec0Tom Wai-Hong Tam """ 1346d03b126f07e8ad45c04e3d7d562b03778feec0Tom Wai-Hong Tam Servo based firmware rollback test. 1446d03b126f07e8ad45c04e3d7d562b03778feec0Tom Wai-Hong Tam 1546d03b126f07e8ad45c04e3d7d562b03778feec0Tom Wai-Hong Tam This test requires a USB disk plugged-in, which contains a Chrome OS test 1646d03b126f07e8ad45c04e3d7d562b03778feec0Tom Wai-Hong Tam image (built by "build_image --test"). On runtime, this test rollbacks 1746d03b126f07e8ad45c04e3d7d562b03778feec0Tom Wai-Hong Tam firmware A and results firmware B boot. It then rollbacks firmware B and 1846d03b126f07e8ad45c04e3d7d562b03778feec0Tom Wai-Hong Tam results recovery boot. 1946d03b126f07e8ad45c04e3d7d562b03778feec0Tom Wai-Hong Tam """ 2046d03b126f07e8ad45c04e3d7d562b03778feec0Tom Wai-Hong Tam version = 1 2146d03b126f07e8ad45c04e3d7d562b03778feec0Tom Wai-Hong Tam 221f115ae9ee3c043e9577e7ce9872dc99323fdcb3Yusuf Mohsinally def initialize(self, host, cmdline_args, dev_mode=False): 231f115ae9ee3c043e9577e7ce9872dc99323fdcb3Yusuf Mohsinally super(firmware_RollbackFirmware, self).initialize(host, cmdline_args) 24fa47407a2de9b579ac98c87c40c2117c334248a4Chun-ting Chang self.backup_firmware() 250cc9a4fe28f221cfc73bb35c30fa79338c4b43f1Tom Wai-Hong Tam self.switcher.setup_mode('dev' if dev_mode else 'normal') 26893bcc0871678b7ba9f70ac1f189e19db8c8db18Tom Wai-Hong Tam self.setup_usbkey(usbkey=True, host=False) 2746d03b126f07e8ad45c04e3d7d562b03778feec0Tom Wai-Hong Tam 2846d03b126f07e8ad45c04e3d7d562b03778feec0Tom Wai-Hong Tam def cleanup(self): 29fa47407a2de9b579ac98c87c40c2117c334248a4Chun-ting Chang self.restore_firmware() 3046d03b126f07e8ad45c04e3d7d562b03778feec0Tom Wai-Hong Tam super(firmware_RollbackFirmware, self).cleanup() 3146d03b126f07e8ad45c04e3d7d562b03778feec0Tom Wai-Hong Tam 32b09a04183cf7017eb4b30ad28f5f81f1204c36b3Vic Yang def run_once(self, dev_mode=False): 33d74e1bbcb1b7ecabea3c9ea536d0123e71d33917Yusuf Mohsinally logging.info("Rollback firmware A.") 343edea989126b84d38135371e780184694892a55aShelley Chen self.check_state((self.checkers.fw_tries_checker, 'A')) 35d74e1bbcb1b7ecabea3c9ea536d0123e71d33917Yusuf Mohsinally self.faft_client.bios.move_version_backward('a') 364777624b69b3b223a5d3c3abe86000783f6c0e96Tom Wai-Hong Tam self.switcher.mode_aware_reboot() 37d74e1bbcb1b7ecabea3c9ea536d0123e71d33917Yusuf Mohsinally 38d74e1bbcb1b7ecabea3c9ea536d0123e71d33917Yusuf Mohsinally logging.info("Expected firmware B boot and rollback firmware B.") 393edea989126b84d38135371e780184694892a55aShelley Chen self.check_state((self.checkers.fw_tries_checker, ('B', False))) 40d74e1bbcb1b7ecabea3c9ea536d0123e71d33917Yusuf Mohsinally self.faft_client.bios.move_version_backward('b') 41e14953ba021b2fcec2bd20bda13e49190b174eb1Shelley Chen self.switcher.simple_reboot() 42e14953ba021b2fcec2bd20bda13e49190b174eb1Shelley Chen if not dev_mode: 43e14953ba021b2fcec2bd20bda13e49190b174eb1Shelley Chen self.switcher.bypass_rec_mode() 44e14953ba021b2fcec2bd20bda13e49190b174eb1Shelley Chen self.switcher.wait_for_client() 45d74e1bbcb1b7ecabea3c9ea536d0123e71d33917Yusuf Mohsinally 46d74e1bbcb1b7ecabea3c9ea536d0123e71d33917Yusuf Mohsinally logging.info("Expected recovery boot and restores firmware A and B.") 47d74e1bbcb1b7ecabea3c9ea536d0123e71d33917Yusuf Mohsinally self.check_state((self.checkers.crossystem_checker, { 48d74e1bbcb1b7ecabea3c9ea536d0123e71d33917Yusuf Mohsinally 'mainfw_type': 'recovery', 49e75f460b26dcae8b0c9043f100a829a324e136ccJulius Werner 'recovery_reason': ( 50e75f460b26dcae8b0c9043f100a829a324e136ccJulius Werner vboot.RECOVERY_REASON['RO_INVALID_RW'], 51e75f460b26dcae8b0c9043f100a829a324e136ccJulius Werner vboot.RECOVERY_REASON['RW_FW_ROLLBACK']), 52d74e1bbcb1b7ecabea3c9ea536d0123e71d33917Yusuf Mohsinally })) 53d74e1bbcb1b7ecabea3c9ea536d0123e71d33917Yusuf Mohsinally self.faft_client.bios.move_version_forward(('a', 'b')) 544777624b69b3b223a5d3c3abe86000783f6c0e96Tom Wai-Hong Tam self.switcher.mode_aware_reboot() 55d74e1bbcb1b7ecabea3c9ea536d0123e71d33917Yusuf Mohsinally 562d02fe309a8620d383e34777c18377ace48faae4Daisuke Nojiri expected_slot = 'B' if self.fw_vboot2 else 'A' 572d02fe309a8620d383e34777c18377ace48faae4Daisuke Nojiri logging.info("Expected firmware " + expected_slot + " boot, done.") 583edea989126b84d38135371e780184694892a55aShelley Chen self.check_state((self.checkers.fw_tries_checker, expected_slot)) 59