csocket.cpp

一个类似QQ的聊天程序--服务器.利用WINSOCK编写,在VISUAL C++调试通过

// CSocket.cpp : implementation file
//

#include "stdafx.h"
#include "Ts.h"
#include "CSocket.h"
#include "TsDlg.h"
#include "Msg.h"


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

/////////////////////////////////////////////////////////////////////////////
// CCSocket

CCSocket::CCSocket(CTsDlg *pDlg)
{
	m_pDlg=pDlg;
	m_pFile=NULL;
	m_pArchiveIn=NULL;
	m_pArchiveOut=NULL;
}

CCSocket::~CCSocket()
{
	m_pDlg=NULL;
	if(m_pArchiveOut!=NULL)
		delete m_pArchiveOut;
	if(m_pArchiveIn!=NULL)
		delete m_pArchiveIn;
	if(m_pFile!=NULL)
		delete m_pFile;
}

void CCSocket::Initialize()   //构造与此套接字相应的CSocketFile对象
{
	m_pFile=new CSocketFile(this,TRUE);
	m_pArchiveIn=new CArchive(m_pFile,CArchive::load);
	m_pArchiveOut=new CArchive(m_pFile,CArchive::store);
}

void CCSocket::SendMessage(CMsg * pMsg)
{
	if(m_pArchiveOut!=NULL)
	{
		pMsg->Serialize(*m_pArchiveOut);
		m_pArchiveOut->Flush();
	}
}

void CCSocket::ReceiveMessage(CMsg * pMsg)
{
	pMsg->Serialize(*m_pArchiveIn);
}

void CCSocket::OnReceive(int nErrorCode)
{
	CSocket::OnAccept(nErrorCode);
	if(nErrorCode==0)
		m_pDlg->OnReceive(this);
}

IMPLEMENT_DYNAMIC(CCSocket,CSocket)

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

/////////////////////////////////////////////////////////////////////////////
// CCSocket member functions