1/*-
2 * Copyright (c) 2008-2011 Robert N. M. Watson
3 * Copyright (c) 2011 Jonathan Anderson
4 * All rights reserved.
5 *
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions
8 * are met:
9 * 1. Redistributions of source code must retain the above copyright
10 *    notice, this list of conditions and the following disclaimer.
11 * 2. Redistributions in binary form must reproduce the above copyright
12 *    notice, this list of conditions and the following disclaimer in the
13 *    documentation and/or other materials provided with the distribution.
14 *
15 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
16 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
17 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
18 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
19 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
20 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
21 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
22 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
23 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
24 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
25 * SUCH DAMAGE.
26 */
27
28#include <sys/cdefs.h>
29#include <sys/wait.h>
30
31#include <err.h>
32#include <inttypes.h>
33#include <stdio.h>
34#include <stdlib.h>
35#include <string.h>
36#include <unistd.h>
37
38#include "cap_test.h"
39
40/* Initialize a named test. Requires test_NAME() function to be declared. */
41#define	TEST_INIT(name)	{ #name, test_##name, FAILED }
42
43/* All of the tests that can be run. */
44struct test all_tests[] = {
45	TEST_INIT(capmode),
46	TEST_INIT(capabilities),
47	TEST_INIT(fcntl),
48	TEST_INIT(pdfork),
49	TEST_INIT(pdkill),
50	TEST_INIT(relative),
51	TEST_INIT(sysctl),
52};
53int test_count = sizeof(all_tests) / sizeof(struct test);
54
55int
56main(int argc, char *argv[])
57{
58
59	/*
60	 * If no tests have been specified at the command line, run them all.
61	 */
62	if (argc == 1) {
63		printf("1..%d\n", test_count);
64
65		for (int i = 0; i < test_count; i++)
66			execute(i + 1, all_tests + i);
67		return (0);
68	}
69
70	/*
71	 * Otherwise, run only the specified tests.
72	 */
73	printf("1..%d\n", argc - 1);
74	for (int i = 1; i < argc; i++)
75	{
76		int found = 0;
77		for (int j = 0; j < test_count; j++) {
78			if (strncmp(argv[i], all_tests[j].t_name,
79			    strlen(argv[i])) == 0) {
80				found = 1;
81				execute(i, all_tests + j);
82				break;
83			}
84		}
85
86		if (found == 0)
87			errx(-1, "No such test '%s'", argv[i]);
88	}
89
90	return (0);
91}
92
93int
94execute(int id, struct test *t) {
95	int result;
96
97	pid_t pid = fork();
98	if (pid < 0)
99		err(-1, "fork");
100	if (pid) {
101		/* Parent: wait for result from child. */
102		int status;
103		while (waitpid(pid, &status, 0) != pid) {}
104		if (WIFEXITED(status))
105			result = WEXITSTATUS(status);
106		else
107			result = FAILED;
108	} else {
109		/* Child process: run the test. */
110		exit(t->t_run());
111	}
112
113	printf("%s %d - %s\n",
114		(result == PASSED) ? "ok" : "not ok",
115		id, t->t_name);
116
117	return (result);
118}
119