网络编程（53）—— Windows下使用WSAAsyncSelect实现窗口处理socket消息

一、引言

       上一文中我们介绍了使用WSAEventSelect实现异步通知IO的方法，本文我们主要讨论下使用WSAAsyncSelect处理socket的方法。本文的主要目标，是创建一个带界面的回声服务端，接收并返回客户端传过来的字符串，并在界面上显示该字符串。为此，我们将采用MFC的编程环境，建立如下的对话框程序：

二、WSAAsyncSelect函数

        WSAAsyncSelect也是windows下一种异步的select，它可以注册IO事件，当发生注册的IO事件时，它会发送一个我们自定义的消息给我们的窗口，而我们在窗口的消息处理函数中就可以处理这些消息了。它的原型如下：

int WSAAsyncSelect(  __in SOCKET s,  __in HWND hWnd,  __in unsigned int wMsg,  __in long lEvent);

s —— 用于监视的socket。

hWnd —— 用来接收socket消息的窗口。

wMsg —— 是我们自定义的消息，一般情况下使用“WM_USER+ 数字”的形式进行定义。

lEvent —— 进行监视的IO事件，包含如下几种：

FD_READ：套接字可读通知。

FD_WRITE：可写通知。

FD_ACCEPT：服务器接收连接的通知。

FD_CONNECT：有客户连接通知。

FD_OOB：外带数据到达通知。

FD_CLOSE：套接字关闭通知。

FD_QOS：服务质量发生变化通知。

FD_GROUP_QOS：组服务质量发生变化通知。

FD_ROUTING_INTERFACE_CHANGE：与路由器接口发生变化的通知。

FD_ADDRESS_LIST_CHANGE：本地地址列表发生变化的通知。

 

返回值 —— 正常情况下返回0，出现错误时返回错误码。

        我们知道Windows系统中消息都会携带两个参数wParam和lParam，我们在进行窗口的消息处理时往往会利用这两个参数。WSAAsyncSelect发送的socket消息也不例外，它的wParam参数携带的是发生IO事件的socket。而lParam低字节表明已发生的事件。高字节包含错误代码。我们可以使用WSAGETSElECTERROR宏来读取lParam的高字节获取错误码，使用WSAGETSELECTEVENT宏读取lParam的低字节来获取发生的事件类型。

三、编程步骤

3.1 自定义消息

        我们先自定义一个socket消息以用WSAAsyncSelect发送：

#define WM_SOCKET WM_USER+1

3.2 注册服务端的socket

       我们先经过一般步骤创建服务端的socket，并然后用WSAAsyncSelect进行注册：

    

    WSAStartup(MAKEWORD(2,2),&m_wsaData);    m_servSock=socket(AF_INET,SOCK_STREAM,0);       memset(&m_servAddr,0,sizeof(m_servAddr));    m_servAddr.sin_family=AF_INET;    m_servAddr.sin_addr.s_addr=htonl(INADDR_ANY);    m_servAddr.sin_port=htons(atoi("8888"));     bind(m_servSock,(SOCKADDR*)&m_servAddr,sizeof(m_servAddr));     listen(m_servSock,5);     WSAAsyncSelect(m_servSock,this->m_hWnd,WM_SOCKET,FD_ACCEPT|FD_READ);

        最后一句代码中，我们使用WSAAsyncSelect注册了服务端的socketm_servSock，WSAAsyncSelect的第二个参数填写了我们窗口的m_hWnd值，第三个参数是我们自定义的消息，而第四个参数填写的是注册的IO事件，分别表示接受链接和有数据可读。注册完成之后，一旦有新的客户端连接，系统就会发送一个WM_SOCKET消息给我们的窗口。

3.3 处理socket消息

        为了能够处理socket消息，我们先引入窗口的消息处理函数，从类向导中添加虚函数OnWndMsg，在OnWndMsg中我们进行socket消息的处理：

BOOL CWSAAsyncSelectServDlg::OnWndMsg(UINT message,WPARAM wParam, LPARAM lParam, LRESULT* pResult){    switch(message)    {        case WM_SOCKET:            {                SOCKETsock = (SOCKET) wParam;                if(m_servSock == sock)                {                    //获?取¨?                    m_clntAddrSz=sizeof(m_clntAddr);                                         m_clntSock=accept(m_servSock,(SOCKADDR*)&m_clntAddr,&m_clntAddrSz);                    WSAAsyncSelect(m_clntSock,this->m_hWnd,WM_SOCKET,FD_READ);                    //将?m_clntSock和¨ªm_clntAddr存ä?到Ì?映®3射¦?中D                    m_sockMap.SetAt(m_clntSock,m_clntAddr);                    PrintClientConnectMsg();                                   }                else                {                    m_strLen= recv(sock,buf,BUF_SIZE - 1,0);                    send(sock,buf,m_strLen,0);                    buf[m_strLen]=0;                    PrintClientSendMsg(sock);                }            }            break;        default:            break;    }     return CDialogEx::OnWndMsg(message, wParam, lParam,pResult);}

        从上述函数的WM_SOCKETswitch分支可以看到，我们通过wParam获取到了发生IO事件的socket，如果有需要还可以通过lParam获取到错误码或者事件类型。获取到socket之后，先判断socket是不是服务端的m_servSock。如果是，表明有新的客户端链接，接收连接并打印客户端信息；如果不是则是由客户端传来了数据，调用recv接收数据并send给客户端，最后打印数据。

        下面是用来打印客户端信息和客户端数据的两个函数：

void CWSAAsyncSelectServDlg::PrintClientConnectMsg(void){    CString str;    CTimelocTime=CTime::GetCurrentTime();    //获?取¨?当Ì¡À前¡ã时º¡À间?    CStringstrTime=locTime.Format("%Y-%m-%d%H:%M:%S");//将?当Ì¡À前¡ã时º¡À间?转Áa换?成¨¦CString类¤¨¤型¨ª    str.Format("%s Client %s:%d Connectted",strTime,inet_ntoa(m_clntAddr.sin_addr),m_clntAddr.sin_port);    this->m_list.AddString(str);}  void CWSAAsyncSelectServDlg::PrintClientSendMsg(SOCKET sock){    CString str;    m_clntAddr=m_sockMap[sock];    CTimelocTime=CTime::GetCurrentTime();      CStringstrTime=locTime.Format("%Y-%m-%d%H:%M:%S");//将?当Ì¡À前¡ã时º¡À间?转Áa换?成¨¦CString类¤¨¤型¨ª    str.Format("%s Client %s:%d say:%s",strTime,inet_ntoa(m_clntAddr.sin_addr),m_clntAddr.sin_port,buf);    this->m_list.AddString(str);}

运行程序，效果如下，服务端不单返回了客户端传过来的信息，还对信息进行了打印。

客户端1：

客户端2：

 

 服务端：

 Github位置：
https://github.com/HymanLiuTS/NetDevelopment
克隆本项目：
git clone [e-mail]:HymanLiuTS/NetDevelopment.git
获取本文源代码：
git checkout NL53