Add Disable folder thumbnails mod

mods/disable-folder-thumbnails.wh.cpp

@@ -0,0 +1,253 @@
+// ==WindhawkMod==
+// @id              disable-folder-thumbnails
+// @name            Disable Folder Thumbnails
+// @description     Disable folder thumbnails while preserving file thumbnails.
+// @version         1.0
+// @author          Anixx
+// @github          https://github.com/Anixx
+// @include         *
+// @compilerOptions -lole32 -luuid -lshell32
+// ==/WindhawkMod==
+
+// ==WindhawkModReadme==
+/*
+# Disable Folder Thumbnails
+
+Disables thumbnail extraction and retrieval for ordinary filesystem
+folders both in Explorer and file choosing dialogs. 
+This mod is particularly useful if you are using a custom folder icon.
+
+File thumbnail requests are passed through unchanged. Not-thumbnail icon overlays are unaffected.
+
+*/
+// ==/WindhawkModReadme==
+
+#include <windows.h>
+#include <shobjidl.h>
+#include <thumbcache.h>
+#include <shlobj.h>
+
+using GetThumbnail_t = HRESULT(STDMETHODCALLTYPE*)(
+    IThumbnailCache* self,
+    IShellItem* shellItem,
+    UINT requestedSize,
+    WTS_FLAGS flags,
+    ISharedBitmap** thumbnail,
+    WTS_CACHEFLAGS* cacheFlags,
+    WTS_THUMBNAILID* thumbnailId
+);
+
+static GetThumbnail_t g_originalGetThumbnail = nullptr;
+
+static bool IsPlainFilesystemFolder(IShellItem* item)
+{
+    if (!item) {
+        return false;
+    }
+
+    constexpr SFGAOF mask =
+        SFGAO_FOLDER | SFGAO_FILESYSTEM | SFGAO_STREAM | SFGAO_LINK;
+
+    SFGAOF attributes = 0;
+    if (FAILED(item->GetAttributes(mask, &attributes))) {
+        return false;
+    }
+
+    return (attributes & (SFGAO_FOLDER | SFGAO_FILESYSTEM)) ==
+               (SFGAO_FOLDER | SFGAO_FILESYSTEM) &&
+           !(attributes & (SFGAO_STREAM | SFGAO_LINK));
+}
+
+// True for ordinary filesystem folders and shortcuts targeting them.
+static bool IsFilesystemFolder(IShellItem* item)
+{
+    if (!item) {
+        return false;
+    }
+
+    if (IsPlainFilesystemFolder(item)) {
+        return true;
+    }
+
+    SFGAOF attributes = 0;
+    if (FAILED(item->GetAttributes(SFGAO_LINK, &attributes)) ||
+        !(attributes & SFGAO_LINK)) {
+        return false;
+    }
+
+    // Load the shortcut itself, without executing it.
+    PWSTR shortcutPath = nullptr;
+    if (FAILED(item->GetDisplayName(
+            SIGDN_FILESYSPATH, &shortcutPath))) {
+        return false;
+    }
+
+    IShellLinkW* link = nullptr;
+    HRESULT hr = CoCreateInstance(
+        CLSID_ShellLink,
+        nullptr,
+        CLSCTX_INPROC_SERVER,
+        IID_PPV_ARGS(&link)
+    );
+
+    if (FAILED(hr)) {
+        CoTaskMemFree(shortcutPath);
+        return false;
+    }
+
+    IPersistFile* persistFile = nullptr;
+    hr = link->QueryInterface(IID_PPV_ARGS(&persistFile));
+
+    if (SUCCEEDED(hr)) {
+        hr = persistFile->Load(shortcutPath, STGM_READ);
+        persistFile->Release();
+    }
+
+    CoTaskMemFree(shortcutPath);
+
+    bool targetIsFolder = false;
+
+    if (SUCCEEDED(hr)) {
+        // Use the stored target. Don't call Resolve(), which could
+        // search for missing targets or introduce delays and UI.
+        PIDLIST_ABSOLUTE targetIdList = nullptr;
+        hr = link->GetIDList(&targetIdList);
+
+        if (SUCCEEDED(hr) && targetIdList) {
+            IShellItem* targetItem = nullptr;
+            hr = SHCreateItemFromIDList(
+                targetIdList,
+                IID_PPV_ARGS(&targetItem)
+            );
+
+            if (SUCCEEDED(hr)) {
+                targetIsFolder = IsPlainFilesystemFolder(targetItem);
+                targetItem->Release();
+            }
+        }
+
+        CoTaskMemFree(targetIdList);
+    }
+
+    link->Release();
+    return targetIsFolder;
+}
+
+static HRESULT STDMETHODCALLTYPE GetThumbnail_Hook(
+    IThumbnailCache* self,
+    IShellItem* shellItem,
+    UINT requestedSize,
+    WTS_FLAGS flags,
+    ISharedBitmap** thumbnail,
+    WTS_CACHEFLAGS* cacheFlags,
+    WTS_THUMBNAILID* thumbnailId)
+{
+    if (IsFilesystemFolder(shellItem)) {
+        if (thumbnail) {
+            *thumbnail = nullptr;
+        }
+
+        if (cacheFlags) {
+            *cacheFlags = static_cast<WTS_CACHEFLAGS>(0);
+        }
+
+        if (thumbnailId) {
+            ZeroMemory(thumbnailId, sizeof(*thumbnailId));
+        }
+
+        // Reject both cached and newly generated folder thumbnails.
+        // Explorer can then use its normal icon fallback.
+        return WTS_E_FAILEDEXTRACTION;
+    }
+
+    return g_originalGetThumbnail(
+        self,
+        shellItem,
+        requestedSize,
+        flags,
+        thumbnail,
+        cacheFlags,
+        thumbnailId
+    );
+}
+
+BOOL Wh_ModInit()
+{
+    Wh_Log(L"Initializing folder-thumbnail suppression");
+
+    HRESULT initHr = CoInitializeEx(nullptr, COINIT_MULTITHREADED);
+    bool mustUninitialize = SUCCEEDED(initHr);
+
+    if (FAILED(initHr) && initHr != RPC_E_CHANGED_MODE) {
+        Wh_Log(L"CoInitializeEx failed: 0x%08X",
+               static_cast<UINT>(initHr));
+        return FALSE;
+    }
+
+    // CLSID_LocalThumbnailCache:
+    // {50EF4544-AC9F-4A8E-B21B-8A26180DB13F}
+    static const CLSID thumbnailCacheClsid = {
+        0x50EF4544,
+        0xAC9F,
+        0x4A8E,
+        {0xB2, 0x1B, 0x8A, 0x26, 0x18, 0x0D, 0xB1, 0x3F}
+    };
+
+    IThumbnailCache* cache = nullptr;
+
+    HRESULT hr = CoCreateInstance(
+        thumbnailCacheClsid,
+        nullptr,
+        CLSCTX_INPROC_SERVER,
+        IID_PPV_ARGS(&cache)
+    );
+
+    if (FAILED(hr)) {
+        Wh_Log(L"Creating thumbnail cache failed: 0x%08X",
+               static_cast<UINT>(hr));
+
+        if (mustUninitialize) {
+            CoUninitialize();
+        }
+
+        return FALSE;
+    }
+
+    // IUnknown occupies slots 0–2.
+    // IThumbnailCache::GetThumbnail occupies slot 3.
+    void** vtable = *reinterpret_cast<void***>(cache);
+    void* target = vtable[3];
+
+    // Keep the implementation module loaded for the process lifetime.
+    // Otherwise COM could unload it after releasing our cache instance.
+    HMODULE implementationModule = nullptr;
+
+    BOOL pinned = GetModuleHandleExW(
+        GET_MODULE_HANDLE_EX_FLAG_FROM_ADDRESS |
+            GET_MODULE_HANDLE_EX_FLAG_PIN,
+        reinterpret_cast<LPCWSTR>(target),
+        &implementationModule
+    );
+
+    BOOL hooked = FALSE;
+
+    if (pinned) {
+        hooked = Wh_SetFunctionHook(
+            target,
+            reinterpret_cast<void*>(GetThumbnail_Hook),
+            reinterpret_cast<void**>(&g_originalGetThumbnail)
+        );
+    } else {
+        Wh_Log(L"Couldn't pin thumbnail-cache module: %lu",
+               GetLastError());
+    }
+
+    cache->Release();
+
+    if (mustUninitialize) {
+        CoUninitialize();
+    }
+
+
+    return hooked;
+}