WSAAsyncSelect 模型允许应用程序以Windows 消息的形式接收网络事件通知。这个模型是为了适应Windows 的消息驱动环境而设置的,现在许多对性能要求不高的网络应用程序都采用WSAAsyncSelect 模型,MFC(Microsoft Foundation Class,Microsoft 基础类库)中的CSocket 类也使用了它。
WSAAsyncSelect 函数自动把套接字设为非阻塞模式,并且为套接字绑定一个窗口句柄,当有网络事件发生时,便向这个窗口发送消息。
int WSAAsyncSelect(
SOCKET s, // 需要设置的套接字句柄
HWND hWnd, // 指定一个窗口句柄,套接字的通知消息将被发送到与其对应的窗口过程中
u_int wMsg, // 网络事件到来时接收到的消息ID,可以在WM_USER 以上的数值中任意选择一个用作ID
long lEvent // 指定哪些通知码需要发送
);
最后一个参数lEvent 指定了要发送的通知码,可以是如下取值的组合。
FD_READ:套接字接收到对方发送过来的数据包,表明这时可以去读套接字了。
FD_WRITE:数据缓冲区满后再次变空时,WinSock 接口通过该通知码通知应用程序。表示可以继续发送数据了(短时间内发送数据过多,便会造成数据缓冲区变满)。
FD_ACCEPT:监听中的套接字检测到有连接进入。
FD_CONNECT:如果用套接字去连接对方的主机,当连接动作完成以后会接收到这个通知码。
FD_CLOSE:检测到套接字对应的连接被关闭。
LRESULT CALLBACK WindowProc(HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam);
wParam 参数指定了发生网络事件的套接字句柄,lParam 参数的低字位指定了发生的网络事件,高字位包含了任何可能出现的错误代码,可以使用宏WSAGETSELECTERROR 和WSAGETSELECTEVENT 将这些信息取出。如果没有错误发生,出错代码为0,程序可以继续检查通知码,以确定发生的网络事件
#include <WINSOCK2.H>
#include <STDIO.H>
#pragma comment(lib, "Ws2_32.lib")
#define WM_SOCKET WM_USER + 101 // 自定义消息
LRESULT CALLBACK WindowProc(HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam);
int main()
{
char szClassName[] = "MainWClass";
WNDCLASSEX wndclass;
wndclass.cbSize = sizeof(wndclass);
wndclass.style = CS_HREDRAW|CS_VREDRAW;
wndclass.lpfnWndProc = WindowProc;
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 = szClassName ;
wndclass.hIconSm = NULL;
::RegisterClassEx(&wndclass);
// 创建主窗口(不显示)
HWND hWnd = ::CreateWindowEx( 0,szClassName,"WSAAsyncSelect",
WS_OVERLAPPEDWINDOW, CW_USEDEFAULT,
CW_USEDEFAULT,CW_USEDEFAULT,CW_USEDEFAULT,
NULL,NULL,NULL,NULL);
if(hWnd == NULL)
{
::MessageBox(NULL, "创建窗口出错!", "error", MB_OK);
return -1;
}
WSADATA wsaData;
WSAStartup(MAKEWORD(2,2), &wsaData);
USHORT nPort = 9000; // 此服务器监听的端口号
// 创建监听套节字
SOCKET sListen = ::socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
sockaddr_in sin;
sin.sin_family = AF_INET;
sin.sin_port = htons(nPort);
sin.sin_addr.S_un.S_addr = INADDR_ANY;
// 绑定套节字到本地机器
if(::bind(sListen, (sockaddr*)&sin, sizeof(sin)) == SOCKET_ERROR)
{
printf(" Failed bind() \n");
return -1;
}
// 将套接字设为窗口通知消息类型, 关注连接和关闭通知
::WSAAsyncSelect(sListen, hWnd, WM_SOCKET, FD_ACCEPT|FD_CLOSE);
// 进入监听模式
::listen(sListen, 5);
printf("server port %d is listenning ... \n", nPort);
MSG msg;
while(::GetMessage(&msg, NULL, 0, 0))
{
::TranslateMessage(&msg);
::DispatchMessage(&msg);
}
WSACleanup();
return msg.wParam;
}
LRESULT CALLBACK WindowProc(HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
{
switch (uMsg)
{
case WM_SOCKET:
{
// 取得有事件发生的套节字句柄
SOCKET s = wParam;
// 查看是否出错
if(WSAGETSELECTERROR(lParam))
{
::closesocket(s);
printf("[sock %d] one error is occur, error code %d\n", s, WSAGetLastError());
return 0;
}
// 处理发生的事件
switch(WSAGETSELECTEVENT(lParam))
{
case FD_ACCEPT: // 监听中的套接字检测到有连接进入
{
SOCKET client = ::accept(s, NULL, NULL);
// 将套接字设为窗口通知消息类型, 关注读写和关闭通知
::WSAAsyncSelect(client, hWnd, WM_SOCKET, FD_READ|FD_WRITE|FD_CLOSE);
printf("[sock %d] one client connect\n", client);
}
break;
case FD_WRITE:
{
}
break;
case FD_READ:
{
char szText[1024] = { 0 };
if(::recv(s, szText, 1024, 0) == -1)
{
printf("[sock %d] one error occur when recv, error code %d\n", WSAGetLastError());
::closesocket(s);
}
else
printf("[sock %d] recv data: %s\n", s, szText);
}
break;
case FD_CLOSE:
{
::closesocket(s);
printf("[sock %d] close socket\n", s);
}
break;
}
}
return 0;
case WM_DESTROY:
::PostQuitMessage(0) ;
return 0 ;
}
return ::DefWindowProc(hWnd, uMsg, wParam, lParam);
}