]>
Commit | Line | Data |
---|---|---|
576fb1ed VB |
1 | /* |
2 | * Copyright (c) 2011 The Chromium OS Authors. | |
3 | * | |
4 | * See file CREDITS for list of people who contributed to this | |
5 | * project. | |
6 | * | |
7 | * This program is free software; you can redistribute it and/or | |
8 | * modify it under the terms of the GNU General Public License as | |
9 | * published by the Free Software Foundation; either version 2 of | |
10 | * the License, or (at your option) any later version. | |
11 | * | |
12 | * This program is distributed in the hope that it will be useful, | |
13 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
14 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
15 | * GNU General Public License for more details. | |
16 | * | |
17 | * You should have received a copy of the GNU General Public License | |
18 | * along with this program; if not, write to the Free Software | |
19 | * Foundation, Inc., 59 Temple Place, Suite 330, Boston, | |
20 | * MA 02111-1307 USA | |
21 | */ | |
22 | ||
23 | #include <common.h> | |
24 | #include <command.h> | |
25 | #include <tpm.h> | |
26 | ||
27 | #define MAX_TRANSACTION_SIZE 30 | |
28 | ||
29 | /* | |
30 | * tpm_write() expects a variable number of parameters: the internal address | |
31 | * followed by data to write, byte by byte. | |
32 | * | |
33 | * Returns 0 on success or -1 on errors (wrong arguments or TPM failure). | |
34 | */ | |
35 | static int tpm_process(int argc, char * const argv[], cmd_tbl_t *cmdtp) | |
36 | { | |
37 | u8 tpm_buffer[MAX_TRANSACTION_SIZE]; | |
38 | u32 write_size, read_size; | |
39 | char *p; | |
40 | int rv = -1; | |
41 | ||
42 | for (write_size = 0; write_size < argc; write_size++) { | |
43 | u32 datum = simple_strtoul(argv[write_size], &p, 0); | |
44 | if (*p || (datum > 0xff)) { | |
45 | printf("\n%s: bad data value\n\n", argv[write_size]); | |
46 | cmd_usage(cmdtp); | |
47 | return rv; | |
48 | } | |
49 | tpm_buffer[write_size] = (u8)datum; | |
50 | } | |
51 | ||
52 | read_size = sizeof(tpm_buffer); | |
53 | if (!tis_sendrecv(tpm_buffer, write_size, tpm_buffer, &read_size)) { | |
54 | int i; | |
55 | puts("Got TPM response:\n"); | |
56 | for (i = 0; i < read_size; i++) | |
57 | printf(" %2.2x", tpm_buffer[i]); | |
58 | puts("\n"); | |
59 | rv = 0; | |
60 | } else { | |
61 | puts("tpm command failed\n"); | |
62 | } | |
63 | return rv; | |
64 | } | |
65 | ||
eea3f4d3 LS |
66 | #define CHECK(exp) do { \ |
67 | int _rv = exp; \ | |
68 | if (_rv) { \ | |
69 | printf("CHECK: %s %d %x\n", #exp, __LINE__, _rv);\ | |
70 | } \ | |
71 | } while (0) | |
72 | ||
73 | static int tpm_process_stress(int repeat_count) | |
576fb1ed | 74 | { |
eea3f4d3 | 75 | int i; |
576fb1ed | 76 | int rv = 0; |
eea3f4d3 LS |
77 | u8 request[] = {0x0, 0xc1, |
78 | 0x0, 0x0, 0x0, 0x16, | |
79 | 0x0, 0x0, 0x0, 0x65, | |
80 | 0x0, 0x0, 0x0, 0x4, | |
81 | 0x0, 0x0, 0x0, 0x4, | |
82 | 0x0, 0x0, 0x1, 0x9}; | |
83 | u8 response[MAX_TRANSACTION_SIZE]; | |
84 | u32 rlength = MAX_TRANSACTION_SIZE; | |
85 | ||
86 | CHECK(tis_init()); | |
87 | ||
88 | for (i = 0; i < repeat_count; i++) { | |
89 | CHECK(tis_open()); | |
90 | rv = tis_sendrecv(request, sizeof(request), response, &rlength); | |
91 | if (rv) { | |
92 | printf("tpm test failed at step %d with 0x%x\n", i, rv); | |
93 | CHECK(tis_close()); | |
94 | break; | |
95 | } | |
96 | CHECK(tis_close()); | |
97 | if ((response[6] || response[7] || response[8] || response[9]) | |
98 | && response[9] != 0x26) { | |
99 | /* Ignore postinit errors */ | |
100 | printf("tpm command failed at step %d\n" | |
101 | "tpm error code: %02x%02x%02x%02x\n", i, | |
102 | response[6], response[7], | |
103 | response[8], response[9]); | |
104 | rv = -1; | |
105 | break; | |
106 | } | |
107 | } | |
108 | return rv; | |
109 | } | |
576fb1ed | 110 | |
eea3f4d3 LS |
111 | |
112 | static int do_tpm_many(cmd_tbl_t *cmdtp, int flag, | |
113 | int argc, char * const argv[], int repeat_count) | |
114 | ||
115 | { | |
116 | int rv = 0; | |
117 | ||
118 | if (argc < 7 && repeat_count == 0) { | |
576fb1ed VB |
119 | puts("command should be at least six bytes in size\n"); |
120 | return -1; | |
121 | } | |
122 | ||
eea3f4d3 LS |
123 | if (repeat_count > 0) { |
124 | rv = tpm_process_stress(repeat_count); | |
125 | return rv; | |
126 | } | |
127 | ||
576fb1ed VB |
128 | if (tis_init()) { |
129 | puts("tis_init() failed!\n"); | |
130 | return -1; | |
131 | } | |
132 | ||
133 | if (tis_open()) { | |
134 | puts("tis_open() failed!\n"); | |
135 | return -1; | |
136 | } | |
137 | ||
138 | rv = tpm_process(argc - 1, argv + 1, cmdtp); | |
139 | ||
140 | if (tis_close()) { | |
141 | puts("tis_close() failed!\n"); | |
142 | rv = -1; | |
143 | } | |
144 | ||
145 | return rv; | |
146 | } | |
147 | ||
eea3f4d3 LS |
148 | |
149 | static int do_tpm(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[]) | |
150 | { | |
151 | return do_tpm_many(cmdtp, flag, argc, argv, 0); | |
152 | } | |
153 | ||
154 | ||
576fb1ed VB |
155 | U_BOOT_CMD(tpm, MAX_TRANSACTION_SIZE, 1, do_tpm, |
156 | "<byte> [<byte> ...] - write data and read response", | |
157 | "send arbitrary data (at least 6 bytes) to the TPM " | |
158 | "device and read the response" | |
159 | ); | |
eea3f4d3 LS |
160 | |
161 | static int do_tpm_stress(cmd_tbl_t *cmdtp, int flag, | |
162 | int argc, char * const argv[]) | |
163 | { | |
164 | long unsigned int n; | |
165 | int rv; | |
166 | ||
167 | if (argc != 2) { | |
168 | puts("usage: tpm_stress <count>\n"); | |
169 | return -1; | |
170 | } | |
171 | ||
172 | rv = strict_strtoul(argv[1], 10, &n); | |
173 | if (rv) { | |
174 | puts("tpm_stress: bad count"); | |
175 | return -1; | |
176 | } | |
177 | ||
178 | return do_tpm_many(cmdtp, flag, argc, argv, n); | |
179 | } | |
180 | ||
181 | U_BOOT_CMD(tpm_stress, 2, 1, do_tpm_stress, | |
182 | "<n> - stress-test communication with TPM", | |
183 | "Repeat a TPM transaction (request-response) N times" | |
184 | ); |