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