-rw-r--r-- | src/dudki.cc | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/src/dudki.cc b/src/dudki.cc index 91a3342..1f95be4 100644 --- a/src/dudki.cc +++ b/src/dudki.cc | |||
@@ -1,71 +1,73 @@ | |||
1 | #include <unistd.h> | 1 | #include <unistd.h> |
2 | #include <signal.h> | 2 | #include <signal.h> |
3 | #include <syslog.h> | 3 | #include <syslog.h> |
4 | #include <errno.h> | 4 | #include <errno.h> |
5 | #include <iostream> | 5 | #include <iostream> |
6 | #include <fstream> | 6 | #include <fstream> |
7 | #include <stdexcept> | 7 | #include <stdexcept> |
8 | #include <cstring> | ||
9 | #include <stdlib.h> | ||
8 | using namespace std; | 10 | using namespace std; |
9 | #include "configuration.h" | 11 | #include "configuration.h" |
10 | #include "util.h" | 12 | #include "util.h" |
11 | 13 | ||
12 | #include "config.h" | 14 | #include "config.h" |
13 | #ifdef HAVE_GETOPT_H | 15 | #ifdef HAVE_GETOPT_H |
14 | # include <getopt.h> | 16 | # include <getopt.h> |
15 | #endif | 17 | #endif |
16 | 18 | ||
17 | #ifndef DEFAULT_CONF_FILE | 19 | #ifndef DEFAULT_CONF_FILE |
18 | # define DEFAULT_CONF_FILE "/etc/dudki.conf" | 20 | # define DEFAULT_CONF_FILE "/etc/dudki.conf" |
19 | #endif | 21 | #endif |
20 | 22 | ||
21 | #define PHEADER PACKAGE " Version " VERSION | 23 | #define PHEADER PACKAGE " Version " VERSION |
22 | #define PCOPY "Copyright (c) 2004-2006 Klever Group" | 24 | #define PCOPY "Copyright (c) 2004-2006 Klever Group" |
23 | 25 | ||
24 | bool finishing = false; | 26 | bool finishing = false; |
25 | bool restarting = false; | 27 | bool restarting = false; |
26 | static char **_argv = NULL; | 28 | static char **_argv = NULL; |
27 | 29 | ||
28 | static void lethal_signal_handler(int signum) { | 30 | static void lethal_signal_handler(int signum) { |
29 | syslog(LOG_NOTICE,"Lethal signal received. Terminating."); | 31 | syslog(LOG_NOTICE,"Lethal signal received. Terminating."); |
30 | finishing = true; | 32 | finishing = true; |
31 | } | 33 | } |
32 | static void sighup_handler(int signum) { | 34 | static void sighup_handler(int signum) { |
33 | syslog(LOG_NOTICE,"SUGHUP received, reloading."); | 35 | syslog(LOG_NOTICE,"SUGHUP received, reloading."); |
34 | restarting = finishing = true; | 36 | restarting = finishing = true; |
35 | } | 37 | } |
36 | 38 | ||
37 | void check_herd(configuration& config) { | 39 | void check_herd(configuration& config) { |
38 | process::prepare_herd(); | 40 | process::prepare_herd(); |
39 | for(processes_t::iterator i=config.processes.begin();i!=config.processes.end();++i) | 41 | for(processes_t::iterator i=config.processes.begin();i!=config.processes.end();++i) |
40 | i->second.check(i->first,config); | 42 | i->second.check(i->first,config); |
41 | process::unprepare_herd(); | 43 | process::unprepare_herd(); |
42 | } | 44 | } |
43 | 45 | ||
44 | void signal_self(const configuration& config,int signum) { | 46 | void signal_self(const configuration& config,int signum) { |
45 | ifstream pids(config.pidfile.c_str(),ios::in); | 47 | ifstream pids(config.pidfile.c_str(),ios::in); |
46 | if(!pids) | 48 | if(!pids) |
47 | throw runtime_error("Can't detect running instance"); | 49 | throw runtime_error("Can't detect running instance"); |
48 | pid_t pid = 0; | 50 | pid_t pid = 0; |
49 | pids >> pid; | 51 | pids >> pid; |
50 | if(!pid) | 52 | if(!pid) |
51 | throw runtime_error("Can't detect running instance"); | 53 | throw runtime_error("Can't detect running instance"); |
52 | if(pid==getpid()) | 54 | if(pid==getpid()) |
53 | throw 0; | 55 | throw 0; |
54 | if(kill(pid,signum)) | 56 | if(kill(pid,signum)) |
55 | throw runtime_error("Failed to signal running instance"); | 57 | throw runtime_error("Failed to signal running instance"); |
56 | } | 58 | } |
57 | 59 | ||
58 | int main(int argc,char **argv) { | 60 | int main(int argc,char **argv) { |
59 | try { | 61 | try { |
60 | _argv = new char*[argc+1]; | 62 | _argv = new char*[argc+1]; |
61 | if(!_argv) | 63 | if(!_argv) |
62 | throw runtime_error("memory allocation problem at the very start"); | 64 | throw runtime_error("memory allocation problem at the very start"); |
63 | memmove(_argv,argv,sizeof(*_argv)*(argc+1)); | 65 | memmove(_argv,argv,sizeof(*_argv)*(argc+1)); |
64 | string config_file = DEFAULT_CONF_FILE; | 66 | string config_file = DEFAULT_CONF_FILE; |
65 | enum { | 67 | enum { |
66 | op_default, | 68 | op_default, |
67 | op_work, | 69 | op_work, |
68 | op_signal, | 70 | op_signal, |
69 | op_ensure, | 71 | op_ensure, |
70 | op_test | 72 | op_test |
71 | } op = op_default; | 73 | } op = op_default; |