[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

const struct block_drvr block_drvr[] = {
#if defined(CONFIG_CMD_IDE)
	{ .name = "ide", },
#endif
#if defined(CONFIG_CMD_SATA)
	{.name = "sata", },
#endif
#if defined(CONFIG_SCSI)
	{ .name = "scsi", },
#endif
#if defined(CONFIG_CMD_USB) && defined(CONFIG_USB_STORAGE)
	{ .name = "usb", },
#endif
#if defined(CONFIG_MMC)
	{
		.name = "mmc",
		.select_hwpart = mmc_select_hwpart,
	},
#endif
#if defined(CONFIG_SYSTEMACE)
	{ .name = "ace", },
#endif
#if defined(CONFIG_SANDBOX)
	{ .name = "host", },
#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)
{
	const struct block_drvr *drvr = block_drvr;
	int (*select_hwpart)(int dev_num, int hwpart);
	char *name;
	int ret;

	if (!ifname)
		return NULL;

	name = drvr->name;
#ifdef CONFIG_NEEDS_MANUAL_RELOC
	name += gd->reloc_off;
#endif
	while (drvr->name) {
		name = drvr->name;
		select_hwpart = drvr->select_hwpart;
#ifdef CONFIG_NEEDS_MANUAL_RELOC
		name += gd->reloc_off;
		if (select_hwpart)
			select_hwpart += gd->reloc_off;
#endif
		if (strncmp(ifname, name, strlen(name)) == 0) {
			struct blk_desc *dev_desc;

			dev_desc = blk_get_devnum_by_typename(name, dev);
			if (!dev_desc)
				return NULL;
			if (hwpart == 0 && !select_hwpart)
				return dev_desc;
			if (!select_hwpart)
				return NULL;
			ret = select_hwpart(dev_desc->devnum, hwpart);
			if (ret < 0)
				return NULL;
			return dev_desc;
		}
		drvr++;
	}
	return NULL;
}

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