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

/*
 * (C) Copyright 2001
 * Wolfgang Denk, DENX Software Engineering, [email protected].
 *
 * 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:
		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) || \
	defined(CONFIG_AMIGA_PARTITION) || \
	defined(CONFIG_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;
	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;

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