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