-rw-r--r-- | src/dudki.cc | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/src/dudki.cc b/src/dudki.cc index 3c50e56..b769109 100644 --- a/src/dudki.cc +++ b/src/dudki.cc | |||
@@ -199,46 +199,51 @@ int main(int argc,char **argv) { | |||
199 | }catch(exception& e) { | 199 | }catch(exception& e) { |
200 | exit(1); | 200 | exit(1); |
201 | } | 201 | } |
202 | case op_ensure: | 202 | case op_ensure: |
203 | try { | 203 | try { |
204 | signal_self(config,0); | 204 | signal_self(config,0); |
205 | break; | 205 | break; |
206 | }catch(exception& e) { | 206 | }catch(exception& e) { |
207 | syslog(LOG_NOTICE,"The dudki process is down, taking its place"); | 207 | syslog(LOG_NOTICE,"The dudki process is down, taking its place"); |
208 | config.daemonize = true; | 208 | config.daemonize = true; |
209 | }catch(int zero) { | 209 | }catch(int zero) { |
210 | // we throw zero in case we're ensuring that this very process is running. | 210 | // we throw zero in case we're ensuring that this very process is running. |
211 | // we don't have to daemonize if we're daemonic. | 211 | // we don't have to daemonize if we're daemonic. |
212 | config.daemonize = false; | 212 | config.daemonize = false; |
213 | } | 213 | } |
214 | case op_default: | 214 | case op_default: |
215 | case op_work: | 215 | case op_work: |
216 | { | 216 | { |
217 | if(config.daemonize) { | 217 | if(config.daemonize) { |
218 | pid_t pf = fork(); | 218 | pid_t pf = fork(); |
219 | if(pf<0) | 219 | if(pf<0) |
220 | throw runtime_error(string(__PRETTY_FUNCTION__)+": failed to fork()"); | 220 | throw runtime_error(string(__PRETTY_FUNCTION__)+": failed to fork()"); |
221 | if(pf) { | 221 | if(pf) { |
222 | _exit(0); | 222 | _exit(0); |
223 | } | 223 | } |
224 | } | 224 | } |
225 | pid_file pidfile; | 225 | pid_file pidfile; |
226 | pidfile.set(config.pidfile); | 226 | pidfile.set(config.pidfile); |
227 | signal(SIGINT,lethal_signal_handler); | 227 | signal(SIGINT,lethal_signal_handler); |
228 | signal(SIGABRT,lethal_signal_handler); | 228 | signal(SIGABRT,lethal_signal_handler); |
229 | signal(SIGTERM,lethal_signal_handler); | 229 | signal(SIGTERM,lethal_signal_handler); |
230 | signal(SIGHUP,sighup_handler); | 230 | signal(SIGHUP,sighup_handler); |
231 | sigset_t sset; | ||
232 | sigemptyset(&sset); | ||
233 | sigaddset(&sset,SIGINT); sigaddset(&sset,SIGABRT); | ||
234 | sigaddset(&sset,SIGTERM); sigaddset(&sset,SIGHUP); | ||
235 | sigprocmask(SIG_UNBLOCK,&sset,NULL); | ||
231 | while(!finishing) { | 236 | while(!finishing) { |
232 | check_herd(config); | 237 | check_herd(config); |
233 | sleep(config.check_interval); | 238 | sleep(config.check_interval); |
234 | } | 239 | } |
235 | } | 240 | } |
236 | break; | 241 | break; |
237 | default: | 242 | default: |
238 | throw runtime_error(string(__PRETTY_FUNCTION__)+": internal error"); | 243 | throw runtime_error(string(__PRETTY_FUNCTION__)+": internal error"); |
239 | } | 244 | } |
240 | }catch(exception& e) { | 245 | }catch(exception& e) { |
241 | cerr << "Oops: " << e.what() << endl; | 246 | cerr << "Oops: " << e.what() << endl; |
242 | return 1; | 247 | return 1; |
243 | } | 248 | } |
244 | } | 249 | } |