u-boot/common/cmd_part.c
Paul Kocialkowski 8607c4f127 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>
2015-06-19 16:46:47 -04:00

217 lines
4.4 KiB
C

/*
* Copyright (c) 2012, NVIDIA CORPORATION. All rights reserved.
*
* made from cmd_ext2, which was:
*
* (C) Copyright 2004
* esd gmbh <www.esd-electronics.com>
* Reinhard Arlt <reinhard.arlt@esd-electronics.com>
*
* made from cmd_reiserfs by
*
* (C) Copyright 2003 - 2004
* Sysgo Real-Time Solutions, AG <www.elinos.com>
* Pavel Bartusek <pba@sysgo.com>
*
* SPDX-License-Identifier: GPL-2.0+
*/
#include <common.h>
#include <config.h>
#include <command.h>
#include <part.h>
#include <vsprintf.h>
#ifndef CONFIG_PARTITION_UUIDS
#error CONFIG_PARTITION_UUIDS must be enabled for CONFIG_CMD_PART to be enabled
#endif
static int do_part_uuid(int argc, char * const argv[])
{
int part;
block_dev_desc_t *dev_desc;
disk_partition_t info;
if (argc < 2)
return CMD_RET_USAGE;
if (argc > 3)
return CMD_RET_USAGE;
part = get_device_and_partition(argv[0], argv[1], &dev_desc, &info, 0);
if (part < 0)
return 1;
if (argc > 2)
setenv(argv[2], info.uuid);
else
printf("%s\n", info.uuid);
return 0;
}
static int do_part_list(int argc, char * const argv[])
{
int ret;
block_dev_desc_t *desc;
char *var = NULL;
bool bootable = false;
int i;
if (argc < 2)
return CMD_RET_USAGE;
if (argc > 2) {
for (i = 2; i < argc ; i++) {
if (argv[i][0] == '-') {
if (!strcmp(argv[i], "-bootable")) {
bootable = true;
} else {
printf("Unknown option %s\n", argv[i]);
return CMD_RET_USAGE;
}
} else {
var = argv[i];
break;
}
}
/* Loops should have been exited at the last argument, which
* as it contained the variable */
if (argc != i + 1)
return CMD_RET_USAGE;
}
ret = get_device(argv[0], argv[1], &desc);
if (ret < 0)
return 1;
if (var != NULL) {
int p;
char str[512] = { '\0', };
disk_partition_t info;
for (p = 1; p < 128; p++) {
char t[5];
int r = get_partition_info(desc, p, &info);
if (r != 0)
continue;
if (bootable && !info.bootable)
continue;
sprintf(t, "%s%d", str[0] ? " " : "", p);
strcat(str, t);
}
setenv(var, str);
return 0;
}
print_part(desc);
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)
return CMD_RET_USAGE;
if (!strcmp(argv[1], "uuid"))
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;
}
U_BOOT_CMD(
part, CONFIG_SYS_MAXARGS, 1, do_part,
"disk partition related commands",
"uuid <interface> <dev>:<part>\n"
" - print partition UUID\n"
"part uuid <interface> <dev>:<part> <varname>\n"
" - set environment variable to partition UUID\n"
"part list <interface> <dev>\n"
" - 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)\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)"
);