windows socket网络编程–事件选择模型

事件选择模型概述

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

API详解

WSAEVENT WSAAPI WSACreateEvent();

返回值
如果未发生错误, WSACreateEvent 将返回事件对象的句柄。 否则,返回值WSA_INVALID_EVENT。
作用
创建新的事件对象

int WSAAPI WSAEventSelect(
  SOCKET   s,            //标识套接字的描述符。
  WSAEVENT hEventObject, //标识要与指定FD_XXX网络事件集关联的事件对象的句柄。
  long     lNetworkEvents//一个位掩码,指定应用程序感兴趣的FD_XXX网络事件的组合。
);

返回值
如果应用程序的网络事件的规范和关联的事件对象成功,则返回值为零。 否则,返回值SOCKET_ERROR。
作用
给事件绑上socket与操作码,并投递给操作系统,应用程序便可以在事件上等待了。

事件类型 含义
FD_READ 应用程序想接收是否有可读的通知
FD_WRITE 应用程序想接收是否有可写的通知
FD_OOB 应用程序想接收是否有OOB数据抵达通知
FD_ACCEPT 应用程序想接收与传入连接有关的通知
FD_CONNECT 应用程序想接收一个已完成连接的通知或者一个多点join操作的通知
FD_CLOSE 应用程序想接收与套接字关闭有关的通知
DWORD WSAAPI WSAWaitForMultipleEvents(
  DWORD          cEvents,
  const WSAEVENT *lphEvents,
  BOOL           fWaitAll,
  DWORD          dwTimeout,
  BOOL           fAlertable
);

cEvents
lphEvents 指向的数组中的事件对象句柄数。 事件对象句柄的最大数目 是WSA_MAXIMUM_WAIT_EVENTS。 必须指定一个或多个事件。

lphEvents
指向事件对象句柄数组的指针。 数组可以包含不同类型的对象的句柄。 如果 fWaitAll 参数设置为 TRUE,则它可能不包含同一句柄的多个副本。 如果在等待仍在挂起时关闭其中一个句柄,则未定义 WSAWaitForMultipleEvents 的行为。

fWaitAll
一个指定等待类型的值。 如果为 TRUE,则当 发出 lphEvents 数组中所有对象的状态时,函数将返回。 如果为 FALSE,则函数在发出任何事件对象的信号时返回。 在后一种情况下,返回值减 去WSA_WAIT_EVENT_0 指示导致函数返回其状态的事件对象的索引。 如果在调用期间发出了多个事件对象的信号,则这是信号事件对象的数组索引,其索引值为所有信号事件对象的最小索引值。

dwTimeout
超时间隔(以毫秒为单位)。 WSAWaitForMultipleEvents 如果超时间隔过期,即使 不满足 fWaitAll 参数指定的条件,也会返回。 如果 dwTimeout 参数为零, WSAWaitForMultipleEvents 将测试指定事件对象的状态并立即返回。 如果 dwTimeoutWSA_INFINITE, WSAWaitForMultipleEvents 将永远等待;也就是说,超时间隔永远不会过期。

fAlertable
一个值,该值指定线程是否处于可警报的等待状态,以便系统可以执行I/O完成例程。 如果为TRUE,则线程处于可警报的等待状态,当系统执行 I/O 完成例程时, WSAWaitForMultipleEvents 可以返回。 在这种情况下,将返回 WSA_WAIT_IO_COMPLETION ,并且等待的事件尚未发出信号。 应用程序必须再次调用 WSAWaitForMultipleEvents 函数。 如果为FALSE,则线程不会处于可警报的等待状态,并且不会执行 I/O 完成例程。

int WSAAPI WSAEnumNetworkEvents(
  SOCKET             s,              //标识套接字
  WSAEVENT           hEventObject,   //用于标识要重置的关联事件对象的可选句柄
  LPWSANETWORKEVENTS lpNetworkEvents //指向 WSANETWORKEVENTS 结构的指针,该结构填充了发生的网络事件记录和任何关联的错误代码
);

返回值
如果操作成功,则返回值为零。 否则,返回值SOCKET_ERROR
作用
枚举出与事件对象相关联的套接字发生了哪些信号,结果放在WSANETWORKEVENTS结构体中

工作原理

流程大致是这样:

  1. 定义一个socket数组和event数组
  2. 每一个socket操作关联一个event对象
  3. 调用WSAWaitForMultipleEvents函数等待事件的触发
  4. 调用WSAEnumNetworkEvents函数查看是哪个一个事件,根据事件找到相应的socket,然后进行相应的处理:比如数据显示等,同时,记得要将那个event重置为无信号状态。
  5. 循环步骤3和4,直到服务器退出。
    流程图

代码实现

服务端

UINT  CMFCWSAEventDlg::ThreadProc(LPVOID lparam) {
	// TODO: 在此添加控件通知处理程序代码
	CMFCWSAEventDlg* p = (CMFCWSAEventDlg*)lparam;
	SocketInit socketInit;
	SOCKET socketServer = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
	if (socketServer == INVALID_SOCKET) {
		AfxMessageBox(_T("套接字创建失败"));
		closesocket(socketServer);
		WSACleanup();
	}
	sockaddr_in sock;
	sock.sin_family = AF_INET;
	sock.sin_port = htons(5678);
	sock.sin_addr.S_un.S_addr = INADDR_ANY;
	int n = sizeof(sock);
	if (bind(socketServer, (sockaddr*)&sock, sizeof(sock)) == SOCKET_ERROR) {
		AfxMessageBox(_T("监听失败"));
		closesocket(socketServer);
		WSACleanup();

	}
	if (listen(socketServer, SOMAXCONN) == SOCKET_ERROR) {
		AfxMessageBox(_T("监听失败"));
		closesocket(socketServer);
		WSACleanup();
	}
	p->showText.SetWindowText("开始监听\r\n");
	
	// 创建事件对象,并关联到新的套节字
	WSAEVENT event = ::WSACreateEvent();
	::WSAEventSelect(socketServer, event, FD_ACCEPT | FD_CLOSE);
	// 添加到表中
	p->eventArray[p->nEventTotal] = event;
	p->sockArray[p->nEventTotal] = socketServer;
	p->nEventTotal++;
	CString str;
	sockaddr_in addrRemote;
	
	while (1){

		// 在所有事件对象上等待
		int nIndex = ::WSAWaitForMultipleEvents(p->nEventTotal, p->eventArray, FALSE, WSA_INFINITE, FALSE);
		// 对每个事件调用WSAWaitForMultipleEvents函数,以便确定它的状态
		nIndex = nIndex - WSA_WAIT_EVENT_0;

		for (int i = nIndex; i < p->nEventTotal; i++)
		{
			nIndex = ::WSAWaitForMultipleEvents(1, &p->eventArray[i], TRUE, 1000, FALSE);
			if (nIndex == WSA_WAIT_FAILED || nIndex == WSA_WAIT_TIMEOUT)
			{
				continue;
			}
			else
			{
				// 获取到来的通知消息,WSAEnumNetworkEvents函数会自动重置受信事件
				WSANETWORKEVENTS event;
				
				::WSAEnumNetworkEvents(p->sockArray[i], p->eventArray[i], &event);
				if (event.lNetworkEvents & FD_ACCEPT)                // 处理FD_ACCEPT通知消息
				{
					if (event.iErrorCode[FD_ACCEPT_BIT] == 0)
					{
						if (p->nEventTotal > WSA_MAXIMUM_WAIT_EVENTS)
						{
							p->showText.SetSel(-1);
							p->showText.ReplaceSel("时间太长");
							continue;
						}
						int nAddrLen = sizeof(addrRemote);
						SOCKET sNew = ::accept(p->sockArray[i], (SOCKADDR*)&addrRemote, &nAddrLen);
						//MessageBox("已连接");
						int nLen = p->showText.GetWindowTextLengthA();
						//p->showText.SetWindowText()
						
						str.Format("%s建立连接\r\n", ::inet_ntoa(addrRemote.sin_addr));
						p->showText.SetSel(-1);
						p->showText.ReplaceSel(str);
						WSAEVENT event = ::WSACreateEvent();
						::WSAEventSelect(sNew, event, FD_READ | FD_CLOSE | FD_WRITE);
						// 添加到表中
						p->eventArray[p->nEventTotal] = event;
						p->sockArray[p->nEventTotal] = sNew;
						p->nEventTotal++;
					}
				}
				else if (event.lNetworkEvents & FD_READ)         // 处理FD_READ通知消息
				{
					if (event.iErrorCode[FD_READ_BIT] == 0)
					{
						//char szText[256];
						char szText[1024] = { 0 };
						//memset(szText, 0, sizeof(szText));
						int nlen = strlen(szText);
						int nRecv = ::recv(p->sockArray[i], szText,1024, 0);
						
						//AfxMessageBox(nRecv);
						if (nRecv > 0)
						{
							szText[nRecv] = '\0';
							str.Format("%s发来了一条消息:%s\r\n", ::inet_ntoa(addrRemote.sin_addr), szText);
							p->showText.SetSel(-1);
							p->showText.ReplaceSel(str);
							//szText[0] = '\0';

							// 向客户端发送数据
							char *sendText =  getallprime(1000);
							if (::send(p->sockArray[i], sendText, strlen(sendText), 0) > 0)
							{
								p->showText.SetSel(-1);
								p->showText.ReplaceSel("已发送结果\r\n");
							}
						}
					}
				}
				else if (event.lNetworkEvents & FD_CLOSE)        // 处理FD_CLOSE通知消息
				{
					if (event.iErrorCode[FD_CLOSE_BIT] == 0)
					{
						::closesocket(p->sockArray[i]);
						for (int j = i; j < p->nEventTotal - 1; j++)
						{
							p->eventArray[j] = p->eventArray[j + 1];
							p->sockArray[j] = p->sockArray[j + 1];
						}
						p->nEventTotal--;
					}
					p->showText.SetSel(-1);
					p->showText.ReplaceSel("关闭连接\r\n");
				}
			}
		}
	}

}
Tags: