From 957c8b98e408b04694ff004986a52ca20e8f35a7 Mon Sep 17 00:00:00 2001
From: qhy040404 <qhy040404@163.com>
Date: Fri, 17 Jan 2025 17:11:25 +0800
Subject: [PATCH] add window hook entry

---
 lib/src/dllmain.cpp | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/lib/src/dllmain.cpp b/lib/src/dllmain.cpp
index 2e54aa8..69e5d0e 100644
--- a/lib/src/dllmain.cpp
+++ b/lib/src/dllmain.cpp
@@ -192,6 +192,11 @@ DWORD __stdcall ThreadProc(LPVOID hInstance)
 // DLL entry point
 BOOL __stdcall DllMain(HMODULE hInstance, DWORD fdwReason, LPVOID lpReserved)
 {
+	// check injectee
+	if (!GetModuleHandleW(L"mhypbase.dll"))
+	{
+		return TRUE;
+	}
 
 	if (fdwReason == DLL_PROCESS_ATTACH)
 	{
@@ -202,3 +207,14 @@ BOOL __stdcall DllMain(HMODULE hInstance, DWORD fdwReason, LPVOID lpReserved)
 
 	return TRUE;
 }
+
+static LRESULT WINAPI YaeGetWindowHookImpl(int code, WPARAM wParam, LPARAM lParam)
+{
+	return CallNextHookEx(NULL, code, wParam, lParam);
+}
+
+EXTERN_C __declspec(dllexport) HRESULT WINAPI YaeGetWindowHook(_Out_ HOOKPROC* pHookProc)
+{
+	*pHookProc = YaeGetWindowHookImpl;
+	return S_OK;
+}
\ No newline at end of file