Blame view

lib/test-string_helpers.c 11.7 KB
16c7fa058   Andy Shevchenko   lib/string_helper...
1
2
3
4
5
6
7
  /*
   * Test cases for lib/string_helpers.c module.
   */
  #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
  
  #include <linux/init.h>
  #include <linux/kernel.h>
c8250381c   Andy Shevchenko   lib / string_help...
8
  #include <linux/slab.h>
16c7fa058   Andy Shevchenko   lib/string_helper...
9
10
11
12
  #include <linux/module.h>
  #include <linux/random.h>
  #include <linux/string.h>
  #include <linux/string_helpers.h>
45ff337a5   Andy Shevchenko   lib / string_help...
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
  static __init bool test_string_check_buf(const char *name, unsigned int flags,
  					 char *in, size_t p,
  					 char *out_real, size_t q_real,
  					 char *out_test, size_t q_test)
  {
  	if (q_real == q_test && !memcmp(out_test, out_real, q_test))
  		return true;
  
  	pr_warn("Test '%s' failed: flags = %u
  ", name, flags);
  
  	print_hex_dump(KERN_WARNING, "Input: ", DUMP_PREFIX_NONE, 16, 1,
  		       in, p, true);
  	print_hex_dump(KERN_WARNING, "Expected: ", DUMP_PREFIX_NONE, 16, 1,
  		       out_test, q_test, true);
  	print_hex_dump(KERN_WARNING, "Got: ", DUMP_PREFIX_NONE, 16, 1,
  		       out_real, q_real, true);
  
  	return false;
  }
16c7fa058   Andy Shevchenko   lib/string_helper...
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
  struct test_string {
  	const char *in;
  	const char *out;
  	unsigned int flags;
  };
  
  static const struct test_string strings[] __initconst = {
  	{
  		.in = "\\f\\ \
  \\r\\t\\v",
  		.out = "\f\\ 
  \r\t\v",
  		.flags = UNESCAPE_SPACE,
  	},
  	{
  		.in = "\\40\\1\\387\\0064\\05\\040\\8a\\110\\777",
  		.out = " \001\00387\0064\005 \\8aH?7",
  		.flags = UNESCAPE_OCTAL,
  	},
  	{
  		.in = "\\xv\\xa\\x2c\\xD\\x6f2",
  		.out = "\\xv
  ,\ro2",
  		.flags = UNESCAPE_HEX,
  	},
  	{
  		.in = "\\h\\\\\\\"\\a\\e\\",
  		.out = "\\h\\\"\a\e\\",
  		.flags = UNESCAPE_SPECIAL,
  	},
  };
45ff337a5   Andy Shevchenko   lib / string_help...
64
65
  static void __init test_string_unescape(const char *name, unsigned int flags,
  					bool inplace)
16c7fa058   Andy Shevchenko   lib/string_helper...
66
  {
c8250381c   Andy Shevchenko   lib / string_help...
67
68
69
70
71
72
73
74
  	int q_real = 256;
  	char *in = kmalloc(q_real, GFP_KERNEL);
  	char *out_test = kmalloc(q_real, GFP_KERNEL);
  	char *out_real = kmalloc(q_real, GFP_KERNEL);
  	int i, p = 0, q_test = 0;
  
  	if (!in || !out_test || !out_real)
  		goto out;
16c7fa058   Andy Shevchenko   lib/string_helper...
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
  
  	for (i = 0; i < ARRAY_SIZE(strings); i++) {
  		const char *s = strings[i].in;
  		int len = strlen(strings[i].in);
  
  		/* Copy string to in buffer */
  		memcpy(&in[p], s, len);
  		p += len;
  
  		/* Copy expected result for given flags */
  		if (flags & strings[i].flags) {
  			s = strings[i].out;
  			len = strlen(strings[i].out);
  		}
  		memcpy(&out_test[q_test], s, len);
  		q_test += len;
  	}
  	in[p++] = '\0';
  
  	/* Call string_unescape and compare result */
  	if (inplace) {
  		memcpy(out_real, in, p);
  		if (flags == UNESCAPE_ANY)
  			q_real = string_unescape_any_inplace(out_real);
  		else
  			q_real = string_unescape_inplace(out_real, flags);
  	} else if (flags == UNESCAPE_ANY) {
  		q_real = string_unescape_any(in, out_real, q_real);
  	} else {
  		q_real = string_unescape(in, out_real, q_real, flags);
  	}
45ff337a5   Andy Shevchenko   lib / string_help...
106
107
  	test_string_check_buf(name, flags, in, p - 1, out_real, q_real,
  			      out_test, q_test);
c8250381c   Andy Shevchenko   lib / string_help...
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
  out:
  	kfree(out_real);
  	kfree(out_test);
  	kfree(in);
  }
  
  struct test_string_1 {
  	const char *out;
  	unsigned int flags;
  };
  
  #define	TEST_STRING_2_MAX_S1		32
  struct test_string_2 {
  	const char *in;
  	struct test_string_1 s1[TEST_STRING_2_MAX_S1];
  };
  
  #define	TEST_STRING_2_DICT_0		NULL
  static const struct test_string_2 escape0[] __initconst = {{
  	.in = "\f\\ 
  \r\t\v",
  	.s1 = {{
  		.out = "\\f\\ \
  \\r\\t\\v",
  		.flags = ESCAPE_SPACE,
  	},{
  		.out = "\\f\\134\\040\
  \\r\\t\\v",
  		.flags = ESCAPE_SPACE | ESCAPE_OCTAL,
  	},{
  		.out = "\\f\\x5c\\x20\
  \\r\\t\\v",
  		.flags = ESCAPE_SPACE | ESCAPE_HEX,
  	},{
  		/* terminator */
  	}},
  },{
  	.in = "\\h\\\"\a\e\\",
  	.s1 = {{
  		.out = "\\\\h\\\\\"\\a\\e\\\\",
  		.flags = ESCAPE_SPECIAL,
  	},{
  		.out = "\\\\\\150\\\\\\042\\a\\e\\\\",
  		.flags = ESCAPE_SPECIAL | ESCAPE_OCTAL,
  	},{
  		.out = "\\\\\\x68\\\\\\x22\\a\\e\\\\",
  		.flags = ESCAPE_SPECIAL | ESCAPE_HEX,
  	},{
  		/* terminator */
  	}},
  },{
  	.in = "\eb \\C\007\"\x90\r]",
  	.s1 = {{
  		.out = "\eb \\C\007\"\x90\\r]",
  		.flags = ESCAPE_SPACE,
  	},{
  		.out = "\\eb \\\\C\\a\"\x90\r]",
  		.flags = ESCAPE_SPECIAL,
  	},{
  		.out = "\\eb \\\\C\\a\"\x90\\r]",
  		.flags = ESCAPE_SPACE | ESCAPE_SPECIAL,
  	},{
  		.out = "\\033\\142\\040\\134\\103\\007\\042\\220\\015\\135",
  		.flags = ESCAPE_OCTAL,
  	},{
  		.out = "\\033\\142\\040\\134\\103\\007\\042\\220\\r\\135",
  		.flags = ESCAPE_SPACE | ESCAPE_OCTAL,
  	},{
  		.out = "\\e\\142\\040\\\\\\103\\a\\042\\220\\015\\135",
  		.flags = ESCAPE_SPECIAL | ESCAPE_OCTAL,
  	},{
  		.out = "\\e\\142\\040\\\\\\103\\a\\042\\220\\r\\135",
  		.flags = ESCAPE_SPACE | ESCAPE_SPECIAL | ESCAPE_OCTAL,
  	},{
  		.out = "\eb \\C\007\"\x90\r]",
  		.flags = ESCAPE_NP,
  	},{
  		.out = "\eb \\C\007\"\x90\\r]",
  		.flags = ESCAPE_SPACE | ESCAPE_NP,
  	},{
  		.out = "\\eb \\C\\a\"\x90\r]",
  		.flags = ESCAPE_SPECIAL | ESCAPE_NP,
  	},{
  		.out = "\\eb \\C\\a\"\x90\\r]",
  		.flags = ESCAPE_SPACE | ESCAPE_SPECIAL | ESCAPE_NP,
  	},{
  		.out = "\\033b \\C\\007\"\\220\\015]",
  		.flags = ESCAPE_OCTAL | ESCAPE_NP,
  	},{
  		.out = "\\033b \\C\\007\"\\220\\r]",
  		.flags = ESCAPE_SPACE | ESCAPE_OCTAL | ESCAPE_NP,
  	},{
  		.out = "\\eb \\C\\a\"\\220\\r]",
  		.flags = ESCAPE_SPECIAL | ESCAPE_SPACE | ESCAPE_OCTAL |
  			 ESCAPE_NP,
  	},{
  		.out = "\\x1bb \\C\\x07\"\\x90\\x0d]",
  		.flags = ESCAPE_NP | ESCAPE_HEX,
  	},{
  		/* terminator */
  	}},
  },{
  	/* terminator */
  }};
  
  #define	TEST_STRING_2_DICT_1		"b\\ \t\r"
  static const struct test_string_2 escape1[] __initconst = {{
  	.in = "\f\\ 
  \r\t\v",
  	.s1 = {{
  		.out = "\f\\134\\040
  \\015\\011\v",
  		.flags = ESCAPE_OCTAL,
  	},{
  		.out = "\f\\x5c\\x20
  \\x0d\\x09\v",
  		.flags = ESCAPE_HEX,
  	},{
  		/* terminator */
  	}},
  },{
  	.in = "\\h\\\"\a\e\\",
  	.s1 = {{
  		.out = "\\134h\\134\"\a\e\\134",
  		.flags = ESCAPE_OCTAL,
  	},{
  		/* terminator */
  	}},
  },{
  	.in = "\eb \\C\007\"\x90\r]",
  	.s1 = {{
  		.out = "\e\\142\\040\\134C\007\"\x90\\015]",
  		.flags = ESCAPE_OCTAL,
  	},{
  		/* terminator */
  	}},
  },{
  	/* terminator */
  }};
1e0f548b2   Vadim Pasternak   lib/test-string_h...
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
  static const struct test_string strings_upper[] __initconst = {
  	{
  		.in = "abcdefgh1234567890test",
  		.out = "ABCDEFGH1234567890TEST",
  	},
  	{
  		.in = "abCdeFgH1234567890TesT",
  		.out = "ABCDEFGH1234567890TEST",
  	},
  };
  
  static const struct test_string strings_lower[] __initconst = {
  	{
  		.in = "ABCDEFGH1234567890TEST",
  		.out = "abcdefgh1234567890test",
  	},
  	{
  		.in = "abCdeFgH1234567890TesT",
  		.out = "abcdefgh1234567890test",
  	},
  };
c8250381c   Andy Shevchenko   lib / string_help...
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
  static __init const char *test_string_find_match(const struct test_string_2 *s2,
  						 unsigned int flags)
  {
  	const struct test_string_1 *s1 = s2->s1;
  	unsigned int i;
  
  	if (!flags)
  		return s2->in;
  
  	/* Test cases are NULL-aware */
  	flags &= ~ESCAPE_NULL;
  
  	/* ESCAPE_OCTAL has a higher priority */
  	if (flags & ESCAPE_OCTAL)
  		flags &= ~ESCAPE_HEX;
  
  	for (i = 0; i < TEST_STRING_2_MAX_S1 && s1->out; i++, s1++)
  		if (s1->flags == flags)
  			return s1->out;
  	return NULL;
  }
41416f233   Rasmus Villemoes   lib/string_helper...
289
290
291
292
293
294
295
296
297
298
299
300
  static __init void
  test_string_escape_overflow(const char *in, int p, unsigned int flags, const char *esc,
  			    int q_test, const char *name)
  {
  	int q_real;
  
  	q_real = string_escape_mem(in, p, NULL, 0, flags, esc);
  	if (q_real != q_test)
  		pr_warn("Test '%s' failed: flags = %u, osz = 0, expected %d, got %d
  ",
  			name, flags, q_test, q_real);
  }
c8250381c   Andy Shevchenko   lib / string_help...
301
302
303
304
  static __init void test_string_escape(const char *name,
  				      const struct test_string_2 *s2,
  				      unsigned int flags, const char *esc)
  {
41416f233   Rasmus Villemoes   lib/string_helper...
305
306
307
  	size_t out_size = 512;
  	char *out_test = kmalloc(out_size, GFP_KERNEL);
  	char *out_real = kmalloc(out_size, GFP_KERNEL);
c8250381c   Andy Shevchenko   lib / string_help...
308
  	char *in = kmalloc(256, GFP_KERNEL);
c8250381c   Andy Shevchenko   lib / string_help...
309
  	int p = 0, q_test = 0;
41416f233   Rasmus Villemoes   lib/string_helper...
310
  	int q_real;
c8250381c   Andy Shevchenko   lib / string_help...
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
  
  	if (!out_test || !out_real || !in)
  		goto out;
  
  	for (; s2->in; s2++) {
  		const char *out;
  		int len;
  
  		/* NULL injection */
  		if (flags & ESCAPE_NULL) {
  			in[p++] = '\0';
  			out_test[q_test++] = '\\';
  			out_test[q_test++] = '0';
  		}
  
  		/* Don't try strings that have no output */
  		out = test_string_find_match(s2, flags);
  		if (!out)
  			continue;
  
  		/* Copy string to in buffer */
  		len = strlen(s2->in);
  		memcpy(&in[p], s2->in, len);
  		p += len;
  
  		/* Copy expected result for given flags */
  		len = strlen(out);
  		memcpy(&out_test[q_test], out, len);
  		q_test += len;
  	}
41416f233   Rasmus Villemoes   lib/string_helper...
341
  	q_real = string_escape_mem(in, p, out_real, out_size, flags, esc);
c8250381c   Andy Shevchenko   lib / string_help...
342
343
344
  
  	test_string_check_buf(name, flags, in, p, out_real, q_real, out_test,
  			      q_test);
41416f233   Rasmus Villemoes   lib/string_helper...
345
346
  
  	test_string_escape_overflow(in, p, flags, esc, q_test, name);
c8250381c   Andy Shevchenko   lib / string_help...
347
348
349
350
351
  out:
  	kfree(in);
  	kfree(out_real);
  	kfree(out_test);
  }
943ba6503   Vitaly Kuznetsov   lib/test-string_h...
352
  #define string_get_size_maxbuf 16
72676bb53   Vitaly Kuznetsov   lib/test-string_h...
353
  #define test_string_get_size_one(size, blk_size, exp_result10, exp_result2)    \
943ba6503   Vitaly Kuznetsov   lib/test-string_h...
354
  	do {                                                                   \
72676bb53   Vitaly Kuznetsov   lib/test-string_h...
355
356
357
358
  		BUILD_BUG_ON(sizeof(exp_result10) >= string_get_size_maxbuf);  \
  		BUILD_BUG_ON(sizeof(exp_result2) >= string_get_size_maxbuf);   \
  		__test_string_get_size((size), (blk_size), (exp_result10),     \
  				       (exp_result2));                         \
943ba6503   Vitaly Kuznetsov   lib/test-string_h...
359
  	} while (0)
72676bb53   Vitaly Kuznetsov   lib/test-string_h...
360
361
362
363
364
  static __init void test_string_get_size_check(const char *units,
  					      const char *exp,
  					      char *res,
  					      const u64 size,
  					      const u64 blk_size)
943ba6503   Vitaly Kuznetsov   lib/test-string_h...
365
  {
72676bb53   Vitaly Kuznetsov   lib/test-string_h...
366
  	if (!memcmp(res, exp, strlen(exp) + 1))
943ba6503   Vitaly Kuznetsov   lib/test-string_h...
367
  		return;
72676bb53   Vitaly Kuznetsov   lib/test-string_h...
368
369
370
371
372
373
  	res[string_get_size_maxbuf - 1] = '\0';
  
  	pr_warn("Test 'test_string_get_size' failed!
  ");
  	pr_warn("string_get_size(size = %llu, blk_size = %llu, units = %s)
  ",
943ba6503   Vitaly Kuznetsov   lib/test-string_h...
374
  		size, blk_size, units);
72676bb53   Vitaly Kuznetsov   lib/test-string_h...
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
  	pr_warn("expected: '%s', got '%s'
  ", exp, res);
  }
  
  static __init void __test_string_get_size(const u64 size, const u64 blk_size,
  					  const char *exp_result10,
  					  const char *exp_result2)
  {
  	char buf10[string_get_size_maxbuf];
  	char buf2[string_get_size_maxbuf];
  
  	string_get_size(size, blk_size, STRING_UNITS_10, buf10, sizeof(buf10));
  	string_get_size(size, blk_size, STRING_UNITS_2, buf2, sizeof(buf2));
  
  	test_string_get_size_check("STRING_UNITS_10", exp_result10, buf10,
  				   size, blk_size);
  
  	test_string_get_size_check("STRING_UNITS_2", exp_result2, buf2,
  				   size, blk_size);
943ba6503   Vitaly Kuznetsov   lib/test-string_h...
394
395
396
397
  }
  
  static __init void test_string_get_size(void)
  {
72676bb53   Vitaly Kuznetsov   lib/test-string_h...
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
  	/* small values */
  	test_string_get_size_one(0, 512, "0 B", "0 B");
  	test_string_get_size_one(1, 512, "512 B", "512 B");
  	test_string_get_size_one(1100, 1, "1.10 kB", "1.07 KiB");
  
  	/* normal values */
  	test_string_get_size_one(16384, 512, "8.39 MB", "8.00 MiB");
  	test_string_get_size_one(500118192, 512, "256 GB", "238 GiB");
  	test_string_get_size_one(8192, 4096, "33.6 MB", "32.0 MiB");
  
  	/* weird block sizes */
  	test_string_get_size_one(3000, 1900, "5.70 MB", "5.44 MiB");
  
  	/* huge values */
  	test_string_get_size_one(U64_MAX, 4096, "75.6 ZB", "64.0 ZiB");
  	test_string_get_size_one(4096, U64_MAX, "75.6 ZB", "64.0 ZiB");
943ba6503   Vitaly Kuznetsov   lib/test-string_h...
414
  }
1e0f548b2   Vadim Pasternak   lib/test-string_h...
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
  static void __init test_string_upper_lower(void)
  {
  	char *dst;
  	int i;
  
  	for (i = 0; i < ARRAY_SIZE(strings_upper); i++) {
  		const char *s = strings_upper[i].in;
  		int len = strlen(strings_upper[i].in) + 1;
  
  		dst = kmalloc(len, GFP_KERNEL);
  		if (!dst)
  			return;
  
  		string_upper(dst, s);
  		if (memcmp(dst, strings_upper[i].out, len)) {
  			pr_warn("Test 'string_upper' failed : expected %s, got %s!
  ",
  				strings_upper[i].out, dst);
  			kfree(dst);
  			return;
  		}
  		kfree(dst);
  	}
  
  	for (i = 0; i < ARRAY_SIZE(strings_lower); i++) {
  		const char *s = strings_lower[i].in;
  		int len = strlen(strings_lower[i].in) + 1;
  
  		dst = kmalloc(len, GFP_KERNEL);
  		if (!dst)
  			return;
  
  		string_lower(dst, s);
  		if (memcmp(dst, strings_lower[i].out, len)) {
  			pr_warn("Test 'string_lower failed : : expected %s, got %s!
  ",
  				strings_lower[i].out, dst);
  			kfree(dst);
  			return;
  		}
  		kfree(dst);
  	}
  }
16c7fa058   Andy Shevchenko   lib/string_helper...
458
459
460
461
462
463
464
  static int __init test_string_helpers_init(void)
  {
  	unsigned int i;
  
  	pr_info("Running tests...
  ");
  	for (i = 0; i < UNESCAPE_ANY + 1; i++)
45ff337a5   Andy Shevchenko   lib / string_help...
465
466
467
  		test_string_unescape("unescape", i, false);
  	test_string_unescape("unescape inplace",
  			     get_random_int() % (UNESCAPE_ANY + 1), true);
16c7fa058   Andy Shevchenko   lib/string_helper...
468

c8250381c   Andy Shevchenko   lib / string_help...
469
470
471
472
473
474
475
  	/* Without dictionary */
  	for (i = 0; i < (ESCAPE_ANY_NP | ESCAPE_HEX) + 1; i++)
  		test_string_escape("escape 0", escape0, i, TEST_STRING_2_DICT_0);
  
  	/* With dictionary */
  	for (i = 0; i < (ESCAPE_ANY_NP | ESCAPE_HEX) + 1; i++)
  		test_string_escape("escape 1", escape1, i, TEST_STRING_2_DICT_1);
943ba6503   Vitaly Kuznetsov   lib/test-string_h...
476
477
  	/* Test string_get_size() */
  	test_string_get_size();
1e0f548b2   Vadim Pasternak   lib/test-string_h...
478
479
  	/* Test string upper(), string_lower() */
  	test_string_upper_lower();
16c7fa058   Andy Shevchenko   lib/string_helper...
480
481
482
483
  	return -EINVAL;
  }
  module_init(test_string_helpers_init);
  MODULE_LICENSE("Dual BSD/GPL");