xref: /linux/tools/testing/selftests/bpf/progs/fexit_test.c (revision 48dea9a700c8728cc31a1dd44588b97578de86ee)
1 // SPDX-License-Identifier: GPL-2.0
2 /* Copyright (c) 2019 Facebook */
3 #include <linux/bpf.h>
4 #include <bpf/bpf_helpers.h>
5 #include <bpf/bpf_tracing.h>
6 
7 char _license[] SEC("license") = "GPL";
8 
9 __u64 test1_result = 0;
10 SEC("fexit/bpf_fentry_test1")
11 int BPF_PROG(test1, int a, int ret)
12 {
13 	test1_result = a == 1 && ret == 2;
14 	return 0;
15 }
16 
17 __u64 test2_result = 0;
18 SEC("fexit/bpf_fentry_test2")
19 int BPF_PROG(test2, int a, __u64 b, int ret)
20 {
21 	test2_result = a == 2 && b == 3 && ret == 5;
22 	return 0;
23 }
24 
25 __u64 test3_result = 0;
26 SEC("fexit/bpf_fentry_test3")
27 int BPF_PROG(test3, char a, int b, __u64 c, int ret)
28 {
29 	test3_result = a == 4 && b == 5 && c == 6 && ret == 15;
30 	return 0;
31 }
32 
33 __u64 test4_result = 0;
34 SEC("fexit/bpf_fentry_test4")
35 int BPF_PROG(test4, void *a, char b, int c, __u64 d, int ret)
36 {
37 	test4_result = a == (void *)7 && b == 8 && c == 9 && d == 10 &&
38 		ret == 34;
39 	return 0;
40 }
41 
42 __u64 test5_result = 0;
43 SEC("fexit/bpf_fentry_test5")
44 int BPF_PROG(test5, __u64 a, void *b, short c, int d, __u64 e, int ret)
45 {
46 	test5_result = a == 11 && b == (void *)12 && c == 13 && d == 14 &&
47 		e == 15 && ret == 65;
48 	return 0;
49 }
50 
51 __u64 test6_result = 0;
52 SEC("fexit/bpf_fentry_test6")
53 int BPF_PROG(test6, __u64 a, void *b, short c, int d, void *e, __u64 f, int ret)
54 {
55 	test6_result = a == 16 && b == (void *)17 && c == 18 && d == 19 &&
56 		e == (void *)20 && f == 21 && ret == 111;
57 	return 0;
58 }
59 
60 struct bpf_fentry_test_t {
61 	struct bpf_fentry_test *a;
62 };
63 
64 __u64 test7_result = 0;
65 SEC("fexit/bpf_fentry_test7")
66 int BPF_PROG(test7, struct bpf_fentry_test_t *arg)
67 {
68 	if (arg == 0)
69 		test7_result = 1;
70 	return 0;
71 }
72 
73 __u64 test8_result = 0;
74 SEC("fexit/bpf_fentry_test8")
75 int BPF_PROG(test8, struct bpf_fentry_test_t *arg)
76 {
77 	if (arg->a == 0)
78 		test8_result = 1;
79 	return 0;
80 }
81