clientsocket.cpp

VC++基础教程《VC开发入门教学》光盘

// ClientSocket.cpp : implementation file
//

#include "stdafx.h"
#include "Server.h"
#include "ClientSocket.h"

#include "Msg.h"
#include "ServerDlg.h"

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

/////////////////////////////////////////////////////////////////////////////
// CClientSocket

CClientSocket::CClientSocket()	// 构造函数
{
	m_pSocketFile = NULL;
	m_pArchiveIn = NULL;
	m_pArchiveOut = NULL;
	m_bClose = false;
}

CClientSocket::~CClientSocket()		// 析构函数
{
	if(m_pArchiveIn)
		delete m_pArchiveIn;
	if(m_pArchiveOut)
		delete m_pArchiveOut;
	if(m_pSocketFile)
		delete m_pSocketFile;
}


// Do not edit the following lines, which are needed by ClassWizard.
#if 0
BEGIN_MESSAGE_MAP(CClientSocket, CSocket)
	//{{AFX_MSG_MAP(CClientSocket)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()
#endif	// 0

/////////////////////////////////////////////////////////////////////////////
// CClientSocket member functions

void CClientSocket::Init(CServerDlg *pDlg)		// 初始化成员变量
{
	m_pSocketFile = new CSocketFile(this);
	m_pArchiveIn = new CArchive(m_pSocketFile, CArchive::load);
	m_pArchiveOut = new CArchive(m_pSocketFile, CArchive::store);
	m_bClose = false;
	m_pDlg = pDlg;
}

void CClientSocket::OnReceive(int nErrorCode)	// 重载OnReceive函数
{
	CSocket::OnReceive(nErrorCode);
	do
	{
		CMsg msg;		// 声明一个CMsg类消息对象
		msg.Serialize(*m_pArchiveIn);	// 串行化消息对象
		m_pDlg->UpdateData(TRUE);
		m_pDlg->m_strShow += "\r\n";
		m_pDlg->m_strShow += msg.m_strMsg;
		m_pDlg->UpdateData(FALSE);		// 更新显示
		m_pDlg->m_Show.LineScroll(m_pDlg->m_Show.GetLineCount());				
		for(POSITION pos=m_pDlg->m_plConn.GetHeadPosition(); pos!=NULL;)
		{	// 向其他客户端套接字发送消息对象
			CClientSocket * s = (CClientSocket*)m_pDlg->m_plConn.GetNext(pos);
			if(s->m_hSocket!=this->m_hSocket)
			{
				s->SendMsg(&msg);
			}
		}
	}
	while (!m_pArchiveIn->IsBufferEmpty());	
}

void CClientSocket::SendMsg(CMsg * pMsg)	// 发送消息对象
{
	if (m_pArchiveOut != NULL)
	{
		pMsg->Serialize(*m_pArchiveOut);	// 串行化消息对象
		m_pArchiveOut->Flush();			// 发送
	}
	else
	{
		m_bClose = true;		// 确定客户端套接字已关闭
		CloseSocket();			// 关闭相应的客户端套接字
		m_pDlg->CloseSessionSocket();	// 关闭与之关联的会话
	}
}

void CClientSocket::CloseSocket()	// 关闭客户端套接字
{
	if (m_pArchiveIn)	{ delete m_pArchiveIn; m_pArchiveIn = NULL; }
	if(m_pArchiveOut)	{ delete m_pArchiveOut; m_pArchiveOut = NULL; }
	if(m_pSocketFile)	{ delete m_pSocketFile; m_pSocketFile = NULL; }
	Close();	// 关闭套接字
	m_bClose = true;
}

void CClientSocket::OnClose(int nErrorCode)		// 重载OnClose函数
{
	m_bClose = true;		// 确定客户端套接字已关闭
	CloseSocket();			// 关闭相应的客户端套接字
	m_pDlg->CloseSessionSocket();	// 关闭与之关联的会话	
	CSocket::OnClose(nErrorCode);
}