/*
   * (C) Copyright 2003
   * Kyle Harris, kharris@nexus-tech.net
   *

   * SPDX-License-Identifier:	GPL-2.0+

   */
  
  #include <common.h>
  #include <command.h>

  #include <console.h>

  #include <mmc.h>

  static int curr_device = -1;

  
  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("Bus Speed: %d
  ", mmc->clock);

  #if CONFIG_IS_ENABLED(MMC_VERBOSE)

  	printf("Mode : %s
  ", mmc_mode_name(mmc->selected_mode));

  	mmc_dump_capabilities("card capabilities", mmc->card_caps);
  	mmc_dump_capabilities("host capabilities", mmc->host_caps);
  #endif

  	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" : "");

  #if CONFIG_IS_ENABLED(MMC_WRITE)

  	puts("Erase Group Size: ");
  	print_size(((u64)mmc->erase_grp_size) << 9, "
  ");

  #endif

  	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);

  #if CONFIG_IS_ENABLED(MMC_HW_PARTITIONING)

  		puts("HC WP Group Size: ");
  		print_size(((u64)mmc->hc_wp_grp_size) << 9, "
  ");

  #endif

  		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 */

  #ifndef CONFIG_BLK

  	original_part = mmc->block_dev.hwpart;

  #else
  	original_part = mmc_get_blk_desc(mmc)->hwpart;
  #endif

  	if (blk_select_hwpart_devnum(IF_TYPE_MMC, curr_device, MMC_PART_RPMB) !=
  	    0)

  		return CMD_RET_FAILURE;

  	ret = cp->cmd(cmdtp, flag, argc, argv);

  	/* Return to original partition */

  	if (blk_select_hwpart_devnum(IF_TYPE_MMC, curr_device, original_part) !=
  	    0)

  		return CMD_RET_FAILURE;

  	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 = blk_dread(mmc_get_blk_desc(mmc), blk, cnt, addr);

  	printf("%d blocks read: %s
  ", n, (n == cnt) ? "OK" : "ERROR");

  	return (n == cnt) ? CMD_RET_SUCCESS : CMD_RET_FAILURE;
  }

  
  #if CONFIG_IS_ENABLED(MMC_WRITE)

  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 = blk_dwrite(mmc_get_blk_desc(mmc), 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 = blk_derase(mmc_get_blk_desc(mmc), blk, cnt);

  	printf("%d blocks erased: %s
  ", n, (n == cnt) ? "OK" : "ERROR");

  	return (n == cnt) ? CMD_RET_SUCCESS : CMD_RET_FAILURE;
  }

  #endif

  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[])
  {

  	struct blk_desc *mmc_dev;

  	struct mmc *mmc;

  	mmc = init_mmc_device(curr_device, false);

  	if (!mmc)
  		return CMD_RET_FAILURE;

  	mmc_dev = blk_get_devnum_by_type(IF_TYPE_MMC, curr_device);

  	if (mmc_dev != NULL && mmc_dev->type != DEV_TYPE_UNKNOWN) {

  		part_print(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 = blk_select_hwpart_devnum(IF_TYPE_MMC, 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_get_blk_desc(mmc)->hwpart);

  	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;
  }

  #if CONFIG_IS_ENABLED(MMC_HW_PARTITIONING)

  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;
  	}
  }

  #endif

  #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 mmc_partconf_print(struct mmc *mmc)
  {
  	u8 ack, access, part;
  
  	if (mmc->part_config == MMCPART_NOAVAILABLE) {
  		printf("No part_config info for ver. 0x%x
  ", mmc->version);
  		return CMD_RET_FAILURE;
  	}
  
  	access = EXT_CSD_EXTRACT_PARTITION_ACCESS(mmc->part_config);
  	ack = EXT_CSD_EXTRACT_BOOT_ACK(mmc->part_config);
  	part = EXT_CSD_EXTRACT_BOOT_PART(mmc->part_config);
  
  	printf("EXT_CSD[179], PARTITION_CONFIG:
  "
  		"BOOT_ACK: 0x%x
  "
  		"BOOT_PARTITION_ENABLE: 0x%x
  "
  		"PARTITION_ACCESS: 0x%x
  ", ack, part, access);
  
  	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 != 2 && argc != 5)

  		return CMD_RET_USAGE;

  	dev = simple_strtoul(argv[1], 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;
  	}

  	if (argc == 2)
  		return mmc_partconf_print(mmc);
  
  	ack = simple_strtoul(argv[2], NULL, 10);
  	part_num = simple_strtoul(argv[3], NULL, 10);
  	access = simple_strtoul(argv[4], NULL, 10);

  	/* 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) {

  		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[1], 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;
  }

  #ifdef CONFIG_CMD_BKOPS_ENABLE
  static int do_mmc_bkops_enable(cmd_tbl_t *cmdtp, int flag,
  				   int argc, char * const argv[])
  {
  	int dev;
  	struct mmc *mmc;
  
  	if (argc != 2)
  		return CMD_RET_USAGE;
  
  	dev = simple_strtoul(argv[1], NULL, 10);
  
  	mmc = init_mmc_device(dev, false);
  	if (!mmc)
  		return CMD_RET_FAILURE;
  
  	if (IS_SD(mmc)) {
  		puts("BKOPS_EN only exists on eMMC
  ");
  		return CMD_RET_FAILURE;
  	}
  
  	return mmc_set_bkops_enable(mmc);
  }
  #endif

  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, "", ""),

  #if CONFIG_IS_ENABLED(MMC_WRITE)

  	U_BOOT_CMD_MKENT(write, 4, 0, do_mmc_write, "", ""),
  	U_BOOT_CMD_MKENT(erase, 3, 0, do_mmc_erase, "", ""),

  #endif

  	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, "", ""),

  #if CONFIG_IS_ENABLED(MMC_HW_PARTITIONING)

  	U_BOOT_CMD_MKENT(hwpartition, 28, 0, do_mmc_hwpartition, "", ""),

  #endif

  #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, "", ""),

  #ifdef CONFIG_CMD_BKOPS_ENABLE
  	U_BOOT_CMD_MKENT(bkops-enable, 2, 0, do_mmc_bkops_enable, "", ""),
  #endif

  };
  
  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]
  "
  	" - Show or 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
  "

  #ifdef CONFIG_CMD_BKOPS_ENABLE
  	"mmc bkops-enable <dev> - enable background operations handshake on device
  "
  	"   WARNING: This is a write-once setting.
  "
  #endif

  	);

  
  /* 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"
  );