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. */
48 struct 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 };
57 int test_count = sizeof(all_tests) / sizeof(struct test);
58 
59 int
60 main(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 
97 int
98 execute(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