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