catch SIGSEGV as well; register sa_handlers only if its SIG_DLF before

1 files changed, 36 insertions, 0 deletions

diff --git a/src/sig_handlers.c b/src/sig_handlers.c
new file mode 100644
index 0000000..601e454
--- /dev/null
+++ b/src/sig_handlers.c
@@ -0,0 +1,36 @@
+#include <signal.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+
+static void abnormal_termination_handler(int signo) {
+	exit(signo);
+}
+
+static void __attribute__((constructor)) register_handlers(void)
+{
+	struct sigaction sa, old_sa;
+	sa.sa_handler = abnormal_termination_handler;
+	sigemptyset(&sa.sa_mask);
+
+	sigaction(SIGABRT, NULL, &old_sa);
+	if (old_sa.sa_handler == SIG_DFL) {
+		if (sigaction(SIGABRT, &sa, NULL) == -1) {
+			perror("sigaction");
+			exit(1);
+		}
+	} else {
+		fprintf(stderr, "SIGABRT handler already set");
+	}
+
+	sigaction(SIGSEGV, NULL, &old_sa);
+	if (old_sa.sa_handler == SIG_DFL) {
+		if (sigaction(SIGSEGV, &sa, NULL) == -1) {
+			perror("sigaction");
+			exit(1);
+		}
+	} else {
+		fprintf(stderr, "SIGSEGV handler already set");
+	}
+}
+