6 #include <kunit/test.h>
55 					  struct kunit *test, size_t buf_size)  in build_aa_ext_struct()  argument
60 	buf = kunit_kzalloc(test, buf_size, GFP_USER);  in build_aa_ext_struct()
61 	KUNIT_EXPECT_NOT_ERR_OR_NULL(test, buf);  in build_aa_ext_struct()
63 	e = kunit_kmalloc(test, sizeof(*e), GFP_USER);  in build_aa_ext_struct()
64 	KUNIT_EXPECT_NOT_ERR_OR_NULL(test, e);  in build_aa_ext_struct()
111 static int policy_unpack_test_init(struct kunit *test)  in policy_unpack_test_init()  argument
116 	puf = kunit_kmalloc(test, sizeof(*puf), GFP_USER);  in policy_unpack_test_init()
117 	KUNIT_EXPECT_NOT_ERR_OR_NULL(test, puf);  in policy_unpack_test_init()
120 	puf->e = build_aa_ext_struct(puf, test, e_size);  in policy_unpack_test_init()
122 	test->priv = puf;  in policy_unpack_test_init()
126 static void policy_unpack_test_inbounds_when_inbounds(struct kunit *test)  in policy_unpack_test_inbounds_when_inbounds()  argument
128 	struct policy_unpack_fixture *puf = test->priv;  in policy_unpack_test_inbounds_when_inbounds()
130 	KUNIT_EXPECT_TRUE(test, aa_inbounds(puf->e, 0));  in policy_unpack_test_inbounds_when_inbounds()
131 	KUNIT_EXPECT_TRUE(test, aa_inbounds(puf->e, puf->e_size / 2));  in policy_unpack_test_inbounds_when_inbounds()
132 	KUNIT_EXPECT_TRUE(test, aa_inbounds(puf->e, puf->e_size));  in policy_unpack_test_inbounds_when_inbounds()
135 static void policy_unpack_test_inbounds_when_out_of_bounds(struct kunit *test)  in policy_unpack_test_inbounds_when_out_of_bounds()  argument
137 	struct policy_unpack_fixture *puf = test->priv;  in policy_unpack_test_inbounds_when_out_of_bounds()
139 	KUNIT_EXPECT_FALSE(test, aa_inbounds(puf->e, puf->e_size + 1));  in policy_unpack_test_inbounds_when_out_of_bounds()
142 static void policy_unpack_test_unpack_array_with_null_name(struct kunit *test)  in policy_unpack_test_unpack_array_with_null_name()  argument
144 	struct policy_unpack_fixture *puf = test->priv;  in policy_unpack_test_unpack_array_with_null_name()
149 	KUNIT_EXPECT_TRUE(test, aa_unpack_array(puf->e, NULL, &array_size));  in policy_unpack_test_unpack_array_with_null_name()
150 	KUNIT_EXPECT_EQ(test, array_size, (u16)TEST_ARRAY_SIZE);  in policy_unpack_test_unpack_array_with_null_name()
151 	KUNIT_EXPECT_PTR_EQ(test, puf->e->pos,  in policy_unpack_test_unpack_array_with_null_name()
155 static void policy_unpack_test_unpack_array_with_name(struct kunit *test)  in policy_unpack_test_unpack_array_with_name()  argument
157 	struct policy_unpack_fixture *puf = test->priv;  in policy_unpack_test_unpack_array_with_name()
163 	KUNIT_EXPECT_TRUE(test, aa_unpack_array(puf->e, name, &array_size));  in policy_unpack_test_unpack_array_with_name()
164 	KUNIT_EXPECT_EQ(test, array_size, (u16)TEST_ARRAY_SIZE);  in policy_unpack_test_unpack_array_with_name()
165 	KUNIT_EXPECT_PTR_EQ(test, puf->e->pos,  in policy_unpack_test_unpack_array_with_name()
169 static void policy_unpack_test_unpack_array_out_of_bounds(struct kunit *test)  in policy_unpack_test_unpack_array_out_of_bounds()  argument
171 	struct policy_unpack_fixture *puf = test->priv;  in policy_unpack_test_unpack_array_out_of_bounds()
178 	KUNIT_EXPECT_FALSE(test, aa_unpack_array(puf->e, name, &array_size));  in policy_unpack_test_unpack_array_out_of_bounds()
179 	KUNIT_EXPECT_PTR_EQ(test, puf->e->pos,  in policy_unpack_test_unpack_array_out_of_bounds()
183 static void policy_unpack_test_unpack_blob_with_null_name(struct kunit *test)  in policy_unpack_test_unpack_blob_with_null_name()  argument
185 	struct policy_unpack_fixture *puf = test->priv;  in policy_unpack_test_unpack_blob_with_null_name()
192 	KUNIT_ASSERT_EQ(test, size, TEST_BLOB_DATA_SIZE);  in policy_unpack_test_unpack_blob_with_null_name()
193 	KUNIT_EXPECT_TRUE(test,  in policy_unpack_test_unpack_blob_with_null_name()
197 static void policy_unpack_test_unpack_blob_with_name(struct kunit *test)  in policy_unpack_test_unpack_blob_with_name()  argument
199 	struct policy_unpack_fixture *puf = test->priv;  in policy_unpack_test_unpack_blob_with_name()
206 	KUNIT_ASSERT_EQ(test, size, TEST_BLOB_DATA_SIZE);  in policy_unpack_test_unpack_blob_with_name()
207 	KUNIT_EXPECT_TRUE(test,  in policy_unpack_test_unpack_blob_with_name()
211 static void policy_unpack_test_unpack_blob_out_of_bounds(struct kunit *test)  in policy_unpack_test_unpack_blob_out_of_bounds()  argument
213 	struct policy_unpack_fixture *puf = test->priv;  in policy_unpack_test_unpack_blob_out_of_bounds()
225 	KUNIT_EXPECT_EQ(test, size, 0);  in policy_unpack_test_unpack_blob_out_of_bounds()
226 	KUNIT_EXPECT_PTR_EQ(test, puf->e->pos, start);  in policy_unpack_test_unpack_blob_out_of_bounds()
229 static void policy_unpack_test_unpack_str_with_null_name(struct kunit *test)  in policy_unpack_test_unpack_str_with_null_name()  argument
231 	struct policy_unpack_fixture *puf = test->priv;  in policy_unpack_test_unpack_str_with_null_name()
238 	KUNIT_EXPECT_EQ(test, size, strlen(TEST_STRING_DATA) + 1);  in policy_unpack_test_unpack_str_with_null_name()
239 	KUNIT_EXPECT_STREQ(test, string, TEST_STRING_DATA);  in policy_unpack_test_unpack_str_with_null_name()
242 static void policy_unpack_test_unpack_str_with_name(struct kunit *test)  in policy_unpack_test_unpack_str_with_name()  argument
244 	struct policy_unpack_fixture *puf = test->priv;  in policy_unpack_test_unpack_str_with_name()
250 	KUNIT_EXPECT_EQ(test, size, strlen(TEST_STRING_DATA) + 1);  in policy_unpack_test_unpack_str_with_name()
251 	KUNIT_EXPECT_STREQ(test, string, TEST_STRING_DATA);  in policy_unpack_test_unpack_str_with_name()
254 static void policy_unpack_test_unpack_str_out_of_bounds(struct kunit *test)  in policy_unpack_test_unpack_str_out_of_bounds()  argument
256 	struct policy_unpack_fixture *puf = test->priv;  in policy_unpack_test_unpack_str_out_of_bounds()
266 	KUNIT_EXPECT_EQ(test, size, 0);  in policy_unpack_test_unpack_str_out_of_bounds()
267 	KUNIT_EXPECT_PTR_EQ(test, puf->e->pos, start);  in policy_unpack_test_unpack_str_out_of_bounds()
270 static void policy_unpack_test_unpack_strdup_with_null_name(struct kunit *test)  in policy_unpack_test_unpack_strdup_with_null_name()  argument
272 	struct policy_unpack_fixture *puf = test->priv;  in policy_unpack_test_unpack_strdup_with_null_name()
279 	KUNIT_EXPECT_EQ(test, size, strlen(TEST_STRING_DATA) + 1);  in policy_unpack_test_unpack_strdup_with_null_name()
280 	KUNIT_EXPECT_FALSE(test,  in policy_unpack_test_unpack_strdup_with_null_name()
283 	KUNIT_EXPECT_STREQ(test, string, TEST_STRING_DATA);  in policy_unpack_test_unpack_strdup_with_null_name()
286 static void policy_unpack_test_unpack_strdup_with_name(struct kunit *test)  in policy_unpack_test_unpack_strdup_with_name()  argument
288 	struct policy_unpack_fixture *puf = test->priv;  in policy_unpack_test_unpack_strdup_with_name()
294 	KUNIT_EXPECT_EQ(test, size, strlen(TEST_STRING_DATA) + 1);  in policy_unpack_test_unpack_strdup_with_name()
295 	KUNIT_EXPECT_FALSE(test,  in policy_unpack_test_unpack_strdup_with_name()
298 	KUNIT_EXPECT_STREQ(test, string, TEST_STRING_DATA);  in policy_unpack_test_unpack_strdup_with_name()
301 static void policy_unpack_test_unpack_strdup_out_of_bounds(struct kunit *test)  in policy_unpack_test_unpack_strdup_out_of_bounds()  argument
303 	struct policy_unpack_fixture *puf = test->priv;  in policy_unpack_test_unpack_strdup_out_of_bounds()
313 	KUNIT_EXPECT_EQ(test, size, 0);  in policy_unpack_test_unpack_strdup_out_of_bounds()
314 	KUNIT_EXPECT_NULL(test, string);  in policy_unpack_test_unpack_strdup_out_of_bounds()
315 	KUNIT_EXPECT_PTR_EQ(test, puf->e->pos, start);  in policy_unpack_test_unpack_strdup_out_of_bounds()
318 static void policy_unpack_test_unpack_nameX_with_null_name(struct kunit *test)  in policy_unpack_test_unpack_nameX_with_null_name()  argument
320 	struct policy_unpack_fixture *puf = test->priv;  in policy_unpack_test_unpack_nameX_with_null_name()
327 	KUNIT_EXPECT_TRUE(test, success);  in policy_unpack_test_unpack_nameX_with_null_name()
328 	KUNIT_EXPECT_PTR_EQ(test, puf->e->pos,  in policy_unpack_test_unpack_nameX_with_null_name()
332 static void policy_unpack_test_unpack_nameX_with_wrong_code(struct kunit *test)  in policy_unpack_test_unpack_nameX_with_wrong_code()  argument
334 	struct policy_unpack_fixture *puf = test->priv;  in policy_unpack_test_unpack_nameX_with_wrong_code()
341 	KUNIT_EXPECT_FALSE(test, success);  in policy_unpack_test_unpack_nameX_with_wrong_code()
342 	KUNIT_EXPECT_PTR_EQ(test, puf->e->pos,  in policy_unpack_test_unpack_nameX_with_wrong_code()
346 static void policy_unpack_test_unpack_nameX_with_name(struct kunit *test)  in policy_unpack_test_unpack_nameX_with_name()  argument
348 	struct policy_unpack_fixture *puf = test->priv;  in policy_unpack_test_unpack_nameX_with_name()
356 	KUNIT_EXPECT_TRUE(test, success);  in policy_unpack_test_unpack_nameX_with_name()
357 	KUNIT_EXPECT_PTR_EQ(test, puf->e->pos,  in policy_unpack_test_unpack_nameX_with_name()
361 static void policy_unpack_test_unpack_nameX_with_wrong_name(struct kunit *test)  in policy_unpack_test_unpack_nameX_with_wrong_name()  argument
363 	struct policy_unpack_fixture *puf = test->priv;  in policy_unpack_test_unpack_nameX_with_wrong_name()
371 	KUNIT_EXPECT_FALSE(test, success);  in policy_unpack_test_unpack_nameX_with_wrong_name()
372 	KUNIT_EXPECT_PTR_EQ(test, puf->e->pos,  in policy_unpack_test_unpack_nameX_with_wrong_name()
376 static void policy_unpack_test_unpack_u16_chunk_basic(struct kunit *test)  in policy_unpack_test_unpack_u16_chunk_basic()  argument
378 	struct policy_unpack_fixture *puf = test->priv;  in policy_unpack_test_unpack_u16_chunk_basic()
392 	KUNIT_EXPECT_PTR_EQ(test, chunk,  in policy_unpack_test_unpack_u16_chunk_basic()
394 	KUNIT_EXPECT_EQ(test, size, TEST_U16_DATA);  in policy_unpack_test_unpack_u16_chunk_basic()
395 	KUNIT_EXPECT_PTR_EQ(test, puf->e->pos, (chunk + TEST_U16_DATA));  in policy_unpack_test_unpack_u16_chunk_basic()
399 		struct kunit *test)  in policy_unpack_test_unpack_u16_chunk_out_of_bounds_1()  argument
401 	struct policy_unpack_fixture *puf = test->priv;  in policy_unpack_test_unpack_u16_chunk_out_of_bounds_1()
409 	KUNIT_EXPECT_EQ(test, size, 0);  in policy_unpack_test_unpack_u16_chunk_out_of_bounds_1()
410 	KUNIT_EXPECT_NULL(test, chunk);  in policy_unpack_test_unpack_u16_chunk_out_of_bounds_1()
411 	KUNIT_EXPECT_PTR_EQ(test, puf->e->pos, puf->e->end - 1);  in policy_unpack_test_unpack_u16_chunk_out_of_bounds_1()
415 		struct kunit *test)  in policy_unpack_test_unpack_u16_chunk_out_of_bounds_2()  argument
417 	struct policy_unpack_fixture *puf = test->priv;  in policy_unpack_test_unpack_u16_chunk_out_of_bounds_2()
431 	KUNIT_EXPECT_EQ(test, size, 0);  in policy_unpack_test_unpack_u16_chunk_out_of_bounds_2()
432 	KUNIT_EXPECT_NULL(test, chunk);  in policy_unpack_test_unpack_u16_chunk_out_of_bounds_2()
433 	KUNIT_EXPECT_PTR_EQ(test, puf->e->pos, puf->e->start + TEST_U16_OFFSET);  in policy_unpack_test_unpack_u16_chunk_out_of_bounds_2()
436 static void policy_unpack_test_unpack_u32_with_null_name(struct kunit *test)  in policy_unpack_test_unpack_u32_with_null_name()  argument
438 	struct policy_unpack_fixture *puf = test->priv;  in policy_unpack_test_unpack_u32_with_null_name()
446 	KUNIT_EXPECT_TRUE(test, success);  in policy_unpack_test_unpack_u32_with_null_name()
447 	KUNIT_EXPECT_EQ(test, data, TEST_U32_DATA);  in policy_unpack_test_unpack_u32_with_null_name()
448 	KUNIT_EXPECT_PTR_EQ(test, puf->e->pos,  in policy_unpack_test_unpack_u32_with_null_name()
452 static void policy_unpack_test_unpack_u32_with_name(struct kunit *test)  in policy_unpack_test_unpack_u32_with_name()  argument
454 	struct policy_unpack_fixture *puf = test->priv;  in policy_unpack_test_unpack_u32_with_name()
463 	KUNIT_EXPECT_TRUE(test, success);  in policy_unpack_test_unpack_u32_with_name()
464 	KUNIT_EXPECT_EQ(test, data, TEST_U32_DATA);  in policy_unpack_test_unpack_u32_with_name()
465 	KUNIT_EXPECT_PTR_EQ(test, puf->e->pos,  in policy_unpack_test_unpack_u32_with_name()
469 static void policy_unpack_test_unpack_u32_out_of_bounds(struct kunit *test)  in policy_unpack_test_unpack_u32_out_of_bounds()  argument
471 	struct policy_unpack_fixture *puf = test->priv;  in policy_unpack_test_unpack_u32_out_of_bounds()
481 	KUNIT_EXPECT_FALSE(test, success);  in policy_unpack_test_unpack_u32_out_of_bounds()
482 	KUNIT_EXPECT_PTR_EQ(test, puf->e->pos,  in policy_unpack_test_unpack_u32_out_of_bounds()
486 static void policy_unpack_test_unpack_u64_with_null_name(struct kunit *test)  in policy_unpack_test_unpack_u64_with_null_name()  argument
488 	struct policy_unpack_fixture *puf = test->priv;  in policy_unpack_test_unpack_u64_with_null_name()
496 	KUNIT_EXPECT_TRUE(test, success);  in policy_unpack_test_unpack_u64_with_null_name()
497 	KUNIT_EXPECT_EQ(test, data, TEST_U64_DATA);  in policy_unpack_test_unpack_u64_with_null_name()
498 	KUNIT_EXPECT_PTR_EQ(test, puf->e->pos,  in policy_unpack_test_unpack_u64_with_null_name()
502 static void policy_unpack_test_unpack_u64_with_name(struct kunit *test)  in policy_unpack_test_unpack_u64_with_name()  argument
504 	struct policy_unpack_fixture *puf = test->priv;  in policy_unpack_test_unpack_u64_with_name()
513 	KUNIT_EXPECT_TRUE(test, success);  in policy_unpack_test_unpack_u64_with_name()
514 	KUNIT_EXPECT_EQ(test, data, TEST_U64_DATA);  in policy_unpack_test_unpack_u64_with_name()
515 	KUNIT_EXPECT_PTR_EQ(test, puf->e->pos,  in policy_unpack_test_unpack_u64_with_name()
519 static void policy_unpack_test_unpack_u64_out_of_bounds(struct kunit *test)  in policy_unpack_test_unpack_u64_out_of_bounds()  argument
521 	struct policy_unpack_fixture *puf = test->priv;  in policy_unpack_test_unpack_u64_out_of_bounds()
531 	KUNIT_EXPECT_FALSE(test, success);  in policy_unpack_test_unpack_u64_out_of_bounds()
532 	KUNIT_EXPECT_PTR_EQ(test, puf->e->pos,  in policy_unpack_test_unpack_u64_out_of_bounds()
536 static void policy_unpack_test_unpack_X_code_match(struct kunit *test)  in policy_unpack_test_unpack_X_code_match()  argument
538 	struct policy_unpack_fixture *puf = test->priv;  in policy_unpack_test_unpack_X_code_match()
541 	KUNIT_EXPECT_TRUE(test, success);  in policy_unpack_test_unpack_X_code_match()
542 	KUNIT_EXPECT_TRUE(test, puf->e->pos == puf->e->start + 1);  in policy_unpack_test_unpack_X_code_match()
545 static void policy_unpack_test_unpack_X_code_mismatch(struct kunit *test)  in policy_unpack_test_unpack_X_code_mismatch()  argument
547 	struct policy_unpack_fixture *puf = test->priv;  in policy_unpack_test_unpack_X_code_mismatch()
550 	KUNIT_EXPECT_FALSE(test, success);  in policy_unpack_test_unpack_X_code_mismatch()
551 	KUNIT_EXPECT_TRUE(test, puf->e->pos == puf->e->start);  in policy_unpack_test_unpack_X_code_mismatch()
554 static void policy_unpack_test_unpack_X_out_of_bounds(struct kunit *test)  in policy_unpack_test_unpack_X_out_of_bounds()  argument
556 	struct policy_unpack_fixture *puf = test->priv;  in policy_unpack_test_unpack_X_out_of_bounds()
562 	KUNIT_EXPECT_FALSE(test, success);  in policy_unpack_test_unpack_X_out_of_bounds()