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. */
44 struct 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 };
53 int test_count = sizeof(all_tests) / sizeof(struct test);
54 
55 int
56 main(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 
93 int
94 execute(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