summaryrefslogtreecommitdiff
path: root/src/supervision/s6-svscan.c
diff options
context:
space:
mode:
Diffstat (limited to 'src/supervision/s6-svscan.c')
-rw-r--r--src/supervision/s6-svscan.c158
1 files changed, 84 insertions, 74 deletions
diff --git a/src/supervision/s6-svscan.c b/src/supervision/s6-svscan.c
index bf338d5..7016236 100644
--- a/src/supervision/s6-svscan.c
+++ b/src/supervision/s6-svscan.c
@@ -23,7 +23,7 @@
#include <s6/config.h>
#include <s6/s6-supervise.h>
-#define USAGE "s6-svscan [ -S | -s ] [ -c maxservices ] [ -t timeout ] [ -d notif ] [ -X consoleholder ] [ dir ]"
+#define USAGE "s6-svscan [ -c maxservices ] [ -t timeout ] [ -d notif ] [ -X consoleholder ] [ dir ]"
#define dieusage() strerr_dieusage(100, USAGE)
#define FINISH_PROG S6_SVSCAN_CTLDIR "/finish"
@@ -50,11 +50,11 @@ static struct svinfo_s *services ;
static unsigned int max = 500 ;
static unsigned int n = 0 ;
static tain_t deadline, defaulttimeout ;
-static char const *finish_arg = "reboot" ;
static int wantreap = 1 ;
static int wantscan = 1 ;
static unsigned int wantkill = 0 ;
static int cont = 1 ;
+static int waitall = 1 ;
static unsigned int consoleholder = 0 ;
static void restore_console (void)
@@ -102,56 +102,69 @@ static void killthem (void)
wantkill = 0 ;
}
-static void term (void)
+static inline void closethem (void)
{
- cont = 0 ;
- wantkill = 3 ;
+ unsigned int i = 0 ;
+ for (; i < n ; i++) if (services[i].flaglog)
+ {
+ if (services[i].p[1] >= 0) close(services[i].p[1]) ;
+ if (services[i].p[0] >= 0) close(services[i].p[0]) ;
+ }
}
-static void hup (void)
+static inline void waitthem (void)
{
- cont = 0 ;
- wantkill = 1 ;
+ unsigned int m = 0 ;
+ unsigned int i = 0 ;
+ pid_t pids[n << 1] ;
+ for (; i < n ; i++)
+ {
+ if (services[i].pid[0])
+ pids[m++] = services[i].pid[0] ;
+ if (services[i].flaglog && services[i].pid[1])
+ pids[m++] = services[i].pid[1] ;
+ }
+ if (!waitn(pids, m))
+ strerr_warnwu1sys("wait for all s6-supervise processes") ;
}
-static void quit (void)
+static inline void chld (void)
+{
+ wantreap = 1 ;
+}
+
+static inline void alrm (void)
+{
+ wantscan = 1 ;
+}
+
+static inline void abrt (void)
{
cont = 0 ;
- wantkill = 7 ;
+ waitall = 0 ;
}
-static void intr (void)
+static void hup (void)
{
- finish_arg = "reboot" ;
- term() ;
+ wantkill = 2 ;
+ wantscan = 1 ;
}
-static void usr1 (void)
+static void term (void)
{
- finish_arg = "poweroff" ;
- term() ;
+ cont = 0 ;
+ waitall = 1 ;
+ wantkill = 3 ;
}
-static void handle_signals (void)
+static void quit (void)
{
- for (;;)
- {
- switch (selfpipe_read())
- {
- case -1 : panic("selfpipe_read") ;
- case 0 : return ;
- case SIGCHLD : wantreap = 1 ; break ;
- case SIGALRM : wantscan = 1 ; break ;
- case SIGABRT : cont = 0 ; break ;
- case SIGTERM : term() ; break ;
- case SIGHUP : hup() ; break ;
- case SIGQUIT : quit() ; break ;
- case SIGINT : intr() ; break ;
- }
- }
+ cont = 0 ;
+ waitall = 1 ;
+ wantkill = 7 ;
}
-static void handle_diverted_signals (void)
+static void handle_signals (void)
{
for (;;)
{
@@ -160,9 +173,9 @@ static void handle_diverted_signals (void)
{
case -1 : panic("selfpipe_read") ;
case 0 : return ;
- case SIGCHLD : wantreap = 1 ; break ;
- case SIGALRM : wantscan = 1 ; break ;
- case SIGABRT : cont = 0 ; break ;
+ case SIGCHLD : chld() ; break ;
+ case SIGALRM : alrm() ; break ;
+ case SIGABRT : abrt() ; break ;
default :
{
char const *name = sig_name(sig) ;
@@ -172,7 +185,16 @@ static void handle_diverted_signals (void)
memcpy(fn, SIGNAL_PROG, SIGNAL_PROG_LEN) ;
memcpy(fn + SIGNAL_PROG_LEN, name, len + 1) ;
if (!child_spawn0(newargv[0], newargv, (char const **)environ))
- strerr_warnwu2sys("spawn ", newargv[0]) ;
+ {
+ if (errno != ENOENT) strerr_warnwu2sys("spawn ", newargv[0]) ;
+ switch (sig)
+ {
+ case SIGHUP : hup() ; break ;
+ case SIGINT :
+ case SIGTERM : term() ; break ;
+ case SIGQUIT : quit() ; break ;
+ }
+ }
}
}
}
@@ -188,22 +210,15 @@ static void handle_control (int fd)
else if (!r) break ;
else switch (c)
{
- case 'p' : finish_arg = "poweroff" ; break ;
- case 'h' : hup() ; return ;
- case 'r' : finish_arg = "reboot" ; break ;
- case 'a' : wantscan = 1 ; break ;
- case 't' : term() ; return ;
- case 's' : finish_arg = "halt" ; break ;
- case 'z' : wantreap = 1 ; break ;
- case 'b' : cont = 0 ; return ;
+ case 'z' : chld() ; break ;
+ case 'a' : alrm() ; break ;
+ case 'b' : abrt() ; break ;
+ case 'h' : hup() ; break ;
+ case 'i' :
+ case 't' : term() ; break ;
+ case 'q' : quit() ; break ;
case 'n' : wantkill = 2 ; break ;
case 'N' : wantkill = 6 ; break ;
- case '6' :
- case 'i' : intr() ; return ;
- case 'q' : quit() ; return ;
- case '0' : finish_arg = "halt" ; term() ; return ;
- case '7' : usr1() ; return ;
- case '8' : finish_arg = "other" ; term() ; return ;
default :
{
char s[2] = { c, 0 } ;
@@ -329,7 +344,7 @@ static void retrydirlater (void)
if (tain_less(&a, &deadline)) deadline = a ;
}
-static void check (char const *name)
+static inline void check (char const *name)
{
struct stat st ;
size_t namelen ;
@@ -420,7 +435,7 @@ static void check (char const *name)
}
}
-static void scan (void)
+static inline void scan (void)
{
unsigned int i = 0 ;
DIR *dir ;
@@ -468,7 +483,6 @@ static void scan (void)
int main (int argc, char const *const *argv)
{
iopause_fd x[2] = { { -1, IOPAUSE_READ, 0 }, { -1, IOPAUSE_READ, 0 } } ;
- int divertsignals = 0 ;
unsigned int notif = 0 ;
PROG = "s6-svscan" ;
{
@@ -476,14 +490,12 @@ int main (int argc, char const *const *argv)
unsigned int t = 0 ;
for (;;)
{
- int opt = subgetopt_r(argc, argv, "Sst:c:d:X:", &l) ;
+ int opt = subgetopt_r(argc, argv, "c:t:d:X:", &l) ;
if (opt == -1) break ;
switch (opt)
{
- case 'S' : divertsignals = 0 ; break ;
- case 's' : divertsignals = 1 ; break ;
- case 't' : if (uint0_scan(l.arg, &t)) break ;
case 'c' : if (uint0_scan(l.arg, &max)) break ;
+ case 't' : if (uint0_scan(l.arg, &t)) break ;
case 'd' :
if (!uint0_scan(l.arg, &notif)) dieusage() ;
if (notif < 3) strerr_dief1x(100, "notification fd must be 3 or more") ;
@@ -521,22 +533,19 @@ int main (int argc, char const *const *argv)
sigemptyset(&set) ;
sigaddset(&set, SIGCHLD) ;
sigaddset(&set, SIGALRM) ;
- sigaddset(&set, SIGTERM) ;
- sigaddset(&set, SIGHUP) ;
- sigaddset(&set, SIGQUIT) ;
sigaddset(&set, SIGABRT) ;
+ sigaddset(&set, SIGHUP) ;
sigaddset(&set, SIGINT) ;
- if (divertsignals)
- {
- sigaddset(&set, SIGUSR1) ;
- sigaddset(&set, SIGUSR2) ;
+ sigaddset(&set, SIGTERM) ;
+ sigaddset(&set, SIGQUIT) ;
+ sigaddset(&set, SIGUSR1) ;
+ sigaddset(&set, SIGUSR2) ;
#ifdef SIGPWR
- sigaddset(&set, SIGPWR) ;
+ sigaddset(&set, SIGPWR) ;
#endif
#ifdef SIGWINCH
- sigaddset(&set, SIGWINCH) ;
+ sigaddset(&set, SIGWINCH) ;
#endif
- }
if (selfpipe_trapset(&set) < 0) strerr_diefu1sys(111, "trap signals") ;
}
if (notif)
@@ -572,8 +581,7 @@ int main (int argc, char const *const *argv)
errno = EIO ;
panic("check internal pipes") ;
}
- if (x[0].revents & IOPAUSE_READ)
- divertsignals ? handle_diverted_signals() : handle_signals() ;
+ if (x[0].revents & IOPAUSE_READ) handle_signals() ;
if (x[1].revents & IOPAUSE_READ) handle_control(x[1].fd) ;
}
}
@@ -581,14 +589,16 @@ int main (int argc, char const *const *argv)
/* Finish phase. */
- selfpipe_finish() ;
killthem() ;
+ closethem() ;
restore_console() ;
- reap() ;
+ if (waitall) waitthem() ; else { chld() ; reap() ; }
+ selfpipe_finish() ;
}
{
- char const *eargv[3] = { FINISH_PROG, finish_arg, 0 } ;
+ char const *eargv[2] = { FINISH_PROG, 0 } ;
execv(eargv[0], (char **)eargv) ;
+ if (errno != ENOENT) panicnosp("exec finish script " FINISH_PROG) ;
}
- panicnosp("exec finish script " FINISH_PROG) ;
+ _exit(0) ;
}
generated by cgit v1.2.3 (git 2.25.1) at 2024-11-10 07:48:30 +0000