WSAAsyncSelect模式

Winsock 提供了一个有用的异步I/O模型。利用这个模型，应用程序可在一个套接字上，接收以Windows消息为基础的网络事件通知。具体的做法是在建好一个套接字后，调用WSAAsyncSelect函数。该模型最早出现于Winsock的1.1版本中，用于帮助应用程序开发者面向一些早期的16位 Windows平台（如Windows for Workgroups），适应其“落后”的多任务消息环境。应用程序仍可从这种模型中得到好处，特别是它们用一个标准的Windows例程（常称为 "WndProc"），对窗口消息进行管理的时候。该模型亦得到了Microsoft Foundation Class（微软基本类，MFC）对象CSocket的采纳。

 

[cpp]  view plain copy print ?

1. #include <winsock2.h>  
2. #include <tchar.h>  
3. #define PORT      5150  
4. #define MSGSIZE   1024  
5. #define WM_SOCKET WM_USER+0  
6. #pragma comment(lib, "ws2_32.lib")  
7. LRESULT CALLBACK WndProc(HWND, UINT, WPARAM, LPARAM);  
8. int WINAPI WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, PSTR szCmdLine, int iCmdShow)  
9. {  
10. static TCHAR szAppName[] = _T("AsyncSelect Model");  
11. HWND         hwnd ;  
12. MSG          msg ;  
13. WNDCLASS     wndclass ;  
14. wndclass.style         = CS_HREDRAW | CS_VREDRAW ;  
15. wndclass.lpfnWndProc   = WndProc ;  
16. wndclass.cbClsExtra    = 0 ;  
17. wndclass.cbWndExtra    = 0 ;  
18. wndclass.hInstance     = hInstance ;  
19. wndclass.hIcon         = LoadIcon (NULL, IDI_APPLICATION) ;  
20. wndclass.hCursor       = LoadCursor (NULL, IDC_ARROW) ;  
21. wndclass.hbrBackground = (HBRUSH) GetStockObject (WHITE_BRUSH) ;  
22. wndclass.lpszMenuName = NULL ;  
23. wndclass.lpszClassName = szAppName ;  
24. if (!RegisterClass(&wndclass))  
25. {  
26.     MessageBox (NULL, TEXT ("This program requires Windows NT!"), szAppName, MB_ICONERROR) ;  
27.     return 0 ;  
28. }  
29. hwnd = CreateWindow (szAppName,                  // window class name  
30.                        TEXT ("AsyncSelect Model"), // window caption  
31.                        WS_OVERLAPPEDWINDOW,        // window style  
32.                        CW_USEDEFAULT,              // initial x position  
33.                        CW_USEDEFAULT,              // initial y position  
34.                        CW_USEDEFAULT,              // initial x size  
35.                        CW_USEDEFAULT,              // initial y size  
36.                        NULL,                       // parent window handle  
37.                        NULL,                       // window menu handle  
38.                        hInstance,                  // program instance handle  
39.                        NULL) ;                     // creation parameters  
40. ShowWindow(hwnd, iCmdShow);  
41. UpdateWindow(hwnd);  
42. while (GetMessage(&msg, NULL, 0, 0)) //获取消息  
43. {  
44.     TranslateMessage(&msg) ;  //转化键盘信息  
45.     DispatchMessage(&msg) ; //将信息发送到指定窗体  
46. }  
47.   
48. return msg.wParam;  
49. }  
50. LRESULT CALLBACK WndProc (HWND hwnd, UINT message, WPARAM wParam, LPARAM lParam)  
51. {  
52. WSADATA       wsd;  
53. static SOCKET sListen;  
54. SOCKET        sClient;  
55. SOCKADDR_IN   local, client;  
56. int           ret, iAddrSize = sizeof(client);  
57. char          szMessage[MSGSIZE];  
58. switch (message)  
59. {  
60. case WM_CREATE:  
61.     // Initialize Windows Socket library  
62.    WSAStartup(0x0202, &wsd);  
63.     
64.    // Create listening socket  
65.     sListen = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);  
66.       
67.    // Bind  
68.     local.sin_addr.S_un.S_addr = htonl(INADDR_ANY);  
69.    local.sin_family = AF_INET;  
70.    local.sin_port = htons(PORT);  
71.    bind(sListen, (struct sockaddr *)&local, sizeof(local));  
72.     
73.    // Listen  
74.     listen(sListen, 3);  
75.     // Associate listening socket with FD_ACCEPT event  
76.    WSAAsyncSelect(sListen, hwnd, WM_SOCKET, FD_ACCEPT);  
77.    return 0;  
78. case WM_DESTROY:  
79.     closesocket(sListen);  
80.     WSACleanup();  
81.     PostQuitMessage(0);  
82.     return 0;  
83.   
84. case WM_SOCKET:  
85.     if (WSAGETSELECTERROR(lParam))  
86.     {  
87.       closesocket(wParam);  
88.       break;  
89.     }  
90.       
91.     switch (WSAGETSELECTEVENT(lParam))  
92.     {  
93.     case FD_ACCEPT:  
94.       // Accept a connection from client  
95.       sClient = accept(wParam, (struct sockaddr *)&client, &iAddrSize);  
96.         
97.       // Associate client socket with FD_READ and FD_CLOSE event  
98.       WSAAsyncSelect(sClient, hwnd, WM_SOCKET, FD_READ | FD_CLOSE);  
99.       break;  
100.     case FD_READ:  
101.       ret = recv(wParam, szMessage, MSGSIZE, 0);  
102.       if (ret == 0 || ret == SOCKET_ERROR && WSAGetLastError() == WSAECONNRESET)  
103.       {  
104.         closesocket(wParam);  
105.       }  
106.       else  
107.       {  
108.         szMessage[ret] = '\0';  
109.         send(wParam, szMessage, strlen(szMessage), 0);  
110.       }  
111.       break;  
112.         
113.     case FD_CLOSE:  
114.       closesocket(wParam);        
115.       break;  
116.     }  
117.     return 0;  
118. }  
119. //把我们不处理的信息交给系统作默认处理  
120. return DefWindowProc(hwnd, message, wParam, lParam);  
121. }  

 

在我看来，WSAAsyncSelect是最简单的一种Winsock I/O模型（之所以说它简单是因为一个主线程就搞定了）。使用Raw Windows API写过窗口类应用程序的人应该都能看得懂。这里，我们需要做的仅仅是：
1.在WM_CREATE消息处理函数中，初始化Windows Socket library，创建监听套接字，绑定，监听，并且调用WSAAsyncSelect函数表示我们关心在监听套接字上发生的FD_ACCEPT事件；
2.自定义一个消息WM_SOCKET，一旦在我们所关心的套接字（监听套接字和客户端套接字）上发生了某个事件，系统就会调用WndProc并且message参数被设置为WM_SOCKET；
3.在WM_SOCKET的消息处理函数中，分别对FD_ACCEPT、FD_READ和FD_CLOSE事件进行处理；
4.在窗口销毁消息(WM_DESTROY)的处理函数中，我们关闭监听套接字，清除Windows Socket library

下面这张用于WSAAsyncSelect函数的网络事件类型表可以让你对各个网络事件有更清楚的认识：
表1

FD_READ	应用程序想要接收有关是否可读的通知，以便读入数据
FD_WRITE	应用程序想要接收有关是否可写的通知，以便写入数据
FD_OOB	应用程序想接收是否有带外（OOB）数据抵达的通知
FD_ACCEPT	应用程序想接收与进入连接有关的通知
FD_CONNECT	应用程序想接收与一次连接或者多点join操作完成的通知
FD_CLOSE	应用程序想接收与套接字关闭有关的通知
FD_QOS	应用程序想接收套接字“服务质量”（QoS）发生更改的通知
FD_GROUP_QOS	应用程序想接收套接字组“服务质量”发生更改的通知（现在没什么用处，为未来套接字组的使用保留）
FD_ROUTING_INTERFACE_CHANGE	应用程序想接收在指定的方向上，与路由接口发生变化的通知
FD_ADDRESS_LIST_CHANGE	应用程序想接收针对套接字的协议家族，本地地址列表发生变化的通知