完成端口模型服务端

windows  完成端口模型服务端

#include<iostream>
#include<winsock2.h>
#pragma comment(lib,"WS2_32.lib")
using namespace std;
const int ServerPort=4399;
const int MSGSIZE=1024;
typedef enum
{
 RECV_POST,
 SEND_POST
}OPERATION_TYPE;
typedef struct 
{
 WSAOVERLAPPED wsaOverlapped;
 WSABUF wsaBuffer;
 char dataBuffer[MSGSIZE];
 DWORD recvedDataSize;
 DWORD flags;
 OPERATION_TYPE operationType;
}COM_IO_OPERATION_DATA,*PIO_OPERATION_DATA;
typedef struct _completionKey
{
   SOCKET sock;
   SOCKADDR_IN sockAddr;
}COMPLETION_KEY,*PCOMPLITON_KEY;
int LoadSocketLib()
{
 WORD wVersionRequested;
 WSADATA wsaData;
 int err;
 
 wVersionRequested = MAKEWORD( 2, 2 );
 
 err = WSAStartup( wVersionRequested, &wsaData );
 if ( err != 0 ) {                       
  return -1;
 }
 if ( LOBYTE( wsaData.wVersion ) != 2 ||
        HIBYTE( wsaData.wVersion ) != 2 ) {
  WSACleanup( );
  return -1; 
 }
 return 0;
}
DWORD WINAPI HandleThread(LPVOID completionPortID);
BOOL AsyncRecvData(SOCKET clientSOCK);
int main()
{
 system("color 2f");
    //加载套接字库
 if(LoadSocketLib()!=0)
 {
  cout<<"Load winsock failed"<<endl;
        return -1;
 }
 //创建完成端口
    HANDLE completionPort=CreateIoCompletionPort(INVALID_HANDLE_VALUE,NULL,0,0);
 if(completionPort==NULL)
 {
  cout<<"创建完成端口失败"<<endl;
  return -1;
 }
 SYSTEM_INFO systemInfo;
 GetSystemInfo(&systemInfo);
 DWORD dw_ThreadId;//线程ID
 //创建线程池
 for(int i=0;i<systemInfo.dwNumberOfProcessors;i++)
 {
  CreateThread(NULL,0,HandleThread,completionPort,0,&dw_ThreadId);
 }
    //创建套接字、绑定、设为监听状态
 SOCKET serverSock=socket(AF_INET,SOCK_STREAM,IPPROTO_TCP);
 if(serverSock==INVALID_SOCKET)
 {
  cout<<"create socket failed"<<endl;
  return -1;
 }
 SOCKADDR_IN serverAddr;
 serverAddr.sin_addr.S_un.S_addr=htonl(INADDR_ANY);
 serverAddr.sin_family=AF_INET;
 serverAddr.sin_port=htons(ServerPort);
 if(SOCKET_ERROR==bind(serverSock,(SOCKADDR*)&serverAddr,sizeof(SOCKADDR_IN)))
 {
  cout<<"binding failed"<<endl;
  return -1;
 }
 if(SOCKET_ERROR==listen(serverSock,5))
 {
  cout<<"listen failed"<<endl;
  return -1;
 }
 //接受客户端连接请求，并关联到完成端口
 SOCKADDR_IN clientAddr;
 int addrSize=sizeof(SOCKADDR_IN);
// PIO_OPERATION_DATA lpper_IO_Data=NULL;
 cout<<"/t/t/t"<<"...Server begin..."<<endl;
 PCOMPLITON_KEY  pCompletion=NULL;
 while(1)
 {
  SOCKET clientSock=accept(serverSock,(SOCKADDR*)&clientAddr,&addrSize);
  if(clientSock==INVALID_SOCKET)
  {
   closesocket(clientSock);
   continue;
  }
  //pCompletion=(PCOMPLITON_KEY)malloc(sizeof(COMPLETION_KEY));
     // pCompletion->sock=clientSock;
  //pCompletion->sockAddr=clientAddr;
  cout<<"the client addr is..."<<inet_ntoa(clientAddr.sin_addr)<<"/t"<<ntohs(clientAddr.sin_port)<<endl;
  if(NULL==CreateIoCompletionPort((HANDLE)clientSock,completionPort,/*(DWORD)pCompletion*/(DWORD)clientSock,0))
  {
   cout<<"套接字关联失败"<<endl;
   closesocket(clientSock);
   continue;
  }
  AsyncRecvData(clientSock);//发起异步接受操作
 }
    PostQueuedCompletionStatus(completionPort,0xFFFFFFFF,0,NULL);
 CloseHandle(completionPort);
 closesocket(serverSock);
    WSACleanup();
 return 0;
}
BOOL AsyncRecvData(SOCKET clientSock)
{
 PIO_OPERATION_DATA lpper_IO_Data=NULL;
 lpper_IO_Data=(PIO_OPERATION_DATA)HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,sizeof(PIO_OPERATION_DATA));//从堆中分配内存块给关联结构体
    if(lpper_IO_Data==NULL)
 {
  cout<<"allocate failed"<<endl;
  closesocket(clientSock);
  //continue;
  return false;
 }
 ZeroMemory(&lpper_IO_Data->wsaOverlapped,sizeof(lpper_IO_Data->wsaOverlapped));
 lpper_IO_Data->wsaBuffer.len=MSGSIZE;
 lpper_IO_Data->wsaBuffer.buf=lpper_IO_Data->dataBuffer;
 lpper_IO_Data->operationType=RECV_POST;
 //异步调用，因此当服务器刚开启的时候第一次激发异步接受Recv会失败，
    if(SOCKET_ERROR==WSARecv(clientSock,&(lpper_IO_Data->wsaBuffer),1,&lpper_IO_Data->recvedDataSize,&lpper_IO_Data->flags,&lpper_IO_Data->wsaOverlapped,NULL))
 {  
 
  if(WSAGetLastError()!=ERROR_IO_PENDING)
  {
   cout<<"WSARecv 函数调用失败"<<WSAGetLastError<<endl;//71A23CCE
   closesocket(clientSock);
   
  }
 }
 return true;
}
DWORD WINAPI HandleThread(LPVOID completionPortID)
{
    HANDLE completionPort=(HANDLE)completionPortID;
 DWORD dwBytesTransferred;
 SOCKET clientSock;
 PIO_OPERATION_DATA lp_IO_Data=NULL;
    //从队列中取出活跃套接字
 GetQueuedCompletionStatus(completionPort,&dwBytesTransferred,(unsigned long*)&clientSock,(LPWSAOVERLAPPED*)&lp_IO_Data,INFINITE);
    
 if(dwBytesTransferred==0xFFFFFFFF)
 {//没有活跃套接字
  return 0;
 }
 if(lp_IO_Data->operationType==RECV_POST)
 {
  if(dwBytesTransferred==0)
  {//客户端关闭
   cout<<"客户端关闭"<<endl;
   closesocket(clientSock);
   HeapFree(GetProcessHeap(),0,lp_IO_Data);
  }
  else
  {//处理客户端请求
     lp_IO_Data->dataBuffer[dwBytesTransferred]='/0';
           //send(clientSock,lp_IO_Data->dataBuffer,dwBytesTransferred,0);
     WSASend(clientSock,&lp_IO_Data->wsaBuffer,1,&lp_IO_Data->recvedDataSize,lp_IO_Data->flags,&lp_IO_Data->wsaOverlapped,NULL);
     memset(lp_IO_Data,0,sizeof(COM_IO_OPERATION_DATA));
     lp_IO_Data->wsaBuffer.buf=lp_IO_Data->dataBuffer;
     lp_IO_Data->wsaBuffer.len=MSGSIZE;
     lp_IO_Data->operationType=RECV_POST;
     WSARecv(clientSock,&lp_IO_Data->wsaBuffer,1,&lp_IO_Data->recvedDataSize,&lp_IO_Data->flags,&lp_IO_Data->wsaOverlapped,NULL);
  }
 }
 return 0;
}