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