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