套接字I/O模型之WSAEventSelect

转自 http://blog.csdn.net/wanjingwei/article/details/4306609

今天我又学习了一种新的套接字I/O模型------WSAEventSelect，他与WSAAsyncSelect一样也是一种异步事件通知模型，不同的是WSAAsyncSelect是与窗口句柄关联在一起的，必须要要窗口才行，而WSAEventSelect是与事件对象关联的。这个模型的基本思路是为感兴趣的一组网络事件创建一个事件对象，再调用WSAEventSelect函数将网络事件和事件对象关联起来。当网络事件发生时，winsock使响应的事件对象受信，在事件对象上等待的函数就会立即返回。之后调用WSAEnumNetworkEvents函数便可获得到底发生了什么网络事件（FD_READ/FD_ACCEPT/FD_CLOSE等等）。

用到的函数有：

WSACreateEvent  、WSAEventSelect、WSAWaitForMultipleEvents、WSAEnumNetworkEvents

等，这里只详细介绍下WSAWaitForMultipleEvents函数

关联了事件对象后就可以用WSAWaitForMultipleEvents函数在一个或多个事件对象上等待了，当所等待的事件对象受信或者指定的时间过去了，此函数返回。

WSAWaitForMultipleEvents(

DWORD cEVents;                            //指定下面lpEvents所指的数组中事件对象句柄的个数

const WSAEVENT*  lpEvents；             //指向一个事件对象句柄的数组

BOOL fWaitAll；                     //指定是否等待所有的事件对象都变成受信状态（为TRUE：是；FALSE:否)

DWORD dwTimeout;            //指定要等待的时间，可以为WSA_INFINITE

BOOL fAlertable;                  //设为FALSE

);

函数最多可以支持WSA_MAXIMUM_WAIT_EVENTS个对象,他的大小是64.该函数会等待网络事件的发生,如果过了指定了时间(dwTimeOut)则返回WSA_WAIT_TIMEOUT,如果在规定的时间内有事件发生,则返回该事件对象的索引(注意:在程序中要想得到发生的事件的真正索引需得用返回值减去WSA_WAIT_EVENT_0),调用失败返回WSA_WAIT_FAILED.如果将参数fWaitAll设置成false如果有多个网络事件发生该函数也只返回一个事件对象索引,并且该事件是在事件句柄数组中最前面的一个.解决方法是循环调用该函数处理后面的受信事件. 
该函数的第一个参数是后面事件对象句柄数组的大小,第二个是个事件对象句柄数组,最后一个设置成false即可. 
   一旦事件对象受信那么找到与之对应的套接字,然后调用 int WSAEnumNetWorkEvent(SOCKET s, WSAEVENT hEventHandle, LPWSANETWORKEVENTS *LPWSANETWORKEVENTS)可以查看发生的网络事件,第一个参数和相应的网络事件标识做与运算就可.第二参数是返回的错误信息。

下面给分别给出用WSAEventSelect模型写出的TCP/UDP服务器例子.

TCP例子：

[cpp] view plaincopy

1. // 事件句柄和套节字句柄表
2. WSAEVENT    eventArray[WSA_MAXIMUM_WAIT_EVENTS];
3. SOCKET      sockArray[WSA_MAXIMUM_WAIT_EVENTS];
4. int nEventTotal = 0;
5. USHORT nPort = 6000;    // 此服务器监听的端口号
6. // 创建监听套节字
7. SOCKET sListen = ::socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
8. sockaddr_in sin;
9. sin.sin_family = AF_INET;
10. sin.sin_port = htons(nPort);
11. sin.sin_addr.S_un.S_addr = INADDR_ANY;
12. if(::bind(sListen, (sockaddr*)&sin, sizeof(sin)) == SOCKET_ERROR)
13. {
14. printf(" Failed bind() /n");
15. return -1;
16. }
17. ::listen(sListen, 5);
18. // 创建事件对象，并关联到新的套节字
19. WSAEVENT event = ::WSACreateEvent();
20. ::WSAEventSelect(sListen, event, FD_ACCEPT|FD_CLOSE);
21. // 添加到表中
22. eventArray[nEventTotal] = event;
23. sockArray[nEventTotal] = sListen;
24. nEventTotal++;
25. // 处理网络事件
26. while(TRUE)
27. {
28. // 在所有事件对象上等待
29. int nIndex = ::WSAWaitForMultipleEvents(nEventTotal, eventArray, FALSE, WSA_INFINITE, FALSE);
30. // 对每个事件调用WSAWaitForMultipleEvents函数，以便确定它的状态
31. nIndex = nIndex - WSA_WAIT_EVENT_0;//发生的事件对象的索引，一般是句柄数组中最前面的那一个，然后再用循环依次处理后面的事件对象
32. for(int i=nIndex; i<nEventTotal; i++)
33. {
34. int ret;
35. ret = ::WSAWaitForMultipleEvents(1, &eventArray[i], TRUE, 1000, FALSE);
36. if(ret == WSA_WAIT_FAILED || ret == WSA_WAIT_TIMEOUT)
37. {
38. continue;
39. }
40. else
41. {
42. // 获取到来的通知消息，WSAEnumNetworkEvents函数会自动重置受信事件
43. WSANETWORKEVENTS event;
44. ::WSAEnumNetworkEvents(sockArray[i], eventArray[i], &event);
45. if(event.lNetworkEvents & FD_ACCEPT)                // 处理FD_ACCEPT通知消息
46. {
47. if(event.iErrorCode[FD_ACCEPT_BIT] == 0)
48. {
49. if(nEventTotal > WSA_MAXIMUM_WAIT_EVENTS)
50. {
51. printf(" Too many connections! /n");
52. continue;
53. }
54. SOCKET sNew = ::accept(sockArray[i], NULL, NULL);
55. WSAEVENT event = ::WSACreateEvent();
56. ::WSAEventSelect(sNew, event, FD_READ|FD_CLOSE|FD_WRITE);
57. // 添加到表中
58. eventArray[nEventTotal] = event;
59. sockArray[nEventTotal] = sNew;
60. nEventTotal++;
61. }
62. }
63. else if(event.lNetworkEvents & FD_READ)         // 处理FD_READ通知消息
64. {
65. if(event.iErrorCode[FD_READ_BIT] == 0)
66. {
67. char szText[256];
68. int nRecv = ::recv(sockArray[i], szText, strlen(szText), 0);
69. if(nRecv > 0)
70. {
71. szText[nRecv] = ‘/0‘;
72. printf("接收到数据：%s /n", szText);
73. }
74. }
75. }
76. else if(event.lNetworkEvents & FD_CLOSE)        // 处理FD_CLOSE通知消息
77. {
78. if(event.iErrorCode[FD_CLOSE_BIT] == 0)
79. {
80. ::closesocket(sockArray[i]);
81. for(int j=i; j<nEventTotal-1; j++)
82. {
83. sockArray[j] = sockArray[j+1];
84. sockArray[j] = sockArray[j+1];
85. }
86. nEventTotal--;
87. }
88. }
89. else if(event.lNetworkEvents & FD_WRITE)        // 处理FD_WRITE通知消息
90. {
91. }
92. }
93. }
94. }

TCP例子就是在监听套接字上关联一个事件对象以及FD_ACCEPT|FD_CLOSE网络事件。

下面是UDP的例子：

[cpp] view plaincopy

1. // 事件句柄和套节字句柄表
2. WSAEVENT    eventArray[WSA_MAXIMUM_WAIT_EVENTS];
3. SOCKET      sockArray[WSA_MAXIMUM_WAIT_EVENTS];
4. int nEventTotal = 0;
5. USHORT nPort = 6000;    // 此服务器监听的端口号
6. // 创建监听套节字
7. SOCKET s = ::socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
8. sockaddr_in sin;
9. sin.sin_family = AF_INET;
10. sin.sin_port = htons(nPort);
11. sin.sin_addr.S_un.S_addr = INADDR_ANY;
12. if(::bind(s, (sockaddr*)&sin, sizeof(sin)) == SOCKET_ERROR)
13. {
14. printf(" Failed bind() /n");
15. return -1;
16. }
17. // 创建事件对象，并关联到新的套节字
18. WSAEVENT event = ::WSACreateEvent();
19. ::WSAEventSelect(s, event, FD_READ|FD_CLOSE);
20. // 添加到表中
21. eventArray[nEventTotal] = event;
22. sockArray[nEventTotal] = s;
23. nEventTotal++;
24. // 处理网络事件
25. while(TRUE)
26. {
27. // 在所有事件对象上等待
28. int nIndex = ::WSAWaitForMultipleEvents(nEventTotal, eventArray, FALSE, WSA_INFINITE, FALSE);
29. // 对每个事件调用WSAWaitForMultipleEvents函数，以便确定它的状态
30. nIndex = nIndex - WSA_WAIT_EVENT_0;
31. for(int i=nIndex; i<nEventTotal; i++)
32. {
33. int ret;
34. ret = ::WSAWaitForMultipleEvents(1, &eventArray[i], TRUE, 1000, FALSE);
35. if(ret == WSA_WAIT_FAILED || ret == WSA_WAIT_TIMEOUT)
36. {
37. continue;
38. }
39. else
40. {
41. // 获取到来的通知消息，WSAEnumNetworkEvents函数会自动重置受信事件
42. WSANETWORKEVENTS event;
43. ::WSAEnumNetworkEvents(sockArray[i], eventArray[i], &event);
44. if(event.lNetworkEvents & FD_READ)          // 处理FD_READ通知消息
45. {
46. if(event.iErrorCode[FD_READ_BIT] == 0)
47. {
48. char szText[256];
49. int nRecv = ::recv(sockArray[i], szText, strlen(szText), 0);
50. if(nRecv > 0)
51. {
52. szText[nRecv] = ‘/0‘;
53. printf("接收到数据：%s /n", szText);
54. }
55. }
56. }
57. }
58. }
59. }

UDP例子就是在一个普通套接字上关联一个事件对象以及FD_READ网络事件。