util: Add become_daemon.h
[sfrench/samba-autobuild/.git] / lib / util / become_daemon.c
index d940cd7d33eb7021ef8cf65a0ad77c75dd927100..7b84299dc2841788c15f94ae5515a8732194caea 100644 (file)
    along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
-#include "includes.h"
+#include "replace.h"
 #include "system/filesys.h"
 #include "system/locale.h"
-#if HAVE_SYSTEMD
+#if defined(HAVE_LIBSYSTEMD_DAEMON) || defined(HAVE_LIBSYSTEMD)
 #include <systemd/sd-daemon.h>
 #endif
 
+#include "close_low_fd.h"
+#include "debug.h"
+
+#include "become_daemon.h"
+
 /*******************************************************************
  Close the low 3 fd's and open dev/null in their place.
 ********************************************************************/
 
-_PUBLIC_ int close_low_fd(int fd)
-{
-#ifndef VALGRIND
-       int ret, dev_null;
-
-       dev_null = open("/dev/null", O_RDWR, 0);
-
-       if ((dev_null == -1) && (errno = ENFILE)) {
-               /*
-                * Try to free up an fd
-                */
-               ret = close(fd);
-               if (ret != 0) {
-                       return errno;
-               }
-       }
-
-       dev_null = open("/dev/null", O_RDWR, 0);
-       if (dev_null == -1) {
-               dev_null = open("/dev/null", O_WRONLY, 0);
-       }
-       if (dev_null == -1) {
-               return errno;
-       }
-
-       if (dev_null == fd) {
-               /*
-                * This can happen in the ENFILE case above
-                */
-               return 0;
-       }
-
-       ret = dup2(dev_null, fd);
-       if (ret == -1) {
-               int err = errno;
-               close(dev_null);
-               return err;
-       }
-       close(dev_null);
-#endif
-       return 0;
-}
-
 _PUBLIC_ void close_low_fds(bool stdin_too, bool stdout_too, bool stderr_too)
 {
 
@@ -111,9 +73,9 @@ _PUBLIC_ void become_daemon(bool do_fork, bool no_process_group, bool log_stdout
        if (do_fork) {
                newpid = fork();
                if (newpid) {
-#if HAVE_SYSTEMD
+#if defined(HAVE_LIBSYSTEMD_DAEMON) || defined(HAVE_LIBSYSTEMD)
                        sd_notifyf(0, "READY=0\nSTATUS=Starting process...\nMAINPID=%lu", (unsigned long) newpid);
-#endif /* HAVE_SYSTEMD */
+#endif /* HAVE_LIBSYSTEMD_DAEMON */
                        _exit(0);
                }
        }
@@ -140,7 +102,7 @@ _PUBLIC_ void become_daemon(bool do_fork, bool no_process_group, bool log_stdout
 
 _PUBLIC_ void exit_daemon(const char *msg, int error)
 {
-#ifdef HAVE_SYSTEMD
+#if defined(HAVE_LIBSYSTEMD_DAEMON) || defined(HAVE_LIBSYSTEMD)
        if (msg == NULL) {
                msg = strerror(error);
        }
@@ -159,9 +121,20 @@ _PUBLIC_ void daemon_ready(const char *name)
        if (name == NULL) {
                name = "Samba";
        }
-#ifdef HAVE_SYSTEMD
+#if defined(HAVE_LIBSYSTEMD_DAEMON) || defined(HAVE_LIBSYSTEMD)
        sd_notifyf(0, "READY=1\nSTATUS=%s: ready to serve connections...", name);
 #endif
        DEBUG(0, ("STATUS=daemon '%s' finished starting up and ready to serve "
                  "connections\n", name));
 }
+
+_PUBLIC_ void daemon_status(const char *name, const char *msg)
+{
+       if (name == NULL) {
+               name = "Samba";
+       }
+#if defined(HAVE_LIBSYSTEMD_DAEMON) || defined(HAVE_LIBSYSTEMD)
+       sd_notifyf(0, "\nSTATUS=%s: %s", name, msg);
+#endif
+       DEBUG(0, ("STATUS=daemon '%s' : %s", name, msg));
+}