#include "precomp.h"

Include dependency graph for folders.cpp:

Functions
static void	InitIconOverlays (void)

BOOL	GetIconOverlay (LPCITEMIDLIST pidl, WCHAR wTemp, int pIndex)

Variables
WCHAR	swShell32Name [MAX_PATH]

DWORD	NumIconOverlayHandlers = 0

IShellIconOverlayIdentifier **	Handlers = NULL

Function Documentation

◆ GetIconOverlay()

BOOL GetIconOverlay	(	LPCITEMIDLIST	pidl,
		WCHAR *	wTemp,
		int *	pIndex
	)

Definition at line 88 of file folders.cpp.

{
    DWORD Index;
    HRESULT hResult;
    int Priority;
    int HighestPriority;
    ULONG IconIndex;
    ULONG Flags;
    WCHAR szPath[MAX_PATH];
 
    if(!SHGetPathFromIDListW(pidl, szPath))
        return FALSE;
 
    if (!Handlers)
        InitIconOverlays();
 
    HighestPriority = 101;
    IconIndex = NumIconOverlayHandlers;
    for(Index = 0; Index < NumIconOverlayHandlers; Index++)
    {
        hResult = Handlers[Index]->IsMemberOf(szPath, SFGAO_FILESYSTEM);
        if (hResult == S_OK)
        {
            hResult = Handlers[Index]->GetPriority(&Priority);
            if (hResult == S_OK)
            {
                if (Priority < HighestPriority)
                {
                    HighestPriority = Priority;
                    IconIndex = Index;
                }
            }
        }
    }
 
    if (IconIndex == NumIconOverlayHandlers)
        return FALSE;
 
    hResult = Handlers[IconIndex]->GetOverlayInfo(wTemp, MAX_PATH, pIndex, &Flags);
 
    if (hResult == S_OK)
        return TRUE;
    else
        return FALSE;
}

◆ InitIconOverlays()

static void InitIconOverlays ( void )

static

Definition at line 27 of file folders.cpp.

{
    HKEY hKey;
    DWORD dwIndex, dwResult, dwSize;
    WCHAR szName[MAX_PATH];
    WCHAR szValue[100];
    CLSID clsid;
 
    if (RegOpenKeyExW(HKEY_LOCAL_MACHINE, L"SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Explorer\\ShellIconOverlayIdentifiers", 0, KEY_READ, &hKey) != ERROR_SUCCESS)
        return;
 
    if (RegQueryInfoKeyW(hKey, NULL, NULL, NULL, &dwResult, NULL, NULL, NULL, NULL, NULL, NULL, NULL) != ERROR_SUCCESS)
    {
        RegCloseKey(hKey);
        return;
    }
 
    Handlers = (IShellIconOverlayIdentifier **)HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, dwResult * sizeof(IShellIconOverlayIdentifier*));
    if (!Handlers)
    {
        RegCloseKey(hKey);
        return;
    }
 
    dwIndex = 0;
 
    CoInitialize(0);
 
    do
    {
        dwSize = sizeof(szName) / sizeof(WCHAR);
        dwResult = RegEnumKeyExW(hKey, dwIndex, szName, &dwSize, NULL, NULL, NULL, NULL);
 
        if (dwResult == ERROR_NO_MORE_ITEMS)
            break;
 
        if (dwResult == ERROR_SUCCESS)
        {
            dwSize = sizeof(szValue) / sizeof(WCHAR);
            if (RegGetValueW(hKey, szName, NULL, RRF_RT_REG_SZ, NULL, szValue, &dwSize) == ERROR_SUCCESS)
            {
                CComPtr<IShellIconOverlayIdentifier> Overlay;
 
                CLSIDFromString(szValue, &clsid);
                dwResult = CoCreateInstance(clsid, NULL, CLSCTX_INPROC_SERVER, IID_PPV_ARG(IShellIconOverlayIdentifier, &Overlay));
                if (dwResult == S_OK)
                {
                    Handlers[NumIconOverlayHandlers] = Overlay.Detach();
                    NumIconOverlayHandlers++;
                }
            }
        }
 
        dwIndex++;
 
    } while(1);
 
    RegCloseKey(hKey);
}