From: Eric Dumazet <[email protected]>

Two put_user() in rseq_update_cpu_id() are replaced
by a pair of unsafe_put_user() with appropriate surroundings.

This removes one stac/clac pair on x86 in fast path.

Signed-off-by: Eric Dumazet <[email protected]>
Cc: Mathieu Desnoyers <[email protected]>
Cc: Peter Zijlstra <[email protected]>
Cc: "Paul E. McKenney" <[email protected]>
Cc: Boqun Feng <[email protected]>
Cc: Arjun Roy <[email protected]>
Cc: Ingo Molnar <[email protected]>
---
 kernel/rseq.c | 15 +++++++++++----
 1 file changed, 11 insertions(+), 4 deletions(-)

diff --git a/kernel/rseq.c b/kernel/rseq.c
index 
a4f86a9d6937cdfa2f13d1dcc9be863c1943d06f..f020f18f512a3f6241c3c9b104ce50e4d2c6188c
 100644
--- a/kernel/rseq.c
+++ b/kernel/rseq.c
@@ -84,13 +84,20 @@
 static int rseq_update_cpu_id(struct task_struct *t)
 {
        u32 cpu_id = raw_smp_processor_id();
+       struct rseq __user *rseq = t->rseq;
 
-       if (put_user(cpu_id, &t->rseq->cpu_id_start))
-               return -EFAULT;
-       if (put_user(cpu_id, &t->rseq->cpu_id))
-               return -EFAULT;
+       if (!user_write_access_begin(rseq, sizeof(*rseq)))
+               goto efault;
+       unsafe_put_user(cpu_id, &rseq->cpu_id_start, efault_end);
+       unsafe_put_user(cpu_id, &rseq->cpu_id, efault_end);
+       user_write_access_end();
        trace_rseq_update(t);
        return 0;
+
+efault_end:
+       user_write_access_end();
+efault:
+       return -EFAULT;
 }
 
 static int rseq_reset_rseq_cpu_id(struct task_struct *t)
-- 
2.31.1.295.g9ea45b61b8-goog

Reply via email to