Restructure repository to include all source folders
Move git root from Client/ to src/ to track all source code: - Client: Game client source (moved to Client/Client/) - Server: Game server source - GameTools: Development tools - CryptoSource: Encryption utilities - database: Database scripts - Script: Game scripts - rylCoder_16.02.2008_src: Legacy coder tools - GMFont, Game: Additional resources 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
@@ -0,0 +1,835 @@
|
||||
#include "stdafx.h"
|
||||
|
||||
#include "ManageClientDispatch.h"
|
||||
|
||||
#include <Network/Session/Session.h>
|
||||
#include <Network/Packet/ManagePacketCmd.h>
|
||||
#include <Setup/SetupClient.h>
|
||||
|
||||
#include <Network/Dispatch/SingleDispatchStorage.h>
|
||||
#include <Network/Dispatch/SendManagePacket.h>
|
||||
#include <Log/ServerLog.h>
|
||||
|
||||
#include <tlhelp32.h>
|
||||
#include <mmsystem.h>
|
||||
|
||||
#include <string>
|
||||
#include <vector>
|
||||
|
||||
#pragma comment(lib, "winmm.lib")
|
||||
|
||||
/*
|
||||
// forward decl
|
||||
HANDLE GetProcessHandle(const char* szProcessFullPathName, unsigned long* lpdwPID);
|
||||
*/
|
||||
|
||||
enum ManageClientConst
|
||||
{
|
||||
MANAGE_CLIENT_DEFAULT_DISPATCH_NUM = 10
|
||||
};
|
||||
|
||||
|
||||
CManageClientDispatch::RunExtraInfo::RunExtraInfo(const ServerManage::RunInfo& runInfo,
|
||||
HANDLE hProcess, unsigned long dwProcessID)
|
||||
: m_RunInfo(runInfo), m_hProcess(hProcess), m_dwProcessID(dwProcessID),
|
||||
m_hWnd(0), m_dwCustomStatus(0), m_dwLastSetWindowHandleTime(0)
|
||||
{
|
||||
char szCommand[MAX_PATH];
|
||||
|
||||
_snprintf(szCommand, MAX_PATH - 1, "%s", m_RunInfo.m_szOption);
|
||||
szCommand[MAX_PATH - 1] = 0;
|
||||
|
||||
char* szData = strtok(szCommand, " \t\r\n");
|
||||
while(0 != szData)
|
||||
{
|
||||
m_CommandSet.insert(szData);
|
||||
szData = strtok(0, " \t\r\n");
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
CSingleDispatch& CManageClientDispatch::GetDispatchTable()
|
||||
{
|
||||
static CSingleDispatch singleDispatch;
|
||||
return singleDispatch;
|
||||
}
|
||||
|
||||
|
||||
CManageClientDispatch::CManageClientDispatch(CSession& Session)
|
||||
: CRylServerDispatch(Session, MANAGE_CLIENT_DEFAULT_DISPATCH_NUM),
|
||||
m_hPatchFile(INVALID_HANDLE_VALUE),
|
||||
m_bDoPatchNow(false)
|
||||
{
|
||||
TCHAR* szTempFileName = _T("./mytmp_patch.tmp");
|
||||
|
||||
memset(m_szTempPatchFileName, 0, sizeof(TCHAR) * MAX_PATH);
|
||||
_tcscpy(m_szTempPatchFileName, szTempFileName);
|
||||
|
||||
TCHAR szFullPathModuleName[MAX_PATH];
|
||||
|
||||
GetModuleFileName(NULL, szFullPathModuleName, MAX_PATH - 1);
|
||||
szFullPathModuleName[MAX_PATH - 1] = 0;
|
||||
|
||||
TCHAR szDrive[MAX_PATH];
|
||||
TCHAR szPath[MAX_PATH];
|
||||
TCHAR szName[MAX_PATH];
|
||||
TCHAR szExtension[MAX_PATH];
|
||||
|
||||
_tsplitpath(szFullPathModuleName, szDrive, szPath, szName, szExtension);
|
||||
|
||||
_sntprintf(m_szTempPatchFileName, MAX_PATH - 1,
|
||||
_T("%s\\%s\\%s"), szDrive, szPath, szTempFileName);
|
||||
m_szTempPatchFileName[MAX_PATH - 1] = 0;
|
||||
|
||||
DETLOG1(g_Log, "this:0x%p/CManageClientDispatch Created", this);
|
||||
}
|
||||
|
||||
|
||||
CManageClientDispatch::~CManageClientDispatch()
|
||||
{
|
||||
if(INVALID_HANDLE_VALUE != m_hPatchFile)
|
||||
{
|
||||
// <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD> <20><><EFBFBD><EFBFBD> <20><><EFBFBD><EFBFBD> <20>ִ<EFBFBD>. <20>ݴ´<DDB4>.
|
||||
ERRLOG2(g_Log, "this:0x%0p/HANDLE:0x%p/PatchFile is not closed. close now", this, m_hPatchFile);
|
||||
CloseHandle(m_hPatchFile);
|
||||
}
|
||||
|
||||
DETLOG1(g_Log, "this:0x%p/CManageClientDispatch Destroyed", this);
|
||||
}
|
||||
|
||||
|
||||
void CManageClientDispatch::Connected()
|
||||
{
|
||||
GetDispatchTable().SetDispatch(this);
|
||||
DETLOG1(g_Log, "this:0x%p/CManageClientDispatch Connected", this);
|
||||
}
|
||||
|
||||
|
||||
void CManageClientDispatch::Disconnected()
|
||||
{
|
||||
GetDispatchTable().RemoveDispatch(this);
|
||||
DETLOG1(g_Log, "this:0x%p/CManageClientDispatch Disconnected", this);
|
||||
}
|
||||
|
||||
|
||||
bool CManageClientDispatch::DispatchPacket(PktBase* lpPktBase)
|
||||
{
|
||||
PktBase::LengthType nPacketLen = lpPktBase->GetLen();
|
||||
|
||||
if(nPacketLen < sizeof(ServerManage::ManageCommand))
|
||||
{
|
||||
ERRLOG3(g_Log, "IP:%15s/PacketLen:%d/Cmd:0x%02x/Invalid packet length or command",
|
||||
GetSession().GetRemoteAddr().get_addr_string(), lpPktBase->GetLen(), lpPktBase->GetCmd());
|
||||
}
|
||||
else
|
||||
{
|
||||
ServerManage::ManageCommand* lpManageCommand =
|
||||
reinterpret_cast<ServerManage::ManageCommand*>(lpPktBase);
|
||||
|
||||
if(nPacketLen != sizeof(ServerManage::ManageCommand) + lpManageCommand->usDataLen)
|
||||
{
|
||||
ERRLOG3(g_Log, "IP:%15s/PacketLen:%d/Cmd:0x%02x/Invalid packet length",
|
||||
GetSession().GetRemoteAddr().get_addr_string(), lpPktBase->GetLen(), lpPktBase->GetCmd());
|
||||
}
|
||||
else
|
||||
{
|
||||
TCHAR* szWindowName = 0;
|
||||
|
||||
char* lpData = (0 != lpManageCommand->usDataLen) ?
|
||||
reinterpret_cast<char*>(lpManageCommand + 1) : 0;
|
||||
|
||||
const TCHAR* szErrorMessage = 0;
|
||||
|
||||
switch(lpManageCommand->GetCmd())
|
||||
{
|
||||
case ServerManage::CMD::UpdateRunList: // <20><><EFBFBD><EFBFBD> <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD> <20>ش<EFBFBD> (ManageServer -> ManageClient, ManageTool), Ack<63><6B><EFBFBD><EFBFBD>.
|
||||
ProcessUpdateRunList(lpData, lpManageCommand->usDataLen, lpManageCommand->usFlags);
|
||||
break;
|
||||
|
||||
case ServerManage::CMD::RunProcess: // <20><><EFBFBD><EFBFBD><EFBFBD>Ѵ<EFBFBD>. (ManageTool -> ManageServer -> ManageClient), Ack<63><6B> <20>ݴ<EFBFBD> <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD> <20><><EFBFBD><EFBFBD>
|
||||
ProcessRunServer(lpManageCommand->dwRunID);
|
||||
break;
|
||||
|
||||
case ServerManage::CMD::QuitProcess: // <20><><EFBFBD><EFBFBD><EFBFBD>Ѵ<EFBFBD>. (ManageTool -> ManageServer -> ManageClient), Ack<63><6B> <20>ݴ<EFBFBD> <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD> <20><><EFBFBD><EFBFBD>
|
||||
ProcessQuitServer(lpManageCommand->dwRunID);
|
||||
break;
|
||||
|
||||
case ServerManage::CMD::ForceTerminate: // <20><><EFBFBD><EFBFBD> <20><><EFBFBD><EFBFBD><EFBFBD>Ѵ<EFBFBD>. (ManageTool -> ManageServer -> ManageClient), Ack<63><6B> <20>ݴ<EFBFBD> <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD> <20><><EFBFBD><EFBFBD>
|
||||
ProcessTerminateServer(lpManageCommand->dwRunID);
|
||||
break;
|
||||
|
||||
case ServerManage::CMD::RelayCommand: // <20><>Ŷ Ŀ<>ǵ带 <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD> <20><EFBFBD><DEBD><EFBFBD><EFBFBD><EFBFBD> <20>Ἥ <20>߰<EFBFBD><DFB0>ؼ<EFBFBD> <20>ѱ<EFBFBD><D1B1><EFBFBD>.
|
||||
|
||||
/*
|
||||
if(!ProcessRelayCommand(lpManageCommand->nMessage,
|
||||
lpManageCommand->wParam, lpManageCommand->lParam,
|
||||
szWindowName, lpData, lpManageCommand->usDataLen, lpManageCommand->cFlag))
|
||||
{
|
||||
szErrorMessage = _T("RelayCommand : RelayCommand Failed");
|
||||
}
|
||||
*/
|
||||
break;
|
||||
|
||||
case ServerManage::CMD::AutoPatch: // <20>ڱ<EFBFBD> <20>ڽ<EFBFBD><DABD><EFBFBD> <20><>ġ<EFBFBD>Ѵ<EFBFBD>
|
||||
|
||||
if(!ProcessAutoPatch(lpData, lpManageCommand->usDataLen, lpManageCommand->usFlags))
|
||||
{
|
||||
szErrorMessage = _T("AutoPatch : AutoPatch packet process failed");
|
||||
}
|
||||
break;
|
||||
|
||||
case ServerManage::CMD::ChangeSetup: // <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD> <20>ٲٰ<D9B2> <20><><EFBFBD><EFBFBD><EFBFBD>Ѵ<EFBFBD>.
|
||||
|
||||
if(!ManageSetup::ClientSetup::GetInstance().SerializeIn(lpData, lpManageCommand->usDataLen))
|
||||
{
|
||||
szErrorMessage = _T("ChangeSetup : Change setup failed");
|
||||
}
|
||||
else
|
||||
{
|
||||
ManageSetup::ClientSetup::GetInstance().Save();
|
||||
}
|
||||
break;
|
||||
|
||||
case ServerManage::CMD::ReloadSetup: // <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD> <20><><EFBFBD>ε<EFBFBD><CEB5>Ѵ<EFBFBD>.
|
||||
|
||||
ManageSetup::ClientSetup::GetInstance().Load();
|
||||
break;
|
||||
|
||||
|
||||
case ServerManage::CMD::ExecuteCommand:
|
||||
ProcessExecuteConsoleCommand(lpManageCommand);
|
||||
break;
|
||||
|
||||
default:
|
||||
|
||||
szErrorMessage = _T("Unknown message : Process failed");
|
||||
break;
|
||||
}
|
||||
|
||||
if(0 != szErrorMessage)
|
||||
{
|
||||
ERRLOG4(g_Log, "IP:%15s/DetailCmd:0x%02x/Error:%d/%s",
|
||||
GetSession().GetRemoteAddr().get_addr_string(),
|
||||
lpPktBase->GetCmd(), GetLastError(), szErrorMessage);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
bool CManageClientDispatch::ProcessRelayCommand(unsigned int nMessage, WPARAM wParam, LPARAM lParam,
|
||||
const char* szWindowName, const char* lpData,
|
||||
unsigned short usDataLen, unsigned char cFlag)
|
||||
{
|
||||
if(0 != szWindowName)
|
||||
{
|
||||
HWND hWnd = FindWindow(0, szWindowName);
|
||||
if(0 != hWnd)
|
||||
{
|
||||
if(0 != lpData && WM_COPYDATA == nMessage)
|
||||
{
|
||||
COPYDATASTRUCT copyDataStruct;
|
||||
|
||||
copyDataStruct.dwData = 0;
|
||||
copyDataStruct.cbData = usDataLen;
|
||||
copyDataStruct.lpData = (void*)lpData;
|
||||
|
||||
// <20><EFBFBD><DEBD><EFBFBD><EFBFBD><EFBFBD> <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD>.
|
||||
SendMessage(hWnd, nMessage, wParam,
|
||||
reinterpret_cast<LPARAM>(©DataStruct));
|
||||
}
|
||||
else
|
||||
{
|
||||
SendMessage(hWnd, nMessage, wParam, lParam);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
|
||||
|
||||
bool CManageClientDispatch::ProcessAutoPatch(const char* lpData, unsigned short usDataLen, unsigned short usFlag)
|
||||
{
|
||||
switch(usFlag)
|
||||
{
|
||||
case ServerManage::ManageCommand::AUTOPATCH_OPENFILE:
|
||||
|
||||
if(INVALID_HANDLE_VALUE != m_hPatchFile)
|
||||
{
|
||||
// <20><>ġ <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD> <20><><EFBFBD><EFBFBD> <20><><EFBFBD><EFBFBD>. <20>ݴ´<DDB4>.
|
||||
ERRLOG2(g_Log, "this:0x%0p/HANDLE:0x%p/PatchFile is not closed. close now for create new one",
|
||||
this, m_hPatchFile);
|
||||
|
||||
CloseHandle(m_hPatchFile);
|
||||
}
|
||||
|
||||
// <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD> <20><><EFBFBD><EFBFBD><EFBFBD>Ѵ<EFBFBD>.
|
||||
m_hPatchFile = CreateFile(m_szTempPatchFileName, GENERIC_WRITE, 0, 0,
|
||||
CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, 0);
|
||||
|
||||
if(INVALID_HANDLE_VALUE == m_hPatchFile)
|
||||
{
|
||||
ERRLOG2(g_Log, "this:0x%p/Error:%d/Patch tempfile create failed",
|
||||
this, GetLastError());
|
||||
}
|
||||
break;
|
||||
|
||||
case ServerManage::ManageCommand::AUTOPATCH_DATA:
|
||||
|
||||
if(INVALID_HANDLE_VALUE != m_hPatchFile)
|
||||
{
|
||||
unsigned long dwWritten = 0;
|
||||
unsigned long dwTotalWritten = 0;
|
||||
|
||||
while(usDataLen != dwTotalWritten)
|
||||
{
|
||||
if(!WriteFile(m_hPatchFile, lpData, usDataLen, &dwWritten, 0))
|
||||
{
|
||||
ERRLOG3(g_Log, "this:0x%p/HANDLE:0x%p/Error:%d/Patchfile write failed.",
|
||||
this, m_hPatchFile, GetLastError());
|
||||
|
||||
CloseHandle(m_hPatchFile);
|
||||
m_hPatchFile = INVALID_HANDLE_VALUE;
|
||||
|
||||
DeleteFile(m_szTempPatchFileName);
|
||||
break;
|
||||
}
|
||||
|
||||
dwTotalWritten += dwWritten;
|
||||
}
|
||||
|
||||
if(0 == usDataLen && INVALID_HANDLE_VALUE != m_hPatchFile)
|
||||
{
|
||||
// <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD> <20><><EFBFBD><EFBFBD> <20>Ҵ<DEBE>. <20><><EFBFBD><EFBFBD> <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD> <20><><EFBFBD><EFBFBD><EFBFBD>ϰ<EFBFBD> <20><>ġ<EFBFBD><C4A1> <20><><EFBFBD><EFBFBD><EFBFBD>Ѵ<EFBFBD>.
|
||||
m_bDoPatchNow = true;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
ERRLOG1(g_Log, "this:0x%p/Patchfile is invalid", this);
|
||||
}
|
||||
break;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
|
||||
|
||||
bool CManageClientDispatch::ProcessUpdateRunList(const char* szData, unsigned short usDataLen,
|
||||
unsigned short usFlags)
|
||||
{
|
||||
if(ServerManage::SEND_RUNINFO_START == usFlags)
|
||||
{
|
||||
// <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD> <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD> Process<73>鿡 <20><><EFBFBD>ؼ<EFBFBD> <20><><EFBFBD><EFBFBD> <20>ڵ<EFBFBD><DAB5><EFBFBD> <20>ݴ´<DDB4>.
|
||||
RunTable::iterator pos = m_RunTable.begin();
|
||||
RunTable::iterator end = m_RunTable.end();
|
||||
|
||||
for(; pos != end; ++pos)
|
||||
{
|
||||
RunExtraInfo& runExtraInfo = pos->second;
|
||||
|
||||
if(0 != runExtraInfo.m_hProcess)
|
||||
{
|
||||
CloseHandle(runExtraInfo.m_hProcess);
|
||||
}
|
||||
}
|
||||
|
||||
m_RunTable.clear();
|
||||
}
|
||||
else if(ServerManage::SEND_RUNINFO_NOW == usFlags)
|
||||
{
|
||||
// RunTable<6C><65> <20><><EFBFBD><EFBFBD><EFBFBD>ϰ<EFBFBD>, <20><><EFBFBD>μ<EFBFBD><CEBC><EFBFBD><EFBFBD><EFBFBD> <20><><EFBFBD><EFBFBD>.
|
||||
const ServerManage::RunInfo* lpRunInfo = reinterpret_cast<const ServerManage::RunInfo*>(szData);
|
||||
const ServerManage::RunInfo* lpEndRunInfo = lpRunInfo + (usDataLen / sizeof(ServerManage::RunInfo));
|
||||
|
||||
for(; lpRunInfo != lpEndRunInfo; ++lpRunInfo)
|
||||
{
|
||||
m_RunTable.insert(RunTable::value_type(lpRunInfo->m_dwRunID,
|
||||
RunExtraInfo(*lpRunInfo, 0, 0)));
|
||||
}
|
||||
}
|
||||
else if(ServerManage::SEND_RUNINFO_FINISH == usFlags)
|
||||
{
|
||||
// <20><><EFBFBD>μ<EFBFBD><CEBC><EFBFBD> <20><><EFBFBD><EFBFBD> <20><><EFBFBD><EFBFBD> <20><>Ŷ<EFBFBD><C5B6> <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD>.
|
||||
CheckProcessStatus();
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
bool CManageClientDispatch::ProcessRunServer(unsigned long dwRunID)
|
||||
{
|
||||
RunTable::iterator pos = m_RunTable.find(dwRunID);
|
||||
|
||||
if(pos != m_RunTable.end())
|
||||
{
|
||||
RunExtraInfo& runExInfo = pos->second;
|
||||
|
||||
TCHAR szDrive[MAX_PATH];
|
||||
TCHAR szPath[MAX_PATH];
|
||||
|
||||
TCHAR szDefaultPath[MAX_PATH];
|
||||
|
||||
_splitpath(runExInfo.m_RunInfo.m_szPath, szDrive, szPath, 0, 0);
|
||||
|
||||
_snprintf(szDefaultPath, MAX_PATH - 1, "%s%s", szDrive, szPath);
|
||||
szDefaultPath[MAX_PATH - 1] = 0;
|
||||
|
||||
STARTUPINFO si;
|
||||
PROCESS_INFORMATION pi;
|
||||
|
||||
ZeroMemory(&si, sizeof(si));
|
||||
si.cb = sizeof(si);
|
||||
|
||||
ZeroMemory(&pi, sizeof(pi));
|
||||
|
||||
// Start the child process.
|
||||
|
||||
|
||||
/*
|
||||
if(CreateProcess(runExInfo.m_RunInfo.m_szPath, runExInfo.m_RunInfo.m_szOption, 0, 0,
|
||||
FALSE, 0, 0, szDefaultPath, &si, &pi))
|
||||
*/
|
||||
char szExecutePath[MAX_PATH * 2];
|
||||
_snprintf(szExecutePath, MAX_PATH * 2 - 1, "\"%s\" %s",
|
||||
runExInfo.m_RunInfo.m_szPath, runExInfo.m_RunInfo.m_szOption);
|
||||
|
||||
szExecutePath[MAX_PATH * 2 - 1] = 0;
|
||||
|
||||
if(CreateProcess(0, szExecutePath, 0, 0,
|
||||
FALSE, 0, 0, szDefaultPath, &si, &pi))
|
||||
|
||||
{
|
||||
runExInfo.m_hProcess = pi.hProcess;
|
||||
runExInfo.m_dwProcessID = pi.dwProcessId;
|
||||
|
||||
DETLOG4(g_Log, "Execute file success, RunID(%u), Path(%s), Option(%s), DefaultPath(%s)",
|
||||
runExInfo.m_RunInfo.m_dwRunID, runExInfo.m_RunInfo.m_szPath,
|
||||
runExInfo.m_RunInfo.m_szOption, szDefaultPath);
|
||||
}
|
||||
else
|
||||
{
|
||||
ERRLOG5(g_Log, "Cannot execute file : CreateProcess failed(%d), RunID(%u), "
|
||||
"Path(%s), Option(%s), DefaultPath(%s)",
|
||||
WSAGetLastError(), runExInfo.m_RunInfo.m_dwRunID,
|
||||
runExInfo.m_RunInfo.m_szPath, runExInfo.m_RunInfo.m_szOption, szDefaultPath);
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
ERRLOG1(g_Log, "Cannot execute file : Unknown RunID(%d)", dwRunID);
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
bool CManageClientDispatch::ProcessQuitServer(unsigned long dwRunID)
|
||||
{
|
||||
RunTable::iterator pos = m_RunTable.find(dwRunID);
|
||||
|
||||
if(pos != m_RunTable.end())
|
||||
{
|
||||
RunExtraInfo& runExInfo = pos->second;
|
||||
|
||||
if(0 != runExInfo.m_hWnd)
|
||||
{
|
||||
TCHAR szWindowName[MAX_PATH];
|
||||
GetWindowText(runExInfo.m_hWnd, szWindowName, MAX_PATH - 1);
|
||||
szWindowName[MAX_PATH - 1] = 0;
|
||||
|
||||
SendMessage(runExInfo.m_hWnd, WM_DESTROY, 0, 0);
|
||||
|
||||
DETLOG6(g_Log, "Send quit message : HANDLE(0x%p), PID(0x%08x), WindowName(%s), "
|
||||
"RunID(%u), Path(%s), Option(%s)",
|
||||
runExInfo.m_hProcess, runExInfo.m_dwProcessID, szWindowName,
|
||||
runExInfo.m_RunInfo.m_dwRunID, runExInfo.m_RunInfo.m_szPath,
|
||||
runExInfo.m_RunInfo.m_szOption);
|
||||
}
|
||||
else
|
||||
{
|
||||
ERRLOG5(g_Log, "Cannot send quit message (Get window failed) : "
|
||||
"HANDLE(0x%p), PID(0x%08x), RunID(%u), Path(%s), Option(%s)",
|
||||
runExInfo.m_hProcess, runExInfo.m_dwProcessID,
|
||||
runExInfo.m_RunInfo.m_dwRunID, runExInfo.m_RunInfo.m_szPath,
|
||||
runExInfo.m_RunInfo.m_szOption);
|
||||
}
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
bool CManageClientDispatch::ProcessTerminateServer(unsigned long dwRunID)
|
||||
{
|
||||
RunTable::iterator pos = m_RunTable.find(dwRunID);
|
||||
|
||||
if(pos != m_RunTable.end())
|
||||
{
|
||||
RunExtraInfo& runExInfo = pos->second;
|
||||
|
||||
if(0 != runExInfo.m_hProcess)
|
||||
{
|
||||
if(TerminateProcess(runExInfo.m_hProcess, -1))
|
||||
{
|
||||
DETLOG5(g_Log, "Terminate process. HANDLE(0x%p), PID(0x%08x), "
|
||||
"RunID(%u), Path(%s), Option(%s)",
|
||||
runExInfo.m_hProcess, runExInfo.m_dwProcessID,
|
||||
runExInfo.m_RunInfo.m_dwRunID, runExInfo.m_RunInfo.m_szPath,
|
||||
runExInfo.m_RunInfo.m_szOption);
|
||||
}
|
||||
else
|
||||
{
|
||||
ERRLOG6(g_Log, "Cannot terminate process. Err(%d), HANDLE(0x%p), PID(0x%08x), "
|
||||
"RunID(%u), Path(%s), Option(%s)",
|
||||
GetLastError(), runExInfo.m_hProcess, runExInfo.m_dwProcessID,
|
||||
runExInfo.m_RunInfo.m_dwRunID, runExInfo.m_RunInfo.m_szPath,
|
||||
runExInfo.m_RunInfo.m_szOption);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
|
||||
bool CManageClientDispatch::ProcessExecuteConsoleCommand(ServerManage::ManageCommand* lpManageCmd)
|
||||
{
|
||||
RunTable::iterator pos = m_RunTable.find(lpManageCmd->dwRunID);
|
||||
|
||||
if(pos != m_RunTable.end())
|
||||
{
|
||||
RunExtraInfo& runExInfo = pos->second;
|
||||
|
||||
if(0 != runExInfo.m_hWnd)
|
||||
{
|
||||
COPYDATASTRUCT copyDataStruct;
|
||||
|
||||
copyDataStruct.dwData = 0;
|
||||
copyDataStruct.cbData = lpManageCmd->GetLen();
|
||||
copyDataStruct.lpData = lpManageCmd;
|
||||
|
||||
SendMessage(runExInfo.m_hWnd, WM_COPYDATA, 0, (LPARAM)©DataStruct);
|
||||
}
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
|
||||
void CManageClientDispatch::CheckProcessStatus()
|
||||
{
|
||||
const int MAX_PROCESS_STATUS_SEND_ONCE =
|
||||
(PktMaxLen - sizeof(ServerManage::ManageCommand)) / sizeof(ServerManage::ProcessStatus);
|
||||
|
||||
ServerManage::ProcessStatus procesStatus[MAX_PROCESS_STATUS_SEND_ONCE];
|
||||
|
||||
RunTable::iterator pos = m_RunTable.begin();
|
||||
RunTable::iterator end = m_RunTable.end();
|
||||
|
||||
for(; pos != end; ++pos)
|
||||
{
|
||||
RunExtraInfo& runExtraInfo = pos->second;
|
||||
|
||||
// üũ<C3BC>ϰ<EFBFBD>, PID<49><44> <20>ִµ<D6B4> <20><><EFBFBD>μ<EFBFBD><CEBC><EFBFBD><EFBFBD><EFBFBD> <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD> <20><><EFBFBD><EFBFBD> <20><><EFBFBD>α<CEB1><D7B7><EFBFBD>, <20><><EFBFBD>μ<EFBFBD><CEBC><EFBFBD><EFBFBD><EFBFBD> <20><><EFBFBD><EFBFBD>.
|
||||
if(0 != runExtraInfo.m_dwProcessID && 0 != runExtraInfo.m_hWnd
|
||||
&& 0 == runExtraInfo.m_hProcess)
|
||||
{
|
||||
runExtraInfo.m_hProcess = OpenProcess(PROCESS_ALL_ACCESS,
|
||||
FALSE, runExtraInfo.m_dwProcessID);
|
||||
}
|
||||
}
|
||||
|
||||
unsigned long dwExitCode = 0;
|
||||
unsigned long dwErrorCode = 0;
|
||||
unsigned short usSendProcessStatus = 0;
|
||||
|
||||
pos = m_RunTable.begin();
|
||||
end = m_RunTable.end();
|
||||
|
||||
for(; pos != end; ++pos)
|
||||
{
|
||||
RunExtraInfo& runExtraInfo = pos->second;
|
||||
|
||||
// Default - Deactivated Process.
|
||||
ServerManage::ProcessStatus* lpCurrentProcessStatus = procesStatus + usSendProcessStatus;
|
||||
memset(lpCurrentProcessStatus, 0, sizeof(ServerManage::ProcessStatus));
|
||||
|
||||
lpCurrentProcessStatus->m_dwRunID = runExtraInfo.m_RunInfo.m_dwRunID;
|
||||
lpCurrentProcessStatus->m_dwStatusFlags = 0;
|
||||
lpCurrentProcessStatus->m_dwCurrentTime = timeGetTime();
|
||||
lpCurrentProcessStatus->m_dwCustomStatus = runExtraInfo.m_dwCustomStatus;
|
||||
lpCurrentProcessStatus->m_dwLastSetWindowHandleTime = runExtraInfo.m_dwLastSetWindowHandleTime;
|
||||
|
||||
if(0 != runExtraInfo.m_hProcess)
|
||||
{
|
||||
GetProcessTimes(runExtraInfo.m_hProcess,
|
||||
&lpCurrentProcessStatus->m_CreationTIme,
|
||||
&lpCurrentProcessStatus->m_ExitTime,
|
||||
&lpCurrentProcessStatus->m_KernelTime,
|
||||
&lpCurrentProcessStatus->m_UserTime);
|
||||
|
||||
GetProcessMemoryInfo(runExtraInfo.m_hProcess,
|
||||
&lpCurrentProcessStatus->m_ProcessMemoryCounters,
|
||||
sizeof(PROCESS_MEMORY_COUNTERS));
|
||||
|
||||
if(!GetExitCodeProcess(runExtraInfo.m_hProcess, &dwExitCode))
|
||||
{
|
||||
ERRLOG3(g_Log, "GetExitCodeProcess failed(%d) : RunID(%u), Path(%s)",
|
||||
dwErrorCode, runExtraInfo.m_RunInfo.m_dwRunID,
|
||||
runExtraInfo.m_RunInfo.m_szPath);
|
||||
}
|
||||
else if(dwExitCode != STILL_ACTIVE)
|
||||
{
|
||||
DETLOG4(g_Log, "Close process handle : HANDLE(0x%p), PID(0x%08x), RunID(%u), Path(%s)",
|
||||
runExtraInfo.m_hProcess, runExtraInfo.m_dwProcessID,
|
||||
runExtraInfo.m_RunInfo.m_dwRunID, runExtraInfo.m_RunInfo.m_szPath);
|
||||
|
||||
CloseHandle(runExtraInfo.m_hProcess);
|
||||
runExtraInfo.m_hProcess = 0;
|
||||
runExtraInfo.m_dwProcessID = 0;
|
||||
runExtraInfo.m_hWnd = 0;
|
||||
runExtraInfo.m_dwLastSetWindowHandleTime = 0;
|
||||
}
|
||||
else
|
||||
{
|
||||
lpCurrentProcessStatus->m_dwStatusFlags =
|
||||
ServerManage::PROCESS_RUNNING;
|
||||
}
|
||||
}
|
||||
|
||||
++usSendProcessStatus;
|
||||
|
||||
if(usSendProcessStatus == MAX_PROCESS_STATUS_SEND_ONCE)
|
||||
{
|
||||
ServerManage::SendManagePacket(GetSession(), ServerManage::CMD::ProcessStatus,
|
||||
0, 0, 0, 0, sizeof(ServerManage::ProcessStatus) * usSendProcessStatus, 0, procesStatus, 0);
|
||||
|
||||
usSendProcessStatus = 0;
|
||||
}
|
||||
}
|
||||
|
||||
if(0 < usSendProcessStatus)
|
||||
{
|
||||
ServerManage::SendManagePacket(GetSession(), ServerManage::CMD::ProcessStatus,
|
||||
0, 0, 0, 0, sizeof(ServerManage::ProcessStatus) * usSendProcessStatus, 0, procesStatus, 0);
|
||||
|
||||
usSendProcessStatus = 0;
|
||||
}
|
||||
}
|
||||
|
||||
bool CManageClientDispatch::SetAppData(unsigned long dwRunID, HWND hWnd,
|
||||
unsigned long dwPID, unsigned long dwCustomStatus)
|
||||
{
|
||||
RunTable::iterator pos = m_RunTable.find(dwRunID);
|
||||
RunTable::iterator end = m_RunTable.end();
|
||||
|
||||
if(pos != end)
|
||||
{
|
||||
RunExtraInfo& runExtraInfo = pos->second;
|
||||
|
||||
runExtraInfo.m_dwProcessID = dwPID;
|
||||
runExtraInfo.m_hWnd = hWnd;
|
||||
runExtraInfo.m_dwCustomStatus = dwCustomStatus;
|
||||
runExtraInfo.m_dwLastSetWindowHandleTime = timeGetTime();
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
|
||||
bool CManageClientDispatch::GetRunIDFromString(const char* szAppName, const char* szCommandLine,
|
||||
unsigned long& dwRunID_Out)
|
||||
{
|
||||
char szProcessName[MAX_PATH];
|
||||
char szProcessSnapName[MAX_PATH];
|
||||
|
||||
_snprintf(szProcessSnapName, MAX_PATH - 1, "%s", szAppName);
|
||||
szProcessSnapName[MAX_PATH - 1] = 0;
|
||||
_strupr(szProcessSnapName);
|
||||
|
||||
char szCommandLineData[MAX_PATH * 2];
|
||||
_snprintf(szCommandLineData, MAX_PATH * 2 - 1, "%s", szCommandLine);
|
||||
szCommandLineData[MAX_PATH * 2 - 1] = 0;
|
||||
|
||||
RunExtraInfo::CommandSet commandSet;
|
||||
|
||||
char* szData = 0;
|
||||
|
||||
if('\"' == szCommandLineData[0])
|
||||
{
|
||||
szData = strtok(szCommandLineData, "\"");
|
||||
szData = strtok(0, "\"");
|
||||
|
||||
if(0 != szData)
|
||||
{
|
||||
// <20><><EFBFBD>ϸ<EFBFBD><CFB8><EFBFBD> <20>и<EFBFBD><D0B8><EFBFBD> Argument<6E><74> <20>Ƽ<DEBE> ó<><C3B3>..
|
||||
char szRealCommandLine[MAX_PATH];
|
||||
_snprintf(szRealCommandLine, MAX_PATH - 1, "%s", szData);
|
||||
szRealCommandLine[MAX_PATH - 1] = 0;
|
||||
|
||||
szData = strtok(szRealCommandLine, " \t\r\n");
|
||||
while(0 != szData)
|
||||
{
|
||||
commandSet.insert(szData);
|
||||
szData = strtok(0, " \t\r\n");
|
||||
}
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
szData = strtok(szCommandLineData, " \t\r\n");
|
||||
|
||||
// ù<><C3B9>° Argument<6E><74> <20><><EFBFBD>ϸ<EFBFBD><CFB8>̹Ƿ<CCB9> <20><><EFBFBD><EFBFBD>.
|
||||
szData = strtok(0, " \t\r\n");
|
||||
while(0 != szData)
|
||||
{
|
||||
commandSet.insert(szData);
|
||||
szData = strtok(0, " \t\r\n");
|
||||
}
|
||||
}
|
||||
|
||||
RunTable::iterator pos = m_RunTable.begin();
|
||||
RunTable::iterator end = m_RunTable.end();
|
||||
|
||||
for(; pos != end; ++pos)
|
||||
{
|
||||
RunExtraInfo& runExtraInfo = pos->second;
|
||||
|
||||
_snprintf(szProcessName, MAX_PATH - 1, "%s", runExtraInfo.m_RunInfo.m_szPath);
|
||||
szProcessName[MAX_PATH - 1] = 0;
|
||||
_strupr(szProcessName);
|
||||
|
||||
if(0 == strncmp(szProcessName, szProcessSnapName, MAX_PATH))
|
||||
{
|
||||
if((runExtraInfo.m_CommandSet.size() == commandSet.size()
|
||||
&& std::equal(runExtraInfo.m_CommandSet.begin(),
|
||||
runExtraInfo.m_CommandSet.end(), commandSet.begin())))
|
||||
{
|
||||
dwRunID_Out = runExtraInfo.m_RunInfo.m_dwRunID;
|
||||
return true;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
dwRunID_Out = 0;
|
||||
return false;
|
||||
}
|
||||
|
||||
|
||||
bool CManageClientDispatch::GetRunIDFromPID(unsigned long dwPID, unsigned long& dwRunID_Out)
|
||||
{
|
||||
RunTable::iterator pos = m_RunTable.begin();
|
||||
RunTable::iterator end = m_RunTable.end();
|
||||
|
||||
for(; pos != end; ++pos)
|
||||
{
|
||||
RunExtraInfo& runExtraInfo = pos->second;
|
||||
|
||||
if(runExtraInfo.m_dwProcessID == dwPID)
|
||||
{
|
||||
dwRunID_Out = runExtraInfo.m_RunInfo.m_dwRunID;
|
||||
return true;
|
||||
}
|
||||
}
|
||||
|
||||
dwRunID_Out = 0;
|
||||
return false;
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
/*
|
||||
HANDLE GetProcessHandle(const char* szProcessFullPathName, unsigned long* lpdwPID)
|
||||
{
|
||||
HANDLE hProcess = INVALID_HANDLE_VALUE;
|
||||
|
||||
if(0 == lpdwPID)
|
||||
{
|
||||
return hProcess;
|
||||
}
|
||||
|
||||
// Take a snapshot of all processes in the system.
|
||||
HANDLE hProcessSnap = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
|
||||
|
||||
if (hProcessSnap == INVALID_HANDLE_VALUE)
|
||||
{
|
||||
return hProcess;
|
||||
}
|
||||
|
||||
// Fill in the size of the structure before using it.
|
||||
PROCESSENTRY32 pe32 = {0};
|
||||
pe32.dwSize = sizeof(PROCESSENTRY32);
|
||||
|
||||
// Walk the snapshot of the processes, and for each process,
|
||||
// display information.
|
||||
|
||||
char szProcessName[MAX_PATH];
|
||||
char szProcessSnapName[MAX_PATH];
|
||||
|
||||
_snprintf(szProcessName, MAX_PATH - 1, "%s", szProcessFullPathName);
|
||||
szProcessName[MAX_PATH - 1] = 0;
|
||||
|
||||
_strupr(szProcessName);
|
||||
|
||||
bool bFound = false;
|
||||
|
||||
if (Process32First(hProcessSnap, &pe32))
|
||||
{
|
||||
do
|
||||
{
|
||||
MODULEENTRY32 me32 = {0};
|
||||
|
||||
HANDLE hModuleSnap = CreateToolhelp32Snapshot(TH32CS_SNAPMODULE, pe32.th32ProcessID);
|
||||
if (hModuleSnap != INVALID_HANDLE_VALUE)
|
||||
{
|
||||
me32.dwSize = sizeof(MODULEENTRY32);
|
||||
|
||||
if (Module32First(hModuleSnap, &me32))
|
||||
{
|
||||
do
|
||||
{
|
||||
_snprintf(szProcessSnapName, MAX_PATH - 1, "%s", me32.szExePath);
|
||||
szProcessSnapName[MAX_PATH - 1] = 0;
|
||||
|
||||
_strupr(szProcessSnapName);
|
||||
|
||||
if(0 == strcmp(szProcessName, szProcessSnapName))
|
||||
{
|
||||
// Get the actual priority class.
|
||||
hProcess = OpenProcess (PROCESS_ALL_ACCESS, FALSE, pe32.th32ProcessID);
|
||||
if(INVALID_HANDLE_VALUE == hProcess)
|
||||
{
|
||||
ERRLOG2(g_Log, "Process open failed(%d) : %s",
|
||||
GetLastError(), me32.szExePath);
|
||||
}
|
||||
|
||||
*lpdwPID = pe32.th32ProcessID;
|
||||
bFound = true;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
while (!bFound && Module32Next(hModuleSnap, &me32));
|
||||
}
|
||||
|
||||
CloseHandle (hModuleSnap);
|
||||
}
|
||||
}
|
||||
|
||||
while (!bFound && Process32Next(hProcessSnap, &pe32));
|
||||
}
|
||||
|
||||
CloseHandle(hProcessSnap);
|
||||
return hProcess;
|
||||
}
|
||||
|
||||
*/
|
||||
@@ -0,0 +1,90 @@
|
||||
#ifndef _GM_MANAGE_CLIENT_DISPATCH_H_
|
||||
#define _GM_MANAGE_CLIENT_DISPATCH_H_
|
||||
|
||||
#pragma warning(disable:4800)
|
||||
|
||||
#include <map>
|
||||
#include <set>
|
||||
#include <boost/pool/pool_alloc.hpp>
|
||||
#include <Network/Dispatch/RylServerDispatch.h>
|
||||
#include <Network/Dispatch/SingleDispatchStorage.h>
|
||||
#include <Network/Packet/ManagePacketCmd.h>
|
||||
|
||||
// CManageClientDispatch<63><68> <20><><EFBFBD><EFBFBD> Ŭ<><C5AC><EFBFBD>̾<EFBFBD>Ʈ<EFBFBD><C6AE> <20><> <20><><EFBFBD>̺귯<CCBA><EAB7AF><EFBFBD><EFBFBD> <20><>ũ<EFBFBD><C5A9>.
|
||||
#pragma comment(lib, "psapi.lib")
|
||||
|
||||
class CManageClientDispatch : public CRylServerDispatch
|
||||
{
|
||||
public:
|
||||
|
||||
struct RunExtraInfo
|
||||
{
|
||||
typedef std::set<std::string> CommandSet;
|
||||
|
||||
ServerManage::RunInfo m_RunInfo;
|
||||
HANDLE m_hProcess;
|
||||
HWND m_hWnd;
|
||||
unsigned long m_dwProcessID;
|
||||
unsigned long m_dwCustomStatus;
|
||||
unsigned long m_dwLastSetWindowHandleTime;
|
||||
|
||||
CommandSet m_CommandSet;
|
||||
|
||||
RunExtraInfo();
|
||||
RunExtraInfo(const ServerManage::RunInfo& runInfo,
|
||||
HANDLE hProcess, unsigned long dwProcessID);
|
||||
};
|
||||
|
||||
// Key - RunID
|
||||
typedef std::map<unsigned long, RunExtraInfo, std::less<unsigned long>,
|
||||
boost::pool_allocator<std::pair<unsigned long, RunExtraInfo> > > RunTable;
|
||||
|
||||
static CSingleDispatch& GetDispatchTable();
|
||||
|
||||
CManageClientDispatch(CSession& Session);
|
||||
virtual ~CManageClientDispatch();
|
||||
|
||||
void CheckProcessStatus();
|
||||
|
||||
bool DoPatchNow() const { return m_bDoPatchNow; }
|
||||
const TCHAR* GetTempPatchFileName() const { return m_szTempPatchFileName; }
|
||||
|
||||
bool SetAppData(unsigned long dwRunID, HWND hWnd,
|
||||
unsigned long dwPID, unsigned long dwCustomStatus);
|
||||
|
||||
bool GetRunIDFromPID(unsigned long dwPID, unsigned long& dwRunID_Out);
|
||||
|
||||
bool GetRunIDFromString(const char* szAppName,
|
||||
const char* szCommandLine, unsigned long& dwRunID_Out);
|
||||
|
||||
RunTable& GetRunTable() { return m_RunTable; }
|
||||
|
||||
private:
|
||||
|
||||
virtual void Connected();
|
||||
virtual void Disconnected();
|
||||
|
||||
virtual bool DispatchPacket(PktBase* lpPktBase);
|
||||
|
||||
bool ProcessUpdateRunList(const char* szData, unsigned short usDataLen, unsigned short usFlags);
|
||||
bool ProcessRunServer(unsigned long dwRunID);
|
||||
bool ProcessQuitServer(unsigned long dwRunID);
|
||||
bool ProcessTerminateServer(unsigned long dwRunID);
|
||||
bool ProcessExecuteConsoleCommand(ServerManage::ManageCommand* lpManageCmd);
|
||||
|
||||
bool ProcessRelayCommand(unsigned int nMessage, WPARAM wParam, LPARAM lParam,
|
||||
const char* szWindowName, const char* lpData, unsigned short usDataLen, unsigned char cFlag);
|
||||
|
||||
bool ProcessAutoPatch(const char* lpData, unsigned short usDataLen, unsigned short usFlag);
|
||||
|
||||
// Member variable;
|
||||
HANDLE m_hPatchFile; // <20><>ġ <20><><EFBFBD><EFBFBD> <20>ڵ<EFBFBD>.
|
||||
TCHAR m_szTempPatchFileName[MAX_PATH]; // <20>ӽ<EFBFBD> <20><>ġ <20><><EFBFBD><EFBFBD> <20≯<EFBFBD>.
|
||||
bool m_bDoPatchNow; // <20><>ġ <20>غ<EFBFBD><D8BA><EFBFBD> <20>Ǿ<EFBFBD><C7BE><EFBFBD><EFBFBD><EFBFBD> <20><><EFBFBD>θ<EFBFBD> <20>˾ƿ<CBBE>.
|
||||
|
||||
|
||||
RunTable m_RunTable;
|
||||
};
|
||||
|
||||
|
||||
#endif
|
||||
Reference in New Issue
Block a user