1# Copyright 2014 The Chromium OS Authors. All rights reserved. 2# Use of this source code is governed by a BSD-style license that can be 3# found in the LICENSE file. 4 5import logging 6 7from autotest_lib.server.cros.faft.firmware_test import FirmwareTest 8 9 10class firmware_TPMKernelVersion(FirmwareTest): 11 """Dev mode will not corrupt kernel and firmware version stored in the TPM. 12 13 Automate 1.2.4 Check Kernel version in TPM is not corrupted. 14 """ 15 version = 1 16 NEEDS_SERVO_USB = True 17 18 def initialize(self, host, cmdline_args): 19 """Initialize the test""" 20 super(firmware_TPMKernelVersion, self).initialize(host, cmdline_args) 21 self.switcher.setup_mode('dev') 22 # Use the USB key for Ctrl-U dev boot, not recovery. 23 self.setup_usbkey(usbkey=True, host=False, used_for_recovery=False) 24 self.original_dev_boot_usb = self.faft_client.system.get_dev_boot_usb() 25 logging.info('Original dev_boot_usb value: %s', 26 str(self.original_dev_boot_usb)) 27 28 29 def dut_run_cmd(self, command): 30 """Execute command on DUT. 31 32 @param command: shell command to be executed on DUT. 33 @returns command output. 34 """ 35 logging.info('Execute %s', command) 36 output = self.faft_client.system.run_shell_command_get_output(command) 37 logging.info('Output %s', output) 38 return output 39 40 def cleanup(self): 41 """Reboot device from SSD.""" 42 try: 43 self.ensure_dev_internal_boot(self.original_dev_boot_usb) 44 except Exception as e: 45 logging.error("Caught exception: %s", str(e)) 46 super(firmware_TPMKernelVersion, self).cleanup() 47 48 def run_once(self): 49 """Main test logic""" 50 # Get the kernel version and its datakey version. 51 # TODO(twreid): Verify the results. 52 version = self.faft_client.tpm.get_kernel_version() 53 logging.info('Kernel version: %d', version) 54 version = self.faft_client.tpm.get_kernel_datakey_version() 55 logging.info('Kernel datakey version: %d', version) 56 57 self.dut_run_cmd('crossystem kernkey_vfy dev_boot_usb') 58 # Boot CrOS from USB 59 self.faft_client.system.set_dev_boot_usb(1) 60 self.switcher.simple_reboot() 61 self.switcher.bypass_dev_boot_usb() 62 self.switcher.wait_for_client() 63 64 # Check that DUT is booted from USB. 65 self.check_state((self.checkers.dev_boot_usb_checker, (True, True), 66 'Device not booted from USB image properly.')) 67 68 out = self.dut_run_cmd('crossystem tpm_kernver tpm_fwver') 69 (kernver, fwver) = out[0].split(' ') 70 logging.info('tpm_kernver=%s tpm_fwver=%s', kernver, fwver) 71 assert kernver.lower() != '0xffffffff' 72 assert fwver.lower() != '0xffffffff' 73