|
MFC对话框(UNICODE环境下)实现SOCKET服务端源码示例
下面是将 `AfxBeginThread` 中的功能提取到一个单独函数中的改进代码。我们将创建一个新的成员函数 `AcceptClientConnections` ,并在 `OnBnClickedStart` 中调用它。这样可以使代码更加清晰和易于维护。
### 修改后的服务端代码 (Server.cpp)
- // Server.cpp : Defines the entry point for the application.
- #include "stdafx.h"
- #include "Server.h"
- #include <afxsock.h>
- #include <iostream>
- #include "ServerDlg.h"
- class CMyServerApp : public CWinApp
- {
- public:
- virtual BOOL InitInstance();
- };
- CMyServerApp theApp;
- BOOL CMyServerApp::InitInstance()
- {
- AfxSocketInit(); // 初始化Socket库
- CServerDlg dlg; // 创建对话框
- m_pMainWnd = &dlg; // 设置主窗口
- dlg.DoModal(); // 显示对话框
- return FALSE; // 退出应用程序
- }
- // 服务器对话框类
- class CServerDlg : public CDialogEx
- {
- public:
- CServerDlg(CWnd* pParent = nullptr); // 标准构造函数
- virtual BOOL OnInitDialog(); // 初始化对话框
- void OnBnClickedStart(); // 开始按钮处理函数
- void OnBnClickedStop(); // 停止按钮处理函数
- void AcceptClientConnections(); // 接受客户端连接的函数
- DECLARE_MESSAGE_MAP()
- private:
- CSocket serverSocket; // 服务器Socket
- BOOL isRunning; // 服务器运行状态
- };
- CServerDlg::CServerDlg(CWnd* pParent /*=nullptr*/)
- : CDialogEx(IDD_SERVER_DIALOG, pParent), isRunning(FALSE)
- {
- }
- BEGIN_MESSAGE_MAP(CServerDlg, CDialogEx)
- ON_BN_CLICKED(IDC_BUTTON_START, &CServerDlg::OnBnClickedStart)
- ON_BN_CLICKED(IDC_BUTTON_STOP, &CServerDlg::OnBnClickedStop)
- END_MESSAGE_MAP()
- BOOL CServerDlg::OnInitDialog()
- {
- CDialogEx::OnInitDialog();
- return TRUE; // 返回 TRUE,除非将焦点设置到控件
- }
- void CServerDlg::OnBnClickedStart()
- {
- if (!isRunning)
- {
- if (!serverSocket.Create(12345)) // 创建监听端口
- {
- AfxMessageBox(_T("无法创建服务器Socket"));
- return;
- }
- serverSocket.Listen(); // 开始监听
- isRunning = TRUE;
- AfxMessageBox(_T("服务器已启动"));
- // 创建线程来接受连接
- AfxBeginThread([](LPVOID pParam) -> UINT {
- CServerDlg* pDlg = (CServerDlg*)pParam;
- pDlg->AcceptClientConnections(); // 调用接受连接的函数
- return 0; // 线程结束
- }, (LPVOID)this); // 将对话框指针传递给线程
- }
- }
- void CServerDlg::AcceptClientConnections()
- {
- while (true)
- {
- CSocket* pClientSocket = new CSocket;
- if (serverSocket.Accept(*pClientSocket)) // 接受客户端连接
- {
- // 获取客户端IP地址
- sockaddr_in clientAddr;
- int addrLen = sizeof(clientAddr);
- pClientSocket->GetPeerName((sockaddr*)&clientAddr, &addrLen); // 获取客户端地址
- // 将IP地址转换为CString
- CStringA ipAddressA(inet_ntoa(clientAddr.sin_addr)); // 使用CStringA处理char*
- CString ipAddress(ipAddressA); // 转换为CString
- CString message;
- message.Format(_T("客户端连接: %s"), ipAddress); // 格式化输出信息
- AfxMessageBox(message); // 显示客户端IP地址
- // 处理客户端请求
- TCHAR buffer[1024];
- int nReceived = pClientSocket->Receive(buffer, sizeof(buffer) - sizeof(TCHAR)); // 接收数据
- buffer[nReceived / sizeof(TCHAR)] = '\0'; // 添加字符串结束符
- // 处理接收到的信息(使用MakeUpper转换为大写)
- CString responseMessage = _T("处理后的信息: ");
- CString receivedMessage(buffer); // 将接收到的内容转换为CString
- receivedMessage.MakeUpper(); // 将接收到的内容转换为大写
- responseMessage += receivedMessage; // 拼接处理后的信息
- pClientSocket->Send(responseMessage, responseMessage.GetLength() * sizeof(TCHAR)); // 发送处理后的信息
- pClientSocket->Close(); // 关闭客户端Socket
- }
- else
- {
- delete pClientSocket; // 释放内存
- }
- }
- }
- void CServerDlg::OnBnClickedStop()
- {
- if (isRunning)
- {
- serverSocket.Close(); // 关闭服务器Socket
- isRunning = FALSE;
- AfxMessageBox(_T("服务器已停止"));
- }
- }
复制代码
### 代码说明
1. **提取函数**:
- 创建了一个新的成员函数 `AcceptClientConnections()` ,用于处理客户端连接的逻辑。
- 在 `OnBnClickedStart()` 中,使用 `AfxBeginThread` 创建线程,并在其中调用 `AcceptClientConnections()` 函数。
2. **线程参数**:
- 将 `this` 指针传递给线程,以便在线程中调用对话框的成员函数。
### 编译和运行
1. 创建一个 MFC 对话框应用程序项目。
2. 将上述代码放入 `Server.cpp` 文件中。
3. 在资源文件中添加按钮并设置它们的 ID(例如 `IDC_BUTTON_START` 和 `IDC_BUTTON_STOP` )。
4. 编译并运行服务器应用程序。
### 注意事项
- 确保防火墙允许访问指定的端口(在这个例子中是 12345)。
- 运行该应用程序后,点击“开始”按钮以启动服务器,点击“停止”按钮以停止服务器。
- 处理逻辑仍然是将接收到的信息转换为大写,您可以根据需求修改处理逻辑。
|
|