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