e04doc.cpp

这是一个用VC开发的聊天室程序

// E04Doc.cpp : implementation of the CE04Doc class
//

#include "stdafx.h"
#include "E04.h"

#include "E04Doc.h"
#include "E04View.h"
#include "ChannelDlg.h"

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

/////////////////////////////////////////////////////////////////////////////
// CE04Doc

IMPLEMENT_DYNCREATE(CE04Doc, CDocument)

BEGIN_MESSAGE_MAP(CE04Doc, CDocument)
	//{{AFX_MSG_MAP(CE04Doc)
	ON_COMMAND(ID_FILE_START, OnFileStart)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CE04Doc construction/destruction

CE04Doc::CE04Doc()
{
	// TODO: add one-time construction code here
	m_clientSocket=NULL;
	m_listenSocket=NULL;
}

CE04Doc::~CE04Doc()
{
	POSITION pos=m_list.GetHeadPosition();
	while (pos!=NULL)
	{
		CClientSocket* client=(CClientSocket*)m_list.GetNext(pos);
		delete client;
	}
	if (m_listenSocket!=NULL)
	{
		m_listenSocket->Close();
		delete m_listenSocket;
	}
}

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

	((CEditView*)m_viewList.GetHead())->SetWindowText(NULL);

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

	return TRUE;
}

/////////////////////////////////////////////////////////////////////////////
// CE04Doc serialization

void CE04Doc::Serialize(CArchive& ar)
{
	// CEditView contains an edit control which handles all serialization
	((CEditView*)m_viewList.GetHead())->SerializeRaw(ar);
}

/////////////////////////////////////////////////////////////////////////////
// CE04Doc diagnostics

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

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

/////////////////////////////////////////////////////////////////////////////
// CE04Doc commands

void CE04Doc::OnFileStart() 
{
	// TODO: Add your command handler code here
	CChannelDlg dlg;
	if (dlg.DoModal()!=IDOK) return;
	if (m_listenSocket!=NULL)
	{
		m_listenSocket->Close();
		delete m_listenSocket;
	}

	m_listenSocket=new CListenSocket(this);
	if (!m_listenSocket->Create(atoi(dlg.m_channel)+BASE_CHANNEL))
	{
		AfxMessageBox("创建Socket失败!");
		m_listenSocket=NULL;
		return;
	}
	if (!m_listenSocket->Listen())
	{
		AfxMessageBox("Socket侦听异常!");
		delete m_listenSocket;
		m_listenSocket=NULL;
		return;
	}

	POSITION pos=m_list.GetHeadPosition();
	while (pos!=NULL)
	{
		CClientSocket* client=(CClientSocket*)m_list.GetNext(pos);
		delete client;
	}
}

void CE04Doc::AcceptClient()
{
	m_clientSocket=new CClientSocket(this);
	if (!m_listenSocket->Accept(*m_clientSocket))
	{
		AfxMessageBox("Socket接收出现异常!");
		delete m_clientSocket;
		m_clientSocket=NULL;
		return;
	}
#ifdef _DEBUG
	AfxMessageBox("有用户加入!");
#endif
	m_list.AddTail(m_clientSocket);
}

void CE04Doc::ReadMessage(CClientSocket *socket)
{
	char buffer[MAX_BUFFER_SIZE];
	int len=socket->Receive(buffer,MAX_BUFFER_SIZE-1);
	if (len<1)
	{
		AfxMessageBox("接收信息出现异常!");
		return;
	}

	buffer[len]=0;
	ShowMessage(socket,buffer);
	SendMessage(socket,buffer);
}

void CE04Doc::ShowMessage(CClientSocket *socket,char* buffer)
{
	POSITION pos=GetFirstViewPosition();
	while (pos!=NULL)
	{
		CE04View* eView=(CE04View*)GetNextView(pos);
		CEdit& edit=eView->GetEditCtrl();

		CString addr;
		UINT port;
		socket->GetPeerName(addr,port);
		char msg[MAX_BUFFER_SIZE];
		wsprintf(msg,"%d[%s]:%s\r\n",port,addr,buffer);

		int len=eView->GetWindowTextLength();
		edit.SetSel(len,len);
		edit.ReplaceSel(msg);
	}
}

void CE04Doc::SendMessage(CClientSocket *socket,char* buffer)
{
	POSITION pos=m_list.GetHeadPosition();
	while (pos!=NULL)
	{
		CClientSocket* client=(CClientSocket*)m_list.GetNext(pos);
		if (client==socket) continue;
		client->Send(buffer,strlen(buffer));
	}
}