一、引言

       上一文中我們介紹了使用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);
}

        下面是用來列印客戶端資訊和客戶端資料的兩個函式：

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 git@github.com:HymanLiuTS/NetDevelopment.git獲取本文原始碼：
git checkout NL53