/*
   * (C) Copyright 2003
   * Kyle Harris, kharris@nexus-tech.net
   *
   * 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
  ");
  			return 1;
  		}
  
  		curr_device = dev;
  		printf("mmc%d is available
  ", curr_device);
  	} else if (strcmp(argv[1], "device") == 0) {
  		if (argc == 2) {
  			if (curr_device < 0) {
  				puts("No MMC device available
  ");
  				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
  ", 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
  "
  	"mmc device [dev] - show or set current device"
  );
  #else /* !CONFIG_GENERIC_MMC */
  
  static void print_mmcinfo(struct mmc *mmc)
  {
  	int i;
  
  	printf("Device: %s
  ", mmc->cfg->name);
  	printf("Manufacturer ID: %x
  ", mmc->cid[0] >> 24);
  	printf("OEM: %x
  ", (mmc->cid[0] >> 8) & 0xffff);
  	printf("Name: %c%c%c%c%c 
  ", 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
  ", mmc->tran_speed);
  	printf("Rd Block Len: %d
  ", mmc->read_bl_len);
  
  	printf("%s version %d.%d", IS_SD(mmc) ? "SD" : "MMC",
  			EXTRACT_SDMMC_MAJOR_VERSION(mmc->version),
  			EXTRACT_SDMMC_MINOR_VERSION(mmc->version));
  	if (EXTRACT_SDMMC_CHANGE_VERSION(mmc->version) != 0)
  		printf(".%d", EXTRACT_SDMMC_CHANGE_VERSION(mmc->version));
  	printf("
  ");
  
  	printf("High Capacity: %s
  ", mmc->high_capacity ? "Yes" : "No");
  	puts("Capacity: ");
  	print_size(mmc->capacity, "
  ");
  
  	printf("Bus Width: %d-bit%s
  ", mmc->bus_width,
  			mmc->ddr_mode ? " DDR" : "");
  
  	puts("Erase Group Size: ");
  	print_size(((u64)mmc->erase_grp_size) << 9, "
  ");
  
  	if (!IS_SD(mmc) && mmc->version >= MMC_VERSION_4_41) {
  		bool has_enh = (mmc->part_support & ENHNCD_SUPPORT) != 0;
  		bool usr_enh = has_enh && (mmc->part_attr & EXT_CSD_ENH_USR);
  
  		puts("HC WP Group Size: ");
  		print_size(((u64)mmc->hc_wp_grp_size) << 9, "
  ");
  
  		puts("User Capacity: ");
  		print_size(mmc->capacity_user, usr_enh ? " ENH" : "");
  		if (mmc->wr_rel_set & EXT_CSD_WR_DATA_REL_USR)
  			puts(" WRREL
  ");
  		else
  			putc('
  ');
  		if (usr_enh) {
  			puts("User Enhanced Start: ");
  			print_size(mmc->enh_user_start, "
  ");
  			puts("User Enhanced Size: ");
  			print_size(mmc->enh_user_size, "
  ");
  		}
  		puts("Boot Capacity: ");
  		print_size(mmc->capacity_boot, has_enh ? " ENH
  " : "
  ");
  		puts("RPMB Capacity: ");
  		print_size(mmc->capacity_rpmb, has_enh ? " ENH
  " : "
  ");
  
  		for (i = 0; i < ARRAY_SIZE(mmc->capacity_gp); i++) {
  			bool is_enh = has_enh &&
  				(mmc->part_attr & EXT_CSD_ENH_GP(i));
  			if (mmc->capacity_gp[i]) {
  				printf("GP%i Capacity: ", i+1);
  				print_size(mmc->capacity_gp[i],
  					   is_enh ? " ENH" : "");
  				if (mmc->wr_rel_set & EXT_CSD_WR_DATA_REL_GP(i))
  					puts(" WRREL
  ");
  				else
  					putc('
  ');
  			}
  		}
  	}
  }
  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
  ", 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
  ");
  			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 !
  "
  	     "         Use this command only if you are sure of what you are doing,
  "
  	     "Really perform the key programming? <y/N> ");
  	if (confirm_yesno())
  		return 1;
  
  	puts("Authentication key programming aborted
  ");
  	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 ?
  ");
  		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("
  MMC 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 == 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("
  MMC 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 == 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
  ", 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
  ");
  		return CMD_RET_FAILURE;
  	}
  	if (mmc->version < MMC_VERSION_4_41) {
  		printf("RPMB not supported before version 4.41
  ");
  		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("
  MMC 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 == 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("
  MMC write: dev # %d, block # %d, count %d ... ",
  	       curr_device, blk, cnt);
  
  	if (mmc_getwp(mmc) == 1) {
  		printf("Error: card is write protected!
  ");
  		return CMD_RET_FAILURE;
  	}
  	n = mmc->block_dev.block_write(curr_device, blk, cnt, addr);
  	printf("%d blocks written: %s
  ", 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("
  MMC erase: dev # %d, block # %d, count %d ... ",
  	       curr_device, blk, cnt);
  
  	if (mmc_getwp(mmc) == 1) {
  		printf("Error: card is write protected!
  ");
  		return CMD_RET_FAILURE;
  	}
  	n = mmc->block_dev.block_erase(curr_device, blk, cnt);
  	printf("%d blocks erased: %s
  ", 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!
  ");
  	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
  ",
  			       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
  ",
  	       part, (!ret) ? "OK" : "ERROR");
  	if (ret)
  		return 1;
  
  	curr_device = dev;
  	if (mmc->part_config == MMCPART_NOAVAILABLE)
  		printf("mmc%d is current device
  ", curr_device);
  	else
  		printf("mmc%d(part %d) is current device
  ",
  		       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('
  ');
  	return CMD_RET_SUCCESS;
  }
  
  static int parse_hwpart_user(struct mmc_hwpart_conf *pconf,
  			     int argc, char * const argv[])
  {
  	int i = 0;
  
  	memset(&pconf->user, 0, sizeof(pconf->user));
  
  	while (i < argc) {
  		if (!strcmp(argv[i], "enh")) {
  			if (i + 2 >= argc)
  				return -1;
  			pconf->user.enh_start =
  				simple_strtoul(argv[i+1], NULL, 10);
  			pconf->user.enh_size =
  				simple_strtoul(argv[i+2], NULL, 10);
  			i += 3;
  		} else if (!strcmp(argv[i], "wrrel")) {
  			if (i + 1 >= argc)
  				return -1;
  			pconf->user.wr_rel_change = 1;
  			if (!strcmp(argv[i+1], "on"))
  				pconf->user.wr_rel_set = 1;
  			else if (!strcmp(argv[i+1], "off"))
  				pconf->user.wr_rel_set = 0;
  			else
  				return -1;
  			i += 2;
  		} else {
  			break;
  		}
  	}
  	return i;
  }
  
  static int parse_hwpart_gp(struct mmc_hwpart_conf *pconf, int pidx,
  			   int argc, char * const argv[])
  {
  	int i;
  
  	memset(&pconf->gp_part[pidx], 0, sizeof(pconf->gp_part[pidx]));
  
  	if (1 >= argc)
  		return -1;
  	pconf->gp_part[pidx].size = simple_strtoul(argv[0], NULL, 10);
  
  	i = 1;
  	while (i < argc) {
  		if (!strcmp(argv[i], "enh")) {
  			pconf->gp_part[pidx].enhanced = 1;
  			i += 1;
  		} else if (!strcmp(argv[i], "wrrel")) {
  			if (i + 1 >= argc)
  				return -1;
  			pconf->gp_part[pidx].wr_rel_change = 1;
  			if (!strcmp(argv[i+1], "on"))
  				pconf->gp_part[pidx].wr_rel_set = 1;
  			else if (!strcmp(argv[i+1], "off"))
  				pconf->gp_part[pidx].wr_rel_set = 0;
  			else
  				return -1;
  			i += 2;
  		} else {
  			break;
  		}
  	}
  	return i;
  }
  
  static int do_mmc_hwpartition(cmd_tbl_t *cmdtp, int flag,
  			      int argc, char * const argv[])
  {
  	struct mmc *mmc;
  	struct mmc_hwpart_conf pconf = { };
  	enum mmc_hwpart_conf_mode mode = MMC_HWPART_CONF_CHECK;
  	int i, r, pidx;
  
  	mmc = init_mmc_device(curr_device, false);
  	if (!mmc)
  		return CMD_RET_FAILURE;
  
  	if (argc < 1)
  		return CMD_RET_USAGE;
  	i = 1;
  	while (i < argc) {
  		if (!strcmp(argv[i], "user")) {
  			i++;
  			r = parse_hwpart_user(&pconf, argc-i, &argv[i]);
  			if (r < 0)
  				return CMD_RET_USAGE;
  			i += r;
  		} else if (!strncmp(argv[i], "gp", 2) &&
  			   strlen(argv[i]) == 3 &&
  			   argv[i][2] >= '1' && argv[i][2] <= '4') {
  			pidx = argv[i][2] - '1';
  			i++;
  			r = parse_hwpart_gp(&pconf, pidx, argc-i, &argv[i]);
  			if (r < 0)
  				return CMD_RET_USAGE;
  			i += r;
  		} else if (!strcmp(argv[i], "check")) {
  			mode = MMC_HWPART_CONF_CHECK;
  			i++;
  		} else if (!strcmp(argv[i], "set")) {
  			mode = MMC_HWPART_CONF_SET;
  			i++;
  		} else if (!strcmp(argv[i], "complete")) {
  			mode = MMC_HWPART_CONF_COMPLETE;
  			i++;
  		} else {
  			return CMD_RET_USAGE;
  		}
  	}
  
  	puts("Partition configuration:
  ");
  	if (pconf.user.enh_size) {
  		puts("\tUser Enhanced Start: ");
  		print_size(((u64)pconf.user.enh_start) << 9, "
  ");
  		puts("\tUser Enhanced Size: ");
  		print_size(((u64)pconf.user.enh_size) << 9, "
  ");
  	} else {
  		puts("\tNo enhanced user data area
  ");
  	}
  	if (pconf.user.wr_rel_change)
  		printf("\tUser partition write reliability: %s
  ",
  		       pconf.user.wr_rel_set ? "on" : "off");
  	for (pidx = 0; pidx < 4; pidx++) {
  		if (pconf.gp_part[pidx].size) {
  			printf("\tGP%i Capacity: ", pidx+1);
  			print_size(((u64)pconf.gp_part[pidx].size) << 9,
  				   pconf.gp_part[pidx].enhanced ?
  				   " ENH
  " : "
  ");
  		} else {
  			printf("\tNo GP%i partition
  ", pidx+1);
  		}
  		if (pconf.gp_part[pidx].wr_rel_change)
  			printf("\tGP%i write reliability: %s
  ", pidx+1,
  			       pconf.gp_part[pidx].wr_rel_set ? "on" : "off");
  	}
  
  	if (!mmc_hwpart_config(mmc, &pconf, mode)) {
  		if (mode == MMC_HWPART_CONF_COMPLETE)
  			puts("Partitioning successful, "
  			     "power-cycle to make effective
  ");
  		return CMD_RET_SUCCESS;
  	} else {
  		puts("Failed!
  ");
  		return CMD_RET_FAILURE;
  	}
  }
  
  #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
  ");
  		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
  ");
  		return CMD_RET_FAILURE;
  	}
  
  	if (mmc_boot_partition_size_change(mmc, bootsize, rpmbsize)) {
  		printf("EMMC boot partition Size change Failed.
  ");
  		return CMD_RET_FAILURE;
  	}
  
  	printf("EMMC boot partition Size %d MB
  ", bootsize);
  	printf("EMMC RPMB partition Size %d MB
  ", 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
  ");
  		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
  ");
  		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
  ");
  		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
  ", curr_device);
  		return CMD_RET_FAILURE;
  	}
  	ret = mmc_set_dsr(mmc, val);
  	printf("set dsr %s
  ", (!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, "", ""),
  	U_BOOT_CMD_MKENT(hwpartition, 28, 0, do_mmc_hwpartition, "", ""),
  #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
  ");
  			return CMD_RET_FAILURE;
  		}
  	}
  	return cp->cmd(cmdtp, flag, argc, argv);
  }
  
  U_BOOT_CMD(
  	mmc, 29, 1, do_mmcops,
  	"MMC sub system",
  	"info - display info of the current MMC device
  "
  	"mmc read addr blk# cnt
  "
  	"mmc write addr blk# cnt
  "
  	"mmc erase blk# cnt
  "
  	"mmc rescan
  "
  	"mmc part - lists available partition on current mmc device
  "
  	"mmc dev [dev] [part] - show or set current mmc device [partition]
  "
  	"mmc list - lists available devices
  "
  	"mmc hwpartition [args...] - does hardware partitioning
  "
  	"  arguments (sizes in 512-byte blocks):
  "
  	"    [user [enh start cnt] [wrrel {on|off}]] - sets user data area attributes
  "
  	"    [gp1|gp2|gp3|gp4 cnt [enh] [wrrel {on|off}]] - general purpose partition
  "
  	"    [check|set|complete] - mode, complete set partitioning completed
  "
  	"  WARNING: Partitioning is a write-once setting once it is set to complete.
  "
  	"  Power cycling is required to initialize partitions after set to complete.
  "
  #ifdef CONFIG_SUPPORT_EMMC_BOOT
  	"mmc bootbus dev boot_bus_width reset_boot_bus_width boot_mode
  "
  	" - Set the BOOT_BUS_WIDTH field of the specified device
  "
  	"mmc bootpart-resize <dev> <boot part size MB> <RPMB part size MB>
  "
  	" - Change sizes of boot and RPMB partitions of specified device
  "
  	"mmc partconf dev boot_ack boot_partition partition_access
  "
  	" - Change the bits of the PARTITION_CONFIG field of the specified device
  "
  	"mmc rst-function dev value
  "
  	" - Change the RST_n_FUNCTION field of the specified device
  "
  	"   WARNING: This is a write-once field and 0 / 1 / 2 are the only valid values.
  "
  #endif
  #ifdef CONFIG_SUPPORT_EMMC_RPMB
  	"mmc rpmb read addr blk# cnt [address of auth-key] - block size is 256 bytes
  "
  	"mmc rpmb write addr blk# cnt <address of auth-key> - block size is 256 bytes
  "
  	"mmc rpmb key <address of auth-key> - program the RPMB authentication key.
  "
  	"mmc rpmb counter - read the value of the write counter
  "
  #endif
  	"mmc setdsr <value> - set DSR register value
  "
  	);
  
  /* 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 */