swaylock: Support keyboard and pointer disconnects and reconnects

2 files changed, 13 insertions, 5 deletions

diff --git a/swaylock/main.c b/swaylock/main.c
index ed8c5607..d1384c6f 100644
--- a/swaylock/main.c
+++ b/swaylock/main.c
@@ -277,7 +277,7 @@ static void handle_global(void *data, struct wl_registry *registry,
                                 &wl_shm_interface, 1);
         } else if (strcmp(interface, wl_seat_interface.name) == 0) {
                 struct wl_seat *seat = wl_registry_bind(
-                                registry, name, &wl_seat_interface, 1);
+                                registry, name, &wl_seat_interface, 3);
                 wl_seat_add_listener(seat, &seat_listener, state);
         } else if (strcmp(interface, zwlr_layer_shell_v1_interface.name) == 0) {
                 state->layer_shell = wl_registry_bind(
diff --git a/swaylock/seat.c b/swaylock/seat.c
index c2630d87..22dd9360 100644
--- a/swaylock/seat.c
+++ b/swaylock/seat.c
@@ -145,13 +145,21 @@ static const struct wl_pointer_listener pointer_listener = {
 static void seat_handle_capabilities(void *data, struct wl_seat *wl_seat,
                 enum wl_seat_capability caps) {
         struct swaylock_state *state = data;
+        if (state->pointer) {
+                wl_pointer_release(state->pointer);
+                state->pointer = NULL;
+        }
+        if (state->keyboard) {
+                wl_keyboard_release(state->keyboard);
+                state->keyboard = NULL;
+        }
         if ((caps & WL_SEAT_CAPABILITY_POINTER)) {
-                struct wl_pointer *pointer = wl_seat_get_pointer(wl_seat);
-                wl_pointer_add_listener(pointer, &pointer_listener, NULL);
+                state->pointer = wl_seat_get_pointer(wl_seat);
+                wl_pointer_add_listener(state->pointer, &pointer_listener, NULL);
         }
         if ((caps & WL_SEAT_CAPABILITY_KEYBOARD)) {
-                struct wl_keyboard *keyboard = wl_seat_get_keyboard(wl_seat);
-                wl_keyboard_add_listener(keyboard, &keyboard_listener, state);
+                state->keyboard = wl_seat_get_keyboard(wl_seat);
+                wl_keyboard_add_listener(state->keyboard, &keyboard_listener, state);
         }
 }