chatserverdoc.cpp

利用CSocket类编写的一个网络聊天程序,加深对CSocket类的基本应用的认识.

// ChatServerDoc.cpp : implementation of the CChatServerDoc class
//

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

#include "ChatServerDoc.h"

#include "MainFrm.h"
#include"PortDlg.h"
#include "ChattersListView.h"
#include "ChatView.h"
#include "ListeningSocket.h"
#include "ClientSocket.h"
#include "Msg.h"

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

/////////////////////////////////////////////////////////////////////////////
// CChatServerDoc

IMPLEMENT_DYNCREATE(CChatServerDoc, CDocument)

BEGIN_MESSAGE_MAP(CChatServerDoc, CDocument)
	//{{AFX_MSG_MAP(CChatServerDoc)
	ON_COMMAND(ID_SETPORT, OnSetport)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

#define LEAVING_CHAT          1
#define SENDING_CHATTERS_LIST 2
#define SENDING_NICKNAME      3
#define NORMAL_MESSAGE        4
#define USED_NAME			  5
/////////////////////////////////////////////////////////////////////////////
// CChatServerDoc construction/destruction

CChatServerDoc::CChatServerDoc()
{
	// TODO: add one-time construction code here
	m_pSocket = NULL;
	m_msgList.RemoveAll();
	m_connectionList.RemoveAll();
	bIsNewChatter = TRUE;
	m_ChattersList = "";
}

CChatServerDoc::~CChatServerDoc()
{
}

BOOL CChatServerDoc::OnNewDocument()
{
	if (!CDocument::OnNewDocument())
		return FALSE;

	// TODO: add reinitialization code here
	// (SDI documents will reuse this document)

	return TRUE;
}



/////////////////////////////////////////////////////////////////////////////
// CChatServerDoc serialization

void CChatServerDoc::Serialize(CArchive& ar)
{
	if (ar.IsStoring())
	{
		// TODO: add storing code here
	}
	else
	{
		// TODO: add loading code here
	}
}

/////////////////////////////////////////////////////////////////////////////
// CChatServerDoc diagnostics

#ifdef _DEBUG
void CChatServerDoc::AssertValid() const
{
	CDocument::AssertValid();
}

void CChatServerDoc::Dump(CDumpContext& dc) const
{
	CDocument::Dump(dc);
}
#endif //_DEBUG

/////////////////////////////////////////////////////////////////////////////
// CChatServerDoc commands

void CChatServerDoc::OnSetport() 
{
	// TODO: Add your command handler code here
	CPortDlg dlg;
	if(dlg.DoModal() == IDOK)
	{
		//创建一个新的类进行监听
		m_pSocket = new CListeningSocket(this);
		if (m_pSocket->Create(dlg.m_Port))
		{
			if (m_pSocket->Listen())
				return;
		}
	}
}

void CChatServerDoc::ProcessAccept()
{
	//创建一个新的socket来处理与客户端的数据交互
	CClientSocket* pSocket = new CClientSocket(this);
	//如果接受客户端的连接请求
	if (m_pSocket->Accept(*pSocket))
	{
		//初始化
		pSocket->Init();
		//将此连接socket加入连接链表中
		m_connectionList.AddTail(pSocket);
	}	
	//如果不接受连接
	else
		delete pSocket;
}

void CChatServerDoc::ProcessReceive(CClientSocket* pSocket)
{
	CMsg* pMsg;  //接收的消息
	CString sName;  //用户名
	BOOL bIsUsedName = FALSE;  //用户名是否已使用
	bIsNewChatter = FALSE;  //是否新用户

	do
	{
		//接收消息
		pMsg = ReadMsg(pSocket);
		//如果此消息是发送的新的用户名（在用户登录时客户端先发此消息）
		if(pMsg->code == SENDING_NICKNAME)
		{
			sName = pMsg->m_strText;
			bIsNewChatter = TRUE;
			//检查此用户名是否已使用
			if(IsUsedName(sName))
				bIsUsedName = TRUE;
		}
		//如果连接已关闭则退出
		if (pMsg->m_bClose)
		{
			break;
		}
	}
	while (!pSocket->m_pArchiveIn->IsBufferEmpty());

	//如果是用户登录
	if(bIsNewChatter)
	{
		//如果此用户名没有被使用
		if(!bIsUsedName)
		{
			//发送消息通知各客户端
			pMsg->code = SENDING_CHATTERS_LIST;
			m_ChattersList += sName + ":";
			pMsg->m_strText = m_ChattersList;
			SendForNewcomer(pMsg);
		}
		//如果用户名已使用
		else
		{
			//发消息通知此客户端
			CMsg* msg = new CMsg;
			msg->Init();
			msg->code =  USED_NAME;
			CString string;
			string.Format("用户名已被使用！");
			msg->m_strText = string;
			SendMsg(pSocket, msg);
		}
		bIsNewChatter = FALSE;
		bIsUsedName = FALSE;
	}

	//如果消息表明该客户端已关闭连接
	if (pMsg->m_bClose)
	{
		//发送消息通知各客户端
		SendToAllClients(NORMAL_MESSAGE);
		//在视图中将该用户删除
		DeleteChatter(pMsg);
		//关闭连接
		CloseSocket(pSocket);
		//通知各客户端更新用户列表
		SendToAllClients(SENDING_CHATTERS_LIST);
	}
	//否则是普通消息，发送给各客户端
	else
		SendToAllClients(NORMAL_MESSAGE);
}

CMsg* CChatServerDoc::ReadMsg(CClientSocket* pSocket)
{
	static CMsg msg;
	TRY
	{
		//接收数据
		pSocket->ReceiveMsg(&msg);
		//如果是新用户登录则更新用户列表
		if(msg.code == SENDING_NICKNAME)
		{
			UpdateChattersListView(msg.m_strText , pSocket);
		}
		//如果是普通信息则更新聊天信息列表，并将信息加入到m_msgList链表中
		if(msg.code == NORMAL_MESSAGE)
		{
			UpdateMessageView(msg.m_strText);
			m_msgList.AddTail(msg.m_strText);
		}
	}
	CATCH(CFileException, e)
	{
		CString strTemp;
		strTemp.Format("无法读取数据！");
		UpdateMessageView(strTemp);
		msg.m_bClose = TRUE;
	}
	END_CATCH
	return &msg;
}

void CChatServerDoc::SendMsg(CClientSocket* pSocket, CMsg* pMsg)
{
	TRY
	{
		//调用CClientSocket的函数发送消息
		pSocket->SendMsg(pMsg);
	}
	CATCH(CFileException, e)
	{
		CString strTemp;
		strTemp.Format("无法发送数据！");
		UpdateMessageView(strTemp);
	}
	END_CATCH
}

BOOL CChatServerDoc::IsUsedName(CString sNickName)
{
	CString tempStr, tempList, sName;
	tempStr = "";
	tempList = m_ChattersList;
	do
	{
		//利用“：”得到名字并查找（客户端发送的消息前面都有用户名加“：”）
		sName = tempList.Left(tempList.Find(":", 0));
		tempList = tempList.Mid(tempList.Find(":", 0) + 1);
		//找到则返回真
		if(sName == sNickName)
			return TRUE;
	}while(tempList.Find(":" , 0) != -1);
	//未被使用则返回假
	return FALSE;
}

CMsg* CChatServerDoc::AssembleMsg(CClientSocket* pSocket , int nCode)
{
	static CMsg msg;
	msg.Init();
	for (POSITION pos1 = m_msgList.FindIndex(pSocket->m_nMsgCount); pos1 != NULL;)
	{
		//信息内容
		msg.m_strText = m_msgList.GetNext(pos1);	 
		//信息类型
		msg.code = nCode;
	}
	pSocket->m_nMsgCount = m_msgList.GetCount();
	return &msg;
}

void CChatServerDoc::SendForNewcomer(CMsg* pMsg)
{
	for(POSITION pos = m_connectionList.GetHeadPosition(); pos != NULL;)
	{
		//对每个客户端都发送信息
		CClientSocket* pSocket = (CClientSocket*)m_connectionList.GetNext(pos);
		if (pSocket != NULL)
			SendMsg(pSocket, pMsg);
	}
}

void CChatServerDoc::SendToAllClients(int nCode)
{
	for(POSITION pos = m_connectionList.GetHeadPosition(); pos != NULL;)
	{
		//得到每个客户端
		CClientSocket* pSocket = (CClientSocket*)m_connectionList.GetNext(pos);
		//封装信息
		CMsg* pMsg = AssembleMsg(pSocket, nCode);
		//发送信息
		if (pMsg != NULL)
			SendMsg(pSocket, pMsg);
	}
}

void CChatServerDoc::UpdateMessageView(LPCTSTR lpszMessage)
{
	for(POSITION pos=GetFirstViewPosition();pos!=NULL;)
	{
		CView* pView = GetNextView(pos);
		CChatView* pChatView = DYNAMIC_DOWNCAST(CChatView, pView);
		if (pChatView != NULL)
			pChatView->ShowMessage(lpszMessage);
	}
}

void CChatServerDoc::UpdateChattersListView(CString sName , CClientSocket* pSocket)
{
	CString sIPAddress;
	UINT iPort;
	//得到IP地址和端口号
	pSocket->GetPeerName(sIPAddress , iPort);	
	for(POSITION pos=GetFirstViewPosition();pos!=NULL;)
	{
		CView* pView = GetNextView(pos);
		CChattersListView* pChattersListView = DYNAMIC_DOWNCAST(CChattersListView, pView);
		//更新用户列表
		if (pChattersListView != NULL)
			pChattersListView->AddChatter(sName , sIPAddress , iPort);
	}	
}

void CChatServerDoc::CloseSocket(CClientSocket* pSocket)
{
	//关闭连接
	pSocket->Close();
	POSITION pos,temp;
	for(pos = m_connectionList.GetHeadPosition(); pos != NULL;)
	{
		temp = pos;
		CClientSocket* pSock = (CClientSocket*)m_connectionList.GetNext(pos);
		//将此socket从连接链表中删除
		if (pSock == pSocket)
		{
			m_connectionList.RemoveAt(temp);
			break;
		}
	}
	delete pSocket;
	//如果没有连接则清空连接链表
	if(m_connectionList.GetCount() == 0)
		m_msgList.RemoveAll();
}

void CChatServerDoc::DeleteChatter(CMsg* pMsg)
{
	//得到要删除的用户名
	CString sNickName = pMsg->m_strText.Left(pMsg->m_strText.Find(":", 0));
	POSITION pos;
	for(pos=GetFirstViewPosition();pos!=NULL;)
	{
		CView* pView = GetNextView(pos);
		CChattersListView* pChattersListView = DYNAMIC_DOWNCAST(CChattersListView, pView);
		//在用户列表视图中将其删除
		if (pChattersListView != NULL)
			pChattersListView->DeleteChatter(sNickName);
	}	
	
	CString tempStr, sName, tempList;
	tempStr = "";
	tempList = m_ChattersList;
	//得到除去此用户名之外的所有用户名
	do
	{
		sName = tempList.Left(tempList.Find(":", 0));
		tempList = tempList.Mid(tempList.Find(":", 0) + 1);
		if(sName != sNickName)
			tempStr += sName + ":";
	}while(tempList.Find(":" , 0) != -1);
	m_ChattersList = tempStr;
	//加入到信息链表和信息内容中
	//接下来要对各客户端发送这些用户名，以使各客户端更新用户列表
	pMsg->m_strText = m_ChattersList;
	m_msgList.AddTail(m_ChattersList);
}

void CChatServerDoc::DeleteContents()
{
	if(m_pSocket == (CListeningSocket*)NULL)
		return;
	//删除监听socket
	delete m_pSocket;
	m_pSocket = NULL;

	CString temp;
	temp.Format("服务器已关闭");
	m_msgList.AddTail(temp);

	//对当前连接的每个客户端发送信息后关闭与该客户端的连接
	while(!m_connectionList.IsEmpty())
	{
		CClientSocket* pSocket = (CClientSocket*)m_connectionList.RemoveHead();
		CMsg* pMsg = AssembleMsg(pSocket , NORMAL_MESSAGE);
		pMsg->m_bClose = TRUE;
		SendMsg(pSocket, pMsg);
		pSocket->ShutDown(2);
		delete pSocket;
	}
	//清空所有信息
	m_msgList.RemoveAll();
	CDocument::DeleteContents();
}