Windows Socket的异步选择函数WSAAsyncSelect()提供了对网络事件基于消息机制的异步存取,通过 该函数可以注册应用程序感兴趣的网络事件,当登记的网络事件 发生时,相应的窗口将收到一个信息,消息中指示了发生的网络事件及其相关信息。
int WSAAsyncSelect(socket s,HWND hWnd,unsigned int wMsg,long lEvent);
s表示需要事件通知的套接字,hWnd为接收消息的窗口句柄,wMsg为要接收的信息,lEvent为掩码,指定应用程序感兴趣的网络事件组合,其部分取值及含义如下:
FD_READ 期望在套接字s收到数据(读准备好)时接到通知
FD_WRITE 期望在套接字s可发送数据(写准备好)时接到通知
FD_ACCEPT 期望在套接字s上有到来的连接请求时接到通知
FD_CONNECT 期望在套接字s上连接建立完成时接到通知
FD_CLOSE 期望在套接字s关闭时接到通知
注销对套接字s上发生的网络事件的消息发送,可以通过将参数lEvent设为0来调用
int iResult=WSAAsyncSelect(s,hWnd,0,0);
具体实例及代码:
【1】在WM_CREATE消息处理函数中,初始化WindowsSocket library,创建监听套接字,绑定,监听,并且调用WSAAsyncSelect函数表示我们关心在监听套接 字上发生的FD_ACCEPT事件;
【2】自定义一个消息WM_SOCKET,一旦在我们所关心的套接字(监听套接字和客户端套接字)上发生了某个事件,系统就会调用WndProc并且message参数被设置为WM_SOCKET;
【3】在WM_SOCKET的消息处理函数中,分别对FD_ACCEPT、FD_READ和FD_CLOSE事件进行处理;
【4】在窗口销毁消息(WM_DESTROY)的处理函数中,我们关闭监听套接字,清除WindowsSocket library
- #include <winsock2.h>
- #include <tchar.h>
- #define PORT 5150
- #define MSGSIZE 1024
- #define WM_SOCKET WM_USER+0
- #pragma comment(lib, "ws2_32.lib")
- LRESULT CALLBACK WndProc(HWND, UINT, WPARAM, LPARAM);
- int WINAPI WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, PSTR szCmdLine, int iCmdShow)
- {
- static TCHAR szAppName[] = _T("AsyncSelect Model");
- HWND hwnd ;
- MSG msg ;
- WNDCLASS wndclass ;
- wndclass.style = CS_HREDRAW | CS_VREDRAW ;
- wndclass.lpfnWndProc = WndProc ;
- wndclass.cbClsExtra = 0 ;
- wndclass.cbWndExtra = 0 ;
- wndclass.hInstance = hInstance ;
- wndclass.hIcon = LoadIcon (NULL, IDI_APPLICATION) ;
- wndclass.hCursor = LoadCursor (NULL, IDC_ARROW) ;
- wndclass.hbrBackground = (HBRUSH) GetStockObject (WHITE_BRUSH) ;
- wndclass.lpszMenuName = NULL ;
- wndclass.lpszClassName = szAppName ;
- if (!RegisterClass(&wndclass))
- {
- MessageBox (NULL, TEXT ("This program requires Windows NT!"), szAppName, MB_ICONERROR) ;
- return 0 ;
- }
- hwnd = CreateWindow (szAppName, // window class name
- TEXT ("AsyncSelect Model"), // window caption
- WS_OVERLAPPEDWINDOW, // window style
- CW_USEDEFAULT, // initial x position
- CW_USEDEFAULT, // initial y position
- CW_USEDEFAULT, // initial x size
- CW_USEDEFAULT, // initial y size
- NULL, // parent window handle
- NULL, // window menu handle
- hInstance, // program instance handle
- NULL) ; // creation parameters
- ShowWindow(hwnd, iCmdShow);
- UpdateWindow(hwnd);
- while (GetMessage(&msg, NULL, 0, 0)) //获取消息
- {
- TranslateMessage(&msg) ; //转化键盘信息
- DispatchMessage(&msg) ; //将信息发送到指定窗体
- }
- return msg.wParam;
- }
- LRESULT CALLBACK WndProc (HWND hwnd, UINT message, WPARAM wParam, LPARAM lParam)
- {
- WSADATA wsd;
- static SOCKET sListen;
- SOCKET sClient;
- SOCKADDR_IN local, client;
- int ret, iAddrSize = sizeof(client);
- char szMessage[MSGSIZE];
- switch (message)
- {
- case WM_CREATE:
- // Initialize Windows Socket library
- WSAStartup(0x0202, &wsd);
- // Create listening socket
- sListen = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
- // Bind
- local.sin_addr.S_un.S_addr = htonl(INADDR_ANY);
- local.sin_family = AF_INET;
- local.sin_port = htons(PORT);
- bind(sListen, (struct sockaddr *)&local, sizeof(local));
- // Listen
- listen(sListen, 3);
- // Associate listening socket with FD_ACCEPT event
- WSAAsyncSelect(sListen, hwnd, WM_SOCKET, FD_ACCEPT);
- return 0;
- case WM_DESTROY:
- closesocket(sListen);
- WSACleanup();
- PostQuitMessage(0);
- return 0;
- case WM_SOCKET:
- if (WSAGETSELECTERROR(lParam))
- {
- closesocket(wParam);
- break;
- }
- switch (WSAGETSELECTEVENT(lParam))
- {
- case FD_ACCEPT:
- // Accept a connection from client
- sClient = accept(wParam, (struct sockaddr *)&client, &iAddrSize);
- // Associate client socket with FD_READ and FD_CLOSE event
- WSAAsyncSelect(sClient, hwnd, WM_SOCKET, FD_READ | FD_CLOSE);
- break;
- case FD_READ:
- ret = recv(wParam, szMessage, MSGSIZE, 0);
- if (ret == 0 || ret == SOCKET_ERROR && WSAGetLastError() == WSAECONNRESET)
- {
- closesocket(wParam);
- }
- else
- {
- szMessage[ret] = '\0';
- send(wParam, szMessage, strlen(szMessage), 0);
- }
- break;
- case FD_CLOSE:
- closesocket(wParam);
- break;
- }
- return 0;
- }
- //把我们不处理的信息交给系统作默认处理
- return DefWindowProc(hwnd, message, wParam, lParam);
- }