799 lines
25 KiB

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