Commit | Line | Data |
---|---|---|
5fd003f5 DD |
1 | /* |
2 | * Test cases for printf facility. | |
3 | */ | |
4 | ||
5 | #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt | |
6 | ||
7 | #include <linux/bitmap.h> | |
8 | #include <linux/init.h> | |
9 | #include <linux/kernel.h> | |
10 | #include <linux/module.h> | |
11 | #include <linux/printk.h> | |
12 | #include <linux/slab.h> | |
13 | #include <linux/string.h> | |
14 | ||
15 | static unsigned total_tests __initdata; | |
16 | static unsigned failed_tests __initdata; | |
17 | ||
18 | static char pbl_buffer[PAGE_SIZE] __initdata; | |
19 | ||
20 | ||
21 | static bool __init | |
22 | __check_eq_uint(const char *srcfile, unsigned int line, | |
23 | const unsigned int exp_uint, unsigned int x) | |
24 | { | |
25 | if (exp_uint != x) { | |
3aa56885 | 26 | pr_err("[%s:%u] expected %u, got %u\n", |
5fd003f5 DD |
27 | srcfile, line, exp_uint, x); |
28 | return false; | |
29 | } | |
30 | return true; | |
31 | } | |
32 | ||
33 | ||
34 | static bool __init | |
35 | __check_eq_bitmap(const char *srcfile, unsigned int line, | |
3aa56885 YN |
36 | const unsigned long *exp_bmap, const unsigned long *bmap, |
37 | unsigned int nbits) | |
5fd003f5 | 38 | { |
5fd003f5 DD |
39 | if (!bitmap_equal(exp_bmap, bmap, nbits)) { |
40 | pr_warn("[%s:%u] bitmaps contents differ: expected \"%*pbl\", got \"%*pbl\"\n", | |
41 | srcfile, line, | |
3aa56885 | 42 | nbits, exp_bmap, nbits, bmap); |
5fd003f5 DD |
43 | return false; |
44 | } | |
45 | return true; | |
46 | } | |
47 | ||
48 | static bool __init | |
49 | __check_eq_pbl(const char *srcfile, unsigned int line, | |
50 | const char *expected_pbl, | |
51 | const unsigned long *bitmap, unsigned int nbits) | |
52 | { | |
53 | snprintf(pbl_buffer, sizeof(pbl_buffer), "%*pbl", nbits, bitmap); | |
54 | if (strcmp(expected_pbl, pbl_buffer)) { | |
55 | pr_warn("[%s:%u] expected \"%s\", got \"%s\"\n", | |
56 | srcfile, line, | |
57 | expected_pbl, pbl_buffer); | |
58 | return false; | |
59 | } | |
60 | return true; | |
61 | } | |
62 | ||
3aa56885 YN |
63 | static bool __init |
64 | __check_eq_u32_array(const char *srcfile, unsigned int line, | |
65 | const u32 *exp_arr, unsigned int exp_len, | |
66 | const u32 *arr, unsigned int len) __used; | |
5fd003f5 DD |
67 | static bool __init |
68 | __check_eq_u32_array(const char *srcfile, unsigned int line, | |
69 | const u32 *exp_arr, unsigned int exp_len, | |
70 | const u32 *arr, unsigned int len) | |
71 | { | |
72 | if (exp_len != len) { | |
73 | pr_warn("[%s:%u] array length differ: expected %u, got %u\n", | |
74 | srcfile, line, | |
75 | exp_len, len); | |
76 | return false; | |
77 | } | |
78 | ||
79 | if (memcmp(exp_arr, arr, len*sizeof(*arr))) { | |
80 | pr_warn("[%s:%u] array contents differ\n", srcfile, line); | |
81 | print_hex_dump(KERN_WARNING, " exp: ", DUMP_PREFIX_OFFSET, | |
82 | 32, 4, exp_arr, exp_len*sizeof(*exp_arr), false); | |
83 | print_hex_dump(KERN_WARNING, " got: ", DUMP_PREFIX_OFFSET, | |
84 | 32, 4, arr, len*sizeof(*arr), false); | |
85 | return false; | |
86 | } | |
87 | ||
88 | return true; | |
89 | } | |
90 | ||
91 | #define __expect_eq(suffix, ...) \ | |
92 | ({ \ | |
93 | int result = 0; \ | |
94 | total_tests++; \ | |
95 | if (!__check_eq_ ## suffix(__FILE__, __LINE__, \ | |
96 | ##__VA_ARGS__)) { \ | |
97 | failed_tests++; \ | |
98 | result = 1; \ | |
99 | } \ | |
100 | result; \ | |
101 | }) | |
102 | ||
103 | #define expect_eq_uint(...) __expect_eq(uint, ##__VA_ARGS__) | |
104 | #define expect_eq_bitmap(...) __expect_eq(bitmap, ##__VA_ARGS__) | |
105 | #define expect_eq_pbl(...) __expect_eq(pbl, ##__VA_ARGS__) | |
106 | #define expect_eq_u32_array(...) __expect_eq(u32_array, ##__VA_ARGS__) | |
107 | ||
ee3527bd AS |
108 | static void __init test_zero_clear(void) |
109 | { | |
110 | DECLARE_BITMAP(bmap, 1024); | |
111 | ||
112 | /* Known way to set all bits */ | |
113 | memset(bmap, 0xff, 128); | |
114 | ||
115 | expect_eq_pbl("0-22", bmap, 23); | |
116 | expect_eq_pbl("0-1023", bmap, 1024); | |
117 | ||
118 | /* single-word bitmaps */ | |
119 | bitmap_clear(bmap, 0, 9); | |
120 | expect_eq_pbl("9-1023", bmap, 1024); | |
121 | ||
122 | bitmap_zero(bmap, 35); | |
123 | expect_eq_pbl("64-1023", bmap, 1024); | |
124 | ||
125 | /* cross boundaries operations */ | |
126 | bitmap_clear(bmap, 79, 19); | |
127 | expect_eq_pbl("64-78,98-1023", bmap, 1024); | |
128 | ||
129 | bitmap_zero(bmap, 115); | |
130 | expect_eq_pbl("128-1023", bmap, 1024); | |
131 | ||
132 | /* Zeroing entire area */ | |
133 | bitmap_zero(bmap, 1024); | |
134 | expect_eq_pbl("", bmap, 1024); | |
135 | } | |
136 | ||
5fd003f5 DD |
137 | static void __init test_zero_fill_copy(void) |
138 | { | |
139 | DECLARE_BITMAP(bmap1, 1024); | |
140 | DECLARE_BITMAP(bmap2, 1024); | |
141 | ||
142 | bitmap_zero(bmap1, 1024); | |
143 | bitmap_zero(bmap2, 1024); | |
144 | ||
145 | /* single-word bitmaps */ | |
146 | expect_eq_pbl("", bmap1, 23); | |
147 | ||
148 | bitmap_fill(bmap1, 19); | |
149 | expect_eq_pbl("0-18", bmap1, 1024); | |
150 | ||
151 | bitmap_copy(bmap2, bmap1, 23); | |
152 | expect_eq_pbl("0-18", bmap2, 1024); | |
153 | ||
154 | bitmap_fill(bmap2, 23); | |
155 | expect_eq_pbl("0-22", bmap2, 1024); | |
156 | ||
157 | bitmap_copy(bmap2, bmap1, 23); | |
158 | expect_eq_pbl("0-18", bmap2, 1024); | |
159 | ||
160 | bitmap_zero(bmap1, 23); | |
161 | expect_eq_pbl("", bmap1, 1024); | |
162 | ||
163 | /* multi-word bitmaps */ | |
164 | bitmap_zero(bmap1, 1024); | |
165 | expect_eq_pbl("", bmap1, 1024); | |
166 | ||
167 | bitmap_fill(bmap1, 109); | |
168 | expect_eq_pbl("0-108", bmap1, 1024); | |
169 | ||
170 | bitmap_copy(bmap2, bmap1, 1024); | |
171 | expect_eq_pbl("0-108", bmap2, 1024); | |
172 | ||
173 | bitmap_fill(bmap2, 1024); | |
174 | expect_eq_pbl("0-1023", bmap2, 1024); | |
175 | ||
176 | bitmap_copy(bmap2, bmap1, 1024); | |
177 | expect_eq_pbl("0-108", bmap2, 1024); | |
178 | ||
179 | /* the following tests assume a 32- or 64-bit arch (even 128b | |
180 | * if we care) | |
181 | */ | |
182 | ||
183 | bitmap_fill(bmap2, 1024); | |
184 | bitmap_copy(bmap2, bmap1, 109); /* ... but 0-padded til word length */ | |
185 | expect_eq_pbl("0-108,128-1023", bmap2, 1024); | |
186 | ||
187 | bitmap_fill(bmap2, 1024); | |
188 | bitmap_copy(bmap2, bmap1, 97); /* ... but aligned on word length */ | |
189 | expect_eq_pbl("0-108,128-1023", bmap2, 1024); | |
190 | ||
191 | bitmap_zero(bmap2, 97); /* ... but 0-padded til word length */ | |
192 | expect_eq_pbl("128-1023", bmap2, 1024); | |
193 | } | |
194 | ||
6df0d464 YN |
195 | #define PARSE_TIME 0x1 |
196 | ||
197 | struct test_bitmap_parselist{ | |
198 | const int errno; | |
199 | const char *in; | |
200 | const unsigned long *expected; | |
201 | const int nbits; | |
202 | const int flags; | |
203 | }; | |
204 | ||
60ef6900 YN |
205 | static const unsigned long exp[] __initconst = { |
206 | BITMAP_FROM_U64(1), | |
207 | BITMAP_FROM_U64(2), | |
208 | BITMAP_FROM_U64(0x0000ffff), | |
209 | BITMAP_FROM_U64(0xffff0000), | |
210 | BITMAP_FROM_U64(0x55555555), | |
211 | BITMAP_FROM_U64(0xaaaaaaaa), | |
212 | BITMAP_FROM_U64(0x11111111), | |
213 | BITMAP_FROM_U64(0x22222222), | |
214 | BITMAP_FROM_U64(0xffffffff), | |
215 | BITMAP_FROM_U64(0xfffffffe), | |
8185f570 GU |
216 | BITMAP_FROM_U64(0x3333333311111111ULL), |
217 | BITMAP_FROM_U64(0xffffffff77777777ULL) | |
60ef6900 YN |
218 | }; |
219 | ||
220 | static const unsigned long exp2[] __initconst = { | |
8185f570 GU |
221 | BITMAP_FROM_U64(0x3333333311111111ULL), |
222 | BITMAP_FROM_U64(0xffffffff77777777ULL) | |
60ef6900 | 223 | }; |
6df0d464 YN |
224 | |
225 | static const struct test_bitmap_parselist parselist_tests[] __initconst = { | |
60ef6900 YN |
226 | #define step (sizeof(u64) / sizeof(unsigned long)) |
227 | ||
6df0d464 | 228 | {0, "0", &exp[0], 8, 0}, |
60ef6900 YN |
229 | {0, "1", &exp[1 * step], 8, 0}, |
230 | {0, "0-15", &exp[2 * step], 32, 0}, | |
231 | {0, "16-31", &exp[3 * step], 32, 0}, | |
232 | {0, "0-31:1/2", &exp[4 * step], 32, 0}, | |
233 | {0, "1-31:1/2", &exp[5 * step], 32, 0}, | |
234 | {0, "0-31:1/4", &exp[6 * step], 32, 0}, | |
235 | {0, "1-31:1/4", &exp[7 * step], 32, 0}, | |
236 | {0, "0-31:4/4", &exp[8 * step], 32, 0}, | |
237 | {0, "1-31:4/4", &exp[9 * step], 32, 0}, | |
238 | {0, "0-31:1/4,32-63:2/4", &exp[10 * step], 64, 0}, | |
239 | {0, "0-31:3/4,32-63:4/4", &exp[11 * step], 64, 0}, | |
6df0d464 YN |
240 | |
241 | {0, "0-31:1/4,32-63:2/4,64-95:3/4,96-127:4/4", exp2, 128, 0}, | |
242 | ||
243 | {0, "0-2047:128/256", NULL, 2048, PARSE_TIME}, | |
244 | ||
245 | {-EINVAL, "-1", NULL, 8, 0}, | |
246 | {-EINVAL, "-0", NULL, 8, 0}, | |
247 | {-EINVAL, "10-1", NULL, 8, 0}, | |
248 | {-EINVAL, "0-31:10/1", NULL, 8, 0}, | |
249 | }; | |
250 | ||
251 | static void __init test_bitmap_parselist(void) | |
252 | { | |
253 | int i; | |
254 | int err; | |
255 | cycles_t cycles; | |
256 | DECLARE_BITMAP(bmap, 2048); | |
257 | ||
258 | for (i = 0; i < ARRAY_SIZE(parselist_tests); i++) { | |
259 | #define ptest parselist_tests[i] | |
260 | ||
261 | cycles = get_cycles(); | |
262 | err = bitmap_parselist(ptest.in, bmap, ptest.nbits); | |
263 | cycles = get_cycles() - cycles; | |
264 | ||
265 | if (err != ptest.errno) { | |
266 | pr_err("test %d: input is %s, errno is %d, expected %d\n", | |
267 | i, ptest.in, err, ptest.errno); | |
268 | continue; | |
269 | } | |
270 | ||
271 | if (!err && ptest.expected | |
272 | && !__bitmap_equal(bmap, ptest.expected, ptest.nbits)) { | |
273 | pr_err("test %d: input is %s, result is 0x%lx, expected 0x%lx\n", | |
274 | i, ptest.in, bmap[0], *ptest.expected); | |
275 | continue; | |
276 | } | |
277 | ||
278 | if (ptest.flags & PARSE_TIME) | |
279 | pr_err("test %d: input is '%s' OK, Time: %llu\n", | |
280 | i, ptest.in, | |
281 | (unsigned long long)cycles); | |
282 | } | |
283 | } | |
284 | ||
3aa56885 | 285 | static void __init test_bitmap_arr32(void) |
5fd003f5 | 286 | { |
3aa56885 YN |
287 | unsigned int nbits, next_bit, len = sizeof(exp) * 8; |
288 | u32 arr[sizeof(exp) / 4]; | |
289 | DECLARE_BITMAP(bmap2, len); | |
290 | ||
291 | memset(arr, 0xa5, sizeof(arr)); | |
292 | ||
293 | for (nbits = 0; nbits < len; ++nbits) { | |
294 | bitmap_to_arr32(arr, exp, nbits); | |
295 | bitmap_from_arr32(bmap2, arr, nbits); | |
296 | expect_eq_bitmap(bmap2, exp, nbits); | |
297 | ||
298 | next_bit = find_next_bit(bmap2, | |
299 | round_up(nbits, BITS_PER_LONG), nbits); | |
300 | if (next_bit < round_up(nbits, BITS_PER_LONG)) | |
301 | pr_err("bitmap_copy_arr32(nbits == %d:" | |
302 | " tail is not safely cleared: %d\n", | |
303 | nbits, next_bit); | |
304 | ||
305 | if (nbits < len - 32) | |
306 | expect_eq_uint(arr[DIV_ROUND_UP(nbits, 32)], | |
307 | 0xa5a5a5a5); | |
5fd003f5 DD |
308 | } |
309 | } | |
310 | ||
3cc78125 MW |
311 | static void noinline __init test_mem_optimisations(void) |
312 | { | |
313 | DECLARE_BITMAP(bmap1, 1024); | |
314 | DECLARE_BITMAP(bmap2, 1024); | |
315 | unsigned int start, nbits; | |
316 | ||
317 | for (start = 0; start < 1024; start += 8) { | |
318 | memset(bmap1, 0x5a, sizeof(bmap1)); | |
319 | memset(bmap2, 0x5a, sizeof(bmap2)); | |
320 | for (nbits = 0; nbits < 1024 - start; nbits += 8) { | |
321 | bitmap_set(bmap1, start, nbits); | |
322 | __bitmap_set(bmap2, start, nbits); | |
323 | if (!bitmap_equal(bmap1, bmap2, 1024)) | |
324 | printk("set not equal %d %d\n", start, nbits); | |
325 | if (!__bitmap_equal(bmap1, bmap2, 1024)) | |
326 | printk("set not __equal %d %d\n", start, nbits); | |
327 | ||
328 | bitmap_clear(bmap1, start, nbits); | |
329 | __bitmap_clear(bmap2, start, nbits); | |
330 | if (!bitmap_equal(bmap1, bmap2, 1024)) | |
331 | printk("clear not equal %d %d\n", start, nbits); | |
332 | if (!__bitmap_equal(bmap1, bmap2, 1024)) | |
333 | printk("clear not __equal %d %d\n", start, | |
334 | nbits); | |
335 | } | |
336 | } | |
337 | } | |
338 | ||
5fd003f5 DD |
339 | static int __init test_bitmap_init(void) |
340 | { | |
ee3527bd | 341 | test_zero_clear(); |
5fd003f5 | 342 | test_zero_fill_copy(); |
3aa56885 | 343 | test_bitmap_arr32(); |
6df0d464 | 344 | test_bitmap_parselist(); |
3cc78125 | 345 | test_mem_optimisations(); |
5fd003f5 DD |
346 | |
347 | if (failed_tests == 0) | |
348 | pr_info("all %u tests passed\n", total_tests); | |
349 | else | |
350 | pr_warn("failed %u out of %u tests\n", | |
351 | failed_tests, total_tests); | |
352 | ||
353 | return failed_tests ? -EINVAL : 0; | |
354 | } | |
355 | ||
356 | static void __exit test_bitmap_cleanup(void) | |
357 | { | |
358 | } | |
359 | ||
360 | module_init(test_bitmap_init); | |
361 | module_exit(test_bitmap_cleanup); | |
362 | ||
363 | MODULE_AUTHOR("david decotigny <david.decotigny@googlers.com>"); | |
364 | MODULE_LICENSE("GPL"); |