3 files changed, 14 insertions, 13 deletions

diff --git a/arch/powerpc/boot/flatdevtree_misc.c b/arch/powerpc/boot/flatdevtree_misc.c
index c7f9adbf827d..04da38fa477f 100644
--- a/arch/powerpc/boot/flatdevtree_misc.c
+++ b/arch/powerpc/boot/flatdevtree_misc.c
@@ -33,13 +33,9 @@ static int ft_setprop(const void *phandle, const char *propname,
         return ft_set_prop(&cxt, phandle, propname, buf, buflen);
 }
 
-static void ft_pack(void)
+static unsigned long ft_finalize(void)
 {
         ft_end_tree(&cxt);
-}
-
-static unsigned long ft_addr(void)
-{
         return (unsigned long)cxt.bph;
 }
 
@@ -48,8 +44,7 @@ int ft_init(void *dt_blob, unsigned int max_size, unsigned int max_find_device)
         dt_ops.finddevice = ft_finddevice;
         dt_ops.getprop = ft_getprop;
         dt_ops.setprop = ft_setprop;
-        dt_ops.ft_pack = ft_pack;
-        dt_ops.ft_addr = ft_addr;
+        dt_ops.finalize = ft_finalize;
 
         return ft_open(&cxt, dt_blob, max_size, max_find_device,
                         platform_ops.realloc);
diff --git a/arch/powerpc/boot/main.c b/arch/powerpc/boot/main.c
index 630a453e16fa..6f6b50d238b6 100644
--- a/arch/powerpc/boot/main.c
+++ b/arch/powerpc/boot/main.c
@@ -298,6 +298,7 @@ void start(unsigned long a1, unsigned long a2, void *promptr, void *sp)
 {
         kernel_entry_t kentry;
         char cmdline[COMMAND_LINE_SIZE];
+        unsigned long ft_addr = 0;
 
         memset(__bss_start, 0, _end - __bss_start);
         memset(&platform_ops, 0, sizeof(platform_ops));
@@ -328,14 +329,20 @@ void start(unsigned long a1, unsigned long a2, void *promptr, void *sp)
                 set_cmdline(cmdline);
         }
 
+        printf("Finalizing device tree...");
+        if (dt_ops.finalize)
+                ft_addr = dt_ops.finalize();
+        if (ft_addr)
+                printf(" flat tree at 0x%lx\n\r", ft_addr);
+        else
+                printf(" using OF tree (promptr=%p)\n\r", promptr);
+
         if (console_ops.close)
                 console_ops.close();
 
         kentry = (kernel_entry_t) vmlinux.addr;
-        if (_dtb_end > _dtb_start) {
-                dt_ops.ft_pack();
-                kentry(dt_ops.ft_addr(), 0, NULL);
-        }
+        if (ft_addr)
+                kentry(ft_addr, 0, NULL);
         else
                 /* XXX initrd addr/size should be passed in properties */
                 kentry(initrd.addr, initrd.size, promptr);
diff --git a/arch/powerpc/boot/ops.h b/arch/powerpc/boot/ops.h
index 59832fb0f276..8abb6516bb7c 100644
--- a/arch/powerpc/boot/ops.h
+++ b/arch/powerpc/boot/ops.h
@@ -35,8 +35,7 @@ struct dt_ops {
                         const int buflen);
         int     (*setprop)(const void *phandle, const char *name,
                         const void *buf, const int buflen);
-        void    (*ft_pack)(void);
-        unsigned long (*ft_addr)(void);
+        unsigned long (*finalize)(void);
 };
 extern struct dt_ops dt_ops;