Socket异步编程之基于消息

本例采用Socket实现局域网通信。

开发环境：XP+VS2005+MFC

源码已上传到CSDN资源中：http://download.csdn.net/detail/segen_jaa/4638910，也可私信给我，单独发送。

关键API：WSAAsyncSelect，以网络事件为基础，基于消息实现异步通信。

说明：同一socket上，两次调用WSAAsyncSelect，第二次注册消息会覆盖第一次消息。

1、服务端

效果如下图。

主对话框，响应FD_ACCEPT（接受客户端）|FD_READ（获取客户端信息）|FD_CLOSE（客户端关闭）消息。

[cpp] view plain copy print ?

1. #define WM_NETWORK WM_USER+100
3. LRESULT OnNetwork(WPARAM wParam, LPARAM lParam);
5. ON_MESSAGE(WM_NETWORK,&CAsyncServerDlg::OnNetwork)
7. {
8. ......
9. //注册感兴趣的网络事件
10. int nAsyncRet = WSAAsyncSelect(m_sListen, m_hWnd, WM_NETWORK, FD_ACCEPT|FD_READ|FD_CLOSE);
11. if (SOCKET_ERROR == nAsyncRet)
12. {
13. CAsyncFunc::SetAppendText(this, IDC_EDIT_DESC, _T("注册网络事件失败！"));
15. return;
16. }
17. ......
18. }
20. LRESULT CAsyncServerDlg::OnNetwork(WPARAM wParam, LPARAM lParam)
21. {
22. SOCKET sClient = (SOCKET)wParam;
23. WORD netEvent = WSAGETSELECTEVENT(lParam);
24. WORD error = WSAGETSELECTERROR(lParam);
25. if (error!=0)
26. {
27. CString strErrorInfo = _T("");
28. strErrorInfo.Format(_T("Error code: %d"), error);
29. CAsyncFunc::SetAppendText(this, IDC_EDIT_DESC, strErrorInfo);
31. return -1;
32. }
34. switch(netEvent)
35. {
36. case FD_ACCEPT: OnAccept(); break;
37. case FD_READ: OnFDRead(sClient); break;
38. case FD_CLOSE: OnFDClose(sClient); break;
39. }
41. return 0;
42. }
44. void CAsyncServerDlg::OnFDRead(SOCKET sClient)
45. {
46. //该客户端首次收到消息，启动对话框
47. CAsyncTalkMng::GetInstance()->StartTalkDlg(sClient, this);
48. }
50. void CAsyncServerDlg::OnFDClose(SOCKET sClient)
51. {
52. //删除列表条目
53. int nItemCount = m_ClientList.GetCount();
54. for (int i = 0; i < nItemCount; i++)
55. {
56. SOCKET sTemp = (SOCKET)m_ClientList.GetItemData(i);
57. if (sTemp == sClient)
58. {
59. m_ClientList.DeleteString(i);
60. break;
61. }
62. }
64. //关闭连接
65. CAsyncTalkMng::GetInstance()->RemoveClient(sClient);
67. //通知对话框，连接已关闭
68. CAsyncTalkMng::GetInstance()->StopTalkDlg(sClient);
69. }
71. void CAsyncServerDlg::OnAccept()
72. {
73. struct sockaddr_in client;
74. int iAddrSize = sizeof(client);
75. SOCKET sClient = accept(m_sListen, (struct sockaddr*)&client, &iAddrSize);
76. if (sClient == INVALID_SOCKET)
77. {
78. CString strErrorInfo = _T("");
79. strErrorInfo.Format(_T("accept() failed:%d"), WSAGetLastError());
81. CAsyncFunc::SetAppendText(this, IDC_EDIT_DESC, strErrorInfo);
83. return;
84. }
86. char* sClientIP = inet_ntoa(client.sin_addr);
87. wstring wsClientIP = CAsyncFunc::AnsiToUnicode(string(sClientIP));
88. CString strClientIP = wsClientIP.c_str();
90. CString strIPAddr = _T("");
91. strIPAddr.Format(_T("%s:%d"), strClientIP, ntohs(client.sin_port));
93. CString strTipInfo = _T("");
94. strTipInfo += _T("accepted client:");
95. strTipInfo += strIPAddr;
96. CAsyncFunc::SetAppendText(this, IDC_EDIT_DESC, strTipInfo);
98. StClientSock* pClientSock = new StClientSock;
99. pClientSock->sClient = sClient;
100. pClientSock->strIPAddr = strIPAddr;
102. CAsyncTalkMng::GetInstance()->AddClient(pClientSock);
104. int nItemIndex = m_ClientList.AddString(strIPAddr);
106. m_ClientList.SetItemData(nItemIndex, sClient);
107. }

#define WM_NETWORK WM_USER+100

LRESULT OnNetwork(WPARAM wParam, LPARAM lParam);

ON_MESSAGE(WM_NETWORK,&CAsyncServerDlg::OnNetwork)

{
    ......  
    //注册感兴趣的网络事件
    int nAsyncRet = WSAAsyncSelect(m_sListen, m_hWnd, WM_NETWORK, FD_ACCEPT|FD_READ|FD_CLOSE);
    if (SOCKET_ERROR == nAsyncRet)
    {
        CAsyncFunc::SetAppendText(this, IDC_EDIT_DESC, _T("注册网络事件失败！"));

        return;
    }
    ......
}

LRESULT CAsyncServerDlg::OnNetwork(WPARAM wParam, LPARAM lParam)  
{  
    SOCKET sClient = (SOCKET)wParam;
    WORD netEvent = WSAGETSELECTEVENT(lParam);
    WORD error = WSAGETSELECTERROR(lParam);
    if (error!=0)
    {
        CString strErrorInfo = _T("");
        strErrorInfo.Format(_T("Error code: %d"), error);
        CAsyncFunc::SetAppendText(this, IDC_EDIT_DESC, strErrorInfo);

        return -1;
    }

    switch(netEvent)
    {
    case FD_ACCEPT: OnAccept(); break;
    case FD_READ: OnFDRead(sClient); break;
    case FD_CLOSE: OnFDClose(sClient); break;
    }

    return 0;
}

void CAsyncServerDlg::OnFDRead(SOCKET sClient)
{
    //该客户端首次收到消息，启动对话框
    CAsyncTalkMng::GetInstance()->StartTalkDlg(sClient, this);
}

void CAsyncServerDlg::OnFDClose(SOCKET sClient)
{
    //删除列表条目
    int nItemCount = m_ClientList.GetCount();
    for (int i = 0; i < nItemCount; i++)
    {
        SOCKET sTemp = (SOCKET)m_ClientList.GetItemData(i);
        if (sTemp == sClient)
        {
            m_ClientList.DeleteString(i);
            break;
        }
    }

    //关闭连接
    CAsyncTalkMng::GetInstance()->RemoveClient(sClient);

    //通知对话框，连接已关闭
    CAsyncTalkMng::GetInstance()->StopTalkDlg(sClient);
}

void CAsyncServerDlg::OnAccept()
{
    struct sockaddr_in client;
    int iAddrSize = sizeof(client);
    SOCKET sClient = accept(m_sListen, (struct sockaddr*)&client, &iAddrSize);
    if (sClient == INVALID_SOCKET)
    {
        CString strErrorInfo = _T("");
        strErrorInfo.Format(_T("accept() failed:%d"), WSAGetLastError());

        CAsyncFunc::SetAppendText(this, IDC_EDIT_DESC, strErrorInfo);

        return;
    }

    char* sClientIP = inet_ntoa(client.sin_addr);
    wstring wsClientIP = CAsyncFunc::AnsiToUnicode(string(sClientIP));
    CString strClientIP = wsClientIP.c_str();

    CString strIPAddr = _T("");
    strIPAddr.Format(_T("%s:%d"), strClientIP, ntohs(client.sin_port));

    CString strTipInfo = _T("");
    strTipInfo += _T("accepted client:");
    strTipInfo += strIPAddr;
    CAsyncFunc::SetAppendText(this, IDC_EDIT_DESC, strTipInfo);

    StClientSock* pClientSock = new StClientSock;
    pClientSock->sClient = sClient;
    pClientSock->strIPAddr = strIPAddr;

    CAsyncTalkMng::GetInstance()->AddClient(pClientSock);

    int nItemIndex = m_ClientList.AddString(strIPAddr);

    m_ClientList.SetItemData(nItemIndex, sClient);
}

聊天框，响应FD_READ（获取客户端信息）|FD_CLOSE（客户端关闭）消息。

[cpp] view plain copy print ?

1. #define WM_NETWORK_TALK WM_USER+101
3. LRESULT OnNetwork(WPARAM wParam, LPARAM lParam);
4. void OnFDRead();
5. void OnFDClose(SOCKET sClient);
7. ON_MESSAGE(WM_NETWORK_TALK,&CAsyncTalkDlg::OnNetwork)
9. void CAsyncTalkDlg::InitSocketMsg()
10. {
11. int nAsyncRet = WSAAsyncSelect(clientSock.sClient, m_hWnd, WM_NETWORK_TALK, FD_READ|FD_CLOSE);
12. if (SOCKET_ERROR == nAsyncRet)
13. {
14. CAsyncFunc::SetAppendText(this, IDC_EDIT_DESC, _T("注册网络事件失败！"));
16. return;
17. }
18. }
20. LRESULT CAsyncTalkDlg::OnNetwork(WPARAM wParam, LPARAM lParam)
21. {
22. WORD netEvent = WSAGETSELECTEVENT(lParam);
23. WORD error = WSAGETSELECTERROR(lParam);
24. if (error!=0)
25. {
26. CString strErrorInfo = _T("");
27. strErrorInfo.Format(_T("Error code: %d"), error);
28. CAsyncFunc::SetAppendText(this, IDC_EDIT_DESC, strErrorInfo);
30. return -1;
31. }
33. switch(netEvent)
34. {
35. case FD_READ: OnFDRead(); break;
36. case FD_CLOSE: OnFDClose((SOCKET)wParam); break;
37. }
39. return 0;
40. }
42. void CAsyncTalkDlg::OnFDRead()
43. {
44. char szBuff[DEFAULT_BUFFER] = {0};
46. int ret = recv(clientSock.sClient, szBuff, DEFAULT_BUFFER, 0);
47. if (0 == ret)
48. {
49. return;
50. }
51. else if (SOCKET_ERROR == ret)
52. {
53. CAsyncFunc::SetAppendText(this, IDC_EDIT_RECV, _T("recv() failed"));
54. return;
55. }
56. szBuff[ret] = '\0';
58. wstring wsRecvInfo = CAsyncFunc::AnsiToUnicode(string(szBuff));
59. CString strRecvInfo = _T("");
60. strRecvInfo += clientSock.strIPAddr;
61. strRecvInfo += _T(":");
62. strRecvInfo += (CString)(wsRecvInfo.c_str());
64. CAsyncFunc::SetAppendText(this, IDC_EDIT_RECV, strRecvInfo);
65. }
67. void CAsyncTalkDlg::OnFDClose(SOCKET sClient)
68. {
69. if (NULL == m_pMsgFlow)
70. {
71. return;
72. }
74. m_pMsgFlow->OnFDClose(sClient);
75. }

#define WM_NETWORK_TALK WM_USER+101

LRESULT OnNetwork(WPARAM wParam, LPARAM lParam);
void OnFDRead();
    void OnFDClose(SOCKET sClient);

ON_MESSAGE(WM_NETWORK_TALK,&CAsyncTalkDlg::OnNetwork)

void CAsyncTalkDlg::InitSocketMsg()
{
    int nAsyncRet = WSAAsyncSelect(clientSock.sClient, m_hWnd, WM_NETWORK_TALK, FD_READ|FD_CLOSE);
    if (SOCKET_ERROR == nAsyncRet)
    {
        CAsyncFunc::SetAppendText(this, IDC_EDIT_DESC, _T("注册网络事件失败！"));

        return;
    }
}

LRESULT CAsyncTalkDlg::OnNetwork(WPARAM wParam, LPARAM lParam)  
{
    WORD netEvent = WSAGETSELECTEVENT(lParam);
    WORD error = WSAGETSELECTERROR(lParam);
    if (error!=0)
    {
        CString strErrorInfo = _T("");
        strErrorInfo.Format(_T("Error code: %d"), error);
        CAsyncFunc::SetAppendText(this, IDC_EDIT_DESC, strErrorInfo);

        return -1;
    }

    switch(netEvent)
    {
    case FD_READ: OnFDRead(); break;
    case FD_CLOSE: OnFDClose((SOCKET)wParam); break;
    }

    return 0;
}

void CAsyncTalkDlg::OnFDRead()
{
    char szBuff[DEFAULT_BUFFER] = {0};

    int ret = recv(clientSock.sClient, szBuff, DEFAULT_BUFFER, 0);
    if (0 == ret)
    {
        return;
    }
    else if (SOCKET_ERROR == ret)
    {
        CAsyncFunc::SetAppendText(this, IDC_EDIT_RECV, _T("recv() failed"));
        return;
    }
    szBuff[ret] = '\0';

    wstring wsRecvInfo = CAsyncFunc::AnsiToUnicode(string(szBuff));
    CString strRecvInfo = _T("");
    strRecvInfo += clientSock.strIPAddr;
    strRecvInfo += _T(":");
    strRecvInfo += (CString)(wsRecvInfo.c_str());

    CAsyncFunc::SetAppendText(this, IDC_EDIT_RECV, strRecvInfo);
}

void CAsyncTalkDlg::OnFDClose(SOCKET sClient)
{
    if (NULL == m_pMsgFlow)
    {
        return;
    }

    m_pMsgFlow->OnFDClose(sClient);
}

2、客户端

效果图如下。

主要代码，响应FD_READ（获取服务端消息）|FD_CLOSE（服务端socket关闭）消息。

[cpp] view plain copy print ?

1. #define WM_NETWORK WM_USER+100
3. LRESULT OnNetwork(WPARAM wParam, LPARAM lParam);
4. void OnFDRead();
5. void OnFDClose();
7. ON_MESSAGE(WM_NETWORK,&CAsyncClientDlg::OnNetwork)
9. {
10. ......
11. //注册感兴趣的网络事件
12. int nAsyncRet = WSAAsyncSelect(m_sClient, m_hWnd, WM_NETWORK, FD_READ|FD_CLOSE);
13. if (SOCKET_ERROR == nAsyncRet)
14. {
15. CAsyncFunc::SetAppendText(this, IDC_EDIT_RECV, _T("注册网络事件失败！"));
17. return;
18. }
19. ......
20. }
22. LRESULT CAsyncClientDlg::OnNetwork(WPARAM wParam, LPARAM lParam)
23. {
24. //SOCKET s = (SOCKET)wParam;
25. WORD netEvent = WSAGETSELECTEVENT(lParam);
26. WORD error = WSAGETSELECTERROR(lParam);
27. if (error!=0)
28. {
29. CString strErrorInfo = _T("");
30. strErrorInfo.Format(_T("Error code: %d"), error);
31. CAsyncFunc::SetAppendText(this, IDC_EDIT_RECV, strErrorInfo);
33. return -1;
34. }
36. switch(netEvent)
37. {
38. case FD_READ: OnFDRead(); break;
39. case FD_CLOSE: OnFDClose(); break;
40. }
42. return 0;
43. }
45. void CAsyncClientDlg::OnFDRead()
46. {
47. char szBuff[DEFAULT_BUFFER] = {0};
49. int ret = recv(m_sClient, szBuff, DEFAULT_BUFFER, 0);
50. if (0 == ret)
51. {
52. return;
53. }
54. else if (SOCKET_ERROR == ret)
55. {
56. CAsyncFunc::SetAppendText(this, IDC_EDIT_RECV, _T("recv() failed"));
57. return;
58. }
59. szBuff[ret] = '\0';
61. wstring wsRecvInfo = CAsyncFunc::AnsiToUnicode(string(szBuff));
62. CString strRecvInfo = _T("");
63. strRecvInfo += _T("服务端:");
64. strRecvInfo += (CString)(wsRecvInfo.c_str());
66. CAsyncFunc::SetAppendText(this, IDC_EDIT_RECV, strRecvInfo);
67. }
69. void CAsyncClientDlg::OnFDClose()
70. {
71. closesocket(m_sClient);
72. m_bSocketClose = true;
74. CAsyncFunc::SetAppendText(this, IDC_EDIT_RECV, _T("服务端已关闭"));
76. GetDlgItem(IDC_EDIT_SEND)->EnableWindow(FALSE);
77. GetDlgItem(IDC_BTN_SEND)->EnableWindow(FALSE);
78. }

#define WM_NETWORK WM_USER+100

LRESULT OnNetwork(WPARAM wParam, LPARAM lParam);
void OnFDRead();
void OnFDClose();

ON_MESSAGE(WM_NETWORK,&CAsyncClientDlg::OnNetwork)

{
    ......
    //注册感兴趣的网络事件
    int nAsyncRet = WSAAsyncSelect(m_sClient, m_hWnd, WM_NETWORK, FD_READ|FD_CLOSE);
    if (SOCKET_ERROR == nAsyncRet)
    {
        CAsyncFunc::SetAppendText(this, IDC_EDIT_RECV, _T("注册网络事件失败！"));

        return;
    }
    ......
}

LRESULT CAsyncClientDlg::OnNetwork(WPARAM wParam, LPARAM lParam)
{
    //SOCKET s = (SOCKET)wParam;
    WORD netEvent = WSAGETSELECTEVENT(lParam);
    WORD error = WSAGETSELECTERROR(lParam);
    if (error!=0)
    {
        CString strErrorInfo = _T("");
        strErrorInfo.Format(_T("Error code: %d"), error);
        CAsyncFunc::SetAppendText(this, IDC_EDIT_RECV, strErrorInfo);

        return -1;
    }

    switch(netEvent)
    {
    case FD_READ: OnFDRead(); break;
    case FD_CLOSE: OnFDClose(); break;
    }

    return 0;
}

void CAsyncClientDlg::OnFDRead()  
{
    char szBuff[DEFAULT_BUFFER] = {0};

    int ret = recv(m_sClient, szBuff, DEFAULT_BUFFER, 0);
    if (0 == ret)
    {
        return;
    }
    else if (SOCKET_ERROR == ret)
    {
        CAsyncFunc::SetAppendText(this, IDC_EDIT_RECV, _T("recv() failed"));
        return;
    }
    szBuff[ret] = '\0';

    wstring wsRecvInfo = CAsyncFunc::AnsiToUnicode(string(szBuff));
    CString strRecvInfo = _T("");
    strRecvInfo += _T("服务端:");
    strRecvInfo += (CString)(wsRecvInfo.c_str());

    CAsyncFunc::SetAppendText(this, IDC_EDIT_RECV, strRecvInfo);
}

void CAsyncClientDlg::OnFDClose()
{
    closesocket(m_sClient);
    m_bSocketClose = true;

    CAsyncFunc::SetAppendText(this, IDC_EDIT_RECV, _T("服务端已关闭"));

    GetDlgItem(IDC_EDIT_SEND)->EnableWindow(FALSE);
    GetDlgItem(IDC_BTN_SEND)->EnableWindow(FALSE);
}