xref: /linux/drivers/char/tpm/tpm2-cmd.c (revision a3a02a52bcfcbcc4a637d4b68bf1bc391c9fad02)
1 // SPDX-License-Identifier: GPL-2.0-only
2 /*
3  * Copyright (C) 2014, 2015 Intel Corporation
4  *
5  * Authors:
6  * Jarkko Sakkinen <jarkko.sakkinen@linux.intel.com>
7  *
8  * Maintained by: <tpmdd-devel@lists.sourceforge.net>
9  *
10  * This file contains TPM2 protocol implementations of the commands
11  * used by the kernel internally.
12  */
13 
14 #include "tpm.h"
15 #include <crypto/hash_info.h>
16 
17 static struct tpm2_hash tpm2_hash_map[] = {
18 	{HASH_ALGO_SHA1, TPM_ALG_SHA1},
19 	{HASH_ALGO_SHA256, TPM_ALG_SHA256},
20 	{HASH_ALGO_SHA384, TPM_ALG_SHA384},
21 	{HASH_ALGO_SHA512, TPM_ALG_SHA512},
22 	{HASH_ALGO_SM3_256, TPM_ALG_SM3_256},
23 };
24 
25 int tpm2_get_timeouts(struct tpm_chip *chip)
26 {
27 	/* Fixed timeouts for TPM2 */
28 	chip->timeout_a = msecs_to_jiffies(TPM2_TIMEOUT_A);
29 	chip->timeout_b = msecs_to_jiffies(TPM2_TIMEOUT_B);
30 	chip->timeout_c = msecs_to_jiffies(TPM2_TIMEOUT_C);
31 	chip->timeout_d = msecs_to_jiffies(TPM2_TIMEOUT_D);
32 
33 	/* PTP spec timeouts */
34 	chip->duration[TPM_SHORT] = msecs_to_jiffies(TPM2_DURATION_SHORT);
35 	chip->duration[TPM_MEDIUM] = msecs_to_jiffies(TPM2_DURATION_MEDIUM);
36 	chip->duration[TPM_LONG] = msecs_to_jiffies(TPM2_DURATION_LONG);
37 
38 	/* Key creation commands long timeouts */
39 	chip->duration[TPM_LONG_LONG] =
40 		msecs_to_jiffies(TPM2_DURATION_LONG_LONG);
41 
42 	chip->flags |= TPM_CHIP_FLAG_HAVE_TIMEOUTS;
43 
44 	return 0;
45 }
46 
47 /**
48  * tpm2_ordinal_duration_index() - returns an index to the chip duration table
49  * @ordinal: TPM command ordinal.
50  *
51  * The function returns an index to the chip duration table
52  * (enum tpm_duration), that describes the maximum amount of
53  * time the chip could take to return the result for a  particular ordinal.
54  *
55  * The values of the MEDIUM, and LONG durations are taken
56  * from the PC Client Profile (PTP) specification (750, 2000 msec)
57  *
58  * LONG_LONG is for commands that generates keys which empirically takes
59  * a longer time on some systems.
60  *
61  * Return:
62  * * TPM_MEDIUM
63  * * TPM_LONG
64  * * TPM_LONG_LONG
65  * * TPM_UNDEFINED
66  */
67 static u8 tpm2_ordinal_duration_index(u32 ordinal)
68 {
69 	switch (ordinal) {
70 	/* Startup */
71 	case TPM2_CC_STARTUP:                 /* 144 */
72 		return TPM_MEDIUM;
73 
74 	case TPM2_CC_SELF_TEST:               /* 143 */
75 		return TPM_LONG;
76 
77 	case TPM2_CC_GET_RANDOM:              /* 17B */
78 		return TPM_LONG;
79 
80 	case TPM2_CC_SEQUENCE_UPDATE:         /* 15C */
81 		return TPM_MEDIUM;
82 	case TPM2_CC_SEQUENCE_COMPLETE:       /* 13E */
83 		return TPM_MEDIUM;
84 	case TPM2_CC_EVENT_SEQUENCE_COMPLETE: /* 185 */
85 		return TPM_MEDIUM;
86 	case TPM2_CC_HASH_SEQUENCE_START:     /* 186 */
87 		return TPM_MEDIUM;
88 
89 	case TPM2_CC_VERIFY_SIGNATURE:        /* 177 */
90 		return TPM_LONG_LONG;
91 
92 	case TPM2_CC_PCR_EXTEND:              /* 182 */
93 		return TPM_MEDIUM;
94 
95 	case TPM2_CC_HIERARCHY_CONTROL:       /* 121 */
96 		return TPM_LONG;
97 	case TPM2_CC_HIERARCHY_CHANGE_AUTH:   /* 129 */
98 		return TPM_LONG;
99 
100 	case TPM2_CC_GET_CAPABILITY:          /* 17A */
101 		return TPM_MEDIUM;
102 
103 	case TPM2_CC_NV_READ:                 /* 14E */
104 		return TPM_LONG;
105 
106 	case TPM2_CC_CREATE_PRIMARY:          /* 131 */
107 		return TPM_LONG_LONG;
108 	case TPM2_CC_CREATE:                  /* 153 */
109 		return TPM_LONG_LONG;
110 	case TPM2_CC_CREATE_LOADED:           /* 191 */
111 		return TPM_LONG_LONG;
112 
113 	default:
114 		return TPM_UNDEFINED;
115 	}
116 }
117 
118 /**
119  * tpm2_calc_ordinal_duration() - calculate the maximum command duration
120  * @chip:    TPM chip to use.
121  * @ordinal: TPM command ordinal.
122  *
123  * The function returns the maximum amount of time the chip could take
124  * to return the result for a particular ordinal in jiffies.
125  *
126  * Return: A maximal duration time for an ordinal in jiffies.
127  */
128 unsigned long tpm2_calc_ordinal_duration(struct tpm_chip *chip, u32 ordinal)
129 {
130 	unsigned int index;
131 
132 	index = tpm2_ordinal_duration_index(ordinal);
133 
134 	if (index != TPM_UNDEFINED)
135 		return chip->duration[index];
136 	else
137 		return msecs_to_jiffies(TPM2_DURATION_DEFAULT);
138 }
139 
140 
141 struct tpm2_pcr_read_out {
142 	__be32	update_cnt;
143 	__be32	pcr_selects_cnt;
144 	__be16	hash_alg;
145 	u8	pcr_select_size;
146 	u8	pcr_select[TPM2_PCR_SELECT_MIN];
147 	__be32	digests_cnt;
148 	__be16	digest_size;
149 	u8	digest[];
150 } __packed;
151 
152 /**
153  * tpm2_pcr_read() - read a PCR value
154  * @chip:	TPM chip to use.
155  * @pcr_idx:	index of the PCR to read.
156  * @digest:	PCR bank and buffer current PCR value is written to.
157  * @digest_size_ptr:	pointer to variable that stores the digest size.
158  *
159  * Return: Same as with tpm_transmit_cmd.
160  */
161 int tpm2_pcr_read(struct tpm_chip *chip, u32 pcr_idx,
162 		  struct tpm_digest *digest, u16 *digest_size_ptr)
163 {
164 	int i;
165 	int rc;
166 	struct tpm_buf buf;
167 	struct tpm2_pcr_read_out *out;
168 	u8 pcr_select[TPM2_PCR_SELECT_MIN] = {0};
169 	u16 digest_size;
170 	u16 expected_digest_size = 0;
171 
172 	if (pcr_idx >= TPM2_PLATFORM_PCR)
173 		return -EINVAL;
174 
175 	if (!digest_size_ptr) {
176 		for (i = 0; i < chip->nr_allocated_banks &&
177 		     chip->allocated_banks[i].alg_id != digest->alg_id; i++)
178 			;
179 
180 		if (i == chip->nr_allocated_banks)
181 			return -EINVAL;
182 
183 		expected_digest_size = chip->allocated_banks[i].digest_size;
184 	}
185 
186 	rc = tpm_buf_init(&buf, TPM2_ST_NO_SESSIONS, TPM2_CC_PCR_READ);
187 	if (rc)
188 		return rc;
189 
190 	pcr_select[pcr_idx >> 3] = 1 << (pcr_idx & 0x7);
191 
192 	tpm_buf_append_u32(&buf, 1);
193 	tpm_buf_append_u16(&buf, digest->alg_id);
194 	tpm_buf_append_u8(&buf, TPM2_PCR_SELECT_MIN);
195 	tpm_buf_append(&buf, (const unsigned char *)pcr_select,
196 		       sizeof(pcr_select));
197 
198 	rc = tpm_transmit_cmd(chip, &buf, 0, "attempting to read a pcr value");
199 	if (rc)
200 		goto out;
201 
202 	out = (struct tpm2_pcr_read_out *)&buf.data[TPM_HEADER_SIZE];
203 	digest_size = be16_to_cpu(out->digest_size);
204 	if (digest_size > sizeof(digest->digest) ||
205 	    (!digest_size_ptr && digest_size != expected_digest_size)) {
206 		rc = -EINVAL;
207 		goto out;
208 	}
209 
210 	if (digest_size_ptr)
211 		*digest_size_ptr = digest_size;
212 
213 	memcpy(digest->digest, out->digest, digest_size);
214 out:
215 	tpm_buf_destroy(&buf);
216 	return rc;
217 }
218 
219 /**
220  * tpm2_pcr_extend() - extend a PCR value
221  *
222  * @chip:	TPM chip to use.
223  * @pcr_idx:	index of the PCR.
224  * @digests:	list of pcr banks and corresponding digest values to extend.
225  *
226  * Return: Same as with tpm_transmit_cmd.
227  */
228 int tpm2_pcr_extend(struct tpm_chip *chip, u32 pcr_idx,
229 		    struct tpm_digest *digests)
230 {
231 	struct tpm_buf buf;
232 	int rc;
233 	int i;
234 
235 	rc = tpm2_start_auth_session(chip);
236 	if (rc)
237 		return rc;
238 
239 	rc = tpm_buf_init(&buf, TPM2_ST_SESSIONS, TPM2_CC_PCR_EXTEND);
240 	if (rc) {
241 		tpm2_end_auth_session(chip);
242 		return rc;
243 	}
244 
245 	tpm_buf_append_name(chip, &buf, pcr_idx, NULL);
246 	tpm_buf_append_hmac_session(chip, &buf, 0, NULL, 0);
247 
248 	tpm_buf_append_u32(&buf, chip->nr_allocated_banks);
249 
250 	for (i = 0; i < chip->nr_allocated_banks; i++) {
251 		tpm_buf_append_u16(&buf, digests[i].alg_id);
252 		tpm_buf_append(&buf, (const unsigned char *)&digests[i].digest,
253 			       chip->allocated_banks[i].digest_size);
254 	}
255 
256 	tpm_buf_fill_hmac_session(chip, &buf);
257 	rc = tpm_transmit_cmd(chip, &buf, 0, "attempting extend a PCR value");
258 	rc = tpm_buf_check_hmac_response(chip, &buf, rc);
259 
260 	tpm_buf_destroy(&buf);
261 
262 	return rc;
263 }
264 
265 struct tpm2_get_random_out {
266 	__be16 size;
267 	u8 buffer[TPM_MAX_RNG_DATA];
268 } __packed;
269 
270 /**
271  * tpm2_get_random() - get random bytes from the TPM RNG
272  *
273  * @chip:	a &tpm_chip instance
274  * @dest:	destination buffer
275  * @max:	the max number of random bytes to pull
276  *
277  * Return:
278  *   size of the buffer on success,
279  *   -errno otherwise (positive TPM return codes are masked to -EIO)
280  */
281 int tpm2_get_random(struct tpm_chip *chip, u8 *dest, size_t max)
282 {
283 	struct tpm2_get_random_out *out;
284 	struct tpm_header *head;
285 	struct tpm_buf buf;
286 	u32 recd;
287 	u32 num_bytes = max;
288 	int err;
289 	int total = 0;
290 	int retries = 5;
291 	u8 *dest_ptr = dest;
292 	off_t offset;
293 
294 	if (!num_bytes || max > TPM_MAX_RNG_DATA)
295 		return -EINVAL;
296 
297 	err = tpm2_start_auth_session(chip);
298 	if (err)
299 		return err;
300 
301 	err = tpm_buf_init(&buf, 0, 0);
302 	if (err) {
303 		tpm2_end_auth_session(chip);
304 		return err;
305 	}
306 
307 	do {
308 		tpm_buf_reset(&buf, TPM2_ST_SESSIONS, TPM2_CC_GET_RANDOM);
309 		tpm_buf_append_hmac_session_opt(chip, &buf, TPM2_SA_ENCRYPT
310 						| TPM2_SA_CONTINUE_SESSION,
311 						NULL, 0);
312 		tpm_buf_append_u16(&buf, num_bytes);
313 		tpm_buf_fill_hmac_session(chip, &buf);
314 		err = tpm_transmit_cmd(chip, &buf,
315 				       offsetof(struct tpm2_get_random_out,
316 						buffer),
317 				       "attempting get random");
318 		err = tpm_buf_check_hmac_response(chip, &buf, err);
319 		if (err) {
320 			if (err > 0)
321 				err = -EIO;
322 			goto out;
323 		}
324 
325 		head = (struct tpm_header *)buf.data;
326 		offset = TPM_HEADER_SIZE;
327 		/* Skip the parameter size field: */
328 		if (be16_to_cpu(head->tag) == TPM2_ST_SESSIONS)
329 			offset += 4;
330 
331 		out = (struct tpm2_get_random_out *)&buf.data[offset];
332 		recd = min_t(u32, be16_to_cpu(out->size), num_bytes);
333 		if (tpm_buf_length(&buf) <
334 		    TPM_HEADER_SIZE +
335 		    offsetof(struct tpm2_get_random_out, buffer) +
336 		    recd) {
337 			err = -EFAULT;
338 			goto out;
339 		}
340 		memcpy(dest_ptr, out->buffer, recd);
341 
342 		dest_ptr += recd;
343 		total += recd;
344 		num_bytes -= recd;
345 	} while (retries-- && total < max);
346 
347 	tpm_buf_destroy(&buf);
348 	tpm2_end_auth_session(chip);
349 
350 	return total ? total : -EIO;
351 out:
352 	tpm_buf_destroy(&buf);
353 	tpm2_end_auth_session(chip);
354 	return err;
355 }
356 
357 /**
358  * tpm2_flush_context() - execute a TPM2_FlushContext command
359  * @chip:	TPM chip to use
360  * @handle:	context handle
361  */
362 void tpm2_flush_context(struct tpm_chip *chip, u32 handle)
363 {
364 	struct tpm_buf buf;
365 	int rc;
366 
367 	rc = tpm_buf_init(&buf, TPM2_ST_NO_SESSIONS, TPM2_CC_FLUSH_CONTEXT);
368 	if (rc) {
369 		dev_warn(&chip->dev, "0x%08x was not flushed, out of memory\n",
370 			 handle);
371 		return;
372 	}
373 
374 	tpm_buf_append_u32(&buf, handle);
375 
376 	tpm_transmit_cmd(chip, &buf, 0, "flushing context");
377 	tpm_buf_destroy(&buf);
378 }
379 EXPORT_SYMBOL_GPL(tpm2_flush_context);
380 
381 struct tpm2_get_cap_out {
382 	u8 more_data;
383 	__be32 subcap_id;
384 	__be32 property_cnt;
385 	__be32 property_id;
386 	__be32 value;
387 } __packed;
388 
389 /**
390  * tpm2_get_tpm_pt() - get value of a TPM_CAP_TPM_PROPERTIES type property
391  * @chip:		a &tpm_chip instance
392  * @property_id:	property ID.
393  * @value:		output variable.
394  * @desc:		passed to tpm_transmit_cmd()
395  *
396  * Return:
397  *   0 on success,
398  *   -errno or a TPM return code otherwise
399  */
400 ssize_t tpm2_get_tpm_pt(struct tpm_chip *chip, u32 property_id,  u32 *value,
401 			const char *desc)
402 {
403 	struct tpm2_get_cap_out *out;
404 	struct tpm_buf buf;
405 	int rc;
406 
407 	rc = tpm_buf_init(&buf, TPM2_ST_NO_SESSIONS, TPM2_CC_GET_CAPABILITY);
408 	if (rc)
409 		return rc;
410 	tpm_buf_append_u32(&buf, TPM2_CAP_TPM_PROPERTIES);
411 	tpm_buf_append_u32(&buf, property_id);
412 	tpm_buf_append_u32(&buf, 1);
413 	rc = tpm_transmit_cmd(chip, &buf, 0, NULL);
414 	if (!rc) {
415 		out = (struct tpm2_get_cap_out *)
416 			&buf.data[TPM_HEADER_SIZE];
417 		/*
418 		 * To prevent failing boot up of some systems, Infineon TPM2.0
419 		 * returns SUCCESS on TPM2_Startup in field upgrade mode. Also
420 		 * the TPM2_Getcapability command returns a zero length list
421 		 * in field upgrade mode.
422 		 */
423 		if (be32_to_cpu(out->property_cnt) > 0)
424 			*value = be32_to_cpu(out->value);
425 		else
426 			rc = -ENODATA;
427 	}
428 	tpm_buf_destroy(&buf);
429 	return rc;
430 }
431 EXPORT_SYMBOL_GPL(tpm2_get_tpm_pt);
432 
433 /**
434  * tpm2_shutdown() - send a TPM shutdown command
435  *
436  * Sends a TPM shutdown command. The shutdown command is used in call
437  * sites where the system is going down. If it fails, there is not much
438  * that can be done except print an error message.
439  *
440  * @chip:		a &tpm_chip instance
441  * @shutdown_type:	TPM_SU_CLEAR or TPM_SU_STATE.
442  */
443 void tpm2_shutdown(struct tpm_chip *chip, u16 shutdown_type)
444 {
445 	struct tpm_buf buf;
446 	int rc;
447 
448 	rc = tpm_buf_init(&buf, TPM2_ST_NO_SESSIONS, TPM2_CC_SHUTDOWN);
449 	if (rc)
450 		return;
451 	tpm_buf_append_u16(&buf, shutdown_type);
452 	tpm_transmit_cmd(chip, &buf, 0, "stopping the TPM");
453 	tpm_buf_destroy(&buf);
454 }
455 
456 /**
457  * tpm2_do_selftest() - ensure that all self tests have passed
458  *
459  * @chip: TPM chip to use
460  *
461  * Return: Same as with tpm_transmit_cmd.
462  *
463  * The TPM can either run all self tests synchronously and then return
464  * RC_SUCCESS once all tests were successful. Or it can choose to run the tests
465  * asynchronously and return RC_TESTING immediately while the self tests still
466  * execute in the background. This function handles both cases and waits until
467  * all tests have completed.
468  */
469 static int tpm2_do_selftest(struct tpm_chip *chip)
470 {
471 	struct tpm_buf buf;
472 	int full;
473 	int rc;
474 
475 	for (full = 0; full < 2; full++) {
476 		rc = tpm_buf_init(&buf, TPM2_ST_NO_SESSIONS, TPM2_CC_SELF_TEST);
477 		if (rc)
478 			return rc;
479 
480 		tpm_buf_append_u8(&buf, full);
481 		rc = tpm_transmit_cmd(chip, &buf, 0,
482 				      "attempting the self test");
483 		tpm_buf_destroy(&buf);
484 
485 		if (rc == TPM2_RC_TESTING)
486 			rc = TPM2_RC_SUCCESS;
487 		if (rc == TPM2_RC_INITIALIZE || rc == TPM2_RC_SUCCESS)
488 			return rc;
489 	}
490 
491 	return rc;
492 }
493 
494 /**
495  * tpm2_probe() - probe for the TPM 2.0 protocol
496  * @chip:	a &tpm_chip instance
497  *
498  * Send an idempotent TPM 2.0 command and see whether there is TPM2 chip in the
499  * other end based on the response tag. The flag TPM_CHIP_FLAG_TPM2 is set by
500  * this function if this is the case.
501  *
502  * Return:
503  *   0 on success,
504  *   -errno otherwise
505  */
506 int tpm2_probe(struct tpm_chip *chip)
507 {
508 	struct tpm_header *out;
509 	struct tpm_buf buf;
510 	int rc;
511 
512 	rc = tpm_buf_init(&buf, TPM2_ST_NO_SESSIONS, TPM2_CC_GET_CAPABILITY);
513 	if (rc)
514 		return rc;
515 	tpm_buf_append_u32(&buf, TPM2_CAP_TPM_PROPERTIES);
516 	tpm_buf_append_u32(&buf, TPM_PT_TOTAL_COMMANDS);
517 	tpm_buf_append_u32(&buf, 1);
518 	rc = tpm_transmit_cmd(chip, &buf, 0, NULL);
519 	/* We ignore TPM return codes on purpose. */
520 	if (rc >=  0) {
521 		out = (struct tpm_header *)buf.data;
522 		if (be16_to_cpu(out->tag) == TPM2_ST_NO_SESSIONS)
523 			chip->flags |= TPM_CHIP_FLAG_TPM2;
524 	}
525 	tpm_buf_destroy(&buf);
526 	return 0;
527 }
528 EXPORT_SYMBOL_GPL(tpm2_probe);
529 
530 static int tpm2_init_bank_info(struct tpm_chip *chip, u32 bank_index)
531 {
532 	struct tpm_bank_info *bank = chip->allocated_banks + bank_index;
533 	struct tpm_digest digest = { .alg_id = bank->alg_id };
534 	int i;
535 
536 	/*
537 	 * Avoid unnecessary PCR read operations to reduce overhead
538 	 * and obtain identifiers of the crypto subsystem.
539 	 */
540 	for (i = 0; i < ARRAY_SIZE(tpm2_hash_map); i++) {
541 		enum hash_algo crypto_algo = tpm2_hash_map[i].crypto_id;
542 
543 		if (bank->alg_id != tpm2_hash_map[i].tpm_id)
544 			continue;
545 
546 		bank->digest_size = hash_digest_size[crypto_algo];
547 		bank->crypto_id = crypto_algo;
548 		return 0;
549 	}
550 
551 	bank->crypto_id = HASH_ALGO__LAST;
552 
553 	return tpm2_pcr_read(chip, 0, &digest, &bank->digest_size);
554 }
555 
556 struct tpm2_pcr_selection {
557 	__be16  hash_alg;
558 	u8  size_of_select;
559 	u8  pcr_select[3];
560 } __packed;
561 
562 ssize_t tpm2_get_pcr_allocation(struct tpm_chip *chip)
563 {
564 	struct tpm2_pcr_selection pcr_selection;
565 	struct tpm_buf buf;
566 	void *marker;
567 	void *end;
568 	void *pcr_select_offset;
569 	u32 sizeof_pcr_selection;
570 	u32 nr_possible_banks;
571 	u32 nr_alloc_banks = 0;
572 	u16 hash_alg;
573 	u32 rsp_len;
574 	int rc;
575 	int i = 0;
576 
577 	rc = tpm_buf_init(&buf, TPM2_ST_NO_SESSIONS, TPM2_CC_GET_CAPABILITY);
578 	if (rc)
579 		return rc;
580 
581 	tpm_buf_append_u32(&buf, TPM2_CAP_PCRS);
582 	tpm_buf_append_u32(&buf, 0);
583 	tpm_buf_append_u32(&buf, 1);
584 
585 	rc = tpm_transmit_cmd(chip, &buf, 9, "get tpm pcr allocation");
586 	if (rc)
587 		goto out;
588 
589 	nr_possible_banks = be32_to_cpup(
590 		(__be32 *)&buf.data[TPM_HEADER_SIZE + 5]);
591 
592 	chip->allocated_banks = kcalloc(nr_possible_banks,
593 					sizeof(*chip->allocated_banks),
594 					GFP_KERNEL);
595 	if (!chip->allocated_banks) {
596 		rc = -ENOMEM;
597 		goto out;
598 	}
599 
600 	marker = &buf.data[TPM_HEADER_SIZE + 9];
601 
602 	rsp_len = be32_to_cpup((__be32 *)&buf.data[2]);
603 	end = &buf.data[rsp_len];
604 
605 	for (i = 0; i < nr_possible_banks; i++) {
606 		pcr_select_offset = marker +
607 			offsetof(struct tpm2_pcr_selection, size_of_select);
608 		if (pcr_select_offset >= end) {
609 			rc = -EFAULT;
610 			break;
611 		}
612 
613 		memcpy(&pcr_selection, marker, sizeof(pcr_selection));
614 		hash_alg = be16_to_cpu(pcr_selection.hash_alg);
615 
616 		pcr_select_offset = memchr_inv(pcr_selection.pcr_select, 0,
617 					       pcr_selection.size_of_select);
618 		if (pcr_select_offset) {
619 			chip->allocated_banks[nr_alloc_banks].alg_id = hash_alg;
620 
621 			rc = tpm2_init_bank_info(chip, nr_alloc_banks);
622 			if (rc < 0)
623 				break;
624 
625 			nr_alloc_banks++;
626 		}
627 
628 		sizeof_pcr_selection = sizeof(pcr_selection.hash_alg) +
629 			sizeof(pcr_selection.size_of_select) +
630 			pcr_selection.size_of_select;
631 		marker = marker + sizeof_pcr_selection;
632 	}
633 
634 	chip->nr_allocated_banks = nr_alloc_banks;
635 out:
636 	tpm_buf_destroy(&buf);
637 
638 	return rc;
639 }
640 
641 int tpm2_get_cc_attrs_tbl(struct tpm_chip *chip)
642 {
643 	struct tpm_buf buf;
644 	u32 nr_commands;
645 	__be32 *attrs;
646 	u32 cc;
647 	int i;
648 	int rc;
649 
650 	rc = tpm2_get_tpm_pt(chip, TPM_PT_TOTAL_COMMANDS, &nr_commands, NULL);
651 	if (rc)
652 		goto out;
653 
654 	if (nr_commands > 0xFFFFF) {
655 		rc = -EFAULT;
656 		goto out;
657 	}
658 
659 	chip->cc_attrs_tbl = devm_kcalloc(&chip->dev, 4, nr_commands,
660 					  GFP_KERNEL);
661 	if (!chip->cc_attrs_tbl) {
662 		rc = -ENOMEM;
663 		goto out;
664 	}
665 
666 	rc = tpm_buf_init(&buf, TPM2_ST_NO_SESSIONS, TPM2_CC_GET_CAPABILITY);
667 	if (rc)
668 		goto out;
669 
670 	tpm_buf_append_u32(&buf, TPM2_CAP_COMMANDS);
671 	tpm_buf_append_u32(&buf, TPM2_CC_FIRST);
672 	tpm_buf_append_u32(&buf, nr_commands);
673 
674 	rc = tpm_transmit_cmd(chip, &buf, 9 + 4 * nr_commands, NULL);
675 	if (rc) {
676 		tpm_buf_destroy(&buf);
677 		goto out;
678 	}
679 
680 	if (nr_commands !=
681 	    be32_to_cpup((__be32 *)&buf.data[TPM_HEADER_SIZE + 5])) {
682 		rc = -EFAULT;
683 		tpm_buf_destroy(&buf);
684 		goto out;
685 	}
686 
687 	chip->nr_commands = nr_commands;
688 
689 	attrs = (__be32 *)&buf.data[TPM_HEADER_SIZE + 9];
690 	for (i = 0; i < nr_commands; i++, attrs++) {
691 		chip->cc_attrs_tbl[i] = be32_to_cpup(attrs);
692 		cc = chip->cc_attrs_tbl[i] & 0xFFFF;
693 
694 		if (cc == TPM2_CC_CONTEXT_SAVE || cc == TPM2_CC_FLUSH_CONTEXT) {
695 			chip->cc_attrs_tbl[i] &=
696 				~(GENMASK(2, 0) << TPM2_CC_ATTR_CHANDLES);
697 			chip->cc_attrs_tbl[i] |= 1 << TPM2_CC_ATTR_CHANDLES;
698 		}
699 	}
700 
701 	tpm_buf_destroy(&buf);
702 
703 out:
704 	if (rc > 0)
705 		rc = -ENODEV;
706 	return rc;
707 }
708 EXPORT_SYMBOL_GPL(tpm2_get_cc_attrs_tbl);
709 
710 /**
711  * tpm2_startup - turn on the TPM
712  * @chip: TPM chip to use
713  *
714  * Normally the firmware should start the TPM. This function is provided as a
715  * workaround if this does not happen. A legal case for this could be for
716  * example when a TPM emulator is used.
717  *
718  * Return: same as tpm_transmit_cmd()
719  */
720 
721 static int tpm2_startup(struct tpm_chip *chip)
722 {
723 	struct tpm_buf buf;
724 	int rc;
725 
726 	dev_info(&chip->dev, "starting up the TPM manually\n");
727 
728 	rc = tpm_buf_init(&buf, TPM2_ST_NO_SESSIONS, TPM2_CC_STARTUP);
729 	if (rc < 0)
730 		return rc;
731 
732 	tpm_buf_append_u16(&buf, TPM2_SU_CLEAR);
733 	rc = tpm_transmit_cmd(chip, &buf, 0, "attempting to start the TPM");
734 	tpm_buf_destroy(&buf);
735 
736 	return rc;
737 }
738 
739 /**
740  * tpm2_auto_startup - Perform the standard automatic TPM initialization
741  *                     sequence
742  * @chip: TPM chip to use
743  *
744  * Returns 0 on success, < 0 in case of fatal error.
745  */
746 int tpm2_auto_startup(struct tpm_chip *chip)
747 {
748 	int rc;
749 
750 	rc = tpm2_get_timeouts(chip);
751 	if (rc)
752 		goto out;
753 
754 	rc = tpm2_do_selftest(chip);
755 	if (rc && rc != TPM2_RC_INITIALIZE)
756 		goto out;
757 
758 	if (rc == TPM2_RC_INITIALIZE) {
759 		rc = tpm2_startup(chip);
760 		if (rc)
761 			goto out;
762 
763 		rc = tpm2_do_selftest(chip);
764 		if (rc)
765 			goto out;
766 	}
767 
768 	rc = tpm2_get_cc_attrs_tbl(chip);
769 	if (rc == TPM2_RC_FAILURE || (rc < 0 && rc != -ENOMEM)) {
770 		dev_info(&chip->dev,
771 			 "TPM in field failure mode, requires firmware upgrade\n");
772 		chip->flags |= TPM_CHIP_FLAG_FIRMWARE_UPGRADE;
773 		rc = 0;
774 	}
775 
776 	if (rc)
777 		goto out;
778 
779 	rc = tpm2_sessions_init(chip);
780 
781 out:
782 	/*
783 	 * Infineon TPM in field upgrade mode will return no data for the number
784 	 * of supported commands.
785 	 */
786 	if (rc == TPM2_RC_UPGRADE || rc == -ENODATA) {
787 		dev_info(&chip->dev, "TPM in field upgrade mode, requires firmware upgrade\n");
788 		chip->flags |= TPM_CHIP_FLAG_FIRMWARE_UPGRADE;
789 		rc = 0;
790 	}
791 
792 	if (rc > 0)
793 		rc = -ENODEV;
794 	return rc;
795 }
796 
797 int tpm2_find_cc(struct tpm_chip *chip, u32 cc)
798 {
799 	u32 cc_mask;
800 	int i;
801 
802 	cc_mask = 1 << TPM2_CC_ATTR_VENDOR | GENMASK(15, 0);
803 	for (i = 0; i < chip->nr_commands; i++)
804 		if (cc == (chip->cc_attrs_tbl[i] & cc_mask))
805 			return i;
806 
807 	return -1;
808 }
809