]> git.sur5r.net Git - i3/i3lock/blob - i3lock.c
28a93f081e1312dc874833bff04318bdcc4c2f84
[i3/i3lock] / i3lock.c
1 /*
2  * vim:ts=4:sw=4:expandtab
3  *
4  * © 2010-2013 Michael Stapelberg
5  *
6  * See LICENSE for licensing information
7  *
8  */
9 #include <stdio.h>
10 #include <stdlib.h>
11 #include <string.h>
12 #include <unistd.h>
13 #include <stdbool.h>
14 #include <stdint.h>
15 #include <xcb/xcb.h>
16 #include <xcb/dpms.h>
17 #include <err.h>
18 #include <assert.h>
19 #include <security/pam_appl.h>
20 #include <X11/Xlib-xcb.h>
21 #include <getopt.h>
22 #include <string.h>
23 #include <ev.h>
24 #include <sys/mman.h>
25 #include <X11/XKBlib.h>
26 #include <X11/extensions/XKBfile.h>
27 #include <xkbcommon/xkbcommon.h>
28 #include <cairo.h>
29 #include <cairo/cairo-xcb.h>
30
31 #include "i3lock.h"
32 #include "xcb.h"
33 #include "cursors.h"
34 #include "unlock_indicator.h"
35 #include "xinerama.h"
36
37 /* We need this for libxkbfile */
38 static Display *display;
39 char color[7] = "ffffff";
40 uint32_t last_resolution[2];
41 xcb_window_t win;
42 static xcb_cursor_t cursor;
43 static pam_handle_t *pam_handle;
44 int input_position = 0;
45 /* Holds the password you enter (in UTF-8). */
46 static char password[512];
47 static bool beep = false;
48 bool debug_mode = false;
49 static bool dpms = false;
50 bool unlock_indicator = true;
51 static bool dont_fork = false;
52 struct ev_loop *main_loop;
53 static struct ev_timer *clear_pam_wrong_timeout;
54 extern unlock_state_t unlock_state;
55 extern pam_state_t pam_state;
56
57 static struct xkb_state *xkb_state;
58 static struct xkb_context *xkb_context;
59 static struct xkb_keymap *xkb_keymap;
60
61 cairo_surface_t *img = NULL;
62 bool tile = false;
63 bool ignore_empty_password = false;
64
65 /* isutf, u8_dec © 2005 Jeff Bezanson, public domain */
66 #define isutf(c) (((c) & 0xC0) != 0x80)
67
68 /*
69  * Decrements i to point to the previous unicode glyph
70  *
71  */
72 void u8_dec(char *s, int *i) {
73     (void)(isutf(s[--(*i)]) || isutf(s[--(*i)]) || isutf(s[--(*i)]) || --(*i));
74 }
75
76 static void turn_monitors_on(void) {
77     if (dpms)
78         dpms_set_mode(conn, XCB_DPMS_DPMS_MODE_ON);
79 }
80
81 static void turn_monitors_off(void) {
82     if (dpms)
83         dpms_set_mode(conn, XCB_DPMS_DPMS_MODE_OFF);
84 }
85
86 /*
87  * Loads the XKB keymap from the X11 server and feeds it to xkbcommon.
88  * Necessary so that we can properly let xkbcommon track the keyboard state and
89  * translate keypresses to utf-8.
90  *
91  * Ideally, xkbcommon would ship something like this itself, but as of now
92  * (version 0.2.0), it doesn’t.
93  *
94  * TODO: Once xcb-xkb is enabled by default and released, we should port this
95  * code to xcb-xkb. See also https://github.com/xkbcommon/libxkbcommon/issues/1
96  *
97  */
98 static bool load_keymap(void) {
99     bool ret = false;
100     XkbFileInfo result;
101     memset(&result, '\0', sizeof(result));
102     result.xkb = XkbGetKeyboard(display, XkbAllMapComponentsMask, XkbUseCoreKbd);
103     if (result.xkb == NULL) {
104         fprintf(stderr, "[i3lock] XKB: XkbGetKeyboard failed\n");
105         return false;
106     }
107
108     FILE *temp = tmpfile();
109     if (temp == NULL) {
110         fprintf(stderr, "[i3lock] could not create tempfile\n");
111         return false;
112     }
113
114     bool ok = XkbWriteXKBKeymap(temp, &result, false, false, NULL, NULL);
115     if (!ok) {
116         fprintf(stderr, "[i3lock] XkbWriteXKBKeymap failed\n");
117         goto out;
118     }
119
120     rewind(temp);
121
122     if (xkb_context == NULL) {
123         if ((xkb_context = xkb_context_new(0)) == NULL) {
124             fprintf(stderr, "[i3lock] could not create xkbcommon context\n");
125             goto out;
126         }
127     }
128
129     if (xkb_keymap != NULL)
130         xkb_keymap_unref(xkb_keymap);
131
132     if ((xkb_keymap = xkb_keymap_new_from_file(xkb_context, temp, XKB_KEYMAP_FORMAT_TEXT_V1, 0)) == NULL) {
133         fprintf(stderr, "[i3lock] xkb_keymap_new_from_file failed\n");
134         goto out;
135     }
136
137     struct xkb_state *new_state = xkb_state_new(xkb_keymap);
138     if (new_state == NULL) {
139         fprintf(stderr, "[i3lock] xkb_state_new failed\n");
140         goto out;
141     }
142
143     /* Get the initial modifier state to be in sync with the X server.
144      * See https://github.com/xkbcommon/libxkbcommon/issues/1 for why we ignore
145      * the base and latched fields. */
146     XkbStateRec state_rec;
147     XkbGetState(display, XkbUseCoreKbd, &state_rec);
148
149     xkb_state_update_mask(new_state,
150         0, 0, state_rec.locked_mods,
151         0, 0, state_rec.locked_group);
152
153     if (xkb_state != NULL)
154         xkb_state_unref(xkb_state);
155     xkb_state = new_state;
156
157     ret = true;
158 out:
159     XkbFreeKeyboard(result.xkb, XkbAllComponentsMask, true);
160     fclose(temp);
161     return ret;
162 }
163
164 /*
165  * Clears the memory which stored the password to be a bit safer against
166  * cold-boot attacks.
167  *
168  */
169 static void clear_password_memory(void) {
170     /* A volatile pointer to the password buffer to prevent the compiler from
171      * optimizing this out. */
172     volatile char *vpassword = password;
173     for (int c = 0; c < sizeof(password); c++)
174         /* We store a non-random pattern which consists of the (irrelevant)
175          * index plus (!) the value of the beep variable. This prevents the
176          * compiler from optimizing the calls away, since the value of 'beep'
177          * is not known at compile-time. */
178         vpassword[c] = c + (int)beep;
179 }
180
181
182 /*
183  * Resets pam_state to STATE_PAM_IDLE 2 seconds after an unsuccesful
184  * authentication event.
185  *
186  */
187 static void clear_pam_wrong(EV_P_ ev_timer *w, int revents) {
188     DEBUG("clearing pam wrong\n");
189     pam_state = STATE_PAM_IDLE;
190     unlock_state = STATE_STARTED;
191     redraw_screen();
192
193     /* Now free this timeout. */
194     ev_timer_stop(main_loop, clear_pam_wrong_timeout);
195     free(clear_pam_wrong_timeout);
196     clear_pam_wrong_timeout = NULL;
197 }
198
199 static void clear_input(void) {
200     input_position = 0;
201     clear_password_memory();
202     password[input_position] = '\0';
203
204     /* Hide the unlock indicator after a bit if the password buffer is
205      * empty. */
206     start_clear_indicator_timeout();
207     unlock_state = STATE_BACKSPACE_ACTIVE;
208     redraw_screen();
209     unlock_state = STATE_KEY_PRESSED;
210 }
211
212 static void input_done(void) {
213     if (clear_pam_wrong_timeout) {
214         ev_timer_stop(main_loop, clear_pam_wrong_timeout);
215         free(clear_pam_wrong_timeout);
216         clear_pam_wrong_timeout = NULL;
217     }
218
219     pam_state = STATE_PAM_VERIFY;
220     redraw_screen();
221
222     if (pam_authenticate(pam_handle, 0) == PAM_SUCCESS) {
223         DEBUG("successfully authenticated\n");
224         clear_password_memory();
225         /* Turn the screen on, as it may have been turned off
226          * on release of the 'enter' key. */
227         turn_monitors_on();
228         exit(0);
229     }
230
231     if (debug_mode)
232         fprintf(stderr, "Authentication failure\n");
233
234     pam_state = STATE_PAM_WRONG;
235     clear_input();
236     redraw_screen();
237
238     /* Clear this state after 2 seconds (unless the user enters another
239      * password during that time). */
240     ev_now_update(main_loop);
241     if ((clear_pam_wrong_timeout = calloc(sizeof(struct ev_timer), 1))) {
242         ev_timer_init(clear_pam_wrong_timeout, clear_pam_wrong, 2.0, 0.);
243         ev_timer_start(main_loop, clear_pam_wrong_timeout);
244     }
245
246     /* Cancel the clear_indicator_timeout, it would hide the unlock indicator
247      * too early. */
248     stop_clear_indicator_timeout();
249
250     /* beep on authentication failure, if enabled */
251     if (beep) {
252         xcb_bell(conn, 100);
253         xcb_flush(conn);
254     }
255 }
256
257 /*
258  * Called when the user releases a key. We need to leave the Mode_switch
259  * state when the user releases the Mode_switch key.
260  *
261  */
262 static void handle_key_release(xcb_key_release_event_t *event) {
263     xkb_state_update_key(xkb_state, event->detail, XKB_KEY_UP);
264 }
265
266 static void redraw_timeout(EV_P_ ev_timer *w, int revents) {
267     redraw_screen();
268
269     ev_timer_stop(main_loop, w);
270     free(w);
271 }
272
273 /*
274  * Handle key presses. Fixes state, then looks up the key symbol for the
275  * given keycode, then looks up the key symbol (as UCS-2), converts it to
276  * UTF-8 and stores it in the password array.
277  *
278  */
279 static void handle_key_press(xcb_key_press_event_t *event) {
280     xkb_keysym_t ksym;
281     char buffer[128];
282     int n;
283     bool ctrl;
284
285     ksym = xkb_state_key_get_one_sym(xkb_state, event->detail);
286     ctrl = xkb_state_mod_name_is_active(xkb_state, "Control", XKB_STATE_MODS_DEPRESSED);
287     xkb_state_update_key(xkb_state, event->detail, XKB_KEY_DOWN);
288
289     /* The buffer will be null-terminated, so n >= 2 for 1 actual character. */
290     memset(buffer, '\0', sizeof(buffer));
291     n = xkb_keysym_to_utf8(ksym, buffer, sizeof(buffer));
292
293     switch (ksym) {
294     case XKB_KEY_Return:
295     case XKB_KEY_KP_Enter:
296     case XKB_KEY_XF86ScreenSaver:
297         if (ignore_empty_password && input_position == 0) {
298             clear_input();
299             return;
300         }
301         password[input_position] = '\0';
302         unlock_state = STATE_KEY_PRESSED;
303         redraw_screen();
304         input_done();
305         return;
306
307     case XKB_KEY_u:
308         if (ctrl) {
309             DEBUG("C-u pressed\n");
310             clear_input();
311             return;
312         }
313         break;
314
315     case XKB_KEY_Escape:
316         clear_input();
317         return;
318
319     case XKB_KEY_BackSpace:
320         if (input_position == 0)
321             return;
322
323         /* decrement input_position to point to the previous glyph */
324         u8_dec(password, &input_position);
325         password[input_position] = '\0';
326
327         /* Hide the unlock indicator after a bit if the password buffer is
328          * empty. */
329         start_clear_indicator_timeout();
330         unlock_state = STATE_BACKSPACE_ACTIVE;
331         redraw_screen();
332         unlock_state = STATE_KEY_PRESSED;
333         return;
334     }
335
336     if ((input_position + 8) >= sizeof(password))
337         return;
338
339 #if 0
340     /* FIXME: handle all of these? */
341     printf("is_keypad_key = %d\n", xcb_is_keypad_key(sym));
342     printf("is_private_keypad_key = %d\n", xcb_is_private_keypad_key(sym));
343     printf("xcb_is_cursor_key = %d\n", xcb_is_cursor_key(sym));
344     printf("xcb_is_pf_key = %d\n", xcb_is_pf_key(sym));
345     printf("xcb_is_function_key = %d\n", xcb_is_function_key(sym));
346     printf("xcb_is_misc_function_key = %d\n", xcb_is_misc_function_key(sym));
347     printf("xcb_is_modifier_key = %d\n", xcb_is_modifier_key(sym));
348 #endif
349
350     if (n < 2)
351         return;
352
353     /* store it in the password array as UTF-8 */
354     memcpy(password+input_position, buffer, n-1);
355     input_position += n-1;
356     DEBUG("current password = %.*s\n", input_position, password);
357
358     unlock_state = STATE_KEY_ACTIVE;
359     redraw_screen();
360     unlock_state = STATE_KEY_PRESSED;
361
362     struct ev_timer *timeout = calloc(sizeof(struct ev_timer), 1);
363     if (timeout) {
364         ev_timer_init(timeout, redraw_timeout, 0.25, 0.);
365         ev_timer_start(main_loop, timeout);
366     }
367
368     stop_clear_indicator_timeout();
369 }
370
371 /*
372  * A visibility notify event will be received when the visibility (= can the
373  * user view the complete window) changes, so for example when a popup overlays
374  * some area of the i3lock window.
375  *
376  * In this case, we raise our window on top so that the popup (or whatever is
377  * hiding us) gets hidden.
378  *
379  */
380 static void handle_visibility_notify(xcb_connection_t *conn,
381     xcb_visibility_notify_event_t *event) {
382     if (event->state != XCB_VISIBILITY_UNOBSCURED) {
383         uint32_t values[] = { XCB_STACK_MODE_ABOVE };
384         xcb_configure_window(conn, event->window, XCB_CONFIG_WINDOW_STACK_MODE, values);
385         xcb_flush(conn);
386     }
387 }
388
389 /*
390  * Called when the keyboard mapping changes. We update our symbols.
391  *
392  */
393 static void handle_mapping_notify(xcb_mapping_notify_event_t *event) {
394     /* We ignore errors — if the new keymap cannot be loaded it’s better if the
395      * screen stays locked and the user intervenes by using killall i3lock. */
396     (void)load_keymap();
397 }
398
399 /*
400  * Called when the properties on the root window change, e.g. when the screen
401  * resolution changes. If so we update the window to cover the whole screen
402  * and also redraw the image, if any.
403  *
404  */
405 void handle_screen_resize(void) {
406     xcb_get_geometry_cookie_t geomc;
407     xcb_get_geometry_reply_t *geom;
408     geomc = xcb_get_geometry(conn, screen->root);
409     if ((geom = xcb_get_geometry_reply(conn, geomc, 0)) == NULL)
410         return;
411
412     if (last_resolution[0] == geom->width &&
413         last_resolution[1] == geom->height) {
414         free(geom);
415         return;
416     }
417
418     last_resolution[0] = geom->width;
419     last_resolution[1] = geom->height;
420
421     free(geom);
422
423     redraw_screen();
424
425     uint32_t mask = XCB_CONFIG_WINDOW_WIDTH | XCB_CONFIG_WINDOW_HEIGHT;
426     xcb_configure_window(conn, win, mask, last_resolution);
427     xcb_flush(conn);
428
429     xinerama_query_screens();
430     redraw_screen();
431 }
432
433 /*
434  * Callback function for PAM. We only react on password request callbacks.
435  *
436  */
437 static int conv_callback(int num_msg, const struct pam_message **msg,
438                          struct pam_response **resp, void *appdata_ptr)
439 {
440     if (num_msg == 0)
441         return 1;
442
443     /* PAM expects an array of responses, one for each message */
444     if ((*resp = calloc(num_msg, sizeof(struct pam_response))) == NULL) {
445         perror("calloc");
446         return 1;
447     }
448
449     for (int c = 0; c < num_msg; c++) {
450         if (msg[c]->msg_style != PAM_PROMPT_ECHO_OFF &&
451             msg[c]->msg_style != PAM_PROMPT_ECHO_ON)
452             continue;
453
454         /* return code is currently not used but should be set to zero */
455         resp[c]->resp_retcode = 0;
456         if ((resp[c]->resp = strdup(password)) == NULL) {
457             perror("strdup");
458             return 1;
459         }
460     }
461
462     return 0;
463 }
464
465 /*
466  * This callback is only a dummy, see xcb_prepare_cb and xcb_check_cb.
467  * See also man libev(3): "ev_prepare" and "ev_check" - customise your event loop
468  *
469  */
470 static void xcb_got_event(EV_P_ struct ev_io *w, int revents) {
471     /* empty, because xcb_prepare_cb and xcb_check_cb are used */
472 }
473
474 /*
475  * Flush before blocking (and waiting for new events)
476  *
477  */
478 static void xcb_prepare_cb(EV_P_ ev_prepare *w, int revents) {
479     xcb_flush(conn);
480 }
481
482 /*
483  * Instead of polling the X connection socket we leave this to
484  * xcb_poll_for_event() which knows better than we can ever know.
485  *
486  */
487 static void xcb_check_cb(EV_P_ ev_check *w, int revents) {
488     xcb_generic_event_t *event;
489
490     while ((event = xcb_poll_for_event(conn)) != NULL) {
491         if (event->response_type == 0) {
492             xcb_generic_error_t *error = (xcb_generic_error_t*)event;
493             if (debug_mode)
494                 fprintf(stderr, "X11 Error received! sequence 0x%x, error_code = %d\n",
495                         error->sequence, error->error_code);
496             free(event);
497             continue;
498         }
499
500         /* Strip off the highest bit (set if the event is generated) */
501         int type = (event->response_type & 0x7F);
502         switch (type) {
503             case XCB_KEY_PRESS:
504                 handle_key_press((xcb_key_press_event_t*)event);
505                 break;
506
507             case XCB_KEY_RELEASE:
508                 handle_key_release((xcb_key_release_event_t*)event);
509
510                 /* If this was the backspace or escape key we are back at an
511                  * empty input, so turn off the screen if DPMS is enabled */
512                 if (input_position == 0)
513                     turn_monitors_off();
514
515                 break;
516
517             case XCB_VISIBILITY_NOTIFY:
518                 handle_visibility_notify(conn, (xcb_visibility_notify_event_t*)event);
519                 break;
520
521             case XCB_MAP_NOTIFY:
522                 if (!dont_fork) {
523                     /* After the first MapNotify, we never fork again. We don’t
524                      * expect to get another MapNotify, but better be sure… */
525                     dont_fork = true;
526
527                     /* In the parent process, we exit */
528                     if (fork() != 0)
529                         exit(0);
530
531                     ev_loop_fork(EV_DEFAULT);
532                 }
533                 break;
534
535             case XCB_MAPPING_NOTIFY:
536                 handle_mapping_notify((xcb_mapping_notify_event_t*)event);
537                 break;
538
539             case XCB_CONFIGURE_NOTIFY:
540                 handle_screen_resize();
541                 break;
542         }
543
544         free(event);
545     }
546 }
547
548 /*
549  * This function is called from a fork()ed child and will raise the i3lock
550  * window when the window is obscured, even when the main i3lock process is
551  * blocked due to PAM.
552  *
553  */
554 static void raise_loop(xcb_window_t window) {
555     xcb_connection_t *conn;
556     xcb_generic_event_t *event;
557     int screens;
558
559     if ((conn = xcb_connect(NULL, &screens)) == NULL ||
560         xcb_connection_has_error(conn))
561         errx(EXIT_FAILURE, "Cannot open display\n");
562
563     /* We need to know about the window being obscured or getting destroyed. */
564     xcb_change_window_attributes(conn, window, XCB_CW_EVENT_MASK,
565         (uint32_t[]){
566             XCB_EVENT_MASK_VISIBILITY_CHANGE |
567             XCB_EVENT_MASK_STRUCTURE_NOTIFY
568         });
569     xcb_flush(conn);
570
571     DEBUG("Watching window 0x%08x\n", window);
572     while ((event = xcb_wait_for_event(conn)) != NULL) {
573         if (event->response_type == 0) {
574             xcb_generic_error_t *error = (xcb_generic_error_t*)event;
575             DEBUG("X11 Error received! sequence 0x%x, error_code = %d\n",
576                  error->sequence, error->error_code);
577             free(event);
578             continue;
579         }
580         /* Strip off the highest bit (set if the event is generated) */
581         int type = (event->response_type & 0x7F);
582         DEBUG("Read event of type %d\n", type);
583         switch (type) {
584             case XCB_VISIBILITY_NOTIFY:
585                 handle_visibility_notify(conn, (xcb_visibility_notify_event_t*)event);
586                 break;
587             case XCB_UNMAP_NOTIFY:
588                 DEBUG("UnmapNotify for 0x%08x\n", (((xcb_unmap_notify_event_t*)event)->window));
589                 if (((xcb_unmap_notify_event_t*)event)->window == window)
590                     exit(EXIT_SUCCESS);
591                 break;
592             case XCB_DESTROY_NOTIFY:
593                 DEBUG("DestroyNotify for 0x%08x\n", (((xcb_destroy_notify_event_t*)event)->window));
594                 if (((xcb_destroy_notify_event_t*)event)->window == window)
595                     exit(EXIT_SUCCESS);
596                 break;
597             default:
598                 DEBUG("Unhandled event type %d\n", type);
599                 break;
600         }
601         free(event);
602     }
603 }
604
605 int main(int argc, char *argv[]) {
606     char *username;
607     char *image_path = NULL;
608     int ret;
609     struct pam_conv conv = {conv_callback, NULL};
610     int curs_choice = CURS_NONE;
611     int o;
612     int optind = 0;
613     struct option longopts[] = {
614         {"version", no_argument, NULL, 'v'},
615         {"nofork", no_argument, NULL, 'n'},
616         {"beep", no_argument, NULL, 'b'},
617         {"dpms", no_argument, NULL, 'd'},
618         {"color", required_argument, NULL, 'c'},
619         {"pointer", required_argument, NULL , 'p'},
620         {"debug", no_argument, NULL, 0},
621         {"help", no_argument, NULL, 'h'},
622         {"no-unlock-indicator", no_argument, NULL, 'u'},
623         {"image", required_argument, NULL, 'i'},
624         {"tiling", no_argument, NULL, 't'},
625         {"ignore-empty-password", no_argument, NULL, 'e'},
626         {NULL, no_argument, NULL, 0}
627     };
628
629     if ((username = getenv("USER")) == NULL)
630         errx(1, "USER environment variable not set, please set it.\n");
631
632     while ((o = getopt_long(argc, argv, "hvnbdc:p:ui:te", longopts, &optind)) != -1) {
633         switch (o) {
634         case 'v':
635             errx(EXIT_SUCCESS, "version " VERSION " © 2010-2012 Michael Stapelberg");
636         case 'n':
637             dont_fork = true;
638             break;
639         case 'b':
640             beep = true;
641             break;
642         case 'd':
643             dpms = true;
644             break;
645         case 'c': {
646             char *arg = optarg;
647
648             /* Skip # if present */
649             if (arg[0] == '#')
650                 arg++;
651
652             if (strlen(arg) != 6 || sscanf(arg, "%06[0-9a-fA-F]", color) != 1)
653                 errx(1, "color is invalid, it must be given in 3-byte hexadecimal format: rrggbb\n");
654
655             break;
656         }
657         case 'u':
658             unlock_indicator = false;
659             break;
660         case 'i':
661             image_path = strdup(optarg);
662             break;
663         case 't':
664             tile = true;
665             break;
666         case 'p':
667             if (!strcmp(optarg, "win")) {
668                 curs_choice = CURS_WIN;
669             } else if (!strcmp(optarg, "default")) {
670                 curs_choice = CURS_DEFAULT;
671             } else {
672                 errx(1, "i3lock: Invalid pointer type given. Expected one of \"win\" or \"default\".\n");
673             }
674             break;
675         case 'e':
676             ignore_empty_password = true;
677             break;
678         case 0:
679             if (strcmp(longopts[optind].name, "debug") == 0)
680                 debug_mode = true;
681             break;
682         default:
683             errx(1, "Syntax: i3lock [-v] [-n] [-b] [-d] [-c color] [-u] [-p win|default]"
684             " [-i image.png] [-t] [-e]"
685             );
686         }
687     }
688
689     /* We need (relatively) random numbers for highlighting a random part of
690      * the unlock indicator upon keypresses. */
691     srand(time(NULL));
692
693     /* Initialize PAM */
694     ret = pam_start("i3lock", username, &conv, &pam_handle);
695     if (ret != PAM_SUCCESS)
696         errx(EXIT_FAILURE, "PAM: %s", pam_strerror(pam_handle, ret));
697
698 /* Using mlock() as non-super-user seems only possible in Linux. Users of other
699  * operating systems should use encrypted swap/no swap (or remove the ifdef and
700  * run i3lock as super-user). */
701 #if defined(__linux__)
702     /* Lock the area where we store the password in memory, we don’t want it to
703      * be swapped to disk. Since Linux 2.6.9, this does not require any
704      * privileges, just enough bytes in the RLIMIT_MEMLOCK limit. */
705     if (mlock(password, sizeof(password)) != 0)
706         err(EXIT_FAILURE, "Could not lock page in memory, check RLIMIT_MEMLOCK");
707 #endif
708
709     /* Initialize connection to X11 */
710     if ((display = XOpenDisplay(NULL)) == NULL)
711         errx(EXIT_FAILURE, "Could not connect to X11, maybe you need to set DISPLAY?");
712     XSetEventQueueOwner(display, XCBOwnsEventQueue);
713     conn = XGetXCBConnection(display);
714
715     /* Double checking that connection is good and operatable with xcb */
716     if (xcb_connection_has_error(conn))
717         errx(EXIT_FAILURE, "Could not connect to X11, maybe you need to set DISPLAY?");
718
719     /* When we cannot initially load the keymap, we better exit */
720     if (!load_keymap())
721         errx(EXIT_FAILURE, "Could not load keymap");
722
723     xinerama_init();
724     xinerama_query_screens();
725
726     /* if DPMS is enabled, check if the X server really supports it */
727     if (dpms) {
728         xcb_dpms_capable_cookie_t dpmsc = xcb_dpms_capable(conn);
729         xcb_dpms_capable_reply_t *dpmsr;
730         if ((dpmsr = xcb_dpms_capable_reply(conn, dpmsc, NULL))) {
731             if (!dpmsr->capable) {
732                 if (debug_mode)
733                     fprintf(stderr, "Disabling DPMS, X server not DPMS capable\n");
734                 dpms = false;
735             }
736             free(dpmsr);
737         }
738     }
739
740     screen = xcb_setup_roots_iterator(xcb_get_setup(conn)).data;
741
742     last_resolution[0] = screen->width_in_pixels;
743     last_resolution[1] = screen->height_in_pixels;
744
745     xcb_change_window_attributes(conn, screen->root, XCB_CW_EVENT_MASK,
746             (uint32_t[]){ XCB_EVENT_MASK_STRUCTURE_NOTIFY });
747
748     if (image_path) {
749         /* Create a pixmap to render on, fill it with the background color */
750         img = cairo_image_surface_create_from_png(image_path);
751         /* In case loading failed, we just pretend no -i was specified. */
752         if (cairo_surface_status(img) != CAIRO_STATUS_SUCCESS) {
753             fprintf(stderr, "Could not load image \"%s\": %s\n",
754                     image_path, cairo_status_to_string(cairo_surface_status(img)));
755             img = NULL;
756         }
757     }
758
759     /* Pixmap on which the image is rendered to (if any) */
760     xcb_pixmap_t bg_pixmap = draw_image(last_resolution);
761
762     /* open the fullscreen window, already with the correct pixmap in place */
763     win = open_fullscreen_window(conn, screen, color, bg_pixmap);
764     xcb_free_pixmap(conn, bg_pixmap);
765
766     pid_t pid = fork();
767     /* The pid == -1 case is intentionally ignored here:
768      * While the child process is useful for preventing other windows from
769      * popping up while i3lock blocks, it is not critical. */
770     if (pid == 0) {
771         /* Child */
772         close(xcb_get_file_descriptor(conn));
773         raise_loop(win);
774         exit(EXIT_SUCCESS);
775     }
776
777     cursor = create_cursor(conn, screen, win, curs_choice);
778
779     grab_pointer_and_keyboard(conn, screen, cursor);
780     /* Load the keymap again to sync the current modifier state. Since we first
781      * loaded the keymap, there might have been changes, but starting from now,
782      * we should get all key presses/releases due to having grabbed the
783      * keyboard. */
784     (void)load_keymap();
785
786     turn_monitors_off();
787
788     /* Initialize the libev event loop. */
789     main_loop = EV_DEFAULT;
790     if (main_loop == NULL)
791         errx(EXIT_FAILURE, "Could not initialize libev. Bad LIBEV_FLAGS?\n");
792
793     struct ev_io *xcb_watcher = calloc(sizeof(struct ev_io), 1);
794     struct ev_check *xcb_check = calloc(sizeof(struct ev_check), 1);
795     struct ev_prepare *xcb_prepare = calloc(sizeof(struct ev_prepare), 1);
796
797     ev_io_init(xcb_watcher, xcb_got_event, xcb_get_file_descriptor(conn), EV_READ);
798     ev_io_start(main_loop, xcb_watcher);
799
800     ev_check_init(xcb_check, xcb_check_cb);
801     ev_check_start(main_loop, xcb_check);
802
803     ev_prepare_init(xcb_prepare, xcb_prepare_cb);
804     ev_prepare_start(main_loop, xcb_prepare);
805
806     /* Invoke the event callback once to catch all the events which were
807      * received up until now. ev will only pick up new events (when the X11
808      * file descriptor becomes readable). */
809     ev_invoke(main_loop, xcb_check, 0);
810     ev_loop(main_loop, 0);
811 }