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 "replace.h"
  25 #include "system/filesys.h"
  26 #include "system/locale.h"
  27 #if defined(HAVE_LIBSYSTEMD_DAEMON) || defined(HAVE_LIBSYSTEMD)
  28 #include <systemd/sd-daemon.h>
  29 #endif
  30
  31 #include "close_low_fd.h"
  32 #include "debug.h"
  33
  34 #include "become_daemon.h"
  35
  36 /*******************************************************************
  37  Close the low 3 fd's and open dev/null in their place.
  38 ********************************************************************/
  39
  40 _PUBLIC_ void close_low_fds(bool stdin_too, bool stdout_too, bool stderr_too)
  41 {
  42
  43         if (stdin_too) {
  44                 int ret = close_low_fd(0);
  45                 if (ret != 0) {
  46                         DEBUG(0, ("%s: close_low_fd(0) failed: %s\n",
  47                                   __func__, strerror(ret)));
  48                 }
  49         }
  50         if (stdout_too) {
  51                 int ret = close_low_fd(1);
  52                 if (ret != 0) {
  53                         DEBUG(0, ("%s: close_low_fd(1) failed: %s\n",
  54                                   __func__, strerror(ret)));
  55                 }
  56         }
  57         if (stderr_too) {
  58                 int ret = close_low_fd(2);
  59                 if (ret != 0) {
  60                         DEBUG(0, ("%s: close_low_fd(2) failed: %s\n",
  61                                   __func__, strerror(ret)));
  62                 }
  63         }
  64 }
  65
  66 /****************************************************************************
  67  Become a daemon, discarding the controlling terminal.
  68 ****************************************************************************/
  69
  70 _PUBLIC_ void become_daemon(bool do_fork, bool no_process_group, bool log_stdout)
  71 {
  72         pid_t newpid;
  73         if (do_fork) {
  74                 newpid = fork();
  75                 if (newpid) {
  76 #if defined(HAVE_LIBSYSTEMD_DAEMON) || defined(HAVE_LIBSYSTEMD)
  77                         sd_notifyf(0, "READY=0\nSTATUS=Starting process...\nMAINPID=%lu", (unsigned long) newpid);
  78 #endif /* HAVE_LIBSYSTEMD_DAEMON */
  79                         _exit(0);
  80                 }
  81         }
  82
  83         /* detach from the terminal */
  84 #ifdef HAVE_SETSID
  85         if (!no_process_group) setsid();
  86 #elif defined(TIOCNOTTY)
  87         if (!no_process_group) {
  88                 int i = open("/dev/tty", O_RDWR, 0);
  89                 if (i != -1) {
  90                         ioctl(i, (int) TIOCNOTTY, (char *)0);
  91                         close(i);
  92                 }
  93         }
  94 #endif /* HAVE_SETSID */
  95
  96         /* Close fd's 0,1,2 as appropriate. Needed if started by rsh. */
  97         /* stdin must be open if we do not fork, for monitoring for
  98          * close.  stdout must be open if we are logging there, and we
  99          * never close stderr (but debug might dup it onto a log file) */
 100         close_low_fds(do_fork, !log_stdout, false);
 101 }
 102
 103 _PUBLIC_ void exit_daemon(const char *msg, int error)
 104 {
 105 #if defined(HAVE_LIBSYSTEMD_DAEMON) || defined(HAVE_LIBSYSTEMD)
 106         if (msg == NULL) {
 107                 msg = strerror(error);
 108         }
 109
 110         sd_notifyf(0, "STATUS=daemon failed to start: %s\n"
 111                                   "ERRNO=%i",
 112                                   msg,
 113                                   error);
 114 #endif
 115         DEBUG(0, ("STATUS=daemon failed to start: %s, error code %d\n", msg, error));
 116         exit(1);
 117 }
 118
 119 _PUBLIC_ void daemon_ready(const char *name)
 120 {
 121         if (name == NULL) {
 122                 name = "Samba";
 123         }
 124 #if defined(HAVE_LIBSYSTEMD_DAEMON) || defined(HAVE_LIBSYSTEMD)
 125         sd_notifyf(0, "READY=1\nSTATUS=%s: ready to serve connections...", name);
 126 #endif
 127         DEBUG(0, ("STATUS=daemon '%s' finished starting up and ready to serve "
 128                   "connections\n", name));
 129 }
 130
 131 _PUBLIC_ void daemon_status(const char *name, const char *msg)
 132 {
 133         if (name == NULL) {
 134                 name = "Samba";
 135         }
 136 #if defined(HAVE_LIBSYSTEMD_DAEMON) || defined(HAVE_LIBSYSTEMD)
 137         sd_notifyf(0, "\nSTATUS=%s: %s", name, msg);
 138 #endif
 139         DEBUG(0, ("STATUS=daemon '%s' : %s", name, msg));
 140 }