1 // SPDX-License-Identifier: GPL-2.0
2 /* Copyright (c) 2022 Meta Platforms, Inc. and affiliates. */
3
4 #include <test_progs.h>
5 #include "tracing_struct.skel.h"
6 #include "tracing_struct_many_args.skel.h"
7 #include "tracing_struct_int128.skel.h"
8
test_struct_args(void)9 static void test_struct_args(void)
10 {
11 struct tracing_struct *skel;
12 int err;
13
14 skel = tracing_struct__open_and_load();
15 if (!ASSERT_OK_PTR(skel, "tracing_struct__open_and_load"))
16 return;
17
18 err = tracing_struct__attach(skel);
19 if (!ASSERT_OK(err, "tracing_struct__attach"))
20 goto destroy_skel;
21
22 ASSERT_OK(trigger_module_test_read(256), "trigger_read");
23
24 ASSERT_EQ(skel->bss->t1_a_a, 2, "t1:a.a");
25 ASSERT_EQ(skel->bss->t1_a_b, 3, "t1:a.b");
26 ASSERT_EQ(skel->bss->t1_b, 1, "t1:b");
27 ASSERT_EQ(skel->bss->t1_c, 4, "t1:c");
28
29 ASSERT_EQ(skel->bss->t1_nregs, 4, "t1 nregs");
30 ASSERT_EQ(skel->bss->t1_reg0, 2, "t1 reg0");
31 ASSERT_EQ(skel->bss->t1_reg1, 3, "t1 reg1");
32 ASSERT_EQ(skel->bss->t1_reg2, 1, "t1 reg2");
33 ASSERT_EQ(skel->bss->t1_reg3, 4, "t1 reg3");
34 ASSERT_EQ(skel->bss->t1_ret, 10, "t1 ret");
35
36 ASSERT_EQ(skel->bss->t2_a, 1, "t2:a");
37 ASSERT_EQ(skel->bss->t2_b_a, 2, "t2:b.a");
38 ASSERT_EQ(skel->bss->t2_b_b, 3, "t2:b.b");
39 ASSERT_EQ(skel->bss->t2_c, 4, "t2:c");
40 ASSERT_EQ(skel->bss->t2_ret, 10, "t2 ret");
41
42 ASSERT_EQ(skel->bss->t3_a, 1, "t3:a");
43 ASSERT_EQ(skel->bss->t3_b, 4, "t3:b");
44 ASSERT_EQ(skel->bss->t3_c_a, 2, "t3:c.a");
45 ASSERT_EQ(skel->bss->t3_c_b, 3, "t3:c.b");
46 ASSERT_EQ(skel->bss->t3_ret, 10, "t3 ret");
47
48 ASSERT_EQ(skel->bss->t4_a_a, 10, "t4:a.a");
49 ASSERT_EQ(skel->bss->t4_b, 1, "t4:b");
50 ASSERT_EQ(skel->bss->t4_c, 2, "t4:c");
51 ASSERT_EQ(skel->bss->t4_d, 3, "t4:d");
52 ASSERT_EQ(skel->bss->t4_e_a, 2, "t4:e.a");
53 ASSERT_EQ(skel->bss->t4_e_b, 3, "t4:e.b");
54 ASSERT_EQ(skel->bss->t4_ret, 21, "t4 ret");
55
56 ASSERT_EQ(skel->bss->t5_ret, 1, "t5 ret");
57
58 ASSERT_EQ(skel->bss->t6, 1, "t6 ret");
59
60 destroy_skel:
61 tracing_struct__destroy(skel);
62 }
63
test_struct_many_args(void)64 static void test_struct_many_args(void)
65 {
66 struct tracing_struct_many_args *skel;
67 int err;
68
69 skel = tracing_struct_many_args__open_and_load();
70 if (!ASSERT_OK_PTR(skel, "tracing_struct_many_args__open_and_load"))
71 return;
72
73 err = tracing_struct_many_args__attach(skel);
74 if (!ASSERT_OK(err, "tracing_struct_many_args__attach"))
75 goto destroy_skel;
76
77 ASSERT_OK(trigger_module_test_read(256), "trigger_read");
78
79 ASSERT_EQ(skel->bss->t7_a, 16, "t7:a");
80 ASSERT_EQ(skel->bss->t7_b, 17, "t7:b");
81 ASSERT_EQ(skel->bss->t7_c, 18, "t7:c");
82 ASSERT_EQ(skel->bss->t7_d, 19, "t7:d");
83 ASSERT_EQ(skel->bss->t7_e, 20, "t7:e");
84 ASSERT_EQ(skel->bss->t7_f_a, 21, "t7:f.a");
85 ASSERT_EQ(skel->bss->t7_f_b, 22, "t7:f.b");
86 ASSERT_EQ(skel->bss->t7_ret, 133, "t7 ret");
87
88 ASSERT_EQ(skel->bss->t8_a, 16, "t8:a");
89 ASSERT_EQ(skel->bss->t8_b, 17, "t8:b");
90 ASSERT_EQ(skel->bss->t8_c, 18, "t8:c");
91 ASSERT_EQ(skel->bss->t8_d, 19, "t8:d");
92 ASSERT_EQ(skel->bss->t8_e, 20, "t8:e");
93 ASSERT_EQ(skel->bss->t8_f_a, 21, "t8:f.a");
94 ASSERT_EQ(skel->bss->t8_f_b, 22, "t8:f.b");
95 ASSERT_EQ(skel->bss->t8_g, 23, "t8:g");
96 ASSERT_EQ(skel->bss->t8_ret, 156, "t8 ret");
97
98 ASSERT_EQ(skel->bss->t9_a, 16, "t9:a");
99 ASSERT_EQ(skel->bss->t9_b, 17, "t9:b");
100 ASSERT_EQ(skel->bss->t9_c, 18, "t9:c");
101 ASSERT_EQ(skel->bss->t9_d, 19, "t9:d");
102 ASSERT_EQ(skel->bss->t9_e, 20, "t9:e");
103 ASSERT_EQ(skel->bss->t9_f, 21, "t9:f");
104 ASSERT_EQ(skel->bss->t9_g, 22, "t9:f");
105 ASSERT_EQ(skel->bss->t9_h_a, 23, "t9:h.a");
106 ASSERT_EQ(skel->bss->t9_h_b, 24, "t9:h.b");
107 ASSERT_EQ(skel->bss->t9_h_c, 25, "t9:h.c");
108 ASSERT_EQ(skel->bss->t9_h_d, 26, "t9:h.d");
109 ASSERT_EQ(skel->bss->t9_i, 27, "t9:i");
110 ASSERT_EQ(skel->bss->t9_ret, 258, "t9 ret");
111
112 destroy_skel:
113 tracing_struct_many_args__destroy(skel);
114 }
115
test_int128_args(void)116 static void test_int128_args(void)
117 {
118 /*
119 * __int128 arguments are passed in a register pair on x86_64 and
120 * arm64, which the trampoline packs into two context slots. Other
121 * architectures pass a __int128 differently (e.g. s390x passes larger
122 * arguments by reference), so only exercise this on x86_64 and arm64.
123 */
124 #if defined(__x86_64__) || defined(__aarch64__)
125 struct tracing_struct_int128 *skel;
126 int err;
127
128 skel = tracing_struct_int128__open_and_load();
129 if (!ASSERT_OK_PTR(skel, "tracing_struct_int128__open_and_load"))
130 return;
131
132 err = tracing_struct_int128__attach(skel);
133 if (!ASSERT_OK(err, "tracing_struct_int128__attach"))
134 goto destroy_skel;
135
136 ASSERT_OK(trigger_module_test_read(256), "trigger_read");
137
138 ASSERT_EQ(skel->bss->t_b, 2, "t:b");
139 ASSERT_EQ(skel->bss->t_c, 3, "t:c");
140 ASSERT_EQ(skel->bss->t_ret, 6, "t ret");
141
142 destroy_skel:
143 tracing_struct_int128__destroy(skel);
144 #else
145 test__skip();
146 #endif
147 }
148
test_union_args(void)149 static void test_union_args(void)
150 {
151 struct tracing_struct *skel;
152 int err;
153
154 skel = tracing_struct__open_and_load();
155 if (!ASSERT_OK_PTR(skel, "tracing_struct__open_and_load"))
156 return;
157
158 err = tracing_struct__attach(skel);
159 if (!ASSERT_OK(err, "tracing_struct__attach"))
160 goto out;
161
162 ASSERT_OK(trigger_module_test_read(256), "trigger_read");
163
164 ASSERT_EQ(skel->bss->ut1_a_a, 1, "ut1:a.arg.a");
165 ASSERT_EQ(skel->bss->ut1_b, 4, "ut1:b");
166 ASSERT_EQ(skel->bss->ut1_c, 5, "ut1:c");
167
168 ASSERT_EQ(skel->bss->ut2_a, 6, "ut2:a");
169 ASSERT_EQ(skel->bss->ut2_b_a, 2, "ut2:b.arg.a");
170 ASSERT_EQ(skel->bss->ut2_b_b, 3, "ut2:b.arg.b");
171
172 out:
173 tracing_struct__destroy(skel);
174 }
175
test_tracing_struct(void)176 void test_tracing_struct(void)
177 {
178 if (test__start_subtest("struct_args"))
179 test_struct_args();
180 if (test__start_subtest("struct_many_args"))
181 test_struct_many_args();
182 if (test__start_subtest("int128_args"))
183 test_int128_args();
184 if (test__start_subtest("union_args"))
185 test_union_args();
186 }
187