ATL异步Socket通信实现方案，包含客户端收发功能和对话框数据显示

源码资源网

ATL异步Socket通信实现方案，包含客户端收发功能和对话框数据显示
代码说明：1.使用WSAAsyncSelect实现异步事件通知 2.线程安全的环形缓冲区处理数据接收 3.对话框实时显示收发状态 4.需配套资源文件定义IDD_MAINDLG对话框及控件ID
ATLSocketClient.h

1. 
   
2. #pragma once
   
3. #include <atlbase.h>
   
4. #include <atlwin.h>
   
5. #include <winsock2.h>
   
6. #include <vector>
   
7. #include <mutex>
   
8. 
   
9. #define WM_SOCKET_NOTIFY (WM_USER + 200)
   
10. #define RECV_BUFFER_SIZE 8192
    
11. 
    
12. class CATLSocketClient : public CWindowImpl<CATLSocketClient> {
    
13. public:
    
14.     DECLARE_WND_CLASS(NULL)
    
15. 
    
16.     BEGIN_MSG_MAP(CATLSocketClient)
    
17.         MESSAGE_HANDLER(WM_SOCKET_NOTIFY, OnSocketEvent)
    
18.     END_MSG_MAP()
    
19. 
    
20.     BOOL Connect(LPCTSTR szServer, int nPort);
    
21.     BOOL SendData(const CString& strData);
    
22.     void SetMessageWindow(HWND hWnd) { m_hMsgWnd = hWnd; }
    
23. 
    
24. private:
    
25.     LRESULT OnSocketEvent(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled);
    
26.     void ProcessReceivedData();
    
27.    
    
28.     SOCKET m_socket = INVALID_SOCKET;
    
29.     HWND m_hMsgWnd = NULL;
    
30.     std::vector<char> m_recvBuffer;
    
31.     std::mutex m_bufferMutex;
    
32. };
    

复制代码

ATLSocketClient.cpp

1. #include "stdafx.h"
   
2. #include "ATLSocketClient.h"
   
3. 
   
4. BOOL CATLSocketClient::Connect(LPCTSTR szServer, int nPort) {
   
5.     WSADATA wsa;
   
6.     if (WSAStartup(MAKEWORD(2,2), &wsa) != 0)
   
7.         return FALSE;
   
8. 
   
9.     m_socket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
   
10.     if (m_socket == INVALID_SOCKET)
    
11.         return FALSE;
    
12. 
    
13.     sockaddr_in serverAddr = {0};
    
14.     serverAddr.sin_family = AF_INET;
    
15.     serverAddr.sin_port = htons(nPort);
    
16.     InetPton(AF_INET, szServer, &serverAddr.sin_addr);
    
17. 
    
18.     if (connect(m_socket, (SOCKADDR*)&serverAddr, sizeof(serverAddr)) == SOCKET_ERROR) {
    
19.         closesocket(m_socket);
    
20.         return FALSE;
    
21.     }
    
22. 
    
23.     WSAAsyncSelect(m_socket, m_hWnd, WM_SOCKET_NOTIFY, FD_READ | FD_WRITE | FD_CLOSE);
    
24.     return TRUE;
    
25. }
    
26. 
    
27. BOOL CATLSocketClient::SendData(const CString& strData) {
    
28.     if (m_socket == INVALID_SOCKET) return FALSE;
    
29.    
    
30.     int nLen = strData.GetLength();
    
31.     int nSent = send(m_socket, CT2A(strData), nLen, 0);
    
32.     return nSent == nLen;
    
33. }
    
34. 
    
35. LRESULT CATLSocketClient::OnSocketEvent(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled) {
    
36.     if (WSAGETSELECTERROR(lParam)) {
    
37.         PostMessage(m_hMsgWnd, WM_SOCKET_NOTIFY, 0, (LPARAM)_T("Socket error"));
    
38.         return 0;
    
39.     }
    
40. 
    
41.     switch(WSAGETSELECTEVENT(lParam)) {
    
42.     case FD_READ: {
    
43.         char tempBuf[RECV_BUFFER_SIZE];
    
44.         int len = recv(m_socket, tempBuf, RECV_BUFFER_SIZE-1, 0);
    
45.         if (len > 0) {
    
46.             std::lock_guard<std::mutex> lock(m_bufferMutex);
    
47.             m_recvBuffer.insert(m_recvBuffer.end(), tempBuf, tempBuf + len);
    
48.             PostMessage(m_hMsgWnd, WM_SOCKET_NOTIFY, FD_READ, 0);
    
49.         }
    
50.         break;
    
51.     }
    
52.     case FD_CLOSE:
    
53.         PostMessage(m_hMsgWnd, WM_SOCKET_NOTIFY, FD_CLOSE, (LPARAM)_T("Connection closed"));
    
54.         break;
    
55.     }
    
56.     return 0;
    
57. }
    

复制代码

MainDlg.h

1. once
   
2. #include "ATLSocketClient.h"
   
3. 
   
4. class CMainDlg : public CDialogImpl<CMainDlg> {
   
5. public:
   
6.     enum { IDD = IDD_MAINDLG };
   
7. 
   
8.     BEGIN_MSG_MAP(CMainDlg)
   
9.         MESSAGE_HANDLER(WM_INITDIALOG, OnInitDialog)
   
10.         MESSAGE_HANDLER(WM_SOCKET_NOTIFY, OnSocketNotify)
    
11.         COMMAND_ID_HANDLER(IDC_BTN_CONNECT, OnBtnConnect)
    
12.         COMMAND_ID_HANDLER(IDC_BTN_SEND, OnBtnSend)
    
13.     END_MSG_MAP()
    
14. 
    
15.     LRESULT OnInitDialog(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled);
    
16.     LRESULT OnSocketNotify(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled);
    
17.     LRESULT OnBtnConnect(WORD wNotifyCode, WORD wID, HWND hWndCtl, BOOL& bHandled);
    
18.     LRESULT OnBtnSend(WORD wNotifyCode, WORD wID, HWND hWndCtl, BOOL& bHandled);
    
19. 
    
20. private:
    
21.     CATLSocketClient m_socketClient;
    
22. };
    

复制代码

MainDlg.cpp

1. "stdafx.h"
   
2. #include "MainDlg.h"
   
3. 
   
4. LRESULT CMainDlg::OnInitDialog(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled) {
   
5.     m_socketClient.Create(m_hWnd);
   
6.     m_socketClient.SetMessageWindow(m_hWnd);
   
7.     return TRUE;
   
8. }
   
9. 
   
10. LRESULT CMainDlg::OnSocketNotify(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled) {
    
11.     switch(wParam) {
    
12.     case FD_READ: {
    
13.         CString strRecv;
    
14.         {
    
15.             std::lock_guard<std::mutex> lock(m_socketClient.m_bufferMutex);
    
16.             strRecv = CA2T(m_socketClient.m_recvBuffer.data());
    
17.             m_socketClient.m_recvBuffer.clear();
    
18.         }
    
19.         CString strCurrent;
    
20.         GetDlgItemText(IDC_EDIT_RECV, strCurrent);
    
21.         SetDlgItemText(IDC_EDIT_RECV, strCurrent + _T("\r\n") + strRecv);
    
22.         break;
    
23.     }
    
24.     case FD_CLOSE:
    
25.         SetDlgItemText(IDC_EDIT_STATUS, (LPCTSTR)lParam);
    
26.         break;
    
27.     default:
    
28.         SetDlgItemText(IDC_EDIT_STATUS, _T("Unknown socket event"));
    
29.     }
    
30.     return 0;
    
31. }
    
32. 
    
33. LRESULT CMainDlg::OnBtnConnect(WORD wNotifyCode, WORD wID, HWND hWndCtl, BOOL& bHandled) {
    
34.     CString strServer;
    
35.     GetDlgItemText(IDC_EDIT_SERVER, strServer);
    
36.     int nPort = GetDlgItemInt(IDC_EDIT_PORT);
    
37.    
    
38.     if (m_socketClient.Connect(strServer, nPort)) {
    
39.         SetDlgItemText(IDC_EDIT_STATUS, _T("Connected successfully"));
    
40.     } else {
    
41.         SetDlgItemText(IDC_EDIT_STATUS, _T("Connection failed"));
    
42.     }
    
43.     return 0;
    
44. }
    
45. 
    
46. LRESULT CMainDlg::OnBtnSend(WORD wNotifyCode, WORD wID, HWND hWndCtl, BOOL& bHandled) {
    
47.     CString strSend;
    
48.     GetDlgItemText(IDC_EDIT_SEND, strSend);
    
49.     if (!strSend.IsEmpty()) {
    
50.         if (m_socketClient.SendData(strSend)) {
    
51.             SetDlgItemText(IDC_EDIT_STATUS, _T("Data sent successfully"));
    
52.         } else {
    
53.             SetDlgItemText(IDC_EDIT_STATUS, _T("Send failed"));
    
54.         }
    
55.     }
    
56.     return 0;
    
57. }
    

复制代码