mirror of
https://github.com/gnif/LookingGlass.git
synced 2024-12-24 14:33:38 +00:00
411a6b1e49
This change not only exposes and allows use of NtDelayExecution, but also moves the code to set the system timer resolution.
222 lines
5.4 KiB
C
222 lines
5.4 KiB
C
/**
|
|
* Looking Glass
|
|
* Copyright (C) 2017-2021 The Looking Glass Authors
|
|
* https://looking-glass.io
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify it
|
|
* under the terms of the GNU General Public License as published by the Free
|
|
* Software Foundation; either version 2 of the License, or (at your option)
|
|
* any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful, but WITHOUT
|
|
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
|
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
|
|
* more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License along
|
|
* with this program; if not, write to the Free Software Foundation, Inc., 59
|
|
* Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
*/
|
|
|
|
#include "windows/mousehook.h"
|
|
#include "windows/delay.h"
|
|
#include "common/windebug.h"
|
|
#include "platform.h"
|
|
|
|
#include <windows.h>
|
|
#include <stdbool.h>
|
|
|
|
struct mouseHook
|
|
{
|
|
bool installed;
|
|
HHOOK hook;
|
|
MouseHookFn callback;
|
|
int x, y;
|
|
HANDLE event , updateEvent;
|
|
HANDLE thread, updateThread;
|
|
};
|
|
|
|
static struct mouseHook mouseHook = { 0 };
|
|
|
|
// forwards
|
|
static LRESULT WINAPI mouseHook_hook(int nCode, WPARAM wParam, LPARAM lParam);
|
|
|
|
static bool switchDesktopAndHook(void)
|
|
{
|
|
HDESK desk = OpenInputDesktop(0, FALSE, GENERIC_ALL);
|
|
if (!desk)
|
|
{
|
|
DEBUG_WINERROR("Failed to OpenInputDesktop", GetLastError());
|
|
return false;
|
|
}
|
|
|
|
if (!SetThreadDesktop(desk))
|
|
{
|
|
DEBUG_WINERROR("Failed to SetThreadDesktop", GetLastError());
|
|
CloseDesktop(desk);
|
|
return false;
|
|
}
|
|
CloseDesktop(desk);
|
|
|
|
POINT position;
|
|
GetCursorPos(&position);
|
|
|
|
mouseHook.x = position.x;
|
|
mouseHook.y = position.y;
|
|
mouseHook.callback(position.x, position.y);
|
|
|
|
mouseHook.hook = SetWindowsHookEx(WH_MOUSE_LL, mouseHook_hook, NULL, 0);
|
|
if (!mouseHook.hook)
|
|
{
|
|
DEBUG_WINERROR("Failed to install the mouse hook", GetLastError());
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
static VOID WINAPI winEventProc(HWINEVENTHOOK hWinEventHook, DWORD event,
|
|
HWND hwnd, LONG idObject, LONG idChild, DWORD idEventThread, DWORD dwmsEventTime)
|
|
{
|
|
switch (event)
|
|
{
|
|
case EVENT_SYSTEM_DESKTOPSWITCH:
|
|
UnhookWindowsHookEx(mouseHook.hook);
|
|
switchDesktopAndHook();
|
|
break;
|
|
}
|
|
}
|
|
|
|
static DWORD WINAPI updateThreadProc(LPVOID lParam)
|
|
{
|
|
HANDLE events[2] = {
|
|
mouseHook.event,
|
|
mouseHook.updateEvent
|
|
};
|
|
|
|
while(true)
|
|
{
|
|
switch(WaitForMultipleObjects(2, events, FALSE, INFINITE))
|
|
{
|
|
case WAIT_OBJECT_0:
|
|
DEBUG_INFO("Mouse hook update thread received quit request");
|
|
return 0;
|
|
|
|
case WAIT_OBJECT_0 + 1:
|
|
mouseHook.callback(mouseHook.x, mouseHook.y);
|
|
|
|
// limit this to 1000Hz, who has a mouse that updates faster anyway?
|
|
delayExecution(1.0f);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
static DWORD WINAPI threadProc(LPVOID lParam) {
|
|
if (mouseHook.installed)
|
|
{
|
|
DEBUG_WARN("Mouse hook already installed");
|
|
return 0;
|
|
}
|
|
|
|
mouseHook.callback = (MouseHookFn)lParam;
|
|
if (!switchDesktopAndHook())
|
|
return 0;
|
|
|
|
mouseHook.installed = true;
|
|
|
|
HWINEVENTHOOK eventHook = SetWinEventHook(
|
|
EVENT_SYSTEM_DESKTOPSWITCH, EVENT_SYSTEM_DESKTOPSWITCH, NULL,
|
|
winEventProc, 0, 0, WINEVENT_OUTOFCONTEXT
|
|
);
|
|
if (!eventHook)
|
|
{
|
|
DEBUG_WINERROR("Failed to SetWinEventHook", GetLastError());
|
|
goto exit;
|
|
}
|
|
|
|
MSG msg;
|
|
while (true) {
|
|
switch (MsgWaitForMultipleObjects(1, &mouseHook.event, FALSE, INFINITE, QS_ALLINPUT)) {
|
|
case WAIT_OBJECT_0:
|
|
DEBUG_INFO("Mouse hook thread received quit request");
|
|
PostQuitMessage(0);
|
|
break;
|
|
case WAIT_OBJECT_0 + 1:
|
|
while (PeekMessage(&msg, NULL, 0, 0, PM_REMOVE))
|
|
{
|
|
if (msg.message == WM_QUIT)
|
|
goto exit;
|
|
TranslateMessage(&msg);
|
|
DispatchMessage(&msg);
|
|
}
|
|
break;
|
|
default:
|
|
goto exit;
|
|
}
|
|
}
|
|
|
|
exit:
|
|
if (eventHook) UnhookWinEvent(eventHook);
|
|
UnhookWindowsHookEx(mouseHook.hook);
|
|
mouseHook.installed = false;
|
|
return 0;
|
|
}
|
|
|
|
void mouseHook_install(MouseHookFn callback)
|
|
{
|
|
if (!mouseHook.event)
|
|
{
|
|
mouseHook.event = CreateEventA(NULL, FALSE, FALSE, NULL);
|
|
if (!mouseHook.event)
|
|
{
|
|
DEBUG_WINERROR("Failed to create mouse hook uninstall event",
|
|
GetLastError());
|
|
return;
|
|
}
|
|
}
|
|
|
|
if (!mouseHook.updateEvent)
|
|
{
|
|
mouseHook.updateEvent = CreateEventA(NULL, FALSE, FALSE, NULL);
|
|
if (!mouseHook.event)
|
|
{
|
|
DEBUG_WINERROR("Failed to create mouse hook update event",
|
|
GetLastError());
|
|
return;
|
|
}
|
|
}
|
|
|
|
mouseHook.thread =
|
|
CreateThread(NULL, 0, threadProc, callback, 0, NULL);
|
|
|
|
mouseHook.updateThread =
|
|
CreateThread(NULL, 0, updateThreadProc, 0, 0, NULL);
|
|
}
|
|
|
|
void mouseHook_remove(void)
|
|
{
|
|
if (!mouseHook.event)
|
|
return;
|
|
|
|
SetEvent(mouseHook.event);
|
|
WaitForSingleObject(mouseHook.thread , INFINITE);
|
|
WaitForSingleObject(mouseHook.updateThread, INFINITE);
|
|
CloseHandle(mouseHook.thread);
|
|
CloseHandle(mouseHook.updateThread);
|
|
}
|
|
|
|
static LRESULT WINAPI mouseHook_hook(int nCode, WPARAM wParam, LPARAM lParam)
|
|
{
|
|
if (nCode == HC_ACTION && wParam == WM_MOUSEMOVE)
|
|
{
|
|
MSLLHOOKSTRUCT *msg = (MSLLHOOKSTRUCT *)lParam;
|
|
if (mouseHook.x != msg->pt.x || mouseHook.y != msg->pt.y)
|
|
{
|
|
mouseHook.x = msg->pt.x;
|
|
mouseHook.y = msg->pt.y;
|
|
SetEvent(mouseHook.updateEvent);
|
|
}
|
|
}
|
|
return CallNextHookEx(mouseHook.hook, nCode, wParam, lParam);
|
|
}
|