Implement "s6 process"

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

1 files changed, 63 insertions, 0 deletions

diff --git a/src/s6/process_util.c b/src/s6/process_util.c
new file mode 100644
index 0000000..34b9196
--- /dev/null
+++ b/src/s6/process_util.c
@@ -0,0 +1,63 @@
+/* ISC license. */
+
+#include <string.h>
+#include <sys/stat.h>
+#include <errno.h>
+
+#include <skalibs/posixplz.h>
+#include <skalibs/strerr.h>
+#include <skalibs/cspawn.h>
+#include <skalibs/djbunix.h>
+
+#include <s6/config.h>
+
+#include "s6-internal.h"
+
+static int check_service (char const *name, size_t scandirlen)
+{
+  struct stat st ;
+  size_t namelen = strlen(name) ;
+  char path[scandirlen + namelen + 2] ;
+  memcpy(path, g->scandir, scandirlen) ;
+  path[scandirlen] = '/' ;
+  memcpy(path + scandirlen + 1, name, namelen) ;
+  path[scandirlen + 1 + namelen] = 0 ;
+  return stat(path, &st) == -1 ? errno == ENOENT ? 0 : -1 : !!S_ISDIR(st.st_mode) ;
+}
+
+void process_check_services (char const *const *argv, size_t argc)
+{
+  size_t scandirlen = strlen(g->scandir) ;
+  for (size_t i = 0 ; i < argc ; i++)
+  {
+    int r = check_service(argv[i], scandirlen) ;
+    if (r == -1)
+      strerr_diefu4sys(111, "stat ", g->scandir, "/", argv[i]) ;
+    else if (!r)
+      strerr_dief3x(100, argv[i], "is not registered as a supervised service in ", g->scandir) ;
+  }
+}
+
+int process_send_svc (char const *svcopt, char const *const *argv, size_t argc)
+{
+  char const *newargv[5] = { S6_EXTBINPREFIX "s6-svc", svcopt, "--", 0, 0 } ;
+  size_t scandirlen = strlen(g->scandir) ;
+  int wstat ;
+  pid_t pids[argc] ;
+
+  for (size_t i = 0 ; i < argc ; i++)
+  {
+    size_t arglen = strlen(argv[i]) ;
+    char path[scandirlen + arglen + 2] ;
+    memcpy(path, g->scandir, scandirlen) ;
+    path[scandirlen] = '/' ;
+    memcpy(path + scandirlen + 1, argv[i], arglen) ;
+    path[scandirlen + 1 + arglen] = 0 ;
+    newargv[3] = path ;
+    pids[i] = cspawn(newargv[0], newargv, (char const *const *)environ, 0, 0, 0) ;
+    if (!pids[i])
+      strerr_diefu4sys(111, "spawn ", newargv[0], " command for service ", argv[i]) ;
+  }
+  waitn_posix(pids, argc, &wstat) ;
+  return 0 ;
+}