commandqueue.cpp

来自「一个支持FTP,SFTP的客户端程序」· C++ 代码 · 共 930 行 · 第 1/2 页

CPP
930 行
字号
12
// FileZilla - a Windows ftp client

// Copyright (C) 2002-2004 - Tim Kosse <tim.kosse@gmx.de>

// This program is free software; you can redistribute it and/or
// modify it under the terms of the GNU General Public License
// as published by the Free Software Foundation; either version 2
// of the License, or (at your option) any later version.

// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.

// You should have received a copy of the GNU General Public License
// along with this program; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.

// CommandQueue.cpp: Implementierung der Klasse CCommandQueue.
//
//////////////////////////////////////////////////////////////////////

#include "stdafx.h"
#include "CommandQueue.h"
#include "mainfrm.h"
#include "queueview.h"
#include "queuectrl.h"
#include "statusview.h"
#include "ftplistctrl.h"
#include "entersomething.h"
#include "version.h"
#include "localview2.h"

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

//////////////////////////////////////////////////////////////////////
// Konstruktion/Destruktion
//////////////////////////////////////////////////////////////////////

CCommandQueue::CCommandQueue(CFileZillaApi *pFileZillaApi)
{
	ASSERT(pFileZillaApi);
	m_pFileZillaApi=pFileZillaApi;
	m_nRetryCount=0;
	
	m_pLastServer=0;
	m_pLastServerPath=0;

	m_bLocked=0;
}

CCommandQueue::~CCommandQueue()
{
	delete m_pLastServer;
	delete m_pLastServerPath;
}

BOOL CCommandQueue::Connect(const t_server &server)
{
	m_nRetryCount=0; //Reset retry count

	//Remember the last server
	COptions::SetOption(OPTION_LASTSERVERHOST,server.host);
	COptions::SetOption(OPTION_LASTSERVERPORT,server.port);
	COptions::SetOption(OPTION_LASTSERVERUSER,server.user);
	COptions::SetOption(OPTION_LASTSERVERACCOUNT,server.account);
	COptions::SetOption(OPTION_LASTSERVERDONTREMEMBERPASS,server.bDontRememberPass?1:0);
	COptions::SetOption(OPTION_LASTSERVERUTF8,server.nUTF8);
	COptions::SetOption(OPTION_LASTSERVERNAME,server.name);
	if (!server.bDontRememberPass && !_ttoi(COptions::GetOption(OPTION_RUNINSECUREMODE)))
		COptions::SetOption(OPTION_LASTSERVERPASS,CCrypt::encrypt(server.pass));
	else
		COptions::SetOption(OPTION_LASTSERVERPASS,"");
	COptions::SetOption(OPTION_LASTSERVERFWBYPASS,server.fwbypass);			
	COptions::SetOption(OPTION_LASTSERVERPATH,"");
	if (m_pLastServerPath)
	{
		delete m_pLastServerPath;
		m_pLastServerPath=0;
	}
	COptions::SetOption(OPTION_LASTSERVERTYPE, server.nServerType);
	COptions::SetOption(OPTION_LASTSERVERTRANSFERMODE, server.nPasv);
	if (!m_pLastServer)
		m_pLastServer=new t_server;
	*m_pLastServer=server;
	
	t_command command;
	command.id = FZ_COMMAND_CONNECT;
	command.server = server;
	if (!m_CommandList.empty())
	{
		m_CommandList.push_back(command);
		return TRUE;
	}
	else
	{
		int res=m_pFileZillaApi->Connect(server);
		if (res==FZ_REPLY_WOULDBLOCK)
		{
			m_RetryServerList.AddServer(server);
			m_CommandList.push_back(command);
		
			CString title;
			CString str;
			if (server.name!="")
				str=server.name + _T(" (") + server.host + _T(")");
			else
				str=server.host;
			title.Format(IDS_TITLE_CONNECTING, str);
			SetWindowText( _T("FileZilla - ") + title);
			return TRUE;
		}
		else if (res==FZ_REPLY_OK)
		{
			CString title;
			CString str;
			if (server.name!="")
				str=server.name + _T(" (") + server.host + _T(")");
			else
				str=server.host;
			title.Format(IDS_TITLE_CONNECTED, str);
			SetWindowText( _T("FileZilla - ") + title);
			return TRUE;
		}
		else
			return FALSE;
	}
}

void CCommandQueue::ProcessReply(int nReplyCode, LPARAM lParam)
{
	int nOldSize = m_CommandList.size() - 1;
	if (nReplyCode&FZ_REPLY_OK || nReplyCode&FZ_REPLY_ERROR || nReplyCode&FZ_REPLY_CANCEL || nReplyCode&FZ_REPLY_DISCONNECTED)
	{
		int nOldCommand=0;
		CMainFrame *pMainFrame=DYNAMIC_DOWNCAST(CMainFrame,AfxGetMainWnd());
		if (!lParam)
		{ //Out of order message, connection may have been closed by remote side
			if (nReplyCode & FZ_REPLY_DISCONNECTED && m_CommandList.empty())
			{
				//Failed to connect, retry if possible
				if (m_nRetryCount < 0)
					m_nRetryCount *= -1;
				m_nRetryCount++;
				if (m_nRetryCount <= COptions::GetOptionVal(OPTION_NUMRETRIES))
				{
					t_server server;
					CServerPath path;
					if (GetLastServer(server, path))
					{
						t_command command;
						command.id = FZ_COMMAND_CONNECT;
						command.server = server;
						m_CommandList.push_front(command);
						List(path, FZ_LIST_USECACHE, TRUE);
						OutputRetryMessage();
						CString title;
						CString str;
						if (server.name!="")
							str = server.name + _T(" (") + server.host + _T(")");
						else
							str = server.host;
						title.Format(IDS_TITLE_CONNECTING, str);
						SetWindowText( _T("FileZilla - ") + title);
					}
				}
				else
					SetWindowText(GetVersionString());
			}
			else if (nReplyCode & FZ_REPLY_DISCONNECTED)
				SetWindowText(GetVersionString());
			return;
		}
		else
			if (!m_CommandList.empty())
			{
				t_command command = m_CommandList.front();
				m_CommandList.pop_front();

				nOldCommand=command.id;
				if (command.id==FZ_COMMAND_FILETRANSFER)
				{
					t_transferfile transferfile=command.transferfile;
					
					//Dispatch result to the command queue
					pMainFrame->GetQueuePane()->GetListCtrl()->ProcessReply(nReplyCode);
					if (transferfile.get)
						pMainFrame->GetLocalPane2()->RefreshFile(transferfile.localfile);
					if (!m_CommandList.empty())
						return;
				}
				else if (command.id==FZ_COMMAND_CONNECT)
				{
					if (nReplyCode&FZ_REPLY_CRITICALERROR)
					{
						if (m_pLastServer)
						{
							if (m_pLastServer->bDontRememberPass)
								m_pLastServer->pass="";
						}
					}
					else if (!(nReplyCode&(FZ_REPLY_OK|FZ_REPLY_CANCEL)))
					{
						//Failed to connect, retry if possible
						if (m_nRetryCount < 0)
							m_nRetryCount *= -1;
						m_nRetryCount++;
						if (m_nRetryCount<=COptions::GetOptionVal(OPTION_NUMRETRIES))
						{
							m_CommandList.push_front(command);
							OutputRetryMessage();
							return;
						}
					}
					if (nReplyCode&FZ_REPLY_OK)
					{
						CString title;
						CString str;
						if (command.server.name!="")
							str=command.server.name + _T(" (") + command.server.host + _T(")");
						else
							str=command.server.host;
						title.Format(IDS_TITLE_CONNECTED, str);
						SetWindowText( _T("FileZilla - ") + title);
					}
				}
				else if (command.id==FZ_COMMAND_LIST)
				{
					if (nReplyCode!=FZ_REPLY_OK)
						((CFtpListCtrl *)pMainFrame->GetFtpPane()->GetListCtrl())->ListComplete(FALSE, command.path, command.param1);
					else
						((CFtpListCtrl *)pMainFrame->GetFtpPane()->GetListCtrl())->ListComplete(TRUE, command.path, command.param1);
				}
			}
		if (nOldSize)
			while (!m_CommandList.empty())
			{
				t_command command = m_CommandList.front();
				int res = m_pFileZillaApi->Command(&command);
				if (res==FZ_REPLY_WOULDBLOCK)
				{
					if (command.id==FZ_COMMAND_CONNECT)
					{
						m_RetryServerList.AddServer(command.server);
						CString title;
						CString str;
						if (command.server.name!="")
							str=command.server.name + _T(" (") + command.server.host + _T(")");
						else
							str=command.server.host;
						title.Format(IDS_TITLE_CONNECTING, str);
						SetWindowText( _T("FileZilla - ") + title);
					}
					return;
				}
				else
				{
					m_CommandList.pop_front();
					if (command.id == FZ_COMMAND_LIST)
					{
						((CFtpListCtrl *)pMainFrame->GetFtpPane()->GetListCtrl())->ListComplete((res == FZ_REPLY_OK)?TRUE:FALSE, command.path, command.param1);
					}
					else if (command.id == FZ_COMMAND_FILETRANSFER)
					{
						pMainFrame->GetQueuePane()->GetListCtrl()->ProcessReply(FZ_REPLY_ERROR);
						break;
					}
				}
			}
		if (m_CommandList.empty())
		{
			pMainFrame->GetQueuePane()->GetListCtrl()->TransferQueue(
				pMainFrame->GetQueuePane()->GetListCtrl()->DoProcessQueue()
				);
		}

		if (m_CommandList.empty() && nOldCommand!=FZ_COMMAND_LIST)
			if (((CFtpListCtrl *)pMainFrame->GetFtpPane()->GetListCtrl())->HasUnsureEntries())
				List(FZ_LIST_USECACHE|FZ_LIST_EXACT);	

		if (!IsConnected() && !IsBusy())
			SetWindowText(GetVersionString());
	}
	else
	{
		CMainFrame *pMainFrame = DYNAMIC_DOWNCAST(CMainFrame, AfxGetMainWnd());
		CString str;
		str.Format(_T("ProcessReply(%d) unknown reply code"), 1);
		pMainFrame->GetStatusPane()->ShowStatus(str, FZ_LOG_INFO);
	}
}


BOOL CCommandQueue::List(int nListMode /*=FZ_LIST_USECACHE*/, BOOL bAddToQueue /*=FALSE*/) //Lists current folder
{
	t_command command;
	command.id = FZ_COMMAND_LIST;
	command.param4 = nListMode;
	if (nListMode&FZ_LIST_USECACHE && !bAddToQueue)
	{
		int res=m_pFileZillaApi->List(nListMode|(m_CommandList.empty()?0:FZ_LIST_FORCECACHE));
		if (res==FZ_REPLY_OK)
			return TRUE;
		else if (res==FZ_REPLY_WOULDBLOCK)
		{
			m_CommandList.push_back(command);
			return TRUE;
		}
		else
			return FALSE;
	}
	
	if (!m_CommandList.empty())
	{
		if (!bAddToQueue)
			return FALSE;
		m_CommandList.push_back(command);
		return TRUE;
	}
	else
	{
		int res = m_pFileZillaApi->List(nListMode);
		if (res==FZ_REPLY_WOULDBLOCK)
		{
			m_CommandList.push_back(command);
			return TRUE;
		}
		else if (res==FZ_REPLY_OK)
			return TRUE;
		else
			return FALSE;
	}
}

BOOL CCommandQueue::List(CServerPath path, int nListMode /*=FZ_LIST_USECACHE*/, BOOL bAddToQueue /*=FALSE*/)
{
	t_command command;
	command.id = FZ_COMMAND_LIST;
	command.path = path;
	command.param4 = nListMode;
	if (nListMode&FZ_LIST_USECACHE && !bAddToQueue)
	{
		int res = m_pFileZillaApi->List(path,nListMode|(m_CommandList.empty()?0:FZ_LIST_FORCECACHE));
		if (res == FZ_REPLY_OK)
			return TRUE;
		else if (res == FZ_REPLY_WOULDBLOCK)
		{
			m_CommandList.push_back(command);
			return TRUE;
		}
		else
			return FALSE;
	}	
	if (!m_CommandList.empty())
	{
		if (!bAddToQueue)
			return FALSE;
		m_CommandList.push_back(command);
		return TRUE;
	}
	else
	{
		int res=m_pFileZillaApi->List(path,nListMode);
		if (res==FZ_REPLY_WOULDBLOCK)
		{
			m_CommandList.push_back(command);
			return TRUE;
		}
		else if (res==FZ_REPLY_OK)
			return TRUE;
		else
			return FALSE;
	}
}

int CCommandQueue::List(CServerPath parent, CString dirname, int nListMode /*=FZ_LIST_USECACHE*/, BOOL bAddToQueue /*=FALSE*/)
{
	t_command command;
	command.id = FZ_COMMAND_LIST;
	command.path = parent;
	command.param1 = dirname;
	command.param4 = nListMode;	
	if (nListMode&FZ_LIST_USECACHE && !bAddToQueue)
	{
		int res = m_pFileZillaApi->List(parent, dirname, nListMode|(m_CommandList.empty()?0:FZ_LIST_FORCECACHE));
		if (res==FZ_REPLY_OK)
			return TRUE;
		else if (res==FZ_REPLY_WOULDBLOCK)
		{
			m_CommandList.push_back(command);
			return TRUE;
		}
		else
			return FALSE;
	}
	
	if (!m_CommandList.empty())
	{
		if (!bAddToQueue)
			return FALSE;
		m_CommandList.push_back(command);
		return TRUE;
	}
	else
	{
		int res=m_pFileZillaApi->List(parent, dirname, nListMode);
		if (res==FZ_REPLY_WOULDBLOCK)
		{
			m_CommandList.push_back(command);
			return TRUE;
		}
		else if (res==FZ_REPLY_OK)
			return TRUE;
		else
			return FALSE;
	}
}

BOOL CCommandQueue::IsBusy()
{
	if (m_CommandList.empty())
		return FALSE;
	return TRUE;
}

BOOL CCommandQueue::FileTransfer(const t_transferfile TransferFile)
{
	t_command command;
	command.id=FZ_COMMAND_FILETRANSFER;
	command.transferfile = TransferFile;
	if (!m_CommandList.empty())
	{
		m_CommandList.push_back(command);
		return TRUE;
	}
	else
	{
		t_server server;
		int res=m_pFileZillaApi->GetCurrentServer(server);
		if (!IsConnected() || res!=FZ_REPLY_OK)
		{
			BOOL res=Connect(TransferFile.server);
			if (res)
			{
				m_CommandList.push_back(command);
				return TRUE;
			}
			else
				return FALSE;
		}
		else if (server!=TransferFile.server)
		{
			Disconnect();
			BOOL res=Connect(TransferFile.server);
			if (res)
			{
				m_CommandList.push_back(command);
				return TRUE;
			}
			else
				return FALSE;
12

commandqueue.cpp - 源码说明

本页面展示了「一个支持FTP,SFTP的客户端程序」中的 commandqueue.cpp 源码文件，采用 C++ 编程语言编写，共 930 行代码。您可以在线阅读完整代码内容，也可以返回资源详情页下载完整源码包进行本地学习和开发。

虫虫开发者社区收录了大量与FTP相关的技术资源，包括源代码、技术文档、电路图等，是电子工程师和嵌入式开发者的专业学习平台。

⌨️ 快捷键说明

复制代码Ctrl + C
搜索代码Ctrl + F
全屏模式F11
增大字号Ctrl + =
减小字号Ctrl + -
显示快捷键?