clientsock.cpp

一个完整的数字硬盘录像机系统软件

// clntsock.cpp : implementation of the CClientSocket class

#include "stdafx.h"
#include "GtMpeg.h"
#include "GtMpegWnd.h"
#include "ClientSock.h"
#include "CommandMsg.h"
#include "DataMsg.h"
/////////////////////////////////////////////////////////////////////////////
//
/////////////////////////////////////////////////////////////////////////////
IMPLEMENT_DYNAMIC(CClientSocket, CSocket)
/////////////////////////////////////////////////////////////////////////////
//
/////////////////////////////////////////////////////////////////////////////
CClientSocket::CClientSocket()
{
	m_pFile = NULL;
	m_pArchiveIn = NULL;
	m_pArchiveOut = NULL;
	m_sClientIP=_T("");

}
/////////////////////////////////////////////////////////////////////////////
//
/////////////////////////////////////////////////////////////////////////////
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::Abort()
{
	if (m_pArchiveOut != NULL)
	{
		m_pArchiveOut->Abort();
		delete m_pArchiveOut;
		m_pArchiveOut = NULL;
	}
}
/////////////////////////////////////////////////////////////////////////////
//
/////////////////////////////////////////////////////////////////////////////
void CClientSocket::SendMsg(CDataMsg* pMsg)
{
	if (m_pArchiveOut != NULL)
	{
		pMsg->Serialize(*m_pArchiveOut);
		m_pArchiveOut->Flush();
	}
}
/////////////////////////////////////////////////////////////////////////////
//
/////////////////////////////////////////////////////////////////////////////
void CClientSocket::ReceiveMsg(CCommandMsg* pMsg)
{
	pMsg->Serialize(*m_pArchiveIn);
}
/////////////////////////////////////////////////////////////////////////////
//
/////////////////////////////////////////////////////////////////////////////
void CClientSocket::OnReceive(int nErrorCode)
{
	CSocket::OnReceive(nErrorCode);
	CGtMpegWnd * pWnd=(CGtMpegWnd *)theApp.m_pMainWnd;
	pWnd->ProcessPendingRead(this);
}
/////////////////////////////////////////////////////////////////////////////
//
/////////////////////////////////////////////////////////////////////////////
CClientSocket::~CClientSocket()
{
	if (m_pArchiveOut != NULL)
		delete m_pArchiveOut;
	if (m_pArchiveIn != NULL)
		delete m_pArchiveIn;
	if (m_pFile != NULL)
		delete m_pFile;
}
/////////////////////////////////////////////////////////////////////////////
//
/////////////////////////////////////////////////////////////////////////////
#ifdef _DEBUG
void CClientSocket::AssertValid() const
{
	CSocket::AssertValid();
}
void CClientSocket::Dump(CDumpContext& dc) const
{
	CSocket::Dump(dc);
}
#endif //_DEBUG
/////////////////////////////////////////////////////////////////////////////
//
/////////////////////////////////////////////////////////////////////////////
CString  CClientSocket::GetClientIP()
{
	   CString sClientIP;
	   UINT nPort;
	   GetPeerName(sClientIP,nPort); 
	   return sClientIP;
}