Implement "s6 process"

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

1 files changed, 108 insertions, 0 deletions

diff --git a/src/s6/process_status.c b/src/s6/process_status.c
new file mode 100644
index 0000000..2240c85
--- /dev/null
+++ b/src/s6/process_status.c
@@ -0,0 +1,108 @@
+/* ISC license. */
+
+#include <string.h>
+#include <sys/wait.h>
+#include <errno.h>
+
+#include <skalibs/posixplz.h>
+#include <skalibs/uint64.h>
+#include <skalibs/env.h>
+#include <skalibs/buffer.h>
+#include <skalibs/strerr.h>
+#include <skalibs/gol.h>
+#include <skalibs/cspawn.h>
+#include <skalibs/djbunix.h>
+
+#include <s6/config.h>
+
+#include "s6-internal.h"
+
+#define USAGE "s6 process status [ -l|--with-logs ] services..."
+#define dieusage() strerr_dieusage(100, USAGE)
+
+static int spawn_and_wait (char const *const *argv)
+{
+  int wstat ;
+  pid_t r ;
+  pid_t pid = cspawn(argv[0], argv, (char const *const *)environ, 0, 0, 0) ;
+  if (!pid)
+  {
+    if (g->verbosity) strerr_warnwu2sys("spawn ", argv[0]) ;
+    return 1 ;
+  }
+  r = wait_pid(pid, &wstat) ;
+  if (r != pid)
+  {
+    if (g->verbosity) strerr_warnwu2sys("wait for ", argv[0]) ;
+    return 1 ;
+  }
+  return !!WIFSIGNALED(wstat) || !!WEXITSTATUS(wstat) ;
+}
+
+static int do_status (char const *dir, int withlog)
+{
+  int e ;
+  char const *argv[4] = { S6_EXTBINPREFIX "s6-svstat", "--", dir, 0 } ;
+  size_t dirlen = strlen(dir) ;
+  buffer_puts(buffer_1, dir) ;
+  buffer_putsflush(buffer_1, ": ") ;
+  e = spawn_and_wait(argv) ;
+  if (withlog && (dirlen < 5 || strcmp(dir + dirlen - 4, "/log")))
+  {
+    struct stat st ;
+    char log[dirlen + 5] ;
+    memcpy(log, dir, dirlen) ;
+    memcpy(log + dirlen, "/log", 5) ;
+    if (stat(log, &st) < 0)
+    {
+      if (errno != ENOENT)
+      {
+        if (g->verbosity) strerr_warnwu2sys("stat", log) ;
+        e = 1 ;
+      }
+    }
+    else if (S_ISDIR(st.st_mode))
+    {
+      argv[2] = log ;
+      buffer_puts(buffer_1, log) ;
+      buffer_putsflush(buffer_1, ": ") ;
+      e |= spawn_and_wait(argv) ;
+    }
+  }
+  return e ;
+}
+
+
+enum process_status_golb_e
+{
+  PROCESS_STATUS_GOLB_WITHLOGS,
+  PROCESS_STATUS_GOLB_N
+} ;
+
+static gol_bool const process_status_golb[1] =
+{
+  { .so = 'l', .lo = "with-logs", .set = 1, .mask = 1 << PROCESS_STATUS_GOLB_WITHLOGS }
+} ;
+
+
+int process_status (char const *const *argv)
+{
+  size_t scandirlen = strlen(g->scandir) ;
+  uint64_t golb = 0 ;
+  int e = 0 ;
+  PROG = "s6 process status" ;
+
+  argv += gol_argv(argv, process_status_golb, 1, 0, 0, &golb, 0) ;
+  if (!argv) dieusage() ;
+  process_check_services(argv, env_len(argv)) ;
+  for (; *argv ; argv++)
+  {
+    size_t len = strlen(*argv) ;
+    char path[scandirlen + len + 2] ;
+    memcpy(path, g->scandir, scandirlen) ;
+    path[scandirlen] = '/' ;
+    memcpy(path + scandirlen + 1, *argv, len+1) ;
+    if (do_status(path, !!(golb & 1 << PROCESS_STATUS_GOLB_WITHLOGS))) e = 1 ;
+  }
+  return e ;
+}