/* WARNING: This file was generated by dkct. Changes you make here will be lost if dkct is run again! You should modify the original source and run dkct on it. Original source: dk4dmt.ctr */ /* Copyright (C) 2016-2017, Dirk Krause Redistribution and use in source and binary forms, with or without modification, are permitted provided that the following conditions are met: * Redistributions of source code must retain the above copyright notice, this list of conditions and the following disclaimer. * Redistributions in binary form must reproduce the above opyright notice, this list of conditions and the following disclaimer in the documentation and/or other materials provided with the distribution. * Neither the name of the author nor the names of contributors may be used to endorse or promote products derived from this software without specific prior written permission. THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. */ /** @file dk4dmt.c The dk4dmt module. */ #line 101 "dk4dmt.ctr" #include "dk4conf.h" #include #if DK4_HAVE_SYS_TYPES_H #ifndef SYS_TYPES_H_INCLUDED #include #define SYS_TYPES_H_INCLUDED 1 #endif #endif #if DK4_HAVE_SYS_STAT_H #ifndef SYS_STAT_H_INCLUDED #include #define SYS_STAT_H_INCLUDED 1 #endif #endif #if DK4_HAVE_STDLIB_H #ifndef STDLIB_H_INCLUDED #include #define STDLIB_H_INCLUDED 1 #endif #endif #if DK4_HAVE_UNISTD_H #ifndef UNISTD_H_INCLUDED #include #define UNISTD_H_INCLUDED 1 #endif #endif #if DK4_HAVE_FCNTL_H #ifndef FCNTL_H_INCLUDED #include #define FCNTL_H_INCLUDED 1 #endif #endif #if DK4_HAVE_SIGNAL_H #ifndef SIGNAL_H_INCLUDED #include #define SIGNAL_H_INCLUDED 1 #endif #endif #if DK4_HAVE_IO_H #ifndef IO_H_INCLUDED #include #define IO_H_INCLUDED 1 #endif #endif #if DK4_HAVE_SYSLOG_H #ifndef SYSLOG_H_INCLUDED #include #define SYSLOG_H_INCLUDED 1 #endif #endif #include "dk4types.h" #include "dk4dmt.h" #include "dk4mem.h" #include "dk4mao8d.h" #include "dk4stat8.h" #include "dk4str8.h" #include "dk4fopc8.h" #include "dk4mkdh8.h" #line 175 "dk4dmt.ctr" static const char * const dk4dmt_kw[] = { /* 0 */ "\n", /* 1 */ ": ", /* 2 */ "Failed to create background process!", /* 3 */ "Failed to detach from terminal (missing setsid()/setpgpr())!", /* 4 */ "Failed to detach from terminal (setsid() failed)!", /* 5 */ "Failed to create daemon process!", /* 6 */ "Failed to reset signal handlers (sigemptyset)!", /* 7 */ "Failed to reset signal handlers (sigaction)!", /* 8 */ "Neither sigaction() nor sigset() available!", /* 9 */ "Failed to unblock signals (sigemptyset)!", /* 10 */ "Failed to unblock signals (sigprocmask)!", /* 11 */ "PID file already exists:\n\"", /* 12 */ "\"!", /* 13 */ "Neither setsid() nor setpgrp() available!", /* 14 */ "Failed to detach from terminal (setsid)!", /* 15 */ "Failed to write PID file: \"", /* 16 */ "\"!", /* 17 */ "dk4dmt", /* 18 */ "Failed to open /dev/null for read access!", /* 19 */ "Failed to duplicate /dev/null to standard input!", /* 20 */ "Failed to open /dev/null for write access!", /* 21 */ "Failed to duplicate /dev/null to standard output!", /* 22 */ "Failed to duplicate /dev/null to standard error output!", /* 23 */ "Failed to change into root directory!", /* 24 */ "Failed to remove PID file \"", /* 25 */ "\"!", NULL #line 273 "dk4dmt.ctr" }; static void dk4dmt_syslog_1(const char *prgname, int slf, size_t idx) { #if DK4_HAVE_SYSLOG openlog(((NULL != prgname) ? (prgname) : (dk4dmt_kw[17])), LOG_PID, slf); syslog(LOG_ERR, "%s", dk4dmt_kw[idx]); closelog(); #endif } static void dk4dmt_syslog_3( const char *prgname, int slf, size_t i1, size_t i2, const char *str ) { #if DK4_HAVE_SYSLOG openlog(((NULL != prgname) ? (prgname) : (dk4dmt_kw[17])), LOG_PID, slf); if (NULL != str) { syslog(LOG_ERR, "%s%s%s", dk4dmt_kw[i1], str, dk4dmt_kw[i2]); } else { syslog(LOG_ERR, "%s%s", dk4dmt_kw[i1], dk4dmt_kw[i2]); } closelog(); #endif } static void dk4dmt_log_1(const char *prgname, int slf, int lse, size_t i) { if (0 != lse) { if (NULL != prgname) { fputs(prgname, stderr); fputs(dk4dmt_kw[1], stderr); } fputs(dk4dmt_kw[i], stderr); fputs(dk4dmt_kw[0], stderr); fflush(stderr); } else { dk4dmt_syslog_1(prgname, slf, i); } } static void dk4dmt_log_3( const char *prgname, int slf, int lse, size_t i1, size_t i2, const char *str ) { if (0 != lse) { if (NULL != prgname) { fputs(prgname, stderr); fputs(dk4dmt_kw[1], stderr); } fputs(dk4dmt_kw[i1], stderr); if (NULL != str) { fputs(str, stderr); } fputs(dk4dmt_kw[i2], stderr); fputs(dk4dmt_kw[0], stderr); fflush(stderr); } else { dk4dmt_syslog_3(prgname, slf, i1, i2, str); } } /** Reset all signals to SIG_DFL. @param reason Address of error code variable (1=sigemptyset failed, 2=sigaction failed, 3=no signal handling). @return 1 on success, 0 on error. */ static int dk4dmt_reset_all_signals(int *reason) { #line 361 "dk4dmt.ctr" #if DK4_HAVE_SIGACTION struct sigaction sigact; /* Signal information for sigaction() */ #endif int back = 1; /* Function result */ int maxsig; /* Maximum signal + 1 */ int i; /* Current signal to modify */ int go; #ifdef _NSIG maxsig = _NSIG; #else maxsig = 16; #endif for (i = 1; ((i < maxsig) && (1 == back)); i++) { go = 1; #ifdef SIGKILL if (SIGKILL == i) { go = 0; } #else if (9 == i) { go = 0; } #endif #ifdef SIGSTOP if (SIGSTOP == i) { go = 0; } #else if (19 == i) { go = 0; } #endif #line 388 "dk4dmt.ctr" if (0 != go) { #if DK4_HAVE_SIGACTION DK4_MEMRES(&sigact, sizeof(sigact)); sigact.sa_handler = SIG_DFL; sigact.sa_flags = 0; if (0 != sigemptyset(&(sigact.sa_mask))) { if (16 > i) { back = 0; if (NULL != reason) { #line 397 "dk4dmt.ctr" *reason = 1; } } } if (0 != sigaction(i, &sigact, NULL)) { if (16 > i) { back = 0; if (NULL != reason) { #line 405 "dk4dmt.ctr" *reason = 2; } } } #else #if DK4_HAVE_SIGSET sigset(i, SIG_DFL); #else #if DK4_HAVE_SIGNAL signal(i, SIG_DFL); #else back = 0; if (NULL != reason) { *reason = 3; } #endif #endif #endif } } #line 424 "dk4dmt.ctr" return back; } /** Obtain maximum file descriptor number. @return Maximum file descriptor. */ static int dk4dmt_max_file_descriptor(void) { #if DK4_HAVE_SYS_RESOURCE_H && DK4_HAVE_GETRLIMIT && defined(RLIMIT_NOFILE) struct rlimit rl; #endif int back = 1024; #if DK4_HAVE_SYS_RESOURCE_H && DK4_HAVE_GETRLIMIT && defined(RLIMIT_NOFILE) if (0 == getrlimit(RLIMIT_NOFILE, &rl)) { #if defined(RLIM_INFINITY) if (RLIM_INFINITY != rl.rlim_max) { #endif back = rl.rlim_max; #if defined(RLIM_INFINITY) } #endif } #endif return back; } /** Close all file descriptors except 0 (stdin), 1 (stdout) and 2 (stderr). */ static void dk4dmt_close_non_std_file_descriptors(void) { int maxfd; int fd; maxfd = dk4dmt_max_file_descriptor(); for (fd = 3; fd < maxfd; fd++) { (void)close(fd); } } /** Unblock all signals. @param reason Address of error code variable (1=sigemptyset failed, 2=sigprocmask failed). @return 1 on success, 0 on error. */ static int dk4dmt_unblock_all_signals(int *reason) { sigset_t set; int back = 1; if (0 != sigemptyset(&set)) { back = 0; if (NULL != reason) { *reason = 1; } } if (0 != sigprocmask(SIG_SETMASK, &set, NULL)) { back = 0; if (NULL != reason) { *reason = 2; } } return back; } int dk4dmt_parent_before_fork( const char *prgname, const char *pidfilename, int logstderr, int slf ) { dk4_stat_t stb; int reason = 0; int back = 1; /* Close file descriptors except 0, 1 and 2 */ dk4dmt_close_non_std_file_descriptors(); /* Reset all signal actions to default handlers */ if (0 == dk4dmt_reset_all_signals(&reason)) { back = 0; if (0 != logstderr) { switch (reason) { case 1: { /* ERROR: sigemptyset() failed */ dk4dmt_log_1(prgname, slf, logstderr, 6); } break; case 2: { /* ERROR: sigaction() failed */ dk4dmt_log_1(prgname, slf, logstderr, 7); } break; default: { /* ERROR: No sigaction()/sigset()/signal() function */ dk4dmt_log_1(prgname, slf, logstderr, 8); } break; } } } /* Unblock all signals */ if (0 == dk4dmt_unblock_all_signals(&reason)) { back = 0; if (0 != logstderr) { switch (reason) { case 1: { /* ERROR: sigemptyset() failed */ dk4dmt_log_1(prgname, slf, logstderr, 9); } break; case 2: { /* ERROR: sigprocmask() failed */ dk4dmt_log_1(prgname, slf, logstderr, 10); } break; } } } if ((0 != back) && (NULL != pidfilename)) { if (0 != dk4stat_c8(&stb, pidfilename, NULL)) { back = 0; /* ERROR: PID file exists */ dk4dmt_log_3(prgname, slf, logstderr, 11, 12, pidfilename); } } return back; } int dk4dmt_intermediate_before_fork(const char *prgname, int slf) { int back = 1; #if DK4_HAVE_SETSID if ((pid_t)(-1) == setsid()) { back = 0; dk4dmt_syslog_1(prgname, slf, 14); } #else #if DK4_HAVE_SETPGRP setpgrp(); #else back = 0; dk4dmt_syslog_1(prgname, slf, 13); #endif #endif return back; } int dk4dmt_daemon_after_fork(const char *prgname, const char *pidfilename, int slf) { char pidbuf[16*sizeof(dk4_um_t)]; FILE *fipo; int fdin; /* File descriptor for input */ int fdout; /* File descriptor for output */ int res; /* Operation result */ int back = 1; if (NULL != pidfilename) { res = dk4ma_write_c8_decimal_unsigned( pidbuf, sizeof(pidbuf), (dk4_um_t)getpid(), 0, NULL ); if (0 != res) { if (0 != dk4str8_cat_s(pidbuf, sizeof(pidbuf), dk4dmt_kw[0], NULL)) { (void)dk4mkdir_hierarchy_c8(pidfilename, 0, NULL); fipo = dk4fopen_c8(pidfilename, "w", DK4_FOPEN_SC_PRIVILEGED, NULL); if (NULL != fipo) { fputs(pidbuf, fipo); fclose(fipo); } else { dk4dmt_syslog_3(prgname, slf, 15, 16, pidfilename); back = 0; } } } } /* Connect /dev/null to stdin stout stderr or close them */ fdin = open("/dev/null", O_RDONLY); if (-1 < fdin) { if (-1 == dup2(fdin, 0)) { back = 0; dk4dmt_syslog_1(prgname, slf, 19); } close(fdin); } else { back = 0; dk4dmt_syslog_1(prgname, slf, 18); } fdout = open("/dev/null", O_WRONLY); if (-1 < fdout) { if (-1 == dup2(fdout, 1)) { back = 0; dk4dmt_syslog_1(prgname, slf, 21); } if (-1 == dup2(fdout, 2)) { back = 0; dk4dmt_syslog_1(prgname, slf, 22); } close(fdout); } else { back = 0; dk4dmt_syslog_1(prgname, slf, 20); } /* Use exact permissions when creating new files or directories. */ umask(0); /* Change to root directory, so the previous current directory can be unmounted without blocking. */ if (0 != chdir("/")) { back = 0; dk4dmt_syslog_1(prgname, slf, 23); } /* Indicate success or error. */ return back; } void dk4dmt_daemon_end(const char *prgname, const char *pidfilename, int slf) { if (NULL != pidfilename) { if (0 != unlink(pidfilename)) { dk4dmt_syslog_3(prgname, slf, 24, 25, pidfilename); } } }