#include <CDrivesFolder.h>

Inheritance diagram for CDrivesFolder:

[legend]

Collaboration diagram for CDrivesFolder:

[legend]

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 678 of file CDrivesFolder.cpp.

{
    pidlRoot = NULL;
    m_DriveDisplayMode = -1;
}

◆ ~CDrivesFolder()

CDrivesFolder::~CDrivesFolder ( )

Definition at line 684 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 793 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 828 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 1433 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 840 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 929 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 782 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 1209 of file CDrivesFolder.cpp.

{
    FIXME("(%p)\n", this);
    return E_NOTIMPL;
}

◆ FinalConstruct()

HRESULT WINAPI CDrivesFolder::FinalConstruct ( )

Definition at line 690 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 964 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 1356 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 1380 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 1215 of file CDrivesFolder.cpp.

{
    TRACE("(%p)\n", this);
 
    if (pSort)
        *pSort = 0;
    if (pDisplay)
        *pDisplay = 0;
    return S_OK;
}

◆ GetDefaultColumnState()

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

override

Implements IShellFolder2.

Definition at line 1226 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 1203 of file CDrivesFolder.cpp.

{
    FIXME("(%p)\n", this);
    return E_NOTIMPL;
}

◆ GetDetailsEx()

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

override

Implements IShellFolder2.

Definition at line 1236 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 1265 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 1079 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.

86{ 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 1022 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 1372 of file CDrivesFolder.cpp.

{
    return S_OK;
}

◆ MapColumnToSCID()

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

override

Implements IShellFolder2.

Definition at line 1342 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 1404 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 707 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]))
    {
        // "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 1193 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 1179 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 1394 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

Public Member Functions

Static Public Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

◆ CDrivesFolder()

◆ ~CDrivesFolder()

Member Function Documentation

◆ BindToObject()

◆ BindToStorage()

◆ CallBack()

◆ CompareIDs()

◆ CreateViewObject()

◆ EnumObjects()

◆ EnumSearches()

◆ FinalConstruct()

◆ GetAttributesOf()

◆ GetClassID()

◆ GetCurFolder()

◆ GetDefaultColumn()

◆ GetDefaultColumnState()

◆ GetDefaultSearchGUID()

◆ GetDetailsEx()

◆ GetDetailsOf()

◆ GetDisplayNameOf()

◆ GetEnumFlags()

◆ GetUIObjectOf()

◆ Initialize()

◆ MapColumnToSCID()

◆ MessageSFVCB()

◆ ParseDisplayName()

◆ SetDriveLabel()

◆ SetNameOf()

◆ ShouldShow()

Member Data Documentation

◆ m_DriveDisplayMode

◆ m_regFolder

◆ pidlRoot