git.sur5r.net Git - i3/i3lock/blob - 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 #include <time.h>
  22 #include <sys/time.h>
  23
  24 #include "cursors.h"
  25 #include "unlock_indicator.h"
  26
  27 extern auth_state_t auth_state;
  28
  29 xcb_connection_t *conn;
  30 xcb_screen_t *screen;
  31
  32 #define curs_invisible_width 8
  33 #define curs_invisible_height 8
  34
  35 static unsigned char curs_invisible_bits[] = {
  36     0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00};
  37
  38 #define curs_windows_width 11
  39 #define curs_windows_height 19
  40
  41 static unsigned char curs_windows_bits[] = {
  42     0xfe, 0x07, 0xfc, 0x07, 0xfa, 0x07, 0xf6, 0x07, 0xee, 0x07, 0xde, 0x07,
  43     0xbe, 0x07, 0x7e, 0x07, 0xfe, 0x06, 0xfe, 0x05, 0x3e, 0x00, 0xb6, 0x07,
  44     0x6a, 0x07, 0x6c, 0x07, 0xde, 0x06, 0xdf, 0x06, 0xbf, 0x05, 0xbf, 0x05,
  45     0x7f, 0x06};
  46
  47 #define mask_windows_width 11
  48 #define mask_windows_height 19
  49
  50 static unsigned char mask_windows_bits[] = {
  51     0x01, 0x00, 0x03, 0x00, 0x07, 0x00, 0x0f, 0x00, 0x1f, 0x00, 0x3f, 0x00,
  52     0x7f, 0x00, 0xff, 0x00, 0xff, 0x01, 0xff, 0x03, 0xff, 0x07, 0x7f, 0x00,
  53     0xf7, 0x00, 0xf3, 0x00, 0xe1, 0x01, 0xe0, 0x01, 0xc0, 0x03, 0xc0, 0x03,
  54     0x80, 0x01};
  55
  56 static uint32_t get_colorpixel(char *hex) {
  57     char strgroups[3][3] = {{hex[0], hex[1], '\0'},
  58                             {hex[2], hex[3], '\0'},
  59                             {hex[4], hex[5], '\0'}};
  60     uint32_t rgb16[3] = {(strtol(strgroups[0], NULL, 16)),
  61                          (strtol(strgroups[1], NULL, 16)),
  62                          (strtol(strgroups[2], NULL, 16))};
  63
  64     return (rgb16[0] << 16) + (rgb16[1] << 8) + rgb16[2];
  65 }
  66
  67 xcb_visualtype_t *get_root_visual_type(xcb_screen_t *screen) {
  68     xcb_visualtype_t *visual_type = NULL;
  69     xcb_depth_iterator_t depth_iter;
  70     xcb_visualtype_iterator_t visual_iter;
  71
  72     for (depth_iter = xcb_screen_allowed_depths_iterator(screen);
  73          depth_iter.rem;
  74          xcb_depth_next(&depth_iter)) {
  75         for (visual_iter = xcb_depth_visuals_iterator(depth_iter.data);
  76              visual_iter.rem;
  77              xcb_visualtype_next(&visual_iter)) {
  78             if (screen->root_visual != visual_iter.data->visual_id)
  79                 continue;
  80
  81             visual_type = visual_iter.data;
  82             return visual_type;
  83         }
  84     }
  85
  86     return NULL;
  87 }
  88
  89 xcb_pixmap_t create_bg_pixmap(xcb_connection_t *conn, xcb_screen_t *scr, u_int32_t *resolution, char *color) {
  90     xcb_pixmap_t bg_pixmap = xcb_generate_id(conn);
  91     xcb_create_pixmap(conn, scr->root_depth, bg_pixmap, scr->root,
  92                       resolution[0], resolution[1]);
  93
  94     /* Generate a Graphics Context and fill the pixmap with background color
  95      * (for images that are smaller than your screen) */
  96     xcb_gcontext_t gc = xcb_generate_id(conn);
  97     uint32_t values[] = {get_colorpixel(color)};
  98     xcb_create_gc(conn, gc, bg_pixmap, XCB_GC_FOREGROUND, values);
  99     xcb_rectangle_t rect = {0, 0, resolution[0], resolution[1]};
 100     xcb_poly_fill_rectangle(conn, bg_pixmap, gc, 1, &rect);
 101     xcb_free_gc(conn, gc);
 102
 103     return bg_pixmap;
 104 }
 105
 106 xcb_window_t open_fullscreen_window(xcb_connection_t *conn, xcb_screen_t *scr, char *color, xcb_pixmap_t pixmap) {
 107     uint32_t mask = 0;
 108     uint32_t values[3];
 109     xcb_window_t win = xcb_generate_id(conn);
 110
 111     if (pixmap == XCB_NONE) {
 112         mask |= XCB_CW_BACK_PIXEL;
 113         values[0] = get_colorpixel(color);
 114     } else {
 115         mask |= XCB_CW_BACK_PIXMAP;
 116         values[0] = pixmap;
 117     }
 118
 119     mask |= XCB_CW_OVERRIDE_REDIRECT;
 120     values[1] = 1;
 121
 122     mask |= XCB_CW_EVENT_MASK;
 123     values[2] = XCB_EVENT_MASK_EXPOSURE |
 124                 XCB_EVENT_MASK_KEY_PRESS |
 125                 XCB_EVENT_MASK_KEY_RELEASE |
 126                 XCB_EVENT_MASK_VISIBILITY_CHANGE |
 127                 XCB_EVENT_MASK_STRUCTURE_NOTIFY;
 128
 129     xcb_create_window(conn,
 130                       XCB_COPY_FROM_PARENT,
 131                       win,       /* the window id */
 132                       scr->root, /* parent == root */
 133                       0, 0,
 134                       scr->width_in_pixels,
 135                       scr->height_in_pixels, /* dimensions */
 136                       0,                     /* border = 0, we draw our own */
 137                       XCB_WINDOW_CLASS_INPUT_OUTPUT,
 138                       XCB_WINDOW_CLASS_COPY_FROM_PARENT, /* copy visual from parent */
 139                       mask,
 140                       values);
 141
 142     char *name = "i3lock";
 143     xcb_change_property(conn,
 144                         XCB_PROP_MODE_REPLACE,
 145                         win,
 146                         XCB_ATOM_WM_NAME,
 147                         XCB_ATOM_STRING,
 148                         8,
 149                         strlen(name),
 150                         name);
 151
 152     xcb_change_property(conn,
 153                         XCB_PROP_MODE_REPLACE,
 154                         win,
 155                         XCB_ATOM_WM_CLASS,
 156                         XCB_ATOM_STRING,
 157                         8,
 158                         2 * (strlen("i3lock") + 1),
 159                         "i3lock\0i3lock\0");
 160
 161     /* Map the window (= make it visible) */
 162     xcb_map_window(conn, win);
 163
 164     /* Raise window (put it on top) */
 165     values[0] = XCB_STACK_MODE_ABOVE;
 166     xcb_configure_window(conn, win, XCB_CONFIG_WINDOW_STACK_MODE, values);
 167
 168     /* Ensure that the window is created and set up before returning */
 169     xcb_aux_sync(conn);
 170
 171     return win;
 172 }
 173
 174 /*
 175  * Repeatedly tries to grab pointer and keyboard (up to the specified number of
 176  * tries).
 177  *
 178  * Returns true if the grab succeeded, false if not.
 179  *
 180  */
 181 bool grab_pointer_and_keyboard(xcb_connection_t *conn, xcb_screen_t *screen, xcb_cursor_t cursor, int tries) {
 182     xcb_grab_pointer_cookie_t pcookie;
 183     xcb_grab_pointer_reply_t *preply;
 184
 185     xcb_grab_keyboard_cookie_t kcookie;
 186     xcb_grab_keyboard_reply_t *kreply;
 187
 188     const suseconds_t screen_redraw_timeout = 100000; /* 100ms */
 189
 190     /* Using few variables to trigger a redraw_screen() if too many tries */
 191     bool redrawn = false;
 192     struct timeval start;
 193     if (gettimeofday(&start, NULL) == -1) {
 194         err(EXIT_FAILURE, "gettimeofday");
 195     }
 196
 197     while (tries-- > 0) {
 198         pcookie = xcb_grab_pointer(
 199             conn,
 200             false,               /* get all pointer events specified by the following mask */
 201             screen->root,        /* grab the root window */
 202             XCB_NONE,            /* which events to let through */
 203             XCB_GRAB_MODE_ASYNC, /* pointer events should continue as normal */
 204             XCB_GRAB_MODE_ASYNC, /* keyboard mode */
 205             XCB_NONE,            /* confine_to = in which window should the cursor stay */
 206             cursor,              /* we change the cursor to whatever the user wanted */
 207             XCB_CURRENT_TIME);
 208
 209         if ((preply = xcb_grab_pointer_reply(conn, pcookie, NULL)) &&
 210             preply->status == XCB_GRAB_STATUS_SUCCESS) {
 211             free(preply);
 212             break;
 213         }
 214
 215         /* In case the grab failed, we still need to free the reply */
 216         free(preply);
 217
 218         /* Make this quite a bit slower */
 219         usleep(50);
 220
 221         struct timeval now;
 222         if (gettimeofday(&now, NULL) == -1) {
 223             err(EXIT_FAILURE, "gettimeofday");
 224         }
 225
 226         struct timeval elapsed;
 227         timersub(&now, &start, &elapsed);
 228
 229         if (!redrawn &&
 230             (tries % 100) == 0 &&
 231             elapsed.tv_usec >= screen_redraw_timeout) {
 232             redraw_screen();
 233             redrawn = true;
 234         }
 235     }
 236
 237     while (tries-- > 0) {
 238         kcookie = xcb_grab_keyboard(
 239             conn,
 240             true,         /* report events */
 241             screen->root, /* grab the root window */
 242             XCB_CURRENT_TIME,
 243             XCB_GRAB_MODE_ASYNC, /* process events as normal, do not require sync */
 244             XCB_GRAB_MODE_ASYNC);
 245
 246         if ((kreply = xcb_grab_keyboard_reply(conn, kcookie, NULL)) &&
 247             kreply->status == XCB_GRAB_STATUS_SUCCESS) {
 248             free(kreply);
 249             break;
 250         }
 251
 252         /* In case the grab failed, we still need to free the reply */
 253         free(kreply);
 254
 255         /* Make this quite a bit slower */
 256         usleep(50);
 257
 258         struct timeval now;
 259         if (gettimeofday(&now, NULL) == -1) {
 260             err(EXIT_FAILURE, "gettimeofday");
 261         }
 262
 263         struct timeval elapsed;
 264         timersub(&now, &start, &elapsed);
 265
 266         /* Trigger a screen redraw if 100ms elapsed */
 267         if (!redrawn &&
 268             (tries % 100) == 0 &&
 269             elapsed.tv_usec >= screen_redraw_timeout) {
 270             redraw_screen();
 271             redrawn = true;
 272         }
 273     }
 274
 275     return (tries > 0);
 276 }
 277
 278 xcb_cursor_t create_cursor(xcb_connection_t *conn, xcb_screen_t *screen, xcb_window_t win, int choice) {
 279     xcb_pixmap_t bitmap;
 280     xcb_pixmap_t mask;
 281     xcb_cursor_t cursor;
 282
 283     unsigned char *curs_bits;
 284     unsigned char *mask_bits;
 285     int curs_w, curs_h;
 286
 287     switch (choice) {
 288         case CURS_NONE:
 289             curs_bits = curs_invisible_bits;
 290             mask_bits = curs_invisible_bits;
 291             curs_w = curs_invisible_width;
 292             curs_h = curs_invisible_height;
 293             break;
 294         case CURS_WIN:
 295             curs_bits = curs_windows_bits;
 296             mask_bits = mask_windows_bits;
 297             curs_w = curs_windows_width;
 298             curs_h = curs_windows_height;
 299             break;
 300         case CURS_DEFAULT:
 301         default:
 302             return XCB_NONE; /* XCB_NONE is xcb's way of saying "don't change the cursor" */
 303     }
 304
 305     bitmap = xcb_create_pixmap_from_bitmap_data(conn,
 306                                                 win,
 307                                                 curs_bits,
 308                                                 curs_w,
 309                                                 curs_h,
 310                                                 1,
 311                                                 screen->white_pixel,
 312                                                 screen->black_pixel,
 313                                                 NULL);
 314
 315     mask = xcb_create_pixmap_from_bitmap_data(conn,
 316                                               win,
 317                                               mask_bits,
 318                                               curs_w,
 319                                               curs_h,
 320                                               1,
 321                                               screen->white_pixel,
 322                                               screen->black_pixel,
 323                                               NULL);
 324
 325     cursor = xcb_generate_id(conn);
 326
 327     xcb_create_cursor(conn,
 328                       cursor,
 329                       bitmap,
 330                       mask,
 331                       65535, 65535, 65535,
 332                       0, 0, 0,
 333                       0, 0);
 334
 335     xcb_free_pixmap(conn, bitmap);
 336     xcb_free_pixmap(conn, mask);
 337
 338     return cursor;
 339 }
 340
 341 static xcb_atom_t _NET_ACTIVE_WINDOW = XCB_NONE;
 342 void _init_net_active_window(xcb_connection_t *conn) {
 343     if (_NET_ACTIVE_WINDOW != XCB_NONE) {
 344         /* already initialized */
 345         return;
 346     }
 347     xcb_generic_error_t *err;
 348     xcb_intern_atom_reply_t *atom_reply = xcb_intern_atom_reply(
 349         conn,
 350         xcb_intern_atom(conn, 0, strlen("_NET_ACTIVE_WINDOW"), "_NET_ACTIVE_WINDOW"),
 351         &err);
 352     if (atom_reply == NULL) {
 353         fprintf(stderr, "X11 Error %d\n", err->error_code);
 354         free(err);
 355         return;
 356     }
 357     _NET_ACTIVE_WINDOW = atom_reply->atom;
 358     free(atom_reply);
 359 }
 360
 361 xcb_window_t find_focused_window(xcb_connection_t *conn, const xcb_window_t root) {
 362     xcb_window_t result = XCB_NONE;
 363
 364     _init_net_active_window(conn);
 365
 366     xcb_get_property_reply_t *prop_reply = xcb_get_property_reply(
 367         conn,
 368         xcb_get_property_unchecked(
 369             conn, false, root, _NET_ACTIVE_WINDOW, XCB_GET_PROPERTY_TYPE_ANY, 0, 1 /* word */),
 370         NULL);
 371     if (prop_reply == NULL) {
 372         goto out;
 373     }
 374     if (xcb_get_property_value_length(prop_reply) == 0) {
 375         goto out_prop;
 376     }
 377     if (prop_reply->type != XCB_ATOM_WINDOW) {
 378         goto out_prop;
 379     }
 380
 381     result = *((xcb_window_t *)xcb_get_property_value(prop_reply));
 382
 383 out_prop:
 384     free(prop_reply);
 385 out:
 386     return result;
 387 }
 388
 389 void set_focused_window(xcb_connection_t *conn, const xcb_window_t root, const xcb_window_t window) {
 390     xcb_client_message_event_t ev;
 391     memset(&ev, '\0', sizeof(xcb_client_message_event_t));
 392
 393     _init_net_active_window(conn);
 394
 395     ev.response_type = XCB_CLIENT_MESSAGE;
 396     ev.window = window;
 397     ev.type = _NET_ACTIVE_WINDOW;
 398     ev.format = 32;
 399     ev.data.data32[0] = 2; /* 2 = pager */
 400
 401     xcb_send_event(conn, false, root, XCB_EVENT_MASK_SUBSTRUCTURE_REDIRECT, (char *)&ev);
 402     xcb_flush(conn);
 403 }