diff --git a/src/test/setup.h b/src/test/setup.h index 20936cc..bcf3035 100644 --- a/src/test/setup.h +++ b/src/test/setup.h @@ -1,12 +1,13 @@ -#if defined(WIN32) && defined(SNMALLOC_CI_BUILD) -# include -# include -# include -# include -# include +#if defined(SNMALLOC_CI_BUILD) +# if defined(WIN32) +# include +# include +# include +# include +# include // Has to come after the PAL. -# include -# pragma comment(lib, "dbghelp.lib") +# include +# pragma comment(lib, "dbghelp.lib") void print_stack_trace() { @@ -94,6 +95,20 @@ void setup() // Disable OS level dialog boxes during CI. SetErrorMode(SEM_FAILCRITICALERRORS | SEM_NOGPFAULTERRORBOX); } +# else +# include +# include +void error_handle(int signal) +{ + UNUSED(signal); + snmalloc::error("Seg Fault"); + _exit(1); +} +void setup() +{ + signal(SIGSEGV, error_handle); +} +# endif #else void setup() {} #endif