common/cmd_tpm_test.c

   1 /*
   2  * Copyright (c) 2015 Google, Inc
   3  *
   4  * SPDX-License-Identifier:     GPL-2.0+
   5  */
   6
   7 #include <common.h>
   8 #include <command.h>
   9 #include <environment.h>
  10 #include <tpm.h>
  11
  12 /* Prints error and returns on failure */
  13 #define TPM_CHECK(tpm_command) do { \
  14         uint32_t result; \
  15         \
  16         result = (tpm_command); \
  17         if (result != TPM_SUCCESS) { \
  18                 printf("TEST FAILED: line %d: " #tpm_command ": 0x%x\n", \
  19                         __LINE__, result); \
  20                 return result; \
  21         } \
  22 } while (0)
  23
  24 #define INDEX0                  0xda70
  25 #define INDEX1                  0xda71
  26 #define INDEX2                  0xda72
  27 #define INDEX3                  0xda73
  28 #define INDEX_INITIALISED       0xda80
  29 #define PHYS_PRESENCE           4
  30 #define PRESENCE                8
  31
  32 static uint32_t TlclStartupIfNeeded(void)
  33 {
  34         uint32_t result = tpm_startup(TPM_ST_CLEAR);
  35
  36         return result == TPM_INVALID_POSTINIT ? TPM_SUCCESS : result;
  37 }
  38
  39 static int test_timer(void)
  40 {
  41         printf("get_timer(0) = %lu\n", get_timer(0));
  42         return 0;
  43 }
  44
  45 static uint32_t tpm_get_flags(uint8_t *disable, uint8_t *deactivated,
  46                               uint8_t *nvlocked)
  47 {
  48         struct tpm_permanent_flags pflags;
  49         uint32_t result;
  50
  51         result = tpm_get_permanent_flags(&pflags);
  52         if (result)
  53                 return result;
  54         if (disable)
  55                 *disable = pflags.disable;
  56         if (deactivated)
  57                 *deactivated = pflags.deactivated;
  58         if (nvlocked)
  59                 *nvlocked = pflags.nv_locked;
  60         debug("TPM: Got flags disable=%d, deactivated=%d, nvlocked=%d\n",
  61               pflags.disable, pflags.deactivated, pflags.nv_locked);
  62
  63         return 0;
  64 }
  65
  66 static uint32_t tpm_set_global_lock(void)
  67 {
  68         uint32_t x;
  69
  70         debug("TPM: Set global lock\n");
  71         return tpm_nv_write_value(INDEX0, (uint8_t *)&x, 0);
  72 }
  73
  74 static uint32_t tpm_nv_write_value_lock(uint32_t index)
  75 {
  76         debug("TPM: Write lock 0x%x\n", index);
  77
  78         return tpm_nv_write_value(index, NULL, 0);
  79 }
  80
  81 static uint32_t tpm_nv_set_locked(void)
  82 {
  83         debug("TPM: Set NV locked\n");
  84
  85         return tpm_nv_define_space(TPM_NV_INDEX_LOCK, 0, 0);
  86 }
  87
  88 static int tpm_is_owned(void)
  89 {
  90         uint8_t response[TPM_PUBEK_SIZE];
  91         uint32_t result;
  92
  93         result = tpm_read_pubek(response, sizeof(response));
  94
  95         return result != TPM_SUCCESS;
  96 }
  97
  98 static int test_early_extend(void)
  99 {
 100         uint8_t value_in[20];
 101         uint8_t value_out[20];
 102
 103         printf("Testing earlyextend ...");
 104         tpm_init();
 105         TPM_CHECK(tpm_startup(TPM_ST_CLEAR));
 106         TPM_CHECK(tpm_continue_self_test());
 107         TPM_CHECK(tpm_extend(1, value_in, value_out));
 108         printf("done\n");
 109         return 0;
 110 }
 111
 112 static int test_early_nvram(void)
 113 {
 114         uint32_t x;
 115
 116         printf("Testing earlynvram ...");
 117         tpm_init();
 118         TPM_CHECK(tpm_startup(TPM_ST_CLEAR));
 119         TPM_CHECK(tpm_continue_self_test());
 120         TPM_CHECK(tpm_tsc_physical_presence(PRESENCE));
 121         TPM_CHECK(tpm_nv_read_value(INDEX0, (uint8_t *)&x, sizeof(x)));
 122         printf("done\n");
 123         return 0;
 124 }
 125
 126 static int test_early_nvram2(void)
 127 {
 128         uint32_t x;
 129
 130         printf("Testing earlynvram2 ...");
 131         tpm_init();
 132         TPM_CHECK(tpm_startup(TPM_ST_CLEAR));
 133         TPM_CHECK(tpm_continue_self_test());
 134         TPM_CHECK(tpm_tsc_physical_presence(PRESENCE));
 135         TPM_CHECK(tpm_nv_write_value(INDEX0, (uint8_t *)&x, sizeof(x)));
 136         printf("done\n");
 137         return 0;
 138 }
 139
 140 static int test_enable(void)
 141 {
 142         uint8_t disable = 0, deactivated = 0;
 143
 144         printf("Testing enable ...\n");
 145         tpm_init();
 146         TPM_CHECK(TlclStartupIfNeeded());
 147         TPM_CHECK(tpm_self_test_full());
 148         TPM_CHECK(tpm_tsc_physical_presence(PRESENCE));
 149         TPM_CHECK(tpm_get_flags(&disable, &deactivated, NULL));
 150         printf("\tdisable is %d, deactivated is %d\n", disable, deactivated);
 151         TPM_CHECK(tpm_physical_enable());
 152         TPM_CHECK(tpm_physical_set_deactivated(0));
 153         TPM_CHECK(tpm_get_flags(&disable, &deactivated, NULL));
 154         printf("\tdisable is %d, deactivated is %d\n", disable, deactivated);
 155         if (disable == 1 || deactivated == 1)
 156                 printf("\tfailed to enable or activate\n");
 157         printf("\tdone\n");
 158         return 0;
 159 }
 160
 161 #define reboot() do { \
 162         printf("\trebooting...\n"); \
 163         reset_cpu(0); \
 164 } while (0)
 165
 166 static int test_fast_enable(void)
 167 {
 168         uint8_t disable = 0, deactivated = 0;
 169         int i;
 170
 171         printf("Testing fastenable ...\n");
 172         tpm_init();
 173         TPM_CHECK(TlclStartupIfNeeded());
 174         TPM_CHECK(tpm_self_test_full());
 175         TPM_CHECK(tpm_tsc_physical_presence(PRESENCE));
 176         TPM_CHECK(tpm_get_flags(&disable, &deactivated, NULL));
 177         printf("\tdisable is %d, deactivated is %d\n", disable, deactivated);
 178         for (i = 0; i < 2; i++) {
 179                 TPM_CHECK(tpm_force_clear());
 180                 TPM_CHECK(tpm_get_flags(&disable, &deactivated, NULL));
 181                 printf("\tdisable is %d, deactivated is %d\n", disable,
 182                        deactivated);
 183                 assert(disable == 1 && deactivated == 1);
 184                 TPM_CHECK(tpm_physical_enable());
 185                 TPM_CHECK(tpm_physical_set_deactivated(0));
 186                 TPM_CHECK(tpm_get_flags(&disable, &deactivated, NULL));
 187                 printf("\tdisable is %d, deactivated is %d\n", disable,
 188                        deactivated);
 189                 assert(disable == 0 && deactivated == 0);
 190         }
 191         printf("\tdone\n");
 192         return 0;
 193 }
 194
 195 static int test_global_lock(void)
 196 {
 197         uint32_t zero = 0;
 198         uint32_t result;
 199         uint32_t x;
 200
 201         printf("Testing globallock ...\n");
 202         tpm_init();
 203         TPM_CHECK(TlclStartupIfNeeded());
 204         TPM_CHECK(tpm_self_test_full());
 205         TPM_CHECK(tpm_tsc_physical_presence(PRESENCE));
 206         TPM_CHECK(tpm_nv_read_value(INDEX0, (uint8_t *)&x, sizeof(x)));
 207         TPM_CHECK(tpm_nv_write_value(INDEX0, (uint8_t *)&zero,
 208                                      sizeof(uint32_t)));
 209         TPM_CHECK(tpm_nv_read_value(INDEX1, (uint8_t *)&x, sizeof(x)));
 210         TPM_CHECK(tpm_nv_write_value(INDEX1, (uint8_t *)&zero,
 211                                      sizeof(uint32_t)));
 212         TPM_CHECK(tpm_set_global_lock());
 213         /* Verifies that write to index0 fails */
 214         x = 1;
 215         result = tpm_nv_write_value(INDEX0, (uint8_t *)&x, sizeof(x));
 216         assert(result == TPM_AREA_LOCKED);
 217         TPM_CHECK(tpm_nv_read_value(INDEX0, (uint8_t *)&x, sizeof(x)));
 218         assert(x == 0);
 219         /* Verifies that write to index1 is still possible */
 220         x = 2;
 221         TPM_CHECK(tpm_nv_write_value(INDEX1, (uint8_t *)&x, sizeof(x)));
 222         TPM_CHECK(tpm_nv_read_value(INDEX1, (uint8_t *)&x, sizeof(x)));
 223         assert(x == 2);
 224         /* Turns off PP */
 225         tpm_tsc_physical_presence(PHYS_PRESENCE);
 226         /* Verifies that write to index1 fails */
 227         x = 3;
 228         result = tpm_nv_write_value(INDEX1, (uint8_t *)&x, sizeof(x));
 229         assert(result == TPM_BAD_PRESENCE);
 230         TPM_CHECK(tpm_nv_read_value(INDEX1, (uint8_t *)&x, sizeof(x)));
 231         assert(x == 2);
 232         printf("\tdone\n");
 233         return 0;
 234 }
 235
 236 static int test_lock(void)
 237 {
 238         printf("Testing lock ...\n");
 239         tpm_init();
 240         tpm_startup(TPM_ST_CLEAR);
 241         tpm_self_test_full();
 242         tpm_tsc_physical_presence(PRESENCE);
 243         tpm_nv_write_value_lock(INDEX0);
 244         printf("\tLocked 0x%x\n", INDEX0);
 245         printf("\tdone\n");
 246         return 0;
 247 }
 248
 249 static void initialise_spaces(void)
 250 {
 251         uint32_t zero = 0;
 252         uint32_t perm = TPM_NV_PER_WRITE_STCLEAR | TPM_NV_PER_PPWRITE;
 253
 254         printf("\tInitialising spaces\n");
 255         tpm_nv_set_locked();  /* useful only the first time */
 256         tpm_nv_define_space(INDEX0, perm, 4);
 257         tpm_nv_write_value(INDEX0, (uint8_t *)&zero, 4);
 258         tpm_nv_define_space(INDEX1, perm, 4);
 259         tpm_nv_write_value(INDEX1, (uint8_t *)&zero, 4);
 260         tpm_nv_define_space(INDEX2, perm, 4);
 261         tpm_nv_write_value(INDEX2, (uint8_t *)&zero, 4);
 262         tpm_nv_define_space(INDEX3, perm, 4);
 263         tpm_nv_write_value(INDEX3, (uint8_t *)&zero, 4);
 264         perm = TPM_NV_PER_READ_STCLEAR | TPM_NV_PER_WRITE_STCLEAR |
 265                 TPM_NV_PER_PPWRITE;
 266         tpm_nv_define_space(INDEX_INITIALISED, perm, 1);
 267 }
 268
 269 static int test_readonly(void)
 270 {
 271         uint8_t c;
 272         uint32_t index_0, index_1, index_2, index_3;
 273         int read0, read1, read2, read3;
 274
 275         printf("Testing readonly ...\n");
 276         tpm_init();
 277         tpm_startup(TPM_ST_CLEAR);
 278         tpm_self_test_full();
 279         tpm_tsc_physical_presence(PRESENCE);
 280         /*
 281          * Checks if initialisation has completed by trying to read-lock a
 282          * space that's created at the end of initialisation
 283          */
 284         if (tpm_nv_read_value(INDEX_INITIALISED, &c, 0) == TPM_BADINDEX) {
 285                 /* The initialisation did not complete */
 286                 initialise_spaces();
 287         }
 288
 289         /* Checks if spaces are OK or messed up */
 290         read0 = tpm_nv_read_value(INDEX0, (uint8_t *)&index_0, sizeof(index_0));
 291         read1 = tpm_nv_read_value(INDEX1, (uint8_t *)&index_1, sizeof(index_1));
 292         read2 = tpm_nv_read_value(INDEX2, (uint8_t *)&index_2, sizeof(index_2));
 293         read3 = tpm_nv_read_value(INDEX3, (uint8_t *)&index_3, sizeof(index_3));
 294         if (read0 || read1 || read2 || read3) {
 295                 printf("Invalid contents\n");
 296                 return 0;
 297         }
 298
 299         /*
 300          * Writes space, and locks it.  Then attempts to write again.
 301          * I really wish I could use the imperative.
 302          */
 303         index_0 += 1;
 304         if (tpm_nv_write_value(INDEX0, (uint8_t *)&index_0, sizeof(index_0) !=
 305                 TPM_SUCCESS)) {
 306                 error("\tcould not write index 0\n");
 307         }
 308         tpm_nv_write_value_lock(INDEX0);
 309         if (tpm_nv_write_value(INDEX0, (uint8_t *)&index_0, sizeof(index_0)) ==
 310                         TPM_SUCCESS)
 311                 error("\tindex 0 is not locked\n");
 312
 313         printf("\tdone\n");
 314         return 0;
 315 }
 316
 317 static int test_redefine_unowned(void)
 318 {
 319         uint32_t perm;
 320         uint32_t result;
 321         uint32_t x;
 322
 323         printf("Testing redefine_unowned ...");
 324         tpm_init();
 325         TPM_CHECK(TlclStartupIfNeeded());
 326         TPM_CHECK(tpm_self_test_full());
 327         TPM_CHECK(tpm_tsc_physical_presence(PRESENCE));
 328         assert(!tpm_is_owned());
 329
 330         /* Ensures spaces exist. */
 331         TPM_CHECK(tpm_nv_read_value(INDEX0, (uint8_t *)&x, sizeof(x)));
 332         TPM_CHECK(tpm_nv_read_value(INDEX1, (uint8_t *)&x, sizeof(x)));
 333
 334         /* Redefines spaces a couple of times. */
 335         perm = TPM_NV_PER_PPWRITE | TPM_NV_PER_GLOBALLOCK;
 336         TPM_CHECK(tpm_nv_define_space(INDEX0, perm, 2 * sizeof(uint32_t)));
 337         TPM_CHECK(tpm_nv_define_space(INDEX0, perm, sizeof(uint32_t)));
 338         perm = TPM_NV_PER_PPWRITE;
 339         TPM_CHECK(tpm_nv_define_space(INDEX1, perm, 2 * sizeof(uint32_t)));
 340         TPM_CHECK(tpm_nv_define_space(INDEX1, perm, sizeof(uint32_t)));
 341
 342         /* Sets the global lock */
 343         tpm_set_global_lock();
 344
 345         /* Verifies that index0 cannot be redefined */
 346         result = tpm_nv_define_space(INDEX0, perm, sizeof(uint32_t));
 347         assert(result == TPM_AREA_LOCKED);
 348
 349         /* Checks that index1 can */
 350         TPM_CHECK(tpm_nv_define_space(INDEX1, perm, 2 * sizeof(uint32_t)));
 351         TPM_CHECK(tpm_nv_define_space(INDEX1, perm, sizeof(uint32_t)));
 352
 353         /* Turns off PP */
 354         tpm_tsc_physical_presence(PHYS_PRESENCE);
 355
 356         /* Verifies that neither index0 nor index1 can be redefined */
 357         result = tpm_nv_define_space(INDEX0, perm, sizeof(uint32_t));
 358         assert(result == TPM_BAD_PRESENCE);
 359         result = tpm_nv_define_space(INDEX1, perm, sizeof(uint32_t));
 360         assert(result == TPM_BAD_PRESENCE);
 361
 362         printf("done\n");
 363         return 0;
 364 }
 365
 366 #define PERMPPGL (TPM_NV_PER_PPWRITE | TPM_NV_PER_GLOBALLOCK)
 367 #define PERMPP TPM_NV_PER_PPWRITE
 368
 369 static int test_space_perm(void)
 370 {
 371         uint32_t perm;
 372
 373         printf("Testing spaceperm ...");
 374         tpm_init();
 375         TPM_CHECK(TlclStartupIfNeeded());
 376         TPM_CHECK(tpm_continue_self_test());
 377         TPM_CHECK(tpm_tsc_physical_presence(PRESENCE));
 378         TPM_CHECK(tpm_get_permissions(INDEX0, &perm));
 379         assert((perm & PERMPPGL) == PERMPPGL);
 380         TPM_CHECK(tpm_get_permissions(INDEX1, &perm));
 381         assert((perm & PERMPP) == PERMPP);
 382         printf("done\n");
 383         return 0;
 384 }
 385
 386 static int test_startup(void)
 387 {
 388         uint32_t result;
 389         printf("Testing startup ...\n");
 390
 391         tpm_init();
 392         result = tpm_startup(TPM_ST_CLEAR);
 393         if (result != 0 && result != TPM_INVALID_POSTINIT)
 394                 printf("\ttpm startup failed with 0x%x\n", result);
 395         result = tpm_get_flags(NULL, NULL, NULL);
 396         if (result != 0)
 397                 printf("\ttpm getflags failed with 0x%x\n", result);
 398         printf("\texecuting SelfTestFull\n");
 399         tpm_self_test_full();
 400         result = tpm_get_flags(NULL, NULL, NULL);
 401         if (result != 0)
 402                 printf("\ttpm getflags failed with 0x%x\n", result);
 403         printf("\tdone\n");
 404         return 0;
 405 }
 406
 407 /*
 408  * Runs [op] and ensures it returns success and doesn't run longer than
 409  * [time_limit] in milliseconds.
 410  */
 411 #define TTPM_CHECK(op, time_limit) do { \
 412         ulong start, time; \
 413         uint32_t __result; \
 414         \
 415         start = get_timer(0); \
 416         __result = op; \
 417         if (__result != TPM_SUCCESS) { \
 418                 printf("\t" #op ": error 0x%x\n", __result); \
 419                 return -1; \
 420         } \
 421         time = get_timer(start); \
 422         printf("\t" #op ": %lu ms\n", time); \
 423         if (time > (ulong)time_limit) { \
 424                 printf("\t" #op " exceeded " #time_limit " ms\n"); \
 425         } \
 426 } while (0)
 427
 428
 429 static int test_timing(void)
 430 {
 431         uint32_t x;
 432         uint8_t in[20], out[20];
 433
 434         printf("Testing timing ...");
 435         tpm_init();
 436         TTPM_CHECK(TlclStartupIfNeeded(), 50);
 437         TTPM_CHECK(tpm_continue_self_test(), 100);
 438         TTPM_CHECK(tpm_self_test_full(), 1000);
 439         TTPM_CHECK(tpm_tsc_physical_presence(PRESENCE), 100);
 440         TTPM_CHECK(tpm_nv_write_value(INDEX0, (uint8_t *)&x, sizeof(x)), 100);
 441         TTPM_CHECK(tpm_nv_read_value(INDEX0, (uint8_t *)&x, sizeof(x)), 100);
 442         TTPM_CHECK(tpm_extend(0, in, out), 200);
 443         TTPM_CHECK(tpm_set_global_lock(), 50);
 444         TTPM_CHECK(tpm_tsc_physical_presence(PHYS_PRESENCE), 100);
 445         printf("done\n");
 446         return 0;
 447 }
 448
 449 #define TPM_MAX_NV_WRITES_NOOWNER 64
 450
 451 static int test_write_limit(void)
 452 {
 453         printf("Testing writelimit ...\n");
 454         int i;
 455         uint32_t result;
 456
 457         tpm_init();
 458         TPM_CHECK(TlclStartupIfNeeded());
 459         TPM_CHECK(tpm_self_test_full());
 460         TPM_CHECK(tpm_tsc_physical_presence(PRESENCE));
 461         TPM_CHECK(tpm_force_clear());
 462         TPM_CHECK(tpm_physical_enable());
 463         TPM_CHECK(tpm_physical_set_deactivated(0));
 464
 465         for (i = 0; i < TPM_MAX_NV_WRITES_NOOWNER + 2; i++) {
 466                 printf("\twriting %d\n", i);
 467                 result = tpm_nv_write_value(INDEX0, (uint8_t *)&i, sizeof(i));
 468                 switch (result) {
 469                 case TPM_SUCCESS:
 470                         break;
 471                 case TPM_MAXNVWRITES:
 472                         assert(i >= TPM_MAX_NV_WRITES_NOOWNER);
 473                 default:
 474                         error("\tunexpected error code %d (0x%x)\n",
 475                               result, result);
 476                 }
 477         }
 478
 479         /* Reset write count */
 480         TPM_CHECK(tpm_force_clear());
 481         TPM_CHECK(tpm_physical_enable());
 482         TPM_CHECK(tpm_physical_set_deactivated(0));
 483
 484         /* Try writing again. */
 485         TPM_CHECK(tpm_nv_write_value(INDEX0, (uint8_t *)&i, sizeof(i)));
 486         printf("\tdone\n");
 487         return 0;
 488 }
 489
 490 #define VOIDTEST(XFUNC) \
 491         int do_test_##XFUNC(cmd_tbl_t *cmd_tbl, int flag, int argc, \
 492         char * const argv[]) \
 493         { \
 494                 return test_##XFUNC(); \
 495         }
 496
 497 #define VOIDENT(XNAME) \
 498         U_BOOT_CMD_MKENT(XNAME, 0, 1, do_test_##XNAME, "", ""),
 499
 500 VOIDTEST(early_extend)
 501 VOIDTEST(early_nvram)
 502 VOIDTEST(early_nvram2)
 503 VOIDTEST(enable)
 504 VOIDTEST(fast_enable)
 505 VOIDTEST(global_lock)
 506 VOIDTEST(lock)
 507 VOIDTEST(readonly)
 508 VOIDTEST(redefine_unowned)
 509 VOIDTEST(space_perm)
 510 VOIDTEST(startup)
 511 VOIDTEST(timing)
 512 VOIDTEST(write_limit)
 513 VOIDTEST(timer)
 514
 515 static cmd_tbl_t cmd_cros_tpm_sub[] = {
 516         VOIDENT(early_extend)
 517         VOIDENT(early_nvram)
 518         VOIDENT(early_nvram2)
 519         VOIDENT(enable)
 520         VOIDENT(fast_enable)
 521         VOIDENT(global_lock)
 522         VOIDENT(lock)
 523         VOIDENT(readonly)
 524         VOIDENT(redefine_unowned)
 525         VOIDENT(space_perm)
 526         VOIDENT(startup)
 527         VOIDENT(timing)
 528         VOIDENT(write_limit)
 529         VOIDENT(timer)
 530 };
 531
 532 static int do_tpmtest(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
 533 {
 534         cmd_tbl_t *c;
 535
 536         printf("argc = %d, argv = ", argc);
 537         do {
 538                 int i = 0;
 539
 540                 for (i = 0; i < argc; i++)
 541                         printf(" %s", argv[i]);
 542                         printf("\n------\n");
 543                 } while (0);
 544         argc--;
 545         argv++;
 546         c = find_cmd_tbl(argv[0], cmd_cros_tpm_sub,
 547                          ARRAY_SIZE(cmd_cros_tpm_sub));
 548         return c ? c->cmd(cmdtp, flag, argc, argv) : cmd_usage(cmdtp);
 549 }
 550
 551 U_BOOT_CMD(tpmtest, 2, 1, do_tpmtest, "TPM tests",
 552         "\n\tearly_extend\n"
 553         "\tearly_nvram\n"
 554         "\tearly_nvram2\n"
 555         "\tenable\n"
 556         "\tfast_enable\n"
 557         "\tglobal_lock\n"
 558         "\tlock\n"
 559         "\treadonly\n"
 560         "\tredefine_unowned\n"
 561         "\tspace_perm\n"
 562         "\tstartup\n"
 563         "\ttiming\n"
 564         "\twrite_limit\n");