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