1 files changed, 7 insertions, 1 deletions

diff --git a/swayidle/main.c b/swayidle/main.c
index 60235280..ad9c84c9 100644
--- a/swayidle/main.c
+++ b/swayidle/main.c
@@ -340,7 +340,13 @@ void sig_handler(int signal) {
 }
 
 static int display_event(int fd, uint32_t mask, void *data) {
-        wl_display_dispatch(state.display);
+        if (mask & WL_EVENT_HANGUP) {
+                sway_terminate(0);
+        }
+        if (wl_display_dispatch(state.display) < 0) {
+                wlr_log_errno(L_ERROR, "wl_display_dispatch failed, exiting");
+                sway_terminate(0);
+        };
         return 0;
 }