Simon Glass | 3bb4db4 | 2022-08-30 21:05:36 -0600 | [diff] [blame] | 1 | // SPDX-License-Identifier: GPL-2.0+ |
| 2 | /* |
| 3 | * Copyright 2022 Google LLC |
| 4 | * Written by Simon Glass <sjg@chromium.org> |
| 5 | */ |
| 6 | |
| 7 | #include <common.h> |
| 8 | #include <dm.h> |
| 9 | #include <tpm_api.h> |
| 10 | #include <dm/test.h> |
| 11 | #include <test/test.h> |
| 12 | #include <test/ut.h> |
| 13 | |
Simon Glass | 4fef657 | 2023-02-21 06:24:51 -0700 | [diff] [blame^] | 14 | /* |
| 15 | * get_tpm_version() - Get a TPM of the given version |
| 16 | * |
| 17 | * @version: Version to get |
| 18 | * @devp: Returns the TPM device |
| 19 | * Returns: 0 if OK, -ENODEV if not found |
| 20 | */ |
| 21 | static int get_tpm_version(enum tpm_version version, struct udevice **devp) |
| 22 | { |
| 23 | struct udevice *dev; |
| 24 | |
| 25 | /* |
| 26 | * For now we have to probe each TPM, since the version is set up in |
| 27 | * of_to_plat(). We could require TPMs to declare their version when |
| 28 | * probed, to avoid this |
| 29 | */ |
| 30 | uclass_foreach_dev_probe(UCLASS_TPM, dev) { |
| 31 | if (tpm_get_version(dev) == version) { |
| 32 | *devp = dev; |
| 33 | return 0; |
| 34 | } |
| 35 | } |
| 36 | |
| 37 | return -ENODEV; |
| 38 | } |
| 39 | |
| 40 | /* Basic test of initing a TPM */ |
| 41 | static int test_tpm_init(struct unit_test_state *uts, enum tpm_version version) |
| 42 | { |
| 43 | struct udevice *dev; |
| 44 | |
| 45 | /* check probe success */ |
| 46 | ut_assertok(get_tpm_version(version, &dev)); |
| 47 | |
| 48 | ut_assertok(tpm_init(dev)); |
| 49 | |
| 50 | return 0; |
| 51 | } |
| 52 | |
Simon Glass | 3bb4db4 | 2022-08-30 21:05:36 -0600 | [diff] [blame] | 53 | static int dm_test_tpm(struct unit_test_state *uts) |
| 54 | { |
Simon Glass | 4fef657 | 2023-02-21 06:24:51 -0700 | [diff] [blame^] | 55 | ut_assertok(test_tpm_init(uts, TPM_V1)); |
| 56 | ut_assertok(test_tpm_init(uts, TPM_V2)); |
| 57 | |
| 58 | return 0; |
| 59 | } |
| 60 | DM_TEST(dm_test_tpm, UT_TESTF_SCAN_FDT); |
| 61 | |
| 62 | /* Test report_state */ |
| 63 | static int dm_test_tpm_report_state(struct unit_test_state *uts) |
| 64 | { |
Simon Glass | 3bb4db4 | 2022-08-30 21:05:36 -0600 | [diff] [blame] | 65 | struct udevice *dev; |
| 66 | char buf[50]; |
| 67 | |
| 68 | /* check probe success */ |
Simon Glass | 4fef657 | 2023-02-21 06:24:51 -0700 | [diff] [blame^] | 69 | ut_assertok(get_tpm_version(TPM_V2, &dev)); |
Simon Glass | 3bb4db4 | 2022-08-30 21:05:36 -0600 | [diff] [blame] | 70 | |
| 71 | ut_assert(tpm_report_state(dev, buf, sizeof(buf))); |
| 72 | ut_asserteq_str("init_done=0", buf); |
| 73 | |
Ilias Apalodimas | 1b11de7 | 2023-02-18 17:21:22 +0200 | [diff] [blame] | 74 | ut_assertok(tpm_auto_start(dev)); |
Simon Glass | 3bb4db4 | 2022-08-30 21:05:36 -0600 | [diff] [blame] | 75 | |
| 76 | ut_assert(tpm_report_state(dev, buf, sizeof(buf))); |
| 77 | ut_asserteq_str("init_done=1", buf); |
| 78 | |
| 79 | return 0; |
| 80 | } |
Simon Glass | 4fef657 | 2023-02-21 06:24:51 -0700 | [diff] [blame^] | 81 | DM_TEST(dm_test_tpm_report_state, UT_TESTF_SCAN_FDT); |