1 // SPDX-License-Identifier: GPL-2.0-only
2 #include <time.h>
3 #include <linux/bpf.h>
4 #include <bpf/bpf_helpers.h>
5 
6 struct timer {
7 	struct bpf_timer t;
8 };
9 
10 struct lock {
11 	struct bpf_spin_lock l;
12 };
13 
14 struct {
15 	__uint(type, BPF_MAP_TYPE_ARRAY);
16 	__uint(max_entries, 1);
17 	__type(key, __u32);
18 	__type(value, struct timer);
19 } timers SEC(".maps");
20 
21 struct {
22 	__uint(type, BPF_MAP_TYPE_ARRAY);
23 	__uint(max_entries, 1);
24 	__type(key, __u32);
25 	__type(value, struct lock);
26 } locks SEC(".maps");
27 
timer_cb(void * map,int * key,struct timer * timer)28 static int timer_cb(void *map, int *key, struct timer *timer)
29 {
30 	return 0;
31 }
32 
timer_work(void)33 static void timer_work(void)
34 {
35 	struct timer *timer;
36 	const int key = 0;
37 
38 	timer  = bpf_map_lookup_elem(&timers, &key);
39 	if (timer) {
40 		bpf_timer_init(&timer->t, &timers, CLOCK_MONOTONIC);
41 		bpf_timer_set_callback(&timer->t, timer_cb);
42 		bpf_timer_start(&timer->t, 10E9, 0);
43 		bpf_timer_cancel(&timer->t);
44 	}
45 }
46 
spin_lock_work(void)47 static void spin_lock_work(void)
48 {
49 	const int key = 0;
50 	struct lock *lock;
51 
52 	lock = bpf_map_lookup_elem(&locks, &key);
53 	if (lock) {
54 		bpf_spin_lock(&lock->l);
55 		bpf_spin_unlock(&lock->l);
56 	}
57 }
58 
59 SEC("?raw_tp/sys_enter")
raw_tp_timer(void * ctx)60 int raw_tp_timer(void *ctx)
61 {
62 	timer_work();
63 
64 	return 0;
65 }
66 
67 SEC("?tp/syscalls/sys_enter_nanosleep")
tp_timer(void * ctx)68 int tp_timer(void *ctx)
69 {
70 	timer_work();
71 
72 	return 0;
73 }
74 
75 SEC("?kprobe")
kprobe_timer(void * ctx)76 int kprobe_timer(void *ctx)
77 {
78 	timer_work();
79 
80 	return 0;
81 }
82 
83 SEC("?perf_event")
perf_event_timer(void * ctx)84 int perf_event_timer(void *ctx)
85 {
86 	timer_work();
87 
88 	return 0;
89 }
90 
91 SEC("?raw_tp/sys_enter")
raw_tp_spin_lock(void * ctx)92 int raw_tp_spin_lock(void *ctx)
93 {
94 	spin_lock_work();
95 
96 	return 0;
97 }
98 
99 SEC("?tp/syscalls/sys_enter_nanosleep")
tp_spin_lock(void * ctx)100 int tp_spin_lock(void *ctx)
101 {
102 	spin_lock_work();
103 
104 	return 0;
105 }
106 
107 SEC("?kprobe")
kprobe_spin_lock(void * ctx)108 int kprobe_spin_lock(void *ctx)
109 {
110 	spin_lock_work();
111 
112 	return 0;
113 }
114 
115 SEC("?perf_event")
perf_event_spin_lock(void * ctx)116 int perf_event_spin_lock(void *ctx)
117 {
118 	spin_lock_work();
119 
120 	return 0;
121 }
122 
123 const char LICENSE[] SEC("license") = "GPL";
124