diff options
-rw-r--r-- | test/tpm_test/subcmd.py | 2 | ||||
-rwxr-xr-x | test/tpm_test/tpmtest.py | 3 | ||||
-rw-r--r-- | test/tpm_test/upgrade_test.py | 68 |
3 files changed, 71 insertions, 2 deletions
diff --git a/test/tpm_test/subcmd.py b/test/tpm_test/subcmd.py index 916767222d..33a4ab21d7 100644 --- a/test/tpm_test/subcmd.py +++ b/test/tpm_test/subcmd.py @@ -10,7 +10,7 @@ AES = 0 HASH = 1 RSA = 2 EC = 3 - +FW_UPGRADE = 4 # The same exception class used by all tpmtest modules. class TpmTestError(Exception): diff --git a/test/tpm_test/tpmtest.py b/test/tpm_test/tpmtest.py index 46c03ef23b..4b92ea3ce9 100755 --- a/test/tpm_test/tpmtest.py +++ b/test/tpm_test/tpmtest.py @@ -24,7 +24,7 @@ import ftdi_spi_tpm import hash_test import rsa_test import subcmd - +import upgrade_test # Extension command for dcypto testing EXT_CMD = 0xbaccd00a @@ -137,6 +137,7 @@ if __name__ == '__main__': ecc_test.ecc_test(t) hash_test.hash_test(t) rsa_test.rsa_test(t) + upgrade_test.upgrade(t) except subcmd.TpmTestError as e: exc_file, exc_line = traceback.extract_tb(sys.exc_traceback)[-1][:2] print('\nError in %s:%s: ' % (os.path.basename(exc_file), exc_line), e) diff --git a/test/tpm_test/upgrade_test.py b/test/tpm_test/upgrade_test.py new file mode 100644 index 0000000000..db70b5ce76 --- /dev/null +++ b/test/tpm_test/upgrade_test.py @@ -0,0 +1,68 @@ +#!/usr/bin/python +# Copyright 2016 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. + +from __future__ import print_function + +import hashlib +import os +import struct + +import subcmd +import utils + + +def upgrade(tpm): + """Exercise the upgrade command. + + The target expect the upgrade extension command to have the following + structure: + + cmd 1 value of FW_UPGRADE + digest 4 first 4 bytes of sha1 of the remainder of the message + block_base 4 address of the block to write + data var + + Args: + tpm: a properly initialized tpmtest.TPM object + Raises: + subcmd.TpmTestError: In case of various test problems + """ + cmd = struct.pack('>I', 0) # address + cmd += struct.pack('>I', 0) # data (a noop) + wrapped_response = tpm.command(tpm.wrap_ext_command(subcmd.FW_UPGRADE, cmd)) + base_str = tpm.unwrap_ext_response(subcmd.FW_UPGRADE, wrapped_response) + if len(base_str) < 4: + raise subcmd.TpmTestError('Initialization error %d' % + ord(base_str[0])) + base = struct.unpack('>I', base_str)[0] + if base == 0x84000: + fname = 'build/cr50/RW/ec.RW_B.flat' + elif base == 0x44000: + fname = 'build/cr50/RW/ec.RW.flat' + else: + raise subcmd.TpmTestError('Unknown base address 0x%x' % base) + fname = os.path.join(os.path.dirname(__file__), '../..', fname) + data = open(fname, 'r').read() + transferred = 0 + block_size = 1024 + + while transferred < len(data): + tx_size = min(block_size, len(data) - transferred) + chunk = data[transferred:transferred+tx_size] + cmd = struct.pack('>I', base) # address + h = hashlib.sha1() + h.update(cmd) + h.update(chunk) + cmd = h.digest()[0:4] + cmd + chunk + resp = tpm.unwrap_ext_response(subcmd.FW_UPGRADE, + tpm.command(tpm.wrap_ext_command( + subcmd.FW_UPGRADE, cmd))) + code = ord(resp[0]) + if code: + raise subcmd.TpmTestError('%x - resp %d' % (base, code)) + base += tx_size + transferred += tx_size + + print('%sSUCCESS: Firmware upgrade' % (utils.cursor_back())) |