CDrivesFolder Class Reference

#include <CDrivesFolder.h>

Inheritance diagram for CDrivesFolder:

Collaboration diagram for CDrivesFolder:

Public Member Functions
	CDrivesFolder ()
	~CDrivesFolder ()
HRESULT WINAPI	FinalConstruct ()
STDMETHOD()	ParseDisplayName (HWND hwndOwner, LPBC pbc, LPOLESTR lpszDisplayName, DWORD *pchEaten, PIDLIST_RELATIVE *ppidl, DWORD *pdwAttributes) override
STDMETHOD()	EnumObjects (HWND hwndOwner, DWORD dwFlags, LPENUMIDLIST *ppEnumIDList) override
STDMETHOD()	BindToObject (PCUIDLIST_RELATIVE pidl, LPBC pbcReserved, REFIID riid, LPVOID *ppvOut) override
STDMETHOD()	BindToStorage (PCUIDLIST_RELATIVE pidl, LPBC pbcReserved, REFIID riid, LPVOID *ppvOut) override
STDMETHOD()	CompareIDs (LPARAM lParam, PCUIDLIST_RELATIVE pidl1, PCUIDLIST_RELATIVE pidl2) override
STDMETHOD()	CreateViewObject (HWND hwndOwner, REFIID riid, LPVOID *ppvOut) override
STDMETHOD()	GetAttributesOf (UINT cidl, PCUITEMID_CHILD_ARRAY apidl, DWORD *rgfInOut) override
STDMETHOD()	GetUIObjectOf (HWND hwndOwner, UINT cidl, PCUITEMID_CHILD_ARRAY apidl, REFIID riid, UINT *prgfInOut, LPVOID *ppvOut) override
STDMETHOD()	GetDisplayNameOf (PCUITEMID_CHILD pidl, DWORD dwFlags, LPSTRRET strRet) override
STDMETHOD()	SetNameOf (HWND hwndOwner, PCUITEMID_CHILD pidl, LPCOLESTR lpName, DWORD dwFlags, PITEMID_CHILD *pPidlOut) override
STDMETHOD()	GetDefaultSearchGUID (GUID *pguid) override
STDMETHOD()	EnumSearches (IEnumExtraSearch **ppenum) override
STDMETHOD()	GetDefaultColumn (DWORD dwRes, ULONG *pSort, ULONG *pDisplay) override
STDMETHOD()	GetDefaultColumnState (UINT iColumn, DWORD *pcsFlags) override
STDMETHOD()	GetDetailsEx (PCUITEMID_CHILD pidl, const SHCOLUMNID *pscid, VARIANT *pv) override
STDMETHOD()	GetDetailsOf (PCUITEMID_CHILD pidl, UINT iColumn, SHELLDETAILS *psd) override
STDMETHOD()	MapColumnToSCID (UINT column, SHCOLUMNID *pscid) override
STDMETHOD()	GetClassID (CLSID *lpClassId) override
STDMETHOD()	Initialize (PCIDLIST_ABSOLUTE pidl) override
STDMETHOD()	GetCurFolder (PIDLIST_ABSOLUTE *pidl) override
STDMETHOD()	CallBack (IShellFolder *psf, HWND hwndOwner, IDataObject *pdtobj, UINT uMsg, WPARAM wParam, LPARAM lParam) override
STDMETHODIMP	MessageSFVCB (UINT uMsg, WPARAM wParam, LPARAM lParam) override
STDMETHOD()	ShouldShow (IShellFolder *psf, PCIDLIST_ABSOLUTE pidlFolder, PCUITEMID_CHILD pidlItem) override
STDMETHODIMP	GetEnumFlags (IShellFolder *, PCIDLIST_ABSOLUTE, HWND *, DWORD *) 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	GetDefaultSearchGUID ([out] GUID *lpguid)
HRESULT	EnumSearches ([out] IEnumExtraSearch **ppenum)
HRESULT	GetDefaultColumn ([in] DWORD dwReserved, [out] ULONG *pSort, [out] ULONG *pDisplay)
HRESULT	GetDefaultColumnState ([in] UINT iColumn, [out] SHCOLSTATEF *pcsFlags)
HRESULT	GetDetailsEx ([in] PCUITEMID_CHILD pidl, [in] const SHCOLUMNID *pscid, [out] VARIANT *pv)
HRESULT	GetDetailsOf ([in] PCUITEMID_CHILD pidl, [in] UINT iColumn, [out] SHELLDETAILS *psd)
HRESULT	MapColumnToSCID ([in] UINT iColumn, [in] SHCOLUMNID *pscid)
Public Member Functions inherited from IShellFolder
HRESULT	ParseDisplayName ([in] HWND hwndOwner, [in] LPBC pbcReserved, [in, string] LPOLESTR lpszDisplayName, [out] ULONG *pchEaten, [out] PIDLIST_RELATIVE *ppidl, [in, out, unique] ULONG *pdwAttributes)
HRESULT	EnumObjects ([in] HWND hwndOwner, [in] SHCONTF grfFlags, [out] IEnumIDList **ppenumIDList)
HRESULT	BindToObject ([in] PCUIDLIST_RELATIVE pidl, [in] LPBC pbcReserved, [in] REFIID riid, [out, iid_is(riid)] void **ppvOut)
HRESULT	BindToStorage ([in] PCUIDLIST_RELATIVE pidl, [in] LPBC pbcReserved, [in] REFIID riid, [out, iid_is(riid)] void **ppvObj)
HRESULT	CompareIDs ([in] LPARAM lParam, [in] PCUIDLIST_RELATIVE pidl1, [in] PCUIDLIST_RELATIVE pidl2)
HRESULT	CreateViewObject ([in] HWND hwndOwner, [in] REFIID riid, [out, iid_is(riid)] void **ppvOut)
HRESULT	GetAttributesOf ([in] UINT cidl, [in, size_is(cidl)] PCUITEMID_CHILD_ARRAY apidl, [in, out] SFGAOF *rgfInOut)
HRESULT	GetUIObjectOf ([in] HWND hwndOwner, [in] UINT cidl, [in, size_is(cidl)] PCUITEMID_CHILD_ARRAY apidl, [in] REFIID riid, [in, out, unique] UINT *prgfInOut, [out, iid_is(riid)] void **ppvOut)
HRESULT	GetDisplayNameOf ([in] PCUITEMID_CHILD pidl, [in] SHGDNF uFlags, [out] STRRET *lpName)
HRESULT	SetNameOf ([in] HWND hwnd, [in] PCUITEMID_CHILD pidl, [in, string] LPCOLESTR lpszName, [in] SHGDNF uFlags, [out] PITEMID_CHILD *ppidlOut)
Public Member Functions inherited from IUnknown
HRESULT	QueryInterface ([in] REFIID riid, [out, iid_is(riid)] void **ppvObject)
ULONG	AddRef ()
ULONG	Release ()
HRESULT	GetCurFolder ([out] PIDLIST_ABSOLUTE *ppidl)
HRESULT	Initialize ([in] PCIDLIST_ABSOLUTE pidl)
HRESULT	GetClassID ([out] CLSID *pClassID)
HRESULT	CallBack ([in] IShellFolder *psf, [in] HWND hwndOwner, [in] IDataObject *pdtobj, [in] UINT uMsg, [in] WPARAM wParam, [in] LPARAM lParam)
HRESULT	ShouldShow ([in] IShellFolder *psf, [in, unique] PCIDLIST_ABSOLUTE pidlFolder, [in] PCUITEMID_CHILD pidlItem)
HRESULT	GetEnumFlags ([in] IShellFolder *psf, [in] PCIDLIST_ABSOLUTE pidlFolder, [out] HWND *phwnd, [in, out] DWORD *pgrfFlags)

Static Public Member Functions
static HRESULT	SetDriveLabel (HWND hwndOwner, PCWSTR DrivePath, PCWSTR Label)
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 Attributes
LPITEMIDLIST	pidlRoot
CComPtr< IShellFolder2 >	m_regFolder
INT8	m_DriveDisplayMode

Additional Inherited Members
Public Types inherited from IShellFolder2
typedef struct IShellFolder2::SHCOLUMNID *	LPSHCOLUMNID
typedef const SHCOLUMNID *	LPCSHCOLUMNID
Public Types inherited from IShellFolder
enum	tagSHGDN {   SHGDN_NORMAL = 0 , SHGDN_INFOLDER = 1 , SHGDN_FOREDITING = 0x1000 , SHGDN_INCLUDE_NONFILESYS = 0x2000 ,   SHGDN_FORADDRESSBAR = 0x4000 , SHGDN_FORPARSING = 0x8000 }
enum	tagSHCONTF {   SHCONTF_FOLDERS = 32 , SHCONTF_NONFOLDERS = 64 , SHCONTF_INCLUDEHIDDEN = 128 , SHCONTF_INIT_ON_FIRST_NEXT = 256 ,   SHCONTF_NETPRINTERSRCH = 512 , SHCONTF_SHAREABLE = 1024 , SHCONTF_STORAGE = 2048 , SHCONTF_NAVIGATION_ENUM = 0x01000 ,   SHCONTF_FASTITEMS = 0x02000 , SHCONTF_FLATLIST = 0x04000 , SHCONTF_ENABLE_ASYNC = 0x08000 , SHCONTF_INCLUDESUPERHIDDEN = 0x10000 }
typedef IShellFolder *	LPSHELLFOLDER
typedef enum IShellFolder::tagSHGDN	SHGNO
typedef DWORD	SHGDNF
typedef DWORD	SHCONTF
typedef ULONG	SFGAOF
Public Types inherited from IUnknown
typedef IUnknown *	LPUNKNOWN
Public Types inherited from IPersistFolder2
typedef IPersistFolder2 *	LPPERSISTFOLDER2
Public Types inherited from IPersistFolder
typedef IPersistFolder *	LPPERSISTFOLDER
Public Types inherited from IPersist
typedef IPersist *	LPPERSIST
Public Attributes inherited from ATL::CComObjectRootBase
LONG	m_dwRef

Detailed Description

Definition at line 26 of file CDrivesFolder.h.

Constructor & Destructor Documentation

CDrivesFolder()

CDrivesFolder::CDrivesFolder

(

)

Definition at line 689 of file CDrivesFolder.cpp.

{
    pidlRoot = NULL;
    m_DriveDisplayMode = -1;
}

~CDrivesFolder()

CDrivesFolder::~CDrivesFolder

(

)

Definition at line 695 of file CDrivesFolder.cpp.

{
    TRACE("-- destroying IShellFolder(%p)\n", this);
    SHFree(pidlRoot);
}

Member Function Documentation

BindToObject()

HRESULT WINAPI CDrivesFolder::BindToObject ( PCUIDLIST_RELATIVE  pidl, LPBC  pbcReserved, REFIID  riid, LPVOID *  ppvOut  )

override

Definition at line 807 of file CDrivesFolder.cpp.

{
    TRACE("(%p)->(pidl=%p,%p,%s,%p)\n", this,
          pidl, pbcReserved, shdebugstr_guid(&riid), ppvOut);
 
    if (!pidl)
        return E_INVALIDARG;
 
    if (_ILIsSpecialFolder(pidl))
        return m_regFolder->BindToObject(pidl, pbcReserved, riid, ppvOut);
 
    CHAR* pchDrive = _ILGetDataPointer(pidl)->u.drive.szDriveName;
 
    PERSIST_FOLDER_TARGET_INFO pfti = {0};
    pfti.dwAttributes = -1;
    pfti.csidl = -1;
    pfti.szTargetParsingName[0] = *pchDrive;
    pfti.szTargetParsingName[1] = L':';
    pfti.szTargetParsingName[2] = L'\\';
 
    HRESULT hr = SHELL32_BindToSF(pidlRoot,
                                  &pfti,
                                  pidl,
                                  &CLSID_ShellFSFolder,
                                  riid,
                                  ppvOut);
    if (FAILED_UNEXPECTEDLY(hr))
        return hr;
 
    return S_OK;
}

Referenced by GetUIObjectOf(), and ParseDisplayName().

BindToStorage()

HRESULT WINAPI CDrivesFolder::BindToStorage ( PCUIDLIST_RELATIVE  pidl, LPBC  pbcReserved, REFIID  riid, LPVOID *  ppvOut  )

override

Definition at line 842 of file CDrivesFolder.cpp.

{
    FIXME("(%p)->(pidl=%p,%p,%s,%p) stub\n", this,
          pidl, pbcReserved, shdebugstr_guid (&riid), ppvOut);
 
    *ppvOut = NULL;
    return E_NOTIMPL;
}

CallBack()

HRESULT WINAPI CDrivesFolder::CallBack ( IShellFolder *  psf, HWND  hwndOwner, IDataObject *  pdtobj, UINT  uMsg, WPARAM  wParam, LPARAM  lParam  )

override

Implements IContextMenuCB.

Definition at line 1444 of file CDrivesFolder.cpp.

{
    enum { IDC_PROPERTIES };
    /* no data object means no selection */
    if (!pdtobj)
    {
        if (uMsg == DFM_INVOKECOMMAND && wParam == IDC_PROPERTIES)
        {
            // "System" properties
            return SHELL_ExecuteControlPanelCPL(hwndOwner, L"sysdm.cpl") ? S_OK : E_FAIL;
        }
        else if (uMsg == DFM_MERGECONTEXTMENU) // TODO: DFM_MERGECONTEXTMENU_BOTTOM
        {
            QCMINFO *pqcminfo = (QCMINFO *)lParam;
            HMENU hpopup = CreatePopupMenu();
            _InsertMenuItemW(hpopup, 0, TRUE, IDC_PROPERTIES, MFT_STRING, MAKEINTRESOURCEW(IDS_PROPERTIES), MFS_ENABLED);
            pqcminfo->idCmdFirst = Shell_MergeMenus(pqcminfo->hmenu, hpopup, pqcminfo->indexMenu, pqcminfo->idCmdFirst, pqcminfo->idCmdLast, MM_ADDSEPARATOR);
            DestroyMenu(hpopup);
            return S_OK;
        }
    }
    return SHELL32_DefaultContextMenuCallBack(psf, pdtobj, uMsg);
}

CompareIDs()

HRESULT WINAPI CDrivesFolder::CompareIDs ( LPARAM  lParam, PCUIDLIST_RELATIVE  pidl1, PCUIDLIST_RELATIVE  pidl2  )

override

Implements IShellFolder.

Definition at line 854 of file CDrivesFolder.cpp.

{
    HRESULT hres;
    UINT iColumn = LOWORD(lParam);
 
    if (iColumn >= _countof(MyComputerSFHeader))
        return E_INVALIDARG;
 
    if (!pidl1 || !pidl2)
    {
        ERR("Got null pidl pointer (%Ix %p %p)!\n", lParam, pidl1, pidl2);
        return E_INVALIDARG;
    }
 
    CHAR* pszDrive1 = _ILIsDrive(pidl1) ? _ILGetDataPointer(pidl1)->u.drive.szDriveName : NULL;
    if (!pszDrive1 && !IsRegItem(pidl1) && FAILED_UNEXPECTEDLY(E_INVALIDARG))
        return E_INVALIDARG;
    CHAR* pszDrive2 = _ILIsDrive(pidl2) ? _ILGetDataPointer(pidl2)->u.drive.szDriveName : NULL;
    if (!pszDrive2 && !IsRegItem(pidl2) && FAILED_UNEXPECTEDLY(E_INVALIDARG))
        return E_INVALIDARG;
 
    int result;
    switch (MyComputerSFHeader[iColumn].colnameid)
    {
        case IDS_SHV_COLUMN_NAME:
        {
            if (!pszDrive1 && !pszDrive2)
                return SHELL32_CompareDetails(this, lParam, pidl1, pidl2);
            else if (pszDrive1 && pszDrive2)
                result = _stricmp(pszDrive1, pszDrive2);
            else
                result = (int)!pszDrive1 - (int)!pszDrive2; // Sort drives first
            hres = MAKE_COMPARE_HRESULT(result);
            break;
        }
 
        case IDS_SHV_COLUMN_DISK_CAPACITY:
        case IDS_SHV_COLUMN_DISK_AVAILABLE:
        {
            if (!pszDrive1 || !pszDrive2)
            {
                hres = MAKE_COMPARE_HRESULT((int)!pszDrive1 - (int)!pszDrive2);
                break;
            }
 
            ULARGE_INTEGER Drive1Available, Drive1Total, Drive2Available, Drive2Total;
            BOOL bValid1 = FALSE, bValid2 = FALSE;
 
            if (GetVolumeInformationA(pszDrive1, NULL, 0, NULL, NULL, NULL, NULL, 0))
                bValid1 = GetDiskFreeSpaceExA(pszDrive1, &Drive1Available, &Drive1Total, NULL);
            else
                Drive1Available.QuadPart = Drive1Total.QuadPart = 0;
 
            if (GetVolumeInformationA(pszDrive2, NULL, 0, NULL, NULL, NULL, NULL, 0))
                bValid2 = GetDiskFreeSpaceExA(pszDrive2, &Drive2Available, &Drive2Total, NULL);
            else
                Drive2Available.QuadPart = Drive2Total.QuadPart = 0;
 
            LARGE_INTEGER Diff;
            if (MyComputerSFHeader[iColumn].colnameid == IDS_SHV_COLUMN_DISK_CAPACITY) /* Size */
                Diff.QuadPart = Drive1Total.QuadPart - Drive2Total.QuadPart;
            else /* Size available */
                Diff.QuadPart = Drive1Available.QuadPart - Drive2Available.QuadPart;
 
            if (bValid1 != bValid2)
                hres = MAKE_COMPARE_HRESULT((int)!bValid1 - (int)!bValid2);
            else
                hres = MAKE_COMPARE_HRESULT(Diff.QuadPart);
 
            break;
        }
        case IDS_SHV_COLUMN_TYPE:
        case IDS_SHV_COLUMN_COMMENTS:
        {
            /* We want to return immediately because SHELL32_CompareDetails also compares children. */
            return SHELL32_CompareDetails(this, lParam, pidl1, pidl2);
        }
        DEFAULT_UNREACHABLE;
    }
 
    if (HRESULT_CODE(hres) == 0)
        return SHELL32_CompareChildren(this, lParam, pidl1, pidl2);
 
    return hres;
}

CreateViewObject()

HRESULT WINAPI CDrivesFolder::CreateViewObject ( HWND  hwndOwner, REFIID  riid, LPVOID *  ppvOut  )

override

Definition at line 943 of file CDrivesFolder.cpp.

{
    CComPtr<IShellView> pShellView;
    HRESULT hr = E_INVALIDARG;
 
    TRACE("(%p)->(hwnd=%p,%s,%p)\n", this,
          hwndOwner, shdebugstr_guid (&riid), ppvOut);
 
    if (!ppvOut)
        return hr;
 
    *ppvOut = NULL;
 
    if (IsEqualIID(riid, IID_IDropTarget))
    {
        WARN("IDropTarget not implemented\n");
        hr = E_NOTIMPL;
    }
    else if (IsEqualIID(riid, IID_IContextMenu))
    {
        DEFCONTEXTMENU dcm = { hwndOwner, this, pidlRoot, this };
        hr = SHCreateDefaultContextMenu(&dcm, riid, ppvOut);
    }
    else if (IsEqualIID(riid, IID_IShellView))
    {
        SFV_CREATE sfvparams = { sizeof(SFV_CREATE), this, NULL, this };
        hr = SHCreateShellFolderView(&sfvparams, (IShellView**)ppvOut);
    }
    TRACE("-- (%p)->(interface=%p)\n", this, ppvOut);
    return hr;
}

EnumObjects()

HRESULT WINAPI CDrivesFolder::EnumObjects ( HWND  hwndOwner, DWORD  dwFlags, LPENUMIDLIST *  ppEnumIDList  )

override

Definition at line 796 of file CDrivesFolder.cpp.

{
    CComPtr<IEnumIDList> pRegEnumerator;
    m_regFolder->EnumObjects(hwndOwner, dwFlags, &pRegEnumerator);
 
    return ShellObjectCreatorInit<CDrivesFolderEnum>(hwndOwner, dwFlags, pRegEnumerator, IID_PPV_ARG(IEnumIDList, ppEnumIDList));
}

EnumSearches()

HRESULT WINAPI CDrivesFolder::EnumSearches ( IEnumExtraSearch **  ppenum )

override

Implements IShellFolder2.

Definition at line 1225 of file CDrivesFolder.cpp.

{
    *ppenum = NULL;
    return E_NOTIMPL;
}

FinalConstruct()

HRESULT WINAPI CDrivesFolder::FinalConstruct

(

)

Definition at line 701 of file CDrivesFolder.cpp.

{
    pidlRoot = _ILCreateMyComputer();    /* my qualified pidl */
    if (pidlRoot == NULL)
        return E_OUTOFMEMORY;
 
    REGFOLDERINITDATA RegInit = { static_cast<IShellFolder*>(this), &g_RegFolderInfo };
    HRESULT hr = CRegFolder_CreateInstance(&RegInit,
                                           pidlRoot,
                                           IID_PPV_ARG(IShellFolder2, &m_regFolder));
 
    return hr;
}

GetAttributesOf()

HRESULT WINAPI CDrivesFolder::GetAttributesOf ( UINT  cidl, PCUITEMID_CHILD_ARRAY  apidl, DWORD *  rgfInOut  )

override

Definition at line 978 of file CDrivesFolder.cpp.

{
    TRACE("(%p)->(cidl=%d apidl=%p mask=%p (0x%08x))\n",
          this, cidl, apidl, rgfInOut, rgfInOut ? *rgfInOut : 0);
 
    if (cidl && !apidl)
        return E_INVALIDARG;
 
    if (*rgfInOut == 0)
        *rgfInOut = ~0;
 
    if(cidl == 0)
        *rgfInOut &= dwComputerAttributes;
    else
    {
        for (UINT i = 0; i < cidl; ++i)
        {
            if (_ILIsDrive(apidl[i]))
            {
                *rgfInOut &= dwDriveAttributes;
 
                if (_ILGetDriveType(apidl[i]) == DRIVE_CDROM)
                    *rgfInOut &= ~SFGAO_CANRENAME; // CD-ROM drive cannot rename
            }
            else if (IsRegItem(apidl[i], CLSID_ControlPanel))
            {
                *rgfInOut &= dwControlPanelAttributes;
            }
            else if (_ILIsSpecialFolder(*apidl))
            {
                m_regFolder->GetAttributesOf(1, &apidl[i], rgfInOut);
            }
            else
            {
                ERR("Got unknown pidl type!\n");
            }
        }
    }
 
    /* make sure SFGAO_VALIDATE is cleared, some apps depend on that */
    *rgfInOut &= ~SFGAO_VALIDATE;
 
    TRACE("-- result=0x%08x\n", *rgfInOut);
    return S_OK;
}

Referenced by ParseDisplayName().

GetClassID()

HRESULT WINAPI CDrivesFolder::GetClassID ( CLSID *  lpClassId )

override

Implements IPersist.

Definition at line 1367 of file CDrivesFolder.cpp.

{
    TRACE("(%p)\n", this);
 
    if (!lpClassId)
        return E_POINTER;
 
    *lpClassId = CLSID_MyComputer;
    return S_OK;
}

GetCurFolder()

HRESULT WINAPI CDrivesFolder::GetCurFolder ( PIDLIST_ABSOLUTE *  pidl )

override

Implements IPersistFolder2.

Definition at line 1391 of file CDrivesFolder.cpp.

{
    TRACE("(%p)->(%p)\n", this, pidl);
 
    if (!pidl)
        return E_INVALIDARG; /* xp doesn't have this check and crashes on NULL */
 
    *pidl = ILClone(pidlRoot);
    return S_OK;
}

GetDefaultColumn()

HRESULT WINAPI CDrivesFolder::GetDefaultColumn ( DWORD  dwRes, ULONG *  pSort, ULONG *  pDisplay  )

override

Implements IShellFolder2.

Definition at line 1231 of file CDrivesFolder.cpp.

{
    TRACE("(%p)\n", this);
    return E_NOTIMPL; // Not required when column 0 is our default.
}

GetDefaultColumnState()

HRESULT WINAPI CDrivesFolder::GetDefaultColumnState ( UINT  iColumn, DWORD *  pcsFlags  )

override

Implements IShellFolder2.

Definition at line 1237 of file CDrivesFolder.cpp.

{
    TRACE("(%p)\n", this);
 
    if (!pcsFlags || iColumn >= _countof(MyComputerSFHeader))
        return E_INVALIDARG;
    *pcsFlags = MyComputerSFHeader[iColumn].colstate;
    return S_OK;
}

GetDefaultSearchGUID()

HRESULT WINAPI CDrivesFolder::GetDefaultSearchGUID ( GUID *  pguid )

override

Implements IShellFolder2.

Definition at line 1217 of file CDrivesFolder.cpp.

{
    if (LOBYTE(GetVersion()) >= 6)
        return E_NOTIMPL;
    *pguid = CLSID_ShellSearchExt;
    return S_OK;
}

GetDetailsEx()

HRESULT WINAPI CDrivesFolder::GetDetailsEx ( PCUITEMID_CHILD  pidl, const SHCOLUMNID *  pscid, VARIANT *  pv  )

override

Implements IShellFolder2.

Definition at line 1247 of file CDrivesFolder.cpp.

{
    const CLSID *pCLSID = IsRegItem(pidl);
    if (pscid->fmtid == FMTID_ShellDetails)
    {
        switch (pscid->pid)
        {
            case PID_DESCRIPTIONID:
            {
                if (pCLSID)
                    return SHELL_CreateSHDESCRIPTIONID(pv, SHDID_ROOT_REGITEM, pCLSID);
                UINT id = SHDID_COMPUTER_OTHER;
                switch (_ILGetDriveType(pidl))
                {
                    case DRIVE_REMOVABLE: id = _ILGetRemovableTypeId(pidl); break;
                    case DRIVE_FIXED:     id = SHDID_COMPUTER_FIXED; break;
                    case DRIVE_REMOTE:    id = SHDID_COMPUTER_NETDRIVE; break;
                    case DRIVE_CDROM:     id = SHDID_COMPUTER_CDROM; break;
                    case DRIVE_RAMDISK:   id = SHDID_COMPUTER_RAMDISK; break;
                }
                return SHELL_CreateSHDESCRIPTIONID(pv, id, &CLSID_NULL);
            }
        }
    }
    if (pCLSID)
        return m_regFolder->GetDetailsEx(pidl, pscid, pv);
    return SH32_GetDetailsOfPKeyAsVariant(this, pidl, pscid, pv, FALSE);
}

GetDetailsOf()

HRESULT WINAPI CDrivesFolder::GetDetailsOf ( PCUITEMID_CHILD  pidl, UINT  iColumn, SHELLDETAILS *  psd  )

override

Implements IShellFolder2.

Definition at line 1276 of file CDrivesFolder.cpp.

{
    HRESULT hr;
 
    TRACE("(%p)->(%p %i %p)\n", this, pidl, iColumn, psd);
 
    if (!psd || iColumn >= _countof(MyComputerSFHeader))
        return E_INVALIDARG;
 
    if (!pidl)
    {
        psd->fmt = MyComputerSFHeader[iColumn].fmt;
        psd->cxChar = MyComputerSFHeader[iColumn].cxChar;
        return SHSetStrRet(&psd->str, MyComputerSFHeader[iColumn].colnameid);
    }
    else if (!_ILIsDrive(pidl))
    {
        switch (MyComputerSFHeader[iColumn].colnameid)
        {
            case IDS_SHV_COLUMN_NAME:
                return m_regFolder->GetDetailsOf(pidl, SHFSF_COL_NAME, psd);
            case IDS_SHV_COLUMN_TYPE:
                return m_regFolder->GetDetailsOf(pidl, SHFSF_COL_TYPE, psd);
            case IDS_SHV_COLUMN_DISK_CAPACITY:
            case IDS_SHV_COLUMN_DISK_AVAILABLE:
                return SHSetStrRetEmpty(&psd->str);
            case IDS_SHV_COLUMN_COMMENTS:
                return m_regFolder->GetDetailsOf(pidl, SHFSF_COL_COMMENT, psd);
            DEFAULT_UNREACHABLE;
        }
    }
    else
    {
        ULARGE_INTEGER ulTotalBytes, ulFreeBytes;
        WCHAR szDrive[8];
        INT8 DriveNum = GetDrivePath(pidl, szDrive);
        UINT DriveType = GetCachedDriveType(DriveNum);
        if (DriveType > DRIVE_RAMDISK)
            DriveType = DRIVE_FIXED;
 
        switch (MyComputerSFHeader[iColumn].colnameid)
        {
            case IDS_SHV_COLUMN_NAME:
                hr = GetDisplayNameOf(pidl, SHGDN_NORMAL | SHGDN_INFOLDER, &psd->str);
                break;
            case IDS_SHV_COLUMN_TYPE:
                if (DriveType == DRIVE_UNKNOWN)
                    hr = SHSetStrRetEmpty(&psd->str);
                else if (DriveType == DRIVE_REMOVABLE && IsFloppyDrive(szDrive))
                    hr = SHSetStrRet(&psd->str, IDS_DRIVE_FLOPPY);
                else
                    hr = SHSetStrRet(&psd->str, iDriveTypeIds[DriveType]);
                break;
            case IDS_SHV_COLUMN_DISK_CAPACITY:
            case IDS_SHV_COLUMN_DISK_AVAILABLE:
                psd->str.cStr[0] = 0x00;
                psd->str.uType = STRRET_CSTR;
                if (GetVolumeInformationW(szDrive, NULL, 0, NULL, NULL, NULL, NULL, 0))
                {
                    GetDiskFreeSpaceExW(szDrive, &ulFreeBytes, &ulTotalBytes, NULL);
                    if (iColumn == 2)
                        StrFormatByteSize64A(ulTotalBytes.QuadPart, psd->str.cStr, MAX_PATH);
                    else
                        StrFormatByteSize64A(ulFreeBytes.QuadPart, psd->str.cStr, MAX_PATH);
                }
                hr = S_OK;
                break;
            case IDS_SHV_COLUMN_COMMENTS:
                hr = SHSetStrRetEmpty(&psd->str); /* FIXME: comments */
                break;
            DEFAULT_UNREACHABLE;
        }
    }
 
    return hr;
}

GetDisplayNameOf()

HRESULT WINAPI CDrivesFolder::GetDisplayNameOf ( PCUITEMID_CHILD  pidl, DWORD  dwFlags, LPSTRRET  strRet  )

override

Definition at line 1093 of file CDrivesFolder.cpp.

{
    WCHAR szDrive[8];
 
    TRACE("(%p)->(pidl=%p,0x%08x,%p)\n", this, pidl, dwFlags, strRet);
    pdump (pidl);
 
    if (!strRet)
        return E_INVALIDARG;
 
    if (!_ILIsPidlSimple (pidl))
    {
        return SHELL32_GetDisplayNameOfChild(this, pidl, dwFlags, strRet);
    }
    else if (_ILIsSpecialFolder(pidl))
    {
        return m_regFolder->GetDisplayNameOf(pidl, dwFlags, strRet);
    }
    else if (GetDrivePath(pidl, szDrive) < 0)
    {
        ERR("Wrong pidl type\n");
        return E_INVALIDARG;
    }
 
    PWSTR pszPath = (LPWSTR)CoTaskMemAlloc((MAX_PATH + 1) * sizeof(WCHAR));
    if (!pszPath)
        return E_OUTOFMEMORY;
    pszPath[0] = UNICODE_NULL;
    szDrive[0] &= ~32; // Always uppercase
 
    /* long view "lw_name (C:)" */
    BOOL bEditLabel = GET_SHGDN_RELATION(dwFlags) == SHGDN_INFOLDER && (dwFlags & SHGDN_FOREDITING);
    if (!(dwFlags & SHGDN_FORPARSING))
    {
        if (m_DriveDisplayMode < 0)
        {
            DWORD err, type, data, cb = sizeof(data);
            err = SHRegGetUSValueW(L"Software\\Microsoft\\Windows\\CurrentVersion\\Explorer",
                                   L"ShowDriveLettersFirst", &type, &data, &cb, FALSE, NULL, 0);
            m_DriveDisplayMode = (!err && type == REG_DWORD && cb == sizeof(data)) ? (BYTE)data : 0;
        }
        BOOL bRemoteFirst = m_DriveDisplayMode == 1;
        BOOL bNoLetter = m_DriveDisplayMode == 2;
        BOOL bAllFirst = m_DriveDisplayMode == 4;
        PWSTR pszLabel = pszPath;
 
        if (!bNoLetter && (bAllFirst || (bRemoteFirst && GetCachedDriveType(szDrive) == DRIVE_REMOTE)))
        {
            bNoLetter = TRUE; // Handling the letter now, don't append it again later
            if (!bEditLabel)
                pszLabel += wsprintfW(pszPath, L"(%c:) ", szDrive[0]);
        }
 
        if (GetDriveLabel(szDrive, pszLabel, MAX_PATH - 7) != S_OK && !bEditLabel)
        {
            UINT ResId = 0, DrvType = GetCachedDriveType(szDrive);
            if (DrvType == DRIVE_REMOVABLE)
                ResId = IsFloppyDrive(szDrive) ? IDS_DRIVE_FLOPPY : IDS_DRIVE_REMOVABLE;
            else if (DrvType < _countof(iDriveTypeIds))
                ResId = iDriveTypeIds[DrvType];
 
            if (ResId)
            {
                UINT len = LoadStringW(shell32_hInstance, ResId, pszLabel, MAX_PATH - 7);
                if (len > MAX_PATH - 7)
                    pszLabel[MAX_PATH-7] = UNICODE_NULL;
            }
        }
 
        if (!*pszLabel && !bEditLabel) // No label nor fallback description, use SHGDN_FORPARSING
            *pszPath = UNICODE_NULL;
        else if (!bNoLetter && !bEditLabel)
            wsprintfW(pszPath + wcslen(pszPath), L" (%c:)", szDrive[0]);
    }
 
    if (!*pszPath && !bEditLabel) // SHGDN_FORPARSING or failure above (except editing empty label)
    {
        if (GET_SHGDN_RELATION(dwFlags) == SHGDN_INFOLDER)
            szDrive[2] = UNICODE_NULL; // Remove backslash
        wcscpy(pszPath, szDrive);
    }
    strRet->uType = STRRET_WSTR;
    strRet->pOleStr = pszPath;
 
    TRACE("-- (%p)->(%s)\n", this, strRet->uType == STRRET_CSTR ? strRet->cStr : debugstr_w(strRet->pOleStr));
    return S_OK;
}

Referenced by GetDetailsOf().

GetEnumFlags()

STDMETHODIMP CDrivesFolder::GetEnumFlags ( IShellFolder *  , PCIDLIST_ABSOLUTE  , HWND *  , DWORD *    )

inlineoverride

Implements IFolderFilter.

Definition at line 86 of file CDrivesFolder.h.

{ return E_NOTIMPL; }

GetUIObjectOf()

HRESULT WINAPI CDrivesFolder::GetUIObjectOf ( HWND  hwndOwner, UINT  cidl, PCUITEMID_CHILD_ARRAY  apidl, REFIID  riid, UINT *  prgfInOut, LPVOID *  ppvOut  )

override

Definition at line 1036 of file CDrivesFolder.cpp.

{
    LPVOID pObj = NULL;
    HRESULT hr = E_INVALIDARG;
 
    TRACE("(%p)->(%p,%u,apidl=%p,%s,%p,%p)\n", this,
          hwndOwner, cidl, apidl, shdebugstr_guid (&riid), prgfInOut, ppvOut);
 
    if (!ppvOut)
        return hr;
 
    *ppvOut = NULL;
 
    if (IsEqualIID (riid, IID_IContextMenu) && (cidl >= 1))
    {
        if (_ILIsDrive(apidl[0]))
            hr = CDrivesContextMenu_CreateInstance(pidlRoot, hwndOwner, cidl, apidl, static_cast<IShellFolder*>(this), (IContextMenu**)&pObj);
        else
            hr = m_regFolder->GetUIObjectOf(hwndOwner, cidl, apidl, riid, prgfInOut, &pObj);
    }
    else if (IsEqualIID (riid, IID_IDataObject) && (cidl >= 1))
    {
        hr = IDataObject_Constructor(hwndOwner,
                                     pidlRoot, apidl, cidl, TRUE, (IDataObject **)&pObj);
    }
    else if ((IsEqualIID (riid, IID_IExtractIconA) || IsEqualIID (riid, IID_IExtractIconW)) && (cidl == 1))
    {
        if (_ILIsDrive(apidl[0]))
            hr = CDrivesExtractIcon_CreateInstance(this, apidl[0], riid, &pObj);
        else
            hr = m_regFolder->GetUIObjectOf(hwndOwner, cidl, apidl, riid, prgfInOut, &pObj);
    }
    else if (IsEqualIID (riid, IID_IDropTarget) && (cidl == 1))
    {
        CComPtr<IShellFolder> psfChild;
        hr = this->BindToObject(apidl[0], NULL, IID_PPV_ARG(IShellFolder, &psfChild));
        if (FAILED_UNEXPECTEDLY(hr))
            return hr;
 
        return psfChild->CreateViewObject(NULL, riid, ppvOut);
    }
    else
        hr = E_NOINTERFACE;
 
    if (SUCCEEDED(hr) && !pObj)
        hr = E_OUTOFMEMORY;
 
    *ppvOut = pObj;
    TRACE("(%p)->hr=0x%08x\n", this, hr);
    return hr;
}

Initialize()

HRESULT WINAPI CDrivesFolder::Initialize ( PCIDLIST_ABSOLUTE  pidl )

override

Implements IPersistFolder.

Definition at line 1383 of file CDrivesFolder.cpp.

{
    return S_OK;
}

MapColumnToSCID()

HRESULT WINAPI CDrivesFolder::MapColumnToSCID ( UINT  column, SHCOLUMNID *  pscid  )

override

Implements IShellFolder2.

Definition at line 1353 of file CDrivesFolder.cpp.

{
    switch (column < _countof(MyComputerSFHeader) ? MyComputerSFHeader[column].colnameid : ~0UL)
    {
        case IDS_SHV_COLUMN_NAME: return MakeSCID(*pscid, FMTID_Storage, PID_STG_NAME);
        case IDS_SHV_COLUMN_TYPE: return MakeSCID(*pscid, FMTID_Storage, PID_STG_STORAGETYPE);
        case IDS_SHV_COLUMN_COMMENTS: return MakeSCID(*pscid, FMTID_SummaryInformation, PIDSI_COMMENTS);
    }
    return E_INVALIDARG;
}

MessageSFVCB()

STDMETHODIMP CDrivesFolder::MessageSFVCB ( UINT  uMsg, WPARAM  wParam, LPARAM  lParam  )

override

Definition at line 1415 of file CDrivesFolder.cpp.

{
    switch (uMsg)
    {
        case SFVM_FSNOTIFY:
            if (lParam == SHCNE_DRIVEADD && wParam)
            {
                g_IsFloppyCache = 0;
                INT8 drive = GetDriveNumber(((PIDLIST_ABSOLUTE*)wParam)[0]);
                if (drive >= 0 && ((1UL << drive) & SHRestricted(REST_NODRIVES)))
                    return S_FALSE;
            }
            else if (lParam == SHCNE_DRIVEREMOVED)
            {
                g_IsFloppyCache = 0;
            }
            break;
        #if ROSPOLICY_DRIVESFOLDER_DEFLARGEICONS
        case SFVM_DEFVIEWMODE:
            *((FOLDERVIEWMODE*)lParam) = FVM_ICON;
            return S_OK;
        #endif
    }
    return E_NOTIMPL;
}

ParseDisplayName()

HRESULT WINAPI CDrivesFolder::ParseDisplayName ( HWND  hwndOwner, LPBC  pbc, LPOLESTR  lpszDisplayName, DWORD *  pchEaten, PIDLIST_RELATIVE *  ppidl, DWORD *  pdwAttributes  )

override

Definition at line 718 of file CDrivesFolder.cpp.

{
    HRESULT hr = E_INVALIDARG;
 
    TRACE("(%p)->(HWND=%p,%p,%p=%s,%p,pidl=%p,%p)\n", this,
          hwndOwner, pbc, lpszDisplayName, debugstr_w (lpszDisplayName),
          pchEaten, ppidl, pdwAttributes);
 
    if (!ppidl)
        return hr;
 
    *ppidl = NULL;
 
    if (!lpszDisplayName)
        return hr;
 
    /* handle CLSID paths */
    if (lpszDisplayName[0] == L':' && lpszDisplayName[1] == L':')
    {
        return m_regFolder->ParseDisplayName(hwndOwner, pbc, lpszDisplayName, pchEaten, ppidl,
                                             pdwAttributes);
    }
 
    if (((L'A' <= lpszDisplayName[0] && lpszDisplayName[0] <= L'Z') ||
         (L'a' <= lpszDisplayName[0] && lpszDisplayName[0] <= L'z')) &&
        lpszDisplayName[1] == L':' && (lpszDisplayName[2] == L'\\' || !lpszDisplayName[2]))
    {
        if (!lpszDisplayName[2] && LOBYTE(GetVersion()) < 6)
            return E_INVALIDARG; // NT5 does not support parsing "c:"
 
        // "C:\..."
        WCHAR szRoot[8];
        PathBuildRootW(szRoot, ((*lpszDisplayName - 1) & 0x1F));
 
        if (SHIsFileSysBindCtx(pbc, NULL) != S_OK && !(BindCtx_GetMode(pbc, 0) & STGM_CREATE))
        {
            if (::GetDriveType(szRoot) == DRIVE_NO_ROOT_DIR)
                return HRESULT_FROM_WIN32(ERROR_PATH_NOT_FOUND);
        }
 
        CComHeapPtr<ITEMIDLIST> pidlTemp(_ILCreateDrive(szRoot));
        if (!pidlTemp)
            return E_OUTOFMEMORY;
 
        if (lpszDisplayName[2] && lpszDisplayName[3])
        {
            CComPtr<IShellFolder> pChildFolder;
            hr = BindToObject(pidlTemp, pbc, IID_PPV_ARG(IShellFolder, &pChildFolder));
            if (FAILED_UNEXPECTEDLY(hr))
                return hr;
 
            ULONG chEaten;
            CComHeapPtr<ITEMIDLIST> pidlChild;
            hr = pChildFolder->ParseDisplayName(hwndOwner, pbc, &lpszDisplayName[3], &chEaten,
                                                &pidlChild, pdwAttributes);
            if (FAILED_UNEXPECTEDLY(hr))
                return hr;
 
            hr = SHILCombine(pidlTemp, pidlChild, ppidl);
        }
        else
        {
            *ppidl = pidlTemp.Detach();
            if (pdwAttributes && *pdwAttributes)
                GetAttributesOf(1, (PCUITEMID_CHILD_ARRAY)ppidl, pdwAttributes);
            hr = S_OK;
        }
    }
 
    TRACE("(%p)->(-- ret=0x%08x)\n", this, hr);
 
    return hr;
}

SetDriveLabel()

HRESULT CDrivesFolder::SetDriveLabel ( HWND  hwndOwner, PCWSTR  DrivePath, PCWSTR  Label  )

static

Definition at line 1207 of file CDrivesFolder.cpp.

{
    HRESULT hr = SetVolumeLabelW(DrivePath, *Label ? Label : NULL) ? S_OK : HResultFromWin32(GetLastError());
    if (SUCCEEDED(hr))
        SHChangeNotify(SHCNE_RENAMEFOLDER, SHCNF_PATHW, DrivePath, DrivePath); // DisplayName changed
    else if (hwndOwner)
        SHELL_ErrorBox(hwndOwner, hr);
    return hr;
}

Referenced by CDrvDefExt::GeneralPageProc(), and SetNameOf().

SetNameOf()

HRESULT WINAPI CDrivesFolder::SetNameOf ( HWND  hwndOwner, PCUITEMID_CHILD  pidl, LPCOLESTR  lpName, DWORD  dwFlags, PITEMID_CHILD *  pPidlOut  )

override

Implements IShellFolder.

Definition at line 1193 of file CDrivesFolder.cpp.

{
    if (_ILIsDrive(pidl))
    {
        WCHAR szDrive[8];
        HRESULT hr = GetDrivePath(pidl, szDrive) >= 0 ? SetDriveLabel(hwndOwner, szDrive, lpName) : E_FAIL;
        if (pPidlOut)
            *pPidlOut = SUCCEEDED(hr) ? _ILCreateDrive(szDrive) : NULL;
        return hr;
    }
    return m_regFolder->SetNameOf(hwndOwner, pidl, lpName, dwFlags, pPidlOut);
}

ShouldShow()

HRESULT WINAPI CDrivesFolder::ShouldShow ( IShellFolder *  psf, PCIDLIST_ABSOLUTE  pidlFolder, PCUITEMID_CHILD  pidlItem  )

override

Implements IFolderFilter.

Definition at line 1405 of file CDrivesFolder.cpp.

{
    if (const CLSID* pClsid = IsRegItem(pidlItem))
        return SHELL32_IsShellFolderNamespaceItemHidden(L"HideMyComputerIcons", *pClsid) ? S_FALSE : S_OK;
    return S_OK;
}

Member Data Documentation

m_DriveDisplayMode

INT8 CDrivesFolder::m_DriveDisplayMode

private

Definition at line 39 of file CDrivesFolder.h.

Referenced by CDrivesFolder(), and GetDisplayNameOf().

m_regFolder

CComPtr<IShellFolder2> CDrivesFolder::m_regFolder

private

Definition at line 38 of file CDrivesFolder.h.

Referenced by BindToObject(), EnumObjects(), FinalConstruct(), GetAttributesOf(), GetDetailsEx(), GetDetailsOf(), GetDisplayNameOf(), GetUIObjectOf(), ParseDisplayName(), and SetNameOf().

pidlRoot

LPITEMIDLIST CDrivesFolder::pidlRoot

private

Definition at line 37 of file CDrivesFolder.h.

Referenced by BindToObject(), CDrivesFolder(), CreateViewObject(), FinalConstruct(), GetCurFolder(), GetUIObjectOf(), and ~CDrivesFolder().

---

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

• dll/win32/shell32/folders/CDrivesFolder.h
• dll/win32/shell32/folders/CDrivesFolder.cpp