extras/keymap/keymap.c

   1 /*
   2  * keymap - dump keymap of an evdev device or set a new keymap from a file
   3  *
   4  * Based on keyfuzz by Lennart Poettering <mzqrovna@0pointer.net>
   5  * Adapted for udev-extras by Martin Pitt <martin.pitt@ubuntu.com>
   6  *
   7  * Copyright (C) 2006, Lennart Poettering
   8  * Copyright (C) 2009, Canonical Ltd.
   9  *
  10  * keymap is free software; you can redistribute it and/or modify it
  11  * under the terms of the GNU General Public License as published by
  12  * the Free Software Foundation; either version 2 of the License, or
  13  * (at your option) any later version.
  14  *
  15  * keymap is distributed in the hope that it will be useful, but
  16  * WITHOUT ANY WARRANTY; without even the implied warranty of
  17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  18  * General Public License for more details.
  19  *
  20  * You should have received a copy of the GNU General Public License
  21  * along with keymap; if not, write to the Free Software Foundation,
  22  * Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA.
  23  */
  24
  25 #include <stdio.h>
  26 #include <stdlib.h>
  27 #include <string.h>
  28 #include <stdint.h>
  29 #include <ctype.h>
  30 #include <unistd.h>
  31 #include <errno.h>
  32 #include <limits.h>
  33 #include <fcntl.h>
  34 #include <getopt.h>
  35 #include <sys/ioctl.h>
  36 #include <linux/limits.h>
  37 #include <linux/input.h>
  38
  39 const struct key* lookup_key (const char *str, unsigned int len);
  40
  41 #include "keys-from-name.h"
  42 #include "keys-to-name.h"
  43
  44 #define MAX_SCANCODES 1024
  45
  46 static int evdev_open(const char *dev)
  47 {
  48         int fd;
  49         char fn[PATH_MAX];
  50
  51         if (strncmp(dev, "/dev", 4) != 0) {
  52                 snprintf(fn, sizeof(fn), "/dev/%s", dev);
  53                 dev = fn;
  54         }
  55
  56         if ((fd = open(dev, O_RDWR)) < 0) {
  57                 fprintf(stderr, "error open('%s'): %m\n", dev);
  58                 return -1;
  59         }
  60         return fd;
  61 }
  62
  63 static int evdev_get_keycode(int fd, int scancode, int e)
  64 {
  65         int codes[2];
  66
  67         codes[0] = scancode;
  68         if (ioctl(fd, EVIOCGKEYCODE, codes) < 0) {
  69                 if (e && errno == EINVAL) {
  70                         return -2;
  71                 } else {
  72                         fprintf(stderr, "EVIOCGKEYCODE: %m\n");
  73                         return -1;
  74                 }
  75         }
  76         return codes[1];
  77 }
  78
  79 static int evdev_set_keycode(int fd, int scancode, int keycode)
  80 {
  81         int codes[2];
  82
  83         codes[0] = scancode;
  84         codes[1] = keycode;
  85
  86         if (ioctl(fd, EVIOCSKEYCODE, codes) < 0) {
  87                 fprintf(stderr, "EVIOCSKEYCODE: %m\n");
  88                 return -1;
  89         }
  90         return 0;
  91 }
  92
  93 static int evdev_driver_version(int fd, char *v, size_t l)
  94 {
  95         int version;
  96
  97         if (ioctl(fd, EVIOCGVERSION, &version)) {
  98                 fprintf(stderr, "EVIOCGVERSION: %m\n");
  99                 return -1;
 100         }
 101
 102         snprintf(v, l, "%i.%i.%i.", version >> 16, (version >> 8) & 0xff, version & 0xff);
 103         return 0;
 104 }
 105
 106 static int evdev_device_name(int fd, char *n, size_t l)
 107 {
 108         if (ioctl(fd, EVIOCGNAME(l), n) < 0) {
 109                 fprintf(stderr, "EVIOCGNAME: %m\n");
 110                 return -1;
 111         }
 112         return 0;
 113 }
 114
 115 /* Return a lower-case string with KEY_ prefix removed */
 116 static const char* format_keyname(const char* key) {
 117         static char result[101];
 118         const char* s;
 119         int len;
 120
 121         for (s = key+4, len = 0; *s && len < 100; ++len, ++s)
 122                 result[len] = tolower(*s);
 123         result[len] = '\0';
 124         return result;
 125 }
 126
 127 static int dump_table(int fd) {
 128         char version[256], name[256];
 129         int scancode, r = -1;
 130
 131         if (evdev_driver_version(fd, version, sizeof(version)) < 0)
 132                 goto fail;
 133
 134         if (evdev_device_name(fd, name, sizeof(name)) < 0)
 135                 goto fail;
 136
 137         printf("### evdev %s, driver '%s'\n", version, name);
 138
 139         r = 0;
 140         for (scancode = 0; scancode < MAX_SCANCODES; scancode++) {
 141                 int keycode;
 142
 143                 if ((keycode = evdev_get_keycode(fd, scancode, 1)) < 0) {
 144                         if (keycode != -2)
 145                                 r = -1;
 146                         break;
 147                 }
 148
 149                 if (keycode < KEY_MAX && key_names[keycode])
 150                         printf("0x%03x %s\n", scancode, format_keyname(key_names[keycode]));
 151                 else
 152                         printf("0x%03x 0x%03x\n", scancode, keycode);
 153         }
 154 fail:
 155         return r;
 156 }
 157
 158 static void set_key(int fd, const char* scancode_str, const char* keyname)
 159 {
 160         unsigned scancode;
 161         char *endptr;
 162         char t[105] = "KEY_UNKNOWN";
 163         const struct key *k;
 164
 165         scancode = (unsigned) strtol(scancode_str, &endptr, 0);
 166         if (*endptr != '\0') {
 167                 fprintf(stderr, "ERROR: Invalid scancode\n");
 168                 exit(1);
 169         }
 170
 171         snprintf(t, sizeof(t), "KEY_%s", keyname);
 172
 173         if (!(k = lookup_key(t, strlen(t)))) {
 174                 fprintf(stderr, "ERROR: Unknown key name '%s'\n", keyname);
 175                 exit(1);
 176         }
 177
 178         if (evdev_set_keycode(fd, scancode, k->id) < 0)
 179                 fprintf(stderr, "setting scancode 0x%2X to key code %i failed\n",
 180                         scancode, k->id);
 181         else
 182                 printf("setting scancode 0x%2X to key code %i\n",
 183                         scancode, k->id);
 184 }
 185
 186 static int merge_table(int fd, const char *filename) {
 187         int r = 0;
 188         int line = 0;
 189         FILE* f;
 190
 191         f = fopen(filename, "r");
 192         if (!f) {
 193                 perror(filename);
 194                 r = -1;
 195                 goto fail;
 196         }
 197
 198         while (!feof(f)) {
 199                 char s[256], *p;
 200                 int scancode, new_keycode, old_keycode;
 201
 202                 if (!fgets(s, sizeof(s), f))
 203                         break;
 204
 205                 line++;
 206                 p = s+strspn(s, "\t ");
 207                 if (*p == '#' || *p == '\n')
 208                         continue;
 209
 210                 if (sscanf(p, "%i %i", &scancode, &new_keycode) != 2) {
 211                         char t[105] = "KEY_UNKNOWN";
 212                         const struct key *k;
 213
 214                         if (sscanf(p, "%i %100s", &scancode, t+4) != 2) {
 215                                 fprintf(stderr, "WARNING: Parse failure at line %i, ignoring.\n", line);
 216                                 r = -1;
 217                                 continue;
 218                         }
 219
 220                         if (!(k = lookup_key(t, strlen(t)))) {
 221                                 fprintf(stderr, "WARNING: Unknown key '%s' at line %i, ignoring.\n", t, line);
 222                                 r = -1;
 223                                 continue;
 224                         }
 225
 226                         new_keycode = k->id;
 227                 }
 228
 229
 230                 if ((old_keycode = evdev_get_keycode(fd, scancode, 0)) < 0) {
 231                         r = -1;
 232                         goto fail;
 233                 }
 234
 235                 if (evdev_set_keycode(fd, scancode, new_keycode) < 0) {
 236                         r = -1;
 237                         goto fail;
 238                 }
 239
 240                 if (new_keycode != old_keycode)
 241                         fprintf(stderr, "Remapped scancode 0x%02x to 0x%02x (prior: 0x%02x)\n",
 242                                 scancode, new_keycode, old_keycode);
 243         }
 244 fail:
 245         return r;
 246 }
 247
 248 static const char* default_keymap_path(const char* path)
 249 {
 250         static char result[PATH_MAX];
 251
 252         /* If keymap file is given without a path, assume udev directory; must end with '/' * */
 253         if (!strchr(path, '/')) {
 254                 snprintf(result, sizeof(result), "%s%s", LIBEXECDIR "/keymaps/", path);
 255                 return result;
 256         }
 257         return path;
 258 }
 259
 260 /* read one event; return 1 if valid */
 261 static int read_event(int fd, struct input_event* ev)
 262 {
 263         int ret;
 264         ret = read(fd, ev, sizeof(struct input_event));
 265
 266         if (ret < 0) {
 267                 perror("read");
 268                 return 0;
 269         }
 270         if (ret != sizeof(struct input_event)) {
 271                 fprintf(stderr, "did not get enough data for event struct, aborting\n");
 272                 return 0;
 273         }
 274
 275         return 1;
 276 }
 277
 278 static void print_key(uint32_t scancode, uint16_t keycode, int has_scan, int has_key)
 279 {
 280         const char *keyname;
 281
 282         /* ignore key release events */
 283         if (has_key == 1)
 284                 return;
 285
 286         if (has_key == 0 && has_scan != 0) {
 287                 fprintf(stderr, "got scan code event 0x%02X without a key code event\n",
 288                         scancode);
 289                 return;
 290         }
 291
 292         if (has_scan != 0)
 293                 printf("scan code: 0x%02X   ", scancode);
 294         else
 295                 printf("(no scan code received)  ");
 296
 297         keyname = key_names[keycode];
 298         if (keyname != NULL)
 299                 printf("key code: %s\n", format_keyname(keyname));
 300         else
 301                 printf("key code: %03X\n", keycode);
 302 }
 303
 304 static void interactive(int fd)
 305 {
 306         struct input_event ev;
 307         uint32_t last_scan = 0;
 308         uint16_t last_key = 0;
 309         int has_scan; /* boolean */
 310         int has_key; /* 0: none, 1: release, 2: press */
 311
 312         /* grab input device */
 313         ioctl(fd, EVIOCGRAB, 1);
 314         puts("Press ESC to finish, or Control-C if this device is not your primary keyboard");
 315
 316         has_scan = has_key = 0;
 317         while (read_event(fd, &ev)) {
 318                 /* Drivers usually send the scan code first, then the key code,
 319                  * then a SYN. Some drivers (like thinkpad_acpi) send the key
 320                  * code first, and some drivers might not send SYN events, so
 321                  * keep a robust state machine which can deal with any of those
 322                  */
 323
 324                 if (ev.type == EV_MSC && ev.code == MSC_SCAN) {
 325                         if (has_scan) {
 326                                 fputs("driver did not send SYN event in between key events; previous event:\n",
 327                                       stderr);
 328                                 print_key(last_scan, last_key, has_scan, has_key);
 329                                 has_key = 0;
 330                         }
 331
 332                         last_scan = ev.value;
 333                         has_scan = 1;
 334                         /*printf("--- got scan %u; has scan %i key %i\n", last_scan, has_scan, has_key); */
 335                 }
 336                 else if (ev.type == EV_KEY) {
 337                         if (has_key) {
 338                                 fputs("driver did not send SYN event in between key events; previous event:\n",
 339                                       stderr);
 340                                 print_key(last_scan, last_key, has_scan, has_key);
 341                                 has_scan = 0;
 342                         }
 343
 344                         last_key = ev.code;
 345                         has_key = 1 + ev.value;
 346                         /*printf("--- got key %hu; has scan %i key %i\n", last_key, has_scan, has_key);*/
 347
 348                         /* Stop on ESC */
 349                         if (ev.code == KEY_ESC && ev.value == 0)
 350                                 break;
 351                 }
 352                 else if (ev.type == EV_SYN) {
 353                         /*printf("--- got SYN; has scan %i key %i\n", has_scan, has_key);*/
 354                         print_key(last_scan, last_key, has_scan, has_key);
 355
 356                         has_scan = has_key = 0;
 357                 }
 358
 359         }
 360
 361         /* release input device */
 362         ioctl(fd, EVIOCGRAB, 0);
 363 }
 364
 365 static void help(int error)
 366 {
 367         const char* h = "Usage: keymap <event device> [<map file>]\n"
 368                         "       keymap <event device> scancode keyname [...]\n"
 369                         "       keymap -i <event device>\n";
 370         if (error) {
 371                 fputs(h, stderr);
 372                 exit(2);
 373         } else {
 374                 fputs(h, stdout);
 375                 exit(0);
 376         }
 377 }
 378
 379 int main(int argc, char **argv)
 380 {
 381         static const struct option options[] = {
 382                 { "help", no_argument, NULL, 'h' },
 383                 { "interactive", no_argument, NULL, 'i' },
 384                 {}
 385         };
 386         int fd = -1;
 387         int opt_interactive = 0;
 388         int i;
 389
 390         while (1) {
 391                 int option;
 392
 393                 option = getopt_long(argc, argv, "hi", options, NULL);
 394                 if (option == -1)
 395                         break;
 396
 397                 switch (option) {
 398                 case 'h':
 399                         help(0);
 400
 401                 case 'i':
 402                         opt_interactive = 1;
 403                         break;
 404                 default:
 405                         return 1;
 406                 }
 407         }
 408
 409         if (argc < optind+1)
 410                 help (1);
 411
 412         if ((fd = evdev_open(argv[optind])) < 0)
 413                 return 3;
 414
 415         /* one argument (device): dump or interactive */
 416         if (argc == optind+1) {
 417                 if (opt_interactive)
 418                         interactive(fd);
 419                 else
 420                         dump_table(fd);
 421                 return 0;
 422         }
 423
 424         /* two arguments (device, mapfile): set map file */
 425         if (argc == optind+2) {
 426                 merge_table(fd, default_keymap_path(argv[optind+1]));
 427                 return 0;
 428         }
 429
 430         /* more arguments (device, scancode/keyname pairs): set keys directly */
 431         if ((argc - optind - 1) % 2 == 0) {
 432                 for (i = optind+1; i < argc; i += 2)
 433                         set_key(fd, argv[i], argv[i+1]);
 434                 return 0;
 435         }
 436
 437         /* invalid number of arguments */
 438         help(1);
 439         return 1; /* not reached */
 440 }