make glib loop play well with SIGINT/SIGTERM

ssfdust · Jun 27, 2024 · a681aa1 · a681aa1
1 parent 5f1be14
commit a681aa1
Showing 1 changed file with 17 additions and 4 deletions.
diff --git a/src/glib-main-loop.cpp b/src/glib-main-loop.cpp
@@ -14,11 +14,19 @@
 static gboolean on_wayland_fd_event(gint fd, GIOCondition condition,
     gpointer user_data);
 
+static int glib_signal_handler(gpointer raw_glib_loop)
+{
+    Glib::RefPtr<Glib::MainLoop> g_loop = Glib::wrap((GMainLoop*)raw_glib_loop);
+    g_loop->quit();
+    return G_SOURCE_REMOVE;
+}
+
 namespace wf
 {
 class glib_main_loop_t : public wf::plugin_interface_t
 {
     Glib::RefPtr<Glib::MainLoop> g_loop;
+    wf::wl_idle_call idle_shutdown;
 
   public:
     void init()
@@ -48,7 +56,10 @@ class glib_main_loop_t : public wf::plugin_interface_t
 
         g_loop = Glib::MainLoop::create();
         wf::get_core().connect(&glib_loop_run);
-        wf::get_core().connect(&glib_loop_quit);
+        // wf::get_core().connect(&glib_loop_quit);
+
+        g_unix_signal_add_full(G_PRIORITY_DEFAULT, SIGTERM, glib_signal_handler, g_loop.get(), NULL);
+        g_unix_signal_add_full(G_PRIORITY_DEFAULT, SIGINT, glib_signal_handler, g_loop.get(), NULL);
     }
 
     void handle_wayland_fd_in(GIOCondition flag)
@@ -73,13 +84,15 @@ class glib_main_loop_t : public wf::plugin_interface_t
         g_unix_fd_add(fd, G_IO_HUP, on_wayland_fd_event, this);
 
         g_loop->run();
+        idle_shutdown.run_once([=] ()
+        {
+            wf::get_core().shutdown();
+        });
     };
 
     wf::signal::connection_t<wf::core_shutdown_signal> glib_loop_quit = [=] (auto)
     {
-        auto display = wf::get_core().display;
-        wl_display_destroy_clients(display);
-        wl_display_destroy(display);
+        wf::get_core().shutdown();
         std::exit(0);
     };