summaryrefslogtreecommitdiffabout
path: root/src
Unidiff
Diffstat (limited to 'src') (more/less context) (ignore whitespace changes)
-rw-r--r--src/dudki.cc2
1 files changed, 1 insertions, 1 deletions
diff --git a/src/dudki.cc b/src/dudki.cc
index 3af5372..91a3342 100644
--- a/src/dudki.cc
+++ b/src/dudki.cc
@@ -1,292 +1,292 @@
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>
8using namespace std; 8using namespace std;
9#include "configuration.h" 9#include "configuration.h"
10#include "util.h" 10#include "util.h"
11 11
12#include "config.h" 12#include "config.h"
13#ifdef HAVE_GETOPT_H 13#ifdef HAVE_GETOPT_H
14# include <getopt.h> 14# include <getopt.h>
15#endif 15#endif
16 16
17#ifndef DEFAULT_CONF_FILE 17#ifndef DEFAULT_CONF_FILE
18# define DEFAULT_CONF_FILE "/etc/dudki.conf" 18# define DEFAULT_CONF_FILE "/etc/dudki.conf"
19#endif 19#endif
20 20
21#define PHEADER PACKAGE " Version " VERSION 21#define PHEADER PACKAGE " Version " VERSION
22#define PCOPY "Copyright (c) 2004 Klever Group" 22#define PCOPY "Copyright (c) 2004-2006 Klever Group"
23 23
24bool finishing = false; 24bool finishing = false;
25bool restarting = false; 25bool restarting = false;
26static char **_argv = NULL; 26static char **_argv = NULL;
27 27
28static void lethal_signal_handler(int signum) { 28static void lethal_signal_handler(int signum) {
29 syslog(LOG_NOTICE,"Lethal signal received. Terminating."); 29 syslog(LOG_NOTICE,"Lethal signal received. Terminating.");
30 finishing = true; 30 finishing = true;
31} 31}
32static void sighup_handler(int signum) { 32static void sighup_handler(int signum) {
33 syslog(LOG_NOTICE,"SUGHUP received, reloading."); 33 syslog(LOG_NOTICE,"SUGHUP received, reloading.");
34 restarting = finishing = true; 34 restarting = finishing = true;
35} 35}
36 36
37void check_herd(configuration& config) { 37void check_herd(configuration& config) {
38 process::prepare_herd(); 38 process::prepare_herd();
39 for(processes_t::iterator i=config.processes.begin();i!=config.processes.end();++i) 39 for(processes_t::iterator i=config.processes.begin();i!=config.processes.end();++i)
40 i->second.check(i->first,config); 40 i->second.check(i->first,config);
41 process::unprepare_herd(); 41 process::unprepare_herd();
42} 42}
43 43
44void signal_self(const configuration& config,int signum) { 44void signal_self(const configuration& config,int signum) {
45 ifstream pids(config.pidfile.c_str(),ios::in); 45 ifstream pids(config.pidfile.c_str(),ios::in);
46 if(!pids) 46 if(!pids)
47 throw runtime_error("Can't detect running instance"); 47 throw runtime_error("Can't detect running instance");
48 pid_t pid = 0; 48 pid_t pid = 0;
49 pids >> pid; 49 pids >> pid;
50 if(!pid) 50 if(!pid)
51 throw runtime_error("Can't detect running instance"); 51 throw runtime_error("Can't detect running instance");
52 if(pid==getpid()) 52 if(pid==getpid())
53 throw 0; 53 throw 0;
54 if(kill(pid,signum)) 54 if(kill(pid,signum))
55 throw runtime_error("Failed to signal running instance"); 55 throw runtime_error("Failed to signal running instance");
56} 56}
57 57
58int main(int argc,char **argv) { 58int main(int argc,char **argv) {
59 try { 59 try {
60 _argv = new char*[argc+1]; 60 _argv = new char*[argc+1];
61 if(!_argv) 61 if(!_argv)
62 throw runtime_error("memory allocation problem at the very start"); 62 throw runtime_error("memory allocation problem at the very start");
63 memmove(_argv,argv,sizeof(*_argv)*(argc+1)); 63 memmove(_argv,argv,sizeof(*_argv)*(argc+1));
64 string config_file = DEFAULT_CONF_FILE; 64 string config_file = DEFAULT_CONF_FILE;
65 enum { 65 enum {
66 op_default, 66 op_default,
67 op_work, 67 op_work,
68 op_signal, 68 op_signal,
69 op_ensure, 69 op_ensure,
70 op_test 70 op_test
71 } op = op_default; 71 } op = op_default;
72 int op_signum = 0; 72 int op_signum = 0;
73 while(true) { 73 while(true) {
74 #defineSHORTOPTSTRING "f:hVLrkcets:" 74 #defineSHORTOPTSTRING "f:hVLrkcets:"
75#ifdef HAVE_GETOPT_LONG 75#ifdef HAVE_GETOPT_LONG
76 static struct option opts[] = { 76 static struct option opts[] = {
77 { "help", no_argument, 0, 'h' }, 77 { "help", no_argument, 0, 'h' },
78 { "usage", no_argument, 0, 'h' }, 78 { "usage", no_argument, 0, 'h' },
79 { "version", no_argument, 0, 'V' }, 79 { "version", no_argument, 0, 'V' },
80 { "license", no_argument, 0, 'L' }, 80 { "license", no_argument, 0, 'L' },
81 { "config", required_argument, 0, 'f' }, 81 { "config", required_argument, 0, 'f' },
82 { "kill", no_argument, 0, 'k' }, 82 { "kill", no_argument, 0, 'k' },
83 { "reload", no_argument, 0, 'r' }, 83 { "reload", no_argument, 0, 'r' },
84 { "signal", required_argument, 0, 's' }, 84 { "signal", required_argument, 0, 's' },
85 { "check", no_argument, 0, 'c' }, 85 { "check", no_argument, 0, 'c' },
86 { "ensure", no_argument, 0, 'e' }, 86 { "ensure", no_argument, 0, 'e' },
87 { "test", no_argument, 0, 't' }, 87 { "test", no_argument, 0, 't' },
88 { NULL, 0, 0, 0 } 88 { NULL, 0, 0, 0 }
89 }; 89 };
90 int c = getopt_long(argc,argv,SHORTOPTSTRING,opts,NULL); 90 int c = getopt_long(argc,argv,SHORTOPTSTRING,opts,NULL);
91#else /* !HAVE_GETOPT_LONG */ 91#else /* !HAVE_GETOPT_LONG */
92 int c = getopt(argc,argv,SHORTOPTSTRING); 92 int c = getopt(argc,argv,SHORTOPTSTRING);
93#endif /* /HAVE_GETOPT_LONG */ 93#endif /* /HAVE_GETOPT_LONG */
94 if(c==-1) 94 if(c==-1)
95 break; 95 break;
96 switch(c) { 96 switch(c) {
97 case 'h': 97 case 'h':
98 cerr << PHEADER << endl 98 cerr << PHEADER << endl
99 << PCOPY << endl << endl 99 << PCOPY << endl << endl
100 << " " << argv[0] << " [options] [processes]" << endl << endl << 100 << " " << argv[0] << " [options] [processes]" << endl << endl <<
101#ifdef HAVE_GETOPT_LONG 101#ifdef HAVE_GETOPT_LONG
102 " -h, --help\n" 102 " -h, --help\n"
103 " --usage display this text\n" 103 " --usage display this text\n"
104 " -V, --version display version number\n" 104 " -V, --version display version number\n"
105 " -L, --license show license\n" 105 " -L, --license show license\n"
106 " -f filename, --config=filename\n" 106 " -f filename, --config=filename\n"
107 " specify the configuration file to use\n" 107 " specify the configuration file to use\n"
108 "\n" 108 "\n"
109 " -k, --kill stop running instance (send SIGTERM)\n" 109 " -k, --kill stop running instance (send SIGTERM)\n"
110 " -r, --reload reload running instance (send SIGHUP)\n" 110 " -r, --reload reload running instance (send SIGHUP)\n"
111 " -s signum, --signal=signum\n" 111 " -s signum, --signal=signum\n"
112 " send the specified signal to the running process\n" 112 " send the specified signal to the running process\n"
113 " -c, --check check if the process is running\n" 113 " -c, --check check if the process is running\n"
114 " (the above commands operate on dudki itself if no\n" 114 " (the above commands operate on dudki itself if no\n"
115 " process name has been specified)\n" 115 " process name has been specified)\n"
116 " -e, --ensure ensure that dudki is running\n" 116 " -e, --ensure ensure that dudki is running\n"
117 " -t, --test test configuration file and exit" 117 " -t, --test test configuration file and exit"
118#else /* !HAVE_GETOPT_LONG */ 118#else /* !HAVE_GETOPT_LONG */
119 " -h display this text\n" 119 " -h display this text\n"
120 " -V display version number\n" 120 " -V display version number\n"
121 " -L show license\n" 121 " -L show license\n"
122 " -f filename specify the configuration file to use\n" 122 " -f filename specify the configuration file to use\n"
123 "\n" 123 "\n"
124 " -k stop running instance (send SIGTERM)\n" 124 " -k stop running instance (send SIGTERM)\n"
125 " -r reload running instance (send SIGHUP)\n" 125 " -r reload running instance (send SIGHUP)\n"
126 " -s signum send the specified signal to the running process\n" 126 " -s signum send the specified signal to the running process\n"
127 " -c check if the process is running\n" 127 " -c check if the process is running\n"
128 " (the above commands operate on dudki itself if no\n" 128 " (the above commands operate on dudki itself if no\n"
129 " process name has been specified)\n" 129 " process name has been specified)\n"
130 " -e ensure that dudki is running\n" 130 " -e ensure that dudki is running\n"
131 " -t test configuration file and exit" 131 " -t test configuration file and exit"
132#endif /* /HAVE_GETOPT_LONG */ 132#endif /* /HAVE_GETOPT_LONG */
133 << endl; 133 << endl;
134 exit(0); 134 exit(0);
135 break; 135 break;
136 case 'V': 136 case 'V':
137 cerr << VERSION << endl; 137 cerr << VERSION << endl;
138 exit(0); 138 exit(0);
139 break; 139 break;
140 case 'L': 140 case 'L':
141 extern const char *COPYING; 141 extern const char *COPYING;
142 cerr << COPYING << endl; 142 cerr << COPYING << endl;
143 exit(0); 143 exit(0);
144 break; 144 break;
145 case 'f': 145 case 'f':
146 config_file = optarg; 146 config_file = optarg;
147 break; 147 break;
148 case 'k': 148 case 'k':
149 if(op!=op_default) { 149 if(op!=op_default) {
150 cerr << "Can't obey two or more orders at once" << endl; 150 cerr << "Can't obey two or more orders at once" << endl;
151 exit(1); 151 exit(1);
152 } 152 }
153 op = op_signal; op_signum = SIGTERM; 153 op = op_signal; op_signum = SIGTERM;
154 break; 154 break;
155 case 'r': 155 case 'r':
156 if(op!=op_default) { 156 if(op!=op_default) {
157 cerr << "Can't obey two or more orders at once" << endl; 157 cerr << "Can't obey two or more orders at once" << endl;
158 exit(1); 158 exit(1);
159 } 159 }
160 op = op_signal; op_signum = SIGHUP; 160 op = op_signal; op_signum = SIGHUP;
161 break; 161 break;
162 case 'c': 162 case 'c':
163 if(op!=op_default) { 163 if(op!=op_default) {
164 cerr << "Can't obey two or more orders at once" << endl; 164 cerr << "Can't obey two or more orders at once" << endl;
165 exit(1); 165 exit(1);
166 } 166 }
167 op = op_signal; op_signum = 0; 167 op = op_signal; op_signum = 0;
168 break; 168 break;
169 case 'e': 169 case 'e':
170 if(op!=op_default) { 170 if(op!=op_default) {
171 cerr << "Can't obey two or more orders at once" << endl; 171 cerr << "Can't obey two or more orders at once" << endl;
172 exit(1); 172 exit(1);
173 } 173 }
174 op = op_ensure; 174 op = op_ensure;
175 break; 175 break;
176 case 't': 176 case 't':
177 if(op!=op_default) { 177 if(op!=op_default) {
178 cerr << "Can't obey two or more orders at once" << endl; 178 cerr << "Can't obey two or more orders at once" << endl;
179 exit(1); 179 exit(1);
180 } 180 }
181 op = op_test; 181 op = op_test;
182 break; 182 break;
183 case 's': 183 case 's':
184 if(op!=op_default) { 184 if(op!=op_default) {
185 cerr << "Can't obey two or more orders at once" << endl; 185 cerr << "Can't obey two or more orders at once" << endl;
186 exit(1); 186 exit(1);
187 } 187 }
188 op = op_signal; 188 op = op_signal;
189 errno = 0; 189 errno = 0;
190 op_signum = strtol(optarg,NULL,0); 190 op_signum = strtol(optarg,NULL,0);
191 if(errno) { 191 if(errno) {
192 cerr << "Can't obtain the signal value" << endl; 192 cerr << "Can't obtain the signal value" << endl;
193 exit(1); 193 exit(1);
194 } 194 }
195 break; 195 break;
196 default: 196 default:
197 cerr << "Huh??" << endl; 197 cerr << "Huh??" << endl;
198 exit(1); 198 exit(1);
199 break; 199 break;
200 } 200 }
201 } 201 }
202 const char *sid = *argv; 202 const char *sid = *argv;
203 const char *t; 203 const char *t;
204 while(t = index(sid,'/')) { 204 while(t = index(sid,'/')) {
205 sid = t; sid++; 205 sid = t; sid++;
206 } 206 }
207 openlog(sid,LOG_CONS|LOG_PERROR|LOG_PID,LOG_DAEMON); 207 openlog(sid,LOG_CONS|LOG_PERROR|LOG_PID,LOG_DAEMON);
208 configuration config; 208 configuration config;
209 config.parse(config_file); 209 config.parse(config_file);
210 switch(op) { 210 switch(op) {
211 case op_test: 211 case op_test:
212 cerr << "Configuration OK" << endl; 212 cerr << "Configuration OK" << endl;
213 break; 213 break;
214 case op_signal: 214 case op_signal:
215 try { 215 try {
216 if(optind>=argc) { 216 if(optind>=argc) {
217 signal_self(config,op_signum); 217 signal_self(config,op_signum);
218 }else{ 218 }else{
219 int failures = 0; 219 int failures = 0;
220 for(int narg=optind;narg<argc;narg++) { 220 for(int narg=optind;narg<argc;narg++) {
221 try { 221 try {
222 processes_t::const_iterator i = config.processes.find(argv[narg]); 222 processes_t::const_iterator i = config.processes.find(argv[narg]);
223 if(i==config.processes.end()) 223 if(i==config.processes.end())
224 throw runtime_error("no such process configured"); 224 throw runtime_error("no such process configured");
225 if(op_signum) 225 if(op_signum)
226 i->second.signal(op_signum); 226 i->second.signal(op_signum);
227 else 227 else
228 i->second.check(); 228 i->second.check();
229 }catch(exception& e) { 229 }catch(exception& e) {
230 cerr << "dudki(" << argv[narg] << "): " << e.what() << endl; 230 cerr << "dudki(" << argv[narg] << "): " << e.what() << endl;
231 failures++; 231 failures++;
232 } 232 }
233 } 233 }
234 if(failures) 234 if(failures)
235 throw runtime_error("not all processes have been successfully signaled"); 235 throw runtime_error("not all processes have been successfully signaled");
236 } 236 }
237 exit(0); 237 exit(0);
238 }catch(exception& e) { 238 }catch(exception& e) {
239 exit(1); 239 exit(1);
240 } 240 }
241 break; 241 break;
242 case op_ensure: 242 case op_ensure:
243 try { 243 try {
244 signal_self(config,0); 244 signal_self(config,0);
245 break; 245 break;
246 }catch(exception& e) { 246 }catch(exception& e) {
247 syslog(LOG_NOTICE,"The dudki process is down, taking its place"); 247 syslog(LOG_NOTICE,"The dudki process is down, taking its place");
248 config.daemonize = true; 248 config.daemonize = true;
249 }catch(int zero) { 249 }catch(int zero) {
250 // we throw zero in case we're ensuring that this very process is running. 250 // we throw zero in case we're ensuring that this very process is running.
251 // we don't have to daemonize if we're daemonic. 251 // we don't have to daemonize if we're daemonic.
252 config.daemonize = false; 252 config.daemonize = false;
253 } 253 }
254 case op_default: 254 case op_default:
255 case op_work: 255 case op_work:
256 { 256 {
257 if(config.daemonize) { 257 if(config.daemonize) {
258 pid_t pf = fork(); 258 pid_t pf = fork();
259 if(pf<0) 259 if(pf<0)
260 throw runtime_error(string(__PRETTY_FUNCTION__)+": failed to fork()"); 260 throw runtime_error(string(__PRETTY_FUNCTION__)+": failed to fork()");
261 if(pf) { 261 if(pf) {
262 _exit(0); 262 _exit(0);
263 } 263 }
264 } 264 }
265 pid_file pidfile; 265 pid_file pidfile;
266 pidfile.set(config.pidfile); 266 pidfile.set(config.pidfile);
267 signal(SIGINT,lethal_signal_handler); 267 signal(SIGINT,lethal_signal_handler);
268 signal(SIGABRT,lethal_signal_handler); 268 signal(SIGABRT,lethal_signal_handler);
269 signal(SIGTERM,lethal_signal_handler); 269 signal(SIGTERM,lethal_signal_handler);
270 signal(SIGHUP,sighup_handler); 270 signal(SIGHUP,sighup_handler);
271 sigset_t sset; 271 sigset_t sset;
272 sigemptyset(&sset); 272 sigemptyset(&sset);
273 sigaddset(&sset,SIGINT); sigaddset(&sset,SIGABRT); 273 sigaddset(&sset,SIGINT); sigaddset(&sset,SIGABRT);
274 sigaddset(&sset,SIGTERM); sigaddset(&sset,SIGHUP); 274 sigaddset(&sset,SIGTERM); sigaddset(&sset,SIGHUP);
275 sigprocmask(SIG_UNBLOCK,&sset,NULL); 275 sigprocmask(SIG_UNBLOCK,&sset,NULL);
276 while(!finishing) { 276 while(!finishing) {
277 check_herd(config); 277 check_herd(config);
278 sleep(config.check_interval); 278 sleep(config.check_interval);
279 } 279 }
280 if(restarting) 280 if(restarting)
281 execvp(_argv[0],_argv); 281 execvp(_argv[0],_argv);
282 } 282 }
283 break; 283 break;
284 default: 284 default:
285 throw runtime_error(string(__PRETTY_FUNCTION__)+": internal error"); 285 throw runtime_error(string(__PRETTY_FUNCTION__)+": internal error");
286 } 286 }
287 }catch(exception& e) { 287 }catch(exception& e) {
288 cerr << "Oops: " << e.what() << endl; 288 cerr << "Oops: " << e.what() << endl;
289 return 1; 289 return 1;
290 } 290 }
291 exit(0); 291 exit(0);
292} 292}