]> git.sur5r.net Git - u-boot/blob - cmd/mmc.c
fastboot: Enable flashing by default on sunxi
[u-boot] / 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 );