Check allocation errors in wlr_output_layout

This commit is contained in:
emersion 2017-10-31 11:29:53 +01:00
parent 37b1e8bd25
commit 78245a3998
No known key found for this signature in database
GPG key ID: 0FDE7BE0E88F5E48

View file

@ -8,14 +8,14 @@
#include <assert.h> #include <assert.h>
struct wlr_output_layout_state { struct wlr_output_layout_state {
struct wlr_box *_box; struct wlr_box box;
}; };
struct wlr_output_layout_output_state { struct wlr_output_layout_output_state {
struct wlr_output_layout *layout; struct wlr_output_layout *layout;
struct wlr_output_layout_output *l_output; struct wlr_output_layout_output *l_output;
struct wlr_box *_box; struct wlr_box box;
bool auto_configured; bool auto_configured;
struct wl_listener resolution; struct wl_listener resolution;
@ -25,8 +25,14 @@ struct wlr_output_layout_output_state {
struct wlr_output_layout *wlr_output_layout_create() { struct wlr_output_layout *wlr_output_layout_create() {
struct wlr_output_layout *layout = struct wlr_output_layout *layout =
calloc(1, sizeof(struct wlr_output_layout)); calloc(1, sizeof(struct wlr_output_layout));
if (layout == NULL) {
return NULL;
}
layout->state = calloc(1, sizeof(struct wlr_output_layout_state)); layout->state = calloc(1, sizeof(struct wlr_output_layout_state));
layout->state->_box = calloc(1, sizeof(struct wlr_box)); if (layout->state == NULL) {
free(layout);
return NULL;
}
wl_list_init(&layout->outputs); wl_list_init(&layout->outputs);
wl_signal_init(&layout->events.change); wl_signal_init(&layout->events.change);
@ -40,7 +46,6 @@ static void wlr_output_layout_output_destroy(
wl_list_remove(&l_output->state->resolution.link); wl_list_remove(&l_output->state->resolution.link);
wl_list_remove(&l_output->state->output_destroy.link); wl_list_remove(&l_output->state->output_destroy.link);
wl_list_remove(&l_output->link); wl_list_remove(&l_output->link);
free(l_output->state->_box);
free(l_output->state); free(l_output->state);
free(l_output); free(l_output);
} }
@ -52,25 +57,24 @@ void wlr_output_layout_destroy(struct wlr_output_layout *layout) {
wl_signal_emit(&layout->events.destroy, layout); wl_signal_emit(&layout->events.destroy, layout);
struct wlr_output_layout_output *_output, *temp = NULL; struct wlr_output_layout_output *output, *temp = NULL;
wl_list_for_each_safe(_output, temp, &layout->outputs, link) { wl_list_for_each_safe(output, temp, &layout->outputs, link) {
wlr_output_layout_output_destroy(_output); wlr_output_layout_output_destroy(output);
} }
free(layout->state->_box);
free(layout->state); free(layout->state);
free(layout); free(layout);
} }
static struct wlr_box *wlr_output_layout_output_get_box( static struct wlr_box *wlr_output_layout_output_get_box(
struct wlr_output_layout_output *l_output) { struct wlr_output_layout_output *l_output) {
l_output->state->_box->x = l_output->x; l_output->state->box.x = l_output->x;
l_output->state->_box->y = l_output->y; l_output->state->box.y = l_output->y;
int width, height; int width, height;
wlr_output_effective_resolution(l_output->output, &width, &height); wlr_output_effective_resolution(l_output->output, &width, &height);
l_output->state->_box->width = width; l_output->state->box.width = width;
l_output->state->_box->height = height; l_output->state->box.height = height;
return l_output->state->_box; return &l_output->state->box;
} }
/** /**
@ -138,18 +142,24 @@ static void handle_output_destroy(struct wl_listener *listener, void *data) {
static struct wlr_output_layout_output *wlr_output_layout_output_create( static struct wlr_output_layout_output *wlr_output_layout_output_create(
struct wlr_output_layout *layout, struct wlr_output *output) { struct wlr_output_layout *layout, struct wlr_output *output) {
struct wlr_output_layout_output *l_output; struct wlr_output_layout_output *l_output =
l_output= calloc(1, sizeof(struct wlr_output_layout_output)); calloc(1, sizeof(struct wlr_output_layout_output));
if (l_output == NULL) {
return NULL;
}
l_output->state = calloc(1, sizeof(struct wlr_output_layout_output_state)); l_output->state = calloc(1, sizeof(struct wlr_output_layout_output_state));
if (l_output->state == NULL) {
free(l_output);
return NULL;
}
l_output->state->l_output = l_output; l_output->state->l_output = l_output;
l_output->state->_box = calloc(1, sizeof(struct wlr_box));
l_output->state->layout = layout; l_output->state->layout = layout;
l_output->output = output; l_output->output = output;
wl_list_insert(&layout->outputs, &l_output->link); wl_list_insert(&layout->outputs, &l_output->link);
wl_signal_add(&output->events.resolution, &l_output->state->resolution); wl_signal_add(&output->events.resolution, &l_output->state->resolution);
l_output->state->resolution.notify = handle_output_resolution; l_output->state->resolution.notify = handle_output_resolution;
wl_signal_add(&output->events.destroy, &l_output->state->output_destroy); wl_signal_add(&output->events.destroy, &l_output->state->output_destroy);
l_output->state->output_destroy.notify = handle_output_destroy; l_output->state->output_destroy.notify = handle_output_destroy;
@ -162,6 +172,10 @@ void wlr_output_layout_add(struct wlr_output_layout *layout,
wlr_output_layout_get(layout, output); wlr_output_layout_get(layout, output);
if (!l_output) { if (!l_output) {
l_output = wlr_output_layout_output_create(layout, output); l_output = wlr_output_layout_output_create(layout, output);
if (!l_output) {
wlr_log(L_ERROR, "Failed to create wlr_output_layout_output");
return;
}
} }
l_output->x = x; l_output->x = x;
l_output->y = y; l_output->y = y;
@ -238,8 +252,8 @@ void wlr_output_layout_move(struct wlr_output_layout *layout,
void wlr_output_layout_remove(struct wlr_output_layout *layout, void wlr_output_layout_remove(struct wlr_output_layout *layout,
struct wlr_output *output) { struct wlr_output *output) {
struct wlr_output_layout_output *l_output; struct wlr_output_layout_output *l_output =
l_output= wlr_output_layout_get(layout, output); wlr_output_layout_get(layout, output);
if (l_output) { if (l_output) {
wlr_output_layout_output_destroy(l_output); wlr_output_layout_output_destroy(l_output);
wlr_output_layout_reconfigure(layout); wlr_output_layout_reconfigure(layout);
@ -324,12 +338,12 @@ struct wlr_box *wlr_output_layout_get_box(
} }
} }
layout->state->_box->x = min_x; layout->state->box.x = min_x;
layout->state->_box->y = min_y; layout->state->box.y = min_y;
layout->state->_box->width = max_x - min_x; layout->state->box.width = max_x - min_x;
layout->state->_box->height = max_y - min_y; layout->state->box.height = max_y - min_y;
return layout->state->_box; return &layout->state->box;
} }
// not reached // not reached
@ -339,9 +353,12 @@ void wlr_output_layout_add_auto(struct wlr_output_layout *layout,
struct wlr_output *output) { struct wlr_output *output) {
struct wlr_output_layout_output *l_output = struct wlr_output_layout_output *l_output =
wlr_output_layout_get(layout, output); wlr_output_layout_get(layout, output);
if (!l_output) { if (!l_output) {
l_output = wlr_output_layout_output_create(layout, output); l_output = wlr_output_layout_output_create(layout, output);
if (!l_output) {
wlr_log(L_ERROR, "Failed to create wlr_output_layout_output");
return;
}
} }
l_output->state->auto_configured = true; l_output->state->auto_configured = true;