IOCP完成端口的一个简单封装类

2017-09-13 20:43:56来源:CSDN作者:sanzhixiong86人点击

分享
///////////////////////////////////////////////////////////////////////////////////////   Iocp 头文件 #pragma once #include <winsock2.h>#pragma comment( lib, "ws2_32.lib" ) const int OP_READ = 0;const int OP_WRITE = 1;const int OP_ACCEPT = 2; /*     OVERLAPPEDPLUS 结构体设计思路     OVERLAPPED 是一个固定的用于处理网络消息事件返回值的结构体变量     在完成端口和重叠I/O模型里用于返回消息事件的结果     因为在处理网络消息的时候,发送的是一个返回值的结构体指针,只要结构体     的前面部分满足系统的要求,在系统操作成功的时候也就会把这个结构体指针     发回给用户,我们只要在系统定义的结构体后面扩展一些自己的东西,就可以     很轻松的确定该消息是谁发过来的。     不过好像完成端口在设计的时候也满足了这样的需求,所以在这里我只是放入     一些与系统连接有关的数据,用户需要存放的数据这里就不在存放     这里存储与系统相关的数据有:     socket     OpCode 本次消息的操作类型(在完成端口的操作里面,是以消息通知系统,         读数据/写数据,都是要发这样的消息结构体过去的,所以如果系统要同时         进行读写操作的话,就需要有一个变量来区分操作了)      WSABUF   wbuf;                  //   读写缓冲区结构体变量     DWORD    dwBytes, dwFlags; //   一些在读写时用到的标志性变量     char buf[4096];                  //   自己的缓冲区     上面的4个变量存放的是一些与消息相关的数据,都是一些操作上用到的,     这些东西都是固定的,具体作用需要参考一下完成端口相关函数的参数接口*/struct OVERLAPPEDPLUS{     OVERLAPPED    ol;     SOCKET        s;     int OpCode;     WSABUF   wbuf;     DWORD    dwBytes, dwFlags;     char buf[4096];}; class CIOCP{protected:     HANDLE g_hwThread;     //   工作线程句柄     DWORD m_wthreadID;     HANDLE g_haThread;     //   连接线程句柄     DWORD m_athreadID;public:     bool m_workThread;     bool m_acceptThread;     HANDLE m_hIocp;             //   完成端口的句柄     SOCKET m_sSocket;    public:     CIOCP(void);     ~CIOCP(void);     virtual void OnRead(void * p, char *buf, int len){};     virtual void OnAccept(SOCKET socket);     virtual void OnClose(void * p){};     bool SetIoCompletionPort(SOCKET socket, void *p, char *buf = NULL, int len = 0);         //   把一个socket与一个自定义的结构体关联到完成端口(相当于把socket与一个结构体变量进行绑定),         //   这样当发送上面3种网络事件的时候,该结构体变量会再传回给程序         //   这样就可以区分当前网络事件是那个socket发出的     bool Init(void);     bool Listen(int port);     static DWORD __stdcall WorkThread(LPVOID Param);     static DWORD __stdcall AcceptThread(LPVOID Param);}; class CIOCPClient: public CIOCP{protected:     SOCKET m_socket;public:     bool Connect(char *ip, int port);     void Send(char *buf, int len);};           ////////////////////////////////////////////////////////////////////////////////////////////   Iocp 实现文件 #include "StdAfx.h"#include "iocp.h" static bool bInit = false; DWORD __stdcall CIOCP::WorkThread(LPVOID Param){     CIOCP * pthis = (CIOCP *)Param;      void * re;     OVERLAPPED * pOverlap;     DWORD berByte;     while(pthis->m_workThread)     {         int ret;         ret = GetQueuedCompletionStatus(pthis->m_hIocp, &berByte, (LPDWORD)&re, (LPOVERLAPPED *)&pOverlap, INFINITE);          if (ret == ERROR_SUCCESS)         {          }          if (berByte == 0)         {              //   客户端断开连接              pthis->OnClose(re);              OVERLAPPEDPLUS *olp = (OVERLAPPEDPLUS *)pOverlap;              closesocket(olp->s);              delete olp;        //   释放 与socket绑定的结构体变量              continue;         }          if (re == NULL) return 0;          OVERLAPPEDPLUS *olp = (OVERLAPPEDPLUS *)pOverlap;          switch(olp->OpCode)         {         case OP_READ:              pthis->OnRead(re, olp->wbuf.buf, berByte);     //   调用 OnRead() 通知应用程序,服务器收到来自客户端的网络数据              WSARecv(olp->s, &olp->wbuf, 1, &olp->dwBytes, &olp->dwFlags, &olp->ol, NULL); //   继续调用一个接收的 I/O 异步请求              break;         default:              break;         }     }     return 0;} DWORD __stdcall CIOCP::AcceptThread(LPVOID Param){     CIOCP * pthis = (CIOCP *)Param;     while(pthis->m_acceptThread)     {         SOCKET client;         if ((client= accept(pthis->m_sSocket, NULL, NULL)) == INVALID_SOCKET)         {              //   错误处理         }         pthis->OnAccept(client);    //   调用 OnAccept()通知应用程序有新客户端连接             }     return 1;} CIOCP::CIOCP(void){} CIOCP::~CIOCP(void){} bool CIOCP::Init(void){     if (bInit)         return true;      WSADATA wsd;     if (WSAStartup(MAKEWORD(2,2), &wsd) != 0)         return false;      bInit = true;     return true;} bool CIOCP::Listen(int port){     if (!bInit)         if (!Init())              return false;      m_sSocket = socket(AF_INET, SOCK_STREAM, 0);      if (m_sSocket == INVALID_SOCKET)         return false;      //SOCKADDR_IN addr;     sockaddr_in addr;     addr.sin_family = AF_INET;     addr.sin_port = htons(port);     addr.sin_addr.S_un.S_addr = htonl(INADDR_ANY);     //addr.sin_addr.S_un.S_addr = inet_addr(ip);      if (bind(m_sSocket, (struct sockaddr *)&addr, sizeof(addr)) == SOCKET_ERROR)         return false;      if (listen(m_sSocket, 10) == SOCKET_ERROR)         return false;      if ((m_hIocp = CreateIoCompletionPort(INVALID_HANDLE_VALUE, NULL, NULL, 0)) == NULL)     //   创建完成端口的句柄         return false;      this->m_acceptThread = true;     g_haThread = CreateThread(NULL, 0, AcceptThread, (LPVOID)this, 0, &m_athreadID);    //   创建连接线程,用来接收客户端的连接      this->m_workThread = true;     g_hwThread = CreateThread(NULL, 0, WorkThread, (LPVOID)this, 0, &m_wthreadID); //   创建工作线程,用来处理完成端口消息的     return true;} bool CIOCP::SetIoCompletionPort(SOCKET socket, void *p, char *buf, int len){     if (CreateIoCompletionPort((HANDLE)socket, m_hIocp, (ULONG_PTR)p, 0) == NULL)         return false;      OVERLAPPEDPLUS *olp = new OVERLAPPEDPLUS;     memset(olp, 0, sizeof(OVERLAPPEDPLUS));     olp->s = socket;     if (buf)     {         //   这里可以使用用户自定义的缓冲区地址,如果用户不想设置,也可以采用默认分配的缓冲区         olp->wbuf.buf = buf;         olp->wbuf.len = len;     }     else     {         olp->wbuf.buf = olp->buf;         olp->wbuf.len = 4096;     }     olp->OpCode = OP_READ;     int ret = WSARecv(olp->s, &olp->wbuf, 1, &olp->dwBytes, &olp->dwFlags, &olp->ol, NULL);     if (ret == SOCKET_ERROR)         if (WSAGetLastError() != ERROR_IO_PENDING)              return false;     return true;} void CIOCP::OnAccept(SOCKET socket){     this->SetIoCompletionPort(socket, NULL);} //===================================================================================bool CIOCPClient::Connect(char *ip, int port){         //   连接服务器     if (!bInit)         if (!Init())              return false;      //   初始化连接socket     m_socket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);     if (m_socket == SOCKET_ERROR)     {//       printf("cocket Create fail");         return false;     }      // 填写服务器地址信息     // 端口为1982     // IP地址为INADDR_ANY,注意使用htonl将IP地址转换为网络格式ServerAddr.sin_family = AF_INET;     sockaddr_in ClientAddr;     ClientAddr.sin_family = AF_INET;     ClientAddr.sin_port = htons(port);        ClientAddr.sin_addr.s_addr = inet_addr(ip);      // 绑定监听端口     bind(m_socket, (SOCKADDR *)&ClientAddr, sizeof(ClientAddr));      if (connect(m_socket, (SOCKADDR *)&ClientAddr, sizeof(ClientAddr)) == SOCKET_ERROR)     {         return false;     }         if ((m_hIocp = CreateIoCompletionPort(INVALID_HANDLE_VALUE, NULL, NULL, 0)) == NULL)     //   创建完成端口的句柄         return false;      this->m_workThread = true;     g_hwThread = CreateThread(NULL, 0, WorkThread, (LPVOID)this, 0, &m_wthreadID); //   创建工作线程,用来处理完成端口消息的      this->SetIoCompletionPort(m_socket, &m_socket);    //   设置完成端口监听的socket     return true;} void CIOCPClient::Send(char *buf, int len){     send(m_socket, buf, len, 0);}


///////////////////////////////////////////////////////////////////////////////////// IOCPclient 应用代码 #include "stdafx.h"#include "IOCP.h"#include "TClientSocket.h"class Iocp :public CIOCPClient{      void OnRead(void * p, char *buf, int len)      {          printf(buf);          Sleep(1000);          this->Send(buf, len);      }}; int _tmain(int argc, _TCHAR* argv[]){     Iocp iocp;     iocp.Init();     iocp.Connect("127.0.0.1", 4311);     iocp.Send("test/0", 5);         gets(new char[1000]);     return 0;} 


最新文章

123

最新摄影

微信扫一扫

第七城市微信公众平台