Merge pull request #111 from martinRenou/handle_kill_sig

src/main.cpp

@@ -9,6 +9,14 @@
 
 #include <memory>
 #include <iostream>
+#include <signal.h>
+
+#ifdef __GNUC__
+#include <stdio.h>
+#include <execinfo.h>
+#include <stdlib.h>
+#include <unistd.h>
+#endif
 
 #include "xeus/xkernel.hpp"
 #include "xeus/xkernel_configuration.hpp"
@@ -17,6 +25,26 @@
 #include "xeus-sqlite/xeus_sqlite_interpreter.hpp"
 #include "xeus-sqlite/xeus_sqlite_config.hpp"
 
+#ifdef __GNUC__
+void handler(int sig)
+{
+    void* array[10];
+
+    // get void*'s for all entries on the stack
+    size_t size = backtrace(array, 10);
+
+    // print out all the frames to stderr
+    fprintf(stderr, "Error: signal %d:\n", sig);
+    backtrace_symbols_fd(array, size, STDERR_FILENO);
+    exit(1);
+}
+#endif
+
+void stop_handler(int /*sig*/)
+{
+    exit(0);
+}
+
 bool should_print_version(int argc, char* argv[])
 {
     for (int i = 0; i < argc; ++i)
@@ -56,6 +84,16 @@ int main(int argc, char* argv[])
         return 0;
     }
 
+    // Registering SIGSEGV handler
+#ifdef __GNUC__
+    std::clog << "registering handler for SIGSEGV" << std::endl;
+    signal(SIGSEGV, handler);
+
+    // Registering SIGINT and SIGKILL handlers
+    signal(SIGKILL, stop_handler);
+#endif
+    signal(SIGINT, stop_handler);
+
     // Load configuration file
     std::string file_name = extract_filename(argc, argv);