diff --git a/include/wlr/types/wlr_data_device.h b/include/wlr/types/wlr_data_device.h index 0b425e88..ba067a62 100644 --- a/include/wlr/types/wlr_data_device.h +++ b/include/wlr/types/wlr_data_device.h @@ -98,6 +98,8 @@ struct wlr_drag_icon { struct wl_signal destroy; } events; + struct wl_listener surface_destroy; + void *data; }; diff --git a/types/data_device/wlr_drag.c b/types/data_device/wlr_drag.c index 93614774..3e3f094e 100644 --- a/types/data_device/wlr_drag.c +++ b/types/data_device/wlr_drag.c @@ -106,6 +106,13 @@ out: wl_signal_emit_mutable(&drag->events.focus, drag); } +static void drag_icon_destroy(struct wlr_drag_icon *icon) { + icon->drag->icon = NULL; + wl_list_remove(&icon->surface_destroy.link); + wl_signal_emit_mutable(&icon->events.destroy, icon); + free(icon); +} + static void drag_destroy(struct wlr_drag *drag) { if (drag->cancelling) { return; @@ -145,7 +152,7 @@ static void drag_destroy(struct wlr_drag *drag) { } if (drag->icon != NULL) { - wlr_surface_destroy_role_object(drag->icon->surface); + drag_icon_destroy(drag->icon); } free(drag); } @@ -356,20 +363,18 @@ static void drag_icon_surface_role_commit(struct wlr_surface *surface) { } } -static void drag_icon_surface_role_destroy(struct wlr_surface *surface) { - assert(surface->role == &drag_icon_surface_role); - struct wlr_drag_icon *icon = surface->role_data; - - wl_signal_emit_mutable(&icon->events.destroy, icon); - free(icon); -} - const struct wlr_surface_role drag_icon_surface_role = { .name = "wl_data_device-icon", .commit = drag_icon_surface_role_commit, - .destroy = drag_icon_surface_role_destroy, }; +static void drag_icon_handle_surface_destroy(struct wl_listener *listener, + void *data) { + struct wlr_drag_icon *icon = + wl_container_of(listener, icon, surface_destroy); + drag_icon_destroy(icon); +} + static struct wlr_drag_icon *drag_icon_create(struct wlr_drag *drag, struct wlr_surface *surface) { struct wlr_drag_icon *icon = calloc(1, sizeof(struct wlr_drag_icon)); @@ -382,7 +387,8 @@ static struct wlr_drag_icon *drag_icon_create(struct wlr_drag *drag, wl_signal_init(&icon->events.destroy); - icon->surface->role_data = icon; + icon->surface_destroy.notify = drag_icon_handle_surface_destroy; + wl_signal_add(&surface->events.destroy, &icon->surface_destroy); drag_icon_surface_role_commit(surface);