summarylogtreecommitdiffstats
path: root/keyBlocker.cpp
blob: 5030c5be4fcfb6096d2681b5673e5234aa508c7c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
#include <X11/Xlib.h>
#include <X11/XKBlib.h>
#include <stdio.h>
#include <dlfcn.h>
#include <xcb/xcb.h>
#include <unordered_set>

struct pair_hash
{
public:
	template <typename T, typename U>
	auto operator()(const std::pair<T, U> &x) const
	{
		return std::hash<T>()(x.first) ^ std::hash<U>()(x.second);
	}
};

static std::unordered_set<std::pair<int, unsigned int>, pair_hash> blocked_hotkeys;

extern "C" int XGrabKey(
	Display *display,
	int keycode,
	unsigned int modifiers,
	Window grab_window,
	Bool owner_events,
	int pointer_mode,
	int keyboard_mode)
{

	blocked_hotkeys.insert({keycode, modifiers});
	printf("Hotkey registration blocked: keycode=%d modifiers=%d\n", keycode, modifiers);
	return 0;
}

extern "C" int XUngrabKey(
	Display *display,
	int keycode,
	unsigned int modifiers,
	Window grab_window)
{
	blocked_hotkeys.erase({keycode, modifiers});
	printf("Hotkey unregistration blocked: keycode=%d modifiers=%d\n", keycode, modifiers);
	return 0;
}

static bool filter_xcb_event(xcb_connection_t *c, xcb_generic_event_t *event)
{
	if (event != nullptr)
	{
		if (event->response_type == XCB_KEY_PRESS)
		{
			auto key_press = reinterpret_cast<xcb_key_press_event_t *>(event);
			if (blocked_hotkeys.find({key_press->detail, key_press->state}) != blocked_hotkeys.end())
			{
				return false;
			}
		}
		else if (event->response_type == XCB_KEY_RELEASE)
		{
			auto key_release = reinterpret_cast<xcb_key_release_event_t *>(event);
			if (blocked_hotkeys.find({key_release->detail, key_release->state}) != blocked_hotkeys.end())
			{
				return false;
			}
		}
	}
	return true;
}

extern "C" xcb_generic_event_t *xcb_wait_for_event(xcb_connection_t *c)
{
	static auto origin = reinterpret_cast<decltype(&xcb_wait_for_event)>(dlsym(RTLD_NEXT, "xcb_wait_for_event"));
	xcb_generic_event_t *event;
	do
	{
		event = origin(c);
	} while (!filter_xcb_event(c, event));
	return event;
}

extern "C" xcb_generic_event_t *xcb_poll_for_event(xcb_connection_t *c)
{
	static auto origin = reinterpret_cast<decltype(&xcb_poll_for_event)>(dlsym(RTLD_NEXT, "xcb_poll_for_event"));
	xcb_generic_event_t *event;
	do
	{
		event = origin(c);
	} while (!filter_xcb_event(c, event));
	return event;
}