tpm: fix potential buffer overruns caused by bit glitches on the bus
[sfrench/cifs-2.6.git] / drivers / char / tpm / tpm2-cmd.c
1 /*
2  * Copyright (C) 2014, 2015 Intel Corporation
3  *
4  * Authors:
5  * Jarkko Sakkinen <jarkko.sakkinen@linux.intel.com>
6  *
7  * Maintained by: <tpmdd-devel@lists.sourceforge.net>
8  *
9  * This file contains TPM2 protocol implementations of the commands
10  * used by the kernel internally.
11  *
12  * This program is free software; you can redistribute it and/or
13  * modify it under the terms of the GNU General Public License
14  * as published by the Free Software Foundation; version 2
15  * of the License.
16  */
17
18 #include "tpm.h"
19 #include <crypto/hash_info.h>
20 #include <keys/trusted-type.h>
21
22 enum tpm2_object_attributes {
23         TPM2_OA_USER_WITH_AUTH          = BIT(6),
24 };
25
26 enum tpm2_session_attributes {
27         TPM2_SA_CONTINUE_SESSION        = BIT(0),
28 };
29
30 struct tpm2_startup_in {
31         __be16  startup_type;
32 } __packed;
33
34 struct tpm2_self_test_in {
35         u8      full_test;
36 } __packed;
37
38 struct tpm2_get_tpm_pt_in {
39         __be32  cap_id;
40         __be32  property_id;
41         __be32  property_cnt;
42 } __packed;
43
44 struct tpm2_get_tpm_pt_out {
45         u8      more_data;
46         __be32  subcap_id;
47         __be32  property_cnt;
48         __be32  property_id;
49         __be32  value;
50 } __packed;
51
52 struct tpm2_get_random_in {
53         __be16  size;
54 } __packed;
55
56 struct tpm2_get_random_out {
57         __be16  size;
58         u8      buffer[TPM_MAX_RNG_DATA];
59 } __packed;
60
61 union tpm2_cmd_params {
62         struct  tpm2_startup_in         startup_in;
63         struct  tpm2_self_test_in       selftest_in;
64         struct  tpm2_get_tpm_pt_in      get_tpm_pt_in;
65         struct  tpm2_get_tpm_pt_out     get_tpm_pt_out;
66         struct  tpm2_get_random_in      getrandom_in;
67         struct  tpm2_get_random_out     getrandom_out;
68 };
69
70 struct tpm2_cmd {
71         tpm_cmd_header          header;
72         union tpm2_cmd_params   params;
73 } __packed;
74
75 struct tpm2_hash {
76         unsigned int crypto_id;
77         unsigned int tpm_id;
78 };
79
80 static struct tpm2_hash tpm2_hash_map[] = {
81         {HASH_ALGO_SHA1, TPM2_ALG_SHA1},
82         {HASH_ALGO_SHA256, TPM2_ALG_SHA256},
83         {HASH_ALGO_SHA384, TPM2_ALG_SHA384},
84         {HASH_ALGO_SHA512, TPM2_ALG_SHA512},
85         {HASH_ALGO_SM3_256, TPM2_ALG_SM3_256},
86 };
87
88 /*
89  * Array with one entry per ordinal defining the maximum amount
90  * of time the chip could take to return the result. The values
91  * of the SHORT, MEDIUM, and LONG durations are taken from the
92  * PC Client Profile (PTP) specification.
93  */
94 static const u8 tpm2_ordinal_duration[TPM2_CC_LAST - TPM2_CC_FIRST + 1] = {
95         TPM_UNDEFINED,          /* 11F */
96         TPM_UNDEFINED,          /* 120 */
97         TPM_LONG,               /* 121 */
98         TPM_UNDEFINED,          /* 122 */
99         TPM_UNDEFINED,          /* 123 */
100         TPM_UNDEFINED,          /* 124 */
101         TPM_UNDEFINED,          /* 125 */
102         TPM_UNDEFINED,          /* 126 */
103         TPM_UNDEFINED,          /* 127 */
104         TPM_UNDEFINED,          /* 128 */
105         TPM_LONG,               /* 129 */
106         TPM_UNDEFINED,          /* 12a */
107         TPM_UNDEFINED,          /* 12b */
108         TPM_UNDEFINED,          /* 12c */
109         TPM_UNDEFINED,          /* 12d */
110         TPM_UNDEFINED,          /* 12e */
111         TPM_UNDEFINED,          /* 12f */
112         TPM_UNDEFINED,          /* 130 */
113         TPM_UNDEFINED,          /* 131 */
114         TPM_UNDEFINED,          /* 132 */
115         TPM_UNDEFINED,          /* 133 */
116         TPM_UNDEFINED,          /* 134 */
117         TPM_UNDEFINED,          /* 135 */
118         TPM_UNDEFINED,          /* 136 */
119         TPM_UNDEFINED,          /* 137 */
120         TPM_UNDEFINED,          /* 138 */
121         TPM_UNDEFINED,          /* 139 */
122         TPM_UNDEFINED,          /* 13a */
123         TPM_UNDEFINED,          /* 13b */
124         TPM_UNDEFINED,          /* 13c */
125         TPM_UNDEFINED,          /* 13d */
126         TPM_MEDIUM,             /* 13e */
127         TPM_UNDEFINED,          /* 13f */
128         TPM_UNDEFINED,          /* 140 */
129         TPM_UNDEFINED,          /* 141 */
130         TPM_UNDEFINED,          /* 142 */
131         TPM_LONG,               /* 143 */
132         TPM_MEDIUM,             /* 144 */
133         TPM_UNDEFINED,          /* 145 */
134         TPM_UNDEFINED,          /* 146 */
135         TPM_UNDEFINED,          /* 147 */
136         TPM_UNDEFINED,          /* 148 */
137         TPM_UNDEFINED,          /* 149 */
138         TPM_UNDEFINED,          /* 14a */
139         TPM_UNDEFINED,          /* 14b */
140         TPM_UNDEFINED,          /* 14c */
141         TPM_UNDEFINED,          /* 14d */
142         TPM_LONG,               /* 14e */
143         TPM_UNDEFINED,          /* 14f */
144         TPM_UNDEFINED,          /* 150 */
145         TPM_UNDEFINED,          /* 151 */
146         TPM_UNDEFINED,          /* 152 */
147         TPM_UNDEFINED,          /* 153 */
148         TPM_UNDEFINED,          /* 154 */
149         TPM_UNDEFINED,          /* 155 */
150         TPM_UNDEFINED,          /* 156 */
151         TPM_UNDEFINED,          /* 157 */
152         TPM_UNDEFINED,          /* 158 */
153         TPM_UNDEFINED,          /* 159 */
154         TPM_UNDEFINED,          /* 15a */
155         TPM_UNDEFINED,          /* 15b */
156         TPM_MEDIUM,             /* 15c */
157         TPM_UNDEFINED,          /* 15d */
158         TPM_UNDEFINED,          /* 15e */
159         TPM_UNDEFINED,          /* 15f */
160         TPM_UNDEFINED,          /* 160 */
161         TPM_UNDEFINED,          /* 161 */
162         TPM_UNDEFINED,          /* 162 */
163         TPM_UNDEFINED,          /* 163 */
164         TPM_UNDEFINED,          /* 164 */
165         TPM_UNDEFINED,          /* 165 */
166         TPM_UNDEFINED,          /* 166 */
167         TPM_UNDEFINED,          /* 167 */
168         TPM_UNDEFINED,          /* 168 */
169         TPM_UNDEFINED,          /* 169 */
170         TPM_UNDEFINED,          /* 16a */
171         TPM_UNDEFINED,          /* 16b */
172         TPM_UNDEFINED,          /* 16c */
173         TPM_UNDEFINED,          /* 16d */
174         TPM_UNDEFINED,          /* 16e */
175         TPM_UNDEFINED,          /* 16f */
176         TPM_UNDEFINED,          /* 170 */
177         TPM_UNDEFINED,          /* 171 */
178         TPM_UNDEFINED,          /* 172 */
179         TPM_UNDEFINED,          /* 173 */
180         TPM_UNDEFINED,          /* 174 */
181         TPM_UNDEFINED,          /* 175 */
182         TPM_UNDEFINED,          /* 176 */
183         TPM_LONG,               /* 177 */
184         TPM_UNDEFINED,          /* 178 */
185         TPM_UNDEFINED,          /* 179 */
186         TPM_MEDIUM,             /* 17a */
187         TPM_LONG,               /* 17b */
188         TPM_UNDEFINED,          /* 17c */
189         TPM_UNDEFINED,          /* 17d */
190         TPM_UNDEFINED,          /* 17e */
191         TPM_UNDEFINED,          /* 17f */
192         TPM_UNDEFINED,          /* 180 */
193         TPM_UNDEFINED,          /* 181 */
194         TPM_MEDIUM,             /* 182 */
195         TPM_UNDEFINED,          /* 183 */
196         TPM_UNDEFINED,          /* 184 */
197         TPM_MEDIUM,             /* 185 */
198         TPM_MEDIUM,             /* 186 */
199         TPM_UNDEFINED,          /* 187 */
200         TPM_UNDEFINED,          /* 188 */
201         TPM_UNDEFINED,          /* 189 */
202         TPM_UNDEFINED,          /* 18a */
203         TPM_UNDEFINED,          /* 18b */
204         TPM_UNDEFINED,          /* 18c */
205         TPM_UNDEFINED,          /* 18d */
206         TPM_UNDEFINED,          /* 18e */
207         TPM_UNDEFINED           /* 18f */
208 };
209
210 struct tpm2_pcr_read_out {
211         __be32  update_cnt;
212         __be32  pcr_selects_cnt;
213         __be16  hash_alg;
214         u8      pcr_select_size;
215         u8      pcr_select[TPM2_PCR_SELECT_MIN];
216         __be32  digests_cnt;
217         __be16  digest_size;
218         u8      digest[];
219 } __packed;
220
221 /**
222  * tpm2_pcr_read() - read a PCR value
223  * @chip:       TPM chip to use.
224  * @pcr_idx:    index of the PCR to read.
225  * @res_buf:    buffer to store the resulting hash.
226  *
227  * Return: Same as with tpm_transmit_cmd.
228  */
229 int tpm2_pcr_read(struct tpm_chip *chip, int pcr_idx, u8 *res_buf)
230 {
231         int rc;
232         struct tpm_buf buf;
233         struct tpm2_pcr_read_out *out;
234         u8 pcr_select[TPM2_PCR_SELECT_MIN] = {0};
235
236         if (pcr_idx >= TPM2_PLATFORM_PCR)
237                 return -EINVAL;
238
239         rc = tpm_buf_init(&buf, TPM2_ST_NO_SESSIONS, TPM2_CC_PCR_READ);
240         if (rc)
241                 return rc;
242
243         pcr_select[pcr_idx >> 3] = 1 << (pcr_idx & 0x7);
244
245         tpm_buf_append_u32(&buf, 1);
246         tpm_buf_append_u16(&buf, TPM2_ALG_SHA1);
247         tpm_buf_append_u8(&buf, TPM2_PCR_SELECT_MIN);
248         tpm_buf_append(&buf, (const unsigned char *)pcr_select,
249                        sizeof(pcr_select));
250
251         rc = tpm_transmit_cmd(chip, NULL, buf.data, PAGE_SIZE, 0, 0,
252                         res_buf ? "attempting to read a pcr value" : NULL);
253         if (rc == 0 && res_buf) {
254                 out = (struct tpm2_pcr_read_out *)&buf.data[TPM_HEADER_SIZE];
255                 memcpy(res_buf, out->digest, SHA1_DIGEST_SIZE);
256         }
257
258         tpm_buf_destroy(&buf);
259         return rc;
260 }
261
262 struct tpm2_null_auth_area {
263         __be32  handle;
264         __be16  nonce_size;
265         u8  attributes;
266         __be16  auth_size;
267 } __packed;
268
269 /**
270  * tpm2_pcr_extend() - extend a PCR value
271  *
272  * @chip:       TPM chip to use.
273  * @pcr_idx:    index of the PCR.
274  * @count:      number of digests passed.
275  * @digests:    list of pcr banks and corresponding digest values to extend.
276  *
277  * Return: Same as with tpm_transmit_cmd.
278  */
279 int tpm2_pcr_extend(struct tpm_chip *chip, int pcr_idx, u32 count,
280                     struct tpm2_digest *digests)
281 {
282         struct tpm_buf buf;
283         struct tpm2_null_auth_area auth_area;
284         int rc;
285         int i;
286         int j;
287
288         if (count > ARRAY_SIZE(chip->active_banks))
289                 return -EINVAL;
290
291         rc = tpm_buf_init(&buf, TPM2_ST_SESSIONS, TPM2_CC_PCR_EXTEND);
292         if (rc)
293                 return rc;
294
295         tpm_buf_append_u32(&buf, pcr_idx);
296
297         auth_area.handle = cpu_to_be32(TPM2_RS_PW);
298         auth_area.nonce_size = 0;
299         auth_area.attributes = 0;
300         auth_area.auth_size = 0;
301
302         tpm_buf_append_u32(&buf, sizeof(struct tpm2_null_auth_area));
303         tpm_buf_append(&buf, (const unsigned char *)&auth_area,
304                        sizeof(auth_area));
305         tpm_buf_append_u32(&buf, count);
306
307         for (i = 0; i < count; i++) {
308                 for (j = 0; j < ARRAY_SIZE(tpm2_hash_map); j++) {
309                         if (digests[i].alg_id != tpm2_hash_map[j].tpm_id)
310                                 continue;
311                         tpm_buf_append_u16(&buf, digests[i].alg_id);
312                         tpm_buf_append(&buf, (const unsigned char
313                                               *)&digests[i].digest,
314                                hash_digest_size[tpm2_hash_map[j].crypto_id]);
315                 }
316         }
317
318         rc = tpm_transmit_cmd(chip, NULL, buf.data, PAGE_SIZE, 0, 0,
319                               "attempting extend a PCR value");
320
321         tpm_buf_destroy(&buf);
322
323         return rc;
324 }
325
326
327 #define TPM2_GETRANDOM_IN_SIZE \
328         (sizeof(struct tpm_input_header) + \
329          sizeof(struct tpm2_get_random_in))
330
331 static const struct tpm_input_header tpm2_getrandom_header = {
332         .tag = cpu_to_be16(TPM2_ST_NO_SESSIONS),
333         .length = cpu_to_be32(TPM2_GETRANDOM_IN_SIZE),
334         .ordinal = cpu_to_be32(TPM2_CC_GET_RANDOM)
335 };
336
337 /**
338  * tpm2_get_random() - get random bytes from the TPM RNG
339  *
340  * @chip: TPM chip to use
341  * @out: destination buffer for the random bytes
342  * @max: the max number of bytes to write to @out
343  *
344  * Return:
345  *    Size of the output buffer, or -EIO on error.
346  */
347 int tpm2_get_random(struct tpm_chip *chip, u8 *out, size_t max)
348 {
349         struct tpm2_cmd cmd;
350         u32 recd, rlength;
351         u32 num_bytes;
352         int err;
353         int total = 0;
354         int retries = 5;
355         u8 *dest = out;
356
357         num_bytes = min_t(u32, max, sizeof(cmd.params.getrandom_out.buffer));
358
359         if (!out || !num_bytes ||
360             max > sizeof(cmd.params.getrandom_out.buffer))
361                 return -EINVAL;
362
363         do {
364                 cmd.header.in = tpm2_getrandom_header;
365                 cmd.params.getrandom_in.size = cpu_to_be16(num_bytes);
366
367                 err = tpm_transmit_cmd(chip, NULL, &cmd, sizeof(cmd),
368                                        offsetof(struct tpm2_get_random_out,
369                                                 buffer),
370                                        0, "attempting get random");
371                 if (err)
372                         break;
373
374                 recd = min_t(u32, be16_to_cpu(cmd.params.getrandom_out.size),
375                              num_bytes);
376                 rlength = be32_to_cpu(cmd.header.out.length);
377                 if (rlength < offsetof(struct tpm2_get_random_out, buffer) +
378                               recd)
379                         return -EFAULT;
380                 memcpy(dest, cmd.params.getrandom_out.buffer, recd);
381
382                 dest += recd;
383                 total += recd;
384                 num_bytes -= recd;
385         } while (retries-- && total < max);
386
387         return total ? total : -EIO;
388 }
389
390 #define TPM2_GET_TPM_PT_IN_SIZE \
391         (sizeof(struct tpm_input_header) + \
392          sizeof(struct tpm2_get_tpm_pt_in))
393
394 #define TPM2_GET_TPM_PT_OUT_BODY_SIZE \
395          sizeof(struct tpm2_get_tpm_pt_out)
396
397 static const struct tpm_input_header tpm2_get_tpm_pt_header = {
398         .tag = cpu_to_be16(TPM2_ST_NO_SESSIONS),
399         .length = cpu_to_be32(TPM2_GET_TPM_PT_IN_SIZE),
400         .ordinal = cpu_to_be32(TPM2_CC_GET_CAPABILITY)
401 };
402
403 /**
404  * tpm2_flush_context_cmd() - execute a TPM2_FlushContext command
405  * @chip: TPM chip to use
406  * @payload: the key data in clear and encrypted form
407  * @options: authentication values and other options
408  *
409  * Return: same as with tpm_transmit_cmd
410  */
411 void tpm2_flush_context_cmd(struct tpm_chip *chip, u32 handle,
412                             unsigned int flags)
413 {
414         struct tpm_buf buf;
415         int rc;
416
417         rc = tpm_buf_init(&buf, TPM2_ST_NO_SESSIONS, TPM2_CC_FLUSH_CONTEXT);
418         if (rc) {
419                 dev_warn(&chip->dev, "0x%08x was not flushed, out of memory\n",
420                          handle);
421                 return;
422         }
423
424         tpm_buf_append_u32(&buf, handle);
425
426         (void) tpm_transmit_cmd(chip, NULL, buf.data, PAGE_SIZE, 0, flags,
427                                 "flushing context");
428
429         tpm_buf_destroy(&buf);
430 }
431
432 /**
433  * tpm_buf_append_auth() - append TPMS_AUTH_COMMAND to the buffer.
434  *
435  * @buf: an allocated tpm_buf instance
436  * @session_handle: session handle
437  * @nonce: the session nonce, may be NULL if not used
438  * @nonce_len: the session nonce length, may be 0 if not used
439  * @attributes: the session attributes
440  * @hmac: the session HMAC or password, may be NULL if not used
441  * @hmac_len: the session HMAC or password length, maybe 0 if not used
442  */
443 static void tpm2_buf_append_auth(struct tpm_buf *buf, u32 session_handle,
444                                  const u8 *nonce, u16 nonce_len,
445                                  u8 attributes,
446                                  const u8 *hmac, u16 hmac_len)
447 {
448         tpm_buf_append_u32(buf, 9 + nonce_len + hmac_len);
449         tpm_buf_append_u32(buf, session_handle);
450         tpm_buf_append_u16(buf, nonce_len);
451
452         if (nonce && nonce_len)
453                 tpm_buf_append(buf, nonce, nonce_len);
454
455         tpm_buf_append_u8(buf, attributes);
456         tpm_buf_append_u16(buf, hmac_len);
457
458         if (hmac && hmac_len)
459                 tpm_buf_append(buf, hmac, hmac_len);
460 }
461
462 /**
463  * tpm2_seal_trusted() - seal the payload of a trusted key
464  *
465  * @chip: TPM chip to use
466  * @payload: the key data in clear and encrypted form
467  * @options: authentication values and other options
468  *
469  * Return: < 0 on error and 0 on success.
470  */
471 int tpm2_seal_trusted(struct tpm_chip *chip,
472                       struct trusted_key_payload *payload,
473                       struct trusted_key_options *options)
474 {
475         unsigned int blob_len;
476         struct tpm_buf buf;
477         u32 hash, rlength;
478         int i;
479         int rc;
480
481         for (i = 0; i < ARRAY_SIZE(tpm2_hash_map); i++) {
482                 if (options->hash == tpm2_hash_map[i].crypto_id) {
483                         hash = tpm2_hash_map[i].tpm_id;
484                         break;
485                 }
486         }
487
488         if (i == ARRAY_SIZE(tpm2_hash_map))
489                 return -EINVAL;
490
491         rc = tpm_buf_init(&buf, TPM2_ST_SESSIONS, TPM2_CC_CREATE);
492         if (rc)
493                 return rc;
494
495         tpm_buf_append_u32(&buf, options->keyhandle);
496         tpm2_buf_append_auth(&buf, TPM2_RS_PW,
497                              NULL /* nonce */, 0,
498                              0 /* session_attributes */,
499                              options->keyauth /* hmac */,
500                              TPM_DIGEST_SIZE);
501
502         /* sensitive */
503         tpm_buf_append_u16(&buf, 4 + TPM_DIGEST_SIZE + payload->key_len + 1);
504
505         tpm_buf_append_u16(&buf, TPM_DIGEST_SIZE);
506         tpm_buf_append(&buf, options->blobauth, TPM_DIGEST_SIZE);
507         tpm_buf_append_u16(&buf, payload->key_len + 1);
508         tpm_buf_append(&buf, payload->key, payload->key_len);
509         tpm_buf_append_u8(&buf, payload->migratable);
510
511         /* public */
512         tpm_buf_append_u16(&buf, 14 + options->policydigest_len);
513         tpm_buf_append_u16(&buf, TPM2_ALG_KEYEDHASH);
514         tpm_buf_append_u16(&buf, hash);
515
516         /* policy */
517         if (options->policydigest_len) {
518                 tpm_buf_append_u32(&buf, 0);
519                 tpm_buf_append_u16(&buf, options->policydigest_len);
520                 tpm_buf_append(&buf, options->policydigest,
521                                options->policydigest_len);
522         } else {
523                 tpm_buf_append_u32(&buf, TPM2_OA_USER_WITH_AUTH);
524                 tpm_buf_append_u16(&buf, 0);
525         }
526
527         /* public parameters */
528         tpm_buf_append_u16(&buf, TPM2_ALG_NULL);
529         tpm_buf_append_u16(&buf, 0);
530
531         /* outside info */
532         tpm_buf_append_u16(&buf, 0);
533
534         /* creation PCR */
535         tpm_buf_append_u32(&buf, 0);
536
537         if (buf.flags & TPM_BUF_OVERFLOW) {
538                 rc = -E2BIG;
539                 goto out;
540         }
541
542         rc = tpm_transmit_cmd(chip, NULL, buf.data, PAGE_SIZE, 4, 0,
543                               "sealing data");
544         if (rc)
545                 goto out;
546
547         blob_len = be32_to_cpup((__be32 *) &buf.data[TPM_HEADER_SIZE]);
548         if (blob_len > MAX_BLOB_SIZE) {
549                 rc = -E2BIG;
550                 goto out;
551         }
552         rlength = be32_to_cpu(((struct tpm2_cmd *)&buf)->header.out.length);
553         if (rlength < TPM_HEADER_SIZE + 4 + blob_len) {
554                 rc = -EFAULT;
555                 goto out;
556         }
557
558         memcpy(payload->blob, &buf.data[TPM_HEADER_SIZE + 4], blob_len);
559         payload->blob_len = blob_len;
560
561 out:
562         tpm_buf_destroy(&buf);
563
564         if (rc > 0) {
565                 if (tpm2_rc_value(rc) == TPM2_RC_HASH)
566                         rc = -EINVAL;
567                 else
568                         rc = -EPERM;
569         }
570
571         return rc;
572 }
573
574 /**
575  * tpm2_load_cmd() - execute a TPM2_Load command
576  *
577  * @chip: TPM chip to use
578  * @payload: the key data in clear and encrypted form
579  * @options: authentication values and other options
580  * @blob_handle: returned blob handle
581  * @flags: tpm transmit flags
582  *
583  * Return: 0 on success.
584  *        -E2BIG on wrong payload size.
585  *        -EPERM on tpm error status.
586  *        < 0 error from tpm_transmit_cmd.
587  */
588 static int tpm2_load_cmd(struct tpm_chip *chip,
589                          struct trusted_key_payload *payload,
590                          struct trusted_key_options *options,
591                          u32 *blob_handle, unsigned int flags)
592 {
593         struct tpm_buf buf;
594         unsigned int private_len;
595         unsigned int public_len;
596         unsigned int blob_len;
597         int rc;
598
599         private_len = be16_to_cpup((__be16 *) &payload->blob[0]);
600         if (private_len > (payload->blob_len - 2))
601                 return -E2BIG;
602
603         public_len = be16_to_cpup((__be16 *) &payload->blob[2 + private_len]);
604         blob_len = private_len + public_len + 4;
605         if (blob_len > payload->blob_len)
606                 return -E2BIG;
607
608         rc = tpm_buf_init(&buf, TPM2_ST_SESSIONS, TPM2_CC_LOAD);
609         if (rc)
610                 return rc;
611
612         tpm_buf_append_u32(&buf, options->keyhandle);
613         tpm2_buf_append_auth(&buf, TPM2_RS_PW,
614                              NULL /* nonce */, 0,
615                              0 /* session_attributes */,
616                              options->keyauth /* hmac */,
617                              TPM_DIGEST_SIZE);
618
619         tpm_buf_append(&buf, payload->blob, blob_len);
620
621         if (buf.flags & TPM_BUF_OVERFLOW) {
622                 rc = -E2BIG;
623                 goto out;
624         }
625
626         rc = tpm_transmit_cmd(chip, NULL, buf.data, PAGE_SIZE, 4, flags,
627                               "loading blob");
628         if (!rc)
629                 *blob_handle = be32_to_cpup(
630                         (__be32 *) &buf.data[TPM_HEADER_SIZE]);
631
632 out:
633         tpm_buf_destroy(&buf);
634
635         if (rc > 0)
636                 rc = -EPERM;
637
638         return rc;
639 }
640
641 /**
642  * tpm2_unseal_cmd() - execute a TPM2_Unload command
643  *
644  * @chip: TPM chip to use
645  * @payload: the key data in clear and encrypted form
646  * @options: authentication values and other options
647  * @blob_handle: blob handle
648  * @flags: tpm_transmit_cmd flags
649  *
650  * Return: 0 on success
651  *         -EPERM on tpm error status
652  *         < 0 error from tpm_transmit_cmd
653  */
654 static int tpm2_unseal_cmd(struct tpm_chip *chip,
655                            struct trusted_key_payload *payload,
656                            struct trusted_key_options *options,
657                            u32 blob_handle, unsigned int flags)
658 {
659         struct tpm_buf buf;
660         u16 data_len;
661         u8 *data;
662         int rc;
663         u32 rlength;
664
665         rc = tpm_buf_init(&buf, TPM2_ST_SESSIONS, TPM2_CC_UNSEAL);
666         if (rc)
667                 return rc;
668
669         tpm_buf_append_u32(&buf, blob_handle);
670         tpm2_buf_append_auth(&buf,
671                              options->policyhandle ?
672                              options->policyhandle : TPM2_RS_PW,
673                              NULL /* nonce */, 0,
674                              TPM2_SA_CONTINUE_SESSION,
675                              options->blobauth /* hmac */,
676                              TPM_DIGEST_SIZE);
677
678         rc = tpm_transmit_cmd(chip, NULL, buf.data, PAGE_SIZE, 6, flags,
679                               "unsealing");
680         if (rc > 0)
681                 rc = -EPERM;
682
683         if (!rc) {
684                 data_len = be16_to_cpup(
685                         (__be16 *) &buf.data[TPM_HEADER_SIZE + 4]);
686                 if (data_len < MIN_KEY_SIZE ||  data_len > MAX_KEY_SIZE + 1) {
687                         rc = -EFAULT;
688                         goto out;
689                 }
690
691                 rlength = be32_to_cpu(((struct tpm2_cmd *)&buf)
692                                         ->header.out.length);
693                 if (rlength < TPM_HEADER_SIZE + 6 + data_len) {
694                         rc = -EFAULT;
695                         goto out;
696                 }
697                 data = &buf.data[TPM_HEADER_SIZE + 6];
698
699                 memcpy(payload->key, data, data_len - 1);
700                 payload->key_len = data_len - 1;
701                 payload->migratable = data[data_len - 1];
702         }
703
704 out:
705         tpm_buf_destroy(&buf);
706         return rc;
707 }
708
709 /**
710  * tpm2_unseal_trusted() - unseal the payload of a trusted key
711  *
712  * @chip: TPM chip to use
713  * @payload: the key data in clear and encrypted form
714  * @options: authentication values and other options
715  *
716  * Return: Same as with tpm_transmit_cmd.
717  */
718 int tpm2_unseal_trusted(struct tpm_chip *chip,
719                         struct trusted_key_payload *payload,
720                         struct trusted_key_options *options)
721 {
722         u32 blob_handle;
723         int rc;
724
725         mutex_lock(&chip->tpm_mutex);
726         rc = tpm2_load_cmd(chip, payload, options, &blob_handle,
727                            TPM_TRANSMIT_UNLOCKED);
728         if (rc)
729                 goto out;
730
731         rc = tpm2_unseal_cmd(chip, payload, options, blob_handle,
732                              TPM_TRANSMIT_UNLOCKED);
733         tpm2_flush_context_cmd(chip, blob_handle, TPM_TRANSMIT_UNLOCKED);
734 out:
735         mutex_unlock(&chip->tpm_mutex);
736         return rc;
737 }
738
739 /**
740  * tpm2_get_tpm_pt() - get value of a TPM_CAP_TPM_PROPERTIES type property
741  * @chip:               TPM chip to use.
742  * @property_id:        property ID.
743  * @value:              output variable.
744  * @desc:               passed to tpm_transmit_cmd()
745  *
746  * Return: Same as with tpm_transmit_cmd.
747  */
748 ssize_t tpm2_get_tpm_pt(struct tpm_chip *chip, u32 property_id,  u32 *value,
749                         const char *desc)
750 {
751         struct tpm2_cmd cmd;
752         int rc;
753
754         cmd.header.in = tpm2_get_tpm_pt_header;
755         cmd.params.get_tpm_pt_in.cap_id = cpu_to_be32(TPM2_CAP_TPM_PROPERTIES);
756         cmd.params.get_tpm_pt_in.property_id = cpu_to_be32(property_id);
757         cmd.params.get_tpm_pt_in.property_cnt = cpu_to_be32(1);
758
759         rc = tpm_transmit_cmd(chip, NULL, &cmd, sizeof(cmd),
760                               TPM2_GET_TPM_PT_OUT_BODY_SIZE, 0, desc);
761         if (!rc)
762                 *value = be32_to_cpu(cmd.params.get_tpm_pt_out.value);
763
764         return rc;
765 }
766 EXPORT_SYMBOL_GPL(tpm2_get_tpm_pt);
767
768 #define TPM2_SHUTDOWN_IN_SIZE \
769         (sizeof(struct tpm_input_header) + \
770          sizeof(struct tpm2_startup_in))
771
772 static const struct tpm_input_header tpm2_shutdown_header = {
773         .tag = cpu_to_be16(TPM2_ST_NO_SESSIONS),
774         .length = cpu_to_be32(TPM2_SHUTDOWN_IN_SIZE),
775         .ordinal = cpu_to_be32(TPM2_CC_SHUTDOWN)
776 };
777
778 /**
779  * tpm2_shutdown() - send shutdown command to the TPM chip
780  *
781  * @chip:               TPM chip to use.
782  * @shutdown_type:      shutdown type. The value is either
783  *                      TPM_SU_CLEAR or TPM_SU_STATE.
784  */
785 void tpm2_shutdown(struct tpm_chip *chip, u16 shutdown_type)
786 {
787         struct tpm2_cmd cmd;
788         int rc;
789
790         cmd.header.in = tpm2_shutdown_header;
791         cmd.params.startup_in.startup_type = cpu_to_be16(shutdown_type);
792
793         rc = tpm_transmit_cmd(chip, NULL, &cmd, sizeof(cmd), 0, 0,
794                               "stopping the TPM");
795
796         /* In places where shutdown command is sent there's no much we can do
797          * except print the error code on a system failure.
798          */
799         if (rc < 0 && rc != -EPIPE)
800                 dev_warn(&chip->dev, "transmit returned %d while stopping the TPM",
801                          rc);
802 }
803
804 /*
805  * tpm2_calc_ordinal_duration() - maximum duration for a command
806  *
807  * @chip:       TPM chip to use.
808  * @ordinal:    command code number.
809  *
810  * Return: maximum duration for a command
811  */
812 unsigned long tpm2_calc_ordinal_duration(struct tpm_chip *chip, u32 ordinal)
813 {
814         int index = TPM_UNDEFINED;
815         int duration = 0;
816
817         if (ordinal >= TPM2_CC_FIRST && ordinal <= TPM2_CC_LAST)
818                 index = tpm2_ordinal_duration[ordinal - TPM2_CC_FIRST];
819
820         if (index != TPM_UNDEFINED)
821                 duration = chip->duration[index];
822
823         if (duration <= 0)
824                 duration = 2 * 60 * HZ;
825
826         return duration;
827 }
828 EXPORT_SYMBOL_GPL(tpm2_calc_ordinal_duration);
829
830 #define TPM2_SELF_TEST_IN_SIZE \
831         (sizeof(struct tpm_input_header) + \
832          sizeof(struct tpm2_self_test_in))
833
834 static const struct tpm_input_header tpm2_selftest_header = {
835         .tag = cpu_to_be16(TPM2_ST_NO_SESSIONS),
836         .length = cpu_to_be32(TPM2_SELF_TEST_IN_SIZE),
837         .ordinal = cpu_to_be32(TPM2_CC_SELF_TEST)
838 };
839
840 /**
841  * tpm2_do_selftest() - ensure that all self tests have passed
842  *
843  * @chip: TPM chip to use
844  *
845  * Return: Same as with tpm_transmit_cmd.
846  *
847  * The TPM can either run all self tests synchronously and then return
848  * RC_SUCCESS once all tests were successful. Or it can choose to run the tests
849  * asynchronously and return RC_TESTING immediately while the self tests still
850  * execute in the background. This function handles both cases and waits until
851  * all tests have completed.
852  */
853 static int tpm2_do_selftest(struct tpm_chip *chip)
854 {
855         int rc;
856         unsigned int delay_msec = 10;
857         long duration;
858         struct tpm2_cmd cmd;
859
860         duration = jiffies_to_msecs(
861                 tpm2_calc_ordinal_duration(chip, TPM2_CC_SELF_TEST));
862
863         while (1) {
864                 cmd.header.in = tpm2_selftest_header;
865                 cmd.params.selftest_in.full_test = 0;
866
867                 rc = tpm_transmit_cmd(chip, NULL, &cmd, TPM2_SELF_TEST_IN_SIZE,
868                                       0, 0, "continue selftest");
869
870                 if (rc != TPM2_RC_TESTING || delay_msec >= duration)
871                         break;
872
873                 /* wait longer than before */
874                 delay_msec *= 2;
875                 tpm_msleep(delay_msec);
876         }
877
878         return rc;
879 }
880
881 /**
882  * tpm2_probe() - probe TPM 2.0
883  * @chip: TPM chip to use
884  *
885  * Return: < 0 error and 0 on success.
886  *
887  * Send idempotent TPM 2.0 command and see whether TPM 2.0 chip replied based on
888  * the reply tag.
889  */
890 int tpm2_probe(struct tpm_chip *chip)
891 {
892         struct tpm2_cmd cmd;
893         int rc;
894
895         cmd.header.in = tpm2_get_tpm_pt_header;
896         cmd.params.get_tpm_pt_in.cap_id = cpu_to_be32(TPM2_CAP_TPM_PROPERTIES);
897         cmd.params.get_tpm_pt_in.property_id = cpu_to_be32(0x100);
898         cmd.params.get_tpm_pt_in.property_cnt = cpu_to_be32(1);
899
900         rc = tpm_transmit_cmd(chip, NULL, &cmd, sizeof(cmd), 0, 0, NULL);
901         if (rc <  0)
902                 return rc;
903
904         if (be16_to_cpu(cmd.header.out.tag) == TPM2_ST_NO_SESSIONS)
905                 chip->flags |= TPM_CHIP_FLAG_TPM2;
906
907         return 0;
908 }
909 EXPORT_SYMBOL_GPL(tpm2_probe);
910
911 struct tpm2_pcr_selection {
912         __be16  hash_alg;
913         u8  size_of_select;
914         u8  pcr_select[3];
915 } __packed;
916
917 static ssize_t tpm2_get_pcr_allocation(struct tpm_chip *chip)
918 {
919         struct tpm2_pcr_selection pcr_selection;
920         struct tpm_buf buf;
921         void *marker;
922         void *end;
923         void *pcr_select_offset;
924         unsigned int count;
925         u32 sizeof_pcr_selection;
926         u32 rsp_len;
927         int rc;
928         int i = 0;
929
930         rc = tpm_buf_init(&buf, TPM2_ST_NO_SESSIONS, TPM2_CC_GET_CAPABILITY);
931         if (rc)
932                 return rc;
933
934         tpm_buf_append_u32(&buf, TPM2_CAP_PCRS);
935         tpm_buf_append_u32(&buf, 0);
936         tpm_buf_append_u32(&buf, 1);
937
938         rc = tpm_transmit_cmd(chip, NULL, buf.data, PAGE_SIZE, 9, 0,
939                               "get tpm pcr allocation");
940         if (rc)
941                 goto out;
942
943         count = be32_to_cpup(
944                 (__be32 *)&buf.data[TPM_HEADER_SIZE + 5]);
945
946         if (count > ARRAY_SIZE(chip->active_banks)) {
947                 rc = -ENODEV;
948                 goto out;
949         }
950
951         marker = &buf.data[TPM_HEADER_SIZE + 9];
952
953         rsp_len = be32_to_cpup((__be32 *)&buf.data[2]);
954         end = &buf.data[rsp_len];
955
956         for (i = 0; i < count; i++) {
957                 pcr_select_offset = marker +
958                         offsetof(struct tpm2_pcr_selection, size_of_select);
959                 if (pcr_select_offset >= end) {
960                         rc = -EFAULT;
961                         break;
962                 }
963
964                 memcpy(&pcr_selection, marker, sizeof(pcr_selection));
965                 chip->active_banks[i] = be16_to_cpu(pcr_selection.hash_alg);
966                 sizeof_pcr_selection = sizeof(pcr_selection.hash_alg) +
967                         sizeof(pcr_selection.size_of_select) +
968                         pcr_selection.size_of_select;
969                 marker = marker + sizeof_pcr_selection;
970         }
971
972 out:
973         if (i < ARRAY_SIZE(chip->active_banks))
974                 chip->active_banks[i] = TPM2_ALG_ERROR;
975
976         tpm_buf_destroy(&buf);
977
978         return rc;
979 }
980
981 static int tpm2_get_cc_attrs_tbl(struct tpm_chip *chip)
982 {
983         struct tpm_buf buf;
984         u32 nr_commands;
985         __be32 *attrs;
986         u32 cc;
987         int i;
988         int rc;
989
990         rc = tpm2_get_tpm_pt(chip, TPM_PT_TOTAL_COMMANDS, &nr_commands, NULL);
991         if (rc)
992                 goto out;
993
994         if (nr_commands > 0xFFFFF) {
995                 rc = -EFAULT;
996                 goto out;
997         }
998
999         chip->cc_attrs_tbl = devm_kzalloc(&chip->dev, 4 * nr_commands,
1000                                           GFP_KERNEL);
1001
1002         rc = tpm_buf_init(&buf, TPM2_ST_NO_SESSIONS, TPM2_CC_GET_CAPABILITY);
1003         if (rc)
1004                 goto out;
1005
1006         tpm_buf_append_u32(&buf, TPM2_CAP_COMMANDS);
1007         tpm_buf_append_u32(&buf, TPM2_CC_FIRST);
1008         tpm_buf_append_u32(&buf, nr_commands);
1009
1010         rc = tpm_transmit_cmd(chip, NULL, buf.data, PAGE_SIZE,
1011                               9 + 4 * nr_commands, 0, NULL);
1012         if (rc) {
1013                 tpm_buf_destroy(&buf);
1014                 goto out;
1015         }
1016
1017         if (nr_commands !=
1018             be32_to_cpup((__be32 *)&buf.data[TPM_HEADER_SIZE + 5])) {
1019                 tpm_buf_destroy(&buf);
1020                 goto out;
1021         }
1022
1023         chip->nr_commands = nr_commands;
1024
1025         attrs = (__be32 *)&buf.data[TPM_HEADER_SIZE + 9];
1026         for (i = 0; i < nr_commands; i++, attrs++) {
1027                 chip->cc_attrs_tbl[i] = be32_to_cpup(attrs);
1028                 cc = chip->cc_attrs_tbl[i] & 0xFFFF;
1029
1030                 if (cc == TPM2_CC_CONTEXT_SAVE || cc == TPM2_CC_FLUSH_CONTEXT) {
1031                         chip->cc_attrs_tbl[i] &=
1032                                 ~(GENMASK(2, 0) << TPM2_CC_ATTR_CHANDLES);
1033                         chip->cc_attrs_tbl[i] |= 1 << TPM2_CC_ATTR_CHANDLES;
1034                 }
1035         }
1036
1037         tpm_buf_destroy(&buf);
1038
1039 out:
1040         if (rc > 0)
1041                 rc = -ENODEV;
1042         return rc;
1043 }
1044
1045 /**
1046  * tpm2_auto_startup - Perform the standard automatic TPM initialization
1047  *                     sequence
1048  * @chip: TPM chip to use
1049  *
1050  * Returns 0 on success, < 0 in case of fatal error.
1051  */
1052 int tpm2_auto_startup(struct tpm_chip *chip)
1053 {
1054         int rc;
1055
1056         rc = tpm_get_timeouts(chip);
1057         if (rc)
1058                 goto out;
1059
1060         rc = tpm2_do_selftest(chip);
1061         if (rc != 0 && rc != TPM2_RC_INITIALIZE) {
1062                 dev_err(&chip->dev, "TPM self test failed\n");
1063                 goto out;
1064         }
1065
1066         if (rc == TPM2_RC_INITIALIZE) {
1067                 rc = tpm_startup(chip);
1068                 if (rc)
1069                         goto out;
1070
1071                 rc = tpm2_do_selftest(chip);
1072                 if (rc) {
1073                         dev_err(&chip->dev, "TPM self test failed\n");
1074                         goto out;
1075                 }
1076         }
1077
1078         rc = tpm2_get_pcr_allocation(chip);
1079         if (rc)
1080                 goto out;
1081
1082         rc = tpm2_get_cc_attrs_tbl(chip);
1083
1084 out:
1085         if (rc > 0)
1086                 rc = -ENODEV;
1087         return rc;
1088 }
1089
1090 int tpm2_find_cc(struct tpm_chip *chip, u32 cc)
1091 {
1092         int i;
1093
1094         for (i = 0; i < chip->nr_commands; i++)
1095                 if (cc == (chip->cc_attrs_tbl[i] & GENMASK(15, 0)))
1096                         return i;
1097
1098         return -1;
1099 }