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