lib/util/become_daemon.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    Samba utility functions
   4    Copyright (C) Andrew Tridgell 1992-1998
   5    Copyright (C) Jeremy Allison 2001-2002
   6    Copyright (C) Simo Sorce 2001
   7    Copyright (C) Jim McDonough (jmcd@us.ibm.com)  2003.
   8    Copyright (C) James J Myers 2003
   9
  10    This program is free software; you can redistribute it and/or modify
  11    it under the terms of the GNU General Public License as published by
  12    the Free Software Foundation; either version 3 of the License, or
  13    (at your option) any later version.
  14
  15    This program is distributed in the hope that it will be useful,
  16    but WITHOUT ANY WARRANTY; without even the implied warranty of
  17    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18    GNU General Public License for more details.
  19
  20    You should have received a copy of the GNU General Public License
  21    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  22 */
  23
  24 #include "includes.h"
  25 #include "system/filesys.h"
  26 #include "system/locale.h"
  27 #if HAVE_SYSTEMD
  28 #include <systemd/sd-daemon.h>
  29 #endif
  30
  31 /*******************************************************************
  32  Close the low 3 fd's and open dev/null in their place.
  33 ********************************************************************/
  34
  35 _PUBLIC_ int close_low_fd(int fd)
  36 {
  37 #ifndef VALGRIND
  38         int ret, dev_null;
  39
  40         dev_null = open("/dev/null", O_RDWR, 0);
  41
  42         if ((dev_null == -1) && (errno = ENFILE)) {
  43                 /*
  44                  * Try to free up an fd
  45                  */
  46                 ret = close(fd);
  47                 if (ret != 0) {
  48                         return errno;
  49                 }
  50         }
  51
  52         dev_null = open("/dev/null", O_RDWR, 0);
  53         if (dev_null == -1) {
  54                 dev_null = open("/dev/null", O_WRONLY, 0);
  55         }
  56         if (dev_null == -1) {
  57                 return errno;
  58         }
  59
  60         if (dev_null == fd) {
  61                 /*
  62                  * This can happen in the ENFILE case above
  63                  */
  64                 return 0;
  65         }
  66
  67         ret = dup2(dev_null, fd);
  68         if (ret == -1) {
  69                 int err = errno;
  70                 close(dev_null);
  71                 return err;
  72         }
  73         close(dev_null);
  74 #endif
  75         return 0;
  76 }
  77
  78 _PUBLIC_ void close_low_fds(bool stdin_too, bool stdout_too, bool stderr_too)
  79 {
  80
  81         if (stdin_too) {
  82                 int ret = close_low_fd(0);
  83                 if (ret != 0) {
  84                         DEBUG(0, ("%s: close_low_fd(0) failed: %s\n",
  85                                   __func__, strerror(ret)));
  86                 }
  87         }
  88         if (stdout_too) {
  89                 int ret = close_low_fd(1);
  90                 if (ret != 0) {
  91                         DEBUG(0, ("%s: close_low_fd(1) failed: %s\n",
  92                                   __func__, strerror(ret)));
  93                 }
  94         }
  95         if (stderr_too) {
  96                 int ret = close_low_fd(2);
  97                 if (ret != 0) {
  98                         DEBUG(0, ("%s: close_low_fd(2) failed: %s\n",
  99                                   __func__, strerror(ret)));
 100                 }
 101         }
 102 }
 103
 104 /****************************************************************************
 105  Become a daemon, discarding the controlling terminal.
 106 ****************************************************************************/
 107
 108 _PUBLIC_ void become_daemon(bool do_fork, bool no_process_group, bool log_stdout)
 109 {
 110         pid_t newpid;
 111         if (do_fork) {
 112                 newpid = fork();
 113                 if (newpid) {
 114 #if HAVE_SYSTEMD
 115                         sd_notifyf(0, "READY=0\nSTATUS=Starting process...\nMAINPID=%lu", (unsigned long) newpid);
 116 #endif /* HAVE_SYSTEMD */
 117                         _exit(0);
 118                 }
 119         }
 120
 121         /* detach from the terminal */
 122 #ifdef HAVE_SETSID
 123         if (!no_process_group) setsid();
 124 #elif defined(TIOCNOTTY)
 125         if (!no_process_group) {
 126                 int i = open("/dev/tty", O_RDWR, 0);
 127                 if (i != -1) {
 128                         ioctl(i, (int) TIOCNOTTY, (char *)0);
 129                         close(i);
 130                 }
 131         }
 132 #endif /* HAVE_SETSID */
 133
 134         /* Close fd's 0,1,2 as appropriate. Needed if started by rsh. */
 135         /* stdin must be open if we do not fork, for monitoring for
 136          * close.  stdout must be open if we are logging there, and we
 137          * never close stderr (but debug might dup it onto a log file) */
 138         close_low_fds(do_fork, !log_stdout, false);
 139 }
 140
 141 _PUBLIC_ void exit_daemon(const char *msg, int error)
 142 {
 143 #ifdef HAVE_SYSTEMD
 144         if (msg == NULL) {
 145                 msg = strerror(error);
 146         }
 147
 148         sd_notifyf(0, "STATUS=daemon failed to start: %s\n"
 149                                   "ERRNO=%i",
 150                                   msg,
 151                                   error);
 152 #endif
 153         DEBUG(0, ("STATUS=daemon failed to start: %s, error code %d\n", msg, error));
 154         exit(1);
 155 }
 156
 157 _PUBLIC_ void daemon_ready(const char *name)
 158 {
 159         if (name == NULL) {
 160                 name = "Samba";
 161         }
 162 #ifdef HAVE_SYSTEMD
 163         sd_notifyf(0, "READY=1\nSTATUS=%s: ready to serve connections...", name);
 164 #endif
 165         DEBUG(0, ("STATUS=daemon '%s' finished starting up and ready to serve "
 166                   "connections\n", name));
 167 }