tools/testing/selftests/nolibc/nolibc-test.c

   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 */
  52 static char **test_envp;
  53
  54 /* will be used to test initialization of argv */
  55 static char **test_argv;
  56
  57 /* will be used to test initialization of argc */
  58 static int test_argc;
  59
  60 /* will be used by some test cases as readable file, please don't write it */
  61 static const char *argv0;
  62
  63 /* will be used by constructor tests */
  64 static int constructor_test_value;
  65
  66 /* definition of a series of tests */
  67 struct test {
  68         const char *name;              /* test name */
  69         int (*func)(int min, int max); /* handler */
  70 };
  71
  72 #ifndef _NOLIBC_STDLIB_H
  73 char *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  */
  89 static 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
 134 static 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
 149 enum RESULT {
 150         OK,
 151         FAIL,
 152         SKIPPED,
 153 };
 154
 155 static 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
 179 static __attribute__((unused))
 180 int 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
 193 static __attribute__((unused))
 194 int 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
 207 static __attribute__((unused))
 208 int 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
 221 static __attribute__((unused))
 222 int 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
 235 static __attribute__((unused))
 236 int 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
 249 static __attribute__((unused))
 250 int 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
 263 static __attribute__((unused))
 264 int 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
 277 static __attribute__((unused))
 278 int 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
 291 static __attribute__((unused))
 292 int 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
 311 static __attribute__((unused))
 312 int 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
 331 static __attribute__((unused))
 332 int 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
 354 static __attribute__((unused))
 355 int 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
 378 static __attribute__((unused))
 379 int 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
 397 static __attribute__((unused))
 398 int 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
 415 static __attribute__((unused))
 416 int 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
 433 static __attribute__((unused))
 434 int 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
 451 static __attribute__((unused))
 452 int 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
 464 static __attribute__((unused))
 465 int 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
 478 static __attribute__((unused))
 479 int 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
 492 static __attribute__((unused))
 493 int 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
 508 static __attribute__((unused))
 509 int 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
 531 static __attribute__((unused))
 532 int 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
 550 static __attribute__((unused))
 551 int 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
 569 static __attribute__((unused))
 570 int 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
 588 static __attribute__((unused))
 589 int 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))
 610 static void constructor1(void)
 611 {
 612         constructor_test_value = 1;
 613 }
 614
 615 __attribute__((constructor))
 616 static void constructor2(void)
 617 {
 618         constructor_test_value *= 2;
 619 }
 620
 621 int 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 */
 670 int 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
 688 int 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
 717 int 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
 742 int 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
 764 int 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
 817 end:
 818         close(fd);
 819         return !!ret;
 820 }
 821
 822 int 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
 843 int 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  */
 874 int 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
 986 int 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
1072 static 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
1123 static 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
1152 static 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
1164 static 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) */
1219 int 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 */
1276 static 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
1286 static 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
1315 int 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 }