使用MFC Socket 发送结构体消息

我是在原博主的代码上修改而来，增加宽字符。

原博主地址：

http://blog.csdn.net/yejiansnake/archive/2008/03/13/2175778.aspx

 

 

 

 

学习MFC还没有运用过CSocket，所以照着学了一下。

再把一些东西整理了一下，让程序支持宽字符传输。

 

下面是客户端的主要代码：

// ChatLitClientDlg.cpp : 实现文件 // #include "stdafx.h" #include "ChatLitClient.h" #include "ChatLitClientDlg.h" #include "afxdialogex.h" #ifdef _DEBUG #define new DEBUG_NEW #endif // 用于应用程序“关于”菜单项的 CAboutDlg 对话框 class CAboutDlg : public CDialogEx { public: CAboutDlg(); // 对话框数据 enum { IDD = IDD_ABOUTBOX }; protected: virtual void DoDataExchange(CDataExchange* pDX); // DDX/DDV 支持 // 实现 protected: DECLARE_MESSAGE_MAP() }; CAboutDlg::CAboutDlg() : CDialogEx(CAboutDlg::IDD) { } void CAboutDlg::DoDataExchange(CDataExchange* pDX) { CDialogEx::DoDataExchange(pDX); } BEGIN_MESSAGE_MAP(CAboutDlg, CDialogEx) END_MESSAGE_MAP() // CChatLitClientDlg 对话框 CChatLitClientDlg::CChatLitClientDlg(CWnd* pParent /*=NULL*/) : CDialogEx(CChatLitClientDlg::IDD, pParent) { m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME); } void CChatLitClientDlg::DoDataExchange(CDataExchange* pDX) { CDialogEx::DoDataExchange(pDX); } BEGIN_MESSAGE_MAP(CChatLitClientDlg, CDialogEx) ON_WM_SYSCOMMAND() ON_WM_PAINT() ON_WM_QUERYDRAGICON() ON_BN_CLICKED(IDC_BUTTON_SEND, &CChatLitClientDlg::OnBnClickedButtonSend) END_MESSAGE_MAP() // CChatLitClientDlg 消息处理程序 BOOL CChatLitClientDlg::OnInitDialog() { CDialogEx::OnInitDialog(); // 将“关于...”菜单项添加到系统菜单中。 // IDM_ABOUTBOX 必须在系统命令范围内。 ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX); ASSERT(IDM_ABOUTBOX < 0xF000); CMenu* pSysMenu = GetSystemMenu(FALSE); if (pSysMenu != NULL) { BOOL bNameValid; CString strAboutMenu; bNameValid = strAboutMenu.LoadString(IDS_ABOUTBOX); ASSERT(bNameValid); if (!strAboutMenu.IsEmpty()) { pSysMenu->AppendMenu(MF_SEPARATOR); pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu); } } // 设置此对话框的图标。当应用程序主窗口不是对话框时，框架将自动 // 执行此操作 SetIcon(m_hIcon, TRUE); // 设置大图标 SetIcon(m_hIcon, FALSE); // 设置小图标 // TODO: 在此添加额外的初始化代码 return TRUE; // 除非将焦点设置到控件，否则返回 TRUE } void CChatLitClientDlg::OnSysCommand(UINT nID, LPARAM lParam) { if ((nID & 0xFFF0) == IDM_ABOUTBOX) { CAboutDlg dlgAbout; dlgAbout.DoModal(); } else { CDialogEx::OnSysCommand(nID, lParam); } } // 如果向对话框添加最小化按钮，则需要下面的代码 // 来绘制该图标。对于使用文档/视图模型的 MFC 应用程序， // 这将由框架自动完成。 void CChatLitClientDlg::OnPaint() { if (IsIconic()) { CPaintDC dc(this); // 用于绘制的设备上下文 SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0); // 使图标在工作区矩形中居中 int cxIcon = GetSystemMetrics(SM_CXICON); int cyIcon = GetSystemMetrics(SM_CYICON); CRect rect; GetClientRect(&rect); int x = (rect.Width() - cxIcon + 1) / 2; int y = (rect.Height() - cyIcon + 1) / 2; // 绘制图标 dc.DrawIcon(x, y, m_hIcon); } else { CDialogEx::OnPaint(); } } //当用户拖动最小化窗口时系统调用此函数取得光标 //显示。 HCURSOR CChatLitClientDlg::OnQueryDragIcon() { return static_cast<HCURSOR>(m_hIcon); } void CChatLitClientDlg::OnBnClickedButtonSend() { // TODO: 在此添加控件通知处理程序代码 AfxSocketInit(); CSocket aSocket; CString strIP; CString strPort; CString strText; this->GetDlgItem(IDC_EDIT_IP)->GetWindowText(strIP); this->GetDlgItem(IDC_EDIT_PORT)->GetWindowText(strPort); this->GetDlgItem(IDC_EDIT_TEXT)->GetWindowText(strText); if(!aSocket.Create()) { TCHAR szMsg[1024] = {0}; wsprintf(szMsg, L"create faild: %d", aSocket.GetLastError()); AfxMessageBox(szMsg); return; } int nPort = _ttoi(strPort); if(aSocket.Connect(strIP, nPort)) { TCHAR szRecValue[1024] = {0}; // DWORD dwNum = WideCharToMultiByte(CP_OEMCP,NULL,strText,-1,NULL,0,NULL,FALSE); char *psText; psText = new char[dwNum]; if(!psText) { delete []psText; } WideCharToMultiByte (CP_OEMCP,NULL,strText,-1,psText,dwNum,NULL,FALSE); // aSocket.Send(psText, dwNum); delete []psText; aSocket.Receive((void *)szRecValue, 1024); AfxMessageBox(szRecValue); } else { TCHAR szMsg[1024] = {0}; wsprintf(szMsg, L"create faild: %d", aSocket.GetLastError()); AfxMessageBox(szMsg); } aSocket.Close(); }

下面是服务器主要代码：

// CharLitServerDlg.cpp : 实现文件 // #include "stdafx.h" #include "CharLitServer.h" #include "CharLitServerDlg.h" #include "afxdialogex.h" #ifdef _DEBUG #define new DEBUG_NEW #endif CWinThread *pThread = NULL; bool m_exit = false; // 用于应用程序“关于”菜单项的 CAboutDlg 对话框 class CAboutDlg : public CDialogEx { public: CAboutDlg(); // 对话框数据 enum { IDD = IDD_ABOUTBOX }; protected: virtual void DoDataExchange(CDataExchange* pDX); // DDX/DDV 支持 // 实现 protected: DECLARE_MESSAGE_MAP() }; CAboutDlg::CAboutDlg() : CDialogEx(CAboutDlg::IDD) { } void CAboutDlg::DoDataExchange(CDataExchange* pDX) { CDialogEx::DoDataExchange(pDX); } BEGIN_MESSAGE_MAP(CAboutDlg, CDialogEx) END_MESSAGE_MAP() // CCharLitServerDlg 对话框 CCharLitServerDlg::CCharLitServerDlg(CWnd* pParent /*=NULL*/) : CDialogEx(CCharLitServerDlg::IDD, pParent) { m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME); } void CCharLitServerDlg::DoDataExchange(CDataExchange* pDX) { CDialogEx::DoDataExchange(pDX); } BEGIN_MESSAGE_MAP(CCharLitServerDlg, CDialogEx) ON_WM_SYSCOMMAND() ON_WM_PAINT() ON_WM_QUERYDRAGICON() ON_BN_CLICKED(IDC_BTN_OPEN, &CCharLitServerDlg::OnBnClickedBtnOpen) ON_BN_CLICKED(IDC_BTN_CLOSE, &CCharLitServerDlg::OnBnClickedBtnClose) END_MESSAGE_MAP() // CCharLitServerDlg 消息处理程序 BOOL CCharLitServerDlg::OnInitDialog() { CDialogEx::OnInitDialog(); // 将“关于...”菜单项添加到系统菜单中。 // IDM_ABOUTBOX 必须在系统命令范围内。 ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX); ASSERT(IDM_ABOUTBOX < 0xF000); CMenu* pSysMenu = GetSystemMenu(FALSE); if (pSysMenu != NULL) { BOOL bNameValid; CString strAboutMenu; bNameValid = strAboutMenu.LoadString(IDS_ABOUTBOX); ASSERT(bNameValid); if (!strAboutMenu.IsEmpty()) { pSysMenu->AppendMenu(MF_SEPARATOR); pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu); } } // 设置此对话框的图标。当应用程序主窗口不是对话框时，框架将自动 // 执行此操作 SetIcon(m_hIcon, TRUE); // 设置大图标 SetIcon(m_hIcon, FALSE); // 设置小图标 // TODO: 在此添加额外的初始化代码 return TRUE; // 除非将焦点设置到控件，否则返回 TRUE } void CCharLitServerDlg::OnSysCommand(UINT nID, LPARAM lParam) { if ((nID & 0xFFF0) == IDM_ABOUTBOX) { CAboutDlg dlgAbout; dlgAbout.DoModal(); } else { CDialogEx::OnSysCommand(nID, lParam); } } // 如果向对话框添加最小化按钮，则需要下面的代码 // 来绘制该图标。对于使用文档/视图模型的 MFC 应用程序， // 这将由框架自动完成。 void CCharLitServerDlg::OnPaint() { if (IsIconic()) { CPaintDC dc(this); // 用于绘制的设备上下文 SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0); // 使图标在工作区矩形中居中 int cxIcon = GetSystemMetrics(SM_CXICON); int cyIcon = GetSystemMetrics(SM_CYICON); CRect rect; GetClientRect(&rect); int x = (rect.Width() - cxIcon + 1) / 2; int y = (rect.Height() - cyIcon + 1) / 2; // 绘制图标 dc.DrawIcon(x, y, m_hIcon); } else { CDialogEx::OnPaint(); } } //当用户拖动最小化窗口时系统调用此函数取得光标 //显示。 HCURSOR CCharLitServerDlg::OnQueryDragIcon() { return static_cast<HCURSOR>(m_hIcon); } unsigned int StartServer(LPVOID lParam) { if (!AfxSocketInit()) { AfxMessageBox(_T("Failed!")); return 1; } m_exit = false; CCharLitServerDlg *aDlg = (CCharLitServerDlg *)lParam; CString strPort; aDlg->GetDlgItemText(IDC_EDIT_PORT, strPort); UINT nPort = _ttoi(strPort); //socket------------------------------------------------ CSocket aSocket, serverSocket; if (!aSocket.Socket()) { TCHAR szError[256] = {0}; wsprintf(szError, L"Create Faild: %d", GetLastError()); AfxMessageBox(szError); return 1; } BOOL bOptVal = TRUE; int bOptLen = sizeof(BOOL); aSocket.SetSockOpt(SO_REUSEADDR, (void *)&bOptVal, bOptLen, SOL_SOCKET); if (!aSocket.Bind(nPort)) { TCHAR szError[256] = {0}; wsprintf(szError, L"Bind Faild: %d", GetLastError()); AfxMessageBox(szError); return 1; } if(!aSocket.Listen(10)) { TCHAR szError[256] = {0}; wsprintf(szError, L"Listen Faild: %d", GetLastError()); AfxMessageBox(szError); return 1; } CString strText; aDlg->GetDlgItemText(IDC_EDIT_CHATLOG, strText); strText += "Server Start! /r/n"; aDlg->SetDlgItemText(IDC_EDIT_CHATLOG, strText); while(!m_exit) { if(!aSocket.Accept(serverSocket)) { continue; } else { char szRecvMsg[1024] = {0}; TCHAR szOutMsg[256] = {0}; serverSocket.Receive(szRecvMsg, 1024); // DWORD dwNum = MultiByteToWideChar (CP_ACP, 0, szRecvMsg, -1, NULL, 0); wchar_t *pwText; pwText = new wchar_t[dwNum]; if(!pwText) { delete []pwText; } MultiByteToWideChar (CP_ACP, 0, szRecvMsg, -1, pwText, dwNum); // wsprintf(szOutMsg, L"Receive Msg: %s /r/n", pwText); delete []pwText; aDlg->GetDlgItemText(IDC_EDIT_CHATLOG, strText); strText += szOutMsg; aDlg->SetDlgItemText(IDC_EDIT_CHATLOG, strText); serverSocket.Send(L"Have Receive The Msg", wcslen(L"Have Receive The Msg")*2); serverSocket.Close(); } } aSocket.Close(); serverSocket.Close(); aDlg->GetDlgItemText(IDC_EDIT_CHATLOG, strText); strText += "Have Close!"; aDlg->SetDlgItemText(IDC_EDIT_CHATLOG, strText); return 0; } void StopServer() { m_exit = true; } void CCharLitServerDlg::SetDlgStatus(bool isOpen) { this->GetDlgItem(IDC_BTN_OPEN)->EnableWindow(!isOpen); this->GetDlgItem(IDC_BTN_CLOSE)->EnableWindow(isOpen); this->GetDlgItem(IDC_EDIT_PORT)->EnableWindow(!isOpen); } void CCharLitServerDlg::OnBnClickedBtnOpen() { // TODO: 在此添加控件通知处理程序代码 pThread = new CWinThread(StartServer, (LPVOID)this); pThread->CreateThread(CREATE_SUSPENDED); pThread->ResumeThread(); SetDlgStatus(true); } void CCharLitServerDlg::OnBnClickedBtnClose() { // TODO: 在此添加控件通知处理程序代码 StopServer(); pThread->SuspendThread(); delete pThread; pThread = NULL; SetDlgStatus(false); }

工程在：

http://download.csdn.net/source/2988928

 

续：

感觉这样传送消息不是很舒服，因为可能会出现复杂的通信，光传送文本显然不是很合适。我就试着传输结构体类型的数据过去，最初我使用了CString作为结构体的文本内容容器，一直无法实现，看了很多相关内容的讲解，我想应该是CString本身就包含了很多变长信息导致每次都失败。最后使用了下面的方法测试成功，下次再进行更全面的补充。

首先结构体声明如下：

struct stMSG { TCHAR msg[256]; int msgSize; }; 

发送端这样修改代码：

stMSG newMGS; memset(&newMGS, 0, sizeof(newMGS)); memcpy(&newMGS.msg, strText.GetBuffer(), sizeof(newMGS.msg)); newMGS.msgSize = sizeof(newMGS); aSocket.Send(&newMGS, newMGS.msgSize); aSocket.Receive((void *)szRecValue, 1024);

服务端这样修改代码：

stMSG *recvMSG = (stMSG*)&szRecvMsg; strText = recvMSG->msg;

大致就实现了。

不足之处望见谅。