[people/ms/u-boot.git] / disk / part.c

/*
 * (C) Copyright 2001
 * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
 *
 * SPDX-License-Identifier:	GPL-2.0+
 */

#include <common.h>
#include <command.h>
#include <errno.h>
#include <ide.h>
#include <malloc.h>
#include <part.h>
#include <ubifs_uboot.h>

#undef	PART_DEBUG

#ifdef	PART_DEBUG
#define	PRINTF(fmt,args...)	printf (fmt ,##args)
#else
#define PRINTF(fmt,args...)
#endif

DECLARE_GLOBAL_DATA_PTR;

#ifdef HAVE_BLOCK_DEVICE
static struct part_driver *part_driver_lookup_type(int part_type)
{
	struct part_driver *drv =
		ll_entry_start(struct part_driver, part_driver);
	const int n_ents = ll_entry_count(struct part_driver, part_driver);
	struct part_driver *entry;

	for (entry = drv; entry != drv + n_ents; entry++) {
		if (part_type == entry->part_type)
			return entry;
	}

	/* Not found */
	return NULL;
}

static struct blk_desc *get_dev_hwpart(const char *ifname, int dev, int hwpart)
{
	struct blk_desc *dev_desc;
	int ret;

	dev_desc = blk_get_devnum_by_typename(ifname, dev);
	if (!dev_desc) {
		debug("%s: No device for iface '%s', dev %d\n", __func__,
		      ifname, dev);
		return NULL;
	}
	ret = blk_dselect_hwpart(dev_desc, hwpart);
	if (ret) {
		debug("%s: Failed to select h/w partition: err-%d\n", __func__,
		      ret);
		return NULL;
	}

	return dev_desc;
}

struct blk_desc *blk_get_dev(const char *ifname, int dev)
{
	return get_dev_hwpart(ifname, dev, 0);
}
#else
struct blk_desc *get_dev_hwpart(const char *ifname, int dev, int hwpart)
{
	return NULL;
}

struct blk_desc *blk_get_dev(const char *ifname, int dev)
{
	return NULL;
}
#endif

#ifdef HAVE_BLOCK_DEVICE

/* ------------------------------------------------------------------------- */
/*
 * reports device info to the user
 */

#ifdef CONFIG_LBA48
typedef uint64_t lba512_t;
#else
typedef lbaint_t lba512_t;
#endif

/*
 * Overflowless variant of (block_count * mul_by / div_by)
 * when div_by > mul_by
 */
static lba512_t lba512_muldiv(lba512_t block_count, lba512_t mul_by, lba512_t div_by)
{
	lba512_t bc_quot, bc_rem;

	/* x * m / d == x / d * m + (x % d) * m / d */
	bc_quot = block_count / div_by;
	bc_rem  = block_count - div_by * bc_quot;
	return bc_quot * mul_by + (bc_rem * mul_by) / div_by;
}

void dev_print (struct blk_desc *dev_desc)
{
	lba512_t lba512; /* number of blocks if 512bytes block size */

	if (dev_desc->type == DEV_TYPE_UNKNOWN) {
		puts ("not available\n");
		return;
	}

	switch (dev_desc->if_type) {
	case IF_TYPE_SCSI:
		printf ("(%d:%d) Vendor: %s Prod.: %s Rev: %s\n",
			dev_desc->target,dev_desc->lun,
			dev_desc->vendor,
			dev_desc->product,
			dev_desc->revision);
		break;
	case IF_TYPE_ATAPI:
	case IF_TYPE_IDE:
	case IF_TYPE_SATA:
		printf ("Model: %s Firm: %s Ser#: %s\n",
			dev_desc->vendor,
			dev_desc->revision,
			dev_desc->product);
		break;
	case IF_TYPE_SD:
	case IF_TYPE_MMC:
	case IF_TYPE_USB:
	case IF_TYPE_NVME:
		printf ("Vendor: %s Rev: %s Prod: %s\n",
			dev_desc->vendor,
			dev_desc->revision,
			dev_desc->product);
		break;
	case IF_TYPE_DOC:
		puts("device type DOC\n");
		return;
	case IF_TYPE_UNKNOWN:
		puts("device type unknown\n");
		return;
	default:
		printf("Unhandled device type: %i\n", dev_desc->if_type);
		return;
	}
	puts ("            Type: ");
	if (dev_desc->removable)
		puts ("Removable ");
	switch (dev_desc->type & 0x1F) {
	case DEV_TYPE_HARDDISK:
		puts ("Hard Disk");
		break;
	case DEV_TYPE_CDROM:
		puts ("CD ROM");
		break;
	case DEV_TYPE_OPDISK:
		puts ("Optical Device");
		break;
	case DEV_TYPE_TAPE:
		puts ("Tape");
		break;
	default:
		printf ("# %02X #", dev_desc->type & 0x1F);
		break;
	}
	puts ("\n");
	if (dev_desc->lba > 0L && dev_desc->blksz > 0L) {
		ulong mb, mb_quot, mb_rem, gb, gb_quot, gb_rem;
		lbaint_t lba;

		lba = dev_desc->lba;

		lba512 = (lba * (dev_desc->blksz/512));
		/* round to 1 digit */
		/* 2048 = (1024 * 1024) / 512 MB */
		mb = lba512_muldiv(lba512, 10, 2048);

		mb_quot	= mb / 10;
		mb_rem	= mb - (10 * mb_quot);

		gb = mb / 1024;
		gb_quot	= gb / 10;
		gb_rem	= gb - (10 * gb_quot);
#ifdef CONFIG_LBA48
		if (dev_desc->lba48)
			printf ("            Supports 48-bit addressing\n");
#endif
#if defined(CONFIG_SYS_64BIT_LBA)
		printf ("            Capacity: %lu.%lu MB = %lu.%lu GB (%llu x %lu)\n",
			mb_quot, mb_rem,
			gb_quot, gb_rem,
			lba,
			dev_desc->blksz);
#else
		printf ("            Capacity: %lu.%lu MB = %lu.%lu GB (%lu x %lu)\n",
			mb_quot, mb_rem,
			gb_quot, gb_rem,
			(ulong)lba,
			dev_desc->blksz);
#endif
	} else {
		puts ("            Capacity: not available\n");
	}
}
#endif

#ifdef HAVE_BLOCK_DEVICE

void part_init(struct blk_desc *dev_desc)
{
	struct part_driver *drv =
		ll_entry_start(struct part_driver, part_driver);
	const int n_ents = ll_entry_count(struct part_driver, part_driver);
	struct part_driver *entry;

	blkcache_invalidate(dev_desc->if_type, dev_desc->devnum);

	dev_desc->part_type = PART_TYPE_UNKNOWN;
	for (entry = drv; entry != drv + n_ents; entry++) {
		int ret;

		ret = entry->test(dev_desc);
		debug("%s: try '%s': ret=%d\n", __func__, entry->name, ret);
		if (!ret) {
			dev_desc->part_type = entry->part_type;
			break;
		}
	}
}

static void print_part_header(const char *type, struct blk_desc *dev_desc)
{
#if CONFIG_IS_ENABLED(MAC_PARTITION) || \
	CONFIG_IS_ENABLED(DOS_PARTITION) || \
	CONFIG_IS_ENABLED(ISO_PARTITION) || \
	CONFIG_IS_ENABLED(AMIGA_PARTITION) || \
	CONFIG_IS_ENABLED(EFI_PARTITION)
	puts ("\nPartition Map for ");
	switch (dev_desc->if_type) {
	case IF_TYPE_IDE:
		puts ("IDE");
		break;
	case IF_TYPE_SATA:
		puts ("SATA");
		break;
	case IF_TYPE_SCSI:
		puts ("SCSI");
		break;
	case IF_TYPE_ATAPI:
		puts ("ATAPI");
		break;
	case IF_TYPE_USB:
		puts ("USB");
		break;
	case IF_TYPE_DOC:
		puts ("DOC");
		break;
	case IF_TYPE_MMC:
		puts ("MMC");
		break;
	case IF_TYPE_HOST:
		puts ("HOST");
		break;
	case IF_TYPE_NVME:
		puts ("NVMe");
		break;
	default:
		puts ("UNKNOWN");
		break;
	}
	printf (" device %d  --   Partition Type: %s\n\n",
			dev_desc->devnum, type);
#endif /* any CONFIG_..._PARTITION */
}

void part_print(struct blk_desc *dev_desc)
{
	struct part_driver *drv;

	drv = part_driver_lookup_type(dev_desc->part_type);
	if (!drv) {
		printf("## Unknown partition table type %x\n",
		       dev_desc->part_type);
		return;
	}

	PRINTF("## Testing for valid %s partition ##\n", drv->name);
	print_part_header(drv->name, dev_desc);
	if (drv->print)
		drv->print(dev_desc);
}

#endif /* HAVE_BLOCK_DEVICE */

int part_get_info(struct blk_desc *dev_desc, int part,
		       disk_partition_t *info)
{
#ifdef HAVE_BLOCK_DEVICE
	struct part_driver *drv;

#if CONFIG_IS_ENABLED(PARTITION_UUIDS)
	/* The common case is no UUID support */
	info->uuid[0] = 0;
#endif
#ifdef CONFIG_PARTITION_TYPE_GUID
	info->type_guid[0] = 0;
#endif

	drv = part_driver_lookup_type(dev_desc->part_type);
	if (!drv) {
		debug("## Unknown partition table type %x\n",
		      dev_desc->part_type);
		return -EPROTONOSUPPORT;
	}
	if (!drv->get_info) {
		PRINTF("## Driver %s does not have the get_info() method\n",
		       drv->name);
		return -ENOSYS;
	}
	if (drv->get_info(dev_desc, part, info) == 0) {
		PRINTF("## Valid %s partition found ##\n", drv->name);
		return 0;
	}
#endif /* HAVE_BLOCK_DEVICE */

	return -1;
}

int blk_get_device_by_str(const char *ifname, const char *dev_hwpart_str,
			  struct blk_desc **dev_desc)
{
	char *ep;
	char *dup_str = NULL;
	const char *dev_str, *hwpart_str;
	int dev, hwpart;

	hwpart_str = strchr(dev_hwpart_str, '.');
	if (hwpart_str) {
		dup_str = strdup(dev_hwpart_str);
		dup_str[hwpart_str - dev_hwpart_str] = 0;
		dev_str = dup_str;
		hwpart_str++;
	} else {
		dev_str = dev_hwpart_str;
		hwpart = 0;
	}

	dev = simple_strtoul(dev_str, &ep, 16);
	if (*ep) {
		printf("** Bad device specification %s %s **\n",
		       ifname, dev_str);
		dev = -EINVAL;
		goto cleanup;
	}

	if (hwpart_str) {
		hwpart = simple_strtoul(hwpart_str, &ep, 16);
		if (*ep) {
			printf("** Bad HW partition specification %s %s **\n",
			    ifname, hwpart_str);
			dev = -EINVAL;
			goto cleanup;
		}
	}

	*dev_desc = get_dev_hwpart(ifname, dev, hwpart);
	if (!(*dev_desc) || ((*dev_desc)->type == DEV_TYPE_UNKNOWN)) {
		printf("** Bad device %s %s **\n", ifname, dev_hwpart_str);
		dev = -ENOENT;
		goto cleanup;
	}

#ifdef HAVE_BLOCK_DEVICE
	/*
	 * Updates the partition table for the specified hw partition.
	 * Does not need to be done for hwpart 0 since it is default and
	 * already loaded.
	 */
	if(hwpart != 0)
		part_init(*dev_desc);
#endif

cleanup:
	free(dup_str);
	return dev;
}

#define PART_UNSPECIFIED -2
#define PART_AUTO -1
int blk_get_device_part_str(const char *ifname, const char *dev_part_str,
			     struct blk_desc **dev_desc,
			     disk_partition_t *info, int allow_whole_dev)
{
	int ret = -1;
	const char *part_str;
	char *dup_str = NULL;
	const char *dev_str;
	int dev;
	char *ep;
	int p;
	int part;
	disk_partition_t tmpinfo;

#ifdef CONFIG_SANDBOX
	/*
	 * Special-case a pseudo block device "hostfs", to allow access to the
	 * host's own filesystem.
	 */
	if (0 == strcmp(ifname, "hostfs")) {
		*dev_desc = NULL;
		info->start = 0;
		info->size = 0;
		info->blksz = 0;
		info->bootable = 0;
		strcpy((char *)info->type, BOOT_PART_TYPE);
		strcpy((char *)info->name, "Sandbox host");
#if CONFIG_IS_ENABLED(PARTITION_UUIDS)
		info->uuid[0] = 0;
#endif
#ifdef CONFIG_PARTITION_TYPE_GUID
		info->type_guid[0] = 0;
#endif

		return 0;
	}
#endif

#ifdef CONFIG_CMD_UBIFS
	/*
	 * Special-case ubi, ubi goes through a mtd, rathen then through
	 * a regular block device.
	 */
	if (0 == strcmp(ifname, "ubi")) {
		if (!ubifs_is_mounted()) {
			printf("UBIFS not mounted, use ubifsmount to mount volume first!\n");
			return -1;
		}

		*dev_desc = NULL;
		memset(info, 0, sizeof(*info));
		strcpy((char *)info->type, BOOT_PART_TYPE);
		strcpy((char *)info->name, "UBI");
#if CONFIG_IS_ENABLED(PARTITION_UUIDS)
		info->uuid[0] = 0;
#endif
		return 0;
	}
#endif

	/* If no dev_part_str, use bootdevice environment variable */
	if (!dev_part_str || !strlen(dev_part_str) ||
	    !strcmp(dev_part_str, "-"))
		dev_part_str = getenv("bootdevice");

	/* If still no dev_part_str, it's an error */
	if (!dev_part_str) {
		printf("** No device specified **\n");
		goto cleanup;
	}

	/* Separate device and partition ID specification */
	part_str = strchr(dev_part_str, ':');
	if (part_str) {
		dup_str = strdup(dev_part_str);
		dup_str[part_str - dev_part_str] = 0;
		dev_str = dup_str;
		part_str++;
	} else {
		dev_str = dev_part_str;
	}

	/* Look up the device */
	dev = blk_get_device_by_str(ifname, dev_str, dev_desc);
	if (dev < 0)
		goto cleanup;

	/* Convert partition ID string to number */
	if (!part_str || !*part_str) {
		part = PART_UNSPECIFIED;
	} else if (!strcmp(part_str, "auto")) {
		part = PART_AUTO;
	} else {
		/* Something specified -> use exactly that */
		part = (int)simple_strtoul(part_str, &ep, 16);
		/*
		 * Less than whole string converted,
		 * or request for whole device, but caller requires partition.
		 */
		if (*ep || (part == 0 && !allow_whole_dev)) {
			printf("** Bad partition specification %s %s **\n",
			    ifname, dev_part_str);
			goto cleanup;
		}
	}

	/*
	 * No partition table on device,
	 * or user requested partition 0 (entire device).
	 */
	if (((*dev_desc)->part_type == PART_TYPE_UNKNOWN) ||
	    (part == 0)) {
		if (!(*dev_desc)->lba) {
			printf("** Bad device size - %s %s **\n", ifname,
			       dev_str);
			goto cleanup;
		}

		/*
		 * If user specified a partition ID other than 0,
		 * or the calling command only accepts partitions,
		 * it's an error.
		 */
		if ((part > 0) || (!allow_whole_dev)) {
			printf("** No partition table - %s %s **\n", ifname,
			       dev_str);
			goto cleanup;
		}

		(*dev_desc)->log2blksz = LOG2((*dev_desc)->blksz);

		info->start = 0;
		info->size = (*dev_desc)->lba;
		info->blksz = (*dev_desc)->blksz;
		info->bootable = 0;
		strcpy((char *)info->type, BOOT_PART_TYPE);
		strcpy((char *)info->name, "Whole Disk");
#if CONFIG_IS_ENABLED(PARTITION_UUIDS)
		info->uuid[0] = 0;
#endif
#ifdef CONFIG_PARTITION_TYPE_GUID
		info->type_guid[0] = 0;
#endif

		ret = 0;
		goto cleanup;
	}

	/*
	 * Now there's known to be a partition table,
	 * not specifying a partition means to pick partition 1.
	 */
	if (part == PART_UNSPECIFIED)
		part = 1;

	/*
	 * If user didn't specify a partition number, or did specify something
	 * other than "auto", use that partition number directly.
	 */
	if (part != PART_AUTO) {
		ret = part_get_info(*dev_desc, part, info);
		if (ret) {
			printf("** Invalid partition %d **\n", part);
			goto cleanup;
		}
	} else {
		/*
		 * Find the first bootable partition.
		 * If none are bootable, fall back to the first valid partition.
		 */
		part = 0;
		for (p = 1; p <= MAX_SEARCH_PARTITIONS; p++) {
			ret = part_get_info(*dev_desc, p, info);
			if (ret)
				continue;

			/*
			 * First valid partition, or new better partition?
			 * If so, save partition ID.
			 */
			if (!part || info->bootable)
				part = p;

			/* Best possible partition? Stop searching. */
			if (info->bootable)
				break;

			/*
			 * We now need to search further for best possible.
			 * If we what we just queried was the best so far,
			 * save the info since we over-write it next loop.
			 */
			if (part == p)
				tmpinfo = *info;
		}
		/* If we found any acceptable partition */
		if (part) {
			/*
			 * If we searched all possible partition IDs,
			 * return the first valid partition we found.
			 */
			if (p == MAX_SEARCH_PARTITIONS + 1)
				*info = tmpinfo;
		} else {
			printf("** No valid partitions found **\n");
			ret = -1;
			goto cleanup;
		}
	}
	if (strncmp((char *)info->type, BOOT_PART_TYPE, sizeof(info->type)) != 0) {
		printf("** Invalid partition type \"%.32s\""
			" (expect \"" BOOT_PART_TYPE "\")\n",
			info->type);
		ret  = -1;
		goto cleanup;
	}

	(*dev_desc)->log2blksz = LOG2((*dev_desc)->blksz);

	ret = part;
	goto cleanup;

cleanup:
	free(dup_str);
	return ret;
}

int part_get_info_by_name(struct blk_desc *dev_desc, const char *name,
	disk_partition_t *info)
{
	struct part_driver *first_drv =
		ll_entry_start(struct part_driver, part_driver);
	const int n_drvs = ll_entry_count(struct part_driver, part_driver);
	struct part_driver *part_drv;

	for (part_drv = first_drv; part_drv != first_drv + n_drvs; part_drv++) {
		int ret;
		int i;
		for (i = 1; i < part_drv->max_entries; i++) {
			ret = part_drv->get_info(dev_desc, i, info);
			if (ret != 0) {
				/* no more entries in table */
				break;
			}
			if (strcmp(name, (const char *)info->name) == 0) {
				/* matched */
				return i;
			}
		}
	}
	return -1;
}

void part_set_generic_name(const struct blk_desc *dev_desc,
	int part_num, char *name)
{
	char *devtype;

	switch (dev_desc->if_type) {
	case IF_TYPE_IDE:
	case IF_TYPE_SATA:
	case IF_TYPE_ATAPI:
		devtype = "hd";
		break;
	case IF_TYPE_SCSI:
		devtype = "sd";
		break;
	case IF_TYPE_USB:
		devtype = "usbd";
		break;
	case IF_TYPE_DOC:
		devtype = "docd";
		break;
	case IF_TYPE_MMC:
	case IF_TYPE_SD:
		devtype = "mmcsd";
		break;
	default:
		devtype = "xx";
		break;
	}

	sprintf(name, "%s%c%d", devtype, 'a' + dev_desc->devnum, part_num);
}
Commit	Line	Data
affae2bf WD	1	/*
	2	* (C) Copyright 2001
	3	* Wolfgang Denk, DENX Software Engineering, wd@denx.de.
	4	*
1a459660	5	* SPDX-License-Identifier: GPL-2.0+
affae2bf WD	6	*/
	7
	8	#include <common.h>
	9	#include <command.h>
96e5b03c	10	#include <errno.h>
affae2bf	11	#include <ide.h>
10a37fd7	12	#include <malloc.h>
735dd97b	13	#include <part.h>
251cee0d	14	#include <ubifs_uboot.h>
affae2bf WD	15
	16	#undef PART_DEBUG
	17
	18	#ifdef PART_DEBUG
	19	#define PRINTF(fmt,args...) printf (fmt ,##args)
	20	#else
	21	#define PRINTF(fmt,args...)
	22	#endif
	23
751bb571	24	DECLARE_GLOBAL_DATA_PTR;
751bb571	25
0c9c8fb5	26	#ifdef HAVE_BLOCK_DEVICE
96e5b03c SG	27	static struct part_driver *part_driver_lookup_type(int part_type)
	28	{
	29	struct part_driver *drv =
	30	ll_entry_start(struct part_driver, part_driver);
	31	const int n_ents = ll_entry_count(struct part_driver, part_driver);
	32	struct part_driver *entry;
	33
	34	for (entry = drv; entry != drv + n_ents; entry++) {
	35	if (part_type == entry->part_type)
	36	return entry;
	37	}
	38
	39	/* Not found */
	40	return NULL;
	41	}
	42
4101f687	43	static struct blk_desc get_dev_hwpart(const char ifname, int dev, int hwpart)
735dd97b	44	{
6dd9faf8 SG	45	struct blk_desc *dev_desc;
6dd9faf8 SG	46	int ret;
735dd97b	47
6dd9faf8 SG	48	dev_desc = blk_get_devnum_by_typename(ifname, dev);
	49	if (!dev_desc) {
	50	debug("%s: No device for iface '%s', dev %d\n", __func__,
	51	ifname, dev);
7e71dc68	52	return NULL;
735dd97b	53	}
6dd9faf8 SG	54	ret = blk_dselect_hwpart(dev_desc, hwpart);
	55	if (ret) {
	56	debug("%s: Failed to select h/w partition: err-%d\n", __func__,
	57	ret);
	58	return NULL;
	59	}
	60
	61	return dev_desc;
735dd97b	62	}
336b6f90	63
db1d9e78	64	struct blk_desc blk_get_dev(const char ifname, int dev)
336b6f90	65	{
ecdd57e2	66	return get_dev_hwpart(ifname, dev, 0);
336b6f90	67	}
735dd97b	68	#else
4101f687	69	struct blk_desc get_dev_hwpart(const char ifname, int dev, int hwpart)
336b6f90 SW	70	{
	71	return NULL;
	72	}
	73
db1d9e78	74	struct blk_desc blk_get_dev(const char ifname, int dev)
735dd97b GL	75	{
	76	return NULL;
	77	}
	78	#endif
	79
0c9c8fb5	80	#ifdef HAVE_BLOCK_DEVICE
affae2bf WD	81
	82	/* ------------------------------------------------------------------------- */
	83	/*
	84	* reports device info to the user
	85	*/
69a2a4d9	86
42dfe7a1	87	#ifdef CONFIG_LBA48
69a2a4d9	88	typedef uint64_t lba512_t;
c40b2956	89	#else
69a2a4d9	90	typedef lbaint_t lba512_t;
c40b2956	91	#endif
affae2bf	92
69a2a4d9 ST	93	/*
	94	* Overflowless variant of (block_count * mul_by / div_by)
	95	* when div_by > mul_by
	96	*/
214b3f31	97	static lba512_t lba512_muldiv(lba512_t block_count, lba512_t mul_by, lba512_t div_by)
69a2a4d9 ST	98	{
	99	lba512_t bc_quot, bc_rem;
	100
	101	/* x * m / d == x / d * m + (x % d) * m / d */
	102	bc_quot = block_count / div_by;
	103	bc_rem = block_count - div_by * bc_quot;
	104	return bc_quot * mul_by + (bc_rem * mul_by) / div_by;
	105	}
	106
4101f687	107	void dev_print (struct blk_desc *dev_desc)
69a2a4d9 ST	108	{
	109	lba512_t lba512; /* number of blocks if 512bytes block size */
	110
af75a45d WD	111	if (dev_desc->type == DEV_TYPE_UNKNOWN) {
	112	puts ("not available\n");
	113	return;
	114	}
	115
8ec6e332	116	switch (dev_desc->if_type) {
574b3195 DZ	117	case IF_TYPE_SCSI:
	118	printf ("(%d:%d) Vendor: %s Prod.: %s Rev: %s\n",
	119	dev_desc->target,dev_desc->lun,
affae2bf	120	dev_desc->vendor,
574b3195 DZ	121	dev_desc->product,
	122	dev_desc->revision);
	123	break;
6e24a1eb	124	case IF_TYPE_ATAPI:
574b3195 DZ	125	case IF_TYPE_IDE:
574b3195 DZ	126	case IF_TYPE_SATA:
c7057b52 DL	127	printf ("Model: %s Firm: %s Ser#: %s\n",
	128	dev_desc->vendor,
	129	dev_desc->revision,
	130	dev_desc->product);
574b3195	131	break;
6e24a1eb RB	132	case IF_TYPE_SD:
6e24a1eb RB	133	case IF_TYPE_MMC:
47bebe34	134	case IF_TYPE_USB:
ffab6945	135	case IF_TYPE_NVME:
47bebe34 NCL	136	printf ("Vendor: %s Rev: %s Prod: %s\n",
	137	dev_desc->vendor,
	138	dev_desc->revision,
	139	dev_desc->product);
	140	break;
6e24a1eb RB	141	case IF_TYPE_DOC:
	142	puts("device type DOC\n");
	143	return;
8ec6e332	144	case IF_TYPE_UNKNOWN:
6e24a1eb RB	145	puts("device type unknown\n");
6e24a1eb RB	146	return;
574b3195	147	default:
6e24a1eb	148	printf("Unhandled device type: %i\n", dev_desc->if_type);
574b3195	149	return;
affae2bf WD	150	}
	151	puts (" Type: ");
	152	if (dev_desc->removable)
	153	puts ("Removable ");
	154	switch (dev_desc->type & 0x1F) {
726c0f1e DZ	155	case DEV_TYPE_HARDDISK:
	156	puts ("Hard Disk");
	157	break;
	158	case DEV_TYPE_CDROM:
	159	puts ("CD ROM");
	160	break;
	161	case DEV_TYPE_OPDISK:
	162	puts ("Optical Device");
	163	break;
	164	case DEV_TYPE_TAPE:
	165	puts ("Tape");
	166	break;
	167	default:
	168	printf ("# %02X #", dev_desc->type & 0x1F);
	169	break;
affae2bf WD	170	}
affae2bf WD	171	puts ("\n");
33699df1	172	if (dev_desc->lba > 0L && dev_desc->blksz > 0L) {
affae2bf	173	ulong mb, mb_quot, mb_rem, gb, gb_quot, gb_rem;
c40b2956	174	lbaint_t lba;
6e592385 WD	175
6e592385 WD	176	lba = dev_desc->lba;
affae2bf	177
c40b2956	178	lba512 = (lba * (dev_desc->blksz/512));
affae2bf	179	/* round to 1 digit */
214b3f31 PM	180	/* 2048 = (1024 * 1024) / 512 MB */
214b3f31 PM	181	mb = lba512_muldiv(lba512, 10, 2048);
69a2a4d9	182
affae2bf WD	183	mb_quot = mb / 10;
	184	mb_rem = mb - (10 * mb_quot);
	185
	186	gb = mb / 1024;
	187	gb_quot = gb / 10;
	188	gb_rem = gb - (10 * gb_quot);
42dfe7a1	189	#ifdef CONFIG_LBA48
6e592385	190	if (dev_desc->lba48)
c40b2956 WD	191	printf (" Supports 48-bit addressing\n");
c40b2956 WD	192	#endif
4b142feb	193	#if defined(CONFIG_SYS_64BIT_LBA)
139f7b1d	194	printf (" Capacity: %lu.%lu MB = %lu.%lu GB (%llu x %lu)\n",
c40b2956 WD	195	mb_quot, mb_rem,
	196	gb_quot, gb_rem,
	197	lba,
	198	dev_desc->blksz);
	199	#else
139f7b1d	200	printf (" Capacity: %lu.%lu MB = %lu.%lu GB (%lu x %lu)\n",
affae2bf WD	201	mb_quot, mb_rem,
affae2bf WD	202	gb_quot, gb_rem,
c40b2956	203	(ulong)lba,
affae2bf	204	dev_desc->blksz);
c40b2956	205	#endif
affae2bf WD	206	} else {
	207	puts (" Capacity: not available\n");
	208	}
	209	}
b3aff0cb	210	#endif
affae2bf	211
0c9c8fb5	212	#ifdef HAVE_BLOCK_DEVICE
affae2bf	213
3e8bd469	214	void part_init(struct blk_desc *dev_desc)
affae2bf	215	{
96e5b03c SG	216	struct part_driver *drv =
	217	ll_entry_start(struct part_driver, part_driver);
	218	const int n_ents = ll_entry_count(struct part_driver, part_driver);
	219	struct part_driver *entry;
affae2bf	220
e40cf34a EN	221	blkcache_invalidate(dev_desc->if_type, dev_desc->devnum);
e40cf34a EN	222
99d2c205	223	dev_desc->part_type = PART_TYPE_UNKNOWN;
96e5b03c SG	224	for (entry = drv; entry != drv + n_ents; entry++) {
	225	int ret;
	226
	227	ret = entry->test(dev_desc);
	228	debug("%s: try '%s': ret=%d\n", __func__, entry->name, ret);
	229	if (!ret) {
	230	dev_desc->part_type = entry->part_type;
	231	break;
	232	}
	233	}
affae2bf WD	234	}
affae2bf WD	235
96e5b03c SG	236	static void print_part_header(const char type, struct blk_desc dev_desc)
96e5b03c SG	237	{
f18fa31c	238	#if CONFIG_IS_ENABLED(MAC_PARTITION) \|\| \
b0cf7339	239	CONFIG_IS_ENABLED(DOS_PARTITION) \|\| \
1acc0087	240	CONFIG_IS_ENABLED(ISO_PARTITION) \|\| \
863c5b6c	241	CONFIG_IS_ENABLED(AMIGA_PARTITION) \|\| \
bd42a942	242	CONFIG_IS_ENABLED(EFI_PARTITION)
affae2bf WD	243	puts ("\nPartition Map for ");
affae2bf WD	244	switch (dev_desc->if_type) {
726c0f1e DZ	245	case IF_TYPE_IDE:
	246	puts ("IDE");
	247	break;
	248	case IF_TYPE_SATA:
	249	puts ("SATA");
	250	break;
	251	case IF_TYPE_SCSI:
	252	puts ("SCSI");
	253	break;
	254	case IF_TYPE_ATAPI:
	255	puts ("ATAPI");
	256	break;
	257	case IF_TYPE_USB:
	258	puts ("USB");
	259	break;
	260	case IF_TYPE_DOC:
	261	puts ("DOC");
	262	break;
8f3b9642 LW	263	case IF_TYPE_MMC:
	264	puts ("MMC");
	265	break;
f4d8de48	266	case IF_TYPE_HOST:
ffab6945 ZZ	267	puts ("HOST");
	268	break;
	269	case IF_TYPE_NVME:
	270	puts ("NVMe");
f4d8de48	271	break;
726c0f1e DZ	272	default:
	273	puts ("UNKNOWN");
	274	break;
affae2bf WD	275	}
affae2bf WD	276	printf (" device %d -- Partition Type: %s\n\n",
bcce53d0	277	dev_desc->devnum, type);
0c9c8fb5	278	#endif /* any CONFIG_..._PARTITION */
96e5b03c	279	}
0c9c8fb5	280
3e8bd469	281	void part_print(struct blk_desc *dev_desc)
affae2bf	282	{
96e5b03c	283	struct part_driver *drv;
affae2bf	284
96e5b03c SG	285	drv = part_driver_lookup_type(dev_desc->part_type);
	286	if (!drv) {
	287	printf("## Unknown partition table type %x\n",
	288	dev_desc->part_type);
affae2bf	289	return;
affae2bf	290	}
96e5b03c SG	291
	292	PRINTF("## Testing for valid %s partition ##\n", drv->name);
	293	print_part_header(drv->name, dev_desc);
	294	if (drv->print)
	295	drv->print(dev_desc);
affae2bf WD	296	}
affae2bf WD	297
0c9c8fb5	298	#endif /* HAVE_BLOCK_DEVICE */
2f501646	299
3e8bd469	300	int part_get_info(struct blk_desc *dev_desc, int part,
3f9eb6e1	301	disk_partition_t *info)
2f501646	302	{
0c9c8fb5	303	#ifdef HAVE_BLOCK_DEVICE
96e5b03c	304	struct part_driver *drv;
2f501646	305
b331cd62	306	#if CONFIG_IS_ENABLED(PARTITION_UUIDS)
894bfbbf SW	307	/* The common case is no UUID support */
	308	info->uuid[0] = 0;
	309	#endif
7561b258 PD	310	#ifdef CONFIG_PARTITION_TYPE_GUID
	311	info->type_guid[0] = 0;
	312	#endif
894bfbbf	313
96e5b03c SG	314	drv = part_driver_lookup_type(dev_desc->part_type);
	315	if (!drv) {
	316	debug("## Unknown partition table type %x\n",
	317	dev_desc->part_type);
	318	return -EPROTONOSUPPORT;
	319	}
	320	if (!drv->get_info) {
2ae67aec NM	321	PRINTF("## Driver %s does not have the get_info() method\n",
2ae67aec NM	322	drv->name);
96e5b03c SG	323	return -ENOSYS;
	324	}
	325	if (drv->get_info(dev_desc, part, info) == 0) {
	326	PRINTF("## Valid %s partition found ##\n", drv->name);
	327	return 0;
2f501646	328	}
0c9c8fb5	329	#endif /* HAVE_BLOCK_DEVICE */
2f501646 SW	330
	331	return -1;
	332	}
99d2c205	333
ebac37cf SG	334	int blk_get_device_by_str(const char ifname, const char dev_hwpart_str,
ebac37cf SG	335	struct blk_desc **dev_desc)
2023e608 SW	336	{
2023e608 SW	337	char *ep;
336b6f90 SW	338	char *dup_str = NULL;
	339	const char dev_str, hwpart_str;
	340	int dev, hwpart;
	341
	342	hwpart_str = strchr(dev_hwpart_str, '.');
	343	if (hwpart_str) {
	344	dup_str = strdup(dev_hwpart_str);
	345	dup_str[hwpart_str - dev_hwpart_str] = 0;
	346	dev_str = dup_str;
	347	hwpart_str++;
	348	} else {
	349	dev_str = dev_hwpart_str;
ecdd57e2	350	hwpart = 0;
336b6f90	351	}
2023e608 SW	352
	353	dev = simple_strtoul(dev_str, &ep, 16);
	354	if (*ep) {
	355	printf(" Bad device specification %s %s \n",
	356	ifname, dev_str);
1598dfcb	357	dev = -EINVAL;
336b6f90 SW	358	goto cleanup;
	359	}
	360
	361	if (hwpart_str) {
	362	hwpart = simple_strtoul(hwpart_str, &ep, 16);
	363	if (*ep) {
	364	printf(" Bad HW partition specification %s %s \n",
	365	ifname, hwpart_str);
1598dfcb	366	dev = -EINVAL;
336b6f90 SW	367	goto cleanup;
336b6f90 SW	368	}
2023e608 SW	369	}
2023e608 SW	370
336b6f90	371	*dev_desc = get_dev_hwpart(ifname, dev, hwpart);
2023e608	372	if (!(dev_desc) \|\| ((dev_desc)->type == DEV_TYPE_UNKNOWN)) {
336b6f90	373	printf(" Bad device %s %s \n", ifname, dev_hwpart_str);
1598dfcb	374	dev = -ENOENT;
336b6f90	375	goto cleanup;
2023e608 SW	376	}
2023e608 SW	377
99e7fc8a ET	378	#ifdef HAVE_BLOCK_DEVICE
	379	/*
	380	* Updates the partition table for the specified hw partition.
	381	* Does not need to be done for hwpart 0 since it is default and
	382	* already loaded.
	383	*/
	384	if(hwpart != 0)
3e8bd469	385	part_init(*dev_desc);
99e7fc8a ET	386	#endif
99e7fc8a ET	387
336b6f90 SW	388	cleanup:
336b6f90 SW	389	free(dup_str);
2023e608 SW	390	return dev;
	391	}
	392
10a37fd7 SW	393	#define PART_UNSPECIFIED -2
10a37fd7 SW	394	#define PART_AUTO -1
e35929e4	395	int blk_get_device_part_str(const char ifname, const char dev_part_str,
4101f687	396	struct blk_desc **dev_desc,
10a37fd7	397	disk_partition_t *info, int allow_whole_dev)
99d2c205	398	{
10a37fd7 SW	399	int ret = -1;
	400	const char *part_str;
	401	char *dup_str = NULL;
	402	const char *dev_str;
99d2c205	403	int dev;
10a37fd7 SW	404	char *ep;
	405	int p;
	406	int part;
	407	disk_partition_t tmpinfo;
	408
afc1744e	409	#ifdef CONFIG_SANDBOX
4d907025	410	/*
3f9eb6e1	411	* Special-case a pseudo block device "hostfs", to allow access to the
4d907025 SW	412	* host's own filesystem.
	413	*/
	414	if (0 == strcmp(ifname, "hostfs")) {
	415	*dev_desc = NULL;
	416	info->start = 0;
	417	info->size = 0;
	418	info->blksz = 0;
	419	info->bootable = 0;
	420	strcpy((char *)info->type, BOOT_PART_TYPE);
	421	strcpy((char *)info->name, "Sandbox host");
b331cd62	422	#if CONFIG_IS_ENABLED(PARTITION_UUIDS)
4d907025 SW	423	info->uuid[0] = 0;
4d907025 SW	424	#endif
7561b258 PD	425	#ifdef CONFIG_PARTITION_TYPE_GUID
	426	info->type_guid[0] = 0;
	427	#endif
4d907025 SW	428
	429	return 0;
	430	}
afc1744e	431	#endif
4d907025	432
251cee0d HG	433	#ifdef CONFIG_CMD_UBIFS
	434	/*
	435	* Special-case ubi, ubi goes through a mtd, rathen then through
	436	* a regular block device.
	437	*/
	438	if (0 == strcmp(ifname, "ubi")) {
	439	if (!ubifs_is_mounted()) {
	440	printf("UBIFS not mounted, use ubifsmount to mount volume first!\n");
	441	return -1;
	442	}
	443
	444	*dev_desc = NULL;
	445	memset(info, 0, sizeof(*info));
	446	strcpy((char *)info->type, BOOT_PART_TYPE);
	447	strcpy((char *)info->name, "UBI");
b331cd62	448	#if CONFIG_IS_ENABLED(PARTITION_UUIDS)
251cee0d HG	449	info->uuid[0] = 0;
	450	#endif
	451	return 0;
	452	}
	453	#endif
	454
10a37fd7	455	/* If no dev_part_str, use bootdevice environment variable */
a10973e7 SW	456	if (!dev_part_str \|\| !strlen(dev_part_str) \|\|
a10973e7 SW	457	!strcmp(dev_part_str, "-"))
10a37fd7 SW	458	dev_part_str = getenv("bootdevice");
	459
	460	/* If still no dev_part_str, it's an error */
	461	if (!dev_part_str) {
	462	printf(" No device specified \n");
	463	goto cleanup;
	464	}
	465
	466	/* Separate device and partition ID specification */
	467	part_str = strchr(dev_part_str, ':');
	468	if (part_str) {
	469	dup_str = strdup(dev_part_str);
	470	dup_str[part_str - dev_part_str] = 0;
	471	dev_str = dup_str;
	472	part_str++;
	473	} else {
	474	dev_str = dev_part_str;
99d2c205 RH	475	}
99d2c205 RH	476
10a37fd7	477	/* Look up the device */
ebac37cf	478	dev = blk_get_device_by_str(ifname, dev_str, dev_desc);
10a37fd7 SW	479	if (dev < 0)
	480	goto cleanup;
	481
	482	/* Convert partition ID string to number */
	483	if (!part_str \|\| !*part_str) {
	484	part = PART_UNSPECIFIED;
	485	} else if (!strcmp(part_str, "auto")) {
	486	part = PART_AUTO;
	487	} else {
	488	/* Something specified -> use exactly that */
	489	part = (int)simple_strtoul(part_str, &ep, 16);
	490	/*
	491	* Less than whole string converted,
	492	* or request for whole device, but caller requires partition.
	493	*/
	494	if (*ep \|\| (part == 0 && !allow_whole_dev)) {
	495	printf(" Bad partition specification %s %s \n",
	496	ifname, dev_part_str);
	497	goto cleanup;
	498	}
	499	}
	500
	501	/*
	502	* No partition table on device,
	503	* or user requested partition 0 (entire device).
	504	*/
	505	if (((*dev_desc)->part_type == PART_TYPE_UNKNOWN) \|\|
	506	(part == 0)) {
	507	if (!(*dev_desc)->lba) {
	508	printf(" Bad device size - %s %s \n", ifname,
	509	dev_str);
	510	goto cleanup;
	511	}
99d2c205	512
10a37fd7 SW	513	/*
	514	* If user specified a partition ID other than 0,
	515	* or the calling command only accepts partitions,
	516	* it's an error.
	517	*/
	518	if ((part > 0) \|\| (!allow_whole_dev)) {
	519	printf(" No partition table - %s %s \n", ifname,
	520	dev_str);
	521	goto cleanup;
99d2c205	522	}
10a37fd7	523
50ffc3b6 LY	524	(dev_desc)->log2blksz = LOG2((dev_desc)->blksz);
50ffc3b6 LY	525
99d2c205	526	info->start = 0;
10a37fd7 SW	527	info->size = (*dev_desc)->lba;
	528	info->blksz = (*dev_desc)->blksz;
	529	info->bootable = 0;
6ab6a650 SW	530	strcpy((char *)info->type, BOOT_PART_TYPE);
6ab6a650 SW	531	strcpy((char *)info->name, "Whole Disk");
b331cd62	532	#if CONFIG_IS_ENABLED(PARTITION_UUIDS)
10a37fd7 SW	533	info->uuid[0] = 0;
10a37fd7 SW	534	#endif
7561b258 PD	535	#ifdef CONFIG_PARTITION_TYPE_GUID
	536	info->type_guid[0] = 0;
	537	#endif
99d2c205	538
10a37fd7 SW	539	ret = 0;
10a37fd7 SW	540	goto cleanup;
99d2c205 RH	541	}
99d2c205 RH	542
10a37fd7 SW	543	/*
	544	* Now there's known to be a partition table,
	545	* not specifying a partition means to pick partition 1.
	546	*/
	547	if (part == PART_UNSPECIFIED)
	548	part = 1;
	549
	550	/*
	551	* If user didn't specify a partition number, or did specify something
	552	* other than "auto", use that partition number directly.
	553	*/
	554	if (part != PART_AUTO) {
3e8bd469	555	ret = part_get_info(*dev_desc, part, info);
10a37fd7 SW	556	if (ret) {
	557	printf(" Invalid partition %d \n", part);
	558	goto cleanup;
	559	}
	560	} else {
	561	/*
	562	* Find the first bootable partition.
	563	* If none are bootable, fall back to the first valid partition.
	564	*/
	565	part = 0;
	566	for (p = 1; p <= MAX_SEARCH_PARTITIONS; p++) {
3e8bd469	567	ret = part_get_info(*dev_desc, p, info);
10a37fd7 SW	568	if (ret)
	569	continue;
	570
	571	/*
	572	* First valid partition, or new better partition?
	573	* If so, save partition ID.
	574	*/
	575	if (!part \|\| info->bootable)
	576	part = p;
	577
	578	/* Best possible partition? Stop searching. */
	579	if (info->bootable)
	580	break;
	581
	582	/*
	583	* We now need to search further for best possible.
	584	* If we what we just queried was the best so far,
	585	* save the info since we over-write it next loop.
	586	*/
	587	if (part == p)
	588	tmpinfo = *info;
	589	}
	590	/* If we found any acceptable partition */
	591	if (part) {
	592	/*
	593	* If we searched all possible partition IDs,
	594	* return the first valid partition we found.
	595	*/
	596	if (p == MAX_SEARCH_PARTITIONS + 1)
	597	*info = tmpinfo;
10a37fd7 SW	598	} else {
10a37fd7 SW	599	printf(" No valid partitions found \n");
71bba424	600	ret = -1;
10a37fd7 SW	601	goto cleanup;
10a37fd7 SW	602	}
99d2c205 RH	603	}
	604	if (strncmp((char *)info->type, BOOT_PART_TYPE, sizeof(info->type)) != 0) {
	605	printf("** Invalid partition type \"%.32s\""
	606	" (expect \"" BOOT_PART_TYPE "\")\n",
	607	info->type);
10a37fd7 SW	608	ret = -1;
10a37fd7 SW	609	goto cleanup;
99d2c205 RH	610	}
99d2c205 RH	611
50ffc3b6 LY	612	(dev_desc)->log2blksz = LOG2((dev_desc)->blksz);
50ffc3b6 LY	613
10a37fd7 SW	614	ret = part;
10a37fd7 SW	615	goto cleanup;
99d2c205	616
10a37fd7 SW	617	cleanup:
	618	free(dup_str);
	619	return ret;
99d2c205	620	}
87b8530f PK	621
	622	int part_get_info_by_name(struct blk_desc dev_desc, const char name,
	623	disk_partition_t *info)
	624	{
	625	struct part_driver *first_drv =
	626	ll_entry_start(struct part_driver, part_driver);
	627	const int n_drvs = ll_entry_count(struct part_driver, part_driver);
	628	struct part_driver *part_drv;
	629
	630	for (part_drv = first_drv; part_drv != first_drv + n_drvs; part_drv++) {
	631	int ret;
	632	int i;
	633	for (i = 1; i < part_drv->max_entries; i++) {
	634	ret = part_drv->get_info(dev_desc, i, info);
	635	if (ret != 0) {
	636	/* no more entries in table */
	637	break;
	638	}
	639	if (strcmp(name, (const char *)info->name) == 0) {
	640	/* matched */
88b6329c	641	return i;
87b8530f PK	642	}
	643	}
	644	}
	645	return -1;
	646	}
da2ee24d PK	647
	648	void part_set_generic_name(const struct blk_desc *dev_desc,
	649	int part_num, char *name)
	650	{
	651	char *devtype;
	652
	653	switch (dev_desc->if_type) {
	654	case IF_TYPE_IDE:
	655	case IF_TYPE_SATA:
	656	case IF_TYPE_ATAPI:
	657	devtype = "hd";
	658	break;
	659	case IF_TYPE_SCSI:
	660	devtype = "sd";
	661	break;
	662	case IF_TYPE_USB:
	663	devtype = "usbd";
	664	break;
	665	case IF_TYPE_DOC:
	666	devtype = "docd";
	667	break;
	668	case IF_TYPE_MMC:
	669	case IF_TYPE_SD:
	670	devtype = "mmcsd";
	671	break;
	672	default:
	673	devtype = "xx";
	674	break;
	675	}
	676
	677	sprintf(name, "%s%c%d", devtype, 'a' + dev_desc->devnum, part_num);
	678	}