ATL对话框异步Socket通信实现，包含客户端/服务端完整交互功能

源码资源网

ATL对话框异步Socket通信实现，包含客户端/服务端完整交互功能
代码说明：1.完整实现客户端/服务端通信 2.使用自定义消息处理Socket事件 3.对话框实时显示通信内容 4.支持多客户端连接
ClientSocket.h

1. 
   
2. #pragma once
   
3. #include <atlbase.h>
   
4. #include <atlwin.h>
   
5. #include <atlcoll.h>
   
6. 
   
7. #define WM_SOCKET_MSG (WM_USER + 100)
   
8. 
   
9. class CClientSocket : public CAsyncSocket {
   
10. public:
    
11.     void SetMsgWindow(HWND hWnd) { m_hMsgWnd = hWnd; }
    
12.    
    
13.     BOOL ConnectToServer(LPCTSTR szAddress, UINT nPort) {
    
14.         if(!Create()) return FALSE;
    
15.         return Connect(szAddress, nPort);
    
16.     }
    
17. 
    
18.     void SendData(const CString& strData) {
    
19.         if(GetState() == connected) {
    
20.             Send(CT2A(strData), strData.GetLength());
    
21.         }
    
22.     }
    
23. 
    
24. protected:
    
25.     virtual void OnReceive(int nErrorCode) {
    
26.         if(nErrorCode == 0) {
    
27.             char szBuffer[4096];
    
28.             int nRead = Receive(szBuffer, sizeof(szBuffer)-1);
    
29.             if(nRead > 0 && m_hMsgWnd) {
    
30.                 szBuffer[nRead] = '\0';
    
31.                 ::PostMessage(m_hMsgWnd, WM_SOCKET_MSG, 0, (LPARAM)new CString(CA2T(szBuffer)));
    
32.             }
    
33.         }
    
34.         CAsyncSocket::OnReceive(nErrorCode);
    
35.     }
    
36. 
    
37. private:
    
38.     HWND m_hMsgWnd = NULL;
    
39. };
    

复制代码

ServerSocket.h

1. #pragma once
   
2. #include <atlbase.h>
   
3. 
   
4. class CServerSocket : public CAsyncSocket {
   
5. public:
   
6.     void StartServer(UINT nPort) {
   
7.         Create(nPort);
   
8.         Listen();
   
9.     }
   
10. 
    
11.     virtual void OnAccept(int nErrorCode) {
    
12.         if(nErrorCode == 0) {
    
13.             CClientSocket* pSocket = new CClientSocket;
    
14.             if(Accept(*pSocket)) {
    
15.                 pSocket->AsyncSelect(FD_READ | FD_CLOSE);
    
16.             }
    
17.         }
    
18.     }
    
19. };
    

复制代码

MainDlg.h

1. once
   
2. #include "ClientSocket.h"
   
3. #include "ServerSocket.h"
   
4. 
   
5. class CMainDlg : public CDialogImpl<CMainDlg> {
   
6. public:
   
7.     enum { IDD = IDD_MAINDLG };
   
8. 
   
9.     BEGIN_MSG_MAP(CMainDlg)
   
10.         MESSAGE_HANDLER(WM_INITDIALOG, OnInitDialog)
    
11.         COMMAND_ID_HANDLER(IDC_BTN_CONNECT, OnConnect)
    
12.         COMMAND_ID_HANDLER(IDC_BTN_SEND, OnSend)
    
13.         COMMAND_ID_HANDLER(IDC_BTN_START_SERVER, OnStartServer)
    
14.         MESSAGE_HANDLER(WM_SOCKET_MSG, OnSocketMessage)
    
15.     END_MSG_MAP()
    
16. 
    
17.     LRESULT OnInitDialog(UINT, WPARAM, LPARAM, BOOL&);
    
18.     LRESULT OnConnect(WORD, WORD, HWND, BOOL&);
    
19.     LRESULT OnSend(WORD, WORD, HWND, BOOL&);
    
20.     LRESULT OnStartServer(WORD, WORD, HWND, BOOL&);
    
21.     LRESULT OnSocketMessage(UINT, WPARAM, LPARAM, BOOL&);
    
22. 
    
23. private:
    
24.     void AppendMessage(LPCTSTR lpszMessage);
    
25.     CClientSocket m_clientSocket;
    
26.     CServerSocket m_serverSocket;
    
27. };
    

复制代码

MainDlg.cpp

1. #include "stdafx.h"
   
2. #include "MainDlg.h"
   
3. 
   
4. LRESULT CMainDlg::OnInitDialog(UINT, WPARAM, LPARAM, BOOL&) {
   
5.     CenterWindow();
   
6.     m_clientSocket.SetMsgWindow(m_hWnd);
   
7.     return TRUE;
   
8. }
   
9. 
   
10. void CMainDlg::AppendMessage(LPCTSTR lpszMessage) {
    
11.     CString strCurrent;
    
12.     GetDlgItemText(IDC_EDIT_RECV, strCurrent);
    
13.     strCurrent += lpszMessage;
    
14.     strCurrent += _T("\r\n");
    
15.     SetDlgItemText(IDC_EDIT_RECV, strCurrent);
    
16. }
    
17. 
    
18. LRESULT CMainDlg::OnSocketMessage(UINT, WPARAM, LPARAM lParam, BOOL&) {
    
19.     CString* pStr = reinterpret_cast<CString*>(lParam);
    
20.     if(pStr) {
    
21.         AppendMessage(*pStr);
    
22.         delete pStr;
    
23.     }
    
24.     return 0;
    
25. }
    
26. 
    
27. LRESULT CMainDlg::OnConnect(WORD, WORD, HWND, BOOL&) {
    
28.     CString strAddr;
    
29.     GetDlgItemText(IDC_EDIT_ADDRESS, strAddr);
    
30.     UINT nPort = GetDlgItemInt(IDC_EDIT_PORT);
    
31.    
    
32.     if(m_clientSocket.ConnectToServer(strAddr, nPort)) {
    
33.         AppendMessage(_T("连接服务器成功"));
    
34.     }
    
35.     return 0;
    
36. }
    
37. 
    
38. LRESULT CMainDlg::OnSend(WORD, WORD, HWND, BOOL&) {
    
39.     CString strMsg;
    
40.     GetDlgItemText(IDC_EDIT_MESSAGE, strMsg);
    
41.     m_clientSocket.SendData(strMsg);
    
42.     AppendMessage(_T("发送: ") + strMsg);
    
43.     return 0;
    
44. }
    
45. 
    
46. LRESULT CMainDlg::OnStartServer(WORD, WORD, HWND, BOOL&) {
    
47.     UINT nPort = GetDlgItemInt(IDC_EDIT_SERVER_PORT);
    
48.     if(m_serverSocket.StartServer(nPort)) {
    
49.         AppendMessage(_T("服务器启动成功，监听端口: ") + CString(nPort));
    
50.     }
    
51.     return 0;
    
52. }
    

复制代码