common/cmd_nand.c

   1 /*
   2  * Driver for NAND support, Rick Bronson
   3  * borrowed heavily from:
   4  * (c) 1999 Machine Vision Holdings, Inc.
   5  * (c) 1999, 2000 David Woodhouse <dwmw2@infradead.org>
   6  *
   7  * Ported 'dynenv' to 'nand env.oob' command
   8  * (C) 2010 Nanometrics, Inc.
   9  * 'dynenv' -- Dynamic environment offset in NAND OOB
  10  * (C) Copyright 2006-2007 OpenMoko, Inc.
  11  * Added 16-bit nand support
  12  * (C) 2004 Texas Instruments
  13  */
  14
  15 #include <common.h>
  16 #include <linux/mtd/mtd.h>
  17 #include <command.h>
  18 #include <watchdog.h>
  19 #include <malloc.h>
  20 #include <asm/byteorder.h>
  21 #include <jffs2/jffs2.h>
  22 #include <nand.h>
  23
  24 #if defined(CONFIG_CMD_MTDPARTS)
  25
  26 /* partition handling routines */
  27 int mtdparts_init(void);
  28 int id_parse(const char *id, const char **ret_id, u8 *dev_type, u8 *dev_num);
  29 int find_dev_and_part(const char *id, struct mtd_device **dev,
  30                       u8 *part_num, struct part_info **part);
  31 #endif
  32
  33 static int nand_dump(nand_info_t *nand, ulong off, int only_oob)
  34 {
  35         int i;
  36         u_char *datbuf, *oobbuf, *p;
  37
  38         datbuf = malloc(nand->writesize + nand->oobsize);
  39         oobbuf = malloc(nand->oobsize);
  40         if (!datbuf || !oobbuf) {
  41                 puts("No memory for page buffer\n");
  42                 return 1;
  43         }
  44         off &= ~(nand->writesize - 1);
  45         loff_t addr = (loff_t) off;
  46         struct mtd_oob_ops ops;
  47         memset(&ops, 0, sizeof(ops));
  48         ops.datbuf = datbuf;
  49         ops.oobbuf = oobbuf; /* must exist, but oob data will be appended to ops.datbuf */
  50         ops.len = nand->writesize;
  51         ops.ooblen = nand->oobsize;
  52         ops.mode = MTD_OOB_RAW;
  53         i = nand->read_oob(nand, addr, &ops);
  54         if (i < 0) {
  55                 printf("Error (%d) reading page %08lx\n", i, off);
  56                 free(datbuf);
  57                 free(oobbuf);
  58                 return 1;
  59         }
  60         printf("Page %08lx dump:\n", off);
  61         i = nand->writesize >> 4;
  62         p = datbuf;
  63
  64         while (i--) {
  65                 if (!only_oob)
  66                         printf("\t%02x %02x %02x %02x %02x %02x %02x %02x"
  67                                "  %02x %02x %02x %02x %02x %02x %02x %02x\n",
  68                                p[0], p[1], p[2], p[3], p[4], p[5], p[6], p[7],
  69                                p[8], p[9], p[10], p[11], p[12], p[13], p[14],
  70                                p[15]);
  71                 p += 16;
  72         }
  73         puts("OOB:\n");
  74         i = nand->oobsize >> 3;
  75         while (i--) {
  76                 printf("\t%02x %02x %02x %02x %02x %02x %02x %02x\n",
  77                        p[0], p[1], p[2], p[3], p[4], p[5], p[6], p[7]);
  78                 p += 8;
  79         }
  80         free(datbuf);
  81         free(oobbuf);
  82
  83         return 0;
  84 }
  85
  86 /* ------------------------------------------------------------------------- */
  87
  88 static inline int str2long(char *p, ulong *num)
  89 {
  90         char *endptr;
  91
  92         *num = simple_strtoul(p, &endptr, 16);
  93         return (*p != '\0' && *endptr == '\0') ? 1 : 0;
  94 }
  95
  96 static int
  97 arg_off_size(int argc, char * const argv[], nand_info_t *nand, ulong *off, size_t *size)
  98 {
  99         int idx = nand_curr_device;
 100 #if defined(CONFIG_CMD_MTDPARTS)
 101         struct mtd_device *dev;
 102         struct part_info *part;
 103         u8 pnum;
 104
 105         if (argc >= 1 && !(str2long(argv[0], off))) {
 106                 if ((mtdparts_init() == 0) &&
 107                     (find_dev_and_part(argv[0], &dev, &pnum, &part) == 0)) {
 108                         if (dev->id->type != MTD_DEV_TYPE_NAND) {
 109                                 puts("not a NAND device\n");
 110                                 return -1;
 111                         }
 112                         *off = part->offset;
 113                         if (argc >= 2) {
 114                                 if (!(str2long(argv[1], (ulong *)size))) {
 115                                         printf("'%s' is not a number\n", argv[1]);
 116                                         return -1;
 117                                 }
 118                                 if (*size > part->size)
 119                                         *size = part->size;
 120                         } else {
 121                                 *size = part->size;
 122                         }
 123                         idx = dev->id->num;
 124                         *nand = nand_info[idx];
 125                         goto out;
 126                 }
 127         }
 128 #endif
 129
 130         if (argc >= 1) {
 131                 if (!(str2long(argv[0], off))) {
 132                         printf("'%s' is not a number\n", argv[0]);
 133                         return -1;
 134                 }
 135         } else {
 136                 *off = 0;
 137         }
 138
 139         if (argc >= 2) {
 140                 if (!(str2long(argv[1], (ulong *)size))) {
 141                         printf("'%s' is not a number\n", argv[1]);
 142                         return -1;
 143                 }
 144         } else {
 145                 *size = nand->size - *off;
 146         }
 147
 148 #if defined(CONFIG_CMD_MTDPARTS)
 149 out:
 150 #endif
 151         printf("device %d ", idx);
 152         if (*size == nand->size)
 153                 puts("whole chip\n");
 154         else
 155                 printf("offset 0x%lx, size 0x%zx\n", *off, *size);
 156         return 0;
 157 }
 158
 159 #ifdef CONFIG_CMD_NAND_LOCK_UNLOCK
 160 static void print_status(ulong start, ulong end, ulong erasesize, int status)
 161 {
 162         printf("%08lx - %08lx: %08lx blocks %s%s%s\n",
 163                 start,
 164                 end - 1,
 165                 (end - start) / erasesize,
 166                 ((status & NAND_LOCK_STATUS_TIGHT) ?  "TIGHT " : ""),
 167                 ((status & NAND_LOCK_STATUS_LOCK) ?  "LOCK " : ""),
 168                 ((status & NAND_LOCK_STATUS_UNLOCK) ?  "UNLOCK " : ""));
 169 }
 170
 171 static void do_nand_status(nand_info_t *nand)
 172 {
 173         ulong block_start = 0;
 174         ulong off;
 175         int last_status = -1;
 176
 177         struct nand_chip *nand_chip = nand->priv;
 178         /* check the WP bit */
 179         nand_chip->cmdfunc(nand, NAND_CMD_STATUS, -1, -1);
 180         printf("device is %swrite protected\n",
 181                 (nand_chip->read_byte(nand) & 0x80 ?
 182                 "NOT " : ""));
 183
 184         for (off = 0; off < nand->size; off += nand->erasesize) {
 185                 int s = nand_get_lock_status(nand, off);
 186
 187                 /* print message only if status has changed */
 188                 if (s != last_status && off != 0) {
 189                         print_status(block_start, off, nand->erasesize,
 190                                         last_status);
 191                         block_start = off;
 192                 }
 193                 last_status = s;
 194         }
 195         /* Print the last block info */
 196         print_status(block_start, off, nand->erasesize, last_status);
 197 }
 198 #endif
 199
 200 #ifdef CONFIG_ENV_OFFSET_OOB
 201 unsigned long nand_env_oob_offset;
 202
 203 int do_nand_env_oob(cmd_tbl_t *cmdtp, nand_info_t *nand,
 204                                  int argc, char * const argv[])
 205 {
 206         int ret;
 207         uint32_t oob_buf[ENV_OFFSET_SIZE/sizeof(uint32_t)];
 208
 209         char *cmd = argv[1];
 210
 211         if (!strcmp(cmd, "get")) {
 212                 ret = get_nand_env_oob(nand, &nand_env_oob_offset);
 213                 if (!ret)
 214                         printf("0x%08lx\n", nand_env_oob_offset);
 215                 else
 216                         return 1;
 217         } else if (!strcmp(cmd, "set")) {
 218                 ulong addr;
 219                 size_t dummy_size;
 220                 struct mtd_oob_ops ops;
 221
 222                 if (argc < 3)
 223                         goto usage;
 224
 225                 if (arg_off_size(argc-2, argv + 2, nand, &addr,
 226                                                   &dummy_size) < 0) {
 227                         printf("Offset or partition name expected\n");
 228                         return 1;
 229                 }
 230
 231                 if (nand->oobavail < ENV_OFFSET_SIZE) {
 232                         printf("Insufficient available OOB bytes: %d OOB bytes"
 233                             " available but %d required for env.oob support\n",
 234                                                                 nand->oobavail,
 235                                                               ENV_OFFSET_SIZE);
 236                         return 1;
 237                 }
 238
 239                 if ((addr & (nand->erasesize - 1)) != 0) {
 240                         printf("Environment offset must be block-aligned\n");
 241                         return 1;
 242                 }
 243
 244                 ops.datbuf = NULL;
 245                 ops.mode = MTD_OOB_AUTO;
 246                 ops.ooboffs = 0;
 247                 ops.ooblen = ENV_OFFSET_SIZE;
 248                 ops.oobbuf = (void *) oob_buf;
 249
 250                 oob_buf[0] = ENV_OOB_MARKER;
 251                 oob_buf[1] = addr / nand->erasesize;
 252
 253                 ret = nand->write_oob(nand, ENV_OFFSET_SIZE, &ops);
 254                 if (!ret) {
 255                         ret = get_nand_env_oob(nand, &nand_env_oob_offset);
 256                         if (ret) {
 257                                 printf("Error reading env offset in OOB\n");
 258                                 return ret;
 259                         }
 260
 261                         if (addr != nand_env_oob_offset) {
 262                                 printf("Verification of env offset in OOB "
 263                                        "failed: 0x%08lx expected but got "
 264                                        "0x%08lx\n", addr, nand_env_oob_offset);
 265                                 return 1;
 266                         }
 267                 } else {
 268                         printf("Error writing OOB block 0\n");
 269                         return ret;
 270                 }
 271         } else {
 272                 goto usage;
 273         }
 274
 275         return ret;
 276
 277 usage:
 278         cmd_usage(cmdtp);
 279         return 1;
 280 }
 281
 282 #endif
 283
 284 static void nand_print_info(int idx)
 285 {
 286         nand_info_t *nand = &nand_info[idx];
 287         struct nand_chip *chip = nand->priv;
 288         printf("Device %d: ", idx);
 289         if (chip->numchips > 1)
 290                 printf("%dx ", chip->numchips);
 291         printf("%s, sector size %u KiB\n",
 292                nand->name, nand->erasesize >> 10);
 293 }
 294
 295 int do_nand(cmd_tbl_t * cmdtp, int flag, int argc, char * const argv[])
 296 {
 297         int i, dev, ret = 0;
 298         ulong addr, off;
 299         size_t size;
 300         char *cmd, *s;
 301         nand_info_t *nand;
 302 #ifdef CONFIG_SYS_NAND_QUIET
 303         int quiet = CONFIG_SYS_NAND_QUIET;
 304 #else
 305         int quiet = 0;
 306 #endif
 307         const char *quiet_str = getenv("quiet");
 308
 309         /* at least two arguments please */
 310         if (argc < 2)
 311                 goto usage;
 312
 313         if (quiet_str)
 314                 quiet = simple_strtoul(quiet_str, NULL, 0) != 0;
 315
 316         cmd = argv[1];
 317
 318         if (strcmp(cmd, "info") == 0) {
 319
 320                 putc('\n');
 321                 for (i = 0; i < CONFIG_SYS_MAX_NAND_DEVICE; i++) {
 322                         if (nand_info[i].name)
 323                                 nand_print_info(i);
 324                 }
 325                 return 0;
 326         }
 327
 328         if (strcmp(cmd, "device") == 0) {
 329
 330                 if (argc < 3) {
 331                         putc('\n');
 332                         if ((nand_curr_device < 0) ||
 333                             (nand_curr_device >= CONFIG_SYS_MAX_NAND_DEVICE))
 334                                 puts("no devices available\n");
 335                         else
 336                                 nand_print_info(nand_curr_device);
 337                         return 0;
 338                 }
 339                 dev = (int)simple_strtoul(argv[2], NULL, 10);
 340                 if (dev < 0 || dev >= CONFIG_SYS_MAX_NAND_DEVICE || !nand_info[dev].name) {
 341                         puts("No such device\n");
 342                         return 1;
 343                 }
 344                 printf("Device %d: %s", dev, nand_info[dev].name);
 345                 puts("... is now current device\n");
 346                 nand_curr_device = dev;
 347
 348 #ifdef CONFIG_SYS_NAND_SELECT_DEVICE
 349                 /*
 350                  * Select the chip in the board/cpu specific driver
 351                  */
 352                 board_nand_select_device(nand_info[dev].priv, dev);
 353 #endif
 354
 355                 return 0;
 356         }
 357
 358         if (strcmp(cmd, "bad") != 0 && strcmp(cmd, "erase") != 0 &&
 359             strncmp(cmd, "dump", 4) != 0 &&
 360             strncmp(cmd, "read", 4) != 0 && strncmp(cmd, "write", 5) != 0 &&
 361             strcmp(cmd, "scrub") != 0 && strcmp(cmd, "markbad") != 0 &&
 362             strcmp(cmd, "biterr") != 0 &&
 363             strcmp(cmd, "lock") != 0 && strcmp(cmd, "unlock") != 0
 364 #ifdef CONFIG_ENV_OFFSET_OOB
 365             && strcmp(cmd, "env.oob") != 0
 366 #endif
 367             )
 368                 goto usage;
 369
 370 #ifdef CONFIG_ENV_OFFSET_OOB
 371         /* this command operates only on the first nand device */
 372         if (strcmp(cmd, "env.oob") == 0)
 373               return do_nand_env_oob(cmdtp, &nand_info[0], argc - 1, argv + 1);
 374 #endif
 375
 376         /* the following commands operate on the current device */
 377         if (nand_curr_device < 0 || nand_curr_device >= CONFIG_SYS_MAX_NAND_DEVICE ||
 378             !nand_info[nand_curr_device].name) {
 379                 puts("\nno devices available\n");
 380                 return 1;
 381         }
 382         nand = &nand_info[nand_curr_device];
 383
 384         if (strcmp(cmd, "bad") == 0) {
 385                 printf("\nDevice %d bad blocks:\n", nand_curr_device);
 386                 for (off = 0; off < nand->size; off += nand->erasesize)
 387                         if (nand_block_isbad(nand, off))
 388                                 printf("  %08lx\n", off);
 389                 return 0;
 390         }
 391
 392         /*
 393          * Syntax is:
 394          *   0    1     2       3    4
 395          *   nand erase [clean] [off size]
 396          */
 397         if (strcmp(cmd, "erase") == 0 || strcmp(cmd, "scrub") == 0) {
 398                 nand_erase_options_t opts;
 399                 /* "clean" at index 2 means request to write cleanmarker */
 400                 int clean = argc > 2 && !strcmp("clean", argv[2]);
 401                 int o = clean ? 3 : 2;
 402                 int scrub = !strcmp(cmd, "scrub");
 403
 404                 printf("\nNAND %s: ", scrub ? "scrub" : "erase");
 405                 /* skip first two or three arguments, look for offset and size */
 406                 if (arg_off_size(argc - o, argv + o, nand, &off, &size) != 0)
 407                         return 1;
 408
 409                 memset(&opts, 0, sizeof(opts));
 410                 opts.offset = off;
 411                 opts.length = size;
 412                 opts.jffs2  = clean;
 413                 opts.quiet  = quiet;
 414
 415                 if (scrub) {
 416                         puts("Warning: "
 417                              "scrub option will erase all factory set "
 418                              "bad blocks!\n"
 419                              "         "
 420                              "There is no reliable way to recover them.\n"
 421                              "         "
 422                              "Use this command only for testing purposes "
 423                              "if you\n"
 424                              "         "
 425                              "are sure of what you are doing!\n"
 426                              "\nReally scrub this NAND flash? <y/N>\n");
 427
 428                         if (getc() == 'y') {
 429                                 puts("y");
 430                                 if (getc() == '\r')
 431                                         opts.scrub = 1;
 432                                 else {
 433                                         puts("scrub aborted\n");
 434                                         return -1;
 435                                 }
 436                         } else {
 437                                 puts("scrub aborted\n");
 438                                 return -1;
 439                         }
 440                 }
 441                 ret = nand_erase_opts(nand, &opts);
 442                 printf("%s\n", ret ? "ERROR" : "OK");
 443
 444                 return ret == 0 ? 0 : 1;
 445         }
 446
 447         if (strncmp(cmd, "dump", 4) == 0) {
 448                 if (argc < 3)
 449                         goto usage;
 450
 451                 s = strchr(cmd, '.');
 452                 off = (int)simple_strtoul(argv[2], NULL, 16);
 453
 454                 if (s != NULL && strcmp(s, ".oob") == 0)
 455                         ret = nand_dump(nand, off, 1);
 456                 else
 457                         ret = nand_dump(nand, off, 0);
 458
 459                 return ret == 0 ? 1 : 0;
 460
 461         }
 462
 463         if (strncmp(cmd, "read", 4) == 0 || strncmp(cmd, "write", 5) == 0) {
 464                 int read;
 465
 466                 if (argc < 4)
 467                         goto usage;
 468
 469                 addr = (ulong)simple_strtoul(argv[2], NULL, 16);
 470
 471                 read = strncmp(cmd, "read", 4) == 0; /* 1 = read, 0 = write */
 472                 printf("\nNAND %s: ", read ? "read" : "write");
 473                 if (arg_off_size(argc - 3, argv + 3, nand, &off, &size) != 0)
 474                         return 1;
 475
 476                 s = strchr(cmd, '.');
 477                 if (!s || !strcmp(s, ".jffs2") ||
 478                     !strcmp(s, ".e") || !strcmp(s, ".i")) {
 479                         if (read)
 480                                 ret = nand_read_skip_bad(nand, off, &size,
 481                                                          (u_char *)addr);
 482                         else
 483                                 ret = nand_write_skip_bad(nand, off, &size,
 484                                                           (u_char *)addr);
 485                 } else if (!strcmp(s, ".oob")) {
 486                         /* out-of-band data */
 487                         mtd_oob_ops_t ops = {
 488                                 .oobbuf = (u8 *)addr,
 489                                 .ooblen = size,
 490                                 .mode = MTD_OOB_RAW
 491                         };
 492
 493                         if (read)
 494                                 ret = nand->read_oob(nand, off, &ops);
 495                         else
 496                                 ret = nand->write_oob(nand, off, &ops);
 497                 } else {
 498                         printf("Unknown nand command suffix '%s'.\n", s);
 499                         return 1;
 500                 }
 501
 502                 printf(" %zu bytes %s: %s\n", size,
 503                        read ? "read" : "written", ret ? "ERROR" : "OK");
 504
 505                 return ret == 0 ? 0 : 1;
 506         }
 507
 508         if (strcmp(cmd, "markbad") == 0) {
 509                 argc -= 2;
 510                 argv += 2;
 511
 512                 if (argc <= 0)
 513                         goto usage;
 514
 515                 while (argc > 0) {
 516                         addr = simple_strtoul(*argv, NULL, 16);
 517
 518                         if (nand->block_markbad(nand, addr)) {
 519                                 printf("block 0x%08lx NOT marked "
 520                                         "as bad! ERROR %d\n",
 521                                         addr, ret);
 522                                 ret = 1;
 523                         } else {
 524                                 printf("block 0x%08lx successfully "
 525                                         "marked as bad\n",
 526                                         addr);
 527                         }
 528                         --argc;
 529                         ++argv;
 530                 }
 531                 return ret;
 532         }
 533
 534         if (strcmp(cmd, "biterr") == 0) {
 535                 /* todo */
 536                 return 1;
 537         }
 538
 539 #ifdef CONFIG_CMD_NAND_LOCK_UNLOCK
 540         if (strcmp(cmd, "lock") == 0) {
 541                 int tight = 0;
 542                 int status = 0;
 543                 if (argc == 3) {
 544                         if (!strcmp("tight", argv[2]))
 545                                 tight = 1;
 546                         if (!strcmp("status", argv[2]))
 547                                 status = 1;
 548                 }
 549                 if (status) {
 550                         do_nand_status(nand);
 551                 } else {
 552                         if (!nand_lock(nand, tight)) {
 553                                 puts("NAND flash successfully locked\n");
 554                         } else {
 555                                 puts("Error locking NAND flash\n");
 556                                 return 1;
 557                         }
 558                 }
 559                 return 0;
 560         }
 561
 562         if (strcmp(cmd, "unlock") == 0) {
 563                 if (arg_off_size(argc - 2, argv + 2, nand, &off, &size) < 0)
 564                         return 1;
 565
 566                 if (!nand_unlock(nand, off, size)) {
 567                         puts("NAND flash successfully unlocked\n");
 568                 } else {
 569                         puts("Error unlocking NAND flash, "
 570                              "write and erase will probably fail\n");
 571                         return 1;
 572                 }
 573                 return 0;
 574         }
 575 #endif
 576
 577 usage:
 578         cmd_usage(cmdtp);
 579         return 1;
 580 }
 581
 582 U_BOOT_CMD(nand, CONFIG_SYS_MAXARGS, 1, do_nand,
 583         "NAND sub-system",
 584         "info - show available NAND devices\n"
 585         "nand device [dev] - show or set current device\n"
 586         "nand read - addr off|partition size\n"
 587         "nand write - addr off|partition size\n"
 588         "    read/write 'size' bytes starting at offset 'off'\n"
 589         "    to/from memory address 'addr', skipping bad blocks.\n"
 590         "nand erase [clean] [off size] - erase 'size' bytes from\n"
 591         "    offset 'off' (entire device if not specified)\n"
 592         "nand bad - show bad blocks\n"
 593         "nand dump[.oob] off - dump page\n"
 594         "nand scrub - really clean NAND erasing bad blocks (UNSAFE)\n"
 595         "nand markbad off [...] - mark bad block(s) at offset (UNSAFE)\n"
 596         "nand biterr off - make a bit error at offset (UNSAFE)"
 597 #ifdef CONFIG_CMD_NAND_LOCK_UNLOCK
 598         "\n"
 599         "nand lock [tight] [status]\n"
 600         "    bring nand to lock state or display locked pages\n"
 601         "nand unlock [offset] [size] - unlock section"
 602 #endif
 603 #ifdef CONFIG_ENV_OFFSET_OOB
 604         "\n"
 605         "nand env.oob - environment offset in OOB of block 0 of"
 606         "    first device.\n"
 607         "nand env.oob set off|partition - set enviromnent offset\n"
 608         "nand env.oob get - get environment offset"
 609 #endif
 610 );
 611
 612 static int nand_load_image(cmd_tbl_t *cmdtp, nand_info_t *nand,
 613                            ulong offset, ulong addr, char *cmd)
 614 {
 615         int r;
 616         char *ep, *s;
 617         size_t cnt;
 618         image_header_t *hdr;
 619 #if defined(CONFIG_FIT)
 620         const void *fit_hdr = NULL;
 621 #endif
 622
 623         s = strchr(cmd, '.');
 624         if (s != NULL &&
 625             (strcmp(s, ".jffs2") && strcmp(s, ".e") && strcmp(s, ".i"))) {
 626                 printf("Unknown nand load suffix '%s'\n", s);
 627                 show_boot_progress(-53);
 628                 return 1;
 629         }
 630
 631         printf("\nLoading from %s, offset 0x%lx\n", nand->name, offset);
 632
 633         cnt = nand->writesize;
 634         r = nand_read_skip_bad(nand, offset, &cnt, (u_char *) addr);
 635         if (r) {
 636                 puts("** Read error\n");
 637                 show_boot_progress (-56);
 638                 return 1;
 639         }
 640         show_boot_progress (56);
 641
 642         switch (genimg_get_format ((void *)addr)) {
 643         case IMAGE_FORMAT_LEGACY:
 644                 hdr = (image_header_t *)addr;
 645
 646                 show_boot_progress (57);
 647                 image_print_contents (hdr);
 648
 649                 cnt = image_get_image_size (hdr);
 650                 break;
 651 #if defined(CONFIG_FIT)
 652         case IMAGE_FORMAT_FIT:
 653                 fit_hdr = (const void *)addr;
 654                 puts ("Fit image detected...\n");
 655
 656                 cnt = fit_get_size (fit_hdr);
 657                 break;
 658 #endif
 659         default:
 660                 show_boot_progress (-57);
 661                 puts ("** Unknown image type\n");
 662                 return 1;
 663         }
 664         show_boot_progress (57);
 665
 666         r = nand_read_skip_bad(nand, offset, &cnt, (u_char *) addr);
 667         if (r) {
 668                 puts("** Read error\n");
 669                 show_boot_progress (-58);
 670                 return 1;
 671         }
 672         show_boot_progress (58);
 673
 674 #if defined(CONFIG_FIT)
 675         /* This cannot be done earlier, we need complete FIT image in RAM first */
 676         if (genimg_get_format ((void *)addr) == IMAGE_FORMAT_FIT) {
 677                 if (!fit_check_format (fit_hdr)) {
 678                         show_boot_progress (-150);
 679                         puts ("** Bad FIT image format\n");
 680                         return 1;
 681                 }
 682                 show_boot_progress (151);
 683                 fit_print_contents (fit_hdr);
 684         }
 685 #endif
 686
 687         /* Loading ok, update default load address */
 688
 689         load_addr = addr;
 690
 691         /* Check if we should attempt an auto-start */
 692         if (((ep = getenv("autostart")) != NULL) && (strcmp(ep, "yes") == 0)) {
 693                 char *local_args[2];
 694                 extern int do_bootm(cmd_tbl_t *, int, int, char *[]);
 695
 696                 local_args[0] = cmd;
 697                 local_args[1] = NULL;
 698
 699                 printf("Automatic boot of image at addr 0x%08lx ...\n", addr);
 700
 701                 do_bootm(cmdtp, 0, 1, local_args);
 702                 return 1;
 703         }
 704         return 0;
 705 }
 706
 707 int do_nandboot(cmd_tbl_t * cmdtp, int flag, int argc, char * const argv[])
 708 {
 709         char *boot_device = NULL;
 710         int idx;
 711         ulong addr, offset = 0;
 712 #if defined(CONFIG_CMD_MTDPARTS)
 713         struct mtd_device *dev;
 714         struct part_info *part;
 715         u8 pnum;
 716
 717         if (argc >= 2) {
 718                 char *p = (argc == 2) ? argv[1] : argv[2];
 719                 if (!(str2long(p, &addr)) && (mtdparts_init() == 0) &&
 720                     (find_dev_and_part(p, &dev, &pnum, &part) == 0)) {
 721                         if (dev->id->type != MTD_DEV_TYPE_NAND) {
 722                                 puts("Not a NAND device\n");
 723                                 return 1;
 724                         }
 725                         if (argc > 3)
 726                                 goto usage;
 727                         if (argc == 3)
 728                                 addr = simple_strtoul(argv[1], NULL, 16);
 729                         else
 730                                 addr = CONFIG_SYS_LOAD_ADDR;
 731                         return nand_load_image(cmdtp, &nand_info[dev->id->num],
 732                                                part->offset, addr, argv[0]);
 733                 }
 734         }
 735 #endif
 736
 737         show_boot_progress(52);
 738         switch (argc) {
 739         case 1:
 740                 addr = CONFIG_SYS_LOAD_ADDR;
 741                 boot_device = getenv("bootdevice");
 742                 break;
 743         case 2:
 744                 addr = simple_strtoul(argv[1], NULL, 16);
 745                 boot_device = getenv("bootdevice");
 746                 break;
 747         case 3:
 748                 addr = simple_strtoul(argv[1], NULL, 16);
 749                 boot_device = argv[2];
 750                 break;
 751         case 4:
 752                 addr = simple_strtoul(argv[1], NULL, 16);
 753                 boot_device = argv[2];
 754                 offset = simple_strtoul(argv[3], NULL, 16);
 755                 break;
 756         default:
 757 #if defined(CONFIG_CMD_MTDPARTS)
 758 usage:
 759 #endif
 760                 cmd_usage(cmdtp);
 761                 show_boot_progress(-53);
 762                 return 1;
 763         }
 764
 765         show_boot_progress(53);
 766         if (!boot_device) {
 767                 puts("\n** No boot device **\n");
 768                 show_boot_progress(-54);
 769                 return 1;
 770         }
 771         show_boot_progress(54);
 772
 773         idx = simple_strtoul(boot_device, NULL, 16);
 774
 775         if (idx < 0 || idx >= CONFIG_SYS_MAX_NAND_DEVICE || !nand_info[idx].name) {
 776                 printf("\n** Device %d not available\n", idx);
 777                 show_boot_progress(-55);
 778                 return 1;
 779         }
 780         show_boot_progress(55);
 781
 782         return nand_load_image(cmdtp, &nand_info[idx], offset, addr, argv[0]);
 783 }
 784
 785 U_BOOT_CMD(nboot, 4, 1, do_nandboot,
 786         "boot from NAND device",
 787         "[partition] | [[[loadAddr] dev] offset]"
 788 );