itgmania212121/stepmania/src/smlobby/irc.cpp

// irc.cpp

#include "stdafx.h"
#include "irc.h"

using namespace irc;

////////////////////////////////////////////////////////////////////

CIrcMessage::CIrcMessage(const char* lpszCmdLine, bool bIncoming)
	: m_bIncoming(bIncoming)
{
	ParseIrcCommand(lpszCmdLine);
}

CIrcMessage::CIrcMessage(const CIrcMessage& m)
	:	sCommand(m.sCommand),
		parameters(m.parameters),
		m_bIncoming(m.m_bIncoming)
{
	prefix.sNick = m.prefix.sNick;
	prefix.sUser = m.prefix.sUser;
	prefix.sHost = m.prefix.sHost;
}

void CIrcMessage::Reset()
{
	prefix.sNick = prefix.sUser = prefix.sHost = sCommand = "";
	m_bIncoming = false;
	parameters.clear();
}

CIrcMessage& CIrcMessage::operator = (const CIrcMessage& m)
{
	if( &m != this )
	{
		sCommand = m.sCommand;
		parameters = m.parameters;
		prefix.sNick = m.prefix.sNick;
		prefix.sUser = m.prefix.sUser;
		prefix.sHost = m.prefix.sHost;
		m_bIncoming = m.m_bIncoming;
	}
	return *this;
}

CIrcMessage& CIrcMessage::operator = (const char* lpszCmdLine)
{
	Reset();
	ParseIrcCommand(lpszCmdLine);
	return *this;
}

void CIrcMessage::ParseIrcCommand(const char* lpszCmdLine)
{
	const char* p1 = lpszCmdLine;
	const char* p2 = lpszCmdLine;

	ASSERT(lpszCmdLine != NULL);
	ASSERT(*lpszCmdLine);

	// prefix exists ?
	if( *p1 == ':' )
	{ // break prefix into its components (nick!user@host)
		p2 = ++p1;
		while( *p2 && !strchr(" !", *p2) )
			++p2;
		prefix.sNick.assign(p1, p2 - p1);
		if( *p2 != '!' )
			goto end_of_prefix;
		p1 = ++p2;
		while( *p2 && !strchr(" @", *p2) )
			++p2;
		prefix.sUser.assign(p1, p2 - p1);
		if( *p2 != '@' )
			goto end_of_prefix;
		p1 = ++p2;
		while( *p2 && !isspace(*p2) )
			++p2;
		prefix.sHost.assign(p1, p2 - p1);
end_of_prefix :
		while( *p2 && isspace(*p2) )
			++p2;
		p1 = p2;
	}

	// get command
	ASSERT(*p1 != '\0');
	p2 = p1;
	while( *p2 && !isspace(*p2) )
		++p2;
	sCommand.assign(p1, p2 - p1);
	_strupr((char*)sCommand.c_str());
	while( *p2 && isspace(*p2) )
		++p2;
	p1 = p2;

	// get parameters
	while( *p1 )
	{
		if( *p1 == ':' )
		{
			++p1;
			// seek end-of-message
			while( *p2 )
				++p2;
			parameters.push_back(String(p1, p2 - p1));
			break;
		}
		else
		{
			// seek end of parameter
			while( *p2 && !isspace(*p2) )
				++p2;
			parameters.push_back(String(p1, p2 - p1));
			// see next parameter
			while( *p2 && isspace(*p2) )
				++p2;
			p1 = p2;
		}
	} // end parameters loop
}

String CIrcMessage::AsString() const
{
	String s;

	if( prefix.sNick.length() )
	{
		s += ":" + prefix.sNick;
		if( prefix.sUser.length() && prefix.sHost.length() )
			s += "!" + prefix.sUser + "@" + prefix.sHost;
		s += " ";
	}

	s += sCommand;

	for(int i=0; i < parameters.size(); i++)
	{
		s += " ";
		if( i == parameters.size() - 1 ) // is last parameter ?
			s += ":";
		s += parameters[i];
	}

	s += endl;

	return s;
}

////////////////////////////////////////////////////////////////////

CIrcSession::CIrcSession(IIrcSessionMonitor* pMonitor)
	:	m_hThread(NULL)
{
	InitializeCriticalSection(&m_cs);
}

CIrcSession::~CIrcSession()
{
	Disconnect();
	DeleteCriticalSection(&m_cs);
}

bool CIrcSession::Connect(const CIrcSessionInfo& info)
{
	ASSERT(m_hThread==NULL && !m_socket);

	try
	{
//		if( !m_socket.Create() )
//			throw "Failed to create socket!";

		InetAddr addr(info.sServer.c_str(), info.iPort);
		if( !m_socket.Connect(addr) )
		{
			m_socket.Close();
			throw "Failed to connect to host!";
		}

		m_info = info;

		// start receiving messages from host
		m_hThread = CreateThread(NULL, 0, ThreadProc, this, 0, NULL);
		Sleep(100);

		if( info.sPassword.length() )
			m_socket.Send("PASS %s\r\n", info.sPassword.c_str());

		m_socket.Send("NICK %s\r\n", info.sNick.c_str());

		TCHAR szHostName[MAX_PATH];
		DWORD cbHostName = sizeof(szHostName);
		GetComputerName(szHostName, &cbHostName);

		m_socket.Send("USER %s %s %s :%s\r\n", 
			info.sUserID.c_str(), szHostName, "server", info.sFullName.c_str());

	}
	catch( const char* )
	{
		Disconnect();
	}
	catch( ... )
	{
		Disconnect();
	}

	return (bool)m_socket;
}

void CIrcSession::Disconnect(const char* lpszMessage)
{
	static const DWORD dwServerTimeout = 5 * 1000;

	if( !m_hThread )
		return;

	m_socket.Send("QUIT :%s\r\n", lpszMessage ? lpszMessage : "Bye!");

	if( m_hThread && WaitForSingleObject(m_hThread, dwServerTimeout) != WAIT_OBJECT_0 )
	{
		m_socket.Close();
		Sleep(100);
		if( m_hThread && WaitForSingleObject(m_hThread, dwServerTimeout) != WAIT_OBJECT_0 )
		{
			TerminateThread(m_hThread, 1);
			CloseHandle(m_hThread);
			m_hThread = NULL;
			m_info.Reset();
		}
	}
}

void CIrcSession::Notify(const CIrcMessage* pmsg)
{
	// forward message to monitor objects
	EnterCriticalSection(&m_cs);
	for(std::set<IIrcSessionMonitor*>::iterator it = m_monitors.begin();
			it != m_monitors.end();
			it++
			)
	{
		(*it)->OnIrcMessage(pmsg);
	}
	LeaveCriticalSection(&m_cs);
}

void CIrcSession::DoReceive()
{
	CIrcIdentServer m_identServer;
	char chBuf[1024*4+1];
	int cbInBuf = 0;

	if( m_info.bIdentServer )
		m_identServer.Start(m_info.sUserID.c_str());

	while( m_socket )
	{
		int cbRead;
		int nLinesProcessed = 0;

		cbRead = m_socket.Receive((unsigned char*)chBuf+cbInBuf, sizeof(chBuf)-cbInBuf-1);
		if( cbRead <= 0 )
			break;
		cbInBuf += cbRead;
		chBuf[cbInBuf] = '\0';

		char* pStart = chBuf;
		while( *pStart )
		{
			char* pEnd;

			// seek end-of-line
			for(pEnd=pStart; *pEnd && *pEnd != '\r' && *pEnd != '\n'; ++pEnd)
				;
			if( *pEnd == '\0' )
				break; // uncomplete message. stop parsing.

			++nLinesProcessed;

			// replace end-of-line with NULLs and skip
			while( *pEnd == '\r' || *pEnd == '\n' )
				*pEnd++ = '\0';

			if( *pStart )
			{
				// process single message by monitor objects
				CIrcMessage msg(pStart, true);
				Notify(&msg);
			}

			cbInBuf -= pEnd - pStart;
			ASSERT(cbInBuf >= 0);

			pStart = pEnd;
		}

		// discard processed messages
		if( nLinesProcessed != 0 )
			memmove(chBuf, pStart, cbInBuf+1);
	}

	if( m_socket )
		m_socket.Close();

	if( m_info.bIdentServer )
		m_identServer.Stop();

	// notify monitor objects that the connection has been closed
	Notify(NULL);
}

DWORD WINAPI CIrcSession::ThreadProc(LPVOID pparam)
{
	CIrcSession* pThis = (CIrcSession*)pparam;
	try { pThis->DoReceive(); } catch( ... ) {}
	pThis->m_info.Reset();
	CloseHandle(pThis->m_hThread);
	pThis->m_hThread = NULL;
	return 0;
}

void CIrcSession::AddMonitor(IIrcSessionMonitor* pMonitor)
{
	ASSERT(pMonitor != NULL);
	EnterCriticalSection(&m_cs);
	m_monitors.insert(pMonitor);
	LeaveCriticalSection(&m_cs);
}

void CIrcSession::RemoveMonitor(IIrcSessionMonitor* pMonitor)
{
	ASSERT(pMonitor != NULL);
	EnterCriticalSection(&m_cs);
	m_monitors.erase(pMonitor);
	LeaveCriticalSection(&m_cs);
}

////////////////////////////////////////////////////////////////////

CIrcSessionInfo::CIrcSessionInfo()
	:	iPort(0), bIdentServer(false), iIdentServerPort(0), bIsGameHost(false), luSongHash(0L)	
{
}

CIrcSessionInfo::CIrcSessionInfo(const CIrcSessionInfo& si)
	:	sServer(si.sServer),
		sServerName(si.sServerName),
		iPort(si.iPort),
		sNick(si.sNick),
		sUserID(si.sUserID),
		sFullName(si.sFullName),
		sPassword(si.sPassword),
		bIdentServer(si.bIdentServer),
		sIdentServerType(si.sIdentServerType),
		iIdentServerPort(si.iIdentServerPort),
		bIsGameHost(si.bIsGameHost),
		sSongPath(si.sSongPath),
		luSongHash(si.luSongHash),
		sHostName(si.sHostName),
		sHostIP(si.sHostIP)
{
}

void CIrcSessionInfo::Reset()
{
	sServer = "";
	sServerName = "";
	iPort = 0;
	sNick = "";
	sUserID = "";
	sFullName = "";
	sPassword = "";
	bIdentServer = false;
	sIdentServerType = "";
	iIdentServerPort = 0;

	bIsGameHost = false;
	sSongPath = "";
	luSongHash = 0L;
	sHostName = "";
	sHostIP = "";


}

////////////////////////////////////////////////////////////////////

CIrcIdentServer::CIrcIdentServer()
	: m_uiPort(0), m_hThread(NULL)
{
}

CIrcIdentServer::~CIrcIdentServer()
{
	Stop();
}

bool CIrcIdentServer::Start(
				const char* lpszUserID,
				unsigned int uiPort,
				const char* lpszResponseType
				)
{
	if( m_socket )
		return false;

	if( !m_socket.Bind(InetAddr(uiPort)) )
	{
		m_socket.Close();
		return false;
	}

	m_sResponseType = lpszResponseType;
	m_sUserID = lpszUserID;
	m_uiPort = uiPort;

	m_hThread = CreateThread(NULL, 0, ListenProc, this, 0, NULL);
	Sleep(100);

	return true;
}

void CIrcIdentServer::Stop()
{
	if( m_hThread )
	{
		m_socket.Close();
		if( WaitForSingleObject(m_hThread, 5000) != WAIT_OBJECT_0 && m_hThread )
		{
			TerminateThread(m_hThread, 1);
			CloseHandle(m_hThread);
			m_hThread = NULL;
		}
	}
}

void CIrcIdentServer::DoThread()
{
	m_socket.Listen();

	while( (bool)m_socket )
	{
		Socket s = m_socket.Accept();
		if( !s )
			break;

		char szBuf[1024];
		int cbRead = s.Receive((unsigned char*)szBuf, sizeof(szBuf)-1);
		if( cbRead <= 0 )
			continue;
		szBuf[cbRead] = '\0';

		// strip CRLF from query
		for(char* p = szBuf; *p && *p != '\r' && *p != '\n'; ++p)
			;
		*p = '\0';

		s.Send("%s : USERID : %s : %s\r\n", 
			szBuf, m_sResponseType.c_str(), m_sUserID.c_str());
		Sleep(500);
		s.Close();
	}

	m_socket.Close();
}

DWORD WINAPI CIrcIdentServer::ListenProc(LPVOID pparam)
{
	CIrcIdentServer* pThis = (CIrcIdentServer*)pparam;

	try { pThis->DoThread(); } catch( ... ) {}

	pThis->m_sResponseType = "";
	pThis->m_sUserID = "";
	pThis->m_uiPort = 0;

	CloseHandle(pThis->m_hThread);
	pThis->m_hThread = NULL;

	return 0;
}

////////////////////////////////////////////////////////////////////
/*std::vector<unsigned short> CIrcDCCServer::m_usedPorts;
std::vector<HANDLE> CIrcDCCServer::m_hThread;
const unsigned short CIrcDCCServer::kFirstPort = 1024;
const unsigned short CIrcDCCServer::kLastPort = 5000;

////////////////////////////////////////////////////////////////////
CIrcDCCServer::~CIrcDCCServer()
{
	Stop();
}

unsigned short CIrcDCCServer::MakePortReservation()
{
	//Assume the first port number is open
	unsigned short uiPort = kFirstPort;
	
	//if there are no used ports in the list use the first one
	if (m_usedPorts.size() <= 0)
		m_usedPorts.push_back(kFirstPort);

	//next see if the port range is saturated (pigeon hole principal)
	else if ((*m_usedPorts.end() - kFirstPort + 1) >= m_usedPorts.size())
		m_usedPorts.push_back(*m_usedPorts.end() + 1);

	//there must be a find a gap in the used port list
	//since the list is always sorted
	else
	{
		std::vector<unsigned short>::iterator iter;
		for (iter = m_usedPorts.begin(); iter < m_usedPorts.end(); iter++)
		{
			//oops, this port is being used, let's try the next one over
			if (*iter == uiPort) uiPort++;
			//looks like we found a gap, stop here
			else break;
		}

		//Insert our port number into the gap we just found
		m_usedPorts.insert(iter, uiPort);
	}

	return uiPort;
}

void CIrcDCCServer::FreePort(unsigned short port)
{
	//Tell used port list that this port is now open
	std::vector<unsigned short>::iterator iter;
	for (iter = m_usedPorts.begin(); iter < m_usedPorts.end(); iter++)
	{
		//Find the port number, remove it
		if (*iter == port)
		{
			m_usedPorts.erase(iter);
			return;
		}
	}
}

void CIrcDCCServer::FreeThread(HANDLE pThread)
{
	//If thread list is empty, bail
	if (m_hThread.size() <= 0) return;

	//Tell thread ptr list that this thread is done
	std::vector<HANDLE>::iterator iter;
	for (iter = m_hThread.begin(); iter < m_hThread.end(); iter++)
	{
		//is this the the thread were looking for?
		if ( *iter == pThread )
		{
			//Now we can forget about the thread
			CloseHandle(*iter);
			m_hThread.erase(iter);
		}
	}
}

bool CIrcDCCServer::Start(DCCTransferInfo dccinfo)
{
	//Make sure we have a port reservation made if were hosting a file
	if (dccinfo.m_bIsSender && dccinfo.m_uiPort < kFirstPort)
		return false;

	//Remember when we started and who spawned us
	dccinfo.m_ulStartTime = GetTickCount();
	dccinfo.m_pDCCSever = this;

	//Create the appropriate thread, but don't start it just yet
	if (dccinfo.m_bIsSender)
		dccinfo.m_pThread = CreateThread(NULL, 0, DoThreadSend, (void *)&dccinfo, 
											CREATE_SUSPENDED, NULL);
	else
		dccinfo.m_pThread = CreateThread(NULL, 0, DoThreadRecv, (void *)&dccinfo,
											CREATE_SUSPENDED, NULL);

	//Rememeber who what threads we spawned
	m_hThread.push_back(dccinfo.m_pThread);

	//Now we can fire off the thread
	if (-1 == ResumeThread(dccinfo.m_pThread)) return false;
	Sleep(100);

	return true;
}

void CIrcDCCServer::Stop(DWORD timeout, HANDLE hThread)
{
	//make sure we have a threads running
	if (m_hThread.size() <= 0) return;

	//Generate an iterator for the thread list
	std::vector<HANDLE>::iterator iter;

	//if the thread specified is NULL, kill all of the threads in the list
	if (hThread == NULL)
	{
		//Start at the beginning
		iter = m_hThread.begin();
		while(m_hThread.size() > 0)
		{
			//wait for each thread to finish
			if( *iter && WaitForSingleObject(*iter, timeout) != WAIT_OBJECT_0 )
			{
				TerminateThread(*iter, 1);
				CloseHandle(*iter);
			}

			//now we can forget about the thread
			iter = m_hThread.erase(iter);
		}
	}

	//otherwise, find all a specific thread to close
	else
	{
		for (iter = m_hThread.begin(); iter < m_hThread.end(); iter++)
		{
			//is this the the thread were looking for?
			if ( *iter == hThread )
			{
				//wait for each thread to finish
				if( *iter && WaitForSingleObject(*iter, timeout) != WAIT_OBJECT_0 )
				{
					TerminateThread(*iter, 1);
					CloseHandle(*iter);
				}

				//now we can forget about the thread
				m_hThread.erase(iter);
			}
		}
	}
}

DWORD WINAPI CIrcDCCServer::DoThreadRecv(void* dccInfo)
{
	char szBuf[8193];
	unsigned long cbRead, cbSent, replyTotal;
	float fracSent;
	unsigned long seconds, minutes, hours, rate;

	//Make a local copy of the dcc info
	DCCTransferInfo info = *((DCCTransferInfo*)dccInfo);

	//Create a file transfer dialog
	CSendFileDialog* pSendFileDialog = new CSendFileDialog(NULL);
	CWnd* pParent= pSendFileDialog->FromHandle( AfxGetMainWnd()->m_hWnd );
	if (!pParent || !pSendFileDialog || !pSendFileDialog->Create(IDD_FILEXFER, pParent) )
		return false;

	//Setup the status info on the dialog
	pSendFileDialog->m_BytesSent = "0 bytes";
	pSendFileDialog->m_ProgressFile.SetRange(0, 100);
	pSendFileDialog->m_ProgressFile.SetPos(0);
	pSendFileDialog->m_FolderName = info.m_directory;
	pSendFileDialog->m_FileName = info.m_fileName;
	pSendFileDialog->m_RecvrName = info.m_partnerName;
	pSendFileDialog->m_TimeLeft = "Infinite";
	pSendFileDialog->m_XferRate = "0 bytes/sec";
	pSendFileDialog->m_XferStatus = "Waiting to Connect...";
	pSendFileDialog->m_SentRecvd = "Recv'd:";
	pSendFileDialog->m_ToFrom = "From:";
	sprintf(pSendFileDialog->m_Filesize.GetBuffer(32), "%lu bytes", 
			info.m_ulFileSize);

	//Show the dialog and tell it about this server
	pSendFileDialog->ShowWindow(SW_SHOW);

	//Create an active socket to retrieve data from (close listening socket)
	InetAddr addr;
	Socket sock;
	addr.host = htonl(info.m_ulPartnerIP);
	addr.port = htons(info.m_uiPort);
	sock.Connect(addr);
	
	//bail if we can't open the socket
	if ( !sock )
	{
		pSendFileDialog->ShowWindow(SW_HIDE);
		delete pSendFileDialog;
		return 0;
	}

	//Tell user file transfer has started
	pSendFileDialog->m_XferStatus = "Recieving File...";

	//Create a file to write incoming data to
	CString filename = info.m_directory+info.m_fileName;
	FILE* fp = fopen(filename, "wb");
	
	//bail if we can't open the file
	if ( !sock )
	{
		pSendFileDialog->ShowWindow(SW_HIDE);
		delete pSendFileDialog;
		return 0;
	}

	//Start grabbing data blocks
	while(info.m_ulBytesSent < info.m_ulFileSize && !pSendFileDialog->isCanceled())
	{
		//Grab the next buffer
		cbRead = sock.Receive((unsigned char*)szBuf, sizeof(szBuf)-1);
			
		if( cbRead <= 0 ) continue;
		else szBuf[cbRead] = '\0';

		//Write it to file
		fwrite(szBuf, cbRead, 1, fp);

		//update byte count
		info.m_ulBytesSent += cbRead;

		//Update the file transfer statistics
		fracSent = float(info.m_ulBytesSent)/float(info.m_ulFileSize);
		seconds = (GetTickCount() - info.m_ulStartTime)/1000;
		rate = (seconds > 0) ? info.m_ulBytesSent / seconds : 0;
		seconds = (unsigned long)(float(seconds)/fracSent);
		minutes = (seconds/60) % 60;
		hours = (seconds/3600);
		seconds = seconds % 60;

		//Update the file transfer window
		sprintf(pSendFileDialog->m_BytesSent.GetBuffer(64), "%lu bytes", 
				info.m_ulBytesSent);
		sprintf(pSendFileDialog->m_TimeLeft.GetBuffer(32), "%luh%lum%lus", 
				hours, minutes, seconds);
		sprintf(pSendFileDialog->m_XferRate.GetBuffer(64), "%lubytes/sec", rate);
		pSendFileDialog->m_ProgressFile.SetPos(int(fracSent*100.F));

		//Tell the sender how many bytes we received
		replyTotal = htonl(info.m_ulBytesSent);

		//Send out response
		cbSent = sock.Send((unsigned char *)&replyTotal, 4);
	}

	//Close our data transfer socket
	sock.Close();

	//Close our file
	fclose(fp);

	//Clean up file dialog
	pSendFileDialog->ShowWindow(SW_HIDE);
	delete pSendFileDialog;

	//Free this thread from the system
	if (info.m_pDCCSever) 
		info.m_pDCCSever->FreeThread(info.m_pThread);

	return 0;
}

DWORD WINAPI CIrcDCCServer::DoThreadSend(void* dccInfo)
{
	char szBuf[8193];
	unsigned long cbSend, cbRead, numAck;
	float fracSent;
	unsigned long seconds, minutes, hours, rate;
	const k_ulTimeout = 10000;		//30 second timeout

	//Make a local copy of the dcc info
	DCCTransferInfo info = *((DCCTransferInfo*)dccInfo);

	//Create a file transfer dialog
	CSendFileDialog* pSendFileDialog = new CSendFileDialog(NULL);
	ASSERT (pSendFileDialog);

	CWnd* pParent= pSendFileDialog->FromHandle( AfxGetMainWnd()->m_hWnd );
	ASSERT (pParent);

	ASSERT (pSendFileDialog->Create(IDD_FILEXFER, pParent) == TRUE);
	//if (!pParent || !pSendFileDialog || !pSendFileDialog->Create(IDD_FILEXFER, pParent) )
	//	return false;

	//Setup the status info on the dialog
	pSendFileDialog->m_BytesSent = "0 bytes";
	pSendFileDialog->m_ProgressFile.SetRange(0, 100);
	pSendFileDialog->m_ProgressFile.SetPos(0);
	pSendFileDialog->m_FolderName = info.m_directory;
	pSendFileDialog->m_FileName = info.m_fileName;
	pSendFileDialog->m_RecvrName = info.m_partnerName;
	pSendFileDialog->m_TimeLeft = "Infinite";
	pSendFileDialog->m_XferRate = "0 bytes/sec";
	pSendFileDialog->m_XferStatus = "Waiting to Connect...";
	pSendFileDialog->m_SentRecvd = "Sent:";
	pSendFileDialog->m_ToFrom = "To:";
	sprintf(pSendFileDialog->m_Filesize.GetBuffer(32), "%lu bytes", 
			info.m_ulFileSize);

	//Show the dialog and tell it about this server
	pSendFileDialog->ShowWindow(SW_SHOW);

	//Wait for someone to connect to us
	//on a pre-arranged port
	Socket sockListen;
	sockListen.Bind(InetAddr(INADDR_ANY, htons(info.m_uiPort)));
	if (!sockListen.Listen())
	{
		pSendFileDialog->ShowWindow(SW_HIDE);
		delete pSendFileDialog;		
		return 0;
	}

	//Wait for partner to connect to us
	Socket sock;
	while(!sock && (GetTickCount() - info.m_ulStartTime) < k_ulTimeout)
		sock = sockListen.Accept();

	//Make sure socket is valid (we didn't timeout)
	if ( !sock )
	{
		pSendFileDialog->ShowWindow(SW_HIDE);
		delete pSendFileDialog;		
		return 0;
	}

	//Close the socket we were listening on
	sockListen.Close();

	//Tell user file transfer has started
	pSendFileDialog->m_XferStatus = "Sending File...";

	//Create a file to read incoming data from
	CString filename = info.m_directory+info.m_fileName;
	FILE* fp = fopen(filename, "rb");
	ASSERT( fp != NULL );
	if ( !fp )
	{
		pSendFileDialog->ShowWindow(SW_HIDE);
		delete pSendFileDialog;		
		return 0;
	}

	//Start grabbing data blocks
	while( info.m_ulBytesSent < info.m_ulFileSize && !pSendFileDialog->isCanceled() )
	{
		//Pull a data block from the file
		cbRead = fread(szBuf, info.m_uiXferRate, 1, fp);

		//Send the next chunk out
		cbSend = 0;
		while (cbSend <= 0)
			cbSend = sock.Send((unsigned char *)szBuf, cbRead);
		
		//Update sent byte count
		info.m_ulBytesSent += cbSend;

		//Make sure chunk gets acknowledged
		cbRead = 0;
		while( cbRead <= 0)
			cbRead = sock.Receive((unsigned char*)&numAck, 4);

		//Make sure numAck matches current amount of data seny
		//Update that statistics
		if (info.m_ulBytesSent != ntohl(numAck))
		{
			pSendFileDialog->ShowWindow(SW_HIDE);
			delete pSendFileDialog;	
			sock.Close();
			fclose(fp);

			return 0;
		}

		fracSent = float(info.m_ulBytesSent)/float(info.m_ulFileSize);
		seconds = (GetTickCount() - info.m_ulStartTime)/1000;
		rate = info.m_ulBytesSent / seconds;
		seconds = (unsigned long)(float(seconds)/fracSent);
		minutes = (seconds/60) % 60;
		hours = (seconds/3600);
		seconds = seconds % 60;

		//Update the file transfer window
		sprintf(pSendFileDialog->m_BytesSent.GetBuffer(64), "%lu bytes", 
				info.m_ulBytesSent);
		sprintf(pSendFileDialog->m_TimeLeft.GetBuffer(32), "%luh%lum%lus", 
				hours, minutes, seconds);
		sprintf(pSendFileDialog->m_XferRate.GetBuffer(64), "%lubytes/sec", rate);
		pSendFileDialog->m_ProgressFile.SetPos(int(fracSent*100.F));
	}

	//Close our data transfer socket
	sock.Close();

	//Close our file
	fclose(fp);

	//Clean up file dialog
	pSendFileDialog->ShowWindow(SW_HIDE);
	delete pSendFileDialog;

	//Free this thread and port from the system
	if (info.m_pDCCSever) 
	{
		info.m_pDCCSever->FreeThread(info.m_pThread);
		info.m_pDCCSever->FreePort(info.m_uiPort);
	}

	return 0;
}*/

////////////////////////////////////////////////////////////////////

CIrcMonitor::HandlersMap CIrcMonitor::m_handlers;
CIrcMonitor::IrcCommandsMapsListEntry CIrcMonitor::m_handlersMapsListEntry
	= { &CIrcMonitor::m_handlers, NULL };


CIrcMonitor::CIrcMonitor(CIrcSession& session)
	: m_session(session)
{
	m_xPost.SetMonitor(this);
}

CIrcMonitor::~CIrcMonitor()
{
}

void CIrcMonitor::OnIrcMessage(const CIrcMessage* pmsg)
{
	CIrcMessage* pMsgCopy = NULL;
	if( pmsg )
		pMsgCopy = new CIrcMessage(*pmsg);
	m_xPost.Post(0, (LPARAM)pMsgCopy);
}

void CIrcMonitor::OnCrossThreadsMessage(WPARAM wParam, LPARAM lParam)
{
	CIrcMessage* pmsg = (CIrcMessage*)lParam;

	OnIrcAll(pmsg);

	if( pmsg )
	{
		PfnIrcMessageHandler pfn = FindMethod(pmsg->sCommand.c_str());
		if( pfn )
		{
			// call member function. if it returns 'false',
			// call the default handling
			if( !(this->*pfn)(pmsg) )
				OnIrcDefault(pmsg);
		}
		else // handler not found. call default handler
			OnIrcDefault(pmsg);
		delete pmsg;
	}
	else
		OnIrcDisconnected();
}

CIrcMonitor::PfnIrcMessageHandler CIrcMonitor::FindMethod(const char* lpszName)
{
	// call the recursive version with the most derived map
	return FindMethod(GetIrcCommandsMap(), lpszName);
}

CIrcMonitor::PfnIrcMessageHandler CIrcMonitor::FindMethod(IrcCommandsMapsListEntry* pMapsList, const char* lpszName)
{
	HandlersMap::iterator it = pMapsList->pHandlersMap->find(lpszName);
	if( it != pMapsList->pHandlersMap->end() )
		return it->second; // found !
	else if( pMapsList->pBaseHandlersMap )
		return FindMethod(pMapsList->pBaseHandlersMap, lpszName); // try at base class
	return NULL; // not found in any map
}

////////////////////////////////////////////////////////////////////

DECLARE_IRC_MAP(CIrcDefaultMonitor, CIrcMonitor)

CIrcDefaultMonitor::CIrcDefaultMonitor(CIrcSession& session)
	: CIrcMonitor(session)
{
	IRC_MAP_ENTRY(CIrcDefaultMonitor, "NICK", OnIrc_NICK)
	IRC_MAP_ENTRY(CIrcDefaultMonitor, "PING", OnIrc_PING)
	IRC_MAP_ENTRY(CIrcDefaultMonitor, "002", OnIrc_YOURHOST)
	IRC_MAP_ENTRY(CIrcDefaultMonitor, "005", OnIrc_BOUNCE)
}

bool CIrcDefaultMonitor::OnIrc_NICK(const CIrcMessage* pmsg)
{
	if( (m_session.GetInfo().sNick == pmsg->prefix.sNick) && (pmsg->parameters.size() > 0) )
		m_session.m_info.sNick = pmsg->parameters[0];
	return false;
}

bool CIrcDefaultMonitor::OnIrc_PING(const CIrcMessage* pmsg)
{
	char szResponse[100];
	sprintf(szResponse, "PONG %s", pmsg->parameters[0].c_str());
	m_session << CIrcMessage(szResponse);
	return false;
}

bool CIrcDefaultMonitor::OnIrc_YOURHOST(const CIrcMessage* pmsg)
{
	static const char* lpszFmt = "Your host is %[^ \x5b,], running version %s";
	char szHostName[100], szVersion[100];
	if( sscanf(pmsg->parameters[1].c_str(), lpszFmt, &szHostName, &szVersion) > 0 )
		m_session.m_info.sServerName = szHostName;

	return false;
}

bool CIrcDefaultMonitor::OnIrc_BOUNCE(const CIrcMessage* pmsg)
{
	static const char* lpszFmt = "Try server %[^ ,], port %d";
	char szAltServer[100];
	int iAltPort = 0;
	if( sscanf(pmsg->parameters[1].c_str(), lpszFmt, &szAltServer, &iAltPort) == 2 )
	{
	}
	return false;
}