VisualC高级编程技术精粹.rar

				// 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);
				}