Blame view
lib/test_strscpy.c
4.02 KB
0b0600c8c
|
1 2 3 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 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 |
// SPDX-License-Identifier: GPL-2.0+ #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt #include <linux/string.h> #include "../tools/testing/selftests/kselftest_module.h" /* * Kernel module for testing 'strscpy' family of functions. */ KSTM_MODULE_GLOBALS(); /* * tc() - Run a specific test case. * @src: Source string, argument to strscpy_pad() * @count: Size of destination buffer, argument to strscpy_pad() * @expected: Expected return value from call to strscpy_pad() * @terminator: 1 if there should be a terminating null byte 0 otherwise. * @chars: Number of characters from the src string expected to be * written to the dst buffer. * @pad: Number of pad characters expected (in the tail of dst buffer). * (@pad does not include the null terminator byte.) * * Calls strscpy_pad() and verifies the return value and state of the * destination buffer after the call returns. */ static int __init tc(char *src, int count, int expected, int chars, int terminator, int pad) { int nr_bytes_poison; int max_expected; int max_count; int written; char buf[6]; int index, i; const char POISON = 'z'; total_tests++; if (!src) { pr_err("null source string not supported "); return -1; } memset(buf, POISON, sizeof(buf)); /* Future proofing test suite, validate args */ max_count = sizeof(buf) - 2; /* Space for null and to verify overflow */ max_expected = count - 1; /* Space for the null */ if (count > max_count) { pr_err("count (%d) is too big (%d) ... aborting", count, max_count); return -1; } if (expected > max_expected) { pr_warn("expected (%d) is bigger than can possibly be returned (%d)", expected, max_expected); } written = strscpy_pad(buf, src, count); if ((written) != (expected)) { pr_err("%d != %d (written, expected) ", written, expected); goto fail; } if (count && written == -E2BIG) { if (strncmp(buf, src, count - 1) != 0) { pr_err("buffer state invalid for -E2BIG "); goto fail; } if (buf[count - 1] != '\0') { pr_err("too big string is not null terminated correctly "); goto fail; } } for (i = 0; i < chars; i++) { if (buf[i] != src[i]) { pr_err("buf[i]==%c != src[i]==%c ", buf[i], src[i]); goto fail; } } if (terminator) { if (buf[count - 1] != '\0') { pr_err("string is not null terminated correctly "); goto fail; } } for (i = 0; i < pad; i++) { index = chars + terminator + i; if (buf[index] != '\0') { pr_err("padding missing at index: %d ", i); goto fail; } } nr_bytes_poison = sizeof(buf) - chars - terminator - pad; for (i = 0; i < nr_bytes_poison; i++) { index = sizeof(buf) - 1 - i; /* Check from the end back */ if (buf[index] != POISON) { pr_err("poison value missing at index: %d ", i); goto fail; } } return 0; fail: failed_tests++; return -1; } static void __init selftest(void) { /* * tc() uses a destination buffer of size 6 and needs at * least 2 characters spare (one for null and one to check for * overflow). This means we should only call tc() with * strings up to a maximum of 4 characters long and 'count' * should not exceed 4. To test with longer strings increase * the buffer size in tc(). */ /* tc(src, count, expected, chars, terminator, pad) */ KSTM_CHECK_ZERO(tc("a", 0, -E2BIG, 0, 0, 0)); KSTM_CHECK_ZERO(tc("", 0, -E2BIG, 0, 0, 0)); KSTM_CHECK_ZERO(tc("a", 1, -E2BIG, 0, 1, 0)); KSTM_CHECK_ZERO(tc("", 1, 0, 0, 1, 0)); KSTM_CHECK_ZERO(tc("ab", 2, -E2BIG, 1, 1, 0)); KSTM_CHECK_ZERO(tc("a", 2, 1, 1, 1, 0)); KSTM_CHECK_ZERO(tc("", 2, 0, 0, 1, 1)); KSTM_CHECK_ZERO(tc("abc", 3, -E2BIG, 2, 1, 0)); KSTM_CHECK_ZERO(tc("ab", 3, 2, 2, 1, 0)); KSTM_CHECK_ZERO(tc("a", 3, 1, 1, 1, 1)); KSTM_CHECK_ZERO(tc("", 3, 0, 0, 1, 2)); KSTM_CHECK_ZERO(tc("abcd", 4, -E2BIG, 3, 1, 0)); KSTM_CHECK_ZERO(tc("abc", 4, 3, 3, 1, 0)); KSTM_CHECK_ZERO(tc("ab", 4, 2, 2, 1, 1)); KSTM_CHECK_ZERO(tc("a", 4, 1, 1, 1, 2)); KSTM_CHECK_ZERO(tc("", 4, 0, 0, 1, 3)); } KSTM_MODULE_LOADERS(test_strscpy); MODULE_AUTHOR("Tobin C. Harding <tobin@kernel.org>"); MODULE_LICENSE("GPL"); |