[u-boot.git] / common / cmd_mmc.c

/*
 * (C) Copyright 2003
 * Kyle Harris, [email protected]
 *
 * SPDX-License-Identifier:	GPL-2.0+
 */

#include <common.h>
#include <command.h>
#include <mmc.h>

static int curr_device = -1;
#ifndef CONFIG_GENERIC_MMC
int do_mmc (cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
{
	int dev;

	if (argc < 2)
		return CMD_RET_USAGE;

	if (strcmp(argv[1], "init") == 0) {
		if (argc == 2) {
			if (curr_device < 0)
				dev = 1;
			else
				dev = curr_device;
		} else if (argc == 3) {
			dev = (int)simple_strtoul(argv[2], NULL, 10);
		} else {
			return CMD_RET_USAGE;
		}

		if (mmc_legacy_init(dev) != 0) {
			puts("No MMC card found\n");
			return 1;
		}

		curr_device = dev;
		printf("mmc%d is available\n", curr_device);
	} else if (strcmp(argv[1], "device") == 0) {
		if (argc == 2) {
			if (curr_device < 0) {
				puts("No MMC device available\n");
				return 1;
			}
		} else if (argc == 3) {
			dev = (int)simple_strtoul(argv[2], NULL, 10);

#ifdef CONFIG_SYS_MMC_SET_DEV
			if (mmc_set_dev(dev) != 0)
				return 1;
#endif
			curr_device = dev;
		} else {
			return CMD_RET_USAGE;
		}

		printf("mmc%d is current device\n", curr_device);
	} else {
		return CMD_RET_USAGE;
	}

	return 0;
}

U_BOOT_CMD(
	mmc, 3, 1, do_mmc,
	"MMC sub-system",
	"init [dev] - init MMC sub system\n"
	"mmc device [dev] - show or set current device"
);
#else /* !CONFIG_GENERIC_MMC */

static void print_mmcinfo(struct mmc *mmc)
{
	printf("Device: %s\n", mmc->cfg->name);
	printf("Manufacturer ID: %x\n", mmc->cid[0] >> 24);
	printf("OEM: %x\n", (mmc->cid[0] >> 8) & 0xffff);
	printf("Name: %c%c%c%c%c \n", mmc->cid[0] & 0xff,
			(mmc->cid[1] >> 24), (mmc->cid[1] >> 16) & 0xff,
			(mmc->cid[1] >> 8) & 0xff, mmc->cid[1] & 0xff);

	printf("Tran Speed: %d\n", mmc->tran_speed);
	printf("Rd Block Len: %d\n", mmc->read_bl_len);

	printf("%s version %d.%d\n", IS_SD(mmc) ? "SD" : "MMC",
			(mmc->version >> 8) & 0xf, mmc->version & 0xff);

	printf("High Capacity: %s\n", mmc->high_capacity ? "Yes" : "No");
	puts("Capacity: ");
	print_size(mmc->capacity, "\n");

	printf("Bus Width: %d-bit\n", mmc->bus_width);
}
static struct mmc *init_mmc_device(int dev, bool force_init)
{
	struct mmc *mmc;
	mmc = find_mmc_device(dev);
	if (!mmc) {
		printf("no mmc device at slot %x\n", dev);
		return NULL;
	}
	if (force_init)
		mmc->has_init = 0;
	if (mmc_init(mmc))
		return NULL;
	return mmc;
}
static int do_mmcinfo(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
{
	struct mmc *mmc;

	if (curr_device < 0) {
		if (get_mmc_num() > 0)
			curr_device = 0;
		else {
			puts("No MMC device available\n");
			return 1;
		}
	}

	mmc = init_mmc_device(curr_device, false);
	if (!mmc)
		return CMD_RET_FAILURE;

	print_mmcinfo(mmc);
	return CMD_RET_SUCCESS;
}

#ifdef CONFIG_SUPPORT_EMMC_RPMB
static int confirm_key_prog(void)
{
	puts("Warning: Programming authentication key can be done only once !\n"
	     "         Use this command only if you are sure of what you are doing,\n"
	     "Really perform the key programming? <y/N> ");
	if (confirm_yesno())
		return 1;

	puts("Authentication key programming aborted\n");
	return 0;
}
static int do_mmcrpmb_key(cmd_tbl_t *cmdtp, int flag,
			  int argc, char * const argv[])
{
	void *key_addr;
	struct mmc *mmc = find_mmc_device(curr_device);

	if (argc != 2)
		return CMD_RET_USAGE;

	key_addr = (void *)simple_strtoul(argv[1], NULL, 16);
	if (!confirm_key_prog())
		return CMD_RET_FAILURE;
	if (mmc_rpmb_set_key(mmc, key_addr)) {
		printf("ERROR - Key already programmed ?\n");
		return CMD_RET_FAILURE;
	}
	return CMD_RET_SUCCESS;
}
static int do_mmcrpmb_read(cmd_tbl_t *cmdtp, int flag,
			   int argc, char * const argv[])
{
	u16 blk, cnt;
	void *addr;
	int n;
	void *key_addr = NULL;
	struct mmc *mmc = find_mmc_device(curr_device);

	if (argc < 4)
		return CMD_RET_USAGE;

	addr = (void *)simple_strtoul(argv[1], NULL, 16);
	blk = simple_strtoul(argv[2], NULL, 16);
	cnt = simple_strtoul(argv[3], NULL, 16);

	if (argc == 5)
		key_addr = (void *)simple_strtoul(argv[4], NULL, 16);

	printf("\nMMC RPMB read: dev # %d, block # %d, count %d ... ",
	       curr_device, blk, cnt);
	n =  mmc_rpmb_read(mmc, addr, blk, cnt, key_addr);

	printf("%d RPMB blocks read: %s\n", n, (n == cnt) ? "OK" : "ERROR");
	if (n != cnt)
		return CMD_RET_FAILURE;
	return CMD_RET_SUCCESS;
}
static int do_mmcrpmb_write(cmd_tbl_t *cmdtp, int flag,
			    int argc, char * const argv[])
{
	u16 blk, cnt;
	void *addr;
	int n;
	void *key_addr;
	struct mmc *mmc = find_mmc_device(curr_device);

	if (argc != 5)
		return CMD_RET_USAGE;

	addr = (void *)simple_strtoul(argv[1], NULL, 16);
	blk = simple_strtoul(argv[2], NULL, 16);
	cnt = simple_strtoul(argv[3], NULL, 16);
	key_addr = (void *)simple_strtoul(argv[4], NULL, 16);

	printf("\nMMC RPMB write: dev # %d, block # %d, count %d ... ",
	       curr_device, blk, cnt);
	n =  mmc_rpmb_write(mmc, addr, blk, cnt, key_addr);

	printf("%d RPMB blocks written: %s\n", n, (n == cnt) ? "OK" : "ERROR");
	if (n != cnt)
		return CMD_RET_FAILURE;
	return CMD_RET_SUCCESS;
}
static int do_mmcrpmb_counter(cmd_tbl_t *cmdtp, int flag,
			      int argc, char * const argv[])
{
	unsigned long counter;
	struct mmc *mmc = find_mmc_device(curr_device);

	if (mmc_rpmb_get_counter(mmc, &counter))
		return CMD_RET_FAILURE;
	printf("RPMB Write counter= %lx\n", counter);
	return CMD_RET_SUCCESS;
}

static cmd_tbl_t cmd_rpmb[] = {
	U_BOOT_CMD_MKENT(key, 2, 0, do_mmcrpmb_key, "", ""),
	U_BOOT_CMD_MKENT(read, 5, 1, do_mmcrpmb_read, "", ""),
	U_BOOT_CMD_MKENT(write, 5, 0, do_mmcrpmb_write, "", ""),
	U_BOOT_CMD_MKENT(counter, 1, 1, do_mmcrpmb_counter, "", ""),
};

static int do_mmcrpmb(cmd_tbl_t *cmdtp, int flag,
		      int argc, char * const argv[])
{
	cmd_tbl_t *cp;
	struct mmc *mmc;
	char original_part;
	int ret;

	cp = find_cmd_tbl(argv[1], cmd_rpmb, ARRAY_SIZE(cmd_rpmb));

	/* Drop the rpmb subcommand */
	argc--;
	argv++;

	if (cp == NULL || argc > cp->maxargs)
		return CMD_RET_USAGE;
	if (flag == CMD_FLAG_REPEAT && !cp->repeatable)
		return CMD_RET_SUCCESS;

	mmc = init_mmc_device(curr_device, false);
	if (!mmc)
		return CMD_RET_FAILURE;

	if (!(mmc->version & MMC_VERSION_MMC)) {
		printf("It is not a EMMC device\n");
		return CMD_RET_FAILURE;
	}
	if (mmc->version < MMC_VERSION_4_41) {
		printf("RPMB not supported before version 4.41\n");
		return CMD_RET_FAILURE;
	}
	/* Switch to the RPMB partition */
	original_part = mmc->part_num;
	if (mmc->part_num != MMC_PART_RPMB) {
		if (mmc_switch_part(curr_device, MMC_PART_RPMB) != 0)
			return CMD_RET_FAILURE;
		mmc->part_num = MMC_PART_RPMB;
	}
	ret = cp->cmd(cmdtp, flag, argc, argv);

	/* Return to original partition */
	if (mmc->part_num != original_part) {
		if (mmc_switch_part(curr_device, original_part) != 0)
			return CMD_RET_FAILURE;
		mmc->part_num = original_part;
	}
	return ret;
}
#endif

static int do_mmc_read(cmd_tbl_t *cmdtp, int flag,
		       int argc, char * const argv[])
{
	struct mmc *mmc;
	u32 blk, cnt, n;
	void *addr;

	if (argc != 4)
		return CMD_RET_USAGE;

	addr = (void *)simple_strtoul(argv[1], NULL, 16);
	blk = simple_strtoul(argv[2], NULL, 16);
	cnt = simple_strtoul(argv[3], NULL, 16);

	mmc = init_mmc_device(curr_device, false);
	if (!mmc)
		return CMD_RET_FAILURE;

	printf("\nMMC read: dev # %d, block # %d, count %d ... ",
	       curr_device, blk, cnt);

	n = mmc->block_dev.block_read(curr_device, blk, cnt, addr);
	/* flush cache after read */
	flush_cache((ulong)addr, cnt * 512); /* FIXME */
	printf("%d blocks read: %s\n", n, (n == cnt) ? "OK" : "ERROR");

	return (n == cnt) ? CMD_RET_SUCCESS : CMD_RET_FAILURE;
}
static int do_mmc_write(cmd_tbl_t *cmdtp, int flag,
			int argc, char * const argv[])
{
	struct mmc *mmc;
	u32 blk, cnt, n;
	void *addr;

	if (argc != 4)
		return CMD_RET_USAGE;

	addr = (void *)simple_strtoul(argv[1], NULL, 16);
	blk = simple_strtoul(argv[2], NULL, 16);
	cnt = simple_strtoul(argv[3], NULL, 16);

	mmc = init_mmc_device(curr_device, false);
	if (!mmc)
		return CMD_RET_FAILURE;

	printf("\nMMC write: dev # %d, block # %d, count %d ... ",
	       curr_device, blk, cnt);

	if (mmc_getwp(mmc) == 1) {
		printf("Error: card is write protected!\n");
		return CMD_RET_FAILURE;
	}
	n = mmc->block_dev.block_write(curr_device, blk, cnt, addr);
	printf("%d blocks written: %s\n", n, (n == cnt) ? "OK" : "ERROR");

	return (n == cnt) ? CMD_RET_SUCCESS : CMD_RET_FAILURE;
}
static int do_mmc_erase(cmd_tbl_t *cmdtp, int flag,
			int argc, char * const argv[])
{
	struct mmc *mmc;
	u32 blk, cnt, n;

	if (argc != 3)
		return CMD_RET_USAGE;

	blk = simple_strtoul(argv[1], NULL, 16);
	cnt = simple_strtoul(argv[2], NULL, 16);

	mmc = init_mmc_device(curr_device, false);
	if (!mmc)
		return CMD_RET_FAILURE;

	printf("\nMMC erase: dev # %d, block # %d, count %d ... ",
	       curr_device, blk, cnt);

	if (mmc_getwp(mmc) == 1) {
		printf("Error: card is write protected!\n");
		return CMD_RET_FAILURE;
	}
	n = mmc->block_dev.block_erase(curr_device, blk, cnt);
	printf("%d blocks erased: %s\n", n, (n == cnt) ? "OK" : "ERROR");

	return (n == cnt) ? CMD_RET_SUCCESS : CMD_RET_FAILURE;
}
static int do_mmc_rescan(cmd_tbl_t *cmdtp, int flag,
			 int argc, char * const argv[])
{
	struct mmc *mmc;

	mmc = init_mmc_device(curr_device, true);
	if (!mmc)
		return CMD_RET_FAILURE;

	return CMD_RET_SUCCESS;
}
static int do_mmc_part(cmd_tbl_t *cmdtp, int flag,
		       int argc, char * const argv[])
{
	block_dev_desc_t *mmc_dev;
	struct mmc *mmc;

	mmc = init_mmc_device(curr_device, false);
	if (!mmc)
		return CMD_RET_FAILURE;

	mmc_dev = mmc_get_dev(curr_device);
	if (mmc_dev != NULL && mmc_dev->type != DEV_TYPE_UNKNOWN) {
		print_part(mmc_dev);
		return CMD_RET_SUCCESS;
	}

	puts("get mmc type error!\n");
	return CMD_RET_FAILURE;
}
static int do_mmc_dev(cmd_tbl_t *cmdtp, int flag,
		      int argc, char * const argv[])
{
	int dev, part = 0, ret;
	struct mmc *mmc;

	if (argc == 1) {
		dev = curr_device;
	} else if (argc == 2) {
		dev = simple_strtoul(argv[1], NULL, 10);
	} else if (argc == 3) {
		dev = (int)simple_strtoul(argv[1], NULL, 10);
		part = (int)simple_strtoul(argv[2], NULL, 10);
		if (part > PART_ACCESS_MASK) {
			printf("#part_num shouldn't be larger than %d\n",
			       PART_ACCESS_MASK);
			return CMD_RET_FAILURE;
		}
	} else {
		return CMD_RET_USAGE;
	}

	mmc = init_mmc_device(dev, true);
	if (!mmc)
		return CMD_RET_FAILURE;

	ret = mmc_select_hwpart(dev, part);
	printf("switch to partitions #%d, %s\n",
	       part, (!ret) ? "OK" : "ERROR");
	if (ret)
		return 1;

	curr_device = dev;
	if (mmc->part_config == MMCPART_NOAVAILABLE)
		printf("mmc%d is current device\n", curr_device);
	else
		printf("mmc%d(part %d) is current device\n",
		       curr_device, mmc->part_num);

	return CMD_RET_SUCCESS;
}
static int do_mmc_list(cmd_tbl_t *cmdtp, int flag,
		       int argc, char * const argv[])
{
	print_mmc_devices('\n');
	return CMD_RET_SUCCESS;
}
#ifdef CONFIG_SUPPORT_EMMC_BOOT
static int do_mmc_bootbus(cmd_tbl_t *cmdtp, int flag,
			  int argc, char * const argv[])
{
	int dev;
	struct mmc *mmc;
	u8 width, reset, mode;

	if (argc != 5)
		return CMD_RET_USAGE;
	dev = simple_strtoul(argv[1], NULL, 10);
	width = simple_strtoul(argv[2], NULL, 10);
	reset = simple_strtoul(argv[3], NULL, 10);
	mode = simple_strtoul(argv[4], NULL, 10);

	mmc = init_mmc_device(dev, false);
	if (!mmc)
		return CMD_RET_FAILURE;

	if (IS_SD(mmc)) {
		puts("BOOT_BUS_WIDTH only exists on eMMC\n");
		return CMD_RET_FAILURE;
	}

	/* acknowledge to be sent during boot operation */
	return mmc_set_boot_bus_width(mmc, width, reset, mode);
}
static int do_mmc_boot_resize(cmd_tbl_t *cmdtp, int flag,
			      int argc, char * const argv[])
{
	int dev;
	struct mmc *mmc;
	u32 bootsize, rpmbsize;

	if (argc != 4)
		return CMD_RET_USAGE;
	dev = simple_strtoul(argv[1], NULL, 10);
	bootsize = simple_strtoul(argv[2], NULL, 10);
	rpmbsize = simple_strtoul(argv[3], NULL, 10);

	mmc = init_mmc_device(dev, false);
	if (!mmc)
		return CMD_RET_FAILURE;

	if (IS_SD(mmc)) {
		printf("It is not a EMMC device\n");
		return CMD_RET_FAILURE;
	}

	if (mmc_boot_partition_size_change(mmc, bootsize, rpmbsize)) {
		printf("EMMC boot partition Size change Failed.\n");
		return CMD_RET_FAILURE;
	}

	printf("EMMC boot partition Size %d MB\n", bootsize);
	printf("EMMC RPMB partition Size %d MB\n", rpmbsize);
	return CMD_RET_SUCCESS;
}
static int do_mmc_partconf(cmd_tbl_t *cmdtp, int flag,
			   int argc, char * const argv[])
{
	int dev;
	struct mmc *mmc;
	u8 ack, part_num, access;

	if (argc != 5)
		return CMD_RET_USAGE;

	dev = simple_strtoul(argv[1], NULL, 10);
	ack = simple_strtoul(argv[2], NULL, 10);
	part_num = simple_strtoul(argv[3], NULL, 10);
	access = simple_strtoul(argv[4], NULL, 10);

	mmc = init_mmc_device(dev, false);
	if (!mmc)
		return CMD_RET_FAILURE;

	if (IS_SD(mmc)) {
		puts("PARTITION_CONFIG only exists on eMMC\n");
		return CMD_RET_FAILURE;
	}

	/* acknowledge to be sent during boot operation */
	return mmc_set_part_conf(mmc, ack, part_num, access);
}
static int do_mmc_rst_func(cmd_tbl_t *cmdtp, int flag,
			   int argc, char * const argv[])
{
	int dev;
	struct mmc *mmc;
	u8 enable;

	/*
	 * Set the RST_n_ENABLE bit of RST_n_FUNCTION
	 * The only valid values are 0x0, 0x1 and 0x2 and writing
	 * a value of 0x1 or 0x2 sets the value permanently.
	 */
	if (argc != 3)
		return CMD_RET_USAGE;

	dev = simple_strtoul(argv[1], NULL, 10);
	enable = simple_strtoul(argv[2], NULL, 10);

	if (enable > 2 || enable < 0) {
		puts("Invalid RST_n_ENABLE value\n");
		return CMD_RET_USAGE;
	}

	mmc = init_mmc_device(dev, false);
	if (!mmc)
		return CMD_RET_FAILURE;

	if (IS_SD(mmc)) {
		puts("RST_n_FUNCTION only exists on eMMC\n");
		return CMD_RET_FAILURE;
	}

	return mmc_set_rst_n_function(mmc, enable);
}
#endif
static int do_mmc_setdsr(cmd_tbl_t *cmdtp, int flag,
			 int argc, char * const argv[])
{
	struct mmc *mmc;
	u32 val;
	int ret;

	if (argc != 2)
		return CMD_RET_USAGE;
	val = simple_strtoul(argv[2], NULL, 16);

	mmc = find_mmc_device(curr_device);
	if (!mmc) {
		printf("no mmc device at slot %x\n", curr_device);
		return CMD_RET_FAILURE;
	}
	ret = mmc_set_dsr(mmc, val);
	printf("set dsr %s\n", (!ret) ? "OK, force rescan" : "ERROR");
	if (!ret) {
		mmc->has_init = 0;
		if (mmc_init(mmc))
			return CMD_RET_FAILURE;
		else
			return CMD_RET_SUCCESS;
	}
	return ret;
}

static cmd_tbl_t cmd_mmc[] = {
	U_BOOT_CMD_MKENT(info, 1, 0, do_mmcinfo, "", ""),
	U_BOOT_CMD_MKENT(read, 4, 1, do_mmc_read, "", ""),
	U_BOOT_CMD_MKENT(write, 4, 0, do_mmc_write, "", ""),
	U_BOOT_CMD_MKENT(erase, 3, 0, do_mmc_erase, "", ""),
	U_BOOT_CMD_MKENT(rescan, 1, 1, do_mmc_rescan, "", ""),
	U_BOOT_CMD_MKENT(part, 1, 1, do_mmc_part, "", ""),
	U_BOOT_CMD_MKENT(dev, 3, 0, do_mmc_dev, "", ""),
	U_BOOT_CMD_MKENT(list, 1, 1, do_mmc_list, "", ""),
#ifdef CONFIG_SUPPORT_EMMC_BOOT
	U_BOOT_CMD_MKENT(bootbus, 5, 0, do_mmc_bootbus, "", ""),
	U_BOOT_CMD_MKENT(bootpart-resize, 4, 0, do_mmc_boot_resize, "", ""),
	U_BOOT_CMD_MKENT(partconf, 5, 0, do_mmc_partconf, "", ""),
	U_BOOT_CMD_MKENT(rst-function, 3, 0, do_mmc_rst_func, "", ""),
#endif
#ifdef CONFIG_SUPPORT_EMMC_RPMB
	U_BOOT_CMD_MKENT(rpmb, CONFIG_SYS_MAXARGS, 1, do_mmcrpmb, "", ""),
#endif
	U_BOOT_CMD_MKENT(setdsr, 2, 0, do_mmc_setdsr, "", ""),
};

static int do_mmcops(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
{
	cmd_tbl_t *cp;

	cp = find_cmd_tbl(argv[1], cmd_mmc, ARRAY_SIZE(cmd_mmc));

	/* Drop the mmc command */
	argc--;
	argv++;

	if (cp == NULL || argc > cp->maxargs)
		return CMD_RET_USAGE;
	if (flag == CMD_FLAG_REPEAT && !cp->repeatable)
		return CMD_RET_SUCCESS;

	if (curr_device < 0) {
		if (get_mmc_num() > 0) {
			curr_device = 0;
		} else {
			puts("No MMC device available\n");
			return CMD_RET_FAILURE;
		}
	}
	return cp->cmd(cmdtp, flag, argc, argv);
}

U_BOOT_CMD(
	mmc, 7, 1, do_mmcops,
	"MMC sub system",
	"info - display info of the current MMC device\n"
	"mmc read addr blk# cnt\n"
	"mmc write addr blk# cnt\n"
	"mmc erase blk# cnt\n"
	"mmc rescan\n"
	"mmc part - lists available partition on current mmc device\n"
	"mmc dev [dev] [part] - show or set current mmc device [partition]\n"
	"mmc list - lists available devices\n"
#ifdef CONFIG_SUPPORT_EMMC_BOOT
	"mmc bootbus dev boot_bus_width reset_boot_bus_width boot_mode\n"
	" - Set the BOOT_BUS_WIDTH field of the specified device\n"
	"mmc bootpart-resize <dev> <boot part size MB> <RPMB part size MB>\n"
	" - Change sizes of boot and RPMB partitions of specified device\n"
	"mmc partconf dev boot_ack boot_partition partition_access\n"
	" - Change the bits of the PARTITION_CONFIG field of the specified device\n"
	"mmc rst-function dev value\n"
	" - Change the RST_n_FUNCTION field of the specified device\n"
	"   WARNING: This is a write-once field and 0 / 1 / 2 are the only valid values.\n"
#endif
#ifdef CONFIG_SUPPORT_EMMC_RPMB
	"mmc rpmb read addr blk# cnt [address of auth-key] - block size is 256 bytes\n"
	"mmc rpmb write addr blk# cnt <address of auth-key> - block size is 256 bytes\n"
	"mmc rpmb key <address of auth-key> - program the RPMB authentication key.\n"
	"mmc rpmb counter - read the value of the write counter\n"
#endif
	"mmc setdsr <value> - set DSR register value\n"
	);

/* Old command kept for compatibility. Same as 'mmc info' */
U_BOOT_CMD(
	mmcinfo, 1, 0, do_mmcinfo,
	"display MMC info",
	"- display info of the current MMC device"
);

#endif /* !CONFIG_GENERIC_MMC */
Commit	Line	Data
71f95118 WD	1	/*
	2	* (C) Copyright 2003
	3	* Kyle Harris, [email protected]
	4	*
1a459660	5	* SPDX-License-Identifier: GPL-2.0+
71f95118 WD	6	*/
	7
	8	#include <common.h>
	9	#include <command.h>
71f95118 WD	10	#include <mmc.h>
71f95118 WD	11
02e22c2d	12	static int curr_device = -1;
ea6ebe21	13	#ifndef CONFIG_GENERIC_MMC
54841ab5	14	int do_mmc (cmd_tbl_t cmdtp, int flag, int argc, char const argv[])
71f95118	15	{
869f6bf4 MK	16	int dev;
869f6bf4 MK	17
47e26b1b	18	if (argc < 2)
4c12eeb8	19	return CMD_RET_USAGE;
869f6bf4 MK	20
	21	if (strcmp(argv[1], "init") == 0) {
	22	if (argc == 2) {
	23	if (curr_device < 0)
	24	dev = 1;
	25	else
	26	dev = curr_device;
	27	} else if (argc == 3) {
	28	dev = (int)simple_strtoul(argv[2], NULL, 10);
	29	} else {
4c12eeb8	30	return CMD_RET_USAGE;
869f6bf4 MK	31	}
	32
	33	if (mmc_legacy_init(dev) != 0) {
	34	puts("No MMC card found\n");
	35	return 1;
	36	}
	37
	38	curr_device = dev;
	39	printf("mmc%d is available\n", curr_device);
	40	} else if (strcmp(argv[1], "device") == 0) {
	41	if (argc == 2) {
	42	if (curr_device < 0) {
	43	puts("No MMC device available\n");
	44	return 1;
	45	}
	46	} else if (argc == 3) {
	47	dev = (int)simple_strtoul(argv[2], NULL, 10);
	48
	49	#ifdef CONFIG_SYS_MMC_SET_DEV
	50	if (mmc_set_dev(dev) != 0)
	51	return 1;
	52	#endif
	53	curr_device = dev;
	54	} else {
4c12eeb8	55	return CMD_RET_USAGE;
869f6bf4 MK	56	}
	57
	58	printf("mmc%d is current device\n", curr_device);
	59	} else {
4c12eeb8	60	return CMD_RET_USAGE;
71f95118	61	}
869f6bf4	62
71f95118 WD	63	return 0;
	64	}
	65
0d498393	66	U_BOOT_CMD(
869f6bf4 MK	67	mmc, 3, 1, do_mmc,
	68	"MMC sub-system",
	69	"init [dev] - init MMC sub system\n"
a89c33db	70	"mmc device [dev] - show or set current device"
b0fce99b	71	);
3511b4e2	72	#else /* !CONFIG_GENERIC_MMC */
272cc70b AF	73
	74	static void print_mmcinfo(struct mmc *mmc)
	75	{
93bfd616	76	printf("Device: %s\n", mmc->cfg->name);
272cc70b AF	77	printf("Manufacturer ID: %x\n", mmc->cid[0] >> 24);
	78	printf("OEM: %x\n", (mmc->cid[0] >> 8) & 0xffff);
	79	printf("Name: %c%c%c%c%c \n", mmc->cid[0] & 0xff,
	80	(mmc->cid[1] >> 24), (mmc->cid[1] >> 16) & 0xff,
	81	(mmc->cid[1] >> 8) & 0xff, mmc->cid[1] & 0xff);
	82
	83	printf("Tran Speed: %d\n", mmc->tran_speed);
	84	printf("Rd Block Len: %d\n", mmc->read_bl_len);
	85
	86	printf("%s version %d.%d\n", IS_SD(mmc) ? "SD" : "MMC",
64f4a619	87	(mmc->version >> 8) & 0xf, mmc->version & 0xff);
272cc70b AF	88
272cc70b AF	89	printf("High Capacity: %s\n", mmc->high_capacity ? "Yes" : "No");
940e0782 MK	90	puts("Capacity: ");
940e0782 MK	91	print_size(mmc->capacity, "\n");
272cc70b AF	92
	93	printf("Bus Width: %d-bit\n", mmc->bus_width);
	94	}
1ae24a50	95	static struct mmc *init_mmc_device(int dev, bool force_init)
1fd93c6e PA	96	{
	97	struct mmc *mmc;
	98	mmc = find_mmc_device(dev);
	99	if (!mmc) {
	100	printf("no mmc device at slot %x\n", dev);
	101	return NULL;
	102	}
1ae24a50 SW	103	if (force_init)
1ae24a50 SW	104	mmc->has_init = 0;
1fd93c6e PA	105	if (mmc_init(mmc))
	106	return NULL;
	107	return mmc;
	108	}
088f1b19	109	static int do_mmcinfo(cmd_tbl_t cmdtp, int flag, int argc, char const argv[])
272cc70b AF	110	{
272cc70b AF	111	struct mmc *mmc;
272cc70b	112
ea6ebe21 LW	113	if (curr_device < 0) {
	114	if (get_mmc_num() > 0)
	115	curr_device = 0;
	116	else {
	117	puts("No MMC device available\n");
	118	return 1;
	119	}
	120	}
272cc70b	121
1ae24a50	122	mmc = init_mmc_device(curr_device, false);
1fd93c6e PA	123	if (!mmc)
1fd93c6e PA	124	return CMD_RET_FAILURE;
272cc70b	125
1fd93c6e PA	126	print_mmcinfo(mmc);
	127	return CMD_RET_SUCCESS;
	128	}
272cc70b	129
1fd93c6e PA	130	#ifdef CONFIG_SUPPORT_EMMC_RPMB
	131	static int confirm_key_prog(void)
	132	{
	133	puts("Warning: Programming authentication key can be done only once !\n"
	134	" Use this command only if you are sure of what you are doing,\n"
	135	"Really perform the key programming? <y/N> ");
	136	if (confirm_yesno())
ea6ebe21	137	return 1;
1fd93c6e PA	138
	139	puts("Authentication key programming aborted\n");
	140	return 0;
	141	}
	142	static int do_mmcrpmb_key(cmd_tbl_t *cmdtp, int flag,
	143	int argc, char * const argv[])
	144	{
	145	void *key_addr;
	146	struct mmc *mmc = find_mmc_device(curr_device);
	147
	148	if (argc != 2)
	149	return CMD_RET_USAGE;
	150
	151	key_addr = (void *)simple_strtoul(argv[1], NULL, 16);
	152	if (!confirm_key_prog())
	153	return CMD_RET_FAILURE;
	154	if (mmc_rpmb_set_key(mmc, key_addr)) {
	155	printf("ERROR - Key already programmed ?\n");
	156	return CMD_RET_FAILURE;
272cc70b	157	}
1fd93c6e	158	return CMD_RET_SUCCESS;
272cc70b	159	}
1fd93c6e PA	160	static int do_mmcrpmb_read(cmd_tbl_t *cmdtp, int flag,
	161	int argc, char * const argv[])
	162	{
	163	u16 blk, cnt;
	164	void *addr;
	165	int n;
	166	void *key_addr = NULL;
	167	struct mmc *mmc = find_mmc_device(curr_device);
272cc70b	168
1fd93c6e PA	169	if (argc < 4)
1fd93c6e PA	170	return CMD_RET_USAGE;
272cc70b	171
1fd93c6e PA	172	addr = (void *)simple_strtoul(argv[1], NULL, 16);
	173	blk = simple_strtoul(argv[2], NULL, 16);
	174	cnt = simple_strtoul(argv[3], NULL, 16);
	175
	176	if (argc == 5)
	177	key_addr = (void *)simple_strtoul(argv[4], NULL, 16);
	178
	179	printf("\nMMC RPMB read: dev # %d, block # %d, count %d ... ",
	180	curr_device, blk, cnt);
	181	n = mmc_rpmb_read(mmc, addr, blk, cnt, key_addr);
	182
	183	printf("%d RPMB blocks read: %s\n", n, (n == cnt) ? "OK" : "ERROR");
	184	if (n != cnt)
	185	return CMD_RET_FAILURE;
	186	return CMD_RET_SUCCESS;
	187	}
	188	static int do_mmcrpmb_write(cmd_tbl_t *cmdtp, int flag,
	189	int argc, char * const argv[])
272cc70b	190	{
1fd93c6e PA	191	u16 blk, cnt;
	192	void *addr;
	193	int n;
	194	void *key_addr;
	195	struct mmc *mmc = find_mmc_device(curr_device);
6be95ccf	196
1fd93c6e	197	if (argc != 5)
4c12eeb8	198	return CMD_RET_USAGE;
272cc70b	199
1fd93c6e PA	200	addr = (void *)simple_strtoul(argv[1], NULL, 16);
	201	blk = simple_strtoul(argv[2], NULL, 16);
	202	cnt = simple_strtoul(argv[3], NULL, 16);
	203	key_addr = (void *)simple_strtoul(argv[4], NULL, 16);
	204
	205	printf("\nMMC RPMB write: dev # %d, block # %d, count %d ... ",
	206	curr_device, blk, cnt);
	207	n = mmc_rpmb_write(mmc, addr, blk, cnt, key_addr);
	208
	209	printf("%d RPMB blocks written: %s\n", n, (n == cnt) ? "OK" : "ERROR");
	210	if (n != cnt)
	211	return CMD_RET_FAILURE;
	212	return CMD_RET_SUCCESS;
	213	}
	214	static int do_mmcrpmb_counter(cmd_tbl_t *cmdtp, int flag,
	215	int argc, char * const argv[])
	216	{
	217	unsigned long counter;
	218	struct mmc *mmc = find_mmc_device(curr_device);
	219
	220	if (mmc_rpmb_get_counter(mmc, &counter))
	221	return CMD_RET_FAILURE;
	222	printf("RPMB Write counter= %lx\n", counter);
	223	return CMD_RET_SUCCESS;
	224	}
	225
	226	static cmd_tbl_t cmd_rpmb[] = {
	227	U_BOOT_CMD_MKENT(key, 2, 0, do_mmcrpmb_key, "", ""),
	228	U_BOOT_CMD_MKENT(read, 5, 1, do_mmcrpmb_read, "", ""),
	229	U_BOOT_CMD_MKENT(write, 5, 0, do_mmcrpmb_write, "", ""),
	230	U_BOOT_CMD_MKENT(counter, 1, 1, do_mmcrpmb_counter, "", ""),
	231	};
	232
	233	static int do_mmcrpmb(cmd_tbl_t *cmdtp, int flag,
	234	int argc, char * const argv[])
	235	{
	236	cmd_tbl_t *cp;
	237	struct mmc *mmc;
	238	char original_part;
	239	int ret;
	240
	241	cp = find_cmd_tbl(argv[1], cmd_rpmb, ARRAY_SIZE(cmd_rpmb));
	242
	243	/* Drop the rpmb subcommand */
	244	argc--;
	245	argv++;
	246
	247	if (cp == NULL \|\| argc > cp->maxargs)
	248	return CMD_RET_USAGE;
	249	if (flag == CMD_FLAG_REPEAT && !cp->repeatable)
	250	return CMD_RET_SUCCESS;
	251
1ae24a50	252	mmc = init_mmc_device(curr_device, false);
1fd93c6e PA	253	if (!mmc)
	254	return CMD_RET_FAILURE;
	255
	256	if (!(mmc->version & MMC_VERSION_MMC)) {
	257	printf("It is not a EMMC device\n");
	258	return CMD_RET_FAILURE;
	259	}
	260	if (mmc->version < MMC_VERSION_4_41) {
	261	printf("RPMB not supported before version 4.41\n");
	262	return CMD_RET_FAILURE;
ea6ebe21	263	}
1fd93c6e PA	264	/* Switch to the RPMB partition */
	265	original_part = mmc->part_num;
	266	if (mmc->part_num != MMC_PART_RPMB) {
	267	if (mmc_switch_part(curr_device, MMC_PART_RPMB) != 0)
	268	return CMD_RET_FAILURE;
	269	mmc->part_num = MMC_PART_RPMB;
	270	}
	271	ret = cp->cmd(cmdtp, flag, argc, argv);
272cc70b	272
1fd93c6e PA	273	/* Return to original partition */
	274	if (mmc->part_num != original_part) {
	275	if (mmc_switch_part(curr_device, original_part) != 0)
	276	return CMD_RET_FAILURE;
	277	mmc->part_num = original_part;
	278	}
	279	return ret;
	280	}
	281	#endif
9fd38372	282
1fd93c6e PA	283	static int do_mmc_read(cmd_tbl_t *cmdtp, int flag,
	284	int argc, char * const argv[])
	285	{
	286	struct mmc *mmc;
	287	u32 blk, cnt, n;
	288	void *addr;
e85649c7	289
1fd93c6e PA	290	if (argc != 4)
1fd93c6e PA	291	return CMD_RET_USAGE;
ea6ebe21	292
1fd93c6e PA	293	addr = (void *)simple_strtoul(argv[1], NULL, 16);
	294	blk = simple_strtoul(argv[2], NULL, 16);
	295	cnt = simple_strtoul(argv[3], NULL, 16);
272cc70b	296
1ae24a50	297	mmc = init_mmc_device(curr_device, false);
1fd93c6e PA	298	if (!mmc)
1fd93c6e PA	299	return CMD_RET_FAILURE;
ea6ebe21	300
1fd93c6e PA	301	printf("\nMMC read: dev # %d, block # %d, count %d ... ",
1fd93c6e PA	302	curr_device, blk, cnt);
9fd38372	303
1fd93c6e PA	304	n = mmc->block_dev.block_read(curr_device, blk, cnt, addr);
	305	/* flush cache after read */
	306	flush_cache((ulong)addr, cnt * 512); /* FIXME */
	307	printf("%d blocks read: %s\n", n, (n == cnt) ? "OK" : "ERROR");
8f3b9642	308
1fd93c6e PA	309	return (n == cnt) ? CMD_RET_SUCCESS : CMD_RET_FAILURE;
	310	}
	311	static int do_mmc_write(cmd_tbl_t *cmdtp, int flag,
	312	int argc, char * const argv[])
	313	{
	314	struct mmc *mmc;
	315	u32 blk, cnt, n;
	316	void *addr;
8f3b9642	317
1fd93c6e PA	318	if (argc != 4)
1fd93c6e PA	319	return CMD_RET_USAGE;
272cc70b	320
1fd93c6e PA	321	addr = (void *)simple_strtoul(argv[1], NULL, 16);
	322	blk = simple_strtoul(argv[2], NULL, 16);
	323	cnt = simple_strtoul(argv[3], NULL, 16);
bc897b1d	324
1ae24a50	325	mmc = init_mmc_device(curr_device, false);
1fd93c6e PA	326	if (!mmc)
1fd93c6e PA	327	return CMD_RET_FAILURE;
bc897b1d	328
1fd93c6e PA	329	printf("\nMMC write: dev # %d, block # %d, count %d ... ",
1fd93c6e PA	330	curr_device, blk, cnt);
272cc70b	331
1fd93c6e PA	332	if (mmc_getwp(mmc) == 1) {
	333	printf("Error: card is write protected!\n");
	334	return CMD_RET_FAILURE;
	335	}
	336	n = mmc->block_dev.block_write(curr_device, blk, cnt, addr);
	337	printf("%d blocks written: %s\n", n, (n == cnt) ? "OK" : "ERROR");
792970b0	338
1fd93c6e PA	339	return (n == cnt) ? CMD_RET_SUCCESS : CMD_RET_FAILURE;
	340	}
	341	static int do_mmc_erase(cmd_tbl_t *cmdtp, int flag,
	342	int argc, char * const argv[])
	343	{
	344	struct mmc *mmc;
	345	u32 blk, cnt, n;
2a91c913	346
1fd93c6e PA	347	if (argc != 3)
1fd93c6e PA	348	return CMD_RET_USAGE;
792970b0	349
1fd93c6e PA	350	blk = simple_strtoul(argv[1], NULL, 16);
1fd93c6e PA	351	cnt = simple_strtoul(argv[2], NULL, 16);
5a99b9de	352
1ae24a50	353	mmc = init_mmc_device(curr_device, false);
1fd93c6e PA	354	if (!mmc)
1fd93c6e PA	355	return CMD_RET_FAILURE;
5a99b9de	356
1fd93c6e PA	357	printf("\nMMC erase: dev # %d, block # %d, count %d ... ",
1fd93c6e PA	358	curr_device, blk, cnt);
5a99b9de	359
1fd93c6e PA	360	if (mmc_getwp(mmc) == 1) {
	361	printf("Error: card is write protected!\n");
	362	return CMD_RET_FAILURE;
	363	}
	364	n = mmc->block_dev.block_erase(curr_device, blk, cnt);
	365	printf("%d blocks erased: %s\n", n, (n == cnt) ? "OK" : "ERROR");
b01e6fe6	366
1fd93c6e PA	367	return (n == cnt) ? CMD_RET_SUCCESS : CMD_RET_FAILURE;
	368	}
	369	static int do_mmc_rescan(cmd_tbl_t *cmdtp, int flag,
	370	int argc, char * const argv[])
	371	{
	372	struct mmc *mmc;
2a91c913	373
941944e4 SW	374	mmc = init_mmc_device(curr_device, true);
941944e4 SW	375	if (!mmc)
1fd93c6e	376	return CMD_RET_FAILURE;
2a91c913	377
1fd93c6e PA	378	return CMD_RET_SUCCESS;
	379	}
	380	static int do_mmc_part(cmd_tbl_t *cmdtp, int flag,
	381	int argc, char * const argv[])
	382	{
	383	block_dev_desc_t *mmc_dev;
	384	struct mmc *mmc;
	385
1ae24a50	386	mmc = init_mmc_device(curr_device, false);
1fd93c6e PA	387	if (!mmc)
	388	return CMD_RET_FAILURE;
	389
	390	mmc_dev = mmc_get_dev(curr_device);
	391	if (mmc_dev != NULL && mmc_dev->type != DEV_TYPE_UNKNOWN) {
	392	print_part(mmc_dev);
	393	return CMD_RET_SUCCESS;
	394	}
	395
	396	puts("get mmc type error!\n");
	397	return CMD_RET_FAILURE;
	398	}
	399	static int do_mmc_dev(cmd_tbl_t *cmdtp, int flag,
	400	int argc, char * const argv[])
	401	{
60dc58f7	402	int dev, part = 0, ret;
1fd93c6e PA	403	struct mmc *mmc;
	404
	405	if (argc == 1) {
	406	dev = curr_device;
	407	} else if (argc == 2) {
	408	dev = simple_strtoul(argv[1], NULL, 10);
	409	} else if (argc == 3) {
	410	dev = (int)simple_strtoul(argv[1], NULL, 10);
	411	part = (int)simple_strtoul(argv[2], NULL, 10);
	412	if (part > PART_ACCESS_MASK) {
	413	printf("#part_num shouldn't be larger than %d\n",
	414	PART_ACCESS_MASK);
	415	return CMD_RET_FAILURE;
33ace362	416	}
1fd93c6e PA	417	} else {
	418	return CMD_RET_USAGE;
	419	}
33ace362	420
a5710920	421	mmc = init_mmc_device(dev, true);
1fd93c6e PA	422	if (!mmc)
	423	return CMD_RET_FAILURE;
	424
60dc58f7 SW	425	ret = mmc_select_hwpart(dev, part);
	426	printf("switch to partitions #%d, %s\n",
	427	part, (!ret) ? "OK" : "ERROR");
	428	if (ret)
	429	return 1;
	430
1fd93c6e PA	431	curr_device = dev;
	432	if (mmc->part_config == MMCPART_NOAVAILABLE)
	433	printf("mmc%d is current device\n", curr_device);
	434	else
	435	printf("mmc%d(part %d) is current device\n",
	436	curr_device, mmc->part_num);
33ace362	437
1fd93c6e PA	438	return CMD_RET_SUCCESS;
	439	}
	440	static int do_mmc_list(cmd_tbl_t *cmdtp, int flag,
	441	int argc, char * const argv[])
	442	{
	443	print_mmc_devices('\n');
	444	return CMD_RET_SUCCESS;
	445	}
	446	#ifdef CONFIG_SUPPORT_EMMC_BOOT
	447	static int do_mmc_bootbus(cmd_tbl_t *cmdtp, int flag,
	448	int argc, char * const argv[])
	449	{
	450	int dev;
	451	struct mmc *mmc;
	452	u8 width, reset, mode;
	453
	454	if (argc != 5)
	455	return CMD_RET_USAGE;
	456	dev = simple_strtoul(argv[1], NULL, 10);
	457	width = simple_strtoul(argv[2], NULL, 10);
	458	reset = simple_strtoul(argv[3], NULL, 10);
	459	mode = simple_strtoul(argv[4], NULL, 10);
	460
1ae24a50	461	mmc = init_mmc_device(dev, false);
1fd93c6e PA	462	if (!mmc)
	463	return CMD_RET_FAILURE;
	464
	465	if (IS_SD(mmc)) {
	466	puts("BOOT_BUS_WIDTH only exists on eMMC\n");
	467	return CMD_RET_FAILURE;
2a91c913	468	}
ab71188c	469
1fd93c6e PA	470	/* acknowledge to be sent during boot operation */
	471	return mmc_set_boot_bus_width(mmc, width, reset, mode);
	472	}
	473	static int do_mmc_boot_resize(cmd_tbl_t *cmdtp, int flag,
	474	int argc, char * const argv[])
	475	{
	476	int dev;
	477	struct mmc *mmc;
	478	u32 bootsize, rpmbsize;
ab71188c	479
1fd93c6e PA	480	if (argc != 4)
	481	return CMD_RET_USAGE;
	482	dev = simple_strtoul(argv[1], NULL, 10);
	483	bootsize = simple_strtoul(argv[2], NULL, 10);
	484	rpmbsize = simple_strtoul(argv[3], NULL, 10);
	485
1ae24a50	486	mmc = init_mmc_device(dev, false);
1fd93c6e PA	487	if (!mmc)
	488	return CMD_RET_FAILURE;
	489
	490	if (IS_SD(mmc)) {
	491	printf("It is not a EMMC device\n");
	492	return CMD_RET_FAILURE;
ab71188c MN	493	}
ab71188c MN	494
1fd93c6e PA	495	if (mmc_boot_partition_size_change(mmc, bootsize, rpmbsize)) {
	496	printf("EMMC boot partition Size change Failed.\n");
	497	return CMD_RET_FAILURE;
	498	}
e85649c7	499
1fd93c6e PA	500	printf("EMMC boot partition Size %d MB\n", bootsize);
	501	printf("EMMC RPMB partition Size %d MB\n", rpmbsize);
	502	return CMD_RET_SUCCESS;
	503	}
	504	static int do_mmc_partconf(cmd_tbl_t *cmdtp, int flag,
	505	int argc, char * const argv[])
	506	{
	507	int dev;
	508	struct mmc *mmc;
	509	u8 ack, part_num, access;
272cc70b	510
1fd93c6e PA	511	if (argc != 5)
1fd93c6e PA	512	return CMD_RET_USAGE;
272cc70b	513
1fd93c6e PA	514	dev = simple_strtoul(argv[1], NULL, 10);
	515	ack = simple_strtoul(argv[2], NULL, 10);
	516	part_num = simple_strtoul(argv[3], NULL, 10);
	517	access = simple_strtoul(argv[4], NULL, 10);
d23d8d7e	518
1ae24a50	519	mmc = init_mmc_device(dev, false);
1fd93c6e PA	520	if (!mmc)
	521	return CMD_RET_FAILURE;
	522
	523	if (IS_SD(mmc)) {
	524	puts("PARTITION_CONFIG only exists on eMMC\n");
	525	return CMD_RET_FAILURE;
	526	}
	527
	528	/* acknowledge to be sent during boot operation */
	529	return mmc_set_part_conf(mmc, ack, part_num, access);
	530	}
	531	static int do_mmc_rst_func(cmd_tbl_t *cmdtp, int flag,
	532	int argc, char * const argv[])
	533	{
	534	int dev;
	535	struct mmc *mmc;
	536	u8 enable;
	537
	538	/*
	539	* Set the RST_n_ENABLE bit of RST_n_FUNCTION
	540	* The only valid values are 0x0, 0x1 and 0x2 and writing
	541	* a value of 0x1 or 0x2 sets the value permanently.
	542	*/
	543	if (argc != 3)
	544	return CMD_RET_USAGE;
	545
	546	dev = simple_strtoul(argv[1], NULL, 10);
	547	enable = simple_strtoul(argv[2], NULL, 10);
	548
	549	if (enable > 2 \|\| enable < 0) {
	550	puts("Invalid RST_n_ENABLE value\n");
	551	return CMD_RET_USAGE;
	552	}
	553
1ae24a50	554	mmc = init_mmc_device(dev, false);
1fd93c6e PA	555	if (!mmc)
	556	return CMD_RET_FAILURE;
	557
	558	if (IS_SD(mmc)) {
	559	puts("RST_n_FUNCTION only exists on eMMC\n");
	560	return CMD_RET_FAILURE;
	561	}
272cc70b	562
1fd93c6e PA	563	return mmc_set_rst_n_function(mmc, enable);
	564	}
	565	#endif
	566	static int do_mmc_setdsr(cmd_tbl_t *cmdtp, int flag,
	567	int argc, char * const argv[])
	568	{
	569	struct mmc *mmc;
	570	u32 val;
	571	int ret;
	572
	573	if (argc != 2)
	574	return CMD_RET_USAGE;
	575	val = simple_strtoul(argv[2], NULL, 16);
	576
	577	mmc = find_mmc_device(curr_device);
	578	if (!mmc) {
	579	printf("no mmc device at slot %x\n", curr_device);
	580	return CMD_RET_FAILURE;
	581	}
	582	ret = mmc_set_dsr(mmc, val);
	583	printf("set dsr %s\n", (!ret) ? "OK, force rescan" : "ERROR");
	584	if (!ret) {
	585	mmc->has_init = 0;
	586	if (mmc_init(mmc))
	587	return CMD_RET_FAILURE;
	588	else
	589	return CMD_RET_SUCCESS;
272cc70b	590	}
1fd93c6e PA	591	return ret;
	592	}
	593
	594	static cmd_tbl_t cmd_mmc[] = {
	595	U_BOOT_CMD_MKENT(info, 1, 0, do_mmcinfo, "", ""),
	596	U_BOOT_CMD_MKENT(read, 4, 1, do_mmc_read, "", ""),
	597	U_BOOT_CMD_MKENT(write, 4, 0, do_mmc_write, "", ""),
	598	U_BOOT_CMD_MKENT(erase, 3, 0, do_mmc_erase, "", ""),
	599	U_BOOT_CMD_MKENT(rescan, 1, 1, do_mmc_rescan, "", ""),
	600	U_BOOT_CMD_MKENT(part, 1, 1, do_mmc_part, "", ""),
	601	U_BOOT_CMD_MKENT(dev, 3, 0, do_mmc_dev, "", ""),
	602	U_BOOT_CMD_MKENT(list, 1, 1, do_mmc_list, "", ""),
	603	#ifdef CONFIG_SUPPORT_EMMC_BOOT
	604	U_BOOT_CMD_MKENT(bootbus, 5, 0, do_mmc_bootbus, "", ""),
fa7b8851	605	U_BOOT_CMD_MKENT(bootpart-resize, 4, 0, do_mmc_boot_resize, "", ""),
1fd93c6e PA	606	U_BOOT_CMD_MKENT(partconf, 5, 0, do_mmc_partconf, "", ""),
	607	U_BOOT_CMD_MKENT(rst-function, 3, 0, do_mmc_rst_func, "", ""),
	608	#endif
	609	#ifdef CONFIG_SUPPORT_EMMC_RPMB
	610	U_BOOT_CMD_MKENT(rpmb, CONFIG_SYS_MAXARGS, 1, do_mmcrpmb, "", ""),
	611	#endif
	612	U_BOOT_CMD_MKENT(setdsr, 2, 0, do_mmc_setdsr, "", ""),
	613	};
	614
	615	static int do_mmcops(cmd_tbl_t cmdtp, int flag, int argc, char const argv[])
	616	{
	617	cmd_tbl_t *cp;
	618
	619	cp = find_cmd_tbl(argv[1], cmd_mmc, ARRAY_SIZE(cmd_mmc));
	620
	621	/* Drop the mmc command */
	622	argc--;
	623	argv++;
	624
	625	if (cp == NULL \|\| argc > cp->maxargs)
	626	return CMD_RET_USAGE;
	627	if (flag == CMD_FLAG_REPEAT && !cp->repeatable)
	628	return CMD_RET_SUCCESS;
ea6ebe21	629
1fd93c6e PA	630	if (curr_device < 0) {
	631	if (get_mmc_num() > 0) {
	632	curr_device = 0;
	633	} else {
	634	puts("No MMC device available\n");
	635	return CMD_RET_FAILURE;
	636	}
	637	}
	638	return cp->cmd(cmdtp, flag, argc, argv);
272cc70b AF	639	}
	640
	641	U_BOOT_CMD(
1fd93c6e	642	mmc, 7, 1, do_mmcops,
852dbfdd	643	"MMC sub system",
1fd93c6e PA	644	"info - display info of the current MMC device\n"
1fd93c6e PA	645	"mmc read addr blk# cnt\n"
ea6ebe21	646	"mmc write addr blk# cnt\n"
e6f99a56	647	"mmc erase blk# cnt\n"
ea6ebe21 LW	648	"mmc rescan\n"
ea6ebe21 LW	649	"mmc part - lists available partition on current mmc device\n"
bc897b1d	650	"mmc dev [dev] [part] - show or set current mmc device [partition]\n"
2a91c913 A	651	"mmc list - lists available devices\n"
2a91c913 A	652	#ifdef CONFIG_SUPPORT_EMMC_BOOT
5a99b9de TR	653	"mmc bootbus dev boot_bus_width reset_boot_bus_width boot_mode\n"
5a99b9de TR	654	" - Set the BOOT_BUS_WIDTH field of the specified device\n"
f1fd957e TR	655	"mmc bootpart-resize <dev> <boot part size MB> <RPMB part size MB>\n"
f1fd957e TR	656	" - Change sizes of boot and RPMB partitions of specified device\n"
792970b0 TR	657	"mmc partconf dev boot_ack boot_partition partition_access\n"
792970b0 TR	658	" - Change the bits of the PARTITION_CONFIG field of the specified device\n"
33ace362 TR	659	"mmc rst-function dev value\n"
	660	" - Change the RST_n_FUNCTION field of the specified device\n"
	661	" WARNING: This is a write-once field and 0 / 1 / 2 are the only valid values.\n"
3511b4e2	662	#endif
1fd93c6e PA	663	#ifdef CONFIG_SUPPORT_EMMC_RPMB
	664	"mmc rpmb read addr blk# cnt [address of auth-key] - block size is 256 bytes\n"
	665	"mmc rpmb write addr blk# cnt <address of auth-key> - block size is 256 bytes\n"
	666	"mmc rpmb key <address of auth-key> - program the RPMB authentication key.\n"
	667	"mmc rpmb counter - read the value of the write counter\n"
	668	#endif
	669	"mmc setdsr <value> - set DSR register value\n"
2a91c913	670	);
1fd93c6e PA	671
	672	/* Old command kept for compatibility. Same as 'mmc info' */
	673	U_BOOT_CMD(
	674	mmcinfo, 1, 0, do_mmcinfo,
	675	"display MMC info",
	676	"- display info of the current MMC device"
	677	);
	678
2a91c913	679	#endif /* !CONFIG_GENERIC_MMC */