2007-03-12 23:34:52 +00:00
|
|
|
/*
|
|
|
|
Copyright (c) 2001-2006, Gerrit Pape
|
|
|
|
All rights reserved.
|
|
|
|
|
|
|
|
Redistribution and use in source and binary forms, with or without
|
|
|
|
modification, are permitted provided that the following conditions are met:
|
|
|
|
|
|
|
|
1. Redistributions of source code must retain the above copyright notice,
|
|
|
|
this list of conditions and the following disclaimer.
|
|
|
|
2. Redistributions in binary form must reproduce the above copyright
|
|
|
|
notice, this list of conditions and the following disclaimer in the
|
|
|
|
documentation and/or other materials provided with the distribution.
|
|
|
|
3. The name of the author may not be used to endorse or promote products
|
|
|
|
derived from this software without specific prior written permission.
|
|
|
|
|
|
|
|
THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED
|
|
|
|
WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
|
|
|
|
MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO
|
|
|
|
EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
|
|
|
SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
|
|
|
PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
|
|
|
|
OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
|
|
|
|
WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
|
|
|
|
OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
|
|
|
|
ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
|
|
*/
|
|
|
|
|
2008-03-02 12:53:15 +00:00
|
|
|
/* Busyboxed by Denys Vlasenko <vda.linux@googlemail.com> */
|
2006-11-17 18:58:49 +00:00
|
|
|
/* TODO: depends on runit_lib.c - review and reduce/eliminate */
|
|
|
|
|
|
|
|
#include <sys/poll.h>
|
|
|
|
#include <sys/file.h>
|
2007-05-26 19:00:18 +00:00
|
|
|
#include "libbb.h"
|
2006-11-17 18:58:49 +00:00
|
|
|
#include "runit_lib.h"
|
|
|
|
|
2007-08-31 21:45:52 +00:00
|
|
|
#if ENABLE_MONOTONIC_SYSCALL
|
|
|
|
#include <sys/syscall.h>
|
|
|
|
|
|
|
|
/* libc has incredibly messy way of doing this,
|
|
|
|
* typically requiring -lrt. We just skip all this mess */
|
|
|
|
static void gettimeofday_ns(struct timespec *ts)
|
|
|
|
{
|
|
|
|
syscall(__NR_clock_gettime, CLOCK_REALTIME, ts);
|
|
|
|
}
|
|
|
|
#else
|
|
|
|
static void gettimeofday_ns(struct timespec *ts)
|
|
|
|
{
|
|
|
|
if (sizeof(struct timeval) == sizeof(struct timespec)
|
|
|
|
&& sizeof(((struct timeval*)ts)->tv_usec) == sizeof(ts->tv_nsec)
|
|
|
|
) {
|
|
|
|
/* Cheat */
|
|
|
|
gettimeofday((void*)ts, NULL);
|
|
|
|
ts->tv_nsec *= 1000;
|
|
|
|
} else {
|
|
|
|
extern void BUG_need_to_implement_gettimeofday_ns(void);
|
|
|
|
BUG_need_to_implement_gettimeofday_ns();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
|
|
|
/* Compare possibly overflowing unsigned counters */
|
|
|
|
#define LESS(a,b) ((int)((unsigned)(b) - (unsigned)(a)) > 0)
|
|
|
|
|
2006-11-17 18:58:49 +00:00
|
|
|
/* state */
|
|
|
|
#define S_DOWN 0
|
|
|
|
#define S_RUN 1
|
|
|
|
#define S_FINISH 2
|
|
|
|
/* ctrl */
|
|
|
|
#define C_NOOP 0
|
|
|
|
#define C_TERM 1
|
|
|
|
#define C_PAUSE 2
|
|
|
|
/* want */
|
|
|
|
#define W_UP 0
|
|
|
|
#define W_DOWN 1
|
|
|
|
#define W_EXIT 2
|
|
|
|
|
|
|
|
struct svdir {
|
|
|
|
int pid;
|
2007-03-09 22:46:06 +00:00
|
|
|
smallint state;
|
|
|
|
smallint ctrl;
|
|
|
|
smallint want;
|
|
|
|
smallint islog;
|
2007-08-20 17:27:40 +00:00
|
|
|
struct timespec start;
|
2006-11-17 18:58:49 +00:00
|
|
|
int fdlock;
|
|
|
|
int fdcontrol;
|
|
|
|
int fdcontrolwrite;
|
|
|
|
};
|
|
|
|
|
2007-09-28 10:29:17 +00:00
|
|
|
struct globals {
|
|
|
|
smallint haslog;
|
|
|
|
smallint sigterm;
|
|
|
|
smallint pidchanged;
|
2008-02-16 13:20:56 +00:00
|
|
|
struct fd_pair selfpipe;
|
|
|
|
struct fd_pair logpipe;
|
2007-09-28 10:29:17 +00:00
|
|
|
char *dir;
|
|
|
|
struct svdir svd[2];
|
|
|
|
};
|
|
|
|
#define G (*(struct globals*)&bb_common_bufsiz1)
|
|
|
|
#define haslog (G.haslog )
|
|
|
|
#define sigterm (G.sigterm )
|
|
|
|
#define pidchanged (G.pidchanged )
|
|
|
|
#define selfpipe (G.selfpipe )
|
|
|
|
#define logpipe (G.logpipe )
|
|
|
|
#define dir (G.dir )
|
|
|
|
#define svd (G.svd )
|
|
|
|
#define INIT_G() \
|
|
|
|
do { \
|
|
|
|
pidchanged = 1; \
|
|
|
|
} while (0)
|
2006-11-17 18:58:49 +00:00
|
|
|
|
2007-01-29 22:51:58 +00:00
|
|
|
static void fatal2_cannot(const char *m1, const char *m2)
|
2006-11-17 18:58:49 +00:00
|
|
|
{
|
|
|
|
bb_perror_msg_and_die("%s: fatal: cannot %s%s", dir, m1, m2);
|
|
|
|
/* was exiting 111 */
|
|
|
|
}
|
2007-01-29 22:51:58 +00:00
|
|
|
static void fatal_cannot(const char *m)
|
2006-11-17 18:58:49 +00:00
|
|
|
{
|
|
|
|
fatal2_cannot(m, "");
|
|
|
|
/* was exiting 111 */
|
|
|
|
}
|
2007-01-29 22:51:58 +00:00
|
|
|
static void fatal2x_cannot(const char *m1, const char *m2)
|
2006-11-17 18:58:49 +00:00
|
|
|
{
|
|
|
|
bb_error_msg_and_die("%s: fatal: cannot %s%s", dir, m1, m2);
|
|
|
|
/* was exiting 111 */
|
|
|
|
}
|
2007-01-29 22:51:58 +00:00
|
|
|
static void warn_cannot(const char *m)
|
2006-11-17 18:58:49 +00:00
|
|
|
{
|
|
|
|
bb_perror_msg("%s: warning: cannot %s", dir, m);
|
|
|
|
}
|
|
|
|
|
|
|
|
static void s_child(int sig_no)
|
|
|
|
{
|
2008-02-16 13:20:56 +00:00
|
|
|
write(selfpipe.wr, "", 1);
|
2006-11-17 18:58:49 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
static void s_term(int sig_no)
|
|
|
|
{
|
|
|
|
sigterm = 1;
|
2008-02-16 13:20:56 +00:00
|
|
|
write(selfpipe.wr, "", 1); /* XXX */
|
2006-11-17 18:58:49 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
static char *add_str(char *p, const char *to_add)
|
|
|
|
{
|
|
|
|
while ((*p = *to_add) != '\0') {
|
|
|
|
p++;
|
|
|
|
to_add++;
|
|
|
|
}
|
|
|
|
return p;
|
|
|
|
}
|
|
|
|
|
|
|
|
static int open_trunc_or_warn(const char *name)
|
|
|
|
{
|
|
|
|
int fd = open_trunc(name);
|
|
|
|
if (fd < 0)
|
|
|
|
bb_perror_msg("%s: warning: cannot open %s",
|
|
|
|
dir, name);
|
|
|
|
return fd;
|
|
|
|
}
|
|
|
|
|
|
|
|
static void update_status(struct svdir *s)
|
|
|
|
{
|
2007-08-20 17:27:40 +00:00
|
|
|
ssize_t sz;
|
2006-11-17 18:58:49 +00:00
|
|
|
int fd;
|
2007-08-20 17:27:40 +00:00
|
|
|
svstatus_t status;
|
2006-11-17 18:58:49 +00:00
|
|
|
|
|
|
|
/* pid */
|
|
|
|
if (pidchanged) {
|
|
|
|
fd = open_trunc_or_warn("supervise/pid.new");
|
|
|
|
if (fd < 0)
|
|
|
|
return;
|
|
|
|
if (s->pid) {
|
2007-03-09 22:46:06 +00:00
|
|
|
char spid[sizeof(int)*3 + 2];
|
|
|
|
int size = sprintf(spid, "%u\n", (unsigned)s->pid);
|
2006-11-17 18:58:49 +00:00
|
|
|
write(fd, spid, size);
|
|
|
|
}
|
|
|
|
close(fd);
|
2007-03-09 22:46:06 +00:00
|
|
|
if (rename_or_warn("supervise/pid.new",
|
|
|
|
s->islog ? "log/supervise/pid" : "log/supervise/pid"+4))
|
2006-11-17 18:58:49 +00:00
|
|
|
return;
|
|
|
|
pidchanged = 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* stat */
|
|
|
|
fd = open_trunc_or_warn("supervise/stat.new");
|
|
|
|
if (fd < -1)
|
|
|
|
return;
|
|
|
|
|
|
|
|
{
|
|
|
|
char stat_buf[sizeof("finish, paused, got TERM, want down\n")];
|
|
|
|
char *p = stat_buf;
|
|
|
|
switch (s->state) {
|
|
|
|
case S_DOWN:
|
|
|
|
p = add_str(p, "down");
|
|
|
|
break;
|
|
|
|
case S_RUN:
|
|
|
|
p = add_str(p, "run");
|
|
|
|
break;
|
|
|
|
case S_FINISH:
|
|
|
|
p = add_str(p, "finish");
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
if (s->ctrl & C_PAUSE) p = add_str(p, ", paused");
|
|
|
|
if (s->ctrl & C_TERM) p = add_str(p, ", got TERM");
|
|
|
|
if (s->state != S_DOWN)
|
2006-12-26 10:42:51 +00:00
|
|
|
switch (s->want) {
|
2006-11-17 18:58:49 +00:00
|
|
|
case W_DOWN:
|
|
|
|
p = add_str(p, ", want down");
|
|
|
|
break;
|
|
|
|
case W_EXIT:
|
|
|
|
p = add_str(p, ", want exit");
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
*p++ = '\n';
|
|
|
|
write(fd, stat_buf, p - stat_buf);
|
|
|
|
close(fd);
|
|
|
|
}
|
|
|
|
|
2007-03-09 22:46:06 +00:00
|
|
|
rename_or_warn("supervise/stat.new",
|
|
|
|
s->islog ? "log/supervise/stat" : "log/supervise/stat"+4);
|
2006-11-17 18:58:49 +00:00
|
|
|
|
|
|
|
/* supervise compatibility */
|
2007-08-20 17:27:40 +00:00
|
|
|
memset(&status, 0, sizeof(status));
|
|
|
|
status.time_be64 = SWAP_BE64(s->start.tv_sec + 0x400000000000000aULL);
|
|
|
|
status.time_nsec_be32 = SWAP_BE32(s->start.tv_nsec);
|
|
|
|
status.pid_le32 = SWAP_LE32(s->pid);
|
2006-11-17 18:58:49 +00:00
|
|
|
if (s->ctrl & C_PAUSE)
|
2007-08-20 17:27:40 +00:00
|
|
|
status.paused = 1;
|
2006-11-17 18:58:49 +00:00
|
|
|
if (s->want == W_UP)
|
2007-08-20 17:27:40 +00:00
|
|
|
status.want = 'u';
|
2006-11-17 18:58:49 +00:00
|
|
|
else
|
2007-08-20 17:27:40 +00:00
|
|
|
status.want = 'd';
|
2006-11-17 18:58:49 +00:00
|
|
|
if (s->ctrl & C_TERM)
|
2007-08-20 17:27:40 +00:00
|
|
|
status.got_term = 1;
|
|
|
|
status.run_or_finish = s->state;
|
2006-11-17 18:58:49 +00:00
|
|
|
fd = open_trunc_or_warn("supervise/status.new");
|
|
|
|
if (fd < 0)
|
|
|
|
return;
|
2007-08-20 17:27:40 +00:00
|
|
|
sz = write(fd, &status, sizeof(status));
|
|
|
|
close(fd);
|
|
|
|
if (sz != sizeof(status)) {
|
2006-11-17 18:58:49 +00:00
|
|
|
warn_cannot("write supervise/status.new");
|
|
|
|
unlink("supervise/status.new");
|
|
|
|
return;
|
|
|
|
}
|
2007-03-09 22:46:06 +00:00
|
|
|
rename_or_warn("supervise/status.new",
|
|
|
|
s->islog ? "log/supervise/status" : "log/supervise/status"+4);
|
2006-11-17 18:58:49 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
static unsigned custom(struct svdir *s, char c)
|
|
|
|
{
|
|
|
|
int pid;
|
|
|
|
int w;
|
|
|
|
char a[10];
|
|
|
|
struct stat st;
|
|
|
|
char *prog[2];
|
|
|
|
|
|
|
|
if (s->islog) return 0;
|
2007-03-09 22:46:06 +00:00
|
|
|
strcpy(a, "control/?");
|
2006-11-17 18:58:49 +00:00
|
|
|
a[8] = c;
|
|
|
|
if (stat(a, &st) == 0) {
|
|
|
|
if (st.st_mode & S_IXUSR) {
|
|
|
|
pid = fork();
|
|
|
|
if (pid == -1) {
|
|
|
|
warn_cannot("fork for control/?");
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
if (!pid) {
|
2008-02-16 13:20:56 +00:00
|
|
|
if (haslog && dup2(logpipe.wr, 1) == -1)
|
2006-11-17 18:58:49 +00:00
|
|
|
warn_cannot("setup stdout for control/?");
|
|
|
|
prog[0] = a;
|
2007-03-09 22:46:06 +00:00
|
|
|
prog[1] = NULL;
|
2008-01-28 22:45:43 +00:00
|
|
|
execv(a, prog);
|
2006-11-17 18:58:49 +00:00
|
|
|
fatal_cannot("run control/?");
|
|
|
|
}
|
2008-01-02 19:55:04 +00:00
|
|
|
while (safe_waitpid(pid, &w, 0) == -1) {
|
2006-11-17 18:58:49 +00:00
|
|
|
warn_cannot("wait for child control/?");
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
return !wait_exitcode(w);
|
|
|
|
}
|
2007-03-09 22:46:06 +00:00
|
|
|
} else {
|
|
|
|
if (errno != ENOENT)
|
|
|
|
warn_cannot("stat control/?");
|
2006-11-17 18:58:49 +00:00
|
|
|
}
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
static void stopservice(struct svdir *s)
|
|
|
|
{
|
2007-03-09 22:46:06 +00:00
|
|
|
if (s->pid && !custom(s, 't')) {
|
2006-11-17 18:58:49 +00:00
|
|
|
kill(s->pid, SIGTERM);
|
2007-03-09 22:46:06 +00:00
|
|
|
s->ctrl |= C_TERM;
|
2006-11-17 18:58:49 +00:00
|
|
|
update_status(s);
|
|
|
|
}
|
|
|
|
if (s->want == W_DOWN) {
|
|
|
|
kill(s->pid, SIGCONT);
|
2007-03-09 22:46:06 +00:00
|
|
|
custom(s, 'd');
|
|
|
|
return;
|
2006-11-17 18:58:49 +00:00
|
|
|
}
|
|
|
|
if (s->want == W_EXIT) {
|
|
|
|
kill(s->pid, SIGCONT);
|
|
|
|
custom(s, 'x');
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
static void startservice(struct svdir *s)
|
|
|
|
{
|
|
|
|
int p;
|
|
|
|
char *run[2];
|
|
|
|
|
|
|
|
if (s->state == S_FINISH)
|
2007-01-29 22:51:58 +00:00
|
|
|
run[0] = (char*)"./finish";
|
2006-11-17 18:58:49 +00:00
|
|
|
else {
|
2007-01-29 22:51:58 +00:00
|
|
|
run[0] = (char*)"./run";
|
2006-11-17 18:58:49 +00:00
|
|
|
custom(s, 'u');
|
|
|
|
}
|
2007-01-29 22:51:58 +00:00
|
|
|
run[1] = NULL;
|
2006-11-17 18:58:49 +00:00
|
|
|
|
2007-04-10 15:42:06 +00:00
|
|
|
if (s->pid != 0)
|
|
|
|
stopservice(s); /* should never happen */
|
2006-11-17 18:58:49 +00:00
|
|
|
while ((p = fork()) == -1) {
|
|
|
|
warn_cannot("fork, sleeping");
|
|
|
|
sleep(5);
|
|
|
|
}
|
|
|
|
if (p == 0) {
|
|
|
|
/* child */
|
|
|
|
if (haslog) {
|
2008-02-16 13:20:56 +00:00
|
|
|
/* NB: bug alert! right order is close, then dup2 */
|
2006-11-17 18:58:49 +00:00
|
|
|
if (s->islog) {
|
2007-08-18 14:16:39 +00:00
|
|
|
xchdir("./log");
|
2008-02-16 13:20:56 +00:00
|
|
|
close(logpipe.wr);
|
|
|
|
xdup2(logpipe.rd, 0);
|
2006-11-17 18:58:49 +00:00
|
|
|
} else {
|
2008-02-16 13:20:56 +00:00
|
|
|
close(logpipe.rd);
|
|
|
|
xdup2(logpipe.wr, 1);
|
2006-11-17 18:58:49 +00:00
|
|
|
}
|
|
|
|
}
|
2008-02-16 22:58:56 +00:00
|
|
|
bb_signals(0
|
|
|
|
+ (1 << SIGCHLD)
|
|
|
|
+ (1 << SIGTERM)
|
|
|
|
, SIG_DFL);
|
2007-01-27 22:21:52 +00:00
|
|
|
sig_unblock(SIGCHLD);
|
|
|
|
sig_unblock(SIGTERM);
|
2007-04-01 01:18:20 +00:00
|
|
|
execvp(*run, run);
|
2007-03-09 22:46:06 +00:00
|
|
|
fatal2_cannot(s->islog ? "start log/" : "start ", *run);
|
2006-11-17 18:58:49 +00:00
|
|
|
}
|
|
|
|
if (s->state != S_FINISH) {
|
2007-08-20 17:27:40 +00:00
|
|
|
gettimeofday_ns(&s->start);
|
2006-11-17 18:58:49 +00:00
|
|
|
s->state = S_RUN;
|
|
|
|
}
|
|
|
|
s->pid = p;
|
|
|
|
pidchanged = 1;
|
|
|
|
s->ctrl = C_NOOP;
|
|
|
|
update_status(s);
|
|
|
|
}
|
|
|
|
|
|
|
|
static int ctrl(struct svdir *s, char c)
|
|
|
|
{
|
2007-03-09 22:46:06 +00:00
|
|
|
int sig;
|
|
|
|
|
2006-12-26 10:42:51 +00:00
|
|
|
switch (c) {
|
2006-11-17 18:58:49 +00:00
|
|
|
case 'd': /* down */
|
|
|
|
s->want = W_DOWN;
|
|
|
|
update_status(s);
|
2007-08-20 17:27:40 +00:00
|
|
|
if (s->pid && s->state != S_FINISH)
|
|
|
|
stopservice(s);
|
2006-11-17 18:58:49 +00:00
|
|
|
break;
|
|
|
|
case 'u': /* up */
|
|
|
|
s->want = W_UP;
|
|
|
|
update_status(s);
|
2007-08-20 17:27:40 +00:00
|
|
|
if (s->pid == 0)
|
|
|
|
startservice(s);
|
2006-11-17 18:58:49 +00:00
|
|
|
break;
|
|
|
|
case 'x': /* exit */
|
2007-08-20 17:27:40 +00:00
|
|
|
if (s->islog)
|
|
|
|
break;
|
2006-11-17 18:58:49 +00:00
|
|
|
s->want = W_EXIT;
|
|
|
|
update_status(s);
|
2007-03-09 22:46:06 +00:00
|
|
|
/* FALLTHROUGH */
|
2006-11-17 18:58:49 +00:00
|
|
|
case 't': /* sig term */
|
2007-08-20 17:27:40 +00:00
|
|
|
if (s->pid && s->state != S_FINISH)
|
|
|
|
stopservice(s);
|
2006-11-17 18:58:49 +00:00
|
|
|
break;
|
|
|
|
case 'k': /* sig kill */
|
2007-08-20 17:27:40 +00:00
|
|
|
if (s->pid && !custom(s, c))
|
|
|
|
kill(s->pid, SIGKILL);
|
2006-11-17 18:58:49 +00:00
|
|
|
s->state = S_DOWN;
|
|
|
|
break;
|
|
|
|
case 'p': /* sig pause */
|
2007-08-20 17:27:40 +00:00
|
|
|
if (s->pid && !custom(s, c))
|
|
|
|
kill(s->pid, SIGSTOP);
|
2007-03-09 22:46:06 +00:00
|
|
|
s->ctrl |= C_PAUSE;
|
2006-11-17 18:58:49 +00:00
|
|
|
update_status(s);
|
|
|
|
break;
|
|
|
|
case 'c': /* sig cont */
|
2007-08-20 17:27:40 +00:00
|
|
|
if (s->pid && !custom(s, c))
|
|
|
|
kill(s->pid, SIGCONT);
|
|
|
|
if (s->ctrl & C_PAUSE)
|
|
|
|
s->ctrl &= ~C_PAUSE;
|
2006-11-17 18:58:49 +00:00
|
|
|
update_status(s);
|
|
|
|
break;
|
|
|
|
case 'o': /* once */
|
|
|
|
s->want = W_DOWN;
|
|
|
|
update_status(s);
|
2007-08-20 17:27:40 +00:00
|
|
|
if (!s->pid)
|
|
|
|
startservice(s);
|
2006-11-17 18:58:49 +00:00
|
|
|
break;
|
|
|
|
case 'a': /* sig alarm */
|
2007-03-09 22:46:06 +00:00
|
|
|
sig = SIGALRM;
|
|
|
|
goto sendsig;
|
2006-11-17 18:58:49 +00:00
|
|
|
case 'h': /* sig hup */
|
2007-03-09 22:46:06 +00:00
|
|
|
sig = SIGHUP;
|
|
|
|
goto sendsig;
|
2006-11-17 18:58:49 +00:00
|
|
|
case 'i': /* sig int */
|
2007-03-09 22:46:06 +00:00
|
|
|
sig = SIGINT;
|
|
|
|
goto sendsig;
|
2006-11-17 18:58:49 +00:00
|
|
|
case 'q': /* sig quit */
|
2007-03-09 22:46:06 +00:00
|
|
|
sig = SIGQUIT;
|
|
|
|
goto sendsig;
|
2006-11-17 18:58:49 +00:00
|
|
|
case '1': /* sig usr1 */
|
2007-03-09 22:46:06 +00:00
|
|
|
sig = SIGUSR1;
|
|
|
|
goto sendsig;
|
2006-11-17 18:58:49 +00:00
|
|
|
case '2': /* sig usr2 */
|
2007-03-09 22:46:06 +00:00
|
|
|
sig = SIGUSR2;
|
|
|
|
goto sendsig;
|
2006-11-17 18:58:49 +00:00
|
|
|
}
|
|
|
|
return 1;
|
2007-03-09 22:46:06 +00:00
|
|
|
sendsig:
|
|
|
|
if (s->pid && !custom(s, c))
|
|
|
|
kill(s->pid, sig);
|
|
|
|
return 1;
|
2006-11-17 18:58:49 +00:00
|
|
|
}
|
|
|
|
|
2007-10-11 10:05:36 +00:00
|
|
|
int runsv_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
|
2006-11-17 18:58:49 +00:00
|
|
|
int runsv_main(int argc, char **argv)
|
|
|
|
{
|
|
|
|
struct stat s;
|
|
|
|
int fd;
|
|
|
|
int r;
|
|
|
|
char buf[256];
|
|
|
|
|
2007-09-28 10:29:17 +00:00
|
|
|
INIT_G();
|
|
|
|
|
2007-08-20 17:27:40 +00:00
|
|
|
if (!argv[1] || argv[2])
|
|
|
|
bb_show_usage();
|
2006-11-17 18:58:49 +00:00
|
|
|
dir = argv[1];
|
|
|
|
|
2008-02-16 13:20:56 +00:00
|
|
|
xpiped_pair(selfpipe);
|
|
|
|
close_on_exec_on(selfpipe.rd);
|
|
|
|
close_on_exec_on(selfpipe.wr);
|
|
|
|
ndelay_on(selfpipe.rd);
|
|
|
|
ndelay_on(selfpipe.wr);
|
2007-01-11 17:20:00 +00:00
|
|
|
|
2007-01-27 22:21:52 +00:00
|
|
|
sig_block(SIGCHLD);
|
2008-02-16 22:58:56 +00:00
|
|
|
bb_signals_recursive(1 << SIGCHLD, s_child);
|
2007-01-27 22:21:52 +00:00
|
|
|
sig_block(SIGTERM);
|
2008-02-16 22:58:56 +00:00
|
|
|
bb_signals_recursive(1 << SIGTERM, s_term);
|
2006-11-17 18:58:49 +00:00
|
|
|
|
|
|
|
xchdir(dir);
|
2007-03-09 22:46:06 +00:00
|
|
|
/* 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].want = W_UP;
|
|
|
|
/* bss: svd[0].islog = 0; */
|
|
|
|
/* bss: svd[1].pid = 0; */
|
2007-08-20 17:27:40 +00:00
|
|
|
gettimeofday_ns(&svd[0].start);
|
2006-11-17 18:58:49 +00:00
|
|
|
if (stat("down", &s) != -1) svd[0].want = W_DOWN;
|
|
|
|
|
|
|
|
if (stat("log", &s) == -1) {
|
|
|
|
if (errno != ENOENT)
|
|
|
|
warn_cannot("stat ./log");
|
|
|
|
} else {
|
2007-08-20 17:27:40 +00:00
|
|
|
if (!S_ISDIR(s.st_mode)) {
|
|
|
|
errno = 0;
|
|
|
|
warn_cannot("stat log/down: log is not a directory");
|
|
|
|
} else {
|
2006-11-17 18:58:49 +00:00
|
|
|
haslog = 1;
|
|
|
|
svd[1].state = S_DOWN;
|
|
|
|
svd[1].ctrl = C_NOOP;
|
|
|
|
svd[1].want = W_UP;
|
|
|
|
svd[1].islog = 1;
|
2007-08-20 17:27:40 +00:00
|
|
|
gettimeofday_ns(&svd[1].start);
|
2006-11-17 18:58:49 +00:00
|
|
|
if (stat("log/down", &s) != -1)
|
|
|
|
svd[1].want = W_DOWN;
|
2008-02-16 13:20:56 +00:00
|
|
|
xpiped_pair(logpipe);
|
|
|
|
close_on_exec_on(logpipe.rd);
|
|
|
|
close_on_exec_on(logpipe.wr);
|
2006-11-17 18:58:49 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if (mkdir("supervise", 0700) == -1) {
|
2007-03-09 22:46:06 +00:00
|
|
|
r = readlink("supervise", buf, sizeof(buf));
|
2006-11-17 18:58:49 +00:00
|
|
|
if (r != -1) {
|
2007-03-09 22:46:06 +00:00
|
|
|
if (r == sizeof(buf))
|
|
|
|
fatal2x_cannot("readlink ./supervise", ": name too long");
|
2006-11-17 18:58:49 +00:00
|
|
|
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");
|
2007-09-30 23:50:48 +00:00
|
|
|
close_on_exec_on(svd[0].fdlock);
|
2006-11-17 18:58:49 +00:00
|
|
|
if (haslog) {
|
|
|
|
if (mkdir("log/supervise", 0700) == -1) {
|
|
|
|
r = readlink("log/supervise", buf, 256);
|
|
|
|
if (r != -1) {
|
|
|
|
if (r == 256)
|
2007-03-09 22:46:06 +00:00
|
|
|
fatal2x_cannot("readlink ./log/supervise", ": name too long");
|
2006-11-17 18:58:49 +00:00
|
|
|
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");
|
2007-09-30 23:50:48 +00:00
|
|
|
close_on_exec_on(svd[1].fdlock);
|
2006-11-17 18:58:49 +00:00
|
|
|
}
|
|
|
|
|
2007-01-27 22:22:17 +00:00
|
|
|
mkfifo("log/supervise/control"+4, 0600);
|
2006-11-17 18:58:49 +00:00
|
|
|
svd[0].fdcontrol = xopen("log/supervise/control"+4, O_RDONLY|O_NDELAY);
|
2007-09-30 23:50:48 +00:00
|
|
|
close_on_exec_on(svd[0].fdcontrol);
|
2006-11-17 18:58:49 +00:00
|
|
|
svd[0].fdcontrolwrite = xopen("log/supervise/control"+4, O_WRONLY|O_NDELAY);
|
2007-09-30 23:50:48 +00:00
|
|
|
close_on_exec_on(svd[0].fdcontrolwrite);
|
2006-11-17 18:58:49 +00:00
|
|
|
update_status(&svd[0]);
|
|
|
|
if (haslog) {
|
2007-01-27 22:22:17 +00:00
|
|
|
mkfifo("log/supervise/control", 0600);
|
2006-11-17 18:58:49 +00:00
|
|
|
svd[1].fdcontrol = xopen("log/supervise/control", O_RDONLY|O_NDELAY);
|
2007-09-30 23:50:48 +00:00
|
|
|
close_on_exec_on(svd[1].fdcontrol);
|
2006-11-17 18:58:49 +00:00
|
|
|
svd[1].fdcontrolwrite = xopen("log/supervise/control", O_WRONLY|O_NDELAY);
|
2007-09-30 23:50:48 +00:00
|
|
|
close_on_exec_on(svd[1].fdcontrolwrite);
|
2006-11-17 18:58:49 +00:00
|
|
|
update_status(&svd[1]);
|
|
|
|
}
|
2007-01-27 22:22:17 +00:00
|
|
|
mkfifo("log/supervise/ok"+4, 0600);
|
2006-11-17 18:58:49 +00:00
|
|
|
fd = xopen("log/supervise/ok"+4, O_RDONLY|O_NDELAY);
|
2007-09-30 23:50:48 +00:00
|
|
|
close_on_exec_on(fd);
|
2006-11-17 18:58:49 +00:00
|
|
|
if (haslog) {
|
2007-01-27 22:22:17 +00:00
|
|
|
mkfifo("log/supervise/ok", 0600);
|
2006-11-17 18:58:49 +00:00
|
|
|
fd = xopen("log/supervise/ok", O_RDONLY|O_NDELAY);
|
2007-09-30 23:50:48 +00:00
|
|
|
close_on_exec_on(fd);
|
2006-11-17 18:58:49 +00:00
|
|
|
}
|
|
|
|
for (;;) {
|
2007-08-20 17:27:40 +00:00
|
|
|
struct pollfd x[3];
|
|
|
|
unsigned deadline;
|
2006-11-17 18:58:49 +00:00
|
|
|
char ch;
|
|
|
|
|
|
|
|
if (haslog)
|
|
|
|
if (!svd[1].pid && svd[1].want == W_UP)
|
|
|
|
startservice(&svd[1]);
|
|
|
|
if (!svd[0].pid)
|
|
|
|
if (svd[0].want == W_UP || svd[0].state == S_FINISH)
|
|
|
|
startservice(&svd[0]);
|
|
|
|
|
2008-02-16 13:20:56 +00:00
|
|
|
x[0].fd = selfpipe.rd;
|
2007-08-20 17:27:40 +00:00
|
|
|
x[0].events = POLLIN;
|
2006-11-17 18:58:49 +00:00
|
|
|
x[1].fd = svd[0].fdcontrol;
|
2007-08-20 17:27:40 +00:00
|
|
|
x[1].events = POLLIN;
|
|
|
|
/* x[2] is used only if haslog == 1 */
|
|
|
|
x[2].fd = svd[1].fdcontrol;
|
|
|
|
x[2].events = POLLIN;
|
2007-01-27 22:21:52 +00:00
|
|
|
sig_unblock(SIGTERM);
|
|
|
|
sig_unblock(SIGCHLD);
|
2007-08-20 17:27:40 +00:00
|
|
|
poll(x, 2 + haslog, 3600*1000);
|
2007-01-27 22:21:52 +00:00
|
|
|
sig_block(SIGTERM);
|
|
|
|
sig_block(SIGCHLD);
|
2006-11-17 18:58:49 +00:00
|
|
|
|
2008-02-16 13:20:56 +00:00
|
|
|
while (read(selfpipe.rd, &ch, 1) == 1)
|
2007-08-20 17:27:40 +00:00
|
|
|
continue;
|
|
|
|
|
2006-11-17 18:58:49 +00:00
|
|
|
for (;;) {
|
|
|
|
int child;
|
|
|
|
int wstat;
|
2007-01-11 17:20:00 +00:00
|
|
|
|
2008-01-02 19:55:04 +00:00
|
|
|
child = wait_any_nohang(&wstat);
|
2007-08-20 17:27:40 +00:00
|
|
|
if (!child)
|
|
|
|
break;
|
|
|
|
if ((child == -1) && (errno != EINTR))
|
|
|
|
break;
|
2006-11-17 18:58:49 +00:00
|
|
|
if (child == svd[0].pid) {
|
|
|
|
svd[0].pid = 0;
|
|
|
|
pidchanged = 1;
|
2007-03-09 22:46:06 +00:00
|
|
|
svd[0].ctrl &=~ C_TERM;
|
2007-02-18 11:07:43 +00:00
|
|
|
if (svd[0].state != S_FINISH) {
|
2006-11-17 18:58:49 +00:00
|
|
|
fd = open_read("finish");
|
|
|
|
if (fd != -1) {
|
|
|
|
close(fd);
|
|
|
|
svd[0].state = S_FINISH;
|
|
|
|
update_status(&svd[0]);
|
|
|
|
continue;
|
|
|
|
}
|
2007-02-18 11:07:43 +00:00
|
|
|
}
|
2006-11-17 18:58:49 +00:00
|
|
|
svd[0].state = S_DOWN;
|
2007-08-20 17:27:40 +00:00
|
|
|
deadline = svd[0].start.tv_sec + 1;
|
|
|
|
gettimeofday_ns(&svd[0].start);
|
2006-11-17 18:58:49 +00:00
|
|
|
update_status(&svd[0]);
|
2007-08-20 17:27:40 +00:00
|
|
|
if (LESS(svd[0].start.tv_sec, deadline))
|
|
|
|
sleep(1);
|
2006-11-17 18:58:49 +00:00
|
|
|
}
|
|
|
|
if (haslog) {
|
|
|
|
if (child == svd[1].pid) {
|
|
|
|
svd[1].pid = 0;
|
|
|
|
pidchanged = 1;
|
|
|
|
svd[1].state = S_DOWN;
|
2007-03-09 22:46:06 +00:00
|
|
|
svd[1].ctrl &= ~C_TERM;
|
2007-08-20 17:27:40 +00:00
|
|
|
deadline = svd[1].start.tv_sec + 1;
|
|
|
|
gettimeofday_ns(&svd[1].start);
|
2006-11-17 18:58:49 +00:00
|
|
|
update_status(&svd[1]);
|
2007-08-20 17:27:40 +00:00
|
|
|
if (LESS(svd[1].start.tv_sec, deadline))
|
|
|
|
sleep(1);
|
2006-11-17 18:58:49 +00:00
|
|
|
}
|
|
|
|
}
|
2008-02-16 13:20:56 +00:00
|
|
|
} /* for (;;) */
|
2006-11-17 18:58:49 +00:00
|
|
|
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].want == W_EXIT && svd[0].state == S_DOWN) {
|
|
|
|
if (svd[1].pid == 0)
|
|
|
|
_exit(0);
|
|
|
|
if (svd[1].want != W_EXIT) {
|
|
|
|
svd[1].want = W_EXIT;
|
|
|
|
/* stopservice(&svd[1]); */
|
|
|
|
update_status(&svd[1]);
|
2008-02-16 13:20:56 +00:00
|
|
|
close(logpipe.wr);
|
|
|
|
close(logpipe.rd);
|
2006-11-17 18:58:49 +00:00
|
|
|
}
|
|
|
|
}
|
2008-02-16 13:20:56 +00:00
|
|
|
} /* for (;;) */
|
2006-11-17 18:58:49 +00:00
|
|
|
/* not reached */
|
|
|
|
return 0;
|
|
|
|
}
|