aboutsummaryrefslogtreecommitdiffstats
path: root/src/conn-tools/s6-tcpserver4d.c
diff options
context:
space:
mode:
Diffstat (limited to 'src/conn-tools/s6-tcpserver4d.c')
-rw-r--r--src/conn-tools/s6-tcpserver4d.c427
1 files changed, 0 insertions, 427 deletions
diff --git a/src/conn-tools/s6-tcpserver4d.c b/src/conn-tools/s6-tcpserver4d.c
deleted file mode 100644
index 5982816..0000000
--- a/src/conn-tools/s6-tcpserver4d.c
+++ /dev/null
@@ -1,427 +0,0 @@
-/* ISC license. */
-
-#include <string.h>
-#include <stdint.h>
-#include <sys/stat.h>
-#include <sys/wait.h>
-#include <errno.h>
-#include <unistd.h>
-#include <fcntl.h>
-#include <signal.h>
-
-#include <skalibs/posixplz.h>
-#include <skalibs/uint16.h>
-#include <skalibs/uint32.h>
-#include <skalibs/types.h>
-#include <skalibs/allreadwrite.h>
-#include <skalibs/sgetopt.h>
-#include <skalibs/strerr.h>
-#include <skalibs/fmtscan.h>
-#include <skalibs/diuint32.h>
-#include <skalibs/env.h>
-#include <skalibs/cspawn.h>
-#include <skalibs/djbunix.h>
-#include <skalibs/sig.h>
-#include <skalibs/selfpipe.h>
-#include <skalibs/iopause.h>
-#include <skalibs/socket.h>
-#include <skalibs/genset.h>
-#include <skalibs/avltreen.h>
-
-#define ABSOLUTE_MAXCONN 16384
-
-#define USAGE "s6-tcpserver4d [ -v verbosity ] [ -1 ] [ -c maxconn ] [ -C localmaxconn ] prog..."
-
-typedef struct pidi_s pidi, *pidi_ref ;
-struct pidi_s
-{
- pid_t pid ;
- uint32_t i ;
-} ;
-
-static uint32_t maxconn = 40 ;
-static uint32_t localmaxconn = 40 ;
-static uint32_t verbosity = 1 ;
-static int cont = 1 ;
-
-static genset *pidis ;
-#define PIDI(i) genset_p(pidi, pidis, (i))
-#define numconn genset_n(pidis)
-static genset *ipnums ;
-#define IPNUM(i) genset_p(diuint32, ipnums, (i))
-static avltreen *by_ip ;
-static avltreen *by_pid ;
-
-static char fmtmaxconn[UINT32_FMT+1] = "/" ;
-static char fmtlocalmaxconn[UINT32_FMT+1] = "/" ;
-
-
-static inline void dieusage ()
-{
- strerr_dieusage(100, USAGE) ;
-}
-
-static inline void X (void)
-{
- strerr_dief1x(101, "internal inconsistency. Please submit a bug-report.") ;
-}
-
-static void *bypid_dtok (uint32_t d, void *aux)
-{
- genset *g = aux ;
- return &genset_p(pidi, g, d)->pid ;
-}
-
-static int bypid_cmp (void const *a, void const *b, void *aux)
-{
- (void)aux ;
- pid_t aa = *(pid_t const *)a ;
- pid_t bb = *(pid_t const *)b ;
- return aa < bb ? -1 : aa > bb ;
-}
-
-static void *byip_dtok (uint32_t d, void *aux)
-{
- genset *g = aux ;
- return &genset_p(diuint32, g, d)->left ;
-}
-
-static int byip_cmp (void const *a, void const *b, void *aux)
-{
- (void)aux ;
- uint32_t aa = *(uint32_t const *)a ;
- uint32_t bb = *(uint32_t const *)b ;
- return aa < bb ? -1 : aa > bb ;
-}
-
-static inline void log_start (void)
-{
- strerr_warni1x("starting") ;
-}
-
-static inline void log_exit (void)
-{
- strerr_warni1x("exiting") ;
-}
-
-static void log_status (void)
-{
- char fmt[UINT_FMT] ;
- fmt[uint_fmt(fmt, numconn)] = 0 ;
- strerr_warni3x("status: ", fmt, fmtmaxconn) ;
-}
-
-static inline void log_deny (uint32_t ip, uint16_t port, uint32_t num)
-{
- char fmtip[UINT32_FMT] ;
- char fmtport[UINT16_FMT] ;
- char fmtnum[UINT32_FMT] ;
- fmtip[ip4_fmtu32(fmtip, ip)] = 0 ;
- fmtport[uint16_fmt(fmtport, port)] = 0 ;
- fmtnum[uint32_fmt(fmtnum, num)] = 0 ;
- strerr_warni7sys("deny ", fmtip, ":", fmtport, " count ", fmtnum, fmtlocalmaxconn) ;
-}
-
-static inline void log_accept (pid_t pid, uint32_t ip, uint16_t port, uint32_t num)
-{
- char fmtipport[IP4_FMT + UINT16_FMT + 1] ;
- char fmtpid[PID_FMT] ;
- char fmtnum[UINT32_FMT] ;
- size_t n ;
- n = ip4_fmtu32(fmtipport, ip) ;
- fmtipport[n++] = ':' ;
- n += uint16_fmt(fmtipport + n, port) ;
- fmtipport[n] = 0 ;
- fmtnum[uint32_fmt(fmtnum, num)] = 0 ;
- fmtpid[pid_fmt(fmtpid, pid)] = 0 ;
- strerr_warni7x("allow ", fmtipport, " pid ", fmtpid, " count ", fmtnum, fmtlocalmaxconn) ;
-}
-
-static inline void log_close (pid_t pid, uint32_t ip, int w)
-{
- char fmtpid[PID_FMT] ;
- char fmtip[IP4_FMT] = "?" ;
- char fmtw[UINT_FMT] ;
- fmtpid[pid_fmt(fmtpid, pid)] = 0 ;
- fmtip[ip4_fmtu32(fmtip, ip)] = 0 ;
- fmtw[uint_fmt(fmtw, WIFSIGNALED(w) ? WTERMSIG(w) : WEXITSTATUS(w))] = 0 ;
- strerr_warni6x("end pid ", fmtpid, " ip ", fmtip, WIFSIGNALED(w) ? " signal " : " exitcode ", fmtw) ;
-}
-
-static int killthem_iter (void *data, void *aux)
-{
- kill(((pidi *)data)->pid, *(int *)aux) ;
- return 1 ;
-}
-
-static void killthem (int sig)
-{
- genset_iter(pidis, &killthem_iter, &sig) ;
-}
-
-static inline void wait_children (void)
-{
- for (;;)
- {
- uint32_t d ;
- int wstat ;
- pid_t pid = wait_nohang(&wstat) ;
- if (pid < 0)
- if (errno != ECHILD) strerr_diefu1sys(111, "wait_nohang") ;
- else break ;
- else if (!pid) break ;
- if (avltreen_search(by_pid, &pid, &d))
- {
- uint32_t i = PIDI(d)->i ;
- uint32_t ip = IPNUM(i)->left ;
- avltreen_delete(by_pid, &pid) ;
- genset_delete(pidis, d) ;
- if (!--IPNUM(i)->right)
- {
- avltreen_delete(by_ip, &ip) ;
- genset_delete(ipnums, i) ;
- }
- if (verbosity >= 2)
- {
- log_close(pid, ip, wstat) ;
- log_status() ;
- }
- }
- }
-}
-
-static inline void handle_signals (void)
-{
- for (;;) switch (selfpipe_read())
- {
- case -1 : strerr_diefu1sys(111, "read selfpipe") ;
- case 0 : return ;
- case SIGCHLD : wait_children() ; break ;
- case SIGTERM :
- {
- if (verbosity >= 2)
- strerr_warni3x("received ", "SIGTERM,", " quitting") ;
- cont = 0 ;
- break ;
- }
- case SIGHUP :
- {
- if (verbosity >= 2)
- strerr_warni5x("received ", "SIGHUP,", " sending ", "SIGTERM+SIGCONT", " to all connections") ;
- killthem(SIGTERM) ;
- killthem(SIGCONT) ;
- break ;
- }
- case SIGQUIT :
- {
- if (verbosity >= 2)
- strerr_warni6x("received ", "SIGQUIT,", " sending ", "SIGTERM+SIGCONT", " to all connections", " and quitting") ;
- cont = 0 ;
- killthem(SIGTERM) ;
- killthem(SIGCONT) ;
- break ;
- }
- case SIGABRT :
- {
- if (verbosity >= 2)
- strerr_warni6x("received ", "SIGABRT,", " sending ", "SIGKILL", " to all connections", " and quitting") ;
- cont = 0 ;
- killthem(SIGKILL) ;
- break ;
- }
- default : X() ;
- }
-}
-
-static inline void new_connection (int s, uint32_t ip, uint16_t port, char const *const *argv, char const *const *envp, size_t envlen)
-{
- size_t m = 0 ;
- pid_t pid ;
- uint32_t d ;
- uint32_t num = 0 ;
- char fmt[47 + IP4_FMT + UINT16_FMT + UINT_FMT] ;
-
- if (avltreen_search(by_ip, &ip, &d)) num = IPNUM(d)->right ;
- if (num >= localmaxconn)
- {
- log_deny(ip, port, num) ;
- return ;
- }
-
- memcpy(fmt + m, "PROTO=TCP\0TCPREMOTEIP=", 22) ; m += 22 ;
- m += ip4_fmtu32(fmt + m, ip) ;
- fmt[m++] = 0 ;
- memcpy(fmt + m, "TCPREMOTEPORT=", 14) ; m += 14 ;
- m += uint16_fmt(fmt + m, port) ; fmt[m++] = 0 ;
- memcpy(fmt + m, "TCPCONNNUM=", 11) ; m += 11 ;
- m += uint_fmt(fmt + m, num) ; fmt[m++] = 0 ;
-
- {
- cspawn_fileaction fa[2] =
- {
- [0] = { .type = CSPAWN_FA_MOVE, .x = { .fd2 = { [0] = 0, [1] = s } } },
- [1] = { .type = CSPAWN_FA_COPY, .x = { .fd2 = { [0] = 1, [1] = 0 } } }
- } ;
- char const *newenvp[envlen + 5] ;
- env_mergen(newenvp, envlen + 5, envp, envlen, fmt, m, 4) ;
- pid = cspawn(argv[0], argv, newenvp, CSPAWN_FLAGS_SELFPIPE_FINISH, fa, 2) ;
- }
- if (!pid)
- {
- if (verbosity) strerr_warnwu2sys("spawn ", argv[0]) ;
- return ;
- }
-
- if (num) IPNUM(d)->right++ ;
- else
- {
- d = genset_new(ipnums) ;
- IPNUM(d)->left = ip ;
- IPNUM(d)->right = 1 ;
- avltreen_insert(by_ip, d) ;
- }
-
- num = genset_new(pidis) ;
- PIDI(num)->pid = pid ;
- PIDI(num)->i = d ;
- avltreen_insert(by_pid, num) ;
- if (verbosity >= 2)
- {
- log_accept(pid, ip, port, IPNUM(d)->right) ;
- log_status() ;
- }
-}
-
-int main (int argc, char const *const *argv)
-{
- iopause_fd x[2] = { { .events = IOPAUSE_READ }, { .fd = 0, .events = IOPAUSE_READ | IOPAUSE_EXCEPT } } ;
- int flag1 = 0 ;
- PROG = "s6-tcpserver4d" ;
- {
- subgetopt l = SUBGETOPT_ZERO ;
- for (;;)
- {
- int opt = subgetopt_r(argc, argv, "1c:C:v:", &l) ;
- if (opt == -1) break ;
- switch (opt)
- {
- case '1' : flag1 = 1 ; break ;
- case 'c' : if (!uint320_scan(l.arg, &maxconn)) dieusage() ; break ;
- case 'C' : if (!uint320_scan(l.arg, &localmaxconn)) dieusage() ; break ;
- case 'v' : if (!uint320_scan(l.arg, &verbosity)) dieusage() ; break ;
- default : dieusage() ;
- }
- }
- argc -= l.ind ; argv += l.ind ;
- if (!argc || !*argv[0]) dieusage() ;
- {
- struct stat st ;
- if (fstat(0, &st) < 0) strerr_diefu1sys(111, "fstat stdin") ;
- if (!S_ISSOCK(st.st_mode)) strerr_dief1x(100, "stdin is not a socket") ;
- }
- if (coe(0) == -1 || ndelay_on(0) == -1)
- strerr_diefu1sys(111, "set socket flags") ;
- if (flag1)
- {
- if (fcntl(1, F_GETFD) < 0)
- strerr_dief1sys(100, "called with option -1 but stdout said") ;
- }
- else close(1) ;
- if (!maxconn) maxconn = 1 ;
- if (maxconn > ABSOLUTE_MAXCONN) maxconn = ABSOLUTE_MAXCONN ;
- if (localmaxconn > maxconn) localmaxconn = maxconn ;
-
- x[0].fd = selfpipe_init() ;
- if (x[0].fd == -1) strerr_diefu1sys(111, "create selfpipe") ;
- if (!sig_altignore(SIGPIPE)) strerr_diefu1sys(111, "ignore SIGPIPE") ;
- {
- sigset_t set ;
- sigemptyset(&set) ;
- sigaddset(&set, SIGCHLD) ;
- sigaddset(&set, SIGTERM) ;
- sigaddset(&set, SIGHUP) ;
- sigaddset(&set, SIGQUIT) ;
- sigaddset(&set, SIGABRT) ;
- if (!selfpipe_trapset(&set)) strerr_diefu1sys(111, "trap signals") ;
- }
- fmtlocalmaxconn[1+uint32_fmt(fmtlocalmaxconn+1, localmaxconn)] = 0 ;
- }
-
- {
- diuint32 ipnum_storage[maxconn] ;
- uint32_t ipnum_freelist[maxconn] ;
- avlnode byip_storage[maxconn] ;
- uint32_t byip_freelist[maxconn] ;
- pidi pidi_storage[maxconn] ;
- uint32_t pidi_freelist[maxconn] ;
- avlnode bypid_storage[maxconn] ;
- uint32_t bypid_freelist[maxconn] ;
- genset ipnum_info ;
- genset pidi_info ;
- avltreen byip_info ;
- avltreen bypid_info ;
- size_t envlen = env_len((char const *const *)environ) ;
-
- GENSET_init(&ipnum_info, diuint32, ipnum_storage, ipnum_freelist, maxconn) ;
- GENSET_init(&pidi_info, pidi, pidi_storage, pidi_freelist, maxconn) ;
- avltreen_init(&byip_info, byip_storage, byip_freelist, maxconn, &byip_dtok, &byip_cmp, &ipnum_info) ;
- avltreen_init(&bypid_info, bypid_storage, bypid_freelist, maxconn, &bypid_dtok, &bypid_cmp, &pidi_info) ;
- ipnums = &ipnum_info ;
- pidis = &pidi_info ;
- by_ip = &byip_info ;
- by_pid = &bypid_info ;
-
- if (verbosity >= 2)
- {
- fmtmaxconn[1+uint32_fmt(fmtmaxconn+1, maxconn)] = 0 ;
- log_start() ;
- log_status() ;
- }
-
- if (flag1)
- {
- uint16_t port ;
- uint16_t m = 0 ;
- char ip[4] ;
- char fmtport[UINT16_FMT] ;
- if (socket_local4(0, ip, &port) == -1)
- {
- if (verbosity) strerr_warnwu1sys("socket_local4") ;
- }
- else m = uint16_fmt(fmtport, port) ;
- fmtport[m++] = '\n' ;
- allwrite(1, fmtport, m) ;
- close(1) ;
- }
-
- while (cont)
- {
- if (iopause_g(x, 1 + (numconn < maxconn), 0) < 0) strerr_diefu1sys(111, "iopause") ;
- if (x[0].revents & IOPAUSE_EXCEPT) strerr_dief1x(111, "trouble with selfpipe") ;
- if (x[0].revents & IOPAUSE_READ) { handle_signals() ; continue ; }
- if (numconn < maxconn)
- {
- if (x[1].revents & IOPAUSE_EXCEPT) strerr_dief1x(111, "trouble with socket") ;
- if (x[1].revents & IOPAUSE_READ)
- {
- char packedip[4] ;
- uint16_t port ;
- int fd = socket_accept4(x[1].fd, packedip, &port) ;
- if (fd == -1)
- {
- if (verbosity) strerr_warnwu1sys("accept") ;
- }
- else
- {
- uint32_t ip ;
- uint32_unpack_big(packedip, &ip) ;
- new_connection(fd, ip, port, argv, (char const *const *)environ, envlen) ;
- fd_close(fd) ;
- }
- }
- }
- }
- }
- if (verbosity >= 2) log_exit() ;
- return 0 ;
-}