WIN网络编程-WSAEventSelect

//它与WSAAsyncSelect模型类似是因为它也接收FD_XXX类型的网络事件，是

//经由事件对象句柄通知。

//为一组网络事件创建一个事件对象，再调用WSAEventSelect函数将网络事件

//和事件对象关联起来，当网络事件发生是，WINSOCK使相应的事件对象受信，

//在事件对象上等待函数WSAWaitForMultipleEvents就会返回。然后，调

//用WSAEnumNetworkEvents获取网络事件。 

// WSAWaitForMultipleEvents最多支持WSA_MAXIMUM_WAIT_EVENTS=64个对象。

//WSANETWORKEVENTS结构

//lNetworkEvents指定已经发生的网络事件

//iErrorCode[FD_MAX_EVENTS]指定与lNetworkEvents相关的出错代码，未出错为0

//具体使用WSAAsyncSelect模型步骤如下:

//１）创建一个事件句柄表和一个对应的套接字句柄表

//２）每创建一个套接字，就创建一个事件对象，把它们的句柄分别放入上

//面的两个表中，调用WSAAsyncSelect增加它们的关联

//３）调用WSAWaitForMultipleEvents在所有事件对象上等待，返回后，对事

//件句柄表中每个事件调用WSAWaitForMultipleEvents确认哪些套接字有网络

//事件发生。

//4)处理发生的网络事件，继续在事件对象上等待

 

 

 

// initsock.h

#include<winsock2.h>
#pragma comment(lib, "WS2_32")

class CInitSock
{
public:
 CInitSock();
 ~CInitSock();
};

inline CInitSock::CInitSock()
{
 // 初始化WS2_32.dll
 WSADATA wsaData;
 WORD sockVersion = MAKEWORD(2, 2);
 if(::WSAStartup(sockVersion,&wsaData) != 0)
 {
  exit(0);
 }
}

inline CInitSock::~CInitSock()
{
 ::WSACleanup(); 
}

 

//
// WSAEventSelect文件

#include "initsock.h"
#include <stdio.h>
#include <iostream.h>
#include <windows.h>

// 初始化Winsock库
CInitSock theSock;

int main()
{
 // 事件句柄和套节字句柄表
 WSAEVENT eventArray[WSA_MAXIMUM_WAIT_EVENTS];
 SOCKET  sockArray[WSA_MAXIMUM_WAIT_EVENTS];
 int nEventTotal = 0;

 USHORT nPort =4567; // 此服务器监听的端口号

 // 创建监听套节字
 SOCKET sListen = ::socket(AF_INET, SOCK_STREAM,IPPROTO_TCP); 
 sockaddr_in sin;
 sin.sin_family = AF_INET;
 sin.sin_port = htons(nPort);
 sin.sin_addr.S_un.S_addr = INADDR_ANY;
 if(::bind(sListen,(sockaddr*)&sin, sizeof(sin)) ==SOCKET_ERROR)
 {
  printf(" Failed bind()/n");
  return -1;
 }
 ::listen(sListen, 5);

 // 创建事件对象，并关联到新的套节字
 WSAEVENT event = ::WSACreateEvent();
 ::WSAEventSelect(sListen, event,FD_ACCEPT|FD_CLOSE);
 // 添加到表中
 eventArray[nEventTotal] = event;
 sockArray[nEventTotal] =sListen; 
 nEventTotal++;

 // 处理网络事件
 while(TRUE)
 {
  // 在所有事件对象上等待
  int nIndex =::WSAWaitForMultipleEvents(nEventTotal, eventArray, FALSE,WSA_INFINITE, FALSE);
  //对每个事件调用WSAWaitForMultipleEvents函数，以便确定它的状态
  nIndex = nIndex -WSA_WAIT_EVENT_0;
  for(int i=nIndex;i<nEventTotal; i++)
  {
   nIndex =::WSAWaitForMultipleEvents(1, &eventArray[i], TRUE,1000, FALSE);
   if(nIndex ==WSA_WAIT_FAILED || nIndex == WSA_WAIT_TIMEOUT)
   {
    continue;
   }
   else
   {
    //获取到来的通知消息，WSAEnumNetworkEvents函数会自动重置受信事件
    WSANETWORKEVENTSevent;
    ::WSAEnumNetworkEvents(sockArray[i],eventArray[i], &event);
    if(event.lNetworkEvents&FD_ACCEPT)    //处理FD_ACCEPT通知消息
    {
     if(event.iErrorCode[FD_ACCEPT_BIT]== 0)
     {
      if(nEventTotal> WSA_MAXIMUM_WAIT_EVENTS)
      {
       printf("Too many connections! /n");
       continue;
      }
      SOCKETsNew = ::accept(sockArray[i], NULL, NULL);
      WSAEVENTevent = ::WSACreateEvent();
      ::WSAEventSelect(sNew,event, FD_READ|FD_CLOSE|FD_WRITE);
      //添加到表中
      eventArray[nEventTotal]= event;
      sockArray[nEventTotal]= sNew; 
      nEventTotal++;
     }
    }
    elseif(event.lNetworkEvents &FD_READ)   //处理FD_READ通知消息
    {
     if(event.iErrorCode[FD_READ_BIT]== 0)
     {
      charszText[256];
      intnRecv = ::recv(sockArray[i], szText, strlen(szText), 0);
      if(nRecv>0)    
      {
       szText[nRecv]= '/0';
       printf("接收到数据：%s/n", szText);
      }
     }
    }
    elseif(event.lNetworkEvents &FD_CLOSE)  //处理FD_CLOSE通知消息
    {
     if(event.iErrorCode[FD_CLOSE_BIT]== 0)
     {
      ::closesocket(sockArray[i]);
      for(intj=i; j<nEventTotal-1; j++)
      {
       sockArray[j]= sockArray[j+1];
       sockArray[j]= sockArray[j+1]; 
      }
      nEventTotal--;
     }
    }
    elseif(event.lNetworkEvents &FD_WRITE)  //处理FD_WRITE通知消息
    {
    }
   }
  }
 }
 return 0;
}