217 lines
6.8 KiB
C++
217 lines
6.8 KiB
C++
#include "X11_Hook.h"
|
|
#include "../Renderer_Detector.h"
|
|
#include "../dll/dll.h"
|
|
|
|
#ifdef __LINUX__
|
|
#ifndef NO_OVERLAY
|
|
|
|
#include <imgui.h>
|
|
#include <impls/linux/imgui_impl_x11.h>
|
|
|
|
#include <dlfcn.h>
|
|
#include <unistd.h>
|
|
#include <fstream>
|
|
|
|
extern int ImGui_ImplX11_EventHandler(XEvent &event);
|
|
|
|
X11_Hook* X11_Hook::_inst = nullptr;
|
|
|
|
bool X11_Hook::start_hook()
|
|
{
|
|
bool res = true;
|
|
if (!hooked)
|
|
{
|
|
res = false;
|
|
std::ifstream flibrary("/proc/self/maps");
|
|
std::string line;
|
|
while( std::getline(flibrary, line) )
|
|
{
|
|
if( strcasestr(line.c_str(), DLL_NAME) != nullptr )
|
|
{
|
|
_library = dlopen(line.substr(line.find('/')).c_str(), RTLD_NOW);
|
|
|
|
if( _library != nullptr )
|
|
{
|
|
// Hook only this
|
|
_XEventsQueued = (decltype(_XEventsQueued))dlsym(_library, "XEventsQueued");
|
|
// Thoses are needed to ignore some events when overlay is on
|
|
_XPeekEvent = (decltype(_XPeekEvent))dlsym(_library, "XPeekEvent");
|
|
_XNextEvent = (decltype(_XNextEvent))dlsym(_library, "XNextEvent");
|
|
_XKeysymToKeycode = (decltype(_XKeysymToKeycode))dlsym(_library, "XKeysymToKeycode");
|
|
_XLookupKeysym = (decltype(_XLookupKeysym))dlsym(_library, "XLookupKeysym");
|
|
|
|
if( _XEventsQueued != nullptr && _XPeekEvent != nullptr
|
|
&& _XNextEvent != nullptr && _XKeysymToKeycode != nullptr
|
|
&& _XLookupKeysym != nullptr )
|
|
{
|
|
PRINT_DEBUG("Hooked X11\n");
|
|
hooked = true;
|
|
|
|
BeginHook();
|
|
HookFuncs(
|
|
std::make_pair<void**, void*>(&(void*&)_XEventsQueued, (void*)&X11_Hook::MyXEventsQueued)
|
|
);
|
|
|
|
EndHook();
|
|
break;
|
|
}
|
|
dlclose(_library);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return res;
|
|
}
|
|
|
|
void X11_Hook::resetRenderState()
|
|
{
|
|
if (initialized)
|
|
{
|
|
initialized = false;
|
|
ImGui_ImplX11_Shutdown();
|
|
}
|
|
}
|
|
|
|
void X11_Hook::prepareForOverlay(Display *display)
|
|
{
|
|
if (!initialized)
|
|
{
|
|
ImGui_ImplX11_Init(display);
|
|
|
|
ImGuiIO &io = ImGui::GetIO();
|
|
io.KeyMap[ImGuiKey_Tab] = _XKeysymToKeycode(display, XK_Tab);
|
|
io.KeyMap[ImGuiKey_LeftArrow] = _XKeysymToKeycode(display, XK_Left);
|
|
io.KeyMap[ImGuiKey_RightArrow] = _XKeysymToKeycode(display, XK_Right);
|
|
io.KeyMap[ImGuiKey_UpArrow] = _XKeysymToKeycode(display, XK_Up);
|
|
io.KeyMap[ImGuiKey_DownArrow] = _XKeysymToKeycode(display, XK_Down);
|
|
io.KeyMap[ImGuiKey_PageUp] = _XKeysymToKeycode(display, XK_Prior);
|
|
io.KeyMap[ImGuiKey_PageDown] = _XKeysymToKeycode(display, XK_Next);
|
|
io.KeyMap[ImGuiKey_Home] = _XKeysymToKeycode(display, XK_Home);
|
|
io.KeyMap[ImGuiKey_End] = _XKeysymToKeycode(display, XK_End);
|
|
io.KeyMap[ImGuiKey_Insert] = _XKeysymToKeycode(display, XK_Insert);
|
|
io.KeyMap[ImGuiKey_Delete] = _XKeysymToKeycode(display, XK_Delete);
|
|
io.KeyMap[ImGuiKey_Backspace] = _XKeysymToKeycode(display, XK_BackSpace);
|
|
io.KeyMap[ImGuiKey_Space] = _XKeysymToKeycode(display, XK_space);
|
|
io.KeyMap[ImGuiKey_Enter] = _XKeysymToKeycode(display, XK_Return);
|
|
io.KeyMap[ImGuiKey_Escape] = _XKeysymToKeycode(display, XK_Escape);
|
|
io.KeyMap[ImGuiKey_A] = _XKeysymToKeycode(display, XK_A);
|
|
io.KeyMap[ImGuiKey_C] = _XKeysymToKeycode(display, XK_C);
|
|
io.KeyMap[ImGuiKey_V] = _XKeysymToKeycode(display, XK_V);
|
|
io.KeyMap[ImGuiKey_X] = _XKeysymToKeycode(display, XK_X);
|
|
io.KeyMap[ImGuiKey_Y] = _XKeysymToKeycode(display, XK_Y);
|
|
io.KeyMap[ImGuiKey_Z] = _XKeysymToKeycode(display, XK_Z);
|
|
|
|
initialized = true;
|
|
}
|
|
|
|
ImGui_ImplX11_NewFrame();
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////////
|
|
// X11 window hooks
|
|
bool IgnoreEvent(XEvent &event)
|
|
{
|
|
switch(event.type)
|
|
{
|
|
// Keyboard
|
|
case KeyPress: case KeyRelease:
|
|
// MouseButton
|
|
case ButtonPress: case ButtonRelease:
|
|
// Mouse move
|
|
case MotionNotify:
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
int X11_Hook::MyXEventsQueued(Display *display, int mode)
|
|
{
|
|
static Time prev_time = {};
|
|
X11_Hook* inst = X11_Hook::Inst();
|
|
XEvent event = {};
|
|
|
|
int res = inst->_XEventsQueued(display, mode);
|
|
|
|
if( res > 0 )
|
|
{
|
|
/// The XPeekEvent function returns the first event from the event queue,
|
|
/// but it does not remove the event from the queue. If the queue is
|
|
/// empty, XPeekEvent flushes the output buffer and blocks until an event
|
|
/// is received.
|
|
inst->_XPeekEvent(display, &event);
|
|
Steam_Overlay* overlay = get_steam_client()->steam_overlay;
|
|
bool show = overlay->ShowOverlay();
|
|
// Is the event is a key press
|
|
if (event.type == KeyPress)
|
|
{
|
|
// Tab is pressed and was not pressed before
|
|
if (event.xkey.keycode == inst->_XKeysymToKeycode(display, XK_Tab) && event.xkey.state & ShiftMask)
|
|
{
|
|
// if key TAB is help, don't make the overlay flicker :p
|
|
if( event.xkey.time != prev_time)
|
|
{
|
|
overlay->ShowOverlay(!overlay->ShowOverlay());
|
|
|
|
if (overlay->ShowOverlay())
|
|
show = true;
|
|
}
|
|
}
|
|
}
|
|
else if(event.type == KeyRelease && event.xkey.keycode == inst->_XKeysymToKeycode(display, XK_Tab))
|
|
{
|
|
prev_time = event.xkey.time;
|
|
}
|
|
|
|
if (show)
|
|
{
|
|
ImGui_ImplX11_EventHandler(event);
|
|
|
|
if (IgnoreEvent(event))
|
|
{
|
|
inst->_XNextEvent(display, &event);
|
|
res = 0;
|
|
}
|
|
}
|
|
}
|
|
// XEventsQueued returns the num of events available.
|
|
// Usually, games tend to read all events queued before calling again XEventsQueued
|
|
// making us unavailable to intercept undesired events
|
|
return res;
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
X11_Hook::X11_Hook() :
|
|
initialized(false),
|
|
hooked(false)
|
|
{
|
|
//_library = dlopen(DLL_NAME, RTLD_NOW);
|
|
}
|
|
|
|
X11_Hook::~X11_Hook()
|
|
{
|
|
PRINT_DEBUG("X11 Hook removed\n");
|
|
|
|
resetRenderState();
|
|
|
|
//dlclose(_library);
|
|
|
|
_inst = nullptr;
|
|
}
|
|
|
|
X11_Hook* X11_Hook::Inst()
|
|
{
|
|
if (_inst == nullptr)
|
|
_inst = new X11_Hook;
|
|
|
|
return _inst;
|
|
}
|
|
|
|
const char* X11_Hook::get_lib_name() const
|
|
{
|
|
return DLL_NAME;
|
|
}
|
|
|
|
#endif//NO_OVERLAY
|
|
#endif//#__LINUX__
|