Windows Socket五种I/O模型（二）——异步选择

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

 

1. #include <winsock.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  
应用程序想接收针对套接字的协议家族，本地地址列表发生变化的通知