这个是看了孙鑫老师的视频之后,敲出来的代码,虽然老师的视频讲解的已经很好了,但是在vs里运行起来,会碰到一些令人头疼的问题,终于靠着强大的网络,将这些问题一个一个都解决了。
这个基于线程、socket和对话框的小的聊天程序,目的是在于理解多线程,熟悉socket,将其放在对话框中去练习。
后来,我又给对话框添加了响应回车键的事件,就是按回车键,即发送消息,就是像qq那样的,这个看起来很简单,但是实行起来,还得小费一番功夫的。
具体的代码和代码中比较重要的地方,我都做了注释。
源代码如下:
// ChatDlg.h : 头文件
//
#pragma once
#define WM_RECVDATA WM_USER+1
#define WM_RETURNDOWN WM_USER+2//自定义了一个消息,回车消息
struct RECVPARAM
{
SOCKET sock;
HWND hWnd;
};
// CChatDlg 对话框
class CChatDlg : public CDialog
{
// 构造
public:
CChatDlg(CWnd* pParent = NULL); // 标准构造函数
// 对话框数据
enum { IDD = IDD_CHAT_DIALOG };
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();
afx_msg LRESULT OnRecvData(WPARAM wParam,LPARAM lParam);
afx_msg LRESULT OnReturnDown(WPARAM wParam,LPARAM lParam);//声明了一个消息响应函数
DECLARE_MESSAGE_MAP()
public:
bool InitSocket(void);
private:
SOCKET m_socket;
public:
static DWORD WINAPI RecvProc(LPVOID lpParameter);
afx_msg void OnBnClickedBtnSend();
BOOL PreTranslateMessage(MSG *pMsg);
};
// ChatDlg.cpp : 实现文件
//
#include "stdafx.h"
#include "Chat.h"
#include "ChatDlg.h"
#ifdef _DEBUG
#define new DEBUG_NEW
#endif
// 用于应用程序“关于”菜单项的 CAboutDlg 对话框
class CAboutDlg : public CDialog
{
public:
CAboutDlg();
// 对话框数据
enum { IDD = IDD_ABOUTBOX };
protected:
virtual void DoDataExchange(CDataExchange* pDX); // DDX/DDV 支持
// 实现
protected:
DECLARE_MESSAGE_MAP()
};
CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
{
}
void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
CDialog::DoDataExchange(pDX);
}
BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
END_MESSAGE_MAP()
// CChatDlg 对话框
CChatDlg::CChatDlg(CWnd* pParent /*=NULL*/)
: CDialog(CChatDlg::IDD, pParent)
{
m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}
void CChatDlg::DoDataExchange(CDataExchange* pDX)
{
CDialog::DoDataExchange(pDX);
}
BEGIN_MESSAGE_MAP(CChatDlg, CDialog)
ON_WM_SYSCOMMAND()
ON_WM_PAINT()
ON_WM_QUERYDRAGICON()
//}}AFX_MSG_MAP
ON_MESSAGE(WM_RECVDATA,OnRecvData)
ON_MESSAGE(WM_RETURNDOWN,OnReturnDown)//进行消息和消息响应函数的关联
ON_BN_CLICKED(IDC_BTN_SEND, &CChatDlg::OnBnClickedBtnSend)
END_MESSAGE_MAP()
// CChatDlg 消息处理程序
BOOL CChatDlg::OnInitDialog()
{
CDialog::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: 在此添加额外的初始化代码
InitSocket();
//由于作为接收端程序,用recvfrom会处于阻塞状态,因为在这里要用线程
//在线程中要用到两个变量,一个是SOCKET,一个是窗口的句柄(或是编辑框的句柄)
//SOCKET用来接收消息,句柄用来将接收到的消息,进行格式转换后,显示在界面上
//由于CreateThread()函数只能接收一个变量,且是个指针型的变量,所以这里要用结构体
//结构体定义在"ChatDlg.h"中
RECVPARAM *pRecvParam=new RECVPARAM;
pRecvParam->sock=m_socket;
pRecvParam->hWnd=m_hWnd;
HANDLE hThread=CreateThread(NULL,0,RecvProc,(LPVOID)pRecvParam,0,NULL);
//RecvProc方法要声明成静态的,因为此时对象还没有创建
CloseHandle(hThread);
SetDlgItemText(IDC_IPADDRESS1,_T("127.0.0.1"));
return TRUE; // 除非将焦点设置到控件,否则返回 TRUE
}
void CChatDlg::OnSysCommand(UINT nID, LPARAM lParam)
{
if ((nID & 0xFFF0) == IDM_ABOUTBOX)
{
CAboutDlg dlgAbout;
dlgAbout.DoModal();
}
else
{
CDialog::OnSysCommand(nID, lParam);
}
}
// 如果向对话框添加最小化按钮,则需要下面的代码
// 来绘制该图标。对于使用文档/视图模型的 MFC 应用程序,
// 这将由框架自动完成。
void CChatDlg::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
{
CDialog::OnPaint();
}
}
//当用户拖动最小化窗口时系统调用此函数取得光标
//显示。
HCURSOR CChatDlg::OnQueryDragIcon()
{
return static_cast<HCURSOR>(m_hIcon);
}
bool CChatDlg::InitSocket(void)//初始化套接字,进行套接字的绑定
{
m_socket=socket(AF_INET,SOCK_DGRAM,0);
if(INVALID_SOCKET == m_socket)
{
AfxMessageBox("套接字创建失败!");
return false;
}
SOCKADDR_IN addrSock;
addrSock.sin_addr.S_un.S_addr=inet_addr("127.0.0.1");
addrSock.sin_family=AF_INET;
addrSock.sin_port=htons(9527);
int retval=bind(m_socket,(sockaddr*)&addrSock,sizeof(sockaddr));
if(retval==SOCKET_ERROR)
{
AfxMessageBox("绑定失败");
return false;
closesocket(m_socket);
}
return false;
}
DWORD WINAPI CChatDlg::RecvProc(LPVOID lpParameter)//线程函数,定义的是静态的
{
SOCKET sock=((RECVPARAM *)lpParameter)->sock;
HWND hWnd=((RECVPARAM *)lpParameter)->hWnd;
SOCKADDR_IN addrFrom;
int len=sizeof(SOCKADDR);
char recvBuf[100];
char tempBuf[200];
int retval;
while(true)
{
retval=recvfrom(sock,recvBuf,100,0,(sockaddr *)&addrFrom,&len);
if(retval==SOCKET_ERROR)
break;
sprintf_s(tempBuf,200,"%s 说:%s",inet_ntoa(addrFrom.sin_addr),recvBuf);
::PostMessage(hWnd,WM_RECVDATA,0,(LPARAM)tempBuf);
//将tempBuf通过自定义的WM_RECVDATA消息传递给hWnd
//在此处可以充分理解windows的消息映射机制
}
return 0;
}
LRESULT CChatDlg::OnRecvData(WPARAM wParam,LPARAM lParam)//自定义的消息映射函数
{
CString strNew=(char *)lParam;
CString strOld;
GetDlgItemText(IDC_EDIT_RECV,strOld);
strNew+="\r\n";
strOld+=strNew;
SetDlgItemText(IDC_EDIT_RECV,strOld);
return 0;
}
void CChatDlg::OnBnClickedBtnSend()
{
// TODO: Add your control notification handler code here
DWORD dwIP;
((CIPAddressCtrl *)GetDlgItem(IDC_IPADDRESS1))->GetAddress(dwIP);//得到ip控件里的ip地址
SOCKADDR_IN addrTo;
addrTo.sin_addr.S_un.S_addr=htonl(dwIP);
addrTo.sin_family=AF_INET;
addrTo.sin_port=htons(9527);
CString strSend;
GetDlgItemText(IDC_EDIT_SEND,strSend);
sendto(m_socket,strSend,strSend.GetLength()+1,0,(sockaddr *)&addrTo,sizeof(SOCKADDR));
GetDlgItem(IDC_EDIT_SEND)->SetFocus();//设置焦点
SetDlgItemText(IDC_EDIT_SEND,NULL);
}
BOOL CChatDlg::PreTranslateMessage(MSG *pMsg)//拦截消息,判断是否为回车键
{
if(pMsg->message>=WM_KEYFIRST && pMsg->message<=WM_KEYLAST)
{
if(pMsg->wParam==VK_RETURN && pMsg->message==WM_KEYDOWN)
OnReturnDown(0,0);
}
return CDialog::PreTranslateMessage(pMsg);
}
LRESULT CChatDlg::OnReturnDown(WPARAM wParam,LPARAM lParam)//具体的消息响应函数
{
HWND hwnd=::GetFocus();//获得当前焦点所在的控件的句柄
int ID=::GetDlgCtrlID(hwnd);//获得该控件句柄的ID
if(ID==IDC_EDIT_SEND)
{
OnBnClickedBtnSend();
}
return 0;
}
通过写这个小程序,我学会了以下一些东西:
1、线程的创建和在对话框中的使用,以及socket的练习和其在对话框中的使用,以及两者结合到一起到的使用。关键是在哪开线程呢?关键是在接收信息的时候开线程。那从什么时候开始接受消息呢?当然是在窗口创建好就开始接收信息了。所以要在OnInitDialog()函数里,创建线程,并且立即启动线程,开始接收。但由于此时窗口还没有创建好,还处于初始化的阶段,所以不能在此时调用线程函数,故只能将线程函数定义成静态的了。
2、自定义一个消息响应函数的流程:
1)先在头文件中定义一个消息,并且声明一个函数,在vs中,自定义的函数只能是下面这一种类型:
afx_msg LRESULT FunctionName(WPARAM wParam,LPARAM lParam);
2)在.cpp文件,即实现文件中,对消息和函数进行关联:ON_MESSAGE(WM_RETURNDOWN,FunctionName)
3)然后,就是实现具体的消息响应函数了
3、windows中消息的传递机制,说直白一点,就是以消息来驱动函数的执行,就像第2点说的,将一个消息和一个函数进行关联后,若这个消息产生,则会驱动与之相关联的函数的执行。
4、在对话框中,对编辑框中回车键的响应:
1)定义一个回车键的消息和一个与之相关联的响应函数。
2)通过覆盖对话框的PreTranslateMessage()方法,来拦截消息,对消息进行判断,如果是回车键消息,则引发与之相关联的响应函数,作出相应的操做