MFC--网络聊天软件设计--孙鑫--vs2008

MFC--网络聊天软件设计，是孙鑫老师教材的例程，自己在vs2008实现，最终实现局域网中两台PC通信；

代码

// chat.h : PROJECT_NAME 应用程序的主头文件
//

#pragma once

#ifndef __AFXWIN_H__
	#error "在包含此文件之前包含“stdafx.h”以生成 PCH 文件"
#endif

#include "resource.h"		// 主符号


// CchatApp:
// 有关此类的实现，请参阅 chat.cpp
//

class CchatApp : public CWinApp
{
public:
	CchatApp();

// 重写
	public:
	virtual BOOL InitInstance();

// 实现

	DECLARE_MESSAGE_MAP()
};

extern CchatApp theApp;

// chat.cpp : 定义应用程序的类行为。
//

#include "stdafx.h"
#include "chat.h"
#include "chatDlg.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#endif


// CchatApp

BEGIN_MESSAGE_MAP(CchatApp, CWinApp)
	ON_COMMAND(ID_HELP, &CWinApp::OnHelp)
END_MESSAGE_MAP()


// CchatApp 构造

CchatApp::CchatApp()
{
	// TODO: 在此处添加构造代码，
	// 将所有重要的初始化放置在 InitInstance 中
}


// 唯一的一个 CchatApp 对象

CchatApp theApp;


// CchatApp 初始化

BOOL CchatApp::InitInstance()
{
    //加载套接字库
	if(!AfxSocketInit())
	{
		AfxMessageBox(_T("加载套接字库失败！"));
		return FALSE;
	}
	
	
	// 如果一个运行在 Windows XP 上的应用程序清单指定要
	// 使用 ComCtl32.dll 版本 6 或更高版本来启用可视化方式，
	//则需要 InitCommonControlsEx()。否则，将无法创建窗口。
	INITCOMMONCONTROLSEX InitCtrls;
	InitCtrls.dwSize = sizeof(InitCtrls);
	// 将它设置为包括所有要在应用程序中使用的
	// 公共控件类。
	InitCtrls.dwICC = ICC_WIN95_CLASSES;
	InitCommonControlsEx(&InitCtrls);

	CWinApp::InitInstance();

	AfxEnableControlContainer();

	// 标准初始化
	// 如果未使用这些功能并希望减小
	// 最终可执行文件的大小，则应移除下列
	// 不需要的特定初始化例程
	// 更改用于存储设置的注册表项
	// TODO: 应适当修改该字符串，
	// 例如修改为公司或组织名
	SetRegistryKey(_T("应用程序向导生成的本地应用程序"));

	CchatDlg dlg;
	m_pMainWnd = &dlg;
	INT_PTR nResponse = dlg.DoModal();
	if (nResponse == IDOK)
	{
		// TODO: 在此放置处理何时用
		//  “确定”来关闭对话框的代码
	}
	else if (nResponse == IDCANCEL)
	{
		// TODO: 在此放置处理何时用
		//  “取消”来关闭对话框的代码
	}

	// 由于对话框已关闭，所以将返回 FALSE 以便退出应用程序，
	//  而不是启动应用程序的消息泵。
	return FALSE;
}

// chatDlg.h : 头文件
//

#pragma once


// CchatDlg 对话框

#define	WM_RECVDATA		WM_USER+1   //自定义消息

struct RECVPARAM
{
	SOCKET sock;
	HWND hwnd;
};

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();
//-----------------by my -----------
	// 自定义消息映射函数
    afx_msg LRESULT OnRecvData(WPARAM wParam,LPARAM lParam);  

	DECLARE_MESSAGE_MAP()

public:
	afx_msg void OnBnClickedBtnSend();

//-----------------by my -----------
public:
	BOOL InitSocket();   //创建套接字
	static DWORD WINAPI RecvProc(LPVOID lpParameter); //接收数据 显示

private:
	SOCKET m_socket;  //套接字

};

// 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

//-----------by my------------- 
    ON_MESSAGE(WM_RECVDATA,OnRecvData)   // 自定义消息与函数的映射
	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)
	{
		CString strAboutMenu;
		strAboutMenu.LoadString(IDS_ABOUTBOX);
		if (!strAboutMenu.IsEmpty())
		{
			pSysMenu->AppendMenu(MF_SEPARATOR);
			pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
		}
	}

	// 设置此对话框的图标。当应用程序主窗口不是对话框时，框架将自动
	//  执行此操作
	SetIcon(m_hIcon, TRUE);			// 设置大图标
	SetIcon(m_hIcon, FALSE);		// 设置小图标

	// TODO: 在此添加额外的初始化代码
//-----------by my------------- 
	
	InitSocket();//创建套接字

	RECVPARAM *pRecvParam=new RECVPARAM;
	pRecvParam->sock=m_socket;
	pRecvParam->hwnd=m_hWnd;
    //创建线程--接收数据
	HANDLE hThread=CreateThread(NULL,0,RecvProc,
		                        (LPVOID)pRecvParam,0,NULL);
	CloseHandle(hThread);

	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);
}


// TODO: 在此添加额外的代码
//-----------by my-------------

BOOL CchatDlg::InitSocket()      //创建套接字   
{
	m_socket=socket(AF_INET,SOCK_DGRAM,0);  //UDP
	if(INVALID_SOCKET==m_socket)
	{
		MessageBox(_T("套接字创建失败！"));
		return FALSE;
	}

	SOCKADDR_IN addrSock;
	addrSock.sin_family=AF_INET;
	addrSock.sin_port=htons(8800);
	addrSock.sin_addr.S_un.S_addr=htonl(INADDR_ANY);

	int retval;
	retval=bind(m_socket,(SOCKADDR*)&addrSock,sizeof(SOCKADDR));  //绑定
	if(SOCKET_ERROR==retval)
	{
		closesocket(m_socket);
		MessageBox(_T("绑定失败!"));
		return FALSE;
	}

	return TRUE;
}


//-----------接收数据部分--辅线程-----------
DWORD WINAPI CchatDlg::RecvProc(LPVOID lpParameter)
{
	SOCKET sock=((RECVPARAM*)lpParameter)->sock;
	HWND hwnd=((RECVPARAM*)lpParameter)->hwnd;

	delete lpParameter;	  

	SOCKADDR_IN addrFrom;     //接收端的地址信息
	int len=sizeof(SOCKADDR);

	char recvBuf[200];
	char tempBuf[300];
	int retval;

	while(TRUE)
	{
		retval=recvfrom(sock,recvBuf,200,0,(SOCKADDR*)&addrFrom,&len);
		if(SOCKET_ERROR==retval)
			break;
		sprintf(tempBuf,"%s说: %s",inet_ntoa(addrFrom.sin_addr),recvBuf);
		::PostMessage(hwnd,WM_RECVDATA,0,(LPARAM)tempBuf);              //产生自定义消息
	}
	return 0;
}

//自定义消息响应函数
LRESULT CchatDlg::OnRecvData(WPARAM wParam,LPARAM lParam)
{
	
	CString str;
	str=(char*)lParam;
	

   	CString strTemp;
	GetDlgItemText(IDC_EDIT_RECV,strTemp);
	str+="\r\n";
	str+=strTemp;
	SetDlgItemText(IDC_EDIT_RECV,str);

	return 0;
}



//-----------发送数据部分-------------
void CchatDlg::OnBnClickedBtnSend()
{
	// TODO: 在此添加控件通知处理程序代码
	DWORD dwIP;
	((CIPAddressCtrl*)GetDlgItem(IDC_IPADDRESS1))->GetAddress(dwIP);//获取Ip

	SOCKADDR_IN addrTo;                    
	addrTo.sin_family=AF_INET;
	addrTo.sin_port=htons(8800);
	addrTo.sin_addr.S_un.S_addr=htonl(dwIP);

	CString strSend;
	GetDlgItemText(IDC_EDIT_SEND,strSend);   //获取发送编辑框信息

	//-----UniCode 下CString 转 char*------------
	 //获取宽字节字符的大小，大小是按字节计算的
	 int len = WideCharToMultiByte(CP_ACP,0,strSend,strSend.GetLength(),NULL,0,NULL,NULL);  
	 //宽字节编码转换成多字节编码
	 char buffer[200];
	 WideCharToMultiByte(CP_ACP,0,strSend,-1,buffer,len+1,NULL,NULL);

	 sendto(m_socket,buffer,200,0,(SOCKADDR*)&addrTo,sizeof(SOCKADDR));


	SetDlgItemText(IDC_EDIT_SEND,_T(""));
}