return NULL;
 }
 
+/*
+ * This and enable_trace_probe/disable_trace_probe rely on event_mutex
+ * held by the caller, __ftrace_set_clr_event().
+ */
 static int trace_probe_nr_files(struct trace_probe *tp)
 {
-       struct ftrace_event_file **file;
+       struct ftrace_event_file **file = rcu_dereference_raw(tp->files);
        int ret = 0;
 
-       /*
-        * Since all tp->files updater is protected by probe_enable_lock,
-        * we don't need to lock an rcu_read_lock.
-        */
-       file = rcu_dereference_raw(tp->files);
        if (file)
                while (*(file++))
                        ret++;
        return ret;
 }
 
-static DEFINE_MUTEX(probe_enable_lock);
-
 /*
  * Enable trace_probe
  * if the file is NULL, enable "perf" handler, or enable "trace" handler.
 {
        int ret = 0;
 
-       mutex_lock(&probe_enable_lock);
-
        if (file) {
                struct ftrace_event_file **new, **old;
                int n = trace_probe_nr_files(tp);
                              GFP_KERNEL);
                if (!new) {
                        ret = -ENOMEM;
-                       goto out_unlock;
+                       goto out;
                }
                memcpy(new, old, n * sizeof(struct ftrace_event_file *));
                new[n] = file;
                else
                        ret = enable_kprobe(&tp->rp.kp);
        }
-
- out_unlock:
-       mutex_unlock(&probe_enable_lock);
-
+ out:
        return ret;
 }
 
 {
        int ret = 0;
 
-       mutex_lock(&probe_enable_lock);
-
        if (file) {
                struct ftrace_event_file **new, **old;
                int n = trace_probe_nr_files(tp);
                old = rcu_dereference_raw(tp->files);
                if (n == 0 || trace_probe_file_index(tp, file) < 0) {
                        ret = -EINVAL;
-                       goto out_unlock;
+                       goto out;
                }
 
                if (n == 1) {   /* Remove the last file */
                                      GFP_KERNEL);
                        if (!new) {
                                ret = -ENOMEM;
-                               goto out_unlock;
+                               goto out;
                        }
 
                        /* This copy & check loop copies the NULL stopper too */
                else
                        disable_kprobe(&tp->rp.kp);
        }
-
- out_unlock:
-       mutex_unlock(&probe_enable_lock);
-
+ out:
        return ret;
 }
 
 }
 #endif /* CONFIG_PERF_EVENTS */
 
+/*
+ * called by perf_trace_init() or __ftrace_set_clr_event() under event_mutex.
+ *
+ * kprobe_trace_self_tests_init() does enable_trace_probe/disable_trace_probe
+ * lockless, but we can't race with this __init function.
+ */
 static __kprobes
 int kprobe_register(struct ftrace_event_call *event,
                    enum trace_reg type, void *data)
        return NULL;
 }
 
+/*
+ * Nobody but us can call enable_trace_probe/disable_trace_probe at this
+ * stage, we can do this lockless.
+ */
 static __init int kprobe_trace_self_tests_init(void)
 {
        int ret, warn = 0;