CFSDropTarget Class Reference

#include <CFSDropTarget.h>

Inheritance diagram for CFSDropTarget:

Collaboration diagram for CFSDropTarget:

Public Member Functions
	CFSDropTarget ()
	~CFSDropTarget ()
HRESULT	Initialize (LPWSTR PathTarget)
STDMETHOD()	DragEnter (IDataObject *pDataObject, DWORD dwKeyState, POINTL pt, DWORD *pdwEffect) override
STDMETHOD()	DragOver (DWORD dwKeyState, POINTL pt, DWORD *pdwEffect) override
STDMETHOD()	DragLeave () override
STDMETHOD()	Drop (IDataObject *pDataObject, DWORD dwKeyState, POINTL pt, DWORD *pdwEffect) override
STDMETHOD()	SetSite (IUnknown *pUnkSite) override
STDMETHOD()	GetSite (REFIID riid, void **ppvSite) override
Public Member Functions inherited from ATL::CComObjectRootEx< CComMultiThreadModelNoCS >
	~CComObjectRootEx ()
ULONG	InternalAddRef ()
ULONG	InternalRelease ()
void	Lock ()
void	Unlock ()
HRESULT	_AtlInitialConstruct ()
Public Member Functions inherited from ATL::CComObjectRootBase
	CComObjectRootBase ()
	~CComObjectRootBase ()
void	SetVoid (void *)
HRESULT	_AtlFinalConstruct ()
HRESULT	FinalConstruct ()
void	InternalFinalConstructAddRef ()
void	InternalFinalConstructRelease ()
void	FinalRelease ()
HRESULT	DragEnter ([in, unique] IDataObject *pDataObj, [in] DWORD grfKeyState, [in] POINTL pt, [in, out] DWORD *pdwEffect)
HRESULT	DragOver ([in] DWORD grfKeyState, [in] POINTL pt, [in, out] DWORD *pdwEffect)
HRESULT	DragLeave ()
HRESULT	Drop ([in, unique] IDataObject *pDataObj, [in] DWORD grfKeyState, [in] POINTL pt, [in, out] DWORD *pdwEffect)
Public Member Functions inherited from IUnknown
HRESULT	QueryInterface ([in] REFIID riid, [out, iid_is(riid)] void **ppvObject)
ULONG	AddRef ()
ULONG	Release ()
Public Member Functions inherited from IObjectWithSite
HRESULT	SetSite ([in] IUnknown *pUnkSite)
HRESULT	GetSite ([in] REFIID riid, [out, iid_is(riid)] PVOID *ppvSite)

Static Public Member Functions
static UINT	TrackPopupMenu (HMENU hMenu, const POINTL &pt)
Static Public Member Functions inherited from ATL::CComObjectRootBase
static void WINAPI	ObjectMain (bool)
static const struct _ATL_CATMAP_ENTRY *	GetCategoryMap ()
static HRESULT WINAPI	InternalQueryInterface (void *pThis, const _ATL_INTMAP_ENTRY *pEntries, REFIID iid, void **ppvObject)

Private Member Functions
BOOL	_QueryDrop (DWORD dwKeyState, LPDWORD pdwEffect)
HRESULT	_DoDrop (IDataObject *pDataObject, DWORD dwKeyState, POINTL pt, DWORD *pdwEffect)
HRESULT	_CopyItems (IDataObject *pDO, IShellFolder *pSFFrom, UINT cidl, LPCITEMIDLIST *apidl, BOOL bCopy)
BOOL	_GetUniqueFileName (LPCWSTR pwszBasePath, LPCWSTR pwszExt, LPWSTR pwszTarget, BOOL bShortcut)
HRESULT	_GetEffectFromMenu (IDataObject *pDataObject, POINTL pt, DWORD *pdwEffect, DWORD dwAvailableEffects)
HRESULT	_RepositionItems (IShellFolderView *psfv, IDataObject *pDataObject, POINTL pt)

Static Private Member Functions
static DWORD WINAPI	_DoDropThreadProc (LPVOID lpParameter)

Private Attributes
UINT	m_cfShellIDList
BOOL	m_fAcceptFmt
LPWSTR	m_sPathTarget
HWND	m_hwndSite
DWORD	m_grfKeyState
DWORD	m_dwDefaultEffect
DWORD	m_AllowedEffects
CComPtr< IUnknown >	m_site

Additional Inherited Members
Public Types inherited from IDropTarget
typedef IDropTarget *	LPDROPTARGET
Public Types inherited from IUnknown
typedef IUnknown *	LPUNKNOWN
Public Types inherited from IObjectWithSite
typedef IObjectWithSite *	LPOBJECTWITHSITE
Public Attributes inherited from ATL::CComObjectRootBase
LONG	m_dwRef
Public Attributes inherited from IDropTarget
const DWORD	MK_ALT = 0x20
const DWORD	DROPEFFECT_NONE = 0
const DWORD	DROPEFFECT_COPY = 1
const DWORD	DROPEFFECT_MOVE = 2
const DWORD	DROPEFFECT_LINK = 4
const DWORD	DROPEFFECT_SCROLL = 0x80000000
const DWORD	DD_DEFSCROLLINSET = 11
const DWORD	DD_DEFSCROLLDELAY = 50
const DWORD	DD_DEFSCROLLINTERVAL = 50
const DWORD	DD_DEFDRAGDELAY = 200
const DWORD	DD_DEFDRAGMINDIST = 2

Detailed Description

Definition at line 29 of file CFSDropTarget.h.

Constructor & Destructor Documentation

CFSDropTarget()

CFSDropTarget::CFSDropTarget

(

)

Definition at line 157 of file CFSDropTarget.cpp.

                            :
    m_cfShellIDList(0),
    m_fAcceptFmt(FALSE),
    m_sPathTarget(NULL),
    m_hwndSite(NULL),
    m_grfKeyState(0),
    m_AllowedEffects(0)
{
}

~CFSDropTarget()

CFSDropTarget::~CFSDropTarget

(

)

Definition at line 179 of file CFSDropTarget.cpp.

{
    SHFree(m_sPathTarget);
}

Member Function Documentation

_CopyItems()

HRESULT CFSDropTarget::_CopyItems ( IDataObject *  pDO, IShellFolder *  pSFFrom, UINT  cidl, LPCITEMIDLIST *  apidl, BOOL  bCopy  )

private

Definition at line 87 of file CFSDropTarget.cpp.

{
    HRESULT ret;
    WCHAR wszDstPath[MAX_PATH + 1] = {0};
    PWCHAR pwszSrcPathsList = (PWCHAR) HeapAlloc(GetProcessHeap(), 0, MAX_PATH * sizeof(WCHAR) * cidl + 1);
    if (!pwszSrcPathsList)
        return E_OUTOFMEMORY;
 
    PWCHAR pwszListPos = pwszSrcPathsList;
    STRRET strretFrom;
    SHFILEOPSTRUCTW fop;
    BOOL bRenameOnCollision = FALSE;
 
    /* Build a double null terminated list of C strings from source paths */
    for (UINT i = 0; i < cidl; i++)
    {
        ret = pSFFrom->GetDisplayNameOf(apidl[i], SHGDN_FORPARSING, &strretFrom);
        if (FAILED(ret))
            goto cleanup;
 
        ret = StrRetToBufW(&strretFrom, NULL, pwszListPos, MAX_PATH);
        if (FAILED(ret))
            goto cleanup;
 
        pwszListPos += lstrlenW(pwszListPos) + 1;
    }
    /* Append the final null. */
    *pwszListPos = L'\0';
 
    /* Build a double null terminated target (this path) */
    ret = StringCchCopyW(wszDstPath, MAX_PATH, m_sPathTarget);
    if (FAILED(ret))
        goto cleanup;
 
    wszDstPath[lstrlenW(wszDstPath) + 1] = UNICODE_NULL;
 
    /* Set bRenameOnCollision to TRUE if necesssary */
    if (bCopy)
    {
        WCHAR szPath1[MAX_PATH], szPath2[MAX_PATH];
        GetFullPathNameW(pwszSrcPathsList, _countof(szPath1), szPath1, NULL);
        GetFullPathNameW(wszDstPath, _countof(szPath2), szPath2, NULL);
        PathRemoveFileSpecW(szPath1);
        if (_wcsicmp(szPath1, szPath2) == 0)
            bRenameOnCollision = TRUE;
    }
 
    ZeroMemory(&fop, sizeof(fop));
    fop.hwnd = m_hwndSite;
    fop.wFunc = bCopy ? FO_COPY : FO_MOVE;
    fop.pFrom = pwszSrcPathsList;
    fop.pTo = wszDstPath;
    fop.fFlags = GetDefaultFileOpFlags(pDO);
    if (bRenameOnCollision)
        fop.fFlags |= FOF_RENAMEONCOLLISION;
 
    ret = S_OK;
 
    if (SHFileOperationW(&fop))
    {
        ERR("SHFileOperationW failed\n");
        ret = E_FAIL;
    }
 
cleanup:
    HeapFree(GetProcessHeap(), 0, pwszSrcPathsList);
    return ret;
}

Referenced by _DoDrop().

_DoDrop()

HRESULT CFSDropTarget::_DoDrop ( IDataObject *  pDataObject, DWORD  dwKeyState, POINTL  pt, DWORD *  pdwEffect  )

private

Definition at line 559 of file CFSDropTarget.cpp.

{
    TRACE("(%p) performing drop, effect %u\n", this, *pdwEffect);
    FORMATETC fmt;
    FORMATETC fmt2;
    STGMEDIUM medium;
 
    InitFormatEtc (fmt, m_cfShellIDList, TYMED_HGLOBAL);
    InitFormatEtc (fmt2, CF_HDROP, TYMED_HGLOBAL);
 
    HRESULT hr;
    bool bCopy = TRUE;
    bool bLinking = FALSE;
 
    /* Figure out what drop operation we're doing */
    if (pdwEffect)
    {
        TRACE("Current drop effect flag %i\n", *pdwEffect);
        if ((*pdwEffect & DROPEFFECT_MOVE) == DROPEFFECT_MOVE)
            bCopy = FALSE;
        if ((*pdwEffect & DROPEFFECT_LINK) == DROPEFFECT_LINK)
            bLinking = TRUE;
    }
 
    if (SUCCEEDED(hr = pDataObject->QueryGetData(&fmt)))
    {
        hr = pDataObject->GetData(&fmt, &medium);
        TRACE("CFSTR_SHELLIDLIST\n");
        if (FAILED(hr))
        {
            ERR("CFSTR_SHELLIDLIST failed\n");
        }
        /* lock the handle */
        LPIDA lpcida = (LPIDA)GlobalLock(medium.hGlobal);
        if (!lpcida)
        {
            ReleaseStgMedium(&medium);
            return E_FAIL;
        }
 
        /* convert the data into pidl */
        LPITEMIDLIST pidl;
        LPITEMIDLIST *apidl = _ILCopyCidaToaPidl(&pidl, lpcida);
        if (!apidl)
        {
            ReleaseStgMedium(&medium);
            return E_FAIL;
        }
 
        CComPtr<IShellFolder> psfDesktop;
        CComPtr<IShellFolder> psfFrom = NULL;
 
        /* Grab the desktop shell folder */
        hr = SHGetDesktopFolder(&psfDesktop);
        if (FAILED(hr))
        {
            ERR("SHGetDesktopFolder failed\n");
            SHFree(pidl);
            _ILFreeaPidl(apidl, lpcida->cidl);
            ReleaseStgMedium(&medium);
            return E_FAIL;
        }
 
        /* Find source folder, this is where the clipboard data was copied from */
        if (_ILIsDesktop(pidl))
        {
            /* use desktop shell folder */
            psfFrom = psfDesktop;
        }
        else
        {
            hr = psfDesktop->BindToObject(pidl, NULL, IID_PPV_ARG(IShellFolder, &psfFrom));
            if (FAILED(hr))
            {
                ERR("no IShellFolder\n");
                SHFree(pidl);
                _ILFreeaPidl(apidl, lpcida->cidl);
                ReleaseStgMedium(&medium);
                return E_FAIL;
            }
        }
 
        if (bLinking)
        {
            WCHAR wszNewLnk[MAX_PATH];
 
            TRACE("target path = %s\n", debugstr_w(m_sPathTarget));
 
            /* We need to create a link for each pidl in the copied items, so step through the pidls from the clipboard */
            for (UINT i = 0; i < lpcida->cidl; i++)
            {
                SFGAOF att = SHGetAttributes(psfFrom, apidl[i], SFGAO_FOLDER | SFGAO_STREAM | SFGAO_FILESYSTEM);
                CComHeapPtr<ITEMIDLIST_ABSOLUTE> pidlFull;
                hr = SHILCombine(pidl, apidl[i], &pidlFull);
 
                WCHAR targetName[MAX_PATH];
                if (SUCCEEDED(hr))
                {
                    // If the target is a file, we use SHGDN_FORPARSING because "NeverShowExt" will hide the ".lnk" extension.
                    // If the target is a virtual item, we ask for the friendly name because SHGDN_FORPARSING will return a GUID.
                    BOOL UseParsing = (att & (SFGAO_FILESYSTEM | SFGAO_FOLDER)) == SFGAO_FILESYSTEM;
                    DWORD ShgdnFor = UseParsing ? SHGDN_FORPARSING : SHGDN_FOREDITING;
                    hr = DisplayNameOfW(psfFrom, apidl[i], ShgdnFor | SHGDN_INFOLDER, targetName, _countof(targetName));
                }
                if (FAILED_UNEXPECTEDLY(hr))
                {
                    SHELL_ErrorBox(m_hwndSite, hr);
                    break;
                }
                SHELL_StripIllegalFsNameCharacters(targetName);
 
                WCHAR wszCombined[MAX_PATH + _countof(targetName)];
                PathCombineW(wszCombined, m_sPathTarget, targetName);
 
                // Check to see if the source is a link
                BOOL fSourceIsLink = FALSE;
                if (!_wcsicmp(PathFindExtensionW(targetName), L".lnk") && (att & (SFGAO_FOLDER | SFGAO_STREAM)) != SFGAO_FOLDER)
                {
                    fSourceIsLink = TRUE;
                    PathRemoveExtensionW(wszCombined);
                }
 
                // Create a pathname to save the new link.
                _GetUniqueFileName(wszCombined, L".lnk", wszNewLnk, TRUE);
 
                CComPtr<IPersistFile> ppf;
                if (fSourceIsLink)
                {
                    PWSTR pwszTargetFull;
                    hr = SHGetNameFromIDList(pidlFull, SIGDN_DESKTOPABSOLUTEPARSING, &pwszTargetFull);
                    if (!FAILED_UNEXPECTEDLY(hr))
                    {
                        hr = IShellLink_ConstructFromPath(pwszTargetFull, IID_PPV_ARG(IPersistFile, &ppf));
                        FAILED_UNEXPECTEDLY(hr);
                        SHFree(pwszTargetFull);
                    }
                }
                else
                {
                    CComPtr<IShellLinkW> pLink;
                    hr = CShellLink::_CreatorClass::CreateInstance(NULL, IID_PPV_ARG(IShellLinkW, &pLink));
                    if (!FAILED_UNEXPECTEDLY(hr))
                    {
                        hr = pLink->SetIDList(pidlFull);
                        if (!FAILED_UNEXPECTEDLY(hr))
                            hr = pLink->QueryInterface(IID_PPV_ARG(IPersistFile, &ppf));
 
                        PWSTR pwszPath, pSep;
                        if ((att & SFGAO_FILESYSTEM) && SUCCEEDED(SHGetNameFromIDList(pidlFull, SIGDN_FILESYSPATH, &pwszPath)))
                        {
                            if ((pSep = PathFindFileNameW(pwszPath)) > pwszPath)
                            {
                                pSep[-1] = UNICODE_NULL;
                                pLink->SetWorkingDirectory(pwszPath);
                            }
                            SHFree(pwszPath);
                        }
                    }
                }
                if (SUCCEEDED(hr))
                    hr = ppf->Save(wszNewLnk, !fSourceIsLink);
                if (FAILED_UNEXPECTEDLY(hr))
                {
                    SHELL_ErrorBox(m_hwndSite, hr);
                    break;
                }
                SHChangeNotify(SHCNE_CREATE, SHCNF_PATHW, wszNewLnk, NULL);
            }
        }
        else
        {
            hr = _CopyItems(pDataObject, psfFrom, lpcida->cidl, (LPCITEMIDLIST*)apidl, bCopy);
        }
 
        SHFree(pidl);
        _ILFreeaPidl(apidl, lpcida->cidl);
        ReleaseStgMedium(&medium);
    }
    else if (SUCCEEDED(hr = pDataObject->QueryGetData(&fmt2)))
    {
        FORMATETC fmt2;
        InitFormatEtc (fmt2, CF_HDROP, TYMED_HGLOBAL);
        if (SUCCEEDED(hr = pDataObject->GetData(&fmt2, &medium)) /* && SUCCEEDED(pDataObject->GetData(&fmt2, &medium))*/)
        {
            WCHAR wszTargetPath[MAX_PATH + 1];
            LPWSTR pszSrcList;
 
            wcscpy(wszTargetPath, m_sPathTarget);
            //Double NULL terminate.
            wszTargetPath[wcslen(wszTargetPath) + 1] = '\0';
 
            LPDROPFILES lpdf = (LPDROPFILES) GlobalLock(medium.hGlobal);
            if (!lpdf)
            {
                ERR("Error locking global\n");
                return E_FAIL;
            }
            pszSrcList = (LPWSTR) (((byte*) lpdf) + lpdf->pFiles);
            ERR("Source file (just the first) = %s, target path = %s, bCopy: %d\n", debugstr_w(pszSrcList), debugstr_w(wszTargetPath), bCopy);
 
            SHFILEOPSTRUCTW op;
            ZeroMemory(&op, sizeof(op));
            op.pFrom = pszSrcList;
            op.pTo = wszTargetPath;
            op.hwnd = m_hwndSite;
            op.wFunc = bCopy ? FO_COPY : FO_MOVE;
            op.fFlags = GetDefaultFileOpFlags(pDataObject);
            int res = SHFileOperationW(&op);
            if (res)
            {
                ERR("SHFileOperationW failed with 0x%x\n", res);
                hr = E_FAIL;
            }
 
            return hr;
        }
        ERR("Error calling GetData\n");
        hr = E_FAIL;
    }
    else
    {
        ERR("No viable drop format\n");
        hr = E_FAIL;
    }
    return hr;
}

Referenced by Drop().

_DoDropThreadProc()

DWORD WINAPI CFSDropTarget::_DoDropThreadProc ( LPVOID  lpParameter )

staticprivate

Definition at line 787 of file CFSDropTarget.cpp.

{
    _DoDropData *data = static_cast<_DoDropData*>(lpParameter);
    CComPtr<IDataObject> pDataObject;
    HRESULT hr = CoGetInterfaceAndReleaseStream (data->pStream, IID_PPV_ARG(IDataObject, &pDataObject));
 
    if (SUCCEEDED(hr))
    {
        CComPtr<IAsyncOperation> pAsyncOperation;
        hr = data->This->_DoDrop(pDataObject, data->dwKeyState, data->pt, &data->pdwEffect);
        if (SUCCEEDED(pDataObject->QueryInterface(IID_PPV_ARG(IAsyncOperation, &pAsyncOperation))))
        {
            pAsyncOperation->EndOperation(hr, NULL, data->pdwEffect);
        }
    }
    //Release the CFSFolder and data object holds in the copying thread.
    data->This->Release();
    //Release the parameter from the heap.
    HeapFree(GetProcessHeap(), 0, data);
    return 0;
}

Referenced by Drop().

_GetEffectFromMenu()

HRESULT CFSDropTarget::_GetEffectFromMenu ( IDataObject *  pDataObject, POINTL  pt, DWORD *  pdwEffect, DWORD  dwAvailableEffects  )

private

Definition at line 257 of file CFSDropTarget.cpp.

{
    HMENU hmenu = LoadMenuW(shell32_hInstance, MAKEINTRESOURCEW(IDM_DRAGFILE));
    if (!hmenu)
        return E_OUTOFMEMORY;
    HMENU hpopupmenu = GetSubMenu(hmenu, 0);
 
    SHELL_LimitDropEffectToItemAttributes(pDataObject, &dwAvailableEffects);
    if ((dwAvailableEffects & DROPEFFECT_COPY) == 0)
        DeleteMenu(hpopupmenu, IDM_COPYHERE, MF_BYCOMMAND);
    if ((dwAvailableEffects & DROPEFFECT_MOVE) == 0)
        DeleteMenu(hpopupmenu, IDM_MOVEHERE, MF_BYCOMMAND);
    if ((dwAvailableEffects & DROPEFFECT_LINK) == 0 && (dwAvailableEffects & (DROPEFFECT_COPY | DROPEFFECT_MOVE)))
        DeleteMenu(hpopupmenu, IDM_LINKHERE, MF_BYCOMMAND);
 
    GetDefaultCopyMoveEffect();
    if (*pdwEffect & dwAvailableEffects & DROPEFFECT_COPY)
        SetMenuDefaultItem(hpopupmenu, IDM_COPYHERE, FALSE);
    else if (*pdwEffect & dwAvailableEffects & DROPEFFECT_MOVE)
        SetMenuDefaultItem(hpopupmenu, IDM_MOVEHERE, FALSE);
    else if (dwAvailableEffects & DROPEFFECT_LINK)
        SetMenuDefaultItem(hpopupmenu, IDM_LINKHERE, FALSE);
 
    CRegKeyHandleArray keys;
    HDCIA hDCIA = DCIA_Create();
    HDCMA hDCMA = DCMA_Create();
    CComHeapPtr<ITEMIDLIST_ABSOLUTE> pidlFolder(SHELL32_CreateSimpleIDListFromPath(m_sPathTarget, FILE_ATTRIBUTE_DIRECTORY));
    if (hDCMA && hDCIA && pidlFolder)
    {
        AddPidlClassKeysToArray(pidlFolder, keys, keys);
        for (UINT i = 0; i < keys; ++i)
            DCIA_AddShellExSubkey(hDCIA, keys[i], L"DragDropHandlers");
 
        QCMINFO qcmi = { hpopupmenu, 0, DROPIDM_EXTFIRST, DROPIDM_EXTLAST };
        DCMA_InsertMenuItems(hDCMA, hDCIA, pidlFolder, pDataObject, keys, keys, &qcmi, 0, m_site);
    }
 
    UINT uCommand = TrackPopupMenu(hpopupmenu, pt);
 
    HRESULT hr = S_FALSE; // S_FALSE means we did not handle the command
    C_ASSERT(IDM_COPYHERE < DROPIDM_EXTFIRST && IDM_MOVEHERE < DROPIDM_EXTFIRST &&
             IDM_LINKHERE < DROPIDM_EXTFIRST && DROPIDM_EXTFIRST > 0);
    if (uCommand >= DROPIDM_EXTFIRST && uCommand <= DROPIDM_EXTLAST)
    {
        CMINVOKECOMMANDINFO ici = { sizeof(ici), 0, m_hwndSite, MAKEINTRESOURCEA(uCommand - DROPIDM_EXTFIRST) };
        ici.nShow = SW_SHOW;
        if (m_grfKeyState & MK_SHIFT)
            ici.fMask |= CMIC_MASK_SHIFT_DOWN;
        if (m_grfKeyState & MK_CONTROL)
            ici.fMask |= CMIC_MASK_CONTROL_DOWN;
        DCMA_InvokeCommand(hDCMA, &ici);
        hr = S_OK;
        *pdwEffect = DROPEFFECT_NONE;
    }
    else if (uCommand == 0)
    {
        hr = S_OK;
        *pdwEffect = DROPEFFECT_NONE;
    }
    else if (uCommand == IDM_COPYHERE)
        *pdwEffect = DROPEFFECT_COPY;
    else if (uCommand == IDM_MOVEHERE)
        *pdwEffect = DROPEFFECT_MOVE;
    else if (uCommand == IDM_LINKHERE)
        *pdwEffect = DROPEFFECT_LINK;
    else
        hr = E_UNEXPECTED;
 
    DCMA_Destroy(hDCMA);
    DCIA_Destroy(hDCIA);
    DestroyMenu(hmenu);
    return hr;
}

Referenced by Drop().

_GetUniqueFileName()

BOOL CFSDropTarget::_GetUniqueFileName ( LPCWSTR  pwszBasePath, LPCWSTR  pwszExt, LPWSTR  pwszTarget, BOOL  bShortcut  )

private

Definition at line 185 of file CFSDropTarget.cpp.

{
    WCHAR wszLink[40];
 
    if (!bShortcut)
    {
        if (!LoadStringW(shell32_hInstance, IDS_LNK_FILE, wszLink, _countof(wszLink)))
            wszLink[0] = L'\0';
    }
 
    if (!bShortcut)
        _swprintf(pwszTarget, L"%s%s%s", wszLink, pwszBasePath, pwszExt);
    else
        _swprintf(pwszTarget, L"%s%s", pwszBasePath, pwszExt);
 
    for (UINT i = 2; PathFileExistsW(pwszTarget); ++i)
    {
        if (!bShortcut)
            _swprintf(pwszTarget, L"%s%s (%u)%s", wszLink, pwszBasePath, i, pwszExt);
        else
            _swprintf(pwszTarget, L"%s (%u)%s", pwszBasePath, i, pwszExt);
    }
 
    return TRUE;
}

Referenced by _DoDrop().

_QueryDrop()

BOOL CFSDropTarget::_QueryDrop ( DWORD  dwKeyState, LPDWORD  pdwEffect  )

private

Definition at line 214 of file CFSDropTarget.cpp.

{
    /* TODO Windows does different drop effects if dragging across drives.
    i.e., it will copy instead of move if the directories are on different disks. */
    GetDefaultCopyMoveEffect();
 
    DWORD dwEffect = m_dwDefaultEffect;
 
    *pdwEffect = DROPEFFECT_NONE;
 
    if (m_fAcceptFmt) { /* Does our interpretation of the keystate ... */
        *pdwEffect = KeyStateToDropEffect(dwKeyState);
 
        // Transform disallowed move to a copy
        if ((*pdwEffect & D_MOVE) && (m_AllowedEffects & (D_MOVE | D_COPY)) == D_COPY)
            *pdwEffect = D_COPY;
 
        *pdwEffect &= m_AllowedEffects;
 
        if (*pdwEffect == DROPEFFECT_NONE)
            *pdwEffect = dwEffect;
 
        /* ... matches the desired effect ? */
        if (dwEffect & *pdwEffect) {
            return TRUE;
        }
    }
    return FALSE;
}

Referenced by DragOver(), and Drop().

_RepositionItems()

HRESULT CFSDropTarget::_RepositionItems ( IShellFolderView *  psfv, IDataObject *  pDataObject, POINTL  pt  )

private

Definition at line 331 of file CFSDropTarget.cpp.

{
    CComPtr<IFolderView> pfv;
    POINT ptDrag;
    HRESULT hr = psfv->QueryInterface(IID_PPV_ARG(IFolderView, &pfv));
    if (FAILED_UNEXPECTEDLY(hr))
        return hr;
 
    hr = psfv->GetDragPoint(&ptDrag);
    if (FAILED_UNEXPECTEDLY(hr))
        return hr;
 
    PIDLIST_ABSOLUTE pidlFolder;
    PUITEMID_CHILD *apidl;
    UINT cidl;
    hr = SH_GetApidlFromDataObject(pdtobj, &pidlFolder, &apidl, &cidl);
    if (FAILED_UNEXPECTEDLY(hr))
        return hr;
 
    CComHeapPtr<POINT> apt;
    if (!apt.Allocate(cidl))
    {
        SHFree(pidlFolder);
        _ILFreeaPidl(apidl, cidl);
        return E_OUTOFMEMORY;
    }
 
    for (UINT i = 0; i<cidl; i++)
    {
        pfv->GetItemPosition(apidl[i], &apt[i]);
        apt[i].x += pt.x - ptDrag.x;
        apt[i].y += pt.y - ptDrag.y;
    }
 
    pfv->SelectAndPositionItems(cidl, apidl, apt, SVSI_SELECT);
 
    SHFree(pidlFolder);
    _ILFreeaPidl(apidl, cidl);
    return S_OK;
}

Referenced by Drop().

DragEnter()

HRESULT WINAPI CFSDropTarget::DragEnter ( IDataObject *  pDataObject, DWORD  dwKeyState, POINTL  pt, DWORD *  pdwEffect  )

override

Implements IDropTarget.

Definition at line 372 of file CFSDropTarget.cpp.

{
    TRACE("(%p)->(DataObject=%p)\n", this, pDataObject);
 
    const BOOL bAnyKeyMod = dwKeyState & (MK_SHIFT | MK_CONTROL);
    m_AllowedEffects = *pdwEffect;
 
    if (*pdwEffect == DROPEFFECT_NONE)
        return S_OK;
 
    FORMATETC fmt;
    FORMATETC fmt2;
    m_fAcceptFmt = FALSE;
 
    InitFormatEtc (fmt, m_cfShellIDList, TYMED_HGLOBAL);
    InitFormatEtc (fmt2, CF_HDROP, TYMED_HGLOBAL);
 
    if (SUCCEEDED(pDataObject->QueryGetData(&fmt)))
        m_fAcceptFmt = TRUE;
    else if (SUCCEEDED(pDataObject->QueryGetData(&fmt2)))
        m_fAcceptFmt = TRUE;
 
    m_grfKeyState = dwKeyState;
 
    SHELL_LimitDropEffectToItemAttributes(pDataObject, pdwEffect);
    m_AllowedEffects = *pdwEffect;
    m_dwDefaultEffect = m_AllowedEffects;
    switch (*pdwEffect & (D_COPY | D_MOVE | D_LINK))
    {
        case D_COPY | D_MOVE:
            if (dwKeyState & MK_CONTROL)
                m_dwDefaultEffect = D_COPY;
            else
                m_dwDefaultEffect = D_MOVE;
            break;
        case D_COPY | D_MOVE | D_LINK:
            if ((dwKeyState & (MK_SHIFT | MK_CONTROL)) == (MK_SHIFT | MK_CONTROL))
                m_dwDefaultEffect = D_LINK;
            else if ((dwKeyState & (MK_SHIFT | MK_CONTROL)) == MK_CONTROL)
                m_dwDefaultEffect = D_COPY;
            else
                m_dwDefaultEffect = D_MOVE;
            break;
        case D_COPY | D_LINK:
            if ((dwKeyState & (MK_SHIFT | MK_CONTROL)) == (MK_SHIFT | MK_CONTROL))
                m_dwDefaultEffect = D_LINK;
            else
                m_dwDefaultEffect = D_COPY;
            break;
        case D_MOVE | D_LINK:
            if ((dwKeyState & (MK_SHIFT | MK_CONTROL)) == (MK_SHIFT | MK_CONTROL))
                m_dwDefaultEffect = D_LINK;
            else
                m_dwDefaultEffect = D_MOVE;
            break;
    }
 
    STGMEDIUM medium;
    if (SUCCEEDED(pDataObject->GetData(&fmt2, &medium)))
    {
        WCHAR wstrFirstFile[MAX_PATH];
        if (DragQueryFileW((HDROP)medium.hGlobal, 0, wstrFirstFile, _countof(wstrFirstFile)))
        {
            if (!PathIsSameDrive(wstrFirstFile, m_sPathTarget) && m_dwDefaultEffect != D_LINK)
            {
                m_dwDefaultEffect = DROPEFFECT_COPY;
            }
 
            if (!bAnyKeyMod && PathIsDriveRoot(wstrFirstFile) && (m_AllowedEffects & DROPEFFECT_LINK))
            {
                m_dwDefaultEffect = DROPEFFECT_LINK; // Don't copy a drive by default
            }
        }
        ReleaseStgMedium(&medium);
    }
 
    if (!m_fAcceptFmt)
        m_AllowedEffects = DROPEFFECT_NONE;
    else
        *pdwEffect = m_dwDefaultEffect;
    *pdwEffect &= m_AllowedEffects;
 
    return S_OK;
}

DragLeave()

HRESULT WINAPI CFSDropTarget::DragLeave ( )

override

Implements IDropTarget.

Definition at line 473 of file CFSDropTarget.cpp.

{
    TRACE("(%p)\n", this);
 
    m_fAcceptFmt = FALSE;
 
    return S_OK;
}

DragOver()

HRESULT WINAPI CFSDropTarget::DragOver ( DWORD  dwKeyState, POINTL  pt, DWORD *  pdwEffect  )

override

Implements IDropTarget.

Definition at line 458 of file CFSDropTarget.cpp.

{
    TRACE("(%p)\n", this);
 
    if (!pdwEffect)
        return E_INVALIDARG;
 
    m_grfKeyState = dwKeyState;
 
    _QueryDrop(dwKeyState, pdwEffect);
 
    return S_OK;
}

Drop()

HRESULT WINAPI CFSDropTarget::Drop ( IDataObject *  pDataObject, DWORD  dwKeyState, POINTL  pt, DWORD *  pdwEffect  )

override

Implements IDropTarget.

Definition at line 482 of file CFSDropTarget.cpp.

{
    TRACE("(%p) object dropped, effect %u\n", this, *pdwEffect);
 
    if (!pdwEffect)
        return E_INVALIDARG;
 
    IUnknown_GetWindow(m_site, &m_hwndSite);
 
    DWORD dwAvailableEffects = *pdwEffect;
 
    _QueryDrop(dwKeyState, pdwEffect);
 
    TRACE("pdwEffect: 0x%x, m_dwDefaultEffect: 0x%x, dwAvailableEffects: 0x%x\n", *pdwEffect, m_dwDefaultEffect, dwAvailableEffects);
 
    if (m_grfKeyState & MK_RBUTTON)
    {
        HRESULT hr = _GetEffectFromMenu(pDataObject, pt, pdwEffect, dwAvailableEffects);
        if (FAILED_UNEXPECTEDLY(hr) || hr == S_OK)
            return hr;
    }
 
    if (*pdwEffect == DROPEFFECT_MOVE && m_site)
    {
        CComPtr<IShellFolderView> psfv;
        HRESULT hr = IUnknown_QueryService(m_site, SID_SFolderView, IID_PPV_ARG(IShellFolderView, &psfv));
        if (SUCCEEDED(hr) && psfv->IsDropOnSource(this) == S_OK)
        {
            _RepositionItems(psfv, pDataObject, pt);
            return S_OK;
        }
    }
 
    BOOL fIsOpAsync = FALSE;
    CComPtr<IAsyncOperation> pAsyncOperation;
 
    if (SUCCEEDED(pDataObject->QueryInterface(IID_PPV_ARG(IAsyncOperation, &pAsyncOperation))))
    {
        if (SUCCEEDED(pAsyncOperation->GetAsyncMode(&fIsOpAsync)) && fIsOpAsync)
        {
            _DoDropData *data = static_cast<_DoDropData*>(HeapAlloc(GetProcessHeap(), 0, sizeof(_DoDropData)));
            data->This = this;
            // Need to maintain this class in case the window is closed or the class exists temporarily (when dropping onto a folder).
            pDataObject->AddRef();
            pAsyncOperation->StartOperation(NULL);
            CoMarshalInterThreadInterfaceInStream(IID_IDataObject, pDataObject, &data->pStream);
            this->AddRef();
            data->dwKeyState = dwKeyState;
            data->pt = pt;
            // Need to dereference as pdweffect gets freed.
            data->pdwEffect = *pdwEffect;
            SHCreateThread(CFSDropTarget::_DoDropThreadProc, data, CTF_COINIT | CTF_PROCESS_REF, NULL);
            return S_OK;
        }
    }
    return this->_DoDrop(pDataObject, dwKeyState, pt, pdwEffect);
}

GetSite()

HRESULT WINAPI CFSDropTarget::GetSite ( REFIID  riid, void **  ppvSite  )

override

Definition at line 551 of file CFSDropTarget.cpp.

{
    if (!m_site)
        return E_FAIL;
 
    return m_site->QueryInterface(riid, ppvSite);
}

Initialize()

HRESULT CFSDropTarget::Initialize

(

LPWSTR

PathTarget

)

Definition at line 167 of file CFSDropTarget.cpp.

{
    if (!PathTarget)
        return E_UNEXPECTED;
 
    m_cfShellIDList = RegisterClipboardFormatW(CFSTR_SHELLIDLIST);
    if (!m_cfShellIDList)
        return E_FAIL;
 
    return SHStrDupW(PathTarget, &m_sPathTarget);
}

SetSite()

HRESULT WINAPI CFSDropTarget::SetSite ( IUnknown *  pUnkSite )

override

Implements IObjectWithSite.

Definition at line 543 of file CFSDropTarget.cpp.

{
    m_site = pUnkSite;
    return S_OK;
}

TrackPopupMenu()

UINT CFSDropTarget::TrackPopupMenu ( HMENU  hMenu, const POINTL &  pt  )

static

Definition at line 244 of file CFSDropTarget.cpp.

{
    /* We shouldn't use the site window here because the menu should work even when we don't have a site */
    HWND hwndDummy = CreateWindowEx(0, WC_STATIC, NULL, WS_DISABLED | WS_CLIPSIBLINGS | WS_BORDER,
                                    pt.x, pt.y, 1, 1, NULL, NULL, NULL, NULL);
 
    SetForegroundWindow(hwndDummy); // Required for aborting by pressing Esc when dragging from Explorer to desktop
    UINT uCommand = ::TrackPopupMenu(hMenu, TPM_RETURNCMD | TPM_LEFTBUTTON | TPM_RIGHTBUTTON,
                                     pt.x, pt.y, 0, hwndDummy, NULL);
    DestroyWindow(hwndDummy);
    return uCommand;
}

Referenced by _GetEffectFromMenu(), CExeDropHandler::Drop(), and TrackPopupMenu().

Member Data Documentation

m_AllowedEffects

DWORD CFSDropTarget::m_AllowedEffects

private

Definition at line 41 of file CFSDropTarget.h.

Referenced by _QueryDrop(), and DragEnter().

m_cfShellIDList

UINT CFSDropTarget::m_cfShellIDList

private

Definition at line 35 of file CFSDropTarget.h.

Referenced by _DoDrop(), DragEnter(), and Initialize().

m_dwDefaultEffect

DWORD CFSDropTarget::m_dwDefaultEffect

private

Definition at line 40 of file CFSDropTarget.h.

Referenced by _QueryDrop(), DragEnter(), and Drop().

m_fAcceptFmt

BOOL CFSDropTarget::m_fAcceptFmt

private

Definition at line 36 of file CFSDropTarget.h.

Referenced by _QueryDrop(), DragEnter(), and DragLeave().

m_grfKeyState

DWORD CFSDropTarget::m_grfKeyState

private

Definition at line 39 of file CFSDropTarget.h.

Referenced by _GetEffectFromMenu(), DragEnter(), DragOver(), and Drop().

m_hwndSite

HWND CFSDropTarget::m_hwndSite

private

Definition at line 38 of file CFSDropTarget.h.

Referenced by _CopyItems(), _DoDrop(), _GetEffectFromMenu(), and Drop().

m_site

CComPtr<IUnknown> CFSDropTarget::m_site

private

Definition at line 42 of file CFSDropTarget.h.

Referenced by _GetEffectFromMenu(), Drop(), GetSite(), and SetSite().

m_sPathTarget

LPWSTR CFSDropTarget::m_sPathTarget

private

Definition at line 37 of file CFSDropTarget.h.

Referenced by _CopyItems(), _DoDrop(), _GetEffectFromMenu(), DragEnter(), Initialize(), and ~CFSDropTarget().

---

The documentation for this class was generated from the following files:

• dll/win32/shell32/droptargets/CFSDropTarget.h
• dll/win32/shell32/droptargets/CFSDropTarget.cpp