inject: Preserve last-error.

Some applications rely on the last-error from LoadLibraryEx to
detect when KB2533623 is not supported.

1 files changed, 14 insertions, 0 deletions

diff --git a/inject/injectee.cpp b/inject/injectee.cpp
index f76f4635..6417d487 100644
--- a/inject/injectee.cpp
+++ b/inject/injectee.cpp
@@ -104,6 +104,8 @@ MyCreateProcessCommon(BOOL bRet,
         return;
     }
 
+    DWORD dwLastError = GetLastError();
+
     char szDllPath[MAX_PATH];
     GetModuleFileNameA(g_hThisModule, szDllPath, sizeof szDllPath);
 
@@ -114,6 +116,8 @@ MyCreateProcessCommon(BOOL bRet,
     if (!(dwCreationFlags & CREATE_SUSPENDED)) {
         ResumeThread(lpProcessInformation->hThread);
     }
+
+    SetLastError(dwLastError);
 }
 
 static BOOL WINAPI
@@ -680,6 +684,7 @@ static HMODULE WINAPI
 MyLoadLibraryA(LPCSTR lpLibFileName)
 {
     HMODULE hModule = LoadLibraryA(lpLibFileName);
+    DWORD dwLastError = GetLastError();
 
     if (VERBOSITY >= 2) {
         debugPrintf("inject: intercepting %s(\"%s\") = 0x%p\n",
@@ -711,6 +716,7 @@ MyLoadLibraryA(LPCSTR lpLibFileName)
     // Hook all new modules (and not just this one, to pick up any dependencies)
     patchAllModules();
 
+    SetLastError(dwLastError);
     return hModule;
 }
 
@@ -718,6 +724,7 @@ static HMODULE WINAPI
 MyLoadLibraryW(LPCWSTR lpLibFileName)
 {
     HMODULE hModule = LoadLibraryW(lpLibFileName);
+    DWORD dwLastError = GetLastError();
 
     if (VERBOSITY >= 2) {
         debugPrintf("inject: intercepting %s(L\"%S\") = 0x%p\n",
@@ -727,6 +734,7 @@ MyLoadLibraryW(LPCWSTR lpLibFileName)
     // Hook all new modules (and not just this one, to pick up any dependencies)
     patchAllModules();
 
+    SetLastError(dwLastError);
     return hModule;
 }
 
@@ -776,6 +784,7 @@ static HMODULE WINAPI
 MyLoadLibraryExA(LPCSTR lpLibFileName, HANDLE hFile, DWORD dwFlags)
 {
     HMODULE hModule = LoadLibraryExA(lpLibFileName, hFile, adjustFlags(dwFlags));
+    DWORD dwLastError = GetLastError();
 
     if (VERBOSITY >= 2) {
         debugPrintf("inject: intercepting %s(\"%s\", 0x%p, 0x%lx) = 0x%p\n",
@@ -785,6 +794,7 @@ MyLoadLibraryExA(LPCSTR lpLibFileName, HANDLE hFile, DWORD dwFlags)
     // Hook all new modules (and not just this one, to pick up any dependencies)
     patchAllModules();
 
+    SetLastError(dwLastError);
     return hModule;
 }
 
@@ -792,6 +802,7 @@ static HMODULE WINAPI
 MyLoadLibraryExW(LPCWSTR lpLibFileName, HANDLE hFile, DWORD dwFlags)
 {
     HMODULE hModule = LoadLibraryExW(lpLibFileName, hFile, adjustFlags(dwFlags));
+    DWORD dwLastError = GetLastError();
 
     if (VERBOSITY >= 2) {
         debugPrintf("inject: intercepting %s(L\"%S\", 0x%p, 0x%lx) = 0x%p\n",
@@ -801,6 +812,7 @@ MyLoadLibraryExW(LPCWSTR lpLibFileName, HANDLE hFile, DWORD dwFlags)
     // Hook all new modules (and not just this one, to pick up any dependencies)
     patchAllModules();
 
+    SetLastError(dwLastError);
     return hModule;
 }
 
@@ -875,12 +887,14 @@ MyFreeLibrary(HMODULE hModule)
     }
 
     BOOL bRet = FreeLibrary(hModule);
+    DWORD dwLastError = GetLastError();
 
     EnterCriticalSection(&Mutex);
     // TODO: Only clear the modules that have been freed
     g_hHookedModules.clear();
     LeaveCriticalSection(&Mutex);
 
+    SetLastError(dwLastError);
     return bRet;
 }