1 // SPDX-License-Identifier: GPL-2.0
2 // Copyright (c) 2019 Facebook
3 #include <linux/bpf.h>
4 #include <linux/version.h>
5 #include <bpf/bpf_helpers.h>
6 
7 __u32 sig = 0, pid = 0, status = 0, signal_thread = 0;
8 
bpf_send_signal_test(void * ctx)9 static __always_inline int bpf_send_signal_test(void *ctx)
10 {
11 	int ret;
12 
13 	if (status != 0 || sig == 0 || pid == 0)
14 		return 0;
15 
16 	if ((bpf_get_current_pid_tgid() >> 32) == pid) {
17 		if (signal_thread)
18 			ret = bpf_send_signal_thread(sig);
19 		else
20 			ret = bpf_send_signal(sig);
21 		if (ret == 0)
22 			status = 1;
23 	}
24 
25 	return 0;
26 }
27 
28 SEC("tracepoint/syscalls/sys_enter_nanosleep")
send_signal_tp(void * ctx)29 int send_signal_tp(void *ctx)
30 {
31 	return bpf_send_signal_test(ctx);
32 }
33 
34 SEC("tracepoint/sched/sched_switch")
send_signal_tp_sched(void * ctx)35 int send_signal_tp_sched(void *ctx)
36 {
37 	return bpf_send_signal_test(ctx);
38 }
39 
40 SEC("perf_event")
send_signal_perf(void * ctx)41 int send_signal_perf(void *ctx)
42 {
43 	return bpf_send_signal_test(ctx);
44 }
45 
46 char __license[] SEC("license") = "GPL";
47