tools/interactive-x11.c

   1 /*
   2  * Copyright © 2013 Ran Benita <ran234@gmail.com>
   3  *
   4  * Permission is hereby granted, free of charge, to any person obtaining a
   5  * copy of this software and associated documentation files (the "Software"),
   6  * to deal in the Software without restriction, including without limitation
   7  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
   8  * and/or sell copies of the Software, and to permit persons to whom the
   9  * Software is furnished to do so, subject to the following conditions:
  10  *
  11  * The above copyright notice and this permission notice (including the next
  12  * paragraph) shall be included in all copies or substantial portions of the
  13  * Software.
  14  *
  15  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  16  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  17  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
  18  * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  19  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  20  * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
  21  * DEALINGS IN THE SOFTWARE.
  22  */
  23
  24 #include "config.h"
  25
  26 #include <locale.h>
  27 #include <stdbool.h>
  28 #include <stdlib.h>
  29 #include <string.h>
  30
  31 #include <xcb/xkb.h>
  32
  33 #include "xkbcommon/xkbcommon-x11.h"
  34 #include "tools-common.h"
  35
  36 /*
  37  * Note: This program only handles the core keyboard device for now.
  38  * It should be straigtforward to change struct keyboard to a list of
  39  * keyboards with device IDs, as in tools/interactive-evdev.c. This would
  40  * require:
  41  *
  42  * - Initially listing the keyboard devices.
  43  * - Listening to device changes.
  44  * - Matching events to their devices.
  45  *
  46  * XKB itself knows about xinput1 devices, and most requests and events are
  47  * device-specific.
  48  *
  49  * In order to list the devices and react to changes, you need xinput1/2.
  50  * You also need xinput for the key press/release event, since the core
  51  * protocol key press event does not carry a device ID to match on.
  52  */
  53
  54 struct keyboard {
  55     xcb_connection_t *conn;
  56     uint8_t first_xkb_event;
  57     struct xkb_context *ctx;
  58
  59     struct xkb_keymap *keymap;
  60     struct xkb_state *state;
  61     int32_t device_id;
  62 };
  63
  64 static bool terminate;
  65
  66 static int
  67 select_xkb_events_for_device(xcb_connection_t *conn, int32_t device_id)
  68 {
  69     enum {
  70         required_events =
  71             (XCB_XKB_EVENT_TYPE_NEW_KEYBOARD_NOTIFY |
  72              XCB_XKB_EVENT_TYPE_MAP_NOTIFY |
  73              XCB_XKB_EVENT_TYPE_STATE_NOTIFY),
  74
  75         required_nkn_details =
  76             (XCB_XKB_NKN_DETAIL_KEYCODES),
  77
  78         required_map_parts =
  79             (XCB_XKB_MAP_PART_KEY_TYPES |
  80              XCB_XKB_MAP_PART_KEY_SYMS |
  81              XCB_XKB_MAP_PART_MODIFIER_MAP |
  82              XCB_XKB_MAP_PART_EXPLICIT_COMPONENTS |
  83              XCB_XKB_MAP_PART_KEY_ACTIONS |
  84              XCB_XKB_MAP_PART_VIRTUAL_MODS |
  85              XCB_XKB_MAP_PART_VIRTUAL_MOD_MAP),
  86
  87         required_state_details =
  88             (XCB_XKB_STATE_PART_MODIFIER_BASE |
  89              XCB_XKB_STATE_PART_MODIFIER_LATCH |
  90              XCB_XKB_STATE_PART_MODIFIER_LOCK |
  91              XCB_XKB_STATE_PART_GROUP_BASE |
  92              XCB_XKB_STATE_PART_GROUP_LATCH |
  93              XCB_XKB_STATE_PART_GROUP_LOCK),
  94     };
  95
  96     static const xcb_xkb_select_events_details_t details = {
  97         .affectNewKeyboard = required_nkn_details,
  98         .newKeyboardDetails = required_nkn_details,
  99         .affectState = required_state_details,
 100         .stateDetails = required_state_details,
 101     };
 102
 103     xcb_void_cookie_t cookie =
 104         xcb_xkb_select_events_aux_checked(conn,
 105                                           device_id,
 106                                           required_events,    /* affectWhich */
 107                                           0,                  /* clear */
 108                                           0,                  /* selectAll */
 109                                           required_map_parts, /* affectMap */
 110                                           required_map_parts, /* map */
 111                                           &details);          /* details */
 112
 113     xcb_generic_error_t *error = xcb_request_check(conn, cookie);
 114     if (error) {
 115         free(error);
 116         return -1;
 117     }
 118
 119     return 0;
 120 }
 121
 122 static int
 123 update_keymap(struct keyboard *kbd)
 124 {
 125     struct xkb_keymap *new_keymap;
 126     struct xkb_state *new_state;
 127
 128     new_keymap = xkb_x11_keymap_new_from_device(kbd->ctx, kbd->conn,
 129                                                 kbd->device_id,
 130                                                 XKB_KEYMAP_COMPILE_NO_FLAGS);
 131     if (!new_keymap)
 132         goto err_out;
 133
 134     new_state = xkb_x11_state_new_from_device(new_keymap, kbd->conn,
 135                                               kbd->device_id);
 136     if (!new_state)
 137         goto err_keymap;
 138
 139     if (kbd->keymap)
 140         printf("Keymap updated!\n");
 141
 142     xkb_state_unref(kbd->state);
 143     xkb_keymap_unref(kbd->keymap);
 144     kbd->keymap = new_keymap;
 145     kbd->state = new_state;
 146     return 0;
 147
 148 err_keymap:
 149     xkb_keymap_unref(new_keymap);
 150 err_out:
 151     return -1;
 152 }
 153
 154 static int
 155 init_kbd(struct keyboard *kbd, xcb_connection_t *conn, uint8_t first_xkb_event,
 156          int32_t device_id, struct xkb_context *ctx)
 157 {
 158     int ret;
 159
 160     kbd->conn = conn;
 161     kbd->first_xkb_event = first_xkb_event;
 162     kbd->ctx = ctx;
 163     kbd->keymap = NULL;
 164     kbd->state = NULL;
 165     kbd->device_id = device_id;
 166
 167     ret = update_keymap(kbd);
 168     if (ret)
 169         goto err_out;
 170
 171     ret = select_xkb_events_for_device(conn, device_id);
 172     if (ret)
 173         goto err_state;
 174
 175     return 0;
 176
 177 err_state:
 178     xkb_state_unref(kbd->state);
 179     xkb_keymap_unref(kbd->keymap);
 180 err_out:
 181     return -1;
 182 }
 183
 184 static void
 185 deinit_kbd(struct keyboard *kbd)
 186 {
 187     xkb_state_unref(kbd->state);
 188     xkb_keymap_unref(kbd->keymap);
 189 }
 190
 191 static void
 192 process_xkb_event(xcb_generic_event_t *gevent, struct keyboard *kbd)
 193 {
 194     union xkb_event {
 195         struct {
 196             uint8_t response_type;
 197             uint8_t xkbType;
 198             uint16_t sequence;
 199             xcb_timestamp_t time;
 200             uint8_t deviceID;
 201         } any;
 202         xcb_xkb_new_keyboard_notify_event_t new_keyboard_notify;
 203         xcb_xkb_map_notify_event_t map_notify;
 204         xcb_xkb_state_notify_event_t state_notify;
 205     } *event = (union xkb_event *) gevent;
 206
 207     if (event->any.deviceID != kbd->device_id)
 208         return;
 209
 210     /*
 211      * XkbNewKkdNotify and XkbMapNotify together capture all sorts of keymap
 212      * updates (e.g. xmodmap, xkbcomp, setxkbmap), with minimal redundent
 213      * recompilations.
 214      */
 215     switch (event->any.xkbType) {
 216     case XCB_XKB_NEW_KEYBOARD_NOTIFY:
 217         if (event->new_keyboard_notify.changed & XCB_XKB_NKN_DETAIL_KEYCODES)
 218             update_keymap(kbd);
 219         break;
 220
 221     case XCB_XKB_MAP_NOTIFY:
 222         update_keymap(kbd);
 223         break;
 224
 225     case XCB_XKB_STATE_NOTIFY:
 226         xkb_state_update_mask(kbd->state,
 227                               event->state_notify.baseMods,
 228                               event->state_notify.latchedMods,
 229                               event->state_notify.lockedMods,
 230                               event->state_notify.baseGroup,
 231                               event->state_notify.latchedGroup,
 232                               event->state_notify.lockedGroup);
 233         break;
 234     }
 235 }
 236
 237 static void
 238 process_event(xcb_generic_event_t *gevent, struct keyboard *kbd)
 239 {
 240     switch (gevent->response_type) {
 241     case XCB_KEY_PRESS: {
 242         xcb_key_press_event_t *event = (xcb_key_press_event_t *) gevent;
 243         xkb_keycode_t keycode = event->detail;
 244
 245         tools_print_keycode_state(kbd->state, NULL, keycode,
 246                                   XKB_CONSUMED_MODE_XKB,
 247                                   PRINT_ALL_FIELDS);
 248
 249         /* Exit on ESC. */
 250         if (xkb_state_key_get_one_sym(kbd->state, keycode) == XKB_KEY_Escape)
 251             terminate = true;
 252         break;
 253     }
 254     default:
 255         if (gevent->response_type == kbd->first_xkb_event)
 256             process_xkb_event(gevent, kbd);
 257         break;
 258     }
 259 }
 260
 261 static int
 262 loop(xcb_connection_t *conn, struct keyboard *kbd)
 263 {
 264     while (!terminate) {
 265         xcb_generic_event_t *event;
 266
 267         switch (xcb_connection_has_error(conn)) {
 268         case 0:
 269             break;
 270         case XCB_CONN_ERROR:
 271             fprintf(stderr,
 272                     "Closed connection to X server: connection error\n");
 273             return -1;
 274         case XCB_CONN_CLOSED_EXT_NOTSUPPORTED:
 275             fprintf(stderr,
 276                     "Closed connection to X server: extension not supported\n");
 277             return -1;
 278         default:
 279             fprintf(stderr,
 280                     "Closed connection to X server: error code %d\n",
 281                     xcb_connection_has_error(conn));
 282             return -1;
 283         }
 284
 285         event = xcb_wait_for_event(conn);
 286         if (!event) {
 287             continue;
 288         }
 289
 290         process_event(event, kbd);
 291         free(event);
 292     }
 293
 294     return 0;
 295 }
 296
 297 static int
 298 create_capture_window(xcb_connection_t *conn)
 299 {
 300     xcb_generic_error_t *error;
 301     xcb_void_cookie_t cookie;
 302     xcb_screen_t *screen =
 303         xcb_setup_roots_iterator(xcb_get_setup(conn)).data;
 304     xcb_window_t window = xcb_generate_id(conn);
 305     uint32_t values[2] = {
 306         screen->white_pixel,
 307         XCB_EVENT_MASK_KEY_PRESS,
 308     };
 309
 310     cookie = xcb_create_window_checked(conn, XCB_COPY_FROM_PARENT,
 311                                        window, screen->root,
 312                                        10, 10, 100, 100, 1,
 313                                        XCB_WINDOW_CLASS_INPUT_OUTPUT,
 314                                        screen->root_visual,
 315                                        XCB_CW_BACK_PIXEL | XCB_CW_EVENT_MASK,
 316                                        values);
 317     if ((error = xcb_request_check(conn, cookie)) != NULL) {
 318         free(error);
 319         return -1;
 320     }
 321
 322     cookie = xcb_map_window_checked(conn, window);
 323     if ((error = xcb_request_check(conn, cookie)) != NULL) {
 324         free(error);
 325         return -1;
 326     }
 327
 328     return 0;
 329 }
 330
 331 int
 332 main(int argc, char *argv[])
 333 {
 334     int ret;
 335     xcb_connection_t *conn;
 336     uint8_t first_xkb_event;
 337     int32_t core_kbd_device_id;
 338     struct xkb_context *ctx;
 339     struct keyboard core_kbd;
 340
 341     if (argc != 1) {
 342         ret = strcmp(argv[1], "--help");
 343         fprintf(ret ? stderr : stdout, "Usage: %s [--help]\n", argv[0]);
 344         if (ret)
 345             fprintf(stderr, "unrecognized option: %s\n", argv[1]);
 346         return ret ? EXIT_INVALID_USAGE : EXIT_SUCCESS;
 347     }
 348
 349     setlocale(LC_ALL, "");
 350
 351     conn = xcb_connect(NULL, NULL);
 352     if (!conn || xcb_connection_has_error(conn)) {
 353         fprintf(stderr, "Couldn't connect to X server: error code %d\n",
 354                 conn ? xcb_connection_has_error(conn) : -1);
 355         ret = -1;
 356         goto err_out;
 357     }
 358
 359     ret = xkb_x11_setup_xkb_extension(conn,
 360                                       XKB_X11_MIN_MAJOR_XKB_VERSION,
 361                                       XKB_X11_MIN_MINOR_XKB_VERSION,
 362                                       XKB_X11_SETUP_XKB_EXTENSION_NO_FLAGS,
 363                                       NULL, NULL, &first_xkb_event, NULL);
 364     if (!ret) {
 365         fprintf(stderr, "Couldn't setup XKB extension\n");
 366         goto err_conn;
 367     }
 368
 369     ctx = xkb_context_new(XKB_CONTEXT_NO_FLAGS);
 370     if (!ctx) {
 371         ret = -1;
 372         fprintf(stderr, "Couldn't create xkb context\n");
 373         goto err_conn;
 374     }
 375
 376     core_kbd_device_id = xkb_x11_get_core_keyboard_device_id(conn);
 377     if (core_kbd_device_id == -1) {
 378         ret = -1;
 379         fprintf(stderr, "Couldn't find core keyboard device\n");
 380         goto err_ctx;
 381     }
 382
 383     ret = init_kbd(&core_kbd, conn, first_xkb_event, core_kbd_device_id, ctx);
 384     if (ret) {
 385         fprintf(stderr, "Couldn't initialize core keyboard device\n");
 386         goto err_ctx;
 387     }
 388
 389     ret = create_capture_window(conn);
 390     if (ret) {
 391         fprintf(stderr, "Couldn't create a capture window\n");
 392         goto err_core_kbd;
 393     }
 394
 395     tools_disable_stdin_echo();
 396     ret = loop(conn, &core_kbd);
 397     tools_enable_stdin_echo();
 398
 399 err_core_kbd:
 400     deinit_kbd(&core_kbd);
 401 err_ctx:
 402     xkb_context_unref(ctx);
 403 err_conn:
 404     xcb_disconnect(conn);
 405 err_out:
 406     exit(ret == 0 ? EXIT_SUCCESS : EXIT_FAILURE);
 407 }