int main (int argc, char const *const *argv, char const *const *envp)
{
unsigned int backlog = 20 ;
int flagreuse = 1 ;
PROG = "s6-ipcserver-socketbinder" ;
{
subgetopt_t l = SUBGETOPT_ZERO ;
for (;;)
{
register int opt = subgetopt_r(argc, argv, "Ddb:", &l) ;
if (opt == -1) break ;
switch (opt)
{
case 'D' : flagreuse = 0 ; break ;
case 'd' : flagreuse = 1 ; break ;
case 'b' : if (!uint0_scan(l.arg, &backlog)) dieusage() ; break ;
default : dieusage() ;
}
}
argc -= l.ind ; argv += l.ind ;
}
if (argc < 2) dieusage() ;
close(0) ;
if (ipc_stream()) strerr_diefu1sys(111, "create socket") ;
{
mode_t m = umask(0) ;
if ((flagreuse ? ipc_bind_reuse(0, argv[0]) : ipc_bind(0, argv[0])) < 0)
strerr_diefu2sys(111, "bind to ", argv[0]) ;
umask(m) ;
}
if (ipc_listen(0, backlog) < 0) strerr_diefu2sys(111, "listen to ", argv[0]) ;
pathexec_run(argv[1], argv + 1, envp) ;
strerr_dieexec(111, argv[1]) ;
}
int main (int argc, char const *const *argv)
{
stralloc sa = STRALLOC_ZERO ;
int nl = 1 ;
PROG = "s6-dirname" ;
{
subgetopt_t l = SUBGETOPT_ZERO ;
for (;;)
{
register int opt = subgetopt_r(argc, argv, "n", &l) ;
if (opt == -1) break ;
switch (opt)
{
case 'n' : nl = 0 ; break ;
default : strerr_dieusage(100, USAGE) ;
}
}
argc -= l.ind ; argv += l.ind ;
}
if (!argc) strerr_dieusage(100, USAGE) ;
if (!sadirname(&sa, argv[0], str_len(argv[0])))
strerr_diefu2sys(111, "get dirname of ", argv[0]) ;
if (nl && !stralloc_catb(&sa, "\n", 1))
strerr_diefu2sys(111, "get dirname of ", argv[0]) ;
if (allwrite(1, sa.s, sa.len) < sa.len)
strerr_diefu1sys(111, "write to stdout") ;
return 0 ;
}
int main (int argc, char const *const *argv, char const *const *envp)
{
int fd, fd2 ;
unsigned int flags = 0 ;
int what = -1 ;
int changemode = 0 ;
PROG = "redirfd" ;
{
subgetopt_t l = SUBGETOPT_ZERO ;
for (;;)
{
register int opt = subgetopt_r(argc, argv, "rwuacxnb", &l) ;
if (opt == -1) break ;
switch (opt)
{
case 'r' : what = O_RDONLY ; flags &= ~(O_APPEND|O_CREAT|O_TRUNC|O_EXCL) ; break ;
case 'w' : what = O_WRONLY ; flags |= O_CREAT|O_TRUNC ; flags &= ~(O_APPEND|O_EXCL) ; break ;
case 'u' : what = O_RDWR ; flags &= ~(O_APPEND|O_CREAT|O_TRUNC|O_EXCL) ; break ;
case 'a' : what = O_WRONLY ; flags |= O_CREAT|O_APPEND ; flags &= ~(O_TRUNC|O_EXCL) ; break ;
case 'c' : what = O_WRONLY ; flags |= O_APPEND ; flags &= ~(O_CREAT|O_TRUNC|O_EXCL) ; break ;
case 'x' : what = O_WRONLY ; flags |= O_CREAT|O_EXCL ; flags &= ~(O_APPEND|O_TRUNC) ; break ;
case 'n' : flags |= O_NONBLOCK ; break ;
case 'b' : changemode = 1 ; break ;
default : dieusage() ;
}
}
argc -= l.ind ; argv += l.ind ;
}
if ((argc < 3) || (what == -1)) dieusage() ;
if (!uint0_scan(argv[0], (unsigned int *)&fd)) dieusage() ;
flags |= what ;
fd2 = open3(argv[1], flags, 0666) ;
if ((fd2 == -1) && (what == O_WRONLY) && (errno == ENXIO))
{
register int e ;
int fdr = open_read(argv[1]) ;
if (fdr == -1) strerr_diefu2sys(111, "open_read ", argv[1]) ;
fd2 = open3(argv[1], flags, 0666) ;
e = errno ;
fd_close(fdr) ;
errno = e ;
}
if (fd2 == -1) strerr_diefu2sys(111, "open ", argv[1]) ;
if (fd_move(fd, fd2) == -1)
{
char fmt[UINT_FMT] ;
fmt[uint_fmt(fmt, fd2)] = 0 ;
strerr_diefu4sys(111, "move fd ", fmt, " to fd ", argv[0]) ;
}
if (changemode)
{
if (((flags & O_NONBLOCK) ? ndelay_off(fd) : ndelay_on(fd)) < 0)
strerr_diefu1sys(111, "change blocking mode") ;
}
pathexec_run(argv[2], argv+2, envp) ;
strerr_dieexec(111, argv[2]) ;
}
int main (int argc, char const *const *argv)
{
s6_svstatus_t status ;
char fmt[UINT_FMT] ;
int isup, normallyup ;
PROG = "s6-svstat" ;
if (argc < 2) strerr_dieusage(100, USAGE) ;
argv++ ; argc-- ;
if (!s6_svstatus_read(*argv, &status))
strerr_diefu2sys(111, "read status for ", *argv) ;
{
struct stat st ;
unsigned int dirlen = str_len(*argv) ;
char fn[dirlen + 6] ;
byte_copy(fn, dirlen, *argv) ;
byte_copy(fn + dirlen, 6, "/down") ;
if (stat(fn, &st) == -1)
if (errno != ENOENT) strerr_diefu2sys(111, "stat ", fn) ;
else normallyup = 1 ;
else normallyup = 0 ;
}
taia_now_g() ;
if (taia_future(&status.stamp)) taia_copynow(&status.stamp) ;
taia_sub(&status.stamp, &STAMP, &status.stamp) ;
isup = status.pid && !status.flagfinishing ;
if (isup)
{
buffer_putnoflush(buffer_1small,"up (pid ", 8) ;
buffer_putnoflush(buffer_1small, fmt, uint_fmt(fmt, status.pid)) ;
buffer_putnoflush(buffer_1small, ") ", 2) ;
}
else buffer_putnoflush(buffer_1small, "down ", 5) ;
buffer_putnoflush(buffer_1small, fmt, uint64_fmt(fmt, status.stamp.sec.x)) ;
buffer_putnoflush(buffer_1small," seconds", 8) ;
if (isup && !normallyup)
buffer_putnoflush(buffer_1small, ", normally down", 15) ;
if (!isup && normallyup)
buffer_putnoflush(buffer_1small, ", normally up", 13) ;
if (isup && status.flagpaused)
buffer_putnoflush(buffer_1small, ", paused", 8) ;
if (!isup && (status.flagwant == 'u'))
buffer_putnoflush(buffer_1small, ", want up", 10) ;
if (isup && (status.flagwant == 'd'))
buffer_putnoflush(buffer_1small, ", want down", 12) ;
if (buffer_putflush(buffer_1small, "\n", 1) == -1)
strerr_diefu1sys(111, "write to stdout") ;
return 0 ;
}
int main (int argc, char const *const *argv)
{
mode_t mode = 0 ;
unsigned int m ;
PROG = "s6-chmod" ;
if (argc < 3) strerr_dieusage(100, USAGE) ;
if (!uint0_oscan(argv[1], &m)) strerr_dieusage(100, USAGE) ;
if (m & 0001) mode |= S_IXOTH ;
if (m & 0002) mode |= S_IWOTH ;
if (m & 0004) mode |= S_IROTH ;
if (m & 0010) mode |= S_IXGRP ;
if (m & 0020) mode |= S_IWGRP ;
if (m & 0040) mode |= S_IRGRP ;
if (m & 0100) mode |= S_IXUSR ;
if (m & 0200) mode |= S_IWUSR ;
if (m & 0400) mode |= S_IRUSR ;
if (m & 01000) mode |= S_ISVTX ;
if (m & 02000) mode |= S_ISGID ;
if (m & 04000) mode |= S_ISUID ;
if (chmod(argv[2], mode) == -1)
strerr_diefu2sys(111, "change mode of ", argv[2]) ;
return 0 ;
}
int main (int argc, char const *const *argv)
{
unsigned int mode = S_IRUSR|S_IWUSR|S_IRGRP|S_IWGRP|S_IROTH|S_IWOTH ;
PROG = "s6-mkfifo" ;
{
subgetopt_t l = SUBGETOPT_ZERO ;
for (;;)
{
register int opt = subgetopt_r(argc, argv, "m:", &l) ;
if (opt == -1) break ;
switch (opt)
{
case 'm': if (uint0_oscan(l.arg, &mode)) break ;
default : strerr_dieusage(100, USAGE) ;
}
}
argc -= l.ind ; argv += l.ind ;
}
if (!argc) strerr_dieusage(100, USAGE) ;
umask(S_IXUSR|S_IXGRP|S_IXOTH) ;
for (; *argv ; argv++)
if (mkfifo(*argv, mode) < 0)
strerr_diefu2sys(111, "mkfifo ", *argv) ;
return 0 ;
}
int main (int argc, char const *const *argv, char const *const *envp)
{
int incr = 10 ;
int strict = 0 ;
PROG = "s6-nice" ;
{
subgetopt_t l = SUBGETOPT_ZERO ;
for (;;)
{
register int opt = subgetopt_r(argc, argv, "Iin:", &l) ;
if (opt == -1) break ;
switch (opt)
{
case 'I' : strict = 0 ; break ;
case 'i' : strict = 1 ; break ;
case 'n': if (!int_scan(l.arg, &incr)) strerr_dieusage(100, USAGE) ; break ;
default : strerr_dieusage(100, USAGE) ;
}
}
argc -= l.ind ; argv += l.ind ;
}
if (!argc) strerr_dieusage(100, USAGE) ;
errno = 0 ;
if ((nice(incr) < 0) && errno)
{
char fmt[1+UINT_FMT] ;
fmt[int_fmt(fmt, incr)] = 0 ;
if (strict) strerr_diefu2sys(111, "nice to ", fmt) ;
else strerr_warnwu2sys("nice to ", fmt) ;
}
pathexec_run(argv[0], argv, envp) ;
strerr_dieexec((errno == ENOENT) ? 127 : 126, argv[0]) ;
}
void el_execsequence (char const *const *argv1, char const *const *argv2, char const *const *envp)
{
if (!argv2[0])
{
pathexec0_run(argv1, envp) ;
strerr_dieexec(111, argv1[0]) ;
}
else
{
int wstat ;
unsigned int j = 2 ;
#ifdef EXECLINE_OLD_VARNAMES
char fmt[UINT_FMT * 2 + 15] = "?=" ;
#else
char fmt[UINT_FMT + 1] = "?=" ;
#endif
pid_t pid = el_spawn0(argv1[0], argv1, envp) ;
if (!pid) strerr_warnwu2sys("spawn ", argv1[0]) ;
if (wait_pid(pid, &wstat) < 0)
strerr_diefu2sys(111, "wait for ", argv1[0]) ;
j += uint_fmt(fmt + j, wait_status(wstat)) ; fmt[j++] = 0 ;
#ifdef EXECLINE_OLD_VARNAMES
byte_copy(fmt + j, 13, "LASTEXITCODE=") ; j += 13 ;
j += uint_fmt(fmt + j, wait_status(wstat)) ; fmt[j++] = 0 ;
#endif
pathexec_r(argv2, envp, env_len(envp), fmt, j) ;
}
strerr_dieexec(111, argv2[0]) ;
}
int main (int argc, char const *const *argv)
{
stralloc sa = STRALLOC_ZERO ;
int path = 0, nl = 1 ;
PROG = "s6-linkname" ;
{
subgetopt_t l = SUBGETOPT_ZERO ;
for (;;)
{
register int opt = subgetopt_r(argc, argv, "nf", &l) ;
if (opt == -1) break ;
switch(opt)
{
case 'n' : nl = 0 ; break ;
case 'f' : path = 1 ; break ;
default : dieusage() ;
}
}
argv += l.ind ; argc -= l.ind ;
}
if (!argc) dieusage() ;
if ((path ? sarealpath(&sa, *argv) : sareadlink(&sa, *argv)) == -1)
strerr_diefu2sys(111, "resolve ", *argv) ;
if ((buffer_putalign(buffer_1small, sa.s, sa.len) == -1)
|| (nl && (buffer_putalign(buffer_1small, "\n", 1)) == -1)
|| (buffer_flush(buffer_1small) == -1))
strerr_diefu1sys(111, "write to stdout") ;
/* stralloc_free(&sa) ; */
return 0 ;
}
int s6_svc_main (int argc, char const *const *argv, char const *optstring, char const *usage, char const *controldir)
{
char data[DATASIZE] ;
unsigned int datalen = 0 ;
register int r ;
for (;;)
{
register int opt = subgetopt(argc, argv, optstring) ;
if (opt == -1) break ;
if (opt == '?') strerr_dieusage(100, usage) ;
if (datalen >= DATASIZE) strerr_dief1x(100, "too many commands") ;
data[datalen++] = opt ;
}
argc -= subgetopt_here.ind ; argv += subgetopt_here.ind ;
if (!argc) strerr_dieusage(100, usage) ;
{
unsigned int arglen = str_len(*argv) ;
unsigned int cdirlen = str_len(controldir) ;
char tmp[arglen + cdirlen + 10] ;
byte_copy(tmp, arglen, *argv) ;
tmp[arglen] = '/' ;
byte_copy(tmp + arglen + 1, cdirlen, controldir) ;
byte_copy(tmp + arglen + 1 + cdirlen, 9, "/control") ;
r = s6_svc_write(tmp, data, datalen) ;
}
if (r < 0) strerr_diefu2sys(111, "control ", *argv) ;
else if (!r) strerr_diefu3x(100, "control ", *argv, ": supervisor not listening") ;
return 0 ;
}
int main (int argc, char const *const *argv)
{
PROG = "s6-cleanfifodir" ;
if (argc < 2) strerr_dieusage(100, USAGE) ;
if (!ftrigw_clean(argv[1]))
strerr_diefu2sys(111, "clean up fifodir at ", argv[1]) ;
return 0 ;
}
int main (int argc, char const *const *argv, char const *const *envp)
{
int df = 0 ;
PROG = "heredoc" ;
{
subgetopt_t l = SUBGETOPT_ZERO ;
for (;;)
{
register int opt = subgetopt_r(argc, argv, "d", &l) ;
if (opt == -1) break ;
switch (opt)
{
case 'd' : df = 1 ; break ;
default : dieusage() ;
}
}
argc -= l.ind ; argv += l.ind ;
}
if (argc < 3) dieusage() ;
{
int fd[2] ;
unsigned int fdr ;
int pid ;
if (!uint0_scan(argv[0], &fdr)) strerr_dieusage(100, USAGE) ;
if (pipe(fd) < 0) strerr_diefu1sys(111, "pipe") ;
pid = df ? doublefork() : fork() ;
switch (pid)
{
case -1: strerr_diefu2sys(111, df ? "double" : "", "fork") ;
case 0:
{
unsigned int len = str_len(argv[1]) ;
PROG = "heredoc (child)" ;
fd_close(fd[0]) ;
if (allwrite(fd[1], argv[1], len) < len)
strerr_diefu1sys(111, "allwrite") ;
return 0 ;
}
}
fd_close(fd[1]) ;
if (fd_move((int)fdr, fd[0]) == -1)
strerr_diefu2sys(111, "read on fd ", argv[0]) ;
}
pathexec_run(argv[2], argv+2, envp) ;
strerr_dieexec(111, argv[2]) ;
}
int main (int argc, char const *const *argv, char const *const *envp)
{
PROG = "cd" ;
if (argc < 3) strerr_dieusage(100, USAGE) ;
if (chdir(argv[1]) == -1)
strerr_diefu2sys(111, "chdir to ", argv[1]) ;
pathexec_run(argv[2], argv+2, envp) ;
strerr_dieexec(111, argv[2]) ;
}
int main (int argc, char const *const *argv)
{
int fd ;
PROG = "s6-freeramdisk" ;
if (argc < 2) strerr_dieusage(100, USAGE) ;
fd = open(argv[1], O_RDWR) ;
if (fd < 0) strerr_diefu3sys(111, "open ", argv[1], " in read-write mode") ;
if (ioctl(fd, BLKFLSBUF) < 0) strerr_diefu2sys(111, "ioctl ", argv[1]) ;
return 0 ;
}
static inline void notify_systemd (pid_t pid, char const *socketpath)
{
size_t n = 16 ;
char fmt[16 + PID_FMT] = "READY=1\nMAINPID=" ;
int fd = ipc_datagram_b() ;
if (fd < 0) strerr_diefu1sys(111, "create socket") ;
n += pid_fmt(fmt + n, pid) ;
fmt[n++] = '\n' ;
if (!ipc_sendto(fd, fmt, n, socketpath))
strerr_diefu2sys(111, "send notification message to ", socketpath) ;
close(fd) ;
}
int main (int argc, char const *const *argv, char const *const *envp)
{
int uid = -1, gid = -1 ;
PROG = "s6-chown" ;
{
subgetopt_t l = SUBGETOPT_ZERO ;
for (;;)
{
register int opt = subgetopt_r(argc, argv, "Uu:g:", &l) ;
if (opt == -1) break ;
switch (opt)
{
case 'u':
{
unsigned int u ;
if (!uint0_scan(l.arg, &u)) strerr_dieusage(100, USAGE) ;
uid = u ;
break ;
}
case 'g':
{
unsigned int g ;
if (!uint0_scan(l.arg, &g)) strerr_dieusage(100, USAGE) ;
gid = g ;
break ;
}
case 'U':
{
unsigned int x ;
char const *s = env_get2(envp, "UID") ;
if (!s) strerr_dienotset(100, "UID") ;
if (!uint0_scan(s, &x)) strerr_dieinvalid(100, "UID") ;
uid = x ;
s = env_get2(envp, "GID") ;
if (!s) strerr_dienotset(100, "GID") ;
if (!uint0_scan(s, &x)) strerr_dieinvalid(100, "GID") ;
gid = x ;
break ;
}
default : strerr_dieusage(100, USAGE) ;
}
}
argc -= l.ind ; argv += l.ind ;
}
if (!argc) strerr_dieusage(100, USAGE) ;
if (chown(*argv, uid, gid) == -1)
strerr_diefu2sys(111, "chown ", argv[0]) ;
return 0 ;
}
static void trystart (unsigned int i, char const *name, int islog)
{
int pid = fork() ;
switch (pid)
{
case -1 :
tain_addsec_g(&services[i].restartafter[islog], CHECK_RETRY_TIMEOUT) ;
strerr_warnwu2sys("fork for ", name) ;
return ;
case 0 :
{
char const *cargv[3] = { "s6-supervise", name, 0 } ;
PROG = "s6-svscan (child)" ;
selfpipe_finish() ;
if (services[i].flaglog)
if (fd_move(!islog, services[i].p[!islog]) == -1)
strerr_diefu2sys(111, "set fds for ", name) ;
pathexec_run(S6_BINPREFIX "s6-supervise", cargv, (char const **)environ) ;
strerr_dieexec(111, S6_BINPREFIX "s6-supervise") ;
}
}
services[i].pid[islog] = pid ;
}
void s6_svlisten_init (int argc, char const *const *argv, s6_svlisten_t *foo, uint16 *ids, unsigned char *upstate, unsigned char *readystate, tain_t const *deadline)
{
register unsigned int i = 0 ;
foo->n = (unsigned int)argc ;
foo->ids = ids ;
foo->upstate = upstate ;
foo->readystate = readystate ;
if (!ftrigr_startf_g(&foo->a, deadline)) strerr_diefu1sys(111, "ftrigr_startf") ;
for (; i < foo->n ; i++)
{
s6_svstatus_t status = S6_SVSTATUS_ZERO ;
unsigned int len = str_len(argv[i]) ;
char s[len + 1 + sizeof(S6_SUPERVISE_EVENTDIR)] ;
byte_copy(s, len, argv[i]) ;
s[len] = '/' ;
byte_copy(s + len + 1, sizeof(S6_SUPERVISE_EVENTDIR), S6_SUPERVISE_EVENTDIR) ;
foo->ids[i] = ftrigr_subscribe_g(&foo->a, s, "D|u|U|d", FTRIGR_REPEAT, deadline) ;
if (!foo->ids[i]) strerr_diefu2sys(111, "subscribe to events for ", argv[i]) ;
if (!s6_svstatus_read(argv[i], &status)) strerr_diefu1sys(111, "s6_svstatus_read") ;
bitarray_poke(foo->upstate, i, status.pid && !status.flagfinishing) ;
bitarray_poke(foo->readystate, i, status.flagready) ;
}
}
int main (int argc, char const *const *argv)
{
tain_t deadline, tto ;
ftrigr_t a = FTRIGR_ZERO ;
uint16 id ;
char pack[2] = " \n" ;
PROG = "s6-ftrig-wait" ;
{
unsigned int t = 0 ;
for (;;)
{
register int opt = subgetopt(argc, argv, "t:") ;
if (opt == -1) break ;
switch (opt)
{
case 't' : if (uint0_scan(subgetopt_here.arg, &t)) break ;
default : strerr_dieusage(100, USAGE) ;
}
}
if (t) tain_from_millisecs(&tto, t) ;
else tto = tain_infinite_relative ;
argc -= subgetopt_here.ind ; argv += subgetopt_here.ind ;
}
if (argc < 2) strerr_dieusage(100, USAGE) ;
tain_now_g() ;
tain_add_g(&deadline, &tto) ;
if (!ftrigr_startf_g(&a, &deadline)) strerr_diefu1sys(111, "ftrigr_startf") ;
id = ftrigr_subscribe_g(&a, argv[0], argv[1], 0, &deadline) ;
if (!id) strerr_diefu4sys(111, "subscribe to ", argv[0], " with regexp ", argv[1]) ;
if (ftrigr_wait_or_g(&a, &id, 1, &deadline, &pack[0]) == -1)
strerr_diefu2sys((errno == ETIMEDOUT) ? 1 : 111, "match regexp on ", argv[1]) ;
if (allwrite(1, pack, 2) < 2) strerr_diefu1sys(111, "write to stdout") ;
return 0 ;
}
int main (int argc, char const **argv, char const *const *envp)
{
char const *x ;
int argc1 ;
unsigned short okcodes[256] ;
unsigned int nbc = 0 ;
int flagpar = 0, not = 1 ;
PROG = "forx" ;
{
subgetopt_t l = SUBGETOPT_ZERO ;
for (;;)
{
register int opt = subgetopt_r(argc, argv, "epo:x:", &l) ;
if (opt == -1) break ;
switch (opt)
{
case 'e' : break ; /* compat */
case 'p' : flagpar = 1 ; break ;
case 'o' :
not = 0 ;
if (!ushort_scanlist(okcodes, 256, l.arg, &nbc)) dieusage() ;
break ;
case 'x' :
not = 1 ;
if (!ushort_scanlist(okcodes, 256, l.arg, &nbc)) dieusage() ;
break ;
default : dieusage() ;
}
}
argc -= l.ind ; argv += l.ind ;
}
if (argc < 2) dieusage() ;
x = argv[0] ; if (!*x) dieusage() ;
argv++ ; argc-- ;
argc1 = el_semicolon(argv) ;
if (argc1 >= argc) strerr_dief1x(100, "unterminated block") ;
if (!argc1 || (argc1 + 1 == argc)) return 0 ;
{
unsigned int envlen = env_len(envp) ;
unsigned int varlen = str_len(x) ;
unsigned int i = 0 ;
pid_t pids[flagpar ? argc1 : 1] ;
char const *newenv[envlen + 2] ;
for (; i < (unsigned int)argc1 ; i++)
{
pid_t pid ;
unsigned int vallen = str_len(argv[i]) ;
char modif[varlen + vallen + 2] ;
byte_copy(modif, varlen, x) ;
modif[varlen] = '=' ;
byte_copy(modif + varlen + 1, vallen, argv[i]) ;
modif[varlen + vallen + 1] = 0 ;
if (!env_merge(newenv, envlen + 2, envp, envlen, modif, varlen + vallen + 2))
strerr_diefu1sys(111, "build new environment") ;
pid = el_spawn0(argv[argc1+1], argv + argc1 + 1, newenv) ;
if (!pid) strerr_diefu2sys(111, "spawn ", argv[argc1+1]) ;
if (flagpar) pids[i] = pid ;
else
{
int wstat ;
if (wait_pid(pid, &wstat) == -1)
strerr_diefu2sys(111, "wait for ", argv[argc1+1]) ;
if (not == isok(okcodes, nbc, wait_estatus(wstat)))
return wait_estatus(wstat) ;
}
}
if (flagpar)
if (!waitn(pids, argc1)) strerr_diefu1sys(111, "waitn") ;
}
return 0 ;
}
int main (int argc, char const *const *argv) {
tain_t deadline ;
int sfd ;
PROG = "bevt_relayd" ;
if (argc < 2) strerr_dieusage(100, USAGE) ;
if (ndelay_on(0) < 0) strerr_diefu2sys(111, "ndelay_on ", "0") ;
if (ndelay_on(1) < 0) strerr_diefu2sys(111, "ndelay_on ", "1") ;
if (sig_ignore(SIGPIPE) < 0) strerr_diefu1sys(111, "ignore SIGPIPE") ;
sfd = selfpipe_init() ;
if (sfd < 0) strerr_diefu1sys(111, "selfpipe_init") ;
{
sigset_t set ;
sigemptyset(&set) ;
sigaddset(&set, SIGCHLD) ;
sigaddset(&set, SIGTERM) ;
sigaddset(&set, SIGQUIT) ;
sigaddset(&set, SIGHUP) ;
sigaddset(&set, SIGABRT) ;
sigaddset(&set, SIGINT) ;
if (selfpipe_trapset(&set) < 0)
strerr_diefu1sys(111, "trap signals") ;
}
if(bevt_relay_db_init(argv[1])<0)
strerr_diefu1sys(111, "db init failed") ;
tain_now_g() ;
tain_addsec_g(&deadline, 2) ;
if (!skaclient_server_01x_init_g(BEVT_RELAY_BANNER1, BEVT_RELAY_BANNER1_LEN, BEVT_RELAY_BANNER2, BEVT_RELAY_BANNER2_LEN, &deadline))
strerr_diefu1sys(111, "sync with client") ;
for (;;) {
register unsigned int n = 0 ;
iopause_fd x[6 + n] ;
int r ;
if(mfd<0) handle_connect_central();
tain_add_g(&deadline, &tain_infinite_relative) ;
x[0].fd = 0 ; x[0].events = IOPAUSE_EXCEPT | IOPAUSE_READ ;
x[1].fd = 1 ; x[1].events = IOPAUSE_EXCEPT | (unixmessage_sender_isempty(unixmessage_sender_1) ? 0 : IOPAUSE_WRITE ) ;
x[2].fd = sfd ; x[2].events = IOPAUSE_READ ;
x[3].fd = bozclient_sfd(¢ral_client_g); x[3].events = IOPAUSE_READ ;
x[4].fd = bozclient_sfd(¢ral_client_g); x[4].events = (bozclient_siswritable(¢ral_client_g) ? IOPAUSE_WRITE : 0) ;
x[5].fd = unixmessage_sender_fd(unixmessage_sender_x) ; x[5].events = (unixmessage_sender_isempty(unixmessage_sender_x) ? 0 : IOPAUSE_WRITE) ;
r = iopause_g(x, 5 + n, &deadline) ;
if (r < 0) {
cleanup() ;
strerr_diefu1sys(111, "iopause") ;
}
/* client closed */
if ((x[0].revents | x[1].revents) & IOPAUSE_EXCEPT) break ;
/* client is sync reading */
if (x[1].revents & IOPAUSE_WRITE) {
if (!unixmessage_sender_flush(unixmessage_sender_1) && !error_isagain(errno)) {
cleanup() ;
strerr_diefu1sys(111, "flush sync out") ;
}
}
/* client is async rzading */
if (x[5].revents & IOPAUSE_WRITE) {
if (!unixmessage_sender_flush(unixmessage_sender_x) && !error_isagain(errno)) {
cleanup() ;
strerr_diefu1sys(111, "flush async out") ;
}
}
/* signals arrived */
if (x[2].revents & (IOPAUSE_READ | IOPAUSE_EXCEPT)) handle_signals() ;
/* main socket read or close */
if (x[3].revents & IOPAUSE_READ) {
handle_close_central();
continue;
}
/* main socket close */
if (x[4].revents & IOPAUSE_WRITE) {
bozclient_flush(¢ral_client_g);
}
/* main socket close */
if (x[5].revents & IOPAUSE_WRITE) {
if (!unixmessage_sender_flush(unixmessage_sender_x) && !error_isagain(errno)) {
cleanup() ;
strerr_diefu1sys(111, "flush stdout") ;
}
}
/* client is sync writing */
if (!unixmessage_receiver_isempty(unixmessage_receiver_0) || x[0].revents & IOPAUSE_READ)
{
if (unixmessage_handle(unixmessage_receiver_0, &bevt_relay_parse_prot_cmd, 0) < 0)
{
if (errno == EPIPE) break ; /* normal exit */
cleanup() ;
strerr_diefu1sys(111, "handle messages from client") ;
}
}
}
cleanup() ;
return 0 ;
}
int main (int argc, char const *const *argv, char const *const *envp)
{
char data[DATASIZE+1] = "-" ;
unsigned int datalen = 1 ;
unsigned int timeout = 0 ;
char updown[3] = "-\0" ;
PROG = "s6-svc" ;
{
subgetopt_t l = SUBGETOPT_ZERO ;
for (;;)
{
register int opt = subgetopt_r(argc, argv, "abqhkti12pcoduxOXyT:w:", &l) ;
if (opt == -1) break ;
switch (opt)
{
case 'a' :
case 'b' :
case 'q' :
case 'h' :
case 'k' :
case 't' :
case 'i' :
case '1' :
case '2' :
case 'p' :
case 'c' :
case 'o' :
case 'd' :
case 'u' :
case 'x' :
case 'O' :
case 'X' :
case 'y' :
{
if (datalen >= DATASIZE) strerr_dief1x(100, "too many commands") ;
data[datalen++] = opt ;
break ;
}
case 'T' : if (!uint0_scan(l.arg, &timeout)) dieusage() ; break ;
case 'w' :
{
if (byte_chr("dDuUrR", 6, l.arg[0]) >= 6) dieusage() ;
updown[1] = l.arg[0] ;
break ;
}
default : dieusage() ;
}
}
argc -= l.ind ; argv += l.ind ;
}
if (!argc) dieusage() ;
if (argc > 1) strerr_warn1x("ignoring extra arguments") ;
if (datalen <= 1) return 0 ;
if (updown[1] == 'U' || updown[1] == 'R')
{
unsigned int arglen = str_len(argv[0]) ;
char fn[arglen + 17] ;
byte_copy(fn, arglen, argv[0]) ;
byte_copy(fn + arglen, 17, "/notification-fd") ;
if (access(fn, F_OK) < 0)
{
if (errno != ENOENT) strerr_diefu2sys(111, "access ", fn) ;
updown[1] = updown[1] == 'U' ? 'u' : 'r' ;
strerr_warnw2x(fn, " not present - ignoring request for readiness notification") ;
}
}
if (updown[1])
{
char const *newargv[11] ;
unsigned int m = 0 ;
char fmt[UINT_FMT] ;
newargv[m++] = S6_BINPREFIX "s6-svlisten1" ;
newargv[m++] = updown ;
if (timeout)
{
fmt[uint_fmt(fmt, timeout)] = 0 ;
newargv[m++] = "-t" ;
newargv[m++] = fmt ;
}
newargv[m++] = "--" ;
newargv[m++] = argv[0] ;
newargv[m++] = S6_BINPREFIX "s6-svc" ;
newargv[m++] = data ;
newargv[m++] = "--" ;
newargv[m++] = argv[0] ;
newargv[m++] = 0 ;
pathexec_run(newargv[0], newargv, envp) ;
strerr_dieexec(111, newargv[0]) ;
}
else
{
register int r = s6_svc_writectl(argv[0], S6_SUPERVISE_CTLDIR, data + 1, datalen - 1) ;
if (r < 0) strerr_diefu2sys(111, "control ", argv[0]) ;
else if (!r) strerr_diefu3x(100, "control ", argv[0], ": supervisor not listening") ;
}
return 0 ;
}
int main (void)
{
PROG = "s6-ftrigrd" ;
if (ndelay_on(0) < 0) strerr_diefu2sys(111, "ndelay_on ", "0") ;
if (ndelay_on(1) < 0) strerr_diefu2sys(111, "ndelay_on ", "1") ;
if (sig_ignore(SIGPIPE) < 0) strerr_diefu1sys(111, "ignore SIGPIPE") ;
{
tain_t deadline ;
tain_now_g() ;
tain_addsec_g(&deadline, 2) ;
if (!skaclient_server_01x_init_g(FTRIGR_BANNER1, FTRIGR_BANNER1_LEN, FTRIGR_BANNER2, FTRIGR_BANNER2_LEN, &deadline))
strerr_diefu1sys(111, "sync with client") ;
}
for (;;)
{
iopause_fd x[3 + n] ;
unsigned int i = 0 ;
x[0].fd = 0 ; x[0].events = IOPAUSE_EXCEPT | IOPAUSE_READ ;
x[1].fd = 1 ; x[1].events = IOPAUSE_EXCEPT | (unixmessage_sender_isempty(unixmessage_sender_1) ? 0 : IOPAUSE_WRITE) ;
x[2].fd = unixmessage_sender_fd(unixmessage_sender_x) ;
x[2].events = IOPAUSE_EXCEPT | (unixmessage_sender_isempty(unixmessage_sender_x) ? 0 : IOPAUSE_WRITE) ;
for (; i < n ; i++)
{
a[i].xindex = 3 + i ;
x[3+i].fd = a[i].trig.fd ;
x[3+i].events = IOPAUSE_READ ;
}
if (iopause(x, 3 + n, 0, 0) < 0)
{
cleanup() ;
strerr_diefu1sys(111, "iopause") ;
}
/* client closed */
if ((x[0].revents | x[1].revents) & IOPAUSE_EXCEPT) break ;
/* client is reading */
if (x[1].revents & IOPAUSE_WRITE)
if (!unixmessage_sender_flush(unixmessage_sender_1) && !error_isagain(errno))
{
cleanup() ;
strerr_diefu1sys(111, "flush stdout") ;
}
if (x[2].revents & IOPAUSE_WRITE)
if (!unixmessage_sender_flush(unixmessage_sender_x) && !error_isagain(errno))
{
cleanup() ;
strerr_diefu1sys(111, "flush asyncout") ;
}
/* scan listening ftrigs */
for (i = 0 ; i < n ; i++)
{
if (x[a[i].xindex].revents & IOPAUSE_READ)
if (!ftrigio_read(a+i)) remove(i--) ;
}
/* client is writing */
if (!unixmessage_receiver_isempty(unixmessage_receiver_0) || x[0].revents & IOPAUSE_READ)
{
if (unixmessage_handle(unixmessage_receiver_0, &parse_protocol, 0) < 0)
{
if (errno == EPIPE) break ; /* normal exit */
cleanup() ;
strerr_diefu1sys(111, "handle messages from client") ;
}
}
}
cleanup() ;
return 0 ;
}
int main (int argc, char const *const *argv, char const *const *envp)
{
unsigned int nb = 0, ex = 1 ;
unsigned int timeout = 0 ;
PROG = "s6-setlock" ;
for (;;)
{
register int opt = subgetopt(argc, argv, "nNrwt:") ;
if (opt == -1) break ;
switch (opt)
{
case 'n' : nb = 1 ; break ;
case 'N' : nb = 0 ; break ;
case 'r' : ex = 0 ; break ;
case 'w' : ex = 1 ; break ;
case 't' : if (!uint0_scan(subgetopt_here.arg, &timeout)) dieusage() ;
nb = 2 ; break ;
default : dieusage() ;
}
}
argc -= subgetopt_here.ind ; argv += subgetopt_here.ind ;
if (argc < 2) dieusage() ;
if (nb < 2)
{
int fd = open_create(argv[0]) ;
if (fd == -1) strerr_diefu2sys(111, "open_create ", argv[0]) ;
if ((*f[ex][nb])(fd) == -1) strerr_diefu2sys(1, "lock ", argv[0]) ;
}
else
{
char const *cargv[3] = { "s6lockd-helper", argv[0], 0 } ;
char const *cenvp[2] = { ex ? "S6LOCK_EX=1" : 0, 0 } ;
iopause_fd x = { .events = IOPAUSE_READ } ;
tain_t deadline ;
int p[2] ;
unsigned int pid ;
char c ;
tain_now_g() ;
tain_from_millisecs(&deadline, timeout) ;
tain_add_g(&deadline, &deadline) ;
pid = child_spawn(S6_LIBEXECPREFIX "s6lockd-helper", cargv, cenvp, p, 2) ;
if (!pid) strerr_diefu2sys(111, "spawn ", S6_LIBEXECPREFIX "s6lockd-helper") ;
x.fd = p[0] ;
for (;;)
{
register int r = iopause_g(&x, 1, &deadline) ;
if (r < 0) strerr_diefu1sys(111, "iopause") ;
if (!r)
{
kill(pid, SIGTERM) ;
errno = ETIMEDOUT ;
strerr_diefu1sys(1, "acquire lock") ;
}
r = sanitize_read(fd_read(p[0], &c, 1)) ;
if (r < 0) strerr_diefu1sys(111, "read ack from helper") ;
if (r) break ;
}
if (c != '!') strerr_dief1x(111, "helper sent garbage ack") ;
fd_close(p[0]) ;
if (uncoe(p[1]) < 0) strerr_diefu1sys(111, "uncoe fd to helper") ;
}
pathexec_run(argv[1], argv+1, envp) ;
strerr_dieexec(111, argv[1]) ;
}
int main (int argc, char *const *argv)
{
stralloc3 blah = STRALLOC3_ZERO ;
PROG = "s6-update-symlinks" ;
if (argc < 3) strerr_dieusage(100, USAGE) ;
{
register char *const *p = argv + 1 ;
for (; *p ; p++) if (**p != '/') strerr_dieusage(100, USAGE) ;
}
{
register unsigned int i = str_len(argv[1]) ;
while (i && (argv[1][i-1] == '/')) argv[1][--i] = 0 ;
if (!i) strerr_diefu1x(100, "replace root directory") ;
}
if (!makeuniquename(&blah.dst, argv[1], MAGICNEW))
strerr_diefu2sys(111, "make random unique name based on ", argv[1]) ;
if ((unlink(blah.dst.s) == -1) && (errno != ENOENT))
strerr_diefu2sys(111, "unlink ", blah.dst.s) ;
{
char *const *p = argv + 2 ;
for (; *p ; p++)
{
register int r ;
blah.src.len = 0 ;
if (!stralloc_cats(&blah.src, *p) || !stralloc_0(&blah.src))
strerr_diefu1sys(111, "make stralloc") ;
r = addlink(&blah, 0, 0) ;
if (r < 0)
{
stralloc_free(&blah.tmp) ;
stralloc_free(&blah.src) ;
cleanup(&blah.dst, 0) ;
stralloc_free(&blah.dst) ;
if (r == CONFLICT)
strerr_dief4x(100, "destination ", errdst.s, " conflicts with source ", errsrc.s) ;
else
strerr_dief2sys(111, "error processing ", *p) ;
}
}
}
stralloc_free(&blah.tmp) ;
if (rename(blah.dst.s, argv[1]) == -1) /* be atomic if possible */
{
blah.src.len = 0 ;
if (!makeuniquename(&blah.src, argv[1], MAGICOLD))
{
cleanup(&blah.dst, 0) ;
strerr_diefu2sys(111, "make random unique name based on ", argv[1]) ;
}
if (rename(argv[1], blah.src.s) == -1)
{
cleanup(&blah.dst, 0) ;
strerr_diefu4sys(111, "rename ", argv[1], " to ", blah.src.s) ;
}
/* XXX: unavoidable race condition here: argv[1] does not exist */
if (rename(blah.dst.s, argv[1]) == -1)
{
rename(blah.src.s, argv[1]) ;
cleanup(&blah.dst, 0) ;
strerr_diefu4sys(111, "rename ", blah.dst.s, " to ", argv[1]) ;
}
stralloc_free(&blah.dst) ;
if (rm_rf_in_tmp(&blah.src, 0) == -1)
strerr_warnwu2sys("remove old directory ", blah.src.s) ;
stralloc_free(&blah.src) ;
}
return 0 ;
}
int main (int argc, char const *const *argv)
{
tain_t deadline, tto ;
unsigned int dirlen ;
char const *live = S6RC_LIVE_BASE ;
char const *compiled = S6RC_COMPILED_BASE ;
PROG = "s6-rc-init" ;
{
unsigned int t = 0 ;
subgetopt_t l = SUBGETOPT_ZERO ;
for (;;)
{
register int opt = subgetopt_r(argc, argv, "c:l:t:", &l) ;
if (opt == -1) break ;
switch (opt)
{
case 'c' : compiled = l.arg ; break ;
case 'l' : live = l.arg ; break ;
case 't' : if (!uint0_scan(l.arg, &t)) dieusage() ; break ;
default : dieusage() ;
}
}
argc -= l.ind ; argv += l.ind ;
if (t) tain_from_millisecs(&tto, t) ;
else tto = tain_infinite_relative ;
}
if (!argc) dieusage() ;
if (compiled[0] != '/')
strerr_dief2x(100, compiled, " is not an absolute path") ;
if (live[0] != '/')
strerr_dief2x(100, live, " is not an absolute path") ;
if (argv[0][0] != '/')
strerr_dief2x(100, argv[0], " is not an absolute path") ;
tain_now_g() ;
tain_add_g(&deadline, &tto) ;
if (!s6rc_sanitize_dir(&satmp, live, &dirlen)) dienomem() ;
llen = satmp.len ;
if (!stralloc_cats(&satmp, ":initial") || !stralloc_0(&satmp))
strerr_diefu1sys(111, "stralloc_catb") ;
{
int fdlock ;
int fdcompiled ;
int ok ;
s6rc_db_t db ;
unsigned int n ;
char lfn[llen + 13] ;
char cfn[llen + 23] ;
/* Create the real dir, lock it, symlink */
unlink(live) ;
rm_rf(satmp.s) ;
if (mkdir(satmp.s, 0755) < 0) strerr_diefu2sys(111, "mkdir ", satmp.s) ;
if (!s6rc_lock(satmp.s, 2, &fdlock, 0, 0, 0))
{
char tmp[satmp.len] ;
byte_copy(tmp, satmp.len, satmp.s) ;
cleanup() ;
strerr_diefu2sys(111, "take lock on ", tmp) ;
}
byte_copy(lfn, llen, satmp.s) ;
lfn[llen] = 0 ;
if (symlink(satmp.s + dirlen, lfn) < 0)
{
char tmp[satmp.len - dirlen] ;
byte_copy(tmp, satmp.len - dirlen, satmp.s + dirlen) ;
cleanup() ;
strerr_diefu4sys(111, "symlink ", tmp, " to ", lfn) ;
}
/* compiled */
fdcompiled = open_readb(compiled) ;
if (fdcompiled < 0)
{
cleanup() ;
strerr_diefu2sys(111, "open ", compiled) ;
}
byte_copy(lfn + llen, 10, "/compiled") ;
if (symlink(compiled, lfn) < 0)
{
cleanup() ;
strerr_diefu4sys(111, "symlink ", compiled, " to ", lfn) ;
}
/* scandir */
byte_copy(lfn + llen + 1, 8, "scandir") ;
if (symlink(argv[0], lfn) < 0)
{
cleanup() ;
strerr_diefu4sys(111, "symlink ", argv[0], " to ", lfn) ;
}
/* state */
byte_copy(lfn + llen + 1, 6, "state") ;
{
register int r = s6rc_db_read_sizes(fdcompiled, &db) ;
if (r <= 0)
{
cleanup() ;
if (r < 0) strerr_diefu2sys(111, "read database size in ", compiled) ;
else strerr_dief2x(4, "invalid database size in ", compiled) ;
}
close(fdcompiled) ;
n = db.nshort + db.nlong ;
{
char zero[n] ;
byte_zero(zero, n) ;
if (!openwritenclose_unsafe(lfn, zero, n))
{
cleanup() ;
strerr_diefu2sys(111, "write ", lfn) ;
}
}
}
/* servicedirs */
byte_copy(lfn + llen + 1, 12, "servicedirs") ;
byte_copy(cfn, llen + 1, lfn) ;
byte_copy(cfn + llen + 1, 21, "compiled/servicedirs") ;
if (!hiercopy(cfn, lfn))
{
cleanup() ;
strerr_diefu4sys(111, "recursively copy ", cfn, " to ", lfn) ;
}
/* start the supervisors */
lfn[llen] = 0 ;
ok = s6rc_servicedir_manage_g(lfn, &deadline) ;
if (!ok)
{
cleanup() ;
strerr_diefu3sys(111, "supervise service directories in ", lfn, "/servicedirs") ;
}
if (ok & 2)
strerr_warnw3x("s6-svscan not running on ", lfn, "/scandir") ;
}
return 0 ;
}
int main (int argc, char const **argv, char const *const *envp)
{
iopause_fd x[2] = { { -1, IOPAUSE_READ, 0 }, { -1, IOPAUSE_READ, 0 } } ;
tain_t deadline, tto ;
ftrigr_t a = FTRIGR_ZERO ;
int argc1 ;
unsigned int i = 0 ;
char or = 0 ;
PROG = "s6-ftrig-listen" ;
{
unsigned int t = 0 ;
for (;;)
{
register int opt = subgetopt(argc, argv, "aot:") ;
if (opt == -1) break ;
switch (opt)
{
case 'a' : or = 0 ; break ;
case 'o' : or = 1 ; break ;
case 't' : if (uint0_scan(subgetopt_here.arg, &t)) break ;
default : dieusage() ;
}
}
if (t) tain_from_millisecs(&tto, t) ; else tto = tain_infinite_relative ;
argc -= subgetopt_here.ind ; argv += subgetopt_here.ind ;
}
if (argc < 4) dieusage() ;
argc1 = el_semicolon(argv) ;
if (!argc1 || (argc1 & 1) || (argc == argc1 + 1)) dieusage() ;
if (argc1 >= argc) strerr_dief1x(100, "unterminated fifodir+regex block") ;
tain_now_g() ;
tain_add_g(&deadline, &tto) ;
x[0].fd = selfpipe_init() ;
if (x[0].fd < 0) strerr_diefu1sys(111, "selfpipe_init") ;
if (selfpipe_trap(SIGCHLD) < 0) strerr_diefu1sys(111, "selfpipe_trap") ;
if (sig_ignore(SIGPIPE) < 0) strerr_diefu1sys(111, "sig_ignore") ;
if (!ftrigr_startf_g(&a, &deadline)) strerr_diefu1sys(111, "ftrigr_startf") ;
x[1].fd = ftrigr_fd(&a) ;
{
int pid = 0 ;
unsigned int idlen = argc1 >> 1 ;
uint16 ids[idlen] ;
for (; i < idlen ; i++)
{
ids[i] = ftrigr_subscribe_g(&a, argv[i<<1], argv[(i<<1)+1], 0, &deadline) ;
if (!ids[i]) strerr_diefu4sys(111, "subscribe to ", argv[i<<1], " with regexp ", argv[(i<<1)+1]) ;
}
pid = child_spawn0(argv[argc1 + 1], argv + argc1 + 1, envp) ;
if (!pid) strerr_diefu2sys(111, "spawn ", argv[argc1 + 1]) ;
for (;;)
{
register int r ;
i = 0 ;
while (i < idlen)
{
char dummy ;
r = ftrigr_check(&a, ids[i], &dummy) ;
if (r < 0) strerr_diefu1sys(111, "ftrigr_check") ;
else if (!r) i++ ;
else if (or) idlen = 0 ;
else ids[i] = ids[--idlen] ;
}
if (!idlen) break ;
r = iopause_g(x, 2, &deadline) ;
if (r < 0) strerr_diefu1sys(111, "iopause") ;
else if (!r)
{
errno = ETIMEDOUT ;
strerr_diefu1sys(1, "get expected event") ;
}
if (x[0].revents & IOPAUSE_READ) handle_signals() ;
if (x[1].revents & IOPAUSE_READ)
{
if (ftrigr_update(&a) < 0) strerr_diefu1sys(111, "ftrigr_update") ;
}
}
}
return 0 ;
}
int main (void)
{
stralloc indata = STRALLOC_ZERO ;
unsigned int instate = 0 ;
PROG = "s6-ftrigrd" ;
if (ndelay_on(0) < 0) strerr_diefu2sys(111, "ndelay_on ", "0") ;
if (ndelay_on(1) < 0) strerr_diefu2sys(111, "ndelay_on ", "1") ;
if (sig_ignore(SIGPIPE) < 0) strerr_diefu1sys(111, "ignore SIGPIPE") ;
{
struct taia deadline, stamp ;
taia_now(&stamp) ;
taia_addsec(&deadline, &stamp, 2) ;
if (!skaserver2_sync(&asyncout, FTRIGR_BANNER1, FTRIGR_BANNER1_LEN, FTRIGR_BANNER2, FTRIGR_BANNER2_LEN, &deadline, &stamp))
strerr_diefu1sys(111, "sync with client") ;
}
for (;;)
{
register unsigned int n = genalloc_len(ftrigio_t, &a) ;
iopause_fd x[3 + n] ;
unsigned int i = 0 ;
int r ;
x[0].fd = 0 ; x[0].events = IOPAUSE_EXCEPT | IOPAUSE_READ ;
x[1].fd = 1 ; x[1].events = IOPAUSE_EXCEPT | (bufalloc_len(bufalloc_1) ? IOPAUSE_WRITE : 0) ;
x[2].fd = bufalloc_fd(&asyncout) ; x[2].events = IOPAUSE_EXCEPT | (bufalloc_len(&asyncout) ? IOPAUSE_WRITE : 0) ;
for (; i < n ; i++)
{
register ftrigio_t_ref p = genalloc_s(ftrigio_t, &a) + i ;
p->xindex = 3 + i ;
x[3+i].fd = p->trig.fd ;
x[3+i].events = IOPAUSE_READ ;
}
r = iopause(x, 3 + n, 0, 0) ;
if (r < 0)
{
cleanup() ;
strerr_diefu1sys(111, "iopause") ;
}
/* client closed => exit */
if ((x[0].revents | x[1].revents) & IOPAUSE_EXCEPT) break ;
/* client reading => flush pending data */
if (x[1].revents & IOPAUSE_WRITE)
if ((bufalloc_flush(bufalloc_1) == -1) && !error_isagain(errno))
{
cleanup() ;
strerr_diefu1sys(111, "flush stdout") ;
}
if (x[2].revents & IOPAUSE_WRITE)
if ((bufalloc_flush(&asyncout) == -1) && !error_isagain(errno))
{
cleanup() ;
strerr_diefu1sys(111, "flush asyncout") ;
}
/* scan listening ftrigs */
for (i = 0 ; i < genalloc_len(ftrigio_t, &a) ; i++)
{
register ftrigio_t_ref p = genalloc_s(ftrigio_t, &a) + i ;
if (x[p->xindex].revents & IOPAUSE_READ)
{
char c ;
register int r = sanitize_read(fd_read(p->trig.fd, &c, 1)) ;
if (!r) continue ;
if (r < 0)
{
trig(p->id, 'd', errno) ;
remove(i--) ;
}
else if (!sredfa_feed(p->re, &p->dfastate, c))
{
trig(p->id, 'd', ENOEXEC) ;
remove(i--) ;
}
else if (p->dfastate & SREDFA_ACCEPT)
{
trig(p->id, '!', c) ;
if (p->options & FTRIGR_REPEAT)
p->dfastate = SREDFA_START ;
else remove(i--) ;
}
}
}
/* client writing => get data and parse it */
if (buffer_len(buffer_0small) || x[0].revents & IOPAUSE_READ)
{
int r ;
for (;;)
{
uint16 id ;
r = sanitize_read(netstring_get(buffer_0small, &indata, &instate)) ;
if (r <= 0) break ;
if (indata.len < 3)
{
cleanup() ;
strerr_dief1x(100, "invalid client request") ;
}
uint16_unpack_big(indata.s, &id) ;
switch (indata.s[2]) /* protocol parsing */
{
case 'U' : /* unsubscribe */
{
register unsigned int i = genalloc_len(ftrigio_t, &a) ;
for (; i ; i--) if (genalloc_s(ftrigio_t, &a)[i-1].id == id) break ;
if (i) remove(i-1) ;
answer(0) ;
break ;
}
case 'L' : /* subscribe to path and match re */
{
ftrigio_t f = FTRIGIO_ZERO ;
uint32 pathlen, relen ;
if (indata.len < 18)
{
answer(EPROTO) ;
break ;
}
uint32_unpack_big(indata.s + 3, &f.options) ;
uint32_unpack_big(indata.s + 7, &pathlen) ;
uint32_unpack_big(indata.s + 11, &relen) ;
if (((pathlen + relen + 16) != indata.len) || indata.s[15 + pathlen])
{
answer(EPROTO) ;
break ;
}
f.id = id ;
if (!stralloc_0(&indata))
{
answer(errno) ;
break ;
}
f.re = sredfa_new() ;
if (!f.re)
{
answer(errno) ;
break ;
}
if (!sredfa_from_regexp(f.re, indata.s + 16 + pathlen)
|| !ftrig1_make(&f.trig, indata.s + 15))
{
sredfa_delete(f.re) ;
answer(errno) ;
break ;
}
if (!genalloc_append(ftrigio_t, &a, &f))
{
ftrigio_deepfree(&f) ;
answer(errno) ;
break ;
}
answer(0) ;
break ;
}
default :
{
cleanup() ;
strerr_dief1x(100, "invalid client request") ;
}
}
indata.len = 0 ;
} /* end loop: parse input from client */
if (r < 0)
{
if (errno == EPIPE) break ; /* client closed */
else
{
cleanup() ;
strerr_diefu1sys(111, "read a netstring") ;
}
}
} /* end if: stuff to read on stdin */
} /* end loop: main iopause */
cleanup() ;
return 0 ;
}
int main (int argc, char const *const *argv)
{
genalloc list = GENALLOC_ZERO ; /* array of diuint */
char delim = '\t' ;
unsigned long what = 0 ;
PROG = "s6-cut" ;
{
subgetopt_t l = SUBGETOPT_ZERO ;
int flagnodel = 1 ;
for (;;)
{
register int opt = subgetopt_r(argc, argv, "nsb:c:f:d:", &l) ;
if (opt == -1) break ;
switch (opt)
{
case 'n': break ; /* ignored */
case 's': flagnodel = 0 ; break ;
case 'd': delim = *l.arg ; break ;
case 'b':
case 'c':
{
if (what) strerr_dieusage(100, USAGE) ;
what = 2 ;
scanlist(&list, l.arg) ;
break ;
}
case 'f':
{
if (what) strerr_dieusage(100, USAGE) ;
what = 4 ;
scanlist(&list, l.arg) ;
break ;
}
default : strerr_dieusage(100, USAGE) ;
}
}
what += flagnodel ;
argc -= l.ind ; argv += l.ind ;
}
if (!genalloc_len(diuint, &list)) strerr_dieusage(100, USAGE) ;
diuintalloc_normalize(&list) ;
if (!argc)
{
if (!doit(0, genalloc_s(diuint, &list), genalloc_len(diuint, &list), what, delim))
strerr_diefu1sys(111, "cut stdin") ;
}
else
{
for (; *argv ; argv++)
{
if ((*argv[0] == '-') && !*argv[1])
{
if (!doit(0, genalloc_s(diuint, &list), genalloc_len(diuint, &list), what, delim))
strerr_diefu1sys(111, "process stdin") ;
}
else
{
int fd = open_readb(*argv) ;
if (fd == -1)
strerr_diefu3sys(111, "open ", *argv, " for reading") ;
if (!doit(fd, genalloc_s(diuint, &list), genalloc_len(diuint, &list), what, delim))
strerr_diefu2sys(111, "cut ", *argv) ;
fd_close(fd) ;
}
}
}
return 0 ;
}
int main (int argc, char const **argv, char const *const *envp)
{
tain_t deadline, tto ;
int spfd ;
int argc1 ;
int or = 0 ;
int wantup = 1, wantready = 0, wantrestart = 0 ;
PROG = "s6-svlisten" ;
{
subgetopt_t l = SUBGETOPT_ZERO ;
unsigned int t = 0 ;
for (;;)
{
register int opt = subgetopt_r(argc, argv, "uUdDrRaot:", &l) ;
if (opt == -1) break ;
switch (opt)
{
case 'u' : wantup = 1 ; wantready = 0 ; break ;
case 'U' : wantup = 1 ; wantready = 1 ; break ;
case 'd' : wantup = 0 ; wantready = 0 ; break ;
case 'D' : wantup = 0 ; wantready = 1 ; break ;
case 'r' : wantrestart = 1 ; wantready = 0 ; break ;
case 'R' : wantrestart = 1 ; wantready = 1 ; break ;
case 'a' : or = 0 ; break ;
case 'o' : or = 1 ; break ;
case 't' : if (!uint0_scan(l.arg, &t)) dieusage() ; break ;
default : dieusage() ;
}
}
argc -= l.ind ; argv += l.ind ;
if (t) tain_from_millisecs(&tto, t) ; else tto = tain_infinite_relative ;
}
if (argc < 3) dieusage() ;
argc1 = el_semicolon(argv) ;
if (!argc1 || argc == argc1 + 1) dieusage() ;
if (argc1 >= argc) strerr_dief1x(100, "unterminated servicedir block") ;
if (wantrestart && or)
{
or = 0 ;
strerr_warnw3x("-o is unsupported when combined with -", wantready ? "R" : "r", "- using -a instead") ;
}
tain_now_g() ;
tain_add_g(&deadline, &tto) ;
spfd = s6_svlisten_selfpipe_init() ;
{
s6_svlisten_t foo = S6_SVLISTEN_ZERO ;
pid_t pid ;
uint16 ids[argc1] ;
unsigned char upstate[bitarray_div8(argc1)] ;
unsigned char readystate[bitarray_div8(argc1)] ;
s6_svlisten_init(argc1, argv, &foo, ids, upstate, readystate, &deadline) ;
pid = child_spawn0(argv[argc1 + 1], argv + argc1 + 1, envp) ;
if (!pid) strerr_diefu2sys(111, "spawn ", argv[argc1 + 1]) ;
if (wantrestart)
{
argc1 = s6_svlisten_loop(&foo, 0, 1, or, &deadline, spfd, &s6_svlisten_signal_handler) ;
if (argc1) return argc1 ;
wantup = 1 ;
}
return s6_svlisten_loop(&foo, wantup, wantready, or, &deadline, spfd, &s6_svlisten_signal_handler) ;
}
}
