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 TIMEOUT 120
23 #define KILL_TIMEOUT 5
24
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(int (test_function)(void),char * name)97 int test_harness(int (test_function)(void), char *name)
98 {
99 int rc;
100
101 test_start(name);
102 test_set_git_version(GIT_VERSION);
103
104 if (sigaction(SIGINT, &sig_action, NULL)) {
105 perror("sigaction (sigint)");
106 test_error(name);
107 return 1;
108 }
109
110 if (sigaction(SIGALRM, &sig_action, NULL)) {
111 perror("sigaction (sigalrm)");
112 test_error(name);
113 return 1;
114 }
115
116 rc = run_test(test_function, name);
117
118 if (rc == MAGIC_SKIP_RETURN_VALUE) {
119 test_skip(name);
120 /* so that skipped test is not marked as failed */
121 rc = 0;
122 } else
123 test_finish(name, rc);
124
125 return rc;
126 }
127
128 static char auxv[4096];
129
get_auxv_entry(int type)130 void *get_auxv_entry(int type)
131 {
132 ElfW(auxv_t) *p;
133 void *result;
134 ssize_t num;
135 int fd;
136
137 fd = open("/proc/self/auxv", O_RDONLY);
138 if (fd == -1) {
139 perror("open");
140 return NULL;
141 }
142
143 result = NULL;
144
145 num = read(fd, auxv, sizeof(auxv));
146 if (num < 0) {
147 perror("read");
148 goto out;
149 }
150
151 if (num > sizeof(auxv)) {
152 printf("Overflowed auxv buffer\n");
153 goto out;
154 }
155
156 p = (ElfW(auxv_t) *)auxv;
157
158 while (p->a_type != AT_NULL) {
159 if (p->a_type == type) {
160 result = (void *)p->a_un.a_val;
161 break;
162 }
163
164 p++;
165 }
166 out:
167 close(fd);
168 return result;
169 }
170