mirror of
https://github.com/hyprwm/wlroots-hyprland.git
synced 2024-11-22 21:05:58 +01:00
ENOMEM checks: consistently check wl_array_add return
This commit is contained in:
parent
9c163b7d38
commit
3eb4fa15ee
5 changed files with 40 additions and 4 deletions
|
@ -907,6 +907,9 @@ static void data_source_offer(struct wl_client *client,
|
|||
*p = strdup(mime_type);
|
||||
}
|
||||
if (!p || !*p){
|
||||
if (p) {
|
||||
source->mime_types.size -= sizeof *p;
|
||||
}
|
||||
wl_resource_post_no_memory(resource);
|
||||
}
|
||||
}
|
||||
|
|
|
@ -127,6 +127,9 @@ static void source_handle_offer(struct wl_client *client,
|
|||
*p = strdup(mime_type);
|
||||
}
|
||||
if (p == NULL || *p == NULL) {
|
||||
if (p) {
|
||||
source->mime_types.size -= sizeof(*p);
|
||||
}
|
||||
wl_resource_post_no_memory(resource);
|
||||
}
|
||||
}
|
||||
|
|
|
@ -876,6 +876,11 @@ void wlr_seat_keyboard_enter(struct wlr_seat *seat,
|
|||
wl_array_init(&keys);
|
||||
for (size_t i = 0; i < keyboard->num_keycodes; ++i) {
|
||||
uint32_t *p = wl_array_add(&keys, sizeof(uint32_t));
|
||||
if (!p) {
|
||||
wlr_log(L_ERROR, "Cannot allocate memory, skipping keycode: %d\n",
|
||||
keyboard->keycodes[i]);
|
||||
continue;
|
||||
}
|
||||
*p = keyboard->keycodes[i];
|
||||
}
|
||||
uint32_t serial = wl_display_next_serial(seat->display);
|
||||
|
|
|
@ -922,20 +922,36 @@ static void wlr_xdg_toplevel_v6_send_configure(
|
|||
wl_array_init(&states);
|
||||
if (surface->toplevel_state->pending.maximized) {
|
||||
s = wl_array_add(&states, sizeof(uint32_t));
|
||||
if (!s) {
|
||||
wlr_log(L_ERROR, "Could not allocate state for maximized xdg_toplevel");
|
||||
} else {
|
||||
*s = ZXDG_TOPLEVEL_V6_STATE_MAXIMIZED;
|
||||
}
|
||||
}
|
||||
if (surface->toplevel_state->pending.fullscreen) {
|
||||
s = wl_array_add(&states, sizeof(uint32_t));
|
||||
if (!s) {
|
||||
wlr_log(L_ERROR, "Could not allocate state for fullscreen xdg_toplevel");
|
||||
} else {
|
||||
*s = ZXDG_TOPLEVEL_V6_STATE_FULLSCREEN;
|
||||
}
|
||||
}
|
||||
if (surface->toplevel_state->pending.resizing) {
|
||||
s = wl_array_add(&states, sizeof(uint32_t));
|
||||
if (!s) {
|
||||
wlr_log(L_ERROR, "Could not allocate state for resizing xdg_toplevel");
|
||||
} else {
|
||||
*s = ZXDG_TOPLEVEL_V6_STATE_RESIZING;
|
||||
}
|
||||
}
|
||||
if (surface->toplevel_state->pending.activated) {
|
||||
s = wl_array_add(&states, sizeof(uint32_t));
|
||||
if (!s) {
|
||||
wlr_log(L_ERROR, "Could not allocate state for activated xdg_toplevel");
|
||||
} else {
|
||||
*s = ZXDG_TOPLEVEL_V6_STATE_ACTIVATED;
|
||||
}
|
||||
}
|
||||
|
||||
uint32_t width = surface->toplevel_state->pending.width;
|
||||
uint32_t height = surface->toplevel_state->pending.height;
|
||||
|
|
|
@ -55,6 +55,15 @@ static int xwm_read_data_source(int fd, uint32_t mask, void *data) {
|
|||
int current = selection->source_data.size;
|
||||
if (selection->source_data.size < incr_chunk_size) {
|
||||
p = wl_array_add(&selection->source_data, incr_chunk_size);
|
||||
if (!p){
|
||||
wlr_log(L_ERROR, "Could not allocate selection source_data to read into, throwing away some input");
|
||||
/* if we just return now, we'll just be called
|
||||
* again right away - force read something.
|
||||
* 1K on stack is probably fine? */
|
||||
char junk[1024];
|
||||
read(fd, junk, sizeof(junk));
|
||||
return 1;
|
||||
}
|
||||
} else {
|
||||
p = (char *) selection->source_data.data + selection->source_data.size;
|
||||
}
|
||||
|
|
Loading…
Reference in a new issue