lyfz_repos/经典版源码/旧的后台处理/LYFZMsgManage/LYFZReceiveMsg/LYFZReceiveMsg.cpp

/****************************************************************/
/*																*/
/*  LYFZReceiveMsg.cpp												*/
/*																*/
/*  Implementation of the CLYFZReceiveMsg class.						*/
/*	This class is a part of the FTP Server Application			*/
/*																*/
/*  Programmed by LYFZ van der Meer							*/
/*  Copyright LYFZ Software Solutions 2002						*/
/*	http://www.LYFZvandermeer.nl								*/
/*																*/
/*  Last updated: 10 july 2002									*/
/*																*/
/****************************************************************/


#include "stdafx.h"
#include "LYFZReceiveMsg.h"
#include "MyLock.h"
extern int g_port;
CLYFZReceiveMsg *g_pWndServer=NULL;
CLYFZReceiveMsg::CLYFZReceiveMsg()
{
	m_nPort = g_port;
	m_nMaxUsers = 50;
	m_strWelcomeMessage = "Welcome to LYFZ FTP Server";
	m_strGoodbyeMessage = "Bye";
	m_nTimeout = 1;
	m_bRunning = FALSE;
	m_hWnd = NULL;
	m_nConnectionCount = 0;

	// intialize statistics
	m_dwTotalSentBytes = 0;
	m_dwTotalReceivedBytes = 0;
	m_nTotalConnections = 0;
	m_nFilesDownloaded = 0;
	m_nFilesUploaded = 0;
	m_nFailedDownloads = 0;
	m_nFailedUploads = 0;

	m_nSecurityMode = 0;

	m_nStatisticsInterval = 0;

	// load users
	m_UserManager.Serialize(FALSE);

	// load security
	m_SecurityManager.Serialize(FALSE);
}

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

BEGIN_MESSAGE_MAP(CLYFZReceiveMsg, CWnd)
	//{{AFX_MSG_MAP(CLYFZReceiveMsg)
	ON_WM_TIMER()
	//}}AFX_MSG_MAP
	ON_MESSAGE(WM_THREADSTART, OnThreadStart)
	ON_MESSAGE(WM_THREADCLOSE, OnThreadClose)
	ON_MESSAGE(WM_THREADMSG, OnThreadMessage)
END_MESSAGE_MAP()



/********************************************************************/
/*																	*/
/* Function name : Start											*/
/* Description   : Start listining on port 21 and accept new		*/
/*				   connections.										*/
/*																	*/
/********************************************************************/
//在21号端口启动侦听并准备接受新的连接
BOOL CLYFZReceiveMsg::Start()
{
	if (m_bRunning)
		return FALSE;

	// 为消息发送创建一个新的窗口
	if (!CWnd::CreateEx(0, AfxRegisterWndClass(0), "FTP Server Notification Sink", WS_POPUP, 0,0,0,0, NULL, 0))
	{
		AddTraceLine(0, "Failed to create notification window.");
		return FALSE;
	}
	// 创建一个新的套接字
	if (m_ListenSocket.Create(g_port))
	{
		// 启动侦听
		if (m_ListenSocket.Listen())
		{
			g_pWndServer=this;
			//	m_ListenSocket.m_pWndServer = this;
			m_bRunning = TRUE;	

			SetTimer(1, m_nStatisticsInterval, NULL);

			AddTraceLine(0, "FTP Server started on port %d.", g_port);
			return TRUE;
		}
	}
	AddTraceLine(0, "FTP Server failed to listen on port %d.", g_port);

	// 销毁通知窗口
	if (IsWindow(m_hWnd))
		DestroyWindow();
	m_hWnd = NULL;

	return FALSE;
}


/********************************************************************/
/*																	*/
/* Function name : Stop												*/
/* Description   : Stop FTP server.									*/
/*																	*/
/********************************************************************/
//停止FTP服务器
void CLYFZReceiveMsg::Stop()
{
	if (!m_bRunning)
		return;

	// 停止统计计时器
	KillTimer(1);

	m_bRunning = FALSE;	
	m_ListenSocket.Close();

	CConnectThread* pThread = NULL;

	//关闭所有线程
	do
	{
		m_CriticalSection.Lock();

		POSITION pos = m_ThreadList.GetHeadPosition();
		if (pos != NULL)
		{
			pThread = (CConnectThread *)m_ThreadList.GetAt(pos);

			m_CriticalSection.Unlock();

			// 保存线程
			int nThreadID = pThread->m_nThreadID;
			HANDLE hThread = pThread->m_hThread;

			AddTraceLine(0, "[%d] Shutting down thread...", nThreadID);

			// 通知线程停止
			pThread->SetThreadPriority(THREAD_PRIORITY_HIGHEST);
			pThread->PostThreadMessage(WM_QUIT,0,0);

			//等待线程终止
			if (WaitWithMessageLoop(hThread, 5000) == FALSE)
			{
				// 线程不能终止
				AddTraceLine(0, "[%d] Problem while killing thread.", nThreadID);
				//清除线程
				m_CriticalSection.Lock();
				POSITION rmPos = m_ThreadList.Find(pThread);
				if (rmPos != NULL)
					m_ThreadList.RemoveAt(rmPos);
				m_CriticalSection.Unlock();
			}
			else
			{
				AddTraceLine(0, "[%d] Thread successfully stopped.", nThreadID);
			}
		}
		else
		{
			m_CriticalSection.Unlock();	
			pThread = NULL;
		}
	}
	while (pThread != NULL);
	//更新服务器状态
	AddTraceLine(0, "FTP Server stopped.");

	if (IsWindow(m_hWnd))
		DestroyWindow();

	m_hWnd = NULL;
}


/********************************************************************/
/*																	*/
/* Function name : IsActive											*/
/* Description   : Is FTP server active?							*/
/*																	*/
/********************************************************************/
BOOL CLYFZReceiveMsg::IsActive()
{
	return m_bRunning;
}


/********************************************************************/
/*																	*/
/* Function name : SetMaxUsers										*/
/* Description   : Set maximum number of users						*/
/*																	*/
/********************************************************************/
void CLYFZReceiveMsg::SetMaxUsers(int nValue)
{
	m_nMaxUsers = nValue;
}


/********************************************************************/
/*																	*/
/* Function name : SetPort											*/
/* Description   : Set listening port for new connections			*/
/*																	*/
/********************************************************************/
void CLYFZReceiveMsg::SetPort(int nValue)
{
	m_nPort = nValue;
}


/********************************************************************/
/*																	*/
/* Function name : SetTimeout										*/
/* Description   : Set connection timeout							*/
/*																	*/
/********************************************************************/
void CLYFZReceiveMsg::SetTimeout(int nValue)
{
	m_nTimeout = nValue;
}


/********************************************************************/
/*																	*/
/* Function name : SetTimeout										*/
/* Description   : Set connection timeout							*/
/*																	*/
/********************************************************************/
void CLYFZReceiveMsg::SetStatisticsInterval(int nValue)
{
	m_nStatisticsInterval = nValue;
	if (m_nStatisticsInterval != 0)
	{
		KillTimer(1);
		SetTimer(1, nValue, NULL);
	}
	else
	{
		KillTimer(1);
	}
}


/********************************************************************/
/*																	*/
/* Function name : SetWelcomeMessage								*/
/* Description   : Set welcome message								*/
/*																	*/
/********************************************************************/
void CLYFZReceiveMsg::SetWelcomeMessage(LPCTSTR lpszText)
{
	m_strWelcomeMessage = lpszText;
}


/********************************************************************/
/*																	*/
/* Function name : SetGoodbyeMessage								*/
/* Description   : Set goodbye message								*/
/*																	*/
/********************************************************************/
void CLYFZReceiveMsg::SetGoodbyeMessage(LPCTSTR lpszText)
{
	m_strGoodbyeMessage = lpszText;
}


/********************************************************************/
/*																	*/
/* Function name : Initialize										*/
/* Description   : Initialize event sink.							*/
/*																	*/
/********************************************************************/
void CLYFZReceiveMsg::Initialize(CFTPEventSink *pEventSink)
{
	m_pEventSink = pEventSink;
}


/********************************************************************/
/*																	*/
/* Function name : AddTraceLine										*/
/* Description   : FTP status change.								*/
/*																	*/
/********************************************************************/
void CLYFZReceiveMsg::AddTraceLine(int nType, LPCTSTR pstrFormat, ...)
{
	CString str;

	// format and write the data we were given
	va_list args;
	va_start(args, pstrFormat);
	str.FormatV(pstrFormat, args);
	m_pEventSink->OnFTPStatusChange(nType, str);
}


/********************************************************************/
/*																	*/
/* Function name : OnThreadStart									*/
/* Description   : Called when a new thread has started.			*/
/*																	*/
/********************************************************************/
LRESULT CLYFZReceiveMsg::OnThreadStart(WPARAM wParam, LPARAM)
{
	m_nConnectionCount++;
	m_nTotalConnections++;
	CConnectThread *pThread = (CConnectThread *)wParam;


	//	pThread->m_ConnectSocket.GetPeerName(pThread->m_strRemoteHost, port);
	AddTraceLine(0, "[%d] Client connected from %s.", pThread->m_nThreadID, pThread->m_strRemoteHost);

	AfxGetMainWnd()->KillTimer (1);
	AfxGetMainWnd()->SetTimer(1, 5*60*1000, NULL);
	return TRUE;
}


/********************************************************************/
/*																	*/
/* Function name : OnThreadClose									*/
/* Description   : Called when a thread is about to stop.			*/
/*																	*/
/********************************************************************/
LRESULT CLYFZReceiveMsg::OnThreadClose(WPARAM wParam, LPARAM lParam)
{
	m_nConnectionCount--;
	CConnectThread *pThread = (CConnectThread *)wParam;

	AddTraceLine(0, "[%d] Client disconnected from %s.", pThread->m_nThreadID, pThread->m_strRemoteHost);

	m_pEventSink->OnFTPUserDisconnected(pThread->m_nThreadID, pThread->m_ConnectSocket.m_strUserName);
	return TRUE;
}


/********************************************************************/
/*																	*/
/* Function name : OnThreadMessage									*/
/* Description   : Message sent from connect connection.			*/
/*																	*/
/********************************************************************/
LRESULT CLYFZReceiveMsg::OnThreadMessage(WPARAM wParam, LPARAM lParam)
{
	switch(wParam)
	{
	case 0:
		m_dwTotalSentBytes += (int)lParam;
		break;
	case 1:
		m_dwTotalReceivedBytes += (int)lParam;
		break;
	case 2:
		switch(lParam)
		{
		case FTPSTAT_DOWNLOADSUCCEEDED:
			m_nFilesDownloaded++;
			break;
		case FTPSTAT_UPLOADSUCCEEDED:
			m_nFilesUploaded++;
			break;
		case FTPSTAT_DOWNLOADFAILED:
			m_nFailedDownloads++;
			break;
		case FTPSTAT_UPLOADFAILED:
			m_nFailedUploads++;
			break;
		}
		break;
	default:

		break;
	}
	return TRUE;
}


/********************************************************************/
/*																	*/
/* Function name : CheckMaxUsers									*/
/* Description   : Reached maximum number of connections?			*/
/*																	*/
/********************************************************************/
BOOL CLYFZReceiveMsg::CheckMaxUsers()
{
	if (m_nConnectionCount > m_nMaxUsers)
		return TRUE;
	else
		return FALSE;
}


/********************************************************************/
/*																	*/
/* Function name : OnTimer											*/
/* Description   : Update statictics.								*/
/*																	*/
/********************************************************************/
void CLYFZReceiveMsg::OnTimer(UINT nIDEvent) 
{
	// update statictics ?
	if (nIDEvent == 1)
	{
		m_pEventSink->OnFTPSentBytesChange(m_dwTotalSentBytes);
		m_pEventSink->OnFTPReceivedBytesChange(m_dwTotalReceivedBytes);
		m_pEventSink->OnFTPStatisticChange(0, m_nTotalConnections);
		m_pEventSink->OnFTPStatisticChange(1, m_nConnectionCount);
		m_pEventSink->OnFTPStatisticChange(2, m_nFilesDownloaded);
		m_pEventSink->OnFTPStatisticChange(3, m_nFilesUploaded);
		m_pEventSink->OnFTPStatisticChange(4, m_nFailedDownloads);
		m_pEventSink->OnFTPStatisticChange(5, m_nFailedUploads);
	}
	CWnd::OnTimer(nIDEvent);
}


/********************************************************************/
/*																	*/
/* Function name : SetSecurityMode									*/
/* Description   : Set security mode.								*/
/*																	*/
/********************************************************************/
void CLYFZReceiveMsg::SetSecurityMode(BOOL bBlockSpecific)
{
	m_nSecurityMode = bBlockSpecific ? 0 : 1;
}

int FindArray2(CStringArray *pArray, CString Str)
{
	for(int i=0; i<pArray->GetSize (); i++)
	{
		if(pArray->ElementAt (i)==Str)
			return i;
	}
	return -1;
}
extern CStringArray g_ipnoallowarray;
/********************************************************************/
/*																	*/
/* Function name : IsIPAddressAllowed								*/
/* Description   : Check (based on blockmode) if IP is allowed.		*/
/*																	*/
/********************************************************************/
BOOL CLYFZReceiveMsg::IsIPAddressAllowed(LPCTSTR lpszIPAddress)
{
	MyLock lock("IsIPAddressAllowed");
	if(::FindArray2 (&g_ipnoallowarray, lpszIPAddress)!=-1)return 0;
	return 1;
	/*	if (m_nSecurityMode == 0)
	{
	return !m_SecurityManager.IsIPAddressBlocked(lpszIPAddress);
	}
	else
	{
	return m_SecurityManager.IsIPAddressNonBlocked(lpszIPAddress);
	}*/
}