powerpc/boot: Compare _start against ei.loadsize instead ei.memsize

If the vmlinux binary in memory is larger than 4 MiB than it collides
with the initial boot code which is linked at 4 MiB in case of cuBoot.
If the the uncompressed image size (on disk size) is less than 4 MiB
then it would fit. The difference between those two sizes is the bss
section. In cuBoot we have the dtb embedded right after the data
section so it is very likely that the reset of the bss section (in
kernel's start up code) will overwrite the dtb blob. Therefore we
reallocate the dtb. Something similar is allready done to the initrd.

Signed-off-by: Sebastian Andrzej Siewior <bigeasy@linutronix.de>
Acked-by: David Gibson <david@gibson.dropbear.id.au>
Signed-off-by: Benjamin Herrenschmidt <benh@kernel.crashing.org>

1 files changed, 8 insertions, 8 deletions

diff --git a/arch/powerpc/boot/libfdt-wrapper.c b/arch/powerpc/boot/libfdt-wrapper.c
index c541fd8a95d..1daa73f08ba 100644
--- a/arch/powerpc/boot/libfdt-wrapper.c
+++ b/arch/powerpc/boot/libfdt-wrapper.c
@@ -165,6 +165,7 @@ static unsigned long fdt_wrapper_finalize(void)
 void fdt_init(void *blob)
 {
         int err;
+        int bufsize;
 
         dt_ops.finddevice = fdt_wrapper_finddevice;
         dt_ops.getprop = fdt_wrapper_getprop;
@@ -178,16 +179,15 @@ void fdt_init(void *blob)
 
         /* Make sure the dt blob is the right version and so forth */
         fdt = blob;
-        err = fdt_open_into(fdt, fdt, fdt_totalsize(blob));
-        if (err == -FDT_ERR_NOSPACE) {
-                int bufsize = fdt_totalsize(fdt) + 4;
-                buf = malloc(bufsize);
-                err = fdt_open_into(fdt, buf, bufsize);
-        }
+        bufsize = fdt_totalsize(fdt) + 4;
+        buf = malloc(bufsize);
+        if(!buf)
+                fatal("malloc failed. can't relocate the device tree\n\r");
+
+        err = fdt_open_into(fdt, buf, bufsize);
 
         if (err != 0)
                 fatal("fdt_init(): %s\n\r", fdt_strerror(err));
 
-        if (buf)
-                fdt = buf;
+        fdt = buf;
 }