1*5ade3599Smpi /* $OpenBSD: atf-c.c,v 1.1 2020/02/28 12:48:30 mpi Exp $ */
2*5ade3599Smpi
3*5ade3599Smpi #include <sys/wait.h>
4*5ade3599Smpi
5*5ade3599Smpi #include <err.h>
6*5ade3599Smpi #include <errno.h>
7*5ade3599Smpi #include <limits.h>
8*5ade3599Smpi #include <stdarg.h>
9*5ade3599Smpi #include <stdio.h>
10*5ade3599Smpi #include <stdlib.h>
11*5ade3599Smpi #include <pwd.h>
12*5ade3599Smpi #include <unistd.h>
13*5ade3599Smpi
14*5ade3599Smpi #include "atf-c.h"
15*5ade3599Smpi
16*5ade3599Smpi void usage(void);
17*5ade3599Smpi
18*5ade3599Smpi int cleanup;
19*5ade3599Smpi int count;
20*5ade3599Smpi int inspect;
21*5ade3599Smpi int run;
22*5ade3599Smpi int test;
23*5ade3599Smpi
24*5ade3599Smpi int
main(int argc,char * argv[])25*5ade3599Smpi main(int argc, char *argv[])
26*5ade3599Smpi {
27*5ade3599Smpi int ch, test;
28*5ade3599Smpi const char *errstr, *num;
29*5ade3599Smpi
30*5ade3599Smpi while ((ch = getopt(argc, argv, "c:i:nr:")) != -1) {
31*5ade3599Smpi switch(ch) {
32*5ade3599Smpi case 'c':
33*5ade3599Smpi cleanup = 1;
34*5ade3599Smpi num = optarg;
35*5ade3599Smpi break;
36*5ade3599Smpi case 'i':
37*5ade3599Smpi inspect = 1;
38*5ade3599Smpi num = optarg;
39*5ade3599Smpi break;
40*5ade3599Smpi case 'n':
41*5ade3599Smpi count = 1;
42*5ade3599Smpi break;
43*5ade3599Smpi case 'r':
44*5ade3599Smpi run = 1;
45*5ade3599Smpi num = optarg;
46*5ade3599Smpi break;
47*5ade3599Smpi default:
48*5ade3599Smpi usage();
49*5ade3599Smpi }
50*5ade3599Smpi }
51*5ade3599Smpi argc -= optind;
52*5ade3599Smpi argv += optind;
53*5ade3599Smpi
54*5ade3599Smpi if (cleanup + count + inspect + run > 1)
55*5ade3599Smpi usage();
56*5ade3599Smpi
57*5ade3599Smpi if (cleanup || inspect || run) {
58*5ade3599Smpi test = strtonum(num, 1, INT_MAX, &errstr);
59*5ade3599Smpi if (errstr != NULL)
60*5ade3599Smpi errx(1, "test # is %s: %s", errstr, argv[1]);
61*5ade3599Smpi }
62*5ade3599Smpi if (count)
63*5ade3599Smpi printf("%d\n", atf_test(0, 0));
64*5ade3599Smpi else if (cleanup)
65*5ade3599Smpi ATF_CLEANUP(test);
66*5ade3599Smpi else if (run)
67*5ade3599Smpi ATF_RUN(test);
68*5ade3599Smpi else if (inspect)
69*5ade3599Smpi ATF_INSPECT(test);
70*5ade3599Smpi else
71*5ade3599Smpi usage();
72*5ade3599Smpi
73*5ade3599Smpi return 0;
74*5ade3599Smpi }
75*5ade3599Smpi
76*5ade3599Smpi void
usage(void)77*5ade3599Smpi usage(void)
78*5ade3599Smpi {
79*5ade3599Smpi fprintf(stderr, "usage: %s [-n] [-c|i|r test#]\n", getprogname());
80*5ade3599Smpi exit(1);
81*5ade3599Smpi }
82*5ade3599Smpi
83*5ade3599Smpi void
atf_require(int exp,int expected_errno,const char * expstr,const char * src,const int lineno,char * fmt,...)84*5ade3599Smpi atf_require(int exp, int expected_errno, const char *expstr, const char *src,
85*5ade3599Smpi const int lineno, char *fmt, ...)
86*5ade3599Smpi {
87*5ade3599Smpi va_list args;
88*5ade3599Smpi if (!(exp)) {
89*5ade3599Smpi fprintf(stderr, "\n%s:%d: ", src, lineno);
90*5ade3599Smpi if (fmt != NULL) {
91*5ade3599Smpi va_start(args, fmt);
92*5ade3599Smpi vfprintf(stderr, fmt, args);
93*5ade3599Smpi va_end(args);
94*5ade3599Smpi } else {
95*5ade3599Smpi fprintf(stderr, "'%s' evaluated to false\n", expstr);
96*5ade3599Smpi }
97*5ade3599Smpi exit(1);
98*5ade3599Smpi } else if (expected_errno >= 0 && errno != expected_errno) {
99*5ade3599Smpi fprintf(stderr, "\n%s:%d: ", src, lineno);
100*5ade3599Smpi fprintf(stderr, "expected errno %d but got %d instead\n",
101*5ade3599Smpi expected_errno, errno);
102*5ade3599Smpi exit(1);
103*5ade3599Smpi }
104*5ade3599Smpi return;
105*5ade3599Smpi }
106*5ade3599Smpi
107*5ade3599Smpi void
atf_tc_fail(char * fmt,...)108*5ade3599Smpi atf_tc_fail(char *fmt, ...)
109*5ade3599Smpi {
110*5ade3599Smpi va_list args;
111*5ade3599Smpi va_start(args, fmt);
112*5ade3599Smpi verrx(1, fmt, args);
113*5ade3599Smpi }
114