static void trystart (void)
{
int p[2] ;
pid_t pid ;
if (pipecoe(p) < 0)
{
settimeout(60) ;
strerr_warnwu1sys("pipecoe (waiting 60 seconds)") ;
return ;
}
pid = fork() ;
if (pid < 0)
{
settimeout(60) ;
strerr_warnwu1sys("fork (waiting 60 seconds)") ;
fd_close(p[1]) ; fd_close(p[0]) ;
return ;
}
else if (!pid)
{
char const *cargv[2] = { "run", 0 } ;
PROG = "s6-supervise (child)" ;
selfpipe_finish() ;
fd_close(p[0]) ;
if (unlink(S6_SUPERVISE_READY_FILENAME) < 0 && errno != ENOENT)
strerr_warnwu1sys("unlink " S6_SUPERVISE_READY_FILENAME) ;
if (flagsetsid) setsid() ;
execve("./run", (char *const *)cargv, (char *const *)environ) ;
fd_write(p[1], "", 1) ;
strerr_dieexec(127, "run") ;
}
fd_close(p[1]) ;
{
char c ;
switch (fd_read(p[0], &c, 1))
{
case -1 :
fd_close(p[0]) ;
settimeout(60) ;
strerr_warnwu1sys("read pipe (waiting 60 seconds)") ;
kill(pid, SIGKILL) ;
return ;
case 1 :
{
fd_close(p[0]) ;
settimeout(10) ;
strerr_warnwu1x("spawn ./run - waiting 10 seconds") ;
return ;
}
}
}
fd_close(p[0]) ;
settimeout_infinite() ;
state = UP ;
status.pid = pid ;
tain_copynow(&status.stamp) ;
announce() ;
ftrigw_notifyb_nosig(S6_SUPERVISE_EVENTDIR, "u", 1) ;
}
static void panicnosp (char const *errmsg)
{
char const *eargv[2] = { CRASH_PROG, 0 } ;
strerr_warnwu1sys(errmsg) ;
strerr_warnw2x("executing into ", eargv[0]) ;
execve(eargv[0], (char *const *)eargv, (char *const *)environ) ;
/* and if that execve fails, screw it and just die */
strerr_dieexec(111, eargv[0]) ;
}
static void scan (void)
{
unsigned int i = 0 ;
DIR *dir ;
if (!wantscan) return ;
wantscan = 0 ;
dir = opendir(".") ;
if (!dir)
{
strerr_warnwu1sys("opendir .") ;
retrydirlater() ;
return ;
}
for (; i < n ; i++) services[i].flagactive = 0 ;
for (;;)
{
direntry *d ;
errno = 0 ;
d = readdir(dir) ;
if (!d) break ;
check(d->d_name) ;
}
if (errno)
{
strerr_warnwu1sys("readdir .") ;
retrydirlater() ;
}
dir_close(dir) ;
for (i = 0 ; i < n ; i++) if (!services[i].flagactive && !services[i].pid[0])
{
if (services[i].flaglog)
{
if (services[i].pid[1]) continue ;
if (services[i].p[0] >= 0)
{
fd_close(services[i].p[1]) ; services[i].p[1] = -1 ;
fd_close(services[i].p[0]) ; services[i].p[0] = -1 ;
}
}
services[i] = services[--n] ;
}
}
static void uplastup_z (int islast)
{
status.wstat = status.pid ;
status.pid = 0 ;
tain_copynow(&status.stamp) ;
tryfinish(islast) ;
announce() ;
ftrigw_notifyb_nosig(S6_SUPERVISE_EVENTDIR, "d", 1) ;
if (unlink(S6_SUPERVISE_READY_FILENAME) < 0 && errno != ENOENT)
strerr_warnwu1sys("unlink " S6_SUPERVISE_READY_FILENAME) ;
settimeout(5) ;
}
static void check (char const *name)
{
struct stat st ;
unsigned int namelen ;
unsigned int i = 0 ;
if (name[0] == '.') return ;
if (stat(name, &st) == -1)
{
strerr_warnwu2sys("stat ", name) ;
retrydirlater() ;
return ;
}
if (!S_ISDIR(st.st_mode)) return ;
namelen = str_len(name) ;
for (; i < n ; i++) if ((services[i].ino == st.st_ino) && (services[i].dev == st.st_dev)) break ;
if (i < n)
{
if (services[i].flaglog && (services[i].p[0] < 0))
{
/* See BLACK MAGIC above. */
services[i].p[0] = -2 ;
return ;
}
}
else
{
if (n >= max)
{
strerr_warnwu3x("start supervisor for ", name, ": too many services") ;
return ;
}
else
{
struct stat su ;
char tmp[namelen + 5] ;
byte_copy(tmp, namelen, name) ;
byte_copy(tmp + namelen, 5, "/log") ;
if (stat(tmp, &su) < 0)
if (errno == ENOENT) services[i].flaglog = 0 ;
else
{
strerr_warnwu2sys("stat ", tmp) ;
retrydirlater() ;
return ;
}
else if (!S_ISDIR(su.st_mode))
services[i].flaglog = 0 ;
else
{
if (pipecoe(services[i].p) < 0)
{
strerr_warnwu1sys("pipecoe") ;
retrydirlater() ;
return ;
}
services[i].flaglog = 1 ;
}
services[i].ino = st.st_ino ;
services[i].dev = st.st_dev ;
tain_copynow(&services[i].restartafter[0]) ;
tain_copynow(&services[i].restartafter[1]) ;
services[i].pid[0] = 0 ;
services[i].pid[1] = 0 ;
n++ ;
}
}
services[i].flagactive = 1 ;
if (services[i].flaglog && !services[i].pid[1])
{
if (!tain_future(&services[i].restartafter[1]))
{
char tmp[namelen + 5] ;
byte_copy(tmp, namelen, name) ;
byte_copy(tmp + namelen, 5, "/log") ;
trystart(i, tmp, 1) ;
}
else if (tain_less(&services[i].restartafter[1], &deadline))
deadline = services[i].restartafter[1] ;
}
if (!services[i].pid[0])
{
if (!tain_future(&services[i].restartafter[0]))
trystart(i, name, 0) ;
else if (tain_less(&services[i].restartafter[0], &deadline))
deadline = services[i].restartafter[0] ;
}
}
static int doit (char const *s, unsigned int len)
{
if (delimlen)
{
if (!len--)
{
switch (strictness)
{
case 1 :
case 2 :
strerr_warnw1x("empty line") ;
break ;
case 3 :
buffer_flush(buffer_1) ;
strerr_dief1x(100, "empty line") ;
default : break ;
}
return 1 ;
}
if (byte_chr(delim, delimlen, *s) >= delimlen)
{
switch (strictness)
{
case 0 : return 0 ;
case 1 :
{
strerr_warnw1x("invalid starting quote character") ;
return 0 ;
}
case 2 :
{
char fmt[40] ;
unsigned int n = len < 39 ? len+1 : 36 ;
byte_copy(fmt, n, s) ;
if (len >= 39)
{
byte_copy(fmt+n, 3, "...") ;
n += 3 ;
}
fmt[n] = 0 ;
strerr_warnw3x("invalid starting quote character", " in line: ", fmt) ;
return 0 ;
}
case 3 :
{
buffer_flush(buffer_1) ;
strerr_dief1x(100, "invalid starting quote character") ;
}
default : strerr_dief1x(101, "can't happen: unknown strictness") ;
}
}
}
{
unsigned int r, w ;
char d[len] ;
if (!string_unquote_withdelim(d, &w, s + !!delimlen, len, &r, delim, delimlen))
{
switch (strictness)
{
case 0 : return 0 ;
case 1 :
{
strerr_warnwu1sys("unquote") ;
return 0 ;
}
case 2 :
{
char fmt[40] ;
unsigned int n = (len + !!delimlen) < 40 ? (len + !!delimlen) : 36 ;
byte_copy(fmt, n, s) ;
if ((len + !!delimlen) >= 40)
{
byte_copy(fmt+n, 3, "...") ;
n += 3 ;
}
fmt[n] = 0 ;
strerr_warnwu3sys("unquote", " line: ", fmt) ;
return 0 ;
}
case 3 :
{
int e = errno ;
buffer_flush(buffer_1) ;
errno = e ;
strerr_diefu1sys(100, "unquote") ;
}
default : strerr_dief1x(101, "can't happen: unknown strictness") ;
}
}
if (delimlen)
{
if (r == len)
{
switch (strictness)
{
case 0 : return 0 ;
case 1 :
{
strerr_warnwu2x("unquote", ": no ending quote character") ;
return 0 ;
}
case 2 :
{
char fmt[40] ;
unsigned int n = len < 40 ? len : 36 ;
byte_copy(fmt, n, s) ;
if (len >= 40)
{
byte_copy(fmt+n, 3, "...") ;
n += 3 ;
}
fmt[n] = 0 ;
strerr_warnwu5x("unquote", ": no ending quote character", " in ", "line: ", fmt) ;
return 0 ;
}
case 3 :
{
int e = errno ;
buffer_flush(buffer_1) ;
errno = e ;
strerr_diefu2x(100, "unquote", ": no ending quote character") ;
}
default : strerr_dief1x(101, "can't happen: unknown strictness") ;
}
}
else if ((r < len-1) && (strictness >= 2))
{
char fmtnum[UINT_FMT] ;
char fmtden[UINT_FMT] ;
char fmt[40] ;
unsigned int n = len < 39 ? len+1 : 36 ;
byte_copy(fmt, n, s) ;
if (len >= 39)
{
byte_copy(fmt+n, 3, "...") ;
n += 3 ;
}
fmt[n] = 0 ;
fmtnum[uint_fmt(fmtnum, r+1)] = 0 ;
fmtden[uint_fmt(fmtden, len)] = 0 ;
strerr_warnw7x("found ending quote character at position ", fmtnum, "/", fmtden, ", ignoring remainder of ", "line: ", fmt) ;
}
}
if (buffer_putalign(buffer_1, d, w) < (int)w)
strerr_diefu1sys(111, "write to stdout") ;
}
return 1 ;
}
static inline void announce (void)
{
if (!s6_svstatus_write(".", &status))
strerr_warnwu1sys("write status file") ;
}
