From f8de7aa0fd9b2e30fa13a22fc91e963c32c7cf7d Mon Sep 17 00:00:00 2001 From: emersion Date: Thu, 2 Nov 2017 23:14:24 +0100 Subject: [PATCH] Add checks for wl_resource_create --- types/wlr_compositor.c | 23 +++++++++++++--- types/wlr_data_device.c | 16 +++++------ types/wlr_gamma_control.c | 9 +++---- types/wlr_output.c | 8 ++++-- types/wlr_screenshooter.c | 50 ++++++++++++++++++++++------------- types/wlr_seat.c | 23 +++++++++++++--- types/wlr_server_decoration.c | 8 +++--- types/wlr_surface.c | 15 +++++++++-- types/wlr_wl_shell.c | 29 +++++++++++++------- types/wlr_xdg_shell_v6.c | 36 ++++++++++++++++++++----- 10 files changed, 154 insertions(+), 63 deletions(-) diff --git a/types/wlr_compositor.c b/types/wlr_compositor.c index cbb7f138..986b7e60 100644 --- a/types/wlr_compositor.c +++ b/types/wlr_compositor.c @@ -13,10 +13,21 @@ static void destroy_surface_listener(struct wl_listener *listener, void *data) { static void wl_compositor_create_surface(struct wl_client *client, struct wl_resource *resource, uint32_t id) { struct wlr_compositor *compositor = wl_resource_get_user_data(resource); + struct wl_resource *surface_resource = wl_resource_create(client, - &wl_surface_interface, wl_resource_get_version(resource), id); + &wl_surface_interface, wl_resource_get_version(resource), id); + if (surface_resource == NULL) { + wl_resource_post_no_memory(resource); + return; + } + struct wlr_surface *surface = wlr_surface_create(surface_resource, - compositor->renderer); + compositor->renderer); + if (surface == NULL) { + wl_resource_destroy(surface_resource); + wl_resource_post_no_memory(resource); + return; + } surface->compositor_data = compositor; surface->compositor_listener.notify = &destroy_surface_listener; wl_resource_add_destroy_listener(surface_resource, @@ -49,13 +60,17 @@ static void wl_compositor_destroy(struct wl_resource *resource) { } } -static void wl_compositor_bind(struct wl_client *wl_client, void *_compositor, +static void wl_compositor_bind(struct wl_client *wl_client, void *data, uint32_t version, uint32_t id) { - struct wlr_compositor *compositor = _compositor; + struct wlr_compositor *compositor = data; assert(wl_client && compositor); struct wl_resource *wl_resource = wl_resource_create(wl_client, &wl_compositor_interface, version, id); + if (wl_resource == NULL) { + wl_client_post_no_memory(wl_client); + return; + } wl_resource_set_implementation(wl_resource, &wl_compositor_impl, compositor, wl_compositor_destroy); wl_list_insert(&compositor->wl_resources, diff --git a/types/wlr_data_device.c b/types/wlr_data_device.c index 5800c0f3..ee83260e 100644 --- a/types/wlr_data_device.c +++ b/types/wlr_data_device.c @@ -233,6 +233,9 @@ static struct wlr_data_offer *wlr_data_source_send_offer( struct wlr_data_source *source, struct wl_resource *target) { struct wlr_data_offer *offer = calloc(1, sizeof(struct wlr_data_offer)); + if (offer == NULL) { + return NULL; + } offer->resource = wl_resource_create(wl_resource_get_client(target), @@ -781,18 +784,16 @@ data_device_manager_impl = { static void data_device_manager_bind(struct wl_client *client, void *data, uint32_t version, uint32_t id) { - struct wl_resource *resource; - - resource = wl_resource_create(client, - &wl_data_device_manager_interface, - version, id); + struct wl_resource *resource = wl_resource_create(client, + &wl_data_device_manager_interface, + version, id); if (resource == NULL) { wl_client_post_no_memory(client); return; } - wl_resource_set_implementation(resource, - &data_device_manager_impl, NULL, NULL); + wl_resource_set_implementation(resource, &data_device_manager_impl, + NULL, NULL); } struct wlr_data_device_manager *wlr_data_device_manager_create( @@ -807,7 +808,6 @@ struct wlr_data_device_manager *wlr_data_device_manager_create( manager->global = wl_global_create(display, &wl_data_device_manager_interface, 3, NULL, data_device_manager_bind); - if (!manager->global) { wlr_log(L_ERROR, "could not create data device manager wl global"); free(manager); diff --git a/types/wlr_gamma_control.c b/types/wlr_gamma_control.c index e765c2a5..355e56a2 100644 --- a/types/wlr_gamma_control.c +++ b/types/wlr_gamma_control.c @@ -83,8 +83,8 @@ static void gamma_control_manager_get_gamma_control(struct wl_client *client, gamma_control->resource = wl_resource_create(client, &gamma_control_interface, version, id); if (gamma_control->resource == NULL) { - wl_client_post_no_memory(client); free(gamma_control); + wl_client_post_no_memory(client); return; } wlr_log(L_DEBUG, "new gamma_control %p (res %p)", gamma_control, @@ -109,10 +109,9 @@ static struct gamma_control_manager_interface gamma_control_manager_impl = { .get_gamma_control = gamma_control_manager_get_gamma_control, }; -static void gamma_control_manager_bind(struct wl_client *client, - void *_gamma_control_manager, uint32_t version, uint32_t id) { - struct wlr_gamma_control_manager *gamma_control_manager = - _gamma_control_manager; +static void gamma_control_manager_bind(struct wl_client *client, void *data, + uint32_t version, uint32_t id) { + struct wlr_gamma_control_manager *gamma_control_manager = data; assert(client && gamma_control_manager); struct wl_resource *resource = wl_resource_create(client, diff --git a/types/wlr_output.c b/types/wlr_output.c index 24cc5892..5456d064 100644 --- a/types/wlr_output.c +++ b/types/wlr_output.c @@ -91,13 +91,17 @@ static struct wl_output_interface wl_output_impl = { .release = wl_output_release }; -static void wl_output_bind(struct wl_client *wl_client, void *_wlr_output, +static void wl_output_bind(struct wl_client *wl_client, void *data, uint32_t version, uint32_t id) { - struct wlr_output *wlr_output = _wlr_output; + struct wlr_output *wlr_output = data; assert(wl_client && wlr_output); struct wl_resource *wl_resource = wl_resource_create(wl_client, &wl_output_interface, version, id); + if (wl_resource == NULL) { + wl_client_post_no_memory(wl_client); + return; + } wl_resource_set_implementation(wl_resource, &wl_output_impl, wlr_output, wl_output_destroy); wl_list_insert(&wlr_output->wl_resources, diff --git a/types/wlr_screenshooter.c b/types/wlr_screenshooter.c index c6def691..a78c7ad7 100644 --- a/types/wlr_screenshooter.c +++ b/types/wlr_screenshooter.c @@ -51,16 +51,17 @@ static void output_frame_notify(struct wl_listener *listener, void *_data) { } static void screenshooter_shoot(struct wl_client *client, - struct wl_resource *_screenshooter, uint32_t id, - struct wl_resource *_output, struct wl_resource *_buffer) { + struct wl_resource *screenshooter_resource, uint32_t id, + struct wl_resource *output_resource, + struct wl_resource *buffer_resource) { struct wlr_screenshooter *screenshooter = - wl_resource_get_user_data(_screenshooter); - struct wlr_output *output = wl_resource_get_user_data(_output); - if (!wl_shm_buffer_get(_buffer)) { + wl_resource_get_user_data(screenshooter_resource); + struct wlr_output *output = wl_resource_get_user_data(output_resource); + if (!wl_shm_buffer_get(buffer_resource)) { wlr_log(L_ERROR, "Invalid buffer: not a shared memory buffer"); return; } - struct wl_shm_buffer *shm_buffer = wl_shm_buffer_get(_buffer); + struct wl_shm_buffer *shm_buffer = wl_shm_buffer_get(buffer_resource); int32_t width = wl_shm_buffer_get_width(shm_buffer); int32_t height = wl_shm_buffer_get_height(shm_buffer); int32_t stride = wl_shm_buffer_get_stride(shm_buffer); @@ -84,23 +85,31 @@ static void screenshooter_shoot(struct wl_client *client, struct wlr_screenshot *screenshot = calloc(1, sizeof(struct wlr_screenshot)); if (!screenshot) { - wl_client_post_no_memory(client); + wl_resource_post_no_memory(screenshooter_resource); return; } - screenshot->output_resource = _output; + screenshot->output_resource = output_resource; screenshot->output = output; screenshot->screenshooter = screenshooter; screenshot->resource = wl_resource_create(client, - &orbital_screenshot_interface, wl_resource_get_version(_screenshooter), - id); - wlr_log(L_DEBUG, "new screenshot %p (res %p)", screenshot, - screenshot->resource); + &orbital_screenshot_interface, + wl_resource_get_version(screenshooter_resource), id); + if (screenshot->resource == NULL) { + free(screenshot); + wl_resource_post_no_memory(screenshooter_resource); + return; + } wl_resource_set_implementation(screenshot->resource, NULL, screenshot, NULL); + wlr_log(L_DEBUG, "new screenshot %p (res %p)", screenshot, + screenshot->resource); + struct screenshot_state *state = calloc(1, sizeof(struct screenshot_state)); if (!state) { - wl_client_post_no_memory(client); + wl_resource_destroy(screenshot->resource); + free(screenshot); + wl_resource_post_no_memory(screenshooter_resource); return; } state->width = width; @@ -117,13 +126,17 @@ static struct orbital_screenshooter_interface screenshooter_impl = { .shoot = screenshooter_shoot, }; -static void screenshooter_bind(struct wl_client *wl_client, - void *_screenshooter, uint32_t version, uint32_t id) { - struct wlr_screenshooter *screenshooter = _screenshooter; +static void screenshooter_bind(struct wl_client *wl_client, void *data, + uint32_t version, uint32_t id) { + struct wlr_screenshooter *screenshooter = data; assert(wl_client && screenshooter); struct wl_resource *wl_resource = wl_resource_create(wl_client, &orbital_screenshooter_interface, version, id); + if (wl_resource == NULL) { + wl_client_post_no_memory(wl_client); + return; + } wl_resource_set_implementation(wl_resource, &screenshooter_impl, screenshooter, NULL); } @@ -137,13 +150,12 @@ struct wlr_screenshooter *wlr_screenshooter_create(struct wl_display *display, } screenshooter->renderer = renderer; - struct wl_global *wl_global = wl_global_create(display, + screenshooter->wl_global = wl_global_create(display, &orbital_screenshooter_interface, 1, screenshooter, screenshooter_bind); - if (!wl_global) { + if (screenshooter->wl_global == NULL) { free(screenshooter); return NULL; } - screenshooter->wl_global = wl_global; return screenshooter; } diff --git a/types/wlr_seat.c b/types/wlr_seat.c index bdea96ac..76b34d31 100644 --- a/types/wlr_seat.c +++ b/types/wlr_seat.c @@ -65,9 +65,9 @@ static void wl_pointer_destroy(struct wl_resource *resource) { } static void wl_seat_get_pointer(struct wl_client *client, - struct wl_resource *pointer_resource, uint32_t id) { + struct wl_resource *seat_resource, uint32_t id) { struct wlr_seat_client *seat_client = - wl_resource_get_user_data(pointer_resource); + wl_resource_get_user_data(seat_resource); if (!(seat_client->seat->capabilities & WL_SEAT_CAPABILITY_POINTER)) { return; } @@ -78,7 +78,11 @@ static void wl_seat_get_pointer(struct wl_client *client, wl_resource_destroy(seat_client->pointer); } seat_client->pointer = wl_resource_create(client, &wl_pointer_interface, - wl_resource_get_version(pointer_resource), id); + wl_resource_get_version(seat_resource), id); + if (seat_client->pointer == NULL) { + wl_resource_post_no_memory(seat_resource); + return; + } wl_resource_set_implementation(seat_client->pointer, &wl_pointer_impl, seat_client, &wl_pointer_destroy); } @@ -126,6 +130,10 @@ static void wl_seat_get_keyboard(struct wl_client *client, } seat_client->keyboard = wl_resource_create(client, &wl_keyboard_interface, wl_resource_get_version(seat_resource), id); + if (seat_client->keyboard == NULL) { + wl_resource_post_no_memory(seat_resource); + return; + } wl_resource_set_implementation(seat_client->keyboard, &wl_keyboard_impl, seat_client, &wl_keyboard_destroy); @@ -162,6 +170,10 @@ static void wl_seat_get_touch(struct wl_client *client, } seat_client->touch = wl_resource_create(client, &wl_touch_interface, wl_resource_get_version(seat_resource), id); + if (seat_client->touch == NULL) { + wl_resource_post_no_memory(seat_resource); + return; + } wl_resource_set_implementation(seat_client->touch, &wl_touch_impl, seat_client, &wl_touch_destroy); } @@ -213,6 +225,11 @@ static void wl_seat_bind(struct wl_client *client, void *_wlr_seat, } seat_client->wl_resource = wl_resource_create(client, &wl_seat_interface, version, id); + if (seat_client->wl_resource == NULL) { + free(seat_client); + wl_client_post_no_memory(client); + return; + } seat_client->client = client; seat_client->seat = wlr_seat; wl_resource_set_implementation(seat_client->wl_resource, &wl_seat_impl, diff --git a/types/wlr_server_decoration.c b/types/wlr_server_decoration.c index 0664edfb..60cc6369 100644 --- a/types/wlr_server_decoration.c +++ b/types/wlr_server_decoration.c @@ -73,8 +73,8 @@ static void server_decoration_manager_handle_create(struct wl_client *client, decoration->resource = wl_resource_create(client, &org_kde_kwin_server_decoration_interface, version, id); if (decoration->resource == NULL) { - wl_client_post_no_memory(client); free(decoration); + wl_client_post_no_memory(client); return; } wl_resource_set_implementation(decoration->resource, @@ -120,9 +120,9 @@ void server_decoration_manager_destroy_resource(struct wl_resource *resource) { wl_list_remove(wl_resource_get_link(resource)); } -static void server_decoration_manager_bind(struct wl_client *client, - void *_manager, uint32_t version, uint32_t id) { - struct wlr_server_decoration_manager *manager = _manager; +static void server_decoration_manager_bind(struct wl_client *client, void *data, + uint32_t version, uint32_t id) { + struct wlr_server_decoration_manager *manager = data; assert(client && manager); struct wl_resource *resource = wl_resource_create(client, diff --git a/types/wlr_surface.c b/types/wlr_surface.c index 2a8c4d04..48971849 100644 --- a/types/wlr_surface.c +++ b/types/wlr_surface.c @@ -819,14 +819,21 @@ static void subsurface_handle_parent_destroy(struct wl_listener *listener, void wlr_surface_make_subsurface(struct wlr_surface *surface, struct wlr_surface *parent, uint32_t id) { + struct wl_client *client = wl_resource_get_client(surface->resource); assert(surface->subsurface == NULL); struct wlr_subsurface *subsurface = calloc(1, sizeof(struct wlr_subsurface)); if (!subsurface) { + wl_client_post_no_memory(client); return; } subsurface->cached = wlr_surface_state_create(); + if (subsurface->cached == NULL) { + free(subsurface); + wl_client_post_no_memory(client); + return; + } subsurface->synchronized = true; subsurface->surface = surface; @@ -840,10 +847,14 @@ void wlr_surface_make_subsurface(struct wlr_surface *surface, wl_list_insert(&parent->subsurface_pending_list, &subsurface->parent_pending_link); - struct wl_client *client = wl_resource_get_client(surface->resource); - subsurface->resource = wl_resource_create(client, &wl_subsurface_interface, 1, id); + if (subsurface->resource == NULL) { + wlr_surface_state_destroy(subsurface->cached); + free(subsurface); + wl_client_post_no_memory(client); + return; + } wl_resource_set_implementation(subsurface->resource, &subsurface_implementation, subsurface, diff --git a/types/wlr_wl_shell.c b/types/wlr_wl_shell.c index dd1f10b2..fe61075e 100644 --- a/types/wlr_wl_shell.c +++ b/types/wlr_wl_shell.c @@ -506,19 +506,19 @@ static int shell_surface_ping_timeout(void *user_data) { } static void shell_protocol_get_shell_surface(struct wl_client *client, - struct wl_resource *resource, uint32_t id, + struct wl_resource *shell_resource, uint32_t id, struct wl_resource *surface_resource) { struct wlr_surface *surface = wl_resource_get_user_data(surface_resource); if (wlr_surface_set_role(surface, wlr_wl_shell_surface_role, - resource, WL_SHELL_ERROR_ROLE)) { + shell_resource, WL_SHELL_ERROR_ROLE)) { return; } - struct wlr_wl_shell *wl_shell = wl_resource_get_user_data(resource); + struct wlr_wl_shell *wl_shell = wl_resource_get_user_data(shell_resource); struct wlr_wl_shell_surface *wl_surface = calloc(1, sizeof(struct wlr_wl_shell_surface)); if (wl_surface == NULL) { - wl_client_post_no_memory(client); + wl_resource_post_no_memory(shell_resource); return; } wl_list_init(&wl_surface->grab_link); @@ -530,13 +530,20 @@ static void shell_protocol_get_shell_surface(struct wl_client *client, wl_surface->surface = surface; wl_surface->resource = wl_resource_create(client, - &wl_shell_surface_interface, wl_resource_get_version(resource), id); - wlr_log(L_DEBUG, "new wl_shell %p (res %p)", wl_surface, - wl_surface->resource); + &wl_shell_surface_interface, wl_resource_get_version(shell_resource), + id); + if (wl_surface->resource == NULL) { + free(wl_surface); + wl_resource_post_no_memory(shell_resource); + return; + } wl_resource_set_implementation(wl_surface->resource, &shell_surface_impl, wl_surface, shell_surface_resource_destroy); + wlr_log(L_DEBUG, "new wl_shell %p (res %p)", wl_surface, + wl_surface->resource); + wl_signal_init(&wl_surface->events.destroy); wl_signal_init(&wl_surface->events.ping_timeout); wl_signal_init(&wl_surface->events.request_move); @@ -574,13 +581,17 @@ static void shell_destroy(struct wl_resource *resource) { wl_list_remove(wl_resource_get_link(resource)); } -static void shell_bind(struct wl_client *wl_client, void *_wl_shell, +static void shell_bind(struct wl_client *wl_client, void *data, uint32_t version, uint32_t id) { - struct wlr_wl_shell *wl_shell = _wl_shell; + struct wlr_wl_shell *wl_shell = data; assert(wl_client && wl_shell); struct wl_resource *wl_resource = wl_resource_create(wl_client, &wl_shell_interface, version, id); + if (wl_resource == NULL) { + wl_client_post_no_memory(wl_client); + return; + } wl_resource_set_implementation(wl_resource, &shell_impl, wl_shell, shell_destroy); wl_list_insert(&wl_shell->wl_resources, wl_resource_get_link(wl_resource)); diff --git a/types/wlr_xdg_shell_v6.c b/types/wlr_xdg_shell_v6.c index af551b10..fc45bc17 100644 --- a/types/wlr_xdg_shell_v6.c +++ b/types/wlr_xdg_shell_v6.c @@ -343,8 +343,8 @@ static void xdg_shell_create_positioner(struct wl_client *wl_client, wl_resource_get_version(resource), id); if (positioner->resource == NULL) { - wl_client_post_no_memory(wl_client); free(positioner); + wl_client_post_no_memory(wl_client); return; } @@ -480,13 +480,13 @@ static void xdg_surface_get_popup(struct wl_client *client, } if (wlr_surface_set_role(surface->surface, wlr_desktop_xdg_popup_role, - resource, ZXDG_SHELL_V6_ERROR_ROLE)) { + resource, ZXDG_SHELL_V6_ERROR_ROLE)) { return; } surface->popup_state = calloc(1, sizeof(struct wlr_xdg_popup_v6)); if (!surface->popup_state) { - wl_client_post_no_memory(client); + wl_resource_post_no_memory(resource); return; } @@ -495,6 +495,7 @@ static void xdg_surface_get_popup(struct wl_client *client, wl_resource_get_version(resource), id); if (surface->popup_state->resource == NULL) { free(surface->popup_state); + wl_resource_post_no_memory(resource); return; } @@ -735,7 +736,7 @@ static void xdg_surface_get_toplevel(struct wl_client *client, surface->toplevel_state = calloc(1, sizeof(struct wlr_xdg_toplevel_v6)); if (surface->toplevel_state == NULL) { - wl_client_post_no_memory(client); + wl_resource_post_no_memory(resource); return; } @@ -744,6 +745,11 @@ static void xdg_surface_get_toplevel(struct wl_client *client, struct wl_resource *toplevel_resource = wl_resource_create(client, &zxdg_toplevel_v6_interface, wl_resource_get_version(resource), id); + if (toplevel_resource == NULL) { + free(surface->toplevel_state); + wl_resource_post_no_memory(resource); + return; + } surface->toplevel_state->resource = toplevel_resource; @@ -1118,9 +1124,20 @@ static void xdg_shell_get_xdg_surface(struct wl_client *wl_client, surface->resource = wl_resource_create(wl_client, &zxdg_surface_v6_interface, wl_resource_get_version(client_resource), id); + if (surface->resource == NULL) { + free(surface->next_geometry); + free(surface->geometry); + free(surface); + wl_client_post_no_memory(wl_client); + return; + } if (wlr_surface_has_buffer(surface->surface)) { - wl_resource_post_error(surface->resource, + wl_resource_destroy(surface->resource); + free(surface->next_geometry); + free(surface->geometry); + free(surface); + wl_resource_post_error(surface_resource, ZXDG_SURFACE_V6_ERROR_UNCONFIGURED_BUFFER, "xdg_surface must not have a buffer at creation"); return; @@ -1199,9 +1216,9 @@ static int wlr_xdg_client_v6_ping_timeout(void *user_data) { return 1; } -static void xdg_shell_bind(struct wl_client *wl_client, void *_xdg_shell, +static void xdg_shell_bind(struct wl_client *wl_client, void *data, uint32_t version, uint32_t id) { - struct wlr_xdg_shell_v6 *xdg_shell = _xdg_shell; + struct wlr_xdg_shell_v6 *xdg_shell = data; assert(wl_client && xdg_shell); struct wlr_xdg_client_v6 *client = @@ -1215,6 +1232,11 @@ static void xdg_shell_bind(struct wl_client *wl_client, void *_xdg_shell, client->resource = wl_resource_create(wl_client, &zxdg_shell_v6_interface, version, id); + if (client->resource == NULL) { + free(client); + wl_client_post_no_memory(wl_client); + return; + } client->client = wl_client; client->shell = xdg_shell;