clientsocket.cpp

此源代码是一个简单的关于网络聊天室的

// ClientSocket.cpp : implementation file
//

#include "stdafx.h"
#include "Example2_ChatRoom.h"
#include "ClientSocket.h"
#include "Example2_ChatRoomDlg.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_dlg->m_sMsgList+=temp.m_strText;
		m_dlg->SetDlgItemText(IDC_SHOWTEXT,m_dlg->m_sMsgList);
		
		if(temp.m_tag==1&&m_dlg->m_willchating==FALSE)
			//如果有声音过来并且本机的声音设备已经准备好了则首先在本机发出声音
		{	
			memcpy(m_dlg->m_sound.m_cBufferOut,temp.m_buffer,MAX_BUFFER_SIZE);
		}
		int linenum=((CEdit *)(m_dlg->GetDlgItem(IDC_SHOWTEXT)))->GetLineCount();
		((CEdit *)(m_dlg->GetDlgItem(IDC_SHOWTEXT)))->LineScroll(linenum);
		if(!m_dlg->m_bClient)
		{
			for(POSITION pos=m_dlg->m_connectionList.GetHeadPosition();pos!=NULL;)
			{
				CClientSocket * t = (CClientSocket*)m_dlg->m_connectionList.GetNext(pos);
				if(t->m_hSocket!=this->m_hSocket)
				{
					t->SendMessage(&temp);
				}
			}
		}
	}
	while (!m_aSessionIn->IsBufferEmpty());
}

void CClientSocket::Init(CExample2_ChatRoomDlg * dlg)
{
	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_dlg=dlg;
}
BOOL CClientSocket::SendMessage(CMessg * msg)
{
	if (m_aSessionOut != NULL)
	{
		msg->Serialize(*m_aSessionOut);
		m_aSessionOut->Flush();
		return TRUE;
	}
	else
	{
		//对方关闭了连接
		m_bClose=true;
		CloseSocket();
		m_dlg->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_dlg->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;
	switch(tag)
	{
	case WSANOTINITIALISED:
		result="A successful AfxSocketInit must occur before using this API.";
		break;
	case WSAENETDOWN:
		result="the network subsystem failed";
		break;
	case WSAEADDRINUSE:
		result="The specified address is already in use";
		break;
	case WSAEINPROGRESS:
		result="A blocking Windows Sockets call is in progress";
		break;
	case WSAEADDRNOTAVAIL:
		result="The specified address is not available from the local machine";
		break;
	case WSAEAFNOSUPPORT:
		result="Addresses in the specified family cannot be used with this socket";
		break;
	case WSAECONNREFUSED:
		result="The attempt to connect was rejected";
		break;
	case WSAEDESTADDRREQ:
		result="A destination address is required";
		break;
	case WSAEFAULT:
		result="The nSockAddrLen argument is incorrect";
		break;
	case WSAEINVAL:
		result="Invalid host address";
		break;
	case WSAEISCONN:
		result="The socket is already connected.";
		break;
	case WSAEMFILE:
		result="No more file descriptors are available";
		break;
	case WSAENETUNREACH:
		result="The network cannot be reached from this host at this time";
		break;
	case WSAENOBUFS:
		result="No buffer space is available. The socket cannot be connected";
		break;
	case WSAENOTSOCK:
		result="The descriptor is not a socket";
		break;
	case WSAETIMEDOUT:
		result="Attempt to connect timed out without establishing a connection";
		break;
	case WSAEWOULDBLOCK:
		result="The socket is marked as nonblocking and the connection cannot be completed immediately.";
		break;
	default:
		result="unknown error";
	}
	return result;
}