1 // SPDX-License-Identifier: GPL-2.0 2 #include <linux/ftrace.h> 3 #include <linux/tracepoint.h> 4 #include <linux/kernel.h> 5 #include <linux/module.h> 6 #include <linux/init.h> 7 #include <linux/rv.h> 8 #include <rv/instrumentation.h> 9 #include <rv/da_monitor.h> 10 11 #define MODULE_NAME "snep" 12 13 #include <trace/events/sched.h> 14 #include <trace/events/preemptirq.h> 15 #include <rv_trace.h> 16 #include <monitors/sched/sched.h> 17 18 #include "snep.h" 19 20 static struct rv_monitor rv_snep; 21 DECLARE_DA_MON_PER_CPU(snep, unsigned char); 22 23 static void handle_preempt_disable(void *data, unsigned long ip, unsigned long parent_ip) 24 { 25 da_handle_start_event_snep(preempt_disable_snep); 26 } 27 28 static void handle_preempt_enable(void *data, unsigned long ip, unsigned long parent_ip) 29 { 30 da_handle_start_event_snep(preempt_enable_snep); 31 } 32 33 static void handle_schedule_entry(void *data, bool preempt, unsigned long ip) 34 { 35 da_handle_event_snep(schedule_entry_snep); 36 } 37 38 static void handle_schedule_exit(void *data, bool is_switch, unsigned long ip) 39 { 40 da_handle_start_event_snep(schedule_exit_snep); 41 } 42 43 static int enable_snep(void) 44 { 45 int retval; 46 47 retval = da_monitor_init_snep(); 48 if (retval) 49 return retval; 50 51 rv_attach_trace_probe("snep", preempt_disable, handle_preempt_disable); 52 rv_attach_trace_probe("snep", preempt_enable, handle_preempt_enable); 53 rv_attach_trace_probe("snep", sched_entry_tp, handle_schedule_entry); 54 rv_attach_trace_probe("snep", sched_exit_tp, handle_schedule_exit); 55 56 return 0; 57 } 58 59 static void disable_snep(void) 60 { 61 rv_snep.enabled = 0; 62 63 rv_detach_trace_probe("snep", preempt_disable, handle_preempt_disable); 64 rv_detach_trace_probe("snep", preempt_enable, handle_preempt_enable); 65 rv_detach_trace_probe("snep", sched_entry_tp, handle_schedule_entry); 66 rv_detach_trace_probe("snep", sched_exit_tp, handle_schedule_exit); 67 68 da_monitor_destroy_snep(); 69 } 70 71 static struct rv_monitor rv_snep = { 72 .name = "snep", 73 .description = "schedule does not enable preempt.", 74 .enable = enable_snep, 75 .disable = disable_snep, 76 .reset = da_monitor_reset_all_snep, 77 .enabled = 0, 78 }; 79 80 static int __init register_snep(void) 81 { 82 rv_register_monitor(&rv_snep, &rv_sched); 83 return 0; 84 } 85 86 static void __exit unregister_snep(void) 87 { 88 rv_unregister_monitor(&rv_snep); 89 } 90 91 module_init(register_snep); 92 module_exit(unregister_snep); 93 94 MODULE_LICENSE("GPL"); 95 MODULE_AUTHOR("Gabriele Monaco <gmonaco@redhat.com>"); 96 MODULE_DESCRIPTION("snep: schedule does not enable preempt."); 97