1 // SPDX-License-Identifier: GPL-2.0
2 /* Copyright (c) 2022 Meta Platforms, Inc. and affiliates. */
3
4 #include <vmlinux.h>
5 #include <bpf/bpf_tracing.h>
6 #include <bpf/bpf_helpers.h>
7
8 struct bpf_testmod_struct_arg_1 {
9 int a;
10 };
11 struct bpf_testmod_struct_arg_2 {
12 long a;
13 long b;
14 };
15
16 long t1_a_a, t1_a_b, t1_b, t1_c, t1_ret, t1_nregs;
17 __u64 t1_reg0, t1_reg1, t1_reg2, t1_reg3;
18 long t2_a, t2_b_a, t2_b_b, t2_c, t2_ret;
19 long t3_a, t3_b, t3_c_a, t3_c_b, t3_ret;
20 long t4_a_a, t4_b, t4_c, t4_d, t4_e_a, t4_e_b, t4_ret;
21 long t5_ret;
22
23 SEC("fentry/bpf_testmod_test_struct_arg_1")
BPF_PROG2(test_struct_arg_1,struct bpf_testmod_struct_arg_2,a,int,b,int,c)24 int BPF_PROG2(test_struct_arg_1, struct bpf_testmod_struct_arg_2, a, int, b, int, c)
25 {
26 t1_a_a = a.a;
27 t1_a_b = a.b;
28 t1_b = b;
29 t1_c = c;
30 return 0;
31 }
32
33 SEC("fexit/bpf_testmod_test_struct_arg_1")
BPF_PROG2(test_struct_arg_2,struct bpf_testmod_struct_arg_2,a,int,b,int,c,int,ret)34 int BPF_PROG2(test_struct_arg_2, struct bpf_testmod_struct_arg_2, a, int, b, int, c, int, ret)
35 {
36 t1_nregs = bpf_get_func_arg_cnt(ctx);
37 /* a.a */
38 bpf_get_func_arg(ctx, 0, &t1_reg0);
39 /* a.b */
40 bpf_get_func_arg(ctx, 1, &t1_reg1);
41 /* b */
42 bpf_get_func_arg(ctx, 2, &t1_reg2);
43 t1_reg2 = (int)t1_reg2;
44 /* c */
45 bpf_get_func_arg(ctx, 3, &t1_reg3);
46 t1_reg3 = (int)t1_reg3;
47
48 t1_ret = ret;
49 return 0;
50 }
51
52 SEC("fentry/bpf_testmod_test_struct_arg_2")
BPF_PROG2(test_struct_arg_3,int,a,struct bpf_testmod_struct_arg_2,b,int,c)53 int BPF_PROG2(test_struct_arg_3, int, a, struct bpf_testmod_struct_arg_2, b, int, c)
54 {
55 t2_a = a;
56 t2_b_a = b.a;
57 t2_b_b = b.b;
58 t2_c = c;
59 return 0;
60 }
61
62 SEC("fexit/bpf_testmod_test_struct_arg_2")
BPF_PROG2(test_struct_arg_4,int,a,struct bpf_testmod_struct_arg_2,b,int,c,int,ret)63 int BPF_PROG2(test_struct_arg_4, int, a, struct bpf_testmod_struct_arg_2, b, int, c, int, ret)
64 {
65 t2_ret = ret;
66 return 0;
67 }
68
69 SEC("fentry/bpf_testmod_test_struct_arg_3")
BPF_PROG2(test_struct_arg_5,int,a,int,b,struct bpf_testmod_struct_arg_2,c)70 int BPF_PROG2(test_struct_arg_5, int, a, int, b, struct bpf_testmod_struct_arg_2, c)
71 {
72 t3_a = a;
73 t3_b = b;
74 t3_c_a = c.a;
75 t3_c_b = c.b;
76 return 0;
77 }
78
79 SEC("fexit/bpf_testmod_test_struct_arg_3")
BPF_PROG2(test_struct_arg_6,int,a,int,b,struct bpf_testmod_struct_arg_2,c,int,ret)80 int BPF_PROG2(test_struct_arg_6, int, a, int, b, struct bpf_testmod_struct_arg_2, c, int, ret)
81 {
82 t3_ret = ret;
83 return 0;
84 }
85
86 SEC("fentry/bpf_testmod_test_struct_arg_4")
BPF_PROG2(test_struct_arg_7,struct bpf_testmod_struct_arg_1,a,int,b,int,c,int,d,struct bpf_testmod_struct_arg_2,e)87 int BPF_PROG2(test_struct_arg_7, struct bpf_testmod_struct_arg_1, a, int, b,
88 int, c, int, d, struct bpf_testmod_struct_arg_2, e)
89 {
90 t4_a_a = a.a;
91 t4_b = b;
92 t4_c = c;
93 t4_d = d;
94 t4_e_a = e.a;
95 t4_e_b = e.b;
96 return 0;
97 }
98
99 SEC("fexit/bpf_testmod_test_struct_arg_4")
BPF_PROG2(test_struct_arg_8,struct bpf_testmod_struct_arg_1,a,int,b,int,c,int,d,struct bpf_testmod_struct_arg_2,e,int,ret)100 int BPF_PROG2(test_struct_arg_8, struct bpf_testmod_struct_arg_1, a, int, b,
101 int, c, int, d, struct bpf_testmod_struct_arg_2, e, int, ret)
102 {
103 t4_ret = ret;
104 return 0;
105 }
106
107 SEC("fentry/bpf_testmod_test_struct_arg_5")
BPF_PROG2(test_struct_arg_9)108 int BPF_PROG2(test_struct_arg_9)
109 {
110 return 0;
111 }
112
113 SEC("fexit/bpf_testmod_test_struct_arg_5")
BPF_PROG2(test_struct_arg_10,int,ret)114 int BPF_PROG2(test_struct_arg_10, int, ret)
115 {
116 t5_ret = ret;
117 return 0;
118 }
119
120 char _license[] SEC("license") = "GPL";
121