ftrace: prevent ftrace modifications while being kprobe'd, v2

add two missing chunks for ftrace+kprobe.

Signed-off-by: Abhishek Sagar <sagar.abhishek@gmail.com>
Cc: Steven Rostedt <rostedt@goodmis.org>
Signed-off-by: Ingo Molnar <mingo@elte.hu>

1 files changed, 8 insertions, 1 deletions

diff --git a/kernel/trace/ftrace.c b/kernel/trace/ftrace.c
index 85e841335417..0f271c45cd02 100644
--- a/kernel/trace/ftrace.c
+++ b/kernel/trace/ftrace.c
@@ -502,8 +502,12 @@ static void ftrace_replace_code(int enable)
                                 continue;
 
                         /* ignore updates to this record's mcount site */
-                        if (get_kprobe((void *)rec->ip))
+                        if (get_kprobe((void *)rec->ip)) {
+                                freeze_record(rec);
                                 continue;
+                        } else {
+                                unfreeze_record(rec);
+                        }
 
                         failed = __ftrace_replace_code(rec, old, new, enable);
                         if (failed && (rec->flags & FTRACE_FL_CONVERTED)) {
@@ -740,7 +744,10 @@ static int __ftrace_update_code(void *ignore)
                                 ftrace_del_hash(p);
                                 INIT_HLIST_NODE(&p->node);
                                 hlist_add_head(&p->node, &temp_list);
+                                freeze_record(p);
                                 continue;
+                        } else {
+                                unfreeze_record(p);
                         }
 
                         /* convert record (i.e, patch mcount-call with NOP) */