Add s6-socklog

Signed-off-by: Laurent Bercot <ska@appnovation.com>

6 files changed, 345 insertions, 124 deletions

diff --git a/src/daemontools-extras/deps-exe/s6-socklog b/src/daemontools-extras/deps-exe/s6-socklog
new file mode 100644
index 0000000..8e48518
--- /dev/null
+++ b/src/daemontools-extras/deps-exe/s6-socklog
@@ -0,0 +1,4 @@
+lolsyslog.o
+-lskarnet
+${SOCKET_LIB}
+${SYSCLOCK_LIB}
diff --git a/src/daemontools-extras/deps-exe/ucspilogd b/src/daemontools-extras/deps-exe/ucspilogd
index e7187fe..3fe140f 100644
--- a/src/daemontools-extras/deps-exe/ucspilogd
+++ b/src/daemontools-extras/deps-exe/ucspilogd
@@ -1 +1,2 @@
+lolsyslog.o
 -lskarnet
diff --git a/src/daemontools-extras/lolsyslog.c b/src/daemontools-extras/lolsyslog.c
new file mode 100644
index 0000000..a397dae
--- /dev/null
+++ b/src/daemontools-extras/lolsyslog.c
@@ -0,0 +1,90 @@
+/* ISC license. */
+
+#undef INTERNAL_MARK
+#ifndef SYSLOG_NAMES
+#define SYSLOG_NAMES
+#endif
+
+#include <skalibs/nonposix.h>
+
+#include <string.h>
+#include <syslog.h>
+
+#include <skalibs/types.h>
+
+#include "lolsyslog.h"
+
+#ifndef INTERNAL_MARK
+
+typedef struct CODE_s CODE, *CODE_ref ;
+struct CODE_s
+{
+  char *c_name ;
+  unsigned int c_val ;
+} ;
+
+#define LOG_PRI(p) ((p) & LOG_PRIMASK)
+#define LOG_FAC(p) (((p) & LOG_FACMASK) / (LOG_PRIMASK + 1))
+
+static CODE const facilitynames[] =
+{
+  { "kern", LOG_KERN },
+  { "user", LOG_USER },
+  { "mail", LOG_MAIL },
+  { "news", LOG_NEWS },
+  { "uucp", LOG_UUCP },
+  { "daemon", LOG_DAEMON },
+  { "auth", LOG_AUTH },
+  { "cron", LOG_CRON },
+  { "lpr", LOG_LPR },
+#ifdef LOG_SYSLOG
+  { "syslog", LOG_SYSLOG },
+#endif
+#ifdef LOG_AUDIT
+  { "audit", LOG_AUDIT },
+#endif
+  { "local0", LOG_LOCAL0 },
+  { "local1", LOG_LOCAL1 },
+  { "local2", LOG_LOCAL2 },
+  { "local3", LOG_LOCAL3 },
+  { "local4", LOG_LOCAL4 },
+  { "local5", LOG_LOCAL5 },
+  { "local6", LOG_LOCAL6 },
+  { "local7", LOG_LOCAL7 },
+  { 0, -1 }
+} ;
+
+static CODE const prioritynames[] =
+{
+  { "emerg", LOG_EMERG },
+  { "alert", LOG_ALERT },
+  { "crit", LOG_CRIT },
+  { "err", LOG_ERR },
+  { "warning", LOG_WARNING },
+  { "notice", LOG_NOTICE },
+  { "info", LOG_INFO },
+  { "debug", LOG_DEBUG },
+  { 0, -1 }
+} ;
+
+#endif
+
+size_t lolsyslog_string (char *out, char const *in)
+{
+  size_t i ;
+  unsigned int fpr ;
+  int fp ;
+  CODE const *p = facilitynames ;
+
+  if (in[0] != '<' || !(i = uint_scan(in+1, &fpr)) || in[1+i] != '>') return 0 ;
+  fp = LOG_FAC(fpr) << 3 ;
+  for (; p->c_name ; p++) if (p->c_val == fp) break ;
+  out = stpcpy(out, p->c_name ? p->c_name : "unknown") ;
+  *out++ = '.' ;
+
+  fp = LOG_PRI(fpr) ;
+  for (p = prioritynames ; p->c_name ; p++) if (p->c_val == fp) break ;
+  out = stpcpy(out, p->c_name ? p->c_name : "unknown") ;
+  *out++ = ':' ; *out++ = ' ' ; *out++ = 0 ;
+  return i+2 ;
+}
diff --git a/src/daemontools-extras/lolsyslog.h b/src/daemontools-extras/lolsyslog.h
new file mode 100644
index 0000000..7364412
--- /dev/null
+++ b/src/daemontools-extras/lolsyslog.h
@@ -0,0 +1,10 @@
+/* ISC license. */
+
+#ifndef LOLSYSLOG_H
+#define LOLSYSLOG_H
+
+#define LOLSYSLOG_STRING 32
+
+extern size_t lolsyslog_string (char *, char const *) ;
+
+#endif
diff --git a/src/daemontools-extras/s6-socklog.c b/src/daemontools-extras/s6-socklog.c
new file mode 100644
index 0000000..f261916
--- /dev/null
+++ b/src/daemontools-extras/s6-socklog.c
@@ -0,0 +1,220 @@
+/* ISC license. */
+
+#include <stdint.h>
+#include <unistd.h>
+#include <stdlib.h>
+#include <fcntl.h>
+#include <signal.h>
+#include <errno.h>
+#include <limits.h>
+
+#include <skalibs/types.h>
+#include <skalibs/sgetopt.h>
+#include <skalibs/allreadwrite.h>
+#include <skalibs/buffer.h>
+#include <skalibs/error.h>
+#include <skalibs/strerr2.h>
+#include <skalibs/tai.h>
+#include <skalibs/iopause.h>
+#include <skalibs/djbunix.h>
+#include <skalibs/socket.h>
+#include <skalibs/ip46.h>
+#include <skalibs/setgroups.h>
+#include <skalibs/sig.h>
+#include <skalibs/selfpipe.h>
+
+#include "lolsyslog.h"
+
+#define USAGE "s6-socklog [ -d notif ] [ -r ] [ -U | -u uid -g gid -G gidlist ] [ -l linelen ] [ -t lameducktimeout ] [ -x socket | -i ip_port ]"
+#define dieusage() strerr_dieusage(100, USAGE)
+
+static tain lameducktto = TAIN_INFINITE_RELATIVE ;
+static int cont = 1 ;
+
+static inline void handle_signals (void)
+{
+  for (;;) switch (selfpipe_read())
+  {
+    case -1 : strerr_diefu1sys(111, "selfpipe_read()") ;
+    case 0 : return ;
+    case SIGTERM : cont = 0 ; break ;
+    default : break ;
+  }
+}
+
+int main (int argc, char const *const *argv)
+{
+  iopause_fd x[3] = { { .events = IOPAUSE_READ }, { .fd = 1 } } ;
+  int flagraw = 0 ;
+  char const *usock = "/dev/log" ;
+  unsigned int linelen = 1024 ;
+  PROG = "s6-socklog" ;
+  {
+    subgetopt l = SUBGETOPT_ZERO ;
+    unsigned int notif = 0 ;
+    unsigned int t = 0 ;
+    uid_t uid = 0 ;
+    gid_t gid = 0 ;
+    gid_t gids[NGROUPS_MAX + 1] ;
+    size_t gidn = (size_t)-1 ;
+    ip46 ip ;
+    uint16_t port = 514 ;
+    for (;;)
+    {
+      int opt = subgetopt_r(argc, argv, "rd:l:t:u:g:G:Ux:i:", &l) ;
+      if (opt == -1) break ;
+      switch (opt)
+      {
+        case 'r' : flagraw = 1 ; break ;
+        case 'd' : if (!uint0_scan(l.arg, &notif)) dieusage() ; break ;
+        case 'l' : if (!uint0_scan(l.arg, &linelen)) dieusage() ; break ;
+        case 't' : if (!uint0_scan(l.arg, &t)) dieusage() ; break ;
+        case 'u' : if (!uid0_scan(l.arg, &uid)) dieusage() ; break ;
+        case 'g' : if (!gid0_scan(l.arg, &gid)) dieusage() ; break ;
+        case 'G' : if (!gid_scanlist(gids, NGROUPS_MAX, l.arg, &gidn) && *l.arg) dieusage() ; break ;
+        case 'U' :
+        {
+          char const *x = getenv("UID") ;
+          if (!x) strerr_dienotset(100, "UID") ;
+          if (!uid0_scan(x, &uid)) strerr_dieinvalid(100, "UID") ;
+          x = getenv("GID") ;
+          if (!x) strerr_dienotset(100, "GID") ;
+          if (!gid0_scan(x, &gid)) strerr_dieinvalid(100, "GID") ;
+          x = getenv("GIDLIST") ;
+          if (!x) strerr_dienotset(100, "GIDLIST") ;
+          if (!gid_scanlist(gids, NGROUPS_MAX+1, x, &gidn) && *x)
+            strerr_dieinvalid(100, "GIDLIST") ;
+          break ;
+        }
+        case 'x' : usock = l.arg ; break ;
+        case 'i' :
+        {
+          size_t pos = ip46_scan(l.arg, &ip) ;
+          if (!pos) dieusage() ;
+          if (l.arg[pos] == '_' || (!ip46_is6(&ip) && l.arg[pos] == ':'))
+            if (!uint160_scan(l.arg + pos + 1, &port)) dieusage() ;
+          usock = 0 ;
+          break ;
+        }
+        default : dieusage() ;
+      }
+    }
+    argc -= l.ind ; argv += l.ind ;
+
+    if (linelen < 80) linelen = 80 ;
+    if (linelen > 1048576) linelen = 1048576 ;
+    if (t) tain_from_millisecs(&lameducktto, t) ;
+    else lameducktto = tain_infinite_relative ;
+    if (notif)
+    {
+      if (notif < 3) strerr_dief1x(100, "notification fd must be 3 or more") ;
+      if (fcntl(notif, F_GETFD) < 0) strerr_dief1sys(100, "invalid notification fd") ;
+    }
+
+    close(0) ;
+    if (fcntl(1, F_GETFD) < 0) strerr_dief1sys(100, "invalid stdout") ;
+    if (usock)
+    {
+      x[2].fd = ipc_datagram_nbcoe() ;
+      if (x[2].fd == -1) strerr_diefu1sys(111, "create socket") ;
+      if (ipc_bind_reuse_perms(x[2].fd, usock, 0777) == -1)
+        strerr_diefu2sys(111, "bind socket to ", usock) ;
+    }
+    else
+    {
+      x[2].fd = socket_udp46_nbcoe(ip46_is6(&ip)) ;
+      if (x[2].fd == -1) strerr_diefu1sys(111, "create socket") ;
+      if (socket_bind46_reuse(x[2].fd, &ip, port) == -1)
+      {
+        char fmti[IP46_FMT] ;
+        char fmtp[UINT16_FMT] ;
+        fmti[ip46_fmt(fmti, &ip)] = 0 ;
+        fmtp[uint16_fmt(fmtp, port)] = 0 ;
+        strerr_diefu5sys(111, "bind socket to ", "ip ", fmti, " port ", fmtp) ;
+      }
+    }
+
+    if (gidn != (size_t)-1 && setgroups_and_gid(gid ? gid : getegid(), gidn, gids) < 0)
+      strerr_diefu1sys(111, "set supplementary group list") ;
+    if (gid && setgid(gid) < 0)
+      strerr_diefu1sys(111, "setgid") ;
+    if (uid && setuid(uid) < 0)
+      strerr_diefu1sys(111, "setuid") ;
+
+    x[0].fd = selfpipe_init() ;
+    if (x[0].fd == -1) strerr_diefu1sys(111, "init selfpipe") ;
+    if (!sig_altignore(SIGPIPE)) strerr_diefu1sys(111, "ignore SIGPIPE") ;
+    if (!selfpipe_trap(SIGTERM)) strerr_diefu1sys(111, "trap signals") ;
+
+    tain_now_set_stopwatch_g() ;
+
+    if (notif)
+    {
+      fd_write(notif, "\n", 1) ;
+      fd_close(notif) ;
+    }
+  }
+
+  {
+    char outbuf[linelen << 2] ;
+    buffer b1 = BUFFER_INIT(&buffer_write, 1, outbuf, linelen << 2) ;
+    char line[linelen + 1] ;
+    while (cont || buffer_len(&b1))
+    {
+      tain deadline = TAIN_INFINITE ;
+      ssize_t r ;
+      if (!cont) tain_add_g(&deadline, &lameducktto) ;
+      x[1].events = buffer_len(&b1) ? IOPAUSE_WRITE : 0 ;
+      x[2].events = cont && buffer_available(&b1) >= linelen + 80 ? IOPAUSE_READ : 0 ;
+      r = iopause_g(x, 3, &deadline) ;
+      if (r == -1) strerr_diefu1sys(111, "iopause") ;
+      if (!r) return 99 ;
+      if (x[0].revents & IOPAUSE_READ) handle_signals() ;
+      if (x[1].events & x[1].revents & IOPAUSE_WRITE)
+        if (!buffer_flush(&b1) && !error_isagain(errno))
+          strerr_diefu1sys(111, "write to stdout") ;
+      if (x[2].events & x[2].revents & IOPAUSE_READ)
+      {
+        if (usock)
+        {
+          r = sanitize_read(fd_recv(x[2].fd, line, linelen + 1, 0)) ;
+          if (r == -1) strerr_diefu1sys(111, "recv") ;
+        }
+        else
+        {
+          ip46 ip ;
+          uint16 port ;
+          r = sanitize_read(socket_recv46(x[2].fd, line, linelen + 1, &ip, &port)) ;
+          if (r == -1) strerr_diefu1sys(111, "recv") ;
+          if (r)
+          {
+            char fmt[IP46_FMT + UINT16_FMT + 3] ;
+            size_t m = ip46_fmt(fmt, &ip) ;
+            fmt[m++] = '_' ;
+            m += uint16_fmt(fmt, port) ;
+            fmt[m++] = ':' ; fmt[m++] = ' ' ;
+            buffer_putnoflush(&b1, fmt, m) ;
+          }
+        }
+        if (r)
+        {
+          size_t len = r ;
+          size_t pos = 0 ;
+          while (r && (!line[r-1] || line[r-1] == '\n')) r-- ;
+          for (size_t i = 0 ; i < r ; i++)
+            if (!line[i] || line[i] == '\n') line[i] = '~' ;
+          if (!flagraw)
+          {
+            char sbuf[LOLSYSLOG_STRING] ;
+            pos = lolsyslog_string(sbuf, line) ;
+            if (pos) buffer_putsnoflush(&b1, sbuf) ;
+          }
+          buffer_putnoflush(&b1, line + pos, r - pos) ;
+          if (len == linelen+1) buffer_putnoflush(&b1, "...", 3) ;
+          buffer_putnoflush(&b1, "\n", 1) ;
+        }
+      }
+    }
+  }
+  return 0 ;
+}
diff --git a/src/daemontools-extras/ucspilogd.c b/src/daemontools-extras/ucspilogd.c
index c5694d3..0ab6013 100644
--- a/src/daemontools-extras/ucspilogd.c
+++ b/src/daemontools-extras/ucspilogd.c
@@ -1,24 +1,18 @@
 /* ISC license. */
 
-#undef INTERNAL_MARK
-#ifndef SYSLOG_NAMES
-#define SYSLOG_NAMES
-#endif
-
-#include <skalibs/nonposix.h>
 #include <sys/types.h>
 #include <errno.h>
 #include <stdlib.h>
-#include <syslog.h>
-#include <skalibs/types.h>
+
 #include <skalibs/sgetopt.h>
 #include <skalibs/bytestr.h>
 #include <skalibs/buffer.h>
 #include <skalibs/strerr2.h>
 #include <skalibs/stralloc.h>
-#include <skalibs/env.h>
 #include <skalibs/skamisc.h>
 
+#include "lolsyslog.h"
+
 #define USAGE "ucspilogd [ -D default ] [ var... ]"
 #define dieusage() strerr_dieusage(100, USAGE)
 
@@ -27,110 +21,7 @@ static inline void die (void)
   strerr_diefu1sys(111, "write to stdout") ;
 }
 
-
- /*
-   Hack: INTERNAL_MARK is defined by all systems that
-   use the CODE stuff, and not by others (Solaris).
-*/
-
-#ifndef INTERNAL_MARK
-
-typedef struct CODE_s CODE, *CODE_ref ;
-struct CODE_s
-{
-  char *c_name ;
-  unsigned int c_val ;
-} ;
-
-static CODE const prioritynames[] =
-{
-  { "emerg", LOG_EMERG },
-  { "alert", LOG_ALERT },
-  { "crit", LOG_CRIT },
-  { "err", LOG_ERR },
-  { "warning", LOG_WARNING },
-  { "notice", LOG_NOTICE },
-  { "info", LOG_INFO },
-  { "debug", LOG_DEBUG },
-  { 0, -1 }
-} ;
-
-static CODE const facilitynames[] =
-{
-  { "kern", LOG_KERN },
-  { "user", LOG_USER },
-  { "mail", LOG_MAIL },
-  { "news", LOG_NEWS },
-  { "uucp", LOG_UUCP },
-  { "daemon", LOG_DAEMON },
-  { "auth", LOG_AUTH },
-  { "cron", LOG_CRON },
-  { "lpr", LOG_LPR },
-#ifdef LOG_SYSLOG
-  { "syslog", LOG_SYSLOG },
-#endif
-#ifdef LOG_AUDIT
-  { "audit", LOG_AUDIT },
-#endif
-  { "local0", LOG_LOCAL0 },
-  { "local1", LOG_LOCAL1 },
-  { "local2", LOG_LOCAL2 },
-  { "local3", LOG_LOCAL3 },
-  { "local4", LOG_LOCAL4 },
-  { "local5", LOG_LOCAL5 },
-  { "local6", LOG_LOCAL6 },
-  { "local7", LOG_LOCAL7 },
-  { 0, -1 }
-} ;
-
-#define LOG_PRI(p) ((p) & LOG_PRIMASK)
-#define LOG_FAC(p) (((p) & LOG_FACMASK) / (LOG_PRIMASK + 1))
-
-#endif
-
-
-static size_t syslog_names (char const *line)
-{
-  size_t i ;
-  unsigned int fpr ;
-  int fp ;
-  CODE const *p = facilitynames ;
-
-  if (line[0] != '<') return 0 ;
-  i = uint_scan(line+1, &fpr) ;
-  if (!i || (line[i+1] != '>')) return 0 ;
-  i += 2 ;
-  
-  fp = LOG_FAC(fpr) << 3 ;
-  for (; p->c_name ; p++) if (p->c_val == fp) break ;
-  if (p->c_name)
-  {
-    if ((buffer_puts(buffer_1, p->c_name) < 0)
-     || (buffer_put(buffer_1, ".", 1) < 1)) die() ;
-  }
-  else
-  {
-    if (buffer_put(buffer_1, "unknown.", 8) < 8) die() ;
-    i = 0 ;
-  }
-
-  fp = LOG_PRI(fpr) ;
-  for (p = prioritynames ; p->c_name ; p++) if (p->c_val == fp) break ;
-  if (p->c_name)
-  {
-    if ((buffer_puts(buffer_1, p->c_name) < 0)
-     || (buffer_put(buffer_1, ": ", 2) < 2)) die() ;
-  }
-  else
-  {
-    if (buffer_put(buffer_1, "unknown: ", 9) < 9) die() ;
-    i = 0 ;
-  }
-  return i ;
-}
-
-
-int main (int argc, char const *const *argv, char const *const *envp)
+int main (int argc, char const *const *argv)
 {
   char const *d = "<undefined>" ;
   PROG = "ucspilogd" ;
@@ -150,38 +41,43 @@ int main (int argc, char const *const *argv, char const *const *envp)
   }
 
   {
-    char const *envs[argc] ;
     unsigned int i = 0 ;
+    stralloc sa = STRALLOC_ZERO ;
+    char buf[LOLSYSLOG_STRING] ;
+    char const *envs[argc] ;
     for (; i < (unsigned int)argc ; i++)
     {
-      envs[i] = env_get2(envp, argv[i]) ;
+      envs[i] = getenv(argv[i]) ;
       if (!envs[i]) envs[i] = d ;
     }
     for (;;)
     {
       size_t pos = 0 ;
-      satmp.len = 0 ;
+      size_t j ;
+      sa.len = 0 ;
       {
-        int r = skagetlnsep(buffer_0f1, &satmp, "\n", 2) ;
+        int r = skagetlnsep(buffer_0f1, &sa, "\n", 2) ;
         if (r < 0)
         {
-          if (errno != EPIPE || !stralloc_0(&satmp))
+          if (errno != EPIPE || !stralloc_0(&sa))
             strerr_diefu1sys(111, "read from stdin") ;
         }
         if (!r) break ;
       }
-      if (!satmp.len) continue ;
-      satmp.s[satmp.len-1] = '\n' ;
-      if ((satmp.s[0] == '@') && (satmp.len > 26) && (byte_chr(satmp.s, 26, ' ') == 25))
+      if (!sa.len) continue ;
+      sa.s[sa.len-1] = 0 ;
+      if ((sa.s[0] == '@') && (sa.len > 26) && (byte_chr(sa.s, 26, ' ') == 25))
       {
-        if (buffer_put(buffer_1, satmp.s, 26) < 26) die() ;
+        if (buffer_put(buffer_1, sa.s, 26) < 26) die() ;
         pos += 26 ;
       }
       for (i = 0 ; i < (unsigned int)argc ; i++)
         if ((buffer_puts(buffer_1, envs[i]) < 0)
          || (buffer_put(buffer_1, ": ", 2) < 2)) die() ;
-      pos += syslog_names(satmp.s + pos) ;
-      if (buffer_put(buffer_1, satmp.s + pos, satmp.len - pos) < (ssize_t)(satmp.len - pos)) die() ;
+      j = lolsyslog_string(buf, sa.s + pos) ; pos += j ;
+      if (j && buffer_puts(buffer_1, buf) < 0) die() ;
+      sa.s[sa.len-1] = '\n' ;
+      if (buffer_put(buffer_1, sa.s + pos, sa.len - pos) < 0) die() ;
     }
   }
   return 0 ;