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 */
unused(void)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")
BPF_PROG(test1,int a)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")
BPF_PROG(test2,int a)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")
test3(struct pt_regs * ctx)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")
BPF_KRETPROBE(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")
BPF_PROG(test5,int a,int * b,int ret)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")
test6(struct pt_regs * ctx)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")
BPF_UPROBE(test7)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")
BPF_URETPROBE(test8,int ret)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