clientsocket.cpp

来自「网络编程」· C++ 代码 · 共 112 行

// ClientSocket.cpp : implementation file
//

#include "stdafx.h"
//#include "TestSvr.h"
#include "ClientSocket.h"
#include "chatServer.h"
#include "pkg.h"

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

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

CClientSocket::CClientSocket(CChatServer* pServer)
{
	m_pServer = pServer;
	ASSERT(m_pServer);
	m_pFile = NULL;
	m_pArchiveIn = NULL;
	m_pArchiveOut = NULL;

}

CClientSocket::~CClientSocket()
{
	Finalize();
}


// 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::OnReceive(int nErrorCode) 
{
	// TODO: Add your specialized code here and/or call the base class
	CSocket::OnReceive(nErrorCode);
	ASSERT(m_pServer);
	m_pServer->ProcessClientPkg(this);
}





void CClientSocket::Init()
{
	m_pFile = new CSocketFile(this);
	m_pArchiveIn = new CArchive(m_pFile,CArchive::load);
	m_pArchiveOut = new CArchive(m_pFile,CArchive::store);
}

void CClientSocket::ReceivePkg(CPkg *pkg)
{
	ASSERT(pkg);
	try{
		do{
			pkg->Serialize(*m_pArchiveIn);
		}while(!m_pArchiveIn->IsBufferEmpty());
	}catch(CFileException* e){
		m_pArchiveIn->Abort();
	}
}

void CClientSocket::SendPkg(CPkg *pkg)
{
	ASSERT(pkg);
	try{
		if(m_pArchiveOut != NULL){
			pkg->Serialize(*m_pArchiveOut);
			m_pArchiveOut->Flush();
		}
	}catch(CFileException* e){
		m_pArchiveOut->Abort();
	}
}



void CClientSocket::Finalize()
{
	if(m_pArchiveIn != NULL){
		delete m_pArchiveIn;
		m_pArchiveIn = NULL;
	}
	if(m_pArchiveOut != NULL){
		delete m_pArchiveOut;
		m_pArchiveOut = NULL;
	}
	if(m_pFile != NULL){
		delete m_pFile;
		m_pFile = NULL;
	}
	Close();
	
	m_pServer = NULL;
}