添加信号处理

src/main.c

@@ -1 +1,86 @@
-int main(int argc, char *argv[]) { return 0; }
+#include <glib-unix.h>
+#include <glib.h>
+#include <glibconfig.h>
+#include <signal.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+
+#include "backtrace.h"
+#include "buffer.h"
+#include "types.h"
+#include "utils.h"
+
+zdwm_t global_state;
+
+/** argv used to run zdwm */
+static char **zdwm_argv;
+
+/** A pipe that is used to asynchronously handle SIGCHLD */
+static int sigchld_pipe[2];
+
+static void zdwm_atexit(bool restart) { global_state.need_restart = restart; }
+
+static void zdwm_restart(void) {
+  zdwm_atexit(true);
+  execvp(zdwm_argv[0], zdwm_argv);
+  fatal("execv() failed: %s", strerror(errno));
+}
+
+static gboolean restart_on_signal(gpointer data) {
+  zdwm_restart();
+  return TRUE;
+}
+
+static gboolean exit_on_signal(gpointer data) {
+  g_main_loop_quit(global_state.loop);
+  return TRUE;
+}
+
+static void signal_fatal(int signal_number) {
+  buffer_t buffer;
+  backtrace_get(&buffer);
+  fatal("signal %d, dumping backtrace\n%s", signal_number, buffer.s);
+}
+
+/* Signal handler for SIGCHLD. Causes reap_children() to be called. */
+static void signal_child(int signum) {
+  assert(signum == SIGCHLD);
+  int res = write(sigchld_pipe[1], " ", 1);
+  (void)res;
+  assert(res == 1);
+}
+
+int main(int argc, char *argv[]) {
+  p_clear(&global_state, 1);
+
+  zdwm_argv = argv;
+
+  g_unix_signal_add(SIGINT, exit_on_signal, NULL);
+  g_unix_signal_add(SIGTERM, exit_on_signal, NULL);
+  g_unix_signal_add(SIGHUP, restart_on_signal, NULL);
+
+  struct sigaction sa = {.sa_handler = signal_fatal, .sa_flags = SA_RESETHAND};
+  sigemptyset(&sa.sa_mask);
+  sigaction(SIGABRT, &sa, 0);
+  sigaction(SIGBUS, &sa, 0);
+  sigaction(SIGFPE, &sa, 0);
+  sigaction(SIGILL, &sa, 0);
+  sigaction(SIGSEGV, &sa, 0);
+  signal(SIGPIPE, SIG_IGN);
+
+  sa.sa_handler = signal_child;
+  sa.sa_flags = SA_NOCLDSTOP | SA_RESTART;
+  sigaction(SIGCHLD, &sa, 0);
+
+  if (global_state.loop == NULL) {
+    global_state.loop = g_main_loop_new(NULL, FALSE);
+    g_main_loop_run(global_state.loop);
+  }
+  g_main_loop_unref(global_state.loop);
+  global_state.loop = NULL;
+
+  zdwm_atexit(false);
+
+  return EXIT_SUCCESS;
+}