serverdlg.cpp

关于ACE的书籍,对于学习ACE很有帮助,这里只有上册,下册过些时候在传.

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

#include "stdafx.h"
#include "Server.h"
#include "ServerDlg.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()


// CServerDlg 对话框




CServerDlg::CServerDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CServerDlg::IDD, pParent)
	, m_iTotalConns(0)
	, m_ePort(27015)
{
	m_ClientList = NULL;
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}

void CServerDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	DDX_Text(pDX, IDE_PORT, m_ePort);
}

BEGIN_MESSAGE_MAP(CServerDlg, CDialog)
	ON_WM_SYSCOMMAND()
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	//}}AFX_MSG_MAP
	ON_BN_CLICKED(IDB_SETUP, &CServerDlg::OnBnSetup)
	ON_MESSAGE(WM_SOCKET,&CServerDlg::OnSocketEvent) 
	ON_BN_CLICKED(IDB_SYSTEM_SEND, &CServerDlg::OnBnSystemSend)
END_MESSAGE_MAP()


// CServerDlg 消息处理程序

BOOL CServerDlg::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: 在此添加额外的初始化代码

	//SetDlgItemInt(IDE_PORT,27015);
	return TRUE;  // 除非将焦点设置到控件，否则返回 TRUE
}

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

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

void CServerDlg::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 CServerDlg::OnQueryDragIcon()
{
	return static_cast<HCURSOR>(m_hIcon);
}


void CServerDlg::OnBnSetup()
{
	WSADATA wsaData;
	SOCKET ListenSocket;
	sockaddr_in Server;
	CString strErr;

	int iResult = WSAStartup(MAKEWORD(2,2), &wsaData);
	if (iResult != NO_ERROR) {
		MessageBox(L"Error at WSAStartup()\n");
		return ;
	}
	ListenSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
	if (ListenSocket == INVALID_SOCKET) {
		strErr.Format(L"Error at socket(): %ld\n", WSAGetLastError());
		MessageBox(strErr);
		WSACleanup();
		return ;
	}

	Server.sin_family = AF_INET;
	Server.sin_addr.s_addr = inet_addr("127.0.0.1");
	Server.sin_port = htons(m_ePort);

	if (bind( ListenSocket, 
		(SOCKADDR*) &Server, 
		sizeof(Server)) == SOCKET_ERROR) {
			MessageBox(L"bind() failed.\n");
			closesocket(ListenSocket);
			WSACleanup();
			return ;
	}

	if (listen( ListenSocket, SOMAXCONN ) == SOCKET_ERROR) {
		MessageBox(L"Error listening on socket.\n");
		closesocket(ListenSocket);
		WSACleanup();
		return ;
	}

	if((WSAAsyncSelect(ListenSocket, this->m_hWnd, WM_SOCKET, FD_ACCEPT|FD_CLOSE))==SOCKET_ERROR){
		MessageBox(L"WSAAsyncSelect error");
		return ;
	}

	SetDlgItemText(IDE_MESSAGES,L"Setup Success!Waitting For Connectting``````\n");
	GetDlgItem(IDB_SYSTEM_SEND)->EnableWindow(TRUE);
	GetDlgItem(IDE_PORT)->EnableWindow(FALSE);
	GetDlgItem(IDB_SETUP)->EnableWindow(FALSE);

}

LRESULT CServerDlg::OnSocketEvent(WPARAM   wParam,LPARAM   lParam){
	CString strErr;
	CString strMsg;
	SOCKET	Accept;
	char	szBuff[DATA_BUFSIZE];
	char	*cBuffPos = szBuff;
	CClient *lpClient;
	CMsg	*ptrMsg = NULL;

	if (WSAGETSELECTERROR(lParam))
	{
		//FreeClientInformation(wParam);
		//strMsg.Format(L"%d/64",m_iTotalConns);
		//SetDlgItemText(IDS_CLIENTS,strMsg);
		PostMessage(WM_SOCKET,wParam,FD_CLOSE);
	} 
	else
	{
		switch(WSAGETSELECTEVENT(lParam))
		{
		case FD_ACCEPT:

			if ((Accept = accept(wParam, NULL, NULL)) == INVALID_SOCKET)
			{
				strErr.Format(L"accept() failed with error %d\n", WSAGetLastError());
				MessageBox(strErr);
				break;
			}
			CreateClientInformation(Accept);
			strMsg.Format(L"%d/64",m_iTotalConns);
			SetDlgItemText(IDS_CLIENTS,strMsg);
			WSAAsyncSelect(Accept, this->m_hWnd, WM_SOCKET, FD_READ|FD_WRITE|FD_CLOSE);
			break;

		case FD_READ:

			lpClient = GetClientInformation(wParam);

			//recv data
			ptrMsg = new CMsg;
			if (!ptrMsg)break;
			if(!lpClient->RecvMsg())break;
			if(!lpClient->DecodingBuff(ptrMsg)){
				lpClient->RecvBuffRest();
				break;
			}

			if(ptrMsg->MsgHead.eCmdID == CMD_MSG_CHAT){
				strMsg.Format(L"%s Say:\n%s\n",ptrMsg->MsgChat.szSender,ptrMsg->MsgChat.szContent);
			}
			else if (ptrMsg->MsgHead.eCmdID == CMD_LOGIN)
			{
				strMsg.Format(L"%s Login\n",ptrMsg->MsgLogin.szUsername);
				memcpy(lpClient->name,ptrMsg->MsgLogin.szUsername,NAME_LEN*2);
				delete ptrMsg;

				for(CClient *lpTClient = m_ClientList;lpTClient != NULL;lpTClient = lpTClient->Next){
					if (lpTClient == lpClient)continue;
					ptrMsg = new CMsg;
					if(!ptrMsg)break;
					ptrMsg->MsgHead.eCmdID = CMD_NEW;
					ptrMsg->MsgHead.dwToken = 0;
					ptrMsg->MsgHead.dwLength = sizeof CMD_ID + 2*sizeof DWORD+NAME_LEN*2;
					memcpy(ptrMsg->MsgNew.szUsername,lpTClient->name,NAME_LEN*2);
					lpClient->CodingBuff(ptrMsg);
					if(!lpClient->SendMsg()){
						PostMessage(WM_SOCKET,wParam,FD_CLOSE);
						delete ptrMsg;
						break;
					}
					delete ptrMsg;
					lpClient->SendBuffRest();
				}
				ptrMsg = new CMsg;
				if(!ptrMsg)break;
				ptrMsg->MsgHead.eCmdID = CMD_NEW;
				ptrMsg->MsgHead.dwToken = 0;
				ptrMsg->MsgHead.dwLength = sizeof CMD_ID + 2*sizeof DWORD+NAME_LEN*2;
				memcpy(ptrMsg->MsgNew.szUsername,lpClient->name,NAME_LEN*2);
			}

			for(CClient *lpTClient = m_ClientList;lpTClient != NULL;lpTClient = lpTClient->Next){
				if (lpTClient == lpClient)continue;
				lpTClient->CodingBuff(ptrMsg);
				if(!lpTClient->SendMsg()){
					FreeClientInformation(wParam);
				}
				lpTClient->SendBuffRest();
			}
			lpClient->RecvBuffRest();
			((CEdit*)GetDlgItem(IDE_MESSAGES))->SetSel(-1);
			((CEdit*)GetDlgItem(IDE_MESSAGES))->ReplaceSel(strMsg);
			break;
			// DO NOT BREAK HERE SINCE WE GOT A SUCCESSFUL RECV. Go ahead
			// and begin writing data to the client.

		case FD_WRITE:
			lpClient = GetClientInformation(wParam);
			lpClient->m_bWrite = TRUE;
			break;

		case FD_CLOSE:
			lpClient = GetClientInformation(wParam);
			strMsg.Format(L"%s Logout\n",lpClient->name);
			((CEdit*)GetDlgItem(IDE_MESSAGES))->SetSel(-1);
			((CEdit*)GetDlgItem(IDE_MESSAGES))->ReplaceSel(strMsg);

			ptrMsg =new CMsg;
			if(ptrMsg){
				ptrMsg->MsgHead.eCmdID = CMD_LOGOUT;
				ptrMsg->MsgHead.dwToken = 0;
				ptrMsg->MsgHead.dwLength = sizeof CMD_ID + 2*sizeof DWORD+NAME_LEN*2;
				memcpy(ptrMsg->MsgNew.szUsername,lpClient->name,NAME_LEN*2);

				FreeClientInformation(wParam);

				for(CClient *lpTClient = m_ClientList;lpTClient != NULL;lpTClient = lpTClient->Next){
					lpTClient->CodingBuff(ptrMsg);
					if(!lpTClient->SendMsg()){
						FreeClientInformation(lpTClient->m_Socket);
					}
					lpTClient->SendBuffRest();
				}
				delete ptrMsg;
			}
			
			strMsg.Format(L"%d/64",m_iTotalConns);
			SetDlgItemText(IDS_CLIENTS,strMsg);
			break;
		}
	}
	return 0;
}
void CServerDlg::CreateClientInformation(SOCKET s)
{
	CString strMsg;
	CClient *lpCI;

	if ((lpCI = new CClient()) == NULL)
	{
		strMsg.Format(L"GlobalAlloc() failed with error %d\n", GetLastError());
		MessageBox(strMsg);
		return;
	}
	
	// Prepare ClientInfo structure for use.

	lpCI->m_Socket = s;
	lpCI->m_bWrite = FALSE;
	lpCI->m_iMsgNum = 0; 
	lpCI->BytesRECV = 0;
	lpCI->BytesSEND = 0;
	m_iTotalConns++;
	//m_cMsgList.UpdateRefer();
	//m_cMsgList.UpdateClients(true);

	lpCI->Next = m_ClientList;
	m_ClientList = lpCI;
}

void CServerDlg::FreeClientInformation(SOCKET s)
{
	CClient *lpCI = m_ClientList;
	CClient *lpPrevCI = NULL;

	m_iTotalConns--;
	while(lpCI)
	{
		if (lpCI->m_Socket == s)
		{
			if (lpPrevCI)
				lpPrevCI->Next = lpCI->Next;
			else
				m_ClientList = lpCI->Next;

			closesocket(lpCI->m_Socket);
			delete lpCI;
			return;
		}

		lpPrevCI = lpCI;
		lpCI =lpCI->Next;
	}
}

CClient *CServerDlg::GetClientInformation(SOCKET s)
{
	CClient *lpCI = m_ClientList;

	while(lpCI)
	{
		if (lpCI->m_Socket == s)
			return lpCI;

		lpCI = lpCI->Next;
	}

	return NULL;
}

void CServerDlg::OnBnSystemSend()
{

	CString		strMsg;
	DWORD		strLen = 0;
	CMD_ID		MsgType = CMD_MSG_CHAT;
	DWORD		dwLen = 0;

	GetDlgItemText(IDE_SYS_MSG,strMsg);
	strLen = strMsg.GetLength();
	if (strLen <= 0)
	{
		//do not send empty msg
		return ;
	}

	CMsg sysMsg;
	sysMsg.MsgHead.eCmdID = CMD_MSG_CHAT;
	sysMsg.MsgHead.dwToken = 0;
	sysMsg.MsgHead.dwLength = sizeof CMD_ID+3*sizeof DWORD+NAME_LEN*2+(strLen+1)*2;

	memcpy(sysMsg.MsgChat.szSender,L"SYSTEM",sizeof (L"SYSTEM"));
	sysMsg.MsgChat.dwLength = (strLen+1)*2;
	sysMsg.MsgChat.szContent = new wchar_t[strLen+1];
	memcpy(sysMsg.MsgChat.szContent,strMsg,sysMsg.MsgChat.dwLength);
	for(CClient *lpTClient = m_ClientList;lpTClient != NULL;lpTClient = lpTClient->Next){
		lpTClient->CodingBuff(&sysMsg);
		if(!lpTClient->SendMsg()){
			FreeClientInformation(lpTClient->m_Socket);
		}
		lpTClient->SendBuffRest();
	}

	strMsg = L"SYSTEM:"+strMsg+L"\n";
	((CEdit*)GetDlgItem(IDE_MESSAGES))->SetSel(-1);
	((CEdit*)GetDlgItem(IDE_MESSAGES))->ReplaceSel(strMsg);
	SetDlgItemText(IDE_SYS_MSG,L"");
}