1 /*
2 * Copyright 2013, Michael Ellerman, IBM Corp.
3 * Licensed under GPLv2.
4 */
5
6 #include <errno.h>
7 #include <signal.h>
8 #include <stdbool.h>
9 #include <stdio.h>
10 #include <stdlib.h>
11 #include <sys/types.h>
12 #include <sys/wait.h>
13 #include <unistd.h>
14 #include <elf.h>
15 #include <fcntl.h>
16 #include <link.h>
17 #include <sys/stat.h>
18
19 #include "subunit.h"
20 #include "utils.h"
21
22 #define KILL_TIMEOUT 5
23
24 static uint64_t timeout = 120;
25
run_test(int (test_function)(void),char * name)26 int run_test(int (test_function)(void), char *name)
27 {
28 bool terminated;
29 int rc, status;
30 pid_t pid;
31
32 /* Make sure output is flushed before forking */
33 fflush(stdout);
34
35 pid = fork();
36 if (pid == 0) {
37 setpgid(0, 0);
38 exit(test_function());
39 } else if (pid == -1) {
40 perror("fork");
41 return 1;
42 }
43
44 setpgid(pid, pid);
45
46 /* Wake us up in timeout seconds */
47 alarm(timeout);
48 terminated = false;
49
50 wait:
51 rc = waitpid(pid, &status, 0);
52 if (rc == -1) {
53 if (errno != EINTR) {
54 printf("unknown error from waitpid\n");
55 return 1;
56 }
57
58 if (terminated) {
59 printf("!! force killing %s\n", name);
60 kill(-pid, SIGKILL);
61 return 1;
62 } else {
63 printf("!! killing %s\n", name);
64 kill(-pid, SIGTERM);
65 terminated = true;
66 alarm(KILL_TIMEOUT);
67 goto wait;
68 }
69 }
70
71 /* Kill anything else in the process group that is still running */
72 kill(-pid, SIGTERM);
73
74 if (WIFEXITED(status))
75 status = WEXITSTATUS(status);
76 else {
77 if (WIFSIGNALED(status))
78 printf("!! child died by signal %d\n", WTERMSIG(status));
79 else
80 printf("!! child died by unknown cause\n");
81
82 status = 1; /* Signal or other */
83 }
84
85 return status;
86 }
87
sig_handler(int signum)88 static void sig_handler(int signum)
89 {
90 /* Just wake us up from waitpid */
91 }
92
93 static struct sigaction sig_action = {
94 .sa_handler = sig_handler,
95 };
96
test_harness_set_timeout(uint64_t time)97 void test_harness_set_timeout(uint64_t time)
98 {
99 timeout = time;
100 }
101
test_harness(int (test_function)(void),char * name)102 int test_harness(int (test_function)(void), char *name)
103 {
104 int rc;
105
106 test_start(name);
107 test_set_git_version(GIT_VERSION);
108
109 if (sigaction(SIGINT, &sig_action, NULL)) {
110 perror("sigaction (sigint)");
111 test_error(name);
112 return 1;
113 }
114
115 if (sigaction(SIGALRM, &sig_action, NULL)) {
116 perror("sigaction (sigalrm)");
117 test_error(name);
118 return 1;
119 }
120
121 rc = run_test(test_function, name);
122
123 if (rc == MAGIC_SKIP_RETURN_VALUE) {
124 test_skip(name);
125 /* so that skipped test is not marked as failed */
126 rc = 0;
127 } else
128 test_finish(name, rc);
129
130 return rc;
131 }
132