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