1 #include "test/jemalloc_test.h"
2
3 #define TEST_POW2_CEIL(t, suf, pri) do { \
4 unsigned i, pow2; \
5 t x; \
6 \
7 assert_##suf##_eq(pow2_ceil_##suf(0), 0, "Unexpected result"); \
8 \
9 for (i = 0; i < sizeof(t) * 8; i++) { \
10 assert_##suf##_eq(pow2_ceil_##suf(((t)1) << i), ((t)1) \
11 << i, "Unexpected result"); \
12 } \
13 \
14 for (i = 2; i < sizeof(t) * 8; i++) { \
15 assert_##suf##_eq(pow2_ceil_##suf((((t)1) << i) - 1), \
16 ((t)1) << i, "Unexpected result"); \
17 } \
18 \
19 for (i = 0; i < sizeof(t) * 8 - 1; i++) { \
20 assert_##suf##_eq(pow2_ceil_##suf((((t)1) << i) + 1), \
21 ((t)1) << (i+1), "Unexpected result"); \
22 } \
23 \
24 for (pow2 = 1; pow2 < 25; pow2++) { \
25 for (x = (((t)1) << (pow2-1)) + 1; x <= ((t)1) << pow2; \
26 x++) { \
27 assert_##suf##_eq(pow2_ceil_##suf(x), \
28 ((t)1) << pow2, \
29 "Unexpected result, x=%"pri, x); \
30 } \
31 } \
32 } while (0)
33
TEST_BEGIN(test_pow2_ceil_u64)34 TEST_BEGIN(test_pow2_ceil_u64)
35 {
36
37 TEST_POW2_CEIL(uint64_t, u64, FMTu64);
38 }
39 TEST_END
40
TEST_BEGIN(test_pow2_ceil_u32)41 TEST_BEGIN(test_pow2_ceil_u32)
42 {
43
44 TEST_POW2_CEIL(uint32_t, u32, FMTu32);
45 }
46 TEST_END
47
TEST_BEGIN(test_pow2_ceil_zu)48 TEST_BEGIN(test_pow2_ceil_zu)
49 {
50
51 TEST_POW2_CEIL(size_t, zu, "zu");
52 }
53 TEST_END
54
TEST_BEGIN(test_malloc_strtoumax_no_endptr)55 TEST_BEGIN(test_malloc_strtoumax_no_endptr)
56 {
57 int err;
58
59 set_errno(0);
60 assert_ju_eq(malloc_strtoumax("0", NULL, 0), 0, "Unexpected result");
61 err = get_errno();
62 assert_d_eq(err, 0, "Unexpected failure");
63 }
64 TEST_END
65
TEST_BEGIN(test_malloc_strtoumax)66 TEST_BEGIN(test_malloc_strtoumax)
67 {
68 struct test_s {
69 const char *input;
70 const char *expected_remainder;
71 int base;
72 int expected_errno;
73 const char *expected_errno_name;
74 uintmax_t expected_x;
75 };
76 #define ERR(e) e, #e
77 #define KUMAX(x) ((uintmax_t)x##ULL)
78 #define KSMAX(x) ((uintmax_t)(intmax_t)x##LL)
79 struct test_s tests[] = {
80 {"0", "0", -1, ERR(EINVAL), UINTMAX_MAX},
81 {"0", "0", 1, ERR(EINVAL), UINTMAX_MAX},
82 {"0", "0", 37, ERR(EINVAL), UINTMAX_MAX},
83
84 {"", "", 0, ERR(EINVAL), UINTMAX_MAX},
85 {"+", "+", 0, ERR(EINVAL), UINTMAX_MAX},
86 {"++3", "++3", 0, ERR(EINVAL), UINTMAX_MAX},
87 {"-", "-", 0, ERR(EINVAL), UINTMAX_MAX},
88
89 {"42", "", 0, ERR(0), KUMAX(42)},
90 {"+42", "", 0, ERR(0), KUMAX(42)},
91 {"-42", "", 0, ERR(0), KSMAX(-42)},
92 {"042", "", 0, ERR(0), KUMAX(042)},
93 {"+042", "", 0, ERR(0), KUMAX(042)},
94 {"-042", "", 0, ERR(0), KSMAX(-042)},
95 {"0x42", "", 0, ERR(0), KUMAX(0x42)},
96 {"+0x42", "", 0, ERR(0), KUMAX(0x42)},
97 {"-0x42", "", 0, ERR(0), KSMAX(-0x42)},
98
99 {"0", "", 0, ERR(0), KUMAX(0)},
100 {"1", "", 0, ERR(0), KUMAX(1)},
101
102 {"42", "", 0, ERR(0), KUMAX(42)},
103 {" 42", "", 0, ERR(0), KUMAX(42)},
104 {"42 ", " ", 0, ERR(0), KUMAX(42)},
105 {"0x", "x", 0, ERR(0), KUMAX(0)},
106 {"42x", "x", 0, ERR(0), KUMAX(42)},
107
108 {"07", "", 0, ERR(0), KUMAX(7)},
109 {"010", "", 0, ERR(0), KUMAX(8)},
110 {"08", "8", 0, ERR(0), KUMAX(0)},
111 {"0_", "_", 0, ERR(0), KUMAX(0)},
112
113 {"0x", "x", 0, ERR(0), KUMAX(0)},
114 {"0X", "X", 0, ERR(0), KUMAX(0)},
115 {"0xg", "xg", 0, ERR(0), KUMAX(0)},
116 {"0XA", "", 0, ERR(0), KUMAX(10)},
117
118 {"010", "", 10, ERR(0), KUMAX(10)},
119 {"0x3", "x3", 10, ERR(0), KUMAX(0)},
120
121 {"12", "2", 2, ERR(0), KUMAX(1)},
122 {"78", "8", 8, ERR(0), KUMAX(7)},
123 {"9a", "a", 10, ERR(0), KUMAX(9)},
124 {"9A", "A", 10, ERR(0), KUMAX(9)},
125 {"fg", "g", 16, ERR(0), KUMAX(15)},
126 {"FG", "G", 16, ERR(0), KUMAX(15)},
127 {"0xfg", "g", 16, ERR(0), KUMAX(15)},
128 {"0XFG", "G", 16, ERR(0), KUMAX(15)},
129 {"z_", "_", 36, ERR(0), KUMAX(35)},
130 {"Z_", "_", 36, ERR(0), KUMAX(35)}
131 };
132 #undef ERR
133 #undef KUMAX
134 #undef KSMAX
135 unsigned i;
136
137 for (i = 0; i < sizeof(tests)/sizeof(struct test_s); i++) {
138 struct test_s *test = &tests[i];
139 int err;
140 uintmax_t result;
141 char *remainder;
142
143 set_errno(0);
144 result = malloc_strtoumax(test->input, &remainder, test->base);
145 err = get_errno();
146 assert_d_eq(err, test->expected_errno,
147 "Expected errno %s for \"%s\", base %d",
148 test->expected_errno_name, test->input, test->base);
149 assert_str_eq(remainder, test->expected_remainder,
150 "Unexpected remainder for \"%s\", base %d",
151 test->input, test->base);
152 if (err == 0) {
153 assert_ju_eq(result, test->expected_x,
154 "Unexpected result for \"%s\", base %d",
155 test->input, test->base);
156 }
157 }
158 }
159 TEST_END
160
TEST_BEGIN(test_malloc_snprintf_truncated)161 TEST_BEGIN(test_malloc_snprintf_truncated)
162 {
163 #define BUFLEN 15
164 char buf[BUFLEN];
165 size_t result;
166 size_t len;
167 #define TEST(expected_str_untruncated, ...) do { \
168 result = malloc_snprintf(buf, len, __VA_ARGS__); \
169 assert_d_eq(strncmp(buf, expected_str_untruncated, len-1), 0, \
170 "Unexpected string inequality (\"%s\" vs \"%s\")", \
171 buf, expected_str_untruncated); \
172 assert_zu_eq(result, strlen(expected_str_untruncated), \
173 "Unexpected result"); \
174 } while (0)
175
176 for (len = 1; len < BUFLEN; len++) {
177 TEST("012346789", "012346789");
178 TEST("a0123b", "a%sb", "0123");
179 TEST("a01234567", "a%s%s", "0123", "4567");
180 TEST("a0123 ", "a%-6s", "0123");
181 TEST("a 0123", "a%6s", "0123");
182 TEST("a 012", "a%6.3s", "0123");
183 TEST("a 012", "a%*.*s", 6, 3, "0123");
184 TEST("a 123b", "a% db", 123);
185 TEST("a123b", "a%-db", 123);
186 TEST("a-123b", "a%-db", -123);
187 TEST("a+123b", "a%+db", 123);
188 }
189 #undef BUFLEN
190 #undef TEST
191 }
192 TEST_END
193
TEST_BEGIN(test_malloc_snprintf)194 TEST_BEGIN(test_malloc_snprintf)
195 {
196 #define BUFLEN 128
197 char buf[BUFLEN];
198 size_t result;
199 #define TEST(expected_str, ...) do { \
200 result = malloc_snprintf(buf, sizeof(buf), __VA_ARGS__); \
201 assert_str_eq(buf, expected_str, "Unexpected output"); \
202 assert_zu_eq(result, strlen(expected_str), "Unexpected result");\
203 } while (0)
204
205 TEST("hello", "hello");
206
207 TEST("50%, 100%", "50%%, %d%%", 100);
208
209 TEST("a0123b", "a%sb", "0123");
210
211 TEST("a 0123b", "a%5sb", "0123");
212 TEST("a 0123b", "a%*sb", 5, "0123");
213
214 TEST("a0123 b", "a%-5sb", "0123");
215 TEST("a0123b", "a%*sb", -1, "0123");
216 TEST("a0123 b", "a%*sb", -5, "0123");
217 TEST("a0123 b", "a%-*sb", -5, "0123");
218
219 TEST("a012b", "a%.3sb", "0123");
220 TEST("a012b", "a%.*sb", 3, "0123");
221 TEST("a0123b", "a%.*sb", -3, "0123");
222
223 TEST("a 012b", "a%5.3sb", "0123");
224 TEST("a 012b", "a%5.*sb", 3, "0123");
225 TEST("a 012b", "a%*.3sb", 5, "0123");
226 TEST("a 012b", "a%*.*sb", 5, 3, "0123");
227 TEST("a 0123b", "a%*.*sb", 5, -3, "0123");
228
229 TEST("_abcd_", "_%x_", 0xabcd);
230 TEST("_0xabcd_", "_%#x_", 0xabcd);
231 TEST("_1234_", "_%o_", 01234);
232 TEST("_01234_", "_%#o_", 01234);
233 TEST("_1234_", "_%u_", 1234);
234
235 TEST("_1234_", "_%d_", 1234);
236 TEST("_ 1234_", "_% d_", 1234);
237 TEST("_+1234_", "_%+d_", 1234);
238 TEST("_-1234_", "_%d_", -1234);
239 TEST("_-1234_", "_% d_", -1234);
240 TEST("_-1234_", "_%+d_", -1234);
241
242 TEST("_-1234_", "_%d_", -1234);
243 TEST("_1234_", "_%d_", 1234);
244 TEST("_-1234_", "_%i_", -1234);
245 TEST("_1234_", "_%i_", 1234);
246 TEST("_01234_", "_%#o_", 01234);
247 TEST("_1234_", "_%u_", 1234);
248 TEST("_0x1234abc_", "_%#x_", 0x1234abc);
249 TEST("_0X1234ABC_", "_%#X_", 0x1234abc);
250 TEST("_c_", "_%c_", 'c');
251 TEST("_string_", "_%s_", "string");
252 TEST("_0x42_", "_%p_", ((void *)0x42));
253
254 TEST("_-1234_", "_%ld_", ((long)-1234));
255 TEST("_1234_", "_%ld_", ((long)1234));
256 TEST("_-1234_", "_%li_", ((long)-1234));
257 TEST("_1234_", "_%li_", ((long)1234));
258 TEST("_01234_", "_%#lo_", ((long)01234));
259 TEST("_1234_", "_%lu_", ((long)1234));
260 TEST("_0x1234abc_", "_%#lx_", ((long)0x1234abc));
261 TEST("_0X1234ABC_", "_%#lX_", ((long)0x1234ABC));
262
263 TEST("_-1234_", "_%lld_", ((long long)-1234));
264 TEST("_1234_", "_%lld_", ((long long)1234));
265 TEST("_-1234_", "_%lli_", ((long long)-1234));
266 TEST("_1234_", "_%lli_", ((long long)1234));
267 TEST("_01234_", "_%#llo_", ((long long)01234));
268 TEST("_1234_", "_%llu_", ((long long)1234));
269 TEST("_0x1234abc_", "_%#llx_", ((long long)0x1234abc));
270 TEST("_0X1234ABC_", "_%#llX_", ((long long)0x1234ABC));
271
272 TEST("_-1234_", "_%qd_", ((long long)-1234));
273 TEST("_1234_", "_%qd_", ((long long)1234));
274 TEST("_-1234_", "_%qi_", ((long long)-1234));
275 TEST("_1234_", "_%qi_", ((long long)1234));
276 TEST("_01234_", "_%#qo_", ((long long)01234));
277 TEST("_1234_", "_%qu_", ((long long)1234));
278 TEST("_0x1234abc_", "_%#qx_", ((long long)0x1234abc));
279 TEST("_0X1234ABC_", "_%#qX_", ((long long)0x1234ABC));
280
281 TEST("_-1234_", "_%jd_", ((intmax_t)-1234));
282 TEST("_1234_", "_%jd_", ((intmax_t)1234));
283 TEST("_-1234_", "_%ji_", ((intmax_t)-1234));
284 TEST("_1234_", "_%ji_", ((intmax_t)1234));
285 TEST("_01234_", "_%#jo_", ((intmax_t)01234));
286 TEST("_1234_", "_%ju_", ((intmax_t)1234));
287 TEST("_0x1234abc_", "_%#jx_", ((intmax_t)0x1234abc));
288 TEST("_0X1234ABC_", "_%#jX_", ((intmax_t)0x1234ABC));
289
290 TEST("_1234_", "_%td_", ((ptrdiff_t)1234));
291 TEST("_-1234_", "_%td_", ((ptrdiff_t)-1234));
292 TEST("_1234_", "_%ti_", ((ptrdiff_t)1234));
293 TEST("_-1234_", "_%ti_", ((ptrdiff_t)-1234));
294
295 TEST("_-1234_", "_%zd_", ((ssize_t)-1234));
296 TEST("_1234_", "_%zd_", ((ssize_t)1234));
297 TEST("_-1234_", "_%zi_", ((ssize_t)-1234));
298 TEST("_1234_", "_%zi_", ((ssize_t)1234));
299 TEST("_01234_", "_%#zo_", ((ssize_t)01234));
300 TEST("_1234_", "_%zu_", ((ssize_t)1234));
301 TEST("_0x1234abc_", "_%#zx_", ((ssize_t)0x1234abc));
302 TEST("_0X1234ABC_", "_%#zX_", ((ssize_t)0x1234ABC));
303 #undef BUFLEN
304 }
305 TEST_END
306
307 int
main(void)308 main(void)
309 {
310
311 return (test(
312 test_pow2_ceil_u64,
313 test_pow2_ceil_u32,
314 test_pow2_ceil_zu,
315 test_malloc_strtoumax_no_endptr,
316 test_malloc_strtoumax,
317 test_malloc_snprintf_truncated,
318 test_malloc_snprintf));
319 }
320