【网络编程】之八、异步选择WSAAsyncSelect

   大家用这个模型，可以让应用程序在一个套接字上接收以windows消息为基础的网络事件通知。我们想要用这个操作，具体的做法就是我们建立一个套接字，然后调用WSAAsyncSelect函数， 也就是说，这个模型的核心就是我们的这个函数；

来看一下函数原型：

int WSAAsyncSelect(
_In_  SOCKET s,//我们感兴趣的套接字
_In_  HWND hWnd,//窗口的句柄，对于网络事件繁盛后，想要接收到的通知的那个窗口
_In_  unsigned int wMsg,//指定在发生网络事件时，打算接收的消息。
_In_  long lEvent//指定一个位掩码，对应于一系列网络事件的组合。
);

对于最后一个参数，在这里要说一下，他包括的网络事件模型：FD_READ、FD_WRITE、FD_ACCEPT、FD_CONNECT、FD_CLOSE。

到底使用FD_ACCEPT，还是使用FD_CONNECT类型，要取决于应用程序的身份是客户端，还是服务器。我们可以用或运算来一起使用；

FD_READ 应用程序想要接收有关是否可读的通知，以便读入数据

FD_WRITE 应用程序想要接收有关是否可写的通知，以便写入数据

FD_ACCEPT 应用程序想接收与进入连接有关的通知

FD_CONNECT   应用程序想接收与一次连接完成的通知

FD_CLOSE 应用程序想接收与套接字关闭的通知

实例：

WSAAsyncSeltct(s, hwnd, WM_SOCKET, FD_CONNECT | FD_READ | FD_WRITE | FD_CLOSE);

应用程序可以在套接字s上，接收到有关连接，发送，接收以及套接字关闭的一系列通知了。

要注意的是：通过上面的函数原型我们可以看出来，要想使用我们的这个模型，必须要先调用CreateWindow函数来创建一个窗口，然后再为窗口提供一个窗口例程函数(WinProc);

这里要说的注意是：

当你有多个事件的时候，你一定要在套接字上一次注册，一旦你在这个套接字上允许了事件通知，那么以后除非你明确的调用closesocket命令或者由应用程序针对的那个套接字调用了WSAAsyncSelect，那么你就更改了注册的网络事件的类型了，否则的话事件通知就会永远有效。  上面函数你最后一个参数设置为0，效果相当于你停止在套接字上进行的所有网络事件通知。

还有就是你的应用程序针对一个套接字调用了WSAAsyncSelect，那么套接字会从“锁定”模式变成“非锁定”模式。    这样以后，会导致错误。     为了防止错误的产生，应用程序依赖于由WSAAsyncSelect的第三个参数指定的用户自定义窗口消息，来判断网络事件类型何时在套接字上发生，不能盲目的调用。

当你的程序调用WSAAsyncSelect成功后，会在和hWnd窗口句柄对应的窗口以windows消息的形式接受网络事件的通知。

看窗口消息如何定义：

LRESULT CALLBACK WindowProc(
HWND hwnd,//一个窗口句柄，对窗口的调用就是由这个窗口发出的
UINT uMsg,//指定需要对那些消息进行处理。
WPARAM wParam,//指定在上面发生了一个网络事件的套接字。
LPARAM lParam//低字节制定了已经发生的网络事件，高字节包含了可能出现的错误代码。
);

来看一下步骤：当网络消息抵达窗口后，应用程序就会先检查lParam的高字节位，以判断是否是在网络错误。WSAGETSELECTERROR这个宏可返回高字节位包含的错误信息。

然后如果程序没有发现套接字上没产生错误，接着就要调查到底是那个网络事件类型，具体做法是读取lParam的低字节位。  WSAGETSELECTEVENT这个宏返回lParam的低字节部分。

看一下代码：

#include<Windows.h>
#include<tchar.h>

#define MSGSIZE 1024
#define WM_SOCKET WM_USER+100

#pragma comment(lib, "ws2_32.lib")

LRESULT CALLBACK WindowProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam);

int WINAPI WinMain( __in HINSTANCE hInstance, __in_opt HINSTANCE hPrevInstance, __in LPSTR lpCmdLine, __in int nShowCmd )
{
TCHAR classname[] = "AsyncSelect";

WNDCLASS wndclass; //窗口类
wndclass.style = CS_HREDRAW | CS_VREDRAW;//窗口类型
wndclass.lpfnWndProc = WindowProc;//窗口处理函数
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 = classname;//窗口类名为“窗口”
//注册窗口类
if (!RegisterClass(&wndclass))
{
MessageBox(NULL, "register class error", classname, MB_ICONERROR);
return 0;
}
//创建窗口(窗口类名，窗口标题，窗口风格，坐标缺省， 有没有父窗口，有没有子窗口， 创建这个窗口的应用程序当前句柄，不适用)
HWND hwnd = CreateWindow(classname, "AsyncSelect Model", WS_OVERLAPPEDWINDOW,
CW_USEDEFAULT, CW_USEDEFAULT, CW_USEDEFAULT, CW_USEDEFAULT,
NULL, NULL, hInstance, NULL);
//显示和更新
ShowWindow(hwnd, nShowCmd);
UpdateWindow(hwnd);

MSG msg;
//消息循环
while (GetMessage(&msg, NULL, 0, 0))
{
TranslateMessage(&msg);
DispatchMessage(&msg);
}

return msg.wParam;
}

LRESULT CALLBACK WindowProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
{
WSADATA wsd;
static SOCKET slisten;
SOCKET sClient;
SOCKADDR_IN local, client;
int ret, iAddrsize = sizeof(client);
char szMessage[MSGSIZE] = {0};

switch (uMsg)
{
case WM_DESTROY:
{
//退出清理
closesocket(slisten);
WSACleanup();
PostQuitMessage(0);
return 0;
}
case WM_CREATE:
{
//初始化
WORD sockVersion = MAKEWORD(2, 0);
WSAStartup(sockVersion, &wsd);
//创建socket
slisten = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);

local.sin_family = AF_INET;
local.sin_port = htons(8888);
local.sin_addr.S_un.S_addr = htonl(INADDR_ANY);
//绑定socket
if(bind(slisten, (sockaddr*)&local, sizeof(local)) == SOCKET_ERROR)
{
WSACleanup();
return 0;
}
//监听socket
if(listen(slisten, 5) == SOCKET_ERROR)
{
WSACleanup();
return 0;
}
//注册监听socket FD_ACCEPT事件，当socket有连接时，会发送WM_SOCKET消息给窗口
WSAAsyncSelect(slisten, hwnd, WM_SOCKET, FD_ACCEPT);
return 0;
}

case WM_SOCKET:
{
if (WSAGETSELECTERROR(lParam))
{
closesocket(wParam);
break;
}

switch(WSAGETSELECTEVENT(lParam))
{
case FD_ACCEPT:
{
//接受
sClient = accept(wParam, (sockaddr*)&client, &iAddrsize);
//注册客户socket FD_READ和FD_CLOSE事件
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;
}
default:
{
closesocket(wParam);
break;
}
}
}
return 0;
}

return DefWindowProc(hwnd, uMsg, wParam, lParam);
}

2012/9/2

jofranks于南昌