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