aboutsummaryrefslogtreecommitdiff
path: root/loader_wsock32_proxy/dllmain.cpp
blob: 82e0be8bc2b4d1a6e95e9f3ca8c6501b1c3d45cd (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
#include "pch.h"
#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;
		}

		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 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();
	}
}