[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

const struct block_drvr block_drvr[] = {
#if defined(CONFIG_CMD_IDE)
	{ .name = "ide", .get_dev = ide_get_dev, },
#endif
#if defined(CONFIG_CMD_SATA)
	{.name = "sata", .get_dev = sata_get_dev, },
#endif
#if defined(CONFIG_CMD_SCSI)
	{ .name = "scsi", .get_dev = scsi_get_dev, },
#endif
#if defined(CONFIG_CMD_USB) && defined(CONFIG_USB_STORAGE)
	{ .name = "usb", .get_dev = usb_stor_get_dev, },
#endif
#if defined(CONFIG_MMC)
	{
		.name = "mmc",
		.get_dev = mmc_get_dev,
		.select_hwpart = mmc_select_hwpart,
	},
#endif
#if defined(CONFIG_SYSTEMACE)
	{ .name = "ace", .get_dev = systemace_get_dev, },
#endif
#if defined(CONFIG_SANDBOX)
	{ .name = "host", .get_dev = host_get_dev, },
#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;
	struct blk_desc* (*reloc_get_dev)(int dev);
	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;
		reloc_get_dev = drvr->get_dev;
		select_hwpart = drvr->select_hwpart;
#ifdef CONFIG_NEEDS_MANUAL_RELOC
		name += gd->reloc_off;
		reloc_get_dev += gd->reloc_off;
		if (select_hwpart)
			select_hwpart += gd->reloc_off;
#endif
		if (strncmp(ifname, name, strlen(name)) == 0) {
			struct blk_desc *dev_desc = reloc_get_dev(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;

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