实现IExtractIcon获取虚拟文件的黄色文件夹图标与扩展名图标

我遇到了和某微软博客中工程师完全相同的问题，Raymond Chen的回答没给出具体修复方法，只是说要写代码调用系统接口，这让我很困惑。

我的需求明确：

• 正在实现IShellFolder::GetUIObjectOf，目标是返回IExtractIcon接口
• 命名空间中有虚拟项，目录和文件一一对应
• 目录项需要返回标准黄色文件夹图标
• 文件项需要返回对应扩展名（如.txt、.jpg）的系统关联图标

我按照IExtractIcon文档写的代码无法正常工作，当前思路是：

• GetIconLocation中，目录用SHGetFileInfo+SHGFI_USEFILEATTRIBUTES；文件用AssocQueryString获取图标路径和索引
• Extract中，目录返回提前获取的图标，文件返回S_FALSE让Shell自行提取

现在目录部分就卡壳了：Shell要么找不到关联文件，要么拒绝绘制图标（比如Win11资源管理器标签栏的图标）。

当前代码：

STDMETHODIMP CExtractIcon::GetIconLocation (
    UINT uFlags,
    LPTSTR szIconFile,
    UINT cchMax,
    LPINT piIndex,
    UINT *puFlags)
{
    HRESULT hr = E_FAIL;

    if (descriptor.flags.is_a_directory)
    {
        SHFILEINFO sfi = { 0 };

        if (SHGetFileInfo(
            TEXT("DummyFolderPath"),
            FILE_ATTRIBUTE_DIRECTORY,
            &sfi,
            sizeof(sfi),
            SHGFI_USEFILEATTRIBUTES | SHGFI_SYSICONINDEX | SHGFI_ICON | SHGFI_SMALLICON | (uFlags & GIL_OPENICON ? SHGFI_OPENICON : 0))
            )
            this->hIconSmall = sfi.hIcon;

        if (SHGetFileInfo(
            TEXT("DummyFolderPath"),
            FILE_ATTRIBUTE_DIRECTORY,
            &sfi,
            sizeof(sfi),
            SHGFI_USEFILEATTRIBUTES | SHGFI_SYSICONINDEX | SHGFI_ICON //  (uFlags & GIL_OPENICON ? SHGFI_OPENICON : 0)) DEFECT: Might work for large folders too.
            ))
            this->hIconLarge = sfi.hIcon;

        *piIndex = sfi.iIcon;

        *puFlags = GIL_NOTFILENAME;

        hr = S_FALSE;
    }
    else
    {
        DWORD cchOut = cchMax;

        hr = AssocQueryString(
            ASSOCF_NONE,
            ASSOCSTR_DEFAULTICON,
            descriptor.name.locate_reverse('.') ? descriptor.name.indexed_suffix().dump<TCHAR>() : TEXT(""),
            NULL,
            szIconFile,
            &cchOut);
    }
    return hr;
}

STDMETHODIMP CExtractIcon::Extract(LPCTSTR pszRoot,
    UINT nIconIndex,
    HICON *phiconLarge,
    HICON *phiconSmall,
    UINT nIconSize)
{
    *phiconSmall = this->hIconSmall;
    *phiconLarge = this->hIconLarge;

    return S_OK;
}

目录项的正确实现 #

你的目录实现存在三个关键问题：

• 无需提前调用SHGetFileInfo获取图标句柄，Shell可通过系统图标索引自行加载，手动缓存易导致资源泄漏或状态不一致
• GIL_NOTFILENAME需配合GIL_SYSICONINDEX使用，明确告知Shell这是系统图标库的索引
• 无需同时获取大小图标，通过uFlags判断即可返回对应状态的索引

修正后的GetIconLocation目录分支：

if (descriptor.flags.is_a_directory)
{
    SHFILEINFO sfi = { 0 };
    UINT shfiFlags = SHGFI_USEFILEATTRIBUTES | SHGFI_SYSICONINDEX;
    
    // 根据GIL_OPENICON标记返回打开状态的文件夹图标
    if (uFlags & GIL_OPENICON)
    {
        shfiFlags |= SHGFI_OPENICON;
    }

    // 仅获取系统图标索引，不提前提取图标
    if (SHGetFileInfo(TEXT(""), FILE_ATTRIBUTE_DIRECTORY, &sfi, sizeof(sfi), shfiFlags))
    {
        *piIndex = sfi.iIcon;
        *puFlags = GIL_NOTFILENAME | GIL_SYSICONINDEX;
        hr = S_OK;
    }
    else
    {
        hr = HRESULT_FROM_WIN32(GetLastError());
    }
}

对应的Extract方法简化为让Shell自行处理：

STDMETHODIMP CExtractIcon::Extract(LPCTSTR pszRoot,
    UINT nIconIndex,
    HICON *phiconLarge,
    HICON *phiconSmall,
    UINT nIconSize)
{
    if (descriptor.flags.is_a_directory)
    {
        *phiconLarge = NULL;
        *phiconSmall = NULL;
        return S_FALSE; // 让Shell根据系统索引提取图标
    }
    // 文件分支见下文
    ...
}

文件项的正确实现 #

文件部分需注意三个细节：

• AssocQueryString要求扩展名以.开头（如.txt而非txt），无扩展名时用*获取默认文件图标
• 需解析AssocQueryString返回的字符串（格式为路径,索引），分割出图标路径和索引值
• Extract直接返回S_FALSE，让Shell负责图标加载

修正后的GetIconLocation文件分支：

else
{
    CStringW ext;
    if (descriptor.name.locate_reverse('.'))
    {
        ext = L".";
        ext += descriptor.name.indexed_suffix().dump<TCHAR>();
    }
    else
    {
        ext = L"*"; // 无扩展名时使用默认文件图标
    }

    WCHAR szIconPath[MAX_PATH] = {0};
    DWORD cchOut = MAX_PATH;
    HRESULT hr = AssocQueryStringW(
        ASSOCF_INIT_DEFAULTTOSTAR | ASSOCF_NOTRUNCATE,
        ASSOCSTR_DEFAULTICON,
        ext,
        NULL,
        szIconPath,
        &cchOut);

    if (SUCCEEDED(hr))
    {
        // 分割路径与索引，格式为"path,index"
        WCHAR* pComma = wcschr(szIconPath, L',');
        if (pComma != NULL)
        {
            *pComma = L'\0';
            *piIndex = _wtoi(pComma + 1);
        }
        else
        {
            *piIndex = 0;
        }

        // 确保路径长度不超过缓冲区限制
        if (wcslen(szIconPath) < cchMax)
        {
            wcscpy_s(szIconFile, cchMax, szIconPath);
            *puFlags = 0; // 标记为正常文件路径+索引
            hr = S_OK;
        }
        else
        {
            hr = E_INVALIDARG;
        }
    }
}

文件部分的Extract实现：

else
{
    *phiconLarge = NULL;
    *phiconSmall = NULL;
    return S_FALSE; // 让Shell自行提取图标
}

关键注意事项 #

• 统一使用Unicode版本API（SHGetFileInfoW、AssocQueryStringW），避免编码兼容问题
• 不要手动缓存图标句柄，Shell会负责图标加载、缓存和释放，手动处理易引发资源泄漏
• 错误处理时返回对应HRESULT，便于调试定位问题

内容的提问来源于stack exchange，提问作者HackTheStack