]> git.kernelconcepts.de Git - karo-tx-uboot.git/blobdiff - common/image.c
bootm: move lmb into the bootm_headers_t structure
[karo-tx-uboot.git] / common / image.c
index 3911b2f756e77c4a7b7a023bce7198743695fe2c..55c4ccec8bc0fa4b7422d72e50148b00322d901c 100644 (file)
@@ -23,8 +23,6 @@
  * MA 02111-1307 USA
  */
 
-#define DEBUG
-
 #ifndef USE_HOSTCC
 #include <common.h>
 #include <watchdog.h>
 #include <rtc.h>
 #endif
 
-#if defined(CONFIG_FIT)
+#include <image.h>
+
+#if defined(CONFIG_FIT) || defined (CONFIG_OF_LIBFDT)
 #include <fdt.h>
 #include <libfdt.h>
 #include <fdt_support.h>
 #endif
 
+#if defined(CONFIG_FIT)
+#include <u-boot/md5.h>
+#include <sha1.h>
+
+static int fit_check_ramdisk (const void *fit, int os_noffset,
+               uint8_t arch, int verify);
+#endif
+
 #ifdef CONFIG_CMD_BDI
 extern int do_bdinfo(cmd_tbl_t *cmdtp, int flag, int argc, char *argv[]);
 #endif
 
 DECLARE_GLOBAL_DATA_PTR;
 
-static image_header_t* image_get_ramdisk (cmd_tbl_t *cmdtp, int flag,
-               int argc, char *argv[],
-               ulong rd_addr, uint8_t arch, int verify);
+static image_header_t* image_get_ramdisk (ulong rd_addr, uint8_t arch,
+                                               int verify);
 #else
 #include "mkimage.h"
-#endif /* !USE_HOSTCC*/
-
+#include <u-boot/md5.h>
+#include <time.h>
 #include <image.h>
+#endif /* !USE_HOSTCC*/
 
 typedef struct table_entry {
        int     id;             /* as defined in image.h        */
@@ -84,6 +92,7 @@ static table_entry_t uimage_arch[] = {
        {       IH_ARCH_MIPS64,         "mips64",       "MIPS 64 Bit",  },
        {       IH_ARCH_NIOS,           "nios",         "NIOS",         },
        {       IH_ARCH_NIOS2,          "nios2",        "NIOS II",      },
+       {       IH_ARCH_PPC,            "powerpc",      "PowerPC",      },
        {       IH_ARCH_PPC,            "ppc",          "PowerPC",      },
        {       IH_ARCH_S390,           "s390",         "IBM S390",     },
        {       IH_ARCH_SH,             "sh",           "SuperH",       },
@@ -146,7 +155,8 @@ static table_entry_t uimage_comp[] = {
        {       -1,             "",             "",                     },
 };
 
-unsigned long crc32 (unsigned long, const unsigned char *, unsigned int);
+uint32_t crc32 (uint32_t, const unsigned char *, uint);
+uint32_t crc32_wd (uint32_t, const unsigned char *, uint, uint);
 static void genimg_print_size (uint32_t size);
 #if defined(CONFIG_TIMESTAMP) || defined(CONFIG_CMD_DATE) || defined(USE_HOSTCC)
 static void genimg_print_time (time_t timestamp);
@@ -174,39 +184,10 @@ int image_check_dcrc (image_header_t *hdr)
 {
        ulong data = image_get_data (hdr);
        ulong len = image_get_data_size (hdr);
-       ulong dcrc = crc32 (0, (unsigned char *)data, len);
-
-       return (dcrc == image_get_dcrc (hdr));
-}
-
-#ifndef USE_HOSTCC
-int image_check_dcrc_wd (image_header_t *hdr, ulong chunksz)
-{
-       ulong dcrc = 0;
-       ulong len = image_get_data_size (hdr);
-       ulong data = image_get_data (hdr);
-
-#if defined(CONFIG_HW_WATCHDOG) || defined(CONFIG_WATCHDOG)
-       ulong cdata = data;
-       ulong edata = cdata + len;
-
-       while (cdata < edata) {
-               ulong chunk = edata - cdata;
-
-               if (chunk > chunksz)
-                       chunk = chunksz;
-               dcrc = crc32 (dcrc, (unsigned char *)cdata, chunk);
-               cdata += chunk;
-
-               WATCHDOG_RESET ();
-       }
-#else
-       dcrc = crc32 (0, (unsigned char *)data, len);
-#endif
+       ulong dcrc = crc32_wd (0, (unsigned char *)data, len, CHUNKSZ_CRC32);
 
        return (dcrc == image_get_dcrc (hdr));
 }
-#endif /* !USE_HOSTCC */
 
 /**
  * image_multi_count - get component (sub-image) count
@@ -259,7 +240,7 @@ void image_multi_getimg (image_header_t *hdr, ulong idx,
 {
        int i;
        uint32_t *size;
-       ulong offset, tail, count, img_data;
+       ulong offset, count, img_data;
 
        /* get number of component */
        count = image_multi_count (hdr);
@@ -275,19 +256,15 @@ void image_multi_getimg (image_header_t *hdr, ulong idx,
        if (idx < count) {
                *len = uimage_to_cpu (size[idx]);
                offset = 0;
-               tail = 0;
 
                /* go over all indices preceding requested component idx */
                for (i = 0; i < idx; i++) {
-                       /* add up i-th component size */
-                       offset += uimage_to_cpu (size[i]);
-
-                       /* add up alignment for i-th component */
-                       tail += (4 - uimage_to_cpu (size[i]) % 4);
+                       /* add up i-th component size, rounding up to 4 bytes */
+                       offset += (uimage_to_cpu (size[i]) + 3) & ~3 ;
                }
 
                /* calculate idx-th component data address */
-               *data = img_data + offset + tail;
+               *data = img_data + offset;
        } else {
                *len = 0;
                *data = 0;
@@ -306,8 +283,28 @@ static void image_print_type (image_header_t *hdr)
        printf ("%s %s %s (%s)\n", arch, os, type, comp);
 }
 
-static void __image_print_contents (image_header_t *hdr, const char *p)
+/**
+ * image_print_contents - prints out the contents of the legacy format image
+ * @hdr: pointer to the legacy format image header
+ * @p: pointer to prefix string
+ *
+ * image_print_contents() formats a multi line legacy image contents description.
+ * The routine prints out all header fields followed by the size/offset data
+ * for MULTI/SCRIPT images.
+ *
+ * returns:
+ *     no returned results
+ */
+void image_print_contents (image_header_t *hdr)
 {
+       const char *p;
+
+#ifdef USE_HOSTCC
+       p = "";
+#else
+       p = "   ";
+#endif
+
        printf ("%sImage Name:   %.*s\n", p, IH_NMLEN, image_get_name (hdr));
 #if defined(CONFIG_TIMESTAMP) || defined(CONFIG_CMD_DATE) || defined(USE_HOSTCC)
        printf ("%sCreated:      ", p);
@@ -345,23 +342,10 @@ static void __image_print_contents (image_header_t *hdr, const char *p)
        }
 }
 
-inline void image_print_contents (image_header_t *hdr)
-{
-       __image_print_contents (hdr, "   ");
-}
-
-inline void image_print_contents_noindent (image_header_t *hdr)
-{
-       __image_print_contents (hdr, "");
-}
 
 #ifndef USE_HOSTCC
 /**
  * image_get_ramdisk - get and verify ramdisk image
- * @cmdtp: command table pointer
- * @flag: command flag
- * @argc: command argument count
- * @argv: command argument list
  * @rd_addr: ramdisk image start address
  * @arch: expected ramdisk architecture
  * @verify: checksum verification flag
@@ -378,14 +362,10 @@ inline void image_print_contents_noindent (image_header_t *hdr)
  *     pointer to a ramdisk image header, if image was found and valid
  *     otherwise, return NULL
  */
-static image_header_t* image_get_ramdisk (cmd_tbl_t *cmdtp, int flag,
-               int argc, char *argv[],
-               ulong rd_addr, uint8_t arch, int verify)
+static image_header_t* image_get_ramdisk (ulong rd_addr, uint8_t arch,
+                                               int verify)
 {
-       image_header_t *rd_hdr;
-
-       show_boot_progress (9);
-       rd_hdr = (image_header_t *)rd_addr;
+       image_header_t *rd_hdr = (image_header_t *)rd_addr;
 
        if (!image_check_magic (rd_hdr)) {
                puts ("Bad Magic Number\n");
@@ -404,7 +384,7 @@ static image_header_t* image_get_ramdisk (cmd_tbl_t *cmdtp, int flag,
 
        if (verify) {
                puts("   Verifying Checksum ... ");
-               if (!image_check_dcrc_wd (rd_hdr, CHUNKSZ)) {
+               if (!image_check_dcrc (rd_hdr)) {
                        puts ("Bad Data CRC\n");
                        show_boot_progress (-12);
                        return NULL;
@@ -431,15 +411,9 @@ static image_header_t* image_get_ramdisk (cmd_tbl_t *cmdtp, int flag,
 /* Shared dual-format routines */
 /*****************************************************************************/
 #ifndef USE_HOSTCC
-int getenv_verify (void)
-{
-       char *s = getenv ("verify");
-       return (s && (*s == 'n')) ? 0 : 1;
-}
-
-int getenv_autostart (void)
+int getenv_yesno (char *var)
 {
-       char *s = getenv ("autostart");
+       char *s = getenv (var);
        return (s && (*s == 'n')) ? 0 : 1;
 }
 
@@ -451,22 +425,33 @@ ulong getenv_bootm_low(void)
                return tmp;
        }
 
-#ifdef CFG_SDRAM_BASE
+#if defined(CFG_SDRAM_BASE)
        return CFG_SDRAM_BASE;
+#elif defined(CONFIG_ARM)
+       return gd->bd->bi_dram[0].start;
 #else
        return 0;
 #endif
 }
 
-ulong getenv_bootm_size(void)
+phys_size_t getenv_bootm_size(void)
 {
        char *s = getenv ("bootm_size");
        if (s) {
-               ulong tmp = simple_strtoul (s, NULL, 16);
+               phys_size_t tmp;
+#ifdef CFG_64BIT_STRTOUL
+               tmp = (phys_size_t)simple_strtoull (s, NULL, 16);
+#else
+               tmp = (phys_size_t)simple_strtoul (s, NULL, 16);
+#endif
                return tmp;
        }
 
+#if defined(CONFIG_ARM)
+       return gd->bd->bi_dram[0].size;
+#else
        return gd->bd->bi_memsize;
+#endif
 }
 
 void memmove_wd (void *to, void *from, size_t len, ulong chunksz)
@@ -701,7 +686,7 @@ ulong genimg_get_image (ulong img_addr)
                        break;
 #if defined(CONFIG_FIT)
                case IMAGE_FORMAT_FIT:
-                       d_size = fdt_totalsize((void *)ram_addr) - h_size;
+                       d_size = fit_get_size ((const void *)ram_addr) - h_size;
                        debug ("   FIT/FDT format image found at 0x%08lx, size 0x%08lx\n",
                                        ram_addr, d_size);
                        break;
@@ -725,10 +710,28 @@ ulong genimg_get_image (ulong img_addr)
        return ram_addr;
 }
 
+/**
+ * fit_has_config - check if there is a valid FIT configuration
+ * @images: pointer to the bootm command headers structure
+ *
+ * fit_has_config() checks if there is a FIT configuration in use
+ * (if FTI support is present).
+ *
+ * returns:
+ *     0, no FIT support or no configuration found
+ *     1, configuration found
+ */
+int genimg_has_config (bootm_headers_t *images)
+{
+#if defined(CONFIG_FIT)
+       if (images->fit_uname_cfg)
+               return 1;
+#endif
+       return 0;
+}
+
 /**
  * boot_get_ramdisk - main ramdisk handling routine
- * @cmdtp: command table pointer
- * @flag: command flag
  * @argc: command argument count
  * @argv: command argument list
  * @images: pointer to the bootm images structure
@@ -742,14 +745,15 @@ ulong genimg_get_image (ulong img_addr)
  *      - commandline provided address of decicated ramdisk image.
  *
  * returns:
+ *     0, if ramdisk image was found and valid, or skiped
  *     rd_start and rd_end are set to ramdisk start/end addresses if
  *     ramdisk image is found and valid
+ *
+ *     1, if ramdisk image is found but corrupted
  *     rd_start and rd_end are set to 0 if no ramdisk exists
- *     return 1 if ramdisk image is found but corrupted
  */
-int boot_get_ramdisk (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[],
-               bootm_headers_t *images, uint8_t arch,
-               ulong *rd_start, ulong *rd_end)
+int boot_get_ramdisk (int argc, char *argv[], bootm_headers_t *images,
+               uint8_t arch, ulong *rd_start, ulong *rd_end)
 {
        ulong rd_addr, rd_load;
        ulong rd_data, rd_len;
@@ -759,8 +763,15 @@ int boot_get_ramdisk (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[],
        const char      *fit_uname_config = NULL;
        const char      *fit_uname_ramdisk = NULL;
        ulong           default_addr;
+       int             rd_noffset;
+       int             cfg_noffset;
+       const void      *data;
+       size_t          size;
 #endif
 
+       *rd_start = 0;
+       *rd_end = 0;
+
        /*
         * Look for a '-' which indicates to ignore the
         * ramdisk argument
@@ -768,37 +779,65 @@ int boot_get_ramdisk (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[],
        if ((argc >= 3) && (strcmp(argv[2], "-") ==  0)) {
                debug ("## Skipping init Ramdisk\n");
                rd_len = rd_data = 0;
-       } else if (argc >= 3) {
+       } else if (argc >= 3 || genimg_has_config (images)) {
 #if defined(CONFIG_FIT)
-               /*
-                * If the init ramdisk comes from the FIT image and the FIT image
-                * address is omitted in the command line argument, try to use
-                * os FIT image address or default load address.
-                */
-               if (images->fit_uname_os)
-                       default_addr = (ulong)images->fit_hdr_os;
-               else
-                       default_addr = load_addr;
-
-               if (fit_parse_conf (argv[2], default_addr,
-                                       &rd_addr, &fit_uname_config)) {
-                       debug ("*  ramdisk: config '%s' from image at 0x%08lx\n",
-                                       fit_uname_config, rd_addr);
-               } else if (fit_parse_subimage (argv[2], default_addr,
-                                       &rd_addr, &fit_uname_ramdisk)) {
-                       debug ("*  ramdisk: subimage '%s' from image at 0x%08lx\n",
-                                       fit_uname_ramdisk, rd_addr);
-               } else
+               if (argc >= 3) {
+                       /*
+                        * If the init ramdisk comes from the FIT image and
+                        * the FIT image address is omitted in the command
+                        * line argument, try to use os FIT image address or
+                        * default load address.
+                        */
+                       if (images->fit_uname_os)
+                               default_addr = (ulong)images->fit_hdr_os;
+                       else
+                               default_addr = load_addr;
+
+                       if (fit_parse_conf (argv[2], default_addr,
+                                               &rd_addr, &fit_uname_config)) {
+                               debug ("*  ramdisk: config '%s' from image at 0x%08lx\n",
+                                               fit_uname_config, rd_addr);
+                       } else if (fit_parse_subimage (argv[2], default_addr,
+                                               &rd_addr, &fit_uname_ramdisk)) {
+                               debug ("*  ramdisk: subimage '%s' from image at 0x%08lx\n",
+                                               fit_uname_ramdisk, rd_addr);
+                       } else
 #endif
-               {
-                       rd_addr = simple_strtoul(argv[2], NULL, 16);
-                       debug ("*  ramdisk: cmdline image address = 0x%08lx\n",
-                                       rd_addr);
+                       {
+                               rd_addr = simple_strtoul(argv[2], NULL, 16);
+                               debug ("*  ramdisk: cmdline image address = 0x%08lx\n",
+                                               rd_addr);
+                       }
+#if defined(CONFIG_FIT)
+               } else {
+                       /* use FIT configuration provided in first bootm
+                        * command argument
+                        */
+                       rd_addr = (ulong)images->fit_hdr_os;
+                       fit_uname_config = images->fit_uname_cfg;
+                       debug ("*  ramdisk: using config '%s' from image at 0x%08lx\n",
+                                       fit_uname_config, rd_addr);
+
+                       /*
+                        * Check whether configuration has ramdisk defined,
+                        * if not, don't try to use it, quit silently.
+                        */
+                       fit_hdr = (void *)rd_addr;
+                       cfg_noffset = fit_conf_get_node (fit_hdr, fit_uname_config);
+                       if (cfg_noffset < 0) {
+                               debug ("*  ramdisk: no such config\n");
+                               return 1;
+                       }
+
+                       rd_noffset = fit_conf_get_ramdisk_node (fit_hdr, cfg_noffset);
+                       if (rd_noffset < 0) {
+                               debug ("*  ramdisk: no ramdisk in config\n");
+                               return 0;
+                       }
                }
+#endif
 
                /* copy from dataflash if needed */
-               printf ("## Loading init Ramdisk Image at %08lx ...\n",
-                               rd_addr);
                rd_addr = genimg_get_image (rd_addr);
 
                /*
@@ -808,17 +847,15 @@ int boot_get_ramdisk (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[],
                 */
                switch (genimg_get_format ((void *)rd_addr)) {
                case IMAGE_FORMAT_LEGACY:
+                       printf ("## Loading init Ramdisk from Legacy "
+                                       "Image at %08lx ...\n", rd_addr);
 
-                       debug ("*  ramdisk: legacy format image\n");
-
-                       rd_hdr = image_get_ramdisk (cmdtp, flag, argc, argv,
-                                               rd_addr, arch, images->verify);
+                       show_boot_progress (9);
+                       rd_hdr = image_get_ramdisk (rd_addr, arch,
+                                                       images->verify);
 
-                       if (rd_hdr == NULL) {
-                               *rd_start = 0;
-                               *rd_end = 0;
+                       if (rd_hdr == NULL)
                                return 1;
-                       }
 
                        rd_data = image_get_data (rd_hdr);
                        rd_len = image_get_data_size (rd_hdr);
@@ -827,14 +864,78 @@ int boot_get_ramdisk (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[],
 #if defined(CONFIG_FIT)
                case IMAGE_FORMAT_FIT:
                        fit_hdr = (void *)rd_addr;
-                       debug ("*  ramdisk: FIT format image\n");
-                       fit_unsupported_reset ("ramdisk");
-                       return 1;
+                       printf ("## Loading init Ramdisk from FIT "
+                                       "Image at %08lx ...\n", rd_addr);
+
+                       show_boot_progress (120);
+                       if (!fit_check_format (fit_hdr)) {
+                               puts ("Bad FIT ramdisk image format!\n");
+                               show_boot_progress (-120);
+                               return 1;
+                       }
+                       show_boot_progress (121);
+
+                       if (!fit_uname_ramdisk) {
+                               /*
+                                * no ramdisk image node unit name, try to get config
+                                * node first. If config unit node name is NULL
+                                * fit_conf_get_node() will try to find default config node
+                                */
+                               show_boot_progress (122);
+                               cfg_noffset = fit_conf_get_node (fit_hdr, fit_uname_config);
+                               if (cfg_noffset < 0) {
+                                       puts ("Could not find configuration node\n");
+                                       show_boot_progress (-122);
+                                       return 1;
+                               }
+                               fit_uname_config = fdt_get_name (fit_hdr, cfg_noffset, NULL);
+                               printf ("   Using '%s' configuration\n", fit_uname_config);
+
+                               rd_noffset = fit_conf_get_ramdisk_node (fit_hdr, cfg_noffset);
+                               fit_uname_ramdisk = fit_get_name (fit_hdr, rd_noffset, NULL);
+                       } else {
+                               /* get ramdisk component image node offset */
+                               show_boot_progress (123);
+                               rd_noffset = fit_image_get_node (fit_hdr, fit_uname_ramdisk);
+                       }
+                       if (rd_noffset < 0) {
+                               puts ("Could not find subimage node\n");
+                               show_boot_progress (-124);
+                               return 1;
+                       }
+
+                       printf ("   Trying '%s' ramdisk subimage\n", fit_uname_ramdisk);
+
+                       show_boot_progress (125);
+                       if (!fit_check_ramdisk (fit_hdr, rd_noffset, arch, images->verify))
+                               return 1;
+
+                       /* get ramdisk image data address and length */
+                       if (fit_image_get_data (fit_hdr, rd_noffset, &data, &size)) {
+                               puts ("Could not find ramdisk subimage data!\n");
+                               show_boot_progress (-127);
+                               return 1;
+                       }
+                       show_boot_progress (128);
+
+                       rd_data = (ulong)data;
+                       rd_len = size;
+
+                       if (fit_image_get_load (fit_hdr, rd_noffset, &rd_load)) {
+                               puts ("Can't get ramdisk subimage load address!\n");
+                               show_boot_progress (-129);
+                               return 1;
+                       }
+                       show_boot_progress (129);
+
+                       images->fit_hdr_rd = fit_hdr;
+                       images->fit_uname_rd = fit_uname_ramdisk;
+                       images->fit_noffset_rd = rd_noffset;
+                       break;
 #endif
                default:
-                       printf ("Wrong Image Format for %s command\n",
-                                       cmdtp->name);
-                       rd_data = rd_len = 0;
+                       puts ("Wrong Ramdisk Image Format\n");
+                       rd_data = rd_len = rd_load = 0;
                }
 
 #if defined(CONFIG_B2) || defined(CONFIG_EVB4510) || defined(CONFIG_ARMADILLO)
@@ -848,14 +949,14 @@ int boot_get_ramdisk (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[],
 #endif /* CONFIG_B2 || CONFIG_EVB4510 || CONFIG_ARMADILLO */
 
        } else if (images->legacy_hdr_valid &&
-                       image_check_type (images->legacy_hdr_os, IH_TYPE_MULTI)) {
+                       image_check_type (&images->legacy_hdr_os_copy, IH_TYPE_MULTI)) {
                /*
                 * Now check if we have a legacy mult-component image,
                 * get second entry data start address and len.
                 */
                show_boot_progress (13);
                printf ("## Loading init Ramdisk from multi component "
-                               "Image at %08lx ...\n",
+                               "Legacy Image at %08lx ...\n",
                                (ulong)images->legacy_hdr_os);
 
                image_multi_getimg (images->legacy_hdr_os, 1, &rd_data, &rd_len);
@@ -869,8 +970,6 @@ int boot_get_ramdisk (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[],
 
        if (!rd_data) {
                debug ("## No init Ramdisk\n");
-               *rd_start = 0;
-               *rd_end = 0;
        } else {
                *rd_start = rd_data;
                *rd_end = rd_data + rd_len;
@@ -881,7 +980,7 @@ int boot_get_ramdisk (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[],
        return 0;
 }
 
-#if defined(CONFIG_PPC) || defined(CONFIG_M68K)
+#if defined(CONFIG_PPC) || defined(CONFIG_M68K) || defined(CONFIG_SPARC)
 /**
  * boot_ramdisk_high - relocate init ramdisk
  * @lmb: pointer to lmb handle, will be used for memory mgmt
@@ -922,6 +1021,12 @@ int boot_ramdisk_high (struct lmb *lmb, ulong rd_data, ulong rd_len,
                initrd_high = ~0;
        }
 
+
+#ifdef CONFIG_LOGBUFFER
+       /* Prevent initrd from overwriting logbuffer */
+       lmb_reserve(lmb, logbuffer_base() - LOGBUFF_OVERHEAD, LOGBUFF_RESERVE);
+#endif
+
        debug ("## initrd_high = 0x%08lx, copy_to_ram = %d\n",
                        initrd_high, initrd_copy_to_ram);
 
@@ -933,9 +1038,9 @@ int boot_ramdisk_high (struct lmb *lmb, ulong rd_data, ulong rd_len,
                        lmb_reserve(lmb, rd_data, rd_len);
                } else {
                        if (initrd_high)
-                               *initrd_start = lmb_alloc_base (lmb, rd_len, 0x1000, initrd_high);
+                               *initrd_start = (ulong)lmb_alloc_base (lmb, rd_len, 0x1000, initrd_high);
                        else
-                               *initrd_start = lmb_alloc (lmb, rd_len, 0x1000);
+                               *initrd_start = (ulong)lmb_alloc (lmb, rd_len, 0x1000);
 
                        if (*initrd_start == 0) {
                                puts ("ramdisk - allocation error\n");
@@ -965,138 +1070,643 @@ error:
        return -1;
 }
 
+#ifdef CONFIG_OF_LIBFDT
+static void fdt_error (const char *msg)
+{
+       puts ("ERROR: ");
+       puts (msg);
+       puts (" - must RESET the board to recover.\n");
+}
+
+static image_header_t *image_get_fdt (ulong fdt_addr)
+{
+       image_header_t *fdt_hdr = (image_header_t *)fdt_addr;
+
+       image_print_contents (fdt_hdr);
+
+       puts ("   Verifying Checksum ... ");
+       if (!image_check_hcrc (fdt_hdr)) {
+               fdt_error ("fdt header checksum invalid");
+               return NULL;
+       }
+
+       if (!image_check_dcrc (fdt_hdr)) {
+               fdt_error ("fdt checksum invalid");
+               return NULL;
+       }
+       puts ("OK\n");
+
+       if (!image_check_type (fdt_hdr, IH_TYPE_FLATDT)) {
+               fdt_error ("uImage is not a fdt");
+               return NULL;
+       }
+       if (image_get_comp (fdt_hdr) != IH_COMP_NONE) {
+               fdt_error ("uImage is compressed");
+               return NULL;
+       }
+       if (fdt_check_header ((char *)image_get_data (fdt_hdr)) != 0) {
+               fdt_error ("uImage data is not a fdt");
+               return NULL;
+       }
+       return fdt_hdr;
+}
+
 /**
- * boot_get_cmdline - allocate and initialize kernel cmdline
- * @lmb: pointer to lmb handle, will be used for memory mgmt
- * @cmd_start: pointer to a ulong variable, will hold cmdline start
- * @cmd_end: pointer to a ulong variable, will hold cmdline end
- * @bootmap_base: ulong variable, holds offset in physical memory to
- * base of bootmap
+ * fit_check_fdt - verify FIT format FDT subimage
+ * @fit_hdr: pointer to the FIT  header
+ * fdt_noffset: FDT subimage node offset within FIT image
+ * @verify: data CRC verification flag
  *
- * boot_get_cmdline() allocates space for kernel command line below
- * BOOTMAPSZ + bootmap_base address. If "bootargs" U-boot environemnt
- * variable is present its contents is copied to allocated kernel
- * command line.
+ * fit_check_fdt() verifies integrity of the FDT subimage and from
+ * specified FIT image.
  *
  * returns:
- *      0 - success
- *     -1 - failure
+ *     1, on success
+ *     0, on failure
  */
-int boot_get_cmdline (struct lmb *lmb, ulong *cmd_start, ulong *cmd_end,
-                       ulong bootmap_base)
+#if defined(CONFIG_FIT)
+static int fit_check_fdt (const void *fit, int fdt_noffset, int verify)
 {
-       char *cmdline;
-       char *s;
-
-       cmdline = (char *)lmb_alloc_base(lmb, CFG_BARGSIZE, 0xf,
-                                        CFG_BOOTMAPSZ + bootmap_base);
-
-       if (cmdline == NULL)
-               return -1;
-
-       if ((s = getenv("bootargs")) == NULL)
-               s = "";
+       fit_image_print (fit, fdt_noffset, "   ");
 
-       strcpy(cmdline, s);
+       if (verify) {
+               puts ("   Verifying Hash Integrity ... ");
+               if (!fit_image_check_hashes (fit, fdt_noffset)) {
+                       fdt_error ("Bad Data Hash");
+                       return 0;
+               }
+               puts ("OK\n");
+       }
 
-       *cmd_start = (ulong) & cmdline[0];
-       *cmd_end = *cmd_start + strlen(cmdline);
+       if (!fit_image_check_type (fit, fdt_noffset, IH_TYPE_FLATDT)) {
+               fdt_error ("Not a FDT image");
+               return 0;
+       }
 
-       debug ("## cmdline at 0x%08lx ... 0x%08lx\n", *cmd_start, *cmd_end);
+       if (!fit_image_check_comp (fit, fdt_noffset, IH_COMP_NONE)) {
+               fdt_error ("FDT image is compressed");
+               return 0;
+       }
 
-       return 0;
+       return 1;
 }
+#endif /* CONFIG_FIT */
+
+#ifndef CFG_FDT_PAD
+#define CFG_FDT_PAD 0x3000
+#endif
 
 /**
- * boot_get_kbd - allocate and initialize kernel copy of board info
+ * boot_relocate_fdt - relocate flat device tree
  * @lmb: pointer to lmb handle, will be used for memory mgmt
- * @kbd: double pointer to board info data
- * @bootmap_base: ulong variable, holds offset in physical memory to
- * base of bootmap
+ * @bootmap_base: base address of the bootmap region
+ * @of_flat_tree: pointer to a char* variable, will hold fdt start address
+ * @of_size: pointer to a ulong variable, will hold fdt length
  *
- * boot_get_kbd() allocates space for kernel copy of board info data below
- * BOOTMAPSZ + bootmap_base address and kernel board info is initialized with
- * the current u-boot board info data.
+ * boot_relocate_fdt() determines if the of_flat_tree address is within
+ * the bootmap and if not relocates it into that region
+ *
+ * of_flat_tree and of_size are set to final (after relocation) values
  *
  * returns:
  *      0 - success
- *     -1 - failure
+ *      1 - failure
  */
-int boot_get_kbd (struct lmb *lmb, bd_t **kbd, ulong bootmap_base)
+int boot_relocate_fdt (struct lmb *lmb, ulong bootmap_base,
+               char **of_flat_tree, ulong *of_size)
 {
-       *kbd = (bd_t *)lmb_alloc_base(lmb, sizeof(bd_t), 0xf,
-                                     CFG_BOOTMAPSZ + bootmap_base);
-       if (*kbd == NULL)
-               return -1;
+       char    *fdt_blob = *of_flat_tree;
+       ulong   relocate = 0;
+       ulong   of_len = 0;
 
-       **kbd = *(gd->bd);
+       /* nothing to do */
+       if (*of_size == 0)
+               return 0;
 
-       debug ("## kernel board info at 0x%08lx\n", (ulong)*kbd);
+       if (fdt_check_header (fdt_blob) != 0) {
+               fdt_error ("image is not a fdt");
+               goto error;
+       }
 
-#if defined(DEBUG) && defined(CONFIG_CMD_BDI)
-       do_bdinfo(NULL, 0, 0, NULL);
+#ifndef CFG_NO_FLASH
+       /* move the blob if it is in flash (set relocate) */
+       if (addr2info ((ulong)fdt_blob) != NULL)
+               relocate = 1;
 #endif
 
-       return 0;
-}
-#endif /* CONFIG_PPC || CONFIG_M68K */
+       /*
+        * The blob needs to be inside the boot mapping.
+        */
+       if (fdt_blob < (char *)bootmap_base)
+               relocate = 1;
+
+       if ((fdt_blob + *of_size + CFG_FDT_PAD) >=
+                       ((char *)CFG_BOOTMAPSZ + bootmap_base))
+               relocate = 1;
+
+       /* move flattend device tree if needed */
+       if (relocate) {
+               int err;
+               ulong of_start = 0;
+
+               /* position on a 4K boundary before the alloc_current */
+               /* Pad the FDT by a specified amount */
+               of_len = *of_size + CFG_FDT_PAD;
+               of_start = (unsigned long)lmb_alloc_base(lmb, of_len, 0x1000,
+                               (CFG_BOOTMAPSZ + bootmap_base));
+
+               if (of_start == 0) {
+                       puts("device tree - allocation error\n");
+                       goto error;
+               }
 
-#if defined(CONFIG_FIT)
-/*****************************************************************************/
-/* New uImage format routines */
-/*****************************************************************************/
-static int fit_parse_spec (const char *spec, char sepc, ulong addr_curr,
-               ulong *addr, const char **name)
-{
-       const char *sep;
+               debug ("## device tree at 0x%08lX ... 0x%08lX (len=%ld=0x%lX)\n",
+                       (ulong)fdt_blob, (ulong)fdt_blob + *of_size - 1,
+                       of_len, of_len);
 
-       *addr = addr_curr;
-       *name = NULL;
+               printf ("   Loading Device Tree to %08lx, end %08lx ... ",
+                       of_start, of_start + of_len - 1);
 
-       sep = strchr (spec, sepc);
-       if (sep) {
-               if (sep - spec > 0)
-                       *addr = simple_strtoul (spec, NULL, 16);
+               err = fdt_open_into (fdt_blob, (void *)of_start, of_len);
+               if (err != 0) {
+                       fdt_error ("fdt move failed");
+                       goto error;
+               }
+               puts ("OK\n");
 
-               *name = sep + 1;
-               return 1;
+               *of_flat_tree = (char *)of_start;
+               *of_size = of_len;
+       } else {
+               *of_flat_tree = fdt_blob;
+               of_len = (CFG_BOOTMAPSZ + bootmap_base) - (ulong)fdt_blob;
+               lmb_reserve(lmb, (ulong)fdt_blob, of_len);
+               fdt_set_totalsize(*of_flat_tree, of_len);
+
+               *of_size = of_len;
        }
 
+       set_working_fdt_addr(*of_flat_tree);
        return 0;
+
+error:
+       return 1;
 }
 
 /**
- * fit_parse_conf - parse FIT configuration spec
- * @spec: input string, containing configuration spec
- * @add_curr: current image address (to be used as a possible default)
- * @addr: pointer to a ulong variable, will hold FIT image address of a given
- * configuration
- * @conf_name double pointer to a char, will hold pointer to a configuration
- * unit name
- *
- * fit_parse_conf() expects configuration spec in the for of [<addr>]#<conf>,
- * where <addr> is a FIT image address that contains configuration
- * with a <conf> unit name.
+ * boot_get_fdt - main fdt handling routine
+ * @argc: command argument count
+ * @argv: command argument list
+ * @images: pointer to the bootm images structure
+ * @of_flat_tree: pointer to a char* variable, will hold fdt start address
+ * @of_size: pointer to a ulong variable, will hold fdt length
  *
- * Address part is optional, and if omitted default add_curr will
- * be used instead.
+ * boot_get_fdt() is responsible for finding a valid flat device tree image.
+ * Curently supported are the following ramdisk sources:
+ *      - multicomponent kernel/ramdisk image,
+ *      - commandline provided address of decicated ramdisk image.
  *
  * returns:
- *     1 if spec is a valid configuration string,
- *     addr and conf_name are set accordingly
- *     0 otherwise
+ *     0, if fdt image was found and valid, or skipped
+ *     of_flat_tree and of_size are set to fdt start address and length if
+ *     fdt image is found and valid
+ *
+ *     1, if fdt image is found but corrupted
+ *     of_flat_tree and of_size are set to 0 if no fdt exists
  */
-inline int fit_parse_conf (const char *spec, ulong addr_curr,
-               ulong *addr, const char **conf_name)
+int boot_get_fdt (int flag, int argc, char *argv[], bootm_headers_t *images,
+               char **of_flat_tree, ulong *of_size)
 {
-       return fit_parse_spec (spec, '#', addr_curr, addr, conf_name);
-}
+       ulong           fdt_addr;
+       image_header_t  *fdt_hdr;
+       char            *fdt_blob = NULL;
+       ulong           image_start, image_end;
+       ulong           load_start, load_end;
+#if defined(CONFIG_FIT)
+       void            *fit_hdr;
+       const char      *fit_uname_config = NULL;
+       const char      *fit_uname_fdt = NULL;
+       ulong           default_addr;
+       int             cfg_noffset;
+       int             fdt_noffset;
+       const void      *data;
+       size_t          size;
+#endif
 
-/**
- * fit_parse_subimage - parse FIT subimage spec
- * @spec: input string, containing subimage spec
- * @add_curr: current image address (to be used as a possible default)
- * @addr: pointer to a ulong variable, will hold FIT image address of a given
+       *of_flat_tree = NULL;
+       *of_size = 0;
+
+       if (argc > 3 || genimg_has_config (images)) {
+#if defined(CONFIG_FIT)
+               if (argc > 3) {
+                       /*
+                        * If the FDT blob comes from the FIT image and the
+                        * FIT image address is omitted in the command line
+                        * argument, try to use ramdisk or os FIT image
+                        * address or default load address.
+                        */
+                       if (images->fit_uname_rd)
+                               default_addr = (ulong)images->fit_hdr_rd;
+                       else if (images->fit_uname_os)
+                               default_addr = (ulong)images->fit_hdr_os;
+                       else
+                               default_addr = load_addr;
+
+                       if (fit_parse_conf (argv[3], default_addr,
+                                               &fdt_addr, &fit_uname_config)) {
+                               debug ("*  fdt: config '%s' from image at 0x%08lx\n",
+                                               fit_uname_config, fdt_addr);
+                       } else if (fit_parse_subimage (argv[3], default_addr,
+                                               &fdt_addr, &fit_uname_fdt)) {
+                               debug ("*  fdt: subimage '%s' from image at 0x%08lx\n",
+                                               fit_uname_fdt, fdt_addr);
+                       } else
+#endif
+                       {
+                               fdt_addr = simple_strtoul(argv[3], NULL, 16);
+                               debug ("*  fdt: cmdline image address = 0x%08lx\n",
+                                               fdt_addr);
+                       }
+#if defined(CONFIG_FIT)
+               } else {
+                       /* use FIT configuration provided in first bootm
+                        * command argument
+                        */
+                       fdt_addr = (ulong)images->fit_hdr_os;
+                       fit_uname_config = images->fit_uname_cfg;
+                       debug ("*  fdt: using config '%s' from image at 0x%08lx\n",
+                                       fit_uname_config, fdt_addr);
+
+                       /*
+                        * Check whether configuration has FDT blob defined,
+                        * if not quit silently.
+                        */
+                       fit_hdr = (void *)fdt_addr;
+                       cfg_noffset = fit_conf_get_node (fit_hdr,
+                                       fit_uname_config);
+                       if (cfg_noffset < 0) {
+                               debug ("*  fdt: no such config\n");
+                               return 0;
+                       }
+
+                       fdt_noffset = fit_conf_get_fdt_node (fit_hdr,
+                                       cfg_noffset);
+                       if (fdt_noffset < 0) {
+                               debug ("*  fdt: no fdt in config\n");
+                               return 0;
+                       }
+               }
+#endif
+
+               debug ("## Checking for 'FDT'/'FDT Image' at %08lx\n",
+                               fdt_addr);
+
+               /* copy from dataflash if needed */
+               fdt_addr = genimg_get_image (fdt_addr);
+
+               /*
+                * Check if there is an FDT image at the
+                * address provided in the second bootm argument
+                * check image type, for FIT images get a FIT node.
+                */
+               switch (genimg_get_format ((void *)fdt_addr)) {
+               case IMAGE_FORMAT_LEGACY:
+                       /* verify fdt_addr points to a valid image header */
+                       printf ("## Flattened Device Tree from Legacy Image at %08lx\n",
+                                       fdt_addr);
+                       fdt_hdr = image_get_fdt (fdt_addr);
+                       if (!fdt_hdr)
+                               goto error;
+
+                       /*
+                        * move image data to the load address,
+                        * make sure we don't overwrite initial image
+                        */
+                       image_start = (ulong)fdt_hdr;
+                       image_end = image_get_image_end (fdt_hdr);
+
+                       load_start = image_get_load (fdt_hdr);
+                       load_end = load_start + image_get_data_size (fdt_hdr);
+
+                       if ((load_start < image_end) && (load_end > image_start)) {
+                               fdt_error ("fdt overwritten");
+                               goto error;
+                       }
+
+                       debug ("   Loading FDT from 0x%08lx to 0x%08lx\n",
+                                       image_get_data (fdt_hdr), load_start);
+
+                       memmove ((void *)load_start,
+                                       (void *)image_get_data (fdt_hdr),
+                                       image_get_data_size (fdt_hdr));
+
+                       fdt_blob = (char *)load_start;
+                       break;
+               case IMAGE_FORMAT_FIT:
+                       /*
+                        * This case will catch both: new uImage format
+                        * (libfdt based) and raw FDT blob (also libfdt
+                        * based).
+                        */
+#if defined(CONFIG_FIT)
+                       /* check FDT blob vs FIT blob */
+                       if (fit_check_format ((const void *)fdt_addr)) {
+                               /*
+                                * FIT image
+                                */
+                               fit_hdr = (void *)fdt_addr;
+                               printf ("## Flattened Device Tree from FIT Image at %08lx\n",
+                                               fdt_addr);
+
+                               if (!fit_uname_fdt) {
+                                       /*
+                                        * no FDT blob image node unit name,
+                                        * try to get config node first. If
+                                        * config unit node name is NULL
+                                        * fit_conf_get_node() will try to
+                                        * find default config node
+                                        */
+                                       cfg_noffset = fit_conf_get_node (fit_hdr,
+                                                       fit_uname_config);
+
+                                       if (cfg_noffset < 0) {
+                                               fdt_error ("Could not find configuration node\n");
+                                               goto error;
+                                       }
+
+                                       fit_uname_config = fdt_get_name (fit_hdr,
+                                                       cfg_noffset, NULL);
+                                       printf ("   Using '%s' configuration\n",
+                                                       fit_uname_config);
+
+                                       fdt_noffset = fit_conf_get_fdt_node (fit_hdr,
+                                                       cfg_noffset);
+                                       fit_uname_fdt = fit_get_name (fit_hdr,
+                                                       fdt_noffset, NULL);
+                               } else {
+                                       /* get FDT component image node offset */
+                                       fdt_noffset = fit_image_get_node (fit_hdr,
+                                                       fit_uname_fdt);
+                               }
+                               if (fdt_noffset < 0) {
+                                       fdt_error ("Could not find subimage node\n");
+                                       goto error;
+                               }
+
+                               printf ("   Trying '%s' FDT blob subimage\n",
+                                               fit_uname_fdt);
+
+                               if (!fit_check_fdt (fit_hdr, fdt_noffset,
+                                                       images->verify))
+                                       goto error;
+
+                               /* get ramdisk image data address and length */
+                               if (fit_image_get_data (fit_hdr, fdt_noffset,
+                                                       &data, &size)) {
+                                       fdt_error ("Could not find FDT subimage data");
+                                       goto error;
+                               }
+
+                               /* verift that image data is a proper FDT blob */
+                               if (fdt_check_header ((char *)data) != 0) {
+                                       fdt_error ("Subimage data is not a FTD");
+                                       goto error;
+                               }
+
+                               /*
+                                * move image data to the load address,
+                                * make sure we don't overwrite initial image
+                                */
+                               image_start = (ulong)fit_hdr;
+                               image_end = fit_get_end (fit_hdr);
+
+                               if (fit_image_get_load (fit_hdr, fdt_noffset,
+                                                       &load_start) == 0) {
+                                       load_end = load_start + size;
+
+                                       if ((load_start < image_end) &&
+                                                       (load_end > image_start)) {
+                                               fdt_error ("FDT overwritten");
+                                               goto error;
+                                       }
+
+                                       printf ("   Loading FDT from 0x%08lx to 0x%08lx\n",
+                                                       (ulong)data, load_start);
+
+                                       memmove ((void *)load_start,
+                                                       (void *)data, size);
+
+                                       fdt_blob = (char *)load_start;
+                               } else {
+                                       fdt_blob = (char *)data;
+                               }
+
+                               images->fit_hdr_fdt = fit_hdr;
+                               images->fit_uname_fdt = fit_uname_fdt;
+                               images->fit_noffset_fdt = fdt_noffset;
+                               break;
+                       } else
+#endif
+                       {
+                               /*
+                                * FDT blob
+                                */
+                               fdt_blob = (char *)fdt_addr;
+                               debug ("*  fdt: raw FDT blob\n");
+                               printf ("## Flattened Device Tree blob at %08lx\n", (long)fdt_blob);
+                       }
+                       break;
+               default:
+                       fdt_error ("Did not find a cmdline Flattened Device Tree");
+                       goto error;
+               }
+
+               printf ("   Booting using the fdt blob at 0x%x\n", (int)fdt_blob);
+
+       } else if (images->legacy_hdr_valid &&
+                       image_check_type (&images->legacy_hdr_os_copy, IH_TYPE_MULTI)) {
+
+               ulong fdt_data, fdt_len;
+
+               /*
+                * Now check if we have a legacy multi-component image,
+                * get second entry data start address and len.
+                */
+               printf ("## Flattened Device Tree from multi "
+                       "component Image at %08lX\n",
+                       (ulong)images->legacy_hdr_os);
+
+               image_multi_getimg (images->legacy_hdr_os, 2, &fdt_data, &fdt_len);
+               if (fdt_len) {
+
+                       fdt_blob = (char *)fdt_data;
+                       printf ("   Booting using the fdt at 0x%x\n", (int)fdt_blob);
+
+                       if (fdt_check_header (fdt_blob) != 0) {
+                               fdt_error ("image is not a fdt");
+                               goto error;
+                       }
+
+                       if (be32_to_cpu (fdt_totalsize (fdt_blob)) != fdt_len) {
+                               fdt_error ("fdt size != image size");
+                               goto error;
+                       }
+               } else {
+                       debug ("## No Flattened Device Tree\n");
+                       return 0;
+               }
+       } else {
+               debug ("## No Flattened Device Tree\n");
+               return 0;
+       }
+
+       *of_flat_tree = fdt_blob;
+       *of_size = be32_to_cpu (fdt_totalsize (fdt_blob));
+       debug ("   of_flat_tree at 0x%08lx size 0x%08lx\n",
+                       *of_flat_tree, *of_size);
+
+       return 0;
+
+error:
+       *of_flat_tree = 0;
+       *of_size = 0;
+       return 1;
+}
+#endif /* CONFIG_OF_LIBFDT */
+
+/**
+ * boot_get_cmdline - allocate and initialize kernel cmdline
+ * @lmb: pointer to lmb handle, will be used for memory mgmt
+ * @cmd_start: pointer to a ulong variable, will hold cmdline start
+ * @cmd_end: pointer to a ulong variable, will hold cmdline end
+ * @bootmap_base: ulong variable, holds offset in physical memory to
+ * base of bootmap
+ *
+ * boot_get_cmdline() allocates space for kernel command line below
+ * BOOTMAPSZ + bootmap_base address. If "bootargs" U-boot environemnt
+ * variable is present its contents is copied to allocated kernel
+ * command line.
+ *
+ * returns:
+ *      0 - success
+ *     -1 - failure
+ */
+int boot_get_cmdline (struct lmb *lmb, ulong *cmd_start, ulong *cmd_end,
+                       ulong bootmap_base)
+{
+       char *cmdline;
+       char *s;
+
+       cmdline = (char *)(ulong)lmb_alloc_base(lmb, CFG_BARGSIZE, 0xf,
+                                        CFG_BOOTMAPSZ + bootmap_base);
+
+       if (cmdline == NULL)
+               return -1;
+
+       if ((s = getenv("bootargs")) == NULL)
+               s = "";
+
+       strcpy(cmdline, s);
+
+       *cmd_start = (ulong) & cmdline[0];
+       *cmd_end = *cmd_start + strlen(cmdline);
+
+       debug ("## cmdline at 0x%08lx ... 0x%08lx\n", *cmd_start, *cmd_end);
+
+       return 0;
+}
+
+/**
+ * boot_get_kbd - allocate and initialize kernel copy of board info
+ * @lmb: pointer to lmb handle, will be used for memory mgmt
+ * @kbd: double pointer to board info data
+ * @bootmap_base: ulong variable, holds offset in physical memory to
+ * base of bootmap
+ *
+ * boot_get_kbd() allocates space for kernel copy of board info data below
+ * BOOTMAPSZ + bootmap_base address and kernel board info is initialized with
+ * the current u-boot board info data.
+ *
+ * returns:
+ *      0 - success
+ *     -1 - failure
+ */
+int boot_get_kbd (struct lmb *lmb, bd_t **kbd, ulong bootmap_base)
+{
+       *kbd = (bd_t *)(ulong)lmb_alloc_base(lmb, sizeof(bd_t), 0xf,
+                                     CFG_BOOTMAPSZ + bootmap_base);
+       if (*kbd == NULL)
+               return -1;
+
+       **kbd = *(gd->bd);
+
+       debug ("## kernel board info at 0x%08lx\n", (ulong)*kbd);
+
+#if defined(DEBUG) && defined(CONFIG_CMD_BDI)
+       do_bdinfo(NULL, 0, 0, NULL);
+#endif
+
+       return 0;
+}
+#endif /* CONFIG_PPC || CONFIG_M68K */
+#endif /* !USE_HOSTCC */
+
+#if defined(CONFIG_FIT)
+/*****************************************************************************/
+/* New uImage format routines */
+/*****************************************************************************/
+#ifndef USE_HOSTCC
+static int fit_parse_spec (const char *spec, char sepc, ulong addr_curr,
+               ulong *addr, const char **name)
+{
+       const char *sep;
+
+       *addr = addr_curr;
+       *name = NULL;
+
+       sep = strchr (spec, sepc);
+       if (sep) {
+               if (sep - spec > 0)
+                       *addr = simple_strtoul (spec, NULL, 16);
+
+               *name = sep + 1;
+               return 1;
+       }
+
+       return 0;
+}
+
+/**
+ * fit_parse_conf - parse FIT configuration spec
+ * @spec: input string, containing configuration spec
+ * @add_curr: current image address (to be used as a possible default)
+ * @addr: pointer to a ulong variable, will hold FIT image address of a given
+ * configuration
+ * @conf_name double pointer to a char, will hold pointer to a configuration
+ * unit name
+ *
+ * fit_parse_conf() expects configuration spec in the for of [<addr>]#<conf>,
+ * where <addr> is a FIT image address that contains configuration
+ * with a <conf> unit name.
+ *
+ * Address part is optional, and if omitted default add_curr will
+ * be used instead.
+ *
+ * returns:
+ *     1 if spec is a valid configuration string,
+ *     addr and conf_name are set accordingly
+ *     0 otherwise
+ */
+inline int fit_parse_conf (const char *spec, ulong addr_curr,
+               ulong *addr, const char **conf_name)
+{
+       return fit_parse_spec (spec, '#', addr_curr, addr, conf_name);
+}
+
+/**
+ * fit_parse_subimage - parse FIT subimage spec
+ * @spec: input string, containing subimage spec
+ * @add_curr: current image address (to be used as a possible default)
+ * @addr: pointer to a ulong variable, will hold FIT image address of a given
  * subimage
  * @image_name: double pointer to a char, will hold pointer to a subimage name
  *
@@ -1117,6 +1727,1288 @@ inline int fit_parse_subimage (const char *spec, ulong addr_curr,
 {
        return fit_parse_spec (spec, ':', addr_curr, addr, image_name);
 }
-
-#endif /* CONFIG_FIT */
 #endif /* !USE_HOSTCC */
+
+static void fit_get_debug (const void *fit, int noffset,
+               char *prop_name, int err)
+{
+       debug ("Can't get '%s' property from FIT 0x%08lx, "
+               "node: offset %d, name %s (%s)\n",
+               prop_name, (ulong)fit, noffset,
+               fit_get_name (fit, noffset, NULL),
+               fdt_strerror (err));
+}
+
+/**
+ * fit_print_contents - prints out the contents of the FIT format image
+ * @fit: pointer to the FIT format image header
+ * @p: pointer to prefix string
+ *
+ * fit_print_contents() formats a multi line FIT image contents description.
+ * The routine prints out FIT image properties (root node level) follwed by
+ * the details of each component image.
+ *
+ * returns:
+ *     no returned results
+ */
+void fit_print_contents (const void *fit)
+{
+       char *desc;
+       char *uname;
+       int images_noffset;
+       int confs_noffset;
+       int noffset;
+       int ndepth;
+       int count = 0;
+       int ret;
+       const char *p;
+#if defined(CONFIG_TIMESTAMP) || defined(CONFIG_CMD_DATE) || defined(USE_HOSTCC)
+       time_t timestamp;
+#endif
+
+#ifdef USE_HOSTCC
+       p = "";
+#else
+       p = "   ";
+#endif
+
+       /* Root node properties */
+       ret = fit_get_desc (fit, 0, &desc);
+       printf ("%sFIT description: ", p);
+       if (ret)
+               printf ("unavailable\n");
+       else
+               printf ("%s\n", desc);
+
+#if defined(CONFIG_TIMESTAMP) || defined(CONFIG_CMD_DATE) || defined(USE_HOSTCC)
+       ret = fit_get_timestamp (fit, 0, &timestamp);
+       printf ("%sCreated:         ", p);
+       if (ret)
+               printf ("unavailable\n");
+       else
+               genimg_print_time (timestamp);
+#endif
+
+       /* Find images parent node offset */
+       images_noffset = fdt_path_offset (fit, FIT_IMAGES_PATH);
+       if (images_noffset < 0) {
+               printf ("Can't find images parent node '%s' (%s)\n",
+                       FIT_IMAGES_PATH, fdt_strerror (images_noffset));
+               return;
+       }
+
+       /* Process its subnodes, print out component images details */
+       for (ndepth = 0, count = 0, noffset = fdt_next_node (fit, images_noffset, &ndepth);
+            (noffset >= 0) && (ndepth > 0);
+            noffset = fdt_next_node (fit, noffset, &ndepth)) {
+               if (ndepth == 1) {
+                       /*
+                        * Direct child node of the images parent node,
+                        * i.e. component image node.
+                        */
+                       printf ("%s Image %u (%s)\n", p, count++,
+                                       fit_get_name(fit, noffset, NULL));
+
+                       fit_image_print (fit, noffset, p);
+               }
+       }
+
+       /* Find configurations parent node offset */
+       confs_noffset = fdt_path_offset (fit, FIT_CONFS_PATH);
+       if (confs_noffset < 0) {
+               debug ("Can't get configurations parent node '%s' (%s)\n",
+                       FIT_CONFS_PATH, fdt_strerror (confs_noffset));
+               return;
+       }
+
+       /* get default configuration unit name from default property */
+       uname = (char *)fdt_getprop (fit, noffset, FIT_DEFAULT_PROP, NULL);
+       if (uname)
+               printf ("%s Default Configuration: '%s'\n", p, uname);
+
+       /* Process its subnodes, print out configurations details */
+       for (ndepth = 0, count = 0, noffset = fdt_next_node (fit, confs_noffset, &ndepth);
+            (noffset >= 0) && (ndepth > 0);
+            noffset = fdt_next_node (fit, noffset, &ndepth)) {
+               if (ndepth == 1) {
+                       /*
+                        * Direct child node of the configurations parent node,
+                        * i.e. configuration node.
+                        */
+                       printf ("%s Configuration %u (%s)\n", p, count++,
+                                       fit_get_name(fit, noffset, NULL));
+
+                       fit_conf_print (fit, noffset, p);
+               }
+       }
+}
+
+/**
+ * fit_image_print - prints out the FIT component image details
+ * @fit: pointer to the FIT format image header
+ * @image_noffset: offset of the component image node
+ * @p: pointer to prefix string
+ *
+ * fit_image_print() lists all mandatory properies for the processed component
+ * image. If present, hash nodes are printed out as well.
+ *
+ * returns:
+ *     no returned results
+ */
+void fit_image_print (const void *fit, int image_noffset, const char *p)
+{
+       char *desc;
+       uint8_t type, arch, os, comp;
+       size_t size;
+       ulong load, entry;
+       const void *data;
+       int noffset;
+       int ndepth;
+       int ret;
+
+       /* Mandatory properties */
+       ret = fit_get_desc (fit, image_noffset, &desc);
+       printf ("%s  Description:  ", p);
+       if (ret)
+               printf ("unavailable\n");
+       else
+               printf ("%s\n", desc);
+
+       fit_image_get_type (fit, image_noffset, &type);
+       printf ("%s  Type:         %s\n", p, genimg_get_type_name (type));
+
+       fit_image_get_comp (fit, image_noffset, &comp);
+       printf ("%s  Compression:  %s\n", p, genimg_get_comp_name (comp));
+
+       ret = fit_image_get_data (fit, image_noffset, &data, &size);
+
+#ifndef USE_HOSTCC
+       printf ("%s  Data Start:   ", p);
+       if (ret)
+               printf ("unavailable\n");
+       else
+               printf ("0x%08lx\n", (ulong)data);
+#endif
+
+       printf ("%s  Data Size:    ", p);
+       if (ret)
+               printf ("unavailable\n");
+       else
+               genimg_print_size (size);
+
+       /* Remaining, type dependent properties */
+       if ((type == IH_TYPE_KERNEL) || (type == IH_TYPE_STANDALONE) ||
+           (type == IH_TYPE_RAMDISK) || (type == IH_TYPE_FIRMWARE) ||
+           (type == IH_TYPE_FLATDT)) {
+               fit_image_get_arch (fit, image_noffset, &arch);
+               printf ("%s  Architecture: %s\n", p, genimg_get_arch_name (arch));
+       }
+
+       if (type == IH_TYPE_KERNEL) {
+               fit_image_get_os (fit, image_noffset, &os);
+               printf ("%s  OS:           %s\n", p, genimg_get_os_name (os));
+       }
+
+       if ((type == IH_TYPE_KERNEL) || (type == IH_TYPE_STANDALONE)) {
+               ret = fit_image_get_load (fit, image_noffset, &load);
+               printf ("%s  Load Address: ", p);
+               if (ret)
+                       printf ("unavailable\n");
+               else
+                       printf ("0x%08lx\n", load);
+
+               fit_image_get_entry (fit, image_noffset, &entry);
+               printf ("%s  Entry Point:  ", p);
+               if (ret)
+                       printf ("unavailable\n");
+               else
+                       printf ("0x%08lx\n", entry);
+       }
+
+       /* Process all hash subnodes of the component image node */
+       for (ndepth = 0, noffset = fdt_next_node (fit, image_noffset, &ndepth);
+            (noffset >= 0) && (ndepth > 0);
+            noffset = fdt_next_node (fit, noffset, &ndepth)) {
+               if (ndepth == 1) {
+                       /* Direct child node of the component image node */
+                       fit_image_print_hash (fit, noffset, p);
+               }
+       }
+}
+
+/**
+ * fit_image_print_hash - prints out the hash node details
+ * @fit: pointer to the FIT format image header
+ * @noffset: offset of the hash node
+ * @p: pointer to prefix string
+ *
+ * fit_image_print_hash() lists properies for the processed hash node
+ *
+ * returns:
+ *     no returned results
+ */
+void fit_image_print_hash (const void *fit, int noffset, const char *p)
+{
+       char *algo;
+       uint8_t *value;
+       int value_len;
+       int i, ret;
+
+       /*
+        * Check subnode name, must be equal to "hash".
+        * Multiple hash nodes require unique unit node
+        * names, e.g. hash@1, hash@2, etc.
+        */
+       if (strncmp (fit_get_name(fit, noffset, NULL),
+                       FIT_HASH_NODENAME,
+                       strlen(FIT_HASH_NODENAME)) != 0)
+               return;
+
+       debug ("%s  Hash node:    '%s'\n", p,
+                       fit_get_name (fit, noffset, NULL));
+
+       printf ("%s  Hash algo:    ", p);
+       if (fit_image_hash_get_algo (fit, noffset, &algo)) {
+               printf ("invalid/unsupported\n");
+               return;
+       }
+       printf ("%s\n", algo);
+
+       ret = fit_image_hash_get_value (fit, noffset, &value,
+                                       &value_len);
+       printf ("%s  Hash value:   ", p);
+       if (ret) {
+               printf ("unavailable\n");
+       } else {
+               for (i = 0; i < value_len; i++)
+                       printf ("%02x", value[i]);
+               printf ("\n");
+       }
+
+       debug  ("%s  Hash len:     %d\n", p, value_len);
+}
+
+/**
+ * fit_get_desc - get node description property
+ * @fit: pointer to the FIT format image header
+ * @noffset: node offset
+ * @desc: double pointer to the char, will hold pointer to the descrption
+ *
+ * fit_get_desc() reads description property from a given node, if
+ * description is found pointer to it is returened in third call argument.
+ *
+ * returns:
+ *     0, on success
+ *     -1, on failure
+ */
+int fit_get_desc (const void *fit, int noffset, char **desc)
+{
+       int len;
+
+       *desc = (char *)fdt_getprop (fit, noffset, FIT_DESC_PROP, &len);
+       if (*desc == NULL) {
+               fit_get_debug (fit, noffset, FIT_DESC_PROP, len);
+               return -1;
+       }
+
+       return 0;
+}
+
+/**
+ * fit_get_timestamp - get node timestamp property
+ * @fit: pointer to the FIT format image header
+ * @noffset: node offset
+ * @timestamp: pointer to the time_t, will hold read timestamp
+ *
+ * fit_get_timestamp() reads timestamp poperty from given node, if timestamp
+ * is found and has a correct size its value is retured in third call
+ * argument.
+ *
+ * returns:
+ *     0, on success
+ *     -1, on property read failure
+ *     -2, on wrong timestamp size
+ */
+int fit_get_timestamp (const void *fit, int noffset, time_t *timestamp)
+{
+       int len;
+       const void *data;
+
+       data = fdt_getprop (fit, noffset, FIT_TIMESTAMP_PROP, &len);
+       if (data == NULL) {
+               fit_get_debug (fit, noffset, FIT_TIMESTAMP_PROP, len);
+               return -1;
+       }
+       if (len != sizeof (uint32_t)) {
+               debug ("FIT timestamp with incorrect size of (%u)\n", len);
+               return -2;
+       }
+
+       *timestamp = uimage_to_cpu (*((uint32_t *)data));
+       return 0;
+}
+
+/**
+ * fit_image_get_node - get node offset for component image of a given unit name
+ * @fit: pointer to the FIT format image header
+ * @image_uname: component image node unit name
+ *
+ * fit_image_get_node() finds a component image (withing the '/images'
+ * node) of a provided unit name. If image is found its node offset is
+ * returned to the caller.
+ *
+ * returns:
+ *     image node offset when found (>=0)
+ *     negative number on failure (FDT_ERR_* code)
+ */
+int fit_image_get_node (const void *fit, const char *image_uname)
+{
+       int noffset, images_noffset;
+
+       images_noffset = fdt_path_offset (fit, FIT_IMAGES_PATH);
+       if (images_noffset < 0) {
+               debug ("Can't find images parent node '%s' (%s)\n",
+                       FIT_IMAGES_PATH, fdt_strerror (images_noffset));
+               return images_noffset;
+       }
+
+       noffset = fdt_subnode_offset (fit, images_noffset, image_uname);
+       if (noffset < 0) {
+               debug ("Can't get node offset for image unit name: '%s' (%s)\n",
+                       image_uname, fdt_strerror (noffset));
+       }
+
+       return noffset;
+}
+
+/**
+ * fit_image_get_os - get os id for a given component image node
+ * @fit: pointer to the FIT format image header
+ * @noffset: component image node offset
+ * @os: pointer to the uint8_t, will hold os numeric id
+ *
+ * fit_image_get_os() finds os property in a given component image node.
+ * If the property is found, its (string) value is translated to the numeric
+ * id which is returned to the caller.
+ *
+ * returns:
+ *     0, on success
+ *     -1, on failure
+ */
+int fit_image_get_os (const void *fit, int noffset, uint8_t *os)
+{
+       int len;
+       const void *data;
+
+       /* Get OS name from property data */
+       data = fdt_getprop (fit, noffset, FIT_OS_PROP, &len);
+       if (data == NULL) {
+               fit_get_debug (fit, noffset, FIT_OS_PROP, len);
+               *os = -1;
+               return -1;
+       }
+
+       /* Translate OS name to id */
+       *os = genimg_get_os_id (data);
+       return 0;
+}
+
+/**
+ * fit_image_get_arch - get arch id for a given component image node
+ * @fit: pointer to the FIT format image header
+ * @noffset: component image node offset
+ * @arch: pointer to the uint8_t, will hold arch numeric id
+ *
+ * fit_image_get_arch() finds arch property in a given component image node.
+ * If the property is found, its (string) value is translated to the numeric
+ * id which is returned to the caller.
+ *
+ * returns:
+ *     0, on success
+ *     -1, on failure
+ */
+int fit_image_get_arch (const void *fit, int noffset, uint8_t *arch)
+{
+       int len;
+       const void *data;
+
+       /* Get architecture name from property data */
+       data = fdt_getprop (fit, noffset, FIT_ARCH_PROP, &len);
+       if (data == NULL) {
+               fit_get_debug (fit, noffset, FIT_ARCH_PROP, len);
+               *arch = -1;
+               return -1;
+       }
+
+       /* Translate architecture name to id */
+       *arch = genimg_get_arch_id (data);
+       return 0;
+}
+
+/**
+ * fit_image_get_type - get type id for a given component image node
+ * @fit: pointer to the FIT format image header
+ * @noffset: component image node offset
+ * @type: pointer to the uint8_t, will hold type numeric id
+ *
+ * fit_image_get_type() finds type property in a given component image node.
+ * If the property is found, its (string) value is translated to the numeric
+ * id which is returned to the caller.
+ *
+ * returns:
+ *     0, on success
+ *     -1, on failure
+ */
+int fit_image_get_type (const void *fit, int noffset, uint8_t *type)
+{
+       int len;
+       const void *data;
+
+       /* Get image type name from property data */
+       data = fdt_getprop (fit, noffset, FIT_TYPE_PROP, &len);
+       if (data == NULL) {
+               fit_get_debug (fit, noffset, FIT_TYPE_PROP, len);
+               *type = -1;
+               return -1;
+       }
+
+       /* Translate image type name to id */
+       *type = genimg_get_type_id (data);
+       return 0;
+}
+
+/**
+ * fit_image_get_comp - get comp id for a given component image node
+ * @fit: pointer to the FIT format image header
+ * @noffset: component image node offset
+ * @comp: pointer to the uint8_t, will hold comp numeric id
+ *
+ * fit_image_get_comp() finds comp property in a given component image node.
+ * If the property is found, its (string) value is translated to the numeric
+ * id which is returned to the caller.
+ *
+ * returns:
+ *     0, on success
+ *     -1, on failure
+ */
+int fit_image_get_comp (const void *fit, int noffset, uint8_t *comp)
+{
+       int len;
+       const void *data;
+
+       /* Get compression name from property data */
+       data = fdt_getprop (fit, noffset, FIT_COMP_PROP, &len);
+       if (data == NULL) {
+               fit_get_debug (fit, noffset, FIT_COMP_PROP, len);
+               *comp = -1;
+               return -1;
+       }
+
+       /* Translate compression name to id */
+       *comp = genimg_get_comp_id (data);
+       return 0;
+}
+
+/**
+ * fit_image_get_load - get load address property for a given component image node
+ * @fit: pointer to the FIT format image header
+ * @noffset: component image node offset
+ * @load: pointer to the uint32_t, will hold load address
+ *
+ * fit_image_get_load() finds load address property in a given component image node.
+ * If the property is found, its value is returned to the caller.
+ *
+ * returns:
+ *     0, on success
+ *     -1, on failure
+ */
+int fit_image_get_load (const void *fit, int noffset, ulong *load)
+{
+       int len;
+       const uint32_t *data;
+
+       data = fdt_getprop (fit, noffset, FIT_LOAD_PROP, &len);
+       if (data == NULL) {
+               fit_get_debug (fit, noffset, FIT_LOAD_PROP, len);
+               return -1;
+       }
+
+       *load = uimage_to_cpu (*data);
+       return 0;
+}
+
+/**
+ * fit_image_get_entry - get entry point address property for a given component image node
+ * @fit: pointer to the FIT format image header
+ * @noffset: component image node offset
+ * @entry: pointer to the uint32_t, will hold entry point address
+ *
+ * fit_image_get_entry() finds entry point address property in a given component image node.
+ * If the property is found, its value is returned to the caller.
+ *
+ * returns:
+ *     0, on success
+ *     -1, on failure
+ */
+int fit_image_get_entry (const void *fit, int noffset, ulong *entry)
+{
+       int len;
+       const uint32_t *data;
+
+       data = fdt_getprop (fit, noffset, FIT_ENTRY_PROP, &len);
+       if (data == NULL) {
+               fit_get_debug (fit, noffset, FIT_ENTRY_PROP, len);
+               return -1;
+       }
+
+       *entry = uimage_to_cpu (*data);
+       return 0;
+}
+
+/**
+ * fit_image_get_data - get data property and its size for a given component image node
+ * @fit: pointer to the FIT format image header
+ * @noffset: component image node offset
+ * @data: double pointer to void, will hold data property's data address
+ * @size: pointer to size_t, will hold data property's data size
+ *
+ * fit_image_get_data() finds data property in a given component image node.
+ * If the property is found its data start address and size are returned to
+ * the caller.
+ *
+ * returns:
+ *     0, on success
+ *     -1, on failure
+ */
+int fit_image_get_data (const void *fit, int noffset,
+               const void **data, size_t *size)
+{
+       int len;
+
+       *data = fdt_getprop (fit, noffset, FIT_DATA_PROP, &len);
+       if (*data == NULL) {
+               fit_get_debug (fit, noffset, FIT_DATA_PROP, len);
+               *size = 0;
+               return -1;
+       }
+
+       *size = len;
+       return 0;
+}
+
+/**
+ * fit_image_hash_get_algo - get hash algorithm name
+ * @fit: pointer to the FIT format image header
+ * @noffset: hash node offset
+ * @algo: double pointer to char, will hold pointer to the algorithm name
+ *
+ * fit_image_hash_get_algo() finds hash algorithm property in a given hash node.
+ * If the property is found its data start address is returned to the caller.
+ *
+ * returns:
+ *     0, on success
+ *     -1, on failure
+ */
+int fit_image_hash_get_algo (const void *fit, int noffset, char **algo)
+{
+       int len;
+
+       *algo = (char *)fdt_getprop (fit, noffset, FIT_ALGO_PROP, &len);
+       if (*algo == NULL) {
+               fit_get_debug (fit, noffset, FIT_ALGO_PROP, len);
+               return -1;
+       }
+
+       return 0;
+}
+
+/**
+ * fit_image_hash_get_value - get hash value and length
+ * @fit: pointer to the FIT format image header
+ * @noffset: hash node offset
+ * @value: double pointer to uint8_t, will hold address of a hash value data
+ * @value_len: pointer to an int, will hold hash data length
+ *
+ * fit_image_hash_get_value() finds hash value property in a given hash node.
+ * If the property is found its data start address and size are returned to
+ * the caller.
+ *
+ * returns:
+ *     0, on success
+ *     -1, on failure
+ */
+int fit_image_hash_get_value (const void *fit, int noffset, uint8_t **value,
+                               int *value_len)
+{
+       int len;
+
+       *value = (uint8_t *)fdt_getprop (fit, noffset, FIT_VALUE_PROP, &len);
+       if (*value == NULL) {
+               fit_get_debug (fit, noffset, FIT_VALUE_PROP, len);
+               *value_len = 0;
+               return -1;
+       }
+
+       *value_len = len;
+       return 0;
+}
+
+/**
+ * fit_set_timestamp - set node timestamp property
+ * @fit: pointer to the FIT format image header
+ * @noffset: node offset
+ * @timestamp: timestamp value to be set
+ *
+ * fit_set_timestamp() attempts to set timestamp property in the requested
+ * node and returns operation status to the caller.
+ *
+ * returns:
+ *     0, on success
+ *     -1, on property read failure
+ */
+int fit_set_timestamp (void *fit, int noffset, time_t timestamp)
+{
+       uint32_t t;
+       int ret;
+
+       t = cpu_to_uimage (timestamp);
+       ret = fdt_setprop (fit, noffset, FIT_TIMESTAMP_PROP, &t,
+                               sizeof (uint32_t));
+       if (ret) {
+               printf ("Can't set '%s' property for '%s' node (%s)\n",
+                       FIT_TIMESTAMP_PROP, fit_get_name (fit, noffset, NULL),
+                       fdt_strerror (ret));
+               return -1;
+       }
+
+       return 0;
+}
+
+/**
+ * calculate_hash - calculate and return hash for provided input data
+ * @data: pointer to the input data
+ * @data_len: data length
+ * @algo: requested hash algorithm
+ * @value: pointer to the char, will hold hash value data (caller must
+ * allocate enough free space)
+ * value_len: length of the calculated hash
+ *
+ * calculate_hash() computes input data hash according to the requested algorithm.
+ * Resulting hash value is placed in caller provided 'value' buffer, length
+ * of the calculated hash is returned via value_len pointer argument.
+ *
+ * returns:
+ *     0, on success
+ *    -1, when algo is unsupported
+ */
+static int calculate_hash (const void *data, int data_len, const char *algo,
+                       uint8_t *value, int *value_len)
+{
+       if (strcmp (algo, "crc32") == 0 ) {
+               *((uint32_t *)value) = crc32_wd (0, data, data_len,
+                                                       CHUNKSZ_CRC32);
+               *((uint32_t *)value) = cpu_to_uimage (*((uint32_t *)value));
+               *value_len = 4;
+       } else if (strcmp (algo, "sha1") == 0 ) {
+               sha1_csum_wd ((unsigned char *) data, data_len,
+                               (unsigned char *) value, CHUNKSZ_SHA1);
+               *value_len = 20;
+       } else if (strcmp (algo, "md5") == 0 ) {
+               md5_wd ((unsigned char *)data, data_len, value, CHUNKSZ_MD5);
+               *value_len = 16;
+       } else {
+               debug ("Unsupported hash alogrithm\n");
+               return -1;
+       }
+       return 0;
+}
+
+#ifdef USE_HOSTCC
+/**
+ * fit_set_hashes - process FIT component image nodes and calculate hashes
+ * @fit: pointer to the FIT format image header
+ *
+ * fit_set_hashes() adds hash values for all component images in the FIT blob.
+ * Hashes are calculated for all component images which have hash subnodes
+ * with algorithm property set to one of the supported hash algorithms.
+ *
+ * returns
+ *     0, on success
+ *     libfdt error code, on failure
+ */
+int fit_set_hashes (void *fit)
+{
+       int images_noffset;
+       int noffset;
+       int ndepth;
+       int ret;
+
+       /* Find images parent node offset */
+       images_noffset = fdt_path_offset (fit, FIT_IMAGES_PATH);
+       if (images_noffset < 0) {
+               printf ("Can't find images parent node '%s' (%s)\n",
+                       FIT_IMAGES_PATH, fdt_strerror (images_noffset));
+               return images_noffset;
+       }
+
+       /* Process its subnodes, print out component images details */
+       for (ndepth = 0, noffset = fdt_next_node (fit, images_noffset, &ndepth);
+            (noffset >= 0) && (ndepth > 0);
+            noffset = fdt_next_node (fit, noffset, &ndepth)) {
+               if (ndepth == 1) {
+                       /*
+                        * Direct child node of the images parent node,
+                        * i.e. component image node.
+                        */
+                       ret = fit_image_set_hashes (fit, noffset);
+                       if (ret)
+                               return ret;
+               }
+       }
+
+       return 0;
+}
+
+/**
+ * fit_image_set_hashes - calculate/set hashes for given component image node
+ * @fit: pointer to the FIT format image header
+ * @image_noffset: requested component image node
+ *
+ * fit_image_set_hashes() adds hash values for an component image node. All
+ * existing hash subnodes are checked, if algorithm property is set to one of
+ * the supported hash algorithms, hash value is computed and corresponding
+ * hash node property is set, for example:
+ *
+ * Input component image node structure:
+ *
+ * o image@1 (at image_noffset)
+ *   | - data = [binary data]
+ *   o hash@1
+ *     |- algo = "sha1"
+ *
+ * Output component image node structure:
+ *
+ * o image@1 (at image_noffset)
+ *   | - data = [binary data]
+ *   o hash@1
+ *     |- algo = "sha1"
+ *     |- value = sha1(data)
+ *
+ * returns:
+ *     0 on sucess
+ *    <0 on failure
+ */
+int fit_image_set_hashes (void *fit, int image_noffset)
+{
+       const void *data;
+       size_t size;
+       char *algo;
+       uint8_t value[FIT_MAX_HASH_LEN];
+       int value_len;
+       int noffset;
+       int ndepth;
+
+       /* Get image data and data length */
+       if (fit_image_get_data (fit, image_noffset, &data, &size)) {
+               printf ("Can't get image data/size\n");
+               return -1;
+       }
+
+       /* Process all hash subnodes of the component image node */
+       for (ndepth = 0, noffset = fdt_next_node (fit, image_noffset, &ndepth);
+            (noffset >= 0) && (ndepth > 0);
+            noffset = fdt_next_node (fit, noffset, &ndepth)) {
+               if (ndepth == 1) {
+                       /* Direct child node of the component image node */
+
+                       /*
+                        * Check subnode name, must be equal to "hash".
+                        * Multiple hash nodes require unique unit node
+                        * names, e.g. hash@1, hash@2, etc.
+                        */
+                       if (strncmp (fit_get_name(fit, noffset, NULL),
+                                               FIT_HASH_NODENAME,
+                                               strlen(FIT_HASH_NODENAME)) != 0) {
+                               /* Not a hash subnode, skip it */
+                               continue;
+                       }
+
+                       if (fit_image_hash_get_algo (fit, noffset, &algo)) {
+                               printf ("Can't get hash algo property for "
+                                       "'%s' hash node in '%s' image node\n",
+                                       fit_get_name (fit, noffset, NULL),
+                                       fit_get_name (fit, image_noffset, NULL));
+                               return -1;
+                       }
+
+                       if (calculate_hash (data, size, algo, value, &value_len)) {
+                               printf ("Unsupported hash algorithm (%s) for "
+                                       "'%s' hash node in '%s' image node\n",
+                                       algo, fit_get_name (fit, noffset, NULL),
+                                       fit_get_name (fit, image_noffset, NULL));
+                               return -1;
+                       }
+
+                       if (fit_image_hash_set_value (fit, noffset, value,
+                                                       value_len)) {
+                               printf ("Can't set hash value for "
+                                       "'%s' hash node in '%s' image node\n",
+                                       fit_get_name (fit, noffset, NULL),
+                                       fit_get_name (fit, image_noffset, NULL));
+                               return -1;
+                       }
+               }
+       }
+
+       return 0;
+}
+
+/**
+ * fit_image_hash_set_value - set hash value in requested has node
+ * @fit: pointer to the FIT format image header
+ * @noffset: hash node offset
+ * @value: hash value to be set
+ * @value_len: hash value length
+ *
+ * fit_image_hash_set_value() attempts to set hash value in a node at offset
+ * given and returns operation status to the caller.
+ *
+ * returns
+ *     0, on success
+ *     -1, on failure
+ */
+int fit_image_hash_set_value (void *fit, int noffset, uint8_t *value,
+                               int value_len)
+{
+       int ret;
+
+       ret = fdt_setprop (fit, noffset, FIT_VALUE_PROP, value, value_len);
+       if (ret) {
+               printf ("Can't set hash '%s' property for '%s' node (%s)\n",
+                       FIT_VALUE_PROP, fit_get_name (fit, noffset, NULL),
+                       fdt_strerror (ret));
+               return -1;
+       }
+
+       return 0;
+}
+#endif /* USE_HOSTCC */
+
+/**
+ * fit_image_check_hashes - verify data intergity
+ * @fit: pointer to the FIT format image header
+ * @image_noffset: component image node offset
+ *
+ * fit_image_check_hashes() goes over component image hash nodes,
+ * re-calculates each data hash and compares with the value stored in hash
+ * node.
+ *
+ * returns:
+ *     1, if all hashes are valid
+ *     0, otherwise (or on error)
+ */
+int fit_image_check_hashes (const void *fit, int image_noffset)
+{
+       const void      *data;
+       size_t          size;
+       char            *algo;
+       uint8_t         *fit_value;
+       int             fit_value_len;
+       uint8_t         value[FIT_MAX_HASH_LEN];
+       int             value_len;
+       int             noffset;
+       int             ndepth;
+       char            *err_msg = "";
+
+       /* Get image data and data length */
+       if (fit_image_get_data (fit, image_noffset, &data, &size)) {
+               printf ("Can't get image data/size\n");
+               return 0;
+       }
+
+       /* Process all hash subnodes of the component image node */
+       for (ndepth = 0, noffset = fdt_next_node (fit, image_noffset, &ndepth);
+            (noffset >= 0) && (ndepth > 0);
+            noffset = fdt_next_node (fit, noffset, &ndepth)) {
+               if (ndepth == 1) {
+                       /* Direct child node of the component image node */
+
+                       /*
+                        * Check subnode name, must be equal to "hash".
+                        * Multiple hash nodes require unique unit node
+                        * names, e.g. hash@1, hash@2, etc.
+                        */
+                       if (strncmp (fit_get_name(fit, noffset, NULL),
+                                       FIT_HASH_NODENAME,
+                                       strlen(FIT_HASH_NODENAME)) != 0)
+                               continue;
+
+                       if (fit_image_hash_get_algo (fit, noffset, &algo)) {
+                               err_msg = "Can't get hash algo property";
+                               goto error;
+                       }
+                       printf ("%s", algo);
+
+                       if (fit_image_hash_get_value (fit, noffset, &fit_value,
+                                                       &fit_value_len)) {
+                               err_msg = "Can't get hash value property";
+                               goto error;
+                       }
+
+                       if (calculate_hash (data, size, algo, value, &value_len)) {
+                               err_msg = "Unsupported hash algorithm";
+                               goto error;
+                       }
+
+                       if (value_len != fit_value_len) {
+                               err_msg = "Bad hash value len";
+                               goto error;
+                       } else if (memcmp (value, fit_value, value_len) != 0) {
+                               err_msg = "Bad hash value";
+                               goto error;
+                       }
+                       printf ("+ ");
+               }
+       }
+
+       return 1;
+
+error:
+       printf ("%s for '%s' hash node in '%s' image node\n",
+                       err_msg, fit_get_name (fit, noffset, NULL),
+                       fit_get_name (fit, image_noffset, NULL));
+       return 0;
+}
+
+/**
+ * fit_image_check_os - check whether image node is of a given os type
+ * @fit: pointer to the FIT format image header
+ * @noffset: component image node offset
+ * @os: requested image os
+ *
+ * fit_image_check_os() reads image os property and compares its numeric
+ * id with the requested os. Comparison result is returned to the caller.
+ *
+ * returns:
+ *     1 if image is of given os type
+ *     0 otherwise (or on error)
+ */
+int fit_image_check_os (const void *fit, int noffset, uint8_t os)
+{
+       uint8_t image_os;
+
+       if (fit_image_get_os (fit, noffset, &image_os))
+               return 0;
+       return (os == image_os);
+}
+
+/**
+ * fit_image_check_arch - check whether image node is of a given arch
+ * @fit: pointer to the FIT format image header
+ * @noffset: component image node offset
+ * @arch: requested imagearch
+ *
+ * fit_image_check_arch() reads image arch property and compares its numeric
+ * id with the requested arch. Comparison result is returned to the caller.
+ *
+ * returns:
+ *     1 if image is of given arch
+ *     0 otherwise (or on error)
+ */
+int fit_image_check_arch (const void *fit, int noffset, uint8_t arch)
+{
+       uint8_t image_arch;
+
+       if (fit_image_get_arch (fit, noffset, &image_arch))
+               return 0;
+       return (arch == image_arch);
+}
+
+/**
+ * fit_image_check_type - check whether image node is of a given type
+ * @fit: pointer to the FIT format image header
+ * @noffset: component image node offset
+ * @type: requested image type
+ *
+ * fit_image_check_type() reads image type property and compares its numeric
+ * id with the requested type. Comparison result is returned to the caller.
+ *
+ * returns:
+ *     1 if image is of given type
+ *     0 otherwise (or on error)
+ */
+int fit_image_check_type (const void *fit, int noffset, uint8_t type)
+{
+       uint8_t image_type;
+
+       if (fit_image_get_type (fit, noffset, &image_type))
+               return 0;
+       return (type == image_type);
+}
+
+/**
+ * fit_image_check_comp - check whether image node uses given compression
+ * @fit: pointer to the FIT format image header
+ * @noffset: component image node offset
+ * @comp: requested image compression type
+ *
+ * fit_image_check_comp() reads image compression property and compares its
+ * numeric id with the requested compression type. Comparison result is
+ * returned to the caller.
+ *
+ * returns:
+ *     1 if image uses requested compression
+ *     0 otherwise (or on error)
+ */
+int fit_image_check_comp (const void *fit, int noffset, uint8_t comp)
+{
+       uint8_t image_comp;
+
+       if (fit_image_get_comp (fit, noffset, &image_comp))
+               return 0;
+       return (comp == image_comp);
+}
+
+/**
+ * fit_check_format - sanity check FIT image format
+ * @fit: pointer to the FIT format image header
+ *
+ * fit_check_format() runs a basic sanity FIT image verification.
+ * Routine checks for mandatory properties, nodes, etc.
+ *
+ * returns:
+ *     1, on success
+ *     0, on failure
+ */
+int fit_check_format (const void *fit)
+{
+       /* mandatory / node 'description' property */
+       if (fdt_getprop (fit, 0, FIT_DESC_PROP, NULL) == NULL) {
+               debug ("Wrong FIT format: no description\n");
+               return 0;
+       }
+
+#if defined(CONFIG_TIMESTAMP) || defined(CONFIG_CMD_DATE) || defined(USE_HOSTCC)
+       /* mandatory / node 'timestamp' property */
+       if (fdt_getprop (fit, 0, FIT_TIMESTAMP_PROP, NULL) == NULL) {
+               debug ("Wrong FIT format: no description\n");
+               return 0;
+       }
+#endif
+
+       /* mandatory subimages parent '/images' node */
+       if (fdt_path_offset (fit, FIT_IMAGES_PATH) < 0) {
+               debug ("Wrong FIT format: no images parent node\n");
+               return 0;
+       }
+
+       return 1;
+}
+
+/**
+ * fit_conf_get_node - get node offset for configuration of a given unit name
+ * @fit: pointer to the FIT format image header
+ * @conf_uname: configuration node unit name
+ *
+ * fit_conf_get_node() finds a configuration (withing the '/configurations'
+ * parant node) of a provided unit name. If configuration is found its node offset
+ * is returned to the caller.
+ *
+ * When NULL is provided in second argument fit_conf_get_node() will search
+ * for a default configuration node instead. Default configuration node unit name
+ * is retrived from FIT_DEFAULT_PROP property of the '/configurations' node.
+ *
+ * returns:
+ *     configuration node offset when found (>=0)
+ *     negative number on failure (FDT_ERR_* code)
+ */
+int fit_conf_get_node (const void *fit, const char *conf_uname)
+{
+       int noffset, confs_noffset;
+       int len;
+
+       confs_noffset = fdt_path_offset (fit, FIT_CONFS_PATH);
+       if (confs_noffset < 0) {
+               debug ("Can't find configurations parent node '%s' (%s)\n",
+                       FIT_CONFS_PATH, fdt_strerror (confs_noffset));
+               return confs_noffset;
+       }
+
+       if (conf_uname == NULL) {
+               /* get configuration unit name from the default property */
+               debug ("No configuration specified, trying default...\n");
+               conf_uname = (char *)fdt_getprop (fit, confs_noffset, FIT_DEFAULT_PROP, &len);
+               if (conf_uname == NULL) {
+                       fit_get_debug (fit, confs_noffset, FIT_DEFAULT_PROP, len);
+                       return len;
+               }
+               debug ("Found default configuration: '%s'\n", conf_uname);
+       }
+
+       noffset = fdt_subnode_offset (fit, confs_noffset, conf_uname);
+       if (noffset < 0) {
+               debug ("Can't get node offset for configuration unit name: '%s' (%s)\n",
+                       conf_uname, fdt_strerror (noffset));
+       }
+
+       return noffset;
+}
+
+static int __fit_conf_get_prop_node (const void *fit, int noffset,
+               const char *prop_name)
+{
+       char *uname;
+       int len;
+
+       /* get kernel image unit name from configuration kernel property */
+       uname = (char *)fdt_getprop (fit, noffset, prop_name, &len);
+       if (uname == NULL)
+               return len;
+
+       return fit_image_get_node (fit, uname);
+}
+
+/**
+ * fit_conf_get_kernel_node - get kernel image node offset that corresponds to
+ * a given configuration
+ * @fit: pointer to the FIT format image header
+ * @noffset: configuration node offset
+ *
+ * fit_conf_get_kernel_node() retrives kernel image node unit name from
+ * configuration FIT_KERNEL_PROP property and translates it to the node
+ * offset.
+ *
+ * returns:
+ *     image node offset when found (>=0)
+ *     negative number on failure (FDT_ERR_* code)
+ */
+int fit_conf_get_kernel_node (const void *fit, int noffset)
+{
+       return __fit_conf_get_prop_node (fit, noffset, FIT_KERNEL_PROP);
+}
+
+/**
+ * fit_conf_get_ramdisk_node - get ramdisk image node offset that corresponds to
+ * a given configuration
+ * @fit: pointer to the FIT format image header
+ * @noffset: configuration node offset
+ *
+ * fit_conf_get_ramdisk_node() retrives ramdisk image node unit name from
+ * configuration FIT_KERNEL_PROP property and translates it to the node
+ * offset.
+ *
+ * returns:
+ *     image node offset when found (>=0)
+ *     negative number on failure (FDT_ERR_* code)
+ */
+int fit_conf_get_ramdisk_node (const void *fit, int noffset)
+{
+       return __fit_conf_get_prop_node (fit, noffset, FIT_RAMDISK_PROP);
+}
+
+/**
+ * fit_conf_get_fdt_node - get fdt image node offset that corresponds to
+ * a given configuration
+ * @fit: pointer to the FIT format image header
+ * @noffset: configuration node offset
+ *
+ * fit_conf_get_fdt_node() retrives fdt image node unit name from
+ * configuration FIT_KERNEL_PROP property and translates it to the node
+ * offset.
+ *
+ * returns:
+ *     image node offset when found (>=0)
+ *     negative number on failure (FDT_ERR_* code)
+ */
+int fit_conf_get_fdt_node (const void *fit, int noffset)
+{
+       return __fit_conf_get_prop_node (fit, noffset, FIT_FDT_PROP);
+}
+
+/**
+ * fit_conf_print - prints out the FIT configuration details
+ * @fit: pointer to the FIT format image header
+ * @noffset: offset of the configuration node
+ * @p: pointer to prefix string
+ *
+ * fit_conf_print() lists all mandatory properies for the processed
+ * configuration node.
+ *
+ * returns:
+ *     no returned results
+ */
+void fit_conf_print (const void *fit, int noffset, const char *p)
+{
+       char *desc;
+       char *uname;
+       int ret;
+
+       /* Mandatory properties */
+       ret = fit_get_desc (fit, noffset, &desc);
+       printf ("%s  Description:  ", p);
+       if (ret)
+               printf ("unavailable\n");
+       else
+               printf ("%s\n", desc);
+
+       uname = (char *)fdt_getprop (fit, noffset, FIT_KERNEL_PROP, NULL);
+       printf ("%s  Kernel:       ", p);
+       if (uname == NULL)
+               printf ("unavailable\n");
+       else
+               printf ("%s\n", uname);
+
+       /* Optional properties */
+       uname = (char *)fdt_getprop (fit, noffset, FIT_RAMDISK_PROP, NULL);
+       if (uname)
+               printf ("%s  Init Ramdisk: %s\n", p, uname);
+
+       uname = (char *)fdt_getprop (fit, noffset, FIT_FDT_PROP, NULL);
+       if (uname)
+               printf ("%s  FDT:          %s\n", p, uname);
+}
+
+/**
+ * fit_check_ramdisk - verify FIT format ramdisk subimage
+ * @fit_hdr: pointer to the FIT ramdisk header
+ * @rd_noffset: ramdisk subimage node offset within FIT image
+ * @arch: requested ramdisk image architecture type
+ * @verify: data CRC verification flag
+ *
+ * fit_check_ramdisk() verifies integrity of the ramdisk subimage and from
+ * specified FIT image.
+ *
+ * returns:
+ *     1, on success
+ *     0, on failure
+ */
+#ifndef USE_HOSTCC
+static int fit_check_ramdisk (const void *fit, int rd_noffset, uint8_t arch, int verify)
+{
+       fit_image_print (fit, rd_noffset, "   ");
+
+       if (verify) {
+               puts ("   Verifying Hash Integrity ... ");
+               if (!fit_image_check_hashes (fit, rd_noffset)) {
+                       puts ("Bad Data Hash\n");
+                       show_boot_progress (-125);
+                       return 0;
+               }
+               puts ("OK\n");
+       }
+
+       show_boot_progress (126);
+       if (!fit_image_check_os (fit, rd_noffset, IH_OS_LINUX) ||
+           !fit_image_check_arch (fit, rd_noffset, arch) ||
+           !fit_image_check_type (fit, rd_noffset, IH_TYPE_RAMDISK)) {
+               printf ("No Linux %s Ramdisk Image\n",
+                               genimg_get_arch_name(arch));
+               show_boot_progress (-126);
+               return 0;
+       }
+
+       show_boot_progress (127);
+       return 1;
+}
+#endif /* USE_HOSTCC */
+#endif /* CONFIG_FIT */