WinsockIO之WSAAsyncSelect模式

 WSAAsynSelect模型： WSAAsynSelect模型也是一个常用的异步I/O模型。应用程序可以在一个套接字上接收以 WINDOWS消息为基础的网络事件通知。该模型的实现方法是通过调用WSAAsynSelect函 数 自动将套接字设置为非阻塞模式，并向WINDOWS注册一个或多个网络时间，并提供一 个通知时使用的窗口句柄。当注册的事件发生时，对应的窗口将收到一个基于消息的通知。

1.在WM_CREATE消息处理函数中，初始化Windows Socket library，创建监听套接字，绑定，监

听，并且调用WSAAsyncSelect函数表示我们关心在监听套接字上发生的FD_ACCEPT事件；

2.自定义一个消息MM_SOCKET，一旦在我们所关心的套接字（监听套接字和客户端套接字）上

发生了某个事件，系统发送消息（MM_SOCKET)给hWnd指向的窗体，而WndProc函数处理所有

发往窗体的消息，wparam为MM_SOCKET，lparam是发生error和具体事件的组合码。

4、可用WSAGETSELECTERROR(lParam)来查看是否出现错误，若返回值非0，则发生error，

反之，无error。

5、在WM_SOCKET的消息处理中，分别对FD_ACCEPT、FD_READ和FD_CLOSE事件进行处理；

#include <WinSock2.h>
#include <stdio.h>
#include "resource.h"
#include <tchar.h>

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

#define MM_SOCK WM_USER+1

BOOL CALLBACK DlgProc(HWND hDlg, UINT uMsg, WPARAM wParam, LPARAM lParam);

int WINAPI WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLine, int nShowCmd)
{
	HWND hDlg = CreateDialog(hInstance, MAKEINTRESOURCE(IDD_DIALOG1), NULL, DlgProc);
	ShowWindow(hDlg, nShowCmd);
	MSG uMsg;

	while (GetMessage(&uMsg, NULL, 0, 0))
	{
		DispatchMessage(&uMsg);
	}

	return uMsg.wParam;
}

SOCKET sListen;

BOOL CALLBACK DlgProc(HWND hDlg, UINT uMsg, WPARAM wParam, LPARAM lParam)
{
	switch (uMsg)
	{
		case WM_INITDIALOG:
		{
			WSADATA wd;

			if (0 != WSAStartup(MAKEWORD(2, 2), &wd))
			{
				WSACleanup();
			}

			sListen = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);

			if (SOCKET_ERROR == sListen)
			{
				WSACleanup();
			}

			SOCKADDR_IN ListenAddr;
			ListenAddr.sin_family = AF_INET;
			ListenAddr.sin_addr.s_addr = inet_addr("0.0.0.0");
			ListenAddr.sin_port = htons(5051);

			if (SOCKET_ERROR == bind(sListen, (const sockaddr*)&ListenAddr, sizeof(ListenAddr)))
			{
				WSACleanup();
			}

			listen(sListen, 3);

			WSAAsyncSelect(sListen, hDlg, MM_SOCK, FD_ACCEPT | FD_CLOSE);
		}
		break;

		case MM_SOCK:
		{
			if (WSAGETSELECTERROR(lParam))
			{
				break;
			}
			switch (WSAGETSELECTEVENT(lParam))
			{
				case FD_ACCEPT:
				{
					SOCKADDR_IN ListenAddr;
					int nLen = sizeof(ListenAddr);
					SOCKET sClient = accept(sListen, (sockaddr*)&ListenAddr, &nLen);
					WSAAsyncSelect(sClient, hDlg, MM_SOCK, FD_READ | FD_CLOSE);

					TCHAR strBuf[255] = _T("");
					sprintf(strBuf, "IP:%s, PORT:%d Login……", inet_ntoa(ListenAddr.sin_addr), ntohs(ListenAddr.sin_port));

					SendMessage(GetDlgItem(hDlg, IDC_LIST1), LB_ADDSTRING, 0, (LPARAM)strBuf);
				}
				break;

				case FD_READ:
				{
					char szBuf[255] = "";
					recv(wParam, szBuf, 255, 0);
					SendMessage(GetDlgItem(hDlg, IDC_LIST1), LB_ADDSTRING, 0, (LPARAM)szBuf);
					send(wParam, "ok", 3, 0);
				}
				break;

				case FD_CLOSE:
				{
					closesocket(wParam);
				}
				break;

				default:
					break;
			}
		}
		break;

		case WM_SIZE:
		{
			RECT rt;
			GetClientRect(hDlg, &rt);
			MoveWindow(GetDlgItem(hDlg, IDC_LIST1), 8, 8, rt.left - 8, rt.right - 8, TRUE);
		}
		break;

		case WM_CLOSE:
			DestroyWindow(hDlg);
			break;

		case WM_DESTROY:
			PostQuitMessage(0);
			break;

		default:
			break;
	}

	return FALSE;
}