1 // SPDX-License-Identifier: GPL-2.0
2 #include "vmlinux.h"
3 #include <bpf/bpf_helpers.h>
4 #include <bpf/bpf_tracing.h>
5 
6 char _license[] SEC("license") = "GPL";
7 
8 extern const void bpf_fentry_test1 __ksym;
9 extern const void bpf_fentry_test2 __ksym;
10 extern const void bpf_fentry_test3 __ksym;
11 extern const void bpf_fentry_test4 __ksym;
12 extern const void bpf_modify_return_test __ksym;
13 extern const void bpf_fentry_test6 __ksym;
14 extern const void bpf_fentry_test7 __ksym;
15 
16 extern bool CONFIG_X86_KERNEL_IBT __kconfig __weak;
17 
18 /* This function is here to have CONFIG_X86_KERNEL_IBT
19  * used and added to object BTF.
20  */
21 int unused(void)
22 {
23 	return CONFIG_X86_KERNEL_IBT ? 0 : 1;
24 }
25 
26 __u64 test1_result = 0;
27 SEC("fentry/bpf_fentry_test1")
28 int BPF_PROG(test1, int a)
29 {
30 	__u64 addr = bpf_get_func_ip(ctx);
31 
32 	test1_result = (const void *) addr == &bpf_fentry_test1;
33 	return 0;
34 }
35 
36 __u64 test2_result = 0;
37 SEC("fexit/bpf_fentry_test2")
38 int BPF_PROG(test2, int a)
39 {
40 	__u64 addr = bpf_get_func_ip(ctx);
41 
42 	test2_result = (const void *) addr == &bpf_fentry_test2;
43 	return 0;
44 }
45 
46 __u64 test3_result = 0;
47 SEC("kprobe/bpf_fentry_test3")
48 int test3(struct pt_regs *ctx)
49 {
50 	__u64 addr = bpf_get_func_ip(ctx);
51 
52 	test3_result = (const void *) addr == &bpf_fentry_test3;
53 	return 0;
54 }
55 
56 __u64 test4_result = 0;
57 SEC("kretprobe/bpf_fentry_test4")
58 int BPF_KRETPROBE(test4)
59 {
60 	__u64 addr = bpf_get_func_ip(ctx);
61 
62 	test4_result = (const void *) addr == &bpf_fentry_test4;
63 	return 0;
64 }
65 
66 __u64 test5_result = 0;
67 SEC("fmod_ret/bpf_modify_return_test")
68 int BPF_PROG(test5, int a, int *b, int ret)
69 {
70 	__u64 addr = bpf_get_func_ip(ctx);
71 
72 	test5_result = (const void *) addr == &bpf_modify_return_test;
73 	return ret;
74 }
75 
76 __u64 test6_result = 0;
77 SEC("?kprobe")
78 int test6(struct pt_regs *ctx)
79 {
80 	__u64 addr = bpf_get_func_ip(ctx);
81 
82 	test6_result = (const void *) addr == 0;
83 	return 0;
84 }
85 
86 unsigned long uprobe_trigger;
87 
88 __u64 test7_result = 0;
89 SEC("uprobe//proc/self/exe:uprobe_trigger")
90 int BPF_UPROBE(test7)
91 {
92 	__u64 addr = bpf_get_func_ip(ctx);
93 
94 	test7_result = (const void *) addr == (const void *) uprobe_trigger;
95 	return 0;
96 }
97 
98 __u64 test8_result = 0;
99 SEC("uretprobe//proc/self/exe:uprobe_trigger")
100 int BPF_URETPROBE(test8, int ret)
101 {
102 	__u64 addr = bpf_get_func_ip(ctx);
103 
104 	test8_result = (const void *) addr == (const void *) uprobe_trigger;
105 	return 0;
106 }
107