198 lines
4.1 KiB
C++
198 lines
4.1 KiB
C++
// IOCPServer.h: interface for the CIOCPServer class.
|
||
//
|
||
//////////////////////////////////////////////////////////////////////
|
||
|
||
#if !defined(AFX_IOCPSERVER_H__75B80E90_FD25_4FFB_B273_0090AA43BBDF__INCLUDED_)
|
||
#define AFX_IOCPSERVER_H__75B80E90_FD25_4FFB_B273_0090AA43BBDF__INCLUDED_
|
||
|
||
#if _MSC_VER > 1000
|
||
#pragma once
|
||
#endif // _MSC_VER > 1000
|
||
|
||
#include <winsock2.h>
|
||
#include <MSTcpIP.h>
|
||
#pragma comment(lib,"ws2_32.lib")
|
||
#include "Buffer.h"
|
||
//#include "CpuUsage.h"
|
||
|
||
|
||
#include <process.h>
|
||
|
||
#include <afxtempl.h>
|
||
|
||
|
||
|
||
////////////////////////////////////////////////////////////////////
|
||
#define NC_CLIENT_CONNECT 0x0001
|
||
#define NC_CLIENT_DISCONNECT 0x0002
|
||
#define NC_TRANSMIT 0x0003
|
||
#define NC_RECEIVE 0x0004
|
||
#define NC_RECEIVE_COMPLETE 0x0005 // 完整接收
|
||
|
||
class CLock
|
||
{
|
||
public:
|
||
CLock(CRITICAL_SECTION& cs, const CString& strFunc)
|
||
{
|
||
m_strFunc = strFunc;
|
||
m_pcs = &cs;
|
||
Lock();
|
||
}
|
||
~CLock()
|
||
{
|
||
Unlock();
|
||
|
||
}
|
||
|
||
void Unlock()
|
||
{
|
||
LeaveCriticalSection(m_pcs);
|
||
TRACE(_T("LC %d %s\n") , GetCurrentThreadId() , m_strFunc);
|
||
}
|
||
|
||
void Lock()
|
||
{
|
||
TRACE(_T("EC %d %s\n") , GetCurrentThreadId(), m_strFunc);
|
||
EnterCriticalSection(m_pcs);
|
||
}
|
||
|
||
|
||
protected:
|
||
CRITICAL_SECTION* m_pcs;
|
||
CString m_strFunc;
|
||
|
||
};
|
||
|
||
enum IOType
|
||
{
|
||
IOInitialize,
|
||
IORead,
|
||
IOWrite,
|
||
IOIdle
|
||
};
|
||
|
||
|
||
class OVERLAPPEDPLUS
|
||
{
|
||
public:
|
||
OVERLAPPED m_ol;
|
||
IOType m_ioType;
|
||
|
||
OVERLAPPEDPLUS(IOType ioType) {
|
||
ZeroMemory(this, sizeof(OVERLAPPEDPLUS));
|
||
m_ioType = ioType;
|
||
}
|
||
};
|
||
|
||
|
||
struct ClientContext
|
||
{
|
||
SOCKET m_Socket;
|
||
// Store buffers
|
||
CBuffer m_WriteBuffer;
|
||
CBuffer m_CompressionBuffer; // 接收到的压缩的数据
|
||
CBuffer m_DeCompressionBuffer; // 解压后的数据
|
||
|
||
UINT m_Dialog[2]; // 放对话框列表用，第一个int是类型，第二个是CDialog的地址
|
||
UINT m_nTransferProgress;
|
||
// Input Elements for Winsock
|
||
WSABUF m_wsaInBuffer;
|
||
BYTE m_byInBuffer[8192];
|
||
|
||
// Output elements for Winsock
|
||
WSABUF m_wsaOutBuffer;
|
||
HANDLE m_hWriteComplete;
|
||
|
||
|
||
BOOL m_bIsMainSocket; // 是不是主socket
|
||
|
||
ClientContext* m_pWriteContext;
|
||
ClientContext* m_pReadContext;
|
||
};
|
||
|
||
template<>
|
||
inline UINT AFXAPI HashKey(CString & strGuid)
|
||
{
|
||
return HashKey( (LPCTSTR) strGuid);
|
||
}
|
||
|
||
#include "Mapper.h"
|
||
|
||
typedef void (CALLBACK* NOTIFYPROC)(LPVOID, ClientContext*, UINT nCode);
|
||
|
||
typedef CList<ClientContext*, ClientContext* > ContextList;
|
||
|
||
|
||
class CMainFrame;
|
||
|
||
class CIOCPServer
|
||
{
|
||
public:
|
||
void DisconnectAll();
|
||
CIOCPServer();
|
||
virtual ~CIOCPServer();
|
||
|
||
NOTIFYPROC m_pNotifyProc;
|
||
CMainFrame* m_pFrame;
|
||
|
||
bool Initialize(NOTIFYPROC pNotifyProc, CMainFrame* pFrame, UINT nPort);
|
||
|
||
static unsigned __stdcall ListenThreadProc(LPVOID lpVoid);
|
||
static unsigned __stdcall ThreadPoolFunc(LPVOID WorkContext);
|
||
static CRITICAL_SECTION m_cs;
|
||
|
||
void Send(ClientContext* pContext, LPBYTE lpData, UINT nSize);
|
||
void PostRecv(ClientContext* pContext);
|
||
|
||
bool IsRunning();
|
||
void Shutdown();
|
||
void ResetConnection(ClientContext* pContext);
|
||
|
||
protected:
|
||
void InitializeClientRead(ClientContext* pContext);
|
||
BOOL AssociateSocketWithCompletionPort(SOCKET device, HANDLE hCompletionPort, DWORD dwCompletionKey);
|
||
void RemoveStaleClient(ClientContext* pContext, BOOL bGraceful);
|
||
void MoveToFreePool(ClientContext *pContext);
|
||
ClientContext* AllocateContext();
|
||
|
||
LONG m_nWorkerCnt;
|
||
|
||
bool m_bInit;
|
||
bool m_bDisconnectAll;
|
||
|
||
void CloseCompletionPort();
|
||
void OnAccept();
|
||
bool InitializeIOCP(void);
|
||
void Stop();
|
||
|
||
ContextList m_listContexts;
|
||
ContextList m_listFreePool;
|
||
WSAEVENT m_hEvent;
|
||
SOCKET m_socListen;
|
||
HANDLE m_hKillEvent;
|
||
HANDLE m_hThread;
|
||
HANDLE m_hCompletionPort;
|
||
bool m_bTimeToKill;
|
||
|
||
LONG m_nKeepLiveTime; // 心跳超时
|
||
|
||
|
||
|
||
CString GetHostName(SOCKET socket);
|
||
|
||
void CreateStream(ClientContext* pContext);
|
||
|
||
BEGIN_IO_MSG_MAP()
|
||
IO_MESSAGE_HANDLER(IORead, OnClientReading)
|
||
IO_MESSAGE_HANDLER(IOWrite, OnClientWriting)
|
||
IO_MESSAGE_HANDLER(IOInitialize, OnClientInitializing)
|
||
END_IO_MSG_MAP()
|
||
|
||
bool OnClientInitializing (ClientContext* pContext, DWORD dwSize = 0);
|
||
bool OnClientReading (ClientContext* pContext, DWORD dwSize = 0);
|
||
bool OnClientWriting (ClientContext* pContext, DWORD dwSize = 0);
|
||
|
||
};
|
||
|
||
#endif // !defined(AFX_IOCPSERVER_H__75B80E90_FD25_4FFB_B273_0090AA43BBDF__INCLUDED_)
|