kernel/trace/trace_preemptirq.c

   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 */
  20 static DEFINE_PER_CPU(int, tracing_irq_cpu);
  21
  22 void 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 }
  33 EXPORT_SYMBOL(trace_hardirqs_on);
  34 NOKPROBE_SYMBOL(trace_hardirqs_on);
  35
  36 void 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 }
  47 EXPORT_SYMBOL(trace_hardirqs_off);
  48 NOKPROBE_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 }
  61 EXPORT_SYMBOL(trace_hardirqs_on_caller);
  62 NOKPROBE_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 }
  75 EXPORT_SYMBOL(trace_hardirqs_off_caller);
  76 NOKPROBE_SYMBOL(trace_hardirqs_off_caller);
  77 #endif /* CONFIG_TRACE_IRQFLAGS */
  78
  79 #ifdef CONFIG_TRACE_PREEMPT_TOGGLE
  80
  81 void 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
  88 void 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