1*4882a593Smuzhiyun // SPDX-License-Identifier: GPL-2.0
2*4882a593Smuzhiyun /*
3*4882a593Smuzhiyun * Test cases for memcat_p() in lib/memcat_p.c
4*4882a593Smuzhiyun */
5*4882a593Smuzhiyun #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
6*4882a593Smuzhiyun
7*4882a593Smuzhiyun #include <linux/string.h>
8*4882a593Smuzhiyun #include <linux/slab.h>
9*4882a593Smuzhiyun #include <linux/module.h>
10*4882a593Smuzhiyun
11*4882a593Smuzhiyun struct test_struct {
12*4882a593Smuzhiyun int num;
13*4882a593Smuzhiyun unsigned int magic;
14*4882a593Smuzhiyun };
15*4882a593Smuzhiyun
16*4882a593Smuzhiyun #define MAGIC 0xf00ff00f
17*4882a593Smuzhiyun /* Size of each of the NULL-terminated input arrays */
18*4882a593Smuzhiyun #define INPUT_MAX 128
19*4882a593Smuzhiyun /* Expected number of non-NULL elements in the output array */
20*4882a593Smuzhiyun #define EXPECT (INPUT_MAX * 2 - 2)
21*4882a593Smuzhiyun
test_memcat_p_init(void)22*4882a593Smuzhiyun static int __init test_memcat_p_init(void)
23*4882a593Smuzhiyun {
24*4882a593Smuzhiyun struct test_struct **in0, **in1, **out, **p;
25*4882a593Smuzhiyun int err = -ENOMEM, i, r, total = 0;
26*4882a593Smuzhiyun
27*4882a593Smuzhiyun in0 = kcalloc(INPUT_MAX, sizeof(*in0), GFP_KERNEL);
28*4882a593Smuzhiyun if (!in0)
29*4882a593Smuzhiyun return err;
30*4882a593Smuzhiyun
31*4882a593Smuzhiyun in1 = kcalloc(INPUT_MAX, sizeof(*in1), GFP_KERNEL);
32*4882a593Smuzhiyun if (!in1)
33*4882a593Smuzhiyun goto err_free_in0;
34*4882a593Smuzhiyun
35*4882a593Smuzhiyun for (i = 0, r = 1; i < INPUT_MAX - 1; i++) {
36*4882a593Smuzhiyun in0[i] = kmalloc(sizeof(**in0), GFP_KERNEL);
37*4882a593Smuzhiyun if (!in0[i])
38*4882a593Smuzhiyun goto err_free_elements;
39*4882a593Smuzhiyun
40*4882a593Smuzhiyun in1[i] = kmalloc(sizeof(**in1), GFP_KERNEL);
41*4882a593Smuzhiyun if (!in1[i]) {
42*4882a593Smuzhiyun kfree(in0[i]);
43*4882a593Smuzhiyun goto err_free_elements;
44*4882a593Smuzhiyun }
45*4882a593Smuzhiyun
46*4882a593Smuzhiyun /* lifted from test_sort.c */
47*4882a593Smuzhiyun r = (r * 725861) % 6599;
48*4882a593Smuzhiyun in0[i]->num = r;
49*4882a593Smuzhiyun in1[i]->num = -r;
50*4882a593Smuzhiyun in0[i]->magic = MAGIC;
51*4882a593Smuzhiyun in1[i]->magic = MAGIC;
52*4882a593Smuzhiyun }
53*4882a593Smuzhiyun
54*4882a593Smuzhiyun in0[i] = in1[i] = NULL;
55*4882a593Smuzhiyun
56*4882a593Smuzhiyun out = memcat_p(in0, in1);
57*4882a593Smuzhiyun if (!out)
58*4882a593Smuzhiyun goto err_free_all_elements;
59*4882a593Smuzhiyun
60*4882a593Smuzhiyun err = -EINVAL;
61*4882a593Smuzhiyun for (i = 0, p = out; *p && (i < INPUT_MAX * 2 - 1); p++, i++) {
62*4882a593Smuzhiyun total += (*p)->num;
63*4882a593Smuzhiyun
64*4882a593Smuzhiyun if ((*p)->magic != MAGIC) {
65*4882a593Smuzhiyun pr_err("test failed: wrong magic at %d: %u\n", i,
66*4882a593Smuzhiyun (*p)->magic);
67*4882a593Smuzhiyun goto err_free_out;
68*4882a593Smuzhiyun }
69*4882a593Smuzhiyun }
70*4882a593Smuzhiyun
71*4882a593Smuzhiyun if (total) {
72*4882a593Smuzhiyun pr_err("test failed: expected zero total, got %d\n", total);
73*4882a593Smuzhiyun goto err_free_out;
74*4882a593Smuzhiyun }
75*4882a593Smuzhiyun
76*4882a593Smuzhiyun if (i != EXPECT) {
77*4882a593Smuzhiyun pr_err("test failed: expected output size %d, got %d\n",
78*4882a593Smuzhiyun EXPECT, i);
79*4882a593Smuzhiyun goto err_free_out;
80*4882a593Smuzhiyun }
81*4882a593Smuzhiyun
82*4882a593Smuzhiyun for (i = 0; i < INPUT_MAX - 1; i++)
83*4882a593Smuzhiyun if (out[i] != in0[i] || out[i + INPUT_MAX - 1] != in1[i]) {
84*4882a593Smuzhiyun pr_err("test failed: wrong element order at %d\n", i);
85*4882a593Smuzhiyun goto err_free_out;
86*4882a593Smuzhiyun }
87*4882a593Smuzhiyun
88*4882a593Smuzhiyun err = 0;
89*4882a593Smuzhiyun pr_info("test passed\n");
90*4882a593Smuzhiyun
91*4882a593Smuzhiyun err_free_out:
92*4882a593Smuzhiyun kfree(out);
93*4882a593Smuzhiyun err_free_all_elements:
94*4882a593Smuzhiyun i = INPUT_MAX;
95*4882a593Smuzhiyun err_free_elements:
96*4882a593Smuzhiyun for (i--; i >= 0; i--) {
97*4882a593Smuzhiyun kfree(in1[i]);
98*4882a593Smuzhiyun kfree(in0[i]);
99*4882a593Smuzhiyun }
100*4882a593Smuzhiyun
101*4882a593Smuzhiyun kfree(in1);
102*4882a593Smuzhiyun err_free_in0:
103*4882a593Smuzhiyun kfree(in0);
104*4882a593Smuzhiyun
105*4882a593Smuzhiyun return err;
106*4882a593Smuzhiyun }
107*4882a593Smuzhiyun
test_memcat_p_exit(void)108*4882a593Smuzhiyun static void __exit test_memcat_p_exit(void)
109*4882a593Smuzhiyun {
110*4882a593Smuzhiyun }
111*4882a593Smuzhiyun
112*4882a593Smuzhiyun module_init(test_memcat_p_init);
113*4882a593Smuzhiyun module_exit(test_memcat_p_exit);
114*4882a593Smuzhiyun
115*4882a593Smuzhiyun MODULE_LICENSE("GPL");
116