MCOL-4170 Refactor services/systemd units to finish their bootstrap ...

utils/common/pipe.h

@@ -0,0 +1,97 @@
+/* Copyright (C) 2020 MariaDB Corporation
+
+   This program is free software; you can redistribute it and/or
+   modify it under the terms of the GNU General Public License
+   as published by the Free Software Foundation; version 2 of
+   the License.
+
+   This program is distributed in the hope that it will be useful,
+   but WITHOUT ANY WARRANTY; without even the implied warranty of
+   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+   GNU General Public License for more details.
+
+   You should have received a copy of the GNU General Public License
+   along with this program; if not, write to the Free Software
+   Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
+   MA 02110-1301, USA. */
+
+#ifndef PIPE_H_INCLUDED
+#define PIPE_H_INCLUDED
+
+/*
+  A helper class to hold the file descriptors returned from a pipe() call.
+*/
+class Pipe
+{
+  int fd[2];
+public:
+  Pipe()
+  {
+    fd[0]= 0;
+    fd[1]= 0;
+  }
+  ~Pipe()
+  {
+    close();
+  }
+  bool is_open_for_read() const
+  {
+    return fd[0] > 0;
+  }
+  bool is_open_for_write() const
+  {
+    return fd[1] > 0;
+  }
+  bool open()
+  {
+    return ::pipe(fd) == -1;
+  }
+  bool init() // TODO: remove this
+  {
+    return open();
+  }
+  ssize_t read(char *str, size_t nbytes)
+  {
+    return ::read(fd[0], str, nbytes);
+  }
+  ssize_t readtm(const struct timeval &tv, void *buf, size_t nbytes)
+  {
+      fd_set rfds;
+      FD_ZERO(&rfds);
+      FD_SET(fd[0], &rfds);
+      struct timeval tmptv = tv;
+      int retval = select(fd[0] + 1, &rfds, NULL, NULL, &tmptv);
+      if (retval == -1)
+          return -1;
+      if (!retval)
+      {
+        errno= ETIME;
+        return -1;
+      }
+      return ::read(fd[0], buf, nbytes);
+  }
+  ssize_t write(const char *str, size_t nbytes)
+  {
+    return ::write(fd[1], str, nbytes);
+  }
+  ssize_t write(const std::string &str)
+  {
+    return write(str.data(), str.length());
+  }
+  void close()
+  {
+    if (fd[0])
+    {
+      ::close(fd[0]);
+      fd[0]= 0;
+    }
+    if (fd[1])
+    {
+      ::close(fd[1]);
+      fd[1]= 0;
+    }
+  }
+};
+
+
+#endif // PIPE_H_INCLUDED

utils/common/service.h

@@ -0,0 +1,116 @@
+/* Copyright (C) 2020 MariaDB Corporation
+
+   This program is free software; you can redistribute it and/or
+   modify it under the terms of the GNU General Public License
+   as published by the Free Software Foundation; version 2 of
+   the License.
+
+   This program is distributed in the hope that it will be useful,
+   but WITHOUT ANY WARRANTY; without even the implied warranty of
+   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+   GNU General Public License for more details.
+
+   You should have received a copy of the GNU General Public License
+   along with this program; if not, write to the Free Software
+   Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
+   MA 02110-1301, USA. */
+
+#ifndef SERVICE_H_INCLUDED
+#define SERVICE_H_INCLUDED
+
+#include <signal.h>
+#include "pipe.h"
+
+
+class Service
+{
+protected:
+    // The service name, for logging
+    const std::string m_name;
+    // The pipe to send messages from the child to the parent
+    Pipe m_pipe;
+
+
+    static void common_signal_handler_CHLD(int sig)
+    { }
+
+    void InitCommonSignalHandlers()
+    {
+        struct sigaction sa;
+        memset(&sa, 0, sizeof(sa));
+        sa.sa_handler = common_signal_handler_CHLD;
+        sigaction(SIGCHLD, &sa, NULL);
+    }
+
+    int RunForking()
+    {
+        int err;
+        InitCommonSignalHandlers();
+
+        if (m_pipe.open() || (err = fork()) < 0)
+        {
+            // Pipe or fork failed
+            LogErrno();
+            return 1;
+        }
+
+        if (err > 0) // Parent
+          return Parent();
+
+        return Child();
+    }
+
+    virtual int Parent()
+    {
+        char str[100];
+        // Read the message from the child
+        ssize_t nbytes= m_pipe.readtm({120,0}, str, sizeof(str));
+        if (nbytes >= 0)
+        {
+            ParentLogChildMessage(std::string(str, nbytes));
+        }
+        else
+        {
+            // read() failed
+            LogErrno();
+            return 1;
+        }
+        return 0;
+    }
+    
+
+public:
+    Service(const std::string &name)
+     :m_name(name)
+    { }
+    virtual ~Service() { }
+
+    void NotifyServiceStarted()
+    {
+        if (m_pipe.is_open_for_write())
+        {
+            std::ostringstream str;
+            str << m_name << " main process has started";
+            m_pipe.write(str.str());
+        }
+    }
+
+    void NotifyServiceInitializationFailed()
+    {
+        if (m_pipe.is_open_for_write())
+        {
+            std::ostringstream str;
+            str << m_name << " main process initialization failed";
+            m_pipe.write(str.str());
+        }
+    }
+
+    // Used by both Parent and Child to log errors
+    virtual void LogErrno()= 0;
+    // Used by Parent to log an initialization notification message from child
+    virtual void ParentLogChildMessage(const std::string &str)= 0;
+    // The main service process job
+    virtual int Child()= 0;
+};
+
+#endif // SERVICE_H_INCLUDED