ATL对话框中接收Socket数据的完整实现方案

源码资源网

ATL对话框中接收Socket数据的完整实现方案
代码说明：1.使用WSAAsyncSelect实现异步通知 2.通过窗口消息传递数据 3.接收数据自动显示到IDC_EDIT_RECV控件 4.状态信息显示到IDC_EDIT_STATUS
ATLSocket.h

1. 
   
2. #pragma once
   
3. #include <atlbase.h>
   
4. #include <atlwin.h>
   
5. #include <winsock2.h>
   
6. 
   
7. #define WM_SOCKET_MSG (WM_USER + 100)
   
8. #define BUFFER_SIZE 4096
   
9. 
   
10. class CATLSocket : public CWindowImpl<CATLSocket> {
    
11. public:
    
12.     DECLARE_WND_CLASS(NULL)
    
13.    
    
14.     BEGIN_MSG_MAP(CATLSocket)
    
15.         MESSAGE_HANDLER(WM_SOCKET_MSG, OnSocketMessage)
    
16.     END_MSG_MAP()
    
17. 
    
18.     BOOL Init() {
    
19.         WSADATA wsa;
    
20.         return WSAStartup(MAKEWORD(2,2), &wsa) == 0;
    
21.     }
    
22. 
    
23.     void SetReceiver(HWND hWnd) { m_hReceiver = hWnd; }
    
24. 
    
25.     LRESULT OnSocketMessage(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled) {
    
26.         if (WSAGETSELECTERROR(lParam)) {
    
27.             PostMessage(m_hReceiver, WM_SOCKET_MSG, 0, (LPARAM)_T("Socket error"));
    
28.             return 0;
    
29.         }
    
30. 
    
31.         switch(WSAGETSELECTEVENT(lParam)) {
    
32.         case FD_READ: {
    
33.             char buffer[BUFFER_SIZE];
    
34.             int len = recv((SOCKET)wParam, buffer, BUFFER_SIZE-1, 0);
    
35.             if (len > 0) {
    
36.                 buffer[len] = '\0';
    
37.                 CString strMsg = CA2T(buffer);
    
38.                 PostMessage(m_hReceiver, WM_SOCKET_MSG, 1, (LPARAM)new CString(strMsg));
    
39.             }
    
40.             break;
    
41.         }
    
42.         case FD_CLOSE:
    
43.             PostMessage(m_hReceiver, WM_SOCKET_MSG, 0, (LPARAM)_T("Connection closed"));
    
44.             break;
    
45.         }
    
46.         return 0;
    
47.     }
    
48. 
    
49. private:
    
50.     HWND m_hReceiver = NULL;
    
51. };
    

复制代码

MainDlg.h

1. CMainDlg : public CDialogImpl<CMainDlg> {
   
2. public:
   
3.     enum { IDD = IDD_MAIN_DIALOG };
   
4. 
   
5.     BEGIN_MSG_MAP(CMainDlg)
   
6.         MESSAGE_HANDLER(WM_INITDIALOG, OnInitDialog)
   
7.         MESSAGE_HANDLER(WM_SOCKET_MSG, OnSocketMsg)
   
8.         COMMAND_ID_HANDLER(IDCANCEL, OnCancel)
   
9.     END_MSG_MAP()
   
10. 
    
11.     LRESULT OnInitDialog(UINT, WPARAM, LPARAM, BOOL&) {
    
12.         m_socket.Init();
    
13.         m_socket.Create(m_hWnd);
    
14.         m_socket.SetReceiver(m_hWnd);
    
15.         return TRUE;
    
16.     }
    
17. 
    
18.     LRESULT OnSocketMsg(UINT, WPARAM wParam, LPARAM lParam, BOOL&) {
    
19.         if (wParam == 1) { // 数据接收
    
20.             CString* pStr = (CString*)lParam;
    
21.             SetDlgItemText(IDC_EDIT_RECV, *pStr);
    
22.             delete pStr;
    
23.         } else { // 错误/关闭
    
24.             SetDlgItemText(IDC_EDIT_STATUS, (LPCTSTR)lParam);
    
25.         }
    
26.         return 0;
    
27.     }
    
28. 
    
29.     LRESULT OnCancel(WORD, WORD wID, HWND, BOOL&) {
    
30.         EndDialog(wID);
    
31.         return 0;
    
32.     }
    
33. 
    
34. private:
    
35.     CATLSocket m_socket;
    
36. };
    

复制代码