xref: /linux/lib/test_hexdump.c (revision e5c86679d5e864947a52fb31e45a425dea3e7fa9)
1 /*
2  * Test cases for lib/hexdump.c module.
3  */
4 #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
5 
6 #include <linux/init.h>
7 #include <linux/kernel.h>
8 #include <linux/module.h>
9 #include <linux/random.h>
10 #include <linux/string.h>
11 
12 static const unsigned char data_b[] = {
13 	'\xbe', '\x32', '\xdb', '\x7b', '\x0a', '\x18', '\x93', '\xb2',	/* 00 - 07 */
14 	'\x70', '\xba', '\xc4', '\x24', '\x7d', '\x83', '\x34', '\x9b',	/* 08 - 0f */
15 	'\xa6', '\x9c', '\x31', '\xad', '\x9c', '\x0f', '\xac', '\xe9',	/* 10 - 17 */
16 	'\x4c', '\xd1', '\x19', '\x99', '\x43', '\xb1', '\xaf', '\x0c',	/* 18 - 1f */
17 };
18 
19 static const unsigned char data_a[] = ".2.{....p..$}.4...1.....L...C...";
20 
21 static const char * const test_data_1_le[] __initconst = {
22 	"be", "32", "db", "7b", "0a", "18", "93", "b2",
23 	"70", "ba", "c4", "24", "7d", "83", "34", "9b",
24 	"a6", "9c", "31", "ad", "9c", "0f", "ac", "e9",
25 	"4c", "d1", "19", "99", "43", "b1", "af", "0c",
26 };
27 
28 static const char * const test_data_2_le[] __initconst = {
29 	"32be", "7bdb", "180a", "b293",
30 	"ba70", "24c4", "837d", "9b34",
31 	"9ca6", "ad31", "0f9c", "e9ac",
32 	"d14c", "9919", "b143", "0caf",
33 };
34 
35 static const char * const test_data_4_le[] __initconst = {
36 	"7bdb32be", "b293180a", "24c4ba70", "9b34837d",
37 	"ad319ca6", "e9ac0f9c", "9919d14c", "0cafb143",
38 };
39 
40 static const char * const test_data_8_le[] __initconst = {
41 	"b293180a7bdb32be", "9b34837d24c4ba70",
42 	"e9ac0f9cad319ca6", "0cafb1439919d14c",
43 };
44 
45 #define FILL_CHAR	'#'
46 
47 static unsigned total_tests __initdata;
48 static unsigned failed_tests __initdata;
49 
50 static void __init test_hexdump_prepare_test(size_t len, int rowsize,
51 					     int groupsize, char *test,
52 					     size_t testlen, bool ascii)
53 {
54 	char *p;
55 	const char * const *result;
56 	size_t l = len;
57 	int gs = groupsize, rs = rowsize;
58 	unsigned int i;
59 
60 	if (rs != 16 && rs != 32)
61 		rs = 16;
62 
63 	if (l > rs)
64 		l = rs;
65 
66 	if (!is_power_of_2(gs) || gs > 8 || (len % gs != 0))
67 		gs = 1;
68 
69 	if (gs == 8)
70 		result = test_data_8_le;
71 	else if (gs == 4)
72 		result = test_data_4_le;
73 	else if (gs == 2)
74 		result = test_data_2_le;
75 	else
76 		result = test_data_1_le;
77 
78 	/* hex dump */
79 	p = test;
80 	for (i = 0; i < l / gs; i++) {
81 		const char *q = *result++;
82 		size_t amount = strlen(q);
83 
84 		strncpy(p, q, amount);
85 		p += amount;
86 
87 		*p++ = ' ';
88 	}
89 	if (i)
90 		p--;
91 
92 	/* ASCII part */
93 	if (ascii) {
94 		do {
95 			*p++ = ' ';
96 		} while (p < test + rs * 2 + rs / gs + 1);
97 
98 		strncpy(p, data_a, l);
99 		p += l;
100 	}
101 
102 	*p = '\0';
103 }
104 
105 #define TEST_HEXDUMP_BUF_SIZE		(32 * 3 + 2 + 32 + 1)
106 
107 static void __init test_hexdump(size_t len, int rowsize, int groupsize,
108 				bool ascii)
109 {
110 	char test[TEST_HEXDUMP_BUF_SIZE];
111 	char real[TEST_HEXDUMP_BUF_SIZE];
112 
113 	total_tests++;
114 
115 	memset(real, FILL_CHAR, sizeof(real));
116 	hex_dump_to_buffer(data_b, len, rowsize, groupsize, real, sizeof(real),
117 			   ascii);
118 
119 	memset(test, FILL_CHAR, sizeof(test));
120 	test_hexdump_prepare_test(len, rowsize, groupsize, test, sizeof(test),
121 				  ascii);
122 
123 	if (memcmp(test, real, TEST_HEXDUMP_BUF_SIZE)) {
124 		pr_err("Len: %zu row: %d group: %d\n", len, rowsize, groupsize);
125 		pr_err("Result: '%s'\n", real);
126 		pr_err("Expect: '%s'\n", test);
127 		failed_tests++;
128 	}
129 }
130 
131 static void __init test_hexdump_set(int rowsize, bool ascii)
132 {
133 	size_t d = min_t(size_t, sizeof(data_b), rowsize);
134 	size_t len = get_random_int() % d + 1;
135 
136 	test_hexdump(len, rowsize, 4, ascii);
137 	test_hexdump(len, rowsize, 2, ascii);
138 	test_hexdump(len, rowsize, 8, ascii);
139 	test_hexdump(len, rowsize, 1, ascii);
140 }
141 
142 static void __init test_hexdump_overflow(size_t buflen, size_t len,
143 					 int rowsize, int groupsize,
144 					 bool ascii)
145 {
146 	char test[TEST_HEXDUMP_BUF_SIZE];
147 	char buf[TEST_HEXDUMP_BUF_SIZE];
148 	int rs = rowsize, gs = groupsize;
149 	int ae, he, e, f, r;
150 	bool a;
151 
152 	total_tests++;
153 
154 	memset(buf, FILL_CHAR, sizeof(buf));
155 
156 	r = hex_dump_to_buffer(data_b, len, rs, gs, buf, buflen, ascii);
157 
158 	/*
159 	 * Caller must provide the data length multiple of groupsize. The
160 	 * calculations below are made with that assumption in mind.
161 	 */
162 	ae = rs * 2 /* hex */ + rs / gs /* spaces */ + 1 /* space */ + len /* ascii */;
163 	he = (gs * 2 /* hex */ + 1 /* space */) * len / gs - 1 /* no trailing space */;
164 
165 	if (ascii)
166 		e = ae;
167 	else
168 		e = he;
169 
170 	f = min_t(int, e + 1, buflen);
171 	if (buflen) {
172 		test_hexdump_prepare_test(len, rs, gs, test, sizeof(test), ascii);
173 		test[f - 1] = '\0';
174 	}
175 	memset(test + f, FILL_CHAR, sizeof(test) - f);
176 
177 	a = r == e && !memcmp(test, buf, TEST_HEXDUMP_BUF_SIZE);
178 
179 	buf[sizeof(buf) - 1] = '\0';
180 
181 	if (!a) {
182 		pr_err("Len: %zu buflen: %zu strlen: %zu\n",
183 			len, buflen, strnlen(buf, sizeof(buf)));
184 		pr_err("Result: %d '%s'\n", r, buf);
185 		pr_err("Expect: %d '%s'\n", e, test);
186 		failed_tests++;
187 	}
188 }
189 
190 static void __init test_hexdump_overflow_set(size_t buflen, bool ascii)
191 {
192 	unsigned int i = 0;
193 	int rs = (get_random_int() % 2 + 1) * 16;
194 
195 	do {
196 		int gs = 1 << i;
197 		size_t len = get_random_int() % rs + gs;
198 
199 		test_hexdump_overflow(buflen, rounddown(len, gs), rs, gs, ascii);
200 	} while (i++ < 3);
201 }
202 
203 static int __init test_hexdump_init(void)
204 {
205 	unsigned int i;
206 	int rowsize;
207 
208 	rowsize = (get_random_int() % 2 + 1) * 16;
209 	for (i = 0; i < 16; i++)
210 		test_hexdump_set(rowsize, false);
211 
212 	rowsize = (get_random_int() % 2 + 1) * 16;
213 	for (i = 0; i < 16; i++)
214 		test_hexdump_set(rowsize, true);
215 
216 	for (i = 0; i <= TEST_HEXDUMP_BUF_SIZE; i++)
217 		test_hexdump_overflow_set(i, false);
218 
219 	for (i = 0; i <= TEST_HEXDUMP_BUF_SIZE; i++)
220 		test_hexdump_overflow_set(i, true);
221 
222 	if (failed_tests == 0)
223 		pr_info("all %u tests passed\n", total_tests);
224 	else
225 		pr_err("failed %u out of %u tests\n", failed_tests, total_tests);
226 
227 	return failed_tests ? -EINVAL : 0;
228 }
229 module_init(test_hexdump_init);
230 
231 static void __exit test_hexdump_exit(void)
232 {
233 	/* do nothing */
234 }
235 module_exit(test_hexdump_exit);
236 
237 MODULE_AUTHOR("Andy Shevchenko <andriy.shevchenko@linux.intel.com>");
238 MODULE_LICENSE("Dual BSD/GPL");
239