]> git.sur5r.net Git - i3/i3lock/blob - xcb.c
Add support for `xss-lock --transfer-sleep-lock' (#75)
[i3/i3lock] / xcb.c
1 /*
2  * vim:ts=4:sw=4:expandtab
3  *
4  * © 2010 Michael Stapelberg
5  *
6  * xcb.c: contains all functions which use XCB to talk to X11. Mostly wrappers
7  *        around the rather complicated/ugly parts of the XCB API.
8  *
9  */
10 #include <xcb/xcb.h>
11 #include <xcb/xcb_image.h>
12 #include <xcb/xcb_atom.h>
13 #include <xcb/xcb_aux.h>
14 #include <stdio.h>
15 #include <stdlib.h>
16 #include <stdbool.h>
17 #include <string.h>
18 #include <unistd.h>
19 #include <assert.h>
20 #include <err.h>
21
22 #include "cursors.h"
23
24 xcb_connection_t *conn;
25 xcb_screen_t *screen;
26
27 #define curs_invisible_width 8
28 #define curs_invisible_height 8
29
30 static unsigned char curs_invisible_bits[] = {
31     0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00};
32
33 #define curs_windows_width 11
34 #define curs_windows_height 19
35
36 static unsigned char curs_windows_bits[] = {
37     0xfe, 0x07, 0xfc, 0x07, 0xfa, 0x07, 0xf6, 0x07, 0xee, 0x07, 0xde, 0x07,
38     0xbe, 0x07, 0x7e, 0x07, 0xfe, 0x06, 0xfe, 0x05, 0x3e, 0x00, 0xb6, 0x07,
39     0x6a, 0x07, 0x6c, 0x07, 0xde, 0x06, 0xdf, 0x06, 0xbf, 0x05, 0xbf, 0x05,
40     0x7f, 0x06};
41
42 #define mask_windows_width 11
43 #define mask_windows_height 19
44
45 static unsigned char mask_windows_bits[] = {
46     0x01, 0x00, 0x03, 0x00, 0x07, 0x00, 0x0f, 0x00, 0x1f, 0x00, 0x3f, 0x00,
47     0x7f, 0x00, 0xff, 0x00, 0xff, 0x01, 0xff, 0x03, 0xff, 0x07, 0x7f, 0x00,
48     0xf7, 0x00, 0xf3, 0x00, 0xe1, 0x01, 0xe0, 0x01, 0xc0, 0x03, 0xc0, 0x03,
49     0x80, 0x01};
50
51 static uint32_t get_colorpixel(char *hex) {
52     char strgroups[3][3] = {{hex[0], hex[1], '\0'},
53                             {hex[2], hex[3], '\0'},
54                             {hex[4], hex[5], '\0'}};
55     uint32_t rgb16[3] = {(strtol(strgroups[0], NULL, 16)),
56                          (strtol(strgroups[1], NULL, 16)),
57                          (strtol(strgroups[2], NULL, 16))};
58
59     return (rgb16[0] << 16) + (rgb16[1] << 8) + rgb16[2];
60 }
61
62 xcb_visualtype_t *get_root_visual_type(xcb_screen_t *screen) {
63     xcb_visualtype_t *visual_type = NULL;
64     xcb_depth_iterator_t depth_iter;
65     xcb_visualtype_iterator_t visual_iter;
66
67     for (depth_iter = xcb_screen_allowed_depths_iterator(screen);
68          depth_iter.rem;
69          xcb_depth_next(&depth_iter)) {
70         for (visual_iter = xcb_depth_visuals_iterator(depth_iter.data);
71              visual_iter.rem;
72              xcb_visualtype_next(&visual_iter)) {
73             if (screen->root_visual != visual_iter.data->visual_id)
74                 continue;
75
76             visual_type = visual_iter.data;
77             return visual_type;
78         }
79     }
80
81     return NULL;
82 }
83
84 xcb_pixmap_t create_bg_pixmap(xcb_connection_t *conn, xcb_screen_t *scr, u_int32_t *resolution, char *color) {
85     xcb_pixmap_t bg_pixmap = xcb_generate_id(conn);
86     xcb_create_pixmap(conn, scr->root_depth, bg_pixmap, scr->root,
87                       resolution[0], resolution[1]);
88
89     /* Generate a Graphics Context and fill the pixmap with background color
90      * (for images that are smaller than your screen) */
91     xcb_gcontext_t gc = xcb_generate_id(conn);
92     uint32_t values[] = {get_colorpixel(color)};
93     xcb_create_gc(conn, gc, bg_pixmap, XCB_GC_FOREGROUND, values);
94     xcb_rectangle_t rect = {0, 0, resolution[0], resolution[1]};
95     xcb_poly_fill_rectangle(conn, bg_pixmap, gc, 1, &rect);
96     xcb_free_gc(conn, gc);
97
98     return bg_pixmap;
99 }
100
101 xcb_window_t open_fullscreen_window(xcb_connection_t *conn, xcb_screen_t *scr, char *color, xcb_pixmap_t pixmap) {
102     uint32_t mask = 0;
103     uint32_t values[3];
104     xcb_window_t win = xcb_generate_id(conn);
105
106     if (pixmap == XCB_NONE) {
107         mask |= XCB_CW_BACK_PIXEL;
108         values[0] = get_colorpixel(color);
109     } else {
110         mask |= XCB_CW_BACK_PIXMAP;
111         values[0] = pixmap;
112     }
113
114     mask |= XCB_CW_OVERRIDE_REDIRECT;
115     values[1] = 1;
116
117     mask |= XCB_CW_EVENT_MASK;
118     values[2] = XCB_EVENT_MASK_EXPOSURE |
119                 XCB_EVENT_MASK_KEY_PRESS |
120                 XCB_EVENT_MASK_KEY_RELEASE |
121                 XCB_EVENT_MASK_VISIBILITY_CHANGE |
122                 XCB_EVENT_MASK_STRUCTURE_NOTIFY;
123
124     xcb_create_window(conn,
125                       XCB_COPY_FROM_PARENT,
126                       win,       /* the window id */
127                       scr->root, /* parent == root */
128                       0, 0,
129                       scr->width_in_pixels,
130                       scr->height_in_pixels, /* dimensions */
131                       0,                     /* border = 0, we draw our own */
132                       XCB_WINDOW_CLASS_INPUT_OUTPUT,
133                       XCB_WINDOW_CLASS_COPY_FROM_PARENT, /* copy visual from parent */
134                       mask,
135                       values);
136
137     char *name = "i3lock";
138     xcb_change_property(conn,
139                         XCB_PROP_MODE_REPLACE,
140                         win,
141                         XCB_ATOM_WM_NAME,
142                         XCB_ATOM_STRING,
143                         8,
144                         strlen(name),
145                         name);
146
147     /* Map the window (= make it visible) */
148     xcb_map_window(conn, win);
149
150     /* Raise window (put it on top) */
151     values[0] = XCB_STACK_MODE_ABOVE;
152     xcb_configure_window(conn, win, XCB_CONFIG_WINDOW_STACK_MODE, values);
153
154     /* Ensure that the window is created and set up before returning */
155     xcb_aux_sync(conn);
156
157     return win;
158 }
159
160 /*
161  * Repeatedly tries to grab pointer and keyboard (up to 1000 times).
162  *
163  */
164 void grab_pointer_and_keyboard(xcb_connection_t *conn, xcb_screen_t *screen, xcb_cursor_t cursor) {
165     xcb_grab_pointer_cookie_t pcookie;
166     xcb_grab_pointer_reply_t *preply;
167
168     xcb_grab_keyboard_cookie_t kcookie;
169     xcb_grab_keyboard_reply_t *kreply;
170
171     int tries = 10000;
172
173     while (tries-- > 0) {
174         pcookie = xcb_grab_pointer(
175             conn,
176             false,               /* get all pointer events specified by the following mask */
177             screen->root,        /* grab the root window */
178             XCB_NONE,            /* which events to let through */
179             XCB_GRAB_MODE_ASYNC, /* pointer events should continue as normal */
180             XCB_GRAB_MODE_ASYNC, /* keyboard mode */
181             XCB_NONE,            /* confine_to = in which window should the cursor stay */
182             cursor,              /* we change the cursor to whatever the user wanted */
183             XCB_CURRENT_TIME);
184
185         if ((preply = xcb_grab_pointer_reply(conn, pcookie, NULL)) &&
186             preply->status == XCB_GRAB_STATUS_SUCCESS) {
187             free(preply);
188             break;
189         }
190
191         /* Make this quite a bit slower */
192         usleep(50);
193     }
194
195     while (tries-- > 0) {
196         kcookie = xcb_grab_keyboard(
197             conn,
198             true,         /* report events */
199             screen->root, /* grab the root window */
200             XCB_CURRENT_TIME,
201             XCB_GRAB_MODE_ASYNC, /* process events as normal, do not require sync */
202             XCB_GRAB_MODE_ASYNC);
203
204         if ((kreply = xcb_grab_keyboard_reply(conn, kcookie, NULL)) &&
205             kreply->status == XCB_GRAB_STATUS_SUCCESS) {
206             free(kreply);
207             break;
208         }
209
210         /* Make this quite a bit slower */
211         usleep(50);
212     }
213
214     if (tries <= 0)
215         errx(EXIT_FAILURE, "Cannot grab pointer/keyboard");
216 }
217
218 xcb_cursor_t create_cursor(xcb_connection_t *conn, xcb_screen_t *screen, xcb_window_t win, int choice) {
219     xcb_pixmap_t bitmap;
220     xcb_pixmap_t mask;
221     xcb_cursor_t cursor;
222
223     unsigned char *curs_bits;
224     unsigned char *mask_bits;
225     int curs_w, curs_h;
226
227     switch (choice) {
228         case CURS_NONE:
229             curs_bits = curs_invisible_bits;
230             mask_bits = curs_invisible_bits;
231             curs_w = curs_invisible_width;
232             curs_h = curs_invisible_height;
233             break;
234         case CURS_WIN:
235             curs_bits = curs_windows_bits;
236             mask_bits = mask_windows_bits;
237             curs_w = curs_windows_width;
238             curs_h = curs_windows_height;
239             break;
240         case CURS_DEFAULT:
241         default:
242             return XCB_NONE; /* XCB_NONE is xcb's way of saying "don't change the cursor" */
243     }
244
245     bitmap = xcb_create_pixmap_from_bitmap_data(conn,
246                                                 win,
247                                                 curs_bits,
248                                                 curs_w,
249                                                 curs_h,
250                                                 1,
251                                                 screen->white_pixel,
252                                                 screen->black_pixel,
253                                                 NULL);
254
255     mask = xcb_create_pixmap_from_bitmap_data(conn,
256                                               win,
257                                               mask_bits,
258                                               curs_w,
259                                               curs_h,
260                                               1,
261                                               screen->white_pixel,
262                                               screen->black_pixel,
263                                               NULL);
264
265     cursor = xcb_generate_id(conn);
266
267     xcb_create_cursor(conn,
268                       cursor,
269                       bitmap,
270                       mask,
271                       65535, 65535, 65535,
272                       0, 0, 0,
273                       0, 0);
274
275     xcb_free_pixmap(conn, bitmap);
276     xcb_free_pixmap(conn, mask);
277
278     return cursor;
279 }