/*
 EasySFTP - Copyright (C) 2010 Kuri-Applications

 Folder.cpp - implementations of folder-helper classes
 */

#include "StdAfx.h"
#include "ShellDLL.h"
#include "Folder.h"
#include "FoldDrop.h"
#include "FldrMenu.h"

#include "RFolder.h"

static HMENU __stdcall GetSubMenuByID(HMENU hMenu, UINT uID)
{
	MENUITEMINFO mii;
	int nCount;

#ifdef _WIN64
	mii.cbSize = sizeof(mii);
#else
	mii.cbSize = MENUITEMINFO_SIZE_V1;
#endif
	mii.fMask = MIIM_ID | MIIM_SUBMENU;
	nCount = ::GetMenuItemCount(hMenu);
	for (int i = 0; i < nCount; i++)
	{
		::GetMenuItemInfo(hMenu, (UINT) i, TRUE, &mii);
		if (mii.wID == uID)
			return mii.hSubMenu;
	}
	return NULL;
}

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

CEnumFTPItemIDList::CEnumFTPItemIDList(CDelegateMallocData* pMallocData,
	const CMyPtrArrayT<CFTPFileItem>& arrItems,
	SHCONTF grfFlags, IUnknown* pUnkOuter)
	: m_pMallocData(pMallocData)
	, m_arrItems(arrItems)
	, m_grfFlags(grfFlags)
	, m_uPos(0)
	, m_pUnkOuter(pUnkOuter)
{
	m_pMallocData->AddRef();
	if (pUnkOuter)
		pUnkOuter->AddRef();
}

CEnumFTPItemIDList::~CEnumFTPItemIDList()
{
	if (m_pUnkOuter)
		m_pUnkOuter->Release();
	m_pMallocData->Release();
}

STDMETHODIMP CEnumFTPItemIDList::QueryInterface(REFIID riid, void** ppv)
{
	if (!ppv)
		return E_POINTER;
	if (IsEqualIID(riid, IID_IUnknown) ||
		IsEqualIID(riid, IID_IEnumIDList))
	{
		*ppv = this;
		AddRef();
		return S_OK;
	}
	return E_NOINTERFACE;
}

STDMETHODIMP CEnumFTPItemIDList::Next(ULONG celt, PITEMID_CHILD* rgelt, ULONG* pceltFetched)
{
	if (!rgelt)
		return E_POINTER;
	if (!celt)
		return S_OK;
	if (pceltFetched)
		*pceltFetched = 0;
	while (m_uPos < (ULONG) m_arrItems.GetCount())
	{
		CFTPFileItem* pItem = m_arrItems.GetItem((int) (m_uPos++));
		bool bFetch = false;
		if (pItem->IsDirectory())
			bFetch = ((m_grfFlags & SHCONTF_FOLDERS) != 0);
		else
			bFetch = ((m_grfFlags & SHCONTF_NONFOLDERS) != 0);
		if (bFetch && !(m_grfFlags & SHCONTF_INCLUDEHIDDEN))
			bFetch = !pItem->IsHidden();
		if (bFetch)
		{
			*rgelt = ::CreateFileItem(m_pMallocData->pMalloc, pItem);
			if (*rgelt && pceltFetched)
				(*pceltFetched)++;
			rgelt++;
			if (!--celt)
				return S_OK;
		}
	}
	return S_FALSE;
}

STDMETHODIMP CEnumFTPItemIDList::Skip(ULONG celt)
{
	if (!celt)
		return S_OK;
	if (m_uPos + celt > (ULONG) m_arrItems.GetCount())
	{
		m_uPos = (ULONG) m_arrItems.GetCount();
		return S_FALSE;
	}
	m_uPos += celt;
	return S_OK;
}

STDMETHODIMP CEnumFTPItemIDList::Reset()
{
	m_uPos = 0;
	return S_OK;
}

STDMETHODIMP CEnumFTPItemIDList::Clone(IEnumIDList** ppEnum)
{
	CEnumFTPItemIDList* pEnum = new CEnumFTPItemIDList(m_pMallocData, m_arrItems, m_grfFlags, m_pUnkOuter);
	pEnum->m_uPos = m_uPos;
	*ppEnum = pEnum;
	return S_OK;
}

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

// {36089763-B05F-4ed7-B7CB-D2597B21ADA7}
static const GUID GUID_FTPItemColumn =
	{ 0x36089763, 0xb05f, 0x4ed7, { 0xb7, 0xcb, 0xd2, 0x59, 0x7b, 0x21, 0xad, 0xa7 } };
#define PID_FTPITEM_BASE           200
#define STRING_ID_TO_MY_PID(u)     (PID_FTPITEM_BASE + (u) - IDS_HEAD_FILE_NAME + 1)
#define MY_PID_TO_STRING_ID(u)     (IDS_HEAD_FILE_NAME - 1 + (u) - PID_FTPITEM_BASE)
#define PID_FTPITEM_FILE_NAME      STRING_ID_TO_MY_PID(IDS_HEAD_FILE_NAME)
#define PID_FTPITEM_SIZE           STRING_ID_TO_MY_PID(IDS_HEAD_SIZE)
#define PID_FTPITEM_TYPE           STRING_ID_TO_MY_PID(IDS_HEAD_TYPE)
#define PID_FTPITEM_MODIFY_TIME    STRING_ID_TO_MY_PID(IDS_HEAD_MODIFY_TIME)
#define PID_FTPITEM_PERMISSIONS    STRING_ID_TO_MY_PID(IDS_HEAD_PERMISSIONS)
#define PID_FTPITEM_TRANSFER_TYPE  STRING_ID_TO_MY_PID(IDS_HEAD_TRANSFER_TYPE)
static const UINT s_uHeaderItemPIDs[] = {
	PID_FTPITEM_FILE_NAME,
	PID_FTPITEM_SIZE,
	PID_FTPITEM_TYPE,
	PID_FTPITEM_MODIFY_TIME,
	PID_FTPITEM_PERMISSIONS,
	PID_FTPITEM_TRANSFER_TYPE
};

//union SHFILEINFO_UNION
//{
//	SHFILEINFOA a;
//	SHFILEINFOW w;
//};

static DWORD __stdcall GetDummyFileAttribute(CFTPFileItem* pItem)
{
	if (pItem->bWinAttr)
		return pItem->dwAttributes;
	DWORD dwRet = 0;
	{
		register CFTPFileItem* p = pItem;
		while (p->pTargetFile)
			p = p->pTargetFile;
		if (p->type == fitypeDir)
			dwRet |= FILE_ATTRIBUTE_DIRECTORY;
		else
		{
			if (p->permissions.readable || !p->permissions.writable)
				dwRet |= FILE_ATTRIBUTE_READONLY;
		}
	}
	if (!pItem->strFileName.IsEmpty() && *((LPCWSTR) pItem->strFileName) == L'.')
		dwRet |= FILE_ATTRIBUTE_HIDDEN;
	return dwRet ? dwRet : FILE_ATTRIBUTE_NORMAL;
}

// also used in SFilePrp.cpp
extern void __stdcall FileTimeToString(const FILETIME* pft, _StringW& ret)
{
	FILETIME ft;
	SYSTEMTIME st;
	//_StringW str;
	::FileTimeToLocalFileTime(pft, &ft);
	::FileTimeToSystemTime(&ft, &st);

	ret.Format(L"%04hu/%02hu/%02hu %02hu:%02hu:%02hu",
		st.wYear, st.wMonth, st.wDay,
		st.wHour, st.wMinute, st.wSecond);
}

static const WCHAR* s_szSizeType[] = {
	NULL,
	L"KB",
	L"MB",
	L"GB",
	L"TB",
	L"PB",
	L"EB",
	L"ZB",
	L"YB"
};

// also used in Transfer.cpp and SFilePrp.cpp
extern void __stdcall FileSizeToString(ULARGE_INTEGER uli, _StringW& ret)
{
	int t;
	t = 0;
	while (uli.QuadPart >= 2048 && t < 8)
	{
		//uli.QuadPart /= 1024;
		uli.QuadPart >>= 10;
		t++;
	}
	ret.Format(L"%u ", (UINT) uli.LowPart);
	if (t > 0)
		ret += s_szSizeType[t];
	else
	{
		_StringW s;
		s.LoadString(IDS_BYTE);
		ret += s;
	}
}

static void __stdcall FillFileItemInfo(CFTPFileItem* pItem)
{
	SHFILEINFO_UNION sfi;
	DWORD dw;

	if (pItem->iIconIndex != -1)
		return;

	CFTPFileItem* p = pItem;
	while (p->pTargetFile)
		p = p->pTargetFile;
	_StringW str(p->strFileName);
	memset(&sfi.w, 0, sizeof(sfi.w));
	dw = GetDummyFileAttribute(pItem);
	if (!::SHGetFileInfoW(str, dw, &sfi.w, sizeof(sfi.w),
		SHGFI_USEFILEATTRIBUTES | SHGFI_SYSICONINDEX | SHGFI_SMALLICON | SHGFI_TYPENAME))
	{
		memset(&sfi.a, 0, sizeof(sfi.a));
		if (!::SHGetFileInfoA(str, dw, &sfi.a, sizeof(sfi.a),
			SHGFI_USEFILEATTRIBUTES | SHGFI_SYSICONINDEX | SHGFI_SMALLICON | SHGFI_TYPENAME))
			return;
		pItem->iIconIndex = sfi.a.iIcon;
		pItem->strType = sfi.a.szTypeName;
	}
	else
	{
		pItem->iIconIndex = sfi.w.iIcon;
		pItem->strType = sfi.w.szTypeName;
	}
	if (pItem->IsDirectory())
	{
		memset(&sfi.a, 0, sizeof(sfi.a));
		if (!::SHGetFileInfoA(str, dw, &sfi.a, sizeof(sfi.a),
			SHGFI_USEFILEATTRIBUTES | SHGFI_SYSICONINDEX | SHGFI_SMALLICON | SHGFI_OPENICON))
			return;
		pItem->iOpenIconIndex = sfi.a.iIcon;
	}
	else
		pItem->iOpenIconIndex = pItem->iIconIndex;
}

static void __stdcall GetDummyFileItemIcon(LPCWSTR lpszFileName, bool bIsDirectory, int& iIconIndex, int& iOpenIconIndex)
{
	SHFILEINFOA sfiA;
	DWORD dw;

	_StringW str(lpszFileName);
	memset(&sfiA, 0, sizeof(sfiA));
	dw = bIsDirectory ? FILE_ATTRIBUTE_DIRECTORY : FILE_ATTRIBUTE_NORMAL;
		if (!::SHGetFileInfoA(str, dw, &sfiA, sizeof(sfiA),
			SHGFI_USEFILEATTRIBUTES | SHGFI_SYSICONINDEX | SHGFI_SMALLICON))
			return;
		iIconIndex = sfiA.iIcon;
	if (bIsDirectory)
	{
		memset(&sfiA, 0, sizeof(sfiA));
		if (!::SHGetFileInfoA(str, dw, &sfiA, sizeof(sfiA),
			SHGFI_USEFILEATTRIBUTES | SHGFI_SYSICONINDEX | SHGFI_SMALLICON | SHGFI_OPENICON))
			return;
		iOpenIconIndex = sfiA.iIcon;
	}
	else
		iOpenIconIndex = iIconIndex;
}

static void __stdcall FilePermissionsToString(int nPerm, _StringW& ret)
{
	if (!nPerm)
		ret.Empty();
	else
	{
		// S_IFLNK == S_IFCHR | S_IFREG
		if ((nPerm & S_IFLNK) == S_IFLNK)
			ret = L"l";
		// S_IFBLK == S_IFCHR | S_IFDIR
		else if ((nPerm & S_IFBLK) == S_IFBLK)
			ret = L"b";
		else if ((nPerm & S_IFCHR) == S_IFCHR)
			ret = L"c";
		else if ((nPerm & S_IFDIR) == S_IFDIR)
			ret = L"d";
		else
			ret = L"-";
		ret += (nPerm & S_IRUSR) ? L'r' : L'-';
		ret += (nPerm & S_IWUSR) ? L'w' : L'-';
		if (nPerm & S_ISUID)
			ret += (nPerm & S_IXUSR) ? L's' : L'S';
		else
			ret += (nPerm & S_IXUSR) ? L'x' : L'-';
		ret += (nPerm & S_IRGRP) ? L'r' : L'-';
		ret += (nPerm & S_IWGRP) ? L'w' : L'-';
		if (nPerm & S_ISGID)
			ret += (nPerm & S_IXGRP) ? L's' : L'S';
		else
			ret += (nPerm & S_IXGRP) ? L'x' : L'-';
		ret += (nPerm & S_IROTH) ? L'r' : L'-';
		ret += (nPerm & S_IWOTH) ? L'w' : L'-';
		if (nPerm & S_ISVTX)
			ret += (nPerm & S_IXOTH) ? L't' : L'T';
		else
			ret += (nPerm & S_IXOTH) ? L'x' : L'-';
	}
}

static void __stdcall FileAttributesToString(DWORD dwAttribute, _StringW& ret)
{
	ret.Empty();
	if (!(dwAttribute & FILE_ATTRIBUTE_DIRECTORY))
	{
		if (dwAttribute & FILE_ATTRIBUTE_ARCHIVE)
			ret += L'A';
		else
			ret += L'-';
		if (dwAttribute & FILE_ATTRIBUTE_SYSTEM)
			ret += L'S';
		else
			ret += L'-';
		if (dwAttribute & FILE_ATTRIBUTE_HIDDEN)
			ret += L'H';
		else
			ret += L'-';
		if (dwAttribute & FILE_ATTRIBUTE_READONLY)
			ret += L'R';
		else
			ret += L'-';
	}
}

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

CFTPDirectoryBase::CFTPDirectoryBase(
	CDelegateMallocData* pMallocData,
	CFTPDirectoryItem* pItemMe,
	CFTPDirectoryBase* pParent,
	CFTPDirectoryRootBase* pRoot,
	LPCWSTR lpszDirectory)
	: CFolderBase(pMallocData)
	, m_pItemMe(pItemMe)
	, m_pParent(pParent)
	, m_pRoot(pRoot)
	, m_strDirectory(lpszDirectory)
{
	pParent->AddRef();
	pRoot->AddRef();
	m_bIsRoot = false;

	CommonConstruct();
}

CFTPDirectoryBase::CFTPDirectoryBase(
	CDelegateMallocData* pMallocData,
	CFTPDirectoryItem* pItemMe)
	: CFolderBase(pMallocData)
	, m_pItemMe(pItemMe)
{
	m_pParent = NULL;
	m_pRoot = NULL;
	m_bIsRoot = true;

	CommonConstruct();
}

void CFTPDirectoryBase::CommonConstruct()
{
	m_uRef = 1;
	m_bDirReceived = false;
	if (m_pItemMe)
		m_pItemMe->pDirectory = this;

	::InitializeCriticalSection(&m_csRefs);
	::InitializeCriticalSection(&m_csFiles);
	_StringW str;
	str.Format(L"CFTPDirectoryBase::CommonConstruct() for (0x%p), count = 1\n",
		(void*) this);
	OutputDebugString(str);
}

CFTPDirectoryBase::~CFTPDirectoryBase()
{
	if (m_pParent)
		m_pParent->Release();
	if (m_pItemMe)
		m_pItemMe->pDirectory = NULL;
	register int i = m_aDirectories.GetCount();
	while (i--)
	{
		CFTPDirectoryItem* p = m_aDirectories.GetItem(i);
		m_aDirectories.RemoveItem(i);
		//if (p->pDirectory)
		//	p->pDirectory->Release();
		delete p;
	}
	::EnterCriticalSection(&m_csFiles);
	i = m_aFiles.GetCount();
	while (i--)
	{
		CFTPFileItem* pItem = m_aFiles.GetItem(i);
		pItem->Release();
	}
	m_aFiles.RemoveAll();
	::LeaveCriticalSection(&m_csFiles);
	if (!m_bIsRoot)
		m_pRoot->Release();

	::DeleteCriticalSection(&m_csFiles);
	::DeleteCriticalSection(&m_csRefs);
}

STDMETHODIMP_(ULONG) CFTPDirectoryBase::AddRef()
{
	::EnterCriticalSection(&m_csRefs);
	//ULONG ret = (ULONG) ::InterlockedIncrement((LONG*) &m_uRef);
	ULONG ret = ++m_uRef;
	::LeaveCriticalSection(&m_csRefs);
//#ifdef _DEBUG
//	_StringW str;
//	str.Format(L"CFTPDirectoryBase::AddRef() for '%s' (0x%p), count = %lu\n",
//		(LPCWSTR) m_strDirectory, (void*) this, ret);
//	OutputDebugString(str);
//#endif
	return ret;
	//return (ULONG) ::InterlockedIncrement((LONG*) &m_uRef);
}

STDMETHODIMP_(ULONG) CFTPDirectoryBase::Release()
{
	::EnterCriticalSection(&m_csRefs);
	ULONG ret;
	//ret = (ULONG) ::InterlockedDecrement((LONG*) &m_uRef);
	ret = --m_uRef;
	if (!ret && m_pItemMe)
	{
		m_pItemMe->pDirectory = NULL;
		m_pItemMe = NULL;
	}
	::LeaveCriticalSection(&m_csRefs);

//#ifdef _DEBUG
//	_StringW str;
//	str.Format(L"CFTPDirectoryBase::Release() for '%s' (0x%p), count = %lu\n",
//		(LPCWSTR) m_strDirectory, (void*) this, ret);
//	OutputDebugString(str);
//#endif
	if (!ret)
		delete this;
	return ret;
}

STDMETHODIMP CFTPDirectoryBase::QueryInterface(REFIID riid, void** ppv)
{
	if (!ppv)
		return E_POINTER;
	if (IsEqualIID(riid, IID_IExtractIconA) || IsEqualIID(riid, IID_IExtractIconW) ||
		IsEqualIID(riid, IID_IDropTarget) || IsEqualIID(riid, IID_IContextMenu))
		return CreateViewObject(NULL, riid, ppv);
	else if (IsEqualIID(riid, IID_IThumbnailHandlerFactory))
	{
		*ppv = (IThumbnailHandlerFactory*) this;
		AddRef();
		return S_OK;
	}
	else if (IsEqualIID(riid, IID_IEasySFTPDirectory))
	{
		*ppv = (IEasySFTPDirectory*) this;
		AddRef();
		return S_OK;
	}
	return CFolderBase::QueryInterface(riid, ppv);
}

STDMETHODIMP CFTPDirectoryBase::EnumObjects(HWND hWnd, SHCONTF grfFlags, IEnumIDList** ppenumIDList)
{
	if (!ppenumIDList)
		return E_POINTER;
	if (!m_bDirReceived)
	{
		m_hWndOwnerCache = hWnd;
		m_pRoot->m_hWndOwnerCache = hWnd;
		if (!m_pRoot->ReceiveDirectory(hWnd, this, m_strDirectory, &m_bDirReceived))
		{
			*ppenumIDList = NULL;
			return S_FALSE;
		}
	}

	*ppenumIDList = new CEnumFTPItemIDList(m_pMallocData, m_aFiles, grfFlags, (IShellFolder*) this);
	return S_OK;
}

STDMETHODIMP CFTPDirectoryBase::BindToObject(PCUIDLIST_RELATIVE pidl, LPBC pbc, REFIID riid, void** ppv)
{
	if (!pidl || !ppv)
		return E_POINTER;
	//if (pbc)
	//	return E_UNEXPECTED;
	if (IsEqualIID(riid, IID_IExtractIconW) || IsEqualIID(riid, IID_IExtractIconA))
		return E_NOINTERFACE;
	if (!IsEqualIID(riid, IID_IUnknown) &&
		!IsEqualIID(riid, IID_IShellFolder) &&
		!IsEqualIID(riid, IID_IShellFolder2) &&
		!IsEqualIID(riid, IID_IPersist) &&
		!IsEqualIID(riid, IID_IPersistFolder) &&
		!IsEqualIID(riid, IID_IPersistFolder2) &&
		!IsEqualIID(riid, IID_IPersistIDList) &&
		!IsEqualIID(riid, IID_IObjectWithSite) &&
		!IsEqualIID(riid, IID_IEasySFTPDirectory))
		return E_NOINTERFACE;

	CFTPDirectoryItem* pDirItem = GetAlreadyOpenedDirectory(pidl);
	if (pDirItem)
	{
		return pDirItem->pDirectory->QueryInterface(riid, ppv);
	}
	_StringW strRealPath;
	if (!m_pRoot->ValidateDirectory(m_strDirectory, pidl, strRealPath))
		return E_INVALIDARG;
	if (strRealPath.IsEmpty())
		return E_INVALIDARG;

	LPCWSTR lpw = strRealPath;
	CFTPDirectoryBase* pParent;
	if (*lpw == L'/')
	{
		pParent = m_pRoot;
		lpw++;
	}
	else
		pParent = this;
	CFTPDirectoryBase* pRet;
	HRESULT hr = pParent->OpenNewDirectory(lpw, &pRet);
	if (FAILED(hr))
		return hr;
	//{
	//	PIDLIST_ABSOLUTE pidlChild = ::AppendItemIDList(m_pidlMe, pidl);
	//	hr = pRet->Initialize(pidlChild);
	//	::CoTaskMemFree(pidlChild);
	//}
	hr = pRet->QueryInterface(riid, ppv);
	pRet->Release();

	return hr;
}

STDMETHODIMP CFTPDirectoryBase::BindToStorage(PCUIDLIST_RELATIVE pidl, LPBC pbc, REFIID riid, void** ppv)
{
	return E_NOTIMPL;
}

STDMETHODIMP CFTPDirectoryBase::CompareIDs(LPARAM lParam, PCUIDLIST_RELATIVE pidl1, PCUIDLIST_RELATIVE pidl2)
{
	if (!pidl1 || !pidl2)
		return E_POINTER;

	int nColumn;
	SHCOLUMNID scid;
	HRESULT hr;
	if (lParam & SHCIDS_ALLFIELDS)
		nColumn = -1;
	else
	{
		nColumn = (lParam & SHCIDS_COLUMNMASK);
		hr = MapColumnToSCID(nColumn, &scid);
		if (FAILED(hr))
			return hr;
	}

	_StringW str1, str2;
	if (!pidl1->mkid.cb)
		return pidl2->mkid.cb ? MAKE_HRESULT(0, 0, (unsigned short)(short) -1) : MAKE_HRESULT(0, 0, 0);
	else if (!pidl2->mkid.cb)
		return MAKE_HRESULT(0, 0, 1);
	if (!PickupFileName((PCUITEMID_CHILD) pidl1, str1))
		return E_INVALIDARG;
	if (!PickupFileName((PCUITEMID_CHILD) pidl2, str2))
		return E_INVALIDARG;

	CFTPFileItem* pItem1;
	CFTPFileItem* pItem2;
	int r = 0;
	if (nColumn == -1 || scid.pid != PID_FTPITEM_FILE_NAME)
	{
		if (!m_bDirReceived)
		{
			if (!m_pRoot->ReceiveDirectory(m_hWndOwnerCache, this, m_strDirectory, &m_bDirReceived))
				return E_UNEXPECTED;
		}
		pItem1 = GetFileItem(str1);
		pItem2 = GetFileItem(str2);
		if (!pItem1 || !pItem2)
			return E_INVALIDARG;
	}
	else
	{
		if (m_bDirReceived)
		{
			pItem1 = GetFileItem(str1);
			pItem2 = GetFileItem(str2);
			//if (!pItem1 || !pItem2)
			//	return E_INVALIDARG;
		}
		else
			pItem1 = pItem2 = NULL;
	}
	if (pItem1 && pItem2)
	{
		bool bDir1 = pItem1->IsDirectory();
		bool bDir2 = pItem2->IsDirectory();
		if (bDir1 != bDir2)
		{
			if (bDir1)
				r = -1;
			//else if (IsShortcutItem(pItem1) && pItem1->IsDirectory())
			//	return pItem2->type == fitypeDir ? 1 : -1;
			else //if (bDir2)
				r = 1;
		}
	}

	if (r == 0)
	{
		if (nColumn == -1)
			scid.pid = PID_FTPITEM_FILE_NAME;
		while (true)
		{
			switch (scid.pid)
			{
				case PID_FTPITEM_FILE_NAME:
					if (pItem1 && pItem2)
						r = pItem1->strFileName.Compare(pItem2->strFileName);
					else
						r = str1.Compare(str2);
					break;
				case PID_FTPITEM_SIZE:
					if (pItem1->uliSize.QuadPart < pItem2->uliSize.QuadPart)
						r = -1;
					else if (pItem1->uliSize.QuadPart > pItem2->uliSize.QuadPart)
						r = 1;
					else
						r = 0;
					break;
				case PID_FTPITEM_TYPE:
					FillFileItemInfo(pItem1);
					FillFileItemInfo(pItem2);
					r = pItem1->strType.Compare(pItem2->strType);
					break;
				case PID_FTPITEM_MODIFY_TIME:
					r = ::CompareFileTime(&pItem1->ftModifyTime, &pItem2->ftModifyTime);
					break;
				case PID_FTPITEM_PERMISSIONS:
					r = 0;
					break;
			}
			if (nColumn == -1)
			{
				if (r != 0)
					break;
				scid.pid++;
				if (scid.pid > PID_FTPITEM_PERMISSIONS)
					break;
			}
			else
			{
				if (r != 0 || scid.pid == PID_FTPITEM_FILE_NAME)
					break;
				scid.pid = PID_FTPITEM_FILE_NAME;
			}
		}
	}

	if (r != 0)
	{
		r = (r > 0 ? 1 : -1);
		// we must cast to 'unsigned short'
		return MAKE_HRESULT(0, 0, (unsigned short)(short) r);
	}

	if (!((PCUIDLIST_RELATIVE) (((DWORD_PTR) pidl1) + pidl1->mkid.cb))->mkid.cb)
	{
		return (!((PCUIDLIST_RELATIVE) (((DWORD_PTR) pidl1) + pidl1->mkid.cb))->mkid.cb) ?
			MAKE_HRESULT(0, 0, 0) : MAKE_HRESULT(0, 0, (unsigned short)(short) -1);
	}
	else if (!((PCUIDLIST_RELATIVE) (((DWORD_PTR) pidl1) + pidl1->mkid.cb))->mkid.cb)
		return MAKE_HRESULT(0, 0, 1);

	IShellFolder* pChild;
	hr = BindToObject(pidl1, NULL, IID_IShellFolder, (void**) &pChild);
	if (FAILED(hr))
		return hr;
	pidl1 = (PCUIDLIST_RELATIVE) (((DWORD_PTR) pidl1) + pidl1->mkid.cb);
	pidl2 = (PCUIDLIST_RELATIVE) (((DWORD_PTR) pidl2) + pidl2->mkid.cb);
	hr = pChild->CompareIDs(lParam, pidl1, pidl2);
	pChild->Release();
	return hr;
}

STDMETHODIMP CFTPDirectoryBase::CreateViewObject(HWND hWndOwner, REFIID riid, void** ppv)
{
	if (IsEqualIID(riid, IID_IExtractIconA) || IsEqualIID(riid, IID_IExtractIconW))
	{
		CFTPFileItem* pItem = new CFTPFileItem();
		pItem->bWinAttr = false;
		pItem->nUnixMode = S_IFDIR;
		pItem->strFileName = m_pItemMe->strName;
		pItem->iIconIndex = -1;
		CFTPFileItemIcon* pIcon = new CFTPFileItemIcon(pItem);
		HRESULT hr = pIcon->QueryInterface(riid, ppv);
		pIcon->Release();
		pItem->Release();
		return hr;
	}
	else if (IsEqualIID(riid, IID_IDropTarget))
	{
		CFTPDropHandler* pDrop = new CFTPDropHandler(this, hWndOwner);
		*ppv = (IDropTarget*) pDrop;
		return S_OK;
	}
	else if (IsEqualIID(riid, IID_IContextMenu))
	{
		IShellBrowser* pBrowser = GetShellBrowser(hWndOwner);
		CFTPFileDirectoryMenu* pMenu = new CFTPFileDirectoryMenu(this, m_pidlMe, pBrowser);
		if (pBrowser)
			pBrowser->Release();
		*ppv = (IContextMenu*) pMenu;
		return S_OK;
	}
	return CFolderBase::CreateViewObject(hWndOwner, riid, ppv);
}

STDMETHODIMP CFTPDirectoryBase::GetAttributesOf(UINT cidl, PCUITEMID_CHILD_ARRAY apidl, SFGAOF* rgfInOut)
{
	if (!rgfInOut)
		return E_POINTER;
	if (!cidl)
	{
		*rgfInOut = 0;
		return S_OK;
	}
	if (!apidl)
		return E_POINTER;
	SFGAOF ret, f, req;
	req = *rgfInOut;
	if (req & SFGAO_FILESYSTEM)
	{
		*rgfInOut = (req &= ~SFGAO_FILESYSTEM);
		if (!req)
			return S_OK;
	}
	_StringW str;
	for (UINT u = 0; u < cidl; u++)
	{
		if (!::PickupFileName(apidl[u], str))
			return E_INVALIDARG;
		f = SFGAO_CANCOPY | SFGAO_CANDELETE | SFGAO_CANRENAME | SFGAO_HASPROPSHEET;
		if ((req & (SFGAO_BROWSABLE | SFGAO_FOLDER | SFGAO_DROPTARGET | SFGAO_HASSUBFOLDER | SFGAO_HIDDEN | SFGAO_LINK)) != 0)
		{
			CFTPFileItem* p;
			CSFTPFileItem UNALIGNED* pData = (CSFTPFileItem UNALIGNED*) apidl[u];
			if (pData->bHasAttribute)
			{
				if (pData->bIsDirectory)
					f |= SFGAO_BROWSABLE | SFGAO_FOLDER | SFGAO_DROPTARGET | SFGAO_HASSUBFOLDER;
				if (pData->bIsHidden)
					f |= SFGAO_HIDDEN;
				if (pData->bIsShortcut)
					f |= SFGAO_LINK;
			}
			else
			{
				if (!m_bDirReceived)
				{
					if (!m_hWndOwnerCache)
						m_hWndOwnerCache = m_pRoot->m_hWndOwnerCache;
					if (!m_pRoot->ReceiveDirectory(m_hWndOwnerCache, this, m_strDirectory, &m_bDirReceived))
						return E_UNEXPECTED;
				}
				p = GetFileItem(str);
				if (!p)
					return E_INVALIDARG;
				if (p->IsDirectory())
					f |= SFGAO_BROWSABLE | SFGAO_FOLDER | SFGAO_DROPTARGET | SFGAO_HASSUBFOLDER;
				if (p->IsHidden())
					f |= SFGAO_HIDDEN;
				if (p->IsShortcut())
					f |= SFGAO_LINK;
			}
		}
		if (!u)
			ret = f;
		else
			ret &= f;
	}
	*rgfInOut &= ret;
	return S_OK;
}

STDMETHODIMP CFTPDirectoryBase::GetUIObjectOf(HWND hWndOwner, UINT cidl, PCUITEMID_CHILD_ARRAY apidl,
	REFIID riid, UINT* rgfReserved, void** ppv)
{
	if (!ppv)
		return E_POINTER;
	// NOTE: we must set NULL to *ppv, or causes 0xC000041D exception in Windows 7 (x64)
	*ppv = NULL;
	if (!cidl)
		return E_INVALIDARG;
	if (!apidl)
		return E_POINTER;

	bool bIsIcon;
	if (IsEqualIID(riid, IID_IExtractIconA) ||
		IsEqualIID(riid, IID_IExtractIconW))
		bIsIcon = true;
	else
	{
		bIsIcon = false;
		if (!IsEqualIID(riid, IID_IDataObject) &&
			!IsEqualIID(riid, IID_IContextMenu) &&
			!IsEqualIID(riid, IID_IDropTarget) &&
			!IsEqualIID(riid, IID_IThumbnailProvider))
		{
#ifdef _DEBUG
			if (!IsEqualIID(riid, IID_IQueryInfo))
			{
				_StringW str;
				str.Format(L"CFTPDirectoryBase::GetUIObjectOf: unknown interface: {%08lX-%04hX-%04hX-%02X%02X-%02X%02X%02X%02X%02X%02X}\n",
					riid.Data1, riid.Data2, riid.Data3, (UINT) riid.Data4[0], (UINT) riid.Data4[1],
					(UINT) riid.Data4[2], (UINT) riid.Data4[3], (UINT) riid.Data4[4], (UINT) riid.Data4[5],
					(UINT) riid.Data4[6], (UINT) riid.Data4[7]);
				OutputDebugString(str);
			}
#endif
			return E_NOINTERFACE;
		}
	}

	CMyPtrArrayT<CFTPFileItem> aItems;
	_StringW str;
	for (UINT u = 0; u < cidl; u++)
	{
		if (!::PickupFileName(apidl[u], str))
			return E_INVALIDARG;
		if (!m_bDirReceived && !bIsIcon)
		{
			m_hWndOwnerCache = hWndOwner;
			m_pRoot->m_hWndOwnerCache = hWndOwner;
			if (!m_pRoot->ReceiveDirectory(hWndOwner, this, m_strDirectory, &m_bDirReceived))
				return E_UNEXPECTED;
		}
		if (m_bDirReceived)
		{
			CFTPFileItem* p = GetFileItem(str);
			if (!p)
				return E_INVALIDARG;
			aItems.Add(p);
		}
	}

	HRESULT hr = E_NOINTERFACE;

	if (IsEqualIID(riid, IID_IExtractIconA) || IsEqualIID(riid, IID_IExtractIconW))
	{
		if (cidl != 1)
			return E_INVALIDARG;
		CFTPFileItemIcon* pIcon;
		if (m_bDirReceived)
			pIcon = new CFTPFileItemIcon(aItems.GetItem(0));
		else
		{
			CSFTPFileItem UNALIGNED* pFItem = (CSFTPFileItem UNALIGNED*) apidl[0];
			str = pFItem->wchFileName;
			pIcon = new CFTPFileItemIcon(str, pFItem->bIsDirectory != 0);
		}
		hr = pIcon->QueryInterface(riid, ppv);
		pIcon->Release();
	}
	else if (IsEqualIID(riid, IID_IDropTarget))
	{
		if (aItems.GetCount() != 1)
			return E_INVALIDARG;
		if (!aItems.GetItem(0)->IsDirectory())
			return E_INVALIDARG;
		CFTPDirectoryBase* pDir;
		hr = OpenNewDirectory(aItems.GetItem(0)->strFileName, &pDir);
		if (SUCCEEDED(hr))
		{
			hr = pDir->CreateViewObject(hWndOwner, IID_IDropTarget, ppv);
			pDir->Release();
		}
	}
	else if (IsEqualIID(riid, IID_IContextMenu))
	{
		IShellBrowser* pBrowser = GetShellBrowser(hWndOwner);
		CFTPFileItemMenu* pMenu = new CFTPFileItemMenu(this, m_pidlMe, pBrowser, aItems);
		if (pBrowser)
			pBrowser->Release();
		hr = pMenu->QueryInterface(riid, ppv);
		pMenu->Release();
	}
	else if (IsEqualIID(riid, IID_IThumbnailProvider))
	{
		if (!m_pRoot->m_bUseThumbnailPreview)
			return E_NOINTERFACE;
		if (aItems.GetCount() != 1)
			return E_INVALIDARG;
		if (aItems.GetItem(0)->IsDirectory())
			return E_INVALIDARG;
		IThumbnailProvider* pProvider;
		hr = MyCreateThumbnailProviderFromFileName(aItems.GetItem(0)->strFileName, &pProvider);
		if (FAILED(hr))
			return hr;
		IInitializeWithStream* pInitStream;
		hr = pProvider->QueryInterface(IID_IInitializeWithStream, (void**) &pInitStream);
		if (SUCCEEDED(hr))
		{
			IStream* pStream;
			hr = CreateStream(aItems.GetItem(0), &pStream);
			if (SUCCEEDED(hr))
			{
				hr = pInitStream->Initialize(pStream, STGM_READ);
				if (SUCCEEDED(hr))
					hr = pProvider->QueryInterface(riid, ppv);
				pStream->Release();
			}
			pInitStream->Release();
		}
		pProvider->Release();
		return hr;
	}

	if (SUCCEEDED(hr))
		return hr;

	CFTPDataObject* pObj;
	AddRef();
	hr = m_pRoot->GetFTPItemUIObjectOf(hWndOwner, this, aItems, &pObj);
	Release();
	if (SUCCEEDED(hr))
	{
		hr = pObj->QueryInterface(riid, ppv);
		pObj->Release();
		//if (SUCCEEDED(hr))
		//	return hr;
	}

	return hr;
}

STDMETHODIMP CFTPDirectoryBase::GetDisplayNameOf(PCUITEMID_CHILD pidl, SHGDNF uFlags, STRRET* pName)
{
	_StringW str;
	CFTPFileItem* p;
	if (pidl)
	{
		if (!::PickupFileName(pidl, str))
			return E_INVALIDARG;
		if (m_bDirReceived)
		{
			p = GetFileItem(str);
			if (!p)
				return E_INVALIDARG;
		}
		else
			p = NULL;
	}
	else
		p = NULL;
	str.Empty();
	switch (uFlags & 0x0FFF)
	{
		case SHGDN_NORMAL:
			//if (pidl && !p)
			//	return E_INVALIDARG;
			if (uFlags & (SHGDN_FORPARSING | SHGDN_FORADDRESSBAR))
			{
				_StringW str2;
				int nDefPort;
				str = m_pRoot->GetProtocolName(nDefPort);
				str += L"://";
				str += m_pRoot->m_strHostName;
				if (m_pRoot->m_nPort != nDefPort)
				{
					str2.Format(L"%d", m_pRoot->m_nPort);
					str += L':';
					str += str2;
				}
				//str2.Empty();
				//CFTPDirectoryBase* pd = this;
				//while (pd)
				//{
				//	if (pd == (CFTPDirectoryBase*) m_pRoot)
				//		break;
				//	str2.InsertString(pd->m_strDirectory, 0);
				//	str2.InsertChar(L'/', 0);
				//	pd = pd->m_pParent;
				//}
				//str += str2;
				str += m_strDirectory;
				if (((LPCWSTR) m_strDirectory)[m_strDirectory.GetLength() - 1] != L'/')
					str += L'/';
				bool bLastIsNotDelimiter = false;
				if (p)
				{
					str += p->strFileName;
					bLastIsNotDelimiter = true;
				}
				else if (pidl)
				{
					if (!::PickupFileName(pidl, str2))
						return E_INVALIDARG;
					str += str2;
					bLastIsNotDelimiter = true;
				}
				while (true)
				{
					pidl = (PCUITEMID_CHILD) (((DWORD_PTR) pidl) + pidl->mkid.cb);
					if (!pidl->mkid.cb)
						break;
					if (!::PickupFileName(pidl, str2))
						return E_INVALIDARG;
					if (bLastIsNotDelimiter)
						str += L'/';
					else
						bLastIsNotDelimiter = true;
					str += str2;
				}
				break;
			}
			//else
			//{
			//	//_StringW str2;
			//	//str2.Empty();
			//	//CFTPDirectoryBase* pd = this;
			//	//while (pd)
			//	//{
			//	//	if (pd == (CFTPDirectoryBase*) m_pRoot)
			//	//		break;
			//	//	str2.InsertString(pd->m_strDirectory, 0);
			//	//	str2.InsertChar(L'/', 0);
			//	//	pd = pd->m_pParent;
			//	//}
			//	//str += str2;
			//	//str += L'/';
			//	str = m_strDirectory;
			//	if (p)
			//	{
			//		if (((LPCWSTR) m_strDirectory)[m_strDirectory.GetLength() - 1] != L'/')
			//			str += L'/';
			//		str += p->strFileName;
			//	}
			//}
			//break;
		case SHGDN_INFOLDER:
			if (!pidl)
			{
				if (this == (CFTPDirectoryBase*) m_pRoot)
					str = m_pRoot->m_strHostName;
				else
					str = m_strDirectory;
			}
			else if (p)
				str = p->strFileName;
			else if (!::PickupFileName(pidl, str))
				return E_INVALIDARG;
			break;
		default:
			return E_NOTIMPL;
	}

	SIZE_T nSize = sizeof(WCHAR) * (str.GetLength() + 1);
	pName->uType = STRRET_WSTR;
	pName->pOleStr = (LPWSTR) ::CoTaskMemAlloc(nSize);
	memcpy(pName->pOleStr, (LPCWSTR) str, nSize);
	return S_OK;
}

STDMETHODIMP CFTPDirectoryBase::SetNameOf(HWND hWnd, PCUITEMID_CHILD pidl, LPCWSTR pszName, SHGDNF uFlags, PITEMID_CHILD* ppidlOut)
{
	if (!pszName)
		return E_POINTER;
	if (!pidl)
		return E_INVALIDARG;

	_StringW str;
	CFTPFileItem* p;
	if (!::PickupFileName(pidl, str))
	{
		if (ppidlOut)
			*ppidlOut = NULL;
		return E_INVALIDARG;
	}
	p = GetFileItem(str);
	if (!p)
	{
		if (ppidlOut)
			*ppidlOut = NULL;
		return E_INVALIDARG;
	}

	AddRef();
	HRESULT hr = m_pRoot->SetFTPItemNameOf(hWnd, this, p, pszName, uFlags);
	if (SUCCEEDED(hr))
	{
		UpdateRenameFile(p->strFileName, pszName, p->IsDirectory());
		PITEMID_CHILD pidlNewChild;
		p->strFileName = pszName;
		pidlNewChild = ::CreateFileItem(m_pMallocData->pMalloc, p);
		if (pidlNewChild)
		{
			//NotifyUpdate(p->IsDirectory() ? SHCNE_RENAMEFOLDER : SHCNE_RENAMEITEM, pidl, pidlNewChild);
			if (ppidlOut)
				*ppidlOut = pidlNewChild;
		}
	}
	else
	{
		if (ppidlOut)
			*ppidlOut = NULL;
	}
	Release();

	return hr;
}

STDMETHODIMP CFTPDirectoryBase::GetDefaultSearchGUID(GUID* pguid)
{
	return E_NOTIMPL;
}

STDMETHODIMP CFTPDirectoryBase::EnumSearches(IEnumExtraSearch** ppenum)
{
	return E_NOTIMPL;
}

STDMETHODIMP CFTPDirectoryBase::GetDefaultColumn(DWORD dwRes, ULONG* pSort, ULONG* pDisplay)
{
	if (!pSort || !pDisplay)
		return E_POINTER;
	*pSort = *pDisplay = 0;
	return S_OK;
}

STDMETHODIMP CFTPDirectoryBase::GetDefaultColumnState(UINT iColumn, SHCOLSTATEF* pcsFlags)
{
	if (!pcsFlags)
		return E_POINTER;

	SHCOLUMNID scid;
	HRESULT hr = MapColumnToSCID(iColumn, &scid);
	if (FAILED(hr))
		return hr;

	switch (scid.pid)
	{
		case PID_FTPITEM_FILE_NAME:
		case PID_FTPITEM_TYPE:
		case PID_FTPITEM_PERMISSIONS:
		case PID_FTPITEM_TRANSFER_TYPE:
			*pcsFlags = SHCOLSTATE_TYPE_STR | SHCOLSTATE_ONBYDEFAULT;
			break;
		case PID_FTPITEM_SIZE:
			*pcsFlags = SHCOLSTATE_TYPE_INT | SHCOLSTATE_ONBYDEFAULT;
			break;
		case PID_FTPITEM_MODIFY_TIME:
			*pcsFlags = SHCOLSTATE_TYPE_DATE | SHCOLSTATE_ONBYDEFAULT;
			break;
		default:
			return E_INVALIDARG;
	}

	return S_OK;
}

STDMETHODIMP CFTPDirectoryBase::GetDetailsEx(PCUITEMID_CHILD pidl, const SHCOLUMNID* pscid, VARIANT* pv)
{
	if (!IsEqualGUID(pscid->fmtid, GUID_FTPItemColumn))
		return E_INVALIDARG;

	_StringW str;
	if (!::PickupFileName(pidl, str))
		return E_INVALIDARG;
	CFTPFileItem* p = GetFileItem(str);
	if (!p)
		return E_INVALIDARG;

	if (p->iIconIndex == -1)
		FillFileItemInfo(p);

	switch (pscid->pid)
	{
		case PID_FTPITEM_FILE_NAME:
			pv->vt = VT_BSTR;
			pv->bstrVal = ::SysAllocStringLen(p->strFileName, (UINT) p->strFileName.GetLength());
			if (!pv->bstrVal)
				return E_OUTOFMEMORY;
			break;
		case PID_FTPITEM_SIZE:
			pv->vt = VT_UI8;
			pv->ullVal = p->uliSize.QuadPart;
			break;
		case PID_FTPITEM_TYPE:
			pv->vt = VT_BSTR;
			pv->bstrVal = ::SysAllocStringLen(p->strType, (UINT) p->strType.GetLength());
			if (!pv->bstrVal)
				return E_OUTOFMEMORY;
			break;
		case PID_FTPITEM_MODIFY_TIME:
		{
			SYSTEMTIME st;
			::FileTimeToSystemTime(&p->ftModifyTime, &st);
			::SystemTimeToVariantTime(&st, &pv->date);
			pv->vt = VT_DATE;
		}
		break;
		case PID_FTPITEM_PERMISSIONS:
			if (p->bWinAttr)
				::FileAttributesToString(p->dwAttributes, str);
			else
				::FilePermissionsToString(p->nUnixMode, str);
			pv->vt = VT_BSTR;
			pv->bstrVal = ::SysAllocStringLen(str, (UINT) str.GetLength());
			if (!pv->bstrVal)
				return E_OUTOFMEMORY;
			break;
		case PID_FTPITEM_TRANSFER_TYPE:
			if (p->IsDirectory())
				str.LoadString(IDS_TYPE_DIRECTORY);
			else if (IsTextFile(str) == S_OK)
				str.LoadString(IDS_TYPE_TEXT);
			else
				str.LoadString(IDS_TYPE_BINARY);
			pv->vt = VT_BSTR;
			pv->bstrVal = ::SysAllocStringLen(str, (UINT) str.GetLength());
			if (!pv->bstrVal)
				return E_OUTOFMEMORY;
			break;
		default:
			return E_INVALIDARG;
	}

	return S_OK;
}

STDMETHODIMP CFTPDirectoryBase::GetDetailsOf(PCUITEMID_CHILD pidl, UINT iColumn, SHELLDETAILS* psd)
{
	if (!psd)
		return E_POINTER;

	SHCOLUMNID scid;
	HRESULT hr = MapColumnToSCID(iColumn, &scid);
	if (FAILED(hr))
		return hr;

	_StringW str;
	if (!pidl)
	{
		switch (scid.pid)
		{
			case PID_FTPITEM_FILE_NAME:
				psd->fmt = LVCFMT_LEFT;
				break;
			case PID_FTPITEM_SIZE:
				psd->fmt = LVCFMT_RIGHT;
				psd->cxChar = 10;
				break;
			case PID_FTPITEM_TYPE:
				psd->fmt = LVCFMT_LEFT;
				break;
			case PID_FTPITEM_MODIFY_TIME:
				psd->fmt = LVCFMT_LEFT;
				psd->cxChar = 20;
				break;
			case PID_FTPITEM_PERMISSIONS:
				psd->fmt = LVCFMT_LEFT;
				psd->cxChar = 10;
				break;
			case PID_FTPITEM_TRANSFER_TYPE:
				psd->fmt = LVCFMT_LEFT;
				psd->cxChar = 12;
				break;
			default:
				return E_INVALIDARG;
		}
		psd->str.uType = STRRET_WSTR;
		str.LoadString(MY_PID_TO_STRING_ID(scid.pid));
		psd->str.pOleStr = DuplicateCoMemString(str);
		if (!psd->str.pOleStr)
			return E_OUTOFMEMORY;
		return S_OK;
	}

	if (!::PickupFileName(pidl, str))
		return E_INVALIDARG;
	CFTPFileItem* p = GetFileItem(str);
	if (!p)
		return E_INVALIDARG;

	if (p->iIconIndex == -1)
		FillFileItemInfo(p);

	psd->cxChar = 0;
	switch (scid.pid)
	{
		case PID_FTPITEM_FILE_NAME:
			psd->fmt = LVCFMT_LEFT;
			psd->str.uType = STRRET_WSTR;
			psd->str.pOleStr = DuplicateCoMemString(p->strFileName);
			if (!psd->str.pOleStr)
				return E_OUTOFMEMORY;
			break;
		case PID_FTPITEM_SIZE:
			::FileSizeToString(p->uliSize, str);
			psd->fmt = LVCFMT_RIGHT;
			psd->str.uType = STRRET_WSTR;
			psd->str.pOleStr = DuplicateCoMemString(str);
			if (!psd->str.pOleStr)
				return E_OUTOFMEMORY;
			break;
		case PID_FTPITEM_TYPE:
			psd->fmt = LVCFMT_LEFT;
			psd->str.uType = STRRET_WSTR;
			psd->str.pOleStr = DuplicateCoMemString(p->strType);
			if (!psd->str.pOleStr)
				return E_OUTOFMEMORY;
			break;
		case PID_FTPITEM_MODIFY_TIME:
			::FileTimeToString(&p->ftModifyTime, str);
			psd->fmt = LVCFMT_LEFT;
			psd->str.uType = STRRET_WSTR;
			psd->str.pOleStr = DuplicateCoMemString(str);
			if (!psd->str.pOleStr)
				return E_OUTOFMEMORY;
			break;
		case PID_FTPITEM_PERMISSIONS:
			if (p->bWinAttr)
				::FileAttributesToString(p->dwAttributes, str);
			else
				::FilePermissionsToString(p->nUnixMode, str);
			psd->fmt = LVCFMT_LEFT;
			psd->str.uType = STRRET_WSTR;
			psd->str.pOleStr = DuplicateCoMemString(str);
			if (!psd->str.pOleStr)
				return E_OUTOFMEMORY;
			break;
		case PID_FTPITEM_TRANSFER_TYPE:
			if (p->IsDirectory())
				str.LoadString(IDS_TYPE_DIRECTORY);
			else if (IsTextFile(str) == S_OK)
				str.LoadString(IDS_TYPE_TEXT);
			else
				str.LoadString(IDS_TYPE_BINARY);
			psd->fmt = LVCFMT_LEFT;
			psd->str.uType = STRRET_WSTR;
			psd->str.pOleStr = DuplicateCoMemString(str);
			if (!psd->str.pOleStr)
				return E_OUTOFMEMORY;
			break;
		default:
			return E_INVALIDARG;
	}

	return S_OK;
}

STDMETHODIMP CFTPDirectoryBase::MapColumnToSCID(UINT iColumn, SHCOLUMNID* pscid)
{
	if (!pscid)
		return E_POINTER;
	if (iColumn >= sizeof(s_uHeaderItemPIDs) / sizeof(s_uHeaderItemPIDs[0]))
		return E_INVALIDARG;
	pscid->fmtid = GUID_FTPItemColumn;
	pscid->pid = s_uHeaderItemPIDs[iColumn];
	return S_OK;
}


STDMETHODIMP CFTPDirectoryBase::GetIconOf(PCUITEMID_CHILD pidl, UINT flags, int* pIconIndex)
{
	if (!pIconIndex)
		return E_POINTER;

	_StringW str;
	if (!::PickupFileName(pidl, str))
		return E_INVALIDARG;
	if (!m_bDirReceived)
	{
		CSFTPFileItem UNALIGNED* pf = (CSFTPFileItem UNALIGNED*) pidl;
		if (pf->bHasAttribute && pf->bIsDirectory)
		{
			SHFILEINFO_UNION sfi;
			DWORD dw;
			UINT u;
			memset(&sfi.w, 0, sizeof(sfi.w));
			dw = FILE_ATTRIBUTE_DIRECTORY;
			u = SHGFI_USEFILEATTRIBUTES | SHGFI_SYSICONINDEX | SHGFI_LARGEICON;
			if (flags & GIL_OPENICON)
				u |= SHGFI_OPENICON;
			if (pf->bIsHidden)
				dw |= FILE_ATTRIBUTE_HIDDEN;
			if (!::SHGetFileInfoW(str, dw, &sfi.w, sizeof(sfi.w), u))
			{
				memset(&sfi.a, 0, sizeof(sfi.a));
				if (!::SHGetFileInfoA(str, dw, &sfi.a, sizeof(sfi.a), u))
					return E_FAIL;
				*pIconIndex = sfi.a.iIcon;
			}
			else
			{
				*pIconIndex = sfi.w.iIcon;
			}
			return S_OK;
		}
		return S_FALSE;
	}
	CFTPFileItem* p = GetFileItem(str);
	if (!p)
		return E_INVALIDARG;
	if (p->iIconIndex == -1)
		FillFileItemInfo(p);
	*pIconIndex = (flags & GIL_OPENICON) ? p->iOpenIconIndex : p->iIconIndex;
	return S_OK;
}


STDMETHODIMP CFTPDirectoryBase::MessageSFVCB(UINT uMsg, WPARAM wParam, LPARAM lParam)
{
	switch (uMsg)
	{
		case 17:   // SFVM_LISTREFRESHED
			if (wParam)
			{
				::EnterCriticalSection(&m_csFiles);
				int i;
				i = m_aFiles.GetCount();
				while (i--)
					m_aFiles.GetItem(i)->Release();
				m_aFiles.RemoveAll();
				m_bDirReceived = false;
				::LeaveCriticalSection(&m_csFiles);
			}
			return S_OK;
		case SFVM_MERGEMENU:
		{
			LPQCMINFO lpqi = (LPQCMINFO) lParam;
			HMENU hMenuToAdd = GetSubMenuByID(lpqi->hmenu, FCIDM_MENU_HELP);
			if (hMenuToAdd)
			{
				int i, nCount;
				UINT indexMenu;
				MENUITEMINFO mii;
				UINT uMaxID;
				_StringW str;

#ifdef _WIN64
				mii.cbSize = sizeof(mii);
				mii.fMask = MIIM_FTYPE | MIIM_STRING | MIIM_ID | MIIM_STATE;
#else
				mii.cbSize = MENUITEMINFO_SIZE_V1;
				mii.fMask = MIIM_TYPE | MIIM_ID | MIIM_STATE;
#endif
				uMaxID = lpqi->idCmdFirst;
				//indexMenu = lpqi->indexMenu;
				indexMenu = 0;

				HMENU h = ::GetSubMenu(theApp.m_hMenuContext, CXMENU_POPUP_ROOTHELP);
				nCount = ::GetMenuItemCount(h);
				for (i = 0; i < nCount; i++)
				{
					mii.cch = MAX_PATH;
#ifdef _UNICODE
					mii.dwTypeData = str.GetBufferW(MAX_PATH);
#else
					mii.dwTypeData = str.GetBufferA(MAX_PATH);
#endif
					::GetMenuItemInfo(h, (UINT) i, TRUE, &mii);
					mii.wID = (WORD)((UINT) mii.wID - ID_ROOT_IDBASE + lpqi->idCmdFirst);
					if (uMaxID < (UINT) mii.wID)
						uMaxID = (UINT) mii.wID;
					::InsertMenuItem(hMenuToAdd, indexMenu++, TRUE, &mii);
				}
				//mii.fMask = MIIM_TYPE;
				//mii.fType = MFT_SEPARATOR;
				//::InsertMenuItem(hMenuToAdd, indexMenu++, TRUE, &mii);
				lpqi->idCmdFirst = uMaxID - lpqi->idCmdFirst + 1;
			}
			return S_OK;
		}
		case SFVM_INVOKECOMMAND:
		{
			UINT uID = ((UINT) wParam) + ID_ROOT_IDBASE;
			switch (uID)
			{
				case ID_ROOT_SERVER_INFO:
				{
					m_pRoot->ShowServerInfoDialog(m_pRoot->m_hWndOwnerCache);
				}
				return S_OK;
			}
		}
		break;
	}
	return CFolderBase::MessageSFVCB(uMsg, wParam, lParam);
}


STDMETHODIMP CFTPDirectoryBase::GetThumbnailHandler(PCUITEMID_CHILD pidl, LPBC pbc, REFIID riid, void** ppv)
{
	return GetUIObjectOf(NULL, 1, &pidl, riid, NULL, ppv);
}


STDMETHODIMP CFTPDirectoryBase::GetRootDirectory(IEasySFTPDirectory** ppRootDirectory)
{
	return m_pRoot->QueryInterface(IID_IEasySFTPDirectory, (void**) ppRootDirectory);
}


HRESULT CFTPDirectoryBase::CreateStream(CFTPFileItem* pItem, IStream** ppStream)
{
	HRESULT hr = m_pRoot->CreateFTPItemStream(this, pItem, ppStream);
	if (FAILED(hr))
		return hr;
	if (!TEXTMODE_IS_NO_CONVERTION(m_pRoot->m_bTextMode))
	{
		hr = IsTextFile(pItem->strFileName);
		if (SUCCEEDED(hr) && hr == S_OK)
		{
			IStream* pstm2;
			hr = MyCreateTextStream(*ppStream, TEXTMODE_FOR_RECV(m_pRoot->m_bTextMode), &pstm2);
			(*ppStream)->Release();
			if (FAILED(hr))
			{
				*ppStream = NULL;
				return hr;
			}
			*ppStream = pstm2;
		}
	}
	return S_OK;
}

void CFTPDirectoryBase::DeleteFTPItem(CFTPFileItem* pItem)
{
	//PITEMID_CHILD pidl = ::CreateFileItem(m_pMallocData->pMalloc, pItem);
	//if (!pidl)
	//	return;
	//NotifyUpdate(pItem->IsDirectory() ? SHCNE_RMDIR : SHCNE_DELETE, pidl, NULL);
	//::CoTaskMemFree(pidl);
	UpdateRemoveFile(pItem->strFileName, pItem->IsDirectory());
}

void CFTPDirectoryBase::AfterPaste(CFTPDataObject* pObject, DWORD dwEffects)
{
	theApp.CheckClipboardForMoved(pObject, dwEffects);
}

STDMETHODIMP CFTPDirectoryBase::ParseDisplayName2(PIDLIST_RELATIVE pidlParent,
	HWND hWnd,
	LPBC pbc,
	LPWSTR pszDisplayName,
	ULONG* pchEaten,
	PIDLIST_RELATIVE* ppidl,
	ULONG* pdwAttributes)
{
	//bool bSFTPMode;
	ULONG uEaten;
	if (!ppidl)
		return E_POINTER;
	m_hWndOwnerCache = hWnd;
	*ppidl = NULL;

	LPWSTR lpw;
	_StringW strName;
	PIDLIST_RELATIVE pidlCurrent;

	pidlCurrent = NULL;
	uEaten = 0;
	while (true)
	{
		if (*pszDisplayName == L'/')
		{
			pszDisplayName++;
			uEaten++;
		}
		if (!*pszDisplayName)
			break;
		lpw = wcschr(pszDisplayName, L'/');
		if (lpw)
		{
			if (lpw == pszDisplayName)
				continue;
			strName.SetString(pszDisplayName, (DWORD) ((DWORD_PTR) lpw - (DWORD_PTR) pszDisplayName) / sizeof(WCHAR));
		}
		else
			strName = pszDisplayName;

		{
			pszDisplayName += strName.GetLength();
			uEaten += strName.GetLength();

			PITEMID_CHILD pidlChild;
			CFTPFileItem* pItem = NULL;
			if (m_bDirReceived)
				pItem = GetFileItem(strName);
			if (pItem)
				pidlChild = ::CreateFileItem(m_pMallocData->pMalloc, pItem);
			else
				pidlChild = ::CreateDummyFileItem(m_pMallocData->pMalloc, strName);
			if (!pidlChild)
			{
				::CoTaskMemFree(pidlCurrent);
				return E_OUTOFMEMORY;
			}
			PIDLIST_RELATIVE pidl2;
			if (!pidlCurrent && !pidlParent)
				pidl2 = pidlChild;
			else
			{
				pidl2 = (PIDLIST_RELATIVE) ::AppendItemIDList(
					(PCUIDLIST_ABSOLUTE) (pidlCurrent ? pidlCurrent : pidlParent),
					(PCUIDLIST_RELATIVE) pidlChild);
				::CoTaskMemFree(pidlChild);
				if (pidlCurrent)
					::CoTaskMemFree(pidlCurrent);
				if (!pidl2)
					return E_OUTOFMEMORY;
			}
			if (pItem)
			{
				CFTPDirectoryBase* pDir;
				HRESULT hr = OpenNewDirectory(strName, &pDir);
				if (SUCCEEDED(hr))
				{
					ULONG u = 0;
					hr = pDir->ParseDisplayName2(pidl2, hWnd, pbc, pszDisplayName, &u, ppidl, pdwAttributes);
					pDir->Release();
					if (SUCCEEDED(hr))
					{
						uEaten += u;
						::CoTaskMemFree(pidl2);
						if (pdwAttributes)
						{
							//hr = GetAttributesOf(1, (PCUITEMID_CHILD_ARRAY) &pidlCurrent, pdwAttributes);
							*pdwAttributes &= SFGAO_FOLDER;
						}
						if (pchEaten)
							*pchEaten = uEaten;
						return hr;
					}
				}
			}
			pidlCurrent = pidl2;
		}
	}
	if (pdwAttributes)
	{
		//hr = GetAttributesOf(1, (PCUITEMID_CHILD_ARRAY) &pidlCurrent, pdwAttributes);
		*pdwAttributes &= SFGAO_FOLDER;
	}
	if (pchEaten)
		*pchEaten = uEaten;
	if (!pidlCurrent)
		pidlCurrent = (PIDLIST_RELATIVE) ::DuplicateItemIDList((PCUIDLIST_ABSOLUTE) pidlParent);
	*ppidl = pidlCurrent;
	return S_OK;
}

STDMETHODIMP_(void) CFTPDirectoryBase::UpdateItem(PCUITEMID_CHILD pidlOld, PCUITEMID_CHILD pidlNew, LONG lEvent)
{
	CFTPFileItem* pItem = NULL;
	_StringW str;
	if (pidlOld)
	{
		if (!::PickupFileName(pidlOld, str))
			return;
		pItem = GetFileItem(str);
		if (!pItem)
			return;
	}
	str.Empty();
	if (pidlNew)
	{
		if (!::PickupFileName(pidlNew, str))
			return;
	}
	UpdateItem(pItem, str, lEvent);
}

STDMETHODIMP_(void) CFTPDirectoryBase::UpdateItem(CFTPFileItem* pOldItem, LPCWSTR lpszNewItem, LONG lEvent)
{
	switch (lEvent)
	{
		case SHCNE_CREATE:
		case SHCNE_MKDIR:
			break;
		case SHCNE_RENAMEITEM:
		case SHCNE_RENAMEFOLDER:
			pOldItem->strFileName = lpszNewItem;
			break;
		case SHCNE_UPDATEITEM:
		case SHCNE_UPDATEDIR:
			break;
		case SHCNE_DELETE:
		case SHCNE_RMDIR:
		{
			int i = m_aFiles.FindItem(pOldItem);
			m_aFiles.RemoveItem(i);
			pOldItem->Release();
		}
		break;
	}
}


HRESULT CFTPDirectoryBase::OpenNewDirectory(LPCWSTR lpszRelativePath, CFTPDirectoryBase** ppDirectory)
{
	_StringW str;
	LPCWSTR lp = wcschr(lpszRelativePath, L'/');
	if (!lp)
	{
		str = lpszRelativePath;
		lpszRelativePath = NULL;
	}
	else
	{
		str.SetString(lpszRelativePath, (DWORD) (((DWORD_PTR) lp - (DWORD_PTR) lpszRelativePath) / sizeof(WCHAR)));
		lpszRelativePath = lp + 1;
		if (!*lpszRelativePath)
			lpszRelativePath = NULL;
	}
	CFTPFileItem* pItem = NULL;
	if (m_bDirReceived)
	{
		pItem = GetFileItem(str);
		if (pItem)
		{
			if (!pItem->IsDirectory())
				return E_INVALIDARG;
		}
	}

	CFTPDirectoryItem* pDirItem = NULL;
	bool bFound = false;
	for (int i = 0; i < m_aDirectories.GetCount(); i++)
	{
		pDirItem = m_aDirectories.GetItem(i);
		if (pDirItem->strName.Compare(str) == 0)
		{
			bFound = true;
			break;
		}
	}
	if (!bFound)
	{
		pDirItem = new CFTPDirectoryItem();
		pDirItem->pDirectory = NULL;
		pDirItem->strName = str;
	}
	if (!pDirItem->pDirectory)
	{
		PITEMID_CHILD pidlItem;
		if (pItem)
			pidlItem = ::CreateFileItem(m_pRoot->m_pMallocData->pMalloc, pItem);
		else
			pidlItem = ::CreateDummyFileItem(m_pRoot->m_pMallocData->pMalloc, str);
		if (!pidlItem)
		{
			pDirItem->pDirectory = NULL;
			if (!bFound)
				delete pDirItem;
			return E_OUTOFMEMORY;
		}
		((CSFTPFileItem*) pidlItem)->bHasAttribute = true;
		((CSFTPFileItem*) pidlItem)->bIsDirectory = true;
		if (m_strDirectory.GetLength() != 1 || *((LPCWSTR) m_strDirectory) != L'/')
			str.InsertChar(L'/', 0);
		str.InsertString(m_strDirectory, 0);
		HRESULT hr = CreateInstance(pDirItem, this, m_pRoot, str, &pDirItem->pDirectory);
		if (FAILED(hr))
		{
			::CoTaskMemFree(pidlItem);
			pDirItem->pDirectory = NULL;
			if (!bFound)
				delete pDirItem;
			return E_OUTOFMEMORY;
		}
		{
			PIDLIST_ABSOLUTE pidlChild = ::AppendItemIDList(m_pidlMe, pidlItem);
			hr = pDirItem->pDirectory->Initialize(pidlChild);
			::CoTaskMemFree(pidlChild);
			::CoTaskMemFree(pidlItem);
			if (FAILED(hr))
			{
				pDirItem->pDirectory->Release();
				pDirItem->pDirectory = NULL;
				if (!bFound)
					delete pDirItem;
				return E_OUTOFMEMORY;
			}
		}
	}
	else
		pDirItem->pDirectory->AddRef();

	if (lpszRelativePath)
	{
		HRESULT hr = pDirItem->pDirectory->OpenNewDirectory(lpszRelativePath, ppDirectory);
		pDirItem->pDirectory->Release();
		if (SUCCEEDED(hr))
		{
			if (!bFound)
				m_aDirectories.Add(pDirItem);
		}
		else
		{
			//pDirItem->pDirectory = NULL;
			if (!bFound)
				delete pDirItem;
		}
		return hr;
	}
	else
	{
		*ppDirectory = pDirItem->pDirectory;
		//pDirItem->pDirectory->AddRef();
		if (!bFound)
			m_aDirectories.Add(pDirItem);
		return S_OK;
	}
}

CFTPDirectoryItem* CFTPDirectoryBase::GetAlreadyOpenedDirectory(PCUIDLIST_RELATIVE pidlChild)
{
	_StringW str;
	if (!::PickupFileName((PCUITEMID_CHILD) pidlChild, str))
		return NULL;
	pidlChild = (PCUIDLIST_RELATIVE) (((DWORD_PTR) pidlChild) + pidlChild->mkid.cb);
	for (int i = 0; i < m_aDirectories.GetCount(); i++)
	{
		CFTPDirectoryItem* pDirItem = m_aDirectories.GetItem(i);
		if (pDirItem->strName.Compare(str) == 0)
		{
			if (!pDirItem->pDirectory)
				return NULL;
			if (pidlChild->mkid.cb)
				return pDirItem->pDirectory->GetAlreadyOpenedDirectory(pidlChild);
			else
				return pDirItem;
		}
	}
	return NULL;
}

CFTPFileItem* CFTPDirectoryBase::GetFileItem(LPCWSTR lpszName) const
{
	for (int i = 0; i < m_aFiles.GetCount(); i++)
	{
		CFTPFileItem* p = m_aFiles.GetItem(i);
		if (p->strFileName.Compare(lpszName) == 0)
			return p;
	}
	return NULL;
}

void CFTPDirectoryBase::NotifyUpdate(LONG wEventId, LPCWSTR lpszFile1, LPCWSTR lpszFile2)
{
	PIDLIST_RELATIVE pidlR1 = lpszFile1 ? ::CreateFullPathFileItem(m_pMallocData->pMalloc, lpszFile1) : NULL;
	PIDLIST_RELATIVE pidlR2 = lpszFile2 ? ::CreateFullPathFileItem(m_pMallocData->pMalloc, lpszFile2) : NULL;
	PIDLIST_ABSOLUTE pidl1 = pidlR1 ? ::AppendItemIDList(m_pidlMe, pidlR1) : NULL;
	PIDLIST_ABSOLUTE pidl2 = pidlR2 ? ::AppendItemIDList(m_pidlMe, pidlR2) : NULL;
	theApp.MyChangeNotify(wEventId, SHCNF_IDLIST | SHCNF_NOTIFYRECURSIVE | SHCNF_FLUSHNOWAIT, pidl1, pidl2);
	if (pidl1)
		::CoTaskMemFree(pidl1);
	if (pidl2)
		::CoTaskMemFree(pidl2);
	// In registry-emulating mode, the view cannot receive the change notify.
	// So we will update our own views manually.
	if (theApp.m_bEmulateRegMode)
	{
		// pidlR1 and pidlR2 are not actual PITEMID_CHILDren, so
		// we need to parse id-lists to pass real PITEMID_CHILDren to their parents
		CFTPDirectoryBase* pParent = this;
		PCUIDLIST_RELATIVE pidlUR1 = pidlR1;
		PCUIDLIST_RELATIVE pidlUR2 = pidlR2;
		_StringW str;
		pParent->AddRef();
		// only 2 events that use both pidlR1 and pidlR2
		if (wEventId == SHCNE_RENAMEFOLDER || wEventId == SHCNE_RENAMEITEM)
		{
			// pick up the last same parent
			while (pidlUR1 && pidlUR2 && pidlUR1->mkid.cb &&
				pidlUR1->mkid.cb == pidlR2->mkid.cb &&
				memcmp(pidlUR1, pidlUR2, (size_t) pidlUR1->mkid.cb) == 0)
			{
				PickupFileName((PCUITEMID_CHILD) pidlUR1, str);
				CFTPDirectoryBase* pDir2;
				if (FAILED(pParent->OpenNewDirectory(str, &pDir2)))
				{
					pParent->Release();
					pParent = NULL;
					break;
				}
				pParent->Release();
				pParent = pDir2;
				pidlUR1 = (PCUIDLIST_RELATIVE) (((DWORD_PTR) pidlUR1) + pidlUR1->mkid.cb);
				pidlUR2 = (PCUIDLIST_RELATIVE) (((DWORD_PTR) pidlUR2) + pidlUR2->mkid.cb);
			}
		}
		if (pParent)
		{
			// if pidlUR1 and pidlUR2 are the actual PCUITEMID_CHILDren,
			// we can just pass them to pParent->DefViewNotifyUpdate
			if ((wEventId == SHCNE_RENAMEFOLDER || wEventId == SHCNE_RENAMEITEM) &&
				pidlUR1->mkid.cb && pidlUR2->mkid.cb &&
				!(((PCUIDLIST_RELATIVE) (((DWORD_PTR) pidlUR1) + pidlUR1->mkid.cb))->mkid.cb) &&
				!(((PCUIDLIST_RELATIVE) (((DWORD_PTR) pidlUR2) + pidlUR2->mkid.cb))->mkid.cb))
			{
				pParent->DefViewNotifyUpdate(wEventId, (PCUITEMID_CHILD) pidlUR1, (PCUITEMID_CHILD) pidlUR2);
			}
			else
			{
				// picks up and call the method to the last parent of pidlUR1
				if (pidlUR1)
				{
					CFTPDirectoryBase* pDir2 = pParent;
					pDir2->AddRef();
					while (((PCUIDLIST_RELATIVE) (((DWORD_PTR) pidlUR1) + pidlUR1->mkid.cb))->mkid.cb)
					{
						PickupFileName((PCUITEMID_CHILD) pidlUR1, str);
						CFTPDirectoryBase* pDir3;
						if (FAILED(pDir2->OpenNewDirectory(str, &pDir3)))
						{
							pDir2->Release();
							pDir2 = NULL;
							break;
						}
						pDir2->Release();
						pDir2 = pDir3;
						pidlUR1 = (PCUIDLIST_RELATIVE) (((DWORD_PTR) pidlUR1) + pidlUR1->mkid.cb);
					}
					if (pDir2)
					{
						LONG l = wEventId;
						// change event type to the nearest one because
						// the parents of pidlUR1 and pidlUR2 are not same and
						// we cannot pass pidlUR2 to the parent of pidlUR1
						if (l == SHCNE_RENAMEFOLDER)
							l = SHCNE_RMDIR;
						else if (l == SHCNE_RENAMEITEM)
							l = SHCNE_DELETE;
						pDir2->DefViewNotifyUpdate(l, (PCUITEMID_CHILD) pidlUR1, NULL);
						pDir2->Release();
					}
				}
				// picks up and call the method to the last parent of pidlUR2
				if (pidlUR2)
				{
					CFTPDirectoryBase* pDir2 = pParent;
					pDir2->AddRef();
					while (((PCUIDLIST_RELATIVE) (((DWORD_PTR) pidlUR2) + pidlUR2->mkid.cb))->mkid.cb)
					{
						PickupFileName((PCUITEMID_CHILD) pidlUR2, str);
						CFTPDirectoryBase* pDir3;
						if (FAILED(pDir2->OpenNewDirectory(str, &pDir3)))
						{
							pDir2->Release();
							pDir2 = NULL;
							break;
						}
						pDir2->Release();
						pDir2 = pDir3;
						pidlUR2 = (PCUIDLIST_RELATIVE) (((DWORD_PTR) pidlUR2) + pidlUR1->mkid.cb);
					}
					if (pDir2)
					{
						LONG l = wEventId;
						// same as the case of pidlUR1, but
						// the order of arguments is a little different
						if (l == SHCNE_RENAMEFOLDER || l == SHCNE_RENAMEITEM)
						{
							if (l == SHCNE_RENAMEFOLDER)
								l = SHCNE_MKDIR;
							else if (l == SHCNE_RENAMEITEM)
								l = SHCNE_CREATE;
							pidlUR1 = pidlUR2;
							pidlUR2 = NULL;
						}
						else
							pidlUR1 = NULL;
						pDir2->DefViewNotifyUpdate(l, (PCUITEMID_CHILD) pidlUR1, (PCUITEMID_CHILD) pidlUR2);
						pDir2->Release();
					}
				}
				//DefViewNotifyUpdate(wEventId, (PCUITEMID_CHILD) pidlR1, (PCUITEMID_CHILD) pidlR2);
			}
			pParent->Release();
		}
	}
	if (pidlR1)
		::CoTaskMemFree(pidlR1);
	if (pidlR2)
		::CoTaskMemFree(pidlR2);
}

void CFTPDirectoryBase::UpdateNewFile(LPCWSTR lpszFileName, bool bDirectory)
{
	CFTPFileItem* p = m_pRoot->RetrieveFileItem(this, lpszFileName);
	if (p)
	{
		::EnterCriticalSection(&m_csFiles);
		m_aFiles.Add(p);
		::LeaveCriticalSection(&m_csFiles);
	}
	NotifyUpdate(bDirectory ? SHCNE_MKDIR : SHCNE_CREATE, lpszFileName, NULL);
}

void CFTPDirectoryBase::UpdateMoveFile(LPCWSTR lpszFromDir, LPCWSTR lpszFileName, bool bDirectory)
{
	_StringW strFP(lpszFromDir), strTP(m_strDirectory);
	if (m_strDirectory.Compare(lpszFromDir) != 0)
	{
		CFTPFileItem* p = m_pRoot->RetrieveFileItem(this, lpszFileName);
		if (p)
		{
			::EnterCriticalSection(&m_csFiles);
			m_aFiles.Add(p);
			::LeaveCriticalSection(&m_csFiles);
		}
	}
	if (strFP.IsEmpty() || ((LPCWSTR) strFP)[strFP.GetLength() - 1] != L'/')
		strFP += L'/';
	if (strTP.IsEmpty() || ((LPCWSTR) strTP)[strTP.GetLength() - 1] != L'/')
		strTP += L'/';
	strFP += lpszFileName;
	strTP += lpszFileName;
	NotifyUpdate(bDirectory ? SHCNE_RENAMEFOLDER : SHCNE_RENAMEITEM, strFP, strTP);
}

void CFTPDirectoryBase::UpdateRenameFile(LPCWSTR lpszOldFileName, LPCWSTR lpszNewFileName, bool bDirectory)
{
	_StringW strFP(lpszOldFileName);
	//_StringW strFP(m_strDirectory), strTP(m_strDirectory);
	//::EnterCriticalSection(&m_csFiles);
	CFTPFileItem* p = GetFileItem(lpszOldFileName);
	if (p)
		p->strFileName = lpszNewFileName;
	::LeaveCriticalSection(&m_csFiles);
	//if (m_strDirectory.IsEmpty() || ((LPCWSTR) m_strDirectory)[m_strDirectory.GetLength() - 1] != L'/')
	//{
	//	strFP += L'/';
	//	strTP += L'/';
	//}
	//strFP += lpszOldFileName;
	//strTP += lpszNewFileName;
	//NotifyUpdate(bDirectory ? SHCNE_RENAMEFOLDER : SHCNE_RENAMEITEM, strFP, strTP);
	NotifyUpdate(bDirectory ? SHCNE_RENAMEFOLDER : SHCNE_RENAMEITEM, strFP, lpszNewFileName);
}

void CFTPDirectoryBase::UpdateFileAttrs(LPCWSTR lpszFileName, bool bDirectory)
{
	_StringW str(lpszFileName);
	::EnterCriticalSection(&m_csFiles);
	for (int i = 0; i < m_aFiles.GetCount(); i++)
	{
		CFTPFileItem* p = m_aFiles.GetItem(i);
		if (p->strFileName.Compare(lpszFileName) == 0)
		{
			CFTPFileItem* p2 = m_pRoot->RetrieveFileItem(this, lpszFileName);
			if (p2)
			{
				m_aFiles.SetItem(i, p2);
				p->Release();
			}
			break;
		}
	}
	::LeaveCriticalSection(&m_csFiles);
	NotifyUpdate(bDirectory ? SHCNE_UPDATEDIR : SHCNE_UPDATEITEM, str, NULL);
}

void CFTPDirectoryBase::UpdateRemoveFile(LPCWSTR lpszFileName, bool bDirectory)
{
	_StringW str(lpszFileName);
	::EnterCriticalSection(&m_csFiles);
	for (int i = 0; i < m_aFiles.GetCount(); i++)
	{
		CFTPFileItem* p = m_aFiles.GetItem(i);
		if (p->strFileName.Compare(lpszFileName) == 0)
		{
			m_aFiles.RemoveItem(i);
			p->Release();
			break;
		}
	}
	::LeaveCriticalSection(&m_csFiles);
	NotifyUpdate(bDirectory ? SHCNE_RMDIR : SHCNE_DELETE, str, NULL);
}

void CFTPDirectoryBase::RemoveAllFiles()
{
	::EnterCriticalSection(&m_csFiles);
	int i;
	i = m_aFiles.GetCount();
	while (i--)
		m_aFiles.GetItem(i)->Release();
	m_aFiles.RemoveAll();
	::LeaveCriticalSection(&m_csFiles);
	i = m_aDirectories.GetCount();
	while (i--)
	{
		CFTPDirectoryItem* pItem = m_aDirectories.GetItem(i);
		m_aDirectories.RemoveItem(i);
		if (pItem->pDirectory)
		{
			pItem->pDirectory->RemoveAllFiles();
			//pItem->pDirectory->Release();
		}
		if (!pItem->strName.IsEmpty())
		{
			PITEMID_CHILD pidlChild = ::CreateDummyFileItem(m_pMallocData->pMalloc, pItem->strName);
			if (pidlChild)
			{
				NotifyUpdate(SHCNE_RMDIR, pidlChild, NULL);
				m_pMallocData->pMalloc->Free(pidlChild);
			}
		}
		delete pItem;
	}
	//m_aDirectories.RemoveAll();
	m_bDirReceived = false;
}

CFTPDirectoryRootBase::CFTPDirectoryRootBase(CDelegateMallocData* pMallocData, CFTPDirectoryItem* pItemMe)
	: CFTPDirectoryBase(pMallocData, pItemMe)
{
	m_bIsRoot = true;
	m_pRoot = this;
	m_strDirectory = L"/";
	m_bTextMode = TEXTMODE_NO_CONVERT;
	m_nServerCharset = scsUTF8;
	m_nTransferMode = TRANSFER_MODE_AUTO;
	m_arrTextFileType.CopyArray(theApp.m_arrDefTextFileType);
	m_bUseSystemTextFileType = true;
	m_pObjectOnClipboard = NULL;
	m_bUseThumbnailPreview = true;
}

STDMETHODIMP CFTPDirectoryRootBase::QueryInterface(REFIID riid, void** ppv)
{
	if (!ppv)
		return E_POINTER;
	if (IsEqualIID(riid, IID_IExtractIconA) || IsEqualIID(riid, IID_IExtractIconW))
	{
		CEasySFTPRootIcon* pIcon = new CEasySFTPRootIcon(IML_INDEX_NETDRIVE);
		HRESULT hr = pIcon->QueryInterface(riid, ppv);
		pIcon->Release();
		return hr;
	}
	else if (IsEqualIID(riid, IID_IContextMenu))
	{
		*ppv = NULL;
		return E_NOINTERFACE;
	}
	return CFTPDirectoryBase::QueryInterface(riid, ppv);
}

STDMETHODIMP CFTPDirectoryRootBase::GetHostInfo(VARIANT_BOOL* pbIsSFTP, int* pnPort, BSTR* pbstrHostName)
{
	int nPort;
	LPCWSTR lpw;
	lpw = GetProtocolName(nPort);
	if (pbIsSFTP)
		*pbIsSFTP = (_wcsicmp(lpw, L"sftp") == 0);
	if (pnPort)
		*pnPort = nPort;
	if (pbstrHostName)
	{
		*pbstrHostName = ::SysAllocStringLen(m_strHostName, (UINT) m_strHostName.GetLength());
		if (!*pbstrHostName)
			return E_OUTOFMEMORY;
	}
	return S_OK;
}

STDMETHODIMP CFTPDirectoryRootBase::GetTextMode(LONG* pnTextMode)
{
	if (!pnTextMode)
		return E_POINTER;
	*pnTextMode = (LONG) m_bTextMode;
	return S_OK;
}

STDMETHODIMP CFTPDirectoryRootBase::SetTextMode(LONG nTextMode)
{
	if (nTextMode >= 0x100 || nTextMode < 0)
		return E_INVALIDARG;
	m_bTextMode = (BYTE) nTextMode;
	return S_OK;
}

STDMETHODIMP CFTPDirectoryRootBase::GetTransferMode(LONG* pnTransferMode)
{
	if (!pnTransferMode)
		return E_POINTER;
	*pnTransferMode = (LONG) m_nTransferMode;
	return S_OK;
}

STDMETHODIMP CFTPDirectoryRootBase::SetTransferMode(LONG nTransferMode)
{
	if (nTransferMode >= 0x80 || nTransferMode < 0)
		return E_INVALIDARG;
	m_nTransferMode = (BYTE) nTransferMode;
	return S_OK;
}

STDMETHODIMP CFTPDirectoryRootBase::IsTextFile(LPCWSTR lpszFileName)
{
	if (m_nTransferMode == TRANSFER_MODE_TEXT)
		return S_OK;
	else if (m_nTransferMode == TRANSFER_MODE_BINARY)
		return S_FALSE;
	if (!lpszFileName)
		return E_POINTER;

	LPCWSTR lp;
	lp = wcsrchr(lpszFileName, L'\\');
	if (lp)
		lpszFileName = lp + 1;
	lp = wcsrchr(lpszFileName, L'/');
	if (lp)
		lpszFileName = lp + 1;
	for (int i = 0; i < m_arrTextFileType.GetCount(); i++)
	{
		if (::MyMatchWildcardW(lpszFileName, m_arrTextFileType.GetItem(i)))
			return S_OK;
	}
	if (m_bUseSystemTextFileType)
	{
		lp = wcsrchr(lpszFileName, L'.');
		if (lp)
		{
			_StringW str(lp);
			HKEY hKey;
			DWORD dw;
			bool bRet = false;
			if (::RegOpenKeyEx(HKEY_CLASSES_ROOT, str, 0, KEY_QUERY_VALUE, &hKey) == ERROR_SUCCESS)
			{
				dw = MAX_PATH;
				if (::RegQueryValueEx(hKey, _T("PerceivedType"), NULL, NULL,
#ifdef _UNICODE
					(LPBYTE) str.GetBuffer(MAX_PATH),
#else
					(LPBYTE) str.GetBufferA(MAX_PATH),
#endif
					&dw) == ERROR_SUCCESS)
				{
#ifdef _UNICODE
					str.ReleaseBuffer();
#else
					str.ReleaseBufferA();
#endif
					if (str.Compare(L"text", true) == 0)
						bRet = true;
				}
				if (!bRet)
				{
					if (::RegQueryValueEx(hKey, _T("Content Type"), NULL, NULL,
#ifdef _UNICODE
						(LPBYTE) str.GetBuffer(MAX_PATH),
#else
						(LPBYTE) str.GetBufferA(MAX_PATH),
#endif
						&dw) == ERROR_SUCCESS)
					{
#ifdef _UNICODE
						str.ReleaseBuffer();
#else
						str.ReleaseBufferA();
#endif
						lpszFileName = str;
						lp = wcschr(lpszFileName, L'/');
						if (lp && _wcsnicmp(lpszFileName, L"text", ((size_t)((DWORD_PTR) lp - (DWORD_PTR) lpszFileName)) / sizeof(WCHAR)) == 0)
							bRet = true;
					}
				}
				::RegCloseKey(hKey);
			}
			return bRet ? S_OK : S_FALSE;
		}
	}

	return S_FALSE;
}

//void CFTPDirectoryRootBase::BeforeClipboardOperation(IDataObject* pObjectNew)
//{
//}
//
//void CFTPDirectoryRootBase::AfterClipboardOperation(IDataObject* pObjectNew)
//{
//}

void CFTPDirectoryRootBase::ShowServerInfoDialog(HWND hWndOwner)
{
	CServerInfoDialog dlg;
	PreShowServerInfoDialog(&dlg);
	dlg.ModalDialogW(hWndOwner);
}

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

CFTPFileItemIcon::CFTPFileItemIcon(CFTPFileItem* pItem)
	: m_uRef(1)
{
	while (pItem->pTargetFile)
		pItem = pItem->pTargetFile;
	if (pItem->iIconIndex == -1)
		FillFileItemInfo(pItem);
	m_strFileName = pItem->strFileName;
	m_bIsDirectory = pItem->IsDirectory();
	m_iIconIndex = pItem->iIconIndex;
	m_iOpenIconIndex = pItem->iOpenIconIndex;
}

CFTPFileItemIcon::CFTPFileItemIcon(LPCWSTR lpszFileName, bool bIsDirectory)
	: m_uRef(1)
{
	m_strFileName = lpszFileName;
	m_bIsDirectory = bIsDirectory;
	GetDummyFileItemIcon(lpszFileName, bIsDirectory, m_iIconIndex, m_iOpenIconIndex);
}

CFTPFileItemIcon::~CFTPFileItemIcon()
{
}

STDMETHODIMP CFTPFileItemIcon::QueryInterface(REFIID riid, void** ppv)
{
	if (!ppv)
		return E_POINTER;

	// NOTE: we must set NULL to *ppv, or causes 0xC000041D exception in Windows 7 (x64)
	*ppv = NULL;

	if (IsEqualIID(riid, IID_IUnknown) ||
		IsEqualIID(riid, IID_IExtractIconW))
	{
		*ppv = (IExtractIconW*) this;
		AddRef();
		return S_OK;
	}
	else if (IsEqualIID(riid, IID_IExtractIconA))
	{
		*ppv = (IExtractIconA*) this;
		AddRef();
		return S_OK;
	}
	return E_NOINTERFACE;
}

STDMETHODIMP_(ULONG) CFTPFileItemIcon::AddRef()
{
	return (ULONG) ::InterlockedIncrement((LONG*) &m_uRef);
}

STDMETHODIMP_(ULONG) CFTPFileItemIcon::Release()
{
	ULONG u = (ULONG) ::InterlockedDecrement((LONG*) &m_uRef);
	if (!u)
		delete this;
	return u;
}

STDMETHODIMP CFTPFileItemIcon::GetIconLocation(UINT uFlags, LPWSTR pszIconFile, UINT cchMax, int* piIndex, UINT *pwFlags)
{
	if (!piIndex)
		return E_POINTER;
	if (cchMax > 0)
	{
		if (!pszIconFile)
			return E_POINTER;
		DWORD dw = m_strFileName.GetLength();
		if (cchMax >= dw + 1)
			cchMax = dw + 1;
		memcpy(pszIconFile, (LPCWSTR) m_strFileName, sizeof(WCHAR) * (cchMax - 1));
		pszIconFile[cchMax - 1] = 0;
	}
	if ((uFlags & GIL_OPENICON) && m_bIsDirectory)
		*piIndex = 1;
	else
		*piIndex = 0;
	if (pwFlags)
		*pwFlags = GIL_NOTFILENAME;
	return S_OK;
}

HRESULT CFTPFileItemIcon::DoExtract(bool bOpenIcon, HICON* phIconLarge, HICON* phIconSmall, UINT nIconSize)
{
	int iIndex = bOpenIcon ? m_iOpenIconIndex : m_iIconIndex;
	if (phIconLarge)
	{
		if (LOWORD(nIconSize) == 256)
		{
			if (!theApp.m_pimlSysIconJumbo)
			{
				*phIconLarge = NULL;
				return E_NOTIMPL;
			}
			HRESULT hr = theApp.m_pimlSysIconJumbo->GetIcon(iIndex, ILD_TRANSPARENT, phIconLarge);
			if (FAILED(hr))
				return hr;
		}
		if (LOWORD(nIconSize) == 48)
		{
			if (!theApp.m_pimlSysIconExtraLarge)
			{
				*phIconLarge = NULL;
				return E_NOTIMPL;
			}
			HRESULT hr = theApp.m_pimlSysIconExtraLarge->GetIcon(iIndex, ILD_TRANSPARENT, phIconLarge);
			if (FAILED(hr))
				return hr;
		}
		else if (LOWORD(nIconSize) == 32)
			*phIconLarge = ::ImageList_GetIcon(theApp.m_himlSysIconLarge, iIndex, ILD_TRANSPARENT);
		else
		{
			*phIconLarge = NULL;
			return E_NOTIMPL;
		}
		if (!*phIconLarge)
			return E_OUTOFMEMORY;
	}
	if (phIconSmall)
	{
		if (HIWORD(nIconSize) == 16)
			*phIconSmall = ::ImageList_GetIcon(theApp.m_himlSysIconSmall, iIndex, ILD_TRANSPARENT);
		else
		{
			if (phIconLarge)
			{
				::DestroyIcon(*phIconLarge);
				*phIconLarge = NULL;
			}
			*phIconSmall = NULL;
			return E_NOTIMPL;
		}
		if (!*phIconSmall)
		{
			if (phIconLarge)
			{
				::DestroyIcon(*phIconLarge);
				*phIconLarge = NULL;
			}
			return E_OUTOFMEMORY;
		}
	}
	return S_OK;
}

STDMETHODIMP CFTPFileItemIcon::Extract(LPCWSTR pszFile, UINT nIconIndex, HICON* phIconLarge, HICON* phIconSmall, UINT nIconSize)
{
	if (!pszFile)
		return E_POINTER;
	if (m_strFileName.Compare(pszFile) != 0)
		return E_INVALIDARG;
	return DoExtract(nIconIndex == 1, phIconLarge, phIconSmall, nIconSize);
}

STDMETHODIMP CFTPFileItemIcon::GetIconLocation(UINT uFlags, LPSTR pszIconFile, UINT cchMax, int* piIndex, UINT *pwFlags)
{
	if (!piIndex)
		return E_POINTER;
	if (cchMax > 0)
	{
		if (!pszIconFile)
			return E_POINTER;
		DWORD dw = m_strFileName.GetLengthA();
		if (cchMax >= dw + 1)
			cchMax = dw + 1;
		memcpy(pszIconFile, (LPCSTR) m_strFileName, sizeof(CHAR) * (cchMax - 1));
		pszIconFile[cchMax - 1] = 0;
	}
	if ((uFlags & GIL_OPENICON) && m_bIsDirectory)
		*piIndex = 1;
	else
		*piIndex = 0;
	if (pwFlags)
		*pwFlags = 0;
	return S_OK;
}

STDMETHODIMP CFTPFileItemIcon::Extract(LPCSTR pszFile, UINT nIconIndex, HICON* phIconLarge, HICON* phIconSmall, UINT nIconSize)
{
	if (!pszFile)
		return E_POINTER;
	if (m_strFileName.Compare(pszFile) != 0)
		return E_INVALIDARG;
	return DoExtract(nIconIndex == 1, phIconLarge, phIconSmall, nIconSize);
}
