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