]> git.ipfire.org Git - people/ms/u-boot.git/blobdiff - common/cmd_nand.c
moving files from yaffs2/direct/ to yaffs2/ and deleting all symlinks
[people/ms/u-boot.git] / common / cmd_nand.c
index b088150f3bf43825ad7e62a46643d1cc8abcddd1..8b359e01779487eba0e01531387cd5e1371c06e2 100644 (file)
  *
  */
 #include <common.h>
+#include <linux/mtd/mtd.h>
 
-#if (CONFIG_COMMANDS & CFG_CMD_NAND)
+#if defined(CONFIG_CMD_NAND)
 
 #include <command.h>
 #include <watchdog.h>
 #include <malloc.h>
 #include <asm/byteorder.h>
-
-#ifdef CONFIG_SHOW_BOOT_PROGRESS
-# include <status_led.h>
-# define SHOW_BOOT_PROGRESS(arg)       show_boot_progress(arg)
-#else
-# define SHOW_BOOT_PROGRESS(arg)
-#endif
-
 #include <jffs2/jffs2.h>
 #include <nand.h>
 
-#if (CONFIG_COMMANDS & CFG_CMD_JFFS2) && defined(CONFIG_JFFS2_CMDLINE)
+#if defined(CONFIG_CMD_JFFS2) && defined(CONFIG_JFFS2_CMDLINE)
 
 /* parition handling routines */
 int mtdparts_init(void);
 int id_parse(const char *id, const char **ret_id, u8 *dev_type, u8 *dev_num);
 int find_dev_and_part(const char *id, struct mtd_device **dev,
-               u8 *part_num, struct part_info **part);
+                      u8 *part_num, struct part_info **part);
 #endif
 
-extern nand_info_t nand_info[];       /* info for NAND chips */
-
 static int nand_dump_oob(nand_info_t *nand, ulong off)
 {
        return 0;
@@ -57,32 +48,34 @@ static int nand_dump(nand_info_t *nand, ulong off)
        int i;
        u_char *buf, *p;
 
-       buf = malloc(nand->oobblock + nand->oobsize);
+       buf = malloc(nand->writesize + nand->oobsize);
        if (!buf) {
                puts("No memory for page buffer\n");
                return 1;
        }
-       off &= ~(nand->oobblock - 1);
-       i = nand_read_raw(nand, buf, off, nand->oobblock, nand->oobsize);
+       off &= ~(nand->writesize - 1);
+       size_t dummy;
+       loff_t addr = (loff_t) off;
+       i = nand->read(nand, addr, nand->writesize, &dummy, buf);
        if (i < 0) {
-               printf("Error (%d) reading page %08x\n", i, off);
+               printf("Error (%d) reading page %08lx\n", i, off);
                free(buf);
                return 1;
        }
-       printf("Page %08x dump:\n", off);
-       i = nand->oobblock >> 4; p = buf;
+       printf("Page %08lx dump:\n", off);
+       i = nand->writesize >> 4; p = buf;
        while (i--) {
-               printf( "\t%02x %02x %02x %02x %02x %02x %02x %02x"
-                       "  %02x %02x %02x %02x %02x %02x %02x %02x\n",
-                       p[0], p[1], p[2], p[3], p[4], p[5], p[6], p[7],
-                       p[8], p[9], p[10], p[11], p[12], p[13], p[14], p[15]);
+               printf("\t%02x %02x %02x %02x %02x %02x %02x %02x"
+                      "  %02x %02x %02x %02x %02x %02x %02x %02x\n",
+                      p[0], p[1], p[2], p[3], p[4], p[5], p[6], p[7],
+                      p[8], p[9], p[10], p[11], p[12], p[13], p[14], p[15]);
                p += 16;
        }
        puts("OOB:\n");
        i = nand->oobsize >> 3;
        while (i--) {
-               printf( "\t%02x %02x %02x %02x %02x %02x %02x %02x\n",
-                       p[0], p[1], p[2], p[3], p[4], p[5], p[6], p[7]);
+               printf("\t%02x %02x %02x %02x %02x %02x %02x %02x\n",
+                      p[0], p[1], p[2], p[3], p[4], p[5], p[6], p[7]);
                p += 8;
        }
        free(buf);
@@ -101,10 +94,10 @@ static inline int str2long(char *p, ulong *num)
 }
 
 static int
-arg_off_size(int argc, char *argv[], nand_info_t *nand, ulong *off, ulong *size)
+arg_off_size(int argc, char *argv[], nand_info_t *nand, ulong *off, size_t *size)
 {
        int idx = nand_curr_device;
-#if (CONFIG_COMMANDS & CFG_CMD_JFFS2) && defined(CONFIG_JFFS2_CMDLINE)
+#if defined(CONFIG_CMD_JFFS2) && defined(CONFIG_JFFS2_CMDLINE)
        struct mtd_device *dev;
        struct part_info *part;
        u8 pnum;
@@ -118,7 +111,7 @@ arg_off_size(int argc, char *argv[], nand_info_t *nand, ulong *off, ulong *size)
                        }
                        *off = part->offset;
                        if (argc >= 2) {
-                               if (!(str2long(argv[1], size))) {
+                               if (!(str2long(argv[1], (ulong *)size))) {
                                        printf("'%s' is not a number\n", argv[1]);
                                        return -1;
                                }
@@ -144,7 +137,7 @@ arg_off_size(int argc, char *argv[], nand_info_t *nand, ulong *off, ulong *size)
        }
 
        if (argc >= 2) {
-               if (!(str2long(argv[1], size))) {
+               if (!(str2long(argv[1], (ulong *)size))) {
                        printf("'%s' is not a number\n", argv[1]);
                        return -1;
                }
@@ -152,24 +145,29 @@ arg_off_size(int argc, char *argv[], nand_info_t *nand, ulong *off, ulong *size)
                *size = nand->size - *off;
        }
 
-#if (CONFIG_COMMANDS & CFG_CMD_JFFS2) && defined(CONFIG_JFFS2_CMDLINE)
+#if  defined(CONFIG_CMD_JFFS2) && defined(CONFIG_JFFS2_CMDLINE)
 out:
 #endif
        printf("device %d ", idx);
        if (*size == nand->size)
                puts("whole chip\n");
        else
-               printf("offset 0x%x, size 0x%x\n", *off, *size);
+               printf("offset 0x%lx, size 0x%x\n", *off, *size);
        return 0;
 }
 
 int do_nand(cmd_tbl_t * cmdtp, int flag, int argc, char *argv[])
 {
-       int i, dev, ret;
-       ulong addr, off, size;
+       int i, dev, ret = 0;
+       ulong addr, off;
+       size_t size;
        char *cmd, *s;
        nand_info_t *nand;
+#ifdef CFG_NAND_QUIET
+       int quiet = CFG_NAND_QUIET;
+#else
        int quiet = 0;
+#endif
        const char *quiet_str = getenv("quiet");
 
        /* at least two arguments please */
@@ -186,9 +184,9 @@ int do_nand(cmd_tbl_t * cmdtp, int flag, int argc, char *argv[])
                putc('\n');
                for (i = 0; i < CFG_MAX_NAND_DEVICE; i++) {
                        if (nand_info[i].name)
-                               printf("Device %d: %s, sector size %lu KiB\n",
-                                       i, nand_info[i].name,
-                                       nand_info[i].erasesize >> 10);
+                               printf("Device %d: %s, sector size %u KiB\n",
+                                      i, nand_info[i].name,
+                                      nand_info[i].erasesize >> 10);
                }
                return 0;
        }
@@ -201,7 +199,7 @@ int do_nand(cmd_tbl_t * cmdtp, int flag, int argc, char *argv[])
                                puts("\nno devices available\n");
                        else
                                printf("\nDevice %d: %s\n", nand_curr_device,
-                                       nand_info[nand_curr_device].name);
+                                      nand_info[nand_curr_device].name);
                        return 0;
                }
                dev = (int)simple_strtoul(argv[2], NULL, 10);
@@ -243,7 +241,7 @@ int do_nand(cmd_tbl_t * cmdtp, int flag, int argc, char *argv[])
                printf("\nDevice %d bad blocks:\n", nand_curr_device);
                for (off = 0; off < nand->size; off += nand->erasesize)
                        if (nand_block_isbad(nand, off))
-                               printf("  %08x\n", off);
+                               printf("  %08lx\n", off);
                return 0;
        }
 
@@ -333,24 +331,39 @@ int do_nand(cmd_tbl_t * cmdtp, int flag, int argc, char *argv[])
                                /* read */
                                nand_read_options_t opts;
                                memset(&opts, 0, sizeof(opts));
-                               opts.buffer     = (u_char*) addr;
-                               opts.length     = size;
-                               opts.offset     = off;
-                               opts.quiet      = quiet;
-                               ret = nand_read_opts(nand, &opts);
+                               opts.buffer = (u_char*) addr;
+                               opts.length = size;
+                               opts.offset = off;
+                               opts.quiet = quiet;
+/*
+ *  ! BROKEN !
+ *
+ *  TODO: Function must be implemented
+ *
+ *                             ret = nand_read_opts(nand, &opts);
+ */
                        } else {
                                /* write */
-                               nand_write_options_t opts;
+                               mtd_oob_ops_t opts;
                                memset(&opts, 0, sizeof(opts));
-                               opts.buffer     = (u_char*) addr;
-                               opts.length     = size;
-                               opts.offset     = off;
-                               /* opts.forcejffs2 = 1; */
-                               opts.pad        = 1;
-                               opts.blockalign = 1;
-                               opts.quiet      = quiet;
-                               ret = nand_write_opts(nand, &opts);
+                               opts.datbuf = (u_char*) addr;
+                               opts.len = size;
+                               opts.ooblen = 64;
+                               opts.mode = MTD_OOB_AUTO;
+                               ret = nand_write_opts(nand, off, &opts);
                        }
+               } else if (s != NULL && !strcmp(s, ".oob")) {
+                       /* out-of-band data */
+                       mtd_oob_ops_t ops = {
+                               .oobbuf = (u8 *)addr,
+                               .ooblen = size,
+                               .mode = MTD_OOB_RAW
+                       };
+
+                       if (read)
+                               ret = nand->read_oob(nand, off, &ops);
+                       else
+                               ret = nand->write_oob(nand, off, &ops);
                } else {
                        if (read)
                                ret = nand_read(nand, off, &size, (u_char *)addr);
@@ -392,7 +405,12 @@ int do_nand(cmd_tbl_t * cmdtp, int flag, int argc, char *argv[])
                        if (!strcmp("status", argv[2]))
                                status = 1;
                }
-
+/*
+ * ! BROKEN !
+ * 
+ * TODO: must be implemented and tested by someone with HW
+ */
+#if 0
                if (status) {
                        ulong block_start = 0;
                        ulong off;
@@ -403,28 +421,28 @@ int do_nand(cmd_tbl_t * cmdtp, int flag, int argc, char *argv[])
                        nand_chip->cmdfunc (nand, NAND_CMD_STATUS, -1, -1);
                        printf("device is %swrite protected\n",
                               (nand_chip->read_byte(nand) & 0x80 ?
-                               "NOT " : "" ) );
+                              "NOT " : ""));
 
-                       for (off = 0; off < nand->size; off += nand->oobblock) {
+                       for (off = 0; off < nand->size; off += nand->writesize) {
                                int s = nand_get_lock_status(nand, off);
 
                                /* print message only if status has changed
                                 * or at end of chip
                                 */
-                               if (off == nand->size - nand->oobblock
+                               if (off == nand->size - nand->writesize
                                    || (s != last_status && off != 0))  {
 
-                                       printf("%08x - %08x: %8d pages %s%s%s\n",
+                                       printf("%08lx - %08lx: %8d pages %s%s%s\n",
                                               block_start,
                                               off-1,
-                                              (off-block_start)/nand->oobblock,
+                                              (off-block_start)/nand->writesize,
                                               ((last_status & NAND_LOCK_STATUS_TIGHT) ? "TIGHT " : ""),
                                               ((last_status & NAND_LOCK_STATUS_LOCK) ? "LOCK " : ""),
                                               ((last_status & NAND_LOCK_STATUS_UNLOCK) ? "UNLOCK " : ""));
                                }
 
                                last_status = s;
-                      }
+                       }
                } else {
                        if (!nand_lock(nand, tight)) {
                                puts("NAND flash successfully locked\n");
@@ -433,6 +451,7 @@ int do_nand(cmd_tbl_t * cmdtp, int flag, int argc, char *argv[])
                                return 1;
                        }
                }
+#endif
                return 0;
        }
 
@@ -440,6 +459,12 @@ int do_nand(cmd_tbl_t * cmdtp, int flag, int argc, char *argv[])
                if (arg_off_size(argc - 2, argv + 2, nand, &off, &size) < 0)
                        return 1;
 
+/*
+ * ! BROKEN !
+ * 
+ * TODO: must be implemented and tested by someone with HW
+ */
+#if 0
                if (!nand_unlock(nand, off, size)) {
                        puts("NAND flash successfully unlocked\n");
                } else {
@@ -447,6 +472,7 @@ int do_nand(cmd_tbl_t * cmdtp, int flag, int argc, char *argv[])
                             "write and erase will probably fail\n");
                        return 1;
                }
+#endif
                return 0;
        }
 
@@ -456,61 +482,96 @@ usage:
 }
 
 U_BOOT_CMD(nand, 5, 1, do_nand,
-       "nand    - NAND sub-system\n",
-       "info                  - show available NAND devices\n"
-       "nand device [dev]     - show or set current device\n"
-       "nand read[.jffs2]     - addr off|partition size\n"
-       "nand write[.jffs2]    - addr off|partiton size - read/write `size' bytes starting\n"
-       "    at offset `off' to/from memory address `addr'\n"
-       "nand erase [clean] [off size] - erase `size' bytes from\n"
-       "    offset `off' (entire device if not specified)\n"
-       "nand bad - show bad blocks\n"
-       "nand dump[.oob] off - dump page\n"
-       "nand scrub - really clean NAND erasing bad blocks (UNSAFE)\n"
-       "nand markbad off - mark bad block at offset (UNSAFE)\n"
-       "nand biterr off - make a bit error at offset (UNSAFE)\n"
-       "nand lock [tight] [status] - bring nand to lock state or display locked pages\n"
-       "nand unlock [offset] [size] - unlock section\n");
+           "nand - NAND sub-system\n",
+           "info - show available NAND devices\n"
+           "nand device [dev] - show or set current device\n"
+           "nand read[.jffs2] - addr off|partition size\n"
+           "nand write[.jffs2] - addr off|partition size\n"
+           "    read/write 'size' bytes starting at offset 'off'\n"
+           "    to/from memory address 'addr'\n"
+           "nand erase [clean] [off size] - erase 'size' bytes from\n"
+           "    offset 'off' (entire device if not specified)\n"
+           "nand bad - show bad blocks\n"
+           "nand dump[.oob] off - dump page\n"
+           "nand scrub - really clean NAND erasing bad blocks (UNSAFE)\n"
+           "nand markbad off - mark bad block at offset (UNSAFE)\n"
+           "nand biterr off - make a bit error at offset (UNSAFE)\n"
+           "nand lock [tight] [status]\n"
+           "    bring nand to lock state or display locked pages\n"
+           "nand unlock [offset] [size] - unlock section\n");
 
 static int nand_load_image(cmd_tbl_t *cmdtp, nand_info_t *nand,
-                          ulong offset, ulong addr, char *cmd)
+                           ulong offset, ulong addr, char *cmd)
 {
        int r;
-       char *ep;
-       ulong cnt;
+       char *ep, *s;
+       size_t cnt;
        image_header_t *hdr;
+       int jffs2 = 0;
+#if defined(CONFIG_FIT)
+       const void *fit_hdr = NULL;
+#endif
+
+       s = strchr(cmd, '.');
+       if (s != NULL &&
+           (!strcmp(s, ".jffs2") || !strcmp(s, ".e") || !strcmp(s, ".i")))
+               jffs2 = 1;
 
        printf("\nLoading from %s, offset 0x%lx\n", nand->name, offset);
 
-       cnt = nand->oobblock;
+       cnt = nand->writesize;
        r = nand_read(nand, offset, &cnt, (u_char *) addr);
        if (r) {
                puts("** Read error\n");
-               SHOW_BOOT_PROGRESS(-56);
+               show_boot_progress (-56);
                return 1;
        }
-       SHOW_BOOT_PROGRESS(56);
+       show_boot_progress (56);
 
-       hdr = (image_header_t *) addr;
+       switch (genimg_get_format ((void *)addr)) {
+       case IMAGE_FORMAT_LEGACY:
+               hdr = (image_header_t *)addr;
 
-       if (ntohl(hdr->ih_magic) != IH_MAGIC) {
-               printf("\n** Bad Magic Number 0x%x **\n", hdr->ih_magic);
-               SHOW_BOOT_PROGRESS(-57);
-               return 1;
-       }
-       SHOW_BOOT_PROGRESS(57);
+               show_boot_progress (57);
+               image_print_contents (hdr);
 
-       print_image_hdr(hdr);
+               cnt = image_get_image_size (hdr);
+               break;
+#if defined(CONFIG_FIT)
+       case IMAGE_FORMAT_FIT:
+               fit_hdr = (const void *)addr;
+               puts ("Fit image detected...\n");
 
-       cnt = (ntohl(hdr->ih_size) + sizeof (image_header_t));
+               cnt = fit_get_size (fit_hdr);
+               break;
+#endif
+       default:
+               show_boot_progress (-57);
+               puts ("** Unknown image type\n");
+               return 1;
+       }
+       show_boot_progress (57);
 
        r = nand_read(nand, offset, &cnt, (u_char *) addr);
        if (r) {
                puts("** Read error\n");
-               SHOW_BOOT_PROGRESS(-58);
+               show_boot_progress (-58);
                return 1;
        }
-       SHOW_BOOT_PROGRESS(58);
+       show_boot_progress (58);
+
+#if defined(CONFIG_FIT)
+       /* This cannot be done earlier, we need complete FIT image in RAM first */
+       if (genimg_get_format ((void *)addr) == IMAGE_FORMAT_FIT) {
+               if (!fit_check_format (fit_hdr)) {
+                       show_boot_progress (-150);
+                       puts ("** Bad FIT image format\n");
+                       return 1;
+               }
+               show_boot_progress (151);
+               fit_print_contents (fit_hdr);
+       }
+#endif
 
        /* Loading ok, update default load address */
 
@@ -537,7 +598,7 @@ int do_nandboot(cmd_tbl_t * cmdtp, int flag, int argc, char *argv[])
        char *boot_device = NULL;
        int idx;
        ulong addr, offset = 0;
-#if (CONFIG_COMMANDS & CFG_CMD_JFFS2) && defined(CONFIG_JFFS2_CMDLINE)
+#if defined(CONFIG_CMD_JFFS2) && defined(CONFIG_JFFS2_CMDLINE)
        struct mtd_device *dev;
        struct part_info *part;
        u8 pnum;
@@ -553,16 +614,16 @@ int do_nandboot(cmd_tbl_t * cmdtp, int flag, int argc, char *argv[])
                        if (argc > 3)
                                goto usage;
                        if (argc == 3)
-                               addr = simple_strtoul(argv[2], NULL, 16);
+                               addr = simple_strtoul(argv[1], NULL, 16);
                        else
                                addr = CFG_LOAD_ADDR;
                        return nand_load_image(cmdtp, &nand_info[dev->id->num],
-                                              part->offset, addr, argv[0]);
+                                              part->offset, addr, argv[0]);
                }
        }
 #endif
 
-       SHOW_BOOT_PROGRESS(52);
+       show_boot_progress(52);
        switch (argc) {
        case 1:
                addr = CFG_LOAD_ADDR;
@@ -582,39 +643,39 @@ int do_nandboot(cmd_tbl_t * cmdtp, int flag, int argc, char *argv[])
                offset = simple_strtoul(argv[3], NULL, 16);
                break;
        default:
-#if (CONFIG_COMMANDS & CFG_CMD_JFFS2) && defined(CONFIG_JFFS2_CMDLINE)
+#if defined(CONFIG_CMD_JFFS2) && defined(CONFIG_JFFS2_CMDLINE)
 usage:
 #endif
                printf("Usage:\n%s\n", cmdtp->usage);
-               SHOW_BOOT_PROGRESS(-53);
+               show_boot_progress(-53);
                return 1;
        }
 
-       SHOW_BOOT_PROGRESS(53);
+       show_boot_progress(53);
        if (!boot_device) {
                puts("\n** No boot device **\n");
-               SHOW_BOOT_PROGRESS(-54);
+               show_boot_progress(-54);
                return 1;
        }
-       SHOW_BOOT_PROGRESS(54);
+       show_boot_progress(54);
 
        idx = simple_strtoul(boot_device, NULL, 16);
 
        if (idx < 0 || idx >= CFG_MAX_NAND_DEVICE || !nand_info[idx].name) {
                printf("\n** Device %d not available\n", idx);
-               SHOW_BOOT_PROGRESS(-55);
+               show_boot_progress(-55);
                return 1;
        }
-       SHOW_BOOT_PROGRESS(55);
+       show_boot_progress(55);
 
        return nand_load_image(cmdtp, &nand_info[idx], offset, addr, argv[0]);
 }
 
 U_BOOT_CMD(nboot, 4, 1, do_nandboot,
        "nboot   - boot from NAND device\n",
-       "[partition] | [[[loadAddr] dev] offset]\n");
+       "[.jffs2] [partition] | [[[loadAddr] dev] offset]\n");
 
-#endif                         /* (CONFIG_COMMANDS & CFG_CMD_NAND) */
+#endif
 
 #else /* CFG_NAND_LEGACY */
 /*
@@ -627,14 +688,14 @@ U_BOOT_CMD(nboot, 4, 1, do_nandboot,
 #include <asm/io.h>
 #include <watchdog.h>
 
-#ifdef CONFIG_SHOW_BOOT_PROGRESS
+#ifdef CONFIG_show_boot_progress
 # include <status_led.h>
-# define SHOW_BOOT_PROGRESS(arg)       show_boot_progress(arg)
+# define show_boot_progress(arg)       show_boot_progress(arg)
 #else
-# define SHOW_BOOT_PROGRESS(arg)
+# define show_boot_progress(arg)
 #endif
 
-#if (CONFIG_COMMANDS & CFG_CMD_NAND)
+#if defined(CONFIG_CMD_NAND)
 #include <linux/mtd/nand_legacy.h>
 #if 0
 #include <linux/mtd/nand_ids.h>
@@ -669,10 +730,10 @@ void archflashwp(void *archdata, int wp);
 #define CONFIG_MTD_NAND_ECC_JFFS2
 
 /* bits for nand_legacy_rw() `cmd'; or together as needed */
-#define NANDRW_READ    0x01
-#define NANDRW_WRITE   0x00
-#define NANDRW_JFFS2   0x02
-#define NANDRW_JFFS2_SKIP      0x04
+#define NANDRW_READ         0x01
+#define NANDRW_WRITE        0x00
+#define NANDRW_JFFS2       0x02
+#define NANDRW_JFFS2_SKIP   0x04
 
 /*
  * Imports from nand_legacy.c
@@ -783,22 +844,23 @@ int do_nand (cmd_tbl_t * cmdtp, int flag, int argc, char *argv[])
                if (strncmp (argv[1], "read", 4) == 0 ||
                    strncmp (argv[1], "write", 5) == 0) {
                        ulong addr = simple_strtoul (argv[2], NULL, 16);
-                       ulong off = simple_strtoul (argv[3], NULL, 16);
-                       ulong size = simple_strtoul (argv[4], NULL, 16);
+                       off_t off = simple_strtoul (argv[3], NULL, 16);
+                       size_t size = simple_strtoul (argv[4], NULL, 16);
                        int cmd = (strncmp (argv[1], "read", 4) == 0) ?
-                               NANDRW_READ : NANDRW_WRITE;
-                       int ret, total;
+                                 NANDRW_READ : NANDRW_WRITE;
+                       size_t total;
+                       int ret;
                        char *cmdtail = strchr (argv[1], '.');
 
                        if (cmdtail && !strncmp (cmdtail, ".oob", 2)) {
                                /* read out-of-band data */
                                if (cmd & NANDRW_READ) {
                                        ret = nand_read_oob (nand_dev_desc + curr_device,
-                                                            off, size, (size_t *) & total,
+                                                            off, size, &total,
                                                             (u_char *) addr);
                                } else {
                                        ret = nand_write_oob (nand_dev_desc + curr_device,
-                                                             off, size, (size_t *) & total,
+                                                             off, size, &total,
                                                              (u_char *) addr);
                                }
                                return ret;
@@ -828,14 +890,13 @@ int do_nand (cmd_tbl_t * cmdtp, int flag, int argc, char *argv[])
                                return 1;
                        }
 
-                       printf ("\nNAND %s: device %d offset %ld, size %ld ...\n",
+                       printf ("\nNAND %s: device %d offset %ld, size %lu ...\n",
                                (cmd & NANDRW_READ) ? "read" : "write",
-                               curr_device, off, size);
+                               curr_device, off, (ulong)size);
 
                        ret = nand_legacy_rw (nand_dev_desc + curr_device,
                                              cmd, off, size,
-                                             (size_t *) & total,
-                                             (u_char *) addr);
+                                             &total, (u_char *) addr);
 
                        printf (" %d bytes %s: %s\n", total,
                                (cmd & NANDRW_READ) ? "read" : "written",
@@ -894,7 +955,11 @@ int do_nandboot (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])
        ulong offset = 0;
        image_header_t *hdr;
        int rcode = 0;
-       SHOW_BOOT_PROGRESS(52);
+#if defined(CONFIG_FIT)
+       const void *fit_hdr = NULL;
+#endif
+
+       show_boot_progress (52);
        switch (argc) {
        case 1:
                addr = CFG_LOAD_ADDR;
@@ -915,63 +980,84 @@ int do_nandboot (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])
                break;
        default:
                printf ("Usage:\n%s\n", cmdtp->usage);
-               SHOW_BOOT_PROGRESS (-53);
+               show_boot_progress (-53);
                return 1;
        }
 
-       SHOW_BOOT_PROGRESS(53);
+       show_boot_progress (53);
        if (!boot_device) {
                puts ("\n** No boot device **\n");
-               SHOW_BOOT_PROGRESS (-54);
+               show_boot_progress (-54);
                return 1;
        }
-       SHOW_BOOT_PROGRESS(54);
+       show_boot_progress (54);
 
        dev = simple_strtoul(boot_device, &ep, 16);
 
        if ((dev >= CFG_MAX_NAND_DEVICE) ||
            (nand_dev_desc[dev].ChipID == NAND_ChipID_UNKNOWN)) {
                printf ("\n** Device %d not available\n", dev);
-               SHOW_BOOT_PROGRESS (-55);
+               show_boot_progress (-55);
                return 1;
        }
-       SHOW_BOOT_PROGRESS(55);
+       show_boot_progress (55);
 
        printf ("\nLoading from device %d: %s at 0x%lx (offset 0x%lx)\n",
-               dev, nand_dev_desc[dev].name, nand_dev_desc[dev].IO_ADDR,
-               offset);
+           dev, nand_dev_desc[dev].name, nand_dev_desc[dev].IO_ADDR,
+           offset);
 
        if (nand_legacy_rw (nand_dev_desc + dev, NANDRW_READ, offset,
-                       SECTORSIZE, NULL, (u_char *)addr)) {
+                           SECTORSIZE, NULL, (u_char *)addr)) {
                printf ("** Read error on %d\n", dev);
-               SHOW_BOOT_PROGRESS (-56);
+               show_boot_progress (-56);
                return 1;
        }
-       SHOW_BOOT_PROGRESS(56);
-
-       hdr = (image_header_t *)addr;
-
-       if (ntohl(hdr->ih_magic) == IH_MAGIC) {
+       show_boot_progress (56);
 
-               print_image_hdr (hdr);
+       switch (genimg_get_format ((void *)addr)) {
+       case IMAGE_FORMAT_LEGACY:
+               hdr = (image_header_t *)addr;
+               image_print_contents (hdr);
 
-               cnt = (ntohl(hdr->ih_size) + sizeof(image_header_t));
+               cnt = image_get_image_size (hdr);
                cnt -= SECTORSIZE;
-       } else {
-               printf ("\n** Bad Magic Number 0x%x **\n", ntohl(hdr->ih_magic));
-               SHOW_BOOT_PROGRESS (-57);
+               break;
+#if defined(CONFIG_FIT)
+       case IMAGE_FORMAT_FIT:
+               fit_hdr = (const void *)addr;
+               puts ("Fit image detected...\n");
+
+               cnt = fit_get_size (fit_hdr);
+               break;
+#endif
+       default:
+               show_boot_progress (-57);
+               puts ("** Unknown image type\n");
                return 1;
        }
-       SHOW_BOOT_PROGRESS(57);
+       show_boot_progress (57);
 
        if (nand_legacy_rw (nand_dev_desc + dev, NANDRW_READ,
-                       offset + SECTORSIZE, cnt, NULL,
-                       (u_char *)(addr+SECTORSIZE))) {
+                           offset + SECTORSIZE, cnt, NULL,
+                           (u_char *)(addr+SECTORSIZE))) {
                printf ("** Read error on %d\n", dev);
-               SHOW_BOOT_PROGRESS (-58);
+               show_boot_progress (-58);
                return 1;
        }
-       SHOW_BOOT_PROGRESS(58);
+       show_boot_progress (58);
+
+#if defined(CONFIG_FIT)
+       /* This cannot be done earlier, we need complete FIT image in RAM first */
+       if (genimg_get_format ((void *)addr) == IMAGE_FORMAT_FIT) {
+               if (!fit_check_format (fit_hdr)) {
+                       show_boot_progress (-150);
+                       puts ("** Bad FIT image format\n");
+                       return 1;
+               }
+               show_boot_progress (151);
+               fit_print_contents (fit_hdr);
+       }
+#endif
 
        /* Loading ok, update default load address */
 
@@ -999,6 +1085,6 @@ U_BOOT_CMD(
        "loadAddr dev\n"
 );
 
-#endif /* (CONFIG_COMMANDS & CFG_CMD_NAND) */
+#endif
 
 #endif /* CFG_NAND_LEGACY */