上图是程序界面
另外,因为是基于UNICODE,所以发中文也没有问题。
准备工作:
首先应当在CXXXApp::InitInstance()中放入以下代码来实现socket的版本协商(相当于WSAStartup,并且还会自动调用WSACleanUp)。
XXX是自己取的项目名称。
if (!AfxSocketInit())
{
AfxMessageBox(_T("Failed to Initialize Sockets"), MB_OK | MB_ICONSTOP);
return FALSE;
}
不过在这之前还要在stdafx.h中加入#include <afxsock.h>
并且有了这个头文件就不需要去链接ws2_32.lib了。
准备工作做好了之后就可以开始编程了。
套接字初始化工作:
首先在类CXXXDlg中声明一个函数InitSocket()用于初始化一个成员变量:SOCKET m_socket。
在初始化过程中要把套接字与一个地址及端口绑定起来(bind)。
接收数据:
创建一个线程来执行接收数据的过程,因为如果用主线程去接收数据的话,那么接收数据时会一直等待导致程序无法继续运行下去。
不过线程过程函数只能接受一个参数,而接受数据的过程要求两个参数(套接字和窗口句柄)。因此可以通过传结构体指针的方式来实现。因为过程函数的参数是一个指针类型,所以可以通过强制类型转换来实现传送结构体指针。
先在CXXXDlg.h中声明一个结构体叫RECVPARAM,然后加入两个成员变量,一个SOCKET类型,一个HWND类型。
然后在CXXXDlg::OnInitDialog()中动态申请一块该结构体的内存,再将其指针传入过程函数,不过传入的时候要把它类型强制转换成(LPVOID)类型,然后在 过程函数里面将其转换回来即可。
每次接受到一个数据都要发送一个消息来让其显示到对话框上去。因此我们要自定义一个消息WM_RACVDATA,然后在消息响应函数中将新收到的数据与原有的数据一起显示到对话框上。自定义消息的消息响应函数要带两个数据WPARAM和LPARAM。
如果说线程的过程函数作为CXXXDlg的成员函数,那么要在函数前static作为静态函数。
当按下发送按钮时:
要首先获得IP地址控件上的IP地址信息,这个时候要用到CIPAddressCtrl类中的成员函数GetAddress。获得方法可以先通过GetDlgItem来获得该窗口的句柄的指针,然后将其强制转换成CIPAddressCtrl类的指针再调用其成员函数。
获得IP后还要获得编辑窗口的文本,可以调用GetDlgItemText来获取。
然后就可以发送数据了,发送完了要把编辑框的文本给清除掉。
发送接收数据的字符格式问题:
因为recvform和sendto中的发送/接收参数的是char *类型的(sendto是const char*)而VS2017默认是使用Unicode字符集,所以弄不好会出现乱码。不过没关系,在发送/接收的过程中可以直接通过强制类型转换来换成char *,在处理数据的时候再转回WCHAR即可(若换成CString类型,那么在做运算符+的过程中就自动转换成WCHAR类型了)。
// CharRoomDlg.h: 头文件
//
#pragma once
#define WM_RECVDATA WM_USER+1001 //添加自定义消息,小于WM_USER的为系统消息,大于的为自定义消息
typedef struct _RECVPARAM
{
SOCKET socket;
HWND hwnd;
}RECVPARAM;
// CCharRoomDlg 对话框
class CCharRoomDlg : public CDialogEx
{
// 构造
public:
CCharRoomDlg(CWnd* pParent = nullptr); // 标准构造函数
// 对话框数据
#ifdef AFX_DESIGN_TIME
enum { IDD = IDD_CHARROOM_DIALOG };
#endif
protected:
virtual void DoDataExchange(CDataExchange* pDX); // DDX/DDV 支持
// 实现
protected:
HICON m_hIcon;
// 生成的消息映射函数
virtual BOOL OnInitDialog();
afx_msg void OnSysCommand(UINT nID, LPARAM lParam);
afx_msg void OnPaint();
afx_msg HCURSOR OnQueryDragIcon();
DECLARE_MESSAGE_MAP()
public:
// 初始化socket
bool InitSocket();
// 接受数据线程函数
static DWORD WINAPI RecvProc(LPVOID lpParameter);
afx_msg LRESULT OnRecvData(WPARAM wParam, LPARAM lParam); //要加LRESULT
afx_msg void OnBnClickedButton1();
private:
SOCKET m_socket;
};
// CharRoomDlg.cpp: 实现文件
//
#include "stdafx.h"
#include "CharRoom.h"
#include "CharRoomDlg.h"
#include "afxdialogex.h"
#ifdef _DEBUG
#define new DEBUG_NEW
#endif
// 系统自动生成的用于应用程序“关于”菜单项的 CAboutDlg 对话框 的代码这里删除掉了
// CCharRoomDlg 对话框
CCharRoomDlg::CCharRoomDlg(CWnd* pParent /*=nullptr*/)
: CDialogEx(IDD_CHARROOM_DIALOG, pParent)
{
m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}
void CCharRoomDlg::DoDataExchange(CDataExchange* pDX)
{
CDialogEx::DoDataExchange(pDX);
}
BEGIN_MESSAGE_MAP(CCharRoomDlg, CDialogEx)
ON_WM_SYSCOMMAND()
ON_WM_PAINT()
ON_WM_QUERYDRAGICON()
ON_MESSAGE(WM_RECVDATA, OnRecvData) //添加消息映射,将自定义消息与消息函数映射起来,声明响应函数时要加LRESULT
ON_BN_CLICKED(IDC_BUTTON1, &CCharRoomDlg::OnBnClickedButton1)
END_MESSAGE_MAP()
// CCharRoomDlg 消息处理程序
BOOL CCharRoomDlg::OnInitDialog()
{
CDialogEx::OnInitDialog();
// 将“关于...”菜单项添加到系统菜单中。
// IDM_ABOUTBOX 必须在系统命令范围内。
ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
ASSERT(IDM_ABOUTBOX < 0xF000);
CMenu* pSysMenu = GetSystemMenu(FALSE);
if (pSysMenu != nullptr)
{
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: 在此添加额外的初始化代码
InitSocket();
RECVPARAM *pRecvParam = new RECVPARAM;
pRecvParam->hwnd = m_hWnd; //每一个对话框的类都有一个窗口的句柄
pRecvParam->socket = m_socket;
HANDLE hThread = CreateThread(NULL, 0, RecvProc, (LPVOID)pRecvParam, 0, NULL);
CloseHandle(hThread);
return TRUE; // 除非将焦点设置到控件,否则返回 TRUE
}
void CCharRoomDlg::OnSysCommand(UINT nID, LPARAM lParam)
{
if ((nID & 0xFFF0) == IDM_ABOUTBOX)
{
CAboutDlg dlgAbout;
dlgAbout.DoModal();
}
else
{
CDialogEx::OnSysCommand(nID, lParam);
}
}
// 如果向对话框添加最小化按钮,则需要下面的代码
// 来绘制该图标。 对于使用文档/视图模型的 MFC 应用程序,
// 这将由框架自动完成。
void CCharRoomDlg::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 CCharRoomDlg::OnQueryDragIcon()
{
return static_cast<HCURSOR>(m_hIcon);
}
// 初始化socket
bool CCharRoomDlg::InitSocket()
{
// TODO: 在此处添加实现代码.
m_socket = socket(AF_INET, SOCK_DGRAM, 0);
if (m_socket == INVALID_SOCKET) {
MessageBox(_T("Failed to creat Sockets! "));
return false;
}
SOCKADDR_IN addrSrv;
addrSrv.sin_addr.S_un.S_addr = htonl(INADDR_ANY);
addrSrv.sin_family = AF_INET;
addrSrv.sin_port = htons(6000);
int retval;
retval = bind(m_socket, (SOCKADDR*)&addrSrv, sizeof(SOCKADDR));
if (retval == SOCKET_ERROR) {
MessageBox(_T("Failed to bind! "));
return false;
}
return true;
}
// 接受数据线程函数
DWORD WINAPI CCharRoomDlg::RecvProc(LPVOID lpParameter)
{
// TODO: 在此处添加实现代码.
SOCKET socket = ((RECVPARAM*)lpParameter)->socket;
HWND hwnd = ((RECVPARAM*)lpParameter)->hwnd;
SOCKADDR_IN addrFrom;
addrFrom.sin_addr.S_un.S_addr = htonl(INADDR_ANY);
addrFrom.sin_family = AF_INET;
addrFrom.sin_port = htons(6000);
int len = sizeof(SOCKADDR);
WCHAR recvbuf[200];
//char tempbuf[300];
CString tempbuf;
int retval;
while (true)
{
retval = recvfrom(socket, (char *)recvbuf, 201, 0, (SOCKADDR*)&addrFrom, &len);
if (retval == SOCKET_ERROR)
break;
tempbuf.Empty();//清空数据
tempbuf += inet_ntoa(addrFrom.sin_addr);
tempbuf = tempbuf + _T(" say:\r\n") + recvbuf;
//sprintf(tempbuf, "%s say:\n %s", inet_ntoa(addrFrom.sin_addr), recvbuf);
//tempbuf.Format(_T("%s say:\n %s"), winet_ntoa(addrFrom.sin_addr), recvbuf);
::PostMessage(hwnd, WM_RECVDATA, 0, (LPARAM)(LPCTSTR)tempbuf); //为ChatRoom对话框发送RECVDATA消息,给消息带上lParam参数(teampbuf)。用全局函数要加上::
}
return 0;
}
LRESULT CCharRoomDlg::OnRecvData(WPARAM wParam, LPARAM lParam)
{
CString display = (WCHAR *)lParam;
CString temp;
GetDlgItemText(IDC_EDIT_RECV, temp);//存下接受数据的控件中现有的数据
temp += display;
temp += _T("\r\n");
SetDlgItemText(IDC_EDIT_RECV, temp);
return 0;
}
void CCharRoomDlg::OnBnClickedButton1()
{
// TODO: 在此添加控件通知处理程序代码
DWORD dwIpAddr;
((CIPAddressCtrl*)GetDlgItem(IDC_IPADDRESS1))->GetAddress(dwIpAddr);
SOCKADDR_IN addrTo;
addrTo.sin_addr.S_un.S_addr = htonl(dwIpAddr);
addrTo.sin_family = AF_INET;
addrTo.sin_port = htons(6000);
CString sendstr;
GetDlgItemText(IDC_EDIT_SEND, sendstr);
//USES_CONVERSION;
//char *sendbuf = T2A(sendstr);
sendto(m_socket, (const char*)sendstr.GetBuffer(), 201, 0, (SOCKADDR*)&addrTo, sizeof(SOCKADDR));
SetDlgItemText(IDC_EDIT_SEND, _T(""));//清除编辑框的文本
}