mirror of
https://github.com/brain-hackers/u-boot-brain
synced 2024-08-22 01:48:52 +09:00
![Luigi Semenzato](/assets/img/avatar_default.png)
Add a simple command to stress-test a TPM (Trusted Platform Module). Signed-off-by: Luigi Semenzato <semenzato@chromium.org> Signed-off-by: Simon Glass <sjg@chromium.org>
185 lines
4.3 KiB
C
185 lines
4.3 KiB
C
/*
|
|
* Copyright (c) 2011 The Chromium OS Authors.
|
|
*
|
|
* See file CREDITS for list of people who contributed to this
|
|
* project.
|
|
*
|
|
* This program is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU General Public License as
|
|
* published by the Free Software Foundation; either version 2 of
|
|
* the License, or (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, write to the Free Software
|
|
* Foundation, Inc., 59 Temple Place, Suite 330, Boston,
|
|
* MA 02111-1307 USA
|
|
*/
|
|
|
|
#include <common.h>
|
|
#include <command.h>
|
|
#include <tpm.h>
|
|
|
|
#define MAX_TRANSACTION_SIZE 30
|
|
|
|
/*
|
|
* tpm_write() expects a variable number of parameters: the internal address
|
|
* followed by data to write, byte by byte.
|
|
*
|
|
* Returns 0 on success or -1 on errors (wrong arguments or TPM failure).
|
|
*/
|
|
static int tpm_process(int argc, char * const argv[], cmd_tbl_t *cmdtp)
|
|
{
|
|
u8 tpm_buffer[MAX_TRANSACTION_SIZE];
|
|
u32 write_size, read_size;
|
|
char *p;
|
|
int rv = -1;
|
|
|
|
for (write_size = 0; write_size < argc; write_size++) {
|
|
u32 datum = simple_strtoul(argv[write_size], &p, 0);
|
|
if (*p || (datum > 0xff)) {
|
|
printf("\n%s: bad data value\n\n", argv[write_size]);
|
|
cmd_usage(cmdtp);
|
|
return rv;
|
|
}
|
|
tpm_buffer[write_size] = (u8)datum;
|
|
}
|
|
|
|
read_size = sizeof(tpm_buffer);
|
|
if (!tis_sendrecv(tpm_buffer, write_size, tpm_buffer, &read_size)) {
|
|
int i;
|
|
puts("Got TPM response:\n");
|
|
for (i = 0; i < read_size; i++)
|
|
printf(" %2.2x", tpm_buffer[i]);
|
|
puts("\n");
|
|
rv = 0;
|
|
} else {
|
|
puts("tpm command failed\n");
|
|
}
|
|
return rv;
|
|
}
|
|
|
|
#define CHECK(exp) do { \
|
|
int _rv = exp; \
|
|
if (_rv) { \
|
|
printf("CHECK: %s %d %x\n", #exp, __LINE__, _rv);\
|
|
} \
|
|
} while (0)
|
|
|
|
static int tpm_process_stress(int repeat_count)
|
|
{
|
|
int i;
|
|
int rv = 0;
|
|
u8 request[] = {0x0, 0xc1,
|
|
0x0, 0x0, 0x0, 0x16,
|
|
0x0, 0x0, 0x0, 0x65,
|
|
0x0, 0x0, 0x0, 0x4,
|
|
0x0, 0x0, 0x0, 0x4,
|
|
0x0, 0x0, 0x1, 0x9};
|
|
u8 response[MAX_TRANSACTION_SIZE];
|
|
u32 rlength = MAX_TRANSACTION_SIZE;
|
|
|
|
CHECK(tis_init());
|
|
|
|
for (i = 0; i < repeat_count; i++) {
|
|
CHECK(tis_open());
|
|
rv = tis_sendrecv(request, sizeof(request), response, &rlength);
|
|
if (rv) {
|
|
printf("tpm test failed at step %d with 0x%x\n", i, rv);
|
|
CHECK(tis_close());
|
|
break;
|
|
}
|
|
CHECK(tis_close());
|
|
if ((response[6] || response[7] || response[8] || response[9])
|
|
&& response[9] != 0x26) {
|
|
/* Ignore postinit errors */
|
|
printf("tpm command failed at step %d\n"
|
|
"tpm error code: %02x%02x%02x%02x\n", i,
|
|
response[6], response[7],
|
|
response[8], response[9]);
|
|
rv = -1;
|
|
break;
|
|
}
|
|
}
|
|
return rv;
|
|
}
|
|
|
|
|
|
static int do_tpm_many(cmd_tbl_t *cmdtp, int flag,
|
|
int argc, char * const argv[], int repeat_count)
|
|
|
|
{
|
|
int rv = 0;
|
|
|
|
if (argc < 7 && repeat_count == 0) {
|
|
puts("command should be at least six bytes in size\n");
|
|
return -1;
|
|
}
|
|
|
|
if (repeat_count > 0) {
|
|
rv = tpm_process_stress(repeat_count);
|
|
return rv;
|
|
}
|
|
|
|
if (tis_init()) {
|
|
puts("tis_init() failed!\n");
|
|
return -1;
|
|
}
|
|
|
|
if (tis_open()) {
|
|
puts("tis_open() failed!\n");
|
|
return -1;
|
|
}
|
|
|
|
rv = tpm_process(argc - 1, argv + 1, cmdtp);
|
|
|
|
if (tis_close()) {
|
|
puts("tis_close() failed!\n");
|
|
rv = -1;
|
|
}
|
|
|
|
return rv;
|
|
}
|
|
|
|
|
|
static int do_tpm(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
|
|
{
|
|
return do_tpm_many(cmdtp, flag, argc, argv, 0);
|
|
}
|
|
|
|
|
|
U_BOOT_CMD(tpm, MAX_TRANSACTION_SIZE, 1, do_tpm,
|
|
"<byte> [<byte> ...] - write data and read response",
|
|
"send arbitrary data (at least 6 bytes) to the TPM "
|
|
"device and read the response"
|
|
);
|
|
|
|
static int do_tpm_stress(cmd_tbl_t *cmdtp, int flag,
|
|
int argc, char * const argv[])
|
|
{
|
|
long unsigned int n;
|
|
int rv;
|
|
|
|
if (argc != 2) {
|
|
puts("usage: tpm_stress <count>\n");
|
|
return -1;
|
|
}
|
|
|
|
rv = strict_strtoul(argv[1], 10, &n);
|
|
if (rv) {
|
|
puts("tpm_stress: bad count");
|
|
return -1;
|
|
}
|
|
|
|
return do_tpm_many(cmdtp, flag, argc, argv, n);
|
|
}
|
|
|
|
U_BOOT_CMD(tpm_stress, 2, 1, do_tpm_stress,
|
|
"<n> - stress-test communication with TPM",
|
|
"Repeat a TPM transaction (request-response) N times"
|
|
);
|