]> git.sur5r.net Git - u-boot/blobdiff - lib_ppc/bootm.c
Cleanup coding style, update CHANGELOG
[u-boot] / lib_ppc / bootm.c
index 3e89da1c46185fffa580cb9462cd6684a93bfdae..b893146fcb5be2cdb17ddb4941a9f64393d7b046 100644 (file)
@@ -23,7 +23,6 @@
  * MA 02111-1307 USA
  */
 
-#define DEBUG
 
 #include <common.h>
 #include <watchdog.h>
@@ -41,9 +40,9 @@
 #include <fdt_support.h>
 
 static void fdt_error (const char *msg);
-static void get_fdt (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[],
+static int boot_get_fdt (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[],
                bootm_headers_t *images, char **of_flat_tree, ulong *of_size);
-static ulong fdt_relocate (ulong alloc_current,
+static int boot_relocate_fdt (struct lmb *lmb, ulong bootmap_base,
                cmd_tbl_t *cmdtp, int flag, int argc, char *argv[],
                char **of_flat_tree, ulong *of_size);
 #endif
@@ -55,28 +54,55 @@ static ulong fdt_relocate (ulong alloc_current,
 DECLARE_GLOBAL_DATA_PTR;
 
 extern int do_reset (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[]);
+extern ulong get_effective_memsize(void);
 static ulong get_sp (void);
 static void set_clocks_in_mhz (bd_t *kbd);
 
+#ifndef CFG_LINUX_LOWMEM_MAX_SIZE
+#define CFG_LINUX_LOWMEM_MAX_SIZE      (768*1024*1024)
+#endif
+
 void  __attribute__((noinline))
 do_bootm_linux(cmd_tbl_t *cmdtp, int flag, int argc, char *argv[],
                bootm_headers_t *images)
 {
-       ulong   sp, sp_limit, alloc_current;
+       ulong   sp;
 
        ulong   initrd_start, initrd_end;
        ulong   rd_data_start, rd_data_end, rd_len;
+       ulong   size;
 
-       ulong   cmd_start, cmd_end;
+       ulong   cmd_start, cmd_end, bootmap_base;
        bd_t    *kbd;
        ulong   ep = 0;
        void    (*kernel)(bd_t *, ulong, ulong, ulong, ulong);
+       int     ret;
        ulong   of_size = 0;
+       struct lmb *lmb = images->lmb;
 
 #if defined(CONFIG_OF_LIBFDT)
        char    *of_flat_tree = NULL;
 #endif
 
+       bootmap_base = getenv_bootm_low();
+       size = getenv_bootm_size();
+
+#ifdef DEBUG
+       if (((u64)bootmap_base + size) > (CFG_SDRAM_BASE + (u64)gd->ram_size))
+               puts("WARNING: bootm_low + bootm_size exceed total memory\n");
+       if ((bootmap_base + size) > get_effective_memsize())
+               puts("WARNING: bootm_low + bootm_size exceed eff. memory\n");
+#endif
+
+       size = min(size, get_effective_memsize());
+       size = min(size, CFG_LINUX_LOWMEM_MAX_SIZE);
+
+       if (size < getenv_bootm_size()) {
+               ulong base = bootmap_base + size;
+               printf("WARNING: adjusting available memory to %x\n", size);
+               lmb_reserve(lmb, base, getenv_bootm_size() - size);
+       }
+
        /*
         * Booting a (Linux) kernel image
         *
@@ -87,22 +113,34 @@ do_bootm_linux(cmd_tbl_t *cmdtp, int flag, int argc, char *argv[],
         * pointer.
         */
        sp = get_sp();
-       debug ("## Current stack ends at 0x%08lx ", sp);
+       debug ("## Current stack ends at 0x%08lx\n", sp);
 
-       alloc_current = sp_limit = get_boot_sp_limit(sp);
-       debug ("=> set upper limit to 0x%08lx\n", sp_limit);
+       /* adjust sp by 1K to be safe */
+       sp -= 1024;
+       lmb_reserve(lmb, sp, (CFG_SDRAM_BASE + get_effective_memsize() - sp));
 
 #if defined(CONFIG_OF_LIBFDT)
        /* find flattened device tree */
-       get_fdt (cmdtp, flag, argc, argv, images, &of_flat_tree, &of_size);
+       ret = boot_get_fdt (cmdtp, flag, argc, argv, images, &of_flat_tree, &of_size);
+
+       if (ret)
+               goto error;
 #endif
 
        if (!of_size) {
                /* allocate space and init command line */
-               alloc_current = get_boot_cmdline (alloc_current, &cmd_start, &cmd_end);
+               ret = boot_get_cmdline (lmb, &cmd_start, &cmd_end, bootmap_base);
+               if (ret) {
+                       puts("ERROR with allocation of cmdline\n");
+                       goto error;
+               }
 
                /* allocate space for kernel copy of board info */
-               alloc_current = get_boot_kbd (alloc_current, &kbd);
+               ret = boot_get_kbd (lmb, &kbd, bootmap_base);
+               if (ret) {
+                       puts("ERROR with allocation of kernel bd\n");
+                       goto error;
+               }
                set_clocks_in_mhz(kbd);
        }
 
@@ -111,47 +149,51 @@ do_bootm_linux(cmd_tbl_t *cmdtp, int flag, int argc, char *argv[],
                ep = image_get_ep (images->legacy_hdr_os);
 #if defined(CONFIG_FIT)
        } else if (images->fit_uname_os) {
-               fit_unsupported_reset ("PPC linux bootm");
-               do_reset (cmdtp, flag, argc, argv);
+               ret = fit_image_get_entry (images->fit_hdr_os,
+                               images->fit_noffset_os, &ep);
+               if (ret) {
+                       puts ("Can't get entry point property!\n");
+                       goto error;
+               }
 #endif
        } else {
                puts ("Could not find kernel entry point!\n");
-               do_reset (cmdtp, flag, argc, argv);
+               goto error;
        }
        kernel = (void (*)(bd_t *, ulong, ulong, ulong, ulong))ep;
 
        /* find ramdisk */
-       get_ramdisk (cmdtp, flag, argc, argv, images,
-                       IH_ARCH_PPC, &rd_data_start, &rd_data_end);
+       ret = boot_get_ramdisk (argc, argv, images, IH_ARCH_PPC,
+                       &rd_data_start, &rd_data_end);
+       if (ret)
+               goto error;
 
        rd_len = rd_data_end - rd_data_start;
 
-       alloc_current = ramdisk_high (alloc_current, rd_data_start, rd_len,
-                       sp_limit, get_sp (), &initrd_start, &initrd_end);
-
 #if defined(CONFIG_OF_LIBFDT)
-       alloc_current = fdt_relocate (alloc_current,
-                       cmdtp, flag, argc, argv, &of_flat_tree, &of_size);
+       ret = boot_relocate_fdt (lmb, bootmap_base,
+               cmdtp, flag, argc, argv, &of_flat_tree, &of_size);
 
        /*
         * Add the chosen node if it doesn't exist, add the env and bd_t
         * if the user wants it (the logic is in the subroutines).
         */
        if (of_size) {
-               if (fdt_chosen(of_flat_tree, initrd_start, initrd_end, 0) < 0) {
+               /* pass in dummy initrd info, we'll fix up later */
+               if (fdt_chosen(of_flat_tree, rd_data_start, rd_data_end, 0) < 0) {
                        fdt_error ("/chosen node create failed");
-                       do_reset (cmdtp, flag, argc, argv);
+                       goto error;
                }
 #ifdef CONFIG_OF_HAS_UBOOT_ENV
                if (fdt_env(of_flat_tree) < 0) {
                        fdt_error ("/u-boot-env node create failed");
-                       do_reset (cmdtp, flag, argc, argv);
+                       goto error;
                }
 #endif
 #ifdef CONFIG_OF_HAS_BD_T
                if (fdt_bd_t(of_flat_tree) < 0) {
                        fdt_error ("/bd_t node create failed");
-                       do_reset (cmdtp, flag, argc, argv);
+                       goto error;
                }
 #endif
 #ifdef CONFIG_OF_BOARD_SETUP
@@ -161,6 +203,39 @@ do_bootm_linux(cmd_tbl_t *cmdtp, int flag, int argc, char *argv[],
        }
 #endif /* CONFIG_OF_LIBFDT */
 
+       ret = boot_ramdisk_high (lmb, rd_data_start, rd_len, &initrd_start, &initrd_end);
+       if (ret)
+               goto error;
+
+#if defined(CONFIG_OF_LIBFDT)
+       /* fixup the initrd now that we know where it should be */
+       if ((of_flat_tree) && (initrd_start && initrd_end)) {
+               uint64_t addr, size;
+               int  total = fdt_num_mem_rsv(of_flat_tree);
+               int  j;
+
+               /* Look for the dummy entry and delete it */
+               for (j = 0; j < total; j++) {
+                       fdt_get_mem_rsv(of_flat_tree, j, &addr, &size);
+                       if (addr == rd_data_start) {
+                               fdt_del_mem_rsv(of_flat_tree, j);
+                               break;
+                       }
+               }
+
+               ret = fdt_add_mem_rsv(of_flat_tree, initrd_start,
+                                       initrd_end - initrd_start + 1);
+               if (ret < 0) {
+                       printf("fdt_chosen: %s\n", fdt_strerror(ret));
+                       goto error;
+               }
+
+               do_fixup_by_path_u32(of_flat_tree, "/chosen",
+                                       "linux,initrd-start", initrd_start, 0);
+               do_fixup_by_path_u32(of_flat_tree, "/chosen",
+                                       "linux,initrd-end", initrd_end, 0);
+       }
+#endif
        debug ("## Transferring control to Linux (at address %08lx) ...\n",
                (ulong)kernel);
 
@@ -169,6 +244,8 @@ do_bootm_linux(cmd_tbl_t *cmdtp, int flag, int argc, char *argv[],
 #if defined(CFG_INIT_RAM_LOCK) && !defined(CONFIG_E500)
        unlock_ram_in_cache();
 #endif
+       if (!images->autostart)
+               return ;
 
 #if defined(CONFIG_OF_LIBFDT)
        if (of_flat_tree) {     /* device tree; boot new style */
@@ -180,20 +257,30 @@ do_bootm_linux(cmd_tbl_t *cmdtp, int flag, int argc, char *argv[],
                 *   r6: NULL
                 *   r7: NULL
                 */
+               debug ("   Booting using OF flat tree...\n");
                (*kernel) ((bd_t *)of_flat_tree, (ulong)kernel, 0, 0, 0);
                /* does not return */
-       }
+       } else
 #endif
-       /*
-        * Linux Kernel Parameters (passing board info data):
-        *   r3: ptr to board info data
-        *   r4: initrd_start or 0 if no initrd
-        *   r5: initrd_end - unused if r4 is 0
-        *   r6: Start of command line string
-        *   r7: End   of command line string
-        */
-       (*kernel) (kbd, initrd_start, initrd_end, cmd_start, cmd_end);
-       /* does not return */
+       {
+               /*
+                * Linux Kernel Parameters (passing board info data):
+                *   r3: ptr to board info data
+                *   r4: initrd_start or 0 if no initrd
+                *   r5: initrd_end - unused if r4 is 0
+                *   r6: Start of command line string
+                *   r7: End   of command line string
+                */
+               debug ("   Booting using board info...\n");
+               (*kernel) (kbd, initrd_start, initrd_end, cmd_start, cmd_end);
+               /* does not return */
+       }
+       return ;
+
+error:
+       if (images->autostart)
+               do_reset (cmdtp, flag, argc, argv);
+       return ;
 }
 
 static ulong get_sp (void)
@@ -223,7 +310,7 @@ static void set_clocks_in_mhz (bd_t *kbd)
                kbd->bi_cpmfreq /= 1000000L;
                kbd->bi_brgfreq /= 1000000L;
                kbd->bi_sccfreq /= 1000000L;
-               kbd->bi_vco     /= 1000000L;
+               kbd->bi_vco     /= 1000000L;
 #endif
 #if defined(CONFIG_MPC5xxx)
                kbd->bi_ipbfreq /= 1000000L;
@@ -273,7 +360,48 @@ static image_header_t *image_get_fdt (ulong fdt_addr)
        return fdt_hdr;
 }
 
-static void get_fdt (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[],
+/**
+ * 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
+ *
+ * fit_check_fdt() verifies integrity of the FDT subimage and from
+ * specified FIT image.
+ *
+ * returns:
+ *     1, on success
+ *     0, on failure
+ */
+#if defined(CONFIG_FIT)
+static int fit_check_fdt (const void *fit, int fdt_noffset, int verify)
+{
+       fit_image_print (fit, fdt_noffset, "   ");
+
+       if (verify) {
+               puts ("   Verifying Hash Integrity ... ");
+               if (!fit_image_check_hashes (fit, fdt_noffset)) {
+                       fdt_error ("Bad Data Hash");
+                       return 0;
+               }
+               puts ("OK\n");
+       }
+
+       if (!fit_image_check_type (fit, fdt_noffset, IH_TYPE_FLATDT)) {
+               fdt_error ("Not a FDT image");
+               return 0;
+       }
+
+       if (!fit_image_check_comp (fit, fdt_noffset, IH_COMP_NONE)) {
+               fdt_error ("FDT image is compressed");
+               return 0;
+       }
+
+       return 1;
+}
+#endif /* CONFIG_FIT */
+
+static int boot_get_fdt (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[],
                bootm_headers_t *images, char **of_flat_tree, ulong *of_size)
 {
        ulong           fdt_addr;
@@ -282,63 +410,100 @@ static void get_fdt (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[],
        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;
+       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
 
-       if (argc > 3) {
+       *of_flat_tree = NULL;
+       *of_size = 0;
+
+       if (argc > 3 || genimg_has_config (images)) {
 #if defined(CONFIG_FIT)
-               /*
-                * 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
+               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);
+                       {
+                               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",
+               debug ("## Checking for 'FDT'/'FDT Image' at %08lx\n",
                                fdt_addr);
 
                /* copy from dataflash if needed */
-               fdt_addr = gen_get_image (fdt_addr);
+               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 (gen_image_get_format ((void *)fdt_addr)) {
+               switch (genimg_get_format ((void *)fdt_addr)) {
                case IMAGE_FORMAT_LEGACY:
-                       debug ("*  fdt: legacy format image\n");
-
                        /* verify fdt_addr points to a valid image header */
-                       printf ("## Flattened Device Tree Legacy Image at %08lx\n",
+                       printf ("## Flattened Device Tree from Legacy Image at %08lx\n",
                                        fdt_addr);
                        fdt_hdr = image_get_fdt (fdt_addr);
                        if (!fdt_hdr)
-                               do_reset (cmdtp, flag, argc, argv);
+                               goto error;
 
                        /*
                         * move image data to the load address,
@@ -352,13 +517,17 @@ static void get_fdt (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[],
 
                        if ((load_start < image_end) && (load_end > image_start)) {
                                fdt_error ("fdt overwritten");
-                               do_reset (cmdtp, flag, argc, argv);
+                               goto error;
                        }
-                       memmove ((void *)image_get_load (fdt_hdr),
+
+                       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 *)image_get_load (fdt_hdr);
+                       fdt_blob = (char *)load_start;
                        break;
                case IMAGE_FORMAT_FIT:
                        /*
@@ -368,14 +537,101 @@ static void get_fdt (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[],
                         */
 #if defined(CONFIG_FIT)
                        /* check FDT blob vs FIT blob */
-                       if (0) { /* FIXME: call FIT format verification */
+                       if (fit_check_format ((const void *)fdt_addr)) {
                                /*
                                 * FIT image
                                 */
                                fit_hdr = (void *)fdt_addr;
-                               debug ("*  fdt: FIT format image\n");
-                               fit_unsupported_reset ("PPC fdt");
-                               do_reset (cmdtp, flag, argc, argv);
+                               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
                        {
@@ -389,7 +645,7 @@ static void get_fdt (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[],
                        break;
                default:
                        fdt_error ("Did not find a cmdline Flattened Device Tree");
-                       do_reset (cmdtp, flag, argc, argv);
+                       goto error;
                }
 
                printf ("   Booting using the fdt blob at 0x%x\n", fdt_blob);
@@ -415,46 +671,50 @@ static void get_fdt (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[],
 
                        if (fdt_check_header (fdt_blob) != 0) {
                                fdt_error ("image is not a fdt");
-                               do_reset (cmdtp, flag, argc, argv);
+                               goto error;
                        }
 
                        if (be32_to_cpu (fdt_totalsize (fdt_blob)) != fdt_len) {
                                fdt_error ("fdt size != image size");
-                               do_reset (cmdtp, flag, argc, argv);
+                               goto error;
                        }
                } else {
                        fdt_error ("Did not find a Flattened Device Tree "
                                "in a legacy multi-component image");
-                       do_reset (cmdtp, flag, argc, argv);
+                       goto error;
                }
        } else {
                debug ("## No Flattened Device Tree\n");
-               *of_flat_tree = NULL;
-               *of_size = 0;
-               return;
+               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:
+       do_reset (cmdtp, flag, argc, argv);
+       return 1;
 }
 
-static ulong fdt_relocate (ulong alloc_current,
+static int boot_relocate_fdt (struct lmb *lmb, ulong bootmap_base,
                cmd_tbl_t *cmdtp, int flag, int argc, char *argv[],
                char **of_flat_tree, ulong *of_size)
 {
        char    *fdt_blob = *of_flat_tree;
        ulong   relocate = 0;
-       ulong   new_alloc_current;
+       ulong   of_len = 0;
 
        /* nothing to do */
        if (*of_size == 0)
-               return alloc_current;
+               return 0;
 
        if (fdt_check_header (fdt_blob) != 0) {
                fdt_error ("image is not a fdt");
-               do_reset (cmdtp, flag, argc, argv);
+               goto error;
        }
 
 #ifndef CFG_NO_FLASH
@@ -463,25 +723,28 @@ static ulong fdt_relocate (ulong alloc_current,
                relocate = 1;
 #endif
 
-#ifdef CFG_BOOTMAPSZ
        /*
         * The blob must be within CFG_BOOTMAPSZ,
         * so we flag it to be copied if it is not.
         */
        if (fdt_blob >= (char *)CFG_BOOTMAPSZ)
                relocate = 1;
-#endif
+
+       of_len = be32_to_cpu (fdt_totalsize (fdt_blob));
 
        /* move flattend device tree if needed */
        if (relocate) {
                int err;
-               ulong of_start, of_len;
-
-               of_len = *of_size;
+               ulong of_start;
 
                /* position on a 4K boundary before the alloc_current */
-               of_start  = alloc_current - of_len;
-               of_start &= ~(4096 - 1);        /* align on page */
+               of_start = lmb_alloc_base(lmb, of_len, 0x1000,
+                                        (CFG_BOOTMAPSZ + bootmap_base));
+
+               if (of_start == 0) {
+                       puts("device tree - allocation error\n");
+                       goto error;
+               }
 
                debug ("## device tree at 0x%08lX ... 0x%08lX (len=%ld=0x%lX)\n",
                        (ulong)fdt_blob, (ulong)fdt_blob + of_len - 1,
@@ -493,17 +756,19 @@ static ulong fdt_relocate (ulong alloc_current,
                err = fdt_open_into (fdt_blob, (void *)of_start, of_len);
                if (err != 0) {
                        fdt_error ("fdt move failed");
-                       do_reset (cmdtp, flag, argc, argv);
+                       goto error;
                }
                puts ("OK\n");
 
                *of_flat_tree = (char *)of_start;
-               new_alloc_current = of_start;
        } else {
                *of_flat_tree = fdt_blob;
-               new_alloc_current = alloc_current;
+               lmb_reserve(lmb, (ulong)fdt, of_len);
        }
 
-       return new_alloc_current;
+       return 0;
+
+error:
+       return 1;
 }
 #endif