Socket网络编程之WSAAsyncSelect模型

使用异步选择模型，应用程序可在一个套接字上，接收以Windows消息为基础的网络事件通知。具体的做法是在建好一个套接字后，调用WSAAsyncSelect函数。

服务器端代码：

C++代码

// write by larry  

// 2009-8-20   

// This is a server using WSAAsyncSelect model.  

#include "stdafx.h"  

#include <winsock.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 APIENTRY WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLine, int nCmdShow)   

{   

    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, _T("This program requires Windows NT!"), szAppName, MB_ICONERROR);  

        return 0;   

    }   

    hwnd = CreateWindow(szAppName, // window class name   

        _T("AsyncSelect Model"),   // window caption   

        WS_OVERLAPPEDWINDOW,       // window stype   

        CW_USEDEFAULT,             // initial x postion   

        CW_USEDEFAULT,             // initial y postion   

        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, nCmdShow);  

    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 wsaData;  

    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, &wsaData);  

        // Create listening socket   

        sListen = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);  

        // Bind   

        local.sin_family = AF_INET;  

        local.sin_addr.S_un.S_addr = htonl(INADDR_ANY);  

        local.sin_port = htons(PORT);  

        bind(sListen, (sockaddr*)&local, sizeof(SOCKADDR_IN));   

        // 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, (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);  

}  

// writeby larry

//2009-8-20

// Thisis a server using WSAAsyncSelect model.

#include"stdafx.h"

#include<winsock.h>

#include<tchar.h>

#definePORT  5150

#defineMSGSIZE  1024

#defineWM_SOCKET  WM_USER+0

#pragmacomment(lib, "ws2_32.lib")

LRESULTCALLBACK WndProc(HWND, UINT, WPARAM, LPARAM);

intAPIENTRY WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLine,int nCmdShow)

{

    staticTCHAR 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, _T("This programrequires Windows NT!"), szAppName, MB_ICONERROR);

      return 0;

   }

   hwnd = CreateWindow(szAppName, // windowclass name

      _T("AsyncSelect Model"),   // window caption

      WS_OVERLAPPEDWINDOW,       // window stype

      CW_USEDEFAULT,             // initial x postion

      CW_USEDEFAULT,             // initial y postion

      CW_USEDEFAULT,             // initial x size

      CW_USEDEFAULT,             // initial y size

      NULL,                      // parent window handle

      NULL,                      // window menu handle

      hInstance,                 // program instance handle

      NULL);                    // creationparameters

   ShowWindow(hwnd, nCmdShow);

   UpdateWindow(hwnd);

   while (GetMessage(&msg, NULL, 0, 0))

   {

      TranslateMessage(&msg);

      DispatchMessage(&msg);

   }

   return msg.wParam;

}

LRESULTCALLBACK WndProc(HWND hwnd, UINT message, WPARAM wParam, LPARAM lParam)

{

   WSADATA wsaData;

   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, &wsaData);

      // Create listening socket

      sListen = socket(AF_INET, SOCK_STREAM,IPPROTO_TCP);

      // Bind

      local.sin_family = AF_INET;

      local.sin_addr.S_un.S_addr =htonl(INADDR_ANY);

      local.sin_port = htons(PORT);

      bind(sListen, (sockaddr*)&local,sizeof(SOCKADDR_IN));

      // Listen

      listen(sListen, 3);

      // Associate listening socket withFD_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, (sockaddr*)&client,&iAddrSize);

         // Associate client socket with FD_READand 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);

}

服务器端得主要流程：
1.在WM_CREATE消息处理函数中，初始化Windows Socket library，创建监听套接字，绑定，监听，并且调用WSAAsyncSelect函数表示我们关心在监听套接字上发生的FD_ACCEPT事件；
2.自定义一个消息WM_SOCKET，一旦在我们所关心的套接字（监听套接字和客户端套接字）上发生了某个事件，系统就会调用WndProc并且message参数被设置为WM_SOCKET；
3.在WM_SOCKET的消息处理函数中，分别对FD_ACCEPT、FD_READ和FD_CLOSE事件进行处理；
4.在窗口销毁消息(WM_DESTROY)的处理函数中，我们关闭监听套接字，清除Windows Socket library

WSAAsyncSelect函数的网络事件类型可以有以下一种：
FD_READ 应用程序想要接收有关是否可读的通知，以便读入数据
FD_WRITE 应用程序想要接收有关是否可写的通知，以便写入数据
FD_OOB 应用程序想接收是否有带外（OOB）数据抵达的通知
FD_ACCEPT 应用程序想接收与进入连接有关的通知
FD_CONNECT 应用程序想接收与一次连接或者多点join操作完成的通知
FD_CLOSE 应用程序想接收与套接字关闭有关的通知
FD_QOS 应用程序想接收套接字“服务质量”（QoS）发生更改的通知
FD_GROUP_QOS  应用程序想接收套接字组“服务质量”发生更改的通知（现在没什么用处，为未来套接字组的使用保留）
FD_ROUTING_INTERFACE_CHANGE 应用程序想接收在指定的方向上，与路由接口发生变化的通知
FD_ADDRESS_LIST_CHANGE  应用程序想接收针对套接字的协议家族，本地地址列表发生变化的通知