ftrace/x86: Have x86 ftrace use the ftrace_modify_all_code()

To remove duplicate code, have the ftrace arch_ftrace_update_code()
use the generic ftrace_modify_all_code(). This requires that the
default ftrace_replace_code() becomes a weak function so that an
arch may override it.

Signed-off-by: Steven Rostedt <rostedt@goodmis.org>

1 files changed, 2 insertions, 13 deletions

diff --git a/arch/x86/kernel/ftrace.c b/arch/x86/kernel/ftrace.c
index 4243e8bbdcb1..32ff36596ab1 100644
--- a/arch/x86/kernel/ftrace.c
+++ b/arch/x86/kernel/ftrace.c
@@ -435,7 +435,7 @@ static void run_sync(void)
                 local_irq_disable();
 }
 
-static void ftrace_replace_code(int enable)
+void ftrace_replace_code(int enable)
 {
         struct ftrace_rec_iter *iter;
         struct dyn_ftrace *rec;
@@ -493,18 +493,7 @@ void arch_ftrace_update_code(int command)
 {
         modifying_ftrace_code++;
 
-        if (command & FTRACE_UPDATE_CALLS)
-                ftrace_replace_code(1);
-        else if (command & FTRACE_DISABLE_CALLS)
-                ftrace_replace_code(0);
-
-        if (command & FTRACE_UPDATE_TRACE_FUNC)
-                ftrace_update_ftrace_func(ftrace_trace_function);
-
-        if (command & FTRACE_START_FUNC_RET)
-                ftrace_enable_ftrace_graph_caller();
-        else if (command & FTRACE_STOP_FUNC_RET)
-                ftrace_disable_ftrace_graph_caller();
+        ftrace_modify_all_code(command);
 
         modifying_ftrace_code--;
 }