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