ubase/respawn.c

107 lines
1.7 KiB
C

/* See LICENSE file for copyright and license details. */
#include <sys/select.h>
#include <sys/stat.h>
#include <sys/time.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <errno.h>
#include <fcntl.h>
#include <signal.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include "util.h"
static void
sigterm(int sig)
{
if (sig == SIGTERM) {
kill(0, SIGTERM);
_exit(0);
}
}
static void
usage(void)
{
eprintf("usage: respawn [-l fifo] [-d N] cmd [args...]\n");
}
int
main(int argc, char *argv[])
{
char *fifo = NULL;
unsigned int delay = 0;
pid_t pid;
char buf[BUFSIZ];
int savederrno;
int fd;
ssize_t n;
fd_set rdfd;
ARGBEGIN {
case 'd':
delay = estrtol(EARGF(usage()), 0);
break;
case 'l':
fifo = EARGF(usage());
break;
default:
usage();
} ARGEND;
if (argc < 1)
usage();
if (fifo && delay > 0)
usage();
setsid();
signal(SIGTERM, sigterm);
if (fifo) {
/* TODO: we should use O_RDONLY and re-open the fd on EOF */
fd = open(fifo, O_RDWR | O_NONBLOCK);
if (fd < 0)
eprintf("open %s:", fifo);
}
while (1) {
if (fifo) {
FD_ZERO(&rdfd);
FD_SET(fd, &rdfd);
n = select(fd + 1, &rdfd, NULL, NULL, NULL);
if (n < 0)
eprintf("select:");
if (n == 0 || FD_ISSET(fd, &rdfd) == 0)
continue;
while ((n = read(fd, buf, sizeof(buf))) > 0)
;
if (n < 0)
if (errno != EAGAIN)
eprintf("read %s:", fifo);
}
pid = fork();
if (pid < 0)
eprintf("fork:");
switch (pid) {
case 0:
execvp(argv[0], argv);
savederrno = errno;
weprintf("execvp %s:", argv[0]);
_exit(savederrno == ENOENT ? 127 : 126);
break;
default:
waitpid(pid, NULL, 0);
break;
}
if (!fifo)
sleep(delay);
}
/* not reachable */
return 0;
}