WSAEventSelect 模型

Winsock提供了另一种有用的异步事件通知 I/O模型——WSAEventSelect模型。这个模型与 WSAAsyncSelect模型类似,允许应用程序在一个或者多个套接字上接收基于事件的网络通知。它与 WSAAsyncSelect 模型类似是因为它也接收FD_XXX 类型的网络事件,不过并不是依靠Windows 的消息驱动机制,而是经由事件对象句柄通知。 

//
// WSAEventSelect

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

CInitSock initSock;		// 初始化Winsock库

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函数会自动重置受信事件；
				// 找到与之对应的套接字，WSAEnumNetworkEvents查看发生了什么网络事件；
				WSANETWORKEVENTS event;
				::WSAEnumNetworkEvents(sockArray[i], eventArray[i], &event);

				if (event.lNetworkEvents & FD_ACCEPT)	// 处理FD_ACCEPT通知消息
				{
					if (event.iErrorCode[FD_ACCEPT_BIT] == 0)	// FD_ACCEPT未出错
					{
						if (nEventTotal > WSA_MAXIMUM_WAIT_EVENTS) 
						{
							printf(" Too many connections");
							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, strlen(szText), 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; j++)	// 删除数组元素；
						{
							sockArray[j] = sockArray[j+1];
						}
						nEventTotal--;
					}
				}
				else if (event.lNetworkEvents & FD_WRITE)	//  处理FD_WRITE通知消息  
				{
				}
			}
		}
	}

	return 0;
}