aboutsummaryrefslogtreecommitdiff
path: root/loader_wsock32_proxy/dllmain.cpp
diff options
context:
space:
mode:
authorJack <66967891+ASpoonPlaysGames@users.noreply.github.com>2023-12-27 00:32:01 +0000
committerGitHub <noreply@github.com>2023-12-27 01:32:01 +0100
commitf5ab6fb5e8be7b73e6003d4145081d5e0c0ce287 (patch)
tree90f2c6a4885dbd181799e2325cf33588697674e1 /loader_wsock32_proxy/dllmain.cpp
parentbb8ed59f6891b1196c5f5bbe7346cd171c8215fa (diff)
downloadNorthstarLauncher-f5ab6fb5e8be7b73e6003d4145081d5e0c0ce287.tar.gz
NorthstarLauncher-f5ab6fb5e8be7b73e6003d4145081d5e0c0ce287.zip
Folder restructuring from primedev (#624)v1.21.2-rc3v1.21.2
Copies of over the primedev folder structure for easier cherry-picking of further changes Co-authored-by: F1F7Y <filip.bartos07@proton.me>
Diffstat (limited to 'loader_wsock32_proxy/dllmain.cpp')
-rw-r--r--loader_wsock32_proxy/dllmain.cpp182
1 files changed, 0 insertions, 182 deletions
diff --git a/loader_wsock32_proxy/dllmain.cpp b/loader_wsock32_proxy/dllmain.cpp
deleted file mode 100644
index 4cc4f26e..00000000
--- a/loader_wsock32_proxy/dllmain.cpp
+++ /dev/null
@@ -1,182 +0,0 @@
-#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();
- }
-}