[u-boot.git] / cmd / part.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 <[email protected]>
 *
 * made from cmd_reiserfs by
 *
 * (C) Copyright 2003 - 2004
 * Sysgo Real-Time Solutions, AG <www.elinos.com>
 * Pavel Bartusek <[email protected]>
 *
 * SPDX-License-Identifier:	GPL-2.0+
 */

#include <common.h>
#include <config.h>
#include <command.h>
#include <part.h>
#include <vsprintf.h>

static int do_part_uuid(int argc, char * const argv[])
{
	int part;
	struct blk_desc *dev_desc;
	disk_partition_t info;

	if (argc < 2)
		return CMD_RET_USAGE;
	if (argc > 3)
		return CMD_RET_USAGE;

	part = blk_get_device_part_str(argv[0], argv[1], &dev_desc, &info, 0);
	if (part < 0)
		return 1;

	if (argc > 2)
		env_set(argv[2], info.uuid);
	else
		printf("%s\n", info.uuid);

	return 0;
}

static int do_part_list(int argc, char * const argv[])
{
	int ret;
	struct blk_desc *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 = blk_get_device_by_str(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 = part_get_info(desc, p, &info);

			if (r != 0)
				continue;

			if (bootable && !info.bootable)
				continue;

			sprintf(t, "%s%x", str[0] ? " " : "", p);
			strcat(str, t);
		}
		env_set(var, str);
		return 0;
	}

	part_print(desc);

	return 0;
}

static int do_part_start(int argc, char * const argv[])
{
	struct blk_desc *desc;
	disk_partition_t info;
	char buf[512] = { 0 };
	char *endp;
	int part;
	int err;
	int ret;

	if (argc < 3)
		return CMD_RET_USAGE;
	if (argc > 4)
		return CMD_RET_USAGE;

	ret = blk_get_device_by_str(argv[0], argv[1], &desc);
	if (ret < 0)
		return 1;

	part = simple_strtoul(argv[2], &endp, 0);
	if (*endp == '\0') {
		err = part_get_info(desc, part, &info);
		if (err)
			return 1;
	} else {
		part = part_get_info_by_name(desc, argv[2], &info);
		if (part == -1)
			return 1;
	}

	snprintf(buf, sizeof(buf), LBAF, info.start);

	if (argc > 3)
		env_set(argv[3], buf);
	else
		printf("%s\n", buf);

	return 0;
}

static int do_part_size(int argc, char * const argv[])
{
	struct blk_desc *desc;
	disk_partition_t info;
	char buf[512] = { 0 };
	char *endp;
	int part;
	int err;
	int ret;

	if (argc < 3)
		return CMD_RET_USAGE;
	if (argc > 4)
		return CMD_RET_USAGE;

	ret = blk_get_device_by_str(argv[0], argv[1], &desc);
	if (ret < 0)
		return 1;

	part = simple_strtoul(argv[2], &endp, 0);
	if (*endp == '\0') {
		err = part_get_info(desc, part, &info);
		if (err)
			return 1;
	} else {
		part = part_get_info_by_name(desc, argv[2], &info);
		if (part == -1)
			return 1;
	}

	snprintf(buf, sizeof(buf), LBAF, info.size);

	if (argc > 3)
		env_set(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 can be either partition number or partition name\n"
	"part size <interface> <dev> <part> <varname>\n"
	"    - set environment variable to the size of the partition (in blocks)\n"
	"      part can be either partition number or partition name"
);
Commit	Line	Data
5cf41dcc SW	1	/*
	2	* Copyright (c) 2012, NVIDIA CORPORATION. All rights reserved.
	3	*
	4	* made from cmd_ext2, which was:
	5	*
	6	* (C) Copyright 2004
	7	* esd gmbh <www.esd-electronics.com>
	8	* Reinhard Arlt <[email protected]>
	9	*
	10	* made from cmd_reiserfs by
	11	*
	12	* (C) Copyright 2003 - 2004
	13	* Sysgo Real-Time Solutions, AG <www.elinos.com>
	14	* Pavel Bartusek <[email protected]>
	15	*
1a459660	16	* SPDX-License-Identifier: GPL-2.0+
5cf41dcc SW	17	*/
	18
	19	#include <common.h>
	20	#include <config.h>
	21	#include <command.h>
	22	#include <part.h>
	23	#include <vsprintf.h>
	24
0e350f81	25	static int do_part_uuid(int argc, char * const argv[])
5cf41dcc SW	26	{
5cf41dcc SW	27	int part;
4101f687	28	struct blk_desc *dev_desc;
5cf41dcc SW	29	disk_partition_t info;
	30
	31	if (argc < 2)
	32	return CMD_RET_USAGE;
	33	if (argc > 3)
	34	return CMD_RET_USAGE;
	35
e35929e4	36	part = blk_get_device_part_str(argv[0], argv[1], &dev_desc, &info, 0);
5cf41dcc SW	37	if (part < 0)
	38	return 1;
	39
	40	if (argc > 2)
382bee57	41	env_set(argv[2], info.uuid);
5cf41dcc SW	42	else
	43	printf("%s\n", info.uuid);
	44
	45	return 0;
	46	}
	47
0e350f81	48	static int do_part_list(int argc, char * const argv[])
5cf41dcc SW	49	{
5cf41dcc SW	50	int ret;
4101f687	51	struct blk_desc *desc;
0798d6fd SS	52	char *var = NULL;
	53	bool bootable = false;
	54	int i;
5cf41dcc	55
0798d6fd	56	if (argc < 2)
5cf41dcc SW	57	return CMD_RET_USAGE;
5cf41dcc SW	58
0798d6fd SS	59	if (argc > 2) {
	60	for (i = 2; i < argc ; i++) {
	61	if (argv[i][0] == '-') {
	62	if (!strcmp(argv[i], "-bootable")) {
	63	bootable = true;
	64	} else {
	65	printf("Unknown option %s\n", argv[i]);
	66	return CMD_RET_USAGE;
	67	}
	68	} else {
	69	var = argv[i];
	70	break;
	71	}
	72	}
	73
	74	/* Loops should have been exited at the last argument, which
	75	* as it contained the variable */
	76	if (argc != i + 1)
	77	return CMD_RET_USAGE;
	78	}
	79
ebac37cf	80	ret = blk_get_device_by_str(argv[0], argv[1], &desc);
5cf41dcc SW	81	if (ret < 0)
	82	return 1;
	83
0798d6fd	84	if (var != NULL) {
e86df6ef	85	int p;
0798d6fd	86	char str[512] = { '\0', };
a78f78eb	87	disk_partition_t info;
e86df6ef SS	88
e86df6ef SS	89	for (p = 1; p < 128; p++) {
0798d6fd	90	char t[5];
3e8bd469	91	int r = part_get_info(desc, p, &info);
e86df6ef	92
0798d6fd SS	93	if (r != 0)
	94	continue;
	95
	96	if (bootable && !info.bootable)
	97	continue;
	98
653447b4	99	sprintf(t, "%s%x", str[0] ? " " : "", p);
0798d6fd	100	strcat(str, t);
e86df6ef	101	}
382bee57	102	env_set(var, str);
e86df6ef SS	103	return 0;
	104	}
	105
3e8bd469	106	part_print(desc);
5cf41dcc SW	107
	108	return 0;
	109	}
	110
8607c4f1 PK	111	static int do_part_start(int argc, char * const argv[])
8607c4f1 PK	112	{
4101f687	113	struct blk_desc *desc;
8607c4f1 PK	114	disk_partition_t info;
8607c4f1 PK	115	char buf[512] = { 0 };
36df616a	116	char *endp;
8607c4f1 PK	117	int part;
	118	int err;
	119	int ret;
	120
	121	if (argc < 3)
	122	return CMD_RET_USAGE;
	123	if (argc > 4)
	124	return CMD_RET_USAGE;
	125
ebac37cf	126	ret = blk_get_device_by_str(argv[0], argv[1], &desc);
8607c4f1 PK	127	if (ret < 0)
	128	return 1;
	129
36df616a SP	130	part = simple_strtoul(argv[2], &endp, 0);
	131	if (*endp == '\0') {
	132	err = part_get_info(desc, part, &info);
	133	if (err)
	134	return 1;
	135	} else {
	136	part = part_get_info_by_name(desc, argv[2], &info);
	137	if (part == -1)
	138	return 1;
	139	}
8607c4f1 PK	140
	141	snprintf(buf, sizeof(buf), LBAF, info.start);
	142
	143	if (argc > 3)
382bee57	144	env_set(argv[3], buf);
8607c4f1 PK	145	else
	146	printf("%s\n", buf);
	147
	148	return 0;
	149	}
	150
	151	static int do_part_size(int argc, char * const argv[])
	152	{
4101f687	153	struct blk_desc *desc;
8607c4f1 PK	154	disk_partition_t info;
8607c4f1 PK	155	char buf[512] = { 0 };
36df616a	156	char *endp;
8607c4f1 PK	157	int part;
	158	int err;
	159	int ret;
	160
	161	if (argc < 3)
	162	return CMD_RET_USAGE;
	163	if (argc > 4)
	164	return CMD_RET_USAGE;
	165
ebac37cf	166	ret = blk_get_device_by_str(argv[0], argv[1], &desc);
8607c4f1 PK	167	if (ret < 0)
	168	return 1;
	169
36df616a SP	170	part = simple_strtoul(argv[2], &endp, 0);
	171	if (*endp == '\0') {
	172	err = part_get_info(desc, part, &info);
	173	if (err)
	174	return 1;
	175	} else {
	176	part = part_get_info_by_name(desc, argv[2], &info);
	177	if (part == -1)
	178	return 1;
	179	}
8607c4f1 PK	180
	181	snprintf(buf, sizeof(buf), LBAF, info.size);
	182
	183	if (argc > 3)
382bee57	184	env_set(argv[3], buf);
8607c4f1 PK	185	else
	186	printf("%s\n", buf);
	187
	188	return 0;
	189	}
	190
0e350f81	191	static int do_part(cmd_tbl_t cmdtp, int flag, int argc, char const argv[])
5cf41dcc SW	192	{
	193	if (argc < 2)
	194	return CMD_RET_USAGE;
	195
	196	if (!strcmp(argv[1], "uuid"))
	197	return do_part_uuid(argc - 2, argv + 2);
	198	else if (!strcmp(argv[1], "list"))
	199	return do_part_list(argc - 2, argv + 2);
8607c4f1 PK	200	else if (!strcmp(argv[1], "start"))
	201	return do_part_start(argc - 2, argv + 2);
	202	else if (!strcmp(argv[1], "size"))
	203	return do_part_size(argc - 2, argv + 2);
5cf41dcc SW	204
	205	return CMD_RET_USAGE;
	206	}
	207
	208	U_BOOT_CMD(
0798d6fd	209	part, CONFIG_SYS_MAXARGS, 1, do_part,
5cf41dcc	210	"disk partition related commands",
8ca584ec	211	"uuid <interface> <dev>:<part>\n"
5cf41dcc SW	212	" - print partition UUID\n"
	213	"part uuid <interface> <dev>:<part> <varname>\n"
	214	" - set environment variable to partition UUID\n"
	215	"part list <interface> <dev>\n"
e86df6ef	216	" - print a device's partition table\n"
0798d6fd SS	217	"part list <interface> <dev> [flags] <varname>\n"
0798d6fd SS	218	" - set environment variable to the list of partitions\n"
8607c4f1 PK	219	" flags can be -bootable (list only bootable partitions)\n"
	220	"part start <interface> <dev> <part> <varname>\n"
	221	" - set environment variable to the start of the partition (in blocks)\n"
36df616a	222	" part can be either partition number or partition name\n"
8607c4f1	223	"part size <interface> <dev> <part> <varname>\n"
36df616a SP	224	" - set environment variable to the size of the partition (in blocks)\n"
36df616a SP	225	" part can be either partition number or partition name"
5cf41dcc	226	);