WSAEventSelect模型

#include


#include <winsock2.h> // initsock.h 文件
#pragma comment(lib, "WS2_32") // 鏈接到WS2_32.lib
class CInitSock
{
public:
CInitSock(BYTE minorVer = 2, BYTE majorVer = 2)
{ 
// 初始化WS2_32.dll
WSADATA wsaData;
WORD sockVersion = MAKEWORD(minorVer, majorVer);
if(::WSAStartup(sockVersion, &wsaData) != 0)
{ 
exit(0); 
}
}


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


CInitSock initsock;


int main(int argc, char* argv[])
{
// 事件句柄和套接字句柄表
WSAEVENT eventArray[WSA_MAXIMUM_WAIT_EVENTS];
SOCKET sockArray[WSA_MAXIMUM_WAIT_EVENTS];
int nEventTotal = 0;
USHORT nPort = 9000; // 此服務器監聽的端口號

// 創建監聽套接字
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 函數會自動重置受信事件
WSANETWORKEVENTS event;
::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;
}
SOCKET sNew = ::accept(sockArray[i], NULL, NULL);
WSAEVENT event = ::WSACreateEvent();
::WSAEventSelect(sNew, event, FD_READ|FD_CLOSE|FD_WRITE);

// 添加到表中
eventArray[nEventTotal] = event;
sockArray[nEventTotal] = sNew;
nEventTotal++;
}
}
else if(event.lNetworkEvents & FD_READ) // 處理FD_READ 通知消息
{
if(event.iErrorCode[FD_READ_BIT] == 0)
{
char szText[256];

int nRecv = ::recv(sockArray[i], szText, 256, 0);
if(nRecv > 0)
{ 
szText[nRecv] = '\0';
printf("接收到數據：%s \n", szText);
}
}
}
else if(event.lNetworkEvents & FD_CLOSE) // 處理FD_CLOSE 通知消息
{
if(event.iErrorCode[FD_CLOSE_BIT] == 0)
{
::closesocket(sockArray[i]);
for(int j=i; j<nEventTotal-1; j++)
{ 
sockArray[j] = sockArray[j+1];
sockArray[j] = sockArray[j+1];
}
nEventTotal--;
}
}
else if(event.lNetworkEvents & FD_WRITE) // 處理FD_WRITE 通知消息
{
}
}
}
}
return 0;

}

   WSAEventSelect模型的主要的意思是 用事件對象，將套接字和事件對象 進行關聯，一旦事件對象有事情發生便可以做處理 ，此種方法比WSAAsyncSelect的優點是不需要用

窗口做關聯。

以上代碼直接複製粘貼 便可用。