common: cmd_part: start and size sub-commands introduction

This introduces the part start and part size sub-commands. The purpose of these
is to store the start block and size of a partition in a variable, given the
device and partition number.

This allows reading raw data that fits a single partition more easily.
For instance, this could be used to figure out the start block and size of a
kernel partition when a partition table is present, given the partition number.

Signed-off-by: Paul Kocialkowski <contact@paulk.fr>
Acked-by: Stephen Warren <swarren@nvidia.com>
[trini: Change "%lx" to LBAF]
Signed-off-by: Tom Rini <trini@konsulko.com>

diff --git a/common/cmd_part.c b/common/cmd_part.c
index 4bdbf901684979711d156f99b51339e4de1c4a4d..b860624d939cad2d6b364e4af544112f56a7203c 100644 (file)
--- a/common/cmd_part.c
+++ b/common/cmd_part.c
@@ -112,6 +112,74 @@ static int do_part_list(int argc, char * const argv[])
        return 0;
 }
 
+static int do_part_start(int argc, char * const argv[])
+{
+       block_dev_desc_t *desc;
+       disk_partition_t info;
+       char buf[512] = { 0 };
+       int part;
+       int err;
+       int ret;
+
+       if (argc < 3)
+               return CMD_RET_USAGE;
+       if (argc > 4)
+               return CMD_RET_USAGE;
+
+       part = simple_strtoul(argv[2], NULL, 0);
+
+       ret = get_device(argv[0], argv[1], &desc);
+       if (ret < 0)
+               return 1;
+
+       err = get_partition_info(desc, part, &info);
+       if (err)
+               return 1;
+
+       snprintf(buf, sizeof(buf), LBAF, info.start);
+
+       if (argc > 3)
+               setenv(argv[3], buf);
+       else
+               printf("%s\n", buf);
+
+       return 0;
+}
+
+static int do_part_size(int argc, char * const argv[])
+{
+       block_dev_desc_t *desc;
+       disk_partition_t info;
+       char buf[512] = { 0 };
+       int part;
+       int err;
+       int ret;
+
+       if (argc < 3)
+               return CMD_RET_USAGE;
+       if (argc > 4)
+               return CMD_RET_USAGE;
+
+       part = simple_strtoul(argv[2], NULL, 0);
+
+       ret = get_device(argv[0], argv[1], &desc);
+       if (ret < 0)
+               return 1;
+
+       err = get_partition_info(desc, part, &info);
+       if (err)
+               return 1;
+
+       snprintf(buf, sizeof(buf), LBAF, info.size);
+
+       if (argc > 3)
+               setenv(argv[3], buf);
+       else
+               printf("%s\n", buf);
+
+       return 0;
+}
+
 static int do_part(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
 {
        if (argc < 2)
@@ -121,6 +189,10 @@ static int do_part(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
                return do_part_uuid(argc - 2, argv + 2);
        else if (!strcmp(argv[1], "list"))
                return do_part_list(argc - 2, argv + 2);
+       else if (!strcmp(argv[1], "start"))
+               return do_part_start(argc - 2, argv + 2);
+       else if (!strcmp(argv[1], "size"))
+               return do_part_size(argc - 2, argv + 2);
 
        return CMD_RET_USAGE;
 }
@@ -136,5 +208,9 @@ U_BOOT_CMD(
        "    - print a device's partition table\n"
        "part list <interface> <dev> [flags] <varname>\n"
        "    - set environment variable to the list of partitions\n"
-       "      flags can be -bootable (list only bootable partitions)"
+       "      flags can be -bootable (list only bootable partitions)\n"
+       "part start <interface> <dev> <part> <varname>\n"
+       "    - set environment variable to the start of the partition (in blocks)\n"
+       "part size <interface> <dev> <part> <varname>\n"
+       "    - set environment variable to the size of the partition (in blocks)"
 );