www.pudn.com > FiveChess.rar > ClientSocket.cpp

// ClientSocket.cpp : implementation file 
// 
 
#include "stdafx.h" 
#include "ClientSocket.h" 
#include "FiveChess.h" 
#include "MainFrm.h" 
#include "FiveChessView.h" 
 
#ifdef _DEBUG 
#define new DEBUG_NEW 
#undef THIS_FILE 
static char THIS_FILE[] = __FILE__; 
#endif 
 
///////////////////////////////////////////////////////////////////////////// 
// CClientSocket 
 
CClientSocket::CClientSocket() 
{ 
	m_aSessionIn=NULL; 
	m_aSessionOut=NULL; 
	m_sfSocketFile=NULL; 
	m_bInit=false; 
	m_bClose=false; 
 
} 
 
CClientSocket::~CClientSocket() 
{ 
	if(m_aSessionIn) 
		delete m_aSessionIn; 
	if(m_aSessionOut) 
		delete m_aSessionOut; 
	if(m_sfSocketFile) 
		delete m_sfSocketFile; 
} 
 
 
// 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); 
	do 
	{ 
		CMessg temp; 
		temp.Serialize(*m_aSessionIn); 
		m_view->m_sMsgList+=temp.m_strText; 
		m_view->m_outputedit.SetWindowText(m_view->m_sMsgList); 
		m_view->m_iLineCurrentPos=m_view->m_outputedit.GetLineCount(); 
		m_view->m_outputedit.LineScroll(m_view->m_iLineCurrentPos); 
 
		if(m_view->m_match.CanDown(temp.m_x,temp.m_y,m_view->m_who%2+1)) 
		{ 
			m_view->m_turn=temp.m_turn; 
			m_view->Invalidate(FALSE); 
			if(m_view->m_match.IsWin(m_view->m_who%2+1,m_view->m_winpos)) 
			{ 
				m_view->m_bWin=TRUE; 
				m_view->m_bOver=TRUE; 
				m_view->Invalidate(FALSE); 
				AfxMessageBox("你输了"); 
				m_view->m_sMsgList+="你输了"; 
				m_view->m_outputedit.SetWindowText(m_view->m_sMsgList); 
				m_view->m_iLineCurrentPos=m_view->m_outputedit.GetLineCount(); 
				m_view->m_outputedit.LineScroll(m_view->m_iLineCurrentPos); 
			} 
		} 
	} 
	while (!m_aSessionIn->IsBufferEmpty()); 
} 
 
void CClientSocket::Init(CFiveChessView * view) 
{ 
	m_sfSocketFile= new CSocketFile(this); 
	m_aSessionIn=new CArchive(m_sfSocketFile,CArchive::load); 
	m_aSessionOut=new CArchive(m_sfSocketFile,CArchive::store); 
	m_bClose=false; 
	this->m_view=view; 
} 
BOOL CClientSocket::SendMessage(CMessg * msg) 
{ 
	if (m_aSessionOut != NULL) 
	{ 
		msg->Serialize(*m_aSessionOut); 
		m_aSessionOut->Flush(); 
		return TRUE; 
	} 
	else 
	{ 
		//对方关闭了连接 
		m_bClose=true; 
		CloseSocket(); 
//		m_view->CloseSessionSocket(); 
		return FALSE; 
	} 
} 
 
 
void CClientSocket::CloseSocket() 
{ 
	if(m_aSessionIn) 
	{ 
		delete m_aSessionIn; 
		m_aSessionIn=NULL; 
	} 
	if(m_aSessionOut) 
	{ 
		delete m_aSessionOut; 
		m_aSessionOut=NULL; 
	} 
	if(m_sfSocketFile) 
	{ 
		delete m_aSessionOut; 
		m_sfSocketFile=NULL; 
	} 
	Close(); 
	m_bInit=false; 
	m_bClose=true; 
} 
 
void CClientSocket::OnClose(int nErrorCode)  
{ 
	// TODO: Add your specialized code here and/or call the base class 
	m_bClose=true; 
	CloseSocket(); 
//	m_view->CloseSessionSocket();	 
	CSocket::OnClose(nErrorCode); 
} 
 
int CClientSocket::GetLocalHostName(CString &sHostName)	//获得本地计算机名称 
{ 
	char szHostName[256]; 
	int nRetCode; 
	nRetCode=gethostname(szHostName,sizeof(szHostName)); 
	if(nRetCode!=0) 
	{ 
		//产生错误 
		sHostName=_T("没有取得"); 
		return GetLastError(); 
	} 
	sHostName=szHostName; 
	return 0; 
} 
 
int CClientSocket::GetIpAddress(const CString &sHostName, CString &sIpAddress)//获得本地IP 
{ 
	struct hostent FAR * lpHostEnt=gethostbyname(sHostName); 
	if(lpHostEnt==NULL) 
	{ 
		//产生错误 
		sIpAddress=_T(""); 
		return GetLastError(); 
	} 
	//获取IP 
	LPSTR lpAddr=lpHostEnt->h_addr_list[0]; 
	if(lpAddr) 
	{ 
		struct in_addr inAddr; 
		memmove(&inAddr,lpAddr,4); 
		//转换为标准格式 
		sIpAddress=inet_ntoa(inAddr); 
		if(sIpAddress.IsEmpty()) 
			sIpAddress=_T("没有取得"); 
	} 
	return 0; 
} 
int CClientSocket::GetIpAddress(const CString &sHostName, BYTE &f0,BYTE &f1,BYTE &f2,BYTE &f3)//获得本地IP 
{ 
	struct hostent FAR * lpHostEnt=gethostbyname(sHostName); 
	if(lpHostEnt==NULL) 
	{ 
		//产生错误 
		f0=f1=f2=f3=0; 
		return GetLastError(); 
	} 
	//获取IP 
	LPSTR lpAddr=lpHostEnt->h_addr_list[0]; 
	if(lpAddr) 
	{ 
		struct in_addr inAddr; 
		memmove(&inAddr,lpAddr,4); 
		f0=inAddr.S_un.S_un_b.s_b1; 
		f1=inAddr.S_un.S_un_b.s_b2; 
		f2=inAddr.S_un.S_un_b.s_b3; 
		f3=inAddr.S_un.S_un_b.s_b4; 
	} 
	return 0; 
} 
 
CString CClientSocket::ErrorReason(int tag) 
{ 
	CString result; 
	LPVOID lpMsgBuf; 
	FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER|FORMAT_MESSAGE_FROM_SYSTEM |  
		FORMAT_MESSAGE_IGNORE_INSERTS, 
		NULL, 
		tag, 
		MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), // Default language 
		(LPTSTR) &lpMsgBuf, 
		0, 
		NULL); 
	result = (UCHAR *)lpMsgBuf; 
	LocalFree( lpMsgBuf ); 
	return result; 
}