lib/daemonize.c

   1 /*
   2  * This code is mainly taken from Doug Potter's page
   3  *
   4  * http://www-theorie.physik.unizh.ch/~dpotter/howto/daemonize
   5  *
   6  * I contacted him 2007-04-16 about the license for the original code,
   7  * he replied it is Public Domain.  Use the URL above to get the original
   8  * Public Domain version if you want it.
   9  *
  10  * This version is LGPL2 and is (c)2006 - 2013 Andy Green <andy@warmcat.com>
  11  */
  12
  13 #include <stdlib.h>
  14 #include <string.h>
  15 #include <stdio.h>
  16 #include <signal.h>
  17 #include <sys/types.h>
  18 #include <sys/stat.h>
  19 #include <fcntl.h>
  20 #include <limits.h>
  21 #include <unistd.h>
  22 #include <errno.h>
  23
  24 static int pid_daemon;
  25 static char lock_path[PATH_MAX];
  26
  27 static void
  28 child_handler(int signum)
  29 {
  30         int fd;
  31         int len;
  32         int sent;
  33         char sz[20];
  34
  35         switch (signum) {
  36
  37         case SIGALRM: /* timedout daemonizing */
  38                 exit(1);
  39                 break;
  40
  41         case SIGUSR1: /* positive confirmation we daemonized well */
  42                 /* Create the lock file as the current user */
  43
  44                 fd = open(lock_path, O_TRUNC | O_RDWR | O_CREAT, 0640);
  45                 if (fd < 0) {
  46                         fprintf(stderr, "unable to create lock"
  47                                 " file %s, code=%d (%s)",
  48                                 lock_path, errno, strerror(errno));
  49                         exit(1);
  50                 }
  51                 len = sprintf(sz, "%u", pid_daemon);
  52                 sent = write(fd, sz, len);
  53                 if (sent != len)
  54                         fprintf(stderr, "unable write pid to lock"
  55                                 " file %s, code=%d (%s)",
  56                                         lock_path, errno, strerror(errno));
  57
  58                close(fd);
  59                exit(!!(sent == len));
  60
  61         case SIGCHLD: /* daemonization failed */
  62                 exit(1);
  63                 break;
  64         }
  65 }
  66
  67 /*
  68  * You just need to call this from your main(), when it
  69  * returns you are all set "in the background" decoupled
  70  * from the console you were started from.
  71  *
  72  * The process context you called from has been terminated then.
  73  */
  74
  75 int
  76 lws_daemonize(const char *_lock_path)
  77 {
  78         pid_t sid, parent;
  79
  80         /* already a daemon */
  81         if (getppid() == 1)
  82                 return (1);
  83
  84         strncpy(lock_path, _lock_path, sizeof lock_path);
  85         lock_path[sizeof(lock_path) - 1] = '\0';
  86
  87         /* Trap signals that we expect to recieve */
  88         signal(SIGCHLD, child_handler); /* died */
  89         signal(SIGUSR1, child_handler); /* was happy */
  90         signal(SIGALRM, child_handler); /* timeout daemonizing */
  91
  92         /* Fork off the parent process */
  93         pid_daemon = fork();
  94         if (pid_daemon < 0) {
  95                 fprintf(stderr, "unable to fork daemon, code=%d (%s)",
  96                     errno, strerror(errno));
  97                 exit(1);
  98         }
  99
 100         /* If we got a good PID, then we can exit the parent process. */
 101         if (pid_daemon > 0) {
 102
 103                 /*
 104                  * Wait for confirmation signal from the child via
 105                  * SIGCHILD / USR1, or for two seconds to elapse
 106                  * (SIGALRM).  pause() should not return.
 107                  */
 108                 alarm(2);
 109
 110                 pause();
 111                 /* should not be reachable */
 112                 exit(1);
 113         }
 114
 115         /* At this point we are executing as the child process */
 116         parent = getppid();
 117
 118         /* Cancel certain signals */
 119         signal(SIGCHLD, SIG_DFL); /* A child process dies */
 120         signal(SIGTSTP, SIG_IGN); /* Various TTY signals */
 121         signal(SIGTTOU, SIG_IGN);
 122         signal(SIGTTIN, SIG_IGN);
 123         signal(SIGHUP, SIG_IGN); /* Ignore hangup signal */
 124
 125         /* Change the file mode mask */
 126         umask(0);
 127
 128         /* Create a new SID for the child process */
 129         sid = setsid();
 130         if (sid < 0) {
 131                 fprintf(stderr,
 132                         "unable to create a new session, code %d (%s)",
 133                         errno, strerror(errno));
 134                 exit(1);
 135         }
 136
 137         /*
 138          * Change the current working directory.  This prevents the current
 139          * directory from being locked; hence not being able to remove it.
 140          */
 141         if (chdir("/") < 0) {
 142                 fprintf(stderr,
 143                         "unable to change directory to %s, code %d (%s)",
 144                         "/", errno, strerror(errno));
 145                 exit(1);
 146         }
 147
 148         /* Redirect standard files to /dev/null */
 149         if (!freopen("/dev/null", "r", stdin))
 150                 fprintf(stderr, "unable to freopen() stdin, code %d (%s)",
 151                                                        errno, strerror(errno));
 152
 153         if (!freopen("/dev/null", "w", stdout))
 154                 fprintf(stderr, "unable to freopen() stdout, code %d (%s)",
 155                                                        errno, strerror(errno));
 156
 157         if (!freopen("/dev/null", "w", stderr))
 158                 fprintf(stderr, "unable to freopen() stderr, code %d (%s)",
 159                                                        errno, strerror(errno));
 160
 161         /* Tell the parent process that we are A-okay */
 162         kill(parent, SIGUSR1);
 163
 164         /* return to continue what is now "the daemon" */
 165
 166         return (0);
 167 }
 168