git.sur5r.net Git - i3/i3lock/blob - xcb.c

   1 /*
   2  * vim:ts=4:sw=4:expandtab
   3  *
   4  * © 2010-2012 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 xcb_connection_t *conn;
  24 xcb_screen_t *screen;
  25
  26 #define curs_invisible_width 8
  27 #define curs_invisible_height 8
  28
  29 static unsigned char curs_invisible_bits[] = {
  30  0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 };
  31
  32 #define curs_windows_width 11
  33 #define curs_windows_height 19
  34
  35 static unsigned char curs_windows_bits[] = {
  36  0xfe, 0x07, 0xfc, 0x07, 0xfa, 0x07, 0xf6, 0x07, 0xee, 0x07, 0xde, 0x07,
  37  0xbe, 0x07, 0x7e, 0x07, 0xfe, 0x06, 0xfe, 0x05, 0x3e, 0x00, 0xb6, 0x07,
  38  0x6a, 0x07, 0x6c, 0x07, 0xde, 0x06, 0xdf, 0x06, 0xbf, 0x05, 0xbf, 0x05,
  39  0x7f, 0x06 };
  40
  41 #define mask_windows_width 11
  42 #define mask_windows_height 19
  43
  44 static unsigned char mask_windows_bits[] = {
  45  0x01, 0x00, 0x03, 0x00, 0x07, 0x00, 0x0f, 0x00, 0x1f, 0x00, 0x3f, 0x00,
  46  0x7f, 0x00, 0xff, 0x00, 0xff, 0x01, 0xff, 0x03, 0xff, 0x07, 0x7f, 0x00,
  47  0xf7, 0x00, 0xf3, 0x00, 0xe1, 0x01, 0xe0, 0x01, 0xc0, 0x03, 0xc0, 0x03,
  48  0x80, 0x01 };
  49
  50 static uint32_t get_colorpixel(char *hex) {
  51     char strgroups[3][3] = {{hex[0], hex[1], '\0'},
  52                             {hex[2], hex[3], '\0'},
  53                             {hex[4], hex[5], '\0'}};
  54     uint32_t rgb16[3] = {(strtol(strgroups[0], NULL, 16)),
  55                          (strtol(strgroups[1], NULL, 16)),
  56                          (strtol(strgroups[2], NULL, 16))};
  57
  58     return (rgb16[0] << 16) + (rgb16[1] << 8) + rgb16[2];
  59 }
  60
  61 xcb_visualtype_t *get_root_visual_type(xcb_screen_t *screen) {
  62     xcb_visualtype_t *visual_type = NULL;
  63     xcb_depth_iterator_t depth_iter;
  64     xcb_visualtype_iterator_t visual_iter;
  65
  66     for (depth_iter = xcb_screen_allowed_depths_iterator(screen);
  67          depth_iter.rem;
  68          xcb_depth_next(&depth_iter)) {
  69
  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                       24,
 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     /* Map the window (= make it visible) */
 138     xcb_map_window(conn, win);
 139
 140     /* Raise window (put it on top) */
 141     values[0] = XCB_STACK_MODE_ABOVE;
 142     xcb_configure_window(conn, win, XCB_CONFIG_WINDOW_STACK_MODE, values);
 143
 144     return win;
 145 }
 146
 147 /*
 148  * Returns the mask for Mode_switch (to be used for looking up keysymbols by
 149  * keycode).
 150  *
 151  */
 152 uint32_t get_mod_mask(xcb_connection_t *conn, xcb_key_symbols_t *symbols, uint32_t keycode) {
 153     xcb_get_modifier_mapping_reply_t *modmap_r;
 154     xcb_keycode_t *modmap, kc;
 155     xcb_keycode_t *modeswitchcodes = xcb_key_symbols_get_keycode(symbols, keycode);
 156     if (modeswitchcodes == NULL)
 157         return 0;
 158
 159     modmap_r = xcb_get_modifier_mapping_reply(conn, xcb_get_modifier_mapping(conn), NULL);
 160     modmap = xcb_get_modifier_mapping_keycodes(modmap_r);
 161
 162     for (int i = 0; i < 8; i++)
 163         for (int j = 0; j < modmap_r->keycodes_per_modifier; j++) {
 164             kc = modmap[i * modmap_r->keycodes_per_modifier + j];
 165             for (xcb_keycode_t *ktest = modeswitchcodes; *ktest; ktest++) {
 166                 if (*ktest != kc)
 167                     continue;
 168
 169                 free(modeswitchcodes);
 170                 free(modmap_r);
 171                 return (1 << i);
 172             }
 173         }
 174
 175     free(modeswitchcodes);
 176     free(modmap_r);
 177     return 0;
 178 }
 179
 180 void dpms_turn_off_screen(xcb_connection_t *conn) {
 181     xcb_dpms_enable(conn);
 182     xcb_dpms_force_level(conn, XCB_DPMS_DPMS_MODE_OFF);
 183     xcb_flush(conn);
 184 }
 185
 186 /*
 187  * Repeatedly tries to grab pointer and keyboard (up to 1000 times).
 188  *
 189  */
 190 void grab_pointer_and_keyboard(xcb_connection_t *conn, xcb_screen_t *screen, xcb_cursor_t cursor) {
 191     xcb_grab_pointer_cookie_t pcookie;
 192     xcb_grab_pointer_reply_t *preply;
 193
 194     xcb_grab_keyboard_cookie_t kcookie;
 195     xcb_grab_keyboard_reply_t *kreply;
 196
 197     int tries = 10000;
 198
 199     while (tries-- > 0) {
 200         pcookie = xcb_grab_pointer(
 201             conn,
 202             false,               /* get all pointer events specified by the following mask */
 203             screen->root,        /* grab the root window */
 204             XCB_NONE,            /* which events to let through */
 205             XCB_GRAB_MODE_ASYNC, /* pointer events should continue as normal */
 206             XCB_GRAB_MODE_ASYNC, /* keyboard mode */
 207             XCB_NONE,            /* confine_to = in which window should the cursor stay */
 208             cursor,              /* we change the cursor to whatever the user wanted */
 209             XCB_CURRENT_TIME
 210         );
 211
 212         if ((preply = xcb_grab_pointer_reply(conn, pcookie, NULL)) &&
 213             preply->status == XCB_GRAB_STATUS_SUCCESS) {
 214             free(preply);
 215             break;
 216         }
 217
 218         /* Make this quite a bit slower */
 219         usleep(50);
 220     }
 221
 222     while (tries-- > 0) {
 223         kcookie = xcb_grab_keyboard(
 224             conn,
 225             true,                /* report events */
 226             screen->root,        /* grab the root window */
 227             XCB_CURRENT_TIME,
 228             XCB_GRAB_MODE_ASYNC, /* process events as normal, do not require sync */
 229             XCB_GRAB_MODE_ASYNC
 230         );
 231
 232         if ((kreply = xcb_grab_keyboard_reply(conn, kcookie, NULL)) &&
 233             kreply->status == XCB_GRAB_STATUS_SUCCESS) {
 234             free(kreply);
 235             break;
 236         }
 237
 238         /* Make this quite a bit slower */
 239         usleep(50);
 240     }
 241
 242     if (tries <= 0)
 243         errx(EXIT_FAILURE, "Cannot grab pointer/keyboard");
 244 }
 245
 246 xcb_cursor_t create_cursor(xcb_connection_t *conn, xcb_screen_t *screen, xcb_window_t win, int choice) {
 247     xcb_pixmap_t bitmap;
 248     xcb_pixmap_t mask;
 249     xcb_cursor_t cursor;
 250
 251     unsigned char* curs_bits;
 252     unsigned char* mask_bits;
 253     int curs_w, curs_h;
 254
 255     switch (choice) {
 256         case CURS_NONE:
 257             curs_bits = curs_invisible_bits;
 258             mask_bits = curs_invisible_bits;
 259             curs_w = curs_invisible_width;
 260             curs_h = curs_invisible_height;
 261             break;
 262         case CURS_WIN:
 263             curs_bits = curs_windows_bits;
 264             mask_bits = mask_windows_bits;
 265             curs_w = curs_windows_width;
 266             curs_h = curs_windows_height;
 267             break;
 268         case CURS_DEFAULT:
 269         default:
 270             return XCB_NONE; /* XCB_NONE is xcb's way of saying "don't change the cursor" */
 271     }
 272
 273     bitmap = xcb_create_pixmap_from_bitmap_data(conn,
 274                                                 win,
 275                                                 curs_bits,
 276                                                 curs_w,
 277                                                 curs_h,
 278                                                 1,
 279                                                 screen->white_pixel,
 280                                                 screen->black_pixel,
 281                                                 NULL);
 282
 283     mask = xcb_create_pixmap_from_bitmap_data(conn,
 284                                               win,
 285                                               mask_bits,
 286                                               curs_w,
 287                                               curs_h,
 288                                               1,
 289                                               screen->white_pixel,
 290                                               screen->black_pixel,
 291                                               NULL);
 292
 293     cursor = xcb_generate_id(conn);
 294
 295     xcb_create_cursor(conn,
 296                       cursor,
 297                       bitmap,
 298                       mask,
 299                       65535,65535,65535,
 300                       0,0,0,
 301                       0,0);
 302
 303     xcb_free_pixmap(conn, bitmap);
 304     xcb_free_pixmap(conn, mask);
 305
 306     return cursor;
 307 }