CCleanupHandler Struct Reference

#include <CCleanupHandler.hpp>

Collaboration diagram for CCleanupHandler:

Public Member Functions
	CCleanupHandler (CRegKey &subKey, const CStringW &keyName, const GUID &guid)
	~CCleanupHandler ()
void	Deactivate ()
bool	Initialize (LPCWSTR pcwszVolume)
void	ReadProperty (LPCWSTR Name, IPropertyBag *pBag, CComHeapPtr< WCHAR > &storage)
BOOL	HasSettings () const
BOOL	DontShowIfZero () const

Public Attributes
CRegKey	hSubKey
CStringW	KeyName
GUID	Guid
CComHeapPtr< WCHAR >	wszDisplayName
CComHeapPtr< WCHAR >	wszDescription
CComHeapPtr< WCHAR >	wszBtnText
CStringW	IconPath
DWORD	dwFlags
DWORD	Priority
DWORD	StateFlags
CComPtr< IEmptyVolumeCache >	Handler
DWORDLONG	SpaceUsed
bool	ShowHandler
HICON	hIcon

Detailed Description

Definition at line 11 of file CCleanupHandler.hpp.

Constructor & Destructor Documentation

CCleanupHandler()

CCleanupHandler::CCleanupHandler	(	CRegKey &	subKey,
		const CStringW &	keyName,
		const GUID &	guid
	)

Definition at line 11 of file CCleanupHandler.cpp.

    : hSubKey(subKey)
    , KeyName(keyName)
    , Guid(guid)
    , dwFlags(0)
    , Priority(0)
    , StateFlags(0)
    , SpaceUsed(0)
    , ShowHandler(true)
    , hIcon(NULL)
{
}

~CCleanupHandler()

CCleanupHandler::~CCleanupHandler

(

)

Definition at line 24 of file CCleanupHandler.cpp.

{
    Deactivate();
    ::DestroyIcon(hIcon);
}

Member Function Documentation

Deactivate()

void CCleanupHandler::Deactivate

(

)

Definition at line 31 of file CCleanupHandler.cpp.

{
    if (Handler)
    {
        DWORD dwFlags = 0;
        Handler->Deactivate(&dwFlags);
        if (dwFlags & EVCF_REMOVEFROMLIST)
            UNIMPLEMENTED_DBGBREAK();
    }
}

Referenced by ~CCleanupHandler().

DontShowIfZero()

BOOL CCleanupHandler::DontShowIfZero

(

)

const

Definition at line 208 of file CCleanupHandler.cpp.

{
    return !!(dwFlags & EVCF_DONTSHOWIFZERO);
}

HasSettings()

BOOL CCleanupHandler::HasSettings

(

)

const

Definition at line 202 of file CCleanupHandler.cpp.

{
    return !!(dwFlags & EVCF_HASSETTINGS);
}

Initialize()

bool CCleanupHandler::Initialize

(

LPCWSTR

pcwszVolume

)

Definition at line 43 of file CCleanupHandler.cpp.

{
    if (FAILED_UNEXPECTEDLY(
            ::CoCreateInstance(Guid, NULL, CLSCTX_INPROC_SERVER, IID_PPV_ARG(IEmptyVolumeCache, &Handler))))
    {
        return false;
    }
 
    DWORD dwSize = sizeof(Priority);
    if (hSubKey.QueryBinaryValue(L"Priority", &Priority, &dwSize) != ERROR_SUCCESS)
    {
        if (hSubKey.QueryDWORDValue(L"Priority", Priority) != ERROR_SUCCESS)
            Priority = 200;
    }
 
    dwSize = sizeof(StateFlags);
    if (hSubKey.QueryDWORDValue(L"StateFlags", StateFlags) != ERROR_SUCCESS)
        StateFlags = 0;
 
    WCHAR PathBuffer[MAX_PATH] = {};
    ULONG nChars = _countof(PathBuffer);
    if (hSubKey.QueryStringValue(L"IconPath", PathBuffer, &nChars) != ERROR_SUCCESS)
    {
        CStringW Tmp;
        WCHAR GuidStr[50] = {};
        if (StringFromGUID2(Guid, GuidStr, _countof(GuidStr)))
        {
            Tmp.Format(L"CLSID\\%s\\DefaultIcon", GuidStr);
            CRegKey clsid;
            nChars = _countof(PathBuffer);
            if (clsid.Open(HKEY_CLASSES_ROOT, Tmp, KEY_READ) != ERROR_SUCCESS ||
                clsid.QueryStringValue(NULL, PathBuffer, &nChars) != ERROR_SUCCESS)
            {
                PathBuffer[0] = UNICODE_NULL;
            }
        }
    }
    if (!PathBuffer[0])
        StringCchCopyW(PathBuffer, _countof(PathBuffer), L"%systemroot%\\system32\\shell32.dll");
 
    int Index = 0;
    WCHAR *ptr = wcschr(PathBuffer, L',');
    if (ptr)
    {
        *ptr++ = UNICODE_NULL;
        Index = wcstol(ptr, NULL, 10);
    }
    HICON Large, Small;
    UINT Result = ExtractIconExW(PathBuffer, Index, &Large, &Small, 1);
    if (Result < 1)
        Result = ExtractIconExW(L"%systemroot%\\system32\\shell32.dll", 0, &Large, &Small, 1);
    if (Result >= 1)
    {
        hIcon = Small;
        if (!hIcon)
        {
            hIcon = Large;
        }
        else
        {
            ::DestroyIcon(Large);
        }
    }
 
    // These options should come from the command line
    // dwFlags |= EVCF_SETTINGSMODE;
    // dwFlags |= EVCF_OUTOFDISKSPACE;
 
    CComPtr<IEmptyVolumeCache2> spHandler2;
    HRESULT hr = Handler->QueryInterface(IID_PPV_ARG(IEmptyVolumeCache2, &spHandler2));
    if (SUCCEEDED(hr))
    {
        hr = spHandler2->InitializeEx(
            hSubKey, pcwszVolume, KeyName, &wszDisplayName, &wszDescription, &wszBtnText, &dwFlags);
        if (FAILED_UNEXPECTEDLY(hr))
            return false;
 
        // No files to clean will return S_FALSE;
        if (hr != S_OK)
            return false;
    }
    else
    {
        // Observed behavior:
        // When Initialize is called, wszDescription is actually pointing to data
        // wszDescription.AllocateBytes(0x400u);
        hr = Handler->Initialize(hSubKey, pcwszVolume, &wszDisplayName, &wszDescription, &dwFlags);
        if (FAILED_UNEXPECTEDLY(hr))
            return false;
 
        // No files to clean will return S_FALSE;
        if (hr != S_OK)
            return false;
 
        CComPtr<IPropertyBag> spBag;
        WCHAR GuidStr[50] = {};
        nChars = _countof(GuidStr);
        if (hSubKey.QueryStringValue(L"PropertyBag", GuidStr, &nChars) == ERROR_SUCCESS)
        {
            GUID guid = {};
            if (!FAILED_UNEXPECTEDLY(CLSIDFromString(GuidStr, &guid)))
            {
                FAILED_UNEXPECTEDLY(
                    CoCreateInstance(guid, NULL, CLSCTX_INPROC_SERVER, IID_PPV_ARG(IPropertyBag, &spBag)));
            }
        }
        ReadProperty(L"Display", spBag, wszDisplayName);
        ReadProperty(L"Description", spBag, wszDescription);
 
        if (dwFlags & EVCF_HASSETTINGS)
        {
            ReadProperty(L"AdvancedButtonText", spBag, wszBtnText);
        }
    }
 
    if ((dwFlags & EVCF_ENABLEBYDEFAULT) && !(StateFlags & HANDLER_STATE_SELECTED))
    {
        StateFlags |= HANDLER_STATE_SELECTED;
    }
 
    // For convenience
    if (!wszDisplayName)
        SHStrDupW(KeyName, &wszDisplayName);
 
    return true;
}

ReadProperty()

void CCleanupHandler::ReadProperty	(	LPCWSTR	Name,
		IPropertyBag *	pBag,
		CComHeapPtr< WCHAR > &	storage
	)

Definition at line 171 of file CCleanupHandler.cpp.

{
    if (storage)
        return;
 
    if (pBag)
    {
        CComVariant tmp;
        tmp.vt = VT_BSTR;
        HRESULT hr = pBag->Read(Name, &tmp, NULL);
        if (!FAILED_UNEXPECTEDLY(hr) && tmp.vt == VT_BSTR)
        {
            SHStrDupW(tmp.bstrVal, &storage);
        }
    }
 
    if (!storage)
    {
        WCHAR TmpStr[0x200] = {};
        DWORD dwSize = _countof(TmpStr);
 
        if (hSubKey.QueryStringValue(Name, TmpStr, &dwSize) == ERROR_SUCCESS)
        {
            WCHAR ResolvedStr[0x200] = {};
            SHLoadIndirectString(TmpStr, ResolvedStr, _countof(ResolvedStr), NULL);
            SHStrDupW(ResolvedStr, &storage);
        }
    }
}

Referenced by Initialize().

Member Data Documentation

dwFlags

DWORD CCleanupHandler::dwFlags

Definition at line 39 of file CCleanupHandler.hpp.

Guid

GUID CCleanupHandler::Guid

Definition at line 32 of file CCleanupHandler.hpp.

Handler

CComPtr<IEmptyVolumeCache> CCleanupHandler::Handler

Definition at line 43 of file CCleanupHandler.hpp.

Referenced by Deactivate(), and Initialize().

hIcon

HICON CCleanupHandler::hIcon

Definition at line 46 of file CCleanupHandler.hpp.

Referenced by Initialize(), and ~CCleanupHandler().

hSubKey

CRegKey CCleanupHandler::hSubKey

Definition at line 30 of file CCleanupHandler.hpp.

Referenced by Initialize(), and ReadProperty().

IconPath

CStringW CCleanupHandler::IconPath

Definition at line 38 of file CCleanupHandler.hpp.

KeyName

CStringW CCleanupHandler::KeyName

Definition at line 31 of file CCleanupHandler.hpp.

Priority

DWORD CCleanupHandler::Priority

Definition at line 40 of file CCleanupHandler.hpp.

Referenced by Initialize().

ShowHandler

bool CCleanupHandler::ShowHandler

Definition at line 45 of file CCleanupHandler.hpp.

SpaceUsed

DWORDLONG CCleanupHandler::SpaceUsed

Definition at line 44 of file CCleanupHandler.hpp.

StateFlags

DWORD CCleanupHandler::StateFlags

Definition at line 41 of file CCleanupHandler.hpp.

Referenced by Initialize().

wszBtnText

CComHeapPtr<WCHAR> CCleanupHandler::wszBtnText

Definition at line 36 of file CCleanupHandler.hpp.

Referenced by Initialize().

wszDescription

CComHeapPtr<WCHAR> CCleanupHandler::wszDescription

Definition at line 35 of file CCleanupHandler.hpp.

Referenced by Initialize().

wszDisplayName

CComHeapPtr<WCHAR> CCleanupHandler::wszDisplayName

Definition at line 34 of file CCleanupHandler.hpp.

Referenced by Initialize().

---

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

• base/applications/cleanmgr/cleanmgr/CCleanupHandler.hpp
• base/applications/cleanmgr/cleanmgr/CCleanupHandler.cpp