Added ConnectProtocol as module
This commit is contained in:
/dev/sdb
parent
f4469089f9
commit
571a2915d0
@ -24,6 +24,7 @@ namespace Components
|
||||
Loader::Register(new UIScript());
|
||||
Loader::Register(new FastFiles());
|
||||
Loader::Register(new Materials());
|
||||
Loader::Register(new ConnectProtocol());
|
||||
Loader::Register(new Singleton());
|
||||
Loader::Register(new FileSystem());
|
||||
Loader::Register(new QuickPatch());
|
||||
|
||||
@ -44,3 +44,4 @@ namespace Components
|
||||
#include "Modules\AssetHandler.hpp"
|
||||
#include "Modules\Localization.hpp"
|
||||
#include "Modules\MusicalTalent.hpp"
|
||||
#include "Modules\ConnectProtocol.hpp"
|
||||
|
||||
460
src/Components/Modules/ConnectProtocol.cpp
Normal file
460
src/Components/Modules/ConnectProtocol.cpp
Normal file
@ -0,0 +1,460 @@
|
||||
#include "..\..\STDInclude.hpp"
|
||||
#include <psapi.h>
|
||||
|
||||
namespace Components
|
||||
{
|
||||
#define MAX_PROCESSES 1024
|
||||
|
||||
int evaluated = 0;
|
||||
//Declarations SendMessage stuff
|
||||
DWORD proc_id = 0, proc_win_id = 0;
|
||||
HWND console, con_in;
|
||||
|
||||
bool ConnectProtocol::InstallProtocol()
|
||||
{
|
||||
HKEY hKey;
|
||||
LPCTSTR sk = TEXT("SOFTWARE\\Classes\\iw4x\\shell\\open\\command");
|
||||
LPCTSTR data = "URL:iw4x Protocol";
|
||||
LPCTSTR value = TEXT("URL Protocol");
|
||||
HMODULE hModule = GetModuleHandle(NULL);
|
||||
char ownPth[MAX_PATH];
|
||||
char workdir[MAX_PATH];
|
||||
char regred[MAX_PATH];
|
||||
DWORD dwsize = MAX_PATH;
|
||||
|
||||
if (hModule != NULL)
|
||||
{
|
||||
GetModuleFileName(hModule, ownPth, (sizeof(ownPth)));
|
||||
GetModuleFileName(hModule, workdir, (sizeof(workdir)));
|
||||
//////DBG(("EXE Path: %s", ownPth));
|
||||
}
|
||||
else
|
||||
{
|
||||
//////DBG("Cant get executable path");
|
||||
return false;
|
||||
}
|
||||
|
||||
char* endPtr = strstr(workdir, "iw4x.exe");
|
||||
*endPtr = 0;
|
||||
|
||||
////DBG(("EXE Path: %s", ownPth));
|
||||
////DBG(("EXE Path2: %s", workdir));
|
||||
|
||||
SetCurrentDirectory(workdir);
|
||||
|
||||
LONG openRes = RegOpenKeyEx(HKEY_CURRENT_USER, sk, 0, KEY_ALL_ACCESS, &hKey);
|
||||
if (openRes == ERROR_SUCCESS)
|
||||
{
|
||||
//Insert Check of the Key Value here, so the protocol will work even if the game was moved.
|
||||
openRes = RegQueryValueEx(hKey, 0, 0, 0, LPBYTE(regred), &dwsize);
|
||||
if (openRes == ERROR_SUCCESS)
|
||||
{
|
||||
char* endPt = strstr(regred, "\" \"%1\"");
|
||||
*endPt = 0;
|
||||
|
||||
char* regredPtr = regred;
|
||||
regredPtr++;
|
||||
|
||||
////DBG(("Reg Read1: %s", regredPtr));
|
||||
|
||||
|
||||
RegCloseKey(hKey);
|
||||
if (strcmp(regredPtr, ownPth))
|
||||
{
|
||||
////DBG("Protocol changed, reinstall");
|
||||
sk = TEXT("SOFTWARE\\Classes\\iw4x");
|
||||
openRes = RegOpenKeyEx(HKEY_CURRENT_USER, sk, 0, KEY_ALL_ACCESS, &hKey);
|
||||
if (openRes == ERROR_SUCCESS)
|
||||
{
|
||||
////DBG("Protocol is corrupted, reinstall");
|
||||
RegDeleteKey(hKey, 0);
|
||||
RegCloseKey(hKey);
|
||||
|
||||
}
|
||||
|
||||
}
|
||||
else{
|
||||
////DBG("Protocol is already installed");
|
||||
return true;
|
||||
}
|
||||
}
|
||||
else{
|
||||
openRes = RegOpenKeyEx(HKEY_CURRENT_USER, sk, 0, KEY_ALL_ACCESS, &hKey);
|
||||
if (openRes == ERROR_SUCCESS)
|
||||
{
|
||||
////DBG("Protocol is corrupted, reinstall");
|
||||
RegDeleteKey(hKey, 0);
|
||||
RegCloseKey(hKey);
|
||||
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
/*////DBG("Protocol is already installed");
|
||||
return true;*/
|
||||
|
||||
}
|
||||
else
|
||||
{
|
||||
sk = TEXT("SOFTWARE\\Classes\\iw4x");
|
||||
openRes = RegOpenKeyEx(HKEY_CURRENT_USER, sk, 0, KEY_ALL_ACCESS, &hKey);
|
||||
if (openRes == ERROR_SUCCESS)
|
||||
{
|
||||
////DBG("Protocol is corrupted, reinstall");
|
||||
RegDeleteKey(hKey, 0);
|
||||
RegCloseKey(hKey);
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
sk = TEXT("SOFTWARE\\Classes");
|
||||
data = "URL:iw4x Protocol";
|
||||
value = TEXT("URL Protocol");
|
||||
openRes = RegOpenKeyEx(HKEY_CURRENT_USER, sk, 0, KEY_ALL_ACCESS, &hKey);
|
||||
|
||||
if (openRes == ERROR_SUCCESS)
|
||||
{
|
||||
////DBG("Success opening SOFTWARE\\Classes.");
|
||||
sk = TEXT("iw4x");
|
||||
openRes = RegCreateKeyEx(hKey, sk, 0, 0, REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS, 0, &hKey, 0);
|
||||
|
||||
if (openRes == ERROR_SUCCESS)
|
||||
{
|
||||
////DBG("Success creating SOFTWARE\\Classes\\iw3mp.");
|
||||
openRes = RegSetValueEx(hKey, value, 0, REG_SZ, (LPBYTE)data, strlen(data) + 1);
|
||||
|
||||
if (openRes == ERROR_SUCCESS)
|
||||
{
|
||||
////DBG("Success writing URL:iw4x Protocol");
|
||||
data = TEXT("");
|
||||
openRes = RegSetValueEx(hKey, value, 0, REG_SZ, (LPBYTE)data, strlen(data) + 1);
|
||||
|
||||
if (openRes == ERROR_SUCCESS)
|
||||
{
|
||||
////DBG("Success writing URL Protocol");
|
||||
sk = TEXT("DefaultIcon");
|
||||
openRes = RegCreateKeyEx(hKey, sk, 0, 0, REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS, 0, &hKey, 0);
|
||||
|
||||
if (openRes == ERROR_SUCCESS)
|
||||
{
|
||||
////DBG("Success creating SOFTWARE\\Classes\\iw3mp\\DefaultIcon");
|
||||
data = Utils::VA("%s,1", ownPth);
|
||||
openRes = RegSetValueEx(hKey, 0, 0, REG_SZ, (LPBYTE)data, strlen(data) + 1);
|
||||
|
||||
if (openRes == ERROR_SUCCESS)
|
||||
{
|
||||
openRes = RegCloseKey(hKey);
|
||||
|
||||
if (openRes == ERROR_SUCCESS)
|
||||
{
|
||||
sk = TEXT("SOFTWARE\\Classes\\iw4x");
|
||||
openRes = RegOpenKeyEx(HKEY_CURRENT_USER, sk, 0, KEY_ALL_ACCESS, &hKey);
|
||||
|
||||
if (openRes == ERROR_SUCCESS)
|
||||
{
|
||||
sk = TEXT("shell\\open\\command");
|
||||
openRes = RegCreateKeyEx(hKey, sk, 0, 0, REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS, 0, &hKey, 0);
|
||||
|
||||
if (openRes == ERROR_SUCCESS)
|
||||
{
|
||||
data = Utils::VA("\"%s\" \"%s\"", ownPth, "%1");
|
||||
////DBG(("Command is %s", data));
|
||||
openRes = RegSetValueEx(hKey, 0, 0, REG_SZ, (LPBYTE)data, strlen(data) + 1);
|
||||
|
||||
if (openRes == ERROR_SUCCESS)
|
||||
{
|
||||
RegCloseKey(hKey);
|
||||
|
||||
}
|
||||
else
|
||||
{
|
||||
////DBG("Error writing shell command to registry");
|
||||
RegCloseKey(hKey);
|
||||
return false;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
////DBG("Error creating Key shell\\open\\command");
|
||||
RegCloseKey(hKey);
|
||||
return false;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
////DBG("Error opening SOFTWARE\\Classes\\iw3mp");
|
||||
RegCloseKey(hKey);
|
||||
return false;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
////DBG("Error closing DefaultIcon Key");
|
||||
RegCloseKey(hKey);
|
||||
return false;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
////DBG("Error writing EXE Path,1 to DefaultIcon Key");
|
||||
RegCloseKey(hKey);
|
||||
return false;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
////DBG("Error creating subkey DefaultIcon");
|
||||
RegCloseKey(hKey);
|
||||
return false;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
////DBG("Error writing URL Protocol");
|
||||
RegCloseKey(hKey);
|
||||
return false;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
////DBG("Error writing URL:iw4x Protocol Code: %d", openRes);
|
||||
RegCloseKey(hKey);
|
||||
return false;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
////DBG("Error creating key SOFTWARE\\Classes\\iw3mp");
|
||||
RegCloseKey(hKey);
|
||||
return false;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
////DBG("Error opening key.");
|
||||
return false;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
void ConnectProtocol::EvaluateProtocol()
|
||||
{
|
||||
if (evaluated) return;
|
||||
evaluated = 1;
|
||||
|
||||
char* args = GetCommandLineA();
|
||||
char* substr = strstr(args, "iw4x://");
|
||||
|
||||
if (!substr || substr == args)
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
|
||||
substr += 8;
|
||||
char* substr2 = strstr(substr, "/");
|
||||
*substr2 = 0;
|
||||
////DBG(("Connecting to: %s", substr));
|
||||
|
||||
Game::Cbuf_AddText(0, Utils::VA("connect %s;", substr));
|
||||
}
|
||||
|
||||
BOOL ConnectProtocol::InvokeConnect()
|
||||
{
|
||||
|
||||
char* args = GetCommandLineA();
|
||||
char* substr = strstr(args, "iw4x://");
|
||||
|
||||
////DBG("Mutex give us a Handle = %d", mutex);
|
||||
////DBG("Last Error = %d", GetLastError());
|
||||
if (GetLastError() == ERROR_ALREADY_EXISTS)
|
||||
{
|
||||
if (!substr || substr == args)
|
||||
{
|
||||
// Not started using the protocol
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
//Now we have to get the Window Handle of the Console and the Handle of the Input field
|
||||
////DBG("The Game is already running");
|
||||
FindEditHandle("iw4x.exe");
|
||||
|
||||
substr += 7;
|
||||
char* substr2 = strstr(substr, "/");
|
||||
*substr2 = 0;
|
||||
if (proc_id != 0)
|
||||
{
|
||||
|
||||
if (con_in != NULL)
|
||||
{
|
||||
SendMessageA(con_in, WM_SETTEXT, NULL, (LPARAM)Utils::VA("connect %s;", substr));
|
||||
SendMessageA(con_in, WM_CHAR, 0xD, 0);
|
||||
}
|
||||
|
||||
}
|
||||
else
|
||||
{
|
||||
return FALSE;
|
||||
////DBG("Did not find Process by Name iw4x");
|
||||
}
|
||||
|
||||
////DBG("Exit this Instance");
|
||||
return TRUE;
|
||||
|
||||
}
|
||||
else
|
||||
{
|
||||
if (substr && substr != args)
|
||||
{
|
||||
// Skip intro
|
||||
*(BYTE*)0x60BECF = 0xEB;
|
||||
return FALSE;
|
||||
}
|
||||
return FALSE;
|
||||
}
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
|
||||
|
||||
ConnectProtocol::ConnectProtocol()
|
||||
{
|
||||
ConnectProtocol::InstallProtocol();
|
||||
|
||||
}
|
||||
|
||||
|
||||
//Send Connect Command to running iw4x instance
|
||||
BOOL CALLBACK enumWindowsProc(__in HWND hWnd, __in LPARAM lParam) {
|
||||
|
||||
DWORD id = GetWindowThreadProcessId(hWnd, &id);
|
||||
DWORD id2 = GetWindowThreadProcessId(FindWindowFromProcessId(proc_id), &id2);
|
||||
//DBG("process id: %d %d Real ID: %d", id, id2, proc_id);
|
||||
if (id == id2)
|
||||
{
|
||||
//printf("THEY MATCH!process id: %d\n", id);
|
||||
char buffer[256];
|
||||
GetWindowText(hWnd, (LPSTR)buffer, 255);
|
||||
char* endPtr = strstr(buffer, "Console");
|
||||
if (endPtr != 0)
|
||||
{
|
||||
//DBG(("Got Process Window Handle: %d, Window Text = %s", hWnd, buffer));
|
||||
console = hWnd;
|
||||
EnumChildWindows(console, EnumChildProc, NULL);
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
}
|
||||
else{
|
||||
//DBG(("Got no Process Window Handle!!!!!!!!!!!!"));
|
||||
}
|
||||
|
||||
return TRUE;
|
||||
}
|
||||
|
||||
BOOL CALLBACK EnumChildProc(HWND hwnd, LPARAM lParam) {
|
||||
char buffer[256];
|
||||
|
||||
GetClassName(hwnd, (LPSTR)buffer, 255);
|
||||
char* endPtr = strstr(buffer, "Edit");
|
||||
if (endPtr != 0)
|
||||
{
|
||||
con_in = hwnd;
|
||||
//DBG("Got Handle of Edit Field: %d", hwnd);
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
|
||||
return TRUE;
|
||||
|
||||
}
|
||||
|
||||
struct EnumData {
|
||||
DWORD dwProcessId;
|
||||
HWND hWnd;
|
||||
};
|
||||
|
||||
BOOL CALLBACK EnumProc(HWND hWnd, LPARAM lParam) {
|
||||
// Retrieve storage location for communication data
|
||||
EnumData& ed = *(EnumData*)lParam;
|
||||
DWORD dwProcessId = 0x0;
|
||||
// Query process ID for hWnd
|
||||
GetWindowThreadProcessId(hWnd, &dwProcessId);
|
||||
// Apply filter - if you want to implement additional restrictions,
|
||||
// this is the place to do so.
|
||||
if (ed.dwProcessId == dwProcessId) {
|
||||
// Found a window matching the process ID
|
||||
ed.hWnd = hWnd;
|
||||
// Report success
|
||||
SetLastError(ERROR_SUCCESS);
|
||||
// Stop enumeration
|
||||
return FALSE;
|
||||
}
|
||||
// Continue enumeration
|
||||
return TRUE;
|
||||
}
|
||||
|
||||
void FindEditHandle(__in_z LPCTSTR lpcszFileName)
|
||||
{
|
||||
LPDWORD lpdwProcessIds;
|
||||
LPTSTR lpszBaseName;
|
||||
HANDLE hProcess;
|
||||
DWORD i, cdwProcesses, dwProcessId = 0;
|
||||
|
||||
lpdwProcessIds = (LPDWORD)HeapAlloc(GetProcessHeap(), 0, MAX_PROCESSES*sizeof(DWORD));
|
||||
if (lpdwProcessIds != NULL)
|
||||
{
|
||||
if (EnumProcesses(lpdwProcessIds, MAX_PROCESSES*sizeof(DWORD), &cdwProcesses))
|
||||
{
|
||||
lpszBaseName = (LPTSTR)HeapAlloc(GetProcessHeap(), 0, MAX_PATH*sizeof(TCHAR));
|
||||
if (lpszBaseName != NULL)
|
||||
{
|
||||
cdwProcesses /= sizeof(DWORD);
|
||||
for (i = 0; i < cdwProcesses; i++)
|
||||
{
|
||||
hProcess = OpenProcess(PROCESS_QUERY_INFORMATION | PROCESS_VM_READ, FALSE, lpdwProcessIds[i]);
|
||||
if (hProcess != NULL)
|
||||
{
|
||||
if (GetModuleBaseName(hProcess, NULL, lpszBaseName, MAX_PATH) > 0)
|
||||
{
|
||||
//DBG(("Process Name = %s", lpszBaseName));
|
||||
if (!lstrcmpi(lpszBaseName, lpcszFileName))
|
||||
{
|
||||
//DBG("Stage 8");
|
||||
dwProcessId = lpdwProcessIds[i];
|
||||
CloseHandle(hProcess);
|
||||
proc_id = dwProcessId;
|
||||
EnumWindows(enumWindowsProc, NULL);
|
||||
if (con_in != NULL)
|
||||
{
|
||||
break;
|
||||
}
|
||||
|
||||
}
|
||||
}
|
||||
CloseHandle(hProcess);
|
||||
}
|
||||
}
|
||||
HeapFree(GetProcessHeap(), 0, (LPVOID)lpszBaseName);
|
||||
}
|
||||
}
|
||||
HeapFree(GetProcessHeap(), 0, (LPVOID)lpdwProcessIds);
|
||||
}
|
||||
//DBG("Return %d", dwProcessId);
|
||||
//return dwProcessId;
|
||||
}
|
||||
// Main entry
|
||||
HWND FindWindowFromProcessId(DWORD dwProcessId) {
|
||||
EnumData ed = { dwProcessId };
|
||||
if (!EnumWindows(EnumProc, (LPARAM)&ed) &&
|
||||
(GetLastError() == ERROR_SUCCESS)) {
|
||||
return ed.hWnd;
|
||||
}
|
||||
return NULL;
|
||||
}
|
||||
// Helper method for convenience
|
||||
HWND FindWindowFromProcess(HANDLE hProcess) {
|
||||
return FindWindowFromProcessId(GetProcessId(hProcess));
|
||||
}
|
||||
}
|
||||
19
src/Components/Modules/ConnectProtocol.hpp
Normal file
19
src/Components/Modules/ConnectProtocol.hpp
Normal file
@ -0,0 +1,19 @@
|
||||
namespace Components
|
||||
{
|
||||
class ConnectProtocol : public Component
|
||||
{
|
||||
public:
|
||||
ConnectProtocol();
|
||||
void EvaluateProtocol();
|
||||
static BOOL InvokeConnect();
|
||||
private:
|
||||
static bool InstallProtocol();
|
||||
|
||||
};
|
||||
//Additional Functions for InvokeConnect
|
||||
void FindEditHandle(__in_z LPCTSTR lpcszFileName);
|
||||
BOOL CALLBACK EnumProc(HWND hWnd, LPARAM lParam);
|
||||
BOOL CALLBACK EnumChildProc(HWND hwnd, LPARAM lParam);
|
||||
HWND FindWindowFromProcessId(DWORD dwProcessId);
|
||||
HWND FindWindowFromProcess(HANDLE hProcess);
|
||||
}
|
||||
@ -14,10 +14,24 @@ namespace Components
|
||||
if (Dedicated::IsDedicated()) return;
|
||||
|
||||
Singleton::FirstInstance = (CreateMutex(NULL, FALSE, "iw4x_mutex") && GetLastError() != ERROR_ALREADY_EXISTS);
|
||||
|
||||
if (!Singleton::FirstInstance && MessageBoxA(0, "Do you want to start a second instance?", "Game already running", MB_ICONEXCLAMATION | MB_YESNO) == IDNO)
|
||||
//Checking if the instance is for the connect protocol
|
||||
if (!Singleton::FirstInstance)
|
||||
{
|
||||
ExitProcess(0);
|
||||
if (ConnectProtocol::InvokeConnect() == TRUE)
|
||||
{
|
||||
//Connect command was successfuly sent to the first instance, exiting the second game instance now.
|
||||
ExitProcess(0);
|
||||
}
|
||||
else
|
||||
{
|
||||
//No connect command was provided, continuing with normal processing.
|
||||
if (!Singleton::FirstInstance && MessageBoxA(0, "Do you want to start a second instance?", "Game already running", MB_ICONEXCLAMATION | MB_YESNO) == IDNO)
|
||||
{
|
||||
ExitProcess(0);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
Loading…
Reference in New Issue
Block a user