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