Merge https://gitlab.denx.de/u-boot/custodians/u-boot-usb
[oweals/u-boot.git] / cmd / tpm-v2.c
index 38add4f4622c46946dbf313159dac877715dffd3..e6742656f5788d7146f10b799b419e96ef29bd0e 100644 (file)
@@ -5,6 +5,7 @@
  */
 
 #include <common.h>
+#include <command.h>
 #include <dm.h>
 #include <log.h>
 #include <mapmem.h>
 #include <tpm-v2.h>
 #include "tpm-user-utils.h"
 
-static int do_tpm2_startup(cmd_tbl_t *cmdtp, int flag, int argc,
-                          char * const argv[])
+static int do_tpm2_startup(struct cmd_tbl *cmdtp, int flag, int argc,
+                          char *const argv[])
 {
        enum tpm2_startup_types mode;
+       struct udevice *dev;
+       int ret;
 
+       ret = get_tpm(&dev);
+       if (ret)
+               return ret;
        if (argc != 2)
                return CMD_RET_USAGE;
 
@@ -29,14 +35,19 @@ static int do_tpm2_startup(cmd_tbl_t *cmdtp, int flag, int argc,
                return CMD_RET_FAILURE;
        }
 
-       return report_return_code(tpm2_startup(mode));
+       return report_return_code(tpm2_startup(dev, mode));
 }
 
-static int do_tpm2_self_test(cmd_tbl_t *cmdtp, int flag, int argc,
-                            char * const argv[])
+static int do_tpm2_self_test(struct cmd_tbl *cmdtp, int flag, int argc,
+                            char *const argv[])
 {
        enum tpm2_yes_no full_test;
+       struct udevice *dev;
+       int ret;
 
+       ret = get_tpm(&dev);
+       if (ret)
+               return ret;
        if (argc != 2)
                return CMD_RET_USAGE;
 
@@ -49,15 +60,21 @@ static int do_tpm2_self_test(cmd_tbl_t *cmdtp, int flag, int argc,
                return CMD_RET_FAILURE;
        }
 
-       return report_return_code(tpm2_self_test(full_test));
+       return report_return_code(tpm2_self_test(dev, full_test));
 }
 
-static int do_tpm2_clear(cmd_tbl_t *cmdtp, int flag, int argc,
-                        char * const argv[])
+static int do_tpm2_clear(struct cmd_tbl *cmdtp, int flag, int argc,
+                        char *const argv[])
 {
        u32 handle = 0;
        const char *pw = (argc < 3) ? NULL : argv[2];
        const ssize_t pw_sz = pw ? strlen(pw) : 0;
+       struct udevice *dev;
+       int ret;
+
+       ret = get_tpm(&dev);
+       if (ret)
+               return ret;
 
        if (argc < 2 || argc > 3)
                return CMD_RET_USAGE;
@@ -72,11 +89,11 @@ static int do_tpm2_clear(cmd_tbl_t *cmdtp, int flag, int argc,
        else
                return CMD_RET_USAGE;
 
-       return report_return_code(tpm2_clear(handle, pw, pw_sz));
+       return report_return_code(tpm2_clear(dev, handle, pw, pw_sz));
 }
 
-static int do_tpm2_pcr_extend(cmd_tbl_t *cmdtp, int flag, int argc,
-                             char * const argv[])
+static int do_tpm2_pcr_extend(struct cmd_tbl *cmdtp, int flag, int argc,
+                             char *const argv[])
 {
        struct udevice *dev;
        struct tpm_chip_priv *priv;
@@ -88,7 +105,7 @@ static int do_tpm2_pcr_extend(cmd_tbl_t *cmdtp, int flag, int argc,
        if (argc != 3)
                return CMD_RET_USAGE;
 
-       ret = uclass_first_device_err(UCLASS_TPM, &dev);
+       ret = get_tpm(&dev);
        if (ret)
                return ret;
 
@@ -99,15 +116,15 @@ static int do_tpm2_pcr_extend(cmd_tbl_t *cmdtp, int flag, int argc,
        if (index >= priv->pcr_count)
                return -EINVAL;
 
-       rc = tpm2_pcr_extend(index, digest);
+       rc = tpm2_pcr_extend(dev, index, digest);
 
        unmap_sysmem(digest);
 
        return report_return_code(rc);
 }
 
-static int do_tpm_pcr_read(cmd_tbl_t *cmdtp, int flag, int argc,
-                          char * const argv[])
+static int do_tpm_pcr_read(struct cmd_tbl *cmdtp, int flag, int argc,
+                          char *const argv[])
 {
        struct udevice *dev;
        struct tpm_chip_priv *priv;
@@ -119,7 +136,7 @@ static int do_tpm_pcr_read(cmd_tbl_t *cmdtp, int flag, int argc,
        if (argc != 3)
                return CMD_RET_USAGE;
 
-       ret = uclass_first_device_err(UCLASS_TPM, &dev);
+       ret = get_tpm(&dev);
        if (ret)
                return ret;
 
@@ -133,9 +150,9 @@ static int do_tpm_pcr_read(cmd_tbl_t *cmdtp, int flag, int argc,
 
        data = map_sysmem(simple_strtoul(argv[2], NULL, 0), 0);
 
-       rc = tpm2_pcr_read(index, priv->pcr_select_min, data, &updates);
+       rc = tpm2_pcr_read(dev, index, priv->pcr_select_min, data, &updates);
        if (!rc) {
-               printf("PCR #%u content (%d known updates):\n", index, updates);
+               printf("PCR #%u content (%u known updates):\n", index, updates);
                print_byte_string(data, TPM2_DIGEST_LEN);
        }
 
@@ -144,13 +161,19 @@ static int do_tpm_pcr_read(cmd_tbl_t *cmdtp, int flag, int argc,
        return report_return_code(rc);
 }
 
-static int do_tpm_get_capability(cmd_tbl_t *cmdtp, int flag, int argc,
-                                char * const argv[])
+static int do_tpm_get_capability(struct cmd_tbl *cmdtp, int flag, int argc,
+                                char *const argv[])
 {
        u32 capability, property, rc;
        u8 *data;
        size_t count;
        int i, j;
+       struct udevice *dev;
+       int ret;
+
+       ret = get_tpm(&dev);
+       if (ret)
+               return ret;
 
        if (argc != 5)
                return CMD_RET_USAGE;
@@ -160,7 +183,7 @@ static int do_tpm_get_capability(cmd_tbl_t *cmdtp, int flag, int argc,
        data = map_sysmem(simple_strtoul(argv[3], NULL, 0), 0);
        count = simple_strtoul(argv[4], NULL, 0);
 
-       rc = tpm2_get_capability(capability, property, data, count);
+       rc = tpm2_get_capability(dev, capability, property, data, count);
        if (rc)
                goto unmap_data;
 
@@ -181,11 +204,17 @@ unmap_data:
        return report_return_code(rc);
 }
 
-static int do_tpm_dam_reset(cmd_tbl_t *cmdtp, int flag, int argc,
+static int do_tpm_dam_reset(struct cmd_tbl *cmdtp, int flag, int argc,
                            char *const argv[])
 {
        const char *pw = (argc < 2) ? NULL : argv[1];
        const ssize_t pw_sz = pw ? strlen(pw) : 0;
+       struct udevice *dev;
+       int ret;
+
+       ret = get_tpm(&dev);
+       if (ret)
+               return ret;
 
        if (argc > 2)
                return CMD_RET_USAGE;
@@ -193,10 +222,10 @@ static int do_tpm_dam_reset(cmd_tbl_t *cmdtp, int flag, int argc,
        if (pw_sz > TPM2_DIGEST_LEN)
                return -EINVAL;
 
-       return report_return_code(tpm2_dam_reset(pw, pw_sz));
+       return report_return_code(tpm2_dam_reset(dev, pw, pw_sz));
 }
 
-static int do_tpm_dam_parameters(cmd_tbl_t *cmdtp, int flag, int argc,
+static int do_tpm_dam_parameters(struct cmd_tbl *cmdtp, int flag, int argc,
                                 char *const argv[])
 {
        const char *pw = (argc < 5) ? NULL : argv[4];
@@ -208,6 +237,12 @@ static int do_tpm_dam_parameters(cmd_tbl_t *cmdtp, int flag, int argc,
        unsigned long int max_tries;
        unsigned long int recovery_time;
        unsigned long int lockout_recovery;
+       struct udevice *dev;
+       int ret;
+
+       ret = get_tpm(&dev);
+       if (ret)
+               return ret;
 
        if (argc < 4 || argc > 5)
                return CMD_RET_USAGE;
@@ -229,12 +264,12 @@ static int do_tpm_dam_parameters(cmd_tbl_t *cmdtp, int flag, int argc,
        log(LOGC_NONE, LOGL_INFO, "- recoveryTime: %lu\n", recovery_time);
        log(LOGC_NONE, LOGL_INFO, "- lockoutRecovery: %lu\n", lockout_recovery);
 
-       return report_return_code(tpm2_dam_parameters(pw, pw_sz, max_tries,
+       return report_return_code(tpm2_dam_parameters(dev, pw, pw_sz, max_tries,
                                                      recovery_time,
                                                      lockout_recovery));
 }
 
-static int do_tpm_change_auth(cmd_tbl_t *cmdtp, int flag, int argc,
+static int do_tpm_change_auth(struct cmd_tbl *cmdtp, int flag, int argc,
                              char *const argv[])
 {
        u32 handle;
@@ -242,6 +277,12 @@ static int do_tpm_change_auth(cmd_tbl_t *cmdtp, int flag, int argc,
        const char *oldpw = (argc == 3) ? NULL : argv[3];
        const ssize_t newpw_sz = strlen(newpw);
        const ssize_t oldpw_sz = oldpw ? strlen(oldpw) : 0;
+       struct udevice *dev;
+       int ret;
+
+       ret = get_tpm(&dev);
+       if (ret)
+               return ret;
 
        if (argc < 3 || argc > 4)
                return CMD_RET_USAGE;
@@ -260,17 +301,23 @@ static int do_tpm_change_auth(cmd_tbl_t *cmdtp, int flag, int argc,
        else
                return CMD_RET_USAGE;
 
-       return report_return_code(tpm2_change_auth(handle, newpw, newpw_sz,
+       return report_return_code(tpm2_change_auth(dev, handle, newpw, newpw_sz,
                                                   oldpw, oldpw_sz));
 }
 
-static int do_tpm_pcr_setauthpolicy(cmd_tbl_t *cmdtp, int flag, int argc,
-                                   char * const argv[])
+static int do_tpm_pcr_setauthpolicy(struct cmd_tbl *cmdtp, int flag, int argc,
+                                   char *const argv[])
 {
        u32 index = simple_strtoul(argv[1], NULL, 0);
        char *key = argv[2];
        const char *pw = (argc < 4) ? NULL : argv[3];
        const ssize_t pw_sz = pw ? strlen(pw) : 0;
+       struct udevice *dev;
+       int ret;
+
+       ret = get_tpm(&dev);
+       if (ret)
+               return ret;
 
        if (strlen(key) != TPM2_DIGEST_LEN)
                return -EINVAL;
@@ -278,18 +325,24 @@ static int do_tpm_pcr_setauthpolicy(cmd_tbl_t *cmdtp, int flag, int argc,
        if (argc < 3 || argc > 4)
                return CMD_RET_USAGE;
 
-       return report_return_code(tpm2_pcr_setauthpolicy(pw, pw_sz, index,
+       return report_return_code(tpm2_pcr_setauthpolicy(dev, pw, pw_sz, index,
                                                         key));
 }
 
-static int do_tpm_pcr_setauthvalue(cmd_tbl_t *cmdtp, int flag,
-                                  int argc, char * const argv[])
+static int do_tpm_pcr_setauthvalue(struct cmd_tbl *cmdtp, int flag,
+                                  int argc, char *const argv[])
 {
        u32 index = simple_strtoul(argv[1], NULL, 0);
        char *key = argv[2];
        const ssize_t key_sz = strlen(key);
        const char *pw = (argc < 4) ? NULL : argv[3];
        const ssize_t pw_sz = pw ? strlen(pw) : 0;
+       struct udevice *dev;
+       int ret;
+
+       ret = get_tpm(&dev);
+       if (ret)
+               return ret;
 
        if (strlen(key) != TPM2_DIGEST_LEN)
                return -EINVAL;
@@ -297,11 +350,12 @@ static int do_tpm_pcr_setauthvalue(cmd_tbl_t *cmdtp, int flag,
        if (argc < 3 || argc > 4)
                return CMD_RET_USAGE;
 
-       return report_return_code(tpm2_pcr_setauthvalue(pw, pw_sz, index,
+       return report_return_code(tpm2_pcr_setauthvalue(dev, pw, pw_sz, index,
                                                        key, key_sz));
 }
 
-static cmd_tbl_t tpm2_commands[] = {
+static struct cmd_tbl tpm2_commands[] = {
+       U_BOOT_CMD_MKENT(device, 0, 1, do_tpm_device, "", ""),
        U_BOOT_CMD_MKENT(info, 0, 1, do_tpm_info, "", ""),
        U_BOOT_CMD_MKENT(init, 0, 1, do_tpm_init, "", ""),
        U_BOOT_CMD_MKENT(startup, 0, 1, do_tpm2_startup, "", ""),
@@ -319,16 +373,18 @@ static cmd_tbl_t tpm2_commands[] = {
                         do_tpm_pcr_setauthvalue, "", ""),
 };
 
-cmd_tbl_t *get_tpm_commands(unsigned int *size)
+struct cmd_tbl *get_tpm2_commands(unsigned int *size)
 {
        *size = ARRAY_SIZE(tpm2_commands);
 
        return tpm2_commands;
 }
 
-U_BOOT_CMD(tpm, CONFIG_SYS_MAXARGS, 1, do_tpm, "Issue a TPMv2.x command",
+U_BOOT_CMD(tpm2, CONFIG_SYS_MAXARGS, 1, do_tpm, "Issue a TPMv2.x command",
 "<command> [<arguments>]\n"
 "\n"
+"device [num device]\n"
+"    Show all devices or set the specified device\n"
 "info\n"
 "    Show information about the TPM.\n"
 "init\n"