nolibc-test.c 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786
  1. // SPDX-License-Identifier: GPL-2.0
  2. #define _GNU_SOURCE
  3. /* platform-specific include files coming from the compiler */
  4. #include <limits.h>
  5. /* libc-specific include files
  6. * The program may be built in 3 ways:
  7. * $(CC) -nostdlib -include /path/to/nolibc.h => NOLIBC already defined
  8. * $(CC) -nostdlib -I/path/to/nolibc/sysroot => _NOLIBC_* guards are present
  9. * $(CC) with default libc => NOLIBC* never defined
  10. */
  11. #ifndef NOLIBC
  12. #include <stdio.h>
  13. #include <stdlib.h>
  14. #include <string.h>
  15. #ifndef _NOLIBC_STDIO_H
  16. /* standard libcs need more includes */
  17. #include <linux/reboot.h>
  18. #include <sys/io.h>
  19. #include <sys/ioctl.h>
  20. #include <sys/mount.h>
  21. #include <sys/reboot.h>
  22. #include <sys/stat.h>
  23. #include <sys/syscall.h>
  24. #include <sys/sysmacros.h>
  25. #include <sys/time.h>
  26. #include <sys/wait.h>
  27. #include <dirent.h>
  28. #include <errno.h>
  29. #include <fcntl.h>
  30. #include <poll.h>
  31. #include <sched.h>
  32. #include <signal.h>
  33. #include <stdarg.h>
  34. #include <unistd.h>
  35. #endif
  36. #endif
  37. /* will be used by nolibc by getenv() */
  38. char **environ;
  39. /* definition of a series of tests */
  40. struct test {
  41. const char *name; // test name
  42. int (*func)(int min, int max); // handler
  43. };
  44. #ifndef _NOLIBC_STDLIB_H
  45. char *itoa(int i)
  46. {
  47. static char buf[12];
  48. int ret;
  49. ret = snprintf(buf, sizeof(buf), "%d", i);
  50. return (ret >= 0 && ret < sizeof(buf)) ? buf : "#err";
  51. }
  52. #endif
  53. #define CASE_ERR(err) \
  54. case err: return #err
  55. /* returns the error name (e.g. "ENOENT") for common errors, "SUCCESS" for 0,
  56. * or the decimal value for less common ones.
  57. */
  58. const char *errorname(int err)
  59. {
  60. switch (err) {
  61. case 0: return "SUCCESS";
  62. CASE_ERR(EPERM);
  63. CASE_ERR(ENOENT);
  64. CASE_ERR(ESRCH);
  65. CASE_ERR(EINTR);
  66. CASE_ERR(EIO);
  67. CASE_ERR(ENXIO);
  68. CASE_ERR(E2BIG);
  69. CASE_ERR(ENOEXEC);
  70. CASE_ERR(EBADF);
  71. CASE_ERR(ECHILD);
  72. CASE_ERR(EAGAIN);
  73. CASE_ERR(ENOMEM);
  74. CASE_ERR(EACCES);
  75. CASE_ERR(EFAULT);
  76. CASE_ERR(ENOTBLK);
  77. CASE_ERR(EBUSY);
  78. CASE_ERR(EEXIST);
  79. CASE_ERR(EXDEV);
  80. CASE_ERR(ENODEV);
  81. CASE_ERR(ENOTDIR);
  82. CASE_ERR(EISDIR);
  83. CASE_ERR(EINVAL);
  84. CASE_ERR(ENFILE);
  85. CASE_ERR(EMFILE);
  86. CASE_ERR(ENOTTY);
  87. CASE_ERR(ETXTBSY);
  88. CASE_ERR(EFBIG);
  89. CASE_ERR(ENOSPC);
  90. CASE_ERR(ESPIPE);
  91. CASE_ERR(EROFS);
  92. CASE_ERR(EMLINK);
  93. CASE_ERR(EPIPE);
  94. CASE_ERR(EDOM);
  95. CASE_ERR(ERANGE);
  96. CASE_ERR(ENOSYS);
  97. default:
  98. return itoa(err);
  99. }
  100. }
  101. static int pad_spc(int llen, int cnt, const char *fmt, ...)
  102. {
  103. va_list args;
  104. int len;
  105. int ret;
  106. for (len = 0; len < cnt - llen; len++)
  107. putchar(' ');
  108. va_start(args, fmt);
  109. ret = vfprintf(stdout, fmt, args);
  110. va_end(args);
  111. return ret < 0 ? ret : ret + len;
  112. }
  113. /* The tests below are intended to be used by the macroes, which evaluate
  114. * expression <expr>, print the status to stdout, and update the "ret"
  115. * variable to count failures. The functions themselves return the number
  116. * of failures, thus either 0 or 1.
  117. */
  118. #define EXPECT_ZR(cond, expr) \
  119. do { if (!cond) pad_spc(llen, 40, "[SKIPPED]\n"); else ret += expect_zr(expr, llen); } while (0)
  120. static int expect_zr(int expr, int llen)
  121. {
  122. int ret = !(expr == 0);
  123. llen += printf(" = %d ", expr);
  124. pad_spc(llen, 40, ret ? "[FAIL]\n" : " [OK]\n");
  125. return ret;
  126. }
  127. #define EXPECT_NZ(cond, expr, val) \
  128. do { if (!cond) pad_spc(llen, 40, "[SKIPPED]\n"); else ret += expect_nz(expr, llen; } while (0)
  129. static int expect_nz(int expr, int llen)
  130. {
  131. int ret = !(expr != 0);
  132. llen += printf(" = %d ", expr);
  133. pad_spc(llen, 40, ret ? "[FAIL]\n" : " [OK]\n");
  134. return ret;
  135. }
  136. #define EXPECT_EQ(cond, expr, val) \
  137. do { if (!cond) pad_spc(llen, 40, "[SKIPPED]\n"); else ret += expect_eq(expr, llen, val); } while (0)
  138. static int expect_eq(int expr, int llen, int val)
  139. {
  140. int ret = !(expr == val);
  141. llen += printf(" = %d ", expr);
  142. pad_spc(llen, 40, ret ? "[FAIL]\n" : " [OK]\n");
  143. return ret;
  144. }
  145. #define EXPECT_NE(cond, expr, val) \
  146. do { if (!cond) pad_spc(llen, 40, "[SKIPPED]\n"); else ret += expect_ne(expr, llen, val); } while (0)
  147. static int expect_ne(int expr, int llen, int val)
  148. {
  149. int ret = !(expr != val);
  150. llen += printf(" = %d ", expr);
  151. pad_spc(llen, 40, ret ? "[FAIL]\n" : " [OK]\n");
  152. return ret;
  153. }
  154. #define EXPECT_GE(cond, expr, val) \
  155. do { if (!cond) pad_spc(llen, 40, "[SKIPPED]\n"); else ret += expect_ge(expr, llen, val); } while (0)
  156. static int expect_ge(int expr, int llen, int val)
  157. {
  158. int ret = !(expr >= val);
  159. llen += printf(" = %d ", expr);
  160. pad_spc(llen, 40, ret ? "[FAIL]\n" : " [OK]\n");
  161. return ret;
  162. }
  163. #define EXPECT_GT(cond, expr, val) \
  164. do { if (!cond) pad_spc(llen, 40, "[SKIPPED]\n"); else ret += expect_gt(expr, llen, val); } while (0)
  165. static int expect_gt(int expr, int llen, int val)
  166. {
  167. int ret = !(expr > val);
  168. llen += printf(" = %d ", expr);
  169. pad_spc(llen, 40, ret ? "[FAIL]\n" : " [OK]\n");
  170. return ret;
  171. }
  172. #define EXPECT_LE(cond, expr, val) \
  173. do { if (!cond) pad_spc(llen, 40, "[SKIPPED]\n"); else ret += expect_le(expr, llen, val); } while (0)
  174. static int expect_le(int expr, int llen, int val)
  175. {
  176. int ret = !(expr <= val);
  177. llen += printf(" = %d ", expr);
  178. pad_spc(llen, 40, ret ? "[FAIL]\n" : " [OK]\n");
  179. return ret;
  180. }
  181. #define EXPECT_LT(cond, expr, val) \
  182. do { if (!cond) pad_spc(llen, 40, "[SKIPPED]\n"); else ret += expect_lt(expr, llen, val); } while (0)
  183. static int expect_lt(int expr, int llen, int val)
  184. {
  185. int ret = !(expr < val);
  186. llen += printf(" = %d ", expr);
  187. pad_spc(llen, 40, ret ? "[FAIL]\n" : " [OK]\n");
  188. return ret;
  189. }
  190. #define EXPECT_SYSZR(cond, expr) \
  191. do { if (!cond) pad_spc(llen, 40, "[SKIPPED]\n"); else ret += expect_syszr(expr, llen); } while (0)
  192. static int expect_syszr(int expr, int llen)
  193. {
  194. int ret = 0;
  195. if (expr) {
  196. ret = 1;
  197. llen += printf(" = %d %s ", expr, errorname(errno));
  198. llen += pad_spc(llen, 40, "[FAIL]\n");
  199. } else {
  200. llen += printf(" = %d ", expr);
  201. llen += pad_spc(llen, 40, " [OK]\n");
  202. }
  203. return ret;
  204. }
  205. #define EXPECT_SYSEQ(cond, expr, val) \
  206. do { if (!cond) pad_spc(llen, 40, "[SKIPPED]\n"); else ret += expect_syseq(expr, llen, val); } while (0)
  207. static int expect_syseq(int expr, int llen, int val)
  208. {
  209. int ret = 0;
  210. if (expr != val) {
  211. ret = 1;
  212. llen += printf(" = %d %s ", expr, errorname(errno));
  213. llen += pad_spc(llen, 40, "[FAIL]\n");
  214. } else {
  215. llen += printf(" = %d ", expr);
  216. llen += pad_spc(llen, 40, " [OK]\n");
  217. }
  218. return ret;
  219. }
  220. #define EXPECT_SYSNE(cond, expr, val) \
  221. do { if (!cond) pad_spc(llen, 40, "[SKIPPED]\n"); else ret += expect_sysne(expr, llen, val); } while (0)
  222. static int expect_sysne(int expr, int llen, int val)
  223. {
  224. int ret = 0;
  225. if (expr == val) {
  226. ret = 1;
  227. llen += printf(" = %d %s ", expr, errorname(errno));
  228. llen += pad_spc(llen, 40, "[FAIL]\n");
  229. } else {
  230. llen += printf(" = %d ", expr);
  231. llen += pad_spc(llen, 40, " [OK]\n");
  232. }
  233. return ret;
  234. }
  235. #define EXPECT_SYSER(cond, expr, expret, experr) \
  236. do { if (!cond) pad_spc(llen, 40, "[SKIPPED]\n"); else ret += expect_syserr(expr, expret, experr, llen); } while (0)
  237. static int expect_syserr(int expr, int expret, int experr, int llen)
  238. {
  239. int ret = 0;
  240. int _errno = errno;
  241. llen += printf(" = %d %s ", expr, errorname(_errno));
  242. if (expr != expret || _errno != experr) {
  243. ret = 1;
  244. llen += printf(" != (%d %s) ", expret, errorname(experr));
  245. llen += pad_spc(llen, 40, "[FAIL]\n");
  246. } else {
  247. llen += pad_spc(llen, 40, " [OK]\n");
  248. }
  249. return ret;
  250. }
  251. #define EXPECT_PTRZR(cond, expr) \
  252. do { if (!cond) pad_spc(llen, 40, "[SKIPPED]\n"); else ret += expect_ptrzr(expr, llen); } while (0)
  253. static int expect_ptrzr(const void *expr, int llen)
  254. {
  255. int ret = 0;
  256. llen += printf(" = <%p> ", expr);
  257. if (expr) {
  258. ret = 1;
  259. llen += pad_spc(llen, 40, "[FAIL]\n");
  260. } else {
  261. llen += pad_spc(llen, 40, " [OK]\n");
  262. }
  263. return ret;
  264. }
  265. #define EXPECT_PTRNZ(cond, expr) \
  266. do { if (!cond) pad_spc(llen, 40, "[SKIPPED]\n"); else ret += expect_ptrnz(expr, llen); } while (0)
  267. static int expect_ptrnz(const void *expr, int llen)
  268. {
  269. int ret = 0;
  270. llen += printf(" = <%p> ", expr);
  271. if (!expr) {
  272. ret = 1;
  273. llen += pad_spc(llen, 40, "[FAIL]\n");
  274. } else {
  275. llen += pad_spc(llen, 40, " [OK]\n");
  276. }
  277. return ret;
  278. }
  279. #define EXPECT_STRZR(cond, expr) \
  280. do { if (!cond) pad_spc(llen, 40, "[SKIPPED]\n"); else ret += expect_strzr(expr, llen); } while (0)
  281. static int expect_strzr(const char *expr, int llen)
  282. {
  283. int ret = 0;
  284. llen += printf(" = <%s> ", expr);
  285. if (expr) {
  286. ret = 1;
  287. llen += pad_spc(llen, 40, "[FAIL]\n");
  288. } else {
  289. llen += pad_spc(llen, 40, " [OK]\n");
  290. }
  291. return ret;
  292. }
  293. #define EXPECT_STRNZ(cond, expr) \
  294. do { if (!cond) pad_spc(llen, 40, "[SKIPPED]\n"); else ret += expect_strnz(expr, llen); } while (0)
  295. static int expect_strnz(const char *expr, int llen)
  296. {
  297. int ret = 0;
  298. llen += printf(" = <%s> ", expr);
  299. if (!expr) {
  300. ret = 1;
  301. llen += pad_spc(llen, 40, "[FAIL]\n");
  302. } else {
  303. llen += pad_spc(llen, 40, " [OK]\n");
  304. }
  305. return ret;
  306. }
  307. #define EXPECT_STREQ(cond, expr, cmp) \
  308. do { if (!cond) pad_spc(llen, 40, "[SKIPPED]\n"); else ret += expect_streq(expr, llen, cmp); } while (0)
  309. static int expect_streq(const char *expr, int llen, const char *cmp)
  310. {
  311. int ret = 0;
  312. llen += printf(" = <%s> ", expr);
  313. if (strcmp(expr, cmp) != 0) {
  314. ret = 1;
  315. llen += pad_spc(llen, 40, "[FAIL]\n");
  316. } else {
  317. llen += pad_spc(llen, 40, " [OK]\n");
  318. }
  319. return ret;
  320. }
  321. #define EXPECT_STRNE(cond, expr, cmp) \
  322. do { if (!cond) pad_spc(llen, 40, "[SKIPPED]\n"); else ret += expect_strne(expr, llen, cmp); } while (0)
  323. static int expect_strne(const char *expr, int llen, const char *cmp)
  324. {
  325. int ret = 0;
  326. llen += printf(" = <%s> ", expr);
  327. if (strcmp(expr, cmp) == 0) {
  328. ret = 1;
  329. llen += pad_spc(llen, 40, "[FAIL]\n");
  330. } else {
  331. llen += pad_spc(llen, 40, " [OK]\n");
  332. }
  333. return ret;
  334. }
  335. /* declare tests based on line numbers. There must be exactly one test per line. */
  336. #define CASE_TEST(name) \
  337. case __LINE__: llen += printf("%d %s", test, #name);
  338. /* used by some syscall tests below */
  339. int test_getdents64(const char *dir)
  340. {
  341. char buffer[4096];
  342. int fd, ret;
  343. int err;
  344. ret = fd = open(dir, O_RDONLY | O_DIRECTORY, 0);
  345. if (ret < 0)
  346. return ret;
  347. ret = getdents64(fd, (void *)buffer, sizeof(buffer));
  348. err = errno;
  349. close(fd);
  350. errno = err;
  351. return ret;
  352. }
  353. /* Run syscall tests between IDs <min> and <max>.
  354. * Return 0 on success, non-zero on failure.
  355. */
  356. int run_syscall(int min, int max)
  357. {
  358. struct stat stat_buf;
  359. int proc;
  360. int test;
  361. int tmp;
  362. int ret = 0;
  363. void *p1, *p2;
  364. /* <proc> indicates whether or not /proc is mounted */
  365. proc = stat("/proc", &stat_buf) == 0;
  366. for (test = min; test >= 0 && test <= max; test++) {
  367. int llen = 0; // line length
  368. /* avoid leaving empty lines below, this will insert holes into
  369. * test numbers.
  370. */
  371. switch (test + __LINE__ + 1) {
  372. CASE_TEST(getpid); EXPECT_SYSNE(1, getpid(), -1); break;
  373. CASE_TEST(getppid); EXPECT_SYSNE(1, getppid(), -1); break;
  374. #ifdef NOLIBC
  375. CASE_TEST(gettid); EXPECT_SYSNE(1, gettid(), -1); break;
  376. #endif
  377. CASE_TEST(getpgid_self); EXPECT_SYSNE(1, getpgid(0), -1); break;
  378. CASE_TEST(getpgid_bad); EXPECT_SYSER(1, getpgid(-1), -1, ESRCH); break;
  379. CASE_TEST(kill_0); EXPECT_SYSZR(1, kill(getpid(), 0)); break;
  380. CASE_TEST(kill_CONT); EXPECT_SYSZR(1, kill(getpid(), 0)); break;
  381. CASE_TEST(kill_BADPID); EXPECT_SYSER(1, kill(INT_MAX, 0), -1, ESRCH); break;
  382. CASE_TEST(sbrk); if ((p1 = p2 = sbrk(4096)) != (void *)-1) p2 = sbrk(-4096); EXPECT_SYSZR(1, (p2 == (void *)-1) || p2 == p1); break;
  383. CASE_TEST(brk); EXPECT_SYSZR(1, brk(sbrk(0))); break;
  384. CASE_TEST(chdir_root); EXPECT_SYSZR(1, chdir("/")); break;
  385. CASE_TEST(chdir_dot); EXPECT_SYSZR(1, chdir(".")); break;
  386. CASE_TEST(chdir_blah); EXPECT_SYSER(1, chdir("/blah"), -1, ENOENT); break;
  387. CASE_TEST(chmod_net); EXPECT_SYSZR(proc, chmod("/proc/self/net", 0555)); break;
  388. CASE_TEST(chmod_self); EXPECT_SYSER(proc, chmod("/proc/self", 0555), -1, EPERM); break;
  389. CASE_TEST(chown_self); EXPECT_SYSER(proc, chown("/proc/self", 0, 0), -1, EPERM); break;
  390. CASE_TEST(chroot_root); EXPECT_SYSZR(1, chroot("/")); break;
  391. CASE_TEST(chroot_blah); EXPECT_SYSER(1, chroot("/proc/self/blah"), -1, ENOENT); break;
  392. CASE_TEST(chroot_exe); EXPECT_SYSER(proc, chroot("/proc/self/exe"), -1, ENOTDIR); break;
  393. CASE_TEST(close_m1); EXPECT_SYSER(1, close(-1), -1, EBADF); break;
  394. CASE_TEST(close_dup); EXPECT_SYSZR(1, close(dup(0))); break;
  395. CASE_TEST(dup_0); tmp = dup(0); EXPECT_SYSNE(1, tmp, -1); close(tmp); break;
  396. CASE_TEST(dup_m1); tmp = dup(-1); EXPECT_SYSER(1, tmp, -1, EBADF); if (tmp != -1) close(tmp); break;
  397. CASE_TEST(dup2_0); tmp = dup2(0, 100); EXPECT_SYSNE(1, tmp, -1); close(tmp); break;
  398. CASE_TEST(dup2_m1); tmp = dup2(-1, 100); EXPECT_SYSER(1, tmp, -1, EBADF); if (tmp != -1) close(tmp); break;
  399. CASE_TEST(dup3_0); tmp = dup3(0, 100, 0); EXPECT_SYSNE(1, tmp, -1); close(tmp); break;
  400. CASE_TEST(dup3_m1); tmp = dup3(-1, 100, 0); EXPECT_SYSER(1, tmp, -1, EBADF); if (tmp != -1) close(tmp); break;
  401. CASE_TEST(execve_root); EXPECT_SYSER(1, execve("/", (char*[]){ [0] = "/", [1] = NULL }, NULL), -1, EACCES); break;
  402. CASE_TEST(getdents64_root); EXPECT_SYSNE(1, test_getdents64("/"), -1); break;
  403. CASE_TEST(getdents64_null); EXPECT_SYSER(1, test_getdents64("/dev/null"), -1, ENOTDIR); break;
  404. CASE_TEST(gettimeofday_null); EXPECT_SYSZR(1, gettimeofday(NULL, NULL)); break;
  405. #ifdef NOLIBC
  406. CASE_TEST(gettimeofday_bad1); EXPECT_SYSER(1, gettimeofday((void *)1, NULL), -1, EFAULT); break;
  407. CASE_TEST(gettimeofday_bad2); EXPECT_SYSER(1, gettimeofday(NULL, (void *)1), -1, EFAULT); break;
  408. CASE_TEST(gettimeofday_bad2); EXPECT_SYSER(1, gettimeofday(NULL, (void *)1), -1, EFAULT); break;
  409. #endif
  410. CASE_TEST(ioctl_tiocinq); EXPECT_SYSZR(1, ioctl(0, TIOCINQ, &tmp)); break;
  411. CASE_TEST(ioctl_tiocinq); EXPECT_SYSZR(1, ioctl(0, TIOCINQ, &tmp)); break;
  412. CASE_TEST(link_root1); EXPECT_SYSER(1, link("/", "/"), -1, EEXIST); break;
  413. CASE_TEST(link_blah); EXPECT_SYSER(1, link("/proc/self/blah", "/blah"), -1, ENOENT); break;
  414. CASE_TEST(link_dir); EXPECT_SYSER(1, link("/", "/blah"), -1, EPERM); break;
  415. CASE_TEST(link_cross); EXPECT_SYSER(proc, link("/proc/self/net", "/blah"), -1, EXDEV); break;
  416. CASE_TEST(lseek_m1); EXPECT_SYSER(1, lseek(-1, 0, SEEK_SET), -1, EBADF); break;
  417. CASE_TEST(lseek_0); EXPECT_SYSER(1, lseek(0, 0, SEEK_SET), -1, ESPIPE); break;
  418. CASE_TEST(mkdir_root); EXPECT_SYSER(1, mkdir("/", 0755), -1, EEXIST); break;
  419. CASE_TEST(open_tty); EXPECT_SYSNE(1, tmp = open("/dev/null", 0), -1); if (tmp != -1) close(tmp); break;
  420. CASE_TEST(open_blah); EXPECT_SYSER(1, tmp = open("/proc/self/blah", 0), -1, ENOENT); if (tmp != -1) close(tmp); break;
  421. CASE_TEST(poll_null); EXPECT_SYSZR(1, poll(NULL, 0, 0)); break;
  422. CASE_TEST(poll_stdout); EXPECT_SYSNE(1, ({ struct pollfd fds = { 1, POLLOUT, 0}; poll(&fds, 1, 0); }), -1); break;
  423. CASE_TEST(poll_fault); EXPECT_SYSER(1, poll((void *)1, 1, 0), -1, EFAULT); break;
  424. CASE_TEST(read_badf); EXPECT_SYSER(1, read(-1, &tmp, 1), -1, EBADF); break;
  425. CASE_TEST(sched_yield); EXPECT_SYSZR(1, sched_yield()); break;
  426. CASE_TEST(select_null); EXPECT_SYSZR(1, ({ struct timeval tv = { 0 }; select(0, NULL, NULL, NULL, &tv); })); break;
  427. 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;
  428. CASE_TEST(select_fault); EXPECT_SYSER(1, select(1, (void *)1, NULL, NULL, 0), -1, EFAULT); break;
  429. CASE_TEST(stat_blah); EXPECT_SYSER(1, stat("/proc/self/blah", &stat_buf), -1, ENOENT); break;
  430. CASE_TEST(stat_fault); EXPECT_SYSER(1, stat(NULL, &stat_buf), -1, EFAULT); break;
  431. CASE_TEST(symlink_root); EXPECT_SYSER(1, symlink("/", "/"), -1, EEXIST); break;
  432. CASE_TEST(unlink_root); EXPECT_SYSER(1, unlink("/"), -1, EISDIR); break;
  433. CASE_TEST(unlink_blah); EXPECT_SYSER(1, unlink("/proc/self/blah"), -1, ENOENT); break;
  434. CASE_TEST(wait_child); EXPECT_SYSER(1, wait(&tmp), -1, ECHILD); break;
  435. CASE_TEST(waitpid_min); EXPECT_SYSER(1, waitpid(INT_MIN, &tmp, WNOHANG), -1, ESRCH); break;
  436. CASE_TEST(waitpid_child); EXPECT_SYSER(1, waitpid(getpid(), &tmp, WNOHANG), -1, ECHILD); break;
  437. CASE_TEST(write_badf); EXPECT_SYSER(1, write(-1, &tmp, 1), -1, EBADF); break;
  438. CASE_TEST(write_zero); EXPECT_SYSZR(1, write(1, &tmp, 0)); break;
  439. case __LINE__:
  440. return ret; /* must be last */
  441. /* note: do not set any defaults so as to permit holes above */
  442. }
  443. }
  444. return ret;
  445. }
  446. int run_stdlib(int min, int max)
  447. {
  448. int test;
  449. int tmp;
  450. int ret = 0;
  451. void *p1, *p2;
  452. for (test = min; test >= 0 && test <= max; test++) {
  453. int llen = 0; // line length
  454. /* avoid leaving empty lines below, this will insert holes into
  455. * test numbers.
  456. */
  457. switch (test + __LINE__ + 1) {
  458. CASE_TEST(getenv_TERM); EXPECT_STRNZ(1, getenv("TERM")); break;
  459. CASE_TEST(getenv_blah); EXPECT_STRZR(1, getenv("blah")); break;
  460. CASE_TEST(setcmp_blah_blah); EXPECT_EQ(1, strcmp("blah", "blah"), 0); break;
  461. CASE_TEST(setcmp_blah_blah2); EXPECT_NE(1, strcmp("blah", "blah2"), 0); break;
  462. CASE_TEST(setncmp_blah_blah); EXPECT_EQ(1, strncmp("blah", "blah", 10), 0); break;
  463. CASE_TEST(setncmp_blah_blah4); EXPECT_EQ(1, strncmp("blah", "blah4", 4), 0); break;
  464. CASE_TEST(setncmp_blah_blah5); EXPECT_NE(1, strncmp("blah", "blah5", 5), 0); break;
  465. CASE_TEST(setncmp_blah_blah6); EXPECT_NE(1, strncmp("blah", "blah6", 6), 0); break;
  466. CASE_TEST(strchr_foobar_o); EXPECT_STREQ(1, strchr("foobar", 'o'), "oobar"); break;
  467. CASE_TEST(strchr_foobar_z); EXPECT_STRZR(1, strchr("foobar", 'z')); break;
  468. CASE_TEST(strrchr_foobar_o); EXPECT_STREQ(1, strrchr("foobar", 'o'), "obar"); break;
  469. CASE_TEST(strrchr_foobar_z); EXPECT_STRZR(1, strrchr("foobar", 'z')); break;
  470. case __LINE__:
  471. return ret; /* must be last */
  472. /* note: do not set any defaults so as to permit holes above */
  473. }
  474. }
  475. return ret;
  476. }
  477. /* prepare what needs to be prepared for pid 1 (stdio, /dev, /proc, etc) */
  478. int prepare(void)
  479. {
  480. struct stat stat_buf;
  481. /* It's possible that /dev doesn't even exist or was not mounted, so
  482. * we'll try to create it, mount it, or create minimal entries into it.
  483. * We want at least /dev/null and /dev/console.
  484. */
  485. if (stat("/dev/.", &stat_buf) == 0 || mkdir("/dev", 0755) == 0) {
  486. if (stat("/dev/console", &stat_buf) != 0 ||
  487. stat("/dev/null", &stat_buf) != 0) {
  488. /* try devtmpfs first, otherwise fall back to manual creation */
  489. if (mount("/dev", "/dev", "devtmpfs", 0, 0) != 0) {
  490. mknod("/dev/console", 0600 | S_IFCHR, makedev(5, 1));
  491. mknod("/dev/null", 0666 | S_IFCHR, makedev(1, 3));
  492. }
  493. }
  494. }
  495. /* If no /dev/console was found before calling init, stdio is closed so
  496. * we need to reopen it from /dev/console. If it failed above, it will
  497. * still fail here and we cannot emit a message anyway.
  498. */
  499. if (close(dup(1)) == -1) {
  500. int fd = open("/dev/console", O_RDWR);
  501. if (fd >= 0) {
  502. if (fd != 0)
  503. dup2(fd, 0);
  504. if (fd != 1)
  505. dup2(fd, 1);
  506. if (fd != 2)
  507. dup2(fd, 2);
  508. if (fd > 2)
  509. close(fd);
  510. puts("\nSuccessfully reopened /dev/console.");
  511. }
  512. }
  513. /* try to mount /proc if not mounted. Silently fail otherwise */
  514. if (stat("/proc/.", &stat_buf) == 0 || mkdir("/proc", 0755) == 0) {
  515. if (stat("/proc/self", &stat_buf) != 0)
  516. mount("/proc", "/proc", "proc", 0, 0);
  517. }
  518. return 0;
  519. }
  520. /* This is the definition of known test names, with their functions */
  521. static struct test test_names[] = {
  522. /* add new tests here */
  523. { .name = "syscall", .func = run_syscall },
  524. { .name = "stdlib", .func = run_stdlib },
  525. { 0 }
  526. };
  527. int is_setting_valid(char *test)
  528. {
  529. int idx, len, test_len, valid = 0;
  530. char delimiter;
  531. if (!test)
  532. return valid;
  533. test_len = strlen(test);
  534. for (idx = 0; test_names[idx].name; idx++) {
  535. len = strlen(test_names[idx].name);
  536. if (test_len < len)
  537. continue;
  538. if (strncmp(test, test_names[idx].name, len) != 0)
  539. continue;
  540. delimiter = test[len];
  541. if (delimiter != ':' && delimiter != ',' && delimiter != '\0')
  542. continue;
  543. valid = 1;
  544. break;
  545. }
  546. return valid;
  547. }
  548. int main(int argc, char **argv, char **envp)
  549. {
  550. int min = 0;
  551. int max = __INT_MAX__;
  552. int ret = 0;
  553. int err;
  554. int idx;
  555. char *test;
  556. environ = envp;
  557. /* when called as init, it's possible that no console was opened, for
  558. * example if no /dev file system was provided. We'll check that fd#1
  559. * was opened, and if not we'll attempt to create and open /dev/console
  560. * and /dev/null that we'll use for later tests.
  561. */
  562. if (getpid() == 1)
  563. prepare();
  564. /* the definition of a series of tests comes from either argv[1] or the
  565. * "NOLIBC_TEST" environment variable. It's made of a comma-delimited
  566. * series of test names and optional ranges:
  567. * syscall:5-15[:.*],stdlib:8-10
  568. */
  569. test = argv[1];
  570. if (!is_setting_valid(test))
  571. test = getenv("NOLIBC_TEST");
  572. if (is_setting_valid(test)) {
  573. char *comma, *colon, *dash, *value;
  574. do {
  575. comma = strchr(test, ',');
  576. if (comma)
  577. *(comma++) = '\0';
  578. colon = strchr(test, ':');
  579. if (colon)
  580. *(colon++) = '\0';
  581. for (idx = 0; test_names[idx].name; idx++) {
  582. if (strcmp(test, test_names[idx].name) == 0)
  583. break;
  584. }
  585. if (test_names[idx].name) {
  586. /* The test was named, it will be called at least
  587. * once. We may have an optional range at <colon>
  588. * here, which defaults to the full range.
  589. */
  590. do {
  591. min = 0; max = __INT_MAX__;
  592. value = colon;
  593. if (value && *value) {
  594. colon = strchr(value, ':');
  595. if (colon)
  596. *(colon++) = '\0';
  597. dash = strchr(value, '-');
  598. if (dash)
  599. *(dash++) = '\0';
  600. /* support :val: :min-max: :min-: :-max: */
  601. if (*value)
  602. min = atoi(value);
  603. if (!dash)
  604. max = min;
  605. else if (*dash)
  606. max = atoi(dash);
  607. value = colon;
  608. }
  609. /* now's time to call the test */
  610. printf("Running test '%s'\n", test_names[idx].name);
  611. err = test_names[idx].func(min, max);
  612. ret += err;
  613. printf("Errors during this test: %d\n\n", err);
  614. } while (colon && *colon);
  615. } else
  616. printf("Ignoring unknown test name '%s'\n", test);
  617. test = comma;
  618. } while (test && *test);
  619. } else {
  620. /* no test mentioned, run everything */
  621. for (idx = 0; test_names[idx].name; idx++) {
  622. printf("Running test '%s'\n", test_names[idx].name);
  623. err = test_names[idx].func(min, max);
  624. ret += err;
  625. printf("Errors during this test: %d\n\n", err);
  626. }
  627. }
  628. printf("Total number of errors: %d\n", ret);
  629. if (getpid() == 1) {
  630. /* we're running as init, there's no other process on the
  631. * system, thus likely started from a VM for a quick check.
  632. * Exiting will provoke a kernel panic that may be reported
  633. * as an error by Qemu or the hypervisor, while stopping
  634. * cleanly will often be reported as a success. This allows
  635. * to use the output of this program for bisecting kernels.
  636. */
  637. printf("Leaving init with final status: %d\n", !!ret);
  638. if (ret == 0)
  639. reboot(LINUX_REBOOT_CMD_POWER_OFF);
  640. #if defined(__x86_64__)
  641. /* QEMU started with "-device isa-debug-exit -no-reboot" will
  642. * exit with status code 2N+1 when N is written to 0x501. We
  643. * hard-code the syscall here as it's arch-dependent.
  644. */
  645. #if defined(_NOLIBC_SYS_H)
  646. else if (my_syscall3(__NR_ioperm, 0x501, 1, 1) == 0)
  647. #else
  648. else if (ioperm(0x501, 1, 1) == 0)
  649. #endif
  650. asm volatile ("outb %%al, %%dx" :: "d"(0x501), "a"(0));
  651. /* if it does nothing, fall back to the regular panic */
  652. #endif
  653. }
  654. printf("Exiting with status %d\n", !!ret);
  655. return !!ret;
  656. }