1 /* $NetBSD: t_strlen.c,v 1.6 2017/01/14 20:49:24 christos Exp $ */ 2 3 /* 4 * Written by J.T. Conklin <jtc@acorntoolworks.com> 5 * Public domain. 6 */ 7 8 #include <atf-c.h> 9 #include <string.h> 10 #include <unistd.h> 11 #include <stdio.h> 12 #include <stdlib.h> 13 #include <dlfcn.h> 14 #include <unistd.h> 15 16 static void write_num(int); 17 18 static void 19 write_num(int val) 20 { 21 char buf[20]; 22 int i; 23 24 for (i = sizeof buf; --i >= 0;) { 25 buf[i] = '0' + val % 10; 26 val /= 10; 27 if (val == 0) { 28 write(2, buf + i, sizeof buf - i); 29 return; 30 } 31 } 32 write(2, "overflow", 8); 33 } 34 35 ATF_TC(strlen_basic); 36 ATF_TC_HEAD(strlen_basic, tc) 37 { 38 atf_tc_set_md_var(tc, "descr", "Test strlen(3) results"); 39 } 40 41 ATF_TC_BODY(strlen_basic, tc) 42 { 43 void *dl_handle; 44 /* try to trick the compiler */ 45 size_t (*strlen_fn)(const char *); 46 47 unsigned int a, t; 48 size_t len; 49 char buf[64]; 50 51 struct tab { 52 const char* val; 53 size_t len; 54 }; 55 56 const struct tab tab[] = { 57 /* 58 * patterns that check for all combinations of leading and 59 * trailing unaligned characters (on a 64 bit processor) 60 */ 61 62 { "", 0 }, 63 { "a", 1 }, 64 { "ab", 2 }, 65 { "abc", 3 }, 66 { "abcd", 4 }, 67 { "abcde", 5 }, 68 { "abcdef", 6 }, 69 { "abcdefg", 7 }, 70 { "abcdefgh", 8 }, 71 { "abcdefghi", 9 }, 72 { "abcdefghij", 10 }, 73 { "abcdefghijk", 11 }, 74 { "abcdefghijkl", 12 }, 75 { "abcdefghijklm", 13 }, 76 { "abcdefghijklmn", 14 }, 77 { "abcdefghijklmno", 15 }, 78 { "abcdefghijklmnop", 16 }, 79 { "abcdefghijklmnopq", 17 }, 80 { "abcdefghijklmnopqr", 18 }, 81 { "abcdefghijklmnopqrs", 19 }, 82 { "abcdefghijklmnopqrst", 20 }, 83 { "abcdefghijklmnopqrstu", 21 }, 84 { "abcdefghijklmnopqrstuv", 22 }, 85 { "abcdefghijklmnopqrstuvw", 23 }, 86 87 /* 88 * patterns that check for the cases where the expression: 89 * 90 * ((word - 0x7f7f..7f) & 0x8080..80) 91 * 92 * returns non-zero even though there are no zero bytes in 93 * the word. 94 */ 95 96 { "" "\xff\xff\xff\xff\xff\xff\xff\xff" "abcdefgh", 16 }, 97 { "a" "\xff\xff\xff\xff\xff\xff\xff\xff" "bcdefgh", 16 }, 98 { "ab" "\xff\xff\xff\xff\xff\xff\xff\xff" "cdefgh", 16 }, 99 { "abc" "\xff\xff\xff\xff\xff\xff\xff\xff" "defgh", 16 }, 100 { "abcd" "\xff\xff\xff\xff\xff\xff\xff\xff" "efgh", 16 }, 101 { "abcde" "\xff\xff\xff\xff\xff\xff\xff\xff" "fgh", 16 }, 102 { "abcdef" "\xff\xff\xff\xff\xff\xff\xff\xff" "gh", 16 }, 103 { "abcdefg" "\xff\xff\xff\xff\xff\xff\xff\xff" "h", 16 }, 104 { "abcdefgh" "\xff\xff\xff\xff\xff\xff\xff\xff" "", 16 }, 105 }; 106 107 /* 108 * During testing it is useful have the rest of the program 109 * use a known good version! 110 */ 111 dl_handle = dlopen(NULL, RTLD_LAZY); 112 strlen_fn = dlsym(dl_handle, "test_strlen"); 113 if (!strlen_fn) 114 strlen_fn = strlen; 115 116 for (a = 0; a < sizeof(long); ++a) { 117 for (t = 0; t < (sizeof(tab) / sizeof(tab[0])); ++t) { 118 119 memcpy(&buf[a], tab[t].val, tab[t].len + 1); 120 len = strlen_fn(&buf[a]); 121 122 if (len != tab[t].len) { 123 /* Write error without using printf / strlen */ 124 write(2, "alignment ", 10); 125 write_num(a); 126 write(2, ", test ", 7); 127 write_num(t); 128 write(2, ", got len ", 10); 129 write_num(len); 130 write(2, ", not ", 6); 131 write_num(tab[t].len); 132 write(2, ", for '", 7); 133 write(2, tab[t].val, tab[t].len); 134 write(2, "'\n", 2); 135 atf_tc_fail("See stderr for details"); 136 } 137 } 138 } 139 (void)dlclose(dl_handle); 140 } 141 142 ATF_TC(strlen_huge); 143 ATF_TC_HEAD(strlen_huge, tc) 144 { 145 atf_tc_set_md_var(tc, "descr", "Test strlen(3) with huge strings"); 146 } 147 148 ATF_TC_BODY(strlen_huge, tc) 149 { 150 long page; 151 char *str; 152 size_t i; 153 154 page = sysconf(_SC_PAGESIZE); 155 ATF_REQUIRE(page >= 0); 156 157 for (i = 1; i < 1000; i = i + 100) { 158 159 str = malloc(i * page + 1); 160 161 if (str == NULL) 162 continue; 163 164 (void)memset(str, 'x', i * page); 165 str[i * page] = '\0'; 166 167 ATF_REQUIRE(strlen(str) == i * page); 168 free(str); 169 } 170 } 171 172 ATF_TC(strnlen_basic); 173 ATF_TC_HEAD(strnlen_basic, tc) 174 { 175 atf_tc_set_md_var(tc, "descr", "A naive test of strnlen(3)"); 176 } 177 178 ATF_TC_BODY(strnlen_basic, tc) 179 { 180 char buf[1]; 181 182 buf[0] = '\0'; 183 184 ATF_CHECK(strnlen(buf, 000) == 0); 185 ATF_CHECK(strnlen(buf, 111) == 0); 186 187 ATF_CHECK(strnlen("xxx", 0) == 0); 188 ATF_CHECK(strnlen("xxx", 1) == 1); 189 ATF_CHECK(strnlen("xxx", 2) == 2); 190 ATF_CHECK(strnlen("xxx", 3) == 3); 191 ATF_CHECK(strnlen("xxx", 9) == 3); 192 } 193 194 ATF_TP_ADD_TCS(tp) 195 { 196 197 ATF_TP_ADD_TC(tp, strlen_basic); 198 ATF_TP_ADD_TC(tp, strlen_huge); 199 ATF_TP_ADD_TC(tp, strnlen_basic); 200 201 return atf_no_error(); 202 } 203