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