]> git.kernelconcepts.de Git - karo-tx-uboot.git/blob - common/cmd_flash.c
Merge branch 'master' of git://git.denx.de/u-boot-imx
[karo-tx-uboot.git] / common / cmd_flash.c
1 /*
2  * (C) Copyright 2000
3  * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
4  *
5  * See file CREDITS for list of people who contributed to this
6  * project.
7  *
8  * This program is free software; you can redistribute it and/or
9  * modify it under the terms of the GNU General Public License as
10  * published by the Free Software Foundation; either version 2 of
11  * the License, or (at your option) any later version.
12  *
13  * This program is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16  * GNU General Public License for more details.
17  *
18  * You should have received a copy of the GNU General Public License
19  * along with this program; if not, write to the Free Software
20  * Foundation, Inc., 59 Temple Place, Suite 330, Boston,
21  * MA 02111-1307 USA
22  */
23
24 /*
25  * FLASH support
26  */
27 #include <common.h>
28 #include <command.h>
29
30 #ifdef CONFIG_HAS_DATAFLASH
31 #include <dataflash.h>
32 #endif
33
34 #if defined(CONFIG_CMD_JFFS2) && defined(CONFIG_CMD_MTDPARTS)
35 #include <jffs2/jffs2.h>
36
37 /* partition handling routines */
38 int mtdparts_init(void);
39 int mtd_id_parse(const char *id, const char **ret_id, u8 *dev_type, u8 *dev_num);
40 int find_dev_and_part(const char *id, struct mtd_device **dev,
41                 u8 *part_num, struct part_info **part);
42 #endif
43
44 #ifndef CONFIG_SYS_NO_FLASH
45 extern flash_info_t flash_info[];       /* info for FLASH chips */
46
47 /*
48  * The user interface starts numbering for Flash banks with 1
49  * for historical reasons.
50  */
51
52 /*
53  * this routine looks for an abbreviated flash range specification.
54  * the syntax is B:SF[-SL], where B is the bank number, SF is the first
55  * sector to erase, and SL is the last sector to erase (defaults to SF).
56  * bank numbers start at 1 to be consistent with other specs, sector numbers
57  * start at zero.
58  *
59  * returns:     1       - correct spec; *pinfo, *psf and *psl are
60  *                        set appropriately
61  *              0       - doesn't look like an abbreviated spec
62  *              -1      - looks like an abbreviated spec, but got
63  *                        a parsing error, a number out of range,
64  *                        or an invalid flash bank.
65  */
66 static int
67 abbrev_spec (char *str, flash_info_t ** pinfo, int *psf, int *psl)
68 {
69         flash_info_t *fp;
70         int bank, first, last;
71         char *p, *ep;
72
73         if ((p = strchr (str, ':')) == NULL)
74                 return 0;
75         *p++ = '\0';
76
77         bank = simple_strtoul (str, &ep, 10);
78         if (ep == str || *ep != '\0' ||
79                 bank < 1 || bank > CONFIG_SYS_MAX_FLASH_BANKS ||
80                 (fp = &flash_info[bank - 1])->flash_id == FLASH_UNKNOWN)
81                 return -1;
82
83         str = p;
84         if ((p = strchr (str, '-')) != NULL)
85                 *p++ = '\0';
86
87         first = simple_strtoul (str, &ep, 10);
88         if (ep == str || *ep != '\0' || first >= fp->sector_count)
89                 return -1;
90
91         if (p != NULL) {
92                 last = simple_strtoul (p, &ep, 10);
93                 if (ep == p || *ep != '\0' ||
94                         last < first || last >= fp->sector_count)
95                         return -1;
96         } else {
97                 last = first;
98         }
99
100         *pinfo = fp;
101         *psf = first;
102         *psl = last;
103
104         return 1;
105 }
106
107 /*
108  * Take *addr in Flash and adjust it to fall on the end of its sector
109  */
110 int flash_sect_roundb (ulong *addr)
111 {
112         flash_info_t *info;
113         ulong bank, sector_end_addr;
114         char found;
115         int i;
116
117         /* find the end addr of the sector where the *addr is */
118         found = 0;
119         for (bank = 0; bank < CONFIG_SYS_MAX_FLASH_BANKS && !found; ++bank) {
120                 info = &flash_info[bank];
121                 for (i = 0; i < info->sector_count && !found; ++i) {
122                         /* get the end address of the sector */
123                         if (i == info->sector_count - 1) {
124                                 sector_end_addr = info->start[0] +
125                                                                 info->size - 1;
126                         } else {
127                                 sector_end_addr = info->start[i+1] - 1;
128                         }
129
130                         if (*addr <= sector_end_addr &&
131                                                 *addr >= info->start[i]) {
132                                 found = 1;
133                                 /* adjust *addr if necessary */
134                                 if (*addr < sector_end_addr)
135                                         *addr = sector_end_addr;
136                         } /* sector */
137                 } /* bank */
138         }
139         if (!found) {
140                 /* error, addres not in flash */
141                 printf("Error: end address (0x%08lx) not in flash!\n", *addr);
142                 return 1;
143         }
144
145         return 0;
146 }
147
148 /*
149  * This function computes the start and end addresses for both
150  * erase and protect commands. The range of the addresses on which
151  * either of the commands is to operate can be given in two forms:
152  * 1. <cmd> start end - operate on <'start',  'end')
153  * 2. <cmd> start +length - operate on <'start', start + length)
154  * If the second form is used and the end address doesn't fall on the
155  * sector boundary, than it will be adjusted to the next sector boundary.
156  * If it isn't in the flash, the function will fail (return -1).
157  * Input:
158  *    arg1, arg2: address specification (i.e. both command arguments)
159  * Output:
160  *    addr_first, addr_last: computed address range
161  * Return:
162  *    1: success
163  *   -1: failure (bad format, bad address).
164 */
165 static int
166 addr_spec(char *arg1, char *arg2, ulong *addr_first, ulong *addr_last)
167 {
168         char *ep;
169         char len_used; /* indicates if the "start +length" form used */
170
171         *addr_first = simple_strtoul(arg1, &ep, 16);
172         if (ep == arg1 || *ep != '\0')
173                 return -1;
174
175         len_used = 0;
176         if (arg2 && *arg2 == '+'){
177                 len_used = 1;
178                 ++arg2;
179         }
180
181         *addr_last = simple_strtoul(arg2, &ep, 16);
182         if (ep == arg2 || *ep != '\0')
183                 return -1;
184
185         if (len_used){
186                 /*
187                  * *addr_last has the length, compute correct *addr_last
188                  * XXX watch out for the integer overflow! Right now it is
189                  * checked for in both the callers.
190                  */
191                 *addr_last = *addr_first + *addr_last - 1;
192
193                 /*
194                  * It may happen that *addr_last doesn't fall on the sector
195                  * boundary. We want to round such an address to the next
196                  * sector boundary, so that the commands don't fail later on.
197                  */
198
199                 if (flash_sect_roundb(addr_last) > 0)
200                         return -1;
201         } /* "start +length" from used */
202
203         return 1;
204 }
205
206 static int
207 flash_fill_sect_ranges (ulong addr_first, ulong addr_last,
208                         int *s_first, int *s_last,
209                         int *s_count )
210 {
211         flash_info_t *info;
212         ulong bank;
213         int rcode = 0;
214
215         *s_count = 0;
216
217         for (bank=0; bank < CONFIG_SYS_MAX_FLASH_BANKS; ++bank) {
218                 s_first[bank] = -1;     /* first sector to erase        */
219                 s_last [bank] = -1;     /* last  sector to erase        */
220         }
221
222         for (bank=0,info = &flash_info[0];
223              (bank < CONFIG_SYS_MAX_FLASH_BANKS) && (addr_first <= addr_last);
224              ++bank, ++info) {
225                 ulong b_end;
226                 int sect;
227                 short s_end;
228
229                 if (info->flash_id == FLASH_UNKNOWN) {
230                         continue;
231                 }
232
233                 b_end = info->start[0] + info->size - 1;        /* bank end addr */
234                 s_end = info->sector_count - 1;                 /* last sector   */
235
236
237                 for (sect=0; sect < info->sector_count; ++sect) {
238                         ulong end;      /* last address in current sect */
239
240                         end = (sect == s_end) ? b_end : info->start[sect + 1] - 1;
241
242                         if (addr_first > end)
243                                 continue;
244                         if (addr_last < info->start[sect])
245                                 continue;
246
247                         if (addr_first == info->start[sect]) {
248                                 s_first[bank] = sect;
249                         }
250                         if (addr_last  == end) {
251                                 s_last[bank]  = sect;
252                         }
253                 }
254                 if (s_first[bank] >= 0) {
255                         if (s_last[bank] < 0) {
256                                 if (addr_last > b_end) {
257                                         s_last[bank] = s_end;
258                                 } else {
259                                         puts ("Error: end address"
260                                                 " not on sector boundary\n");
261                                         rcode = 1;
262                                         break;
263                                 }
264                         }
265                         if (s_last[bank] < s_first[bank]) {
266                                 puts ("Error: end sector"
267                                         " precedes start sector\n");
268                                 rcode = 1;
269                                 break;
270                         }
271                         sect = s_last[bank];
272                         addr_first = (sect == s_end) ? b_end + 1: info->start[sect + 1];
273                         (*s_count) += s_last[bank] - s_first[bank] + 1;
274                 } else if (addr_first >= info->start[0] && addr_first < b_end) {
275                         puts ("Error: start address not on sector boundary\n");
276                         rcode = 1;
277                         break;
278                 } else if (s_last[bank] >= 0) {
279                         puts ("Error: cannot span across banks when they are"
280                                " mapped in reverse order\n");
281                         rcode = 1;
282                         break;
283                 }
284         }
285
286         return rcode;
287 }
288 #endif /* CONFIG_SYS_NO_FLASH */
289
290 int do_flinfo ( cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])
291 {
292 #ifndef CONFIG_SYS_NO_FLASH
293         ulong bank;
294 #endif
295
296 #ifdef CONFIG_HAS_DATAFLASH
297         dataflash_print_info();
298 #endif
299
300 #ifndef CONFIG_SYS_NO_FLASH
301         if (argc == 1) {        /* print info for all FLASH banks */
302                 for (bank=0; bank <CONFIG_SYS_MAX_FLASH_BANKS; ++bank) {
303                         printf ("\nBank # %ld: ", bank+1);
304
305                         flash_print_info (&flash_info[bank]);
306                 }
307                 return 0;
308         }
309
310         bank = simple_strtoul(argv[1], NULL, 16);
311         if ((bank < 1) || (bank > CONFIG_SYS_MAX_FLASH_BANKS)) {
312                 printf ("Only FLASH Banks # 1 ... # %d supported\n",
313                         CONFIG_SYS_MAX_FLASH_BANKS);
314                 return 1;
315         }
316         printf ("\nBank # %ld: ", bank);
317         flash_print_info (&flash_info[bank-1]);
318 #endif /* CONFIG_SYS_NO_FLASH */
319         return 0;
320 }
321
322 int do_flerase (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])
323 {
324 #ifndef CONFIG_SYS_NO_FLASH
325         flash_info_t *info;
326         ulong bank, addr_first, addr_last;
327         int n, sect_first, sect_last;
328 #if defined(CONFIG_CMD_JFFS2) && defined(CONFIG_CMD_MTDPARTS)
329         struct mtd_device *dev;
330         struct part_info *part;
331         u8 dev_type, dev_num, pnum;
332 #endif
333         int rcode = 0;
334
335         if (argc < 2) {
336                 cmd_usage(cmdtp);
337                 return 1;
338         }
339
340         if (strcmp(argv[1], "all") == 0) {
341                 for (bank=1; bank<=CONFIG_SYS_MAX_FLASH_BANKS; ++bank) {
342                         printf ("Erase Flash Bank # %ld ", bank);
343                         info = &flash_info[bank-1];
344                         rcode = flash_erase (info, 0, info->sector_count-1);
345                 }
346                 return rcode;
347         }
348
349         if ((n = abbrev_spec(argv[1], &info, &sect_first, &sect_last)) != 0) {
350                 if (n < 0) {
351                         puts ("Bad sector specification\n");
352                         return 1;
353                 }
354                 printf ("Erase Flash Sectors %d-%d in Bank # %zu ",
355                         sect_first, sect_last, (info-flash_info)+1);
356                 rcode = flash_erase(info, sect_first, sect_last);
357                 return rcode;
358         }
359
360 #if defined(CONFIG_CMD_JFFS2) && defined(CONFIG_CMD_MTDPARTS)
361         /* erase <part-id> - erase partition */
362         if ((argc == 2) && (mtd_id_parse(argv[1], NULL, &dev_type, &dev_num) == 0)) {
363                 mtdparts_init();
364                 if (find_dev_and_part(argv[1], &dev, &pnum, &part) == 0) {
365                         if (dev->id->type == MTD_DEV_TYPE_NOR) {
366                                 bank = dev->id->num;
367                                 info = &flash_info[bank];
368                                 addr_first = part->offset + info->start[0];
369                                 addr_last = addr_first + part->size - 1;
370
371                                 printf ("Erase Flash Partition %s, "
372                                                 "bank %ld, 0x%08lx - 0x%08lx ",
373                                                 argv[1], bank, addr_first,
374                                                 addr_last);
375
376                                 rcode = flash_sect_erase(addr_first, addr_last);
377                                 return rcode;
378                         }
379
380                         printf("cannot erase, not a NOR device\n");
381                         return 1;
382                 }
383         }
384 #endif
385
386         if (argc != 3) {
387                 cmd_usage(cmdtp);
388                 return 1;
389         }
390
391         if (strcmp(argv[1], "bank") == 0) {
392                 bank = simple_strtoul(argv[2], NULL, 16);
393                 if ((bank < 1) || (bank > CONFIG_SYS_MAX_FLASH_BANKS)) {
394                         printf ("Only FLASH Banks # 1 ... # %d supported\n",
395                                 CONFIG_SYS_MAX_FLASH_BANKS);
396                         return 1;
397                 }
398                 printf ("Erase Flash Bank # %ld ", bank);
399                 info = &flash_info[bank-1];
400                 rcode = flash_erase (info, 0, info->sector_count-1);
401                 return rcode;
402         }
403
404         if (addr_spec(argv[1], argv[2], &addr_first, &addr_last) < 0){
405                 printf ("Bad address format\n");
406                 return 1;
407         }
408
409         if (addr_first >= addr_last) {
410                 cmd_usage(cmdtp);
411                 return 1;
412         }
413
414         rcode = flash_sect_erase(addr_first, addr_last);
415         return rcode;
416 #else
417         return 0;
418 #endif /* CONFIG_SYS_NO_FLASH */
419 }
420
421 #ifndef CONFIG_SYS_NO_FLASH
422 int flash_sect_erase (ulong addr_first, ulong addr_last)
423 {
424         flash_info_t *info;
425         ulong bank;
426 #ifdef CONFIG_SYS_MAX_FLASH_BANKS_DETECT
427         int s_first[CONFIG_SYS_MAX_FLASH_BANKS_DETECT], s_last[CONFIG_SYS_MAX_FLASH_BANKS_DETECT];
428 #else
429         int s_first[CONFIG_SYS_MAX_FLASH_BANKS], s_last[CONFIG_SYS_MAX_FLASH_BANKS];
430 #endif
431         int erased = 0;
432         int planned;
433         int rcode = 0;
434
435         rcode = flash_fill_sect_ranges (addr_first, addr_last,
436                                         s_first, s_last, &planned );
437
438         if (planned && (rcode == 0)) {
439                 for (bank=0,info = &flash_info[0];
440                      (bank < CONFIG_SYS_MAX_FLASH_BANKS) && (rcode == 0);
441                      ++bank, ++info) {
442                         if (s_first[bank]>=0) {
443                                 erased += s_last[bank] - s_first[bank] + 1;
444                                 debug ("Erase Flash from 0x%08lx to 0x%08lx "
445                                         "in Bank # %ld ",
446                                         info->start[s_first[bank]],
447                                         (s_last[bank] == info->sector_count) ?
448                                                 info->start[0] + info->size - 1:
449                                                 info->start[s_last[bank]+1] - 1,
450                                         bank+1);
451                                 rcode = flash_erase (info, s_first[bank], s_last[bank]);
452                         }
453                 }
454                 printf ("Erased %d sectors\n", erased);
455         } else if (rcode == 0) {
456                 puts ("Error: start and/or end address"
457                         " not on sector boundary\n");
458                 rcode = 1;
459         }
460         return rcode;
461 }
462 #endif /* CONFIG_SYS_NO_FLASH */
463
464 int do_protect (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])
465 {
466 #ifndef CONFIG_SYS_NO_FLASH
467         flash_info_t *info;
468         ulong bank;
469         int i, n, sect_first, sect_last;
470 #endif /* CONFIG_SYS_NO_FLASH */
471 #if !defined(CONFIG_SYS_NO_FLASH) || defined(CONFIG_HAS_DATAFLASH)
472         ulong addr_first, addr_last;
473 #endif
474 #if defined(CONFIG_CMD_JFFS2) && defined(CONFIG_CMD_MTDPARTS)
475         struct mtd_device *dev;
476         struct part_info *part;
477         u8 dev_type, dev_num, pnum;
478 #endif
479 #ifdef CONFIG_HAS_DATAFLASH
480         int status;
481 #endif
482         int p;
483         int rcode = 0;
484
485         if (argc < 3) {
486                 cmd_usage(cmdtp);
487                 return 1;
488         }
489
490         if (strcmp(argv[1], "off") == 0) {
491                 p = 0;
492         } else if (strcmp(argv[1], "on") == 0) {
493                 p = 1;
494         } else {
495                 cmd_usage(cmdtp);
496                 return 1;
497         }
498
499 #ifdef CONFIG_HAS_DATAFLASH
500         if ((strcmp(argv[2], "all") != 0) && (strcmp(argv[2], "bank") != 0)) {
501                 addr_first = simple_strtoul(argv[2], NULL, 16);
502                 addr_last  = simple_strtoul(argv[3], NULL, 16);
503
504                 if (addr_dataflash(addr_first) && addr_dataflash(addr_last)) {
505                         status = dataflash_real_protect(p,addr_first,addr_last);
506                         if (status < 0){
507                                 puts ("Bad DataFlash sector specification\n");
508                                 return 1;
509                         }
510                         printf("%sProtect %d DataFlash Sectors\n",
511                                 p ? "" : "Un-", status);
512                         return 0;
513                 }
514         }
515 #endif
516
517 #ifndef CONFIG_SYS_NO_FLASH
518         if (strcmp(argv[2], "all") == 0) {
519                 for (bank=1; bank<=CONFIG_SYS_MAX_FLASH_BANKS; ++bank) {
520                         info = &flash_info[bank-1];
521                         if (info->flash_id == FLASH_UNKNOWN) {
522                                 continue;
523                         }
524                         printf ("%sProtect Flash Bank # %ld\n",
525                                 p ? "" : "Un-", bank);
526
527                         for (i=0; i<info->sector_count; ++i) {
528 #if defined(CONFIG_SYS_FLASH_PROTECTION)
529                                 if (flash_real_protect(info, i, p))
530                                         rcode = 1;
531                                 putc ('.');
532 #else
533                                 info->protect[i] = p;
534 #endif  /* CONFIG_SYS_FLASH_PROTECTION */
535                         }
536 #if defined(CONFIG_SYS_FLASH_PROTECTION)
537                         if (!rcode) puts (" done\n");
538 #endif  /* CONFIG_SYS_FLASH_PROTECTION */
539                 }
540                 return rcode;
541         }
542
543         if ((n = abbrev_spec(argv[2], &info, &sect_first, &sect_last)) != 0) {
544                 if (n < 0) {
545                         puts ("Bad sector specification\n");
546                         return 1;
547                 }
548                 printf("%sProtect Flash Sectors %d-%d in Bank # %zu\n",
549                         p ? "" : "Un-", sect_first, sect_last,
550                         (info-flash_info)+1);
551                 for (i = sect_first; i <= sect_last; i++) {
552 #if defined(CONFIG_SYS_FLASH_PROTECTION)
553                         if (flash_real_protect(info, i, p))
554                                 rcode =  1;
555                         putc ('.');
556 #else
557                         info->protect[i] = p;
558 #endif  /* CONFIG_SYS_FLASH_PROTECTION */
559                 }
560
561 #if defined(CONFIG_SYS_FLASH_PROTECTION)
562                 if (!rcode) puts (" done\n");
563 #endif  /* CONFIG_SYS_FLASH_PROTECTION */
564
565                 return rcode;
566         }
567
568 #if defined(CONFIG_CMD_JFFS2) && defined(CONFIG_CMD_MTDPARTS)
569         /* protect on/off <part-id> */
570         if ((argc == 3) && (mtd_id_parse(argv[2], NULL, &dev_type, &dev_num) == 0)) {
571                 mtdparts_init();
572                 if (find_dev_and_part(argv[2], &dev, &pnum, &part) == 0) {
573                         if (dev->id->type == MTD_DEV_TYPE_NOR) {
574                                 bank = dev->id->num;
575                                 info = &flash_info[bank];
576                                 addr_first = part->offset + info->start[0];
577                                 addr_last = addr_first + part->size - 1;
578
579                                 printf ("%sProtect Flash Partition %s, "
580                                                 "bank %ld, 0x%08lx - 0x%08lx\n",
581                                                 p ? "" : "Un", argv[1],
582                                                 bank, addr_first, addr_last);
583
584                                 rcode = flash_sect_protect (p, addr_first, addr_last);
585                                 return rcode;
586                         }
587
588                         printf("cannot %sprotect, not a NOR device\n",
589                                         p ? "" : "un");
590                         return 1;
591                 }
592         }
593 #endif
594
595         if (argc != 4) {
596                 cmd_usage(cmdtp);
597                 return 1;
598         }
599
600         if (strcmp(argv[2], "bank") == 0) {
601                 bank = simple_strtoul(argv[3], NULL, 16);
602                 if ((bank < 1) || (bank > CONFIG_SYS_MAX_FLASH_BANKS)) {
603                         printf ("Only FLASH Banks # 1 ... # %d supported\n",
604                                 CONFIG_SYS_MAX_FLASH_BANKS);
605                         return 1;
606                 }
607                 printf ("%sProtect Flash Bank # %ld\n",
608                         p ? "" : "Un-", bank);
609                 info = &flash_info[bank-1];
610
611                 if (info->flash_id == FLASH_UNKNOWN) {
612                         puts ("missing or unknown FLASH type\n");
613                         return 1;
614                 }
615                 for (i=0; i<info->sector_count; ++i) {
616 #if defined(CONFIG_SYS_FLASH_PROTECTION)
617                         if (flash_real_protect(info, i, p))
618                                 rcode =  1;
619                         putc ('.');
620 #else
621                         info->protect[i] = p;
622 #endif  /* CONFIG_SYS_FLASH_PROTECTION */
623                 }
624
625 #if defined(CONFIG_SYS_FLASH_PROTECTION)
626                 if (!rcode) puts (" done\n");
627 #endif  /* CONFIG_SYS_FLASH_PROTECTION */
628
629                 return rcode;
630         }
631
632         if (addr_spec(argv[2], argv[3], &addr_first, &addr_last) < 0){
633                 printf("Bad address format\n");
634                 return 1;
635         }
636
637         if (addr_first >= addr_last) {
638                 cmd_usage(cmdtp);
639                 return 1;
640         }
641         rcode = flash_sect_protect (p, addr_first, addr_last);
642 #endif /* CONFIG_SYS_NO_FLASH */
643         return rcode;
644 }
645
646 #ifndef CONFIG_SYS_NO_FLASH
647 int flash_sect_protect (int p, ulong addr_first, ulong addr_last)
648 {
649         flash_info_t *info;
650         ulong bank;
651 #ifdef CONFIG_SYS_MAX_FLASH_BANKS_DETECT
652         int s_first[CONFIG_SYS_MAX_FLASH_BANKS_DETECT], s_last[CONFIG_SYS_MAX_FLASH_BANKS_DETECT];
653 #else
654         int s_first[CONFIG_SYS_MAX_FLASH_BANKS], s_last[CONFIG_SYS_MAX_FLASH_BANKS];
655 #endif
656         int protected, i;
657         int planned;
658         int rcode;
659
660         rcode = flash_fill_sect_ranges( addr_first, addr_last, s_first, s_last, &planned );
661
662         protected = 0;
663
664         if (planned && (rcode == 0)) {
665                 for (bank=0,info = &flash_info[0]; bank < CONFIG_SYS_MAX_FLASH_BANKS; ++bank, ++info) {
666                         if (info->flash_id == FLASH_UNKNOWN) {
667                                 continue;
668                         }
669
670                         if (s_first[bank]>=0 && s_first[bank]<=s_last[bank]) {
671                                 debug ("%sProtecting sectors %d..%d in bank %ld\n",
672                                         p ? "" : "Un-",
673                                         s_first[bank], s_last[bank], bank+1);
674                                 protected += s_last[bank] - s_first[bank] + 1;
675                                 for (i=s_first[bank]; i<=s_last[bank]; ++i) {
676 #if defined(CONFIG_SYS_FLASH_PROTECTION)
677                                         if (flash_real_protect(info, i, p))
678                                                 rcode = 1;
679                                         putc ('.');
680 #else
681                                         info->protect[i] = p;
682 #endif  /* CONFIG_SYS_FLASH_PROTECTION */
683                                 }
684                         }
685                 }
686 #if defined(CONFIG_SYS_FLASH_PROTECTION)
687                 puts (" done\n");
688 #endif  /* CONFIG_SYS_FLASH_PROTECTION */
689
690                 printf ("%sProtected %d sectors\n",
691                         p ? "" : "Un-", protected);
692         } else if (rcode == 0) {
693                 puts ("Error: start and/or end address"
694                         " not on sector boundary\n");
695                 rcode = 1;
696         }
697         return rcode;
698 }
699 #endif /* CONFIG_SYS_NO_FLASH */
700
701
702 /**************************************************/
703 #if defined(CONFIG_CMD_JFFS2) && defined(CONFIG_CMD_MTDPARTS)
704 # define TMP_ERASE      "erase <part-id>\n    - erase partition\n"
705 # define TMP_PROT_ON    "protect on <part-id>\n    - protect partition\n"
706 # define TMP_PROT_OFF   "protect off <part-id>\n    - make partition writable\n"
707 #else
708 # define TMP_ERASE      /* empty */
709 # define TMP_PROT_ON    /* empty */
710 # define TMP_PROT_OFF   /* empty */
711 #endif
712
713 U_BOOT_CMD(
714         flinfo,    2,    1,    do_flinfo,
715         "print FLASH memory information",
716         "\n    - print information for all FLASH memory banks\n"
717         "flinfo N\n    - print information for FLASH memory bank # N"
718 );
719
720 U_BOOT_CMD(
721         erase,   3,   0,  do_flerase,
722         "erase FLASH memory",
723         "start end\n"
724         "    - erase FLASH from addr 'start' to addr 'end'\n"
725         "erase start +len\n"
726         "    - erase FLASH from addr 'start' to the end of sect "
727         "w/addr 'start'+'len'-1\n"
728         "erase N:SF[-SL]\n    - erase sectors SF-SL in FLASH bank # N\n"
729         "erase bank N\n    - erase FLASH bank # N\n"
730         TMP_ERASE
731         "erase all\n    - erase all FLASH banks"
732 );
733
734 U_BOOT_CMD(
735         protect,  4,  0,   do_protect,
736         "enable or disable FLASH write protection",
737         "on  start end\n"
738         "    - protect FLASH from addr 'start' to addr 'end'\n"
739         "protect on start +len\n"
740         "    - protect FLASH from addr 'start' to end of sect "
741         "w/addr 'start'+'len'-1\n"
742         "protect on  N:SF[-SL]\n"
743         "    - protect sectors SF-SL in FLASH bank # N\n"
744         "protect on  bank N\n    - protect FLASH bank # N\n"
745         TMP_PROT_ON
746         "protect on  all\n    - protect all FLASH banks\n"
747         "protect off start end\n"
748         "    - make FLASH from addr 'start' to addr 'end' writable\n"
749         "protect off start +len\n"
750         "    - make FLASH from addr 'start' to end of sect "
751         "w/addr 'start'+'len'-1 wrtable\n"
752         "protect off N:SF[-SL]\n"
753         "    - make sectors SF-SL writable in FLASH bank # N\n"
754         "protect off bank N\n    - make FLASH bank # N writable\n"
755         TMP_PROT_OFF
756         "protect off all\n    - make all FLASH banks writable"
757 );
758
759 #undef  TMP_ERASE
760 #undef  TMP_PROT_ON
761 #undef  TMP_PROT_OFF