1 /* SPDX-License-Identifier: GPL-2.0 */
2 
3 #define _GNU_SOURCE
4 #define _LARGEFILE64_SOURCE
5 
6 /* libc-specific include files
7  * The program may be built in 3 ways:
8  *   $(CC) -nostdlib -include /path/to/nolibc.h => NOLIBC already defined
9  *   $(CC) -nostdlib -I/path/to/nolibc/sysroot  => _NOLIBC_* guards are present
10  *   $(CC) with default libc                    => NOLIBC* never defined
11  */
12 #ifndef NOLIBC
13 #include <stdio.h>
14 #include <stdlib.h>
15 #include <string.h>
16 #ifndef _NOLIBC_STDIO_H
17 /* standard libcs need more includes */
18 #include <sys/auxv.h>
19 #include <sys/io.h>
20 #include <sys/ioctl.h>
21 #include <sys/mman.h>
22 #include <sys/mount.h>
23 #include <sys/prctl.h>
24 #include <sys/reboot.h>
25 #include <sys/resource.h>
26 #include <sys/stat.h>
27 #include <sys/syscall.h>
28 #include <sys/sysmacros.h>
29 #include <sys/time.h>
30 #include <sys/utsname.h>
31 #include <sys/wait.h>
32 #include <dirent.h>
33 #include <errno.h>
34 #include <fcntl.h>
35 #include <poll.h>
36 #include <sched.h>
37 #include <signal.h>
38 #include <stdarg.h>
39 #include <stddef.h>
40 #include <stdint.h>
41 #include <unistd.h>
42 #include <limits.h>
43 #endif
44 #endif
45 
46 #pragma GCC diagnostic ignored "-Wmissing-prototypes"
47 
48 #include "nolibc-test-linkage.h"
49 
50 /* for the type of int_fast16_t and int_fast32_t, musl differs from glibc and nolibc */
51 #define SINT_MAX_OF_TYPE(type) (((type)1 << (sizeof(type) * 8 - 2)) - (type)1 + ((type)1 << (sizeof(type) * 8 - 2)))
52 #define SINT_MIN_OF_TYPE(type) (-SINT_MAX_OF_TYPE(type) - 1)
53 
54 /* will be used to test initialization of environ */
55 static char **test_envp;
56 
57 /* will be used to test initialization of argv */
58 static char **test_argv;
59 
60 /* will be used to test initialization of argc */
61 static int test_argc;
62 
63 /* will be used by some test cases as readable file, please don't write it */
64 static const char *argv0;
65 
66 /* will be used by constructor tests */
67 static int constructor_test_value;
68 
69 static const int is_nolibc =
70 #ifdef NOLIBC
71 	1
72 #else
73 	0
74 #endif
75 ;
76 
77 /* definition of a series of tests */
78 struct test {
79 	const char *name;              /* test name */
80 	int (*func)(int min, int max); /* handler */
81 };
82 
83 #ifndef _NOLIBC_STDLIB_H
itoa(int i)84 char *itoa(int i)
85 {
86 	static char buf[12];
87 	int ret;
88 
89 	ret = snprintf(buf, sizeof(buf), "%d", i);
90 	return (ret >= 0 && ret < sizeof(buf)) ? buf : "#err";
91 }
92 #endif
93 
94 #define CASE_ERR(err) \
95 	case err: return #err
96 
97 /* returns the error name (e.g. "ENOENT") for common errors, "SUCCESS" for 0,
98  * or the decimal value for less common ones.
99  */
errorname(int err)100 static const char *errorname(int err)
101 {
102 	switch (err) {
103 	case 0: return "SUCCESS";
104 	CASE_ERR(EPERM);
105 	CASE_ERR(ENOENT);
106 	CASE_ERR(ESRCH);
107 	CASE_ERR(EINTR);
108 	CASE_ERR(EIO);
109 	CASE_ERR(ENXIO);
110 	CASE_ERR(E2BIG);
111 	CASE_ERR(ENOEXEC);
112 	CASE_ERR(EBADF);
113 	CASE_ERR(ECHILD);
114 	CASE_ERR(EAGAIN);
115 	CASE_ERR(ENOMEM);
116 	CASE_ERR(EACCES);
117 	CASE_ERR(EFAULT);
118 	CASE_ERR(ENOTBLK);
119 	CASE_ERR(EBUSY);
120 	CASE_ERR(EEXIST);
121 	CASE_ERR(EXDEV);
122 	CASE_ERR(ENODEV);
123 	CASE_ERR(ENOTDIR);
124 	CASE_ERR(EISDIR);
125 	CASE_ERR(EINVAL);
126 	CASE_ERR(ENFILE);
127 	CASE_ERR(EMFILE);
128 	CASE_ERR(ENOTTY);
129 	CASE_ERR(ETXTBSY);
130 	CASE_ERR(EFBIG);
131 	CASE_ERR(ENOSPC);
132 	CASE_ERR(ESPIPE);
133 	CASE_ERR(EROFS);
134 	CASE_ERR(EMLINK);
135 	CASE_ERR(EPIPE);
136 	CASE_ERR(EDOM);
137 	CASE_ERR(ERANGE);
138 	CASE_ERR(ENOSYS);
139 	CASE_ERR(EOVERFLOW);
140 	default:
141 		return itoa(err);
142 	}
143 }
144 
align_result(size_t llen)145 static void align_result(size_t llen)
146 {
147 	const size_t align = 64;
148 	char buf[align];
149 	size_t n;
150 
151 	if (llen >= align)
152 		return;
153 
154 	n = align - llen;
155 	memset(buf, ' ', n);
156 	buf[n] = '\0';
157 	fputs(buf, stdout);
158 }
159 
160 enum RESULT {
161 	OK,
162 	FAIL,
163 	SKIPPED,
164 };
165 
result(int llen,enum RESULT r)166 static void result(int llen, enum RESULT r)
167 {
168 	const char *msg;
169 
170 	if (r == OK)
171 		msg = "  [OK]";
172 	else if (r == SKIPPED)
173 		msg = "[SKIPPED]";
174 	else
175 		msg = " [FAIL]";
176 
177 	align_result(llen);
178 	puts(msg);
179 }
180 
181 /* The tests below are intended to be used by the macroes, which evaluate
182  * expression <expr>, print the status to stdout, and update the "ret"
183  * variable to count failures. The functions themselves return the number
184  * of failures, thus either 0 or 1.
185  */
186 
187 #define EXPECT_ZR(cond, expr)				\
188 	do { if (!(cond)) result(llen, SKIPPED); else ret += expect_zr(expr, llen); } while (0)
189 
190 static __attribute__((unused))
expect_zr(int expr,int llen)191 int expect_zr(int expr, int llen)
192 {
193 	int ret = !(expr == 0);
194 
195 	llen += printf(" = %d ", expr);
196 	result(llen, ret ? FAIL : OK);
197 	return ret;
198 }
199 
200 
201 #define EXPECT_NZ(cond, expr, val)			\
202 	do { if (!(cond)) result(llen, SKIPPED); else ret += expect_nz(expr, llen; } while (0)
203 
204 static __attribute__((unused))
expect_nz(int expr,int llen)205 int expect_nz(int expr, int llen)
206 {
207 	int ret = !(expr != 0);
208 
209 	llen += printf(" = %d ", expr);
210 	result(llen, ret ? FAIL : OK);
211 	return ret;
212 }
213 
214 
215 #define EXPECT_EQ(cond, expr, val)				\
216 	do { if (!(cond)) result(llen, SKIPPED); else ret += expect_eq(expr, llen, val); } while (0)
217 
218 static __attribute__((unused))
expect_eq(uint64_t expr,int llen,uint64_t val)219 int expect_eq(uint64_t expr, int llen, uint64_t val)
220 {
221 	int ret = !(expr == val);
222 
223 	llen += printf(" = %lld ", (long long)expr);
224 	result(llen, ret ? FAIL : OK);
225 	return ret;
226 }
227 
228 
229 #define EXPECT_NE(cond, expr, val)				\
230 	do { if (!(cond)) result(llen, SKIPPED); else ret += expect_ne(expr, llen, val); } while (0)
231 
232 static __attribute__((unused))
expect_ne(int expr,int llen,int val)233 int expect_ne(int expr, int llen, int val)
234 {
235 	int ret = !(expr != val);
236 
237 	llen += printf(" = %d ", expr);
238 	result(llen, ret ? FAIL : OK);
239 	return ret;
240 }
241 
242 
243 #define EXPECT_GE(cond, expr, val)				\
244 	do { if (!(cond)) result(llen, SKIPPED); else ret += expect_ge(expr, llen, val); } while (0)
245 
246 static __attribute__((unused))
expect_ge(int expr,int llen,int val)247 int expect_ge(int expr, int llen, int val)
248 {
249 	int ret = !(expr >= val);
250 
251 	llen += printf(" = %d ", expr);
252 	result(llen, ret ? FAIL : OK);
253 	return ret;
254 }
255 
256 
257 #define EXPECT_GT(cond, expr, val)				\
258 	do { if (!(cond)) result(llen, SKIPPED); else ret += expect_gt(expr, llen, val); } while (0)
259 
260 static __attribute__((unused))
expect_gt(int expr,int llen,int val)261 int expect_gt(int expr, int llen, int val)
262 {
263 	int ret = !(expr > val);
264 
265 	llen += printf(" = %d ", expr);
266 	result(llen, ret ? FAIL : OK);
267 	return ret;
268 }
269 
270 
271 #define EXPECT_LE(cond, expr, val)				\
272 	do { if (!(cond)) result(llen, SKIPPED); else ret += expect_le(expr, llen, val); } while (0)
273 
274 static __attribute__((unused))
expect_le(int expr,int llen,int val)275 int expect_le(int expr, int llen, int val)
276 {
277 	int ret = !(expr <= val);
278 
279 	llen += printf(" = %d ", expr);
280 	result(llen, ret ? FAIL : OK);
281 	return ret;
282 }
283 
284 
285 #define EXPECT_LT(cond, expr, val)				\
286 	do { if (!(cond)) result(llen, SKIPPED); else ret += expect_lt(expr, llen, val); } while (0)
287 
288 static __attribute__((unused))
expect_lt(int expr,int llen,int val)289 int expect_lt(int expr, int llen, int val)
290 {
291 	int ret = !(expr < val);
292 
293 	llen += printf(" = %d ", expr);
294 	result(llen, ret ? FAIL : OK);
295 	return ret;
296 }
297 
298 
299 #define EXPECT_SYSZR(cond, expr)				\
300 	do { if (!(cond)) result(llen, SKIPPED); else ret += expect_syszr(expr, llen); } while (0)
301 
302 static __attribute__((unused))
expect_syszr(int expr,int llen)303 int expect_syszr(int expr, int llen)
304 {
305 	int ret = 0;
306 
307 	if (errno == ENOSYS) {
308 		llen += printf(" = ENOSYS");
309 		result(llen, SKIPPED);
310 	} else if (expr) {
311 		ret = 1;
312 		llen += printf(" = %d %s ", expr, errorname(errno));
313 		result(llen, FAIL);
314 	} else {
315 		llen += printf(" = %d ", expr);
316 		result(llen, OK);
317 	}
318 	return ret;
319 }
320 
321 
322 #define EXPECT_SYSEQ(cond, expr, val)				\
323 	do { if (!(cond)) result(llen, SKIPPED); else ret += expect_syseq(expr, llen, val); } while (0)
324 
325 static __attribute__((unused))
expect_syseq(int expr,int llen,int val)326 int expect_syseq(int expr, int llen, int val)
327 {
328 	int ret = 0;
329 
330 	if (expr != val) {
331 		ret = 1;
332 		llen += printf(" = %d %s ", expr, errorname(errno));
333 		result(llen, FAIL);
334 	} else {
335 		llen += printf(" = %d ", expr);
336 		result(llen, OK);
337 	}
338 	return ret;
339 }
340 
341 
342 #define EXPECT_SYSNE(cond, expr, val)				\
343 	do { if (!(cond)) result(llen, SKIPPED); else ret += expect_sysne(expr, llen, val); } while (0)
344 
345 static __attribute__((unused))
expect_sysne(int expr,int llen,int val)346 int expect_sysne(int expr, int llen, int val)
347 {
348 	int ret = 0;
349 
350 	if (errno == ENOSYS) {
351 		llen += printf(" = ENOSYS");
352 		result(llen, SKIPPED);
353 	} else if (expr == val) {
354 		ret = 1;
355 		llen += printf(" = %d %s ", expr, errorname(errno));
356 		result(llen, FAIL);
357 	} else {
358 		llen += printf(" = %d ", expr);
359 		result(llen, OK);
360 	}
361 	return ret;
362 }
363 
364 
365 #define EXPECT_SYSER2(cond, expr, expret, experr1, experr2)		\
366 	do { if (!(cond)) result(llen, SKIPPED); else ret += expect_syserr2(expr, expret, experr1, experr2, llen); } while (0)
367 
368 #define EXPECT_SYSER(cond, expr, expret, experr)			\
369 	EXPECT_SYSER2(cond, expr, expret, experr, 0)
370 
371 static __attribute__((unused))
expect_syserr2(int expr,int expret,int experr1,int experr2,int llen)372 int expect_syserr2(int expr, int expret, int experr1, int experr2, int llen)
373 {
374 	int ret = 0;
375 	int _errno = errno;
376 
377 	llen += printf(" = %d %s ", expr, errorname(_errno));
378 	if (errno == ENOSYS) {
379 		result(llen, SKIPPED);
380 	} else if (expr != expret || (_errno != experr1 && _errno != experr2)) {
381 		ret = 1;
382 		if (experr2 == 0)
383 			llen += printf(" != (%d %s) ", expret, errorname(experr1));
384 		else
385 			llen += printf(" != (%d %s %s) ", expret, errorname(experr1), errorname(experr2));
386 		result(llen, FAIL);
387 	} else {
388 		result(llen, OK);
389 	}
390 	return ret;
391 }
392 
393 
394 #define EXPECT_PTRZR(cond, expr)				\
395 	do { if (!(cond)) result(llen, SKIPPED); else ret += expect_ptrzr(expr, llen); } while (0)
396 
397 static __attribute__((unused))
expect_ptrzr(const void * expr,int llen)398 int expect_ptrzr(const void *expr, int llen)
399 {
400 	int ret = 0;
401 
402 	llen += printf(" = <%p> ", expr);
403 	if (expr) {
404 		ret = 1;
405 		result(llen, FAIL);
406 	} else {
407 		result(llen, OK);
408 	}
409 	return ret;
410 }
411 
412 
413 #define EXPECT_PTRNZ(cond, expr)				\
414 	do { if (!(cond)) result(llen, SKIPPED); else ret += expect_ptrnz(expr, llen); } while (0)
415 
416 static __attribute__((unused))
expect_ptrnz(const void * expr,int llen)417 int expect_ptrnz(const void *expr, int llen)
418 {
419 	int ret = 0;
420 
421 	llen += printf(" = <%p> ", expr);
422 	if (!expr) {
423 		ret = 1;
424 		result(llen, FAIL);
425 	} else {
426 		result(llen, OK);
427 	}
428 	return ret;
429 }
430 
431 #define EXPECT_PTREQ(cond, expr, cmp)				\
432 	do { if (!(cond)) result(llen, SKIPPED); else ret += expect_ptreq(expr, llen, cmp); } while (0)
433 
434 static __attribute__((unused))
expect_ptreq(const void * expr,int llen,const void * cmp)435 int expect_ptreq(const void *expr, int llen, const void *cmp)
436 {
437 	int ret = 0;
438 
439 	llen += printf(" = <%p> ", expr);
440 	if (expr != cmp) {
441 		ret = 1;
442 		result(llen, FAIL);
443 	} else {
444 		result(llen, OK);
445 	}
446 	return ret;
447 }
448 
449 #define EXPECT_PTRNE(cond, expr, cmp)				\
450 	do { if (!(cond)) result(llen, SKIPPED); else ret += expect_ptrne(expr, llen, cmp); } while (0)
451 
452 static __attribute__((unused))
expect_ptrne(const void * expr,int llen,const void * cmp)453 int expect_ptrne(const void *expr, int llen, const void *cmp)
454 {
455 	int ret = 0;
456 
457 	llen += printf(" = <%p> ", expr);
458 	if (expr == cmp) {
459 		ret = 1;
460 		result(llen, FAIL);
461 	} else {
462 		result(llen, OK);
463 	}
464 	return ret;
465 }
466 
467 #define EXPECT_PTRGE(cond, expr, cmp)				\
468 	do { if (!(cond)) result(llen, SKIPPED); else ret += expect_ptrge(expr, llen, cmp); } while (0)
469 
470 static __attribute__((unused))
expect_ptrge(const void * expr,int llen,const void * cmp)471 int expect_ptrge(const void *expr, int llen, const void *cmp)
472 {
473 	int ret = !(expr >= cmp);
474 
475 	llen += printf(" = <%p> ", expr);
476 	result(llen, ret ? FAIL : OK);
477 	return ret;
478 }
479 
480 #define EXPECT_PTRGT(cond, expr, cmp)				\
481 	do { if (!(cond)) result(llen, SKIPPED); else ret += expect_ptrgt(expr, llen, cmp); } while (0)
482 
483 static __attribute__((unused))
expect_ptrgt(const void * expr,int llen,const void * cmp)484 int expect_ptrgt(const void *expr, int llen, const void *cmp)
485 {
486 	int ret = !(expr > cmp);
487 
488 	llen += printf(" = <%p> ", expr);
489 	result(llen, ret ? FAIL : OK);
490 	return ret;
491 }
492 
493 
494 #define EXPECT_PTRLE(cond, expr, cmp)				\
495 	do { if (!(cond)) result(llen, SKIPPED); else ret += expect_ptrle(expr, llen, cmp); } while (0)
496 
497 static __attribute__((unused))
expect_ptrle(const void * expr,int llen,const void * cmp)498 int expect_ptrle(const void *expr, int llen, const void *cmp)
499 {
500 	int ret = !(expr <= cmp);
501 
502 	llen += printf(" = <%p> ", expr);
503 	result(llen, ret ? FAIL : OK);
504 	return ret;
505 }
506 
507 
508 #define EXPECT_PTRLT(cond, expr, cmp)				\
509 	do { if (!(cond)) result(llen, SKIPPED); else ret += expect_ptrlt(expr, llen, cmp); } while (0)
510 
511 static __attribute__((unused))
expect_ptrlt(const void * expr,int llen,const void * cmp)512 int expect_ptrlt(const void *expr, int llen, const void *cmp)
513 {
514 	int ret = !(expr < cmp);
515 
516 	llen += printf(" = <%p> ", expr);
517 	result(llen, ret ? FAIL : OK);
518 	return ret;
519 }
520 
521 #define EXPECT_PTRER2(cond, expr, expret, experr1, experr2)		\
522 	do { if (!(cond)) result(llen, SKIPPED); else ret += expect_ptrerr2(expr, expret, experr1, experr2, llen); } while (0)
523 
524 #define EXPECT_PTRER(cond, expr, expret, experr)			\
525 	EXPECT_PTRER2(cond, expr, expret, experr, 0)
526 
527 static __attribute__((unused))
expect_ptrerr2(const void * expr,const void * expret,int experr1,int experr2,int llen)528 int expect_ptrerr2(const void *expr, const void *expret, int experr1, int experr2, int llen)
529 {
530 	int ret = 0;
531 	int _errno = errno;
532 
533 	llen += printf(" = <%p> %s ", expr, errorname(_errno));
534 	if (expr != expret || (_errno != experr1 && _errno != experr2)) {
535 		ret = 1;
536 		if (experr2 == 0)
537 			llen += printf(" != (<%p> %s) ", expret, errorname(experr1));
538 		else
539 			llen += printf(" != (<%p> %s %s) ", expret, errorname(experr1), errorname(experr2));
540 		result(llen, FAIL);
541 	} else {
542 		result(llen, OK);
543 	}
544 	return ret;
545 }
546 
547 #define EXPECT_STRZR(cond, expr)				\
548 	do { if (!(cond)) result(llen, SKIPPED); else ret += expect_strzr(expr, llen); } while (0)
549 
550 static __attribute__((unused))
expect_strzr(const char * expr,int llen)551 int expect_strzr(const char *expr, int llen)
552 {
553 	int ret = 0;
554 
555 	llen += printf(" = <%s> ", expr ? expr : "(null)");
556 	if (expr) {
557 		ret = 1;
558 		result(llen, FAIL);
559 	} else {
560 		result(llen, OK);
561 	}
562 	return ret;
563 }
564 
565 
566 #define EXPECT_STRNZ(cond, expr)				\
567 	do { if (!(cond)) result(llen, SKIPPED); else ret += expect_strnz(expr, llen); } while (0)
568 
569 static __attribute__((unused))
expect_strnz(const char * expr,int llen)570 int expect_strnz(const char *expr, int llen)
571 {
572 	int ret = 0;
573 
574 	llen += printf(" = <%s> ", expr ? expr : "(null)");
575 	if (!expr) {
576 		ret = 1;
577 		result(llen, FAIL);
578 	} else {
579 		result(llen, OK);
580 	}
581 	return ret;
582 }
583 
584 
585 #define EXPECT_STREQ(cond, expr, cmp)				\
586 	do { if (!(cond)) result(llen, SKIPPED); else ret += expect_streq(expr, llen, cmp); } while (0)
587 
588 static __attribute__((unused))
expect_streq(const char * expr,int llen,const char * cmp)589 int expect_streq(const char *expr, int llen, const char *cmp)
590 {
591 	int ret = 0;
592 
593 	llen += printf(" = <%s> ", expr);
594 	if (strcmp(expr, cmp) != 0) {
595 		ret = 1;
596 		result(llen, FAIL);
597 	} else {
598 		result(llen, OK);
599 	}
600 	return ret;
601 }
602 
603 
604 #define EXPECT_STRNE(cond, expr, cmp)				\
605 	do { if (!(cond)) result(llen, SKIPPED); else ret += expect_strne(expr, llen, cmp); } while (0)
606 
607 static __attribute__((unused))
expect_strne(const char * expr,int llen,const char * cmp)608 int expect_strne(const char *expr, int llen, const char *cmp)
609 {
610 	int ret = 0;
611 
612 	llen += printf(" = <%s> ", expr);
613 	if (strcmp(expr, cmp) == 0) {
614 		ret = 1;
615 		result(llen, FAIL);
616 	} else {
617 		result(llen, OK);
618 	}
619 	return ret;
620 }
621 
622 #define EXPECT_STRBUFEQ(cond, expr, buf, val, cmp)				\
623 	do { if (!(cond)) result(llen, SKIPPED); else ret += expect_str_buf_eq(expr, buf, val, llen, cmp); } while (0)
624 
625 static __attribute__((unused))
expect_str_buf_eq(size_t expr,const char * buf,size_t val,int llen,const char * cmp)626 int expect_str_buf_eq(size_t expr, const char *buf, size_t val, int llen, const char *cmp)
627 {
628 	llen += printf(" = %lu <%s> ", (unsigned long)expr, buf);
629 	if (strcmp(buf, cmp) != 0) {
630 		result(llen, FAIL);
631 		return 1;
632 	}
633 	if (expr != val) {
634 		result(llen, FAIL);
635 		return 1;
636 	}
637 
638 	result(llen, OK);
639 	return 0;
640 }
641 
642 #define EXPECT_STRTOX(cond, func, input, base, expected, chars, expected_errno)				\
643 	do { if (!(cond)) result(llen, SKIPPED); else ret += expect_strtox(llen, func, input, base, expected, chars, expected_errno); } while (0)
644 
645 static __attribute__((unused))
expect_strtox(int llen,void * func,const char * input,int base,intmax_t expected,int expected_chars,int expected_errno)646 int expect_strtox(int llen, void *func, const char *input, int base, intmax_t expected, int expected_chars, int expected_errno)
647 {
648 	char *endptr;
649 	int actual_errno, actual_chars;
650 	intmax_t r;
651 
652 	errno = 0;
653 	if (func == strtol) {
654 		r = strtol(input, &endptr, base);
655 	} else if (func == strtoul) {
656 		r = strtoul(input, &endptr, base);
657 	} else {
658 		result(llen, FAIL);
659 		return 1;
660 	}
661 	actual_errno = errno;
662 	actual_chars = endptr - input;
663 
664 	llen += printf(" %lld = %lld", (long long)expected, (long long)r);
665 	if (r != expected) {
666 		result(llen, FAIL);
667 		return 1;
668 	}
669 	if (expected_chars == -1) {
670 		if (*endptr != '\0') {
671 			result(llen, FAIL);
672 			return 1;
673 		}
674 	} else if (expected_chars != actual_chars) {
675 		result(llen, FAIL);
676 		return 1;
677 	}
678 	if (actual_errno != expected_errno) {
679 		result(llen, FAIL);
680 		return 1;
681 	}
682 
683 	result(llen, OK);
684 	return 0;
685 }
686 
687 /* declare tests based on line numbers. There must be exactly one test per line. */
688 #define CASE_TEST(name) \
689 	case __LINE__: llen += printf("%d %s", test, #name);
690 
691 /* constructors validate that they are executed in definition order */
692 __attribute__((constructor))
constructor1(void)693 static void constructor1(void)
694 {
695 	constructor_test_value |= 1 << 0;
696 }
697 
698 __attribute__((constructor))
constructor2(int argc,char ** argv,char ** envp)699 static void constructor2(int argc, char **argv, char **envp)
700 {
701 	if (argc && argv && envp)
702 		constructor_test_value |= 1 << 1;
703 }
704 
run_startup(int min,int max)705 int run_startup(int min, int max)
706 {
707 	int test;
708 	int ret = 0;
709 	/* kernel at least passes HOME and TERM, shell passes more */
710 	int env_total = 2;
711 	/* checking NULL for argv/argv0, environ and _auxv is not enough, let's compare with sbrk(0) or &end */
712 	extern char end;
713 	char *brk = sbrk(0) != (void *)-1 ? sbrk(0) : &end;
714 	/* differ from nolibc, both glibc and musl have no global _auxv */
715 	const unsigned long *test_auxv = (void *)-1;
716 #ifdef NOLIBC
717 	test_auxv = _auxv;
718 #endif
719 
720 	for (test = min; test >= 0 && test <= max; test++) {
721 		int llen = 0; /* line length */
722 
723 		/* avoid leaving empty lines below, this will insert holes into
724 		 * test numbers.
725 		 */
726 		switch (test + __LINE__ + 1) {
727 		CASE_TEST(argc);             EXPECT_GE(1, test_argc, 1); break;
728 		CASE_TEST(argv_addr);        EXPECT_PTRGT(1, test_argv, brk); break;
729 		CASE_TEST(argv_environ);     EXPECT_PTRLT(1, test_argv, environ); break;
730 		CASE_TEST(argv_total);       EXPECT_EQ(1, environ - test_argv - 1, test_argc ?: 1); break;
731 		CASE_TEST(argv0_addr);       EXPECT_PTRGT(1, argv0, brk); break;
732 		CASE_TEST(argv0_str);        EXPECT_STRNZ(1, argv0 > brk ? argv0 : NULL); break;
733 		CASE_TEST(argv0_len);        EXPECT_GE(1,  argv0 > brk ? strlen(argv0) : 0, 1); break;
734 		CASE_TEST(environ_addr);     EXPECT_PTRGT(1, environ, brk); break;
735 		CASE_TEST(environ_envp);     EXPECT_PTREQ(1, environ, test_envp); break;
736 		CASE_TEST(environ_auxv);     EXPECT_PTRLT(test_auxv != (void *)-1, environ, test_auxv); break;
737 		CASE_TEST(environ_total);    EXPECT_GE(test_auxv != (void *)-1, (void *)test_auxv - (void *)environ - 1, env_total); break;
738 		CASE_TEST(environ_HOME);     EXPECT_PTRNZ(1, getenv("HOME")); break;
739 		CASE_TEST(auxv_addr);        EXPECT_PTRGT(test_auxv != (void *)-1, test_auxv, brk); break;
740 		CASE_TEST(auxv_AT_UID);      EXPECT_EQ(1, getauxval(AT_UID), getuid()); break;
741 		CASE_TEST(constructor);      EXPECT_EQ(is_nolibc, constructor_test_value, 0x3); break;
742 		CASE_TEST(linkage_errno);    EXPECT_PTREQ(1, linkage_test_errno_addr(), &errno); break;
743 		CASE_TEST(linkage_constr);   EXPECT_EQ(1, linkage_test_constructor_test_value, 0x3); break;
744 		case __LINE__:
745 			return ret; /* must be last */
746 		/* note: do not set any defaults so as to permit holes above */
747 		}
748 	}
749 	return ret;
750 }
751 
752 
753 /* used by some syscall tests below */
test_getdents64(const char * dir)754 int test_getdents64(const char *dir)
755 {
756 	char buffer[4096];
757 	int fd, ret;
758 	int err;
759 
760 	ret = fd = open(dir, O_RDONLY | O_DIRECTORY, 0);
761 	if (ret < 0)
762 		return ret;
763 
764 	ret = getdents64(fd, (void *)buffer, sizeof(buffer));
765 	err = errno;
766 	close(fd);
767 
768 	errno = err;
769 	return ret;
770 }
771 
test_dirent(void)772 static int test_dirent(void)
773 {
774 	int comm = 0, cmdline = 0;
775 	struct dirent dirent, *result;
776 	DIR *dir;
777 	int ret;
778 
779 	dir = opendir("/proc/self");
780 	if (!dir)
781 		return 1;
782 
783 	while (1) {
784 		errno = 0;
785 		ret = readdir_r(dir, &dirent, &result);
786 		if (ret != 0)
787 			return 1;
788 		if (!result)
789 			break;
790 
791 		if (strcmp(dirent.d_name, "comm") == 0)
792 			comm++;
793 		else if (strcmp(dirent.d_name, "cmdline") == 0)
794 			cmdline++;
795 	}
796 
797 	if (errno)
798 		return 1;
799 
800 	ret = closedir(dir);
801 	if (ret)
802 		return 1;
803 
804 	if (comm != 1 || cmdline != 1)
805 		return 1;
806 
807 	return 0;
808 }
809 
test_getpagesize(void)810 int test_getpagesize(void)
811 {
812 	int x = getpagesize();
813 	int c;
814 
815 	if (x < 0)
816 		return x;
817 
818 #if defined(__x86_64__) || defined(__i386__) || defined(__i486__) || defined(__i586__) || defined(__i686__)
819 	/*
820 	 * x86 family is always 4K page.
821 	 */
822 	c = (x == 4096);
823 #elif defined(__aarch64__)
824 	/*
825 	 * Linux aarch64 supports three values of page size: 4K, 16K, and 64K
826 	 * which are selected at kernel compilation time.
827 	 */
828 	c = (x == 4096 || x == (16 * 1024) || x == (64 * 1024));
829 #else
830 	/*
831 	 * Assuming other architectures must have at least 4K page.
832 	 */
833 	c = (x >= 4096);
834 #endif
835 
836 	return !c;
837 }
838 
test_fork(void)839 int test_fork(void)
840 {
841 	int status;
842 	pid_t pid;
843 
844 	/* flush the printf buffer to avoid child flush it */
845 	fflush(stdout);
846 	fflush(stderr);
847 
848 	pid = fork();
849 
850 	switch (pid) {
851 	case -1:
852 		return 1;
853 
854 	case 0:
855 		exit(123);
856 
857 	default:
858 		pid = waitpid(pid, &status, 0);
859 
860 		return pid == -1 || !WIFEXITED(status) || WEXITSTATUS(status) != 123;
861 	}
862 }
863 
test_stat_timestamps(void)864 int test_stat_timestamps(void)
865 {
866 	struct stat st;
867 
868 	if (sizeof(st.st_atim.tv_sec) != sizeof(st.st_atime))
869 		return 1;
870 
871 	if (stat("/proc/self/", &st) && stat(argv0, &st) && stat("/", &st))
872 		return 1;
873 
874 	if (st.st_atim.tv_sec != st.st_atime || st.st_atim.tv_nsec > 1000000000)
875 		return 1;
876 
877 	if (st.st_mtim.tv_sec != st.st_mtime || st.st_mtim.tv_nsec > 1000000000)
878 		return 1;
879 
880 	if (st.st_ctim.tv_sec != st.st_ctime || st.st_ctim.tv_nsec > 1000000000)
881 		return 1;
882 
883 	return 0;
884 }
885 
test_uname(void)886 int test_uname(void)
887 {
888 	struct utsname buf;
889 	char osrelease[sizeof(buf.release)];
890 	ssize_t r;
891 	int fd;
892 
893 	memset(&buf.domainname, 'P', sizeof(buf.domainname));
894 
895 	if (uname(&buf))
896 		return 1;
897 
898 	if (strncmp("Linux", buf.sysname, sizeof(buf.sysname)))
899 		return 1;
900 
901 	fd = open("/proc/sys/kernel/osrelease", O_RDONLY);
902 	if (fd == -1)
903 		return 1;
904 
905 	r = read(fd, osrelease, sizeof(osrelease));
906 	if (r == -1)
907 		return 1;
908 
909 	close(fd);
910 
911 	if (osrelease[r - 1] == '\n')
912 		r--;
913 
914 	/* Validate one of the later fields to ensure field sizes are correct */
915 	if (strncmp(osrelease, buf.release, r))
916 		return 1;
917 
918 	/* Ensure the field domainname is set, it is missing from struct old_utsname */
919 	if (strnlen(buf.domainname, sizeof(buf.domainname)) == sizeof(buf.domainname))
920 		return 1;
921 
922 	return 0;
923 }
924 
test_mmap_munmap(void)925 int test_mmap_munmap(void)
926 {
927 	int ret, fd, i, page_size;
928 	void *mem;
929 	size_t file_size, length;
930 	off_t offset, pa_offset;
931 	struct stat stat_buf;
932 	const char * const files[] = {
933 		"/dev/zero",
934 		"/proc/1/exe", "/proc/self/exe",
935 		argv0,
936 		NULL
937 	};
938 
939 	page_size = getpagesize();
940 	if (page_size < 0)
941 		return 1;
942 
943 	/* find a right file to mmap, existed and accessible */
944 	for (i = 0; files[i] != NULL; i++) {
945 		ret = fd = open(files[i], O_RDONLY);
946 		if (ret == -1)
947 			continue;
948 		else
949 			break;
950 	}
951 	if (ret == -1)
952 		return 1;
953 
954 	ret = stat(files[i], &stat_buf);
955 	if (ret == -1)
956 		goto end;
957 
958 	/* file size of the special /dev/zero is 0, let's assign one manually */
959 	if (i == 0)
960 		file_size = 3*page_size;
961 	else
962 		file_size = stat_buf.st_size;
963 
964 	offset = file_size - 1;
965 	if (offset < 0)
966 		offset = 0;
967 	length = file_size - offset;
968 	pa_offset = offset & ~(page_size - 1);
969 
970 	mem = mmap(NULL, length + offset - pa_offset, PROT_READ, MAP_SHARED, fd, pa_offset);
971 	if (mem == MAP_FAILED) {
972 		ret = 1;
973 		goto end;
974 	}
975 
976 	ret = munmap(mem, length + offset - pa_offset);
977 
978 end:
979 	close(fd);
980 	return !!ret;
981 }
982 
test_pipe(void)983 int test_pipe(void)
984 {
985 	const char *const msg = "hello, nolibc";
986 	int pipefd[2];
987 	char buf[32];
988 	size_t len;
989 
990 	if (pipe(pipefd) == -1)
991 		return 1;
992 
993 	write(pipefd[1], msg, strlen(msg));
994 	close(pipefd[1]);
995 	len = read(pipefd[0], buf, sizeof(buf));
996 	close(pipefd[0]);
997 
998 	if (len != strlen(msg))
999 		return 1;
1000 
1001 	return !!memcmp(buf, msg, len);
1002 }
1003 
test_rlimit(void)1004 int test_rlimit(void)
1005 {
1006 	struct rlimit rlim = {
1007 		.rlim_cur = 1 << 20,
1008 		.rlim_max = 1 << 21,
1009 	};
1010 	int ret;
1011 
1012 	ret = setrlimit(RLIMIT_CORE, &rlim);
1013 	if (ret)
1014 		return -1;
1015 
1016 	rlim.rlim_cur = 0;
1017 	rlim.rlim_max = 0;
1018 
1019 	ret = getrlimit(RLIMIT_CORE, &rlim);
1020 	if (ret)
1021 		return -1;
1022 
1023 	if (rlim.rlim_cur != 1 << 20)
1024 		return -1;
1025 	if (rlim.rlim_max != 1 << 21)
1026 		return -1;
1027 
1028 	return 0;
1029 }
1030 
test_openat(void)1031 int test_openat(void)
1032 {
1033 	int dev, null;
1034 
1035 	dev = openat(AT_FDCWD, "/dev", O_DIRECTORY);
1036 	if (dev < 0)
1037 		return -1;
1038 
1039 	null = openat(dev, "null", O_RDONLY);
1040 	close(dev);
1041 	if (null < 0)
1042 		return -1;
1043 
1044 	close(null);
1045 	return 0;
1046 }
1047 
1048 /* Run syscall tests between IDs <min> and <max>.
1049  * Return 0 on success, non-zero on failure.
1050  */
run_syscall(int min,int max)1051 int run_syscall(int min, int max)
1052 {
1053 	struct timeval tv;
1054 	struct timezone tz;
1055 	struct stat stat_buf;
1056 	int euid0;
1057 	int proc;
1058 	int test;
1059 	int tmp;
1060 	int ret = 0;
1061 	void *p1, *p2;
1062 	int has_gettid = 1;
1063 	int has_brk;
1064 
1065 	/* <proc> indicates whether or not /proc is mounted */
1066 	proc = stat("/proc", &stat_buf) == 0;
1067 
1068 	/* this will be used to skip certain tests that can't be run unprivileged */
1069 	euid0 = geteuid() == 0;
1070 
1071 	/* from 2.30, glibc provides gettid() */
1072 #if defined(__GLIBC_MINOR__) && defined(__GLIBC__)
1073 	has_gettid = __GLIBC__ > 2 || (__GLIBC__ == 2 && __GLIBC_MINOR__ >= 30);
1074 #endif
1075 
1076 	/* on musl setting brk()/sbrk() always fails */
1077 	has_brk = brk(0) == 0;
1078 
1079 	for (test = min; test >= 0 && test <= max; test++) {
1080 		int llen = 0; /* line length */
1081 
1082 		/* avoid leaving empty lines below, this will insert holes into
1083 		 * test numbers.
1084 		 */
1085 		switch (test + __LINE__ + 1) {
1086 		CASE_TEST(getpid);            EXPECT_SYSNE(1, getpid(), -1); break;
1087 		CASE_TEST(getppid);           EXPECT_SYSNE(1, getppid(), -1); break;
1088 		CASE_TEST(gettid);            EXPECT_SYSNE(has_gettid, gettid(), -1); break;
1089 		CASE_TEST(getpgid_self);      EXPECT_SYSNE(1, getpgid(0), -1); break;
1090 		CASE_TEST(getpgid_bad);       EXPECT_SYSER(1, getpgid(-1), -1, ESRCH); break;
1091 		CASE_TEST(kill_0);            EXPECT_SYSZR(1, kill(getpid(), 0)); break;
1092 		CASE_TEST(kill_CONT);         EXPECT_SYSZR(1, kill(getpid(), 0)); break;
1093 		CASE_TEST(kill_BADPID);       EXPECT_SYSER(1, kill(INT_MAX, 0), -1, ESRCH); break;
1094 		CASE_TEST(sbrk_0);            EXPECT_PTRNE(has_brk, sbrk(0), (void *)-1); break;
1095 		CASE_TEST(sbrk);              if ((p1 = p2 = sbrk(4096)) != (void *)-1) p2 = sbrk(-4096); EXPECT_SYSZR(has_brk, (p2 == (void *)-1) || p2 == p1); break;
1096 		CASE_TEST(brk);               EXPECT_SYSZR(has_brk, brk(sbrk(0))); break;
1097 		CASE_TEST(chdir_root);        EXPECT_SYSZR(1, chdir("/")); chdir(getenv("PWD")); break;
1098 		CASE_TEST(chdir_dot);         EXPECT_SYSZR(1, chdir(".")); break;
1099 		CASE_TEST(chdir_blah);        EXPECT_SYSER(1, chdir("/blah"), -1, ENOENT); break;
1100 		CASE_TEST(chmod_argv0);       EXPECT_SYSZR(1, chmod(argv0, 0555)); break;
1101 		CASE_TEST(chmod_self);        EXPECT_SYSER(proc, chmod("/proc/self", 0555), -1, EPERM); break;
1102 		CASE_TEST(chown_self);        EXPECT_SYSER(proc, chown("/proc/self", 0, 0), -1, EPERM); break;
1103 		CASE_TEST(chroot_root);       EXPECT_SYSZR(euid0, chroot("/")); break;
1104 		CASE_TEST(chroot_blah);       EXPECT_SYSER(1, chroot("/proc/self/blah"), -1, ENOENT); break;
1105 		CASE_TEST(chroot_exe);        EXPECT_SYSER(1, chroot(argv0), -1, ENOTDIR); break;
1106 		CASE_TEST(close_m1);          EXPECT_SYSER(1, close(-1), -1, EBADF); break;
1107 		CASE_TEST(close_dup);         EXPECT_SYSZR(1, close(dup(0))); break;
1108 		CASE_TEST(dup_0);             tmp = dup(0);  EXPECT_SYSNE(1, tmp, -1); close(tmp); break;
1109 		CASE_TEST(dup_m1);            tmp = dup(-1); EXPECT_SYSER(1, tmp, -1, EBADF); if (tmp != -1) close(tmp); break;
1110 		CASE_TEST(dup2_0);            tmp = dup2(0, 100);  EXPECT_SYSNE(1, tmp, -1); close(tmp); break;
1111 		CASE_TEST(dup2_m1);           tmp = dup2(-1, 100); EXPECT_SYSER(1, tmp, -1, EBADF); if (tmp != -1) close(tmp); break;
1112 		CASE_TEST(dup3_0);            tmp = dup3(0, 100, 0);  EXPECT_SYSNE(1, tmp, -1); close(tmp); break;
1113 		CASE_TEST(dup3_m1);           tmp = dup3(-1, 100, 0); EXPECT_SYSER(1, tmp, -1, EBADF); if (tmp != -1) close(tmp); break;
1114 		CASE_TEST(execve_root);       EXPECT_SYSER(1, execve("/", (char*[]){ [0] = "/", [1] = NULL }, NULL), -1, EACCES); break;
1115 		CASE_TEST(fork);              EXPECT_SYSZR(1, test_fork()); break;
1116 		CASE_TEST(getdents64_root);   EXPECT_SYSNE(1, test_getdents64("/"), -1); break;
1117 		CASE_TEST(getdents64_null);   EXPECT_SYSER(1, test_getdents64("/dev/null"), -1, ENOTDIR); break;
1118 		CASE_TEST(directories);       EXPECT_SYSZR(proc, test_dirent()); break;
1119 		CASE_TEST(gettimeofday_tv);   EXPECT_SYSZR(1, gettimeofday(&tv, NULL)); break;
1120 		CASE_TEST(gettimeofday_tv_tz);EXPECT_SYSZR(1, gettimeofday(&tv, &tz)); break;
1121 		CASE_TEST(getpagesize);       EXPECT_SYSZR(1, test_getpagesize()); break;
1122 		CASE_TEST(ioctl_tiocinq);     EXPECT_SYSZR(1, ioctl(0, TIOCINQ, &tmp)); break;
1123 		CASE_TEST(link_root1);        EXPECT_SYSER(1, link("/", "/"), -1, EEXIST); break;
1124 		CASE_TEST(link_blah);         EXPECT_SYSER(1, link("/proc/self/blah", "/blah"), -1, ENOENT); break;
1125 		CASE_TEST(link_dir);          EXPECT_SYSER(euid0, link("/", "/blah"), -1, EPERM); break;
1126 		CASE_TEST(link_cross);        EXPECT_SYSER(proc, link("/proc/self/cmdline", "/blah"), -1, EXDEV); break;
1127 		CASE_TEST(lseek_m1);          EXPECT_SYSER(1, lseek(-1, 0, SEEK_SET), -1, EBADF); break;
1128 		CASE_TEST(lseek_0);           EXPECT_SYSER(1, lseek(0, 0, SEEK_SET), -1, ESPIPE); break;
1129 		CASE_TEST(mkdir_root);        EXPECT_SYSER(1, mkdir("/", 0755), -1, EEXIST); break;
1130 		CASE_TEST(mmap_bad);          EXPECT_PTRER(1, mmap(NULL, 0, PROT_READ, MAP_PRIVATE, 0, 0), MAP_FAILED, EINVAL); break;
1131 		CASE_TEST(munmap_bad);        EXPECT_SYSER(1, munmap(NULL, 0), -1, EINVAL); break;
1132 		CASE_TEST(mmap_munmap_good);  EXPECT_SYSZR(1, test_mmap_munmap()); break;
1133 		CASE_TEST(open_tty);          EXPECT_SYSNE(1, tmp = open("/dev/null", O_RDONLY), -1); if (tmp != -1) close(tmp); break;
1134 		CASE_TEST(open_blah);         EXPECT_SYSER(1, tmp = open("/proc/self/blah", O_RDONLY), -1, ENOENT); if (tmp != -1) close(tmp); break;
1135 		CASE_TEST(openat_dir);        EXPECT_SYSZR(1, test_openat()); break;
1136 		CASE_TEST(pipe);              EXPECT_SYSZR(1, test_pipe()); break;
1137 		CASE_TEST(poll_null);         EXPECT_SYSZR(1, poll(NULL, 0, 0)); break;
1138 		CASE_TEST(poll_stdout);       EXPECT_SYSNE(1, ({ struct pollfd fds = { 1, POLLOUT, 0}; poll(&fds, 1, 0); }), -1); break;
1139 		CASE_TEST(poll_fault);        EXPECT_SYSER(1, poll(NULL, 1, 0), -1, EFAULT); break;
1140 		CASE_TEST(prctl);             EXPECT_SYSER(1, prctl(PR_SET_NAME, (unsigned long)NULL, 0, 0, 0), -1, EFAULT); break;
1141 		CASE_TEST(read_badf);         EXPECT_SYSER(1, read(-1, &tmp, 1), -1, EBADF); break;
1142 		CASE_TEST(rlimit);            EXPECT_SYSZR(1, test_rlimit()); break;
1143 		CASE_TEST(rmdir_blah);        EXPECT_SYSER(1, rmdir("/blah"), -1, ENOENT); break;
1144 		CASE_TEST(sched_yield);       EXPECT_SYSZR(1, sched_yield()); break;
1145 		CASE_TEST(select_null);       EXPECT_SYSZR(1, ({ struct timeval tv = { 0 }; select(0, NULL, NULL, NULL, &tv); })); break;
1146 		CASE_TEST(select_stdout);     EXPECT_SYSNE(1, ({ fd_set fds; FD_ZERO(&fds); FD_SET(1, &fds); select(2, NULL, &fds, NULL, NULL); }), -1); break;
1147 		CASE_TEST(select_fault);      EXPECT_SYSER(1, select(1, (void *)1, NULL, NULL, 0), -1, EFAULT); break;
1148 		CASE_TEST(stat_blah);         EXPECT_SYSER(1, stat("/proc/self/blah", &stat_buf), -1, ENOENT); break;
1149 		CASE_TEST(stat_fault);        EXPECT_SYSER(1, stat(NULL, &stat_buf), -1, EFAULT); break;
1150 		CASE_TEST(stat_timestamps);   EXPECT_SYSZR(1, test_stat_timestamps()); break;
1151 		CASE_TEST(symlink_root);      EXPECT_SYSER(1, symlink("/", "/"), -1, EEXIST); break;
1152 		CASE_TEST(uname);             EXPECT_SYSZR(proc, test_uname()); break;
1153 		CASE_TEST(uname_fault);       EXPECT_SYSER(1, uname(NULL), -1, EFAULT); break;
1154 		CASE_TEST(unlink_root);       EXPECT_SYSER(1, unlink("/"), -1, EISDIR); break;
1155 		CASE_TEST(unlink_blah);       EXPECT_SYSER(1, unlink("/proc/self/blah"), -1, ENOENT); break;
1156 		CASE_TEST(wait_child);        EXPECT_SYSER(1, wait(&tmp), -1, ECHILD); break;
1157 		CASE_TEST(waitpid_min);       EXPECT_SYSER(1, waitpid(INT_MIN, &tmp, WNOHANG), -1, ESRCH); break;
1158 		CASE_TEST(waitpid_child);     EXPECT_SYSER(1, waitpid(getpid(), &tmp, WNOHANG), -1, ECHILD); break;
1159 		CASE_TEST(write_badf);        EXPECT_SYSER(1, write(-1, &tmp, 1), -1, EBADF); break;
1160 		CASE_TEST(write_zero);        EXPECT_SYSZR(1, write(1, &tmp, 0)); break;
1161 		CASE_TEST(syscall_noargs);    EXPECT_SYSEQ(1, syscall(__NR_getpid), getpid()); break;
1162 		CASE_TEST(syscall_args);      EXPECT_SYSER(1, syscall(__NR_statx, 0, NULL, 0, 0, NULL), -1, EFAULT); break;
1163 		case __LINE__:
1164 			return ret; /* must be last */
1165 		/* note: do not set any defaults so as to permit holes above */
1166 		}
1167 	}
1168 	return ret;
1169 }
1170 
run_stdlib(int min,int max)1171 int run_stdlib(int min, int max)
1172 {
1173 	int test;
1174 	int ret = 0;
1175 
1176 	for (test = min; test >= 0 && test <= max; test++) {
1177 		int llen = 0; /* line length */
1178 
1179 		/* For functions that take a long buffer, like strlcat()
1180 		 * Add some more chars after the \0, to test functions that overwrite the buffer set
1181 		 * the \0 at the exact right position.
1182 		 */
1183 		char buf[10] = "test123456";
1184 		buf[4] = '\0';
1185 
1186 
1187 		/* avoid leaving empty lines below, this will insert holes into
1188 		 * test numbers.
1189 		 */
1190 		switch (test + __LINE__ + 1) {
1191 		CASE_TEST(getenv_TERM);        EXPECT_STRNZ(1, getenv("TERM")); break;
1192 		CASE_TEST(getenv_blah);        EXPECT_STRZR(1, getenv("blah")); break;
1193 		CASE_TEST(setcmp_blah_blah);   EXPECT_EQ(1, strcmp("blah", "blah"), 0); break;
1194 		CASE_TEST(setcmp_blah_blah2);  EXPECT_NE(1, strcmp("blah", "blah2"), 0); break;
1195 		CASE_TEST(setncmp_blah_blah);  EXPECT_EQ(1, strncmp("blah", "blah", 10), 0); break;
1196 		CASE_TEST(setncmp_blah_blah4); EXPECT_EQ(1, strncmp("blah", "blah4", 4), 0); break;
1197 		CASE_TEST(setncmp_blah_blah5); EXPECT_NE(1, strncmp("blah", "blah5", 5), 0); break;
1198 		CASE_TEST(setncmp_blah_blah6); EXPECT_NE(1, strncmp("blah", "blah6", 6), 0); break;
1199 		CASE_TEST(strchr_foobar_o);    EXPECT_STREQ(1, strchr("foobar", 'o'), "oobar"); break;
1200 		CASE_TEST(strchr_foobar_z);    EXPECT_STRZR(1, strchr("foobar", 'z')); break;
1201 		CASE_TEST(strrchr_foobar_o);   EXPECT_STREQ(1, strrchr("foobar", 'o'), "obar"); break;
1202 		CASE_TEST(strrchr_foobar_z);   EXPECT_STRZR(1, strrchr("foobar", 'z')); break;
1203 		CASE_TEST(strlcat_0);          EXPECT_STRBUFEQ(is_nolibc, strlcat(buf, "bar", 0), buf, 3, "test"); break;
1204 		CASE_TEST(strlcat_1);          EXPECT_STRBUFEQ(is_nolibc, strlcat(buf, "bar", 1), buf, 4, "test"); break;
1205 		CASE_TEST(strlcat_5);          EXPECT_STRBUFEQ(is_nolibc, strlcat(buf, "bar", 5), buf, 7, "test"); break;
1206 		CASE_TEST(strlcat_6);          EXPECT_STRBUFEQ(is_nolibc, strlcat(buf, "bar", 6), buf, 7, "testb"); break;
1207 		CASE_TEST(strlcat_7);          EXPECT_STRBUFEQ(is_nolibc, strlcat(buf, "bar", 7), buf, 7, "testba"); break;
1208 		CASE_TEST(strlcat_8);          EXPECT_STRBUFEQ(is_nolibc, strlcat(buf, "bar", 8), buf, 7, "testbar"); break;
1209 		CASE_TEST(strlcpy_0);          EXPECT_STRBUFEQ(is_nolibc, strlcpy(buf, "bar", 0), buf, 3, "test"); break;
1210 		CASE_TEST(strlcpy_1);          EXPECT_STRBUFEQ(is_nolibc, strlcpy(buf, "bar", 1), buf, 3, ""); break;
1211 		CASE_TEST(strlcpy_2);          EXPECT_STRBUFEQ(is_nolibc, strlcpy(buf, "bar", 2), buf, 3, "b"); break;
1212 		CASE_TEST(strlcpy_3);          EXPECT_STRBUFEQ(is_nolibc, strlcpy(buf, "bar", 3), buf, 3, "ba"); break;
1213 		CASE_TEST(strlcpy_4);          EXPECT_STRBUFEQ(is_nolibc, strlcpy(buf, "bar", 4), buf, 3, "bar"); break;
1214 		CASE_TEST(memcmp_20_20);       EXPECT_EQ(1, memcmp("aaa\x20", "aaa\x20", 4), 0); break;
1215 		CASE_TEST(memcmp_20_60);       EXPECT_LT(1, memcmp("aaa\x20", "aaa\x60", 4), 0); break;
1216 		CASE_TEST(memcmp_60_20);       EXPECT_GT(1, memcmp("aaa\x60", "aaa\x20", 4), 0); break;
1217 		CASE_TEST(memcmp_20_e0);       EXPECT_LT(1, memcmp("aaa\x20", "aaa\xe0", 4), 0); break;
1218 		CASE_TEST(memcmp_e0_20);       EXPECT_GT(1, memcmp("aaa\xe0", "aaa\x20", 4), 0); break;
1219 		CASE_TEST(memcmp_80_e0);       EXPECT_LT(1, memcmp("aaa\x80", "aaa\xe0", 4), 0); break;
1220 		CASE_TEST(memcmp_e0_80);       EXPECT_GT(1, memcmp("aaa\xe0", "aaa\x80", 4), 0); break;
1221 		CASE_TEST(limit_int8_max);          EXPECT_EQ(1, INT8_MAX,         (int8_t)          0x7f); break;
1222 		CASE_TEST(limit_int8_min);          EXPECT_EQ(1, INT8_MIN,         (int8_t)          0x80); break;
1223 		CASE_TEST(limit_uint8_max);         EXPECT_EQ(1, UINT8_MAX,        (uint8_t)         0xff); break;
1224 		CASE_TEST(limit_int16_max);         EXPECT_EQ(1, INT16_MAX,        (int16_t)         0x7fff); break;
1225 		CASE_TEST(limit_int16_min);         EXPECT_EQ(1, INT16_MIN,        (int16_t)         0x8000); break;
1226 		CASE_TEST(limit_uint16_max);        EXPECT_EQ(1, UINT16_MAX,       (uint16_t)        0xffff); break;
1227 		CASE_TEST(limit_int32_max);         EXPECT_EQ(1, INT32_MAX,        (int32_t)         0x7fffffff); break;
1228 		CASE_TEST(limit_int32_min);         EXPECT_EQ(1, INT32_MIN,        (int32_t)         0x80000000); break;
1229 		CASE_TEST(limit_uint32_max);        EXPECT_EQ(1, UINT32_MAX,       (uint32_t)        0xffffffff); break;
1230 		CASE_TEST(limit_int64_max);         EXPECT_EQ(1, INT64_MAX,        (int64_t)         0x7fffffffffffffff); break;
1231 		CASE_TEST(limit_int64_min);         EXPECT_EQ(1, INT64_MIN,        (int64_t)         0x8000000000000000); break;
1232 		CASE_TEST(limit_uint64_max);        EXPECT_EQ(1, UINT64_MAX,       (uint64_t)        0xffffffffffffffff); break;
1233 		CASE_TEST(limit_int_least8_max);    EXPECT_EQ(1, INT_LEAST8_MAX,   (int_least8_t)    0x7f); break;
1234 		CASE_TEST(limit_int_least8_min);    EXPECT_EQ(1, INT_LEAST8_MIN,   (int_least8_t)    0x80); break;
1235 		CASE_TEST(limit_uint_least8_max);   EXPECT_EQ(1, UINT_LEAST8_MAX,  (uint_least8_t)   0xff); break;
1236 		CASE_TEST(limit_int_least16_max);   EXPECT_EQ(1, INT_LEAST16_MAX,  (int_least16_t)   0x7fff); break;
1237 		CASE_TEST(limit_int_least16_min);   EXPECT_EQ(1, INT_LEAST16_MIN,  (int_least16_t)   0x8000); break;
1238 		CASE_TEST(limit_uint_least16_max);  EXPECT_EQ(1, UINT_LEAST16_MAX, (uint_least16_t)  0xffff); break;
1239 		CASE_TEST(limit_int_least32_max);   EXPECT_EQ(1, INT_LEAST32_MAX,  (int_least32_t)   0x7fffffff); break;
1240 		CASE_TEST(limit_int_least32_min);   EXPECT_EQ(1, INT_LEAST32_MIN,  (int_least32_t)   0x80000000); break;
1241 		CASE_TEST(limit_uint_least32_max);  EXPECT_EQ(1, UINT_LEAST32_MAX, (uint_least32_t)  0xffffffffU); break;
1242 		CASE_TEST(limit_int_least64_min);   EXPECT_EQ(1, INT_LEAST64_MIN,  (int_least64_t)   0x8000000000000000LL); break;
1243 		CASE_TEST(limit_int_least64_max);   EXPECT_EQ(1, INT_LEAST64_MAX,  (int_least64_t)   0x7fffffffffffffffLL); break;
1244 		CASE_TEST(limit_uint_least64_max);  EXPECT_EQ(1, UINT_LEAST64_MAX, (uint_least64_t)  0xffffffffffffffffULL); break;
1245 		CASE_TEST(limit_int_fast8_max);     EXPECT_EQ(1, INT_FAST8_MAX,    (int_fast8_t)     0x7f); break;
1246 		CASE_TEST(limit_int_fast8_min);     EXPECT_EQ(1, INT_FAST8_MIN,    (int_fast8_t)     0x80); break;
1247 		CASE_TEST(limit_uint_fast8_max);    EXPECT_EQ(1, UINT_FAST8_MAX,   (uint_fast8_t)    0xff); break;
1248 		CASE_TEST(limit_int_fast16_min);    EXPECT_EQ(1, INT_FAST16_MIN,   (int_fast16_t)    SINT_MIN_OF_TYPE(int_fast16_t)); break;
1249 		CASE_TEST(limit_int_fast16_max);    EXPECT_EQ(1, INT_FAST16_MAX,   (int_fast16_t)    SINT_MAX_OF_TYPE(int_fast16_t)); break;
1250 		CASE_TEST(limit_uint_fast16_max);   EXPECT_EQ(1, UINT_FAST16_MAX,  (uint_fast16_t)   UINTPTR_MAX); break;
1251 		CASE_TEST(limit_int_fast32_min);    EXPECT_EQ(1, INT_FAST32_MIN,   (int_fast32_t)    SINT_MIN_OF_TYPE(int_fast32_t)); break;
1252 		CASE_TEST(limit_int_fast32_max);    EXPECT_EQ(1, INT_FAST32_MAX,   (int_fast32_t)    SINT_MAX_OF_TYPE(int_fast32_t)); break;
1253 		CASE_TEST(limit_uint_fast32_max);   EXPECT_EQ(1, UINT_FAST32_MAX,  (uint_fast32_t)   UINTPTR_MAX); break;
1254 		CASE_TEST(limit_int_fast64_min);    EXPECT_EQ(1, INT_FAST64_MIN,   (int_fast64_t)    INT64_MIN); break;
1255 		CASE_TEST(limit_int_fast64_max);    EXPECT_EQ(1, INT_FAST64_MAX,   (int_fast64_t)    INT64_MAX); break;
1256 		CASE_TEST(limit_uint_fast64_max);   EXPECT_EQ(1, UINT_FAST64_MAX,  (uint_fast64_t)   UINT64_MAX); break;
1257 		CASE_TEST(sizeof_long_sane);        EXPECT_EQ(1, sizeof(long) == 8 || sizeof(long) == 4, 1); break;
1258 		CASE_TEST(limit_intptr_min);        EXPECT_EQ(1, INTPTR_MIN,  sizeof(long) == 8 ? (intptr_t)  0x8000000000000000LL  : (intptr_t)  0x80000000); break;
1259 		CASE_TEST(limit_intptr_max);        EXPECT_EQ(1, INTPTR_MAX,  sizeof(long) == 8 ? (intptr_t)  0x7fffffffffffffffLL  : (intptr_t)  0x7fffffff); break;
1260 		CASE_TEST(limit_uintptr_max);       EXPECT_EQ(1, UINTPTR_MAX, sizeof(long) == 8 ? (uintptr_t) 0xffffffffffffffffULL : (uintptr_t) 0xffffffffU); break;
1261 		CASE_TEST(limit_ptrdiff_min);       EXPECT_EQ(1, PTRDIFF_MIN, sizeof(long) == 8 ? (ptrdiff_t) 0x8000000000000000LL  : (ptrdiff_t) 0x80000000); break;
1262 		CASE_TEST(limit_ptrdiff_max);       EXPECT_EQ(1, PTRDIFF_MAX, sizeof(long) == 8 ? (ptrdiff_t) 0x7fffffffffffffffLL  : (ptrdiff_t) 0x7fffffff); break;
1263 		CASE_TEST(limit_size_max);          EXPECT_EQ(1, SIZE_MAX,    sizeof(long) == 8 ? (size_t)    0xffffffffffffffffULL : (size_t)    0xffffffffU); break;
1264 		CASE_TEST(strtol_simple);           EXPECT_STRTOX(1, strtol, "35", 10, 35, -1, 0); break;
1265 		CASE_TEST(strtol_positive);         EXPECT_STRTOX(1, strtol, "+35", 10, 35, -1, 0); break;
1266 		CASE_TEST(strtol_negative);         EXPECT_STRTOX(1, strtol, "-35", 10, -35, -1, 0); break;
1267 		CASE_TEST(strtol_hex_auto);         EXPECT_STRTOX(1, strtol, "0xFF", 0, 255, -1, 0); break;
1268 		CASE_TEST(strtol_base36);           EXPECT_STRTOX(1, strtol, "12yZ", 36, 50507, -1, 0); break;
1269 		CASE_TEST(strtol_cutoff);           EXPECT_STRTOX(1, strtol, "1234567890", 8, 342391, 7, 0); break;
1270 		CASE_TEST(strtol_octal_auto);       EXPECT_STRTOX(1, strtol, "011", 0, 9, -1, 0); break;
1271 		CASE_TEST(strtol_hex_00);           EXPECT_STRTOX(1, strtol, "0x00", 16, 0, -1, 0); break;
1272 		CASE_TEST(strtol_hex_FF);           EXPECT_STRTOX(1, strtol, "FF", 16, 255, -1, 0); break;
1273 		CASE_TEST(strtol_hex_ff);           EXPECT_STRTOX(1, strtol, "ff", 16, 255, -1, 0); break;
1274 		CASE_TEST(strtol_hex_prefix);       EXPECT_STRTOX(1, strtol, "0xFF", 16, 255, -1, 0); break;
1275 		CASE_TEST(strtol_trailer);          EXPECT_STRTOX(1, strtol, "35foo", 10, 35, 2, 0); break;
1276 		CASE_TEST(strtol_overflow);         EXPECT_STRTOX(1, strtol, "0x8000000000000000", 16, LONG_MAX, -1, ERANGE); break;
1277 		CASE_TEST(strtol_underflow);        EXPECT_STRTOX(1, strtol, "-0x8000000000000001", 16, LONG_MIN, -1, ERANGE); break;
1278 		CASE_TEST(strtoul_negative);        EXPECT_STRTOX(1, strtoul, "-0x1", 16, ULONG_MAX, 4, 0); break;
1279 		CASE_TEST(strtoul_overflow);        EXPECT_STRTOX(1, strtoul, "0x10000000000000000", 16, ULONG_MAX, -1, ERANGE); break;
1280 		CASE_TEST(strerror_success);        EXPECT_STREQ(is_nolibc, strerror(0), "errno=0"); break;
1281 		CASE_TEST(strerror_EINVAL);         EXPECT_STREQ(is_nolibc, strerror(EINVAL), "errno=22"); break;
1282 		CASE_TEST(strerror_int_max);        EXPECT_STREQ(is_nolibc, strerror(INT_MAX), "errno=2147483647"); break;
1283 		CASE_TEST(strerror_int_min);        EXPECT_STREQ(is_nolibc, strerror(INT_MIN), "errno=-2147483648"); break;
1284 
1285 		case __LINE__:
1286 			return ret; /* must be last */
1287 		/* note: do not set any defaults so as to permit holes above */
1288 		}
1289 	}
1290 	return ret;
1291 }
1292 
1293 #define EXPECT_VFPRINTF(c, expected, fmt, ...)				\
1294 	ret += expect_vfprintf(llen, c, expected, fmt, ##__VA_ARGS__)
1295 
expect_vfprintf(int llen,int c,const char * expected,const char * fmt,...)1296 static int expect_vfprintf(int llen, int c, const char *expected, const char *fmt, ...)
1297 {
1298 	int ret, pipefd[2];
1299 	ssize_t w, r;
1300 	char buf[100];
1301 	FILE *memfile;
1302 	va_list args;
1303 
1304 	ret = pipe(pipefd);
1305 	if (ret == -1) {
1306 		llen += printf(" pipe() != %s", strerror(errno));
1307 		result(llen, FAIL);
1308 		return 1;
1309 	}
1310 
1311 	memfile = fdopen(pipefd[1], "w");
1312 	if (!memfile) {
1313 		result(llen, FAIL);
1314 		return 1;
1315 	}
1316 
1317 	va_start(args, fmt);
1318 	w = vfprintf(memfile, fmt, args);
1319 	va_end(args);
1320 
1321 	if (w != c) {
1322 		llen += printf(" written(%d) != %d", (int)w, c);
1323 		result(llen, FAIL);
1324 		return 1;
1325 	}
1326 
1327 	fclose(memfile);
1328 
1329 	r = read(pipefd[0], buf, sizeof(buf) - 1);
1330 
1331 	if (r != w) {
1332 		llen += printf(" written(%d) != read(%d)", (int)w, (int)r);
1333 		result(llen, FAIL);
1334 		return 1;
1335 	}
1336 
1337 	buf[r] = '\0';
1338 	llen += printf(" \"%s\" = \"%s\"", expected, buf);
1339 	ret = strncmp(expected, buf, c);
1340 
1341 	result(llen, ret ? FAIL : OK);
1342 	return ret;
1343 }
1344 
test_scanf(void)1345 static int test_scanf(void)
1346 {
1347 	unsigned long long ull;
1348 	unsigned long ul;
1349 	unsigned int u;
1350 	long long ll;
1351 	long l;
1352 	void *p;
1353 	int i;
1354 
1355 	/* return __LINE__ to point to the specific failure */
1356 
1357 	/* test EOF */
1358 	if (sscanf("", "foo") != EOF)
1359 		return __LINE__;
1360 
1361 	/* test simple literal without placeholder */
1362 	if (sscanf("foo", "foo") != 0)
1363 		return __LINE__;
1364 
1365 	/* test single placeholder */
1366 	if (sscanf("123", "%d", &i) != 1)
1367 		return __LINE__;
1368 
1369 	if (i != 123)
1370 		return __LINE__;
1371 
1372 	/* test multiple place holders and separators */
1373 	if (sscanf("a123b456c0x90", "a%db%uc%p", &i, &u, &p) != 3)
1374 		return __LINE__;
1375 
1376 	if (i != 123)
1377 		return __LINE__;
1378 
1379 	if (u != 456)
1380 		return __LINE__;
1381 
1382 	if (p != (void *)0x90)
1383 		return __LINE__;
1384 
1385 	/* test space handling */
1386 	if (sscanf("a    b1", "a b%d", &i) != 1)
1387 		return __LINE__;
1388 
1389 	if (i != 1)
1390 		return __LINE__;
1391 
1392 	/* test literal percent */
1393 	if (sscanf("a%1", "a%%%d", &i) != 1)
1394 		return __LINE__;
1395 
1396 	if (i != 1)
1397 		return __LINE__;
1398 
1399 	/* test stdint.h types */
1400 	if (sscanf("1|2|3|4|5|6",
1401 		   "%d|%ld|%lld|%u|%lu|%llu",
1402 		   &i, &l, &ll, &u, &ul, &ull) != 6)
1403 		return __LINE__;
1404 
1405 	if (i != 1 || l != 2 || ll != 3 ||
1406 	    u != 4 || ul != 5 || ull != 6)
1407 		return __LINE__;
1408 
1409 	return 0;
1410 }
1411 
run_vfprintf(int min,int max)1412 static int run_vfprintf(int min, int max)
1413 {
1414 	int test;
1415 	int ret = 0;
1416 
1417 	for (test = min; test >= 0 && test <= max; test++) {
1418 		int llen = 0; /* line length */
1419 
1420 		/* avoid leaving empty lines below, this will insert holes into
1421 		 * test numbers.
1422 		 */
1423 		switch (test + __LINE__ + 1) {
1424 		CASE_TEST(empty);        EXPECT_VFPRINTF(0, "", ""); break;
1425 		CASE_TEST(simple);       EXPECT_VFPRINTF(3, "foo", "foo"); break;
1426 		CASE_TEST(string);       EXPECT_VFPRINTF(3, "foo", "%s", "foo"); break;
1427 		CASE_TEST(number);       EXPECT_VFPRINTF(4, "1234", "%d", 1234); break;
1428 		CASE_TEST(negnumber);    EXPECT_VFPRINTF(5, "-1234", "%d", -1234); break;
1429 		CASE_TEST(unsigned);     EXPECT_VFPRINTF(5, "12345", "%u", 12345); break;
1430 		CASE_TEST(char);         EXPECT_VFPRINTF(1, "c", "%c", 'c'); break;
1431 		CASE_TEST(hex);          EXPECT_VFPRINTF(1, "f", "%x", 0xf); break;
1432 		CASE_TEST(pointer);      EXPECT_VFPRINTF(3, "0x1", "%p", (void *) 0x1); break;
1433 		CASE_TEST(scanf);        EXPECT_ZR(1, test_scanf()); break;
1434 		case __LINE__:
1435 			return ret; /* must be last */
1436 		/* note: do not set any defaults so as to permit holes above */
1437 		}
1438 	}
1439 	return ret;
1440 }
1441 
smash_stack(void)1442 static int smash_stack(void)
1443 {
1444 	char buf[100];
1445 	volatile char *ptr = buf;
1446 	size_t i;
1447 
1448 	for (i = 0; i < 200; i++)
1449 		ptr[i] = 'P';
1450 
1451 	return 1;
1452 }
1453 
run_protection(int min,int max)1454 static int run_protection(int min __attribute__((unused)),
1455 			  int max __attribute__((unused)))
1456 {
1457 	pid_t pid;
1458 	int llen = 0, ret;
1459 	siginfo_t siginfo = {};
1460 	struct rlimit rlimit = { 0, 0 };
1461 
1462 	llen += printf("0 -fstackprotector ");
1463 
1464 #if !defined(_NOLIBC_STACKPROTECTOR)
1465 	llen += printf("not supported");
1466 	result(llen, SKIPPED);
1467 	return 0;
1468 #endif
1469 
1470 #if defined(_NOLIBC_STACKPROTECTOR)
1471 	if (!__stack_chk_guard) {
1472 		llen += printf("__stack_chk_guard not initialized");
1473 		result(llen, FAIL);
1474 		return 1;
1475 	}
1476 #endif
1477 
1478 	pid = -1;
1479 	pid = fork();
1480 
1481 	switch (pid) {
1482 	case -1:
1483 		llen += printf("fork()");
1484 		result(llen, FAIL);
1485 		return 1;
1486 
1487 	case 0:
1488 		close(STDOUT_FILENO);
1489 		close(STDERR_FILENO);
1490 
1491 		prctl(PR_SET_DUMPABLE, 0, 0, 0, 0);
1492 		setrlimit(RLIMIT_CORE, &rlimit);
1493 		smash_stack();
1494 		return 1;
1495 
1496 	default:
1497 		ret = waitid(P_PID, pid, &siginfo, WEXITED);
1498 
1499 		if (ret != 0 || siginfo.si_signo != SIGCHLD ||
1500 		    siginfo.si_code != CLD_KILLED || siginfo.si_status != SIGABRT) {
1501 			llen += printf("waitid()");
1502 			result(llen, FAIL);
1503 			return 1;
1504 		}
1505 		result(llen, OK);
1506 		return 0;
1507 	}
1508 }
1509 
1510 /* prepare what needs to be prepared for pid 1 (stdio, /dev, /proc, etc) */
prepare(void)1511 int prepare(void)
1512 {
1513 	struct stat stat_buf;
1514 
1515 	/* It's possible that /dev doesn't even exist or was not mounted, so
1516 	 * we'll try to create it, mount it, or create minimal entries into it.
1517 	 * We want at least /dev/null and /dev/console.
1518 	 */
1519 	if (stat("/dev/.", &stat_buf) == 0 || mkdir("/dev", 0755) == 0) {
1520 		if (stat("/dev/console", &stat_buf) != 0 ||
1521 		    stat("/dev/null", &stat_buf) != 0 ||
1522 		    stat("/dev/zero", &stat_buf) != 0) {
1523 			/* try devtmpfs first, otherwise fall back to manual creation */
1524 			if (mount("/dev", "/dev", "devtmpfs", 0, 0) != 0) {
1525 				mknod("/dev/console", 0600 | S_IFCHR, makedev(5, 1));
1526 				mknod("/dev/null",    0666 | S_IFCHR, makedev(1, 3));
1527 				mknod("/dev/zero",    0666 | S_IFCHR, makedev(1, 5));
1528 			}
1529 		}
1530 	}
1531 
1532 	/* If no /dev/console was found before calling init, stdio is closed so
1533 	 * we need to reopen it from /dev/console. If it failed above, it will
1534 	 * still fail here and we cannot emit a message anyway.
1535 	 */
1536 	if (close(dup(1)) == -1) {
1537 		int fd = open("/dev/console", O_RDWR);
1538 
1539 		if (fd >= 0) {
1540 			if (fd != 0)
1541 				dup2(fd, 0);
1542 			if (fd != 1)
1543 				dup2(fd, 1);
1544 			if (fd != 2)
1545 				dup2(fd, 2);
1546 			if (fd > 2)
1547 				close(fd);
1548 			puts("\nSuccessfully reopened /dev/console.");
1549 		}
1550 	}
1551 
1552 	/* try to mount /proc if not mounted. Silently fail otherwise */
1553 	if (stat("/proc/.", &stat_buf) == 0 || mkdir("/proc", 0755) == 0) {
1554 		if (stat("/proc/self", &stat_buf) != 0) {
1555 			/* If not mountable, remove /proc completely to avoid misuse */
1556 			if (mount("none", "/proc", "proc", 0, 0) != 0)
1557 				rmdir("/proc");
1558 		}
1559 	}
1560 
1561 	/* some tests rely on a writable /tmp */
1562 	mkdir("/tmp", 0755);
1563 
1564 	return 0;
1565 }
1566 
1567 /* This is the definition of known test names, with their functions */
1568 static const struct test test_names[] = {
1569 	/* add new tests here */
1570 	{ .name = "startup",    .func = run_startup    },
1571 	{ .name = "syscall",    .func = run_syscall    },
1572 	{ .name = "stdlib",     .func = run_stdlib     },
1573 	{ .name = "vfprintf",   .func = run_vfprintf   },
1574 	{ .name = "protection", .func = run_protection },
1575 	{ 0 }
1576 };
1577 
is_setting_valid(char * test)1578 static int is_setting_valid(char *test)
1579 {
1580 	int idx, len, test_len, valid = 0;
1581 	char delimiter;
1582 
1583 	if (!test)
1584 		return valid;
1585 
1586 	test_len = strlen(test);
1587 
1588 	for (idx = 0; test_names[idx].name; idx++) {
1589 		len = strlen(test_names[idx].name);
1590 		if (test_len < len)
1591 			continue;
1592 
1593 		if (strncmp(test, test_names[idx].name, len) != 0)
1594 			continue;
1595 
1596 		delimiter = test[len];
1597 		if (delimiter != ':' && delimiter != ',' && delimiter != '\0')
1598 			continue;
1599 
1600 		valid = 1;
1601 		break;
1602 	}
1603 
1604 	return valid;
1605 }
1606 
main(int argc,char ** argv,char ** envp)1607 int main(int argc, char **argv, char **envp)
1608 {
1609 	int min = 0;
1610 	int max = INT_MAX;
1611 	int ret = 0;
1612 	int err;
1613 	int idx;
1614 	char *test;
1615 
1616 	argv0 = argv[0];
1617 	test_argc = argc;
1618 	test_argv = argv;
1619 	test_envp = envp;
1620 
1621 	/* when called as init, it's possible that no console was opened, for
1622 	 * example if no /dev file system was provided. We'll check that fd#1
1623 	 * was opened, and if not we'll attempt to create and open /dev/console
1624 	 * and /dev/null that we'll use for later tests.
1625 	 */
1626 	if (getpid() == 1)
1627 		prepare();
1628 
1629 	/* the definition of a series of tests comes from either argv[1] or the
1630 	 * "NOLIBC_TEST" environment variable. It's made of a comma-delimited
1631 	 * series of test names and optional ranges:
1632 	 *    syscall:5-15[:.*],stdlib:8-10
1633 	 */
1634 	test = argv[1];
1635 	if (!is_setting_valid(test))
1636 		test = getenv("NOLIBC_TEST");
1637 
1638 	if (is_setting_valid(test)) {
1639 		char *comma, *colon, *dash, *value;
1640 
1641 		do {
1642 			comma = strchr(test, ',');
1643 			if (comma)
1644 				*(comma++) = '\0';
1645 
1646 			colon = strchr(test, ':');
1647 			if (colon)
1648 				*(colon++) = '\0';
1649 
1650 			for (idx = 0; test_names[idx].name; idx++) {
1651 				if (strcmp(test, test_names[idx].name) == 0)
1652 					break;
1653 			}
1654 
1655 			if (test_names[idx].name) {
1656 				/* The test was named, it will be called at least
1657 				 * once. We may have an optional range at <colon>
1658 				 * here, which defaults to the full range.
1659 				 */
1660 				do {
1661 					min = 0; max = INT_MAX;
1662 					value = colon;
1663 					if (value && *value) {
1664 						colon = strchr(value, ':');
1665 						if (colon)
1666 							*(colon++) = '\0';
1667 
1668 						dash = strchr(value, '-');
1669 						if (dash)
1670 							*(dash++) = '\0';
1671 
1672 						/* support :val: :min-max: :min-: :-max: */
1673 						if (*value)
1674 							min = atoi(value);
1675 						if (!dash)
1676 							max = min;
1677 						else if (*dash)
1678 							max = atoi(dash);
1679 
1680 						value = colon;
1681 					}
1682 
1683 					/* now's time to call the test */
1684 					printf("Running test '%s'\n", test_names[idx].name);
1685 					err = test_names[idx].func(min, max);
1686 					ret += err;
1687 					printf("Errors during this test: %d\n\n", err);
1688 				} while (colon && *colon);
1689 			} else
1690 				printf("Ignoring unknown test name '%s'\n", test);
1691 
1692 			test = comma;
1693 		} while (test && *test);
1694 	} else {
1695 		/* no test mentioned, run everything */
1696 		for (idx = 0; test_names[idx].name; idx++) {
1697 			printf("Running test '%s'\n", test_names[idx].name);
1698 			err = test_names[idx].func(min, max);
1699 			ret += err;
1700 			printf("Errors during this test: %d\n\n", err);
1701 		}
1702 	}
1703 
1704 	printf("Total number of errors: %d\n", ret);
1705 
1706 	if (getpid() == 1) {
1707 		/* we're running as init, there's no other process on the
1708 		 * system, thus likely started from a VM for a quick check.
1709 		 * Exiting will provoke a kernel panic that may be reported
1710 		 * as an error by Qemu or the hypervisor, while stopping
1711 		 * cleanly will often be reported as a success. This allows
1712 		 * to use the output of this program for bisecting kernels.
1713 		 */
1714 		printf("Leaving init with final status: %d\n", !!ret);
1715 		if (ret == 0)
1716 			reboot(RB_POWER_OFF);
1717 #if defined(__x86_64__)
1718 		/* QEMU started with "-device isa-debug-exit -no-reboot" will
1719 		 * exit with status code 2N+1 when N is written to 0x501. We
1720 		 * hard-code the syscall here as it's arch-dependent.
1721 		 */
1722 		else if (syscall(__NR_ioperm, 0x501, 1, 1) == 0)
1723 			__asm__ volatile ("outb %%al, %%dx" :: "d"(0x501), "a"(0));
1724 		/* if it does nothing, fall back to the regular panic */
1725 #endif
1726 	}
1727 
1728 	printf("Exiting with status %d\n", !!ret);
1729 	return !!ret;
1730 }
1731