1 // SPDX-License-Identifier: GPL-2.0 2 /* Copyright (c) 2020 Facebook */ 3 4 #include "vmlinux.h" 5 #include <asm/unistd.h> 6 #include <bpf/bpf_helpers.h> 7 #include <bpf/bpf_tracing.h> 8 #include <bpf/bpf_core_read.h> 9 10 #define MY_TV_NSEC 1337 11 12 bool tp_called = false; 13 bool raw_tp_called = false; 14 bool tp_btf_called = false; 15 bool kprobe_called = false; 16 bool fentry_called = false; 17 18 SEC("tp/syscalls/sys_enter_nanosleep") 19 int handle__tp(struct trace_event_raw_sys_enter *args) 20 { 21 struct __kernel_timespec *ts; 22 long tv_nsec; 23 24 if (args->id != __NR_nanosleep) 25 return 0; 26 27 ts = (void *)args->args[0]; 28 if (bpf_probe_read_user(&tv_nsec, sizeof(ts->tv_nsec), &ts->tv_nsec) || 29 tv_nsec != MY_TV_NSEC) 30 return 0; 31 32 tp_called = true; 33 return 0; 34 } 35 36 SEC("raw_tp/sys_enter") 37 int BPF_PROG(handle__raw_tp, struct pt_regs *regs, long id) 38 { 39 struct __kernel_timespec *ts; 40 long tv_nsec; 41 42 if (id != __NR_nanosleep) 43 return 0; 44 45 ts = (void *)PT_REGS_PARM1_CORE(regs); 46 if (bpf_probe_read_user(&tv_nsec, sizeof(ts->tv_nsec), &ts->tv_nsec) || 47 tv_nsec != MY_TV_NSEC) 48 return 0; 49 50 raw_tp_called = true; 51 return 0; 52 } 53 54 SEC("tp_btf/sys_enter") 55 int BPF_PROG(handle__tp_btf, struct pt_regs *regs, long id) 56 { 57 struct __kernel_timespec *ts; 58 long tv_nsec; 59 60 if (id != __NR_nanosleep) 61 return 0; 62 63 ts = (void *)PT_REGS_PARM1_CORE(regs); 64 if (bpf_probe_read_user(&tv_nsec, sizeof(ts->tv_nsec), &ts->tv_nsec) || 65 tv_nsec != MY_TV_NSEC) 66 return 0; 67 68 tp_btf_called = true; 69 return 0; 70 } 71 72 SEC("kprobe/hrtimer_start_range_ns") 73 int BPF_KPROBE(handle__kprobe, struct hrtimer *timer, ktime_t tim, u64 delta_ns, 74 const enum hrtimer_mode mode) 75 { 76 if (tim == MY_TV_NSEC) 77 kprobe_called = true; 78 return 0; 79 } 80 81 SEC("fentry/hrtimer_start_range_ns") 82 int BPF_PROG(handle__fentry, struct hrtimer *timer, ktime_t tim, u64 delta_ns, 83 const enum hrtimer_mode mode) 84 { 85 if (tim == MY_TV_NSEC) 86 fentry_called = true; 87 return 0; 88 } 89 90 char _license[] SEC("license") = "GPL"; 91