3rd-party/catch2
1
0
Fork 0
mirror of https://github.com/catchorg/Catch2.git synced 2024-11-22 21:36:11 +01:00
Code Issues Packages Projects Releases Wiki Activity

Reset signals immediately after use and re-raise orginal signal instead of just exiting

Browse Source
This commit is contained in:
Phil Nash 2017-01-12 08:44:00 +00:00 committed by Martin Hořeňovský
parent 70f43d719b
commit 40dbdf6cb2
1 changed files with 41 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

84
include/internal/catch_fatal_condition.hpp
View File

@ -12,14 +12,11 @@
namespace Catch { namespace Catch {
// Report the error condition then exit the process // Report the error condition
inline void fatal( std::string const& message, int exitCode ) { inline void reportFatal( std::string const& message, int exitCode ) {
IContext& context = Catch::getCurrentContext(); IContext& context = Catch::getCurrentContext();
IResultCapture* resultCapture = context.getResultCapture(); IResultCapture* resultCapture = context.getResultCapture();
resultCapture->handleFatalErrorCondition( message ); resultCapture->handleFatalErrorCondition( message );
if( Catch::alwaysTrue() ) // avoids "no return" warnings
exit( exitCode );
} }
} // namespace Catch } // namespace Catch
@ -52,7 +49,7 @@ namespace Catch {
static LONG CALLBACK handleVectoredException(PEXCEPTION_POINTERS ExceptionInfo) { static LONG CALLBACK handleVectoredException(PEXCEPTION_POINTERS ExceptionInfo) {
for (int i = 0; i < sizeof(signalDefs) / sizeof(SignalDefs); ++i) { for (int i = 0; i < sizeof(signalDefs) / sizeof(SignalDefs); ++i) {
if (ExceptionInfo->ExceptionRecord->ExceptionCode == signalDefs[i].id) { if (ExceptionInfo->ExceptionRecord->ExceptionCode == signalDefs[i].id) {
fatal(signalDefs[i].name, -i); reportFatal(signalDefs[i].name, -i);
} }
} }
// If its not an exception we care about, pass it along. // If its not an exception we care about, pass it along.
@ -94,7 +91,10 @@ namespace Catch {
namespace Catch { namespace Catch {
struct SignalDefs { int id; const char* name; }; struct SignalDefs {
int id;
const char* name;
};
extern SignalDefs signalDefs[]; extern SignalDefs signalDefs[];
SignalDefs signalDefs[] = { SignalDefs signalDefs[] = {
{ SIGINT, "SIGINT - Terminal interrupt signal" }, { SIGINT, "SIGINT - Terminal interrupt signal" },
@ -107,50 +107,66 @@ namespace Catch {
struct FatalConditionHandler { struct FatalConditionHandler {
static bool isSet;
static struct sigaction oldSigActions [sizeof(signalDefs)/sizeof(SignalDefs)];
static stack_t oldSigStack;
static char altStackMem[SIGSTKSZ];
static void handleSignal( int sig ) { static void handleSignal( int sig ) {
for( std::size_t i = 0; i < sizeof(signalDefs)/sizeof(SignalDefs); ++i ) std::string name = "<unknown signal>";
if( sig == signalDefs[i].id ) for (std::size_t i = 0; i < sizeof(signalDefs) / sizeof(SignalDefs); ++i) {
fatal( signalDefs[i].name, -sig ); SignalDefs &def = signalDefs[i];
fatal( "<unknown signal>", -sig ); if (sig == def.id) {
name = def.name;
sigaction(def.id, &oldSigActions[i], CATCH_NULL);
break;
}
}
reportFatal(name, -sig);
raise( sig );
} }
FatalConditionHandler(): m_isSet(true), m_altStackMem(new char[SIGSTKSZ]) { FatalConditionHandler() {
isSet = true;
stack_t sigStack; stack_t sigStack;
sigStack.ss_sp = m_altStackMem; sigStack.ss_sp = altStackMem;
sigStack.ss_size = SIGSTKSZ; sigStack.ss_size = SIGSTKSZ;
sigStack.ss_flags = 0; sigStack.ss_flags = 0;
sigaltstack(&sigStack, &m_oldSigStack); sigaltstack(&sigStack, &oldSigStack);
struct sigaction sa = { 0 }; struct sigaction sa = { 0 };
sa.sa_handler = handleSignal; sa.sa_handler = handleSignal;
sa.sa_flags = SA_ONSTACK; sa.sa_flags = SA_ONSTACK;
for (std::size_t i = 0; i < sizeof(signalDefs)/sizeof(SignalDefs); ++i) { for (std::size_t i = 0; i < sizeof(signalDefs)/sizeof(SignalDefs); ++i) {
sigaction(signalDefs[i].id, &sa, &m_oldSigActions[i]); sigaction(signalDefs[i].id, &sa, &oldSigActions[i]);
} }
} }
~FatalConditionHandler() {
reset();
delete[] m_altStackMem; ~FatalConditionHandler() {
} reset();
void reset() { }
if( m_isSet ) { void reset() {
// Set signals back to previous values -- hopefully nobody overwrote them in the meantime if( isSet ) {
for( std::size_t i = 0; i < sizeof(signalDefs)/sizeof(SignalDefs); ++i ) { // Set signals back to previous values -- hopefully nobody overwrote them in the meantime
sigaction(signalDefs[i].id, &m_oldSigActions[i], CATCH_NULL); for( std::size_t i = 0; i < sizeof(signalDefs)/sizeof(SignalDefs); ++i ) {
} sigaction(signalDefs[i].id, &oldSigActions[i], CATCH_NULL);
// Return the old stack }
sigaltstack(&m_oldSigStack, CATCH_NULL); // Return the old stack
m_isSet = false; sigaltstack(&oldSigStack, CATCH_NULL);
isSet = false;
} }
} }
bool m_isSet;
// C++03 doesn't allow auto_ptr<T[]>, so we have manage the memory ourselves // C++03 doesn't allow auto_ptr<T[]>, so we have manage the memory ourselves
char* m_altStackMem;
struct sigaction m_oldSigActions [sizeof(signalDefs)/sizeof(SignalDefs)];
stack_t m_oldSigStack;
}; };
bool FatalConditionHandler::isSet = false;
struct sigaction FatalConditionHandler::oldSigActions[sizeof(signalDefs)/sizeof(SignalDefs)] = {} ;//[sizeof(signalDefs)/sizeof(SignalDefs)];
stack_t FatalConditionHandler::oldSigStack = {};
char FatalConditionHandler::altStackMem[SIGSTKSZ] = {};
} // namespace Catch } // namespace Catch
#endif // not Windows #endif // not Windows