cmd/mmc.c

   1 /*
   2  * (C) Copyright 2003
   3  * Kyle Harris, kharris@nexus-tech.net
   4  *
   5  * SPDX-License-Identifier:     GPL-2.0+
   6  */
   7
   8 #include <common.h>
   9 #include <command.h>
  10 #include <console.h>
  11 #include <mmc.h>
  12
  13 static int curr_device = -1;
  14
  15 static void print_mmcinfo(struct mmc *mmc)
  16 {
  17         int i;
  18
  19         printf("Device: %s\n", mmc->cfg->name);
  20         printf("Manufacturer ID: %x\n", mmc->cid[0] >> 24);
  21         printf("OEM: %x\n", (mmc->cid[0] >> 8) & 0xffff);
  22         printf("Name: %c%c%c%c%c \n", mmc->cid[0] & 0xff,
  23                         (mmc->cid[1] >> 24), (mmc->cid[1] >> 16) & 0xff,
  24                         (mmc->cid[1] >> 8) & 0xff, mmc->cid[1] & 0xff);
  25
  26         printf("Tran Speed: %d\n", mmc->tran_speed);
  27         printf("Rd Block Len: %d\n", mmc->read_bl_len);
  28
  29         printf("%s version %d.%d", IS_SD(mmc) ? "SD" : "MMC",
  30                         EXTRACT_SDMMC_MAJOR_VERSION(mmc->version),
  31                         EXTRACT_SDMMC_MINOR_VERSION(mmc->version));
  32         if (EXTRACT_SDMMC_CHANGE_VERSION(mmc->version) != 0)
  33                 printf(".%d", EXTRACT_SDMMC_CHANGE_VERSION(mmc->version));
  34         printf("\n");
  35
  36         printf("High Capacity: %s\n", mmc->high_capacity ? "Yes" : "No");
  37         puts("Capacity: ");
  38         print_size(mmc->capacity, "\n");
  39
  40         printf("Bus Width: %d-bit%s\n", mmc->bus_width,
  41                         mmc->ddr_mode ? " DDR" : "");
  42
  43         puts("Erase Group Size: ");
  44         print_size(((u64)mmc->erase_grp_size) << 9, "\n");
  45
  46         if (!IS_SD(mmc) && mmc->version >= MMC_VERSION_4_41) {
  47                 bool has_enh = (mmc->part_support & ENHNCD_SUPPORT) != 0;
  48                 bool usr_enh = has_enh && (mmc->part_attr & EXT_CSD_ENH_USR);
  49
  50                 puts("HC WP Group Size: ");
  51                 print_size(((u64)mmc->hc_wp_grp_size) << 9, "\n");
  52
  53                 puts("User Capacity: ");
  54                 print_size(mmc->capacity_user, usr_enh ? " ENH" : "");
  55                 if (mmc->wr_rel_set & EXT_CSD_WR_DATA_REL_USR)
  56                         puts(" WRREL\n");
  57                 else
  58                         putc('\n');
  59                 if (usr_enh) {
  60                         puts("User Enhanced Start: ");
  61                         print_size(mmc->enh_user_start, "\n");
  62                         puts("User Enhanced Size: ");
  63                         print_size(mmc->enh_user_size, "\n");
  64                 }
  65                 puts("Boot Capacity: ");
  66                 print_size(mmc->capacity_boot, has_enh ? " ENH\n" : "\n");
  67                 puts("RPMB Capacity: ");
  68                 print_size(mmc->capacity_rpmb, has_enh ? " ENH\n" : "\n");
  69
  70                 for (i = 0; i < ARRAY_SIZE(mmc->capacity_gp); i++) {
  71                         bool is_enh = has_enh &&
  72                                 (mmc->part_attr & EXT_CSD_ENH_GP(i));
  73                         if (mmc->capacity_gp[i]) {
  74                                 printf("GP%i Capacity: ", i+1);
  75                                 print_size(mmc->capacity_gp[i],
  76                                            is_enh ? " ENH" : "");
  77                                 if (mmc->wr_rel_set & EXT_CSD_WR_DATA_REL_GP(i))
  78                                         puts(" WRREL\n");
  79                                 else
  80                                         putc('\n');
  81                         }
  82                 }
  83         }
  84 }
  85 static struct mmc *init_mmc_device(int dev, bool force_init)
  86 {
  87         struct mmc *mmc;
  88         mmc = find_mmc_device(dev);
  89         if (!mmc) {
  90                 printf("no mmc device at slot %x\n", dev);
  91                 return NULL;
  92         }
  93
  94         if (force_init)
  95                 mmc->has_init = 0;
  96         if (mmc_init(mmc))
  97                 return NULL;
  98         return mmc;
  99 }
 100 static int do_mmcinfo(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
 101 {
 102         struct mmc *mmc;
 103
 104         if (curr_device < 0) {
 105                 if (get_mmc_num() > 0)
 106                         curr_device = 0;
 107                 else {
 108                         puts("No MMC device available\n");
 109                         return 1;
 110                 }
 111         }
 112
 113         mmc = init_mmc_device(curr_device, false);
 114         if (!mmc)
 115                 return CMD_RET_FAILURE;
 116
 117         print_mmcinfo(mmc);
 118         return CMD_RET_SUCCESS;
 119 }
 120
 121 #ifdef CONFIG_SUPPORT_EMMC_RPMB
 122 static int confirm_key_prog(void)
 123 {
 124         puts("Warning: Programming authentication key can be done only once !\n"
 125              "         Use this command only if you are sure of what you are doing,\n"
 126              "Really perform the key programming? <y/N> ");
 127         if (confirm_yesno())
 128                 return 1;
 129
 130         puts("Authentication key programming aborted\n");
 131         return 0;
 132 }
 133 static int do_mmcrpmb_key(cmd_tbl_t *cmdtp, int flag,
 134                           int argc, char * const argv[])
 135 {
 136         void *key_addr;
 137         struct mmc *mmc = find_mmc_device(curr_device);
 138
 139         if (argc != 2)
 140                 return CMD_RET_USAGE;
 141
 142         key_addr = (void *)simple_strtoul(argv[1], NULL, 16);
 143         if (!confirm_key_prog())
 144                 return CMD_RET_FAILURE;
 145         if (mmc_rpmb_set_key(mmc, key_addr)) {
 146                 printf("ERROR - Key already programmed ?\n");
 147                 return CMD_RET_FAILURE;
 148         }
 149         return CMD_RET_SUCCESS;
 150 }
 151 static int do_mmcrpmb_read(cmd_tbl_t *cmdtp, int flag,
 152                            int argc, char * const argv[])
 153 {
 154         u16 blk, cnt;
 155         void *addr;
 156         int n;
 157         void *key_addr = NULL;
 158         struct mmc *mmc = find_mmc_device(curr_device);
 159
 160         if (argc < 4)
 161                 return CMD_RET_USAGE;
 162
 163         addr = (void *)simple_strtoul(argv[1], NULL, 16);
 164         blk = simple_strtoul(argv[2], NULL, 16);
 165         cnt = simple_strtoul(argv[3], NULL, 16);
 166
 167         if (argc == 5)
 168                 key_addr = (void *)simple_strtoul(argv[4], NULL, 16);
 169
 170         printf("\nMMC RPMB read: dev # %d, block # %d, count %d ... ",
 171                curr_device, blk, cnt);
 172         n =  mmc_rpmb_read(mmc, addr, blk, cnt, key_addr);
 173
 174         printf("%d RPMB blocks read: %s\n", n, (n == cnt) ? "OK" : "ERROR");
 175         if (n != cnt)
 176                 return CMD_RET_FAILURE;
 177         return CMD_RET_SUCCESS;
 178 }
 179 static int do_mmcrpmb_write(cmd_tbl_t *cmdtp, int flag,
 180                             int argc, char * const argv[])
 181 {
 182         u16 blk, cnt;
 183         void *addr;
 184         int n;
 185         void *key_addr;
 186         struct mmc *mmc = find_mmc_device(curr_device);
 187
 188         if (argc != 5)
 189                 return CMD_RET_USAGE;
 190
 191         addr = (void *)simple_strtoul(argv[1], NULL, 16);
 192         blk = simple_strtoul(argv[2], NULL, 16);
 193         cnt = simple_strtoul(argv[3], NULL, 16);
 194         key_addr = (void *)simple_strtoul(argv[4], NULL, 16);
 195
 196         printf("\nMMC RPMB write: dev # %d, block # %d, count %d ... ",
 197                curr_device, blk, cnt);
 198         n =  mmc_rpmb_write(mmc, addr, blk, cnt, key_addr);
 199
 200         printf("%d RPMB blocks written: %s\n", n, (n == cnt) ? "OK" : "ERROR");
 201         if (n != cnt)
 202                 return CMD_RET_FAILURE;
 203         return CMD_RET_SUCCESS;
 204 }
 205 static int do_mmcrpmb_counter(cmd_tbl_t *cmdtp, int flag,
 206                               int argc, char * const argv[])
 207 {
 208         unsigned long counter;
 209         struct mmc *mmc = find_mmc_device(curr_device);
 210
 211         if (mmc_rpmb_get_counter(mmc, &counter))
 212                 return CMD_RET_FAILURE;
 213         printf("RPMB Write counter= %lx\n", counter);
 214         return CMD_RET_SUCCESS;
 215 }
 216
 217 static cmd_tbl_t cmd_rpmb[] = {
 218         U_BOOT_CMD_MKENT(key, 2, 0, do_mmcrpmb_key, "", ""),
 219         U_BOOT_CMD_MKENT(read, 5, 1, do_mmcrpmb_read, "", ""),
 220         U_BOOT_CMD_MKENT(write, 5, 0, do_mmcrpmb_write, "", ""),
 221         U_BOOT_CMD_MKENT(counter, 1, 1, do_mmcrpmb_counter, "", ""),
 222 };
 223
 224 static int do_mmcrpmb(cmd_tbl_t *cmdtp, int flag,
 225                       int argc, char * const argv[])
 226 {
 227         cmd_tbl_t *cp;
 228         struct mmc *mmc;
 229         char original_part;
 230         int ret;
 231
 232         cp = find_cmd_tbl(argv[1], cmd_rpmb, ARRAY_SIZE(cmd_rpmb));
 233
 234         /* Drop the rpmb subcommand */
 235         argc--;
 236         argv++;
 237
 238         if (cp == NULL || argc > cp->maxargs)
 239                 return CMD_RET_USAGE;
 240         if (flag == CMD_FLAG_REPEAT && !cp->repeatable)
 241                 return CMD_RET_SUCCESS;
 242
 243         mmc = init_mmc_device(curr_device, false);
 244         if (!mmc)
 245                 return CMD_RET_FAILURE;
 246
 247         if (!(mmc->version & MMC_VERSION_MMC)) {
 248                 printf("It is not a EMMC device\n");
 249                 return CMD_RET_FAILURE;
 250         }
 251         if (mmc->version < MMC_VERSION_4_41) {
 252                 printf("RPMB not supported before version 4.41\n");
 253                 return CMD_RET_FAILURE;
 254         }
 255         /* Switch to the RPMB partition */
 256 #ifndef CONFIG_BLK
 257         original_part = mmc->block_dev.hwpart;
 258 #else
 259         original_part = mmc_get_blk_desc(mmc)->hwpart;
 260 #endif
 261         if (blk_select_hwpart_devnum(IF_TYPE_MMC, curr_device, MMC_PART_RPMB) !=
 262             0)
 263                 return CMD_RET_FAILURE;
 264         ret = cp->cmd(cmdtp, flag, argc, argv);
 265
 266         /* Return to original partition */
 267         if (blk_select_hwpart_devnum(IF_TYPE_MMC, curr_device, original_part) !=
 268             0)
 269                 return CMD_RET_FAILURE;
 270         return ret;
 271 }
 272 #endif
 273
 274 static int do_mmc_read(cmd_tbl_t *cmdtp, int flag,
 275                        int argc, char * const argv[])
 276 {
 277         struct mmc *mmc;
 278         u32 blk, cnt, n;
 279         void *addr;
 280
 281         if (argc != 4)
 282                 return CMD_RET_USAGE;
 283
 284         addr = (void *)simple_strtoul(argv[1], NULL, 16);
 285         blk = simple_strtoul(argv[2], NULL, 16);
 286         cnt = simple_strtoul(argv[3], NULL, 16);
 287
 288         mmc = init_mmc_device(curr_device, false);
 289         if (!mmc)
 290                 return CMD_RET_FAILURE;
 291
 292         printf("\nMMC read: dev # %d, block # %d, count %d ... ",
 293                curr_device, blk, cnt);
 294
 295         n = blk_dread(mmc_get_blk_desc(mmc), blk, cnt, addr);
 296         printf("%d blocks read: %s\n", n, (n == cnt) ? "OK" : "ERROR");
 297
 298         return (n == cnt) ? CMD_RET_SUCCESS : CMD_RET_FAILURE;
 299 }
 300 static int do_mmc_write(cmd_tbl_t *cmdtp, int flag,
 301                         int argc, char * const argv[])
 302 {
 303         struct mmc *mmc;
 304         u32 blk, cnt, n;
 305         void *addr;
 306
 307         if (argc != 4)
 308                 return CMD_RET_USAGE;
 309
 310         addr = (void *)simple_strtoul(argv[1], NULL, 16);
 311         blk = simple_strtoul(argv[2], NULL, 16);
 312         cnt = simple_strtoul(argv[3], NULL, 16);
 313
 314         mmc = init_mmc_device(curr_device, false);
 315         if (!mmc)
 316                 return CMD_RET_FAILURE;
 317
 318         printf("\nMMC write: dev # %d, block # %d, count %d ... ",
 319                curr_device, blk, cnt);
 320
 321         if (mmc_getwp(mmc) == 1) {
 322                 printf("Error: card is write protected!\n");
 323                 return CMD_RET_FAILURE;
 324         }
 325         n = blk_dwrite(mmc_get_blk_desc(mmc), blk, cnt, addr);
 326         printf("%d blocks written: %s\n", n, (n == cnt) ? "OK" : "ERROR");
 327
 328         return (n == cnt) ? CMD_RET_SUCCESS : CMD_RET_FAILURE;
 329 }
 330 static int do_mmc_erase(cmd_tbl_t *cmdtp, int flag,
 331                         int argc, char * const argv[])
 332 {
 333         struct mmc *mmc;
 334         u32 blk, cnt, n;
 335
 336         if (argc != 3)
 337                 return CMD_RET_USAGE;
 338
 339         blk = simple_strtoul(argv[1], NULL, 16);
 340         cnt = simple_strtoul(argv[2], NULL, 16);
 341
 342         mmc = init_mmc_device(curr_device, false);
 343         if (!mmc)
 344                 return CMD_RET_FAILURE;
 345
 346         printf("\nMMC erase: dev # %d, block # %d, count %d ... ",
 347                curr_device, blk, cnt);
 348
 349         if (mmc_getwp(mmc) == 1) {
 350                 printf("Error: card is write protected!\n");
 351                 return CMD_RET_FAILURE;
 352         }
 353         n = blk_derase(mmc_get_blk_desc(mmc), blk, cnt);
 354         printf("%d blocks erased: %s\n", n, (n == cnt) ? "OK" : "ERROR");
 355
 356         return (n == cnt) ? CMD_RET_SUCCESS : CMD_RET_FAILURE;
 357 }
 358 static int do_mmc_rescan(cmd_tbl_t *cmdtp, int flag,
 359                          int argc, char * const argv[])
 360 {
 361         struct mmc *mmc;
 362
 363         mmc = init_mmc_device(curr_device, true);
 364         if (!mmc)
 365                 return CMD_RET_FAILURE;
 366
 367         return CMD_RET_SUCCESS;
 368 }
 369 static int do_mmc_part(cmd_tbl_t *cmdtp, int flag,
 370                        int argc, char * const argv[])
 371 {
 372         struct blk_desc *mmc_dev;
 373         struct mmc *mmc;
 374
 375         mmc = init_mmc_device(curr_device, false);
 376         if (!mmc)
 377                 return CMD_RET_FAILURE;
 378
 379         mmc_dev = blk_get_devnum_by_type(IF_TYPE_MMC, curr_device);
 380         if (mmc_dev != NULL && mmc_dev->type != DEV_TYPE_UNKNOWN) {
 381                 part_print(mmc_dev);
 382                 return CMD_RET_SUCCESS;
 383         }
 384
 385         puts("get mmc type error!\n");
 386         return CMD_RET_FAILURE;
 387 }
 388 static int do_mmc_dev(cmd_tbl_t *cmdtp, int flag,
 389                       int argc, char * const argv[])
 390 {
 391         int dev, part = 0, ret;
 392         struct mmc *mmc;
 393
 394         if (argc == 1) {
 395                 dev = curr_device;
 396         } else if (argc == 2) {
 397                 dev = simple_strtoul(argv[1], NULL, 10);
 398         } else if (argc == 3) {
 399                 dev = (int)simple_strtoul(argv[1], NULL, 10);
 400                 part = (int)simple_strtoul(argv[2], NULL, 10);
 401                 if (part > PART_ACCESS_MASK) {
 402                         printf("#part_num shouldn't be larger than %d\n",
 403                                PART_ACCESS_MASK);
 404                         return CMD_RET_FAILURE;
 405                 }
 406         } else {
 407                 return CMD_RET_USAGE;
 408         }
 409
 410         mmc = init_mmc_device(dev, true);
 411         if (!mmc)
 412                 return CMD_RET_FAILURE;
 413
 414         ret = blk_select_hwpart_devnum(IF_TYPE_MMC, dev, part);
 415         printf("switch to partitions #%d, %s\n",
 416                part, (!ret) ? "OK" : "ERROR");
 417         if (ret)
 418                 return 1;
 419
 420         curr_device = dev;
 421         if (mmc->part_config == MMCPART_NOAVAILABLE)
 422                 printf("mmc%d is current device\n", curr_device);
 423         else
 424                 printf("mmc%d(part %d) is current device\n",
 425                        curr_device, mmc_get_blk_desc(mmc)->hwpart);
 426
 427         return CMD_RET_SUCCESS;
 428 }
 429 static int do_mmc_list(cmd_tbl_t *cmdtp, int flag,
 430                        int argc, char * const argv[])
 431 {
 432         print_mmc_devices('\n');
 433         return CMD_RET_SUCCESS;
 434 }
 435
 436 static int parse_hwpart_user(struct mmc_hwpart_conf *pconf,
 437                              int argc, char * const argv[])
 438 {
 439         int i = 0;
 440
 441         memset(&pconf->user, 0, sizeof(pconf->user));
 442
 443         while (i < argc) {
 444                 if (!strcmp(argv[i], "enh")) {
 445                         if (i + 2 >= argc)
 446                                 return -1;
 447                         pconf->user.enh_start =
 448                                 simple_strtoul(argv[i+1], NULL, 10);
 449                         pconf->user.enh_size =
 450                                 simple_strtoul(argv[i+2], NULL, 10);
 451                         i += 3;
 452                 } else if (!strcmp(argv[i], "wrrel")) {
 453                         if (i + 1 >= argc)
 454                                 return -1;
 455                         pconf->user.wr_rel_change = 1;
 456                         if (!strcmp(argv[i+1], "on"))
 457                                 pconf->user.wr_rel_set = 1;
 458                         else if (!strcmp(argv[i+1], "off"))
 459                                 pconf->user.wr_rel_set = 0;
 460                         else
 461                                 return -1;
 462                         i += 2;
 463                 } else {
 464                         break;
 465                 }
 466         }
 467         return i;
 468 }
 469
 470 static int parse_hwpart_gp(struct mmc_hwpart_conf *pconf, int pidx,
 471                            int argc, char * const argv[])
 472 {
 473         int i;
 474
 475         memset(&pconf->gp_part[pidx], 0, sizeof(pconf->gp_part[pidx]));
 476
 477         if (1 >= argc)
 478                 return -1;
 479         pconf->gp_part[pidx].size = simple_strtoul(argv[0], NULL, 10);
 480
 481         i = 1;
 482         while (i < argc) {
 483                 if (!strcmp(argv[i], "enh")) {
 484                         pconf->gp_part[pidx].enhanced = 1;
 485                         i += 1;
 486                 } else if (!strcmp(argv[i], "wrrel")) {
 487                         if (i + 1 >= argc)
 488                                 return -1;
 489                         pconf->gp_part[pidx].wr_rel_change = 1;
 490                         if (!strcmp(argv[i+1], "on"))
 491                                 pconf->gp_part[pidx].wr_rel_set = 1;
 492                         else if (!strcmp(argv[i+1], "off"))
 493                                 pconf->gp_part[pidx].wr_rel_set = 0;
 494                         else
 495                                 return -1;
 496                         i += 2;
 497                 } else {
 498                         break;
 499                 }
 500         }
 501         return i;
 502 }
 503
 504 static int do_mmc_hwpartition(cmd_tbl_t *cmdtp, int flag,
 505                               int argc, char * const argv[])
 506 {
 507         struct mmc *mmc;
 508         struct mmc_hwpart_conf pconf = { };
 509         enum mmc_hwpart_conf_mode mode = MMC_HWPART_CONF_CHECK;
 510         int i, r, pidx;
 511
 512         mmc = init_mmc_device(curr_device, false);
 513         if (!mmc)
 514                 return CMD_RET_FAILURE;
 515
 516         if (argc < 1)
 517                 return CMD_RET_USAGE;
 518         i = 1;
 519         while (i < argc) {
 520                 if (!strcmp(argv[i], "user")) {
 521                         i++;
 522                         r = parse_hwpart_user(&pconf, argc-i, &argv[i]);
 523                         if (r < 0)
 524                                 return CMD_RET_USAGE;
 525                         i += r;
 526                 } else if (!strncmp(argv[i], "gp", 2) &&
 527                            strlen(argv[i]) == 3 &&
 528                            argv[i][2] >= '1' && argv[i][2] <= '4') {
 529                         pidx = argv[i][2] - '1';
 530                         i++;
 531                         r = parse_hwpart_gp(&pconf, pidx, argc-i, &argv[i]);
 532                         if (r < 0)
 533                                 return CMD_RET_USAGE;
 534                         i += r;
 535                 } else if (!strcmp(argv[i], "check")) {
 536                         mode = MMC_HWPART_CONF_CHECK;
 537                         i++;
 538                 } else if (!strcmp(argv[i], "set")) {
 539                         mode = MMC_HWPART_CONF_SET;
 540                         i++;
 541                 } else if (!strcmp(argv[i], "complete")) {
 542                         mode = MMC_HWPART_CONF_COMPLETE;
 543                         i++;
 544                 } else {
 545                         return CMD_RET_USAGE;
 546                 }
 547         }
 548
 549         puts("Partition configuration:\n");
 550         if (pconf.user.enh_size) {
 551                 puts("\tUser Enhanced Start: ");
 552                 print_size(((u64)pconf.user.enh_start) << 9, "\n");
 553                 puts("\tUser Enhanced Size: ");
 554                 print_size(((u64)pconf.user.enh_size) << 9, "\n");
 555         } else {
 556                 puts("\tNo enhanced user data area\n");
 557         }
 558         if (pconf.user.wr_rel_change)
 559                 printf("\tUser partition write reliability: %s\n",
 560                        pconf.user.wr_rel_set ? "on" : "off");
 561         for (pidx = 0; pidx < 4; pidx++) {
 562                 if (pconf.gp_part[pidx].size) {
 563                         printf("\tGP%i Capacity: ", pidx+1);
 564                         print_size(((u64)pconf.gp_part[pidx].size) << 9,
 565                                    pconf.gp_part[pidx].enhanced ?
 566                                    " ENH\n" : "\n");
 567                 } else {
 568                         printf("\tNo GP%i partition\n", pidx+1);
 569                 }
 570                 if (pconf.gp_part[pidx].wr_rel_change)
 571                         printf("\tGP%i write reliability: %s\n", pidx+1,
 572                                pconf.gp_part[pidx].wr_rel_set ? "on" : "off");
 573         }
 574
 575         if (!mmc_hwpart_config(mmc, &pconf, mode)) {
 576                 if (mode == MMC_HWPART_CONF_COMPLETE)
 577                         puts("Partitioning successful, "
 578                              "power-cycle to make effective\n");
 579                 return CMD_RET_SUCCESS;
 580         } else {
 581                 puts("Failed!\n");
 582                 return CMD_RET_FAILURE;
 583         }
 584 }
 585
 586 #ifdef CONFIG_SUPPORT_EMMC_BOOT
 587 static int do_mmc_bootbus(cmd_tbl_t *cmdtp, int flag,
 588                           int argc, char * const argv[])
 589 {
 590         int dev;
 591         struct mmc *mmc;
 592         u8 width, reset, mode;
 593
 594         if (argc != 5)
 595                 return CMD_RET_USAGE;
 596         dev = simple_strtoul(argv[1], NULL, 10);
 597         width = simple_strtoul(argv[2], NULL, 10);
 598         reset = simple_strtoul(argv[3], NULL, 10);
 599         mode = simple_strtoul(argv[4], NULL, 10);
 600
 601         mmc = init_mmc_device(dev, false);
 602         if (!mmc)
 603                 return CMD_RET_FAILURE;
 604
 605         if (IS_SD(mmc)) {
 606                 puts("BOOT_BUS_WIDTH only exists on eMMC\n");
 607                 return CMD_RET_FAILURE;
 608         }
 609
 610         /* acknowledge to be sent during boot operation */
 611         return mmc_set_boot_bus_width(mmc, width, reset, mode);
 612 }
 613 static int do_mmc_boot_resize(cmd_tbl_t *cmdtp, int flag,
 614                               int argc, char * const argv[])
 615 {
 616         int dev;
 617         struct mmc *mmc;
 618         u32 bootsize, rpmbsize;
 619
 620         if (argc != 4)
 621                 return CMD_RET_USAGE;
 622         dev = simple_strtoul(argv[1], NULL, 10);
 623         bootsize = simple_strtoul(argv[2], NULL, 10);
 624         rpmbsize = simple_strtoul(argv[3], NULL, 10);
 625
 626         mmc = init_mmc_device(dev, false);
 627         if (!mmc)
 628                 return CMD_RET_FAILURE;
 629
 630         if (IS_SD(mmc)) {
 631                 printf("It is not a EMMC device\n");
 632                 return CMD_RET_FAILURE;
 633         }
 634
 635         if (mmc_boot_partition_size_change(mmc, bootsize, rpmbsize)) {
 636                 printf("EMMC boot partition Size change Failed.\n");
 637                 return CMD_RET_FAILURE;
 638         }
 639
 640         printf("EMMC boot partition Size %d MB\n", bootsize);
 641         printf("EMMC RPMB partition Size %d MB\n", rpmbsize);
 642         return CMD_RET_SUCCESS;
 643 }
 644
 645 static int mmc_partconf_print(struct mmc *mmc)
 646 {
 647         u8 ack, access, part;
 648
 649         if (mmc->part_config == MMCPART_NOAVAILABLE) {
 650                 printf("No part_config info for ver. 0x%x\n", mmc->version);
 651                 return CMD_RET_FAILURE;
 652         }
 653
 654         access = EXT_CSD_EXTRACT_PARTITION_ACCESS(mmc->part_config);
 655         ack = EXT_CSD_EXTRACT_BOOT_ACK(mmc->part_config);
 656         part = EXT_CSD_EXTRACT_BOOT_PART(mmc->part_config);
 657
 658         printf("EXT_CSD[179], PARTITION_CONFIG:\n"
 659                 "BOOT_ACK: 0x%x\n"
 660                 "BOOT_PARTITION_ENABLE: 0x%x\n"
 661                 "PARTITION_ACCESS: 0x%x\n", ack, part, access);
 662
 663         return CMD_RET_SUCCESS;
 664 }
 665
 666 static int do_mmc_partconf(cmd_tbl_t *cmdtp, int flag,
 667                            int argc, char * const argv[])
 668 {
 669         int dev;
 670         struct mmc *mmc;
 671         u8 ack, part_num, access;
 672
 673         if (argc != 2 && argc != 5)
 674                 return CMD_RET_USAGE;
 675
 676         dev = simple_strtoul(argv[1], NULL, 10);
 677
 678         mmc = init_mmc_device(dev, false);
 679         if (!mmc)
 680                 return CMD_RET_FAILURE;
 681
 682         if (IS_SD(mmc)) {
 683                 puts("PARTITION_CONFIG only exists on eMMC\n");
 684                 return CMD_RET_FAILURE;
 685         }
 686
 687         if (argc == 2)
 688                 return mmc_partconf_print(mmc);
 689
 690         ack = simple_strtoul(argv[2], NULL, 10);
 691         part_num = simple_strtoul(argv[3], NULL, 10);
 692         access = simple_strtoul(argv[4], NULL, 10);
 693
 694         /* acknowledge to be sent during boot operation */
 695         return mmc_set_part_conf(mmc, ack, part_num, access);
 696 }
 697 static int do_mmc_rst_func(cmd_tbl_t *cmdtp, int flag,
 698                            int argc, char * const argv[])
 699 {
 700         int dev;
 701         struct mmc *mmc;
 702         u8 enable;
 703
 704         /*
 705          * Set the RST_n_ENABLE bit of RST_n_FUNCTION
 706          * The only valid values are 0x0, 0x1 and 0x2 and writing
 707          * a value of 0x1 or 0x2 sets the value permanently.
 708          */
 709         if (argc != 3)
 710                 return CMD_RET_USAGE;
 711
 712         dev = simple_strtoul(argv[1], NULL, 10);
 713         enable = simple_strtoul(argv[2], NULL, 10);
 714
 715         if (enable > 2) {
 716                 puts("Invalid RST_n_ENABLE value\n");
 717                 return CMD_RET_USAGE;
 718         }
 719
 720         mmc = init_mmc_device(dev, false);
 721         if (!mmc)
 722                 return CMD_RET_FAILURE;
 723
 724         if (IS_SD(mmc)) {
 725                 puts("RST_n_FUNCTION only exists on eMMC\n");
 726                 return CMD_RET_FAILURE;
 727         }
 728
 729         return mmc_set_rst_n_function(mmc, enable);
 730 }
 731 #endif
 732 static int do_mmc_setdsr(cmd_tbl_t *cmdtp, int flag,
 733                          int argc, char * const argv[])
 734 {
 735         struct mmc *mmc;
 736         u32 val;
 737         int ret;
 738
 739         if (argc != 2)
 740                 return CMD_RET_USAGE;
 741         val = simple_strtoul(argv[1], NULL, 16);
 742
 743         mmc = find_mmc_device(curr_device);
 744         if (!mmc) {
 745                 printf("no mmc device at slot %x\n", curr_device);
 746                 return CMD_RET_FAILURE;
 747         }
 748         ret = mmc_set_dsr(mmc, val);
 749         printf("set dsr %s\n", (!ret) ? "OK, force rescan" : "ERROR");
 750         if (!ret) {
 751                 mmc->has_init = 0;
 752                 if (mmc_init(mmc))
 753                         return CMD_RET_FAILURE;
 754                 else
 755                         return CMD_RET_SUCCESS;
 756         }
 757         return ret;
 758 }
 759
 760 #ifdef CONFIG_CMD_BKOPS_ENABLE
 761 static int do_mmc_bkops_enable(cmd_tbl_t *cmdtp, int flag,
 762                                    int argc, char * const argv[])
 763 {
 764         int dev;
 765         struct mmc *mmc;
 766
 767         if (argc != 2)
 768                 return CMD_RET_USAGE;
 769
 770         dev = simple_strtoul(argv[1], NULL, 10);
 771
 772         mmc = init_mmc_device(dev, false);
 773         if (!mmc)
 774                 return CMD_RET_FAILURE;
 775
 776         if (IS_SD(mmc)) {
 777                 puts("BKOPS_EN only exists on eMMC\n");
 778                 return CMD_RET_FAILURE;
 779         }
 780
 781         return mmc_set_bkops_enable(mmc);
 782 }
 783 #endif
 784
 785 static cmd_tbl_t cmd_mmc[] = {
 786         U_BOOT_CMD_MKENT(info, 1, 0, do_mmcinfo, "", ""),
 787         U_BOOT_CMD_MKENT(read, 4, 1, do_mmc_read, "", ""),
 788         U_BOOT_CMD_MKENT(write, 4, 0, do_mmc_write, "", ""),
 789         U_BOOT_CMD_MKENT(erase, 3, 0, do_mmc_erase, "", ""),
 790         U_BOOT_CMD_MKENT(rescan, 1, 1, do_mmc_rescan, "", ""),
 791         U_BOOT_CMD_MKENT(part, 1, 1, do_mmc_part, "", ""),
 792         U_BOOT_CMD_MKENT(dev, 3, 0, do_mmc_dev, "", ""),
 793         U_BOOT_CMD_MKENT(list, 1, 1, do_mmc_list, "", ""),
 794         U_BOOT_CMD_MKENT(hwpartition, 28, 0, do_mmc_hwpartition, "", ""),
 795 #ifdef CONFIG_SUPPORT_EMMC_BOOT
 796         U_BOOT_CMD_MKENT(bootbus, 5, 0, do_mmc_bootbus, "", ""),
 797         U_BOOT_CMD_MKENT(bootpart-resize, 4, 0, do_mmc_boot_resize, "", ""),
 798         U_BOOT_CMD_MKENT(partconf, 5, 0, do_mmc_partconf, "", ""),
 799         U_BOOT_CMD_MKENT(rst-function, 3, 0, do_mmc_rst_func, "", ""),
 800 #endif
 801 #ifdef CONFIG_SUPPORT_EMMC_RPMB
 802         U_BOOT_CMD_MKENT(rpmb, CONFIG_SYS_MAXARGS, 1, do_mmcrpmb, "", ""),
 803 #endif
 804         U_BOOT_CMD_MKENT(setdsr, 2, 0, do_mmc_setdsr, "", ""),
 805 #ifdef CONFIG_CMD_BKOPS_ENABLE
 806         U_BOOT_CMD_MKENT(bkops-enable, 2, 0, do_mmc_bkops_enable, "", ""),
 807 #endif
 808 };
 809
 810 static int do_mmcops(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
 811 {
 812         cmd_tbl_t *cp;
 813
 814         cp = find_cmd_tbl(argv[1], cmd_mmc, ARRAY_SIZE(cmd_mmc));
 815
 816         /* Drop the mmc command */
 817         argc--;
 818         argv++;
 819
 820         if (cp == NULL || argc > cp->maxargs)
 821                 return CMD_RET_USAGE;
 822         if (flag == CMD_FLAG_REPEAT && !cp->repeatable)
 823                 return CMD_RET_SUCCESS;
 824
 825         if (curr_device < 0) {
 826                 if (get_mmc_num() > 0) {
 827                         curr_device = 0;
 828                 } else {
 829                         puts("No MMC device available\n");
 830                         return CMD_RET_FAILURE;
 831                 }
 832         }
 833         return cp->cmd(cmdtp, flag, argc, argv);
 834 }
 835
 836 U_BOOT_CMD(
 837         mmc, 29, 1, do_mmcops,
 838         "MMC sub system",
 839         "info - display info of the current MMC device\n"
 840         "mmc read addr blk# cnt\n"
 841         "mmc write addr blk# cnt\n"
 842         "mmc erase blk# cnt\n"
 843         "mmc rescan\n"
 844         "mmc part - lists available partition on current mmc device\n"
 845         "mmc dev [dev] [part] - show or set current mmc device [partition]\n"
 846         "mmc list - lists available devices\n"
 847         "mmc hwpartition [args...] - does hardware partitioning\n"
 848         "  arguments (sizes in 512-byte blocks):\n"
 849         "    [user [enh start cnt] [wrrel {on|off}]] - sets user data area attributes\n"
 850         "    [gp1|gp2|gp3|gp4 cnt [enh] [wrrel {on|off}]] - general purpose partition\n"
 851         "    [check|set|complete] - mode, complete set partitioning completed\n"
 852         "  WARNING: Partitioning is a write-once setting once it is set to complete.\n"
 853         "  Power cycling is required to initialize partitions after set to complete.\n"
 854 #ifdef CONFIG_SUPPORT_EMMC_BOOT
 855         "mmc bootbus dev boot_bus_width reset_boot_bus_width boot_mode\n"
 856         " - Set the BOOT_BUS_WIDTH field of the specified device\n"
 857         "mmc bootpart-resize <dev> <boot part size MB> <RPMB part size MB>\n"
 858         " - Change sizes of boot and RPMB partitions of specified device\n"
 859         "mmc partconf dev [boot_ack boot_partition partition_access]\n"
 860         " - Show or change the bits of the PARTITION_CONFIG field of the specified device\n"
 861         "mmc rst-function dev value\n"
 862         " - Change the RST_n_FUNCTION field of the specified device\n"
 863         "   WARNING: This is a write-once field and 0 / 1 / 2 are the only valid values.\n"
 864 #endif
 865 #ifdef CONFIG_SUPPORT_EMMC_RPMB
 866         "mmc rpmb read addr blk# cnt [address of auth-key] - block size is 256 bytes\n"
 867         "mmc rpmb write addr blk# cnt <address of auth-key> - block size is 256 bytes\n"
 868         "mmc rpmb key <address of auth-key> - program the RPMB authentication key.\n"
 869         "mmc rpmb counter - read the value of the write counter\n"
 870 #endif
 871         "mmc setdsr <value> - set DSR register value\n"
 872 #ifdef CONFIG_CMD_BKOPS_ENABLE
 873         "mmc bkops-enable <dev> - enable background operations handshake on device\n"
 874         "   WARNING: This is a write-once setting.\n"
 875 #endif
 876         );
 877
 878 /* Old command kept for compatibility. Same as 'mmc info' */
 879 U_BOOT_CMD(
 880         mmcinfo, 1, 0, do_mmcinfo,
 881         "display MMC info",
 882         "- display info of the current MMC device"
 883 );