Windows socket 之WSAAsyncSelect模型

转载自：http://blog.csdn.net/ithzhang/article/details/8464330

        WSAAsyncSelect模型

     WSAAsyncSelect模型是Windows socket的一个异步IO模型。利用该模型可以接收以Windows消息为基础的网络事件。Windows sockets应用程序在创建套接字后，调用WSAAsyncSelect函数注册感兴趣的网络事件，当该事件发生时Windows窗口收到消息，应用程序就可以对接收到的网络时间进行处理。

 

     WSAAsyncSelect是select模型的异步版本。在应用程序使用select函数时会发生阻塞现象。可以通过select的timeout参数设置阻塞的时间。在设置的时间内，select函数等待，直到一个或多个套接字满足可读或可写的条件。

 

     而WSAAsyncSelect是非阻塞的。Windows sockets程序在调用recv或send之前，调用WSAAsyncSelect注册网络事件。WSAAsyncSelect函数立即返回。当系统中数据准备好时，会向应用程序发送消息。此此消息的处理函数中可以调用recv或send进行接收或发送数据。

 

     WSAAsyncSelect模型与select模型的相同点是它们都可以对多个套接字进行管理。但它们也有不小的区别。首先WSAAsyncSelect模型是异步的，且通知方式不同。更重要的一点是：WSAAsyncSelect模型应用在基于消息的Windows环境下，使用该模型时必须创建窗口，而select模型可以广泛应用在Unix系统，使用该模型不需要创建窗口。最后一点区别：应用程序在调用WSAAsyncSelect函数后，套接字就被设置为非阻塞状态。而使用select函数不改变套接字的工作方式。

 

      WSAAsyncSelect函数。

     该函数告诉系统当网络事件发生时为套接字发送消息。声明如下：

 

 

[html]  view plain copy

1. int WSAAsyncSelect(  
2.   
3.     SOCKET s,  
4.   
5.     HWND hWnd,  
6.   
7.     u_int wMsg,  
8.   
9.     long lEvent);  

 

     s为需要通知的套接字。

     hWnd为当网络事件发生时接收消息的窗口句柄。

     wMsg为当网络事件发生时窗口收到的消息。在此消息的响应函数内对网络事件进行处理。

     lEvent为应用程序感兴趣的网络事件集合。

     应用程序调用该函数后自动将套接字设置为非阻塞模式。通常用户自定义消息应该在WM_USER的基础之上定义。如WM_USER+1，以避免与Windows预定义的消息发生混淆。

     网络事件可以有以下几种：

 

     FD_READ：套接字可读通知。

     FD_WRITE：可写通知。

     FD_ACCEPT：服务器接收连接的通知。

     FD_CONNECT：有客户连接通知。

     FD_OOB：外带数据到达通知。

     FD_CLOSE：套接字关闭通知。

     FD_QOS：服务质量发生变化通知。

     FD_GROUP_QOS：组服务质量发生变化通知。

     FD_ROUTING_INTERFACE_CHANGE：与路由器接口发生变化的通知。

     FD_ADDRESS_LIST_CHANGE：本地地址列表发生变化的通知。

 

     开发人员应向应用程序注册感兴趣的网络事件。可以将它们按位或并传给lEvent函数。如：

 

 

[cpp]  view plain copy

1. WSAAsyncSelect(s,hWnd,WM_SOCKET,FD_CONNECT|FD_READ|FD_CLOSE);  

 

     上述代码表示：当套接字连接到来、有数据可读或这套接字关闭的网络事件发生时，WM_SOCKET消息就会发送给hWnd为句柄的窗口。

     消息处理函数。

     消息处理函数是对网络事件发生时窗口消息的处理。它的声明如下：

 

[cpp]  view plain copy

1. LRESULT CALLBACK WindowProc(  
2.   
3.        HWND hWnd,  
4.   
5.        UINT uMsg,  
6.   
7.        WPARAM wParam,  
8.   
9.        LPARAM lParam)  

 

     hWnd为窗口句柄。

     uMsg为当网络事件发生时的消息。

     wParam为消息参数。该参数表明发生网络事件的套接字。

     lParam也为消息参数。低字节表明已发生的网络事件。高字节包含错误代码。

 

     在Windows sockets应用程序中，当WindowProc接收到网络消息时，在该函数内执行下面的步骤：

     1：读取lParam的高字节，判断是否有错误发生。可以使用WSAGETSElECTERROR宏。

     2：如果没有错误，读取lParam的低字节，检查发生了什么网络事件，可以使用WSAGETSELECTEVENT宏。

 

     WSAGETSElECTERROR和WSAGETSELECTEVENT宏定义如下：

 

[cpp]  view plain copy

1. #define WSAGETSElECTERROR(lParam)  LOWORD(lParam)  
2.   
3. #define WSAGETSELECTEVENT(lParam)  HIWORD(lParam)  

 

     接下来就需要创建窗口和将网络消息与消息处理函数关联起来。如果使用MFC可以使用MFC提供的宏来进行处理。

     注意：多次调用WSAAsyncSelect时，最后一次调用会取消前面注册的网络事件。

 

     因为调用accept接受的套接字和监听套接字具有同样的属性。所以，任何为监听套接字设置的网络事件对接受套接字同样起作用。如果一个监听套接字请求FD_ACCEPT、FD_READ和FD_WRITE网络事件。则在该监听套接字上接受的任何套接字也会请求FD_ACCEPT，FD_READ和FD_WRITE网络事件。

 

     FD_CLOSE网络事件用来判断套接字是否已经关闭。错误代码会指出套接字是从容关闭还是硬关闭。如果为0，为从容关闭。若错误代码为WSAECONNRESET，则套接字是硬关闭。调用closesocket不会投递FD_CLOSE事件。

       发生网络事件的条件。

 

     下列条件下会发生FD_READ事件：

       1：当调用WSAAsyncSelect函数时，如果当前有数据可读。

       2：当数据到达并且没有发送FD_READ网络事件时。

      3：调用recv()或这recvfrom,如果仍有数据可读里。

 

     下列情况下会发生FD_WRITE事件：

       1：调用WSAAsyncSelect函数时，如果能够发送数据时。

       2：connect或者accept函数后，连接已经建立时。

       3：调用send或者sendto函数，返回WSAWOULDBLOCK错误后，再次调用send()或者sendto函数可能成功时。因为此时可能是套接字还处于不可写状态，多次调用直到调用成功为止。

 

     WSAAsyncSelect的优势与不足。

 

     该模型是在基于消息的Windows环境下开发应用程序。开发人员可以像处理其他消息一样，对网络事件进行处理。而且为确保接受所有数据提供了很好的机制。

    不足：由于该模型基于Windows消息机制，必须在应用程序中创建窗口。虽然可以在开发中，确定是否显示该窗口。 由于调用WSAAsyncSelect函数后自动将套接字设置为非阻塞状态，当应用程序接收到网络事件时，未必能够成功返回。这无疑增加了使用该模型的难度。

     接下来展示一个使用如何WSAAsyncSelect模型的例子。该程序使用WSAAsyncSelect模型管理接受的客户端套接字。编码步骤如下：

     1：声明自定义消息。如WM_SOCKET

     2：声明窗口例程。

     3：将自定义消息与消息处理函数相关联。

     4：初始化套接字动态库，创建套接字。

     5：调用WSAAsyncSelect注册感兴趣的网络事件。本例服务器感兴趣的网络事件有FD_ACCEPT和FD_CLOSE。

     6：绑定套接字开始监听。

     一：声明自定义消息：

 

[cpp]  view plain copy

1. #define WM_SOCKET WM_USER+1 //套接字消息。  

     除了声明自定义消息外还需要声明最大字符串长度、服务器监听端口、数据缓冲区。

 

[cpp]  view plain copy

1. #define MAX_STRING 100     //最大字符串长度。  
2.   
3. #define SERVERPORT 5000    //服务器端口。  
4.   
5. #define MAX_SIZE_BUF 1024  //数据缓冲区长度。  

     二：声明消息处理函数并与消息关联：

 

     1：在窗口类头文件中声明消息处理函数。如：

 

[cpp]  view plain copy

1. afx_msg LRESULT onWmSocket(WPARAM wParam, LPARAM lParam);  

 

     2：在消息映射宏中将自定义消息如声明的消息处理函数关联：

 

[cpp]  view plain copy

1. ON_MESSAGE(WM_SOCKET,&onWmSocket)  

 

     3：实现消息处理函数：

[cpp]  view plain copy

1. LRESULT CuserdefinedMessageTestDlg::onWmSocket( WPARAM wParam, LPARAM lParam )  
2. {  
3.     if(WSAGETSELECTERROR(lParam))  
4.     {  
5.         m_list.deleteNode(wParam);//wParam为发生消息的套接字。出现错误，则从链表中将该套接字对应的CClient类对象删除。  
6.         return false;  
7.     }  
8.     else  
9.     {  
10.         switch(WSAGETSELECTEVENT(lParam))  
11.         {  
12.         case FD_ACCEPT://接受客户端连接请求。  
13.             {  
14.                 SOCKET sAccept;  
15.                 if((sAccept==accept(wParam,NULL,NULL)==INVALID_SOCKET))  
16.                     break;  
17.                 m_list.add(sAccept);  
18.                 //在新接受的套接字发生FD_READ,FD_WRITE,FD_CLOSE网络事件发生，发送WM_SOCKET消息；  
19.                 WSAAsyncSelect(sAccept,this->m_hWnd,WM_SOCKET,FD_READ|FD_WRITE|FD_CLOSE);  
20.             }  
21.             break;  
22.         case FD_READ://可读，接收数据。  
23.             {  
24.                 CClient *pClinet=GetClient(wParam);//根据套接字，获取客户端节点。  
25.                 pClient->RecvData();  
26.             }  
27.             break;  
28.         case FD_WRITE://可写，发送数据。  
29.             {  
30.                 CClient*pClient=GetClient(wParam);  
31.                 pClient->SendData();  
32.                   
33.             }  
34.             break;  
35.         case FD_CLOSE://对方关闭套接字连接。  
36.             {  
37.                 if(WSAGETSELECTERROR(lParam)==0)  
38.                 {  
39.                     //从容关闭。  
40.                 }  
41.                 else if(WSAGETSELECTERROR(lParam)==WSAECONNREFUSED)  
42.                 {  
43.                     //硬关闭。  
44.                 }  
45.                 m_list.deleteNode(wParam);  
46.             }  
47.             break;  
48.         default:  
49.             break;  
50.         }  
51.     }  
52.     return 0;  
53. }  

 

     三：创建套接字并注册感兴趣的网络事件.

     1:初始化套接字动态库，并创建套接字。

 

[cpp]  view plain copy

1. WSADATA wsa;  
2.   
3. int ret=WSAStartup(MAKEWORD(2,2),&wsa);  
4.   
5. ListenSocket=socket(AF_INET,SOCK_STREAM,0);  
6.   
7. if(ListenSocket==INVALID_SOCKET)  
8.   
9. {  
10.   
11.    return false;  
12.   
13. }  

     2：注册感兴趣的网络事件：

 

[cpp]  view plain copy

1. WSAAsyncSelect(ListenSocket,this,WM_SOCKET,FD_ACCEPT|FD_CLOSE);  

 

     3:绑定套接字并监听。

 

[cpp]  view plain copy

1. SOCKADDR_IN addr;  
2.     addr.sin_family=AF_INET;  
3.     addr.sin_addr.S_un.S_addr=inet_addr("192.168.1.100");  
4.     addr.sin_port=htons(4000);  
5.     int ret=bind(m_ListenSocket,(SOCKADDR*)&addr,sizeof(addr));  
6.     if(ret==INVALID_SOCKET)  
7.     {  
8.         return false;  
9.     }  
10.     ret=listen(m_ListenSocket,10);  
11.     if(ret==INVALID_SOCKET)  
12.     {  
13.         return false;  
14.     }  

 

    四：退出

 

[cpp]  view plain copy

1. closesocket(m_ListenSocket);  
2.   
3. WSACleanup();  

 

    五：CClient类。

 

     自定义类CClient类用于管理服务器接受客户端的新建套接字。在该类中实现与客户端通信。

 

     六；管理客户端套接字链表。

        当服务器接受一个客户端连接请求后就会创建一个CClient实例。将该实例地址加入链表中。

     以上参考自《精通Windows sockets网络开发--基于Visual C++实现》孙海民著。如有纰漏，请不吝指正！

                                      2012.1.4 于山西大同