Eric Lee / smarc-fsl-linux-kernel

Blame view

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