src-wlib := string.S crt0.S stdio.c main.c flatdevtree.c flatdevtree_misc.c \
                ns16550.c serial.c simple_alloc.c div64.S util.S \
                gunzip_util.c elf_util.c $(zlib) devtree.c oflib.c ofconsole.c \
-               44x.c ebony.c mv64x60.c mpsc.c mv64x60_i2c.c
+               44x.c ebony.c mv64x60.c mpsc.c mv64x60_i2c.c cuboot.c
 src-plat := of.c cuboot-83xx.c cuboot-85xx.c holly.c \
                cuboot-ebony.c treeboot-ebony.c prpmc2800.c
 src-boot := $(src-wlib) $(src-plat) empty.c
 
 
 #include "ops.h"
 #include "stdio.h"
+#include "cuboot.h"
 
 #define TARGET_83xx
 #include "ppcboot.h"
 
 static bd_t bd;
-extern char _end[];
 extern char _dtb_start[], _dtb_end[];
 
 static void platform_fixups(void)
 void platform_init(unsigned long r3, unsigned long r4, unsigned long r5,
                    unsigned long r6, unsigned long r7)
 {
-       unsigned long end_of_ram = bd.bi_memstart + bd.bi_memsize;
-       unsigned long avail_ram = end_of_ram - (unsigned long)_end;
-
-       memcpy(&bd, (bd_t *)r3, sizeof(bd));
-       loader_info.initrd_addr = r4;
-       loader_info.initrd_size = r4 ? r5 - r4 : 0;
-       loader_info.cmdline = (char *)r6;
-       loader_info.cmdline_len = r7 - r6;
-
-       simple_alloc_init(_end, avail_ram - 1024*1024, 32, 64);
+       CUBOOT_INIT();
        ft_init(_dtb_start, _dtb_end - _dtb_start, 32);
        serial_console_init();
        platform_ops.fixups = platform_fixups;
 
 
 #include "ops.h"
 #include "stdio.h"
+#include "cuboot.h"
 
 #define TARGET_85xx
 #include "ppcboot.h"
 
 static bd_t bd;
-extern char _end[];
 extern char _dtb_start[], _dtb_end[];
 
 static void platform_fixups(void)
 void platform_init(unsigned long r3, unsigned long r4, unsigned long r5,
                    unsigned long r6, unsigned long r7)
 {
-       unsigned long end_of_ram = bd.bi_memstart + bd.bi_memsize;
-       unsigned long avail_ram = end_of_ram - (unsigned long)_end;
-
-       memcpy(&bd, (bd_t *)r3, sizeof(bd));
-       loader_info.initrd_addr = r4;
-       loader_info.initrd_size = r4 ? r5 - r4 : 0;
-       loader_info.cmdline = (char *)r6;
-       loader_info.cmdline_len = r7 - r6;
-
-       simple_alloc_init(_end, avail_ram - 1024*1024, 32, 64);
+       CUBOOT_INIT();
        ft_init(_dtb_start, _dtb_end - _dtb_start, 32);
        serial_console_init();
        platform_ops.fixups = platform_fixups;
 
 #include "ops.h"
 #include "stdio.h"
 #include "44x.h"
+#include "cuboot.h"
 
 #define TARGET_44x
 #include "ppcboot.h"
 
 static bd_t bd;
-extern char _end[];
-
-BSS_STACK(4096);
 
 void platform_init(unsigned long r3, unsigned long r4, unsigned long r5,
                    unsigned long r6, unsigned long r7)
 {
-       unsigned long end_of_ram = bd.bi_memstart + bd.bi_memsize;
-       unsigned long avail_ram = end_of_ram - (unsigned long)_end;
-
-       memcpy(&bd, (bd_t *)r3, sizeof(bd));
-       loader_info.initrd_addr = r4;
-       loader_info.initrd_size = r4 ? r5 : 0;
-       loader_info.cmdline = (char *)r6;
-       loader_info.cmdline_len = r7 - r6;
-
-       simple_alloc_init(_end, avail_ram, 32, 64);
-
+       CUBOOT_INIT();
        ebony_init(&bd.bi_enetaddr, &bd.bi_enet1addr);
 }
 
--- /dev/null
+/*
+ * Compatibility for old (not device tree aware) U-Boot versions
+ *
+ * Author: Scott Wood <scottwood@freescale.com>
+ * Consolidated using macros by David Gibson <david@gibson.dropbear.id.au>
+ *
+ * Copyright 2007 David Gibson, IBM Corporation.
+ * Copyright (c) 2007 Freescale Semiconductor, Inc.
+ *
+ * This program is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU General Public License version 2 as published
+ * by the Free Software Foundation.
+ */
+
+#include "ops.h"
+#include "stdio.h"
+
+#include "ppcboot.h"
+
+extern char _end[];
+extern char _dtb_start[], _dtb_end[];
+
+void cuboot_init(unsigned long r4, unsigned long r5,
+                unsigned long r6, unsigned long r7,
+                unsigned long end_of_ram)
+{
+       unsigned long avail_ram = end_of_ram - (unsigned long)_end;
+
+       loader_info.initrd_addr = r4;
+       loader_info.initrd_size = r4 ? r5 - r4 : 0;
+       loader_info.cmdline = (char *)r6;
+       loader_info.cmdline_len = r7 - r6;
+
+       simple_alloc_init(_end, avail_ram - 1024*1024, 32, 64);
+}
 
--- /dev/null
+#ifndef _PPC_BOOT_CUBOOT_H_
+#define _PPC_BOOT_CUBOOT_H_
+
+void cuboot_init(unsigned long r4, unsigned long r5,
+                unsigned long r6, unsigned long r7,
+                unsigned long end_of_ram);
+
+#define CUBOOT_INIT() \
+       do { \
+               memcpy(&bd, (bd_t *)r3, sizeof(bd)); \
+               cuboot_init(r4, r5, r6, r7, bd.bi_memstart + bd.bi_memsize); \
+       } while (0)
+
+#endif /* _PPC_BOOT_CUBOOT_H_ */