Blame src/common/runtest.c

Szabolcs Nagy 814c6f
#include <stdlib.h>
Szabolcs Nagy 814c6f
#include <string.h>
Szabolcs Nagy 814c6f
#include <errno.h>
Szabolcs Nagy 814c6f
#include <signal.h>
Szabolcs Nagy 814c6f
#include <time.h>
Szabolcs Nagy 814c6f
#include <sys/types.h>
Szabolcs Nagy 814c6f
#include <sys/wait.h>
Szabolcs Nagy 814c6f
#include <sys/time.h>
Szabolcs Nagy 814c6f
#include <sys/resource.h>
Szabolcs Nagy 814c6f
#include <unistd.h>
Szabolcs Nagy 814c6f
#include "test.h"
Szabolcs Nagy 814c6f
Szabolcs Nagy 814c6f
static void handler(int s)
Szabolcs Nagy 814c6f
{
Szabolcs Nagy 814c6f
}
Szabolcs Nagy 814c6f
Szabolcs Nagy 7ef8af
static int start(char *wrap, char *argv[])
Szabolcs Nagy 814c6f
{
Szabolcs Nagy 814c6f
	int pid;
Szabolcs Nagy 814c6f
Szabolcs Nagy 814c6f
	pid = fork();
Szabolcs Nagy 814c6f
	if (pid == 0) {
Szabolcs Nagy 157cc4
		t_setrlim(RLIMIT_STACK, 100*1024);
Szabolcs Nagy 7ef8af
		if (*wrap) {
Szabolcs Nagy 7ef8af
			argv--;
Szabolcs Nagy 7ef8af
			argv[0] = wrap;
Szabolcs Nagy 7ef8af
		}
Szabolcs Nagy 7ef8af
		execvp(argv[0], argv);
Szabolcs Nagy 814c6f
		t_error("%s exec failed: %s\n", argv[0], strerror(errno));
Szabolcs Nagy 814c6f
		exit(1);
Szabolcs Nagy 814c6f
	}
Szabolcs Nagy 814c6f
	return pid;
Szabolcs Nagy 814c6f
}
Szabolcs Nagy 814c6f
Szabolcs Nagy 7ef8af
static void usage(char *argv[])
Szabolcs Nagy 7ef8af
{
Szabolcs Nagy 7ef8af
	t_error("usage: %s [-t timeoutsec] [-w wrapcmd] cmd [args..]\n", argv[0]);
Szabolcs Nagy 7ef8af
	exit(-1);
Szabolcs Nagy 7ef8af
}
Szabolcs Nagy 7ef8af
Szabolcs Nagy 814c6f
int main(int argc, char *argv[])
Szabolcs Nagy 814c6f
{
Szabolcs Nagy 7ef8af
	char *wrap = "";
Szabolcs Nagy 7ef8af
	int timeoutsec = 5;
Szabolcs Nagy 7ef8af
	int timeout = 0;
Szabolcs Nagy 814c6f
	int status;
Szabolcs Nagy 814c6f
	sigset_t set;
Szabolcs Nagy 7ef8af
	int opt;
Szabolcs Nagy 814c6f
	int pid;
Szabolcs Nagy 814c6f
Szabolcs Nagy 7ef8af
	while ((opt = getopt(argc, argv, "w:t:")) != -1) {
Szabolcs Nagy 7ef8af
		switch (opt) {
Szabolcs Nagy 7ef8af
		case 'w':
Szabolcs Nagy 7ef8af
			wrap = optarg;
Szabolcs Nagy 7ef8af
			break;
Szabolcs Nagy 7ef8af
		case 't':
Szabolcs Nagy 7ef8af
			timeoutsec = atoi(optarg);
Szabolcs Nagy 7ef8af
			break;
Szabolcs Nagy 7ef8af
		default:
Szabolcs Nagy 7ef8af
			usage(argv);
Szabolcs Nagy 7ef8af
		}
Szabolcs Nagy 814c6f
	}
Szabolcs Nagy 7ef8af
	if (optind >= argc)
Szabolcs Nagy 7ef8af
		usage(argv);
Szabolcs Nagy 7ef8af
	argv += optind;
Szabolcs Nagy 814c6f
	sigemptyset(&set);
Szabolcs Nagy 814c6f
	sigaddset(&set, SIGCHLD);
Szabolcs Nagy 814c6f
	sigprocmask(SIG_BLOCK, &set, 0);
Szabolcs Nagy 814c6f
	signal(SIGCHLD, handler);
Szabolcs Nagy 7ef8af
	pid = start(wrap, argv);
Szabolcs Nagy 7ef8af
	if (pid == -1) {
Szabolcs Nagy 7ef8af
		t_error("%s fork failed: %s\n", argv[0], strerror(errno));
Szabolcs Nagy 7ef8af
		t_printf("FAIL %s [internal]\n", argv[0]);
Szabolcs Nagy 7ef8af
		return -1;
Szabolcs Nagy 7ef8af
	}
Szabolcs Nagy 7ef8af
	if (sigtimedwait(&set, 0, &(struct timespec){timeoutsec,0}) == -1) {
Szabolcs Nagy 814c6f
		if (errno == EAGAIN)
Szabolcs Nagy 814c6f
			timeout = 1;
Szabolcs Nagy 814c6f
		if (kill(pid, SIGKILL) == -1)
Szabolcs Nagy 814c6f
			t_error("%s kill failed: %s\n", argv[0], strerror(errno));
Szabolcs Nagy 814c6f
	}
Szabolcs Nagy 814c6f
	if (waitpid(pid, &status, 0) != pid) {
Szabolcs Nagy 814c6f
		t_error("%s waitpid failed: %s\n", argv[0], strerror(errno));
Szabolcs Nagy 7ef8af
		t_printf("FAIL %s [internal]\n", argv[0]);
Szabolcs Nagy 814c6f
		return -1;
Szabolcs Nagy 814c6f
	}
Szabolcs Nagy 814c6f
	if (WIFEXITED(status)) {
Szabolcs Nagy 814c6f
		if (WEXITSTATUS(status) == 0)
Szabolcs Nagy 29ee9a
			return t_status;
Szabolcs Nagy 814c6f
		t_printf("FAIL %s [status %d]\n", argv[0], WEXITSTATUS(status));
Szabolcs Nagy 814c6f
	} else if (timeout) {
Szabolcs Nagy 814c6f
		t_printf("FAIL %s [timed out]\n", argv[0]);
Szabolcs Nagy 814c6f
	} else if (WIFSIGNALED(status)) {
Szabolcs Nagy 814c6f
		t_printf("FAIL %s [signal %s]\n", argv[0], strsignal(WTERMSIG(status)));
Szabolcs Nagy 814c6f
	} else
Szabolcs Nagy 814c6f
		t_printf("FAIL %s [unknown]\n", argv[0]);
Szabolcs Nagy 814c6f
	return 1;
Szabolcs Nagy 814c6f
}