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 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 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 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 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 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