moving to using unpacked for launcher

1 files changed, 92 insertions, 144 deletions

diff --git a/LauncherInjector/main.cpp b/LauncherInjector/main.cpp
index 86c50804..c335390b 100644
--- a/LauncherInjector/main.cpp
+++ b/LauncherInjector/main.cpp
@@ -8,155 +8,103 @@ namespace fs = std::filesystem;
 
 DWORD GetProcessByName(std::wstring processName)
 {
-	HANDLE snapshot = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
-
-	PROCESSENTRY32 processSnapshotEntry = { 0 };
-	processSnapshotEntry.dwSize = sizeof(PROCESSENTRY32);
-
-	if (snapshot == INVALID_HANDLE_VALUE)
-		return 0;
-
-	if (!Process32First(snapshot, &processSnapshotEntry))
-		return 0;
-
-	while (Process32Next(snapshot, &processSnapshotEntry))
-	{
-		if (!wcscmp(processSnapshotEntry.szExeFile, processName.c_str()))
-		{
-			CloseHandle(snapshot);
-			return processSnapshotEntry.th32ProcessID;
-		}
-	}
-	
-	CloseHandle(snapshot);
-	return 0;
-}
-
-void InjectInjectorIntoProcess(DWORD pid)
-{
-	HANDLE procHandle = OpenProcess(PROCESS_ALL_ACCESS, false, pid);
-
-	std::wstring path = (fs::current_path() / "GameInjector.dll").wstring();
-	size_t length = (path.length() + 1) * 2;
-	LPVOID lpLibName = VirtualAllocEx(procHandle, NULL, length, MEM_COMMIT, PAGE_EXECUTE_READWRITE);
-	WriteProcessMemory(procHandle, lpLibName, path.c_str(), length, 0);
-
-	// load minhook, since origin's loadlibrary won't load it from the tf directly by default
-	std::wstring minhookPath = (fs::current_path() / "MinHook.x86.dll").wstring();
-	size_t minhookLength = (minhookPath.length() + 1) * 2;
-	LPVOID lpMinhookLibName = VirtualAllocEx(procHandle, NULL, minhookLength, MEM_COMMIT, PAGE_EXECUTE_READWRITE);
-	WriteProcessMemory(procHandle, lpMinhookLibName, minhookPath.c_str(), minhookLength, 0);
+    HANDLE snapshot = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
 
-	HMODULE hKernel32 = GetModuleHandleW(L"kernel32.dll");
-	LPTHREAD_START_ROUTINE pLoadLibraryW = (LPTHREAD_START_ROUTINE)GetProcAddress(hKernel32, "LoadLibraryW");
-	HANDLE thread = CreateRemoteThread(procHandle, NULL, 0, pLoadLibraryW, lpMinhookLibName, 0, 0);
-	WaitForSingleObject(thread, INFINITE);
+    PROCESSENTRY32 processSnapshotEntry = { 0 };
+    processSnapshotEntry.dwSize = sizeof(PROCESSENTRY32);
 
-	thread = CreateRemoteThread(procHandle, NULL, 0, pLoadLibraryW, lpLibName, 0, 0);
-	WaitForSingleObject(thread, INFINITE);
+    if (snapshot == INVALID_HANDLE_VALUE)
+        return 0;
 
-	DWORD dwExitCode;
-	GetExitCodeThread(thread, &dwExitCode);
-	std::cout << dwExitCode << std::endl;
+    if (!Process32First(snapshot, &processSnapshotEntry))
+        return 0;
 
-	CloseHandle(procHandle);
+    while (Process32Next(snapshot, &processSnapshotEntry))
+    {
+        if (!wcscmp(processSnapshotEntry.szExeFile, processName.c_str()))
+        {
+            CloseHandle(snapshot);
+            return processSnapshotEntry.th32ProcessID;
+        }
+    }
 
-	std::cout << pid << std::endl;
+    CloseHandle(snapshot);
+    return 0;
 }
 
-void CreateAndHookUnpackedTitanfallProcess()
-{
-	PROCESS_INFORMATION tfPi;
-	memset(&tfPi, 0, sizeof(tfPi));
-	STARTUPINFO si;
-	memset(&si, 0, sizeof(si));
-	CreateProcessA("Titanfall2-unpacked.exe", (LPSTR)"", NULL, NULL, false, CREATE_SUSPENDED, NULL, NULL, (LPSTARTUPINFOA)&si, &tfPi);
-	
-	PROCESS_INFORMATION pi;
-	memset(&pi, 0, sizeof(pi));
-	memset(&si, 0, sizeof(si));
-
-	std::stringstream argStream;
-	argStream << tfPi.dwProcessId;
-	argStream << " ";
-	argStream << tfPi.dwThreadId;
-
-	CreateProcessA("InjectionProxy64.exe", (LPSTR)(argStream.str().c_str()), NULL, NULL, false, 0, NULL, NULL, (LPSTARTUPINFOA)&si, &pi);
-	WaitForSingleObject(pi.hThread, INFINITE);
-
-	CloseHandle(pi.hThread);
-	CloseHandle(tfPi.hThread);
-}
-
-int main()
-{
-	//AllocConsole();
-
-	// check if we're in the titanfall directory
-	if (!fs::exists("Titanfall2.exe") && !fs::exists("Titanfall2-unpacked.exe"))
-	{
-		MessageBox(NULL, L"Titanfall2.exe not found! Please launch from your titanfall 2 directory!", L"", MB_OK);
-		return 1;
-	}
-
-	// check for steam dll and unpacked exe
-	bool unpacked = fs::exists("Titanfall2-unpacked.exe");
-	//bool steamBuild = !unpacked && fs::exists("steam_api64.dll"); // dont actually need to check for this
-	
-	// unpacked origin
-	if (unpacked)
-	{
-		// check origin process
-		DWORD origin = GetProcessByName(L"Origin.exe");
-
-		if (!origin)
-		{
-			// unpacked exe will crash if origin isn't open on launch, so launch it
-			// get origin path from registry, code here is reversed from OriginSDK.dll
-			HKEY key;
-			if (RegOpenKeyExA(HKEY_LOCAL_MACHINE, "SOFTWARE\\WOW6432Node\\Origin", 0, KEY_READ, &key) != ERROR_SUCCESS)
-				return 1;
-
-			char originPath[520];
-			DWORD originPathLength = 520;
-			if (RegQueryValueExA(key, "ClientPath", 0, 0, (LPBYTE)&originPath, &originPathLength) != ERROR_SUCCESS)
-				return 1;
-
-			PROCESS_INFORMATION pi;
-			memset(&pi, 0, sizeof(pi));
-			STARTUPINFO si;
-			memset(&si, 0, sizeof(si));
-			CreateProcessA(originPath, (LPSTR)"", NULL, NULL, false, CREATE_DEFAULT_ERROR_MODE | CREATE_NEW_PROCESS_GROUP, NULL, NULL, (LPSTARTUPINFOA)&si, &pi);
-
-			// bit of a hack, but wait 12.5s to give origin a sec to init
-			// would be nice if we could do this dynamically, but idk how rn
-			Sleep(12500);
-		}
-		
-		CreateAndHookUnpackedTitanfallProcess();
-	}
-	// packed
-	else
-	{
-		// create a titanfall process, this will cause origin to start launching the game
-		// if we're on steam, origin will launch the steam release here, too
-		// we can't hook the titanfall process here unfortunately, since the titanfall process we create here dies when origin stuff starts
-
-		PROCESS_INFORMATION pi;
-		memset(&pi, 0, sizeof(pi));
-		STARTUPINFO si;
-		memset(&si, 0, sizeof(si));
-		CreateProcessA("Titanfall2.exe", (LPSTR)"", NULL, NULL, false, 0, NULL, NULL, (LPSTARTUPINFOA)&si, &pi);
-	
-		// hook launcher
-		DWORD launcherPID;
-		// dont actually need to check for steam, origin launches game no matter what
-		//if (steamBuild)
-		//	while (!(launcherPID = GetProcessByName(L"EASteamProxy.exe"))) Sleep(50);
-		//else
-		while (!(launcherPID = GetProcessByName(L"Origin.exe"))) Sleep(50);
-
-		// injector should clean itself up after its job is done
-		InjectInjectorIntoProcess(launcherPID);
-	}
+#define PROCESS_NAME L"Titanfall2-unpacked.exe"
+#define DLL_NAME L"Northstar.dll"
+
+int main() {
+    if (!fs::exists(PROCESS_NAME))
+    {
+        MessageBoxA(0, "Titanfall2-unpacked.exe not found! Please launch from your titanfall 2 directory and ensure you have Northstar installed correctly!", "", MB_OK);
+        return 1;
+    }
+
+    if (!fs::exists(DLL_NAME))
+    {
+        MessageBoxA(0, "Northstar.dll not found! Please launch from your titanfall 2 directory and ensure you have Northstar installed correctly!", "", MB_OK);
+        return 1;
+    }
+
+    if (!GetProcessByName(L"Origin.exe"))
+    {
+        // unpacked exe will crash if origin isn't open on launch, so launch it
+        // get origin path from registry, code here is reversed from OriginSDK.dll
+        HKEY key;
+        if (RegOpenKeyExA(HKEY_LOCAL_MACHINE, "SOFTWARE\\WOW6432Node\\Origin", 0, KEY_READ, &key) != ERROR_SUCCESS)
+        {
+            MessageBoxA(0, "Error: failed reading origin path!", "", MB_OK);
+            return 1;
+        }
+
+        char originPath[520];
+        DWORD originPathLength = 520;
+        if (RegQueryValueExA(key, "ClientPath", 0, 0, (LPBYTE)&originPath, &originPathLength) != ERROR_SUCCESS)
+        {
+            MessageBoxA(0, "Error: failed reading origin path!", "", MB_OK);
+            return 1;
+        }
+
+        PROCESS_INFORMATION pi;
+        memset(&pi, 0, sizeof(pi));
+        STARTUPINFO si;
+        memset(&si, 0, sizeof(si));
+        CreateProcessA(originPath, (LPSTR)"", NULL, NULL, false, CREATE_DEFAULT_ERROR_MODE | CREATE_NEW_PROCESS_GROUP, NULL, NULL, (LPSTARTUPINFOA)&si, &pi);
+
+        // wait for origin to be ready, this process is created when origin is ready enough to launch game without any errors
+        while (!GetProcessByName(L"OriginClientService.exe"))
+            Sleep(200);
+    }
+
+    STARTUPINFO startupInfo;
+    PROCESS_INFORMATION processInfo;
+
+    memset(&startupInfo, 0, sizeof(startupInfo));
+    memset(&processInfo, 0, sizeof(processInfo));
+
+    CreateProcessW(PROCESS_NAME, (LPWSTR)L"-multiple -novid", NULL, NULL, FALSE, CREATE_SUSPENDED, NULL, NULL, &startupInfo, &processInfo);
+
+    HMODULE hKernel32 = GetModuleHandleW(L"kernel32.dll");
+    LPTHREAD_START_ROUTINE pLoadLibraryW = (LPTHREAD_START_ROUTINE)GetProcAddress(hKernel32, "LoadLibraryW");
+
+    SIZE_T dwLength = (wcslen(DLL_NAME) + 1) * 2;
+    LPVOID lpLibName = VirtualAllocEx(processInfo.hProcess, NULL, dwLength, MEM_COMMIT, PAGE_READWRITE);
+
+    SIZE_T written = 0;
+    WriteProcessMemory(processInfo.hProcess, lpLibName, DLL_NAME, dwLength, &written);
+
+    HANDLE hThread = CreateRemoteThread(processInfo.hProcess, NULL, NULL, pLoadLibraryW, lpLibName, NULL, NULL);
+    WaitForSingleObject(hThread, INFINITE);
+    CloseHandle(hThread);
+
+    ResumeThread(processInfo.hThread);
+
+    VirtualFreeEx(processInfo.hProcess, lpLibName, dwLength, MEM_RELEASE);
+
+    CloseHandle(processInfo.hProcess);
+    CloseHandle(processInfo.hThread);
+
+    return 0;
 }
\ No newline at end of file