chatserverdlg.cpp

文件包含了很多VC实例

// ChatServerDlg.cpp : 实现文件
//

#include "stdafx.h"
#include <atlconv.h>
#include "ChatServer.h"
#include "ChatServerDlg.h"
#include ".\chatserverdlg.h"


#ifdef _DEBUG
#define new DEBUG_NEW
#endif

const int SOCK_TCP	= 0;
const int SOCK_UDP  = 1;

// 用于应用程序“关于”菜单项的 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()


// CChatServerDlg 对话框



CChatServerDlg::CChatServerDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CChatServerDlg::IDD, pParent)
	, m_nSocketType(SOCK_TCP) //默认为TCP socket
	, m_nStrPort(_T("5000")) //默认端口为 5000
{
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}

void CChatServerDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	DDX_Control(pDX, IDC_EDIT_MSGRER, m_ctrlMsgRecord);
	DDX_Control(pDX, IDC_EDIT_MSG, m_ctrlMsg);
	DDX_Control(pDX, IDC_SPINPORT, m_ctrlSpinPort);
	DDX_Text(pDX, IDC_EDITPORT, m_nStrPort);
}

BEGIN_MESSAGE_MAP(CChatServerDlg, CDialog)
	ON_WM_SYSCOMMAND()
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	//}}AFX_MSG_MAP
	ON_BN_CLICKED(IDSTART, OnBnClickedStart)
	ON_BN_CLICKED(IDSTOP, OnBnClickedStop)
	ON_BN_CLICKED(IDSEND, OnBnClickedSend)
	ON_BN_CLICKED(IDEXIT, OnBnClickedExit)
	ON_BN_CLICKED(IDC_RADIOTCP, OnBnClickedRadiotcp)
	ON_BN_CLICKED(IDC_RADIOUDP, OnBnClickedRadioudp)
	ON_WM_DESTROY()
	ON_BN_CLICKED(IDC_BUTTON1, OnBnClickedButton1)
END_MESSAGE_MAP()

// CChatServerDlg 消息处理程序

BOOL CChatServerDlg::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);		// 设置小图标

	//增加其他的初始化工作
	m_ctrlSpinPort.SetRange32( 2000, 6000);//默认变化范围为2000-6000
	GetDlgItem(IDSEND)->EnableWindow( FALSE );
	GetDlgItem(IDSTOP)->EnableWindow( FALSE );

	//设定最大连接的属性
	for(int i=0; i<MAX_CONNECTION; i++)
	{
		m_SocketManager[i].SetMessageWindow( &m_ctrlMsgRecord );
	}

	PickNextAvailable(); 
	//设置socket类型为TCP
	CButton* p = (CButton*)GetDlgItem(IDC_RADIOTCP);
	p->SetCheck(BST_CHECKED);
	return TRUE;  // 除非设置了控件的焦点，否则返回 TRUE
}

void CChatServerDlg::OnSysCommand(UINT nID, LPARAM lParam)
{
	if ((nID & 0xFFF0) == IDM_ABOUTBOX)
	{
		CAboutDlg dlgAbout;
		dlgAbout.DoModal();
	}
	else
	{
		CDialog::OnSysCommand(nID, lParam);
	}
}

// 如果向对话框添加最小化按钮，则需要下面的代码
//  来绘制该图标。对于使用文档/视图模型的 MFC 应用程序，
//  这将由框架自动完成。

void CChatServerDlg::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();
	}
}

///////////////////////////////////////////////////////////////////////////////
// OnUpdateConnection
// This message is sent by server manager to indicate connection status

//事件处理
LRESULT CChatServerDlg::OnUpdateConnection(WPARAM wParam, LPARAM lParam)
{
	UINT uEvent = (UINT) wParam;
	CSocketManager* pManager = reinterpret_cast<CSocketManager*>( lParam );

	// 如果不是TCP，则返回
	if (m_nSocketType != SOCK_TCP)
		return 0L;

	if ( pManager != NULL)
	{
		// 服务器socket已经连接上
		if (uEvent == EVT_CONSUCCESS)
		{
			PickNextAvailable();
			StartServer();
		}
		else if (uEvent == EVT_CONFAILURE || uEvent == EVT_CONDROP)
		{
			pManager->StopComm();
			if (m_pCurServer == NULL)
			{
				PickNextAvailable();
				StartServer();
			}
		}
	}

	return 1L;
}

//当用户拖动最小化窗口时系统调用此函数取得光标显示。
HCURSOR CChatServerDlg::OnQueryDragIcon()
{
	return static_cast<HCURSOR>(m_hIcon);
}

void CChatServerDlg::OnBnClickedStart()
{
	//数据更新
	UpdateData();
	StartServer();
}

void CChatServerDlg::OnBnClickedStop()
{
	// 断开服务器
	m_pCurServer->StopComm();
	if (!m_pCurServer->IsOpen())
	{
		GetDlgItem(IDSTART)->EnableWindow( TRUE );
		PrevDlgCtrl();
		GetDlgItem(IDSTOP)->EnableWindow( FALSE );
		GetDlgItem(IDC_RADIOTCP)->EnableWindow( TRUE );
		GetDlgItem(IDC_RADIOUDP)->EnableWindow( TRUE );
	}
}

void CChatServerDlg::OnBnClickedSend()
{
	// TODO: 在此添加控件通知处理程序代码
		//定义缓冲区
	BYTE byBuffer[256] = { 0 };
	CString strText;
	CString strSpace = " 说： ";
    CString str_add=" 服务器 ";
	// 更新数据
	UpdateData();

	//获得要发送的信息
	m_ctrlMsg.GetWindowText( strText );
	int nLen = strText.GetLength();

	if (nLen > 0)
	{
		//在消息前加消息发送人身份标识
		TCHAR strHost[HOSTNAME_SIZE] = { 0 };
		if (false == CSocketComm::GetLocalName( strHost, sizeof(strHost)/sizeof(TCHAR)))
		{
			AfxMessageBox("获取消息发送人身份标识失败!");
			return;
		}
		//在要发送信息的后面加上回车
		strText += _T("\r\n");
		nLen = strText.GetLength() + strlen(strHost) + strSpace.GetLength()+str_add.GetLength();
		
		
		//如果是UDP协议
		if (m_nSocketType == SOCK_UDP)
		{
			USES_CONVERSION;
			SockAddrIn sin;
			// 发送广播信息
			sin.CreateFrom(_T("255.255.255.255"), m_nStrPort);

			memcpy(byBuffer, (LPSOCKADDR) sin, sin.Size());
			//将要发送的信息复制到缓冲区
			strcpy((LPSTR)&byBuffer[sin.Size()], T2CA(strHost));
			strcat((LPSTR)&byBuffer[sin.Size()], T2CA(str_add));
			strcat((LPSTR)&byBuffer[sin.Size()], T2CA(strSpace));
			strcat((LPSTR)&byBuffer[sin.Size()], T2CA(strText));
			nLen =nLen+ sin.Size();
		}
		else
		{
			USES_CONVERSION;
			strcpy((LPSTR)byBuffer, T2CA(strHost));
			strcat((LPSTR)byBuffer, T2CA(str_add));
			strcat((LPSTR)byBuffer, T2CA(strSpace));
			strcat((LPSTR)byBuffer, T2CA(strText));
		}

		// 发送数据
		if (m_nSocketType == SOCK_UDP)
			m_pCurServer->WriteComm( byBuffer, nLen+1, INFINITE);
		else
		{
			// 发送到所有客户端
			for(int i=0; i<MAX_CONNECTION; i++)
			{
				if (m_SocketManager[i].IsOpen() && (m_pCurServer = &m_SocketManager[i]))
						m_SocketManager[i].WriteComm( byBuffer, nLen+1, INFINITE);
			}
		}
		m_ctrlMsg.SetWindowText("");
	}	
}

void CChatServerDlg::OnBnClickedExit()
{
	CDialog::OnOK();
}

void CChatServerDlg::OnBnClickedRadiotcp()
{
	// TODO: 在此添加控件通知处理程序代码
	m_nSocketType = SOCK_TCP;
}

void CChatServerDlg::OnBnClickedRadioudp()
{
	// TODO: 在此添加控件通知处理程序代码
	m_nSocketType = SOCK_UDP;
}


//获得下一个客户连接
void CChatServerDlg::PickNextAvailable()
{
	m_pCurServer = NULL;
	for(int i=0; i<MAX_CONNECTION; i++)
	{
		if (!(m_SocketManager[i].IsOpen()))
		{
			m_pCurServer = &m_SocketManager[i];
			break;
		}
	}
}


///////////////////////////////////////////////////////////////////////////////
// 启动服务器
bool CChatServerDlg::StartServer()
{
	bool bSuccess = false;
	//判断服务器是否为空
	if (m_pCurServer != NULL)
	{
		//如果是TCP连接
		if (m_nSocketType == SOCK_TCP)
		{
			// no smart addressing - we use connection oriented
			//不设定smart地址
			m_pCurServer->SetSmartAddressing( false );
			//创建TCP socket
			bSuccess = m_pCurServer->CreateSocket( m_nStrPort, AF_INET, SOCK_STREAM, SO_REUSEADDR); // TCP
		}
		//如果是UDP连接
		else
		{
			m_pCurServer->SetSmartAddressing( true );
			//创建UDP socket
			bSuccess = m_pCurServer->CreateSocket( m_nStrPort, AF_INET, SOCK_DGRAM, SO_BROADCAST); // UDP
		}
		//判断是否创建成功
		if (bSuccess && m_pCurServer->WatchComm())
		{
			//设定相关控件属性
			GetDlgItem(IDSEND)->EnableWindow( TRUE );
			GetDlgItem(IDSTOP)->EnableWindow( TRUE );
			NextDlgCtrl();
			GetDlgItem(IDSTART)->EnableWindow( FALSE );
			GetDlgItem(IDC_RADIOTCP)->EnableWindow( FALSE );
			GetDlgItem(IDC_RADIOUDP)->EnableWindow( FALSE );
			CString strServer, strAddr;
			//获得本地机器名
			m_pCurServer->GetLocalName( strServer.GetBuffer(256), 256);
			strServer.ReleaseBuffer();
			//获得本地IP地址
			m_pCurServer->GetLocalAddress( strAddr.GetBuffer(256), 256);
			strAddr.ReleaseBuffer();
			CString strMsg = _T("服务器: ") + strServer;
					strMsg += _T(", @IP地址: ") + strAddr;
					strMsg += _T(" 运行在端口 ") + m_nStrPort + CString("\r\n");
			m_pCurServer->AppendMessage( strMsg );
		}
	}

	return bSuccess;
}


void CChatServerDlg::OnDestroy()
{
	for(int i=0; i<MAX_CONNECTION; i++)
		m_SocketManager[i].StopComm();
	CDialog::OnDestroy();
}

////////////////////////////////////////////////////////////////////////////////
//  重载该消息以响应快捷键 Enter或 ESC 
BOOL CChatServerDlg::PreTranslateMessage(MSG* pMsg)
{
	if (pMsg->message == WM_KEYDOWN)
	{
		int nVirtKey = (int) pMsg->wParam;
		if (nVirtKey == VK_ESCAPE)
			return TRUE;
		if (nVirtKey == VK_RETURN && (GetFocus()->m_hWnd  == m_ctrlMsg.m_hWnd))
		{
			if (m_pCurServer->IsOpen())
				OnBnClickedSend();
			return TRUE;
		}
	}

	return CDialog::PreTranslateMessage(pMsg);
}

void CChatServerDlg::OnBnClickedButton1()
{
	// TODO: 在此添加控件通知处理程序代码
	CAboutDlg dlg;
	dlg.DoModal();
}