IOCP网络模型基本步骤

一、定义结构

   typedef struct PER_HANDLE_DATA 

   {

      SOCKET s;

      SOCKADDR_IN RemoteAddr; 

   }*PPER_HANDLE_DATA;

   typedef struct PER_IO_DATA

   {

     WSAOVERLAPPED ol;

     char buf[12];

     #define OP_READ  1

     #define OP_WRITE 2

     int nOperation;

    }*PPER_IO_DATA;

 

二、创建一个完成端口

   HANDLE hCompletion=CreateIoCompletionPort(INVALID_HANDLE_VALUE,0,0,0);

三、创建合适的工作线程

   SYSTEM_INFO si;

   GetSystemInfo(&si);

   for(int i=0;i<si.dwNumberOfProcessors*2;++i)//如果需要使用WaitForSingleObject之类使线程等待，避免CPU空闲

   CreateThread(NULL,0,WorkThread,(LPVOID)hCompletion,0,0);

四、绑定监听socket

   SOCKET sListen=socket(AF_INET,SOCK_STREAM,0);

   SOCKADDR_IN srvaddr;

   srvaddr.sin_addr.S_un.S_addr=INADDR_ANY;

   srvaddr.sin_family=AF_INET;

   srvaddr.sin_port=htons(4567);

   bind(s,(SOCKADRR*)&srvaddr,sizeof srvaddr);

   listen(s,5);

五、处理客户连接

   while(1)

   {

      SOCKADDR_IN Remoteaddr;

      int nlen=sizeof Remoteaddr;

      SOCKET sAccept=accept(sListen,(SOCKADDR*)Remoteaddr,&nlen);

      PPER_HANDLE_DATA pPerHandle=(PER_HANDLE_DATA)GlobalAlloc(GPTR,sizeof(PER_HANDLE_DATA));

      pPerHandle->s=sAccept;

      memcpy(&pPerHandle->RemoteAddr,Remoteaddr,nlen);

 

      CreateIoCompletionPort((HANDLE)pPerHandle->s,hCompletion,(DWORD)perHandle,0);//关联per数据

      PPER_IO_DATA pPerIO=(PER_IO_DATA)GlobalAlloc(GPTR,sizeof(PER_IO_DATA));

      pPerIO->nOperation=OP_READ;

      WSABUF buf;

      buf.buf=pPerIO->buf;

      buf.len=12;

      DWORD dwRecv;

      DWORD dwFlags=0;

      WSARecv(pPerHandle->s,&buf,1,&dwRecv,&dwFlags,&pPerIO->ol,NULL);//投递第一个请求

   }

六、处理客户请求(工作线程)

   HANDLE hCompletion=(HANDLE)lpParam;

   DWORD dwTrans;

   PPER_HANDLE_DATA pPerHandle;

   PPER_IO_DATA pPerIO;

 

   while(1)

   {

     bool ok=GetQueuedCompletionStatus(hCompletion,&dwTrans,(PULONG_PTR)&pPerHandle,(LPOVERLAPPED*)&pPerIO,WSA_INFINITE);//查看是否有异步IO请求完成 完成返回

     if(!bok)//出错了

     {

       closesocket(pPerHandle->s);

       GlobalFree(pPerHandle);

       GlobalFree(pPerIO);

       continue;

     }

     if(dwTrans==0&&(pPerIO->nOperation==OP_READ||pPerIO->nOperation==OP_WRITE))//连接关闭了

     {

      closesocket(pPerHandle->s);

      GlobalFree(pPerHandle);

      GlobalFree(pPerIO);

      continue; 

     }

     switch(pPerIO->nOperation)

     {

       case OP_READ:

         pPerIO->buf[dwTrans]='/0';

         printf(pPerIO->buf);

         WSABUF buf;

         buf.buf=pPerIO->buf;

         buf.len=12;

         pPerIO->nOperation=OP_READ;

         DWORD nFlags=0;

         WSARecv(pPerHandle->s,&buf,1,&dwTrans,&nFlags,&pPerIO->ol,NULL);//继续投递

         break;

      case OP_WRITE:

      case OP_ACCEPT:

         break;

     }

 

   }