swaynag: exit on wl_display_roundtrip error

fixes loop when sway closes the socket in the middle of querying outputs,
see #5138.

1 files changed, 5 insertions, 1 deletions

diff --git a/swaynag/swaynag.c b/swaynag/swaynag.c
index 06700e47..4e692b14 100644
--- a/swaynag/swaynag.c
+++ b/swaynag/swaynag.c
@@ -444,7 +444,11 @@ void swaynag_setup(struct swaynag *swaynag) {
         assert(swaynag->compositor && swaynag->layer_shell && swaynag->shm);
 
         while (swaynag->querying_outputs > 0) {
-                wl_display_roundtrip(swaynag->display);
+                if (wl_display_roundtrip(swaynag->display) < 0) {
+                        sway_log(SWAY_ERROR, "Error during outputs init.");
+                        swaynag_destroy(swaynag);
+                        exit(EXIT_FAILURE);
+                }
         }
 
         if (!swaynag->output && swaynag->type->output) {