00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028
00029
00030
00031
00032 #include "config.h"
00033
00034 static char rcsid[] not_used =
00035 { "$Id: SignalHandler.cc 16088 2007-03-28 21:42:19Z jimg $"
00036 };
00037
00038 #include <signal.h>
00039
00040 #ifndef WIN32
00041 #include <unistd.h>
00042 #endif
00043
00044 #include <pthread.h>
00045
00046 #include "SignalHandler.h"
00047 #include "util.h"
00048
00049 EventHandler *SignalHandler::d_signal_handlers[NSIG];
00050 Sigfunc *SignalHandler::d_old_handlers[NSIG];
00051 SignalHandler *SignalHandler::d_instance = 0;
00052
00053
00054
00055 static pthread_once_t instance_control = PTHREAD_ONCE_INIT;
00056
00058 void
00059 SignalHandler::initialize_instance()
00060 {
00061
00062 SignalHandler::d_instance = new SignalHandler;
00063 atexit(SignalHandler::delete_instance);
00064 }
00065
00067 void
00068 SignalHandler::delete_instance()
00069 {
00070 if (SignalHandler::d_instance) {
00071 for (int i = 0; i < NSIG; ++i) {
00072 d_signal_handlers[i] = 0;
00073 d_old_handlers[i] = 0;
00074 }
00075
00076 delete SignalHandler::d_instance;
00077 SignalHandler::d_instance = 0;
00078 }
00079 }
00080
00087 void
00088 SignalHandler::dispatcher(int signum)
00089 {
00090
00091 if (SignalHandler::d_signal_handlers[signum] != 0)
00092
00093 SignalHandler::d_signal_handlers[signum]->handle_signal(signum);
00094
00095 Sigfunc *old_handler = SignalHandler::d_old_handlers[signum];
00096 if (old_handler == SIG_IGN || old_handler == SIG_ERR)
00097 return;
00098 else if (old_handler == SIG_DFL) {
00099 switch (signum) {
00100 #ifndef WIN32
00101 case SIGHUP:
00102 case SIGKILL:
00103 case SIGUSR1:
00104 case SIGUSR2:
00105 case SIGPIPE:
00106 case SIGALRM:
00107 #endif
00108 case SIGINT:
00109 case SIGTERM: _exit(EXIT_FAILURE);
00110
00111
00112
00113 default: abort();
00114 }
00115 }
00116 else
00117 old_handler(signum);
00118 }
00119
00121 SignalHandler*
00122 SignalHandler::instance()
00123 {
00124 pthread_once(&instance_control, initialize_instance);
00125
00126 return d_instance;
00127 }
00128
00141 EventHandler *
00142 SignalHandler::register_handler(int signum, EventHandler *eh, bool override)
00143 {
00144
00145 switch (signum) {
00146 #ifndef WIN32
00147 case SIGHUP:
00148 case SIGKILL:
00149 case SIGUSR1:
00150 case SIGUSR2:
00151 case SIGPIPE:
00152 case SIGALRM:
00153 #endif
00154 case SIGINT:
00155 case SIGTERM: break;
00156
00157 default: throw InternalErr(__FILE__, __LINE__,
00158 string("Call to register_handler with unsupported signal (")
00159 + long_to_string(signum) + string(")."));
00160 }
00161
00162
00163 EventHandler *old_eh = SignalHandler::d_signal_handlers[signum];
00164
00165 SignalHandler::d_signal_handlers[signum] = eh;
00166
00167
00168
00169 #ifndef WIN32
00170 struct sigaction sa;
00171 sa.sa_handler = dispatcher;
00172 sigemptyset(&sa.sa_mask);
00173 sa.sa_flags = 0;
00174
00175
00176
00177
00178 if (signum == SIGALRM) {
00179 #ifdef SA_INTERUPT
00180 sa.sa_flags |= SA_INTERUPT;
00181 #endif
00182 }
00183 else {
00184 #ifdef SA_RESTART
00185 sa.sa_flags |= SA_RESTART;
00186 #endif
00187 }
00188
00189 struct sigaction osa;
00190
00191 if (sigaction(signum, &sa, &osa) < 0)
00192 throw InternalErr(__FILE__, __LINE__, "Could not register a signal handler.");
00193
00194
00195
00196
00197
00198 if (override)
00199 SignalHandler::d_old_handlers[signum] = SIG_IGN;
00200 else if (osa.sa_handler != dispatcher)
00201 SignalHandler::d_old_handlers[signum] = osa.sa_handler;
00202 #endif
00203
00204 return old_eh;
00205 }
00206
00210 EventHandler *
00211 SignalHandler::remove_handler(int signum)
00212 {
00213 EventHandler *old_eh = SignalHandler::d_signal_handlers[signum];
00214
00215 SignalHandler::d_signal_handlers[signum] = 0;
00216
00217 return old_eh;
00218 }