1 files changed, 5 insertions, 24 deletions

diff --git a/arch/powerpc/kernel/ftrace.c b/arch/powerpc/kernel/ftrace.c
index 2d182f119d1d..1b12696cca06 100644
--- a/arch/powerpc/kernel/ftrace.c
+++ b/arch/powerpc/kernel/ftrace.c
@@ -23,25 +23,14 @@
 #include <asm/code-patching.h>
 #include <asm/ftrace.h>
 
-#ifdef CONFIG_PPC32
-# define GET_ADDR(addr) addr
-#else
-/* PowerPC64's functions are data that points to the functions */
-# define GET_ADDR(addr) (*(unsigned long *)addr)
-#endif
 
 #ifdef CONFIG_DYNAMIC_FTRACE
-static unsigned int ftrace_nop_replace(void)
-{
-        return PPC_INST_NOP;
-}
-
 static unsigned int
 ftrace_call_replace(unsigned long ip, unsigned long addr, int link)
 {
         unsigned int op;
 
-        addr = GET_ADDR(addr);
+        addr = ppc_function_entry((void *)addr);
 
         /* if (link) set op to 'bl' else 'b' */
         op = create_branch((unsigned int *)ip, addr, link ? 1 : 0);
@@ -49,14 +38,6 @@ ftrace_call_replace(unsigned long ip, unsigned long addr, int link)
         return op;
 }
 
-#ifdef CONFIG_PPC64
-# define _ASM_ALIGN     " .align 3 "
-# define _ASM_PTR       " .llong "
-#else
-# define _ASM_ALIGN     " .align 2 "
-# define _ASM_PTR       " .long "
-#endif
-
 static int
 ftrace_modify_code(unsigned long ip, unsigned int old, unsigned int new)
 {
@@ -197,7 +178,7 @@ __ftrace_make_nop(struct module *mod,
         ptr = ((unsigned long)jmp[0] << 32) + jmp[1];
 
         /* This should match what was called */
-        if (ptr != GET_ADDR(addr)) {
+        if (ptr != ppc_function_entry((void *)addr)) {
                 printk(KERN_ERR "addr does not match %lx\n", ptr);
                 return -EINVAL;
         }
@@ -328,7 +309,7 @@ int ftrace_make_nop(struct module *mod,
         if (test_24bit_addr(ip, addr)) {
                 /* within range */
                 old = ftrace_call_replace(ip, addr, 1);
-                new = ftrace_nop_replace();
+                new = PPC_INST_NOP;
                 return ftrace_modify_code(ip, old, new);
         }
 
@@ -466,7 +447,7 @@ int ftrace_make_call(struct dyn_ftrace *rec, unsigned long addr)
          */
         if (test_24bit_addr(ip, addr)) {
                 /* within range */
-                old = ftrace_nop_replace();
+                old = PPC_INST_NOP;
                 new = ftrace_call_replace(ip, addr, 1);
                 return ftrace_modify_code(ip, old, new);
         }
@@ -570,7 +551,7 @@ void prepare_ftrace_return(unsigned long *parent, unsigned long self_addr)
                 return_hooker = (unsigned long)&mod_return_to_handler;
 #endif
 
-        return_hooker = GET_ADDR(return_hooker);
+        return_hooker = ppc_function_entry((void *)return_hooker);
 
         /*
          * Protect against fault, even if it shouldn't