1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
|
/* This file is part of GNU Pies testsuite.
Copyright (C) 2019-2020 Sergey Poznyakoff
GNU Pies is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 3, or (at your option)
any later version.
GNU Pies is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with GNU Pies. If not, see <http://www.gnu.org/licenses/>. */
#include <config.h>
#include <stdlib.h>
#include <stdio.h>
#include <errno.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <signal.h>
int volatile got_sigchld, got_sigalrm;
void
sighan (int sig)
{
switch (sig)
{
case SIGCHLD:
got_sigchld = 1;
break;
case SIGALRM:
got_sigalrm = 1;
break;
}
}
int
main (int argc, char **argv)
{
char *progname = argv[0];
unsigned long n;
char *p;
pid_t pid, ret;
int status;
if (argc < 3)
{
fprintf (stderr, "usage: %s TIMEOUT COMMAND ARGS...\n", progname);
fprintf (stderr, "Runs command with a timeout.\n");
exit (1);
}
errno = 0;
n = strtoul (argv[1], &p, 10);
if (errno || *p || n == 0)
{
fprintf (stderr, "%s: %s is not a valid timeout\n", progname, argv[1]);
exit (1);
}
argc -= 2;
argv += 2;
signal (SIGALRM, sighan);
signal (SIGCHLD, sighan);
pid = fork ();
if (pid == -1)
{
perror ("fork");
exit (127);
}
if (pid == 0)
{
execvp (argv[0], argv);
perror (argv[0]);
exit (127);
}
alarm (n);
while (1)
{
pause ();
if (got_sigchld)
{
alarm (0);
break;
}
if (got_sigalrm)
{
fprintf (stderr, "%s: timed out\n", progname);
kill (pid, SIGKILL);
exit (127);
}
}
ret = wait (&status);
if (ret != pid)
{
perror ("wait");
exit (127);
}
if (WIFEXITED (status))
return WEXITSTATUS (status);
if (WIFSIGNALED (status))
fprintf (stderr, "%s: %s terminated on signal %d\n", progname, argv[0],
WTERMSIG (status));
else if (WIFSTOPPED (status))
fprintf (stderr, "%s: %s stopped\n", progname, argv[0]);
else
fprintf (stderr, "%s: %s exited with unrecognized status %d\n",
progname, argv[0], status);
return 127;
}
|