www.pudn.com > fanccMSNr.src.rar > MessageDispatcher.cpp


// MessageDispatcher.cpp : 구현 파일입니다. 
// 
 
#include "stdafx.h" 
#include "DumbMessenger.h" 
#include "MessageDispatcher.h" 
#include "messenger/MIMEMessage.hpp" 
#include "messenger/MessageFactory.hpp" 
 
#include  
 
 
// CMessageDispatcher 
CMessageDispatcher::CMessageDispatcher(): 
 listeners() { 
	serverSocket=NULL; 
} 
 
CMessageDispatcher::~CMessageDispatcher() { 
	if(serverSocket!=NULL) { 
		delete serverSocket; 
	} 
} 
 
void CMessageDispatcher::connect(const string &address, unsigned long port) { 
	// 닫히지 않았는데 접속하려는 경우를 배려한다. 
	//Close(); 
 
	if(Create()==0) { 
		int error=GetLastError(); 
		if(error!=WSAEWOULDBLOCK) { 
			TRACE("ERROR in Create() in CMessageDispatcher::connect(). GetLastError==%d\n", error); 
			throw SocketException(); 
		} 
	} 
 
	if(Connect(address.data(), port)==0) { 
		int error=GetLastError(); 
		if(error!=WSAEWOULDBLOCK) { 
			TRACE("ERROR in Connect() in CMessageDispatcher::connect(). GetLastError==%d\n", error); 
			throw SocketException(); 
		} 
	} 
} 
 
namespace { 
	class ServerSocket: public CAsyncSocket { 
	protected: 
		CAsyncSocket *connection; 
	public: 
		ServerSocket(CAsyncSocket &connection) { 
			this->connection=&connection; 
		} 
		virtual void OnAccept(int nErrorCode) { 
			//TODO: handle nErrorCode 
			Accept(*connection); 
		} 
	}; 
} 
 
int CMessageDispatcher::listen(int preferredPort) { 
	if(serverSocket==NULL) { 
		serverSocket=new ServerSocket(*this); 
	} 
	int port=preferredPort; 
	// 닫히지 않았는데 접속하려는 경우를 배려한다. 
	serverSocket->Close(); 
 
	while(serverSocket->Create(port)==0) { 
		int error=GetLastError(); 
		if(error==WSAEPROTONOSUPPORT) { 
			port++; 
			// retry Create() 
			continue; 
		} 
		TRACE("ERROR in Create() in CMessageDispatcher::listen(). GetLastError==%d\n", error); 
		throw SocketException(); 
	} 
 
	if(serverSocket->Listen()==0) { 
		int error=GetLastError(); 
		TRACE("ERROR in Listen() in CMessageDispatcher::listen(). GetLastError==%d\n", error); 
		throw SocketException(); 
	} 
	return port; 
} 
 
void CMessageDispatcher::getAddress(string &address) { 
	CString socketAddress; 
	UINT port; 
	GetSockName(socketAddress, port); 
	address=(LPCSTR)socketAddress; 
} 
 
 
void CMessageDispatcher::addMessageListener(MessageListener &messageListener) { 
	listeners.push_back(&messageListener); 
} 
 
void CMessageDispatcher::removeMessageListener(MessageListener &messageListener) { 
	list::iterator location=find(listeners.begin(), listeners.end(), &messageListener); 
	if(location!=listeners.end()) { 
		listeners.erase(location); 
	} 
} 
 
MessageDispatcher *CMessageDispatcher::newInstance() { 
	return new CMessageDispatcher; 
} 
 
//#include  
 
void CMessageDispatcher::sendMessage(Message &message) { 
	const string &stringMessage=message.toString(); 
	TRACE("send: \"%s\"\n", stringMessage.data()); 
//			static ofstream out("out_log.txt"); 
//			out<< message.toString(); 
 
	sendBytes((const unsigned char *)stringMessage.data(), stringMessage.length()); 
} 
 
bool CMessageDispatcher::sendBytes(const unsigned char *bytes, size_t length) { 
 
	if(Send(bytes, (int)length)==SOCKET_ERROR) { 
		int error=GetLastError(); 
		 
		if(error==WSAEWOULDBLOCK) { 
			return false; 
		} 
		TRACE("ERROR in CMessageDispatcher::sendMessage(). GetLastError==%d\n", error); 
		throw SocketException(); 
	} 
	return true; 
} 
 
 
// CMessageDispatcher 멤버 함수입니다. 
 
void CMessageDispatcher::OnConnect(int nErrorCode) 
{ 
	// 이전 연결에 남아있던 버퍼를 지운다. 
	buffer.erase(buffer.begin(), buffer.end()); 
 
	list::iterator listener; 
	for(listener=listeners.begin(); listener!=listeners.end(); listener++) { 
		if( nErrorCode==0) { 
			(*listener)->connected(); 
		} else { 
			TRACE("About to failed() in CMessageDispatcher::OnConnect() nErrorCode==%d\n", nErrorCode); 
			(*listener)->failed(); 
		} 
	} 
} 
 
void CMessageDispatcher::OnClose(int nErrorCode) 
{ 
	// 에러가 발생했을 때만 호출한다. 
	if(nErrorCode!=0) { 
		list::iterator listener; 
		for(listener=listeners.begin(); listener!=listeners.end(); listener++) { 
 
			TRACE("About to failed() in CMessageDispatcher::OnClose() nErrorCode==%d\n", nErrorCode); 
			(*listener)->failed(); 
		} 
	} 
} 
 
void CMessageDispatcher::OnSend(int nErrorCode) { 
	list::iterator listener; 
 
	// 에러가 발생했을 때만 호출한다. 
	if(nErrorCode!=0) { 
		for(listener=listeners.begin(); listener!=listeners.end(); listener++) { 
 
			TRACE("About to failed() in CMessageDispatcher::OnSend() nErrorCode==%d\n", nErrorCode); 
			(*listener)->failed(); 
		} 
	} 
	for(listener=listeners.begin(); listener!=listeners.end(); listener++) { 
		(*listener)->sendable(); 
	} 
 
} 
 
 
// CAsyncSocket에서 오버라이드 
void CMessageDispatcher::OnReceive(int nErrorCode) 
{ 
	if(nErrorCode!=0) { 
		list::iterator listener; 
		for(listener=listeners.begin(); listener!=listeners.end(); listener++) { 
			TRACE("About to failed() in CMessageDispatcher::OnReceive() nErrorCode==%d\n", nErrorCode); 
			(*listener)->failed(); 
		} 
		return; 
	} 
 
	static const int nByteBuffer=500; 
	char byteBuffer[nByteBuffer+1]; 
	int nReceived = Receive(byteBuffer, nByteBuffer); 
	if(nReceived==0 || nReceived==SOCKET_ERROR) { 
		int error=GetLastError(); 
		if(error!=WSAEWOULDBLOCK) { 
			TRACE("ERROR in CMessageDispatcher::OnReceive(). GetLastError==%d\n", error); 
			// OnReceive에서 읽기가 실패하면 그저 아무 일도 하지 않는다. 
			return; 
		} 
	} 
	byteBuffer[nReceived]='\0'; 
//	TRACE("Bytes from socket: \"%s\"\n", byteBuffer); 
	buffer.append(byteBuffer, nReceived); 
 
	Message *message=NULL; 
	 
	try { 
		while((message=MessageFactory::createMessage(buffer))!=NULL) { 
			TRACE("message arrived: \"%s\"", message->toString().data()); 
//			static ofstream out("in_log.txt"); 
//			out<< message->toString(); 
 
 
			list::iterator listener; 
			for( listener=listeners.begin(); listener!=listeners.end(); listener++) { 
				(*listener)->messageArrived(*message); 
			} 
 
			delete message; 
			message=NULL; 
		} 
	} catch(exception e) { 
		if(message!=NULL) { 
			delete message; 
			message=NULL; 
		} 
		// TODO: handle exception 
	} 
}