1 // SPDX-License-Identifier: GPL-2.0+
14 #if defined(CONFIG_CMD_MTDPARTS)
15 #include <jffs2/jffs2.h>
17 /* partition handling routines */
18 int mtdparts_init(void);
19 int mtd_id_parse(const char *id, const char **ret_id, u8 *dev_type, u8 *dev_num);
20 int find_dev_and_part(const char *id, struct mtd_device **dev,
21 u8 *part_num, struct part_info **part);
24 #ifdef CONFIG_MTD_NOR_FLASH
26 #include <mtd/cfi_flash.h>
27 extern flash_info_t flash_info[]; /* info for FLASH chips */
30 * The user interface starts numbering for Flash banks with 1
31 * for historical reasons.
35 * this routine looks for an abbreviated flash range specification.
36 * the syntax is B:SF[-SL], where B is the bank number, SF is the first
37 * sector to erase, and SL is the last sector to erase (defaults to SF).
38 * bank numbers start at 1 to be consistent with other specs, sector numbers
41 * returns: 1 - correct spec; *pinfo, *psf and *psl are
43 * 0 - doesn't look like an abbreviated spec
44 * -1 - looks like an abbreviated spec, but got
45 * a parsing error, a number out of range,
46 * or an invalid flash bank.
49 abbrev_spec (char *str, flash_info_t ** pinfo, int *psf, int *psl)
52 int bank, first, last;
55 if ((p = strchr (str, ':')) == NULL)
59 bank = simple_strtoul (str, &ep, 10);
60 if (ep == str || *ep != '\0' ||
61 bank < 1 || bank > CONFIG_SYS_MAX_FLASH_BANKS ||
62 (fp = &flash_info[bank - 1])->flash_id == FLASH_UNKNOWN)
66 if ((p = strchr (str, '-')) != NULL)
69 first = simple_strtoul (str, &ep, 10);
70 if (ep == str || *ep != '\0' || first >= fp->sector_count)
74 last = simple_strtoul (p, &ep, 10);
75 if (ep == p || *ep != '\0' ||
76 last < first || last >= fp->sector_count)
90 * Take *addr in Flash and adjust it to fall on the end of its sector
92 int flash_sect_roundb(ulong *addr)
95 ulong bank, sector_end_addr;
99 /* find the end addr of the sector where the *addr is */
101 for (bank = 0; bank < CONFIG_SYS_MAX_FLASH_BANKS && !found; ++bank) {
102 info = &flash_info[bank];
103 for (i = 0; i < info->sector_count && !found; ++i) {
104 /* get the end address of the sector */
105 if (i == info->sector_count - 1) {
106 sector_end_addr = info->start[0] +
109 sector_end_addr = info->start[i+1] - 1;
112 if (*addr <= sector_end_addr &&
113 *addr >= info->start[i]) {
115 /* adjust *addr if necessary */
116 if (*addr < sector_end_addr)
117 *addr = sector_end_addr;
122 /* error, address not in flash */
123 printf("Error: end address (0x%08lx) not in flash!\n", *addr);
131 * This function computes the start and end addresses for both
132 * erase and protect commands. The range of the addresses on which
133 * either of the commands is to operate can be given in two forms:
134 * 1. <cmd> start end - operate on <'start', 'end')
135 * 2. <cmd> start +length - operate on <'start', start + length)
136 * If the second form is used and the end address doesn't fall on the
137 * sector boundary, than it will be adjusted to the next sector boundary.
138 * If it isn't in the flash, the function will fail (return -1).
140 * arg1, arg2: address specification (i.e. both command arguments)
142 * addr_first, addr_last: computed address range
145 * -1: failure (bad format, bad address).
148 addr_spec(char *arg1, char *arg2, ulong *addr_first, ulong *addr_last)
151 char len_used; /* indicates if the "start +length" form used */
153 *addr_first = simple_strtoul(arg1, &ep, 16);
154 if (ep == arg1 || *ep != '\0')
158 if (arg2 && *arg2 == '+'){
163 *addr_last = simple_strtoul(arg2, &ep, 16);
164 if (ep == arg2 || *ep != '\0')
169 * *addr_last has the length, compute correct *addr_last
170 * XXX watch out for the integer overflow! Right now it is
171 * checked for in both the callers.
173 *addr_last = *addr_first + *addr_last - 1;
176 * It may happen that *addr_last doesn't fall on the sector
177 * boundary. We want to round such an address to the next
178 * sector boundary, so that the commands don't fail later on.
181 if (flash_sect_roundb(addr_last) > 0)
183 } /* "start +length" from used */
189 flash_fill_sect_ranges (ulong addr_first, ulong addr_last,
190 int *s_first, int *s_last,
199 for (bank=0; bank < CONFIG_SYS_MAX_FLASH_BANKS; ++bank) {
200 s_first[bank] = -1; /* first sector to erase */
201 s_last [bank] = -1; /* last sector to erase */
204 for (bank=0,info = &flash_info[0];
205 (bank < CONFIG_SYS_MAX_FLASH_BANKS) && (addr_first <= addr_last);
211 if (info->flash_id == FLASH_UNKNOWN) {
215 b_end = info->start[0] + info->size - 1; /* bank end addr */
216 s_end = info->sector_count - 1; /* last sector */
219 for (sect=0; sect < info->sector_count; ++sect) {
220 ulong end; /* last address in current sect */
222 end = (sect == s_end) ? b_end : info->start[sect + 1] - 1;
224 if (addr_first > end)
226 if (addr_last < info->start[sect])
229 if (addr_first == info->start[sect]) {
230 s_first[bank] = sect;
232 if (addr_last == end) {
236 if (s_first[bank] >= 0) {
237 if (s_last[bank] < 0) {
238 if (addr_last > b_end) {
239 s_last[bank] = s_end;
241 puts ("Error: end address"
242 " not on sector boundary\n");
247 if (s_last[bank] < s_first[bank]) {
248 puts ("Error: end sector"
249 " precedes start sector\n");
254 addr_first = (sect == s_end) ? b_end + 1: info->start[sect + 1];
255 (*s_count) += s_last[bank] - s_first[bank] + 1;
256 } else if (addr_first >= info->start[0] && addr_first < b_end) {
257 puts ("Error: start address not on sector boundary\n");
260 } else if (s_last[bank] >= 0) {
261 puts ("Error: cannot span across banks when they are"
262 " mapped in reverse order\n");
270 #endif /* CONFIG_MTD_NOR_FLASH */
272 static int do_flinfo(struct cmd_tbl *cmdtp, int flag, int argc,
275 #ifdef CONFIG_MTD_NOR_FLASH
279 #ifdef CONFIG_MTD_NOR_FLASH
280 if (argc == 1) { /* print info for all FLASH banks */
281 for (bank=0; bank <CONFIG_SYS_MAX_FLASH_BANKS; ++bank) {
282 printf ("\nBank # %ld: ", bank+1);
284 flash_print_info(&flash_info[bank]);
289 bank = simple_strtoul(argv[1], NULL, 16);
290 if ((bank < 1) || (bank > CONFIG_SYS_MAX_FLASH_BANKS)) {
291 printf ("Only FLASH Banks # 1 ... # %d supported\n",
292 CONFIG_SYS_MAX_FLASH_BANKS);
295 printf ("\nBank # %ld: ", bank);
296 flash_print_info(&flash_info[bank - 1]);
297 #endif /* CONFIG_MTD_NOR_FLASH */
301 static int do_flerase(struct cmd_tbl *cmdtp, int flag, int argc,
304 #ifdef CONFIG_MTD_NOR_FLASH
305 flash_info_t *info = NULL;
306 ulong bank, addr_first, addr_last;
307 int n, sect_first = 0, sect_last = 0;
308 #if defined(CONFIG_CMD_MTDPARTS)
309 struct mtd_device *dev;
310 struct part_info *part;
311 u8 dev_type, dev_num, pnum;
316 return CMD_RET_USAGE;
318 if (strcmp(argv[1], "all") == 0) {
319 for (bank=1; bank<=CONFIG_SYS_MAX_FLASH_BANKS; ++bank) {
320 printf ("Erase Flash Bank # %ld ", bank);
321 info = &flash_info[bank-1];
322 rcode = flash_erase(info, 0, info->sector_count - 1);
327 if ((n = abbrev_spec(argv[1], &info, §_first, §_last)) != 0) {
329 puts ("Bad sector specification\n");
332 printf ("Erase Flash Sectors %d-%d in Bank # %zu ",
333 sect_first, sect_last, (info-flash_info)+1);
334 rcode = flash_erase(info, sect_first, sect_last);
338 #if defined(CONFIG_CMD_MTDPARTS)
339 /* erase <part-id> - erase partition */
340 if ((argc == 2) && (mtd_id_parse(argv[1], NULL, &dev_type, &dev_num) == 0)) {
342 if (find_dev_and_part(argv[1], &dev, &pnum, &part) == 0) {
343 if (dev->id->type == MTD_DEV_TYPE_NOR) {
345 info = &flash_info[bank];
346 addr_first = part->offset + info->start[0];
347 addr_last = addr_first + part->size - 1;
349 printf ("Erase Flash Partition %s, "
350 "bank %ld, 0x%08lx - 0x%08lx ",
351 argv[1], bank, addr_first,
354 rcode = flash_sect_erase(addr_first, addr_last);
358 printf("cannot erase, not a NOR device\n");
365 return CMD_RET_USAGE;
367 if (strcmp(argv[1], "bank") == 0) {
368 bank = simple_strtoul(argv[2], NULL, 16);
369 if ((bank < 1) || (bank > CONFIG_SYS_MAX_FLASH_BANKS)) {
370 printf ("Only FLASH Banks # 1 ... # %d supported\n",
371 CONFIG_SYS_MAX_FLASH_BANKS);
374 printf ("Erase Flash Bank # %ld ", bank);
375 info = &flash_info[bank-1];
376 rcode = flash_erase(info, 0, info->sector_count - 1);
380 if (addr_spec(argv[1], argv[2], &addr_first, &addr_last) < 0){
381 printf ("Bad address format\n");
385 if (addr_first >= addr_last)
386 return CMD_RET_USAGE;
388 rcode = flash_sect_erase(addr_first, addr_last);
392 #endif /* CONFIG_MTD_NOR_FLASH */
395 #ifdef CONFIG_MTD_NOR_FLASH
396 int flash_sect_erase(ulong addr_first, ulong addr_last)
400 int s_first[CONFIG_SYS_MAX_FLASH_BANKS], s_last[CONFIG_SYS_MAX_FLASH_BANKS];
405 rcode = flash_fill_sect_ranges (addr_first, addr_last,
406 s_first, s_last, &planned );
408 if (planned && (rcode == 0)) {
409 for (bank=0,info = &flash_info[0];
410 (bank < CONFIG_SYS_MAX_FLASH_BANKS) && (rcode == 0);
412 if (s_first[bank]>=0) {
413 erased += s_last[bank] - s_first[bank] + 1;
414 debug ("Erase Flash from 0x%08lx to 0x%08lx "
416 info->start[s_first[bank]],
417 (s_last[bank] == info->sector_count) ?
418 info->start[0] + info->size - 1:
419 info->start[s_last[bank]+1] - 1,
421 rcode = flash_erase(info, s_first[bank],
426 printf("Erased %d sectors\n", erased);
427 } else if (rcode == 0) {
428 puts ("Error: start and/or end address"
429 " not on sector boundary\n");
434 #endif /* CONFIG_MTD_NOR_FLASH */
436 static int do_protect(struct cmd_tbl *cmdtp, int flag, int argc,
440 #ifdef CONFIG_MTD_NOR_FLASH
441 flash_info_t *info = NULL;
443 int i, n, sect_first = 0, sect_last = 0;
444 #if defined(CONFIG_CMD_MTDPARTS)
445 struct mtd_device *dev;
446 struct part_info *part;
447 u8 dev_type, dev_num, pnum;
449 #endif /* CONFIG_MTD_NOR_FLASH */
450 #if defined(CONFIG_MTD_NOR_FLASH)
452 ulong addr_first, addr_last;
456 return CMD_RET_USAGE;
458 #if defined(CONFIG_MTD_NOR_FLASH)
459 if (strcmp(argv[1], "off") == 0)
461 else if (strcmp(argv[1], "on") == 0)
464 return CMD_RET_USAGE;
467 #ifdef CONFIG_MTD_NOR_FLASH
468 if (strcmp(argv[2], "all") == 0) {
469 for (bank=1; bank<=CONFIG_SYS_MAX_FLASH_BANKS; ++bank) {
470 info = &flash_info[bank-1];
471 if (info->flash_id == FLASH_UNKNOWN) {
474 printf ("%sProtect Flash Bank # %ld\n",
475 p ? "" : "Un-", bank);
477 for (i=0; i<info->sector_count; ++i) {
478 #if defined(CONFIG_SYS_FLASH_PROTECTION)
479 if (flash_real_protect(info, i, p))
483 info->protect[i] = p;
484 #endif /* CONFIG_SYS_FLASH_PROTECTION */
486 #if defined(CONFIG_SYS_FLASH_PROTECTION)
487 if (!rcode) puts (" done\n");
488 #endif /* CONFIG_SYS_FLASH_PROTECTION */
493 if ((n = abbrev_spec(argv[2], &info, §_first, §_last)) != 0) {
495 puts ("Bad sector specification\n");
498 printf("%sProtect Flash Sectors %d-%d in Bank # %zu\n",
499 p ? "" : "Un-", sect_first, sect_last,
500 (info-flash_info)+1);
501 for (i = sect_first; i <= sect_last; i++) {
502 #if defined(CONFIG_SYS_FLASH_PROTECTION)
503 if (flash_real_protect(info, i, p))
507 info->protect[i] = p;
508 #endif /* CONFIG_SYS_FLASH_PROTECTION */
511 #if defined(CONFIG_SYS_FLASH_PROTECTION)
512 if (!rcode) puts (" done\n");
513 #endif /* CONFIG_SYS_FLASH_PROTECTION */
518 #if defined(CONFIG_CMD_MTDPARTS)
519 /* protect on/off <part-id> */
520 if ((argc == 3) && (mtd_id_parse(argv[2], NULL, &dev_type, &dev_num) == 0)) {
522 if (find_dev_and_part(argv[2], &dev, &pnum, &part) == 0) {
523 if (dev->id->type == MTD_DEV_TYPE_NOR) {
525 info = &flash_info[bank];
526 addr_first = part->offset + info->start[0];
527 addr_last = addr_first + part->size - 1;
529 printf ("%sProtect Flash Partition %s, "
530 "bank %ld, 0x%08lx - 0x%08lx\n",
531 p ? "" : "Un", argv[1],
532 bank, addr_first, addr_last);
534 rcode = flash_sect_protect(p, addr_first,
539 printf("cannot %sprotect, not a NOR device\n",
547 return CMD_RET_USAGE;
549 if (strcmp(argv[2], "bank") == 0) {
550 bank = simple_strtoul(argv[3], NULL, 16);
551 if ((bank < 1) || (bank > CONFIG_SYS_MAX_FLASH_BANKS)) {
552 printf ("Only FLASH Banks # 1 ... # %d supported\n",
553 CONFIG_SYS_MAX_FLASH_BANKS);
556 printf ("%sProtect Flash Bank # %ld\n",
557 p ? "" : "Un-", bank);
558 info = &flash_info[bank-1];
560 if (info->flash_id == FLASH_UNKNOWN) {
561 puts ("missing or unknown FLASH type\n");
564 for (i=0; i<info->sector_count; ++i) {
565 #if defined(CONFIG_SYS_FLASH_PROTECTION)
566 if (flash_real_protect(info, i, p))
570 info->protect[i] = p;
571 #endif /* CONFIG_SYS_FLASH_PROTECTION */
574 #if defined(CONFIG_SYS_FLASH_PROTECTION)
575 if (!rcode) puts (" done\n");
576 #endif /* CONFIG_SYS_FLASH_PROTECTION */
581 if (addr_spec(argv[2], argv[3], &addr_first, &addr_last) < 0){
582 printf("Bad address format\n");
586 if (addr_first >= addr_last)
587 return CMD_RET_USAGE;
589 rcode = flash_sect_protect(p, addr_first, addr_last);
590 #endif /* CONFIG_MTD_NOR_FLASH */
594 #ifdef CONFIG_MTD_NOR_FLASH
595 int flash_sect_protect(int p, ulong addr_first, ulong addr_last)
599 int s_first[CONFIG_SYS_MAX_FLASH_BANKS], s_last[CONFIG_SYS_MAX_FLASH_BANKS];
604 rcode = flash_fill_sect_ranges( addr_first, addr_last, s_first, s_last, &planned );
608 if (planned && (rcode == 0)) {
609 for (bank=0,info = &flash_info[0]; bank < CONFIG_SYS_MAX_FLASH_BANKS; ++bank, ++info) {
610 if (info->flash_id == FLASH_UNKNOWN) {
614 if (s_first[bank]>=0 && s_first[bank]<=s_last[bank]) {
615 debug ("%sProtecting sectors %d..%d in bank %ld\n",
617 s_first[bank], s_last[bank], bank+1);
618 protected += s_last[bank] - s_first[bank] + 1;
619 for (i=s_first[bank]; i<=s_last[bank]; ++i) {
620 #if defined(CONFIG_SYS_FLASH_PROTECTION)
621 if (flash_real_protect(info, i, p))
625 info->protect[i] = p;
626 #endif /* CONFIG_SYS_FLASH_PROTECTION */
630 #if defined(CONFIG_SYS_FLASH_PROTECTION)
632 #endif /* CONFIG_SYS_FLASH_PROTECTION */
634 printf ("%sProtected %d sectors\n",
635 p ? "" : "Un-", protected);
636 } else if (rcode == 0) {
637 puts ("Error: start and/or end address"
638 " not on sector boundary\n");
643 #endif /* CONFIG_MTD_NOR_FLASH */
646 /**************************************************/
647 #if defined(CONFIG_CMD_MTDPARTS)
648 # define TMP_ERASE "erase <part-id>\n - erase partition\n"
649 # define TMP_PROT_ON "protect on <part-id>\n - protect partition\n"
650 # define TMP_PROT_OFF "protect off <part-id>\n - make partition writable\n"
652 # define TMP_ERASE /* empty */
653 # define TMP_PROT_ON /* empty */
654 # define TMP_PROT_OFF /* empty */
658 flinfo, 2, 1, do_flinfo,
659 "print FLASH memory information",
660 "\n - print information for all FLASH memory banks\n"
661 "flinfo N\n - print information for FLASH memory bank # N"
665 erase, 3, 0, do_flerase,
666 "erase FLASH memory",
668 " - erase FLASH from addr 'start' to addr 'end'\n"
670 " - erase FLASH from addr 'start' to the end of sect "
671 "w/addr 'start'+'len'-1\n"
672 "erase N:SF[-SL]\n - erase sectors SF-SL in FLASH bank # N\n"
673 "erase bank N\n - erase FLASH bank # N\n"
675 "erase all\n - erase all FLASH banks"
679 protect, 4, 0, do_protect,
680 "enable or disable FLASH write protection",
682 " - protect FLASH from addr 'start' to addr 'end'\n"
683 "protect on start +len\n"
684 " - protect FLASH from addr 'start' to end of sect "
685 "w/addr 'start'+'len'-1\n"
686 "protect on N:SF[-SL]\n"
687 " - protect sectors SF-SL in FLASH bank # N\n"
688 "protect on bank N\n - protect FLASH bank # N\n"
690 "protect on all\n - protect all FLASH banks\n"
691 "protect off start end\n"
692 " - make FLASH from addr 'start' to addr 'end' writable\n"
693 "protect off start +len\n"
694 " - make FLASH from addr 'start' to end of sect "
695 "w/addr 'start'+'len'-1 wrtable\n"
696 "protect off N:SF[-SL]\n"
697 " - make sectors SF-SL writable in FLASH bank # N\n"
698 "protect off bank N\n - make FLASH bank # N writable\n"
700 "protect off all\n - make all FLASH banks writable"