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