ATL服务端实现方案，可确保客户端正确接收FD_READ和FD_CLOSE事件

源码资源网

ATL服务端实现方案，可确保客户端正确接收FD_READ和FD_CLOSE事件
该实现包含完整的消息处理机制，关键点包括：1) 使用WSAAsyncSelect注册网络事件 2) 正确处理FD_ACCEPT/FD_READ/FD_CLOSE事件 3) 错误处理机制。使用时需确保窗口已创建并正确关联消息映射。
ATLSocketServer.h

1. 
   
2. #pragma once
   
3. #include <winsock2.h>
   
4. #include <atlbase.h>
   
5. #include <atlwin.h>
   
6. 
   
7. class CATLSocketServer : public CWindowImpl<CATLSocketServer> {
   
8. public:
   
9.     DECLARE_WND_CLASS(_T("ATL_SOCKET_SERVER"))
   
10. 
    
11.     BEGIN_MSG_MAP(CATLSocketServer)
    
12.         MESSAGE_HANDLER(WM_SOCKET_NOTIFY, OnSocketEvent)
    
13.         MESSAGE_HANDLER(WM_DESTROY, OnDestroy)
    
14.     END_MSG_MAP()
    
15. 
    
16.     BOOL Start(int nPort);
    
17.     LRESULT OnSocketEvent(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled);
    
18.     LRESULT OnDestroy(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled);
    
19. 
    
20. private:
    
21.     SOCKET m_listenSocket = INVALID_SOCKET;
    
22.     SOCKET m_clientSocket = INVALID_SOCKET;
    
23. };
    

复制代码

ATLSocketServer.cpp

1. "stdafx.h"
   
2. #include "ATLSocketServer.h"
   
3. 
   
4. BOOL CATLSocketServer::Start(int nPort) {
   
5.     WSADATA wsa;
   
6.     if (WSAStartup(MAKEWORD(2,2), &wsa) != 0)
   
7.         return FALSE;
   
8. 
   
9.     m_listenSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
   
10.     if (m_listenSocket == INVALID_SOCKET)
    
11.         return FALSE;
    
12. 
    
13.     sockaddr_in serverAddr = {0};
    
14.     serverAddr.sin_family = AF_INET;
    
15.     serverAddr.sin_port = htons(nPort);
    
16.     serverAddr.sin_addr.s_addr = INADDR_ANY;
    
17. 
    
18.     if (bind(m_listenSocket, (SOCKADDR*)&serverAddr, sizeof(serverAddr)) == SOCKET_ERROR) {
    
19.         closesocket(m_listenSocket);
    
20.         return FALSE;
    
21.     }
    
22. 
    
23.     listen(m_listenSocket, SOMAXCONN);
    
24.     WSAAsyncSelect(m_listenSocket, m_hWnd, WM_SOCKET_NOTIFY, FD_ACCEPT);
    
25.     return TRUE;
    
26. }
    
27. 
    
28. LRESULT CATLSocketServer::OnSocketEvent(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled) {
    
29.     SOCKET s = (SOCKET)wParam;
    
30.     int event = WSAGETSELECTEVENT(lParam);
    
31.     int error = WSAGETSELECTERROR(lParam);
    
32. 
    
33.     if (error != 0) {
    
34.         closesocket(s);
    
35.         return 0;
    
36.     }
    
37. 
    
38.     switch (event) {
    
39.     case FD_ACCEPT: {
    
40.         sockaddr_in clientAddr;
    
41.         int addrLen = sizeof(clientAddr);
    
42.         m_clientSocket = accept(s, (SOCKADDR*)&clientAddr, &addrLen);
    
43.         WSAAsyncSelect(m_clientSocket, m_hWnd, WM_SOCKET_NOTIFY, FD_READ | FD_CLOSE);
    
44.         break;
    
45.     }
    
46.     case FD_READ: {
    
47.         char buffer[1024];
    
48.         int bytesRead = recv(s, buffer, sizeof(buffer), 0);
    
49.         if (bytesRead > 0) {
    
50.             // 处理接收到的数据
    
51.             OutputDebugStringA(buffer);
    
52.         }
    
53.         break;
    
54.     }
    
55.     case FD_CLOSE:
    
56.         closesocket(s);
    
57.         if (s == m_clientSocket) m_clientSocket = INVALID_SOCKET;
    
58.         break;
    
59.     }
    
60.     return 0;
    
61. }
    
62. 
    
63. LRESULT CATLSocketServer::OnDestroy(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled) {
    
64.     if (m_listenSocket != INVALID_SOCKET) closesocket(m_listenSocket);
    
65.     if (m_clientSocket != INVALID_SOCKET) closesocket(m_clientSocket);
    
66.     WSACleanup();
    
67.     return 0;
    
68. }
    

复制代码