IOCP的封装类[By Sodme]

[原创]Linux下使用Select模型的两个类Client And Server

如果你被骗了

大 | 中 | 小

2008/03/10 发表时间

16:53

3168

huzhangyou2002

开发语言 » 代码库

在网上搜索到的很早的大宝兄的代码，写的还是满不错的，多采用List实现。希望对大家开发有所帮助。

下载文件 (已下载 911 次)

点击这里下载文件: iocp.rar

部分数据结构：

////////////////////////////////////////
// <a href="http://www.server-development.cn/go.php/tags/IOCP/" target="_blank" title="http://www.server-development.cn/go.php/tags/IOCP/" class="mykeyword">IOCP</a>.h文件

#ifndef __IOCP_H__
#define __IOCP_H__
#include <stdio.h>
#include <winsock2.h>
#include <windows.h>
#include <Mswsock.h>
#define USE_DATA_LONGTH 8012
#define NET_DATA_LONGTH 4096
#define BUFFER_SIZE 1024*4    // I/O请求的缓冲区大小
#define MAX_THREAD  2      // I/O服务线程的数量

// 这是per-I/O数据。它包含了在套节字上处理I/O操作的必要信息
struct CIOCPBuffer
{
  WSAOVERLAPPED ol;

SOCKET sClient;      // AcceptEx接收的客户方套节字

char *buff;        // I/O操作使用的缓冲区
  int nLen;        // buff缓冲区（使用的）大小

ULONG nSequenceNumber;  // 此I/O的序列号

int nOperation;      // 操作类型
#define OP_ACCEPT  1
#define OP_WRITE  2
#define OP_READ    3
  CIOCPBuffer *pNext;
};

// 这是per-Handle数据。它包含了一个套节字的信息
struct CIOCPContext
{
  SOCKET s;            // 套节字句柄

SOCKADDR_IN addrLocal;      // 连接的本地地址
  SOCKADDR_IN addrRemote;      // 连接的远程地址

BOOL bClosing;          // 套节字是否关闭

int nOutstandingRecv;      // 此套节字上抛出的重叠操作的数量
  int nOutstandingSend;

ULONG nReadSequence;      // 安排给接收的下一个序列号
  ULONG nCurrentReadSequence;    // 当前要读的序列号
  CIOCPBuffer *pOutOfOrderReads;  // 记录没有按顺序完成的读I/O
  BYTE* lpBufBegin; //拼包缓冲区头指针
    BYTE* lpBufEnd; //拼包缓冲区尾指针
    BYTE arrayDataBuf[USE_DATA_LONGTH]; //拼包缓冲区
  CRITICAL_SECTION Lock;      // 保护这个结构
  CIOCPContext *pNext;
};

class CIOCPServer   // 处理线程
{
public:
  CIOCPServer();
  ~CIOCPServer();

// 开始服务
  BOOL Start(int nPort = 4567, int nMaxConnections = 2000, 
      int nMaxFreeBuffers = 200, int nMaxFreeContexts = 100, int nInitialReads = 4);
  // 停止服务
  void Shutdown();

// 关闭一个连接和关闭所有连接
  void CloseAConnection(CIOCPContext *pContext);
  void CloseAllConnections();

// 取得当前的连接数量
  ULONG GetCurrentConnection() { return m_nCurrentConnection; }

// 向指定客户发送文本
  BOOL SendText(CIOCPContext *pContext, char *pszText, int nLen);

protected:

// 申请和释放缓冲区对象
  CIOCPBuffer *AllocateBuffer(int nLen);
  void ReleaseBuffer(CIOCPBuffer *pBuffer);

// 申请和释放套节字上下文
  CIOCPContext *AllocateContext(SOCKET s);
  void ReleaseContext(CIOCPContext *pContext);

// 释放空闲缓冲区对象列表和空闲上下文对象列表
  void FreeBuffers();
  void FreeContexts();

// 向连接列表中添加一个连接
  BOOL AddAConnection(CIOCPContext *pContext);

// 插入和移除未决的接受请求
  BOOL InsertPendingAccept(CIOCPBuffer *pBuffer);
  BOOL RemovePendingAccept(CIOCPBuffer *pBuffer);

// 取得下一个要读取的
  CIOCPBuffer *GetNextReadBuffer(CIOCPContext *pContext, CIOCPBuffer *pBuffer);

// 投递接受I/O、发送I/O、接收I/O
  BOOL PostAccept(CIOCPBuffer *pBuffer);
  BOOL PostSend(CIOCPContext *pContext, CIOCPBuffer *pBuffer);
  BOOL PostRecv(CIOCPContext *pContext, CIOCPBuffer *pBuffer);

void HandleIO(DWORD dwKey, CIOCPBuffer *pBuffer, DWORD dwTrans, int nError);

// 事件通知函数
  // 建立了一个新的连接
  virtual void OnConnectionEstablished(CIOCPContext *pContext, CIOCPBuffer *pBuffer) = 0;
  // 一个连接关闭
  virtual void OnConnectionClosing(CIOCPContext *pContext, CIOCPBuffer *pBuffer) = 0;
  // 在一个连接上发生了错误
  virtual void OnConnectionError(CIOCPContext *pContext, CIOCPBuffer *pBuffer, int nError) = 0;
  // 一个连接上的读操作完成
  virtual void OnReadCompleted(CIOCPContext *pContext, CIOCPBuffer *pBuffer)=0;
  // 一个连接上的写操作完成
  virtual void OnWriteCompleted(CIOCPContext *pContext, CIOCPBuffer *pBuffer)=0;
  //拆包处理过程
  virtual bool SplitPacket(CIOCPContext *pContext, CIOCPBuffer *pBuffer)=0;
  virtual void OnPacketError()=0;

protected:

// 记录空闲结构信息
  CIOCPBuffer *m_pFreeBufferList;
  CIOCPContext *m_pFreeContextList;
  int m_nFreeBufferCount;
  int m_nFreeContextCount;  
  CRITICAL_SECTION m_FreeBufferListLock;
  CRITICAL_SECTION m_FreeContextListLock;

// 记录抛出的Accept请求
  CIOCPBuffer *m_pPendingAccepts;   // 抛出请求列表。
  long m_nPendingAcceptCount;
  CRITICAL_SECTION m_PendingAcceptsLock;

// 记录连接列表
  CIOCPContext *m_pConnectionList;
  int m_nCurrentConnection;
  CRITICAL_SECTION m_ConnectionListLock;

// 用于投递Accept请求
  HANDLE m_hAcceptEvent;
  HANDLE m_hRepostEvent;
  LONG m_nRepostCount;

int m_nPort;        // <a href="http://www.server-development.cn/go.php/tags/服务器/" target="_blank" title="http://www.server-development.cn/go.php/tags/服务器/" class="mykeyword">服务器</a>监听的端口

int m_nInitialAccepts;
  int m_nInitialReads;
  int m_nMaxAccepts;
  int m_nMaxSends;
  int m_nMaxFreeBuffers;
  int m_nMaxFreeContexts;
  int m_nMaxConnections;

HANDLE m_hListenThread;      // 监听线程
  HANDLE m_hCompletion;      // 完成端口句柄
  SOCKET m_sListen;        // 监听套节字句柄
  LPFN_ACCEPTEX m_lpfnAcceptEx;  // AcceptEx函数地址
  LPFN_GETACCEPTEXSOCKADDRS m_lpfnGetAcceptExSockaddrs; // GetAcceptExSockaddrs函数地址

BOOL m_bShutDown;    // 用于通知监听线程退出
  BOOL m_bServerStarted;  // 记录服务是否启动

private:  // 线程函数
  static DWORD WINAPI _ListenThreadProc(LPVOID lpParam);
  static DWORD WINAPI _WorkerThreadProc(LPVOID lpParam);
};

#endif // __IOCP_H__

NEO

2008/11/15 21:27

很奇怪这些代码不会碰到10055的错误码吗？从代码上看应该很难避免。

huzhangyou2002 回复于 2008/11/15 22:17

您好，这个类只是大宝兄的demo,真实的类处理起来会干净多了

diablof

2008/07/23 17:58

因为我最近在写一个服务器压力测试程序

需要客户端多线程连接服务器

diablof

2008/07/23 17:56

请问可以用完成端口写服务器么？

还有，为何不能下载？我已经登录了

hollyhock13

2008/06/05 16:57

1.感觉这个类的超时检测部分有点问题
在超时关闭连接SOCKET后,还应该把请求列表中的缓冲区移除掉才对.应该掉用RemovePendingAccept.
2.超时那里应该注意临界区问题.轮询"请求列表"时,没有临界处理,可能导致正在使用的缓冲区Buffer被工作线程移除,释放.
(当然这种情况出现的机率很小)

你认为呢,希望能够指教.

huzhangyou2002 回复于 2008/06/05 19:20

很久以前分析过这几天没有时间看过几天一定认真再看看是否真是如此非常感谢您的关注

分页： 1/1