diff options
Diffstat (limited to 'tools/gsr-global-hotkeys')
-rw-r--r-- | tools/gsr-global-hotkeys/hotplug.c | 8 | ||||
-rw-r--r-- | tools/gsr-global-hotkeys/keyboard_event.c | 407 | ||||
-rw-r--r-- | tools/gsr-global-hotkeys/keyboard_event.h | 78 | ||||
-rw-r--r-- | tools/gsr-global-hotkeys/main.c | 126 |
4 files changed, 332 insertions, 287 deletions
diff --git a/tools/gsr-global-hotkeys/hotplug.c b/tools/gsr-global-hotkeys/hotplug.c index ba3ef9c..5ea2978 100644 --- a/tools/gsr-global-hotkeys/hotplug.c +++ b/tools/gsr-global-hotkeys/hotplug.c @@ -22,7 +22,7 @@ bool hotplug_event_init(hotplug_event *self) { const int fd = socket(PF_NETLINK, SOCK_DGRAM, NETLINK_KOBJECT_UEVENT); if(fd == -1) - return false; /* Not root user */ + return false; if(bind(fd, (void*)&nls, sizeof(struct sockaddr_nl))) { close(fd); @@ -56,19 +56,21 @@ static void hotplug_event_parse_netlink_data(hotplug_event *self, const char *li if(strcmp(line, "SUBSYSTEM=input") == 0) self->subsystem_is_input = true; - if(self->subsystem_is_input && strncmp(line, "DEVNAME=", 8) == 0) + if(self->subsystem_is_input && strncmp(line, "DEVNAME=", 8) == 0) { callback(line+8, userdata); + self->event_is_add = false; + } } } /* Netlink uevent structure is documented here: https://web.archive.org/web/20160127215232/https://www.kernel.org/doc/pending/hotplug.txt */ void hotplug_event_process_event_data(hotplug_event *self, int fd, hotplug_device_added_callback callback, void *userdata) { const int bytes_read = read(fd, self->event_data, sizeof(self->event_data)); - int data_index = 0; if(bytes_read <= 0) return; /* Hotplug data ends with a newline and a null terminator */ + int data_index = 0; while(data_index < bytes_read) { hotplug_event_parse_netlink_data(self, self->event_data + data_index, callback, userdata); data_index += strlen(self->event_data + data_index) + 1; /* Skip null terminator as well */ diff --git a/tools/gsr-global-hotkeys/keyboard_event.c b/tools/gsr-global-hotkeys/keyboard_event.c index bcfd616..b8d94fd 100644 --- a/tools/gsr-global-hotkeys/keyboard_event.c +++ b/tools/gsr-global-hotkeys/keyboard_event.c @@ -11,7 +11,7 @@ #include <fcntl.h> #include <unistd.h> #include <dirent.h> -#include <sys/poll.h> +#include <poll.h> /* LINUX */ #include <linux/input.h> @@ -25,24 +25,6 @@ #define KEY_STATES_SIZE (KEY_MAX/8 + 1) -#define KEYCODE_TO_XKB_KEYCODE(key) ((key) + 8) - -#define XK_Shift_L 0xffe1 /* Left shift */ -#define XK_Shift_R 0xffe2 /* Right shift */ -#define XK_Control_L 0xffe3 /* Left control */ -#define XK_Control_R 0xffe4 /* Right control */ -#define XK_Alt_L 0xffe9 /* Left alt */ -#define XK_Alt_R 0xffea /* Right alt */ -#define XK_Super_L 0xffeb /* Left super */ -#define XK_Super_R 0xffec /* Right super */ - -#define XK_z 0x007a -#define XK_F7 0xffc4 -#define XK_F8 0xffc5 -#define XK_F9 0xffc6 -#define XK_F10 0xffc7 -#define XK_F11 0xffc8 - static inline int count_num_bits_set(unsigned char c) { int n = 0; n += (c & 1); @@ -135,36 +117,44 @@ static void keyboard_event_process_key_state_change(keyboard_event *self, struct } } -static uint32_t keycode_to_keysym(keyboard_event *self, uint16_t keycode) { - const unsigned long xkb_keycode = KEYCODE_TO_XKB_KEYCODE(keycode); - if(self->x_context.display && self->x_context.XKeycodeToKeysym && xkb_keycode <= 255) - return self->x_context.XKeycodeToKeysym(self->x_context.display, xkb_keycode, 0); +/* Return true if a global hotkey is assigned to the key combination */ +static bool keyboard_event_on_key_pressed(keyboard_event *self, const struct input_event *event, uint32_t modifiers) { + if(event->value != KEYBOARD_BUTTON_PRESSED) + return false; + + bool global_hotkey_match = false; + for(int i = 0; i < self->num_global_hotkeys; ++i) { + if(event->code == self->global_hotkeys[i].key && modifiers == self->global_hotkeys[i].modifiers) { + puts(self->global_hotkeys[i].action); + fflush(stdout); + global_hotkey_match = true; + } + } + return global_hotkey_match; +} + +static inline uint32_t set_bit(uint32_t value, uint32_t bit_flag, bool set) { + if(set) + return value | bit_flag; else - return 0; + return value & ~bit_flag; } -/* TODO: Support more keys when needed */ -static uint32_t keysym_to_keycode(uint32_t keysym) { - switch(keysym) { - case XK_Control_L: return KEY_LEFTCTRL; - case XK_Shift_L: return KEY_LEFTSHIFT; - case XK_Alt_L: return KEY_LEFTALT; - case XK_Super_L: return KEY_LEFTMETA; - case XK_Control_R: return KEY_RIGHTCTRL; - case XK_Shift_R: return KEY_RIGHTSHIFT; - case XK_Alt_R: return KEY_RIGHTALT; - case XK_Super_R: return KEY_RIGHTMETA; - case XK_z: return KEY_Z; - case XK_F7: return KEY_F7; - case XK_F8: return KEY_F8; - case XK_F9: return KEY_F9; - case XK_F10: return KEY_F10; - case XK_F11: return KEY_F11; - default: return 0; +static uint32_t keycode_to_modifier_bit(uint32_t keycode) { + switch(keycode) { + case KEY_LEFTSHIFT: return KEYBOARD_MODKEY_LSHIFT; + case KEY_RIGHTSHIFT: return KEYBOARD_MODKEY_RSHIFT; + case KEY_LEFTCTRL: return KEYBOARD_MODKEY_LCTRL; + case KEY_RIGHTCTRL: return KEYBOARD_MODKEY_RCTRL; + case KEY_LEFTALT: return KEYBOARD_MODKEY_LALT; + case KEY_RIGHTALT: return KEYBOARD_MODKEY_RALT; + case KEY_LEFTMETA: return KEYBOARD_MODKEY_LSUPER; + case KEY_RIGHTMETA: return KEYBOARD_MODKEY_RSUPER; } + return 0; } -static void keyboard_event_process_input_event_data(keyboard_event *self, event_extra_data *extra_data, int fd, key_callback callback, void *userdata) { +static void keyboard_event_process_input_event_data(keyboard_event *self, event_extra_data *extra_data, int fd) { struct input_event event; if(read(fd, &event, sizeof(event)) != sizeof(event)) { fprintf(stderr, "Error: failed to read input event data\n"); @@ -183,62 +173,12 @@ static void keyboard_event_process_input_event_data(keyboard_event *self, event_ if(event.type == EV_KEY) { keyboard_event_process_key_state_change(self, event, extra_data, fd); - - uint32_t keycode = event.code; - const uint32_t keysym = keycode_to_keysym(self, event.code); - if(keysym) - keycode = keysym_to_keycode(keysym); - - switch(keycode) { - case KEY_LEFTSHIFT: - self->lshift_button_state = event.value >= 1 ? KEYBOARD_BUTTON_PRESSED : KEYBOARD_BUTTON_RELEASED; - break; - case KEY_RIGHTSHIFT: - self->rshift_button_state = event.value >= 1 ? KEYBOARD_BUTTON_PRESSED : KEYBOARD_BUTTON_RELEASED; - break; - case KEY_LEFTCTRL: - self->lctrl_button_state = event.value >= 1 ? KEYBOARD_BUTTON_PRESSED : KEYBOARD_BUTTON_RELEASED; - break; - case KEY_RIGHTCTRL: - self->rctrl_button_state = event.value >= 1 ? KEYBOARD_BUTTON_PRESSED : KEYBOARD_BUTTON_RELEASED; - break; - case KEY_LEFTALT: - self->lalt_button_state = event.value >= 1 ? KEYBOARD_BUTTON_PRESSED : KEYBOARD_BUTTON_RELEASED; - break; - case KEY_RIGHTALT: - self->ralt_button_state = event.value >= 1 ? KEYBOARD_BUTTON_PRESSED : KEYBOARD_BUTTON_RELEASED; - break; - case KEY_LEFTMETA: - self->lmeta_button_state = event.value >= 1 ? KEYBOARD_BUTTON_PRESSED : KEYBOARD_BUTTON_RELEASED; - break; - case KEY_RIGHTMETA: - self->rmeta_button_state = event.value >= 1 ? KEYBOARD_BUTTON_PRESSED : KEYBOARD_BUTTON_RELEASED; - break; - default: { - const bool shift_pressed = self->lshift_button_state == KEYBOARD_BUTTON_PRESSED || self->rshift_button_state == KEYBOARD_BUTTON_PRESSED; - const bool ctrl_pressed = self->lctrl_button_state == KEYBOARD_BUTTON_PRESSED || self->rctrl_button_state == KEYBOARD_BUTTON_PRESSED; - const bool lalt_pressed = self->lalt_button_state == KEYBOARD_BUTTON_PRESSED; - const bool ralt_pressed = self->ralt_button_state == KEYBOARD_BUTTON_PRESSED; - const bool meta_pressed = self->lmeta_button_state == KEYBOARD_BUTTON_PRESSED || self->rmeta_button_state == KEYBOARD_BUTTON_PRESSED; - //fprintf(stderr, "pressed key: %d, state: %d, shift: %s, ctrl: %s, alt: %s, meta: %s\n", keycode, event.value, - // shift_pressed ? "yes" : "no", ctrl_pressed ? "yes" : "no", alt_pressed ? "yes" : "no", meta_pressed ? "yes" : "no"); - uint32_t modifiers = 0; - if(shift_pressed) - modifiers |= KEYBOARD_MODKEY_SHIFT; - if(ctrl_pressed) - modifiers |= KEYBOARD_MODKEY_CTRL; - if(lalt_pressed) - modifiers |= KEYBOARD_MODKEY_LALT; - if(ralt_pressed) - modifiers |= KEYBOARD_MODKEY_RALT; - if(meta_pressed) - modifiers |= KEYBOARD_MODKEY_SUPER; - - if(!callback(keycode, modifiers, event.value, userdata)) - return; - - break; - } + const uint32_t modifier_bit = keycode_to_modifier_bit(event.code); + if(modifier_bit == 0) { + if(keyboard_event_on_key_pressed(self, &event, self->modifier_button_states)) + return; + } else { + self->modifier_button_states = set_bit(self->modifier_button_states, modifier_bit, event.value >= 1); } } @@ -249,6 +189,36 @@ static void keyboard_event_process_input_event_data(keyboard_event *self, event_ } } +/* Retarded linux takes very long time to close /dev/input/eventN files, even though they are virtual and opened read-only */ +static void* keyboard_event_close_fds_callback(void *userdata) { + keyboard_event *self = userdata; + while(self->running) { + pthread_mutex_lock(&self->close_dev_input_mutex); + for(int i = 0; i < self->num_close_fds; ++i) { + close(self->close_fds[i]); + } + self->num_close_fds = 0; + pthread_mutex_unlock(&self->close_dev_input_mutex); + + usleep(100 * 1000); /* 100 milliseconds */ + } + return NULL; +} + +static bool keyboard_event_try_add_close_fd(keyboard_event *self, int fd) { + bool success = false; + pthread_mutex_lock(&self->close_dev_input_mutex); + if(self->num_close_fds < MAX_CLOSE_FDS) { + self->close_fds[self->num_close_fds] = fd; + ++self->num_close_fds; + success = true; + } else { + success = false; + } + pthread_mutex_unlock(&self->close_dev_input_mutex); + return success; +} + /* Returns -1 if invalid format. Expected |dev_input_filepath| to be in format /dev/input/eventN */ static int get_dev_input_id_from_filepath(const char *dev_input_filepath) { if(strncmp(dev_input_filepath, "/dev/input/event", 16) != 0) @@ -356,7 +326,10 @@ static bool keyboard_event_try_add_device_if_keyboard(keyboard_event *self, cons } } - close(fd); + if(!keyboard_event_try_add_close_fd(self, fd)) { + fprintf(stderr, "Error: failed to add immediately, closing now\n"); + close(fd); + } return false; } @@ -458,12 +431,19 @@ static int setup_virtual_keyboard_input(const char *name) { return fd; } -bool keyboard_event_init(keyboard_event *self, bool poll_stdout_error, bool exclusive_grab, keyboard_grab_type grab_type, x11_context x_context) { +bool keyboard_event_init(keyboard_event *self, bool exclusive_grab, keyboard_grab_type grab_type) { memset(self, 0, sizeof(*self)); - self->stdout_event_index = -1; + self->stdin_event_index = -1; self->hotplug_event_index = -1; self->grab_type = grab_type; - self->x_context = x_context; + self->running = true; + + pthread_mutex_init(&self->close_dev_input_mutex, NULL); + if(pthread_create(&self->close_dev_input_fds_thread, NULL, keyboard_event_close_fds_callback, self) != 0) { + self->close_dev_input_fds_thread = 0; + fprintf(stderr, "Error: failed to create close fds thread\n"); + return false; + } if(exclusive_grab) { self->uinput_fd = setup_virtual_keyboard_input(GSR_UI_VIRTUAL_KEYBOARD_NAME); @@ -471,23 +451,21 @@ bool keyboard_event_init(keyboard_event *self, bool poll_stdout_error, bool excl fprintf(stderr, "Warning: failed to setup virtual keyboard input for exclusive grab. The focused application will receive keys used for global hotkeys\n"); } - if(poll_stdout_error) { - self->event_polls[self->num_event_polls] = (struct pollfd) { - .fd = STDOUT_FILENO, - .events = 0, - .revents = 0 - }; + self->event_polls[self->num_event_polls] = (struct pollfd) { + .fd = STDIN_FILENO, + .events = POLLIN, + .revents = 0 + }; - self->event_extra_data[self->num_event_polls] = (event_extra_data) { - .dev_input_id = -1, - .grabbed = false, - .key_states = NULL, - .num_keys_pressed = 0 - }; + self->event_extra_data[self->num_event_polls] = (event_extra_data) { + .dev_input_id = -1, + .grabbed = false, + .key_states = NULL, + .num_keys_pressed = 0 + }; - self->stdout_event_index = self->num_event_polls; - ++self->num_event_polls; - } + self->stdin_event_index = self->num_event_polls; + ++self->num_event_polls; if(hotplug_event_init(&self->hotplug_ev)) { self->event_polls[self->num_event_polls] = (struct pollfd) { @@ -521,6 +499,13 @@ bool keyboard_event_init(keyboard_event *self, bool poll_stdout_error, bool excl } void keyboard_event_deinit(keyboard_event *self) { + self->running = false; + + for(int i = 0; i < self->num_global_hotkeys; ++i) { + free(self->global_hotkeys[i].action); + } + self->num_global_hotkeys = 0; + if(self->uinput_fd > 0) { close(self->uinput_fd); self->uinput_fd = -1; @@ -534,40 +519,184 @@ void keyboard_event_deinit(keyboard_event *self) { self->num_event_polls = 0; hotplug_event_deinit(&self->hotplug_ev); + + if(self->close_dev_input_fds_thread > 0) { + pthread_join(self->close_dev_input_fds_thread, NULL); + self->close_dev_input_fds_thread = 0; + } + + pthread_mutex_destroy(&self->close_dev_input_mutex); } static void on_device_added_callback(const char *devname, void *userdata) { keyboard_event *keyboard_ev = userdata; - char dev_input_filepath[1024]; + char dev_input_filepath[256]; snprintf(dev_input_filepath, sizeof(dev_input_filepath), "/dev/%s", devname); keyboard_event_try_add_device_if_keyboard(keyboard_ev, dev_input_filepath); } -#define MappingNotify 34 +/* Returns -1 on error */ +static int parse_u8(const char *str, int size) { + if(size <= 0) + return -1; -static void keyboard_event_poll_x11_events(keyboard_event *self) { - if(!self->x_context.display || !self->x_context.XPending || !self->x_context.XNextEvent || !self->x_context.XRefreshKeyboardMapping) - return; + int result = 0; + for(int i = 0; i < size; ++i) { + char c = str[i]; + if(c >= '0' && c <= '9') { + result = result * 10 + (c - '0'); + if(result > 255) + return -1; + } else { + return -1; + } + } + return result; +} - XEvent xev; - while(self->x_context.XPending(self->x_context.display)) { - xev.type = 0; - self->x_context.XNextEvent(self->x_context.display, &xev); - if(xev.type == MappingNotify) - self->x_context.XRefreshKeyboardMapping(xev.data); +static bool keyboard_event_parse_bind_keys(const char *str, int size, uint8_t *key, uint32_t *modifiers) { + *key = 0; + *modifiers = 0; + + const char *number_start = str; + const char *end = str + size; + for(;;) { + const char *next = strchr(number_start, '+'); + if(!next) + next = end; + + const int number_len = next - number_start; + const int number = parse_u8(number_start, number_len); + if(number == -1) { + fprintf(stderr, "Error: bind command keys \"%s\" is in invalid format\n", str); + return false; + } + + const uint32_t modifier_bit = keycode_to_modifier_bit(number); + if(modifier_bit == 0) { + if(*key != 0) { + fprintf(stderr, "Error: can't bind hotkey with multiple non-modifier keys\n"); + return false; + } + *key = number; + } else { + *modifiers = set_bit(*modifiers, modifier_bit, true); + } + + number_start = next + 1; + if(next == end) + break; } + + if(key == 0) { + fprintf(stderr, "Error: can't bind hotkey without a non-modifier key\n"); + return false; + } + + if(modifiers == 0) { + fprintf(stderr, "Error: can't bind hotkey without a modifier\n"); + return false; + } + + return true; } -void keyboard_event_poll_events(keyboard_event *self, int timeout_milliseconds, key_callback callback, void *userdata) { - /* TODO: Add the x11 connection to the below poll? */ - keyboard_event_poll_x11_events(self); +/* |command| is null-terminated */ +static void keyboard_event_parse_stdin_command(keyboard_event *self, const char *command, int command_size) { + if(strncmp(command, "bind ", 5) == 0) { + /* Example: |bind show_hide 20+40| */ + if(self->num_global_hotkeys >= MAX_GLOBAL_HOTKEYS) { + fprintf(stderr, "Error: can't add another hotkey. The maximum number of hotkeys (%d) has been reached\n", MAX_GLOBAL_HOTKEYS); + return; + } + + const char *action_name_end = strchr(command + 5, ' '); + if(!action_name_end) { + fprintf(stderr, "Error: command \"%s\" is in invalid format\n", command); + return; + } + const char *action_name = command + 5; + const int action_name_size = action_name_end - action_name; + + uint8_t key = 0; + uint32_t modifiers = 0; + const char *number_start = action_name_end + 1; + const char *end = command + command_size; + if(!keyboard_event_parse_bind_keys(number_start, end - number_start, &key, &modifiers)) + return; + + char *action = strndup(action_name, action_name_size); + if(!action) { + fprintf(stderr, "Error: failed to duplicate %.*s\n", action_name_size, action_name); + return; + } + + self->global_hotkeys[self->num_global_hotkeys] = (global_hotkey) { + .action = action, + .key = key, + .modifiers = modifiers + }; + ++self->num_global_hotkeys; + fprintf(stderr, "Info: binded hotkey: %s\n", action); + } else if(strncmp(command, "unbind_all", 10) == 0) { + for(int i = 0; i < self->num_global_hotkeys; ++i) { + free(self->global_hotkeys[i].action); + } + self->num_global_hotkeys = 0; + fprintf(stderr, "Info: unbinded all hotkeys\n"); + } else { + fprintf(stderr, "Warning: got invalid command: \"%s\", expected command to start with either \"bind\" or \"unbind_all\"\n", command); + } +} + +static void keyboard_event_process_stdin_command_data(keyboard_event *self, int fd) { + const int num_bytes_to_read = sizeof(self->stdin_command_data) - self->stdin_command_data_size; + if(num_bytes_to_read == 0) { + fprintf(stderr, "Error: failed to read data from stdin, buffer is full. Clearing buffer\n"); + self->stdin_command_data_size = 0; + return; + } + + const ssize_t bytes_read = read(fd, self->stdin_command_data + self->stdin_command_data_size, num_bytes_to_read); + if(bytes_read <= 0) + return; + + const char *command_start = self->stdin_command_data; + const char *search = self->stdin_command_data + self->stdin_command_data_size; + const char *end = search + bytes_read; + self->stdin_command_data_size += bytes_read; + + for(;;) { + char *next = memchr(search, '\n', end - search); + if(!next) + break; + + *next = '\0'; + keyboard_event_parse_stdin_command(self, command_start, next - command_start); + search = next + 1; + command_start = search; + if(next == end) + break; + } + + const int bytes_parsed = command_start - self->stdin_command_data; + if(bytes_parsed > 0) { + self->stdin_command_data_size -= bytes_parsed; + memmove(self->stdin_command_data, command_start, self->stdin_command_data_size); + } +} + +void keyboard_event_poll_events(keyboard_event *self, int timeout_milliseconds) { if(poll(self->event_polls, self->num_event_polls, timeout_milliseconds) <= 0) return; + if(self->stdin_failed) + return; + for(int i = 0; i < self->num_event_polls; ++i) { - if(i == self->stdout_event_index && (self->event_polls[i].revents & (POLLHUP|POLLERR))) - self->stdout_failed = true; + if(i == self->stdin_event_index && (self->event_polls[i].revents & (POLLHUP|POLLERR))) + self->stdin_failed = true; if(self->event_polls[i].revents & POLLHUP) { /* TODO: What if this is the hotplug fd? */ keyboard_event_remove_event(self, i); @@ -581,14 +710,14 @@ void keyboard_event_poll_events(keyboard_event *self, int timeout_milliseconds, if(i == self->hotplug_event_index) { /* Device is added to end of |event_polls| so it's ok to add while iterating it via index */ hotplug_event_process_event_data(&self->hotplug_ev, self->event_polls[i].fd, on_device_added_callback, self); - } else if(i == self->stdout_event_index) { - /* Do nothing, this shouldn't happen anyways since we dont poll for input */ + } else if(i == self->stdin_event_index) { + keyboard_event_process_stdin_command_data(self, self->event_polls[i].fd); } else { - keyboard_event_process_input_event_data(self, &self->event_extra_data[i], self->event_polls[i].fd, callback, userdata); + keyboard_event_process_input_event_data(self, &self->event_extra_data[i], self->event_polls[i].fd); } } } -bool keyboard_event_stdout_has_failed(const keyboard_event *self) { - return self->stdout_failed; +bool keyboard_event_stdin_has_failed(const keyboard_event *self) { + return self->stdin_failed; } diff --git a/tools/gsr-global-hotkeys/keyboard_event.h b/tools/gsr-global-hotkeys/keyboard_event.h index 9904237..a86b3dd 100644 --- a/tools/gsr-global-hotkeys/keyboard_event.h +++ b/tools/gsr-global-hotkeys/keyboard_event.h @@ -10,39 +10,25 @@ #include <stdint.h> /* POSIX */ -#include <sys/poll.h> +#include <poll.h> +#include <pthread.h> /* LINUX */ #include <linux/input-event-codes.h> #define MAX_EVENT_POLLS 32 - -typedef struct { - union { - int type; - unsigned char data[192]; - }; -} XEvent; - -typedef unsigned long (*XKeycodeToKeysym_FUNC)(void *display, unsigned char keycode, int index); -typedef int (*XPending_FUNC)(void *display); -typedef int (*XNextEvent_FUNC)(void *display, XEvent *event_return); -typedef int (*XRefreshKeyboardMapping_FUNC)(void* event_map); - -typedef struct { - void *display; - XKeycodeToKeysym_FUNC XKeycodeToKeysym; - XPending_FUNC XPending; - XNextEvent_FUNC XNextEvent; - XRefreshKeyboardMapping_FUNC XRefreshKeyboardMapping; -} x11_context; +#define MAX_CLOSE_FDS 256 +#define MAX_GLOBAL_HOTKEYS 32 typedef enum { KEYBOARD_MODKEY_LALT = 1 << 0, - KEYBOARD_MODKEY_RALT = 1 << 2, - KEYBOARD_MODKEY_SUPER = 1 << 3, - KEYBOARD_MODKEY_CTRL = 1 << 4, - KEYBOARD_MODKEY_SHIFT = 1 << 5 + KEYBOARD_MODKEY_RALT = 1 << 1, + KEYBOARD_MODKEY_LSUPER = 1 << 2, + KEYBOARD_MODKEY_RSUPER = 1 << 3, + KEYBOARD_MODKEY_LCTRL = 1 << 4, + KEYBOARD_MODKEY_RCTRL = 1 << 5, + KEYBOARD_MODKEY_LSHIFT = 1 << 6, + KEYBOARD_MODKEY_RSHIFT = 1 << 7 } keyboard_modkeys; typedef enum { @@ -63,38 +49,44 @@ typedef enum { } keyboard_grab_type; typedef struct { + uint32_t key; + uint32_t modifiers; /* keyboard_modkeys bitmask */ + char *action; +} global_hotkey; + +typedef struct { struct pollfd event_polls[MAX_EVENT_POLLS]; /* Current size is |num_event_polls| */ event_extra_data event_extra_data[MAX_EVENT_POLLS]; /* Current size is |num_event_polls| */ int num_event_polls; - int stdout_event_index; + int stdin_event_index; int hotplug_event_index; int uinput_fd; - bool stdout_failed; + bool stdin_failed; keyboard_grab_type grab_type; - x11_context x_context; + + pthread_t close_dev_input_fds_thread; + pthread_mutex_t close_dev_input_mutex; + int close_fds[MAX_CLOSE_FDS]; + int num_close_fds; + bool running; + + char stdin_command_data[512]; + int stdin_command_data_size; + + global_hotkey global_hotkeys[MAX_GLOBAL_HOTKEYS]; + int num_global_hotkeys; hotplug_event hotplug_ev; - keyboard_button_state lshift_button_state; - keyboard_button_state rshift_button_state; - keyboard_button_state lctrl_button_state; - keyboard_button_state rctrl_button_state; - keyboard_button_state lalt_button_state; - keyboard_button_state ralt_button_state; - keyboard_button_state lmeta_button_state; - keyboard_button_state rmeta_button_state; + uint32_t modifier_button_states; } keyboard_event; -/* |key| is a KEY_ from linux/input-event-codes.h. |modifiers| is a bitmask of keyboard_modkeys. |press_status| is 0 for released, 1 for pressed and 2 for repeat */ -/* Return true to allow other applications to receive the key input (when using exclusive grab) */ -typedef bool (*key_callback)(uint32_t key, uint32_t modifiers, int press_status, void *userdata); - -bool keyboard_event_init(keyboard_event *self, bool poll_stdout_error, bool exclusive_grab, keyboard_grab_type grab_type, x11_context x_context); +bool keyboard_event_init(keyboard_event *self, bool exclusive_grab, keyboard_grab_type grab_type); void keyboard_event_deinit(keyboard_event *self); /* If |timeout_milliseconds| is -1 then wait until an event is received */ -void keyboard_event_poll_events(keyboard_event *self, int timeout_milliseconds, key_callback callback, void *userdata); -bool keyboard_event_stdout_has_failed(const keyboard_event *self); +void keyboard_event_poll_events(keyboard_event *self, int timeout_milliseconds); +bool keyboard_event_stdin_has_failed(const keyboard_event *self); #endif /* KEYBOARD_EVENT_H */ diff --git a/tools/gsr-global-hotkeys/main.c b/tools/gsr-global-hotkeys/main.c index b64d60f..41e5ca5 100644 --- a/tools/gsr-global-hotkeys/main.c +++ b/tools/gsr-global-hotkeys/main.c @@ -2,42 +2,11 @@ /* C stdlib */ #include <stdio.h> -#include <stdint.h> #include <string.h> +#include <locale.h> /* POSIX */ #include <unistd.h> -#include <dlfcn.h> - -typedef struct { - uint32_t key; - uint32_t modifiers; /* keyboard_modkeys bitmask */ - const char *action; -} global_hotkey; - -#define NUM_GLOBAL_HOTKEYS 6 -static global_hotkey global_hotkeys[NUM_GLOBAL_HOTKEYS] = { - { .key = KEY_Z, .modifiers = KEYBOARD_MODKEY_LALT, .action = "show_hide" }, - { .key = KEY_F9, .modifiers = KEYBOARD_MODKEY_LALT, .action = "record" }, - { .key = KEY_F7, .modifiers = KEYBOARD_MODKEY_LALT, .action = "pause" }, - { .key = KEY_F8, .modifiers = KEYBOARD_MODKEY_LALT, .action = "stream" }, - { .key = KEY_F10, .modifiers = KEYBOARD_MODKEY_LALT | KEYBOARD_MODKEY_SHIFT, .action = "replay_start" }, - { .key = KEY_F10, .modifiers = KEYBOARD_MODKEY_LALT, .action = "replay_save" } -}; - -static bool on_key_callback(uint32_t key, uint32_t modifiers, int press_status, void *userdata) { - (void)userdata; - for(int i = 0; i < NUM_GLOBAL_HOTKEYS; ++i) { - if(key == global_hotkeys[i].key && modifiers == global_hotkeys[i].modifiers) { - if(press_status == 1) { /* 1 == Pressed */ - puts(global_hotkeys[i].action); - fflush(stdout); - } - return false; - } - } - return true; -} static void usage(void) { fprintf(stderr, "usage: gsr-global-hotkeys [--all|--virtual]\n"); @@ -46,77 +15,27 @@ static void usage(void) { fprintf(stderr, " --virtual Grab all virtual devices only.\n"); } -typedef void* (*XOpenDisplay_FUNC)(const char*); -typedef int (*XErrorHandler_FUNC)(void *display, void* error_event); -typedef XErrorHandler_FUNC (*XSetErrorHandler_FUNC)(XErrorHandler_FUNC handler); +static bool is_gsr_global_hotkeys_already_running(void) { + FILE *f = fopen("/proc/bus/input/devices", "rb"); + if(!f) + return false; -static int x_ignore_error(void *display, void *ee) { - (void)display; - (void)ee; - return 0; -} - -static x11_context setup_x11_context(void) { - x11_context x_context = {0}; - XSetErrorHandler_FUNC XSetErrorHandler = NULL; - - void *x11_lib = dlopen("libX11.so.6", RTLD_LAZY); - if(!x11_lib) { - fprintf(stderr, "Warning: dlopen libX11.so.6 failed\n"); - return x_context; - } - - XOpenDisplay_FUNC XOpenDisplay = dlsym(x11_lib, "XOpenDisplay"); - if(!XOpenDisplay) { - fprintf(stderr, "Warning: dlsym XOpenDisplay failed\n"); - goto fail; - } - - x_context.XKeycodeToKeysym = dlsym(x11_lib, "XKeycodeToKeysym"); - if(!x_context.XKeycodeToKeysym) { - fprintf(stderr, "Warning: dlsym XKeycodeToKeysym failed\n"); - goto fail; - } - - x_context.XPending = dlsym(x11_lib, "XPending"); - if(!x_context.XPending) { - fprintf(stderr, "Warning: dlsym XPending failed\n"); - goto fail; - } - - x_context.XNextEvent = dlsym(x11_lib, "XNextEvent"); - if(!x_context.XNextEvent) { - fprintf(stderr, "Warning: dlsym XNextEvent failed\n"); - goto fail; - } - - x_context.XRefreshKeyboardMapping = dlsym(x11_lib, "XRefreshKeyboardMapping"); - if(!x_context.XRefreshKeyboardMapping) { - fprintf(stderr, "Warning: dlsym XRefreshKeyboardMapping failed\n"); - goto fail; - } - - x_context.display = XOpenDisplay(NULL); - if(!x_context.display) { - fprintf(stderr, "Warning: XOpenDisplay failed\n"); - goto fail; + bool virtual_keyboard_running = false; + char line[1024]; + while(fgets(line, sizeof(line), f)) { + if(strstr(line, "gsr-ui virtual keyboard")) { + virtual_keyboard_running = true; + break; + } } - XSetErrorHandler = dlsym(x11_lib, "XSetErrorHandler"); - if(XSetErrorHandler) - XSetErrorHandler(x_ignore_error); - else - fprintf(stderr, "Warning: dlsym XSetErrorHandler failed\n"); - - return x_context; - - fail: - memset(&x_context, 0, sizeof(x_context)); - dlclose(x11_lib); - return x_context; + fclose(f); + return virtual_keyboard_running; } int main(int argc, char **argv) { + setlocale(LC_ALL, "C"); /* Sigh... stupid C */ + keyboard_grab_type grab_type = KEYBOARD_GRAB_TYPE_ALL; if(argc == 2) { const char *grab_type_arg = argv[1]; @@ -135,7 +54,10 @@ int main(int argc, char **argv) { return 1; } - x11_context x_context = setup_x11_context(); + if(is_gsr_global_hotkeys_already_running()) { + fprintf(stderr, "Error: gsr-global-hotkeys is already running\n"); + return 1; + } const uid_t user_id = getuid(); if(geteuid() != 0) { @@ -146,7 +68,7 @@ int main(int argc, char **argv) { } keyboard_event keyboard_ev; - if(!keyboard_event_init(&keyboard_ev, true, true, grab_type, x_context)) { + if(!keyboard_event_init(&keyboard_ev, true, grab_type)) { fprintf(stderr, "Error: failed to setup hotplugging and no keyboard input devices were found\n"); setuid(user_id); return 1; @@ -155,9 +77,9 @@ int main(int argc, char **argv) { fprintf(stderr, "Info: global hotkeys setup, waiting for hotkeys to be pressed\n"); for(;;) { - keyboard_event_poll_events(&keyboard_ev, -1, on_key_callback, NULL); - if(keyboard_event_stdout_has_failed(&keyboard_ev)) { - fprintf(stderr, "Info: stdout closed (parent process likely closed this process), exiting...\n"); + keyboard_event_poll_events(&keyboard_ev, -1); + if(keyboard_event_stdin_has_failed(&keyboard_ev)) { + fprintf(stderr, "Info: stdin closed (parent process likely closed this process), exiting...\n"); break; } } |