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 * $FreeBSD$
28 */
29
30#include <sys/cdefs.h>
31__FBSDID("$FreeBSD$");
32
33#include <sys/wait.h>
34
35#include <err.h>
36#include <inttypes.h>
37#include <stdio.h>
38#include <stdlib.h>
39#include <string.h>
40#include <unistd.h>
41
42#include "cap_test.h"
43
44/* Initialize a named test. Requires test_NAME() function to be declared. */
45#define	TEST_INIT(name)	{ #name, test_##name, FAILED }
46
47/* All of the tests that can be run. */
48struct test all_tests[] = {
49	TEST_INIT(capmode),
50	TEST_INIT(capabilities),
51	TEST_INIT(fcntl),
52	TEST_INIT(pdfork),
53	TEST_INIT(pdkill),
54	TEST_INIT(relative),
55	TEST_INIT(sysctl),
56};
57int test_count = sizeof(all_tests) / sizeof(struct test);
58
59int
60main(int argc, char *argv[])
61{
62
63	/*
64	 * If no tests have been specified at the command line, run them all.
65	 */
66	if (argc == 1) {
67		printf("1..%d\n", test_count);
68
69		for (int i = 0; i < test_count; i++)
70			execute(i + 1, all_tests + i);
71		return (0);
72	}
73
74	/*
75	 * Otherwise, run only the specified tests.
76	 */
77	printf("1..%d\n", argc - 1);
78	for (int i = 1; i < argc; i++)
79	{
80		int found = 0;
81		for (int j = 0; j < test_count; j++) {
82			if (strncmp(argv[i], all_tests[j].t_name,
83			    strlen(argv[i])) == 0) {
84				found = 1;
85				execute(i, all_tests + j);
86				break;
87			}
88		}
89
90		if (found == 0)
91			errx(-1, "No such test '%s'", argv[i]);
92	}
93
94	return (0);
95}
96
97int
98execute(int id, struct test *t) {
99	int result;
100
101	pid_t pid = fork();
102	if (pid < 0)
103		err(-1, "fork");
104	if (pid) {
105		/* Parent: wait for result from child. */
106		int status;
107		while (waitpid(pid, &status, 0) != pid) {}
108		if (WIFEXITED(status))
109			result = WEXITSTATUS(status);
110		else
111			result = FAILED;
112	} else {
113		/* Child process: run the test. */
114		exit(t->t_run());
115	}
116
117	printf("%s %d - %s\n",
118		(result == PASSED) ? "ok" : "not ok",
119		id, t->t_name);
120
121	return (result);
122}
123