#include <signal.h>
#include <errno.h>
#include <string.h>
#include <dfork.h>
#include <dsignal.h>
#include <dlog.h>
#include <dpid.h>
int main(int argc, char *argv[]) {
pid_t pid;
daemon_pid_file_ident = daemon_log_ident = daemon_ident_from_argv0(argv[0]);
if (argc >= 2 && !strcmp(argv[1], "-k")) {
int ret;
#ifdef DAEMON_PID_FILE_KILL_WAIT_AVAILABLE
if ((ret = daemon_pid_file_kill_wait(SIGINT, 5)) < 0)
#else
if ((ret = daemon_pid_file_kill(SIGINT)) < 0)
#endif
daemon_log(LOG_WARNING, "Failed to kill daemon");
return ret < 0 ? 1 : 0;
}
if ((pid = daemon_pid_file_is_running()) >= 0) {
daemon_log(LOG_ERR, "Daemon already running on PID file %u", pid);
return 1;
}
daemon_retval_init();
if ((pid = daemon_fork()) < 0) {
daemon_retval_done();
return 1;
} else if (pid) {
int ret;
if ((ret = daemon_retval_wait(20)) < 0) {
daemon_log(LOG_ERR, "Could not recieve return value from daemon process.");
return 255;
}
daemon_log(ret != 0 ? LOG_ERR : LOG_INFO, "Daemon returned %i as return value.", ret);
return ret;
} else {
int fd, quit = 0;
fd_set fds;
if (daemon_pid_file_create() < 0) {
daemon_log(LOG_ERR, "Could not create PID file (%s).", strerror(errno));
daemon_retval_send(1);
goto finish;
}
if (daemon_signal_init(SIGINT, SIGQUIT, SIGHUP, 0) < 0) {
daemon_log(LOG_ERR, "Could not register signal handlers (%s).", strerror(errno));
daemon_retval_send(2);
goto finish;
}
daemon_retval_send(0);
daemon_log(LOG_INFO, "Sucessfully started");
FD_ZERO(&fds);
FD_SET(fd = daemon_signal_fd(), &fds);
while (!quit) {
fd_set fds2 = fds;
if (select(FD_SETSIZE, &fds2, 0, 0, 0) < 0) {
if (errno == EINTR)
continue;
daemon_log(LOG_ERR, "select(): %s", strerror(errno));
break;
}
if (FD_ISSET(fd, &fds)) {
int sig;
if ((sig = daemon_signal_next()) <= 0) {
daemon_log(LOG_ERR, "daemon_signal_next() failed.");
break;
}
switch (sig) {
case SIGINT:
case SIGQUIT:
daemon_log(LOG_WARNING, "Got SIGINT or SIGQUIT");
quit = 1;
break;
case SIGHUP:
daemon_log(LOG_INFO, "Got a HUP");
break;
}
}
}
finish:
daemon_log(LOG_INFO, "Exiting...");
daemon_signal_done();
daemon_pid_file_remove();
return 0;
}
}