From f237b5c7a702a6641de6d49e9b0bd754246283c9 Mon Sep 17 00:00:00 2001 From: emersion Date: Fri, 29 Dec 2017 22:51:33 +0100 Subject: [PATCH 1/6] Make wlr_data_source abstract. This removes some fields specific to sources coming from clients. This adds some drag'n'drop-related callbacks. --- include/wlr/types/wlr_data_device.h | 38 +++--- types/wlr_data_device.c | 177 ++++++++++++++++------------ types/wlr_seat.c | 35 ++++-- 3 files changed, 149 insertions(+), 101 deletions(-) diff --git a/include/wlr/types/wlr_data_device.h b/include/wlr/types/wlr_data_device.h index 92d15972..54514b4c 100644 --- a/include/wlr/types/wlr_data_device.h +++ b/include/wlr/types/wlr_data_device.h @@ -23,33 +23,39 @@ struct wlr_data_offer { struct wl_resource *resource; struct wlr_data_source *source; - uint32_t dnd_actions; - enum wl_data_device_manager_dnd_action preferred_dnd_action; + uint32_t actions; + enum wl_data_device_manager_dnd_action preferred_action; bool in_ask; struct wl_listener source_destroy; }; struct wlr_data_source { - struct wl_resource *resource; + // source metadata + struct wl_array mime_types; + int32_t actions; + + // source implementation + void (*send)(struct wlr_data_source *source, const char *mime_type, + int32_t fd); + void (*accept)(struct wlr_data_source *source, uint32_t serial, + const char *mime_type); + void (*cancel)(struct wlr_data_source *source); + + // drag'n'drop implementation + void (*dnd_drop)(struct wlr_data_source *source); + void (*dnd_finish)(struct wlr_data_source *source); + void (*dnd_action)(struct wlr_data_source *source, + enum wl_data_device_manager_dnd_action action); + + // source status + bool accepted; struct wlr_data_offer *offer; struct wlr_seat_client *seat_client; - struct wl_array mime_types; - - bool accepted; - - // drag and drop + // drag'n'drop status enum wl_data_device_manager_dnd_action current_dnd_action; - uint32_t dnd_actions; uint32_t compositor_action; - bool actions_set; - - void (*accept)(struct wlr_data_source *source, uint32_t serial, - const char *mime_type); - void (*send)(struct wlr_data_source *source, const char *mime_type, - int32_t fd); - void (*cancel)(struct wlr_data_source *source); struct { struct wl_signal destroy; diff --git a/types/wlr_data_device.c b/types/wlr_data_device.c index a9f54a0b..7c61c0a5 100644 --- a/types/wlr_data_device.c +++ b/types/wlr_data_device.c @@ -14,26 +14,23 @@ WL_DATA_DEVICE_MANAGER_DND_ACTION_ASK) static uint32_t data_offer_choose_action(struct wlr_data_offer *offer) { - uint32_t available_actions, preferred_action = 0; - uint32_t source_actions, offer_actions; - + uint32_t offer_actions, preferred_action = 0; if (wl_resource_get_version(offer->resource) >= WL_DATA_OFFER_ACTION_SINCE_VERSION) { - offer_actions = offer->dnd_actions; - preferred_action = offer->preferred_dnd_action; + offer_actions = offer->actions; + preferred_action = offer->preferred_action; } else { offer_actions = WL_DATA_DEVICE_MANAGER_DND_ACTION_COPY; } - if (wl_resource_get_version(offer->source->resource) >= - WL_DATA_SOURCE_ACTION_SINCE_VERSION) { - source_actions = offer->source->dnd_actions; + uint32_t source_actions; + if (offer->source->actions >= 0) { + source_actions = offer->source->actions; } else { source_actions = WL_DATA_DEVICE_MANAGER_DND_ACTION_COPY; } - available_actions = offer_actions & source_actions; - + uint32_t available_actions = offer_actions & source_actions; if (!available_actions) { return WL_DATA_DEVICE_MANAGER_DND_ACTION_NONE; } @@ -53,14 +50,11 @@ static uint32_t data_offer_choose_action(struct wlr_data_offer *offer) { } static void data_offer_update_action(struct wlr_data_offer *offer) { - uint32_t action; - if (!offer->source) { return; } - action = data_offer_choose_action(offer); - + uint32_t action = data_offer_choose_action(offer); if (offer->source->current_dnd_action == action) { return; } @@ -71,9 +65,8 @@ static void data_offer_update_action(struct wlr_data_offer *offer) { return; } - if (wl_resource_get_version(offer->source->resource) >= - WL_DATA_SOURCE_ACTION_SINCE_VERSION) { - wl_data_source_send_action(offer->source->resource, action); + if (offer->source->dnd_action) { + offer->source->dnd_action(offer->source, action); } if (wl_resource_get_version(offer->resource) >= @@ -82,21 +75,58 @@ static void data_offer_update_action(struct wlr_data_offer *offer) { } } -static void client_data_source_accept(struct wlr_data_source *source, +struct client_data_source { + struct wlr_data_source source; + struct wl_resource *resource; + bool actions_set; +}; + +static void client_data_source_accept(struct wlr_data_source *wlr_source, uint32_t serial, const char *mime_type) { + struct client_data_source *source = (struct client_data_source *)wlr_source; wl_data_source_send_target(source->resource, mime_type); } -static void client_data_source_send(struct wlr_data_source *source, +static void client_data_source_send(struct wlr_data_source *wlr_source, const char *mime_type, int32_t fd) { + struct client_data_source *source = (struct client_data_source *)wlr_source; wl_data_source_send_send(source->resource, mime_type, fd); close(fd); } -static void client_data_source_cancel(struct wlr_data_source *source) { +static void client_data_source_cancel(struct wlr_data_source *wlr_source) { + struct client_data_source *source = (struct client_data_source *)wlr_source; wl_data_source_send_cancelled(source->resource); } +static void client_data_source_dnd_drop(struct wlr_data_source *wlr_source) { + struct client_data_source *source = (struct client_data_source *)wlr_source; + assert(wl_resource_get_version(source->resource) >= + WL_DATA_SOURCE_DND_DROP_PERFORMED_SINCE_VERSION); + wl_data_source_send_dnd_drop_performed(source->resource); +} + +static void client_data_source_dnd_finish(struct wlr_data_source *wlr_source) { + struct client_data_source *source = (struct client_data_source *)wlr_source; + assert(wl_resource_get_version(source->resource) >= + WL_DATA_SOURCE_DND_FINISHED_SINCE_VERSION); + wl_data_source_send_dnd_finished(source->resource); +} + +static void client_data_source_dnd_action(struct wlr_data_source *wlr_source, + enum wl_data_device_manager_dnd_action action) { + struct client_data_source *source = (struct client_data_source *)wlr_source; + assert(wl_resource_get_version(source->resource) >= + WL_DATA_SOURCE_ACTION_SINCE_VERSION); + wl_data_source_send_action(source->resource, action); +} + +static void client_data_source_resource_destroy(struct wl_resource *resource) { + struct client_data_source *source = wl_resource_get_user_data(resource); + wlr_data_source_finish(&source->source); + free(source); +} + static void data_offer_accept(struct wl_client *client, struct wl_resource *resource, uint32_t serial, const char *mime_type) { @@ -108,7 +138,9 @@ static void data_offer_accept(struct wl_client *client, // TODO check that client is currently focused by the input device - offer->source->accept(offer->source, serial, mime_type); + if (offer->source->accept) { + offer->source->accept(offer->source, serial, mime_type); + } offer->source->accepted = (mime_type != NULL); } @@ -128,19 +160,17 @@ static void data_offer_destroy(struct wl_client *client, } static void data_source_notify_finish(struct wlr_data_source *source) { - if (!source->actions_set) { + assert(source->offer); + if (source->actions < 0) { return; } - if (source->offer->in_ask && wl_resource_get_version(source->resource) >= - WL_DATA_SOURCE_ACTION_SINCE_VERSION) { - wl_data_source_send_action(source->resource, - source->current_dnd_action); + if (source->offer->in_ask && source->dnd_action) { + source->dnd_action(source, source->current_dnd_action); } - if (wl_resource_get_version(source->resource) >= - WL_DATA_SOURCE_DND_FINISHED_SINCE_VERSION) { - wl_data_source_send_dnd_finished(source->resource); + if (source->dnd_finish) { + source->dnd_finish(source); } source->offer = NULL; @@ -158,27 +188,27 @@ static void data_offer_finish(struct wl_client *client, } static void data_offer_set_actions(struct wl_client *client, - struct wl_resource *resource, uint32_t dnd_actions, + struct wl_resource *resource, uint32_t actions, uint32_t preferred_action) { struct wlr_data_offer *offer = wl_resource_get_user_data(resource); - if (dnd_actions & ~ALL_ACTIONS) { + if (actions & ~ALL_ACTIONS) { wl_resource_post_error(offer->resource, WL_DATA_OFFER_ERROR_INVALID_ACTION_MASK, - "invalid action mask %x", dnd_actions); + "invalid action mask %x", actions); return; } - if (preferred_action && (!(preferred_action & dnd_actions) || - __builtin_popcount(preferred_action) > 1)) { + if (preferred_action && (!(preferred_action & actions) || + __builtin_popcount(preferred_action) > 1)) { wl_resource_post_error(offer->resource, WL_DATA_OFFER_ERROR_INVALID_ACTION, "invalid action %x", preferred_action); return; } - offer->dnd_actions = dnd_actions; - offer->preferred_dnd_action = preferred_action; + offer->actions = actions; + offer->preferred_action = preferred_action; data_offer_update_action(offer); } @@ -198,14 +228,12 @@ static void data_offer_resource_destroy(struct wl_resource *resource) { // If the drag destination has version < 3, wl_data_offer.finish // won't be called, so do this here as a safety net, because - // we still want the version >=3 drag source to be happy. + // we still want the version >= 3 drag source to be happy. if (wl_resource_get_version(offer->resource) < WL_DATA_OFFER_ACTION_SINCE_VERSION) { data_source_notify_finish(offer->source); - } else if (offer->source->resource && - wl_resource_get_version(offer->source->resource) >= - WL_DATA_SOURCE_DND_FINISHED_SINCE_VERSION) { - wl_data_source_send_cancelled(offer->source->resource); + } else if (offer->source->dnd_finish) { + offer->source->cancel(offer->source); } offer->source->offer = NULL; @@ -316,6 +344,8 @@ static void seat_client_selection_data_source_destroy( void wlr_seat_set_selection(struct wlr_seat *seat, struct wlr_data_source *source, uint32_t serial) { + assert(source->send); + assert(source->cancel); if (seat->selection_source && seat->selection_serial - serial < UINT32_MAX / 2) { return; @@ -433,7 +463,7 @@ static void wlr_drag_set_focus(struct wlr_drag *drag, if (wl_resource_get_version(offer->resource) >= WL_DATA_OFFER_SOURCE_ACTIONS_SINCE_VERSION) { wl_data_offer_send_source_actions(offer->resource, - drag->source->dnd_actions); + drag->source->actions); } offer_resource = offer->resource; @@ -510,18 +540,15 @@ static uint32_t pointer_drag_button(struct wlr_seat_pointer_grab *grab, wl_resource_for_each(resource, &drag->focus_client->data_devices) { wl_data_device_send_drop(resource); } - if (wl_resource_get_version(drag->source->resource) >= - WL_DATA_SOURCE_DND_DROP_PERFORMED_SINCE_VERSION) { - wl_data_source_send_dnd_drop_performed( - drag->source->resource); + if (drag->source->dnd_drop) { + drag->source->dnd_drop(drag->source); } drag->source->offer->in_ask = drag->source->current_dnd_action == WL_DATA_DEVICE_MANAGER_DND_ACTION_ASK; - } else if (wl_resource_get_version(drag->source->resource) >= - WL_DATA_SOURCE_DND_FINISHED_SINCE_VERSION) { - wl_data_source_send_cancelled(drag->source->resource); + } else if (drag->source->dnd_finish) { + drag->source->cancel(drag->source); } } @@ -852,13 +879,6 @@ void data_device_manager_get_data_device(struct wl_client *client, wl_list_insert(&seat_client->data_devices, wl_resource_get_link(resource)); } -static void data_source_resource_destroy(struct wl_resource *resource) { - struct wlr_data_source *source = - wl_resource_get_user_data(resource); - wlr_data_source_finish(source); - free(source); -} - static void data_source_destroy(struct wl_client *client, struct wl_resource *resource) { wl_resource_destroy(resource); @@ -866,10 +886,10 @@ static void data_source_destroy(struct wl_client *client, static void data_source_set_actions(struct wl_client *client, struct wl_resource *resource, uint32_t dnd_actions) { - struct wlr_data_source *source = + struct client_data_source *source = wl_resource_get_user_data(resource); - if (source->actions_set) { + if (source->source.actions >= 0) { wl_resource_post_error(source->resource, WL_DATA_SOURCE_ERROR_INVALID_ACTION_MASK, "cannot set actions more than once"); @@ -883,7 +903,7 @@ static void data_source_set_actions(struct wl_client *client, return; } - if (source->seat_client) { + if (source->source.seat_client) { wl_resource_post_error(source->resource, WL_DATA_SOURCE_ERROR_INVALID_ACTION_MASK, "invalid action change after " @@ -891,22 +911,18 @@ static void data_source_set_actions(struct wl_client *client, return; } - source->dnd_actions = dnd_actions; - source->actions_set = true; + source->source.actions = dnd_actions; } static void data_source_offer(struct wl_client *client, struct wl_resource *resource, const char *mime_type) { - struct wlr_data_source *source = - wl_resource_get_user_data(resource); - char **p; - - p = wl_array_add(&source->mime_types, sizeof(*p)); + struct wlr_data_source *source = wl_resource_get_user_data(resource); + char **p = wl_array_add(&source->mime_types, sizeof(*p)); if (p) { *p = strdup(mime_type); } - if (!p || !*p){ + if (!p || !*p) { if (p) { source->mime_types.size -= sizeof(*p); } @@ -923,6 +939,7 @@ static struct wl_data_source_interface data_source_impl = { void wlr_data_source_init(struct wlr_data_source *source) { wl_array_init(&source->mime_types); wl_signal_init(&source->events.destroy); + source->actions = -1; } void wlr_data_source_finish(struct wlr_data_source *source) { @@ -941,12 +958,13 @@ void wlr_data_source_finish(struct wlr_data_source *source) { static void data_device_manager_create_data_source(struct wl_client *client, struct wl_resource *resource, uint32_t id) { - struct wlr_data_source *source = calloc(1, sizeof(struct wlr_data_source)); + struct client_data_source *source = + calloc(1, sizeof(struct client_data_source)); if (source == NULL) { wl_resource_post_no_memory(resource); return; } - wlr_data_source_init(source); + wlr_data_source_init(&source->source); source->resource = wl_resource_create(client, &wl_data_source_interface, wl_resource_get_version(resource), id); @@ -956,11 +974,24 @@ static void data_device_manager_create_data_source(struct wl_client *client, return; } wl_resource_set_implementation(source->resource, &data_source_impl, - source, data_source_resource_destroy); + source, client_data_source_resource_destroy); - source->accept = client_data_source_accept; - source->send = client_data_source_send; - source->cancel = client_data_source_cancel; + source->source.accept = client_data_source_accept; + source->source.send = client_data_source_send; + source->source.cancel = client_data_source_cancel; + + if (wl_resource_get_version(source->resource) >= + WL_DATA_SOURCE_DND_DROP_PERFORMED_SINCE_VERSION) { + source->source.dnd_drop = client_data_source_dnd_drop; + } + if (wl_resource_get_version(source->resource) >= + WL_DATA_SOURCE_DND_FINISHED_SINCE_VERSION) { + source->source.dnd_finish = client_data_source_dnd_finish; + } + if (wl_resource_get_version(source->resource) >= + WL_DATA_SOURCE_ACTION_SINCE_VERSION) { + source->source.dnd_action = client_data_source_dnd_action; + } } static const struct wl_data_device_manager_interface diff --git a/types/wlr_seat.c b/types/wlr_seat.c index f9e2db5f..ec95053b 100644 --- a/types/wlr_seat.c +++ b/types/wlr_seat.c @@ -343,28 +343,39 @@ static const struct wlr_touch_grab_interface default_touch_grab_impl = { }; -void wlr_seat_destroy(struct wlr_seat *wlr_seat) { - if (!wlr_seat) { +void wlr_seat_destroy(struct wlr_seat *seat) { + if (!seat) { return; } - wl_signal_emit(&wlr_seat->events.destroy, wlr_seat); + wl_signal_emit(&seat->events.destroy, seat); - wl_list_remove(&wlr_seat->display_destroy.link); + wl_list_remove(&seat->display_destroy.link); + + if (seat->selection_source) { + seat->selection_source->cancel(seat->selection_source); + seat->selection_source = NULL; + wl_list_remove(&seat->selection_data_source_destroy.link); + } + if (seat->primary_selection_source) { + seat->primary_selection_source->cancel(seat->primary_selection_source); + seat->primary_selection_source = NULL; + wl_list_remove(&seat->primary_selection_source_destroy.link); + } struct wlr_seat_client *client, *tmp; - wl_list_for_each_safe(client, tmp, &wlr_seat->clients, link) { + wl_list_for_each_safe(client, tmp, &seat->clients, link) { // will destroy other resources as well wl_resource_destroy(client->wl_resource); } - wl_global_destroy(wlr_seat->wl_global); - free(wlr_seat->pointer_state.default_grab); - free(wlr_seat->keyboard_state.default_grab); - free(wlr_seat->touch_state.default_grab); - free(wlr_seat->data_device); - free(wlr_seat->name); - free(wlr_seat); + wl_global_destroy(seat->wl_global); + free(seat->pointer_state.default_grab); + free(seat->keyboard_state.default_grab); + free(seat->touch_state.default_grab); + free(seat->data_device); + free(seat->name); + free(seat); } static void handle_display_destroy(struct wl_listener *listener, void *data) { From 062809723aef85c6eed6b8f9fcb9e3e6f1f1d8b0 Mon Sep 17 00:00:00 2001 From: emersion Date: Fri, 29 Dec 2017 23:01:35 +0100 Subject: [PATCH 2/6] Data source resource now holds a `struct client_data_source *` --- types/wlr_data_device.c | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/types/wlr_data_device.c b/types/wlr_data_device.c index 7c61c0a5..26cf0a87 100644 --- a/types/wlr_data_device.c +++ b/types/wlr_data_device.c @@ -78,7 +78,6 @@ static void data_offer_update_action(struct wlr_data_offer *offer) { struct client_data_source { struct wlr_data_source source; struct wl_resource *resource; - bool actions_set; }; static void client_data_source_accept(struct wlr_data_source *wlr_source, @@ -380,7 +379,7 @@ void wlr_seat_set_selection(struct wlr_seat *seat, static void data_device_set_selection(struct wl_client *client, struct wl_resource *dd_resource, struct wl_resource *source_resource, uint32_t serial) { - struct wlr_data_source *source = NULL; + struct client_data_source *source = NULL; if (source_resource != NULL) { source = wl_resource_get_user_data(source_resource); } @@ -389,7 +388,8 @@ static void data_device_set_selection(struct wl_client *client, wl_resource_get_user_data(dd_resource); // TODO: store serial and check against incoming serial here - wlr_seat_set_selection(seat_client->seat, source, serial); + struct wlr_data_source *wlr_source = (struct wlr_data_source *)source; + wlr_seat_set_selection(seat_client->seat, wlr_source, serial); } static void data_device_release(struct wl_client *client, @@ -916,15 +916,15 @@ static void data_source_set_actions(struct wl_client *client, static void data_source_offer(struct wl_client *client, struct wl_resource *resource, const char *mime_type) { - struct wlr_data_source *source = wl_resource_get_user_data(resource); + struct client_data_source *source = wl_resource_get_user_data(resource); - char **p = wl_array_add(&source->mime_types, sizeof(*p)); + char **p = wl_array_add(&source->source.mime_types, sizeof(*p)); if (p) { *p = strdup(mime_type); } if (!p || !*p) { if (p) { - source->mime_types.size -= sizeof(*p); + source->source.mime_types.size -= sizeof(*p); } wl_resource_post_no_memory(resource); } From b20aed66d6f7556b13c2f22df9fcfb31c37dbc90 Mon Sep 17 00:00:00 2001 From: emersion Date: Fri, 29 Dec 2017 23:23:00 +0100 Subject: [PATCH 3/6] Abstract wlr_primary_selection_source --- include/wlr/types/wlr_primary_selection.h | 10 +- types/wlr_data_device.c | 145 +++++++++++----------- types/wlr_primary_selection.c | 67 ++++++---- 3 files changed, 121 insertions(+), 101 deletions(-) diff --git a/include/wlr/types/wlr_primary_selection.h b/include/wlr/types/wlr_primary_selection.h index b4eceb78..da3c096a 100644 --- a/include/wlr/types/wlr_primary_selection.h +++ b/include/wlr/types/wlr_primary_selection.h @@ -15,16 +15,18 @@ struct wlr_primary_selection_device_manager { struct wlr_primary_selection_offer; struct wlr_primary_selection_source { - struct wl_resource *resource; - struct wlr_primary_selection_offer *offer; - struct wlr_seat_client *seat_client; - + // source metadata struct wl_array mime_types; + // source implementation void (*send)(struct wlr_primary_selection_source *source, const char *mime_type, int32_t fd); void (*cancel)(struct wlr_primary_selection_source *source); + // source status + struct wlr_primary_selection_offer *offer; + struct wlr_seat_client *seat_client; + struct { struct wl_signal destroy; } events; diff --git a/types/wlr_data_device.c b/types/wlr_data_device.c index 26cf0a87..804718f9 100644 --- a/types/wlr_data_device.c +++ b/types/wlr_data_device.c @@ -75,58 +75,6 @@ static void data_offer_update_action(struct wlr_data_offer *offer) { } } -struct client_data_source { - struct wlr_data_source source; - struct wl_resource *resource; -}; - -static void client_data_source_accept(struct wlr_data_source *wlr_source, - uint32_t serial, const char *mime_type) { - struct client_data_source *source = (struct client_data_source *)wlr_source; - wl_data_source_send_target(source->resource, mime_type); -} - -static void client_data_source_send(struct wlr_data_source *wlr_source, - const char *mime_type, int32_t fd) { - struct client_data_source *source = (struct client_data_source *)wlr_source; - wl_data_source_send_send(source->resource, mime_type, fd); - close(fd); -} - -static void client_data_source_cancel(struct wlr_data_source *wlr_source) { - struct client_data_source *source = (struct client_data_source *)wlr_source; - wl_data_source_send_cancelled(source->resource); -} - -static void client_data_source_dnd_drop(struct wlr_data_source *wlr_source) { - struct client_data_source *source = (struct client_data_source *)wlr_source; - assert(wl_resource_get_version(source->resource) >= - WL_DATA_SOURCE_DND_DROP_PERFORMED_SINCE_VERSION); - wl_data_source_send_dnd_drop_performed(source->resource); -} - -static void client_data_source_dnd_finish(struct wlr_data_source *wlr_source) { - struct client_data_source *source = (struct client_data_source *)wlr_source; - assert(wl_resource_get_version(source->resource) >= - WL_DATA_SOURCE_DND_FINISHED_SINCE_VERSION); - wl_data_source_send_dnd_finished(source->resource); -} - -static void client_data_source_dnd_action(struct wlr_data_source *wlr_source, - enum wl_data_device_manager_dnd_action action) { - struct client_data_source *source = (struct client_data_source *)wlr_source; - assert(wl_resource_get_version(source->resource) >= - WL_DATA_SOURCE_ACTION_SINCE_VERSION); - wl_data_source_send_action(source->resource, action); -} - -static void client_data_source_resource_destroy(struct wl_resource *resource) { - struct client_data_source *source = wl_resource_get_user_data(resource); - wlr_data_source_finish(&source->source); - free(source); -} - - static void data_offer_accept(struct wl_client *client, struct wl_resource *resource, uint32_t serial, const char *mime_type) { struct wlr_data_offer *offer = wl_resource_get_user_data(resource); @@ -343,8 +291,11 @@ static void seat_client_selection_data_source_destroy( void wlr_seat_set_selection(struct wlr_seat *seat, struct wlr_data_source *source, uint32_t serial) { - assert(source->send); - assert(source->cancel); + if (source) { + assert(source->send); + assert(source->cancel); + } + if (seat->selection_source && seat->selection_serial - serial < UINT32_MAX / 2) { return; @@ -861,22 +812,50 @@ static void data_device_destroy(struct wl_resource *resource) { wl_list_remove(wl_resource_get_link(resource)); } -void data_device_manager_get_data_device(struct wl_client *client, - struct wl_resource *manager_resource, uint32_t id, - struct wl_resource *seat_resource) { - struct wlr_seat_client *seat_client = - wl_resource_get_user_data(seat_resource); - struct wl_resource *resource = wl_resource_create(client, - &wl_data_device_interface, wl_resource_get_version(manager_resource), - id); - if (resource == NULL) { - wl_resource_post_no_memory(manager_resource); - return; - } - wl_resource_set_implementation(resource, &data_device_impl, seat_client, - &data_device_destroy); - wl_list_insert(&seat_client->data_devices, wl_resource_get_link(resource)); +struct client_data_source { + struct wlr_data_source source; + struct wl_resource *resource; +}; + +static void client_data_source_accept(struct wlr_data_source *wlr_source, + uint32_t serial, const char *mime_type) { + struct client_data_source *source = (struct client_data_source *)wlr_source; + wl_data_source_send_target(source->resource, mime_type); +} + +static void client_data_source_send(struct wlr_data_source *wlr_source, + const char *mime_type, int32_t fd) { + struct client_data_source *source = (struct client_data_source *)wlr_source; + wl_data_source_send_send(source->resource, mime_type, fd); + close(fd); +} + +static void client_data_source_cancel(struct wlr_data_source *wlr_source) { + struct client_data_source *source = (struct client_data_source *)wlr_source; + wl_data_source_send_cancelled(source->resource); +} + +static void client_data_source_dnd_drop(struct wlr_data_source *wlr_source) { + struct client_data_source *source = (struct client_data_source *)wlr_source; + assert(wl_resource_get_version(source->resource) >= + WL_DATA_SOURCE_DND_DROP_PERFORMED_SINCE_VERSION); + wl_data_source_send_dnd_drop_performed(source->resource); +} + +static void client_data_source_dnd_finish(struct wlr_data_source *wlr_source) { + struct client_data_source *source = (struct client_data_source *)wlr_source; + assert(wl_resource_get_version(source->resource) >= + WL_DATA_SOURCE_DND_FINISHED_SINCE_VERSION); + wl_data_source_send_dnd_finished(source->resource); +} + +static void client_data_source_dnd_action(struct wlr_data_source *wlr_source, + enum wl_data_device_manager_dnd_action action) { + struct client_data_source *source = (struct client_data_source *)wlr_source; + assert(wl_resource_get_version(source->resource) >= + WL_DATA_SOURCE_ACTION_SINCE_VERSION); + wl_data_source_send_action(source->resource, action); } static void data_source_destroy(struct wl_client *client, @@ -936,6 +915,12 @@ static struct wl_data_source_interface data_source_impl = { .set_actions = data_source_set_actions, }; +static void data_source_resource_handle_destroy(struct wl_resource *resource) { + struct client_data_source *source = wl_resource_get_user_data(resource); + wlr_data_source_finish(&source->source); + free(source); +} + void wlr_data_source_init(struct wlr_data_source *source) { wl_array_init(&source->mime_types); wl_signal_init(&source->events.destroy); @@ -956,6 +941,25 @@ void wlr_data_source_finish(struct wlr_data_source *source) { wl_array_release(&source->mime_types); } + +void data_device_manager_get_data_device(struct wl_client *client, + struct wl_resource *manager_resource, uint32_t id, + struct wl_resource *seat_resource) { + struct wlr_seat_client *seat_client = + wl_resource_get_user_data(seat_resource); + + struct wl_resource *resource = wl_resource_create(client, + &wl_data_device_interface, wl_resource_get_version(manager_resource), + id); + if (resource == NULL) { + wl_resource_post_no_memory(manager_resource); + return; + } + wl_resource_set_implementation(resource, &data_device_impl, seat_client, + &data_device_destroy); + wl_list_insert(&seat_client->data_devices, wl_resource_get_link(resource)); +} + static void data_device_manager_create_data_source(struct wl_client *client, struct wl_resource *resource, uint32_t id) { struct client_data_source *source = @@ -974,7 +978,7 @@ static void data_device_manager_create_data_source(struct wl_client *client, return; } wl_resource_set_implementation(source->resource, &data_source_impl, - source, client_data_source_resource_destroy); + source, data_source_resource_handle_destroy); source->source.accept = client_data_source_accept; source->source.send = client_data_source_send; @@ -1019,6 +1023,7 @@ void wlr_data_device_manager_destroy(struct wlr_data_device_manager *manager) { return; } wl_list_remove(&manager->display_destroy.link); + // TODO: free wl_resources wl_global_destroy(manager->global); free(manager); } diff --git a/types/wlr_primary_selection.c b/types/wlr_primary_selection.c index 1228e94e..c29856f2 100644 --- a/types/wlr_primary_selection.c +++ b/types/wlr_primary_selection.c @@ -2,23 +2,12 @@ #include #include #include +#include #include #include #include #include -static void client_source_send(struct wlr_primary_selection_source *source, - const char *mime_type, int32_t fd) { - gtk_primary_selection_source_send_send(source->resource, mime_type, fd); - close(fd); -} - -static void client_source_cancel( - struct wlr_primary_selection_source *source) { - gtk_primary_selection_source_send_cancelled(source->resource); -} - - static void offer_handle_receive(struct wl_client *client, struct wl_resource *resource, const char *mime_type, int32_t fd) { struct wlr_primary_selection_offer *offer = @@ -55,8 +44,8 @@ static void offer_resource_handle_destroy(struct wl_resource *resource) { goto out; } - if (offer->source->resource) { - gtk_primary_selection_source_send_cancelled(offer->source->resource); + if (offer->source->cancel) { + offer->source->cancel(offer->source); } offer->source->offer = NULL; @@ -73,6 +62,24 @@ static void offer_handle_source_destroy(struct wl_listener *listener, } +struct client_data_source { + struct wlr_primary_selection_source source; + struct wl_resource *resource; +}; + +static void client_source_send(struct wlr_primary_selection_source *wlr_source, + const char *mime_type, int32_t fd) { + struct client_data_source *source = (struct client_data_source *)wlr_source; + gtk_primary_selection_source_send_send(source->resource, mime_type, fd); + close(fd); +} + +static void client_source_cancel( + struct wlr_primary_selection_source *wlr_source) { + struct client_data_source *source = (struct client_data_source *)wlr_source; + gtk_primary_selection_source_send_cancelled(source->resource); +} + static struct wlr_primary_selection_offer *source_send_offer( struct wlr_primary_selection_source *source, struct wlr_seat_client *target) { @@ -119,16 +126,15 @@ static struct wlr_primary_selection_offer *source_send_offer( static void source_handle_offer(struct wl_client *client, struct wl_resource *resource, const char *mime_type) { - struct wlr_primary_selection_source *source = - wl_resource_get_user_data(resource); + struct client_data_source *source = wl_resource_get_user_data(resource); - char **p = wl_array_add(&source->mime_types, sizeof(*p)); + char **p = wl_array_add(&source->source.mime_types, sizeof(*p)); if (p) { *p = strdup(mime_type); } if (p == NULL || *p == NULL) { if (p) { - source->mime_types.size -= sizeof(*p); + source->source.mime_types.size -= sizeof(*p); } wl_resource_post_no_memory(resource); } @@ -145,9 +151,9 @@ static const struct gtk_primary_selection_source_interface source_impl = { }; static void source_resource_handle_destroy(struct wl_resource *resource) { - struct wlr_primary_selection_source *source = + struct client_data_source *source = wl_resource_get_user_data(resource); - wlr_primary_selection_source_finish(source); + wlr_primary_selection_source_finish(&source->source); free(source); } @@ -197,6 +203,11 @@ static void seat_client_primary_selection_source_destroy( void wlr_seat_set_primary_selection(struct wlr_seat *seat, struct wlr_primary_selection_source *source, uint32_t serial) { + if (source) { + assert(source->send); + assert(source->cancel); + } + if (seat->primary_selection_source && seat->primary_selection_serial - serial < UINT32_MAX / 2) { return; @@ -230,7 +241,7 @@ void wlr_seat_set_primary_selection(struct wlr_seat *seat, static void device_handle_set_selection(struct wl_client *client, struct wl_resource *resource, struct wl_resource *source_resource, uint32_t serial) { - struct wlr_primary_selection_source *source = NULL; + struct client_data_source *source = NULL; if (source_resource != NULL) { source = wl_resource_get_user_data(source_resource); } @@ -239,7 +250,9 @@ static void device_handle_set_selection(struct wl_client *client, wl_resource_get_user_data(resource); // TODO: store serial and check against incoming serial here - wlr_seat_set_primary_selection(seat_client->seat, source, serial); + struct wlr_primary_selection_source *wlr_source = + (struct wlr_primary_selection_source *)source; + wlr_seat_set_primary_selection(seat_client->seat, wlr_source, serial); } static void device_handle_destroy(struct wl_client *client, @@ -280,13 +293,13 @@ void wlr_primary_selection_source_finish( static void device_manager_handle_create_source(struct wl_client *client, struct wl_resource *manager_resource, uint32_t id) { - struct wlr_primary_selection_source *source = - calloc(1, sizeof(struct wlr_primary_selection_source)); + struct client_data_source *source = + calloc(1, sizeof(struct client_data_source)); if (source == NULL) { wl_client_post_no_memory(client); return; } - wlr_primary_selection_source_init(source); + wlr_primary_selection_source_init(&source->source); int version = wl_resource_get_version(manager_resource); source->resource = wl_resource_create(client, @@ -299,8 +312,8 @@ static void device_manager_handle_create_source(struct wl_client *client, wl_resource_set_implementation(source->resource, &source_impl, source, source_resource_handle_destroy); - source->send = client_source_send; - source->cancel = client_source_cancel; + source->source.send = client_source_send; + source->source.cancel = client_source_cancel; } void device_manager_handle_get_device(struct wl_client *client, From 720c1154dc8bc79dd7d99fb5ef86982963645d25 Mon Sep 17 00:00:00 2001 From: emersion Date: Sat, 30 Dec 2017 09:26:48 +0100 Subject: [PATCH 4/6] Fix use-after-free when destroying an offer --- types/wlr_data_device.c | 3 ++- types/wlr_primary_selection.c | 5 +---- 2 files changed, 3 insertions(+), 5 deletions(-) diff --git a/types/wlr_data_device.c b/types/wlr_data_device.c index 804718f9..fc2060e0 100644 --- a/types/wlr_data_device.c +++ b/types/wlr_data_device.c @@ -173,6 +173,8 @@ static void data_offer_resource_destroy(struct wl_resource *resource) { goto out; } + offer->source->offer = NULL; + // If the drag destination has version < 3, wl_data_offer.finish // won't be called, so do this here as a safety net, because // we still want the version >= 3 drag source to be happy. @@ -183,7 +185,6 @@ static void data_offer_resource_destroy(struct wl_resource *resource) { offer->source->cancel(offer->source); } - offer->source->offer = NULL; out: free(offer); } diff --git a/types/wlr_primary_selection.c b/types/wlr_primary_selection.c index c29856f2..491145f4 100644 --- a/types/wlr_primary_selection.c +++ b/types/wlr_primary_selection.c @@ -44,11 +44,8 @@ static void offer_resource_handle_destroy(struct wl_resource *resource) { goto out; } - if (offer->source->cancel) { - offer->source->cancel(offer->source); - } - offer->source->offer = NULL; + out: free(offer); } From 38ed3b42451e350a97f6686324ca2beb10c1dbb3 Mon Sep 17 00:00:00 2001 From: emersion Date: Sat, 30 Dec 2017 09:58:04 +0100 Subject: [PATCH 5/6] seat: rename selection_source to selection_data_source, remove unused data_device --- include/wlr/types/wlr_seat.h | 3 +-- types/wlr_data_device.c | 16 ++++++++-------- types/wlr_seat.c | 7 +++---- xwayland/selection.c | 12 +++++++----- 4 files changed, 19 insertions(+), 19 deletions(-) diff --git a/include/wlr/types/wlr_seat.h b/include/wlr/types/wlr_seat.h index e049d4c7..432e5dc3 100644 --- a/include/wlr/types/wlr_seat.h +++ b/include/wlr/types/wlr_seat.h @@ -178,8 +178,7 @@ struct wlr_seat { uint32_t capabilities; struct timespec last_event; - struct wlr_data_device *data_device; // TODO needed? - struct wlr_data_source *selection_source; + struct wlr_data_source *selection_data_source; uint32_t selection_serial; struct wlr_primary_selection_source *primary_selection_source; diff --git a/types/wlr_data_device.c b/types/wlr_data_device.c index fc2060e0..ff209d9f 100644 --- a/types/wlr_data_device.c +++ b/types/wlr_data_device.c @@ -253,9 +253,9 @@ void wlr_seat_client_send_selection(struct wlr_seat_client *seat_client) { return; } - if (seat_client->seat->selection_source) { + if (seat_client->seat->selection_data_source) { struct wlr_data_offer *offer = wlr_data_source_send_offer( - seat_client->seat->selection_source, seat_client); + seat_client->seat->selection_data_source, seat_client); if (offer == NULL) { return; } @@ -285,7 +285,7 @@ static void seat_client_selection_data_source_destroy( } } - seat->selection_source = NULL; + seat->selection_data_source = NULL; wl_signal_emit(&seat->events.selection, seat); } @@ -297,18 +297,18 @@ void wlr_seat_set_selection(struct wlr_seat *seat, assert(source->cancel); } - if (seat->selection_source && + if (seat->selection_data_source && seat->selection_serial - serial < UINT32_MAX / 2) { return; } - if (seat->selection_source) { - seat->selection_source->cancel(seat->selection_source); - seat->selection_source = NULL; + if (seat->selection_data_source) { + seat->selection_data_source->cancel(seat->selection_data_source); + seat->selection_data_source = NULL; wl_list_remove(&seat->selection_data_source_destroy.link); } - seat->selection_source = source; + seat->selection_data_source = source; seat->selection_serial = serial; struct wlr_seat_client *focused_client = diff --git a/types/wlr_seat.c b/types/wlr_seat.c index ec95053b..79638822 100644 --- a/types/wlr_seat.c +++ b/types/wlr_seat.c @@ -352,9 +352,9 @@ void wlr_seat_destroy(struct wlr_seat *seat) { wl_list_remove(&seat->display_destroy.link); - if (seat->selection_source) { - seat->selection_source->cancel(seat->selection_source); - seat->selection_source = NULL; + if (seat->selection_data_source) { + seat->selection_data_source->cancel(seat->selection_data_source); + seat->selection_data_source = NULL; wl_list_remove(&seat->selection_data_source_destroy.link); } if (seat->primary_selection_source) { @@ -373,7 +373,6 @@ void wlr_seat_destroy(struct wlr_seat *seat) { free(seat->pointer_state.default_grab); free(seat->keyboard_state.default_grab); free(seat->touch_state.default_grab); - free(seat->data_device); free(seat->name); free(seat); } diff --git a/xwayland/selection.c b/xwayland/selection.c index 65599145..060c0baf 100644 --- a/xwayland/selection.c +++ b/xwayland/selection.c @@ -153,7 +153,8 @@ error_out: static void xwm_selection_source_send(struct wlr_xwm_selection *selection, const char *mime_type, int32_t fd) { if (selection == &selection->xwm->clipboard_selection) { - struct wlr_data_source *source = selection->xwm->seat->selection_source; + struct wlr_data_source *source = + selection->xwm->seat->selection_data_source; if (source != NULL) { source->send(source, mime_type, fd); return; @@ -214,7 +215,8 @@ static void xwm_selection_send_timestamp(struct wlr_xwm_selection *selection) { static struct wl_array *xwm_selection_source_get_mime_types( struct wlr_xwm_selection *selection) { if (selection == &selection->xwm->clipboard_selection) { - struct wlr_data_source *source = selection->xwm->seat->selection_source; + struct wlr_data_source *source = + selection->xwm->seat->selection_data_source; if (source != NULL) { return &source->mime_types; } @@ -834,8 +836,8 @@ void xwm_selection_finish(struct wlr_xwm *xwm) { xcb_destroy_window(xwm->xcb_conn, xwm->selection_window); } if (xwm->seat) { - if (xwm->seat->selection_source && - xwm->seat->selection_source->cancel == data_source_cancel) { + if (xwm->seat->selection_data_source && + xwm->seat->selection_data_source->cancel == data_source_cancel) { wlr_seat_set_selection(xwm->seat, NULL, wl_display_next_serial(xwm->xwayland->wl_display)); } @@ -871,7 +873,7 @@ static void seat_handle_selection(struct wl_listener *listener, struct wlr_seat *seat = data; struct wlr_xwm *xwm = wl_container_of(listener, xwm, seat_selection); - struct wlr_data_source *source = seat->selection_source; + struct wlr_data_source *source = seat->selection_data_source; if (source != NULL && source->send == data_source_send) { return; From c599d8024c5788aa4e15a026c46577723c8bb531 Mon Sep 17 00:00:00 2001 From: emersion Date: Wed, 3 Jan 2018 16:46:59 +0100 Subject: [PATCH 6/6] Remove serial validation TODOs --- types/wlr_data_device.c | 1 - types/wlr_primary_selection.c | 1 - 2 files changed, 2 deletions(-) diff --git a/types/wlr_data_device.c b/types/wlr_data_device.c index ff209d9f..a4eb233b 100644 --- a/types/wlr_data_device.c +++ b/types/wlr_data_device.c @@ -339,7 +339,6 @@ static void data_device_set_selection(struct wl_client *client, struct wlr_seat_client *seat_client = wl_resource_get_user_data(dd_resource); - // TODO: store serial and check against incoming serial here struct wlr_data_source *wlr_source = (struct wlr_data_source *)source; wlr_seat_set_selection(seat_client->seat, wlr_source, serial); } diff --git a/types/wlr_primary_selection.c b/types/wlr_primary_selection.c index 491145f4..fafde669 100644 --- a/types/wlr_primary_selection.c +++ b/types/wlr_primary_selection.c @@ -246,7 +246,6 @@ static void device_handle_set_selection(struct wl_client *client, struct wlr_seat_client *seat_client = wl_resource_get_user_data(resource); - // TODO: store serial and check against incoming serial here struct wlr_primary_selection_source *wlr_source = (struct wlr_primary_selection_source *)source; wlr_seat_set_primary_selection(seat_client->seat, wlr_source, serial);