chatserverdlg.cpp

c++ 图表的设计

// ChatServerDlg.cpp : implementation file
//

#include "stdafx.h"
#include "ChatServer.h"
#include "ChatServerDlg.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

/////////////////////////////////////////////////////////////////////////////
// CAboutDlg dialog used for App About

class CAboutDlg : public CDialog
{
public:
	CAboutDlg();

// Dialog Data
	//{{AFX_DATA(CAboutDlg)
	enum { IDD = IDD_ABOUTBOX };
	//}}AFX_DATA

	// ClassWizard generated virtual function overrides
	//{{AFX_VIRTUAL(CAboutDlg)
	protected:
	virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
	//}}AFX_VIRTUAL

// Implementation
protected:
	//{{AFX_MSG(CAboutDlg)
	//}}AFX_MSG
	DECLARE_MESSAGE_MAP()
};

CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
{
	//{{AFX_DATA_INIT(CAboutDlg)
	//}}AFX_DATA_INIT
}

void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CAboutDlg)
	//}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
	//{{AFX_MSG_MAP(CAboutDlg)
		// No message handlers
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CChatServerDlg dialog

CChatServerDlg::CChatServerDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CChatServerDlg::IDD, pParent)
{
	//{{AFX_DATA_INIT(CChatServerDlg)
		// NOTE: the ClassWizard will add member initialization here
	//}}AFX_DATA_INIT
	// Note that LoadIcon does not require a subsequent DestroyIcon in Win32
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);


	m_svr=INVALID_SOCKET;
}

void CChatServerDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CChatServerDlg)
	DDX_Control(pDX, IDC_MSG, m_editMsg);
	DDX_Control(pDX, IDC_LIST_MSG, m_listMessage);
	DDX_Control(pDX, IDC_LIST_USER, m_listUser);
	//}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CChatServerDlg, CDialog)
	//{{AFX_MSG_MAP(CChatServerDlg)
	ON_WM_SYSCOMMAND()
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	ON_BN_CLICKED(ID_START_SVR, OnStartSvr)
	ON_MESSAGE(WM_MSG_DATA,OnSocketData)
	ON_BN_CLICKED(ID_MSG_TO_ALL, OnMsgToAll)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CChatServerDlg message handlers

BOOL CChatServerDlg::OnInitDialog()
{
	CDialog::OnInitDialog();

	// Add "About..." menu item to system menu.

	// IDM_ABOUTBOX must be in the system command range.
	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);
		}
	}

	// Set the icon for this dialog.  The framework does this automatically
	//  when the application's main window is not a dialog
	SetIcon(m_hIcon, TRUE);			// Set big icon
	SetIcon(m_hIcon, FALSE);		// Set small icon
	
	// TODO: Add extra initialization here
	InitCtls();
	return TRUE;  // return TRUE  unless you set the focus to a control
}

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

// If you add a minimize button to your dialog, you will need the code below
//  to draw the icon.  For MFC applications using the document/view model,
//  this is automatically done for you by the framework.

void CChatServerDlg::OnPaint() 
{
	if (IsIconic())
	{
		CPaintDC dc(this); // device context for painting

		SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);

		// Center icon in client rectangle
		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;

		// Draw the icon
		dc.DrawIcon(x, y, m_hIcon);
	}
	else
	{
		CDialog::OnPaint();
	}
}

// The system calls this to obtain the cursor to display while the user drags
//  the minimized window.
HCURSOR CChatServerDlg::OnQueryDragIcon()
{
	return (HCURSOR) m_hIcon;
}

void CChatServerDlg::OnCancel()
{
	CloseSocket();
	CDialog::OnCancel();
}

LRESULT CChatServerDlg::OnSocketData(WPARAM w,LPARAM l)
{
	char data[1024];
	memset(data,0,1024);
	SOCKADDR_IN addr;
	int size=sizeof(SOCKADDR_IN);
	int ir=::recvfrom(m_svr,data,1024,0,(sockaddr*)&addr,&size);
	if(ir==SOCKET_ERROR || ir==0)
		return 0;
	else
	{
		char* pData=data+1;
		switch(data[0])
		{
		case EM_MSG:
			OnMsgData(pData-1,ir);
			break;
		case EM_REG_ID:
			OnRegData(pData,&addr);
			break;
		case EM_DEL_ID:
			OnDelData(pData);
			break;
		case EM_USERS:
			OnUserListData(&addr);
			break;
		default:;
		}
	}
	
	return 0;
}

void CChatServerDlg::OnUserListData(SOCKADDR_IN* pAddr)
{
typedef list<StUser> ListUser;
	if(!pAddr)
		return;

	ListUser::iterator ite=m_listUserInfo.begin();

	int nCount=m_listUserInfo.size();
	int m,n;
	m=nCount/255;
	n=nCount%255;
	char* pData=new char[1+1+255*8];
	for (int i=0;i<m;i++)
	{
		memset(pData,0,1+1+255*8);

		pData[0]=EM_USERS;
		pData[1]=(char)255;
		for (int j=0;j<255;j++)
		{
			StUser st=*ite;
			memcpy(pData+2+j*8,st.name,8);

			++ite;
		}
		::sendto(m_svr,pData,1+1+255*8,0,(sockaddr*)pAddr,sizeof(SOCKADDR_IN));
		
	}

	memset(pData,0,1+1+255*8);

	pData[0]=EM_USERS;
	pData[1]=n;

	for (i=0;i<n;i++)
	{
		StUser st=*ite;
		memcpy(pData+2+i*8,st.name,8);

		++ite;
	}
	::sendto(m_svr,pData,1+1+n*8,0,(sockaddr*)pAddr,sizeof(SOCKADDR_IN));
	delete [] pData;
}

void CChatServerDlg::OnMsgData(char* pData,int nLen)
{
	if(!pData)
		return;

	char from[9],to[9];
	memset(from,0,9);
	memset(to,0,9);
	memcpy(from,pData+1,8);
	memcpy(to,pData+1+8,8);
	CString strMsg;
	strMsg.Format(_T("%s"),pData+1+8+8);
	

	//send msg to reciver
	StUser st;
	if(FindUser(from,st))
	{		
		if(FindUser(to,st))
		{
			::sendto(m_svr,pData,nLen,0,(sockaddr*)&st.addr,sizeof(SOCKADDR_IN));
			AddMsg(from,to,strMsg);
		}
	}
}

void CChatServerDlg::OnRegData(char* pData,SOCKADDR_IN* pAddr)
{
#define REG_OK 0
#define REG_FAILED 1

	if (!pData || !pAddr)
	{
		return;
	}
	else
	{
		StUser st;
		char data[2]={EM_REG_ID,REG_OK};
		if(!FindUser(pData,st))
		{
			strcpy(st.name,pData);
			memcpy(&st.addr,pAddr,sizeof(SOCKADDR_IN));
			m_listUserInfo.push_back(st);

			m_listUser.InsertItem(0,pData);

			ListUser::iterator ite;

			char data[10];
			memset(data,0,10);
			data[0]=EM_USERS;
			data[1]=1;
			memcpy(data+2,pData,8);
			for (ite=m_listUserInfo.begin();ite!=m_listUserInfo.end();++ite)
			{
				StUser st=*ite;
				::sendto(m_svr,data,10,0,(sockaddr*)&st.addr,sizeof(SOCKADDR_IN));	
			}
		}
		else
		{
			data[1]=REG_FAILED;
		}
		::sendto(m_svr,data,2,0,(sockaddr*)pAddr,sizeof(SOCKADDR_IN));
	}
}

void CChatServerDlg::OnDelData(char* pData)
{
	if (!pData)
	{
		return;
	}
	StUser st;
	if(FindUser(pData,st,TRUE))
	{
		LVFINDINFO info;
		info.flags = LVFI_PARTIAL|LVFI_STRING;
		info.psz=pData;
		int index=index=m_listUser.FindItem(&info);
		if(index!=-1)
		{
			m_listUser.DeleteItem(index);
		}

		
		ListUser::iterator ite;
		char data[9];
		memset(data,0,9);
		data[0]=EM_USERS_EXIT;
		memcpy(data+1,pData,8);
		for (ite=m_listUserInfo.begin();ite!=m_listUserInfo.end();++ite)
		{
			StUser st=*ite;
			::sendto(m_svr,data,9,0,(sockaddr*)&st.addr,sizeof(SOCKADDR_IN));	
		}
	}
}

bool CChatServerDlg::FindUser(char* name,StUser& refSt,BOOL remove/*=FALSE*/)
{
	ListUser::iterator ite;
	for (ite=m_listUserInfo.begin();ite!=m_listUserInfo.end();++ite)
	{
		StUser st=*ite;
		if(0==strcmp(st.name,name))
		{
			memcpy(&refSt,&st,sizeof(StUser));
			if(remove)
			{
				m_listUserInfo.erase(ite);
			}
			return true;
		}
	}
	return false;
}

void CChatServerDlg::AddMsg(CString from,CString to,CString msg)
{
	int nCount=m_listMessage.GetItemCount();
	int index=m_listMessage.InsertItem(nCount,from);
	m_listMessage.SetItemText(index,1,to);
	m_listMessage.SetItemText(index,2,msg);
}

void CChatServerDlg::InitCtls()
{
	m_listUser.DeleteAllItems();
	m_listUser.InsertColumn(0,_T("用户"),LVCFMT_LEFT,100);

	m_listMessage.DeleteAllItems();
	m_listMessage.InsertColumn(0,_T("From"),LVCFMT_LEFT,100);
	m_listMessage.InsertColumn(1,_T("To"),LVCFMT_LEFT,100);

	CRect rc;
	m_listMessage.GetClientRect(rc);
	m_listMessage.InsertColumn(2,_T("Message"),LVCFMT_LEFT,rc.Width()-200);

	m_listMessage.SetExtendedStyle(LVS_EX_HEADERDRAGDROP|LVS_EX_GRIDLINES|LVS_EX_FULLROWSELECT);
}

void CChatServerDlg::OnStartSvr() 
{
	// TODO: Add your control notification handler code here
	m_svr=::socket(AF_INET,SOCK_DGRAM,0);
	if(m_svr==INVALID_SOCKET)
	{
		AfxMessageBox(_T("启动服务失败:")+this->GetLastErrorMsg());
	}

	memset(&m_svrAddr,0,sizeof(SOCKADDR_IN));
	m_svrAddr.sin_family=AF_INET;
	m_svrAddr.sin_addr.s_addr=inet_addr("127.0.0.1");
	m_svrAddr.sin_addr.S_un.S_addr=inet_addr("127.0.0.1");
	m_svrAddr.sin_port=htons(16000);

	int ir=::bind(m_svr,(SOCKADDR*)&m_svrAddr,sizeof(SOCKADDR_IN));
	if(ir==SOCKET_ERROR)
	{
		AfxMessageBox(_T("启动服务失败:")+this->GetLastErrorMsg());
		CloseSocket();
	}
	::WSAAsyncSelect(m_svr,this->GetSafeHwnd(),WM_MSG_DATA,FD_READ);
}

CString CChatServerDlg::GetLastErrorMsg()
{
	int ir=::WSAGetLastError();
	CString errMsg;
	errMsg.Format(_T("错误信息:%d"),ir);
	return errMsg;
}


void CChatServerDlg::CloseSocket()
{
	if(m_svr!=INVALID_SOCKET)
		::closesocket(m_svr);
}

void CChatServerDlg::OnMsgToAll() 
{
typedef list<StUser> ListUser;

	if(m_svr==INVALID_SOCKET)
	{
		AfxMessageBox(_T("服务尚未开启!"));
	}
	int nLen=m_editMsg.GetWindowTextLength();
	if(nLen>64 || nLen<1)
	{
		AfxMessageBox(_T("消息太长或者太短"));
	}
	else
	{
		char msgData[128];
		memset(msgData,0,128);
		msgData[0]=EM_ALL_MSG;
		m_editMsg.GetWindowText(msgData+1,128-1);

		
		ListUser::iterator ite;
		for (ite=m_listUserInfo.begin();ite!=m_listUserInfo.end();++ite)
		{
			StUser st=*ite;
			::sendto(m_svr,msgData,128,0,(sockaddr*)&st.addr,sizeof(SOCKADDR_IN));
		}
	}
}