1// SPDX-License-Identifier: GPL-2.0
2/*
3 * preemptoff and irqoff tracepoints
4 *
5 * Copyright (C) Joel Fernandes (Google) <joel@joelfernandes.org>
6 */
7
8#include <linux/kallsyms.h>
9#include <linux/uaccess.h>
10#include <linux/module.h>
11#include <linux/ftrace.h>
12#include <linux/kprobes.h>
13#include "trace.h"
14
15#define CREATE_TRACE_POINTS
16#include <trace/events/preemptirq.h>
17
18#ifdef CONFIG_TRACE_IRQFLAGS
19/* Per-cpu variable to prevent redundant calls when IRQs already off */
20static DEFINE_PER_CPU(int, tracing_irq_cpu);
21
22void trace_hardirqs_on(void)
23{
24 if (this_cpu_read(tracing_irq_cpu)) {
25 if (!in_nmi())
26 trace_irq_enable_rcuidle(CALLER_ADDR0, CALLER_ADDR1);
27 tracer_hardirqs_on(CALLER_ADDR0, CALLER_ADDR1);
28 this_cpu_write(tracing_irq_cpu, 0);
29 }
30
31 lockdep_hardirqs_on(CALLER_ADDR0);
32}
33EXPORT_SYMBOL(trace_hardirqs_on);
34NOKPROBE_SYMBOL(trace_hardirqs_on);
35
36void trace_hardirqs_off(void)
37{
38 if (!this_cpu_read(tracing_irq_cpu)) {
39 this_cpu_write(tracing_irq_cpu, 1);
40 tracer_hardirqs_off(CALLER_ADDR0, CALLER_ADDR1);
41 if (!in_nmi())
42 trace_irq_disable_rcuidle(CALLER_ADDR0, CALLER_ADDR1);
43 }
44
45 lockdep_hardirqs_off(CALLER_ADDR0);
46}
47EXPORT_SYMBOL(trace_hardirqs_off);
48NOKPROBE_SYMBOL(trace_hardirqs_off);
49
50__visible void trace_hardirqs_on_caller(unsigned long caller_addr)
51{
52 if (this_cpu_read(tracing_irq_cpu)) {
53 if (!in_nmi())
54 trace_irq_enable_rcuidle(CALLER_ADDR0, caller_addr);
55 tracer_hardirqs_on(CALLER_ADDR0, caller_addr);
56 this_cpu_write(tracing_irq_cpu, 0);
57 }
58
59 lockdep_hardirqs_on(CALLER_ADDR0);
60}
61EXPORT_SYMBOL(trace_hardirqs_on_caller);
62NOKPROBE_SYMBOL(trace_hardirqs_on_caller);
63
64__visible void trace_hardirqs_off_caller(unsigned long caller_addr)
65{
66 if (!this_cpu_read(tracing_irq_cpu)) {
67 this_cpu_write(tracing_irq_cpu, 1);
68 tracer_hardirqs_off(CALLER_ADDR0, caller_addr);
69 if (!in_nmi())
70 trace_irq_disable_rcuidle(CALLER_ADDR0, caller_addr);
71 }
72
73 lockdep_hardirqs_off(CALLER_ADDR0);
74}
75EXPORT_SYMBOL(trace_hardirqs_off_caller);
76NOKPROBE_SYMBOL(trace_hardirqs_off_caller);
77#endif /* CONFIG_TRACE_IRQFLAGS */
78
79#ifdef CONFIG_TRACE_PREEMPT_TOGGLE
80
81void trace_preempt_on(unsigned long a0, unsigned long a1)
82{
83 if (!in_nmi())
84 trace_preempt_enable_rcuidle(a0, a1);
85 tracer_preempt_on(a0, a1);
86}
87
88void trace_preempt_off(unsigned long a0, unsigned long a1)
89{
90 if (!in_nmi())
91 trace_preempt_disable_rcuidle(a0, a1);
92 tracer_preempt_off(a0, a1);
93}
94#endif
95