0.6.0.1

README.md

@@ -1,4 +1,4 @@
-# wufuc [![a.k.a. kb4012218-19](https://img.shields.io/badge/a.k.a.-kb4012218--19-blue.svg)](../../tree/old-kb4012218-19) [![download count](https://img.shields.io/github/downloads/zeffy/kb4012218-19/total.svg)](https://github.com/zeffy/kb4012218-19/releases/latest)
+# wufuc [![a.k.a. kb4012218-19](https://img.shields.io/badge/a.k.a.-kb4012218--19-blue.svg)](../../tree/old-kb4012218-19) [![downloads](https://img.shields.io/github/downloads/zeffy/wufuc/total.svg) ![downloads before v0.6](https://img.shields.io/badge/downloads%20before%20v0.6-13k-brightgreen.svg)](https://github.com/zeffy/wufuc/releases/latest)
 
 Disables the "Unsupported Hardware" message in Windows Update, and allows you to continue installing updates on Windows 7 and 8.1 systems with Intel Kaby Lake, AMD Ryzen, or other unsupported processors.
 
@@ -24,25 +24,39 @@ My patch takes advantage of this result caching behavior by setting the "hasn't
 
 ## How it works
 
-- On system boot the `wufuc` scheduled task runs under the `NT AUTHORITY\SYSTEM` user.
+- At system boot the wufuc scheduled task runs as the `NT AUTHORITY\SYSTEM` user.
-- `wufuc` determines what service host process the Windows Update service (`wuauserv`) runs in, and injects itself into it.
+- `wufuc` determines what service host group process the Windows Update service runs in (typically `netsvcs`), and injects itself into it.
 - Once injected, it applies a hook to `LoadLibraryEx` that automatically patches `wuaueng.dll` when it is loaded.
 - Any previously loaded `wuaueng.dll` is also patched.
 
-### Several improvements over my script-based approach:		
+### Several improvements over my xdelta3/batch script method:
-- **No system files are modified!***
+
-- Heuristic byte signature patching persists over new updates.		
+- **No system files are modified!**
+- Heuristic-based patching, which means it will usually keep working even after updates.		
 - C is best language.		
-- No external dependencies except for Microsoft Visual C++ 2015 Redistributable.	
+- No external dependencies except for Microsoft Visual C++ 2017 Redistributable.	
 
 ### How to install/uninstall?
 
-Just run move the `wufuc` folder to wherever you want and run `install_wufuc.bat` as administrator.
+Just download the [latest release](https://github.com/zeffy/wufuc/releases/latest), and extract the `wufuc` folder to a permanent location (like `C:\Program Files\wufuc`) and then run `install_wufuc.bat` as administrator. 
 
 To uninstall run `uninstall_wufuc.bat` as administrator. 
 
-To temporarily disable the patch, just go to the Task Scheduler and disable the `wufuc.{ ... }` task, then restart your computer.
+To temporarily disable the patch, just go to the Task Scheduler and disable the `wufuc.{72EEE38B-9997-42BD-85D3-2DD96DA17307}` task, then restart your computer.
 
 ### How do I remove your old patch and use this instead?
 
-I've included a utility script called `repair_wuaueng.dll.bat` that will initiate an sfc scan and attempt to automatically revert any changes made to `wuaueng.dll`.
+I've included a utility script called `repair_wuaueng.dll.bat` that will initiate an sfc scan and revert any changes made to `wuaueng.dll`.
+
+### How to see wufuc's debugging message output?
+
+You will need to download [DebugView](https://technet.microsoft.com/en-us/sysinternals/debugview.aspx) to do this.
+
+The best way to get a log of the entire life-cycle of wufuc is to do the following:
+
+1. Disable wufuc in Task Scheduler.
+2. Restart your computer.
+3. Start `DebugView.exe` as administrator and check `Capture -> Capture Global Win32`.
+4. Enable wufuc in Task Scheduler. 
+5. Run wufuc in Task Scheduler.
+6. Output will be shown in DebugView.

install/install_wufuc.bat

@@ -1,5 +1,5 @@
 @echo off
-title wufuc installer - v0.6
+title wufuc installer - v0.6.0.1
 :: Copyright (C) 2017 zeffy
 
 :: This program is free software: you can redistribute it and/or modify
@@ -15,7 +15,7 @@ title wufuc installer - v0.6
 :: You should have received a copy of the GNU General Public License
 :: along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
-echo Copyright (C) 2017 zeffy
+echo Copyright ^(C^) 2017 zeffy
 echo This program comes with ABSOLUTELY NO WARRANTY.
 echo This is free software, and you are welcome to redistribute it
 echo under certain conditions; see COPYING.txt for details.
@@ -76,7 +76,7 @@ goto :die
 for %%a in (%SUPPORTED_HOTFIXES%) do (
     wmic /output:stdout qfe get hotfixid | find "%%a" >nul && (
         set "INSTALLED_HOTFIX=%%a"
-        echo Detected installed supported update: %%a
+        echo Detected supported installed update: %%a
         goto :confirmation
     )
 )
@@ -84,8 +84,15 @@ for %%a in (%SUPPORTED_HOTFIXES%) do (
 echo.
 echo WARNING - Detected that no supported updates are installed! 
 echo.
-echo This can be a false warning, if you are certain that need wufuc then you
+echo This can be a false warning, sometimes it is caused by the WMI 
-echo can continue (there will be no side effects even if you don't need it)
+echo Win32_QuickFixEngineering class being broken. If you are certain
+echo that you need wufuc, then you can continue ^(there should be no 
+echo side effects even if you don't need it^).
+echo.
+echo This warning could also mean that a new update came out and the 
+echo installer script's list of updates hasn't been updated yet. If 
+echo this is the case and you know which update it is, feel free to
+echo create an issue.  https://github.com/zeffy/wufuc/issues
 
 set /p CONTINUE=Enter 'Y' if you still want to continue: 
 if /I not "%CONTINUE%"=="Y" goto :cancel
@@ -123,7 +130,9 @@ rundll32 "%wufuc_dll%",Rundll32Unload
 schtasks /Run /TN "%wufuc_task%"
 
 echo.
-echo Installed and started wufuc!
+echo Installed and started wufuc, you can now continue installing updates! :^)
+echo.
+echo To uninstall, run uninstall_wufuc.bat as administrator.
 goto :die
 
 :die
@@ -132,7 +141,6 @@ echo Press any key to exit...
 pause >nul
 exit
 
-
 :cancel
 echo.
 echo Canceled by user, press any key to exit...

install/repair_wuaueng.dll.bat

@@ -1,5 +1,5 @@
 @echo off
-title install wufuc (repair wuaueng.dll) - v0.6
+title install wufuc ^(repair wuaueng.dll^) - v0.6.0.1
 :: Copyright (C) 2017 zeffy
 
 :: This program is free software: you can redistribute it and/or modify
@@ -15,7 +15,7 @@ title install wufuc (repair wuaueng.dll) - v0.6
 :: You should have received a copy of the GNU General Public License
 :: along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
-echo Copyright (C) 2017 zeffy
+echo Copyright ^(C^) 2017 zeffy
 echo This program comes with ABSOLUTELY NO WARRANTY.
 echo This is free software, and you are welcome to redistribute it
 echo under certain conditions; see COPYING.txt for details.

install/uninstall_wufuc.bat

@@ -1,5 +1,5 @@
 @echo off
-title wufuc uninstaller - v0.6
+title wufuc uninstaller - v0.6.0.1
 :: Copyright (C) 2017 zeffy
 
 :: This program is free software: you can redistribute it and/or modify
@@ -15,7 +15,7 @@ title wufuc uninstaller - v0.6
 :: You should have received a copy of the GNU General Public License
 :: along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
-echo Copyright (C) 2017 zeffy
+echo Copyright ^(C^) 2017 zeffy
 echo This program comes with ABSOLUTELY NO WARRANTY.
 echo This is free software, and you are welcome to redistribute it
 echo under certain conditions; see COPYING.txt for details.
@@ -56,7 +56,7 @@ rundll32 "%wufuc_dll%",Rundll32Unload
 schtasks /Delete /TN "%wufuc_task%" /F
 
 echo.
-echo Unloaded and uninstalled wufuc!
+echo Unloaded and uninstalled wufuc. :^(
 
 :die
 echo.

wufuc.sln

@@ -10,7 +10,6 @@ Project("{2150E333-8FDC-42A3-9474-1A3956D46DE8}") = "Solution Items", "Solution
 		.gitignore = .gitignore
 		LICENSE = LICENSE
 		README.md = README.md
-		wufuc.xml = wufuc.xml
 	EndProjectSection
 EndProject
 Global

wufuc/core.c

@@ -10,152 +10,151 @@
 #include "util.h"
 
 DWORD WINAPI NewThreadProc(LPVOID lpParam) {
-	SC_HANDLE hSCManager = OpenSCManager(NULL, SERVICES_ACTIVE_DATABASE, SC_MANAGER_CONNECT);
+    SC_HANDLE hSCManager = OpenSCManager(NULL, SERVICES_ACTIVE_DATABASE, SC_MANAGER_CONNECT);
 
-	TCHAR lpBinaryPathName[0x8000];
+    TCHAR lpBinaryPathName[0x8000];
-	QueryServiceBinaryPathName(hSCManager, _T("wuauserv"), lpBinaryPathName, _countof(lpBinaryPathName));
+    get_svcpath(hSCManager, _T("wuauserv"), lpBinaryPathName, _countof(lpBinaryPathName));
 
-	BOOL result = _tcsicmp(GetCommandLine(), lpBinaryPathName);
+    BOOL result = _tcsicmp(GetCommandLine(), lpBinaryPathName);
-	CloseServiceHandle(hSCManager);
+    CloseServiceHandle(hSCManager);
 
-	if (result) {
+    if (result) {
-		return 0;
+        return 0;
-	}
+    }
 
-	SECURITY_ATTRIBUTES sa;
+    SECURITY_ATTRIBUTES sa;
-	sa.nLength = sizeof(sa);
+    sa.nLength = sizeof(sa);
-	ConvertStringSecurityDescriptorToSecurityDescriptor(_T("D:PAI(A;;FA;;;BA)"), SDDL_REVISION_1, &(sa.lpSecurityDescriptor), NULL);
+    ConvertStringSecurityDescriptorToSecurityDescriptor(_T("D:PAI(A;;FA;;;BA)"), SDDL_REVISION_1, &(sa.lpSecurityDescriptor), NULL);
-	sa.bInheritHandle = FALSE;
+    sa.bInheritHandle = FALSE;
 
-	HANDLE hEvent = CreateEvent(&sa, FALSE, FALSE, _T("Global\\wufuc_UnloadEvent"));
+    HANDLE hEvent = CreateEvent(&sa, TRUE, FALSE, _T("Global\\wufuc_UnloadEvent"));
 
-	if (!hEvent) {
+    if (!hEvent) {
-		return 0;
+        return 0;
-	}
+    }
 
-	HANDLE lphThreads[0x1000];
+    DWORD dwProcessId = GetCurrentProcessId();
-	SIZE_T cb;
+    DWORD dwThreadId = GetCurrentThreadId();
-	SuspendProcess(lphThreads, _countof(lphThreads), &cb);
+    HANDLE lphThreads[0x1000];
+    SIZE_T cb;
 
-	HMODULE hm = GetModuleHandle(NULL);
+    SuspendProcessThreads(dwProcessId, dwThreadId, lphThreads, _countof(lphThreads), &cb);
-	DETOUR_IAT(hm, LoadLibraryExA);
-	DETOUR_IAT(hm, LoadLibraryExW);
 
-	DbgPrint("Applied LoadLibraryEx hooks.");
+    HMODULE hm = GetModuleHandle(NULL);
+    DETOUR_IAT(hm, LoadLibraryExA);
+    DETOUR_IAT(hm, LoadLibraryExW);
 
-	HMODULE hwu = GetModuleHandle(_T("wuaueng.dll"));
+    HMODULE hwu = GetModuleHandle(_T("wuaueng.dll"));
-	if (hwu) {
+    if (hwu) {
-		PatchWUModule(hwu);
+        PatchWUModule(hwu);
-	}
+    }
-	ResumeAndCloseThreads(lphThreads, cb);
+    ResumeAndCloseThreads(lphThreads, cb);
 
-	WaitForSingleObject(hEvent, INFINITE);
+    WaitForSingleObject(hEvent, INFINITE);
-	CloseHandle(hEvent);
 
-	DbgPrint("Received wufuc_UnloadEvent, removing hooks.");
+    _tdbgprintf(_T("Received wufuc_UnloadEvent, removing hooks."));
 
-	SuspendProcess(lphThreads, _countof(lphThreads), &cb);
+    SuspendProcessThreads(dwProcessId, dwThreadId, lphThreads, _countof(lphThreads), &cb);
-	RESTORE_IAT(hm, LoadLibraryExA);
+    RESTORE_IAT(hm, LoadLibraryExA);
-	RESTORE_IAT(hm, LoadLibraryExW);
+    RESTORE_IAT(hm, LoadLibraryExW);
-	ResumeAndCloseThreads(lphThreads, cb);
+    ResumeAndCloseThreads(lphThreads, cb);
 
-	DbgPrint("Unloading library. Cya!");
+    _tdbgprintf(_T("Unloading library. Cya!"));
-	FreeLibraryAndExitThread(HINST_THISCOMPONENT, 0);
+    CloseHandle(hEvent);
-	return 0;
+    FreeLibraryAndExitThread(HINST_THISCOMPONENT, 0);
+    return 0;
 }
 
 BOOL IsWUModule(HMODULE hModule) {
-	TCHAR lpBaseName[MAX_PATH + 1];
+    TCHAR lpBaseName[MAX_PATH + 1];
-	GetModuleBaseName(GetCurrentProcess(), hModule, lpBaseName, _countof(lpBaseName));
+    GetModuleBaseName(GetCurrentProcess(), hModule, lpBaseName, _countof(lpBaseName));
-	return !_tcsicmp(lpBaseName, _T("wuaueng.dll"));
+    return !_tcsicmp(lpBaseName, _T("wuaueng.dll"));
 }
 
 BOOL PatchWUModule(HMODULE hModule) {
-	if (!IsWindows7Or8Point1()) {
+    if (!IsWindows7Or8Point1()) {
-		return FALSE;
+        return FALSE;
-	}
+    }
 
-	LPSTR lpszPattern;
+    LPSTR lpszPattern;
-	SIZE_T n1, n2;
+    SIZE_T n1, n2;
 #ifdef _WIN64
-	lpszPattern =
+    lpszPattern =
-		"FFF3"					// push rbx
+        "FFF3"					// push rbx
-		"4883EC??"				// sub rsp,??
+        "4883EC??"				// sub rsp,??
-		"33DB"					// xor ebx,ebx
+        "33DB"					// xor ebx,ebx
-		"391D????????"			// cmp dword ptr ds:[???????????],ebx
+        "391D????????"			// cmp dword ptr ds:[???????????],ebx
-		"7508"					// jnz $+8
+        "7508"					// jnz $+8
-		"8B05????????";			// mov eax,dword ptr ds:[???????????]
+        "8B05????????";			// mov eax,dword ptr ds:[???????????]
-	n1 = 10;
+    n1 = 10;
-	n2 = 18;
+    n2 = 18;
 #elif defined(_WIN32)
-	if (IsWindows8Point1()) {
+    if (IsWindows8Point1()) {
-		lpszPattern =
+        lpszPattern =
-			"8BFF"				// mov edi,edi
+            "8BFF"				// mov edi,edi
-			"51"				// push ecx
+            "51"				// push ecx
-			"833D????????00"	// cmp dword ptr ds:[????????],0
+            "833D????????00"	// cmp dword ptr ds:[????????],0
-			"7507"				// jnz $+7
+            "7507"				// jnz $+7
-			"A1????????";		// mov eax,dword ptr ds:[????????]
+            "A1????????";		// mov eax,dword ptr ds:[????????]
-		n1 = 5;
+        n1 = 5;
-		n2 = 13;
+        n2 = 13;
-	}
+    } else if (IsWindows7()) {
-	else if (IsWindows7()) {
+        lpszPattern =
-		lpszPattern =
+            "833D????????00"	// cmp dword ptr ds:[????????],0
-			"833D????????00"	// cmp dword ptr ds:[????????],0
+            "743E"				// je $+3E
-			"743E"				// je $+3E
+            "E8????????"		// call <wuaueng.IsCPUSupported>
-			"E8????????"		// call <wuaueng.IsCPUSupported>
+            "A3????????";		// mov dword ptr ds:[????????],eax
-			"A3????????";		// mov dword ptr ds:[????????],eax
+        n1 = 2;
-		n1 = 2;
+        n2 = 15;
-		n2 = 15;
+    }
-	}
 #else
-	return FALSE;
+    return FALSE;
 #endif
 
-	MODULEINFO modinfo;
+    MODULEINFO modinfo;
-	GetModuleInformation(GetCurrentProcess(), hModule, &modinfo, sizeof(MODULEINFO));
+    GetModuleInformation(GetCurrentProcess(), hModule, &modinfo, sizeof(MODULEINFO));
 
-	SIZE_T offset;
+    SIZE_T offset;
-	if (!FindPattern(modinfo.lpBaseOfDll, modinfo.SizeOfImage, lpszPattern, 0, &offset)) {
+    if (!FindPattern(modinfo.lpBaseOfDll, modinfo.SizeOfImage, lpszPattern, 0, &offset)) {
-		return FALSE;
+        return FALSE;
-	}
+    }
-	DbgPrint("IsDeviceServiceable(void) matched at %p", (UINT_PTR)modinfo.lpBaseOfDll + offset);
+    SIZE_T rva = (SIZE_T)modinfo.lpBaseOfDll + offset;
+    _tdbgprintf(_T("IsDeviceServiceable(void) matched at %p"), rva);
 
-	DWORD *lpdwResultIsNotCachedOffset = (DWORD *)((UINT_PTR)modinfo.lpBaseOfDll + offset + n1);
+    BOOL *lpbNotRunOnce = (BOOL *)(rva + n1 + sizeof(DWORD) + *(DWORD *)(rva + n1));
-	BOOL *lpbResultIsNotCached = (BOOL *)((UINT_PTR)modinfo.lpBaseOfDll + offset + n1 + sizeof(DWORD) + *lpdwResultIsNotCachedOffset);
+    if (*lpbNotRunOnce) {
-	if (*lpbResultIsNotCached) {
+        *lpbNotRunOnce = FALSE;
-		*lpbResultIsNotCached = FALSE;
+        _tdbgprintf(_T("Patched %p=%d"), lpbNotRunOnce, *lpbNotRunOnce);
-		DbgPrint("Patched %p=%d", lpbResultIsNotCached, *lpbResultIsNotCached);
+    }
-	}
 
-	DWORD *lpdwCachedResultOffset = (DWORD *)((UINT_PTR)modinfo.lpBaseOfDll + offset + n2);
+    BOOL *lpbCachedResult = (BOOL *)(rva + n2 + sizeof(DWORD) + *(DWORD *)(rva + n2));
-	BOOL *lpbCachedResult = (BOOL *)((UINT_PTR)modinfo.lpBaseOfDll + offset + n2 + sizeof(DWORD) + *lpdwCachedResultOffset);
+    if (!*lpbCachedResult) {
-	if (!*lpbCachedResult) {
+        *lpbCachedResult = TRUE;
-		*lpbCachedResult = TRUE;
+        _tdbgprintf(_T("Patched %p=%d"), lpbCachedResult, *lpbCachedResult);
-		DbgPrint("Patched %p=%d", lpbCachedResult, *lpbCachedResult);
+    }
-	}
+    return TRUE;
-	return TRUE;
 }
 
 HMODULE WINAPI _LoadLibraryExA(
-	_In_       LPCSTR  lpFileName,
+    _In_       LPCSTR  lpFileName,
-	_Reserved_ HANDLE  hFile,
+    _Reserved_ HANDLE  hFile,
-	_In_       DWORD   dwFlags
+    _In_       DWORD   dwFlags
 ) {
-	HMODULE result = LoadLibraryExA(lpFileName, hFile, dwFlags);
+    HMODULE result = LoadLibraryExA(lpFileName, hFile, dwFlags);
-	if (IsWUModule(result)) {
+    if (IsWUModule(result)) {
-		PatchWUModule(result);
+        PatchWUModule(result);
-	}
+    }
-	return result;
+    return result;
 }
 
 HMODULE WINAPI _LoadLibraryExW(
-	_In_       LPCWSTR lpFileName,
+    _In_       LPCWSTR lpFileName,
-	_Reserved_ HANDLE  hFile,
+    _Reserved_ HANDLE  hFile,
-	_In_       DWORD   dwFlags
+    _In_       DWORD   dwFlags
 ) {
-	HMODULE result = LoadLibraryExW(lpFileName, hFile, dwFlags);
+    HMODULE result = LoadLibraryExW(lpFileName, hFile, dwFlags);
-	if (IsWUModule(result)) {
+    if (IsWUModule(result)) {
-		PatchWUModule(result);
+        PatchWUModule(result);
-	}
+    }
-	return result;
+    return result;
 };

wufuc/core.h

@@ -5,13 +5,13 @@ DWORD WINAPI NewThreadProc(LPVOID lpParam);
 BOOL PatchWUModule(HMODULE hModule);
 
 HMODULE WINAPI _LoadLibraryExA(
-	_In_       LPCSTR  lpFileName,
+    _In_       LPCSTR  lpFileName,
-	_Reserved_ HANDLE  hFile,
+    _Reserved_ HANDLE  hFile,
-	_In_       DWORD   dwFlags
+    _In_       DWORD   dwFlags
 );
 
 HMODULE WINAPI _LoadLibraryExW(
-	_In_       LPCWSTR lpFileName,
+    _In_       LPCWSTR lpFileName,
-	_Reserved_ HANDLE  hFile,
+    _Reserved_ HANDLE  hFile,
-	_In_       DWORD   dwFlags
+    _In_       DWORD   dwFlags
 );

wufuc/dllmain.c

@@ -2,18 +2,18 @@
 #include "core.h"
 
 BOOL APIENTRY DllMain(HMODULE hModule, DWORD ul_reason_for_call, LPVOID lpReserved) {
-	switch (ul_reason_for_call) {
+    switch (ul_reason_for_call) {
-	case DLL_PROCESS_ATTACH:
+    case DLL_PROCESS_ATTACH:
-	{
+    {
-		DisableThreadLibraryCalls(hModule);
+        DisableThreadLibraryCalls(hModule);
-		HANDLE hThread = CreateThread(NULL, 0, NewThreadProc, NULL, 0, NULL);
+        HANDLE hThread = CreateThread(NULL, 0, NewThreadProc, NULL, 0, NULL);
-		CloseHandle(hThread);
+        CloseHandle(hThread);
-		break;
+        break;
-	}
+    }
-	case DLL_PROCESS_DETACH:
+    case DLL_PROCESS_DETACH:
-		break;
+        break;
-	default:
+    default:
-		break;
+        break;
-	}
+    }
-	return TRUE;
+    return TRUE;
 }

wufuc/entrypoint.c

@@ -6,36 +6,36 @@
 #include "util.h"
 
 void CALLBACK Rundll32Entry(HWND hwnd, HINSTANCE hinst, LPSTR lpszCmdLine, int nCmdShow) {
-	HANDLE hEvent = OpenEvent(SYNCHRONIZE, FALSE, _T("Global\\wufuc_UnloadEvent"));
+    HANDLE hEvent = OpenEvent(SYNCHRONIZE, FALSE, _T("Global\\wufuc_UnloadEvent"));
-	if (hEvent) {
+    if (hEvent) {
-		CloseHandle(hEvent);
+        CloseHandle(hEvent);
-		return;
+        return;
-	}
+    }
-	SC_HANDLE hSCManager = OpenSCManager(NULL, SERVICES_ACTIVE_DATABASE, SC_MANAGER_CONNECT);
+    SC_HANDLE hSCManager = OpenSCManager(NULL, SERVICES_ACTIVE_DATABASE, SC_MANAGER_CONNECT);
-	TCHAR lpGroupName[256];
+    TCHAR lpGroupName[256];
-	DWORD dwProcessId;
+    DWORD dwProcessId;
-	BOOL result = QueryServiceProcessId(hSCManager, _T("wuauserv"), &dwProcessId);
+    BOOL result = get_svcpid(hSCManager, _T("wuauserv"), &dwProcessId);
-	if (!result && GetServiceGroupName(hSCManager, _T("wuauserv"), lpGroupName, _countof(lpGroupName))) {
+    if (!result && get_svcgname(hSCManager, _T("wuauserv"), lpGroupName, _countof(lpGroupName))) {
-		result = FindServiceGroupProcessId(hSCManager, lpGroupName, &dwProcessId);
+        result = get_svcgpid(hSCManager, lpGroupName, &dwProcessId);
-	}
+    }
-	CloseServiceHandle(hSCManager);
+    CloseServiceHandle(hSCManager);
-	if (!result) {
+    if (!result) {
-		return;
+        return;
-	}
+    }
-	TCHAR lpLibFileName[MAX_PATH + 1];
+    TCHAR lpLibFileName[MAX_PATH + 1];
-	GetModuleFileName(HINST_THISCOMPONENT, lpLibFileName, _countof(lpLibFileName));
+    GetModuleFileName(HINST_THISCOMPONENT, lpLibFileName, _countof(lpLibFileName));
 
-	HANDLE hProcess = OpenProcess(PROCESS_ALL_ACCESS, FALSE, dwProcessId);
+    HANDLE hProcess = OpenProcess(PROCESS_ALL_ACCESS, FALSE, dwProcessId);
 
-	InjectLibrary(hProcess, lpLibFileName, _countof(lpLibFileName));
+    InjectLibrary(hProcess, lpLibFileName, _countof(lpLibFileName));
-	CloseHandle(hProcess);
+    CloseHandle(hProcess);
 }
 
 void CALLBACK Rundll32Unload(HWND hwnd, HINSTANCE hinst, LPSTR lpszCmdLine, int nCmdShow) {
-	HANDLE hEvent = OpenEvent(EVENT_MODIFY_STATE, FALSE, _T("Global\\wufuc_UnloadEvent"));
+    HANDLE hEvent = OpenEvent(EVENT_MODIFY_STATE, FALSE, _T("Global\\wufuc_UnloadEvent"));
-	if (hEvent) {
+    if (hEvent) {
-		DbgPrint("Setting wufuc_UnloadEvent...");
+        _tdbgprintf(_T("Setting wufuc_UnloadEvent..."));
-		SetEvent(hEvent);
+        SetEvent(hEvent);
-		CloseHandle(hEvent);
+        CloseHandle(hEvent);
-	}
+    }
 }

wufuc/process.c

@@ -1,139 +0,0 @@
-#include <windows.h>
-#include <stdio.h>
-#include <TlHelp32.h>
-#include <tchar.h>
-#include "util.h"
-#include "process.h"
-
-VOID DetourIAT(HMODULE hModule, LPSTR lpFuncName, LPVOID *lpOldAddress, LPVOID lpNewAddress) {
-	LPVOID *lpAddress = FindIAT(hModule, lpFuncName);
-	if (!lpAddress || *lpAddress == lpNewAddress) {
-		return;
-	}
-
-	DWORD flOldProtect;
-	DWORD flNewProtect = PAGE_READWRITE;
-	VirtualProtect(lpAddress, sizeof(LPVOID), flNewProtect, &flOldProtect);
-	if (lpOldAddress) {
-		*lpOldAddress = *lpAddress;
-	}
-	DbgPrint("%S %p => %p", lpFuncName, *lpAddress, lpNewAddress);
-	*lpAddress = lpNewAddress;
-	VirtualProtect(lpAddress, sizeof(LPVOID), flOldProtect, &flNewProtect);
-}
-
-LPVOID *FindIAT(HMODULE hModule, LPSTR lpFuncName) {
-	PIMAGE_DOS_HEADER dos = (PIMAGE_DOS_HEADER)hModule;
-	PIMAGE_NT_HEADERS nt = (PIMAGE_NT_HEADERS)((LPBYTE)dos + dos->e_lfanew);
-	PIMAGE_IMPORT_DESCRIPTOR desc = (PIMAGE_IMPORT_DESCRIPTOR)((LPBYTE)dos + nt->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_IMPORT].VirtualAddress);
-
-	for (PIMAGE_IMPORT_DESCRIPTOR iid = desc; iid->Name != 0; iid++) {
-		for (int i = 0; *(i + (LPVOID*)(iid->FirstThunk + (SIZE_T)hModule)) != NULL; i++) {
-			LPSTR name = (LPSTR)(*(i + (SIZE_T*)(iid->OriginalFirstThunk + (SIZE_T)hModule)) + (SIZE_T)hModule + 2);
-			const uintptr_t n = (uintptr_t)name;
-			if (!(n & (sizeof(n) == 4 ? 0x80000000 : 0x8000000000000000)) && !_stricmp(lpFuncName, name)) {
-				return i + (LPVOID*)(iid->FirstThunk + (SIZE_T)hModule);
-			}
-		}
-	}
-	return NULL;
-}
-
-BOOL FindPattern(LPCBYTE lpBytes, SIZE_T nNumberOfBytes, LPSTR lpszPattern, SIZE_T nStart, SIZE_T *lpOffset) {
-	SIZE_T nPatternLength = strlen(lpszPattern);
-	SIZE_T nMaskLength = nPatternLength / 2;
-	if (nMaskLength > nNumberOfBytes || nPatternLength % 2) {
-		return FALSE;
-	}
-
-	LPBYTE lpPattern = malloc(nMaskLength * sizeof(BYTE));
-	BOOL *lpbMask = malloc(nMaskLength * sizeof(BOOL));
-
-	LPSTR p = lpszPattern;
-	BOOL valid = TRUE;
-	for (SIZE_T i = 0; i < nMaskLength; i++) {
-		if (lpbMask[i] = strncmp(p, "??", 2)) {
-			if (sscanf_s(p, "%2hhx", &lpPattern[i]) != 1) {
-				valid = FALSE;
-				break;
-			}
-		}
-		p += 2;
-	}
-	BOOL result = FALSE;
-	if (valid) {
-		for (SIZE_T i = nStart; i < nNumberOfBytes - nStart - (nMaskLength - 1); i++) {
-			BOOL found = TRUE;
-			for (SIZE_T j = 0; j < nMaskLength; j++) {
-				if (lpbMask[j] && lpBytes[i + j] != lpPattern[j]) {
-					found = FALSE;
-					break;
-				}
-			}
-			if (found) {
-				*lpOffset = i;
-				result = TRUE;
-				break;
-			}
-		}
-	}
-	free(lpPattern);
-	free(lpbMask);
-	return result;
-}
-
-BOOL InjectLibrary(HANDLE hProcess, LPCTSTR lpLibFileName, DWORD cb) {
-	LPVOID lpBaseAddress = VirtualAllocEx(hProcess, NULL, cb, MEM_RESERVE | MEM_COMMIT, PAGE_READWRITE);
-	if (!WriteProcessMemory(hProcess, lpBaseAddress, lpLibFileName, cb, NULL)) {
-		return FALSE;
-	}
-	DWORD dwProcessId = GetProcessId(hProcess);
-	HANDLE hSnap = CreateToolhelp32Snapshot(TH32CS_SNAPMODULE, dwProcessId);
-	MODULEENTRY32 me;
-	me.dwSize = sizeof(me);
-
-	Module32First(hSnap, &me);
-	do {
-		if (!_tcsicmp(me.szModule, _T("kernel32.dll"))) {
-			break;
-		}
-	} while (Module32Next(hSnap, &me));
-	CloseHandle(hSnap);
-	DbgPrint("Injecting %s into process %d", lpLibFileName, dwProcessId);
-	HANDLE hThread = CreateRemoteThread(hProcess, NULL, 0, (LPTHREAD_START_ROUTINE)GetProcAddress(me.hModule, _CRT_STRINGIZE(LoadLibrary)), lpBaseAddress, 0, NULL);
-	CloseHandle(hThread);
-	return TRUE;
-}
-
-VOID SuspendProcess(HANDLE *lphThreads, SIZE_T dwSize, SIZE_T *lpcb) {
-	HANDLE hSnap = CreateToolhelp32Snapshot(TH32CS_SNAPTHREAD, 0);
-	THREADENTRY32 te;
-	te.dwSize = sizeof(te);
-	Thread32First(hSnap, &te);
-
-	DWORD dwProcessId = GetCurrentProcessId();
-	DWORD dwThreadId = GetCurrentThreadId();
-
-	SIZE_T count = 0;
-
-	do {
-		if (te.th32OwnerProcessID != dwProcessId || te.th32ThreadID == dwThreadId) {
-			continue;
-		}
-		lphThreads[count] = OpenThread(THREAD_SUSPEND_RESUME, FALSE, te.th32ThreadID);
-		SuspendThread(lphThreads[count]);
-		count++;
-	} while (count < dwSize && Thread32Next(hSnap, &te));
-	CloseHandle(hSnap);
-
-	*lpcb = count;
-	DbgPrint("Suspended other threads.");
-}
-
-VOID ResumeAndCloseThreads(HANDLE *lphThreads, SIZE_T cb) {
-	for (SIZE_T i = 0; i < cb; i++) {
-		ResumeThread(lphThreads[i]);
-		CloseHandle(lphThreads[i]);
-	}
-	DbgPrint("Resumed threads.");
-}

wufuc/process.h

@@ -1,20 +0,0 @@
-#pragma once
-
-VOID DetourIAT(HMODULE hModule, LPSTR lpFuncName, LPVOID *lpOldAddress, LPVOID lpNewAddress);
-
-#define DETOUR_IAT(x, y) \
-	LPVOID __LPORIGINAL##y; \
-	DetourIAT(x, #y, &__LPORIGINAL##y, &_##y)
-
-#define RESTORE_IAT(x, y) \
-	DetourIAT(x, #y, NULL, __LPORIGINAL##y)
-
-LPVOID *FindIAT(HMODULE hModule, LPSTR lpFuncName);
-
-BOOL FindPattern(LPCBYTE lpBytes, SIZE_T nNumberOfBytes, LPSTR lpszPattern, SIZE_T nStart, SIZE_T *lpOffset);
-
-BOOL InjectLibrary(HANDLE hProcess, LPCTSTR lpLibFileName, DWORD cb);
-
-VOID SuspendProcess(HANDLE *lphThreads, SIZE_T dwSize, SIZE_T *lpcb);
-
-VOID ResumeAndCloseThreads(HANDLE *lphThreads, SIZE_T dwSize);

wufuc/resource.h

@@ -1,15 +0,0 @@
-//{{NO_DEPENDENCIES}}
-// Microsoft Visual C++ generated include file.
-// Used by version.rc
-//
-
-// Next default values for new objects
-// 
-#ifdef APSTUDIO_INVOKED
-#ifndef APSTUDIO_READONLY_SYMBOLS
-#define _APS_NEXT_RESOURCE_VALUE        101
-#define _APS_NEXT_COMMAND_VALUE         40001
-#define _APS_NEXT_CONTROL_VALUE         1000
-#define _APS_NEXT_SYMED_VALUE           101
-#endif
-#endif

wufuc/service.c

@@ -2,96 +2,90 @@
 #include <tchar.h>
 #include "util.h"
 #include "service.h"
+#include "shellapihelper.h"
 
-BOOL QueryServiceBinaryPathName(SC_HANDLE hSCManager, LPCTSTR lpServiceName, LPTSTR lpBinaryPathName, SIZE_T dwSize) {
+BOOL get_svcpath(SC_HANDLE hSCManager, LPCTSTR lpServiceName, LPTSTR lpBinaryPathName, SIZE_T dwSize) {
-	HANDLE hService = OpenService(hSCManager, lpServiceName, SERVICE_QUERY_CONFIG);
+    HANDLE hService = OpenService(hSCManager, lpServiceName, SERVICE_QUERY_CONFIG);
-	if (!hService) {
+    if (!hService) {
-		return FALSE;
+        return FALSE;
-	}
+    }
 
-	DWORD cbBytesNeeded;
+    DWORD cbBytesNeeded;
-	QueryServiceConfig(hService, NULL, 0, &cbBytesNeeded);
+    QueryServiceConfig(hService, NULL, 0, &cbBytesNeeded);
-	LPQUERY_SERVICE_CONFIG sc = malloc(cbBytesNeeded);
+    LPQUERY_SERVICE_CONFIG sc = malloc(cbBytesNeeded);
-	BOOL result = QueryServiceConfig(hService, sc, cbBytesNeeded, &cbBytesNeeded);
+    BOOL result = QueryServiceConfig(hService, sc, cbBytesNeeded, &cbBytesNeeded);
-	int a = GetLastError();
+    CloseServiceHandle(hService);
-	CloseServiceHandle(hService);
+    if (result) {
-	if (result) {
+        _tcscpy_s(lpBinaryPathName, dwSize, sc->lpBinaryPathName);
-		_tcscpy_s(lpBinaryPathName, dwSize, sc->lpBinaryPathName);
+    }
-	}
+    free(sc);
-	LocalFree(sc);
+    return result;
-	return result;
 }
 
-BOOL QueryServiceProcessId(SC_HANDLE hSCManager, LPCTSTR lpServiceName, DWORD *lpdwProcessId) {
+BOOL get_svcpid(SC_HANDLE hSCManager, LPCTSTR lpServiceName, DWORD *lpdwProcessId) {
-	SC_HANDLE hService = OpenService(hSCManager, lpServiceName, SERVICE_QUERY_STATUS);
+    SC_HANDLE hService = OpenService(hSCManager, lpServiceName, SERVICE_QUERY_STATUS);
-	if (!hService) {
+    if (!hService) {
-		return FALSE;
+        return FALSE;
-	}
+    }
 
-	SERVICE_STATUS_PROCESS lpBuffer;
+    SERVICE_STATUS_PROCESS lpBuffer;
-	DWORD cbBytesNeeded;
+    DWORD cbBytesNeeded;
-	BOOL result = FALSE;
+    BOOL result = FALSE;
-	if (QueryServiceStatusEx(hService, SC_STATUS_PROCESS_INFO, (LPBYTE)&lpBuffer, sizeof(lpBuffer), &cbBytesNeeded) && lpBuffer.dwProcessId) {
+    if (QueryServiceStatusEx(hService, SC_STATUS_PROCESS_INFO, (LPBYTE)&lpBuffer, sizeof(lpBuffer), &cbBytesNeeded) && lpBuffer.dwProcessId) {
-		*lpdwProcessId = lpBuffer.dwProcessId;
+        *lpdwProcessId = lpBuffer.dwProcessId;
-		result = TRUE;
+        result = TRUE;
-		DbgPrint("Found %s pid %d", lpServiceName, *lpdwProcessId);
+    }
-	}
+    CloseServiceHandle(hService);
-	CloseServiceHandle(hService);
+    return result;
-	return result;
 }
 
-BOOL GetServiceGroupName(SC_HANDLE hSCManager, LPCTSTR lpServiceName, LPTSTR lpGroupName, SIZE_T dwSize) {
+BOOL get_svcgname(SC_HANDLE hSCManager, LPCTSTR lpServiceName, LPTSTR lpGroupName, SIZE_T dwSize) {
-	TCHAR lpBinaryPathName[0x8000];
+    TCHAR lpBinaryPathName[0x8000];
-	if (!QueryServiceBinaryPathName(hSCManager, lpServiceName, lpBinaryPathName, _countof(lpBinaryPathName))) {
+    if (!get_svcpath(hSCManager, lpServiceName, lpBinaryPathName, _countof(lpBinaryPathName))) {
-		return FALSE;
+        return FALSE;
-	}
+    }
-	int numArgs;
+    int numArgs;
-	LPWSTR *argv = CommandLineToArgv(lpBinaryPathName, &numArgs);
+    LPWSTR *argv = CommandLineToArgv(lpBinaryPathName, &numArgs);
-	if (numArgs < 3) {
+    if (numArgs < 3) {
-		return FALSE;
+        return FALSE;
-	}
+    }
 
-	TCHAR fname[_MAX_FNAME];
+    TCHAR fname[_MAX_FNAME];
-	_tsplitpath_s(argv[0], NULL, 0, NULL, 0, fname, _countof(fname), NULL, 0);
+    _tsplitpath_s(argv[0], NULL, 0, NULL, 0, fname, _countof(fname), NULL, 0);
 
-	BOOL result = FALSE;
+    BOOL result = FALSE;
-	if (!_tcsicmp(fname, _T("svchost"))) {
+    if (!_tcsicmp(fname, _T("svchost"))) {
-		LPWSTR *p = argv;
+        LPWSTR *p = argv;
-		for (int i = 0; i + 1 < numArgs; i++) {
+        for (int i = 1; i < numArgs; i++) {
-			if (!_tcsicmp(*(p++), _T("-k"))) {
+            if (!_tcsicmp(*(p++), _T("-k"))) {
-				_tcscpy_s(lpGroupName, dwSize, *p);
+                _tcscpy_s(lpGroupName, dwSize, *p);
-				result = TRUE;
+                result = TRUE;
-				DbgPrint("Found %s svc group: %s", lpServiceName, lpGroupName);
+                break;
-				break;
+            }
-			}
+        }
-		}
+    }
-	}
+    return result;
-	return result;
 }
 
-BOOL FindServiceGroupProcessId(SC_HANDLE hSCManager, LPTSTR lpServiceGroupName, DWORD *lpdwProcessId) {
+BOOL get_svcgpid(SC_HANDLE hSCManager, LPTSTR lpServiceGroupName, DWORD *lpdwProcessId) {
-	DWORD uBytes = 0x100000;
+    DWORD uBytes = 0x100000;
-	LPBYTE pvData = malloc(uBytes);
+    LPBYTE pvData = malloc(uBytes);
 
-	RegGetValue(HKEY_LOCAL_MACHINE, _T("SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion\\Svchost"), lpServiceGroupName, RRF_RT_REG_MULTI_SZ, NULL, pvData, &uBytes);
+    RegGetValue(HKEY_LOCAL_MACHINE, _T("SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion\\Svchost"), lpServiceGroupName, RRF_RT_REG_MULTI_SZ, NULL, pvData, &uBytes);
 
-	LPTSTR ptr = (LPTSTR)pvData;
+    BOOL result = FALSE;
-
+    for (LPTSTR p = (LPTSTR)pvData; *p; p += _tcslen(p) + 1) {
-	BOOL result = FALSE;
+        DWORD dwProcessId;
-	while (*ptr) {
+        TCHAR group[256];
-		DWORD dwProcessId;
+        if (get_svcpid(hSCManager, p, &dwProcessId)) {
-		if (QueryServiceProcessId(hSCManager, ptr, &dwProcessId)) {
+            get_svcgname(hSCManager, p, group, _countof(group));
-			TCHAR group[256];
+            result = !_tcsicmp(group, lpServiceGroupName);
-			GetServiceGroupName(hSCManager, ptr, group, _countof(group));
+        }
-			result = !_tcsicmp(group, lpServiceGroupName);
+        if (result) {
-		}
+            *lpdwProcessId = dwProcessId;
-		if (result) {
+            break;
-			DbgPrint("Found %s pid %d", lpServiceGroupName, dwProcessId);
+        }
-			*lpdwProcessId = dwProcessId;
+    }
-			break;
+    free(pvData);
-		}
+    return result;
-		ptr += _tcslen(ptr) + 1;
-	}
-	LocalFree(pvData);
-	return result;
 }

wufuc/service.h

@@ -1,9 +1,9 @@
 #pragma once
 
-BOOL QueryServiceBinaryPathName(SC_HANDLE hSCManager, LPCTSTR lpServiceName, LPTSTR lpBinaryPathName, SIZE_T dwSize);
+BOOL get_svcpath(SC_HANDLE hSCManager, LPCTSTR lpServiceName, LPTSTR lpBinaryPathName, SIZE_T dwSize);
 
-BOOL QueryServiceProcessId(SC_HANDLE hSCManager, LPCTSTR lpServiceName, DWORD *lpdwProcessId);
+BOOL get_svcpid(SC_HANDLE hSCManager, LPCTSTR lpServiceName, DWORD *lpdwProcessId);
 
-BOOL GetServiceGroupName(SC_HANDLE hSCManager, LPCTSTR lpServiceName, LPTSTR lpGroupName, SIZE_T dwSize);
+BOOL get_svcgname(SC_HANDLE hSCManager, LPCTSTR lpServiceName, LPTSTR lpGroupName, SIZE_T dwSize);
 
-BOOL FindServiceGroupProcessId(SC_HANDLE hSCManager, LPTSTR lpServiceGroupName, DWORD *lpdwProcessId);
+BOOL get_svcgpid(SC_HANDLE hSCManager, LPTSTR lpServiceGroupName, DWORD *lpdwProcessId);

wufuc/shellapihelper.h

@@ -0,0 +1,7 @@
+#pragma once
+
+#ifdef UNICODE
+#define CommandLineToArgv  CommandLineToArgvW
+#else
+#define CommandLineToArgv  CommandLineToArgvA
+#endif // !UNICODE

wufuc/util.c

@@ -1,27 +1,164 @@
 #include <Windows.h>
+#include <stdio.h>
 #include <VersionHelpers.h>
+#include <TlHelp32.h>
 #include <tchar.h>
 #include "util.h"
 
 BOOL IsWindows7Or8Point1(void) {
-	return IsWindows7() || IsWindows8Point1();
+    return IsWindows7() || IsWindows8Point1();
 }
 
 BOOL IsWindows7(void) {
-	return IsWindows7OrGreater() && !IsWindows8OrGreater();
+    return IsWindows7OrGreater() && !IsWindows8OrGreater();
 }
 
 BOOL IsWindows8Point1(void) {
-	return IsWindows8Point1OrGreater() && !IsWindows10OrGreater();
+    return IsWindows8Point1OrGreater() && !IsWindows10OrGreater();
 }
 
-//#ifdef _DEBUG
+VOID DetourIAT(HMODULE hModule, LPSTR lpFuncName, LPVOID *lpOldAddress, LPVOID lpNewAddress) {
-VOID _DbgPrint(LPCTSTR format, ...) {
+    LPVOID *lpAddress = FindIAT(hModule, lpFuncName);
-	TCHAR buffer[0x1000];
+    if (!lpAddress || *lpAddress == lpNewAddress) {
-	va_list argptr;
+        return;
-	va_start(argptr, format);
+    }
-	_vstprintf_s(buffer, _countof(buffer), format, argptr);
+
-	va_end(argptr);
+    DWORD flOldProtect;
-	OutputDebugString(buffer);
+    DWORD flNewProtect = PAGE_READWRITE;
+    VirtualProtect(lpAddress, sizeof(LPVOID), flNewProtect, &flOldProtect);
+    if (lpOldAddress) {
+        *lpOldAddress = *lpAddress;
+    }
+    _dbgprintf("%s %p => %p", lpFuncName, *lpAddress, lpNewAddress);
+    *lpAddress = lpNewAddress;
+    VirtualProtect(lpAddress, sizeof(LPVOID), flOldProtect, &flNewProtect);
+}
+
+LPVOID *FindIAT(HMODULE hModule, LPSTR lpFunctionName) {
+    SIZE_T hm = (SIZE_T)hModule;
+
+    for (PIMAGE_IMPORT_DESCRIPTOR iid = (PIMAGE_IMPORT_DESCRIPTOR)(hm + ((PIMAGE_NT_HEADERS)(hm + ((PIMAGE_DOS_HEADER)hm)->e_lfanew))
+        ->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_IMPORT].VirtualAddress); iid->Name; iid++) {
+
+        LPVOID *p;
+        for (SIZE_T i = 0; *(p = i + (LPVOID *)(hm + iid->FirstThunk)); i++) {
+            LPSTR fn = (LPSTR)(hm + *(i + (SIZE_T *)(hm + iid->OriginalFirstThunk)) + 2);
+            if (!((uintptr_t)fn & IMAGE_ORDINAL_FLAG) && !_stricmp(lpFunctionName, fn)) {
+                return p;
+            }
+        }
+    }
+    return NULL;
+}
+
+BOOL FindPattern(LPCBYTE pvData, SIZE_T nNumberOfBytes, LPSTR lpszPattern, SIZE_T nStart, SIZE_T *lpOffset) {
+    SIZE_T length = strlen(lpszPattern);
+    SIZE_T nBytes;
+    if (length % 2 || (nBytes = length / 2) > nNumberOfBytes) {
+        return FALSE;
+    }
+
+    LPBYTE lpBytes = malloc(nBytes * sizeof(BYTE));
+    BOOL *lpbwc = malloc(nBytes * sizeof(BOOL));
+
+    LPSTR p = lpszPattern;
+    BOOL valid = TRUE;
+    for (SIZE_T i = 0; i < nBytes; i++) {
+        if ((lpbwc[i] = strncmp(p, "??", 2)) && sscanf_s(p, "%2hhx", &lpBytes[i]) != 1) {
+            valid = FALSE;
+            break;
+        }
+        p += 2;
+    }
+    BOOL result = FALSE;
+    if (valid) {
+        for (SIZE_T i = nStart; i < nNumberOfBytes - nStart - (nBytes - 1); i++) {
+            BOOL found = TRUE;
+            for (SIZE_T j = 0; j < nBytes; j++) {
+                if (lpbwc[j] && pvData[i + j] != lpBytes[j]) {
+                    found = FALSE;
+                    break;
+                }
+            }
+            if (found) {
+                *lpOffset = i;
+                result = TRUE;
+                break;
+            }
+        }
+    }
+    free(lpBytes);
+    free(lpbwc);
+    return result;
+}
+
+BOOL InjectLibrary(HANDLE hProcess, LPCTSTR lpLibFileName, DWORD cb) {
+    LPVOID lpBaseAddress = VirtualAllocEx(hProcess, NULL, cb, MEM_RESERVE | MEM_COMMIT, PAGE_READWRITE);
+    if (!WriteProcessMemory(hProcess, lpBaseAddress, lpLibFileName, cb, NULL)) {
+        return FALSE;
+    }
+    DWORD dwProcessId = GetProcessId(hProcess);
+    HANDLE hSnap = CreateToolhelp32Snapshot(TH32CS_SNAPMODULE, dwProcessId);
+    MODULEENTRY32 me;
+    me.dwSize = sizeof(me);
+
+    Module32First(hSnap, &me);
+    do {
+        if (!_tcsicmp(me.szModule, _T("kernel32.dll"))) {
+            break;
+        }
+    } while (Module32Next(hSnap, &me));
+    CloseHandle(hSnap);
+    _tdbgprintf(_T("Injecting %s into process %d"), lpLibFileName, dwProcessId);
+    HANDLE hThread = CreateRemoteThread(hProcess, NULL, 0, (LPTHREAD_START_ROUTINE)GetProcAddress(me.hModule, _CRT_STRINGIZE(LoadLibrary)), lpBaseAddress, 0, NULL);
+    CloseHandle(hThread);
+    return TRUE;
+}
+
+VOID SuspendProcessThreads(DWORD dwProcessId, DWORD dwThreadId, HANDLE *lphThreads, SIZE_T dwSize, SIZE_T *lpcb) {
+    HANDLE hSnap = CreateToolhelp32Snapshot(TH32CS_SNAPTHREAD, 0);
+    THREADENTRY32 te;
+    te.dwSize = sizeof(te);
+    Thread32First(hSnap, &te);
+
+    SIZE_T count = 0;
+
+    do {
+        if (te.th32OwnerProcessID != dwProcessId || te.th32ThreadID == dwThreadId) {
+            continue;
+        }
+        lphThreads[count] = OpenThread(THREAD_SUSPEND_RESUME, FALSE, te.th32ThreadID);
+        SuspendThread(lphThreads[count]);
+        count++;
+    } while (count < dwSize && Thread32Next(hSnap, &te));
+    CloseHandle(hSnap);
+
+    *lpcb = count;
+    _tdbgprintf(_T("Suspended other threads."));
+}
+
+VOID ResumeAndCloseThreads(HANDLE *lphThreads, SIZE_T cb) {
+    for (SIZE_T i = 0; i < cb; i++) {
+        ResumeThread(lphThreads[i]);
+        CloseHandle(lphThreads[i]);
+    }
+    _tdbgprintf(_T("Resumed threads."));
+}
+
+VOID _wdbgprintf(LPCWSTR format, ...) {
+    WCHAR buffer[0x1000];
+    va_list argptr;
+    va_start(argptr, format);
+    vswprintf_s(buffer, _countof(buffer), format, argptr);
+    va_end(argptr);
+    OutputDebugStringW(buffer);
+}
+
+VOID _dbgprintf(LPCSTR format, ...) {
+    CHAR buffer[0x1000];
+    va_list argptr;
+    va_start(argptr, format);
+    vsprintf_s(buffer, _countof(buffer), format, argptr);
+    va_end(argptr);
+    OutputDebugStringA(buffer);
 }
-//#endif

wufuc/util.h

@@ -9,16 +9,33 @@ BOOL IsWindows7(void);
 
 BOOL IsWindows8Point1(void);
 
-//#ifdef _DEBUG
+VOID DetourIAT(HMODULE hModule, LPSTR lpFuncName, LPVOID *lpOldAddress, LPVOID lpNewAddress);
-VOID _DbgPrint(LPCTSTR format, ...);
-#define DbgPrint(format, ...) \
-	_DbgPrint(_T(__FUNCTION__) _T(": ") _T(format), ##__VA_ARGS__)
-//#else
-//#define DbgPrint(format, ...)
-//#endif
 
+#define DETOUR_IAT(x, y) \
+	LPVOID __LPORIGINAL##y; \
+	DetourIAT(x, #y, &__LPORIGINAL##y, &_##y)
+
+#define RESTORE_IAT(x, y) \
+	DetourIAT(x, #y, NULL, __LPORIGINAL##y)
+
+LPVOID *FindIAT(HMODULE hModule, LPSTR lpFuncName);
+
+BOOL FindPattern(LPCBYTE lpBytes, SIZE_T nNumberOfBytes, LPSTR lpszPattern, SIZE_T nStart, SIZE_T *lpOffset);
+
+BOOL InjectLibrary(HANDLE hProcess, LPCTSTR lpLibFileName, DWORD cb);
+
+VOID SuspendProcessThreads(DWORD dwProcessId, DWORD dwThreadId, HANDLE *lphThreads, SIZE_T dwSize, SIZE_T *lpcb);
+
+VOID ResumeAndCloseThreads(HANDLE *lphThreads, SIZE_T dwSize);
+
+VOID _wdbgprintf(LPCWSTR format, ...);
+VOID _dbgprintf(LPCSTR format, ...);
+//#ifdef _DEBUG
 #ifdef UNICODE
-#define CommandLineToArgv  CommandLineToArgvW
+#define _tdbgprintf _wdbgprintf
 #else
-#define CommandLineToArgv  CommandLineToArgvA
+#define _tdbgprintf _dbgprintf
 #endif // !UNICODE
+//#else
+//#define _tdbgprintf(format, ...)
+//#endif // !_DEBUG

wufuc/wufuc.vcxproj

@@ -144,6 +144,9 @@
       <GenerateDebugInformation>false</GenerateDebugInformation>
       <AdditionalDependencies>Shlwapi.lib;%(AdditionalDependencies)</AdditionalDependencies>
     </Link>
+    <PostBuildEvent>
+      <Command>copy /Y "$(TargetPath)" "$(SolutionDir)install\$(TargetFileName)"</Command>
+    </PostBuildEvent>
   </ItemDefinitionGroup>
   <ItemDefinitionGroup Condition="'$(Configuration)|$(Platform)'=='Release|x64'">
     <ClCompile>
@@ -163,12 +166,14 @@
       <GenerateDebugInformation>false</GenerateDebugInformation>
       <AdditionalDependencies>Shlwapi.lib;%(AdditionalDependencies)</AdditionalDependencies>
     </Link>
+    <PostBuildEvent>
+      <Command>copy /Y "$(TargetPath)" "$(SolutionDir)install\$(TargetFileName)"</Command>
+    </PostBuildEvent>
   </ItemDefinitionGroup>
   <ItemGroup>
     <ClCompile Include="core.c" />
     <ClCompile Include="dllmain.c" />
     <ClCompile Include="entrypoint.c" />
-    <ClCompile Include="process.c" />
     <ClCompile Include="service.c" />
     <ClCompile Include="util.c" />
   </ItemGroup>
@@ -177,8 +182,8 @@
   </ItemGroup>
   <ItemGroup>
     <ClInclude Include="core.h" />
-    <ClInclude Include="process.h" />
     <ClInclude Include="service.h" />
+    <ClInclude Include="shellapihelper.h" />
     <ClInclude Include="util.h" />
   </ItemGroup>
   <ItemGroup>