93 lines
2.1 KiB
C++
93 lines
2.1 KiB
C++
|
#include "StarSignalHandler.hpp"
|
||
|
|
|
||
|
|
#include <signal.h>
|
||
|
|
|
||
|
|
namespace Star {
|
||
|
|
|
||
|
|
struct SignalHandlerImpl {
|
||
|
|
bool handlingFatal;
|
||
|
|
bool handlingInterrupt;
|
||
|
|
bool interrupted;
|
||
|
|
|
||
|
|
SignalHandlerImpl() : handlingFatal(false), handlingInterrupt(false), interrupted(false) {}
|
||
|
|
|
||
|
|
~SignalHandlerImpl() {
|
||
|
|
setHandleFatal(false);
|
||
|
|
setHandleInterrupt(false);
|
||
|
|
}
|
||
|
|
|
||
|
|
void setHandleFatal(bool b) {
|
||
|
|
handlingFatal = b;
|
||
|
|
if (handlingFatal) {
|
||
|
|
signal(SIGSEGV, handleFatal);
|
||
|
|
signal(SIGILL, handleFatal);
|
||
|
|
signal(SIGFPE, handleFatal);
|
||
|
|
signal(SIGBUS, handleFatal);
|
||
|
|
} else {
|
||
|
|
signal(SIGSEGV, SIG_DFL);
|
||
|
|
signal(SIGILL, SIG_DFL);
|
||
|
|
signal(SIGFPE, SIG_DFL);
|
||
|
|
signal(SIGBUS, SIG_DFL);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
void setHandleInterrupt(bool b) {
|
||
|
|
handlingInterrupt = b;
|
||
|
|
if (handlingInterrupt)
|
||
|
|
signal(SIGINT, handleInterrupt);
|
||
|
|
else
|
||
|
|
signal(SIGINT, SIG_DFL);
|
||
|
|
}
|
||
|
|
|
||
|
|
static void handleFatal(int signum) {
|
||
|
|
if (signum == SIGSEGV)
|
||
|
|
fatalError("Segfault Encountered!", true);
|
||
|
|
else if (signum == SIGILL)
|
||
|
|
fatalError("Illegal Instruction Encountered!", true);
|
||
|
|
else if (signum == SIGFPE)
|
||
|
|
fatalError("Floating Point Exception Encountered!", true);
|
||
|
|
else if (signum == SIGBUS)
|
||
|
|
fatalError("Bus Error Encountered!", true);
|
||
|
|
}
|
||
|
|
|
||
|
|
static void handleInterrupt(int) {
|
||
|
|
if (SignalHandler::s_singleton)
|
||
|
|
SignalHandler::s_singleton->interrupted = true;
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
SignalHandlerImplUPtr SignalHandler::s_singleton;
|
||
|
|
|
||
|
|
SignalHandler::SignalHandler() {
|
||
|
|
if (s_singleton)
|
||
|
|
throw StarException("Singleton SignalHandler has been constructed twice!");
|
||
|
|
|
||
|
|
s_singleton = make_unique<SignalHandlerImpl>();
|
||
|
|
}
|
||
|
|
|
||
|
|
SignalHandler::~SignalHandler() {
|
||
|
|
s_singleton.reset();
|
||
|
|
}
|
||
|
|
|
||
|
|
void SignalHandler::setHandleFatal(bool handleFatal) {
|
||
|
|
s_singleton->setHandleFatal(handleFatal);
|
||
|
|
}
|
||
|
|
|
||
|
|
bool SignalHandler::handlingFatal() const {
|
||
|
|
return s_singleton->handlingFatal;
|
||
|
|
}
|
||
|
|
|
||
|
|
void SignalHandler::setHandleInterrupt(bool handleInterrupt) {
|
||
|
|
s_singleton->setHandleInterrupt(handleInterrupt);
|
||
|
|
}
|
||
|
|
|
||
|
|
bool SignalHandler::handlingInterrupt() const {
|
||
|
|
return s_singleton->handlingInterrupt;
|
||
|
|
}
|
||
|
|
|
||
|
|
bool SignalHandler::interruptCaught() const {
|
||
|
|
return s_singleton->interrupted;
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|