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