Signal Handling: Set up a signal handler for common catastrophic fail…

…ure signals (#165)

* signal handling

---------

Co-authored-by: Charity Kathure <[email protected]>

LogMonitor/src/LogMonitor/Main.cpp

@@ -25,6 +25,27 @@ std::unique_ptr<EventMonitor> g_eventMon(nullptr);
 std::vector<std::shared_ptr<LogFileMonitor>> g_logfileMonitors;
 std::unique_ptr<EtwMonitor> g_etwMon(nullptr);
 
+/// Handle signals.
+///
+/// \return None
+///
+void signalHandler(int signum) {
+    logWriter.TraceError(
+        Utility::FormatString(L"Catastrophic failure! Signal received: %d", signum).c_str()
+    );
+
+    // Terminate the program
+    std::exit(signum);
+}
+
+/// Set up a signal handler
+///
+/// \return None
+///
+void setSignalHandler(int signum, void (*handler)(int)) {
+    std::signal(signum, handler);
+}
+
 BOOL WINAPI ControlHandle(_In_ DWORD dwCtrlType)
 {
     switch (dwCtrlType)
@@ -249,6 +270,14 @@ int __cdecl wmain(int argc, WCHAR *argv[])
         logWriter.TraceError(L"Invalid configuration file.");
     }
 
+    //
+    // Set up a signal handler for common catastrophic failure signals
+    //
+    setSignalHandler(SIGSEGV, signalHandler); // Segmentation fault
+    setSignalHandler(SIGFPE, signalHandler);  // Floating point exception
+    setSignalHandler(SIGILL, signalHandler);  // Illegal instruction
+    setSignalHandler(SIGABRT, signalHandler); // Abort
+
     //
     // Set the Ctrl handler function, that propagates the Ctrl events to the child process.
     //

LogMonitor/src/LogMonitor/pch.h

@@ -23,6 +23,8 @@
 #include <queue>
 #include <map>
 #include <stdexcept>
+#include <csignal>
+#include <cstdlib>
 #include <Windows.h>
 #include <cctype>
 #include <sal.h>