117 lines · c
1// SPDX-License-Identifier: GPL-2.02/*3 * Test cases for memcat_p() in lib/memcat_p.c4 */5#define pr_fmt(fmt) KBUILD_MODNAME ": " fmt6 7#include <linux/string.h>8#include <linux/slab.h>9#include <linux/module.h>10 11struct test_struct {12 int num;13 unsigned int magic;14};15 16#define MAGIC 0xf00ff00f17/* Size of each of the NULL-terminated input arrays */18#define INPUT_MAX 12819/* Expected number of non-NULL elements in the output array */20#define EXPECT (INPUT_MAX * 2 - 2)21 22static 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 91err_free_out:92 kfree(out);93err_free_all_elements:94 i = INPUT_MAX;95err_free_elements:96 for (i--; i >= 0; i--) {97 kfree(in1[i]);98 kfree(in0[i]);99 }100 101 kfree(in1);102err_free_in0:103 kfree(in0);104 105 return err;106}107 108static void __exit test_memcat_p_exit(void)109{110}111 112module_init(test_memcat_p_init);113module_exit(test_memcat_p_exit);114 115MODULE_DESCRIPTION("Test cases for memcat_p() in lib/memcat_p.c");116MODULE_LICENSE("GPL");117