1*9203SMark.Logan@Sun.COM /*******************************************************************************
2*9203SMark.Logan@Sun.COM * Copyright (C) 2004-2008 Intel Corp. All rights reserved.
3*9203SMark.Logan@Sun.COM *
4*9203SMark.Logan@Sun.COM * Redistribution and use in source and binary forms, with or without
5*9203SMark.Logan@Sun.COM * modification, are permitted provided that the following conditions are met:
6*9203SMark.Logan@Sun.COM *
7*9203SMark.Logan@Sun.COM * - Redistributions of source code must retain the above copyright notice,
8*9203SMark.Logan@Sun.COM * this list of conditions and the following disclaimer.
9*9203SMark.Logan@Sun.COM *
10*9203SMark.Logan@Sun.COM * - Redistributions in binary form must reproduce the above copyright notice,
11*9203SMark.Logan@Sun.COM * this list of conditions and the following disclaimer in the documentation
12*9203SMark.Logan@Sun.COM * and/or other materials provided with the distribution.
13*9203SMark.Logan@Sun.COM *
14*9203SMark.Logan@Sun.COM * - Neither the name of Intel Corp. nor the names of its
15*9203SMark.Logan@Sun.COM * contributors may be used to endorse or promote products derived from this
16*9203SMark.Logan@Sun.COM * software without specific prior written permission.
17*9203SMark.Logan@Sun.COM *
18*9203SMark.Logan@Sun.COM * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS ``AS IS''
19*9203SMark.Logan@Sun.COM * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
20*9203SMark.Logan@Sun.COM * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
21*9203SMark.Logan@Sun.COM * ARE DISCLAIMED. IN NO EVENT SHALL Intel Corp. OR THE CONTRIBUTORS
22*9203SMark.Logan@Sun.COM * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
23*9203SMark.Logan@Sun.COM * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
24*9203SMark.Logan@Sun.COM * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
25*9203SMark.Logan@Sun.COM * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
26*9203SMark.Logan@Sun.COM * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
27*9203SMark.Logan@Sun.COM * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
28*9203SMark.Logan@Sun.COM * POSSIBILITY OF SUCH DAMAGE.
29*9203SMark.Logan@Sun.COM *******************************************************************************/
30*9203SMark.Logan@Sun.COM
31*9203SMark.Logan@Sun.COM #ifdef HAVE_CONFIG_H
32*9203SMark.Logan@Sun.COM #include "config.h"
33*9203SMark.Logan@Sun.COM #endif
34*9203SMark.Logan@Sun.COM
35*9203SMark.Logan@Sun.COM #ifdef DAEMON
36*9203SMark.Logan@Sun.COM
37*9203SMark.Logan@Sun.COM #include <cstdio>
38*9203SMark.Logan@Sun.COM #include <cstdlib>
39*9203SMark.Logan@Sun.COM #include <cstring>
40*9203SMark.Logan@Sun.COM #include <cerrno>
41*9203SMark.Logan@Sun.COM #include <csignal>
42*9203SMark.Logan@Sun.COM #include <unistd.h>
43*9203SMark.Logan@Sun.COM #include <sys/types.h>
44*9203SMark.Logan@Sun.COM #include <sys/stat.h>
45*9203SMark.Logan@Sun.COM #include <fcntl.h>
46*9203SMark.Logan@Sun.COM #include <syslog.h>
47*9203SMark.Logan@Sun.COM #include <pwd.h>
48*9203SMark.Logan@Sun.COM
49*9203SMark.Logan@Sun.COM #include "daemonize.h"
50*9203SMark.Logan@Sun.COM
child_handler(int signum)51*9203SMark.Logan@Sun.COM static RETSIGTYPE child_handler(int signum)
52*9203SMark.Logan@Sun.COM {
53*9203SMark.Logan@Sun.COM switch (signum) {
54*9203SMark.Logan@Sun.COM case SIGALRM:
55*9203SMark.Logan@Sun.COM exit(EXIT_FAILURE);
56*9203SMark.Logan@Sun.COM break;
57*9203SMark.Logan@Sun.COM case SIGUSR1:
58*9203SMark.Logan@Sun.COM exit(EXIT_SUCCESS);
59*9203SMark.Logan@Sun.COM break;
60*9203SMark.Logan@Sun.COM case SIGCHLD:
61*9203SMark.Logan@Sun.COM exit(EXIT_FAILURE);
62*9203SMark.Logan@Sun.COM break;
63*9203SMark.Logan@Sun.COM }
64*9203SMark.Logan@Sun.COM }
65*9203SMark.Logan@Sun.COM
daemonize()66*9203SMark.Logan@Sun.COM void daemonize()
67*9203SMark.Logan@Sun.COM {
68*9203SMark.Logan@Sun.COM pid_t pid, sid, parent;
69*9203SMark.Logan@Sun.COM
70*9203SMark.Logan@Sun.COM /* Drop user if there is one, and we were run as root */
71*9203SMark.Logan@Sun.COM if (getuid() == 0 || geteuid() == 0) {
72*9203SMark.Logan@Sun.COM struct passwd *pw = getpwnam(RUN_AS_USER);
73*9203SMark.Logan@Sun.COM if (pw) {
74*9203SMark.Logan@Sun.COM //syslog(LOG_NOTICE, "setting user to " RUN_AS_USER);
75*9203SMark.Logan@Sun.COM setuid(pw->pw_uid);
76*9203SMark.Logan@Sun.COM }
77*9203SMark.Logan@Sun.COM }
78*9203SMark.Logan@Sun.COM
79*9203SMark.Logan@Sun.COM /* Trap signals that we expect to recieve */
80*9203SMark.Logan@Sun.COM signal(SIGCHLD, child_handler);
81*9203SMark.Logan@Sun.COM signal(SIGUSR1, child_handler);
82*9203SMark.Logan@Sun.COM signal(SIGALRM, child_handler);
83*9203SMark.Logan@Sun.COM
84*9203SMark.Logan@Sun.COM /* Fork off the parent process */
85*9203SMark.Logan@Sun.COM pid = fork();
86*9203SMark.Logan@Sun.COM if (pid < 0) {
87*9203SMark.Logan@Sun.COM syslog(LOG_ERR, "unable to fork daemon, code=%d (%s)",
88*9203SMark.Logan@Sun.COM errno, strerror(errno));
89*9203SMark.Logan@Sun.COM exit(EXIT_FAILURE);
90*9203SMark.Logan@Sun.COM }
91*9203SMark.Logan@Sun.COM /* If we got a good PID, then we can exit the parent process. */
92*9203SMark.Logan@Sun.COM if (pid > 0) {
93*9203SMark.Logan@Sun.COM
94*9203SMark.Logan@Sun.COM /* Wait for confirmation from the child via SIGTERM or SIGCHLD, or
95*9203SMark.Logan@Sun.COM for two seconds to elapse (SIGALRM). pause() should not return. */
96*9203SMark.Logan@Sun.COM alarm(2);
97*9203SMark.Logan@Sun.COM pause();
98*9203SMark.Logan@Sun.COM
99*9203SMark.Logan@Sun.COM exit(EXIT_FAILURE);
100*9203SMark.Logan@Sun.COM }
101*9203SMark.Logan@Sun.COM
102*9203SMark.Logan@Sun.COM /* At this point we are executing as the child process */
103*9203SMark.Logan@Sun.COM parent = getppid();
104*9203SMark.Logan@Sun.COM
105*9203SMark.Logan@Sun.COM signal(SIGCHLD, SIG_DFL);
106*9203SMark.Logan@Sun.COM signal(SIGUSR1, SIG_DFL);
107*9203SMark.Logan@Sun.COM signal(SIGALRM, SIG_DFL);
108*9203SMark.Logan@Sun.COM setTerminationHandler();
109*9203SMark.Logan@Sun.COM
110*9203SMark.Logan@Sun.COM /* Create a new SID for the child process */
111*9203SMark.Logan@Sun.COM sid = setsid();
112*9203SMark.Logan@Sun.COM if (sid < 0) {
113*9203SMark.Logan@Sun.COM syslog(LOG_ERR,
114*9203SMark.Logan@Sun.COM "unable to create a new session, code %d (%s)",
115*9203SMark.Logan@Sun.COM errno, strerror(errno));
116*9203SMark.Logan@Sun.COM exit(EXIT_FAILURE);
117*9203SMark.Logan@Sun.COM }
118*9203SMark.Logan@Sun.COM
119*9203SMark.Logan@Sun.COM /* Change the current working directory. This prevents the current
120*9203SMark.Logan@Sun.COM directory from being locked; hence not being able to remove it. */
121*9203SMark.Logan@Sun.COM if ((chdir("/")) < 0) {
122*9203SMark.Logan@Sun.COM syslog(LOG_ERR,
123*9203SMark.Logan@Sun.COM "unable to change directory to %s, code %d (%s)",
124*9203SMark.Logan@Sun.COM "/", errno, strerror(errno));
125*9203SMark.Logan@Sun.COM exit(EXIT_FAILURE);
126*9203SMark.Logan@Sun.COM }
127*9203SMark.Logan@Sun.COM
128*9203SMark.Logan@Sun.COM /* Redirect standard files to /dev/null */
129*9203SMark.Logan@Sun.COM freopen("/dev/null", "r", stdin);
130*9203SMark.Logan@Sun.COM freopen("/dev/null", "w", stdout);
131*9203SMark.Logan@Sun.COM freopen("/dev/null", "w", stderr);
132*9203SMark.Logan@Sun.COM
133*9203SMark.Logan@Sun.COM /* Tell the parent process that we are A-okay */
134*9203SMark.Logan@Sun.COM kill(parent, SIGUSR1);
135*9203SMark.Logan@Sun.COM }
136*9203SMark.Logan@Sun.COM
137*9203SMark.Logan@Sun.COM #endif //DAEMON
138*9203SMark.Logan@Sun.COM
139