wlt: toolkit: fix stuck key repeat bug
[platform/upstream/kmscon.git] / src / wlt_toolkit.c
1 /*
2  * wlt - Toolkit Helper
3  *
4  * Copyright (c) 2012 David Herrmann <dh.herrmann@googlemail.com>
5  *
6  * Permission is hereby granted, free of charge, to any person obtaining
7  * a copy of this software and associated documentation files
8  * (the "Software"), to deal in the Software without restriction, including
9  * without limitation the rights to use, copy, modify, merge, publish,
10  * distribute, sublicense, and/or sell copies of the Software, and to
11  * permit persons to whom the Software is furnished to do so, subject to
12  * the following conditions:
13  *
14  * The above copyright notice and this permission notice shall be included
15  * in all copies or substantial portions of the Software.
16  *
17  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
18  * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
19  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
20  * IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
21  * CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
22  * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
23  * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
24  */
25
26 /*
27  * Wayland Terminal toolkit helpers
28  */
29
30 #include <errno.h>
31 #include <fcntl.h>
32 #include <stdlib.h>
33 #include <string.h>
34 #include <sys/mman.h>
35 #include <unistd.h>
36 #include <wayland-client.h>
37 #include <wayland-cursor.h>
38 #include <xkbcommon/xkbcommon.h>
39 #include "eloop.h"
40 #include "log.h"
41 #include "shl_array.h"
42 #include "shl_dlist.h"
43 #include "shl_hook.h"
44 #include "shl_misc.h"
45 #include "tsm_vte.h"
46 #include "wlt_main.h"
47 #include "wlt_toolkit.h"
48
49 #define LOG_SUBSYSTEM "wlt_toolkit"
50
51 struct wlt_pool {
52         struct wl_shm_pool *w_pool;
53         size_t size;
54         uint8_t *data;
55 };
56
57 enum {
58         STATE_INIT = 0,
59         STATE_RUNNING,
60         STATE_HUP,
61 };
62
63 struct data_offer {
64         unsigned long ref;
65         struct wl_data_offer *w_offer;
66         struct wlt_display *disp;
67         struct shl_array *types;
68 };
69
70 struct wlt_display {
71         unsigned long ref;
72         struct ev_eloop *eloop;
73         struct wl_display *dp;
74         struct ev_fd *dp_fd;
75         struct wl_global_listener *dp_listener;
76         struct shl_hook *listeners;
77         unsigned int state;
78         struct wl_registry *w_registry;
79
80         struct shl_dlist window_list;
81
82         struct wl_compositor *w_comp;
83         struct wl_seat *w_seat;
84         struct wl_shell *w_shell;
85         struct wl_shm *w_shm;
86
87         uint32_t last_serial;
88         uint32_t pointer_enter_serial;
89         struct wl_pointer *w_pointer;
90         struct wlt_window *pointer_focus;
91
92         uint32_t cursor_serial;
93         unsigned int current_cursor;
94         struct wl_surface *w_cursor_surface;
95         struct wl_cursor_theme *cursor_theme;
96         struct wl_cursor *cursors[WLT_CURSOR_NUM];
97
98         struct wl_keyboard *w_keyboard;
99         struct xkb_context *xkb_ctx;
100         struct xkb_keymap *xkb_keymap;
101         struct xkb_state *xkb_state;
102         struct wlt_window *keyboard_focus;
103         struct ev_timer *repeat_timer;
104         uint32_t repeat_sym;
105         uint32_t repeat_ascii;
106
107         struct wl_data_device_manager *w_manager;
108         struct wl_data_device *w_data_dev;
109         struct data_offer *drag_offer;
110         struct data_offer *selection_offer;
111 };
112
113 struct wlt_window {
114         unsigned long ref;
115         struct shl_dlist list;
116         void *data;
117         wlt_window_close_cb close_cb;
118         bool close_pending;
119         struct wlt_display *disp;
120         struct wlt_pool *pool;
121
122         struct wl_surface *w_surface;
123         struct wl_shell_surface *w_shell_surface;
124         struct wl_buffer *w_buffer;
125
126         bool buffer_attached;
127         bool skip_damage;
128         bool need_resize;
129         bool need_frame;
130         bool idle_pending;
131         unsigned int new_width;
132         unsigned int new_height;
133         unsigned int saved_width;
134         unsigned int saved_height;
135         unsigned int resize_edges;
136         bool maximized;
137         bool fullscreen;
138         struct wlt_shm_buffer buffer;
139         struct wl_callback *w_frame;
140
141         struct shl_dlist widget_list;
142 };
143
144 struct wlt_widget {
145         struct shl_dlist list;
146         struct wlt_window *wnd;
147         void *data;
148         wlt_widget_redraw_cb redraw_cb;
149         wlt_widget_destroy_cb destroy_cb;
150         wlt_widget_prepare_resize_cb prepare_resize_cb;
151         wlt_widget_resize_cb resize_cb;
152         wlt_widget_pointer_enter_cb pointer_enter_cb;
153         wlt_widget_pointer_leave_cb pointer_leave_cb;
154         wlt_widget_pointer_motion_cb pointer_motion_cb;
155         wlt_widget_pointer_button_cb pointer_button_cb;
156         wlt_widget_keyboard_cb keyboard_cb;
157 };
158
159 static int wlt_pool_new(struct wlt_pool **out, struct wlt_display *disp,
160                         size_t size)
161 {
162         static const char template[] = "/wlterm-shared-XXXXXX";
163         struct wlt_pool *pool;
164         int fd, ret;
165         const char *path;
166         char *name;
167
168         path = getenv("XDG_RUNTIME_DIR");
169         if (!path) {
170                 log_error("XDG_RUNTIME_DIR not set");
171                 return -EFAULT;
172         }
173
174         pool = malloc(sizeof(*pool));
175         if (!pool)
176                 return -ENOMEM;
177         memset(pool, 0, sizeof(*pool));
178         pool->size = size;
179
180         name = malloc(strlen(path) + sizeof(template));
181         if (!name) {
182                 ret = -ENOMEM;
183                 goto err_free;
184         }
185         strcpy(name, path);
186         strcat(name, template);
187
188         fd = mkostemp(name, O_CLOEXEC);
189         if (fd < 0) {
190                 log_error("cannot create temporary file %s via mkostemp() (%d): %m",
191                           name, errno);
192                 ret = -EFAULT;
193                 free(name);
194                 goto err_free;
195         }
196
197         ret = unlink(name);
198         if (ret)
199                 log_warning("cannot unlink temporary file %s (%d): %m",
200                             name, errno);
201         free(name);
202
203         ret = ftruncate(fd, size);
204         if (ret) {
205                 log_error("cannot truncate temporary file to length %lu (%d): %m",
206                           size, errno);
207                 ret = -EFAULT;
208                 goto err_close;
209         }
210
211         pool->data = mmap(NULL, size, PROT_READ | PROT_WRITE, MAP_SHARED,
212                           fd, 0);
213         if (pool->data == MAP_FAILED) {
214                 log_error("cannot mmap temporary file (%d): %m", errno);
215                 ret = -EFAULT;
216                 goto err_close;
217         }
218
219         pool->w_pool = wl_shm_create_pool(disp->w_shm, fd, size);
220         if (!pool->w_pool) {
221                 log_error("cannot create wayland shm pool object");
222                 ret = -EFAULT;
223                 goto err_map;
224         }
225
226         close(fd);
227         *out = pool;
228         return 0;
229
230 err_map:
231         munmap(pool->data, size);
232 err_close:
233         close(fd);
234 err_free:
235         free(pool);
236         return ret;
237 }
238
239 static void wlt_pool_free(struct wlt_pool *pool)
240 {
241         munmap(pool->data, pool->size);
242         wl_shm_pool_destroy(pool->w_pool);
243         free(pool);
244 }
245
246 /*
247  * These cursor-names are based on:
248  *   https://bugs.kde.org/attachment.cgi?id=67313
249  */
250
251 static char *(*cursors[])[] = {
252         [WLT_CURSOR_NONE] = &(char*[]) {
253                 NULL,
254         },
255         [WLT_CURSOR_TOP] = &(char*[]) {
256                 "top_side",
257                 "n-resize",
258                 NULL,
259         },
260         [WLT_CURSOR_BOTTOM] = &(char*[]) {
261                 "bottom_side",
262                 "s-resize",
263                 NULL,
264         },
265         [WLT_CURSOR_LEFT] = &(char*[]) {
266                 "left_side",
267                 "w-resize",
268                 NULL,
269         },
270         [WLT_CURSOR_RIGHT] = &(char*[]) {
271                 "right_side",
272                 "e-resize",
273                 NULL,
274         },
275         [WLT_CURSOR_TOP_LEFT] = &(char*[]) {
276                 "top_left_corner",
277                 "nw-resize",
278                 NULL,
279         },
280         [WLT_CURSOR_TOP_RIGHT] = &(char*[]) {
281                 "top_right_corner",
282                 "ne-resize",
283                 NULL,
284         },
285         [WLT_CURSOR_BOTTOM_LEFT] = &(char*[]) {
286                 "bottom_left_corner",
287                 "sw-resize",
288                 NULL,
289         },
290         [WLT_CURSOR_BOTTOM_RIGHT] = &(char*[]) {
291                 "bottom_right_corner",
292                 "se-resize",
293                 NULL,
294         },
295         [WLT_CURSOR_DRAGGING] = &(char*[]) {
296                 "grabbing",
297                 "closedhand",
298                 "208530c400c041818281048008011002",
299                 NULL,
300         },
301         [WLT_CURSOR_LEFT_PTR] = &(char*[]) {
302                 "left_ptr",
303                 "default",
304                 "top_left_arrow",
305                 "left-arrow",
306                 NULL,
307         },
308         [WLT_CURSOR_IBEAM] = &(char*[]) {
309                 "xterm",
310                 "ibeam",
311                 "text",
312                 NULL,
313         },
314 };
315
316 static int load_cursors(struct wlt_display *disp)
317 {
318         unsigned int i, j;
319         size_t size = sizeof(cursors) / sizeof(*cursors);
320         struct wl_cursor *cursor;
321
322         disp->w_cursor_surface = wl_compositor_create_surface(disp->w_comp);
323
324         disp->cursor_theme = wl_cursor_theme_load(NULL, 32, disp->w_shm);
325         if (!disp->cursor_theme) {
326                 log_warning("cannot load curdor theme");
327                 return -EFAULT;
328         }
329
330         for (i = 0; i < WLT_CURSOR_NUM; ++i) {
331                 cursor = NULL;
332                 if (i < size && cursors[i]) {
333                         for (j = 0; (*cursors[i])[j]; ++j) {
334                                 cursor = wl_cursor_theme_get_cursor(
335                                                         disp->cursor_theme,
336                                                         (*cursors[i])[j]);
337                                 if (cursor)
338                                         break;
339                         }
340                 }
341
342                 if (!cursor && i != WLT_CURSOR_NONE)
343                         log_warning("cannot load cursor for ID %d", i);
344
345                 disp->cursors[i] = cursor;
346         }
347
348         return 0;
349 }
350
351 static void unload_cursors(struct wlt_display *disp)
352 {
353         if (disp->cursor_theme)
354                 wl_cursor_theme_destroy(disp->cursor_theme);
355         if (disp->w_cursor_surface)
356                 wl_surface_destroy(disp->w_cursor_surface);
357 }
358
359 static void set_cursor(struct wlt_display *disp, unsigned int cursor)
360 {
361         bool force = false;
362         struct wl_cursor *cur;
363         struct wl_cursor_image *image;
364         struct wl_buffer *buffer;
365
366         if (cursor >= WLT_CURSOR_NUM)
367                 cursor = WLT_CURSOR_LEFT_PTR;
368
369         if (disp->pointer_enter_serial > disp->cursor_serial)
370                 force = true;
371
372         if (!force && cursor == disp->current_cursor)
373                 return;
374
375         disp->current_cursor = cursor;
376         disp->cursor_serial = disp->pointer_enter_serial;
377
378         cur = disp->cursors[cursor];
379         if (!cur) {
380                 wl_pointer_set_cursor(disp->w_pointer,
381                                       disp->pointer_enter_serial,
382                                       NULL, 0, 0);
383                 return;
384         }
385
386         image = cur->images[0];
387         buffer = wl_cursor_image_get_buffer(image);
388         if (!buffer) {
389                 log_error("cannot load buffer for cursor image");
390                 return;
391         }
392
393         wl_pointer_set_cursor(disp->w_pointer,
394                               disp->pointer_enter_serial,
395                               disp->w_cursor_surface,
396                               image->hotspot_x, image->hotspot_y);
397         wl_surface_attach(disp->w_cursor_surface, buffer, 0, 0);
398         wl_surface_damage(disp->w_cursor_surface, 0, 0,
399                           image->width, image->height);
400         wl_surface_commit(disp->w_cursor_surface);
401 }
402
403 static void dp_dispatch(struct wlt_display *disp, bool nonblock)
404 {
405         int ret;
406
407         errno = 0;
408         if (nonblock)
409                 ret = wl_display_dispatch_pending(disp->dp);
410         else
411                 ret = wl_display_dispatch(disp->dp);
412
413         if (ret == -1) {
414                 log_error("error during wayland dispatch (%d): %m", errno);
415                 return;
416         } else if (errno == EAGAIN) {
417                 ret = ev_fd_update(disp->dp_fd, EV_READABLE | EV_WRITEABLE);
418                 if (ret)
419                         log_warning("cannot update wayland-fd event-polling modes (%d)",
420                                     ret);
421         } else {
422                 ret = ev_fd_update(disp->dp_fd, EV_READABLE);
423                 if (ret)
424                         log_warning("cannot update wayland-fd event-polling modes (%d)",
425                                     ret);
426         }
427 }
428
429 static void dp_pre_event(struct ev_eloop *eloop, void *unused, void *data)
430 {
431         struct wlt_display *disp = data;
432
433         dp_dispatch(disp, true);
434 }
435
436 static void dp_event(struct ev_fd *fd, int mask, void *data)
437 {
438         struct wlt_display *disp = data;
439
440         if (mask & (EV_HUP | EV_ERR)) {
441                 log_warning("HUP/ERR on wayland socket");
442                 disp->state = STATE_HUP;
443                 shl_hook_call(disp->listeners, disp, (void*)WLT_DISPLAY_HUP);
444                 ev_eloop_rm_fd(disp->dp_fd);
445                 disp->dp_fd = NULL;
446                 return;
447         }
448
449         if (mask & EV_READABLE)
450                 dp_dispatch(disp, false);
451         else
452                 dp_dispatch(disp, true);
453 }
454
455 static void pointer_enter(void *data, struct wl_pointer *w_pointer,
456                           uint32_t serial, struct wl_surface *w_surface,
457                           wl_fixed_t x, wl_fixed_t y)
458 {
459         struct wlt_display *disp = data;
460         struct shl_dlist *iter;
461         struct wlt_window *wnd;
462         struct wlt_widget *widget;
463
464         if (!w_surface)
465                 return;
466
467         shl_dlist_for_each(iter, &disp->window_list) {
468                 wnd = shl_dlist_entry(iter, struct wlt_window, list);
469                 if (wnd->w_surface == w_surface)
470                         break;
471         }
472
473         if (iter == &disp->window_list) {
474                 log_debug("unknown surface");
475                 return;
476         }
477
478         disp->pointer_enter_serial = serial;
479         disp->last_serial = serial;
480         disp->pointer_focus = wnd;
481         shl_dlist_for_each(iter, &wnd->widget_list) {
482                 widget = shl_dlist_entry(iter, struct wlt_widget, list);
483                 if (widget->pointer_enter_cb)
484                         widget->pointer_enter_cb(widget, x >> 8, y >> 8,
485                                                  widget->data);
486         }
487 }
488
489 static void pointer_leave(void *data, struct wl_pointer *w_pointer,
490                           uint32_t serial, struct wl_surface *w_surface)
491 {
492         struct wlt_display *disp = data;
493         struct shl_dlist *iter;
494         struct wlt_window *wnd;
495         struct wlt_widget *widget;
496
497         wnd = disp->pointer_focus;
498         disp->pointer_focus = NULL;
499         disp->last_serial = serial;
500
501         if (!wnd)
502                 return;
503
504         shl_dlist_for_each(iter, &wnd->widget_list) {
505                 widget = shl_dlist_entry(iter, struct wlt_widget, list);
506                 if (widget->pointer_leave_cb)
507                         widget->pointer_leave_cb(widget, widget->data);
508         }
509 }
510
511 static void pointer_motion(void *data, struct wl_pointer *w_pointer,
512                            uint32_t time, wl_fixed_t x, wl_fixed_t y)
513 {
514         struct wlt_display *disp = data;
515         struct shl_dlist *iter;
516         struct wlt_window *wnd;
517         struct wlt_widget *widget;
518
519         wnd = disp->pointer_focus;
520         if (!wnd)
521                 return;
522
523         shl_dlist_for_each(iter, &wnd->widget_list) {
524                 widget = shl_dlist_entry(iter, struct wlt_widget, list);
525                 if (widget->pointer_motion_cb)
526                         widget->pointer_motion_cb(widget, x >> 8, y >> 8,
527                                                   widget->data);
528         }
529 }
530
531 static void pointer_button(void *data, struct wl_pointer *w_pointer,
532                            uint32_t serial, uint32_t time, uint32_t button,
533                            uint32_t state)
534 {
535         struct wlt_display *disp = data;
536         struct shl_dlist *iter;
537         struct wlt_window *wnd;
538         struct wlt_widget *widget;
539
540         wnd = disp->pointer_focus;
541         disp->last_serial = serial;
542
543         if (!wnd)
544                 return;
545
546         shl_dlist_for_each(iter, &wnd->widget_list) {
547                 widget = shl_dlist_entry(iter, struct wlt_widget, list);
548                 if (widget->pointer_button_cb)
549                         widget->pointer_button_cb(widget, button, state,
550                                                   widget->data);
551         }
552 }
553
554 static void pointer_axis(void *data, struct wl_pointer *w_pointer,
555                          uint32_t time, uint32_t axis, wl_fixed_t value)
556 {
557 }
558
559 static const struct wl_pointer_listener pointer_listener = {
560         .enter = pointer_enter,
561         .leave = pointer_leave,
562         .motion = pointer_motion,
563         .button = pointer_button,
564         .axis = pointer_axis,
565 };
566
567 static void keyboard_keymap(void *data, struct wl_keyboard *keyboard,
568                             uint32_t format, int fd, uint32_t size)
569 {
570         struct wlt_display *disp = data;
571         char *map;
572
573         if (format != WL_KEYBOARD_KEYMAP_FORMAT_XKB_V1) {
574                 log_error("invalid keyboard format");
575                 close(fd);
576                 return;
577         }
578
579         map = mmap(NULL, size, PROT_READ, MAP_SHARED, fd, 0);
580         if (map == MAP_FAILED) {
581                 log_error("cannot mmap keyboard keymap");
582                 close(fd);
583                 return;
584         }
585
586         disp->xkb_keymap = xkb_keymap_new_from_string(disp->xkb_ctx, map,
587                                                       XKB_KEYMAP_FORMAT_TEXT_V1,
588                                                       0);
589         munmap(map, size);
590         close(fd);
591
592         if (!disp->xkb_keymap) {
593                 log_error("cannot create xkb keymap");
594                 return;
595         }
596
597         disp->xkb_state = xkb_state_new(disp->xkb_keymap);
598         if (!disp->xkb_state) {
599                 log_error("cannot create XKB state object");
600                 xkb_keymap_unref(disp->xkb_keymap);
601                 disp->xkb_keymap = NULL;
602                 return;
603         }
604 }
605
606 static void keyboard_enter(void *data, struct wl_keyboard *keyboard,
607                            uint32_t serial, struct wl_surface *w_surface,
608                            struct wl_array *keys)
609 {
610         struct wlt_display *disp = data;
611         struct shl_dlist *iter;
612         struct wlt_window *wnd;
613
614         disp->last_serial = serial;
615         if (!disp->xkb_state)
616                 return;
617
618         if (!w_surface)
619                 return;
620
621         shl_dlist_for_each(iter, &disp->window_list) {
622                 wnd = shl_dlist_entry(iter, struct wlt_window, list);
623                 if (wnd->w_surface == w_surface)
624                         break;
625         }
626
627         if (iter == &disp->window_list)
628                 return;
629
630         disp->keyboard_focus = wnd;
631 }
632
633 static void keyboard_leave(void *data, struct wl_keyboard *keyboard,
634                            uint32_t serial, struct wl_surface *surface)
635 {
636         struct wlt_display *disp = data;
637
638         disp->last_serial = serial;
639         disp->keyboard_focus = NULL;
640         ev_timer_update(disp->repeat_timer, NULL);
641 }
642
643 static void keyboard_key(void *data, struct wl_keyboard *keyboard,
644                          uint32_t serial, uint32_t time, uint32_t key,
645                          uint32_t state_w)
646 {
647         struct wlt_display *disp = data;
648         struct wlt_window *wnd = disp->keyboard_focus;
649         uint32_t code, num_syms, ascii;
650         unsigned int mask;
651         enum wl_keyboard_key_state state = state_w;
652         const xkb_keysym_t *syms;
653         xkb_keysym_t sym;
654         struct shl_dlist *iter;
655         struct wlt_widget *widget;
656         struct itimerspec spec;
657         bool handled;
658
659         disp->last_serial = serial;
660         if (!disp->xkb_state)
661                 return;
662
663         code = key + 8;
664         if (!wnd)
665                 return;
666
667         mask = shl_get_xkb_mods(disp->xkb_state);
668         num_syms = xkb_state_key_get_syms(disp->xkb_state, code, &syms);
669         ascii = shl_get_ascii(disp->xkb_state, code, syms, num_syms);
670         sym = XKB_KEY_NoSymbol;
671         if (num_syms == 1)
672                 sym = syms[0];
673
674         handled = false;
675         shl_dlist_for_each(iter, &wnd->widget_list) {
676                 widget = shl_dlist_entry(iter, struct wlt_widget, list);
677                 if (widget->keyboard_cb) {
678                         if (widget->keyboard_cb(widget, mask, sym, ascii, state,
679                                                 handled, widget->data))
680                                 handled = true;
681                 }
682         }
683
684         if (state == WL_KEYBOARD_KEY_STATE_RELEASED) {
685                 ev_timer_update(disp->repeat_timer, NULL);
686         } else if (state == WL_KEYBOARD_KEY_STATE_PRESSED) {
687                 disp->repeat_sym = sym;
688                 disp->repeat_ascii = ascii;
689                 spec.it_interval.tv_sec = 0;
690                 spec.it_interval.tv_nsec = wlt_conf.xkb_repeat_rate * 1000000;
691                 spec.it_value.tv_sec = 0;
692                 spec.it_value.tv_nsec = wlt_conf.xkb_repeat_delay * 1000000;
693                 ev_timer_update(disp->repeat_timer, &spec);
694         }
695 }
696
697 static void repeat_event(struct ev_timer *timer, uint64_t num, void *data)
698 {
699         struct wlt_display *disp = data;
700         struct wlt_window *wnd = disp->keyboard_focus;
701         struct wlt_widget *widget;
702         struct shl_dlist *iter;
703         unsigned int mask;
704         bool handled;
705
706         if (!wnd)
707                 return;
708
709         mask = shl_get_xkb_mods(disp->xkb_state);
710         handled = false;
711         shl_dlist_for_each(iter, &wnd->widget_list) {
712                 widget = shl_dlist_entry(iter, struct wlt_widget, list);
713                 if (widget->keyboard_cb) {
714                         if (widget->keyboard_cb(widget, mask, disp->repeat_sym,
715                                                 disp->repeat_ascii,
716                                                 WL_KEYBOARD_KEY_STATE_PRESSED,
717                                                 handled, widget->data))
718                                 handled = true;
719                 }
720         }
721 }
722
723 static void keyboard_modifiers(void *data, struct wl_keyboard *keyboard,
724                                uint32_t serial, uint32_t mods_depressed,
725                                uint32_t mods_latched, uint32_t mods_locked,
726                                uint32_t group)
727 {
728         struct wlt_display *disp = data;
729
730         disp->last_serial = serial;
731         if (!disp->xkb_state)
732                 return;
733
734         xkb_state_update_mask(disp->xkb_state, mods_depressed, mods_latched,
735                               mods_locked, 0, 0, group);
736 }
737
738 static const struct wl_keyboard_listener keyboard_listener = {
739         .keymap = keyboard_keymap,
740         .enter = keyboard_enter,
741         .leave = keyboard_leave,
742         .key = keyboard_key,
743         .modifiers = keyboard_modifiers,
744 };
745
746 static void data_offer_unref(struct data_offer *offer)
747 {
748         unsigned int i, len;
749
750         if (!offer || !offer->ref || --offer->ref)
751                 return;
752
753         len = shl_array_get_length(offer->types);
754         for (i = 0; i < len; ++i)
755                 free(*SHL_ARRAY_AT(offer->types, char*, i));
756         shl_array_free(offer->types);
757         wl_data_offer_destroy(offer->w_offer);
758         free(offer);
759 }
760
761 static void data_offer_offer(void *data, struct wl_data_offer *w_offer,
762                              const char *type)
763 {
764         char *tmp;
765         int ret;
766         struct data_offer *offer = wl_data_offer_get_user_data(w_offer);
767
768         tmp = strdup(type);
769         if (!tmp)
770                 return;
771
772         ret = shl_array_push(offer->types, &tmp);
773         if (ret) {
774                 free(tmp);
775                 return;
776         }
777 }
778
779 static const struct wl_data_offer_listener data_offer_listener = {
780         .offer = data_offer_offer,
781 };
782
783 static void data_dev_data_offer(void *data, struct wl_data_device *w_dev,
784                                 struct wl_data_offer *w_offer)
785 {
786         struct data_offer *offer;
787         struct wlt_display *disp = data;
788         int ret;
789
790         offer = malloc(sizeof(*offer));
791         if (!offer) {
792                 wl_data_offer_destroy(w_offer);
793                 return;
794         }
795         memset(offer, 0, sizeof(*offer));
796         offer->ref = 1;
797         offer->w_offer = w_offer;
798         offer->disp = disp;
799
800         ret = shl_array_new(&offer->types, sizeof(char*), 4);
801         if (ret) {
802                 wl_data_offer_destroy(w_offer);
803                 free(offer);
804                 return;
805         }
806
807         wl_data_offer_add_listener(w_offer, &data_offer_listener, offer);
808         wl_data_offer_set_user_data(w_offer, offer);
809 }
810
811 static void data_dev_enter(void *data, struct wl_data_device *w_dev,
812                            uint32_t serial, struct wl_surface *w_surface,
813                            wl_fixed_t x, wl_fixed_t y,
814                            struct wl_data_offer *w_offer)
815 {
816         struct wlt_display *disp = data;
817
818         if (disp->drag_offer) {
819                 data_offer_unref(disp->drag_offer);
820                 disp->drag_offer = NULL;
821         }
822
823         if (!w_offer)
824                 return;
825
826         disp->drag_offer = wl_data_offer_get_user_data(w_offer);
827 }
828
829 static void data_dev_leave(void *data, struct wl_data_device *w_dev)
830 {
831         struct wlt_display *disp = data;
832
833         if (disp->drag_offer) {
834                 data_offer_unref(disp->drag_offer);
835                 disp->drag_offer = NULL;
836         }
837 }
838
839 static void data_dev_motion(void *data, struct wl_data_device *w_dev,
840                             uint32_t time, wl_fixed_t x, wl_fixed_t y)
841 {
842 }
843
844 static void data_dev_drop(void *data, struct wl_data_device *w_dev)
845 {
846 }
847
848 static void data_dev_selection(void *data, struct wl_data_device *w_dev,
849                                struct wl_data_offer *w_offer)
850 {
851         struct wlt_display *disp = data;
852
853         if (disp->selection_offer) {
854                 data_offer_unref(disp->selection_offer);
855                 disp->selection_offer = NULL;
856         }
857
858         if (!w_offer)
859                 return;
860
861         disp->selection_offer = wl_data_offer_get_user_data(w_offer);
862 }
863
864 int wlt_display_get_selection_to_fd(struct wlt_display *disp, const char *mime,
865                                     int output_fd)
866 {
867         unsigned int i, num;
868         struct data_offer *offer;
869
870         if (!disp || !mime)
871                 return -EINVAL;
872         if (!disp->selection_offer)
873                 return -ENOENT;
874
875         offer = disp->selection_offer;
876         num = shl_array_get_length(offer->types);
877         for (i = 0; i < num; ++i) {
878                 if (!strcmp(mime, *SHL_ARRAY_AT(offer->types, char*, i)))
879                         break;
880         }
881
882         if (i == num)
883                 return -EAGAIN;
884
885         wl_data_offer_receive(offer->w_offer, mime, output_fd);
886         return 0;
887 }
888
889 int wlt_display_get_selection_fd(struct wlt_display *disp, const char *mime)
890 {
891         int p[2], ret;
892
893         if (pipe2(p, O_CLOEXEC | O_NONBLOCK))
894                 return -EFAULT;
895
896         ret = wlt_display_get_selection_to_fd(disp, mime, p[1]);
897         close(p[1]);
898
899         if (ret) {
900                 close(p[0]);
901                 return ret;
902         }
903
904         return p[0];
905 }
906
907 int wlt_display_new_data_source(struct wlt_display *disp,
908                                 struct wl_data_source **out)
909 {
910         struct wl_data_source *src;
911
912         if (!disp)
913                 return -EINVAL;
914
915         src = wl_data_device_manager_create_data_source(disp->w_manager);
916         if (!src)
917                 return -EFAULT;
918
919         *out = src;
920         return 0;
921 }
922
923 void wlt_display_set_selection(struct wlt_display *disp,
924                                struct wl_data_source *selection)
925 {
926         if (!disp)
927                 return;
928
929         wl_data_device_set_selection(disp->w_data_dev, selection,
930                                      disp->last_serial);
931 }
932
933 static const struct wl_data_device_listener data_dev_listener = {
934         .data_offer = data_dev_data_offer,
935         .enter = data_dev_enter,
936         .leave = data_dev_leave,
937         .motion = data_dev_motion,
938         .drop = data_dev_drop,
939         .selection = data_dev_selection,
940 };
941
942 static void check_ready(struct wlt_display *disp)
943 {
944         if (disp->state > STATE_INIT)
945                 return;
946
947         if (disp->w_comp &&
948             disp->w_seat &&
949             disp->w_shell &&
950             disp->w_shm &&
951             disp->w_pointer &&
952             disp->w_keyboard &&
953             disp->w_manager) {
954                 log_debug("wayland display initialized");
955                 load_cursors(disp);
956
957                 disp->w_data_dev = wl_data_device_manager_get_data_device(
958                                                 disp->w_manager, disp->w_seat);
959                 wl_data_device_add_listener(disp->w_data_dev,
960                                             &data_dev_listener,
961                                             disp);
962
963                 disp->state = STATE_RUNNING;
964                 shl_hook_call(disp->listeners, disp, (void*)WLT_DISPLAY_READY);
965         }
966 }
967
968 static void seat_capabilities(void *data, struct wl_seat *seat,
969                               enum wl_seat_capability caps)
970 {
971         struct wlt_display *disp = data;
972
973         if (caps & WL_SEAT_CAPABILITY_POINTER && !disp->w_pointer) {
974                 disp->w_pointer = wl_seat_get_pointer(seat);
975                 wl_pointer_add_listener(disp->w_pointer, &pointer_listener,
976                                         disp);
977         }
978
979         if (caps & WL_SEAT_CAPABILITY_KEYBOARD && !disp->w_keyboard) {
980                 disp->w_keyboard = wl_seat_get_keyboard(seat);
981                 wl_keyboard_add_listener(disp->w_keyboard, &keyboard_listener,
982                                          disp);
983         }
984
985         check_ready(disp);
986 }
987
988 static const struct wl_seat_listener seat_listener = {
989         .capabilities = seat_capabilities,
990 };
991
992 static void dp_global(void *data, struct wl_registry *registry, uint32_t id,
993                       const char *iface, uint32_t version)
994 {
995         struct wlt_display *disp = data;
996
997         if (!strcmp(iface, "wl_display")) {
998                 log_debug("new wl_display global");
999                 return;
1000         } else if (!strcmp(iface, "wl_compositor")) {
1001                 if (disp->w_comp) {
1002                         log_error("global wl_compositor advertised twice");
1003                         return;
1004                 }
1005                 disp->w_comp = wl_registry_bind(disp->w_registry,
1006                                                 id,
1007                                                 &wl_compositor_interface,
1008                                                 1);
1009                 if (!disp->w_comp) {
1010                         log_error("cannot bind wl_compositor object");
1011                         return;
1012                 }
1013         } else if (!strcmp(iface, "wl_seat")) {
1014                 if (disp->w_seat) {
1015                         log_error("global wl_seat advertised twice");
1016                         return;
1017                 }
1018                 disp->w_seat = wl_registry_bind(disp->w_registry,
1019                                                 id,
1020                                                 &wl_seat_interface,
1021                                                 1);
1022                 if (!disp->w_seat) {
1023                         log_error("cannot bind wl_seat object");
1024                         return;
1025                 }
1026
1027                 wl_seat_add_listener(disp->w_seat, &seat_listener, disp);
1028         } else if (!strcmp(iface, "wl_shell")) {
1029                 if (disp->w_shell) {
1030                         log_error("global wl_shell advertised twice");
1031                         return;
1032                 }
1033                 disp->w_shell = wl_registry_bind(disp->w_registry,
1034                                                  id,
1035                                                  &wl_shell_interface,
1036                                                  1);
1037                 if (!disp->w_shell) {
1038                         log_error("cannot bind wl_shell object");
1039                         return;
1040                 }
1041         } else if (!strcmp(iface, "wl_shm")) {
1042                 if (disp->w_shm) {
1043                         log_error("global wl_shm advertised twice");
1044                         return;
1045                 }
1046                 disp->w_shm = wl_registry_bind(disp->w_registry,
1047                                                id,
1048                                                &wl_shm_interface,
1049                                                1);
1050                 if (!disp->w_shm) {
1051                         log_error("cannot bind wl_shm object");
1052                         return;
1053                 }
1054         } else if (!strcmp(iface, "wl_data_device_manager")) {
1055                 if (disp->w_manager) {
1056                         log_error("global wl_data_device_manager advertised twice");
1057                         return;
1058                 }
1059                 disp->w_manager = wl_registry_bind(disp->w_registry,
1060                                                    id,
1061                                                    &wl_data_device_manager_interface,
1062                                                    1);
1063                 if (!disp->w_manager) {
1064                         log_error("cannot bind wl_data_device_manager_object");
1065                         return;
1066                 }
1067         } else {
1068                 log_debug("ignoring new unknown global %s", iface);
1069                 return;
1070         }
1071
1072         log_debug("new global %s", iface);
1073
1074         check_ready(disp);
1075 }
1076
1077 static const struct wl_registry_listener registry_listener = {
1078         .global = dp_global,
1079         /* TODO: handle .global_remove */
1080 };
1081
1082 int wlt_display_new(struct wlt_display **out,
1083                     struct ev_eloop *eloop)
1084 {
1085         struct wlt_display *disp;
1086         int ret, fd;
1087
1088         if (!out || !eloop)
1089                 return -EINVAL;
1090
1091         log_debug("creating new wlt-display");
1092
1093         disp = malloc(sizeof(*disp));
1094         if (!disp)
1095                 return -ENOMEM;
1096         memset(disp, 0, sizeof(*disp));
1097         disp->ref = 1;
1098         disp->eloop = eloop;
1099         shl_dlist_init(&disp->window_list);
1100
1101         ret = shl_hook_new(&disp->listeners);
1102         if (ret) {
1103                 log_error("cannot create listeners hook");
1104                 goto err_free;
1105         }
1106
1107         disp->dp = wl_display_connect(NULL);
1108         if (!disp->dp) {
1109                 log_error("cannot connect to wayland socket (%d): %m", errno);
1110                 ret = -EFAULT;
1111                 goto err_listener;
1112         }
1113
1114         fd = wl_display_get_fd(disp->dp);
1115         if (fd < 0) {
1116                 log_error("cannot retrieve wayland FD");
1117                 ret = -EFAULT;
1118                 goto err_dp;
1119         }
1120
1121         /* TODO: nonblocking wl_display doesn't work, yet. fix that upstream */
1122         /*
1123         int set;
1124         set = fcntl(fd, F_GETFL);
1125         if (set < 0) {
1126                 log_error("cannot retrieve file flags for wayland FD (%d): %m",
1127                           errno);
1128                 ret = -EFAULT;
1129                 goto err_dp;
1130         }
1131
1132         set |= O_NONBLOCK;
1133         ret = fcntl(fd, F_SETFL, set);
1134         if (ret) {
1135                 log_error("cannot set file flags for wayland FD (%d): %m",
1136                           errno);
1137                 ret = -EFAULT;
1138                 goto err_dp;
1139         }*/
1140
1141         ret = ev_eloop_new_fd(eloop,
1142                               &disp->dp_fd,
1143                               fd,
1144                               EV_READABLE,
1145                               dp_event,
1146                               disp);
1147         if (ret) {
1148                 log_error("cannot create event-fd for wayland display (%d)",
1149                           ret);
1150                 goto err_dp;
1151         }
1152
1153         ret = ev_eloop_register_pre_cb(eloop, dp_pre_event, disp);
1154         if (ret) {
1155                 log_error("cannot register pre-cb (%d)", ret);
1156                 goto err_dp_fd;
1157         }
1158
1159         ret = ev_eloop_new_timer(eloop, &disp->repeat_timer, NULL,
1160                                  repeat_event, disp);
1161         if (ret) {
1162                 log_error("cannot create repeat-timer");
1163                 goto err_pre_cb;
1164         }
1165
1166         disp->w_registry = wl_display_get_registry(disp->dp);
1167         if (!disp->w_registry) {
1168                 log_error("cannot get wayland registry object");
1169                 goto err_timer;
1170         }
1171
1172         wl_registry_add_listener(disp->w_registry, &registry_listener, disp);
1173
1174         disp->xkb_ctx = xkb_context_new(0);
1175         if (!disp->xkb_ctx) {
1176                 log_error("cannot create XKB context");
1177                 goto err_timer;
1178         }
1179
1180         log_debug("wlt-display waiting for globals...");
1181
1182         ev_eloop_ref(disp->eloop);
1183         *out = disp;
1184         return 0;
1185
1186 err_timer:
1187         ev_eloop_rm_timer(disp->repeat_timer);
1188 err_pre_cb:
1189         ev_eloop_unregister_pre_cb(disp->eloop, dp_pre_event, disp);
1190 err_dp_fd:
1191         ev_eloop_rm_fd(disp->dp_fd);
1192 err_dp:
1193         wl_display_disconnect(disp->dp);
1194 err_listener:
1195         shl_hook_free(disp->listeners);
1196 err_free:
1197         free(disp);
1198         return ret;
1199 }
1200
1201 void wlt_display_ref(struct wlt_display *disp)
1202 {
1203         if (!disp || !disp->ref)
1204                 return;
1205
1206         ++disp->ref;
1207 }
1208
1209 void wlt_display_unref(struct wlt_display *disp)
1210 {
1211         if (!disp || !disp->ref || --disp->ref)
1212                 return;
1213
1214         unload_cursors(disp);
1215         wl_display_flush(disp->dp);
1216         wl_display_disconnect(disp->dp);
1217         ev_eloop_rm_timer(disp->repeat_timer);
1218         ev_eloop_unregister_pre_cb(disp->eloop, dp_pre_event, disp);
1219         ev_eloop_rm_fd(disp->dp_fd);
1220         xkb_context_unref(disp->xkb_ctx);
1221         shl_hook_free(disp->listeners);
1222         ev_eloop_unref(disp->eloop);
1223         free(disp);
1224 }
1225
1226 int wlt_display_register_cb(struct wlt_display *disp,
1227                             wlt_display_cb cb, void *data)
1228 {
1229         if (!disp)
1230                 return -EINVAL;
1231
1232         return shl_hook_add_cast(disp->listeners, cb, data, false);
1233 }
1234
1235 void wlt_display_unregister_cb(struct wlt_display *disp,
1236                                wlt_display_cb cb, void *data)
1237 {
1238         if (!disp)
1239                 return;
1240
1241         shl_hook_rm_cast(disp->listeners, cb, data);
1242 }
1243
1244 static void shell_surface_ping(void *data, struct wl_shell_surface *s,
1245                                uint32_t serial)
1246 {
1247         wl_shell_surface_pong(s, serial);
1248 }
1249
1250 static void wlt_window_do_redraw(struct wlt_window *wnd,
1251                                  unsigned int oldw,
1252                                  unsigned int oldh)
1253 {
1254         struct shl_dlist *iter;
1255         struct wlt_widget *widget;
1256         unsigned int x, y, flags;
1257         struct wlt_rect alloc;
1258
1259         flags = 0;
1260         if (wnd->maximized)
1261                 flags |= WLT_WINDOW_MAXIMIZED;
1262         if (wnd->fullscreen)
1263                 flags |= WLT_WINDOW_FULLSCREEN;
1264
1265         alloc.x = 0;
1266         alloc.y = 0;
1267         alloc.width = wnd->buffer.width;
1268         alloc.height = wnd->buffer.height;
1269         shl_dlist_for_each(iter, &wnd->widget_list) {
1270                 widget = shl_dlist_entry(iter, struct wlt_widget, list);
1271                 if (widget->resize_cb)
1272                         widget->resize_cb(widget, flags, &alloc, widget->data);
1273         }
1274
1275         memset(wnd->buffer.data, 0,
1276                wnd->buffer.stride * wnd->buffer.height);
1277
1278         wnd->skip_damage = true;
1279         shl_dlist_for_each(iter, &wnd->widget_list) {
1280                 widget = shl_dlist_entry(iter, struct wlt_widget, list);
1281                 if (widget->redraw_cb)
1282                         widget->redraw_cb(widget, flags, widget->data);
1283         }
1284         wnd->skip_damage = false;
1285
1286         if (!wnd->buffer_attached) {
1287                 wnd->buffer_attached = true;
1288                 if (wnd->resize_edges & WL_SHELL_SURFACE_RESIZE_LEFT)
1289                         x = (int)oldw - wnd->buffer.width;
1290                 else
1291                         x = 0;
1292                 if (wnd->resize_edges & WL_SHELL_SURFACE_RESIZE_TOP)
1293                         y = (int)oldh - wnd->buffer.height;
1294                 else
1295                         y = 0;
1296                 wl_surface_attach(wnd->w_surface, wnd->w_buffer, x, y);
1297                 wnd->resize_edges = 0;
1298         }
1299
1300         wl_surface_damage(wnd->w_surface, 0, 0, wnd->buffer.width,
1301                           wnd->buffer.height);
1302         wl_surface_commit(wnd->w_surface);
1303 }
1304
1305 static int resize_window(struct wlt_window *wnd, unsigned int width,
1306                          unsigned int height)
1307 {
1308         struct shl_dlist *iter;
1309         struct wlt_widget *widget;
1310         struct wl_buffer *old_buffer = NULL;
1311         struct wlt_pool *old_pool = NULL;
1312         size_t nsize;
1313         int ret;
1314         unsigned int oldw, oldh, neww, newh, minw, minh, flags;
1315
1316         if (!width)
1317                 width = wnd->buffer.width;
1318         if (!height)
1319                 height = wnd->buffer.height;
1320
1321         if (!width || !height)
1322                 return -EINVAL;
1323
1324         flags = 0;
1325         if (wnd->maximized)
1326                 flags |= WLT_WINDOW_MAXIMIZED;
1327         if (wnd->fullscreen)
1328                 flags |= WLT_WINDOW_FULLSCREEN;
1329
1330         neww = 0;
1331         newh = 0;
1332         minw = 0;
1333         minh = 0;
1334         shl_dlist_for_each(iter, &wnd->widget_list) {
1335                 widget = shl_dlist_entry(iter, struct wlt_widget, list);
1336                 if (widget->prepare_resize_cb)
1337                         widget->prepare_resize_cb(widget,
1338                                                   flags,
1339                                                   width, height,
1340                                                   &minw, &minh,
1341                                                   &neww, &newh,
1342                                                   widget->data);
1343         }
1344
1345         if (neww)
1346                 width = neww;
1347         if (newh)
1348                 height = newh;
1349
1350         if (width == wnd->buffer.width &&
1351             height == wnd->buffer.height) {
1352                 wlt_window_do_redraw(wnd, width, height);
1353                 return 0;
1354         }
1355
1356         oldw = wnd->buffer.width;
1357         oldh = wnd->buffer.height;
1358
1359         nsize = width * height * 4;
1360         if (wnd->pool && wnd->pool->size >= nsize) {
1361                 old_buffer = wnd->w_buffer;
1362                 wnd->w_buffer = wl_shm_pool_create_buffer(wnd->pool->w_pool,
1363                                                 0, width, height,
1364                                                 width * 4,
1365                                                 WL_SHM_FORMAT_ARGB8888);
1366                 if (!wnd->w_buffer) {
1367                         log_error("cannot create wayland shm buffer");
1368                         wnd->w_buffer = old_buffer;
1369                         return -ENOMEM;
1370                 }
1371         } else {
1372                 old_pool = wnd->pool;
1373                 ret = wlt_pool_new(&wnd->pool, wnd->disp, nsize);
1374                 if (ret) {
1375                         log_error("cannot create memory pool");
1376                         wnd->pool = old_pool;
1377                         return ret;
1378                 }
1379
1380                 old_buffer = wnd->w_buffer;
1381                 wnd->w_buffer = wl_shm_pool_create_buffer(wnd->pool->w_pool,
1382                                                 0, width, height,
1383                                                 width * 4,
1384                                                 WL_SHM_FORMAT_ARGB8888);
1385                 if (!wnd->w_buffer) {
1386                         log_error("cannot create wayland shm buffer");
1387                         wnd->w_buffer = old_buffer;
1388                         wlt_pool_free(wnd->pool);
1389                         wnd->pool = old_pool;
1390                         return -ENOMEM;
1391                 }
1392         }
1393
1394         wnd->buffer.data = wnd->pool->data;
1395         wnd->buffer.width = width;
1396         wnd->buffer.height = height;
1397         wnd->buffer.stride = width * 4;
1398         wnd->buffer_attached = false;
1399
1400         wlt_window_do_redraw(wnd, oldw, oldh);
1401
1402         if (old_buffer)
1403                 wl_buffer_destroy(old_buffer);
1404         if (old_pool)
1405                 wlt_pool_free(old_pool);
1406
1407         return 0;
1408 }
1409
1410 static void frame_callback(void *data, struct wl_callback *w_callback,
1411                            uint32_t time);
1412 static void idle_frame(struct ev_eloop *eloop, void *unused, void *data);
1413
1414 static const struct wl_callback_listener frame_callback_listener = {
1415         .done = frame_callback,
1416 };
1417
1418 static void do_frame(struct wlt_window *wnd)
1419 {
1420         ev_eloop_unregister_idle_cb(wnd->disp->eloop, idle_frame, wnd,
1421                                     EV_NORMAL);
1422
1423         if (wnd->need_resize) {
1424                 wnd->need_frame = true;
1425                 wnd->need_resize = false;
1426                 wnd->w_frame = wl_surface_frame(wnd->w_surface);
1427                 wl_callback_add_listener(wnd->w_frame,
1428                                          &frame_callback_listener, wnd);
1429                 resize_window(wnd, wnd->new_width, wnd->new_height);
1430         } else {
1431                 wnd->need_frame = true;
1432                 wnd->w_frame = wl_surface_frame(wnd->w_surface);
1433                 wl_callback_add_listener(wnd->w_frame,
1434                                          &frame_callback_listener, wnd);
1435                 wlt_window_do_redraw(wnd, wnd->buffer.width,
1436                                      wnd->buffer.height);
1437         }
1438 }
1439
1440 static void schedule_frame(struct wlt_window *wnd);
1441
1442 static void frame_callback(void *data, struct wl_callback *w_callback,
1443                            uint32_t time)
1444 {
1445         struct wlt_window *wnd = data;
1446
1447         wl_callback_destroy(w_callback);
1448         wnd->w_frame = NULL;
1449
1450         wnd->idle_pending = false;
1451         if (wnd->need_frame)
1452                 schedule_frame(wnd);
1453 }
1454
1455 static void idle_frame(struct ev_eloop *eloop, void *unused, void *data)
1456 {
1457         struct wlt_window *wnd = data;
1458
1459         do_frame(wnd);
1460 }
1461
1462 /*
1463  * Buffer Handling and Frame Scheduling
1464  * We use wl_shm for buffer allocation. This means, we have a single buffer
1465  * client side and the server loads it into its backbuffer for rendering. If the
1466  * server does not do this, we are screwed anyway, but that's on behalf of the
1467  * server, so we don't care.
1468  *
1469  * However, this means, when we create a buffer, we need to notify the
1470  * compositor and then wait until the compositor has created its back-buffer,
1471  * before we continue using this buffer. We can use the "frame" callback to get
1472  * notified about this.
1473  *
1474  * The logic we have is:
1475  * You set the boolean flags what action is needed in wlt_window and then call
1476  * "schedule_frame()". If we didn't already do any buffer operations in this
1477  * frame, then this function schedules an idle-callback which then performs
1478  * the requested functions (flags in wlt_window). Afterwards, it sets a marker
1479  * that this frame was already used and schedules a frame-callback.
1480  * If during this time another call to schedule_frame() is made, we do nothing
1481  * but wait for the frame-callback. It will then directly perform all the
1482  * requested functions and reschedule a frame-callback.
1483  * If nothing was schedule for one frame, we have no more interest in
1484  * frame-callbacks and thus we set "need_frame" to false again and don't
1485  * schedule any more frame-callbacks.
1486  */
1487 static void schedule_frame(struct wlt_window *wnd)
1488 {
1489         int ret;
1490
1491         if (!wnd)
1492                 return;
1493
1494         wnd->need_frame = true;
1495
1496         if (wnd->idle_pending)
1497                 return;
1498
1499         ret = ev_eloop_register_idle_cb(wnd->disp->eloop, idle_frame, wnd,
1500                                         EV_NORMAL);
1501         if (ret)
1502                 log_error("cannot schedule idle callback: %d", ret);
1503         else
1504                 wnd->idle_pending = true;
1505 }
1506
1507 static void shell_surface_configure(void *data, struct wl_shell_surface *s,
1508                                     uint32_t edges, int32_t width,
1509                                     int32_t height)
1510 {
1511         struct wlt_window *wnd = data;
1512
1513         if (width <= 0)
1514                 width = 1;
1515         if (height <= 0)
1516                 height = 1;
1517
1518         wnd->resize_edges = edges;
1519         wlt_window_set_size(wnd, width, height);
1520 }
1521
1522 static void shell_surface_popup_done(void *data, struct wl_shell_surface *s)
1523 {
1524 }
1525
1526 static const struct wl_shell_surface_listener shell_surface_listener = {
1527         .ping = shell_surface_ping,
1528         .configure = shell_surface_configure,
1529         .popup_done = shell_surface_popup_done,
1530 };
1531
1532 static void close_window(struct ev_eloop *eloop, void *unused, void *data)
1533 {
1534         struct wlt_window *wnd = data;
1535
1536         ev_eloop_unregister_idle_cb(eloop, close_window, wnd, EV_NORMAL);
1537         wnd->close_pending = false;
1538
1539         if (wnd->close_cb)
1540                 wnd->close_cb(wnd, wnd->data);
1541 }
1542
1543 int wlt_display_create_window(struct wlt_display *disp,
1544                               struct wlt_window **out,
1545                               unsigned int width,
1546                               unsigned int height,
1547                               void *data)
1548 {
1549         struct wlt_window *wnd;
1550         int ret;
1551
1552         if (!disp || !out || !width || !height)
1553                 return -EINVAL;
1554
1555         if (disp->state != STATE_RUNNING) {
1556                 log_error("cannot create window, display is not running but in state %u",
1557                           disp->state);
1558                 return -EBUSY;
1559         }
1560
1561         wnd = malloc(sizeof(*wnd));
1562         if (!wnd)
1563                 return -ENOMEM;
1564         memset(wnd, 0, sizeof(*wnd));
1565         wnd->ref = 1;
1566         wnd->disp = disp;
1567         wnd->data = data;
1568         shl_dlist_init(&wnd->widget_list);
1569
1570         wnd->w_surface = wl_compositor_create_surface(disp->w_comp);
1571         if (!wnd->w_surface) {
1572                 log_error("cannot create wayland surface");
1573                 ret = -EFAULT;
1574                 goto err_free;
1575         }
1576
1577         wnd->w_shell_surface = wl_shell_get_shell_surface(disp->w_shell,
1578                                                           wnd->w_surface);
1579         if (!wnd->w_shell_surface) {
1580                 log_error("cannot retrieve shell-surface for wayland surface");
1581                 ret = -EFAULT;
1582                 goto err_surface;
1583         }
1584
1585         wl_shell_surface_add_listener(wnd->w_shell_surface,
1586                                       &shell_surface_listener, wnd);
1587         wl_shell_surface_set_toplevel(wnd->w_shell_surface);
1588
1589         ret = resize_window(wnd, width, height);
1590         if (ret)
1591                 goto err_shell_surface;
1592
1593         wlt_display_ref(disp);
1594         shl_dlist_link(&disp->window_list, &wnd->list);
1595         *out = wnd;
1596         return 0;
1597
1598 err_shell_surface:
1599         wl_shell_surface_destroy(wnd->w_shell_surface);
1600 err_surface:
1601         wl_surface_destroy(wnd->w_surface);
1602 err_free:
1603         free(wnd);
1604         return ret;
1605 }
1606
1607 void wlt_window_ref(struct wlt_window *wnd)
1608 {
1609         if (!wnd || !wnd->ref)
1610                 return;
1611
1612         ++wnd->ref;
1613 }
1614
1615 void wlt_window_unref(struct wlt_window *wnd)
1616 {
1617         struct wlt_widget *widget;
1618
1619         if (!wnd || !wnd->ref || --wnd->ref)
1620                 return;
1621
1622         while (!shl_dlist_empty(&wnd->widget_list)) {
1623                 widget = shl_dlist_entry(wnd->widget_list.next,
1624                                          struct wlt_widget, list);
1625                 wlt_widget_destroy(widget);
1626         }
1627
1628         if (wnd->close_pending)
1629                 ev_eloop_unregister_idle_cb(wnd->disp->eloop, close_window,
1630                                             wnd, EV_NORMAL);
1631         if (wnd->idle_pending)
1632                 ev_eloop_unregister_idle_cb(wnd->disp->eloop, idle_frame, wnd,
1633                                             EV_NORMAL);
1634         shl_dlist_unlink(&wnd->list);
1635         if (wnd->w_frame)
1636                 wl_callback_destroy(wnd->w_frame);
1637         if (wnd->w_buffer)
1638                 wl_buffer_destroy(wnd->w_buffer);
1639         if (wnd->pool)
1640                 wlt_pool_free(wnd->pool);
1641         wl_shell_surface_destroy(wnd->w_shell_surface);
1642         wl_surface_destroy(wnd->w_surface);
1643         wlt_display_unref(wnd->disp);
1644         free(wnd);
1645 }
1646
1647 int wlt_window_create_widget(struct wlt_window *wnd,
1648                              struct wlt_widget **out,
1649                              void *data)
1650 {
1651         struct wlt_widget *widget;
1652
1653         if (!wnd || !out)
1654                 return -EINVAL;
1655
1656         widget = malloc(sizeof(*widget));
1657         if (!widget)
1658                 return -ENOMEM;
1659         memset(widget, 0, sizeof(*widget));
1660         widget->wnd = wnd;
1661         widget->data = data;
1662
1663         wlt_window_set_size(wnd, wnd->buffer.width, wnd->buffer.height);
1664         shl_dlist_link_tail(&wnd->widget_list, &widget->list);
1665         *out = widget;
1666         return 0;
1667 }
1668
1669 void wlt_window_schedule_redraw(struct wlt_window *wnd)
1670 {
1671         if (!wnd)
1672                 return;
1673
1674         schedule_frame(wnd);
1675 }
1676
1677 void wlt_window_damage(struct wlt_window *wnd,
1678                        struct wlt_rect *damage)
1679 {
1680         if (!wnd || !damage || wnd->skip_damage)
1681                 return;
1682
1683         wl_surface_damage(wnd->w_surface, damage->x, damage->y,
1684                           damage->width, damage->height);
1685         wl_surface_commit(wnd->w_surface);
1686 }
1687
1688 void wlt_window_get_buffer(struct wlt_window *wnd,
1689                            const struct wlt_rect *alloc,
1690                            struct wlt_shm_buffer *buf)
1691 {
1692         struct wlt_shm_buffer *rbuf;
1693
1694         if (!wnd || !buf)
1695                 return;
1696
1697         rbuf = &wnd->buffer;
1698
1699         if (!alloc) {
1700                 memcpy(buf, rbuf, sizeof(*buf));
1701                 return;
1702         }
1703
1704         if (alloc->x >= rbuf->width ||
1705             alloc->y >= rbuf->height) {
1706                 memset(buf, 0, sizeof(*buf));
1707                 return;
1708         }
1709
1710         /* set width */
1711         if (alloc->x + alloc->width > rbuf->width)
1712                 buf->width = rbuf->width - alloc->x;
1713         else
1714                 buf->width = alloc->width;
1715
1716         /* set height */
1717         if (alloc->y + alloc->height > rbuf->height)
1718                 buf->height = rbuf->height - alloc->y;
1719         else
1720                 buf->height = alloc->height;
1721
1722         /* set stride */
1723         buf->stride  = rbuf->stride;
1724
1725         /* set data */
1726         buf->data  = rbuf->data;
1727         buf->data += alloc->y * rbuf->stride;
1728         buf->data += alloc->x * 4;
1729 }
1730
1731 void wlt_window_move(struct wlt_window *wnd)
1732 {
1733         if (!wnd)
1734                 return;
1735
1736         wl_shell_surface_move(wnd->w_shell_surface,
1737                               wnd->disp->w_seat,
1738                               wnd->disp->last_serial);
1739 }
1740
1741 void wlt_window_resize(struct wlt_window *wnd, uint32_t edges)
1742 {
1743         if (!wnd)
1744                 return;
1745
1746         wl_shell_surface_resize(wnd->w_shell_surface,
1747                                 wnd->disp->w_seat,
1748                                 wnd->disp->last_serial,
1749                                 edges);
1750 }
1751
1752 int wlt_window_set_size(struct wlt_window *wnd,
1753                         unsigned int width, unsigned int height)
1754 {
1755         if (!wnd)
1756                 return -EINVAL;
1757
1758         wnd->new_width = width;
1759         wnd->new_height = height;
1760         wnd->need_resize = true;
1761         schedule_frame(wnd);
1762
1763         return 0;
1764 }
1765
1766 void wlt_window_set_cursor(struct wlt_window *wnd, unsigned int cursor)
1767 {
1768         if (!wnd)
1769                 return;
1770
1771         set_cursor(wnd->disp, cursor);
1772 }
1773
1774 void wlt_window_set_close_cb(struct wlt_window *wnd,
1775                              wlt_window_close_cb cb)
1776 {
1777         if (!wnd)
1778                 return;
1779
1780         wnd->close_cb = cb;
1781 }
1782
1783 void wlt_window_close(struct wlt_window *wnd)
1784 {
1785         if (!wnd)
1786                 return;
1787
1788         wnd->close_pending = true;
1789         ev_eloop_register_idle_cb(wnd->disp->eloop, close_window, wnd,
1790                                   EV_NORMAL);
1791 }
1792
1793 void wlt_window_toggle_maximize(struct wlt_window *wnd)
1794 {
1795         if (!wnd)
1796                 return;
1797
1798         if (wnd->maximized) {
1799                 if (!wnd->fullscreen) {
1800                         wl_shell_surface_set_toplevel(wnd->w_shell_surface);
1801                         wlt_window_set_size(wnd, wnd->saved_width,
1802                                             wnd->saved_height);
1803                 }
1804         } else {
1805                 if (!wnd->fullscreen) {
1806                         wnd->saved_width = wnd->buffer.width;
1807                         wnd->saved_height = wnd->buffer.height;
1808                         wl_shell_surface_set_maximized(wnd->w_shell_surface,
1809                                                        NULL);
1810                 }
1811         }
1812
1813         wnd->maximized = !wnd->maximized;
1814 }
1815
1816 void wlt_window_toggle_fullscreen(struct wlt_window *wnd)
1817 {
1818         if (!wnd)
1819                 return;
1820
1821         if (wnd->fullscreen) {
1822                 if (wnd->maximized) {
1823                         wl_shell_surface_set_maximized(wnd->w_shell_surface,
1824                                                        NULL);
1825                 } else {
1826                         wl_shell_surface_set_toplevel(wnd->w_shell_surface);
1827                         wlt_window_set_size(wnd, wnd->saved_width,
1828                                             wnd->saved_height);
1829                 }
1830         } else {
1831                 if (!wnd->maximized) {
1832                         wnd->saved_width = wnd->buffer.width;
1833                         wnd->saved_height = wnd->buffer.height;
1834                 }
1835                 wl_shell_surface_set_fullscreen(wnd->w_shell_surface,
1836                                 WL_SHELL_SURFACE_FULLSCREEN_METHOD_DEFAULT,
1837                                 0, NULL);
1838         }
1839
1840         wnd->fullscreen = !wnd->fullscreen;
1841 }
1842
1843 struct ev_eloop *wlt_window_get_eloop(struct wlt_window *wnd)
1844 {
1845         if (!wnd)
1846                 return NULL;
1847
1848         return wnd->disp->eloop;
1849 }
1850
1851 struct wlt_display *wlt_window_get_display(struct wlt_window *wnd)
1852 {
1853         if (!wnd)
1854                 return NULL;
1855
1856         return wnd->disp;
1857 }
1858
1859 void wlt_widget_destroy(struct wlt_widget *widget)
1860 {
1861         if (!widget)
1862                 return;
1863
1864         if (widget->destroy_cb)
1865                 widget->destroy_cb(widget, widget->data);
1866         shl_dlist_unlink(&widget->list);
1867         free(widget);
1868 }
1869
1870 struct wlt_window *wlt_widget_get_window(struct wlt_widget *widget)
1871 {
1872         if (!widget)
1873                 return NULL;
1874
1875         return widget->wnd;
1876 }
1877
1878 void wlt_widget_set_redraw_cb(struct wlt_widget *widget,
1879                               wlt_widget_redraw_cb cb)
1880 {
1881         if (!widget)
1882                 return;
1883
1884         widget->redraw_cb = cb;
1885 }
1886
1887 void wlt_widget_set_destroy_cb(struct wlt_widget *widget,
1888                                wlt_widget_destroy_cb cb)
1889 {
1890         if (!widget)
1891                 return;
1892
1893         widget->destroy_cb = cb;
1894 }
1895
1896 void wlt_widget_set_resize_cb(struct wlt_widget *widget,
1897                               wlt_widget_prepare_resize_cb prepare_cb,
1898                               wlt_widget_resize_cb cb)
1899 {
1900         if (!widget)
1901                 return;
1902
1903         widget->prepare_resize_cb = prepare_cb;
1904         widget->resize_cb = cb;
1905 }
1906
1907 void wlt_widget_set_pointer_cb(struct wlt_widget *widget,
1908                                wlt_widget_pointer_enter_cb enter_cb,
1909                                wlt_widget_pointer_leave_cb leave_cb,
1910                                wlt_widget_pointer_motion_cb motion_cb,
1911                                wlt_widget_pointer_button_cb button_cb)
1912 {
1913         if (!widget)
1914                 return;
1915
1916         widget->pointer_enter_cb = enter_cb;
1917         widget->pointer_leave_cb = leave_cb;
1918         widget->pointer_motion_cb = motion_cb;
1919         widget->pointer_button_cb = button_cb;
1920 }
1921
1922 void wlt_widget_set_keyboard_cb(struct wlt_widget *widget,
1923                                 wlt_widget_keyboard_cb cb)
1924 {
1925         if (!widget)
1926                 return;
1927
1928         widget->keyboard_cb = cb;
1929 }