DWORD Q3_ConnectToMasterServer(GAME_INFO *pGI, int iMasterIdx)
{
size_t packetlen=0;
char sendbuf[80];
ZeroMemory(sendbuf,sizeof(sendbuf));
SOCKET ConnectSocket;
// UINT_PTR timerProgressWait = SetTimer(g_hWnd,IDT_TIMER_1SECOND,100,NULL);
int i = 0;
unsigned char *packet[MAX_PACKETS];
size_t packet_len[MAX_PACKETS];
for(i=0; i<MAX_PACKETS;i++)
packet[i] = NULL;
i = 0;
char szIP[260];
strcpy(szIP,pGI->szMasterServerIP[iMasterIdx]);
SplitIPandPORT(szIP,pGI->dwMasterServerPORT);
int len = 0;//(int)strlen(sendbuf);
string query;
query = pGI->szMasterQueryString;
ReplaceStrInStr(query,"%PROTOCOL%",pGI->szGameProtocol[iMasterIdx]);
len = UTILZ_ConvertEscapeCodes((TCHAR*)query.c_str(),sendbuf,sizeof(sendbuf));
ConnectSocket = getsockudp(szIP,(unsigned short)pGI->dwMasterServerPORT); // etmaster.idsoftware.com"27950 master server
if(INVALID_SOCKET==ConnectSocket)
{
KillTimer(g_hWnd,IDT_1SECOND);
g_log.AddLogInfo(GS_LOG_ERROR,"Error connecting to server %s:%d!",szIP,pGI->dwMasterServerPORT);
return 1;
}
WSAEVENT hEvent;
hEvent = WSACreateEvent();
if (hEvent == WSA_INVALID_EVENT)
{
KillTimer(g_hWnd,IDT_1SECOND);
g_log.AddLogInfo(GS_LOG_ERROR,"WSACreateEvent() = WSA_INVALID_EVENT");
closesocket(ConnectSocket);
return 1;
}
//
// Make the socket non-blocking and
// associate it with network events
//
int nRet;
nRet = WSAEventSelect(ConnectSocket, hEvent,FD_READ|FD_CONNECT|FD_CLOSE);
if (nRet == SOCKET_ERROR)
{
KillTimer(g_hWnd,IDT_1SECOND);
g_log.AddLogInfo(GS_LOG_ERROR,"EventSelect() = SOCKET_ERROR");
closesocket(ConnectSocket);
WSACloseEvent(hEvent);
return 5;
}
//
// Handle async network events
//
WSANETWORKEVENTS events;
while(1)
{
//
// Wait for something to happen
//
//dbg_print("\nWaitForMultipleEvents()");
DWORD dwRet;
dwRet = WSAWaitForMultipleEvents(1, &hEvent, FALSE,4000,FALSE);
if (dwRet == WSA_WAIT_TIMEOUT)
{
g_log.AddLogInfo(GS_LOG_ERROR,"WSAWaitForMultipleEvents = WSA_WAIT_TIMEOUT");
break;
}
//
// Figure out what happened
//
//g_log.AddLogInfo(0,"\nWSAEnumNetworkEvents()");
nRet = WSAEnumNetworkEvents(ConnectSocket, hEvent, &events);
if (nRet == SOCKET_ERROR)
{
g_log.AddLogInfo(GS_LOG_ERROR,"WSAEnumNetworkEvents() = SOCKET_ERROR");
break;
}
// //
// Handle events //
// //
// Connect event?
if (events.lNetworkEvents & FD_CONNECT)
{
//g_log.AddLogInfo(0,"\nFD_CONNECT: %d", events.iErrorCode[FD_CONNECT_BIT]);
g_log.AddLogInfo(0,"Master server %s",pGI->szMasterServerIP[0]);
g_log.AddLogInfo(0,"Sending packet string: [%s] Length: %d",sendbuf,len);
g_log.AddLogInfo(0,"xml config: %s ",pGI->szMasterQueryString);
if(send(ConnectSocket, sendbuf, len , 0)==SOCKET_ERROR)
{
KillTimer(g_hWnd,IDT_1SECOND);
WSACloseEvent(hEvent);
closesocket(ConnectSocket);
g_log.AddLogInfo(GS_LOG_ERROR,"Error sending master query packet!");
return 2;
}
}
// Read event?
if (events.lNetworkEvents & FD_READ)
{
//g_log.AddLogInfo(0,"\nFD_READ: %d, %d",events.iErrorCode[FD_READ_BIT],i);
// Read the data and write it to stdout
packet[i]=(unsigned char*)ReadPacket(ConnectSocket, &packetlen);
packet_len[i] = packetlen;
i++;
if(i>=MAX_PACKETS)
break;
}
// Close event?
if (events.lNetworkEvents & FD_CLOSE)
{
g_log.AddLogInfo(0,"\nFD_CLOSE: %d",events.iErrorCode[FD_CLOSE_BIT]);
break;
}
/*
// Write event?
if (events.lNetworkEvents & FD_WRITE)
{
g_log.AddLogInfo(0,"\nFD_WRITE: %d",events.iErrorCode[FD_WRITE_BIT]);
}
*/
}
closesocket(ConnectSocket);
WSACloseEvent(hEvent);
DWORD Q3_dwNewTotalServers=0;
for(i=0; i<MAX_PACKETS;i++)
{
if(packet[i] != NULL)
{
Q3_dwNewTotalServers += Q3_ParseServers((char*)packet[i],packet_len[i],pGI);
free(packet[i]);
packet[i]=NULL;
SetStatusText(pGI->iIconIndex,g_lang.GetString("StatusReceivingMaster"),Q3_dwNewTotalServers,pGI->szGAME_NAME);
if(g_bCancel)
{
//ensure to clean up
for(i=0; i<MAX_PACKETS;i++)
{
if(packet[i] != NULL)
{
free(packet[i]);
packet[i]=NULL;
}
}
break;
}
Sleep(50); //give cpu sometime to breath
}
}
// KillTimer(g_hWnd,IDT_TIMER_1SECOND);
pGI->dwTotalServers += Q3_dwNewTotalServers;
return 0;
}
//Create scanning threads
void Initialize_Rescan2(GAME_INFO *pGI, bool (*filterServerItem)(SERVER_INFO *lp,GAME_INFO *pGI, vFILTER_SETS *vFilterSets))
{
g_log.AddLogInfo(GS_LOG_DEBUG,"Entering Initialize_Rescan2 function.");
//SCAN_FilterServerItem = filterServerItem;
vFILTER_SETS vFS;
vFS = pGI->vFilterSets;
vFS.insert(vFS.end(),gm.GetFilterSet(GLOBAL_FILTER).begin(),gm.GetFilterSet(GLOBAL_FILTER).end());
sort(vFS.begin(),vFS.end(),Sort_Filter_By_GroupName);
vSRV_INF::iterator iLst;
pGI->vRefScanSI.clear();
pGI->dwViewFlags |= SCAN_SERVERLIST;
for ( iLst = pGI->vSI.begin( ); iLst != pGI->vSI.end( ); iLst++ )
{
SERVER_INFO *pSI = *iLst;
REF_SERVER_INFO refSI;
refSI.pServerInfo = pSI;
if(filterServerItem!=NULL)
{
if(filterServerItem(pSI,pGI,&vFS))
{
pGI->vRefScanSI.push_back(refSI);
}
}
else
{
pGI->vRefScanSI.push_back(refSI);
}
}
if(pGI->dwViewFlags & SCAN_SERVERLIST)
pGI->dwViewFlags ^= SCAN_SERVERLIST;
g_log.AddLogInfo(GS_LOG_INFO,"Preparing to scan %d servers of a total %d.",pGI->vRefScanSI.size(),pGI->vSI.size());
if(pGI->dwViewFlags & FORCE_SCAN_FILTERED)
pGI->dwViewFlags = 0;
if(g_hwndProgressBar!=NULL)
{
//Initililze Progressbar
SendMessage(g_hwndProgressBar, PBM_SETSTEP, (WPARAM) 1, 0);
SendMessage(g_hwndProgressBar, PBM_SETRANGE, (WPARAM) 0,MAKELPARAM(0,pGI->vRefScanSI.size()));
SendMessage(g_hwndProgressBar, PBM_SETPOS, (WPARAM) 0, 0);
}
HANDLE hThreadIndex[MAX_THREADS];
DWORD dwThreadId[MAX_THREADS];
for(DWORD i=0; i<AppCFG.dwThreads;i++)
hThreadIndex[i]=NULL;
SCANNER_dwThreadCounter=0;
pGI->dwScanIdx = 0;
//Create and setup Continue Event, this is important! Otherwise ETSV can crash.
SCAN_hContinueEvent = CreateEvent(NULL,TRUE,TRUE,"ScanContinueEvent");
if (SCAN_hContinueEvent == NULL)
g_log.AddLogInfo(GS_LOG_DEBUG,"CreateEvent failed (%d)\n", GetLastError());
if (! ResetEvent(SCAN_hContinueEvent) )
g_log.AddLogInfo(GS_LOG_DEBUG,"ResetEvent failed (%d)\n", GetLastError());
DWORD dwMaxThreads = (AppCFG.dwThreads>pGI->vRefScanSI.size())?pGI->vRefScanSI.size():AppCFG.dwThreads;
//---------------------------------
//Multi thread scanning code
//---------------------------------
//Startup the threads!!!
for (DWORD i=0; i<dwMaxThreads;i++)
{
hThreadIndex[SCANNER_dwThreadCounter] = CreateThread( NULL, 0, &Get_ServerStatusThread2, (LPVOID) pGI ,0, &dwThreadId[SCANNER_dwThreadCounter]);
if (hThreadIndex[SCANNER_dwThreadCounter] == NULL)
{
dbg_print("Error creating thread!\n");
}
else
{
//g_log.AddLogInfo(GS_LOG_INFO,"Thread created! %d",i);
EnterCriticalSection( &SCANNER_CSthreadcounter );
SCANNER_dwThreadCounter++;
LeaveCriticalSection( &SCANNER_CSthreadcounter );
//SetThreadName(dwThreadId[SCANNER_dwThreadCounter], "Get_ServerStatusThread2");
}
}
//All threads created in graceful way and counter increased properly
if (! SetEvent(SCAN_hContinueEvent) )
dbg_print("SetEvent failed\n");
//After this this the thread counter can decrease properly with a noncorrupted handle
DWORD dwStartTick = GetTickCount();
DWORD iWaitIndex = 0;
DWORD i=0;
//Wait for all threads to finish...
//g_log.AddLogInfo(GS_LOG_DEBUG,"AppCFG.dwThreads %d",AppCFG.dwThreads);
while(iWaitIndex<dwMaxThreads)
{
DWORD max = ((dwMaxThreads-iWaitIndex)<MAXIMUM_WAIT_OBJECTS)?(dwMaxThreads-iWaitIndex):MAXIMUM_WAIT_OBJECTS;
// g_log.AddLogInfo(GS_LOG_DEBUG,"iWaitIndex: %d, iWaitIndex+max: %d, dwMaxThreads: %d, max:%d",iWaitIndex,iWaitIndex+max,dwMaxThreads,max);
DWORD dwEvent = WaitForMultipleObjects(max, &hThreadIndex[iWaitIndex], TRUE, INFINITE);
//g_log.AddLogInfo(GS_LOG_DEBUG,">iWaitIndex: %d, iWaitIndex+max: %d, dwMaxThreads: %d, nThreads: %d, max:%d",iWaitIndex,iWaitIndex+max,dwMaxThreads, nThreads,max);
switch (dwEvent)
{
case WAIT_OBJECT_0:
//dbg_print("First event was signaled.\n");
break;
case WAIT_TIMEOUT:
g_log.AddLogInfo(GS_LOG_DEBUG,"Wait timed out. @ %s",__FUNCTION__);
break;
// Return value is invalid.
default:
{
g_log.AddLogInfo(GS_LOG_DEBUG,"Error at waiting threads! %s",__FUNCTION__);
}
}
// Close all thread handles upon completion.
for(i=iWaitIndex; i<iWaitIndex+max; i++)
{
if(hThreadIndex[i]!=NULL)
{
CloseHandle(hThreadIndex[i]);
hThreadIndex[i]=NULL;
}
}
iWaitIndex+=MAXIMUM_WAIT_OBJECTS;
}
DWORD dwEndTick = GetTickCount();
g_log.AddLogInfo(0,"All servers is now scanned...%d sec",(dwEndTick-dwStartTick)/1000);
pGI->vRefScanSI.clear();
CloseHandle(SCAN_hContinueEvent);
}
LRESULT CALLBACK CFG_MainProc(HWND hDlg, UINT message, WPARAM wParam, LPARAM lParam)
{
switch (message)
{
case WM_INITDIALOG:
{
g_tvIndexCFG = 0;
g_hwndTree = GetDlgItem(hDlg,IDC_TREE_CONF);
SendMessage(g_hwndTree, TVM_SETIMAGELIST , TVSIL_NORMAL, (LPARAM)g_hImageListIcons);
SendMessage(g_hwndTree, TVM_SETIMAGELIST , TVSIL_STATE, (LPARAM)g_hImageListStates);
HTREEITEM hNewItem;
hNewItem = TreeView_AddItem(27,g_lang.GetString("ConfigGeneral"));
hNewItem = TreeView_AddItem(15,g_lang.GetString("ConfigMinimizer"));
hNewItem = TreeView_AddItem(28,"mIRC");
hNewItem = TreeView_AddItem(3,"Account (XMPP)");
hNewItem = TreeView_AddItem(16,g_lang.GetString("ConfigExtExe"));
hNewItem = TreeView_AddItem(25,g_lang.GetString("ConfigGraphic"));
hNewItem = TreeView_AddItem(13,g_lang.GetString("ConfigNetwork"));
hNewItem = TreeView_AddItem(20 ,g_lang.GetString("ConfigGames"));
if (hNewItem)
TreeView_Select(g_hwndTree, hNewItem, TVGN_CARET);
for(UINT i=0;i<gm.GamesInfo.size();i++)
hNewItem = TreeView_AddItem(gm.GamesInfo[i].iIconIndex,gm.GamesInfo[i].szGAME_SHORTNAME);
TreeView_Select(g_hwndTree, NULL, TVGN_CARET);
for(UINT i=0; i<gm.GamesInfo.size();i++)
{
GamesInfoCFG[i].bActive = gm.GamesInfo[i].bActive;
GamesInfoCFG[i].bUseHTTPServerList[0] = gm.GamesInfo[i].bUseHTTPServerList[0];
GamesInfoCFG[i].dwMasterServerPORT = gm.GamesInfo[i].dwMasterServerPORT;
strcpy(GamesInfoCFG[i].szGAME_NAME, gm.GamesInfo[i].szGAME_NAME);
strcpy(GamesInfoCFG[i].szMasterServerIP[0], gm.GamesInfo[i].szMasterServerIP[0]);
GamesInfoCFG[i].vGAME_INST = gm.GamesInfo[i].vGAME_INST;
}
memcpy(&AppCFGtemp,&AppCFG,sizeof(APP_SETTINGS_NEW));
CFG_g_sMIRCoutputTemp = g_sMIRCoutput;
SetDlgItemText(hDlg,IDOK,g_lang.GetString("Ok"));
SetDlgItemText(hDlg,IDC_BUTTON_DEFAULT,g_lang.GetString("SetDefault"));
SetDlgItemText(hDlg,IDCANCEL,g_lang.GetString("Cancel"));
CenterWindow(hDlg);
CFG_OnTabbedDialogInit(hDlg) ;
return TRUE;
}
case WM_NOTIFY:
{
NMTREEVIEW *lpnmtv;
lpnmtv = (LPNMTREEVIEW)lParam;
switch (wParam)
{
case IDC_TREE_CONF:
{
NMTREEVIEW *pnmtv;
pnmtv = (LPNMTREEVIEW) lParam;
if((lpnmtv->hdr.code == TVN_SELCHANGED) )
{
if((g_bChanged==true) && (pnmtv->action == TVC_BYMOUSE))
CFG_ApplySettings();
CFG_OnSelChanged(hDlg);
}
}
break;
}
break;
}
case WM_COMMAND:
{
switch (LOWORD(wParam))
{
case IDCANCEL:
LocalFree(g_pHdr);
EndDialog(hDlg, LOWORD(wParam));
return TRUE;
case IDOK:
{
CFG_ApplySettings();
if(AppCFGtemp.bAutostart)
AddAutoRun(EXE_PATH);
else
RemoveAutoRun();
if(AppCFGtemp.bUse_minimize)
{
UnregisterHotKey(NULL, HOTKEY_ID);
if (!RegisterHotKey(NULL, HOTKEY_ID, AppCFGtemp.dwMinimizeMODKey ,AppCFGtemp.cMinimizeKey))
{
//probably already registred
MessageBox(NULL,g_lang.GetString("ErrorRegHotkey"),"Hotkey error",NULL);
}
}else
{
UnregisterHotKey(NULL, HOTKEY_ID);
}
for(UINT i=0; i<gm.GamesInfo.size();i++)
{
gm.GamesInfo[i].bActive = GamesInfoCFG[i].bActive;
gm.GamesInfo[i].bUseHTTPServerList[0] = GamesInfoCFG[i].bUseHTTPServerList[0];
gm.GamesInfo[i].dwMasterServerPORT = GamesInfoCFG[i].dwMasterServerPORT;
strcpy(gm.GamesInfo[i].szMasterServerIP[0], GamesInfoCFG[i].szMasterServerIP[0]);
gm.GamesInfo[i].vGAME_INST = GamesInfoCFG[i].vGAME_INST;
}
memcpy(&AppCFG,&AppCFGtemp,sizeof(APP_SETTINGS_NEW));
//memcpy(&GI,&GamesInfoCFG,sizeof(GAME_INFO)*GamesInfo.size());
// ZeroMemory(&GamesInfoCFG,sizeof(GAME_INFO)*GamesInfo.size());
ZeroMemory(&AppCFGtemp,sizeof(APP_SETTINGS_NEW));
HANDLE hThread=NULL;
DWORD dwThreadIdBrowser=0;
hThread = CreateThread( NULL, 0, &CFG_Save,(LPVOID)0 ,0, &dwThreadIdBrowser);
if (hThread == NULL)
{
g_log.AddLogInfo(GS_LOG_WARNING, "CreateThread CFG_Save failed (%d)\n", GetLastError() );
} else
{
CloseHandle( hThread );
}
// CFG_Save(0);
LocalFree(g_pHdr);
EndDialog(hDlg, LOWORD(wParam));
return TRUE;
}
case IDC_BUTTON_DEFAULT:
{
Default_Appsettings();
// Default_GameSettings();
memcpy(&AppCFGtemp,&AppCFG,sizeof(APP_SETTINGS_NEW));
//memcpy(&GamesInfoCFG,&GI,sizeof(GAME_INFO)*GamesInfo.size());
//GamesInfoCFG = GamesInfo;
g_currSelCfg=-1;
CFG_OnSelChanged(hDlg);
return TRUE;
}
break;
}
}
}
return FALSE;
}
