Loading...
1// SPDX-License-Identifier: GPL-2.0
2
3#include <linux/context_tracking.h>
4#include <linux/entry-common.h>
5#include <linux/livepatch.h>
6#include <linux/audit.h>
7
8#define CREATE_TRACE_POINTS
9#include <trace/events/syscalls.h>
10
11/**
12 * enter_from_user_mode - Establish state when coming from user mode
13 *
14 * Syscall/interrupt entry disables interrupts, but user mode is traced as
15 * interrupts enabled. Also with NO_HZ_FULL RCU might be idle.
16 *
17 * 1) Tell lockdep that interrupts are disabled
18 * 2) Invoke context tracking if enabled to reactivate RCU
19 * 3) Trace interrupts off state
20 */
21static __always_inline void enter_from_user_mode(struct pt_regs *regs)
22{
23 arch_check_user_regs(regs);
24 lockdep_hardirqs_off(CALLER_ADDR0);
25
26 CT_WARN_ON(ct_state() != CONTEXT_USER);
27 user_exit_irqoff();
28
29 instrumentation_begin();
30 trace_hardirqs_off_finish();
31 instrumentation_end();
32}
33
34static inline void syscall_enter_audit(struct pt_regs *regs, long syscall)
35{
36 if (unlikely(audit_context())) {
37 unsigned long args[6];
38
39 syscall_get_arguments(current, regs, args);
40 audit_syscall_entry(syscall, args[0], args[1], args[2], args[3]);
41 }
42}
43
44static long syscall_trace_enter(struct pt_regs *regs, long syscall,
45 unsigned long ti_work)
46{
47 long ret = 0;
48
49 /* Handle ptrace */
50 if (ti_work & (_TIF_SYSCALL_TRACE | _TIF_SYSCALL_EMU)) {
51 ret = arch_syscall_enter_tracehook(regs);
52 if (ret || (ti_work & _TIF_SYSCALL_EMU))
53 return -1L;
54 }
55
56 /* Do seccomp after ptrace, to catch any tracer changes. */
57 if (ti_work & _TIF_SECCOMP) {
58 ret = __secure_computing(NULL);
59 if (ret == -1L)
60 return ret;
61 }
62
63 /* Either of the above might have changed the syscall number */
64 syscall = syscall_get_nr(current, regs);
65
66 if (unlikely(ti_work & _TIF_SYSCALL_TRACEPOINT))
67 trace_sys_enter(regs, syscall);
68
69 syscall_enter_audit(regs, syscall);
70
71 return ret ? : syscall;
72}
73
74static __always_inline long
75__syscall_enter_from_user_work(struct pt_regs *regs, long syscall)
76{
77 unsigned long ti_work;
78
79 ti_work = READ_ONCE(current_thread_info()->flags);
80 if (ti_work & SYSCALL_ENTER_WORK)
81 syscall = syscall_trace_enter(regs, syscall, ti_work);
82
83 return syscall;
84}
85
86long syscall_enter_from_user_mode_work(struct pt_regs *regs, long syscall)
87{
88 return __syscall_enter_from_user_work(regs, syscall);
89}
90
91noinstr long syscall_enter_from_user_mode(struct pt_regs *regs, long syscall)
92{
93 long ret;
94
95 enter_from_user_mode(regs);
96
97 instrumentation_begin();
98 local_irq_enable();
99 ret = __syscall_enter_from_user_work(regs, syscall);
100 instrumentation_end();
101
102 return ret;
103}
104
105noinstr void syscall_enter_from_user_mode_prepare(struct pt_regs *regs)
106{
107 enter_from_user_mode(regs);
108 instrumentation_begin();
109 local_irq_enable();
110 instrumentation_end();
111}
112
113/**
114 * exit_to_user_mode - Fixup state when exiting to user mode
115 *
116 * Syscall/interupt exit enables interrupts, but the kernel state is
117 * interrupts disabled when this is invoked. Also tell RCU about it.
118 *
119 * 1) Trace interrupts on state
120 * 2) Invoke context tracking if enabled to adjust RCU state
121 * 3) Invoke architecture specific last minute exit code, e.g. speculation
122 * mitigations, etc.
123 * 4) Tell lockdep that interrupts are enabled
124 */
125static __always_inline void exit_to_user_mode(void)
126{
127 instrumentation_begin();
128 trace_hardirqs_on_prepare();
129 lockdep_hardirqs_on_prepare(CALLER_ADDR0);
130 instrumentation_end();
131
132 user_enter_irqoff();
133 arch_exit_to_user_mode();
134 lockdep_hardirqs_on(CALLER_ADDR0);
135}
136
137/* Workaround to allow gradual conversion of architecture code */
138void __weak arch_do_signal(struct pt_regs *regs) { }
139
140static unsigned long exit_to_user_mode_loop(struct pt_regs *regs,
141 unsigned long ti_work)
142{
143 /*
144 * Before returning to user space ensure that all pending work
145 * items have been completed.
146 */
147 while (ti_work & EXIT_TO_USER_MODE_WORK) {
148
149 local_irq_enable_exit_to_user(ti_work);
150
151 if (ti_work & _TIF_NEED_RESCHED)
152 schedule();
153
154 if (ti_work & _TIF_UPROBE)
155 uprobe_notify_resume(regs);
156
157 if (ti_work & _TIF_PATCH_PENDING)
158 klp_update_patch_state(current);
159
160 if (ti_work & _TIF_SIGPENDING)
161 arch_do_signal(regs);
162
163 if (ti_work & _TIF_NOTIFY_RESUME) {
164 clear_thread_flag(TIF_NOTIFY_RESUME);
165 tracehook_notify_resume(regs);
166 rseq_handle_notify_resume(NULL, regs);
167 }
168
169 /* Architecture specific TIF work */
170 arch_exit_to_user_mode_work(regs, ti_work);
171
172 /*
173 * Disable interrupts and reevaluate the work flags as they
174 * might have changed while interrupts and preemption was
175 * enabled above.
176 */
177 local_irq_disable_exit_to_user();
178 ti_work = READ_ONCE(current_thread_info()->flags);
179 }
180
181 /* Return the latest work state for arch_exit_to_user_mode() */
182 return ti_work;
183}
184
185static void exit_to_user_mode_prepare(struct pt_regs *regs)
186{
187 unsigned long ti_work = READ_ONCE(current_thread_info()->flags);
188
189 lockdep_assert_irqs_disabled();
190
191 if (unlikely(ti_work & EXIT_TO_USER_MODE_WORK))
192 ti_work = exit_to_user_mode_loop(regs, ti_work);
193
194 arch_exit_to_user_mode_prepare(regs, ti_work);
195
196 /* Ensure that the address limit is intact and no locks are held */
197 addr_limit_user_check();
198 lockdep_assert_irqs_disabled();
199 lockdep_sys_exit();
200}
201
202#ifndef _TIF_SINGLESTEP
203static inline bool report_single_step(unsigned long ti_work)
204{
205 return false;
206}
207#else
208/*
209 * If TIF_SYSCALL_EMU is set, then the only reason to report is when
210 * TIF_SINGLESTEP is set (i.e. PTRACE_SYSEMU_SINGLESTEP). This syscall
211 * instruction has been already reported in syscall_enter_from_usermode().
212 */
213#define SYSEMU_STEP (_TIF_SINGLESTEP | _TIF_SYSCALL_EMU)
214
215static inline bool report_single_step(unsigned long ti_work)
216{
217 return (ti_work & SYSEMU_STEP) == _TIF_SINGLESTEP;
218}
219#endif
220
221static void syscall_exit_work(struct pt_regs *regs, unsigned long ti_work)
222{
223 bool step;
224
225 audit_syscall_exit(regs);
226
227 if (ti_work & _TIF_SYSCALL_TRACEPOINT)
228 trace_sys_exit(regs, syscall_get_return_value(current, regs));
229
230 step = report_single_step(ti_work);
231 if (step || ti_work & _TIF_SYSCALL_TRACE)
232 arch_syscall_exit_tracehook(regs, step);
233}
234
235/*
236 * Syscall specific exit to user mode preparation. Runs with interrupts
237 * enabled.
238 */
239static void syscall_exit_to_user_mode_prepare(struct pt_regs *regs)
240{
241 u32 cached_flags = READ_ONCE(current_thread_info()->flags);
242 unsigned long nr = syscall_get_nr(current, regs);
243
244 CT_WARN_ON(ct_state() != CONTEXT_KERNEL);
245
246 if (IS_ENABLED(CONFIG_PROVE_LOCKING)) {
247 if (WARN(irqs_disabled(), "syscall %lu left IRQs disabled", nr))
248 local_irq_enable();
249 }
250
251 rseq_syscall(regs);
252
253 /*
254 * Do one-time syscall specific work. If these work items are
255 * enabled, we want to run them exactly once per syscall exit with
256 * interrupts enabled.
257 */
258 if (unlikely(cached_flags & SYSCALL_EXIT_WORK))
259 syscall_exit_work(regs, cached_flags);
260}
261
262__visible noinstr void syscall_exit_to_user_mode(struct pt_regs *regs)
263{
264 instrumentation_begin();
265 syscall_exit_to_user_mode_prepare(regs);
266 local_irq_disable_exit_to_user();
267 exit_to_user_mode_prepare(regs);
268 instrumentation_end();
269 exit_to_user_mode();
270}
271
272noinstr void irqentry_enter_from_user_mode(struct pt_regs *regs)
273{
274 enter_from_user_mode(regs);
275}
276
277noinstr void irqentry_exit_to_user_mode(struct pt_regs *regs)
278{
279 instrumentation_begin();
280 exit_to_user_mode_prepare(regs);
281 instrumentation_end();
282 exit_to_user_mode();
283}
284
285noinstr irqentry_state_t irqentry_enter(struct pt_regs *regs)
286{
287 irqentry_state_t ret = {
288 .exit_rcu = false,
289 };
290
291 if (user_mode(regs)) {
292 irqentry_enter_from_user_mode(regs);
293 return ret;
294 }
295
296 /*
297 * If this entry hit the idle task invoke rcu_irq_enter() whether
298 * RCU is watching or not.
299 *
300 * Interupts can nest when the first interrupt invokes softirq
301 * processing on return which enables interrupts.
302 *
303 * Scheduler ticks in the idle task can mark quiescent state and
304 * terminate a grace period, if and only if the timer interrupt is
305 * not nested into another interrupt.
306 *
307 * Checking for __rcu_is_watching() here would prevent the nesting
308 * interrupt to invoke rcu_irq_enter(). If that nested interrupt is
309 * the tick then rcu_flavor_sched_clock_irq() would wrongfully
310 * assume that it is the first interupt and eventually claim
311 * quiescient state and end grace periods prematurely.
312 *
313 * Unconditionally invoke rcu_irq_enter() so RCU state stays
314 * consistent.
315 *
316 * TINY_RCU does not support EQS, so let the compiler eliminate
317 * this part when enabled.
318 */
319 if (!IS_ENABLED(CONFIG_TINY_RCU) && is_idle_task(current)) {
320 /*
321 * If RCU is not watching then the same careful
322 * sequence vs. lockdep and tracing is required
323 * as in irq_enter_from_user_mode().
324 */
325 lockdep_hardirqs_off(CALLER_ADDR0);
326 rcu_irq_enter();
327 instrumentation_begin();
328 trace_hardirqs_off_finish();
329 instrumentation_end();
330
331 ret.exit_rcu = true;
332 return ret;
333 }
334
335 /*
336 * If RCU is watching then RCU only wants to check whether it needs
337 * to restart the tick in NOHZ mode. rcu_irq_enter_check_tick()
338 * already contains a warning when RCU is not watching, so no point
339 * in having another one here.
340 */
341 instrumentation_begin();
342 rcu_irq_enter_check_tick();
343 /* Use the combo lockdep/tracing function */
344 trace_hardirqs_off();
345 instrumentation_end();
346
347 return ret;
348}
349
350void irqentry_exit_cond_resched(void)
351{
352 if (!preempt_count()) {
353 /* Sanity check RCU and thread stack */
354 rcu_irq_exit_check_preempt();
355 if (IS_ENABLED(CONFIG_DEBUG_ENTRY))
356 WARN_ON_ONCE(!on_thread_stack());
357 if (need_resched())
358 preempt_schedule_irq();
359 }
360}
361
362noinstr void irqentry_exit(struct pt_regs *regs, irqentry_state_t state)
363{
364 lockdep_assert_irqs_disabled();
365
366 /* Check whether this returns to user mode */
367 if (user_mode(regs)) {
368 irqentry_exit_to_user_mode(regs);
369 } else if (!regs_irqs_disabled(regs)) {
370 /*
371 * If RCU was not watching on entry this needs to be done
372 * carefully and needs the same ordering of lockdep/tracing
373 * and RCU as the return to user mode path.
374 */
375 if (state.exit_rcu) {
376 instrumentation_begin();
377 /* Tell the tracer that IRET will enable interrupts */
378 trace_hardirqs_on_prepare();
379 lockdep_hardirqs_on_prepare(CALLER_ADDR0);
380 instrumentation_end();
381 rcu_irq_exit();
382 lockdep_hardirqs_on(CALLER_ADDR0);
383 return;
384 }
385
386 instrumentation_begin();
387 if (IS_ENABLED(CONFIG_PREEMPTION))
388 irqentry_exit_cond_resched();
389 /* Covers both tracing and lockdep */
390 trace_hardirqs_on();
391 instrumentation_end();
392 } else {
393 /*
394 * IRQ flags state is correct already. Just tell RCU if it
395 * was not watching on entry.
396 */
397 if (state.exit_rcu)
398 rcu_irq_exit();
399 }
400}
1// SPDX-License-Identifier: GPL-2.0
2
3#include <linux/context_tracking.h>
4#include <linux/entry-common.h>
5#include <linux/resume_user_mode.h>
6#include <linux/highmem.h>
7#include <linux/jump_label.h>
8#include <linux/kmsan.h>
9#include <linux/livepatch.h>
10#include <linux/audit.h>
11#include <linux/tick.h>
12
13#include "common.h"
14
15#define CREATE_TRACE_POINTS
16#include <trace/events/syscalls.h>
17
18static inline void syscall_enter_audit(struct pt_regs *regs, long syscall)
19{
20 if (unlikely(audit_context())) {
21 unsigned long args[6];
22
23 syscall_get_arguments(current, regs, args);
24 audit_syscall_entry(syscall, args[0], args[1], args[2], args[3]);
25 }
26}
27
28long syscall_trace_enter(struct pt_regs *regs, long syscall,
29 unsigned long work)
30{
31 long ret = 0;
32
33 /*
34 * Handle Syscall User Dispatch. This must comes first, since
35 * the ABI here can be something that doesn't make sense for
36 * other syscall_work features.
37 */
38 if (work & SYSCALL_WORK_SYSCALL_USER_DISPATCH) {
39 if (syscall_user_dispatch(regs))
40 return -1L;
41 }
42
43 /* Handle ptrace */
44 if (work & (SYSCALL_WORK_SYSCALL_TRACE | SYSCALL_WORK_SYSCALL_EMU)) {
45 ret = ptrace_report_syscall_entry(regs);
46 if (ret || (work & SYSCALL_WORK_SYSCALL_EMU))
47 return -1L;
48 }
49
50 /* Do seccomp after ptrace, to catch any tracer changes. */
51 if (work & SYSCALL_WORK_SECCOMP) {
52 ret = __secure_computing(NULL);
53 if (ret == -1L)
54 return ret;
55 }
56
57 /* Either of the above might have changed the syscall number */
58 syscall = syscall_get_nr(current, regs);
59
60 if (unlikely(work & SYSCALL_WORK_SYSCALL_TRACEPOINT)) {
61 trace_sys_enter(regs, syscall);
62 /*
63 * Probes or BPF hooks in the tracepoint may have changed the
64 * system call number as well.
65 */
66 syscall = syscall_get_nr(current, regs);
67 }
68
69 syscall_enter_audit(regs, syscall);
70
71 return ret ? : syscall;
72}
73
74noinstr void syscall_enter_from_user_mode_prepare(struct pt_regs *regs)
75{
76 enter_from_user_mode(regs);
77 instrumentation_begin();
78 local_irq_enable();
79 instrumentation_end();
80}
81
82/* Workaround to allow gradual conversion of architecture code */
83void __weak arch_do_signal_or_restart(struct pt_regs *regs) { }
84
85/**
86 * exit_to_user_mode_loop - do any pending work before leaving to user space
87 * @regs: Pointer to pt_regs on entry stack
88 * @ti_work: TIF work flags as read by the caller
89 */
90__always_inline unsigned long exit_to_user_mode_loop(struct pt_regs *regs,
91 unsigned long ti_work)
92{
93 /*
94 * Before returning to user space ensure that all pending work
95 * items have been completed.
96 */
97 while (ti_work & EXIT_TO_USER_MODE_WORK) {
98
99 local_irq_enable_exit_to_user(ti_work);
100
101 if (ti_work & _TIF_NEED_RESCHED)
102 schedule();
103
104 if (ti_work & _TIF_UPROBE)
105 uprobe_notify_resume(regs);
106
107 if (ti_work & _TIF_PATCH_PENDING)
108 klp_update_patch_state(current);
109
110 if (ti_work & (_TIF_SIGPENDING | _TIF_NOTIFY_SIGNAL))
111 arch_do_signal_or_restart(regs);
112
113 if (ti_work & _TIF_NOTIFY_RESUME)
114 resume_user_mode_work(regs);
115
116 /* Architecture specific TIF work */
117 arch_exit_to_user_mode_work(regs, ti_work);
118
119 /*
120 * Disable interrupts and reevaluate the work flags as they
121 * might have changed while interrupts and preemption was
122 * enabled above.
123 */
124 local_irq_disable_exit_to_user();
125
126 /* Check if any of the above work has queued a deferred wakeup */
127 tick_nohz_user_enter_prepare();
128
129 ti_work = read_thread_flags();
130 }
131
132 /* Return the latest work state for arch_exit_to_user_mode() */
133 return ti_work;
134}
135
136/*
137 * If SYSCALL_EMU is set, then the only reason to report is when
138 * SINGLESTEP is set (i.e. PTRACE_SYSEMU_SINGLESTEP). This syscall
139 * instruction has been already reported in syscall_enter_from_user_mode().
140 */
141static inline bool report_single_step(unsigned long work)
142{
143 if (work & SYSCALL_WORK_SYSCALL_EMU)
144 return false;
145
146 return work & SYSCALL_WORK_SYSCALL_EXIT_TRAP;
147}
148
149static void syscall_exit_work(struct pt_regs *regs, unsigned long work)
150{
151 bool step;
152
153 /*
154 * If the syscall was rolled back due to syscall user dispatching,
155 * then the tracers below are not invoked for the same reason as
156 * the entry side was not invoked in syscall_trace_enter(): The ABI
157 * of these syscalls is unknown.
158 */
159 if (work & SYSCALL_WORK_SYSCALL_USER_DISPATCH) {
160 if (unlikely(current->syscall_dispatch.on_dispatch)) {
161 current->syscall_dispatch.on_dispatch = false;
162 return;
163 }
164 }
165
166 audit_syscall_exit(regs);
167
168 if (work & SYSCALL_WORK_SYSCALL_TRACEPOINT)
169 trace_sys_exit(regs, syscall_get_return_value(current, regs));
170
171 step = report_single_step(work);
172 if (step || work & SYSCALL_WORK_SYSCALL_TRACE)
173 ptrace_report_syscall_exit(regs, step);
174}
175
176/*
177 * Syscall specific exit to user mode preparation. Runs with interrupts
178 * enabled.
179 */
180static void syscall_exit_to_user_mode_prepare(struct pt_regs *regs)
181{
182 unsigned long work = READ_ONCE(current_thread_info()->syscall_work);
183 unsigned long nr = syscall_get_nr(current, regs);
184
185 CT_WARN_ON(ct_state() != CONTEXT_KERNEL);
186
187 if (IS_ENABLED(CONFIG_PROVE_LOCKING)) {
188 if (WARN(irqs_disabled(), "syscall %lu left IRQs disabled", nr))
189 local_irq_enable();
190 }
191
192 rseq_syscall(regs);
193
194 /*
195 * Do one-time syscall specific work. If these work items are
196 * enabled, we want to run them exactly once per syscall exit with
197 * interrupts enabled.
198 */
199 if (unlikely(work & SYSCALL_WORK_EXIT))
200 syscall_exit_work(regs, work);
201}
202
203static __always_inline void __syscall_exit_to_user_mode_work(struct pt_regs *regs)
204{
205 syscall_exit_to_user_mode_prepare(regs);
206 local_irq_disable_exit_to_user();
207 exit_to_user_mode_prepare(regs);
208}
209
210void syscall_exit_to_user_mode_work(struct pt_regs *regs)
211{
212 __syscall_exit_to_user_mode_work(regs);
213}
214
215__visible noinstr void syscall_exit_to_user_mode(struct pt_regs *regs)
216{
217 instrumentation_begin();
218 __syscall_exit_to_user_mode_work(regs);
219 instrumentation_end();
220 exit_to_user_mode();
221}
222
223noinstr void irqentry_enter_from_user_mode(struct pt_regs *regs)
224{
225 enter_from_user_mode(regs);
226}
227
228noinstr void irqentry_exit_to_user_mode(struct pt_regs *regs)
229{
230 instrumentation_begin();
231 exit_to_user_mode_prepare(regs);
232 instrumentation_end();
233 exit_to_user_mode();
234}
235
236noinstr irqentry_state_t irqentry_enter(struct pt_regs *regs)
237{
238 irqentry_state_t ret = {
239 .exit_rcu = false,
240 };
241
242 if (user_mode(regs)) {
243 irqentry_enter_from_user_mode(regs);
244 return ret;
245 }
246
247 /*
248 * If this entry hit the idle task invoke ct_irq_enter() whether
249 * RCU is watching or not.
250 *
251 * Interrupts can nest when the first interrupt invokes softirq
252 * processing on return which enables interrupts.
253 *
254 * Scheduler ticks in the idle task can mark quiescent state and
255 * terminate a grace period, if and only if the timer interrupt is
256 * not nested into another interrupt.
257 *
258 * Checking for rcu_is_watching() here would prevent the nesting
259 * interrupt to invoke ct_irq_enter(). If that nested interrupt is
260 * the tick then rcu_flavor_sched_clock_irq() would wrongfully
261 * assume that it is the first interrupt and eventually claim
262 * quiescent state and end grace periods prematurely.
263 *
264 * Unconditionally invoke ct_irq_enter() so RCU state stays
265 * consistent.
266 *
267 * TINY_RCU does not support EQS, so let the compiler eliminate
268 * this part when enabled.
269 */
270 if (!IS_ENABLED(CONFIG_TINY_RCU) && is_idle_task(current)) {
271 /*
272 * If RCU is not watching then the same careful
273 * sequence vs. lockdep and tracing is required
274 * as in irqentry_enter_from_user_mode().
275 */
276 lockdep_hardirqs_off(CALLER_ADDR0);
277 ct_irq_enter();
278 instrumentation_begin();
279 kmsan_unpoison_entry_regs(regs);
280 trace_hardirqs_off_finish();
281 instrumentation_end();
282
283 ret.exit_rcu = true;
284 return ret;
285 }
286
287 /*
288 * If RCU is watching then RCU only wants to check whether it needs
289 * to restart the tick in NOHZ mode. rcu_irq_enter_check_tick()
290 * already contains a warning when RCU is not watching, so no point
291 * in having another one here.
292 */
293 lockdep_hardirqs_off(CALLER_ADDR0);
294 instrumentation_begin();
295 kmsan_unpoison_entry_regs(regs);
296 rcu_irq_enter_check_tick();
297 trace_hardirqs_off_finish();
298 instrumentation_end();
299
300 return ret;
301}
302
303void raw_irqentry_exit_cond_resched(void)
304{
305 if (!preempt_count()) {
306 /* Sanity check RCU and thread stack */
307 rcu_irq_exit_check_preempt();
308 if (IS_ENABLED(CONFIG_DEBUG_ENTRY))
309 WARN_ON_ONCE(!on_thread_stack());
310 if (need_resched())
311 preempt_schedule_irq();
312 }
313}
314#ifdef CONFIG_PREEMPT_DYNAMIC
315#if defined(CONFIG_HAVE_PREEMPT_DYNAMIC_CALL)
316DEFINE_STATIC_CALL(irqentry_exit_cond_resched, raw_irqentry_exit_cond_resched);
317#elif defined(CONFIG_HAVE_PREEMPT_DYNAMIC_KEY)
318DEFINE_STATIC_KEY_TRUE(sk_dynamic_irqentry_exit_cond_resched);
319void dynamic_irqentry_exit_cond_resched(void)
320{
321 if (!static_branch_unlikely(&sk_dynamic_irqentry_exit_cond_resched))
322 return;
323 raw_irqentry_exit_cond_resched();
324}
325#endif
326#endif
327
328noinstr void irqentry_exit(struct pt_regs *regs, irqentry_state_t state)
329{
330 lockdep_assert_irqs_disabled();
331
332 /* Check whether this returns to user mode */
333 if (user_mode(regs)) {
334 irqentry_exit_to_user_mode(regs);
335 } else if (!regs_irqs_disabled(regs)) {
336 /*
337 * If RCU was not watching on entry this needs to be done
338 * carefully and needs the same ordering of lockdep/tracing
339 * and RCU as the return to user mode path.
340 */
341 if (state.exit_rcu) {
342 instrumentation_begin();
343 /* Tell the tracer that IRET will enable interrupts */
344 trace_hardirqs_on_prepare();
345 lockdep_hardirqs_on_prepare();
346 instrumentation_end();
347 ct_irq_exit();
348 lockdep_hardirqs_on(CALLER_ADDR0);
349 return;
350 }
351
352 instrumentation_begin();
353 if (IS_ENABLED(CONFIG_PREEMPTION))
354 irqentry_exit_cond_resched();
355
356 /* Covers both tracing and lockdep */
357 trace_hardirqs_on();
358 instrumentation_end();
359 } else {
360 /*
361 * IRQ flags state is correct already. Just tell RCU if it
362 * was not watching on entry.
363 */
364 if (state.exit_rcu)
365 ct_irq_exit();
366 }
367}
368
369irqentry_state_t noinstr irqentry_nmi_enter(struct pt_regs *regs)
370{
371 irqentry_state_t irq_state;
372
373 irq_state.lockdep = lockdep_hardirqs_enabled();
374
375 __nmi_enter();
376 lockdep_hardirqs_off(CALLER_ADDR0);
377 lockdep_hardirq_enter();
378 ct_nmi_enter();
379
380 instrumentation_begin();
381 kmsan_unpoison_entry_regs(regs);
382 trace_hardirqs_off_finish();
383 ftrace_nmi_enter();
384 instrumentation_end();
385
386 return irq_state;
387}
388
389void noinstr irqentry_nmi_exit(struct pt_regs *regs, irqentry_state_t irq_state)
390{
391 instrumentation_begin();
392 ftrace_nmi_exit();
393 if (irq_state.lockdep) {
394 trace_hardirqs_on_prepare();
395 lockdep_hardirqs_on_prepare();
396 }
397 instrumentation_end();
398
399 ct_nmi_exit();
400 lockdep_hardirq_exit();
401 if (irq_state.lockdep)
402 lockdep_hardirqs_on(CALLER_ADDR0);
403 __nmi_exit();
404}