void CQuickConnect::OnSubmit()
{
String strHost;
unsigned short usPort;
if(!CMainMenu::GetHostAndPort(m_GUIElements.pIPEditBox->getText().c_str(), strHost, usPort))
{
g_pClient->GetGUI()->ShowMessageBox("You must enter a valid host and port.", "Error");
return;
}
// Get the password
String strPassword(m_GUIElements.pPasswordEditBox->getText().c_str());
// Set the ip, port and password
CVAR_SET_STRING("ip", strHost);
CVAR_SET_INTEGER("port", usPort);
CVAR_SET_STRING("pass", strPassword);
// Hide the quick connect window
SetVisible(false);
// Call the connect function
CMainMenu::GetSingleton()->OnConnect(strHost, usPort, strPassword);
}
// setWeather(weather)
SQInteger CServerNatives::SetWeather(SQVM * pVM)
{
SQInteger iWeather;
sq_getinteger(pVM, 2, &iWeather);
if(iWeather >= 1 && iWeather <= 10 && iWeather != CVAR_GET_INTEGER("weather"))
{
CVAR_SET_INTEGER("weather", (int)iWeather);
CBitStream bsSend;
bsSend.Write((unsigned char)iWeather);
g_pNetworkManager->RPC(RPC_ScriptingSetPlayerWeather, &bsSend, PRIORITY_HIGH, RELIABILITY_RELIABLE_ORDERED, INVALID_ENTITY_ID, true);
sq_pushbool(pVM, true);
return 1;
}
sq_pushbool(pVM, false);
return 1;
}
bool CSettingsMenu::OnSaveButtonClick(const CEGUI::EventArgs &eventArgs)
{
bool bWindowed = ((CEGUI::Checkbox *)m_GUIElements.pWindowedCheckBox)->isSelected();
bool bShowFPS = ((CEGUI::Checkbox *)m_GUIElements.pFPSCheckBox)->isSelected();
String strNick(m_GUIElements.pNickEditBox->getText().c_str());
String strFont(m_GUIElements.pChatFontEditBox->getText().c_str());
String strFontSize(m_GUIElements.pChatFontSizeEditBox->getText().c_str());
String strBackgroundAlpha(m_GUIElements.pChatBGColorAEditBox->getText().c_str());
String strBackgroundRed(m_GUIElements.pChatBGColorREditBox->getText().c_str());
String strBackgroundGreen(m_GUIElements.pChatBGColorGEditBox->getText().c_str());
String strBackgroundBlue(m_GUIElements.pChatBGColorBEditBox->getText().c_str());
CVAR_SET_BOOL("windowed", bWindowed);
CVAR_SET_BOOL("fps", g_pClient->GetFPSToggle());
g_pClient->SetFPSToggle(bShowFPS);
CVAR_SET_INTEGER("chatfont", strFont.ToInteger());
CVAR_SET_INTEGER("chatsize", strFontSize.ToInteger());
CVAR_SET_INTEGER("chatbga", strBackgroundAlpha.ToInteger());
CVAR_SET_INTEGER("chatbgr", strBackgroundRed.ToInteger());
CVAR_SET_INTEGER("chatbgg", strBackgroundGreen.ToInteger());
CVAR_SET_INTEGER("chatbgb", strBackgroundBlue.ToInteger());
// Init the chat window font and background
g_pClient->GetChatWindow()->InitFontAndBackground();
if(CVAR_GET_STRING("nick").Compare(strNick) != 0)
{
CVAR_SET_STRING("nick", strNick);
g_pClient->SetNick(strNick);
CNetworkManager * pNetworkManager = g_pClient->GetNetworkManager();
if(pNetworkManager && pNetworkManager->IsConnected())
{
CBitStream bsSend;
bsSend.Write(strNick);
pNetworkManager->RPC(RPC_NameChange, &bsSend, PRIORITY_HIGH, RELIABILITY_RELIABLE_ORDERED);
}
}
if(bWindowed != g_pClient->IsWindowedMode())
g_pClient->GetGUI()->ShowMessageBox("You must restart IV:MP for the windowed mode option to take effect", "Information.");
SetVisible(false);
return true;
}
void CClient::OnD3DEndScene()
{
// Check for pause menu
if(m_pNetworkManager)
{
if(CGame::IsMenuActive() && CGame::GetState() == GAME_STATE_INGAME && m_pNetworkManager->IsConnected())
CGame::SetState(GAME_STATE_PAUSE_MENU);
else if(!CGame::IsMenuActive() && CGame::GetState() == GAME_STATE_PAUSE_MENU && m_pNetworkManager->IsConnected())
CGame::SetState(GAME_STATE_INGAME);
}
// Are we in the main menu?
if(CGame::GetState() == GAME_STATE_MAIN_MENU || CGame::GetState() == GAME_STATE_LOADING)
{
// Is the main menu shown?
if(CGame::IsMenuActive())
{
// Clear the screen
m_pDevice->Clear(0, NULL, D3DCLEAR_TARGET, D3DCOLOR_ARGB(255, 0, 0, 0), 1.0, 0);
}
}
if(m_pMainMenu && CGame::GetState() == GAME_STATE_MAIN_MENU || CGame::GetState() == GAME_STATE_IVMP_PAUSE_MENU)
{
if(!m_pMainMenu->IsVisible())
{
if(m_pNetworkManager && m_pNetworkManager->IsConnected())
{
m_pMainMenu->SetDisconnectButtonVisible(true);
m_pMainMenu->SetVisible(true);
}
else
{
m_pMainMenu->SetDisconnectButtonVisible(false);
m_pMainMenu->SetVisible(true);
}
if(CGame::GetState() == GAME_STATE_MAIN_MENU)
{
String strPostPath("/getlatestversion.php");
m_pHttpClient->Get(strPostPath);
if(strcmp(m_strConnectHost,"0.0.0.0"))
{
// Send connect event
m_pMainMenu->OnDirectConnect(m_strConnectHost,m_usConnectPort);
// Remove current connect entries from xml file
CVAR_SET_STRING("currentconnect_server","0.0.0.0");
CVAR_SET_INTEGER("currentconnect_port",9999);
}
}
}
}
else
{
if(m_pMainMenu->IsVisible())
{
m_pMainMenu->SetVisible(false);
m_pChatWindow->SetEnabled(true);
}
}
if(m_pClientScriptManager && m_pClientScriptManager->GetGUIManager())
{
if(CGame::GetState() != GAME_STATE_INGAME)
{
if(!m_pClientScriptManager->GetGUIManager()->IsHidden())
m_pClientScriptManager->GetGUIManager()->Hide();
}
else
{
if(m_pClientScriptManager->GetGUIManager()->IsHidden())
m_pClientScriptManager->GetGUIManager()->Show();
}
}
// If our GUI class exists render it
if(m_pGUI)
m_pGUI->Render();
// If our main menu exists process it
if(m_pMainMenu)
m_pMainMenu->Process();
// If our chat window exist draw it
if(m_pChatWindow && !CGame::IsMenuActive())
m_pChatWindow->Draw();
// If our fps class exists update it
if(m_pFPSCounter)
m_pFPSCounter->Pulse();
// If our credits class exists process it
if(m_pCredits)
m_pCredits->Process();
if(m_p3DLabelManager)
m_p3DLabelManager->Render();
// If our scripting manager exists, call the frame event
if(m_pEvents && !m_pMainMenu->IsVisible())
m_pEvents->Call("frameRender");
#ifdef IVMP_DEBUG
if(m_pDebugView && m_pGUI && m_pLocalPlayer)
m_pDebugView->Draw();
#endif
// Check if our screen shot write failed
if(CScreenShot::IsDone())
{
if(CScreenShot::HasSucceeded())
m_pChatWindow->AddInfoMessage("Screen shot written (%s).", CScreenShot::GetWriteName().Get());
else
m_pChatWindow->AddInfoMessage("Screen shot write failed (%s).", CScreenShot::GetError().Get());
CScreenShot::Reset();
}
if(CGame::GetState() == GAME_STATE_MAIN_MENU)
{
if(m_pHttpClient->IsBusy())
m_pHttpClient->Process();
if(m_pHttpClient->GotData())
{
// Get the data
String * strData = m_pHttpClient->GetData();
// Did we get any data?
if(!strData->IsEmpty())
{
#ifndef IVMP_DEBUG
if(strcmp(strData->Get(), MOD_VERSION_STRING))
{
if(m_pGUI)
m_pGUI->ShowMessageBox(String("A new version of IV:MP is available (Version %s)", strData->Get()).Get(), "New Version available");
}
#endif
}
}
}
// Are we in game?
if(CGame::GetState() == GAME_STATE_INGAME)
{
// Is F4 held down and do we have a network manager?
if(GetAsyncKeyState(VK_F4) && m_pNetworkManager && m_pActorManager)
{
// Get the network statistics
CNetStats * pNetStats = m_pNetworkManager->GetNetClient()->GetNetStats();
// Convert the network statistics to a string
char szNetworkStats[10000];
pNetStats->ToString(szNetworkStats, 2);
// Create the statistics string
String strStats(szNetworkStats);
// Append loaded and unloaded model counts to the stats
// jenksta: too performance heavy to be done every frame
//strStats.AppendF("Models (Loaded/Unload): %d/%d\n", CGame::GetLoadedModelCount(), CGame::GetUnloadedModelCount());
// Append streamed in/out entity counts and streamed in limits to the stats
strStats.AppendF("Players (StreamedIn/StreamedInLimit): %d/%d\n", m_pStreamer->GetStreamedInEntityCountOfType(STREAM_ENTITY_PLAYER)+1, m_pStreamer->GetStreamedInLimitOfType(STREAM_ENTITY_PLAYER));
strStats.AppendF("Vehicles (StreamedIn/StreamedInLimit): %d/%d\n", m_pStreamer->GetStreamedInEntityCountOfType(STREAM_ENTITY_VEHICLE), m_pStreamer->GetStreamedInLimitOfType(STREAM_ENTITY_VEHICLE));
strStats.AppendF("Actors (Actor-Count/Max-Actors): %d/%d\n", m_pActorManager->GetActorCount(), MAX_ACTORS);
strStats.AppendF("Pickups (StreamedIn/StreamedInLimit): %d/%d\n", m_pStreamer->GetStreamedInEntityCountOfType(STREAM_ENTITY_PICKUP), m_pStreamer->GetStreamedInLimitOfType(STREAM_ENTITY_PICKUP));
strStats.AppendF("Objects (StreamedIn/StreamedInLimit): %d/%d\n", m_pStreamer->GetStreamedInEntityCountOfType(STREAM_ENTITY_OBJECT), m_pStreamer->GetStreamedInLimitOfType(STREAM_ENTITY_OBJECT));
strStats.AppendF("Checkpoints (StreamedIn/StreamedInLimit): %d/%d\n", m_pStreamer->GetStreamedInEntityCountOfType(STREAM_ENTITY_CHECKPOINT), m_pStreamer->GetStreamedInLimitOfType(STREAM_ENTITY_CHECKPOINT));
// Draw the string
m_pGUI->DrawText(strStats, CEGUI::Vector2(26, 30), (CEGUI::colour)D3DCOLOR_RGBA(255, 255, 255, 255), m_pGUI->GetFont("tahoma-bold", 10));
m_pChatWindow->SetEnabled(false);
m_bNetworkStatsDisplayed = true;
}
else
{
// If our network stats were displayed
if(m_bNetworkStatsDisplayed)
{
if(m_pChatWindow && !m_pChatWindow->IsEnabled())
m_pChatWindow->SetEnabled(true);
m_bNetworkStatsDisplayed = false;
}
// If our input window exists draw it
if(m_pInputWindow)
m_pInputWindow->Draw();
}
// Update the time
if(m_pTime)
{
// jenksta: Setting this every frame causes the weather to change
// rapidly
unsigned char ucHour = 0, ucMinute = 0;
m_pTime->GetTime(&ucHour, &ucMinute);
CGame::SetTime(ucHour, ucMinute);
CGame::SetDayOfWeek(m_pTime->GetDayOfWeek());
}
#ifdef IVMP_DEBUG
if(m_pVehicleManager)
{
CVector3 vecWorldPosition;
Vector2 vecScreenPosition;
for(EntityId i = 0; i < MAX_VEHICLES; i++)
{
CNetworkVehicle * pVehicle = m_pVehicleManager->Get(i);
if(pVehicle)
{
if(!pVehicle->IsStreamedIn())
continue;
if(!pVehicle->IsOnScreen())
continue;
pVehicle->GetPosition(vecWorldPosition);
if(!CGame::GetScreenPositionFromWorldPosition(vecWorldPosition, vecScreenPosition))
continue;
int iHealth = pVehicle->GetHealth();
int iModel = pVehicle->GetEngineState();
float fPetrol = pVehicle->GetPetrolTankHealth();
CVector3 vecPos; pVehicle->GetPosition(vecPos);
CVector3 vecRot; pVehicle->GetRotation(vecRot);
m_pGUI->DrawText(String("VehicleId %d, Enginestate: %d, Health: %d, PetrolTankHealth: %f\nPosition(%.3f,%.3f,%.3f), Rot(%.3f,%.3f,%.3f)", i, iModel, iHealth, fPetrol, vecPos.fX,vecPos.fY,vecPos.fZ,vecRot.fX,vecRot.fY,vecRot.fZ), CEGUI::Vector2(vecScreenPosition.fX, vecScreenPosition.fY));
}
}
}
#endif
if(m_pNameTags)
m_pNameTags->Draw();
}
}
int WINAPI WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLine, int nShowCmd)
{
// Get the GTA IV install directory from the registry
char szInstallDirectory[MAX_PATH];
bool bFoundCustomDirectory = false, bRenewProtocol = false;
std::string strReNewEntries = lpCmdLine;
if(SharedUtility::ReadRegistryString(HKEY_CURRENT_USER, "Software\\IVMP", "gtaivdir", NULL,
szInstallDirectory, sizeof(szInstallDirectory)) ||
!SharedUtility::Exists(szInstallDirectory))
{
char szProtocolDirectory[MAX_PATH];
CString strCommand = CString("\"%s\" \"%%1\"",SharedUtility::GetAbsolutePath("Client.Launcher.exe"));
if(strcmp(szProtocolDirectory, strCommand.Get()))
bRenewProtocol = true;
}
// Check if protocol 'ivmp' and 'ivmultiplayer' is avaiable in registry
if(!SharedUtility::ReadRegistryString(HKEY_CLASSES_ROOT, "ivmp", NULL, "", NULL, NULL)
|| !SharedUtility::ReadRegistryString(HKEY_CLASSES_ROOT, "ivmultiplayer", NULL, "", NULL, NULL)
|| bRenewProtocol)
{
// Update
SharedUtility::WriteRegistryString(HKEY_CLASSES_ROOT,"ivmp","","IVMultiplayer",strlen("IVMultiplayer"));
SharedUtility::WriteRegistryString(HKEY_CLASSES_ROOT,"ivmultiplayer","","IVMultiplayer",strlen("IVMultiplayer"));
CString strcommand = CString("\"%s\" \"%%1\"",SharedUtility::GetAbsolutePath("Client.Launcher.exe"));
SharedUtility::WriteRegistryString(HKEY_CLASSES_ROOT,"ivmp","Url Protocol","",0);
SharedUtility::WriteRegistryString(HKEY_CLASSES_ROOT,"ivmp\\shell\\open\\command\\","",strcommand.GetData(),strcommand.GetLength());
SharedUtility::WriteRegistryString(HKEY_CLASSES_ROOT,"ivmp\\DefaultIcon","", CString("Client.Launcher.exe,1").GetData(),strlen("Client.Launcher.exe,1"));
SharedUtility::WriteRegistryString(HKEY_CLASSES_ROOT,"ivmultiplayer","Url Protocol","",0);
SharedUtility::WriteRegistryString(HKEY_CLASSES_ROOT,"ivmultiplayer\\shell\\open\\command\\","",strcommand.GetData(),strcommand.GetLength());
SharedUtility::WriteRegistryString(HKEY_CLASSES_ROOT,"ivmultiplayer\\DefaultIcon","", CString("Client.Launcher.exe,1").GetData(),strlen("Client.Launcher.exe,1"));
}
// TODO: Steam registry entry support(or the client should just pick the directory via data browser)
if(!SharedUtility::ReadRegistryString(HKEY_LOCAL_MACHINE, "Software\\Rockstar Games\\Grand Theft Auto IV",
"InstallFolder", NULL, szInstallDirectory, sizeof(szInstallDirectory)) ||
!SharedUtility::Exists(szInstallDirectory)) {
if(!SharedUtility::ReadRegistryString(HKEY_CURRENT_USER, "Software\\IVMP", "gtaivdir", NULL,
szInstallDirectory, sizeof(szInstallDirectory)) ||
!SharedUtility::Exists(szInstallDirectory)) {
if(ShowMessageBox("Failed to retrieve GTA IV install directory from registry. Specify your GTA IV path now?",
(MB_ICONEXCLAMATION | MB_OKCANCEL)) == IDOK) {
// Taken from http://vcfaq.mvps.org/sdk/20.htm
BROWSEINFO browseInfo = { 0 };
browseInfo.lpszTitle = "Pick a Directory";
ITEMIDLIST * pItemIdList = SHBrowseForFolder(&browseInfo);
if(pItemIdList != NULL) {
// Get the name of the selected folder
if(SHGetPathFromIDList(pItemIdList, szInstallDirectory))
bFoundCustomDirectory = true;
// Free any memory used
IMalloc * pIMalloc = 0;
if(SUCCEEDED(SHGetMalloc(&pIMalloc))) {
pIMalloc->Free(pItemIdList);
pIMalloc->Release();
}
}
}
if(!bFoundCustomDirectory) {
ShowMessageBox("Failed to retrieve GTA IV install directory from registry or browser window. Cannot launch IV: Multiplayer.");
return 1;
}
}
}
// Check if we have the 'multiplayer' directory, if not: create it.
char szExecutablePath[MAX_PATH];
sprintf_s(szExecutablePath,SharedUtility::GetAbsolutePath("").Get(),sizeof(MAX_PATH));
CString strMultiplayerPath = szExecutablePath;
strMultiplayerPath.AppendF ("multiplayer");
SharedUtility::CreateDirectoryA(strMultiplayerPath.Get());
strMultiplayerPath = szExecutablePath;
strMultiplayerPath.AppendF ("multiplayer\\common");
SharedUtility::CreateDirectoryA(strMultiplayerPath.Get());
SharedUtility::CreateDirectoryA(strMultiplayerPath.Get());
strMultiplayerPath = szExecutablePath;
strMultiplayerPath.AppendF ("multiplayer\\pc");
SharedUtility::CreateDirectoryA(strMultiplayerPath.Get());
SharedUtility::CreateDirectoryA(strMultiplayerPath.Get());
strMultiplayerPath = szExecutablePath;
strMultiplayerPath.AppendF ("multiplayer\\pc\\data");
SharedUtility::CreateDirectoryA(strMultiplayerPath.Get());
strMultiplayerPath = szExecutablePath;
strMultiplayerPath.AppendF ("multiplayer\\common\\data");
SharedUtility::CreateDirectoryA(strMultiplayerPath.Get());
strMultiplayerPath = szExecutablePath;
strMultiplayerPath.AppendF ("multiplayer\\common\\data\\effects");
SharedUtility::CreateDirectoryA(strMultiplayerPath.Get());
strMultiplayerPath = szExecutablePath;
strMultiplayerPath.AppendF ("multiplayer\\pc\\textures");
SharedUtility::CreateDirectoryA(strMultiplayerPath.Get());
strMultiplayerPath = szExecutablePath;
strMultiplayerPath.AppendF ("multiplayer\\pc\\data\\eflc");
SharedUtility::CreateDirectoryA(strMultiplayerPath.Get());
Sleep(500);
// Check for eflc dir
char szEFLCDirectory[MAX_PATH];
bool bFoundCustomEFLCDirectory = false;
char szUsingEFLC[MAX_PATH];
bool bUsingEFLC;
if(SharedUtility::ReadRegistryString(HKEY_CURRENT_USER, "Software\\IVMP", "usingeflc", "1", szUsingEFLC, sizeof(szUsingEFLC)))
{
SharedUtility::WriteRegistryString(HKEY_CURRENT_USER, "Software\\IVMP", "usingeflc", "1", 1);
bUsingEFLC = true;
}
else
bUsingEFLC = 0;
if(!bUsingEFLC) {
if(ShowMessageBox("Do you want to use EFCL import map function?", MB_ICONQUESTION | MB_YESNO ) == IDYES) {
bUsingEFLC = true;
SharedUtility::WriteRegistryString(HKEY_CURRENT_USER, "Software\\IVMP", "usingeflc", "1", 1);
}
}
if(bUsingEFLC)
{
if(!SharedUtility::ReadRegistryString(HKEY_CURRENT_USER, "Software\\IVMP", "eflcdir", NULL, szEFLCDirectory, sizeof(szEFLCDirectory))
|| !SharedUtility::Exists(szEFLCDirectory)) {
if(SharedUtility::ReadRegistryString(HKEY_LOCAL_MACHINE, "Software\\Rockstar Games\\EFLC", "InstallFolder", NULL, szEFLCDirectory, sizeof(szEFLCDirectory))
|| SharedUtility::Exists(szEFLCDirectory)) {
bFoundCustomEFLCDirectory = true;
}
if(!SharedUtility::ReadRegistryString(HKEY_LOCAL_MACHINE, "Software\\Rockstar Games\\EFLC", "InstallFolder", NULL, szEFLCDirectory, sizeof(szEFLCDirectory))
|| !SharedUtility::Exists(szEFLCDirectory)) {
if(ShowMessageBox("Failed to retrieve GTA IV: EFLC install directory from registry. Specify your GTA IV: EFLC path now? If you do not have EFLC installed, just click No.",
(MB_ICONEXCLAMATION | MB_OKCANCEL)) == IDOK) {
// Taken from http://vcfaq.mvps.org/sdk/20.htm
BROWSEINFO browseInfo = { 0 };
browseInfo.lpszTitle = "Pick a Directory";
ITEMIDLIST * pItemIdList = SHBrowseForFolder(&browseInfo);
if(pItemIdList != NULL) {
// Get the name of the selected folder
if(SHGetPathFromIDList(pItemIdList, szEFLCDirectory))
{
bFoundCustomEFLCDirectory = true;
SharedUtility::WriteRegistryString(HKEY_CURRENT_USER, "Software\\IVMP", "usingeflc", "1", 1);
}
// Free any memory used
IMalloc * pIMalloc = 0;
if(SUCCEEDED(SHGetMalloc(&pIMalloc))) {
pIMalloc->Free(pItemIdList);
pIMalloc->Release();
}
}
}
if (!bFoundCustomEFLCDirectory)
SharedUtility::WriteRegistryString(HKEY_CURRENT_USER, "Software\\IVMP", "usingeflc", "0", 1);
}
}
}
if(bUsingEFLC)
{
CGameFiles::CheckFiles();
PROCESS_INFORMATION ProcessInfo = PROCESS_INFORMATION();
STARTUPINFO StartupInfo;
ZeroMemory(&StartupInfo, sizeof(StartupInfo));
StartupInfo.cb = sizeof StartupInfo;
char * szPath = new char[MAX_PATH];
sprintf(szPath,CString("%s",SharedUtility::GetAbsolutePath("\\IVGameReady.exe").Get()).Get());
CreateProcess(szPath, NULL,NULL,NULL, FALSE, NULL, NULL, NULL, &StartupInfo, &ProcessInfo);
WaitForSingleObject(ProcessInfo.hProcess, INFINITE);
}
// Get the full path to LaunchGTAIV.exe
CString strApplicationPath("%s\\LaunchGTAIV.exe", szInstallDirectory);
// Check if LaunchGTAIV.exe exists
if(!SharedUtility::Exists(strApplicationPath.Get()))
return ShowMessageBox("Failed to find LaunchGTAIV.exe. Cannot launch IV: Multiplayer.");
// If we have a custom directory save it
if(bFoundCustomDirectory)
SharedUtility::WriteRegistryString(HKEY_CURRENT_USER, "Software\\IVMP", "gtaivdir", szInstallDirectory, strlen(szInstallDirectory));
if(bFoundCustomEFLCDirectory)
SharedUtility::WriteRegistryString(HKEY_CURRENT_USER, "Software\\IVMP", "eflcdir", szEFLCDirectory, strlen(szEFLCDirectory));
// Get the full path of the client core
CString strClientCore(SharedUtility::GetAbsolutePath(CLIENT_CORE_NAME DEBUG_SUFFIX LIBRARY_EXTENSION));
// Check if the client core exists
if(!SharedUtility::Exists(strClientCore.Get()))
return ShowMessageBox("Failed to find " CLIENT_CORE_NAME DEBUG_SUFFIX LIBRARY_EXTENSION ". Cannot launch IV: Multiplayer.");
// Get the full path of the launch helper
CString strLaunchHelper(SharedUtility::GetAbsolutePath(CLIENT_LAUNCH_HELPER_NAME DEBUG_SUFFIX LIBRARY_EXTENSION));
// Check if the launch helper exists
if(!SharedUtility::Exists(strLaunchHelper.Get()))
return ShowMessageBox("Failed to find " CLIENT_LAUNCH_HELPER_NAME DEBUG_SUFFIX LIBRARY_EXTENSION". Cannot launch IV: Multiplayer.");
// Check if GTAIV is already running
if(SharedUtility::IsProcessRunning("GTAIV.exe")) {
if(ShowMessageBox("GTAIV is already running and needs to be terminated before IV: Multiplayer can be started. Do you want to do that now?",
MB_ICONQUESTION | MB_YESNO ) == IDYES) {
if(!SharedUtility::_TerminateProcess("GTAIV.exe"))
return ShowMessageBox("GTAIV.exe could not be terminated. Cannot launch IV: Multiplayer.");
}
else
return ShowMessageBox("GTAIV.exe is already running. Cannot launch IV: Multiplayer.");
}
// Check if LaunchGTAIV.exe is already running
if(SharedUtility::IsProcessRunning("LaunchGTAIV.exe")) {
if(ShowMessageBox("LaunchGTAIV is already running and needs to be terminated before IV: Multiplayer can be started. Do you want to do that now?",
MB_ICONQUESTION | MB_YESNO ) == IDYES) {
if(!SharedUtility::_TerminateProcess("LaunchGTAIV.exe")) {
// Wait until we've successfully terminated the process
Sleep(3000);
if(SharedUtility::IsProcessRunning("LaunchGTAIV.exe")) {
if(!SharedUtility::_TerminateProcess("LaunchGTAIV.exe"))
return ShowMessageBox("LaunchGTAIV.exe could not be terminated. Cannot launch IV: Multiplayer.");
}
}
}
else
return ShowMessageBox("LaunchGTAIV.exe is already running. Cannot launch IV: Multiplayer.");
}
// TODO ADD WINDOW COMMANDLINE SUPPORT!
// Check if we have an server connect command
CString strServer, strPort;
std::string strServerCheck = CString(lpCmdLine);
std::size_t sizetCMDFound = strServerCheck.find("-ivmp");// -[1]i[2]v[3]m[4]p[5]*space*[6]***.***.***.***
int iOffset = 0;
bool bCommandFound = false;
CString strNewCommandLine = lpCmdLine;
// Check for shortcut commandline
if(sizetCMDFound != std::string::npos) {
iOffset = 6;
bCommandFound = true;
}
// Check for ivmp protocol
if(!bCommandFound) {
sizetCMDFound = strServerCheck.find("ivmp://"); // i[1]v[2]m[3]p[4]:[5]/[6]/[7]***.***.***
if(sizetCMDFound != std::string::npos) {
iOffset = 7;
bCommandFound = true;
}
}
// Check for ivmultiplayer protocol
if(!bCommandFound) {
sizetCMDFound = strServerCheck.find("ivmultiplayer://");// i[1]v[2]m[3]u[4]l[5]t[6]i[7]p[8]l[9]a[10]y[11]e[12]r[13]:[14]/[15]/[16]***.***.***
if(sizetCMDFound != std::string::npos) {
iOffset = 16;
bCommandFound = true;
}
}
// Open default clientsettings
CSettings::Open(SharedUtility::GetAbsolutePath(CLIENT_SETTINGS_FILE), true, true, true);
// If we have found an direct connect force
if(bCommandFound)
{
std::string strServerInst = strServerCheck.substr(sizetCMDFound+iOffset,strServerCheck.length());
std::size_t sizetCMDFound_2 = strServerInst.find(":");
// Have we an : in our instruction
if(sizetCMDFound_2 != std::string::npos) {
// Grab our connect data
strServer = CString("%s",strServerInst.substr(0,sizetCMDFound_2).c_str());
strPort = CString("%s",strServerInst.substr(sizetCMDFound_2+1,strServerInst.length()).c_str());
// Parse the command line
CSettings::ParseCommandLine(GetCommandLine());
// Write connect data to settings xml
CVAR_SET_STRING("currentconnect_server",strServer.Get());
CVAR_SET_INTEGER("currentconnect_port",strPort.ToInteger());
// Generate new commandline
strNewCommandLine = CString("%s -directconnect", lpCmdLine);
}
else // Something is wrong with our URI
{
if(ShowMessageBox("Something is wrong with your server direct-connect URI, do you want to start IV:MP without direct-connect?", MB_ICONQUESTION | MB_YESNO ) == IDYES)
{
// Set default server direct connect values
CVAR_SET_STRING("currentconnect_server","0.0.0.0");
CVAR_SET_INTEGER("currentconnect_port",9999);
strNewCommandLine = lpCmdLine;
}
else // Terminate IV:MP
{
if(!SharedUtility::_TerminateProcess("Client.Launcher.exe"))
return ShowMessageBox("LaunchGTAIV.exe could not be terminated. Cannot launch IV: Multiplayer.");
}
}
}
else {
CSettings::ParseCommandLine(GetCommandLine());
// If we haven't found a server connect command, delte the old instructions( if the client had crashed before )
CVAR_SET_STRING("currentconnect_server","0.0.0.0");
CVAR_SET_INTEGER("currentconnect_port",9999);
}
// Close settings...
CSettings::Close();
// Generate the command line
CString strCommandLine("%s %s", strApplicationPath.Get(), strNewCommandLine.Get());
// Start LaunchGTAIV.exe
STARTUPINFO siStartupInfo;
PROCESS_INFORMATION piProcessInfo;
memset(&siStartupInfo, 0, sizeof(siStartupInfo));
memset(&piProcessInfo, 0, sizeof(piProcessInfo));
siStartupInfo.cb = sizeof(siStartupInfo);
if(!CreateProcess(strApplicationPath.Get(), (char *)strCommandLine.Get(), NULL, NULL, TRUE, CREATE_SUSPENDED, NULL,
SharedUtility::GetAppPath(), &siStartupInfo, &piProcessInfo)) {
ShowMessageBox("Failed to start LaunchGTAIV.exe. Cannot launch IV: Multiplayer.");
return 1;
}
// Inject LauncherLibrary.dll into LaunchGTAIV.exe
int iReturn = SharedUtility::InjectLibraryIntoProcess(piProcessInfo.hProcess, strLaunchHelper.Get());
// Did the injection fail?
if(iReturn > 0) {
// Terminate the process
TerminateProcess(piProcessInfo.hProcess, 0);
// Show the error message
CString strError("Unknown error. Cannot launch IV: Multiplayer.");
if(iReturn == 1)
strError = "Failed to write library path into remote process. Cannot launch IV: Multiplayer.";
else if(iReturn == 2)
strError = "Failed to create remote thread in remote process. Cannot launch IV: Multiplayer.";
else if(iReturn == 3)
strError = "Failed to open the remote process, Cannot launch IV: Multiplayer.";
ShowMessageBox(strError.Get());
return 1;
}
// Resume the LaunchGTAIV.exe thread
ResumeThread(piProcessInfo.hThread);
return 0;
}
int WINAPI WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLine, int nShowCmd)
{
// Get the GTA IV install directory from the registry
char szInstallDirectory[MAX_PATH];
bool bFoundCustomDirectory = false, bRenewProtocol = false;
std::string strReNewEntries = lpCmdLine;
char szProtocolDirectory[MAX_PATH];
CString strCommand = CString("\"%s\" \"%%1\"",SharedUtility::GetAbsolutePath(MP_START_EXECUTABLE));
if(strcmp(szProtocolDirectory, strCommand.Get()))
bRenewProtocol = true;
// Check if protocol 'ivn' and 'ivnetwork' is avaiable in registry
if(!SharedUtility::ReadRegistryString(HKEY_CLASSES_ROOT, SHORT_URI_LAUNCH_3, NULL, "", NULL, NULL)
|| !SharedUtility::ReadRegistryString(HKEY_CLASSES_ROOT, SHORT_URI_LAUNCH_4, NULL, "", NULL, NULL) || bRenewProtocol) {
SharedUtility::WriteRegistryString(HKEY_CLASSES_ROOT,SHORT_URI_LAUNCH_3,"",CLIENT_CORE_NAME,strlen(CLIENT_CORE_NAME));
SharedUtility::WriteRegistryString(HKEY_CLASSES_ROOT,SHORT_URI_LAUNCH_4,"",CLIENT_CORE_NAME,strlen(CLIENT_CORE_NAME));
CString strcommand = CString("\"%s\" \"%%1\"",SharedUtility::GetAbsolutePath(MP_START_EXECUTABLE));
SharedUtility::WriteRegistryString(HKEY_CLASSES_ROOT,SHORT_URI_LAUNCH_3 ,"Url Protocol","",0);
SharedUtility::WriteRegistryString(HKEY_CLASSES_ROOT,SHORT_URI_LAUNCH_3"\\shell\\open\\command\\","",CString("%s",strcommand.Get()).GetData(),strcommand.GetLength());
SharedUtility::WriteRegistryString(HKEY_CLASSES_ROOT,SHORT_URI_LAUNCH_3"\\DefaultIcon","", CString(MP_START_EXECUTABLE" ,1").GetData(),strlen(MP_START_EXECUTABLE" ,1"));
SharedUtility::WriteRegistryString(HKEY_CLASSES_ROOT,SHORT_URI_LAUNCH_4,"Url Protocol","",0);
SharedUtility::WriteRegistryString(HKEY_CLASSES_ROOT,SHORT_URI_LAUNCH_4"\\shell\\open\\command\\","",CString("%s",strcommand.Get()).GetData(),strcommand.GetLength());
SharedUtility::WriteRegistryString(HKEY_CLASSES_ROOT,SHORT_URI_LAUNCH_4"\\DefaultIcon","", CString(MP_START_EXECUTABLE" ,1").GetData(),strlen(MP_START_EXECUTABLE" ,1"));
}
if(!SharedUtility::ReadRegistryString(HKEY_LOCAL_MACHINE, DEFAULT_REGISTRY_GAME_DIRECTORY,
"InstallFolder", NULL, szInstallDirectory, sizeof(szInstallDirectory)) ||
!SharedUtility::Exists(szInstallDirectory)) {
if(!SharedUtility::ReadRegistryString(HKEY_CURRENT_USER, REGISTRY_AREA, GAME_DIRECTORY, NULL,
szInstallDirectory, sizeof(szInstallDirectory)) ||
!SharedUtility::Exists(szInstallDirectory)) {
if(ShowMessageBox("Failed to retrieve the install directory from "GAME_DEFAULT_EXECUTABLE"'s registry. Specify your game path now?",
(MB_ICONEXCLAMATION | MB_OKCANCEL)) == IDOK) {
BROWSEINFO browseInfo = { 0 };
browseInfo.lpszTitle = MOD_NAME" - Pick a Directory";
ITEMIDLIST * pItemIdList = SHBrowseForFolder(&browseInfo);
if(pItemIdList != NULL) {
// Get the name of the selected folder
if(SHGetPathFromIDList(pItemIdList, szInstallDirectory))
bFoundCustomDirectory = true;
// Free any memory used
IMalloc * pIMalloc = 0;
if(SUCCEEDED(SHGetMalloc(&pIMalloc))) {
pIMalloc->Free(pItemIdList);
pIMalloc->Release();
}
}
}
if(!bFoundCustomDirectory) {
ShowMessageBox("Failed to retrieve the install directory from registry or browser window. Cannot launch "MOD_NAME".");
return 1;
}
}
}
// Create basic directories for extracting files(..)
SharedUtility::CreateBasicMPDirectories();
Sleep(500);
// Get the full path to EFLC.exe
CString strApplicationPath("%s\\"GAME_DEFAULT_EXECUTABLE, szInstallDirectory);
// Check if EFLC.exe exists
if(!SharedUtility::Exists(strApplicationPath.Get()))
return ShowMessageBox("Failed to find "GAME_DEFAULT_EXECUTABLE". Cannot launch "MOD_NAME".");
// If we have a custom directory save it
if(bFoundCustomDirectory)
SharedUtility::WriteRegistryString(HKEY_CURRENT_USER, REGISTRY_AREA, GAME_DIRECTORY, szInstallDirectory, strlen(szInstallDirectory));
// Get the full path of the client core
CString strClientCore(SharedUtility::GetAbsolutePath(CLIENT_CORE_NAME DEBUG_SUFFIX LIBRARY_EXTENSION));
// Check if the client core exists
if(!SharedUtility::Exists(strClientCore.Get()))
return ShowMessageBox("Failed to find " CLIENT_CORE_NAME DEBUG_SUFFIX LIBRARY_EXTENSION ". Cannot launch "MOD_NAME".");
// Get the full path of the launch helper
CString strCore(SharedUtility::GetAbsolutePath(CLIENT_CORE_NAME DEBUG_SUFFIX LIBRARY_EXTENSION));
// Check if the launch helper exists
if (!SharedUtility::Exists(strCore.Get()))
return ShowMessageBox("Failed to find " CLIENT_CORE_NAME DEBUG_SUFFIX LIBRARY_EXTENSION". Cannot launch "MOD_NAME".");
// Get the full path of the launch helper
CString strBass(SharedUtility::GetAbsolutePath("bass.dll"));
// Check if the launch helper exists
if (!SharedUtility::Exists(strBass.Get()))
return ShowMessageBox("Failed to find bass.dll. Cannot launch "MOD_NAME".");
// Check if GTAEFLC is already running
if (SharedUtility::IsProcessRunning(GAME_DEFAULT_EXECUTABLE))
{
if (ShowMessageBox(GAME_DEFAULT_EXECUTABLE" is already running and needs to be terminated before "MOD_NAME" can be started. Do you want to do that now?",
MB_ICONQUESTION | MB_YESNO) == IDYES)
{
if (!SharedUtility::_TerminateProcess(GAME_DEFAULT_EXECUTABLE))
{
if (ShowMessageBox("Do you want to start it?",
MB_ICONQUESTION | MB_YESNO) == IDYES) {
}
else
{
return ShowMessageBox(GAME_DEFAULT_EXECUTABLE" could not be terminated. Cannot launch "MOD_NAME".");
}
}
}
else
return ShowMessageBox(GAME_DEFAULT_EXECUTABLE" is already running. Cannot launch "MOD_NAME".");
}
// TODO ADD WINDOW COMMANDLINE SUPPORT!
// Check if we have an server connect command
CString strServer, strPort;
std::string strServerCheck = CString(lpCmdLine);
std::size_t sizetCMDFound = strServerCheck.find(SHORT_COMMANDLINE_LAUNCH_1);
int iOffset = 0;
bool bCommandFound = false;
CString strNewCommandLine = lpCmdLine;
// Check for shortcut commandline
if(sizetCMDFound != std::string::npos) {
iOffset = 6;
bCommandFound = true;
}
// Check for ivn protocol
if(!bCommandFound) {
sizetCMDFound = strServerCheck.find(SHORT_URI_LAUNCH_1);
if(sizetCMDFound != std::string::npos) {
iOffset = 7;
bCommandFound = true;
}
}
// Check for ivmultiplayer protocol
if(!bCommandFound) {
sizetCMDFound = strServerCheck.find(SHORT_URI_LAUNCH_2);
if(sizetCMDFound != std::string::npos) {
iOffset = 16;
bCommandFound = true;
}
}
// Open default clientsettings
CSettings::Open(SharedUtility::GetAbsolutePath(CLIENT_SETTINGS_FILE), true, true, true);
// If we have found an direct connect force
if(bCommandFound)
{
std::string strServerInst = strServerCheck.substr(sizetCMDFound+iOffset,strServerCheck.length());
std::size_t sizetCMDFound_2 = strServerInst.find(":");
// Have we an : in our instruction
if(sizetCMDFound_2 != std::string::npos) {
// Grab our connect data
strServer = CString("%s",strServerInst.substr(0,sizetCMDFound_2).c_str());
strPort = CString("%s",strServerInst.substr(sizetCMDFound_2+1,strServerInst.length()).c_str());
// Parse the command line
CSettings::ParseCommandLine(GetCommandLine());
// Write connect data to settings xml
CVAR_SET_STRING("currentconnect_server",strServer.Get());
CVAR_SET_INTEGER("currentconnect_port",strPort.ToInteger());
// Generate new commandline
strNewCommandLine = CString("%s -directconnect", lpCmdLine);
}
else // Something is wrong with our URI
{
if(ShowMessageBox("Something is wrong with your server direct-connect URI, do you want to start "MOD_NAME" without direct-connect?", MB_ICONQUESTION | MB_YESNO ) == IDYES)
{
// Set default server direct connect values
CVAR_SET_STRING("currentconnect_server","0.0.0.0");
CVAR_SET_INTEGER("currentconnect_port",9999);
strNewCommandLine = lpCmdLine;
}
else // Terminate IV:N
{
if(!SharedUtility::_TerminateProcess(MP_START_EXECUTABLE))
return ShowMessageBox(MP_START_EXECUTABLE" could not be terminated. Cannot launch IV:Network.");
}
}
}
else {
CSettings::ParseCommandLine(GetCommandLine());
// If we haven't found a server connect command, delte the old instructions(if the client had crashed before)
CVAR_SET_STRING("currentconnect_server","0.0.0.0");
CVAR_SET_INTEGER("currentconnect_port",9999);
}
// Close settings...
CSettings::Close();
// Generate the command line
CString strCommandLine("%s %s", strApplicationPath.Get(), strNewCommandLine.Get());
// Start LaunchGTAIV.exe
STARTUPINFO siStartupInfo;
PROCESS_INFORMATION piProcessInfo;
memset(&siStartupInfo, 0, sizeof(siStartupInfo));
memset(&piProcessInfo, 0, sizeof(piProcessInfo));
siStartupInfo.cb = sizeof(siStartupInfo);
if (!CreateProcess(strApplicationPath.Get(), (char *) strCommandLine.Get(), NULL, NULL, TRUE, CREATE_SUSPENDED, NULL, szInstallDirectory, &siStartupInfo, &piProcessInfo)) {
ShowMessageBox("Failed to start "GAME_DEFAULT_EXECUTABLE". Cannot launch "MOD_NAME".");
return 1;
}
// Inject bass.dll into EFLC.exe
int iReturn = SharedUtility::InjectLibraryIntoProcess(piProcessInfo.hProcess, strBass.Get());
// Inject IVNetwork.dll into EFLC.exe
iReturn += SharedUtility::InjectLibraryIntoProcess(piProcessInfo.hProcess, strCore.Get());
// Did the injection fail?
if(iReturn > 0) {
// Terminate the process
TerminateProcess(piProcessInfo.hProcess, 0);
// Show the error message
CString strError("Unknown error. Cannot launch "MOD_NAME".");
if(iReturn == 1)
strError = "Failed to write library path into remote process. Cannot launch "MOD_NAME".";
else if(iReturn == 2)
strError = "Failed to create remote thread in remote process. Cannot launch "MOD_NAME".";
else if(iReturn == 3)
strError = "Failed to open the remote process, Cannot launch "MOD_NAME".";
ShowMessageBox(strError.Get());
return 1;
}
// Resume the thread
ResumeThread(piProcessInfo.hThread);
return 0;
}
