winsock 完成端口 服务器模型(1)

//IOCPServer.h

#include <winsock2.h>
#include <windows.h>
#include <Mswsock.h>

#define BUFFER_SIZE 1024
#define MAXFREEBUFFERS 200
#define MAXFREECONTEXTS 100
#define MAXCONNECTIONS 2000
#define OP_ACCEPT 1
#define OP_RECV 2
#define OP_SEND 3

#pragma comment(lib,"WS2_32.lib")

//缓冲区对象
struct IOCPBUFFER
{
WSAOVERLAPPED ol; //重叠I/O
SOCKET sClient; //新收到的客户端 (只对Listen套接字有效)
char *buf; //缓冲区指针
int nLen; //buf的长度
ULONG nSequenceNumber; //当前对象的序列号
int nOperationType; //操作类型 :OP_ACCEPT,OP_RECV ,OP_SEND
IOCPBUFFER *pNext; //指向下一个IOCPBUFFER
};

//客户上下文对象
struct IOCPCONTEXT
{
SOCKET s; //套接字句柄
sockaddr_in addrLocal; //连接的本地地址
sockaddr_in addrRemote;//连接的远程地址
int nOutstandingRecv; //此套接字上被抛出的重叠操作数量(recv)
int nOutstandingSend; //此套接字上被抛出的重叠操作数量 (send)
ULONG nReadSequence; //安排给接收下一个序列号
ULONG nCurrentSequence;//当前要读的序列号
CRITICAL_SECTION lock; //保护这个结构锁
IOCPBUFFER *pOutofOrederReads; //记录没有按顺序完成的读I/O
IOCPCONTEXT *pNext; //指向下一个IOCPCONTEXT
};

class IOCPServer
{
public:
IOCPServer(void);
virtual ~IOCPServer(void);
void Start(); //服务起动
bool GetIsServerStart(); //获取程序是否运行
char *GetErrorMsg(); //获取错误信息

public:
void virtual CompletedRecvDate(IOCPBUFFER* pBuffer,IOCPCONTEXT* pContext); //接收数据完成
void virtual CompletedWriteDate(IOCPBUFFER* pBuffer,IOCPCONTEXT* pContext);//写完成
void virtual ConnectionComplete(IOCPCONTEXT *pContext); //pContext客户已连接上
void virtual DisConnection(IOCPCONTEXT *pContext); //pContext客户断开连接
BOOL WriteDate(char *buf,int len,IOCPCONTEXT *pContext);//写数据
void ShutDown();

protected:
void InitSocket(); //初始化
void ResetValue();
IOCPBUFFER* AllocBuffer(int nLen); //从内存池中取一块可用的缓冲块,如果内存池中没有就从内存中分配空间
void FreeBuffer(IOCPBUFFER* pBuffer); //释放缓冲对象
IOCPCONTEXT* AllocContext(SOCKET s); //从客户上下文列表中取一个客户上下文,如果列表为空就分配一个
void FreeContext(IOCPCONTEXT* pContext);//释放客户上下文对象
bool PostAccept(IOCPBUFFER* pBuffer); //投递Accept
bool PostRecv(IOCPBUFFER* pBuffer,IOCPCONTEXT* pContext); //投递Recv
bool PostSend(IOCPBUFFER* pBuffer,IOCPCONTEXT* pContext); //投递write
IOCPBUFFER* GetNextReadBuffer(IOCPCONTEXT* pContext,IOCPBUFFER* pBuffer); //按顺序读取缓冲区中的数据,包重组
bool AddConnection(IOCPCONTEXT* pContext); //添加已连接列表
void CloseConnection(IOCPCONTEXT* pContext);//关闭连接
void InsertToPostAccept(IOCPBUFFER *pBuffer);
void RemoveFormPostAccept(IOCPBUFFER *pBuffer);
void CloseAllConnects(); //关闭所有的连接
void FreeBuffers(); //释放内存池
void FreeContexts(); //释放空闲客户上下文列表

private:
bool m_bIsServerStart; //服务是否正常起动
char *m_pErrorMsg; //程序没有出错的原因

BOOL m_bShutDown;//关闭服务

IOCPBUFFER* m_pFreeBufferList; //指向内存池列表的头指针
CRITICAL_SECTION m_FreeBufferListLock; //保护内池列表锁
int m_FreeBufferCount; //内存池中缓冲区块的数量

IOCPCONTEXT* m_pFreeContextList; //空闲客户上下文列表指针
CRITICAL_SECTION m_FreeContextListLock; //空闲客户上下文列表锁
int m_FreeContextCount; //空闲客户上下文的数量

IOCPCONTEXT* m_pConnectContextList; //被连接上的客户列表
CRITICAL_SECTION m_ConnectContextListLock; //被连接上的客户列表锁
int m_ConnectContextCount;//已连接的数量

LPFN_ACCEPTEX m_lpfnAcceptEx; //指向AcceptEx的指针
LPFN_GETACCEPTEXSOCKADDRS m_lpfnGetAddr; //指向日GetAcceptSockAddrs指针
SOCKET m_sListen; //监听套接字

IOCPBUFFER * m_pPostAcceptList; //被投出的accept
CRITICAL_SECTION m_PostAcceptListLock;

HANDLE m_hCompletion; //完成端口句柄

HANDLE m_hListenThread; //监听线程名柄

HANDLE m_hAcceptEvent; //Accept事件
HANDLE m_hRePostAcceptEvent; //重投Accept事件;

HANDLE m_hWorkThread[2];
private:
//线程函数
static DWORD WINAPI _ListenThreadProc(LPVOID lpParam);
static DWORD WINAPI _WorkerThreadProc(LPVOID lpParam);
};