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
|
#include "pch.h"
#include "loader.h"
#include "../NorthstarDedicatedTest/hookutils.h"
#include <string>
#include <system_error>
#include <sstream>
#include <fstream>
void LibraryLoadError(DWORD dwMessageId, const wchar_t* libName, const wchar_t* location)
{
char text[2048];
std::string message = std::system_category().message(dwMessageId);
sprintf_s(text, "Failed to load the %ls at \"%ls\" (%lu):\n\n%hs", libName, location, dwMessageId, message.c_str());
MessageBoxA(GetForegroundWindow(), text, "Northstar Wsock32 Proxy Error", 0);
}
bool ShouldLoadNorthstar()
{
bool loadNorthstar = !strstr(GetCommandLineA(), "-vanilla");
if (!loadNorthstar)
return loadNorthstar;
auto runNorthstarFile = std::ifstream("run_northstar.txt");
if (runNorthstarFile)
{
std::stringstream runNorthstarFileBuffer;
runNorthstarFileBuffer << runNorthstarFile.rdbuf();
runNorthstarFile.close();
if (runNorthstarFileBuffer.str()._Starts_with("0"))
loadNorthstar = false;
}
return loadNorthstar;
}
bool LoadNorthstar()
{
FARPROC Hook_Init = nullptr;
{
swprintf_s(dllPath, L"%s\\Northstar.dll", exePath);
auto hHookModule = LoadLibraryExW(dllPath, 0, LOAD_WITH_ALTERED_SEARCH_PATH);
if (hHookModule) Hook_Init = GetProcAddress(hHookModule, "InitialiseNorthstar");
if (!hHookModule || Hook_Init == nullptr)
{
LibraryLoadError(GetLastError(), L"Northstar.dll", dllPath);
return false;
}
}
((bool (*)()) Hook_Init)();
return true;
}
typedef int(*LauncherMainType)(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLine, int nCmdShow);
LauncherMainType LauncherMainOriginal;
int LauncherMainHook(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLine, int nCmdShow)
{
if (ShouldLoadNorthstar())
LoadNorthstar();
return LauncherMainOriginal(hInstance, hPrevInstance, lpCmdLine, nCmdShow);
}
bool ProvisionNorthstar()
{
if (MH_Initialize() != MH_OK)
{
MessageBoxA(GetForegroundWindow(), "MH_Initialize failed\nThe game cannot continue and has to exit.", "Northstar Wsock32 Proxy Error", 0);
return false;
}
auto launcherHandle = GetModuleHandleA("launcher.dll");
if (!launcherHandle)
{
MessageBoxA(GetForegroundWindow(), "Launcher isn't loaded yet.\nThe game cannot continue and has to exit.", "Northstar Wsock32 Proxy Error", 0);
return false;
}
HookEnabler hook;
ENABLER_CREATEHOOK(hook, GetProcAddress(launcherHandle, "LauncherMain"), &LauncherMainHook, reinterpret_cast<LPVOID*>(&LauncherMainOriginal));
return true;
}
|
