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/events/internal.h | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/kernel/events/internal.h b/kernel/events/internal.h index 241a2318bfdc..8e4215f8cb93 100644 --- a/kernel/events/internal.h +++ b/kernel/events/internal.h @@ -219,6 +219,18 @@ enum { 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(); @@ -229,6 +241,7 @@ static __always_inline int trace_get_context_bit(void) else return TRACE_CTX_NORMAL; } +#endif /* ARCH_HAS_TASK_CONTEXT */ static inline int get_recursion_context(int *recursion) { -- 2.20.1
