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