1 files changed, 4 insertions, 0 deletions

diff --git a/kernel/trace/trace_branch.c b/kernel/trace/trace_branch.c
index 4ad967453b6f..3ea65cdff30d 100644
--- a/kernel/trace/trace_branch.c
+++ b/kernel/trace/trace_branch.c
@@ -205,6 +205,8 @@ void trace_likely_condition(struct ftrace_likely_data *f, int val, int expect)
 void ftrace_likely_update(struct ftrace_likely_data *f, int val,
                           int expect, int is_constant)
 {
+        unsigned long flags = user_access_save();
+
         /* A constant is always correct */
         if (is_constant) {
                 f->constant++;
@@ -223,6 +225,8 @@ void ftrace_likely_update(struct ftrace_likely_data *f, int val,
                 f->data.correct++;
         else
                 f->data.incorrect++;
+
+        user_access_restore(flags);
 }
 EXPORT_SYMBOL(ftrace_likely_update);