clientsocket.cpp

来自「VisualC高级编程技术精粹.rar」· C++ 代码 · 共 63 行

// ClientSocket.cpp : implementation file
//

#include "stdafx.h"
#include "TCPChatClient.h"
#include "ClientSocket.h"
#include "TCPChatClientDlg.h"
#include "tagHeader.h"

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

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

CClientSocket::CClientSocket()
{

}

CClientSocket::~CClientSocket()
{
}


// 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
	char buff[sizeof(Header)];
	memset(buff,0,sizeof(buff));
	Receive(buff,sizeof(buff));

	Header *header = (Header*)buff;
	int length = header->len;
	char type = header->type;
	
	if (type == SEND_MESSAGE)
	{
		chatDlg->GetMessage();
	}

	if (type == LOGIN_IO)
	{
		chatDlg->UpdateUser();		
	}

	CSocket::OnReceive(nErrorCode);
}