diff options
Diffstat (limited to 'primedev/wsockproxy/dllmain.cpp')
| -rw-r--r-- | primedev/wsockproxy/dllmain.cpp | 182 |
1 files changed, 182 insertions, 0 deletions
diff --git a/primedev/wsockproxy/dllmain.cpp b/primedev/wsockproxy/dllmain.cpp new file mode 100644 index 00000000..4cc4f26e --- /dev/null +++ b/primedev/wsockproxy/dllmain.cpp @@ -0,0 +1,182 @@ +#include "loader.h" + +#include <shlwapi.h> +#include <filesystem> + +HINSTANCE hLThis = 0; +FARPROC p[857]; +HINSTANCE hL = 0; + +bool GetExePathWide(wchar_t* dest, DWORD destSize) +{ + if (!dest) + return NULL; + if (destSize < MAX_PATH) + return NULL; + + DWORD length = GetModuleFileNameW(NULL, dest, destSize); + return length && PathRemoveFileSpecW(dest); +} + +wchar_t exePath[4096]; +wchar_t buffer1[8192]; +wchar_t buffer2[12288]; + +BOOL WINAPI DllMain(HINSTANCE hInst, DWORD reason, LPVOID) +{ + if (reason == DLL_PROCESS_ATTACH) + { + hLThis = hInst; + + if (!GetExePathWide(exePath, 4096)) + { + MessageBoxA( + GetForegroundWindow(), + "Failed getting game directory.\nThe game cannot continue and has to exit.", + "Northstar Wsock32 Proxy Error", + 0); + return true; + } + + SetCurrentDirectoryW(exePath); + + if (!ProvisionNorthstar()) // does not call InitialiseNorthstar yet, will do it on LauncherMain hook + return true; + + // copy the original library for system to our local directory, with changed name so that we can load it + swprintf_s(buffer1, L"%s\\bin\\x64_retail\\wsock32.org.dll", exePath); + GetSystemDirectoryW(buffer2, 4096); + swprintf_s(buffer2, L"%s\\wsock32.dll", buffer2); + try + { + std::filesystem::copy_file(buffer2, buffer1); + } + catch (const std::exception& e1) + { + if (!std::filesystem::exists(buffer1)) + { + // fallback by copying to temp dir... + // because apparently games installed by EA Desktop app don't have write permissions in their directories + auto temp_dir = std::filesystem::temp_directory_path() / L"wsock32.org.dll"; + try + { + std::filesystem::copy_file(buffer2, temp_dir); + } + catch (const std::exception& e2) + { + if (!std::filesystem::exists(temp_dir)) + { + swprintf_s( + buffer2, + L"Failed copying wsock32.dll from system32 to \"%s\"\n\n%S\n\nFurthermore, we failed copying wsock32.dll into " + L"temporary directory at \"%s\"\n\n%S", + buffer1, + e1.what(), + temp_dir.c_str(), + e2.what()); + MessageBoxW(GetForegroundWindow(), buffer2, L"Northstar Wsock32 Proxy Error", 0); + return false; + } + } + swprintf_s(buffer1, L"%s", temp_dir.c_str()); + } + } + hL = LoadLibraryExW(buffer1, 0, LOAD_WITH_ALTERED_SEARCH_PATH); + if (!hL) + { + LibraryLoadError(GetLastError(), L"wsock32.org.dll", buffer1); + return false; + } + + // load the functions to proxy + // it's only some of them, because in case of wsock32 most of the functions can actually be natively redirected + // (see wsock32.def and https://source.winehq.org/WineAPI/wsock32.html) + p[1] = GetProcAddress(hL, "EnumProtocolsA"); + p[2] = GetProcAddress(hL, "EnumProtocolsW"); + p[4] = GetProcAddress(hL, "GetAddressByNameA"); + p[5] = GetProcAddress(hL, "GetAddressByNameW"); + p[17] = GetProcAddress(hL, "WEP"); + p[30] = GetProcAddress(hL, "WSARecvEx"); + p[36] = GetProcAddress(hL, "__WSAFDIsSet"); + p[45] = GetProcAddress(hL, "getnetbyname"); + p[52] = GetProcAddress(hL, "getsockopt"); + p[56] = GetProcAddress(hL, "inet_network"); + p[67] = GetProcAddress(hL, "s_perror"); + p[72] = GetProcAddress(hL, "setsockopt"); + } + + if (reason == DLL_PROCESS_DETACH) + { + FreeLibrary(hL); + return true; + } + + return true; +} + +extern "C" +{ + FARPROC PA = NULL; + int RunASM(); + + void PROXY_EnumProtocolsA() + { + PA = p[1]; + RunASM(); + } + void PROXY_EnumProtocolsW() + { + PA = p[2]; + RunASM(); + } + void PROXY_GetAddressByNameA() + { + PA = p[4]; + RunASM(); + } + void PROXY_GetAddressByNameW() + { + PA = p[5]; + RunASM(); + } + void PROXY_WEP() + { + PA = p[17]; + RunASM(); + } + void PROXY_WSARecvEx() + { + PA = p[30]; + RunASM(); + } + void PROXY___WSAFDIsSet() + { + PA = p[36]; + RunASM(); + } + void PROXY_getnetbyname() + { + PA = p[45]; + RunASM(); + } + void PROXY_getsockopt() + { + PA = p[52]; + RunASM(); + } + void PROXY_inet_network() + { + PA = p[56]; + RunASM(); + } + void PROXY_s_perror() + { + PA = p[67]; + RunASM(); + } + void PROXY_setsockopt() + { + PA = p[72]; + RunASM(); + } +} |