138 lines
3.6 KiB
C++
138 lines
3.6 KiB
C++
// IOCPClient.h: interface for the IOCPClient class.
|
|
//
|
|
//////////////////////////////////////////////////////////////////////
|
|
|
|
#pragma once
|
|
|
|
#ifdef _WIN32
|
|
#include "stdafx.h"
|
|
#include <WinSock2.h>
|
|
#include <MSTcpIP.h>
|
|
#pragma comment(lib,"ws2_32.lib")
|
|
#endif
|
|
|
|
#include "Buffer.h"
|
|
#include "common/commands.h"
|
|
#include "zstd/zstd.h"
|
|
#include "domain_pool.h"
|
|
|
|
#define MAX_RECV_BUFFER 1024*32
|
|
#define MAX_SEND_BUFFER 1024*32
|
|
#define FLAG_LENGTH 5
|
|
#define HDR_LENGTH 13
|
|
|
|
enum { S_STOP = 0, S_RUN, S_END };
|
|
|
|
typedef int (*DataProcessCB)(void* userData, PBYTE szBuffer, ULONG ulLength);
|
|
|
|
class IOCPManager {
|
|
public:
|
|
virtual ~IOCPManager() {}
|
|
virtual BOOL IsAlive() const { return TRUE; }
|
|
virtual BOOL IsReady() const { return TRUE; }
|
|
virtual VOID OnReceive(PBYTE szBuffer, ULONG ulLength) { }
|
|
|
|
static int DataProcess(void* user, PBYTE szBuffer, ULONG ulLength) {
|
|
IOCPManager* m_Manager = (IOCPManager*)user;
|
|
if (nullptr == m_Manager) {
|
|
Mprintf("IOCPManager DataProcess on NULL ptr: %d\n", unsigned(szBuffer[0]));
|
|
return FALSE;
|
|
}
|
|
// 等待子类准备就绪才能处理数据, 1秒足够了
|
|
int i = 0;
|
|
for (; i < 1000 && !m_Manager->IsReady(); ++i)
|
|
Sleep(1);
|
|
if (!m_Manager->IsReady()) {
|
|
Mprintf("IOCPManager DataProcess is NOT ready: %d\n", unsigned(szBuffer[0]));
|
|
return FALSE;
|
|
}
|
|
if (i) {
|
|
Mprintf("IOCPManager DataProcess wait for %dms: %d\n", i, unsigned(szBuffer[0]));
|
|
}
|
|
m_Manager->OnReceive(szBuffer, ulLength);
|
|
return TRUE;
|
|
}
|
|
};
|
|
|
|
class IOCPClient
|
|
{
|
|
public:
|
|
IOCPClient(State& bExit, bool exit_while_disconnect = false);
|
|
virtual ~IOCPClient();
|
|
|
|
int SendLoginInfo(const LOGIN_INFOR& logInfo) {
|
|
LOGIN_INFOR tmp = logInfo;
|
|
int iRet = Send2Server((char*)&tmp, sizeof(LOGIN_INFOR));
|
|
|
|
return iRet;
|
|
}
|
|
virtual BOOL ConnectServer(const char* szServerIP, unsigned short uPort);
|
|
|
|
BOOL Send2Server(const char* szBuffer, ULONG ulOriginalLength) {
|
|
return OnServerSending(szBuffer, ulOriginalLength);
|
|
}
|
|
|
|
void SetServerAddress(const char* szServerIP, unsigned short uPort) {
|
|
m_Domain = szServerIP ? szServerIP : "127.0.0.1";
|
|
m_nHostPort = uPort;
|
|
}
|
|
|
|
BOOL IsRunning() const
|
|
{
|
|
return m_bIsRunning;
|
|
}
|
|
|
|
void SetExit() {
|
|
m_bIsRunning = FALSE;
|
|
}
|
|
|
|
VOID setManagerCallBack(void* Manager, DataProcessCB dataProcess);
|
|
|
|
VOID RunEventLoop(const BOOL &bCondition);
|
|
bool IsConnected() const { return m_bConnected == TRUE; }
|
|
BOOL Reconnect(void* manager) {
|
|
Disconnect();
|
|
if (manager) m_Manager = manager;
|
|
return ConnectServer(NULL, 0);
|
|
}
|
|
State& GetState() {
|
|
return g_bExit;
|
|
}
|
|
protected:
|
|
virtual int ReceiveData(char* buffer, int bufSize, int flags) {
|
|
// TCP版本调用 recv
|
|
return recv(m_sClientSocket, buffer, bufSize - 1, 0);
|
|
}
|
|
virtual VOID Disconnect(); // 函数支持 TCP/UDP
|
|
virtual int SendTo(const char* buf, int len, int flags) {
|
|
return ::send(m_sClientSocket, buf, len, flags);
|
|
}
|
|
BOOL OnServerSending(const char* szBuffer, ULONG ulOriginalLength);
|
|
static DWORD WINAPI WorkThreadProc(LPVOID lParam);
|
|
VOID OnServerReceiving(char* szBuffer, ULONG ulReceivedLength);
|
|
BOOL SendWithSplit(const char* szBuffer, ULONG ulLength, ULONG ulSplitLength);
|
|
|
|
protected:
|
|
sockaddr_in m_ServerAddr;
|
|
char m_szPacketFlag[FLAG_LENGTH + 3];
|
|
SOCKET m_sClientSocket;
|
|
CBuffer m_CompressedBuffer;
|
|
BOOL m_bWorkThread;
|
|
HANDLE m_hWorkThread;
|
|
BOOL m_bIsRunning;
|
|
BOOL m_bConnected;
|
|
|
|
#if USING_CTX
|
|
ZSTD_CCtx* m_Cctx; // 压缩上下文
|
|
ZSTD_DCtx* m_Dctx; // 解压上下文
|
|
#endif
|
|
|
|
State& g_bExit; // 全局状态量
|
|
void* m_Manager; // 用户数据
|
|
DataProcessCB m_DataProcess; // 处理用户数据
|
|
DomainPool m_Domain;
|
|
std::string m_sCurIP;
|
|
int m_nHostPort;
|
|
bool m_exit_while_disconnect;
|
|
};
|