Andy Shevchenko | 64d1d77 | 2015-02-12 15:02:21 -0800 | [diff] [blame] | 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 | |
Andi Kleen | c79574a | 2015-04-15 16:16:39 -0700 | [diff] [blame] | 21 | static const char * const test_data_1_le[] __initconst = { |
Andy Shevchenko | 64d1d77 | 2015-02-12 15:02:21 -0800 | [diff] [blame] | 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 | |
Geert Uytterhoeven | 79e23d5 | 2015-06-25 15:02:11 -0700 | [diff] [blame] | 28 | static const char * const test_data_2_le[] __initconst = { |
Andy Shevchenko | 64d1d77 | 2015-02-12 15:02:21 -0800 | [diff] [blame] | 29 | "32be", "7bdb", "180a", "b293", |
| 30 | "ba70", "24c4", "837d", "9b34", |
| 31 | "9ca6", "ad31", "0f9c", "e9ac", |
| 32 | "d14c", "9919", "b143", "0caf", |
| 33 | }; |
| 34 | |
Geert Uytterhoeven | 79e23d5 | 2015-06-25 15:02:11 -0700 | [diff] [blame] | 35 | static const char * const test_data_4_le[] __initconst = { |
Andy Shevchenko | 64d1d77 | 2015-02-12 15:02:21 -0800 | [diff] [blame] | 36 | "7bdb32be", "b293180a", "24c4ba70", "9b34837d", |
| 37 | "ad319ca6", "e9ac0f9c", "9919d14c", "0cafb143", |
| 38 | }; |
| 39 | |
Geert Uytterhoeven | 79e23d5 | 2015-06-25 15:02:11 -0700 | [diff] [blame] | 40 | static const char * const test_data_8_le[] __initconst = { |
Andy Shevchenko | 64d1d77 | 2015-02-12 15:02:21 -0800 | [diff] [blame] | 41 | "b293180a7bdb32be", "9b34837d24c4ba70", |
| 42 | "e9ac0f9cad319ca6", "0cafb1439919d14c", |
| 43 | }; |
| 44 | |
Andy Shevchenko | 3db4a98 | 2016-01-20 14:58:50 -0800 | [diff] [blame] | 45 | #define FILL_CHAR '#' |
| 46 | |
Andy Shevchenko | 87977ca6b | 2016-01-20 14:58:47 -0800 | [diff] [blame] | 47 | static void __init test_hexdump_prepare_test(size_t len, int rowsize, |
| 48 | int groupsize, char *test, |
| 49 | size_t testlen, bool ascii) |
Andy Shevchenko | 64d1d77 | 2015-02-12 15:02:21 -0800 | [diff] [blame] | 50 | { |
Andy Shevchenko | 64d1d77 | 2015-02-12 15:02:21 -0800 | [diff] [blame] | 51 | char *p; |
Linus Torvalds | 17974c05 | 2015-04-19 13:48:40 -0700 | [diff] [blame] | 52 | const char * const *result; |
Andy Shevchenko | 64d1d77 | 2015-02-12 15:02:21 -0800 | [diff] [blame] | 53 | size_t l = len; |
| 54 | int gs = groupsize, rs = rowsize; |
| 55 | unsigned int i; |
| 56 | |
Andy Shevchenko | 64d1d77 | 2015-02-12 15:02:21 -0800 | [diff] [blame] | 57 | if (rs != 16 && rs != 32) |
| 58 | rs = 16; |
| 59 | |
| 60 | if (l > rs) |
| 61 | l = rs; |
| 62 | |
| 63 | if (!is_power_of_2(gs) || gs > 8 || (len % gs != 0)) |
| 64 | gs = 1; |
| 65 | |
| 66 | if (gs == 8) |
| 67 | result = test_data_8_le; |
| 68 | else if (gs == 4) |
| 69 | result = test_data_4_le; |
| 70 | else if (gs == 2) |
| 71 | result = test_data_2_le; |
| 72 | else |
| 73 | result = test_data_1_le; |
| 74 | |
Andy Shevchenko | 3db4a98 | 2016-01-20 14:58:50 -0800 | [diff] [blame] | 75 | memset(test, FILL_CHAR, testlen); |
Andy Shevchenko | 64d1d77 | 2015-02-12 15:02:21 -0800 | [diff] [blame] | 76 | |
| 77 | /* hex dump */ |
| 78 | p = test; |
| 79 | for (i = 0; i < l / gs; i++) { |
| 80 | const char *q = *result++; |
| 81 | size_t amount = strlen(q); |
| 82 | |
| 83 | strncpy(p, q, amount); |
Andy Shevchenko | 3db4a98 | 2016-01-20 14:58:50 -0800 | [diff] [blame] | 84 | p += amount; |
| 85 | |
| 86 | *p++ = ' '; |
Andy Shevchenko | 64d1d77 | 2015-02-12 15:02:21 -0800 | [diff] [blame] | 87 | } |
| 88 | if (i) |
| 89 | p--; |
| 90 | |
| 91 | /* ASCII part */ |
| 92 | if (ascii) { |
Andy Shevchenko | 3db4a98 | 2016-01-20 14:58:50 -0800 | [diff] [blame] | 93 | do { |
| 94 | *p++ = ' '; |
| 95 | } while (p < test + rs * 2 + rs / gs + 1); |
| 96 | |
Andy Shevchenko | 64d1d77 | 2015-02-12 15:02:21 -0800 | [diff] [blame] | 97 | strncpy(p, data_a, l); |
| 98 | p += l; |
| 99 | } |
| 100 | |
| 101 | *p = '\0'; |
Andy Shevchenko | 87977ca6b | 2016-01-20 14:58:47 -0800 | [diff] [blame] | 102 | } |
| 103 | |
| 104 | #define TEST_HEXDUMP_BUF_SIZE (32 * 3 + 2 + 32 + 1) |
| 105 | |
| 106 | static void __init test_hexdump(size_t len, int rowsize, int groupsize, |
| 107 | bool ascii) |
| 108 | { |
| 109 | char test[TEST_HEXDUMP_BUF_SIZE]; |
| 110 | char real[TEST_HEXDUMP_BUF_SIZE]; |
| 111 | |
| 112 | hex_dump_to_buffer(data_b, len, rowsize, groupsize, real, sizeof(real), |
| 113 | ascii); |
| 114 | |
| 115 | test_hexdump_prepare_test(len, rowsize, groupsize, test, sizeof(test), |
| 116 | ascii); |
Andy Shevchenko | 64d1d77 | 2015-02-12 15:02:21 -0800 | [diff] [blame] | 117 | |
| 118 | if (strcmp(test, real)) { |
| 119 | pr_err("Len: %zu row: %d group: %d\n", len, rowsize, groupsize); |
| 120 | pr_err("Result: '%s'\n", real); |
| 121 | pr_err("Expect: '%s'\n", test); |
| 122 | } |
| 123 | } |
| 124 | |
| 125 | static void __init test_hexdump_set(int rowsize, bool ascii) |
| 126 | { |
| 127 | size_t d = min_t(size_t, sizeof(data_b), rowsize); |
| 128 | size_t len = get_random_int() % d + 1; |
| 129 | |
| 130 | test_hexdump(len, rowsize, 4, ascii); |
| 131 | test_hexdump(len, rowsize, 2, ascii); |
| 132 | test_hexdump(len, rowsize, 8, ascii); |
| 133 | test_hexdump(len, rowsize, 1, ascii); |
| 134 | } |
| 135 | |
Andy Shevchenko | a3d601f | 2016-01-20 14:58:53 -0800 | [diff] [blame] | 136 | static void __init test_hexdump_overflow(size_t buflen, bool ascii) |
Andy Shevchenko | 114fc1a | 2015-02-12 15:02:29 -0800 | [diff] [blame] | 137 | { |
Andy Shevchenko | a3d601f | 2016-01-20 14:58:53 -0800 | [diff] [blame] | 138 | char buf[TEST_HEXDUMP_BUF_SIZE]; |
Andy Shevchenko | 114fc1a | 2015-02-12 15:02:29 -0800 | [diff] [blame] | 139 | const char *t = test_data_1_le[0]; |
Andy Shevchenko | ad27a75 | 2016-01-20 14:58:56 -0800 | [diff] [blame^] | 140 | size_t len = 1; |
| 141 | int rs = 16, gs = 1; |
| 142 | int ae, he, e, r; |
Andy Shevchenko | 114fc1a | 2015-02-12 15:02:29 -0800 | [diff] [blame] | 143 | bool a; |
Andy Shevchenko | 114fc1a | 2015-02-12 15:02:29 -0800 | [diff] [blame] | 144 | |
Andy Shevchenko | 3db4a98 | 2016-01-20 14:58:50 -0800 | [diff] [blame] | 145 | memset(buf, FILL_CHAR, sizeof(buf)); |
Andy Shevchenko | 114fc1a | 2015-02-12 15:02:29 -0800 | [diff] [blame] | 146 | |
Andy Shevchenko | ad27a75 | 2016-01-20 14:58:56 -0800 | [diff] [blame^] | 147 | r = hex_dump_to_buffer(data_b, len, rs, gs, buf, buflen, ascii); |
| 148 | |
| 149 | /* |
| 150 | * Caller must provide the data length multiple of groupsize. The |
| 151 | * calculations below are made with that assumption in mind. |
| 152 | */ |
| 153 | ae = rs * 2 /* hex */ + rs / gs /* spaces */ + 1 /* space */ + len /* ascii */; |
| 154 | he = (gs * 2 /* hex */ + 1 /* space */) * len / gs - 1 /* no trailing space */; |
Andy Shevchenko | 114fc1a | 2015-02-12 15:02:29 -0800 | [diff] [blame] | 155 | |
| 156 | if (ascii) |
Andy Shevchenko | ad27a75 | 2016-01-20 14:58:56 -0800 | [diff] [blame^] | 157 | e = ae; |
Andy Shevchenko | 114fc1a | 2015-02-12 15:02:29 -0800 | [diff] [blame] | 158 | else |
Andy Shevchenko | ad27a75 | 2016-01-20 14:58:56 -0800 | [diff] [blame^] | 159 | e = he; |
Andy Shevchenko | 114fc1a | 2015-02-12 15:02:29 -0800 | [diff] [blame] | 160 | buf[e + 2] = '\0'; |
| 161 | |
Andy Shevchenko | a3d601f | 2016-01-20 14:58:53 -0800 | [diff] [blame] | 162 | if (!buflen) { |
Andy Shevchenko | 3db4a98 | 2016-01-20 14:58:50 -0800 | [diff] [blame] | 163 | a = r == e && buf[0] == FILL_CHAR; |
Andy Shevchenko | a3d601f | 2016-01-20 14:58:53 -0800 | [diff] [blame] | 164 | } else if (buflen < 3) { |
Andy Shevchenko | 114fc1a | 2015-02-12 15:02:29 -0800 | [diff] [blame] | 165 | a = r == e && buf[0] == '\0'; |
Andy Shevchenko | a3d601f | 2016-01-20 14:58:53 -0800 | [diff] [blame] | 166 | } else if (buflen < 4) { |
Andy Shevchenko | 114fc1a | 2015-02-12 15:02:29 -0800 | [diff] [blame] | 167 | a = r == e && !strcmp(buf, t); |
| 168 | } else if (ascii) { |
Andy Shevchenko | a3d601f | 2016-01-20 14:58:53 -0800 | [diff] [blame] | 169 | if (buflen < 51) |
| 170 | a = r == e && buf[buflen - 1] == '\0' && buf[buflen - 2] == FILL_CHAR; |
Andy Shevchenko | 114fc1a | 2015-02-12 15:02:29 -0800 | [diff] [blame] | 171 | else |
| 172 | a = r == e && buf[50] == '\0' && buf[49] == '.'; |
| 173 | } else { |
| 174 | a = r == e && buf[e] == '\0'; |
| 175 | } |
| 176 | |
| 177 | if (!a) { |
Andy Shevchenko | a3d601f | 2016-01-20 14:58:53 -0800 | [diff] [blame] | 178 | pr_err("Len: %zu rc: %u strlen: %zu\n", buflen, r, strlen(buf)); |
Andy Shevchenko | 114fc1a | 2015-02-12 15:02:29 -0800 | [diff] [blame] | 179 | pr_err("Result: '%s'\n", buf); |
| 180 | } |
| 181 | } |
| 182 | |
Andy Shevchenko | 64d1d77 | 2015-02-12 15:02:21 -0800 | [diff] [blame] | 183 | static int __init test_hexdump_init(void) |
| 184 | { |
| 185 | unsigned int i; |
| 186 | int rowsize; |
| 187 | |
| 188 | pr_info("Running tests...\n"); |
| 189 | |
| 190 | rowsize = (get_random_int() % 2 + 1) * 16; |
| 191 | for (i = 0; i < 16; i++) |
| 192 | test_hexdump_set(rowsize, false); |
| 193 | |
| 194 | rowsize = (get_random_int() % 2 + 1) * 16; |
| 195 | for (i = 0; i < 16; i++) |
| 196 | test_hexdump_set(rowsize, true); |
| 197 | |
Andy Shevchenko | a3d601f | 2016-01-20 14:58:53 -0800 | [diff] [blame] | 198 | for (i = 0; i <= TEST_HEXDUMP_BUF_SIZE; i++) |
| 199 | test_hexdump_overflow(i, false); |
Andy Shevchenko | 114fc1a | 2015-02-12 15:02:29 -0800 | [diff] [blame] | 200 | |
Andy Shevchenko | a3d601f | 2016-01-20 14:58:53 -0800 | [diff] [blame] | 201 | for (i = 0; i <= TEST_HEXDUMP_BUF_SIZE; i++) |
| 202 | test_hexdump_overflow(i, true); |
Andy Shevchenko | 114fc1a | 2015-02-12 15:02:29 -0800 | [diff] [blame] | 203 | |
Andy Shevchenko | 64d1d77 | 2015-02-12 15:02:21 -0800 | [diff] [blame] | 204 | return -EINVAL; |
| 205 | } |
| 206 | module_init(test_hexdump_init); |
| 207 | MODULE_LICENSE("Dual BSD/GPL"); |