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;
}
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 ;
}
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 retrydirlater (void)
{
tain_t a ;
tain_addsec_g(&a, DIR_RETRY_TIMEOUT) ;
if (tain_less(&a, &deadline)) deadline = a ;
}
static void reap (void)
{
tain_t nextscan ;
if (!wantreap) return ;
wantreap = 0 ;
tain_addsec_g(&nextscan, 1) ;
for (;;)
{
int wstat ;
int r = wait_nohang(&wstat) ;
if (r < 0)
if (errno != ECHILD) panic("wait_nohang") ;
else break ;
else if (!r) break ;
else
{
register unsigned int i = 0 ;
for (; i < n ; i++)
{
if (services[i].pid[0] == r)
{
services[i].pid[0] = 0 ;
services[i].restartafter[0] = nextscan ;
break ;
}
else if (services[i].pid[1] == r)
{
services[i].pid[1] = 0 ;
services[i].restartafter[1] = nextscan ;
break ;
}
}
if (i == n) continue ;
if (services[i].flagactive)
{
if (tain_less(&nextscan, &deadline)) deadline = nextscan ;
}
else
{
if (services[i].flaglog)
{
/*
BLACK MAGIC:
- we need to close the pipe early:
* as soon as the writer exits so the logger can exit on EOF
* or as soon as the logger exits so the writer can crash on EPIPE
- but if the same service gets reactivated before the second
supervise process exits, ouch: we've lost the pipe
- so we can't reuse the same service even if it gets reactivated
- so we're marking a dying service with a closed pipe
- if the scanner sees a service with p[0] = -1 it won't flag
it as active (and won't restart the dead supervise)
- but if the service gets reactivated we want it to restart
as soon as the 2nd supervise process dies
- so the scanner marks such a process with p[0] = -2
- and the reaper triggers a scan when it finds a -2.
*/
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 ;
}
else if (services[i].p[0] == -2) wantscan = 1 ;
}
if (!services[i].pid[0] && (!services[i].flaglog || !services[i].pid[1]))
services[i] = services[--n] ;
}
}
}
}
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 ;
}
static inline void settimeout (int secs)
{
tain_addsec_g(&deadline, secs) ;
}
