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

【转】完成端口的一个简单封装类

 

/////////////////////////////////////////////////////////////////////////////////////

//   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;

}