int
aa_service_status_write (aa_service_status *svst, const char *dir)
{
size_t len = strlen (dir);
char file[len + 1 + sizeof (AA_SVST_FILENAME)];
mode_t mask;
int r;
int e;
if (!stralloc_ready_tuned (&svst->sa, AA_SVST_FIXED_SIZE, 0, 0, 1))
return -1;
tain_pack (svst->sa.s, &svst->stamp);
uint32_pack (svst->sa.s + 12, (uint32_t) svst->event);
uint32_pack (svst->sa.s + 16, (uint32_t) svst->code);
if (svst->sa.len < AA_SVST_FIXED_SIZE)
svst->sa.len = AA_SVST_FIXED_SIZE;
byte_copy (file, len, dir);
byte_copy (file + len, 1 + sizeof (AA_SVST_FILENAME), "/" AA_SVST_FILENAME);
mask = umask (0033);
if (!openwritenclose_suffix (file, svst->sa.s,
svst->sa.len + ((svst->sa.len > AA_SVST_FIXED_SIZE) ? -1 : 0), ".new"))
r = -1;
else
r = 0;
e = errno;
umask (mask);
tain_now_g ();
errno = e;
return r;
}
static int
it_start (direntry *d, void *data)
{
if (*d->d_name == '.')
return 0;
tain_now_g ();
add_service (d->d_name, NULL);
return 0;
}
int
aa_service_status_read (aa_service_status *svst, const char *dir)
{
size_t len = strlen (dir);
char file[len + 1 + sizeof (AA_SVST_FILENAME)];
uint32_t u;
/* most cases should be w/out a message, so we'll only need FIXED_SIZE and
* one extra byte to NUL-terminate the (empty) message */
if (!stralloc_ready_tuned (&svst->sa, AA_SVST_FIXED_SIZE + 1, 0, 0, 1))
return -1;
byte_copy (file, len, dir);
byte_copy (file + len, 1 + sizeof (AA_SVST_FILENAME), "/" AA_SVST_FILENAME);
if (!openreadfileclose (file, &svst->sa, AA_SVST_FIXED_SIZE + AA_SVST_MAX_MSG_SIZE)
|| svst->sa.len < AA_SVST_FIXED_SIZE)
{
int e = errno;
tain_now_g ();
errno = e;
return -1;
}
tain_now_g ();
if (svst->sa.len >= svst->sa.a
&& !stralloc_ready_tuned (&svst->sa, svst->sa.len + 1, 0, 0, 1))
return -1;
svst->sa.s[svst->sa.len] = '\0';
svst->sa.len++;
tain_unpack (svst->sa.s, &svst->stamp);
uint32_unpack (svst->sa.s + 12, &u);
svst->event = (unsigned int) u;
uint32_unpack (svst->sa.s + 16, &u);
svst->code = (int) u;
return 0;
}
static int handle_connect_central(void) {
tain_t deadline;
mfd = main_socket_open();
if (mfd < 0) {
strerr_warnw1x("open_main_socket") ;
return -1;
}
tain_now_g();
tain_addsec_g(&deadline, 1);
bozclient_start_b_g(¢ral_client_g, ¢ral_client_data_g, mfd,
BEVT_CENTRAL_BANNER1, BEVT_CENTRAL_BANNER1_LEN,
BEVT_CENTRAL_BANNER2, BEVT_CENTRAL_BANNER2_LEN,
&deadline);
return 0;
}
int main (int argc, char const *const *argv)
{
unsigned int verbosity = 1 ;
tain_t deadline ;
PROG = "s6-rc-dryrun" ;
{
unsigned int t = 1000 ;
subgetopt_t l = SUBGETOPT_ZERO ;
for (;;)
{
register int opt = subgetopt_r(argc, argv, "v:t:", &l) ;
if (opt == -1) break ;
switch (opt)
{
case 'v': if (!uint0_scan(l.arg, &verbosity)) dieusage() ; break ;
case 't': if (!uint0_scan(l.arg, &t)) dieusage() ; break ;
default : strerr_dieusage(100, USAGE) ;
}
}
argc -= l.ind ; argv += l.ind ;
tain_from_millisecs(&deadline, t) ;
}
if (!argc) dieusage() ;
if (verbosity)
{
buffer_puts(buffer_1, PROG) ;
buffer_put(buffer_1, ":", 1) ;
for (; *argv ; argv++)
{
buffer_put(buffer_1, " ", 1) ;
buffer_puts(buffer_1, *argv) ;
}
buffer_putflush(buffer_1, "\n", 1) ;
}
tain_now_g() ;
tain_add_g(&deadline, &deadline) ;
deepsleepuntil_g(&deadline) ;
return 0 ;
}
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 ;
}
static inline int run_child (int fd, unsigned int timeout, pid_t pid, char const *s)
{
char dummy[4096] ;
iopause_fd x = { .fd = fd, .events = IOPAUSE_READ } ;
tain_t deadline ;
tain_now_g() ;
if (timeout) tain_from_millisecs(&deadline, timeout) ;
else deadline = tain_infinite_relative ;
tain_add_g(&deadline, &deadline) ;
for (;;)
{
int r = iopause_g(&x, 1, &deadline) ;
if (r < 0) strerr_diefu1sys(111, "iopause") ;
if (!r) return 99 ;
r = sanitize_read(fd_read(fd, dummy, 4096)) ;
if (r < 0)
if (errno == EPIPE) return 1 ;
else strerr_diefu1sys(111, "read from parent") ;
else if (r && memchr(dummy, '\n', r)) break ;
}
close(fd) ;
notify_systemd(pid, s) ;
return 0 ;
}
int main (int argc, char const *const *argv)
{
iopause_fd x[2] = { { -1, IOPAUSE_READ, 0 }, { -1, IOPAUSE_READ, 0 } } ;
PROG = "s6-supervise" ;
if (argc < 2) strerr_dieusage(100, USAGE) ;
if (chdir(argv[1]) < 0) strerr_diefu2sys(111, "chdir to ", argv[1]) ;
{
register unsigned int proglen = str_len(PROG) ;
register unsigned int namelen = str_len(argv[1]) ;
char progname[proglen + namelen + 2] ;
byte_copy(progname, proglen, PROG) ;
progname[proglen] = ' ' ;
byte_copy(progname + proglen + 1, namelen + 1, argv[1]) ;
PROG = progname ;
if (!fd_sanitize()) strerr_diefu1sys(111, "sanitize stdin and stdout") ;
x[1].fd = s6_supervise_lock(S6_SUPERVISE_CTLDIR) ;
if (!ftrigw_fifodir_make(S6_SUPERVISE_EVENTDIR, getegid(), 0))
strerr_diefu2sys(111, "mkfifodir ", S6_SUPERVISE_EVENTDIR) ;
x[0].fd = selfpipe_init() ;
if (x[0].fd == -1) strerr_diefu1sys(111, "init selfpipe") ;
if (sig_ignore(SIGPIPE) < 0) strerr_diefu1sys(111, "ignore SIGPIPE") ;
{
sigset_t set ;
sigemptyset(&set) ;
sigaddset(&set, SIGTERM) ;
sigaddset(&set, SIGHUP) ;
sigaddset(&set, SIGQUIT) ;
sigaddset(&set, SIGCHLD) ;
if (selfpipe_trapset(&set) < 0) strerr_diefu1sys(111, "trap signals") ;
}
if (!ftrigw_clean(S6_SUPERVISE_EVENTDIR))
strerr_warnwu2sys("ftrigw_clean ", S6_SUPERVISE_EVENTDIR) ;
{
struct stat st ;
if (stat("down", &st) == -1)
{
if (errno != ENOENT)
strerr_diefu1sys(111, "stat down") ;
}
else status.flagwantup = 0 ;
if (stat("nosetsid", &st) == -1)
{
if (errno != ENOENT)
strerr_diefu1sys(111, "stat nosetsid") ;
}
else flagsetsid = 0 ;
}
tain_now_g() ;
settimeout(0) ;
tain_copynow(&status.stamp) ;
announce() ;
ftrigw_notifyb_nosig(S6_SUPERVISE_EVENTDIR, "s", 1) ;
while (cont)
{
register int r = iopause_g(x, 2, &deadline) ;
if (r < 0) strerr_diefu1sys(111, "iopause") ;
else if (!r) (*actions[state][V_TIMEOUT])() ;
else
{
if ((x[0].revents | x[1].revents) & IOPAUSE_EXCEPT)
strerr_diefu1x(111, "iopause: trouble with pipes") ;
if (x[0].revents & IOPAUSE_READ) handle_signals() ;
else if (x[1].revents & IOPAUSE_READ) handle_control(x[1].fd) ;
}
}
ftrigw_notifyb_nosig(S6_SUPERVISE_EVENTDIR, "x", 1) ;
}
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 ;
}
static void
reset_service (const char *name, intptr_t mode)
{
aa_service *s;
int si;
int r;
int old_event;
int event;
r = aa_get_service (name, &si, 1);
if (r < 0)
{
aa_put_err (name, errmsg[-r], 1);
return;
}
r = aa_preload_service (si);
if (r < 0)
{
aa_put_err (name, errmsg[-r], 1);
return;
}
s = aa_service (si);
if (aa_service_status_read (&s->st, aa_service_name (s)) < 0 && errno != ENOENT)
{
int e = errno;
aa_put_err (name, "Failed to read service status file: ", 0);
aa_bs_noflush (AA_ERR, error_str (e));
aa_end_err ();
return;
}
if (s->st.type == AA_TYPE_LONGRUN)
{
aa_put_err (name, "Can only reset ont-shot services", 1);
return;
}
/* Starting/Stopping cannot be reset */
if (s->st.event == AA_EVT_STARTING || s->st.event == AA_EVT_STOPPING)
return;
if (mode == MODE_AUTO)
{
if (s->st.event == AA_EVT_STARTING_FAILED || s->st.event == AA_EVT_START_FAILED)
event = AA_EVT_STARTED;
else if (s->st.event == AA_EVT_STOPPING_FAILED || s->st.event == AA_EVT_STOP_FAILED)
event = AA_EVT_STOPPED;
else
return;
}
else
event = (mode == MODE_STARTED) ? AA_EVT_STARTED : AA_EVT_STOPPED;
if (s->st.event == event)
return;
tain_now_g ();
old_event = s->st.event;
s->st.event = event;
s->st.stamp = STAMP;
aa_service_status_set_msg (&s->st, "");
if (aa_service_status_write (&s->st, aa_service_name (s)) < 0)
{
int e = errno;
aa_put_err (name, "Failed to write service status file: ", 0);
aa_bs_noflush (AA_ERR, error_str (e));
aa_end_err ();
}
else
{
aa_put_title (1, name, "", 0);
aa_is_noflush (AA_OUT, ANSI_HIGHLIGHT_OFF);
aa_bs_noflush (AA_OUT, eventmsg[old_event]);
aa_is_noflush (AA_OUT, ANSI_HIGHLIGHT_ON);
aa_bs_noflush (AA_OUT, " -> ");
aa_bs_noflush (AA_OUT, eventmsg[event]);
aa_end_title ();
}
}
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 (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) ;
}
}
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)
{
iopause_fd x[2] = { { -1, IOPAUSE_READ, 0 }, { -1, IOPAUSE_READ, 0 } } ;
PROG = "s6-svscan" ;
{
subgetopt_t l = SUBGETOPT_ZERO ;
unsigned int t = 5000 ;
for (;;)
{
register int opt = subgetopt_r(argc, argv, "t:c:", &l) ;
if (opt == -1) break ;
switch (opt)
{
case 't' : if (uint0_scan(l.arg, &t)) break ;
case 'c' : if (uint0_scan(l.arg, &max)) break ;
default : strerr_dieusage(100, USAGE) ;
}
}
argc -= l.ind ; argv += l.ind ;
if (t) tain_from_millisecs(&defaulttimeout, t) ;
else defaulttimeout = tain_infinite_relative ;
if (max < 2) max = 2 ;
}
/* Init phase.
If something fails here, we can die, because it means that
something is seriously wrong with the system, and we can't
run correctly anyway.
*/
if (argc && (chdir(argv[0]) < 0)) strerr_diefu1sys(111, "chdir") ;
x[1].fd = s6_supervise_lock(S6_SVSCAN_CTLDIR) ;
x[0].fd = selfpipe_init() ;
if (x[0].fd < 0) strerr_diefu1sys(111, "selfpipe_init") ;
if (sig_ignore(SIGPIPE) < 0) strerr_diefu1sys(111, "ignore SIGPIPE") ;
{
sigset_t set ;
sigemptyset(&set) ;
sigaddset(&set, SIGCHLD) ;
sigaddset(&set, SIGALRM) ;
sigaddset(&set, SIGTERM) ;
sigaddset(&set, SIGHUP) ;
sigaddset(&set, SIGQUIT) ;
sigaddset(&set, SIGABRT) ;
sigaddset(&set, SIGINT) ;
sigaddset(&set, SIGUSR1) ;
if (selfpipe_trapset(&set) < 0) strerr_diefu1sys(111, "trap signals") ;
}
{
struct svinfo_s blob[max] ; /* careful with that stack, Eugene */
services = blob ;
tain_now_g() ;
/* Loop phase.
From now on, we must not die.
Temporize on recoverable errors, and panic on serious ones. */
while (cont)
{
int r ;
tain_add_g(&deadline, &defaulttimeout) ;
reap() ;
scan() ;
killthem() ;
r = iopause_g(x, 2, &deadline) ;
if (r < 0) panic("iopause") ;
else if (!r) wantscan = 1 ;
else
{
if ((x[0].revents | x[1].revents) & IOPAUSE_EXCEPT)
{
errno = EIO ;
panic("check internal pipes") ;
}
if (x[0].revents & IOPAUSE_READ) handle_signals() ;
if (x[1].revents & IOPAUSE_READ) handle_control(x[1].fd) ;
}
}
/* Finish phase. */
selfpipe_finish() ;
killthem() ;
reap() ;
}
{
char const *eargv[3] = { FINISH_PROG, finish_arg, 0 } ;
execve(eargv[0], (char **)eargv, (char *const *)environ) ;
}
panicnosp("exec finish script " FINISH_PROG) ;
}
int
main (int argc, char * const argv[])
{
PROG = "aa-start";
const char *path_repo = "/run/services";
const char *path_list = NULL;
int i;
aa_secs_timeout = DEFAULT_TIMEOUT_SECS;
for (;;)
{
struct option longopts[] = {
{ "double-output", no_argument, NULL, 'D' },
{ "help", no_argument, NULL, 'h' },
{ "listdir", required_argument, NULL, 'l' },
{ "dry-list", no_argument, NULL, 'n' },
{ "repodir", required_argument, NULL, 'r' },
{ "timeout", required_argument, NULL, 't' },
{ "version", no_argument, NULL, 'V' },
{ "verbose", no_argument, NULL, 'v' },
{ "no-wants", no_argument, NULL, 'W' },
{ NULL, 0, 0, 0 }
};
int c;
c = getopt_long (argc, argv, "Dhl:nr:t:VvW", longopts, NULL);
if (c == -1)
break;
switch (c)
{
case 'D':
aa_set_double_output (1);
break;
case 'h':
dieusage (0);
case 'l':
unslash (optarg);
path_list = optarg;
break;
case 'n':
if (mode & AA_MODE_IS_DRY)
mode |= AA_MODE_IS_DRY_FULL;
else
mode |= AA_MODE_IS_DRY;
break;
case 'r':
unslash (optarg);
path_repo = optarg;
break;
case 't':
if (!uint0_scan (optarg, &aa_secs_timeout))
aa_strerr_diefu2sys (ERR_IO, "set default timeout to ", optarg);
break;
case 'V':
aa_die_version ();
case 'v':
verbose = 1;
break;
case 'W':
no_wants = 1;
break;
default:
dieusage (1);
}
}
argc -= optind;
argv += optind;
cols = get_cols (1);
is_utf8 = is_locale_utf8 ();
if (!path_list && argc < 1)
dieusage (1);
if (aa_init_repo (path_repo, (mode & AA_MODE_IS_DRY) ? AA_REPO_READ : AA_REPO_WRITE) < 0)
aa_strerr_diefu2sys (ERR_IO, "init repository ", path_repo);
if (path_list)
{
stralloc sa = STRALLOC_ZERO;
int r;
if (*path_list != '/' && *path_list != '.')
stralloc_cats (&sa, LISTDIR_PREFIX);
stralloc_catb (&sa, path_list, strlen (path_list) + 1);
r = aa_scan_dir (&sa, 1, it_start, NULL);
stralloc_free (&sa);
if (r < 0)
aa_strerr_diefu3sys (-r, "read list directory ",
(*path_list != '/' && *path_list != '.') ? LISTDIR_PREFIX : path_list,
(*path_list != '/' && *path_list != '.') ? path_list : "");
}
tain_now_g ();
for (i = 0; i < argc; ++i)
if (str_equal (argv[i], "-"))
{
if (process_names_from_stdin ((names_cb) add_service, NULL) < 0)
aa_strerr_diefu1sys (ERR_IO, "process names from stdin");
}
else
add_service (argv[i], NULL);
mainloop (mode, scan_cb);
if (!(mode & AA_MODE_IS_DRY))
{
aa_bs_noflush (AA_OUT, "\n");
put_title (1, PROG, "Completed.", 1);
aa_show_stat_nb (nb_already, "Already up", ANSI_HIGHLIGHT_GREEN_ON);
aa_show_stat_nb (nb_done, "Started", ANSI_HIGHLIGHT_GREEN_ON);
show_stat_service_names (&ga_timedout, "Timed out", ANSI_HIGHLIGHT_RED_ON);
show_stat_service_names (&ga_failed, "Failed", ANSI_HIGHLIGHT_RED_ON);
show_stat_service_names (&ga_depend, "Dependency failed", ANSI_HIGHLIGHT_RED_ON);
aa_show_stat_names (aa_names.s, &ga_io, "I/O error", ANSI_HIGHLIGHT_RED_ON);
aa_show_stat_names (aa_names.s, &ga_unknown, "Unknown", ANSI_HIGHLIGHT_RED_ON);
aa_show_stat_names (aa_names.s, &ga_skipped, "Skipped", ANSI_HIGHLIGHT_YELLOW_ON);
}
genalloc_free (int, &ga_timedout);
genalloc_free (int, &ga_failed);
genalloc_free (int, &ga_depend);
genalloc_free (size_t, &ga_io);
genalloc_free (size_t, &ga_unknown);
genalloc_free (size_t, &ga_skipped);
genalloc_free (pid_t, &ga_pid);
genalloc_free (int, &aa_tmp_list);
genalloc_free (int, &aa_main_list);
stralloc_free (&aa_names);
genalloc_deepfree (struct progress, &ga_progress, free_progress);
aa_free_services (close_fd);
genalloc_free (iopause_fd, &ga_iop);
return rc;
}
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]) ;
}
