int main(int argc,char **argv)
{
struct stat st;
dir = argv[1];
if (!dir || argv[2])
strerr_die1x(100,"supervise: usage: supervise dir");
if (pipe(selfpipe) == -1)
strerr_die4sys(111,FATAL,"unable to create pipe for ",dir,": ");
coe(selfpipe[0]);
coe(selfpipe[1]);
ndelay_on(selfpipe[0]);
ndelay_on(selfpipe[1]);
sig_block(sig_child);
sig_catch(sig_child,trigger);
if (chdir(dir) == -1)
strerr_die4sys(111,FATAL,"unable to chdir to ",dir,": ");
if (stat("down",&st) != -1)
flagwantup = 0;
else
if (errno != error_noent)
strerr_die4sys(111,FATAL,"unable to stat ",dir,"/down: ");
mkdir("supervise",0700);
fdlock = open_append("supervise/lock");
if ((fdlock == -1) || (lock_exnb(fdlock) == -1))
strerr_die4sys(111,FATAL,"unable to acquire ",dir,"/supervise/lock: ");
coe(fdlock);
fifo_make("supervise/control",0600);
fdcontrol = open_read("supervise/control");
if (fdcontrol == -1)
strerr_die4sys(111,FATAL,"unable to read ",dir,"/supervise/control: ");
coe(fdcontrol);
ndelay_on(fdcontrol); /* shouldn't be necessary */
fdcontrolwrite = open_write("supervise/control");
if (fdcontrolwrite == -1)
strerr_die4sys(111,FATAL,"unable to write ",dir,"/supervise/control: ");
coe(fdcontrolwrite);
pidchange();
announce();
fifo_make("supervise/ok",0600);
fdok = open_read("supervise/ok");
if (fdok == -1)
strerr_die4sys(111,FATAL,"unable to read ",dir,"/supervise/ok: ");
coe(fdok);
if (!flagwant || flagwantup) trystart();
doit();
announce();
_exit(0);
}
void slock(const char *f, unsigned int d, unsigned int x) {
int fd;
if ((fd =open_append(f)) == -1) slock_die("unable to open lock", f, x);
if (d) {
if (lock_ex(fd) == -1) slock_die("unable to lock", f, x);
return;
}
if (lock_exnb(fd) == -1) slock_die("unable to lock", f, x);
}
void restart(struct cyclog *d)
{
struct stat st;
int fd;
int flagprocessed;
if (fchdir(fdstartdir) == -1)
strerr_die2sys(111,FATAL,"unable to switch to starting directory: ");
mkdir(d->dir,0700);
d->fddir = open_read(d->dir);
if ((d->fddir == -1) || (fchdir(d->fddir) == -1))
strerr_die4sys(111,FATAL,"unable to open directory ",d->dir,": ");
coe(d->fddir);
d->fdlock = open_append("lock");
if ((d->fdlock == -1) || (lock_exnb(d->fdlock) == -1))
strerr_die4sys(111,FATAL,"unable to lock directory ",d->dir,": ");
coe(d->fdlock);
if (stat("current",&st) == -1) {
if (errno != error_noent)
strerr_die4sys(111,FATAL,"unable to stat ",d->dir,"/current: ");
}
else
if (st.st_mode & 0100) {
fd = open_append("current");
if (fd == -1)
strerr_die4sys(111,FATAL,"unable to append to ",d->dir,"/current: ");
if (fchmod(fd,0644) == -1)
strerr_die4sys(111,FATAL,"unable to set mode of ",d->dir,"/current: ");
coe(fd);
d->fdcurrent = fd;
d->bytes = st.st_size;
return;
}
unlink("state");
unlink("newstate");
flagprocessed = 0;
if (stat("processed",&st) == -1) {
if (errno != error_noent)
strerr_die4sys(111,FATAL,"unable to stat ",d->dir,"/processed: ");
}
else if (st.st_mode & 0100)
flagprocessed = 1;
if (flagprocessed) {
unlink("previous");
finish(d,"processed","s");
}
else {
unlink("processed");
finish(d,"previous","u");
}
finish(d,"current","u");
fd = open_trunc("state");
if (fd == -1)
strerr_die4sys(111,FATAL,"unable to write to ",d->dir,"/state: ");
close(fd);
fd = open_append("current");
if (fd == -1)
strerr_die4sys(111,FATAL,"unable to write to ",d->dir,"/current: ");
if (fchmod(fd,0644) == -1)
strerr_die4sys(111,FATAL,"unable to set mode of ",d->dir,"/current: ");
coe(fd);
d->fdcurrent = fd;
d->bytes = 0;
}
int runsv_main(int argc UNUSED_PARAM, char **argv)
{
struct stat s;
int fd;
int r;
char buf[256];
INIT_G();
dir = single_argv(argv);
xpiped_pair(selfpipe);
close_on_exec_on(selfpipe.rd);
close_on_exec_on(selfpipe.wr);
ndelay_on(selfpipe.rd);
ndelay_on(selfpipe.wr);
sig_block(SIGCHLD);
bb_signals_recursive_norestart(1 << SIGCHLD, s_child);
sig_block(SIGTERM);
bb_signals_recursive_norestart(1 << SIGTERM, s_term);
xchdir(dir);
/* bss: svd[0].pid = 0; */
if (S_DOWN) svd[0].state = S_DOWN; /* otherwise already 0 (bss) */
if (C_NOOP) svd[0].ctrl = C_NOOP;
if (W_UP) svd[0].sd_want = W_UP;
/* bss: svd[0].islog = 0; */
/* bss: svd[1].pid = 0; */
gettimeofday_ns(&svd[0].start);
if (stat("down", &s) != -1)
svd[0].sd_want = W_DOWN;
if (stat("log", &s) == -1) {
if (errno != ENOENT)
warn_cannot("stat ./log");
} else {
if (!S_ISDIR(s.st_mode)) {
errno = 0;
warn_cannot("stat log/down: log is not a directory");
} else {
haslog = 1;
svd[1].state = S_DOWN;
svd[1].ctrl = C_NOOP;
svd[1].sd_want = W_UP;
svd[1].islog = 1;
gettimeofday_ns(&svd[1].start);
if (stat("log/down", &s) != -1)
svd[1].sd_want = W_DOWN;
xpiped_pair(logpipe);
close_on_exec_on(logpipe.rd);
close_on_exec_on(logpipe.wr);
}
}
if (mkdir("supervise", 0700) == -1) {
r = readlink("supervise", buf, sizeof(buf));
if (r != -1) {
if (r == sizeof(buf))
fatal2x_cannot("readlink ./supervise", ": name too long");
buf[r] = 0;
mkdir(buf, 0700);
} else {
if ((errno != ENOENT) && (errno != EINVAL))
fatal_cannot("readlink ./supervise");
}
}
svd[0].fdlock = xopen3("log/supervise/lock"+4,
O_WRONLY|O_NDELAY|O_APPEND|O_CREAT, 0600);
if (lock_exnb(svd[0].fdlock) == -1)
fatal_cannot("lock supervise/lock");
close_on_exec_on(svd[0].fdlock);
if (haslog) {
if (mkdir("log/supervise", 0700) == -1) {
r = readlink("log/supervise", buf, 256);
if (r != -1) {
if (r == 256)
fatal2x_cannot("readlink ./log/supervise", ": name too long");
buf[r] = 0;
fd = xopen(".", O_RDONLY|O_NDELAY);
xchdir("./log");
mkdir(buf, 0700);
if (fchdir(fd) == -1)
fatal_cannot("change back to service directory");
close(fd);
}
else {
if ((errno != ENOENT) && (errno != EINVAL))
fatal_cannot("readlink ./log/supervise");
}
}
svd[1].fdlock = xopen3("log/supervise/lock",
O_WRONLY|O_NDELAY|O_APPEND|O_CREAT, 0600);
if (lock_ex(svd[1].fdlock) == -1)
fatal_cannot("lock log/supervise/lock");
close_on_exec_on(svd[1].fdlock);
}
mkfifo("log/supervise/control"+4, 0600);
svd[0].fdcontrol = xopen("log/supervise/control"+4, O_RDONLY|O_NDELAY);
close_on_exec_on(svd[0].fdcontrol);
svd[0].fdcontrolwrite = xopen("log/supervise/control"+4, O_WRONLY|O_NDELAY);
close_on_exec_on(svd[0].fdcontrolwrite);
update_status(&svd[0]);
if (haslog) {
mkfifo("log/supervise/control", 0600);
svd[1].fdcontrol = xopen("log/supervise/control", O_RDONLY|O_NDELAY);
close_on_exec_on(svd[1].fdcontrol);
svd[1].fdcontrolwrite = xopen("log/supervise/control", O_WRONLY|O_NDELAY);
close_on_exec_on(svd[1].fdcontrolwrite);
update_status(&svd[1]);
}
mkfifo("log/supervise/ok"+4, 0600);
fd = xopen("log/supervise/ok"+4, O_RDONLY|O_NDELAY);
close_on_exec_on(fd);
if (haslog) {
mkfifo("log/supervise/ok", 0600);
fd = xopen("log/supervise/ok", O_RDONLY|O_NDELAY);
close_on_exec_on(fd);
}
for (;;) {
struct pollfd x[3];
unsigned deadline;
char ch;
if (haslog)
if (!svd[1].pid && svd[1].sd_want == W_UP)
startservice(&svd[1]);
if (!svd[0].pid)
if (svd[0].sd_want == W_UP || svd[0].state == S_FINISH)
startservice(&svd[0]);
x[0].fd = selfpipe.rd;
x[0].events = POLLIN;
x[1].fd = svd[0].fdcontrol;
x[1].events = POLLIN;
/* x[2] is used only if haslog == 1 */
x[2].fd = svd[1].fdcontrol;
x[2].events = POLLIN;
sig_unblock(SIGTERM);
sig_unblock(SIGCHLD);
poll(x, 2 + haslog, 3600*1000);
sig_block(SIGTERM);
sig_block(SIGCHLD);
while (read(selfpipe.rd, &ch, 1) == 1)
continue;
for (;;) {
pid_t child;
int wstat;
child = wait_any_nohang(&wstat);
if (!child)
break;
if ((child == -1) && (errno != EINTR))
break;
if (child == svd[0].pid) {
svd[0].wstat = wstat;
svd[0].pid = 0;
pidchanged = 1;
svd[0].ctrl &= ~C_TERM;
if (svd[0].state != S_FINISH) {
fd = open_read("finish");
if (fd != -1) {
close(fd);
svd[0].state = S_FINISH;
update_status(&svd[0]);
continue;
}
}
svd[0].state = S_DOWN;
deadline = svd[0].start.tv_sec + 1;
gettimeofday_ns(&svd[0].start);
update_status(&svd[0]);
if (LESS(svd[0].start.tv_sec, deadline))
sleep(1);
}
if (haslog) {
if (child == svd[1].pid) {
svd[0].wstat = wstat;
svd[1].pid = 0;
pidchanged = 1;
svd[1].state = S_DOWN;
svd[1].ctrl &= ~C_TERM;
deadline = svd[1].start.tv_sec + 1;
gettimeofday_ns(&svd[1].start);
update_status(&svd[1]);
if (LESS(svd[1].start.tv_sec, deadline))
sleep(1);
}
}
} /* for (;;) */
if (read(svd[0].fdcontrol, &ch, 1) == 1)
ctrl(&svd[0], ch);
if (haslog)
if (read(svd[1].fdcontrol, &ch, 1) == 1)
ctrl(&svd[1], ch);
if (sigterm) {
ctrl(&svd[0], 'x');
sigterm = 0;
}
if (svd[0].sd_want == W_EXIT && svd[0].state == S_DOWN) {
if (svd[1].pid == 0)
_exit(EXIT_SUCCESS);
if (svd[1].sd_want != W_EXIT) {
svd[1].sd_want = W_EXIT;
/* stopservice(&svd[1]); */
update_status(&svd[1]);
close(logpipe.wr);
close(logpipe.rd);
}
}
} /* for (;;) */
/* not reached */
return 0;
}
int main(int argc, char **argv)
{
struct sigaction sa;
if (parse_argv(argc, argv) < 0)
_exit(100);
snprintf(status_files[0], 1024, "%s/lock", status_dir);
snprintf(status_files[1], 1024, "%s/control", status_dir);
snprintf(status_files[2], 1024, "%s/ok", status_dir);
snprintf(status_files[3], 1024, "%s/status", status_dir);
snprintf(status_files[4], 1024, "%s/status.new", status_dir);
snprintf(status_files[5], 1024, "%s/supervise.log", status_dir);
snprintf(status_files[6], 1024, "%s/supervise.log.wf", status_dir);
sa.sa_handler = SIG_IGN;
sigemptyset(&sa.sa_mask);
if (sigaction(SIGHUP, &sa, NULL) < 0)
{
printf("unable to ignore SIGHUP for %s\n", service);
_exit(110);
}
if (mkdir(status_dir, 0700) < 0 && errno != EEXIST)
{
printf("unable to create dir: %s\n", status_dir);
_exit(110);
}
fdlog = open_append(status_files[5]);
if (fdlog == -1)
{
printf("unable to open %s%s", status_dir, "/supervise.log");
_exit(111);
}
coe(fdlog);
fdlogwf = open_append(status_files[6]);
if (fdlogwf == -1)
{
printf("unable to open %s%s", status_dir, "/supervise.log.wf");
_exit(1);
}
coe(fdlogwf);
if (daemon(1, 0) < 0)
{
printf("failed to daemonize supervise!\n");
_exit(111);
}
if (pipe(selfpipe) == -1)
{
write_log(fdlogwf, FATAL, "unable to create pipe for ", service, "\n");
_exit(111);
}
coe(selfpipe[0]);
coe(selfpipe[1]);
ndelay_on(selfpipe[0]);
ndelay_on(selfpipe[1]);
sig_block(sig_child);
sig_catch(sig_child, trigger);
sig_block(sig_alarm);
sig_catch(sig_alarm, timer_handler);
sig_unblock(sig_alarm);
fdlock = open_append(status_files[0]);
if ((fdlock == -1) || (lock_exnb(fdlock) == -1))
{
write_log(fdlogwf, FATAL, "Unable to acquier ", status_dir, "/lock\n");
_exit(111);
}
coe(fdlock);
fifo_make(status_files[1], 0600);
fdcontrol = open_read(status_files[1]);
if (fdcontrol == -1)
{
write_log(fdlogwf, FATAL, "unable to read ", status_dir, "/control\n");
_exit(1);
}
coe(fdcontrol);
ndelay_on(fdcontrol);
fdcontrolwrite = open_write(status_files[1]);
if (fdcontrolwrite == -1)
{
write_log(fdlogwf, FATAL, "unable to write ", status_dir, "/control\n");
_exit(1);
}
coe(fdcontrolwrite);
fifo_make(status_files[2], 0600);
fdok = open_read(status_files[2]);
if (fdok == -1)
{
write_log(fdlogwf, FATAL, "unable to read ", status_dir, "/ok\n");
_exit(1);
}
coe(fdok);
if (!restart_sh[0])
{
parse_conf();
}
pidchange();
announce();
if (!flagwant || flagwantup)
trystart();
doit();
announce();
_exit(0);
}
