介绍
WSAAsyncSelect模型又称为异步选择模型,是为了适应Windows的消息驱动环境而设置的。引入Berkeley套接字后,该模型在Windows Sockets1.1版本中增加,以Windows系统中最常用的消息机制来反馈网络I/O事件,达到了收发数据的异步通知效果。这种模型在现在许多性能要求不高的网络应用程序中应用广泛,MFC中的CSocket类也使用了它。
WSAAsyncSelect模型的相关函数
WSAAsyncSelect模型的核心是WSAAsyncSelect()函数,它自动将套接字设置为非阻塞模式,使Windows应用程序能够接收网络事件消息。该函数的定义如下:
int WSAAsyncSelect(
__in SOCKET s,
__in HWND hWnd,
__in unsigned int wMsg,
__in long lEvent
);
- s:关心某网络事件的套接字;
- hWnd:网络事件发生时用于接收消息的窗口句柄;
- wMsg:网络事件发生时接收到的消息;
- lEvent:套接字感兴趣的网络事件。
WSAAsyncSelect()函数调用后,当检测到lEvent参数指定的网络事件发生时,Windows Sockets实现会向hWnd指定的窗体发送消息wMsg。如果函数成功,则返回0;否则返回SOCKET_ERROR。
参数lEvent可以包含的网络事件如表:
如果对多个网络事件都关心,则可以将以上的事件执行按位或操作,例如,应用程序希望接收到读就绪和连接关闭的通知事件,则对WSAAsyncSelect()函数的lEvent参数同时使用FD_READ和FD_CLOSE事件,代码示例如下:
iResult = WSAAsyncSelect(s, hWnd, wMsg, FD_READ|FD_CLOSE);
这里要注意对同一个套接字的多次WSAAsyncSelect()调用之间的影响。当我们对同一个套接字执行一次WSAAsyncSelect()时,会取消之前对该套接字的WSAAsyncSelect()或WSAEventSelect()调用所声明的网络事件。例如,下面的代码两次调用WSAAsyncSelect()函数,先后为套接字s的不同事件指定了不同的消息:
iResult = WSAAsyncSelect(s, hWnd, wMsg1, FD_READ);
iResult = WSAAsyncSelect(s, hWnd, wMsg2, FD_WRITE);
其结果是,第二次调用WSAAsyncSelect()函数会取消第一次调用对FD_READ事件的注册,也就是说,只有当FD_WRITE事件发生时,套接字s才会收到wMsg2指定的消息,wMsg1消息不会再发出。
因此,对同一个套接字而言,不能为不同的事件指定不同的消息,如果关注多个事件,需要用按位或的操作来实现。
如果要取消指定套接字上的所有通知事件,则可以在调用WSAAsyncSelect()函数时将事件参数lEvent设置为0,例如:
iResult = WSAAsyncSelect(s, hWnd, 0, 0);
WSAAsyncSelect模型的编程框架
WSAAsyncSelect模型为套接字关心的网络事件绑定用户自定义的消息,当网络事件发生时,相应的消息被发送给关心该事件的套接字所在的窗口,从而使应用程序可以对该事件做相应的处理。
在消息驱动下,原来顺序执行的程序改为由两个相对独立的执行部分构成:
主程序框架。这部分主要完成套接字的创建和初始化的工作,根据程序工作的环境不同,主程序框架中的功能可能会有一些差别。如果是基于SDK的应用程序,为了对消息队列进行创建和维护,需要创建维护消息资源的窗口并对消息进行循环转换和分发;如果是基于MFC的应用程序,MFC已对消息映射机制进行了合理的封装,主程序框架部分不需要手工维护消息队列。
消息处理框架。在网络事件发生后,消息产生,该消息被主程序框架捕获到,消息处理框架部分被执行,主要完成消息类型的判断和网络事件的具体处理。在基于SDK的应用程序中,这部分功能是在窗口过程中完成的,窗口过程只有一个;而在基于MFC的应用程序中,这部分功能是在独立的消息处理函数中完成的,且消息处理函数可能会存在多个。
整体来看,基于WSAAsyncSelect模型的网络应用程序的基本流程如下
- 定义套接字网络事件对应的用户消息。
- 如果不存在窗口,则创建窗口和窗口例程支持函数。
- 调用WSAAsyncSelect()函数为套接字设置网络事件、用户消息和消息接收窗口之间的关系。
- 增加消息循环的具体功能,或者添加消息与消息处理函数的映射关系。
- 添加消息处理框架的具体功能,判断是哪个套接字上发生了网络事件。使用WSAGETSELECTEVENT宏了解所发生的网络事件,从而进行相应的处理。
依赖于不同的消息实现机制,在WSAAsyncSelect模型下应用程序的处理过程有较大的差别,我们分SDK环境和MFC环境两种情况举例说明套接字的创建及接收的基本过程。
以下示例实现了基于WSAAsyncSelect模型的面向连接通信服务器,该服务器的主要功能是接收客户使用TCP协议发来的数据,打印接收到的数据的字节数。
在SDK下实现基于WSAAsyncSelect模型的套接字通信服务器
#include <WinSock2.h>
#include <Windows.h>
#include <WS2tcpip.h>
#include <stdlib.h>
#include <stdio.h>
#include <conio.h>
#include<tchar.h>
#pragma comment(lib,"ws2_32.lib")
#define WM_SOCKET WM_USER+101
#define DEFAULT_BUFLEN 512
#define DEFAULT_PORT 27015
LRESULT CALLBACK WndProc(HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam);
HWND CreateWorkWindow(void) {
WNDCLASS wndclass;
TCHAR ClassName[] = L"SelectWindow";//窗口类名
HWND hWnd = NULL;//新建窗口句柄
/*
wndclass.style = CS_HREDRAW | CS_VREDRAW;//窗口类型
wndclass.lpfnWndProc = (WNDPROC)WndProc;//指定窗口处理程序
wndclass.cbClsExtra = 0;//窗口类结构体后面额外分配的字节
wndclass.cbWndExtra = 0;//窗口实例后面额外分配的字节数
wndclass.hInstance = NULL;//窗口类对应实例的句柄
wndclass.hIcon = LoadIcon(NULL, IDC_ARROW);
//光标句柄
wndclass.hbrBackground = (HBRUSH)GetStockObject(WHITE_BRUSH);
// 刷子句柄(白色)
wndclass.lpszMenuName = NULL; // 窗口中无菜单
wndclass.lpszClassName = (WCHAR*)ClassName; // 窗口类名为"SelectWindow"
*/
wndclass.style = CS_HREDRAW | CS_VREDRAW;
wndclass.lpfnWndProc = WndProc;
wndclass.cbClsExtra = 0;
wndclass.cbWndExtra = 0;
wndclass.hInstance = NULL;
wndclass.hIcon = LoadIcon(NULL, IDI_APPLICATION);
wndclass.hCursor = LoadCursor(NULL, IDC_ARROW);
wndclass.hbrBackground = (HBRUSH)GetStockObject(WHITE_BRUSH);
wndclass.lpszMenuName = NULL;
wndclass.lpszClassName = ClassName;
if (RegisterClass(&wndclass)==0)
{
printf("REGISTERCLASS() failed with error code %d\n",GetLastError());
return NULL;
}
hWnd = CreateWindow(ClassName, L"", WS_OVERLAPPEDWINDOW, CW_USEDEFAULT, CW_USEDEFAULT, CW_USEDEFAULT, CW_USEDEFAULT, NULL, NULL, NULL, NULL);
if (hWnd==NULL)
{
printf("CREATEWINDOWS FAILD WITH ERROR CODE %d\n",GetLastError());
return NULL;
}
return hWnd;
}
输入参数:无。
输出参数:窗口句柄。
● 非NULL:表示成功;
● NULL:表示失败。
程序创建窗口时,可以创建预先定义的窗口类或自定义的窗口类。创建自定义的窗口类时,在使用该窗口类前必须注册该窗口类。使用RegisterClass()函数注册窗口类。
第5~21行代码创建了一个窗口结构WNDCLASS,该结构包含了RegisterClass()函数注册的类属性,并调用RegisterClass()函数对窗口结构进行注册。
第22~35行代码调用CreateWindow()函数创建接收消息的窗口。
(2)第二步:定义窗口例程
窗口例程具体实现消息处理框架的
窗口例程具体实现消息处理框架的功能,取出消息,判断消息类型,进行相应的网络处理。
该函数的具体实现代码如下:
int main() {
HWND hWnd;
MSG Msg;
if ((hWnd=CreateWorkWindow())==NULL)
{
printf("create window failed error code %d\n", GetLastError());
return 0;
}
WSADATA wsaData;
int iResult;
SOCKET ServerSocket = INVALID_SOCKET;
iResult = WSAStartup(MAKEWORD(2, 2), &wsaData);
if (iResult!=0)
{
printf("WSASTART fAILED WITH ERROR CODE %d\n",iResult);
WSACleanup();
return 1;
}
ServerSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_IP);
if (ServerSocket == INVALID_SOCKET)
{
printf("socket failed with error: %ld\n", WSAGetLastError());
WSACleanup();
return 1;
}
SOCKADDR_IN addrServ;
addrServ.sin_family = AF_INET;
addrServ.sin_port = htons(DEFAULT_PORT);
addrServ.sin_addr.S_un.S_addr = htonl(INADDR_ANY);
iResult = bind(ServerSocket, (const struct sockaddr*)&addrServ,sizeof(SOCKADDR_IN));
if (iResult == SOCKET_ERROR)
{
printf_s("bind falied with error code %d\n", WSAGetLastError());
closesocket(ServerSocket);
WSACleanup();
return 1;
}
iResult = listen(ServerSocket, SOMAXCONN);
if (iResult == SOCKET_ERROR)
{
printf_s("listen failed with error code %d", iResult);
closesocket(ServerSocket);
WSACleanup();
return -1;
}
printf_s("TCP server starting");
WSAAsyncSelect(ServerSocket,hWnd,WM_SOCKET,FD_ACCEPT);
while (GetMessage(&Msg,NULL,0,0))
{
TranslateMessage(&Msg);
DispatchMessage(&Msg);
}
return 0;
}
HWND hwnd:窗口句柄;
● UINT uMsg:消息;
● WPARAM wParam:消息传入参数,此处传入的是套接字;
● LPARAM lParam:消息传入参数,此处传入的是事件类型。
输出参数:
● TRUE:表示成功;
● FALSE:表示失败。
窗口例程是一个回调函数,网络事件到达后,窗口例程被执行。
第12~13行代码首先检查lParam参数的高位,以判断是否在套接字上发生了网络错误,宏WSAGETSELECTERROR返回高字节包含的错误信息。
第16~59行代码使用宏WSAGETSELECTEVENT读取lParam参数的低位确定发生的网络事件。本示例对FD_ACCEPT、FD_READ、FD_CLOSE三种常见的网络事件都进行了判断和处理。如果是FD_ACCEPT事件发生,则调用accept()函数返回新的连接套接字,并通过WSAAsyncSelect()函数注册该套接字上关心的网络事件;如果是FD_READ事件发生,则在发生事件的套接字上接收数据;如果是FD_CLOSE事件发生,则关闭相应的套接字;如果套接字还对其他网络事件感兴趣,则在swith-case语句中相应增加对其他事件的判断和处理。
(3)第三步:完成程序主框架
程序主框架实现了流式套接字服务器的主体功能,创建接收消息的窗口,初始化套接字,注册套接字感兴趣的网络事件和消息的对应关系,并不断地循环读取消息和分发消息。
该函数的具体实现代码如下:
LRESULT CALLBACK WndProc(HWND hWnd,UINT uMsg,WPARAM wParam,LPARAM lParam) {
sockaddr_in addrClient;
int addrClientLen = sizeof(sockaddr_in);
char recvbuf[DEFAULT_BUFLEN];
int recvbuflen = sizeof(recvbuf);
SOCKET s, AcceptSocket;
int iResult;
if (uMsg== WM_SOCKET)
{
s = wParam;
if (WSAGETSELECTERROR(lParam))
{
printf("socket failed error code %d\n", WSAGETSELECTERROR(lParam));
}
else
{
switch (WSAGETSELECTEVENT(lParam))
{
case FD_ACCEPT:
//检测到有新的连接进入
AcceptSocket = accept(s,(sockaddr FAR*)&addrClient,&addrClientLen);
if (AcceptSocket==INVALID_SOCKET)
{
printf("ACCEPTED \n");
closesocket(s);
}
printf("get new conn\n");
WSAAsyncSelect(AcceptSocket,hWnd,WM_SOCKET,FD_READ|FD_WRITE|FD_CLOSE);
break;
case FD_READ:
memset(recvbuf,0,recvbuflen);
iResult = recv(s, recvbuf, recvbuflen, 0);
if (iResult>=0)
{
printf("\nByte received %d\n", iResult);
}
else
{
printf("recv failed with error code %d",WSAGetLastError());
closesocket(s);
}
case FD_WRITE:
{}
break;
default:
break;
}
}
}
return DefWindowProc(hWnd, uMsg, wParam, lParam);
}
本程序是一个命令行程序,为了使用WSAAsyncSelect模型,需要用户手工创建窗口并维护消息队列。
第9行代码定义了套接字网络事件对应的用户消息WM_SOCKET。
第15~19行代码调用第一步声明的CreateWorkerWindow()函数,用于注册窗口并创建窗口。
第20~60行代码进行基于流式套接字的服务器程序初始化。
第62行代码调用WSAAsyncSelect()函数向窗口hWnd注册了套接字ServerSocket感兴趣的网络事件,并与用户自定义的消息WM_SOCKET关联起来。
第63~67行代码维护了一个消息循环,不断地通过GetMessage()函数取出消息,对消息转换后将消息发送给窗口例程。
在MFC下实现基于WSAAsyncSelect模型的套接字通信服务器
评价
WSAAsyncSelect模型的优点是在系统开销不大的情况下可以同时处理多个客户的网络I/O。但是,消息的运转需要有消息队列,通常消息队列依附于窗口实现,有些应用程序可能并不需要窗口,为了支持消息机制,就必须创建一个窗口来接收消息,这对于一些特殊的应用场合并不适合。另外,在一个窗口中处理大量的消息也可能成为性能的瓶颈