d0p1/ack
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

Fix signal unsafety in testdriver.c.

Browse source
This commit is contained in:
David Given 2016-12-01 22:05:22 +01:00
parent 90547157b4
commit b7de58e34e
1 changed files with 38 additions and 34 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

72
util/build/testrunner.c
View file

@ -10,9 +10,9 @@
static bool timed_out = false; static bool timed_out = false;
static bool child_exited = false; static bool child_exited = false;
static char** command = NULL; static char* const* command = NULL;
static int timeout = 0; static int timeout = 0;
static jmp_buf exit_jmp; static int pid = 0;
static void parse_arguments(int argc, char* const argv[]) static void parse_arguments(int argc, char* const argv[])
{ {
@ -44,7 +44,8 @@ static void parse_arguments(int argc, char* const argv[])
static void sigalrm_cb(int sigraised) static void sigalrm_cb(int sigraised)
{ {
timed_out = true; timed_out = true;
longjmp(exit_jmp, 1); if (pid)
kill(pid, SIGKILL);
} }
int main(int argc, char* const argv[]) int main(int argc, char* const argv[])
@ -53,48 +54,51 @@ int main(int argc, char* const argv[])
const int WRITE = 1; const int WRITE = 1;
int fds[2]; int fds[2];
int pid;
FILE* childin; FILE* childin;
int wstatus; int wstatus;
char buffer[4096]; char buffer[4096];
parse_arguments(argc, argv); parse_arguments(argc, argv);
if (setjmp(exit_jmp) == 0) pipe(fds);
pid = fork();
if (pid == 0)
{ {
/* First time through. */ /* Child */
close(fds[READ]);
close(0);
dup2(fds[WRITE], 1);
dup2(fds[WRITE], 2);
execvp(command[0], command);
_exit(1);
}
else
{
/* Parent */
close(fds[WRITE]);
signal(SIGALRM, sigalrm_cb); signal(SIGALRM, sigalrm_cb);
alarm(timeout); alarm(timeout);
pipe(fds);
pid = fork();
if (pid == 0)
{
/* Child */
close(fds[READ]);
close(0);
dup2(fds[WRITE], 1);
dup2(fds[WRITE], 2);
execvp(command[0], command);
_exit(1);
}
childin = fdopen(fds[READ], "r");
while (!timed_out)
{
if (!fgets(buffer, sizeof(buffer), childin))
break;
fputs(buffer, stdout);
if (strcmp(buffer, "@@FINISHED\n") == 0)
break;
if (strcmp(buffer, "@@FINISHED\r\n") == 0)
break;
}
} }
/* Reached via longjmp, EOF, or seeing a @@FINISHED. */ childin = fdopen(fds[READ], "r");
if (!childin)
fatal("cannot open pipe");
while (!timed_out)
{
if (!fgets(buffer, sizeof(buffer), childin))
break;
fputs(buffer, stdout);
if (strcmp(buffer, "@@FINISHED\n") == 0)
break;
if (strcmp(buffer, "@@FINISHED\r\n") == 0)
break;
}
/* Reached via EOF or seeing a @@FINISHED. */
alarm(0);
kill(pid, SIGKILL); kill(pid, SIGKILL);
waitpid(pid, &wstatus, 0); waitpid(pid, &wstatus, 0);