util: Add become_daemon.h
[sfrench/samba-autobuild/.git] / lib / util / become_daemon.c
index eba0cae..7b84299 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)
+_PUBLIC_ void close_low_fds(bool stdin_too, bool stdout_too, bool stderr_too)
 {
-#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 (stdin_too) {
+               int ret = close_low_fd(0);
                if (ret != 0) {
-                       return errno;
+                       DEBUG(0, ("%s: close_low_fd(0) failed: %s\n",
+                                 __func__, strerror(ret)));
                }
        }
-
-       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)
-{
-#ifndef VALGRIND
-       int fd;
-       int i;
-
-       if (stdin_too)
-               close(0);
-       if (stdout_too)
-               close(1);
-
-       if (stderr_too)
-               close(2);
-
-       /* try and use up these file descriptors, so silly
-               library routines writing to stdout etc won't cause havoc */
-       for (i=0;i<3;i++) {
-               if (i == 0 && !stdin_too)
-                       continue;
-               if (i == 1 && !stdout_too)
-                       continue;
-               if (i == 2 && !stderr_too)
-                       continue;
-
-               fd = open("/dev/null",O_RDWR,0);
-               if (fd < 0)
-                       fd = open("/dev/null",O_WRONLY,0);
-               if (fd < 0) {
-                       DEBUG(0,("Can't open /dev/null\n"));
-                       return;
+       if (stdout_too) {
+               int ret = close_low_fd(1);
+               if (ret != 0) {
+                       DEBUG(0, ("%s: close_low_fd(1) failed: %s\n",
+                                 __func__, strerror(ret)));
                }
-               if (fd != i) {
-                       DEBUG(0,("Didn't get file descriptor %d\n",i));
-                       close(fd);
-                       return;
+       }
+       if (stderr_too) {
+               int ret = close_low_fd(2);
+               if (ret != 0) {
+                       DEBUG(0, ("%s: close_low_fd(2) failed: %s\n",
+                                 __func__, strerror(ret)));
                }
        }
-#endif
 }
 
 /****************************************************************************
@@ -125,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);
                }
        }
@@ -154,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);
        }
@@ -173,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));
+}