1 // SPDX-License-Identifier: GPL-2.0-or-later
2
3 #include <linux/init.h>
4 #include <linux/log2.h>
5 #include <kunit/test.h>
6
7 #include <asm/guest-state-buffer.h>
8
test_creating_buffer(struct kunit * test)9 static void test_creating_buffer(struct kunit *test)
10 {
11 struct kvmppc_gs_buff *gsb;
12 size_t size = 0x100;
13
14 gsb = kvmppc_gsb_new(size, 0, 0, GFP_KERNEL);
15 KUNIT_ASSERT_NOT_ERR_OR_NULL(test, gsb);
16
17 KUNIT_ASSERT_NOT_ERR_OR_NULL(test, gsb->hdr);
18
19 KUNIT_EXPECT_EQ(test, gsb->capacity, roundup_pow_of_two(size));
20 KUNIT_EXPECT_EQ(test, gsb->len, sizeof(__be32));
21
22 kvmppc_gsb_free(gsb);
23 }
24
test_adding_element(struct kunit * test)25 static void test_adding_element(struct kunit *test)
26 {
27 const struct kvmppc_gs_elem *head, *curr;
28 union {
29 __vector128 v;
30 u64 dw[2];
31 } u;
32 int rem;
33 struct kvmppc_gs_buff *gsb;
34 size_t size = 0x1000;
35 int i, rc;
36 u64 data;
37
38 gsb = kvmppc_gsb_new(size, 0, 0, GFP_KERNEL);
39 KUNIT_ASSERT_NOT_ERR_OR_NULL(test, gsb);
40
41 /* Single elements, direct use of __kvmppc_gse_put() */
42 data = 0xdeadbeef;
43 rc = __kvmppc_gse_put(gsb, KVMPPC_GSID_GPR(0), 8, &data);
44 KUNIT_EXPECT_GE(test, rc, 0);
45
46 head = kvmppc_gsb_data(gsb);
47 KUNIT_EXPECT_EQ(test, kvmppc_gse_iden(head), KVMPPC_GSID_GPR(0));
48 KUNIT_EXPECT_EQ(test, kvmppc_gse_len(head), 8);
49 data = 0;
50 memcpy(&data, kvmppc_gse_data(head), 8);
51 KUNIT_EXPECT_EQ(test, data, 0xdeadbeef);
52
53 /* Multiple elements, simple wrapper */
54 rc = kvmppc_gse_put_u64(gsb, KVMPPC_GSID_GPR(1), 0xcafef00d);
55 KUNIT_EXPECT_GE(test, rc, 0);
56
57 u.dw[0] = 0x1;
58 u.dw[1] = 0x2;
59 rc = kvmppc_gse_put_vector128(gsb, KVMPPC_GSID_VSRS(0), &u.v);
60 KUNIT_EXPECT_GE(test, rc, 0);
61 u.dw[0] = 0x0;
62 u.dw[1] = 0x0;
63
64 kvmppc_gsb_for_each_elem(i, curr, gsb, rem) {
65 switch (i) {
66 case 0:
67 KUNIT_EXPECT_EQ(test, kvmppc_gse_iden(curr),
68 KVMPPC_GSID_GPR(0));
69 KUNIT_EXPECT_EQ(test, kvmppc_gse_len(curr), 8);
70 KUNIT_EXPECT_EQ(test, kvmppc_gse_get_be64(curr),
71 0xdeadbeef);
72 break;
73 case 1:
74 KUNIT_EXPECT_EQ(test, kvmppc_gse_iden(curr),
75 KVMPPC_GSID_GPR(1));
76 KUNIT_EXPECT_EQ(test, kvmppc_gse_len(curr), 8);
77 KUNIT_EXPECT_EQ(test, kvmppc_gse_get_u64(curr),
78 0xcafef00d);
79 break;
80 case 2:
81 KUNIT_EXPECT_EQ(test, kvmppc_gse_iden(curr),
82 KVMPPC_GSID_VSRS(0));
83 KUNIT_EXPECT_EQ(test, kvmppc_gse_len(curr), 16);
84 kvmppc_gse_get_vector128(curr, &u.v);
85 KUNIT_EXPECT_EQ(test, u.dw[0], 0x1);
86 KUNIT_EXPECT_EQ(test, u.dw[1], 0x2);
87 break;
88 }
89 }
90 KUNIT_EXPECT_EQ(test, i, 3);
91
92 kvmppc_gsb_reset(gsb);
93 KUNIT_EXPECT_EQ(test, kvmppc_gsb_nelems(gsb), 0);
94 KUNIT_EXPECT_EQ(test, kvmppc_gsb_len(gsb),
95 sizeof(struct kvmppc_gs_header));
96
97 kvmppc_gsb_free(gsb);
98 }
99
test_gs_parsing(struct kunit * test)100 static void test_gs_parsing(struct kunit *test)
101 {
102 struct kvmppc_gs_elem *gse;
103 struct kvmppc_gs_parser gsp = { 0 };
104 struct kvmppc_gs_buff *gsb;
105 size_t size = 0x1000;
106 u64 tmp1, tmp2;
107
108 gsb = kvmppc_gsb_new(size, 0, 0, GFP_KERNEL);
109 KUNIT_ASSERT_NOT_ERR_OR_NULL(test, gsb);
110
111 tmp1 = 0xdeadbeefull;
112 kvmppc_gse_put_u64(gsb, KVMPPC_GSID_GPR(0), tmp1);
113
114 KUNIT_EXPECT_GE(test, kvmppc_gse_parse(&gsp, gsb), 0);
115
116 gse = kvmppc_gsp_lookup(&gsp, KVMPPC_GSID_GPR(0));
117 KUNIT_ASSERT_NOT_ERR_OR_NULL(test, gse);
118
119 tmp2 = kvmppc_gse_get_u64(gse);
120 KUNIT_EXPECT_EQ(test, tmp2, 0xdeadbeefull);
121
122 kvmppc_gsb_free(gsb);
123 }
124
test_gs_bitmap(struct kunit * test)125 static void test_gs_bitmap(struct kunit *test)
126 {
127 struct kvmppc_gs_bitmap gsbm = { 0 };
128 struct kvmppc_gs_bitmap gsbm1 = { 0 };
129 struct kvmppc_gs_bitmap gsbm2 = { 0 };
130 u16 iden;
131 int i, j;
132
133 i = 0;
134 for (u16 iden = KVMPPC_GSID_HOST_STATE_SIZE;
135 iden <= KVMPPC_GSID_PROCESS_TABLE; iden++) {
136 kvmppc_gsbm_set(&gsbm, iden);
137 kvmppc_gsbm_set(&gsbm1, iden);
138 KUNIT_EXPECT_TRUE(test, kvmppc_gsbm_test(&gsbm, iden));
139 kvmppc_gsbm_clear(&gsbm, iden);
140 KUNIT_EXPECT_FALSE(test, kvmppc_gsbm_test(&gsbm, iden));
141 i++;
142 }
143
144 for (u16 iden = KVMPPC_GSID_RUN_INPUT; iden <= KVMPPC_GSID_VPA;
145 iden++) {
146 kvmppc_gsbm_set(&gsbm, iden);
147 kvmppc_gsbm_set(&gsbm1, iden);
148 KUNIT_EXPECT_TRUE(test, kvmppc_gsbm_test(&gsbm, iden));
149 kvmppc_gsbm_clear(&gsbm, iden);
150 KUNIT_EXPECT_FALSE(test, kvmppc_gsbm_test(&gsbm, iden));
151 i++;
152 }
153
154 for (u16 iden = KVMPPC_GSID_GPR(0); iden <= KVMPPC_GSE_DW_REGS_END; iden++) {
155 kvmppc_gsbm_set(&gsbm, iden);
156 kvmppc_gsbm_set(&gsbm1, iden);
157 KUNIT_EXPECT_TRUE(test, kvmppc_gsbm_test(&gsbm, iden));
158 kvmppc_gsbm_clear(&gsbm, iden);
159 KUNIT_EXPECT_FALSE(test, kvmppc_gsbm_test(&gsbm, iden));
160 i++;
161 }
162
163 for (u16 iden = KVMPPC_GSID_CR; iden <= KVMPPC_GSID_PSPB; iden++) {
164 kvmppc_gsbm_set(&gsbm, iden);
165 kvmppc_gsbm_set(&gsbm1, iden);
166 KUNIT_EXPECT_TRUE(test, kvmppc_gsbm_test(&gsbm, iden));
167 kvmppc_gsbm_clear(&gsbm, iden);
168 KUNIT_EXPECT_FALSE(test, kvmppc_gsbm_test(&gsbm, iden));
169 i++;
170 }
171
172 for (u16 iden = KVMPPC_GSID_VSRS(0); iden <= KVMPPC_GSID_VSRS(63);
173 iden++) {
174 kvmppc_gsbm_set(&gsbm, iden);
175 kvmppc_gsbm_set(&gsbm1, iden);
176 KUNIT_EXPECT_TRUE(test, kvmppc_gsbm_test(&gsbm, iden));
177 kvmppc_gsbm_clear(&gsbm, iden);
178 KUNIT_EXPECT_FALSE(test, kvmppc_gsbm_test(&gsbm, iden));
179 i++;
180 }
181
182 for (u16 iden = KVMPPC_GSID_HDAR; iden <= KVMPPC_GSID_ASDR; iden++) {
183 kvmppc_gsbm_set(&gsbm, iden);
184 kvmppc_gsbm_set(&gsbm1, iden);
185 KUNIT_EXPECT_TRUE(test, kvmppc_gsbm_test(&gsbm, iden));
186 kvmppc_gsbm_clear(&gsbm, iden);
187 KUNIT_EXPECT_FALSE(test, kvmppc_gsbm_test(&gsbm, iden));
188 i++;
189 }
190
191 j = 0;
192 kvmppc_gsbm_for_each(&gsbm1, iden)
193 {
194 kvmppc_gsbm_set(&gsbm2, iden);
195 j++;
196 }
197 KUNIT_EXPECT_EQ(test, i, j);
198 KUNIT_EXPECT_MEMEQ(test, &gsbm1, &gsbm2, sizeof(gsbm1));
199 }
200
201 struct kvmppc_gs_msg_test1_data {
202 u64 a;
203 u32 b;
204 struct kvmppc_gs_part_table c;
205 struct kvmppc_gs_proc_table d;
206 struct kvmppc_gs_buff_info e;
207 };
208
test1_get_size(struct kvmppc_gs_msg * gsm)209 static size_t test1_get_size(struct kvmppc_gs_msg *gsm)
210 {
211 size_t size = 0;
212 u16 ids[] = {
213 KVMPPC_GSID_PARTITION_TABLE,
214 KVMPPC_GSID_PROCESS_TABLE,
215 KVMPPC_GSID_RUN_INPUT,
216 KVMPPC_GSID_GPR(0),
217 KVMPPC_GSID_CR,
218 };
219
220 for (int i = 0; i < ARRAY_SIZE(ids); i++)
221 size += kvmppc_gse_total_size(kvmppc_gsid_size(ids[i]));
222 return size;
223 }
224
test1_fill_info(struct kvmppc_gs_buff * gsb,struct kvmppc_gs_msg * gsm)225 static int test1_fill_info(struct kvmppc_gs_buff *gsb,
226 struct kvmppc_gs_msg *gsm)
227 {
228 struct kvmppc_gs_msg_test1_data *data = gsm->data;
229
230 if (kvmppc_gsm_includes(gsm, KVMPPC_GSID_GPR(0)))
231 kvmppc_gse_put_u64(gsb, KVMPPC_GSID_GPR(0), data->a);
232
233 if (kvmppc_gsm_includes(gsm, KVMPPC_GSID_CR))
234 kvmppc_gse_put_u32(gsb, KVMPPC_GSID_CR, data->b);
235
236 if (kvmppc_gsm_includes(gsm, KVMPPC_GSID_PARTITION_TABLE))
237 kvmppc_gse_put_part_table(gsb, KVMPPC_GSID_PARTITION_TABLE,
238 data->c);
239
240 if (kvmppc_gsm_includes(gsm, KVMPPC_GSID_PROCESS_TABLE))
241 kvmppc_gse_put_proc_table(gsb, KVMPPC_GSID_PARTITION_TABLE,
242 data->d);
243
244 if (kvmppc_gsm_includes(gsm, KVMPPC_GSID_RUN_INPUT))
245 kvmppc_gse_put_buff_info(gsb, KVMPPC_GSID_RUN_INPUT, data->e);
246
247 return 0;
248 }
249
test1_refresh_info(struct kvmppc_gs_msg * gsm,struct kvmppc_gs_buff * gsb)250 static int test1_refresh_info(struct kvmppc_gs_msg *gsm,
251 struct kvmppc_gs_buff *gsb)
252 {
253 struct kvmppc_gs_parser gsp = { 0 };
254 struct kvmppc_gs_msg_test1_data *data = gsm->data;
255 struct kvmppc_gs_elem *gse;
256 int rc;
257
258 rc = kvmppc_gse_parse(&gsp, gsb);
259 if (rc < 0)
260 return rc;
261
262 gse = kvmppc_gsp_lookup(&gsp, KVMPPC_GSID_GPR(0));
263 if (gse)
264 data->a = kvmppc_gse_get_u64(gse);
265
266 gse = kvmppc_gsp_lookup(&gsp, KVMPPC_GSID_CR);
267 if (gse)
268 data->b = kvmppc_gse_get_u32(gse);
269
270 return 0;
271 }
272
273 static struct kvmppc_gs_msg_ops gs_msg_test1_ops = {
274 .get_size = test1_get_size,
275 .fill_info = test1_fill_info,
276 .refresh_info = test1_refresh_info,
277 };
278
test_gs_msg(struct kunit * test)279 static void test_gs_msg(struct kunit *test)
280 {
281 struct kvmppc_gs_msg_test1_data test1_data = {
282 .a = 0xdeadbeef,
283 .b = 0x1,
284 };
285 struct kvmppc_gs_msg *gsm;
286 struct kvmppc_gs_buff *gsb;
287
288 gsm = kvmppc_gsm_new(&gs_msg_test1_ops, &test1_data, GSM_SEND,
289 GFP_KERNEL);
290 KUNIT_ASSERT_NOT_ERR_OR_NULL(test, gsm);
291
292 gsb = kvmppc_gsb_new(kvmppc_gsm_size(gsm), 0, 0, GFP_KERNEL);
293 KUNIT_ASSERT_NOT_ERR_OR_NULL(test, gsb);
294
295 kvmppc_gsm_include(gsm, KVMPPC_GSID_PARTITION_TABLE);
296 kvmppc_gsm_include(gsm, KVMPPC_GSID_PROCESS_TABLE);
297 kvmppc_gsm_include(gsm, KVMPPC_GSID_RUN_INPUT);
298 kvmppc_gsm_include(gsm, KVMPPC_GSID_GPR(0));
299 kvmppc_gsm_include(gsm, KVMPPC_GSID_CR);
300
301 kvmppc_gsm_fill_info(gsm, gsb);
302
303 memset(&test1_data, 0, sizeof(test1_data));
304
305 kvmppc_gsm_refresh_info(gsm, gsb);
306 KUNIT_EXPECT_EQ(test, test1_data.a, 0xdeadbeef);
307 KUNIT_EXPECT_EQ(test, test1_data.b, 0x1);
308
309 kvmppc_gsm_free(gsm);
310 }
311
312 static struct kunit_case guest_state_buffer_testcases[] = {
313 KUNIT_CASE(test_creating_buffer),
314 KUNIT_CASE(test_adding_element),
315 KUNIT_CASE(test_gs_bitmap),
316 KUNIT_CASE(test_gs_parsing),
317 KUNIT_CASE(test_gs_msg),
318 {}
319 };
320
321 static struct kunit_suite guest_state_buffer_test_suite = {
322 .name = "guest_state_buffer_test",
323 .test_cases = guest_state_buffer_testcases,
324 };
325
326 kunit_test_suites(&guest_state_buffer_test_suite);
327
328 MODULE_DESCRIPTION("KUnit tests for Guest State Buffer APIs");
329 MODULE_LICENSE("GPL");
330