Use the early task context tracking to detect the current context, if the arch supports it.
Signed-off-by: Daniel Bristot de Oliveira <[email protected]> Cc: Steven Rostedt <[email protected]> Cc: Arnaldo Carvalho de Melo <[email protected]> Cc: Ingo Molnar <[email protected]> Cc: Andy Lutomirski <[email protected]> Cc: Thomas Gleixner <[email protected]> Cc: Borislav Petkov <[email protected]> Cc: Peter Zijlstra <[email protected]> Cc: "H. Peter Anvin" <[email protected]> Cc: "Joel Fernandes (Google)" <[email protected]> Cc: Jiri Olsa <[email protected]> Cc: Namhyung Kim <[email protected]> Cc: Alexander Shishkin <[email protected]> Cc: Tommaso Cucinotta <[email protected]> Cc: Romulo Silva de Oliveira <[email protected]> Cc: Clark Williams <[email protected]> Cc: [email protected] Cc: [email protected] --- kernel/trace/trace.h | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/kernel/trace/trace.h b/kernel/trace/trace.h index 09318748fab8..62ba4bd0e436 100644 --- a/kernel/trace/trace.h +++ b/kernel/trace/trace.h @@ -632,7 +632,18 @@ enum { TRACE_CTX_NORMAL, TRACE_CTX_MAX }; +#ifdef ARCH_HAS_TASK_CONTEXT +static __always_inline int trace_get_context_bit(void) +{ + unsigned long tc = this_cpu_read(task_context); + if (tc) + return tc & TASK_CTX_NMI ? TRACE_CTX_NMI : + tc & TASK_CTX_IRQ ? TRACE_CTX_IRQ : TRACE_CTX_SOFTIRQ; + else + return TRACE_CTX_NORMAL; +} +#else /* ARCH_HAS_TASK_CONTEXT */ static __always_inline int trace_get_context_bit(void) { unsigned long pc = preempt_count(); @@ -643,6 +654,7 @@ static __always_inline int trace_get_context_bit(void) else return TRACE_CTX_NORMAL; } +#endif /* ARCH_HAS_TASK_CONTEXT */ static __always_inline int trace_test_and_set_recursion(int start, int max) { -- 2.20.1
