xref: /openbmc/u-boot/cmd/tpm-v2.c (revision 04b24965)
1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3  * Copyright (c) 2018 Bootlin
4  * Author: Miquel Raynal <miquel.raynal@bootlin.com>
5  */
6 
7 #include <common.h>
8 #include <dm.h>
9 #include <log.h>
10 #include <mapmem.h>
11 #include <tpm-common.h>
12 #include <tpm-v2.h>
13 #include "tpm-user-utils.h"
14 
15 static int do_tpm2_startup(cmd_tbl_t *cmdtp, int flag, int argc,
16 			   char * const argv[])
17 {
18 	enum tpm2_startup_types mode;
19 	struct udevice *dev;
20 	int ret;
21 
22 	ret = get_tpm(&dev);
23 	if (ret)
24 		return ret;
25 	if (argc != 2)
26 		return CMD_RET_USAGE;
27 
28 	if (!strcasecmp("TPM2_SU_CLEAR", argv[1])) {
29 		mode = TPM2_SU_CLEAR;
30 	} else if (!strcasecmp("TPM2_SU_STATE", argv[1])) {
31 		mode = TPM2_SU_STATE;
32 	} else {
33 		printf("Couldn't recognize mode string: %s\n", argv[1]);
34 		return CMD_RET_FAILURE;
35 	}
36 
37 	return report_return_code(tpm2_startup(dev, mode));
38 }
39 
40 static int do_tpm2_self_test(cmd_tbl_t *cmdtp, int flag, int argc,
41 			     char * const argv[])
42 {
43 	enum tpm2_yes_no full_test;
44 	struct udevice *dev;
45 	int ret;
46 
47 	ret = get_tpm(&dev);
48 	if (ret)
49 		return ret;
50 	if (argc != 2)
51 		return CMD_RET_USAGE;
52 
53 	if (!strcasecmp("full", argv[1])) {
54 		full_test = TPMI_YES;
55 	} else if (!strcasecmp("continue", argv[1])) {
56 		full_test = TPMI_NO;
57 	} else {
58 		printf("Couldn't recognize test mode: %s\n", argv[1]);
59 		return CMD_RET_FAILURE;
60 	}
61 
62 	return report_return_code(tpm2_self_test(dev, full_test));
63 }
64 
65 static int do_tpm2_clear(cmd_tbl_t *cmdtp, int flag, int argc,
66 			 char * const argv[])
67 {
68 	u32 handle = 0;
69 	const char *pw = (argc < 3) ? NULL : argv[2];
70 	const ssize_t pw_sz = pw ? strlen(pw) : 0;
71 	struct udevice *dev;
72 	int ret;
73 
74 	ret = get_tpm(&dev);
75 	if (ret)
76 		return ret;
77 
78 	if (argc < 2 || argc > 3)
79 		return CMD_RET_USAGE;
80 
81 	if (pw_sz > TPM2_DIGEST_LEN)
82 		return -EINVAL;
83 
84 	if (!strcasecmp("TPM2_RH_LOCKOUT", argv[1]))
85 		handle = TPM2_RH_LOCKOUT;
86 	else if (!strcasecmp("TPM2_RH_PLATFORM", argv[1]))
87 		handle = TPM2_RH_PLATFORM;
88 	else
89 		return CMD_RET_USAGE;
90 
91 	return report_return_code(tpm2_clear(dev, handle, pw, pw_sz));
92 }
93 
94 static int do_tpm2_pcr_extend(cmd_tbl_t *cmdtp, int flag, int argc,
95 			      char * const argv[])
96 {
97 	struct udevice *dev;
98 	struct tpm_chip_priv *priv;
99 	u32 index = simple_strtoul(argv[1], NULL, 0);
100 	void *digest = map_sysmem(simple_strtoul(argv[2], NULL, 0), 0);
101 	int ret;
102 	u32 rc;
103 
104 	if (argc != 3)
105 		return CMD_RET_USAGE;
106 
107 	ret = get_tpm(&dev);
108 	if (ret)
109 		return ret;
110 
111 	priv = dev_get_uclass_priv(dev);
112 	if (!priv)
113 		return -EINVAL;
114 
115 	if (index >= priv->pcr_count)
116 		return -EINVAL;
117 
118 	rc = tpm2_pcr_extend(dev, index, digest);
119 
120 	unmap_sysmem(digest);
121 
122 	return report_return_code(rc);
123 }
124 
125 static int do_tpm_pcr_read(cmd_tbl_t *cmdtp, int flag, int argc,
126 			   char * const argv[])
127 {
128 	struct udevice *dev;
129 	struct tpm_chip_priv *priv;
130 	u32 index, rc;
131 	unsigned int updates;
132 	void *data;
133 	int ret;
134 
135 	if (argc != 3)
136 		return CMD_RET_USAGE;
137 
138 	ret = get_tpm(&dev);
139 	if (ret)
140 		return ret;
141 
142 	priv = dev_get_uclass_priv(dev);
143 	if (!priv)
144 		return -EINVAL;
145 
146 	index = simple_strtoul(argv[1], NULL, 0);
147 	if (index >= priv->pcr_count)
148 		return -EINVAL;
149 
150 	data = map_sysmem(simple_strtoul(argv[2], NULL, 0), 0);
151 
152 	rc = tpm2_pcr_read(dev, index, priv->pcr_select_min, data, &updates);
153 	if (!rc) {
154 		printf("PCR #%u content (%u known updates):\n", index, updates);
155 		print_byte_string(data, TPM2_DIGEST_LEN);
156 	}
157 
158 	unmap_sysmem(data);
159 
160 	return report_return_code(rc);
161 }
162 
163 static int do_tpm_get_capability(cmd_tbl_t *cmdtp, int flag, int argc,
164 				 char * const argv[])
165 {
166 	u32 capability, property, rc;
167 	u8 *data;
168 	size_t count;
169 	int i, j;
170 	struct udevice *dev;
171 	int ret;
172 
173 	ret = get_tpm(&dev);
174 	if (ret)
175 		return ret;
176 
177 	if (argc != 5)
178 		return CMD_RET_USAGE;
179 
180 	capability = simple_strtoul(argv[1], NULL, 0);
181 	property = simple_strtoul(argv[2], NULL, 0);
182 	data = map_sysmem(simple_strtoul(argv[3], NULL, 0), 0);
183 	count = simple_strtoul(argv[4], NULL, 0);
184 
185 	rc = tpm2_get_capability(dev, capability, property, data, count);
186 	if (rc)
187 		goto unmap_data;
188 
189 	printf("Capabilities read from TPM:\n");
190 	for (i = 0; i < count; i++) {
191 		printf("Property 0x");
192 		for (j = 0; j < 4; j++)
193 			printf("%02x", data[(i * 8) + j]);
194 		printf(": 0x");
195 		for (j = 4; j < 8; j++)
196 			printf("%02x", data[(i * 8) + j]);
197 		printf("\n");
198 	}
199 
200 unmap_data:
201 	unmap_sysmem(data);
202 
203 	return report_return_code(rc);
204 }
205 
206 static int do_tpm_dam_reset(cmd_tbl_t *cmdtp, int flag, int argc,
207 			    char *const argv[])
208 {
209 	const char *pw = (argc < 2) ? NULL : argv[1];
210 	const ssize_t pw_sz = pw ? strlen(pw) : 0;
211 	struct udevice *dev;
212 	int ret;
213 
214 	ret = get_tpm(&dev);
215 	if (ret)
216 		return ret;
217 
218 	if (argc > 2)
219 		return CMD_RET_USAGE;
220 
221 	if (pw_sz > TPM2_DIGEST_LEN)
222 		return -EINVAL;
223 
224 	return report_return_code(tpm2_dam_reset(dev, pw, pw_sz));
225 }
226 
227 static int do_tpm_dam_parameters(cmd_tbl_t *cmdtp, int flag, int argc,
228 				 char *const argv[])
229 {
230 	const char *pw = (argc < 5) ? NULL : argv[4];
231 	const ssize_t pw_sz = pw ? strlen(pw) : 0;
232 	/*
233 	 * No Dictionary Attack Mitigation (DAM) means:
234 	 * maxtries = 0xFFFFFFFF, recovery_time = 1, lockout_recovery = 0
235 	 */
236 	unsigned long int max_tries;
237 	unsigned long int recovery_time;
238 	unsigned long int lockout_recovery;
239 	struct udevice *dev;
240 	int ret;
241 
242 	ret = get_tpm(&dev);
243 	if (ret)
244 		return ret;
245 
246 	if (argc < 4 || argc > 5)
247 		return CMD_RET_USAGE;
248 
249 	if (pw_sz > TPM2_DIGEST_LEN)
250 		return -EINVAL;
251 
252 	if (strict_strtoul(argv[1], 0, &max_tries))
253 		return CMD_RET_USAGE;
254 
255 	if (strict_strtoul(argv[2], 0, &recovery_time))
256 		return CMD_RET_USAGE;
257 
258 	if (strict_strtoul(argv[3], 0, &lockout_recovery))
259 		return CMD_RET_USAGE;
260 
261 	log(LOGC_NONE, LOGL_INFO, "Changing dictionary attack parameters:\n");
262 	log(LOGC_NONE, LOGL_INFO, "- maxTries: %lu", max_tries);
263 	log(LOGC_NONE, LOGL_INFO, "- recoveryTime: %lu\n", recovery_time);
264 	log(LOGC_NONE, LOGL_INFO, "- lockoutRecovery: %lu\n", lockout_recovery);
265 
266 	return report_return_code(tpm2_dam_parameters(dev, pw, pw_sz, max_tries,
267 						      recovery_time,
268 						      lockout_recovery));
269 }
270 
271 static int do_tpm_change_auth(cmd_tbl_t *cmdtp, int flag, int argc,
272 			      char *const argv[])
273 {
274 	u32 handle;
275 	const char *newpw = argv[2];
276 	const char *oldpw = (argc == 3) ? NULL : argv[3];
277 	const ssize_t newpw_sz = strlen(newpw);
278 	const ssize_t oldpw_sz = oldpw ? strlen(oldpw) : 0;
279 	struct udevice *dev;
280 	int ret;
281 
282 	ret = get_tpm(&dev);
283 	if (ret)
284 		return ret;
285 
286 	if (argc < 3 || argc > 4)
287 		return CMD_RET_USAGE;
288 
289 	if (newpw_sz > TPM2_DIGEST_LEN || oldpw_sz > TPM2_DIGEST_LEN)
290 		return -EINVAL;
291 
292 	if (!strcasecmp("TPM2_RH_LOCKOUT", argv[1]))
293 		handle = TPM2_RH_LOCKOUT;
294 	else if (!strcasecmp("TPM2_RH_ENDORSEMENT", argv[1]))
295 		handle = TPM2_RH_ENDORSEMENT;
296 	else if (!strcasecmp("TPM2_RH_OWNER", argv[1]))
297 		handle = TPM2_RH_OWNER;
298 	else if (!strcasecmp("TPM2_RH_PLATFORM", argv[1]))
299 		handle = TPM2_RH_PLATFORM;
300 	else
301 		return CMD_RET_USAGE;
302 
303 	return report_return_code(tpm2_change_auth(dev, handle, newpw, newpw_sz,
304 						   oldpw, oldpw_sz));
305 }
306 
307 static int do_tpm_pcr_setauthpolicy(cmd_tbl_t *cmdtp, int flag, int argc,
308 				    char * const argv[])
309 {
310 	u32 index = simple_strtoul(argv[1], NULL, 0);
311 	char *key = argv[2];
312 	const char *pw = (argc < 4) ? NULL : argv[3];
313 	const ssize_t pw_sz = pw ? strlen(pw) : 0;
314 	struct udevice *dev;
315 	int ret;
316 
317 	ret = get_tpm(&dev);
318 	if (ret)
319 		return ret;
320 
321 	if (strlen(key) != TPM2_DIGEST_LEN)
322 		return -EINVAL;
323 
324 	if (argc < 3 || argc > 4)
325 		return CMD_RET_USAGE;
326 
327 	return report_return_code(tpm2_pcr_setauthpolicy(dev, pw, pw_sz, index,
328 							 key));
329 }
330 
331 static int do_tpm_pcr_setauthvalue(cmd_tbl_t *cmdtp, int flag,
332 				   int argc, char * const argv[])
333 {
334 	u32 index = simple_strtoul(argv[1], NULL, 0);
335 	char *key = argv[2];
336 	const ssize_t key_sz = strlen(key);
337 	const char *pw = (argc < 4) ? NULL : argv[3];
338 	const ssize_t pw_sz = pw ? strlen(pw) : 0;
339 	struct udevice *dev;
340 	int ret;
341 
342 	ret = get_tpm(&dev);
343 	if (ret)
344 		return ret;
345 
346 	if (strlen(key) != TPM2_DIGEST_LEN)
347 		return -EINVAL;
348 
349 	if (argc < 3 || argc > 4)
350 		return CMD_RET_USAGE;
351 
352 	return report_return_code(tpm2_pcr_setauthvalue(dev, pw, pw_sz, index,
353 							key, key_sz));
354 }
355 
356 static cmd_tbl_t tpm2_commands[] = {
357 	U_BOOT_CMD_MKENT(info, 0, 1, do_tpm_info, "", ""),
358 	U_BOOT_CMD_MKENT(init, 0, 1, do_tpm_init, "", ""),
359 	U_BOOT_CMD_MKENT(startup, 0, 1, do_tpm2_startup, "", ""),
360 	U_BOOT_CMD_MKENT(self_test, 0, 1, do_tpm2_self_test, "", ""),
361 	U_BOOT_CMD_MKENT(clear, 0, 1, do_tpm2_clear, "", ""),
362 	U_BOOT_CMD_MKENT(pcr_extend, 0, 1, do_tpm2_pcr_extend, "", ""),
363 	U_BOOT_CMD_MKENT(pcr_read, 0, 1, do_tpm_pcr_read, "", ""),
364 	U_BOOT_CMD_MKENT(get_capability, 0, 1, do_tpm_get_capability, "", ""),
365 	U_BOOT_CMD_MKENT(dam_reset, 0, 1, do_tpm_dam_reset, "", ""),
366 	U_BOOT_CMD_MKENT(dam_parameters, 0, 1, do_tpm_dam_parameters, "", ""),
367 	U_BOOT_CMD_MKENT(change_auth, 0, 1, do_tpm_change_auth, "", ""),
368 	U_BOOT_CMD_MKENT(pcr_setauthpolicy, 0, 1,
369 			 do_tpm_pcr_setauthpolicy, "", ""),
370 	U_BOOT_CMD_MKENT(pcr_setauthvalue, 0, 1,
371 			 do_tpm_pcr_setauthvalue, "", ""),
372 };
373 
374 cmd_tbl_t *get_tpm2_commands(unsigned int *size)
375 {
376 	*size = ARRAY_SIZE(tpm2_commands);
377 
378 	return tpm2_commands;
379 }
380 
381 U_BOOT_CMD(tpm2, CONFIG_SYS_MAXARGS, 1, do_tpm, "Issue a TPMv2.x command",
382 "<command> [<arguments>]\n"
383 "\n"
384 "info\n"
385 "    Show information about the TPM.\n"
386 "init\n"
387 "    Initialize the software stack. Always the first command to issue.\n"
388 "startup <mode>\n"
389 "    Issue a TPM2_Startup command.\n"
390 "    <mode> is one of:\n"
391 "        * TPM2_SU_CLEAR (reset state)\n"
392 "        * TPM2_SU_STATE (preserved state)\n"
393 "self_test <type>\n"
394 "    Test the TPM capabilities.\n"
395 "    <type> is one of:\n"
396 "        * full (perform all tests)\n"
397 "        * continue (only check untested tests)\n"
398 "clear <hierarchy>\n"
399 "    Issue a TPM2_Clear command.\n"
400 "    <hierarchy> is one of:\n"
401 "        * TPM2_RH_LOCKOUT\n"
402 "        * TPM2_RH_PLATFORM\n"
403 "pcr_extend <pcr> <digest_addr>\n"
404 "    Extend PCR #<pcr> with digest at <digest_addr>.\n"
405 "    <pcr>: index of the PCR\n"
406 "    <digest_addr>: address of a 32-byte SHA256 digest\n"
407 "pcr_read <pcr> <digest_addr>\n"
408 "    Read PCR #<pcr> to memory address <digest_addr>.\n"
409 "    <pcr>: index of the PCR\n"
410 "    <digest_addr>: address to store the a 32-byte SHA256 digest\n"
411 "get_capability <capability> <property> <addr> <count>\n"
412 "    Read and display <count> entries indexed by <capability>/<property>.\n"
413 "    Values are 4 bytes long and are written at <addr>.\n"
414 "    <capability>: capability\n"
415 "    <property>: property\n"
416 "    <addr>: address to store <count> entries of 4 bytes\n"
417 "    <count>: number of entries to retrieve\n"
418 "dam_reset [<password>]\n"
419 "    If the TPM is not in a LOCKOUT state, reset the internal error counter.\n"
420 "    <password>: optional password\n"
421 "dam_parameters <max_tries> <recovery_time> <lockout_recovery> [<password>]\n"
422 "    If the TPM is not in a LOCKOUT state, set the DAM parameters\n"
423 "    <maxTries>: maximum number of failures before lockout,\n"
424 "                0 means always locking\n"
425 "    <recoveryTime>: time before decrement of the error counter,\n"
426 "                    0 means no lockout\n"
427 "    <lockoutRecovery>: time of a lockout (before the next try),\n"
428 "                       0 means a reboot is needed\n"
429 "    <password>: optional password of the LOCKOUT hierarchy\n"
430 "change_auth <hierarchy> <new_pw> [<old_pw>]\n"
431 "    <hierarchy>: the hierarchy\n"
432 "    <new_pw>: new password for <hierarchy>\n"
433 "    <old_pw>: optional previous password of <hierarchy>\n"
434 "pcr_setauthpolicy|pcr_setauthvalue <pcr> <key> [<password>]\n"
435 "    Change the <key> to access PCR #<pcr>.\n"
436 "    hierarchy and may be empty.\n"
437 "    /!\\WARNING: untested function, use at your own risks !\n"
438 "    <pcr>: index of the PCR\n"
439 "    <key>: secret to protect the access of PCR #<pcr>\n"
440 "    <password>: optional password of the PLATFORM hierarchy\n"
441 );
442