tools/regression/security/cap_test/cap_test.c

   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 }