[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_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;

	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)
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
c649e3c9	31	#if defined(CONFIG_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
e40cf34a EN	271	blkcache_invalidate(dev_desc->if_type, dev_desc->devnum);
e40cf34a EN	272
99d2c205	273	dev_desc->part_type = PART_TYPE_UNKNOWN;
96e5b03c SG	274	for (entry = drv; entry != drv + n_ents; entry++) {
	275	int ret;
	276
	277	ret = entry->test(dev_desc);
	278	debug("%s: try '%s': ret=%d\n", __func__, entry->name, ret);
	279	if (!ret) {
	280	dev_desc->part_type = entry->part_type;
	281	break;
	282	}
	283	}
affae2bf WD	284	}
affae2bf WD	285
96e5b03c SG	286	static void print_part_header(const char type, struct blk_desc dev_desc)
96e5b03c SG	287	{
0c9c8fb5 GB	288	#if defined(CONFIG_MAC_PARTITION) \|\| \
	289	defined(CONFIG_DOS_PARTITION) \|\| \
	290	defined(CONFIG_ISO_PARTITION) \|\| \
	291	defined(CONFIG_AMIGA_PARTITION) \|\| \
	292	defined(CONFIG_EFI_PARTITION)
affae2bf WD	293	puts ("\nPartition Map for ");
affae2bf WD	294	switch (dev_desc->if_type) {
726c0f1e DZ	295	case IF_TYPE_IDE:
	296	puts ("IDE");
	297	break;
	298	case IF_TYPE_SATA:
	299	puts ("SATA");
	300	break;
	301	case IF_TYPE_SCSI:
	302	puts ("SCSI");
	303	break;
	304	case IF_TYPE_ATAPI:
	305	puts ("ATAPI");
	306	break;
	307	case IF_TYPE_USB:
	308	puts ("USB");
	309	break;
	310	case IF_TYPE_DOC:
	311	puts ("DOC");
	312	break;
8f3b9642 LW	313	case IF_TYPE_MMC:
	314	puts ("MMC");
	315	break;
f4d8de48 HN	316	case IF_TYPE_HOST:
	317	puts("HOST");
	318	break;
726c0f1e DZ	319	default:
	320	puts ("UNKNOWN");
	321	break;
affae2bf WD	322	}
affae2bf WD	323	printf (" device %d -- Partition Type: %s\n\n",
bcce53d0	324	dev_desc->devnum, type);
0c9c8fb5	325	#endif /* any CONFIG_..._PARTITION */
96e5b03c	326	}
0c9c8fb5	327
3e8bd469	328	void part_print(struct blk_desc *dev_desc)
affae2bf	329	{
96e5b03c	330	struct part_driver *drv;
affae2bf	331
96e5b03c SG	332	drv = part_driver_lookup_type(dev_desc->part_type);
	333	if (!drv) {
	334	printf("## Unknown partition table type %x\n",
	335	dev_desc->part_type);
affae2bf	336	return;
affae2bf	337	}
96e5b03c SG	338
	339	PRINTF("## Testing for valid %s partition ##\n", drv->name);
	340	print_part_header(drv->name, dev_desc);
	341	if (drv->print)
	342	drv->print(dev_desc);
affae2bf WD	343	}
affae2bf WD	344
0c9c8fb5	345	#endif /* HAVE_BLOCK_DEVICE */
2f501646	346
3e8bd469	347	int part_get_info(struct blk_desc *dev_desc, int part,
3f9eb6e1	348	disk_partition_t *info)
2f501646	349	{
0c9c8fb5	350	#ifdef HAVE_BLOCK_DEVICE
96e5b03c	351	struct part_driver *drv;
2f501646	352
894bfbbf SW	353	#ifdef CONFIG_PARTITION_UUIDS
	354	/* The common case is no UUID support */
	355	info->uuid[0] = 0;
	356	#endif
7561b258 PD	357	#ifdef CONFIG_PARTITION_TYPE_GUID
	358	info->type_guid[0] = 0;
	359	#endif
894bfbbf	360
96e5b03c SG	361	drv = part_driver_lookup_type(dev_desc->part_type);
	362	if (!drv) {
	363	debug("## Unknown partition table type %x\n",
	364	dev_desc->part_type);
	365	return -EPROTONOSUPPORT;
	366	}
	367	if (!drv->get_info) {
2ae67aec NM	368	PRINTF("## Driver %s does not have the get_info() method\n",
2ae67aec NM	369	drv->name);
96e5b03c SG	370	return -ENOSYS;
	371	}
	372	if (drv->get_info(dev_desc, part, info) == 0) {
	373	PRINTF("## Valid %s partition found ##\n", drv->name);
	374	return 0;
2f501646	375	}
0c9c8fb5	376	#endif /* HAVE_BLOCK_DEVICE */
2f501646 SW	377
	378	return -1;
	379	}
99d2c205	380
ebac37cf SG	381	int blk_get_device_by_str(const char ifname, const char dev_hwpart_str,
ebac37cf SG	382	struct blk_desc **dev_desc)
2023e608 SW	383	{
2023e608 SW	384	char *ep;
336b6f90 SW	385	char *dup_str = NULL;
	386	const char dev_str, hwpart_str;
	387	int dev, hwpart;
	388
	389	hwpart_str = strchr(dev_hwpart_str, '.');
	390	if (hwpart_str) {
	391	dup_str = strdup(dev_hwpart_str);
	392	dup_str[hwpart_str - dev_hwpart_str] = 0;
	393	dev_str = dup_str;
	394	hwpart_str++;
	395	} else {
	396	dev_str = dev_hwpart_str;
ecdd57e2	397	hwpart = 0;
336b6f90	398	}
2023e608 SW	399
	400	dev = simple_strtoul(dev_str, &ep, 16);
	401	if (*ep) {
	402	printf(" Bad device specification %s %s \n",
	403	ifname, dev_str);
336b6f90 SW	404	dev = -1;
	405	goto cleanup;
	406	}
	407
	408	if (hwpart_str) {
	409	hwpart = simple_strtoul(hwpart_str, &ep, 16);
	410	if (*ep) {
	411	printf(" Bad HW partition specification %s %s \n",
	412	ifname, hwpart_str);
	413	dev = -1;
	414	goto cleanup;
	415	}
2023e608 SW	416	}
2023e608 SW	417
336b6f90	418	*dev_desc = get_dev_hwpart(ifname, dev, hwpart);
2023e608	419	if (!(dev_desc) \|\| ((dev_desc)->type == DEV_TYPE_UNKNOWN)) {
336b6f90 SW	420	printf(" Bad device %s %s \n", ifname, dev_hwpart_str);
	421	dev = -1;
	422	goto cleanup;
2023e608 SW	423	}
2023e608 SW	424
99e7fc8a ET	425	#ifdef HAVE_BLOCK_DEVICE
	426	/*
	427	* Updates the partition table for the specified hw partition.
	428	* Does not need to be done for hwpart 0 since it is default and
	429	* already loaded.
	430	*/
	431	if(hwpart != 0)
3e8bd469	432	part_init(*dev_desc);
99e7fc8a ET	433	#endif
99e7fc8a ET	434
336b6f90 SW	435	cleanup:
336b6f90 SW	436	free(dup_str);
2023e608 SW	437	return dev;
	438	}
	439
10a37fd7 SW	440	#define PART_UNSPECIFIED -2
	441	#define PART_AUTO -1
	442	#define MAX_SEARCH_PARTITIONS 16
e35929e4	443	int blk_get_device_part_str(const char ifname, const char dev_part_str,
4101f687	444	struct blk_desc **dev_desc,
10a37fd7	445	disk_partition_t *info, int allow_whole_dev)
99d2c205	446	{
10a37fd7 SW	447	int ret = -1;
	448	const char *part_str;
	449	char *dup_str = NULL;
	450	const char *dev_str;
99d2c205	451	int dev;
10a37fd7 SW	452	char *ep;
	453	int p;
	454	int part;
	455	disk_partition_t tmpinfo;
	456
afc1744e	457	#ifdef CONFIG_SANDBOX
4d907025	458	/*
3f9eb6e1	459	* Special-case a pseudo block device "hostfs", to allow access to the
4d907025 SW	460	* host's own filesystem.
	461	*/
	462	if (0 == strcmp(ifname, "hostfs")) {
	463	*dev_desc = NULL;
	464	info->start = 0;
	465	info->size = 0;
	466	info->blksz = 0;
	467	info->bootable = 0;
	468	strcpy((char *)info->type, BOOT_PART_TYPE);
	469	strcpy((char *)info->name, "Sandbox host");
	470	#ifdef CONFIG_PARTITION_UUIDS
	471	info->uuid[0] = 0;
	472	#endif
7561b258 PD	473	#ifdef CONFIG_PARTITION_TYPE_GUID
	474	info->type_guid[0] = 0;
	475	#endif
4d907025 SW	476
	477	return 0;
	478	}
afc1744e	479	#endif
4d907025	480
251cee0d HG	481	#ifdef CONFIG_CMD_UBIFS
	482	/*
	483	* Special-case ubi, ubi goes through a mtd, rathen then through
	484	* a regular block device.
	485	*/
	486	if (0 == strcmp(ifname, "ubi")) {
	487	if (!ubifs_is_mounted()) {
	488	printf("UBIFS not mounted, use ubifsmount to mount volume first!\n");
	489	return -1;
	490	}
	491
	492	*dev_desc = NULL;
	493	memset(info, 0, sizeof(*info));
	494	strcpy((char *)info->type, BOOT_PART_TYPE);
	495	strcpy((char *)info->name, "UBI");
	496	#ifdef CONFIG_PARTITION_UUIDS
	497	info->uuid[0] = 0;
	498	#endif
	499	return 0;
	500	}
	501	#endif
	502
10a37fd7	503	/* If no dev_part_str, use bootdevice environment variable */
a10973e7 SW	504	if (!dev_part_str \|\| !strlen(dev_part_str) \|\|
a10973e7 SW	505	!strcmp(dev_part_str, "-"))
10a37fd7 SW	506	dev_part_str = getenv("bootdevice");
	507
	508	/* If still no dev_part_str, it's an error */
	509	if (!dev_part_str) {
	510	printf(" No device specified \n");
	511	goto cleanup;
	512	}
	513
	514	/* Separate device and partition ID specification */
	515	part_str = strchr(dev_part_str, ':');
	516	if (part_str) {
	517	dup_str = strdup(dev_part_str);
	518	dup_str[part_str - dev_part_str] = 0;
	519	dev_str = dup_str;
	520	part_str++;
	521	} else {
	522	dev_str = dev_part_str;
99d2c205 RH	523	}
99d2c205 RH	524
10a37fd7	525	/* Look up the device */
ebac37cf	526	dev = blk_get_device_by_str(ifname, dev_str, dev_desc);
10a37fd7 SW	527	if (dev < 0)
	528	goto cleanup;
	529
	530	/* Convert partition ID string to number */
	531	if (!part_str \|\| !*part_str) {
	532	part = PART_UNSPECIFIED;
	533	} else if (!strcmp(part_str, "auto")) {
	534	part = PART_AUTO;
	535	} else {
	536	/* Something specified -> use exactly that */
	537	part = (int)simple_strtoul(part_str, &ep, 16);
	538	/*
	539	* Less than whole string converted,
	540	* or request for whole device, but caller requires partition.
	541	*/
	542	if (*ep \|\| (part == 0 && !allow_whole_dev)) {
	543	printf(" Bad partition specification %s %s \n",
	544	ifname, dev_part_str);
	545	goto cleanup;
	546	}
	547	}
	548
	549	/*
	550	* No partition table on device,
	551	* or user requested partition 0 (entire device).
	552	*/
	553	if (((*dev_desc)->part_type == PART_TYPE_UNKNOWN) \|\|
	554	(part == 0)) {
	555	if (!(*dev_desc)->lba) {
	556	printf(" Bad device size - %s %s \n", ifname,
	557	dev_str);
	558	goto cleanup;
	559	}
99d2c205	560
10a37fd7 SW	561	/*
	562	* If user specified a partition ID other than 0,
	563	* or the calling command only accepts partitions,
	564	* it's an error.
	565	*/
	566	if ((part > 0) \|\| (!allow_whole_dev)) {
	567	printf(" No partition table - %s %s \n", ifname,
	568	dev_str);
	569	goto cleanup;
99d2c205	570	}
10a37fd7	571
50ffc3b6 LY	572	(dev_desc)->log2blksz = LOG2((dev_desc)->blksz);
50ffc3b6 LY	573
99d2c205	574	info->start = 0;
10a37fd7 SW	575	info->size = (*dev_desc)->lba;
	576	info->blksz = (*dev_desc)->blksz;
	577	info->bootable = 0;
6ab6a650 SW	578	strcpy((char *)info->type, BOOT_PART_TYPE);
6ab6a650 SW	579	strcpy((char *)info->name, "Whole Disk");
10a37fd7 SW	580	#ifdef CONFIG_PARTITION_UUIDS
	581	info->uuid[0] = 0;
	582	#endif
7561b258 PD	583	#ifdef CONFIG_PARTITION_TYPE_GUID
	584	info->type_guid[0] = 0;
	585	#endif
99d2c205	586
10a37fd7 SW	587	ret = 0;
10a37fd7 SW	588	goto cleanup;
99d2c205 RH	589	}
99d2c205 RH	590
10a37fd7 SW	591	/*
	592	* Now there's known to be a partition table,
	593	* not specifying a partition means to pick partition 1.
	594	*/
	595	if (part == PART_UNSPECIFIED)
	596	part = 1;
	597
	598	/*
	599	* If user didn't specify a partition number, or did specify something
	600	* other than "auto", use that partition number directly.
	601	*/
	602	if (part != PART_AUTO) {
3e8bd469	603	ret = part_get_info(*dev_desc, part, info);
10a37fd7 SW	604	if (ret) {
	605	printf(" Invalid partition %d \n", part);
	606	goto cleanup;
	607	}
	608	} else {
	609	/*
	610	* Find the first bootable partition.
	611	* If none are bootable, fall back to the first valid partition.
	612	*/
	613	part = 0;
	614	for (p = 1; p <= MAX_SEARCH_PARTITIONS; p++) {
3e8bd469	615	ret = part_get_info(*dev_desc, p, info);
10a37fd7 SW	616	if (ret)
	617	continue;
	618
	619	/*
	620	* First valid partition, or new better partition?
	621	* If so, save partition ID.
	622	*/
	623	if (!part \|\| info->bootable)
	624	part = p;
	625
	626	/* Best possible partition? Stop searching. */
	627	if (info->bootable)
	628	break;
	629
	630	/*
	631	* We now need to search further for best possible.
	632	* If we what we just queried was the best so far,
	633	* save the info since we over-write it next loop.
	634	*/
	635	if (part == p)
	636	tmpinfo = *info;
	637	}
	638	/* If we found any acceptable partition */
	639	if (part) {
	640	/*
	641	* If we searched all possible partition IDs,
	642	* return the first valid partition we found.
	643	*/
	644	if (p == MAX_SEARCH_PARTITIONS + 1)
	645	*info = tmpinfo;
10a37fd7 SW	646	} else {
10a37fd7 SW	647	printf(" No valid partitions found \n");
71bba424	648	ret = -1;
10a37fd7 SW	649	goto cleanup;
10a37fd7 SW	650	}
99d2c205 RH	651	}
	652	if (strncmp((char *)info->type, BOOT_PART_TYPE, sizeof(info->type)) != 0) {
	653	printf("** Invalid partition type \"%.32s\""
	654	" (expect \"" BOOT_PART_TYPE "\")\n",
	655	info->type);
10a37fd7 SW	656	ret = -1;
10a37fd7 SW	657	goto cleanup;
99d2c205 RH	658	}
99d2c205 RH	659
50ffc3b6 LY	660	(dev_desc)->log2blksz = LOG2((dev_desc)->blksz);
50ffc3b6 LY	661
10a37fd7 SW	662	ret = part;
10a37fd7 SW	663	goto cleanup;
99d2c205	664
10a37fd7 SW	665	cleanup:
	666	free(dup_str);
	667	return ret;
99d2c205	668	}