Eradicate Immersive Menus 1.0.0

mods/eradicate-immersive-menus.wh.cpp

@@ -0,0 +1,353 @@
+// ==WindhawkMod==
+// @id              eradicate-immersive-menus
+// @name            Eradicate Immersive Menus
+// @description     Gets rid of immersive menus system-wide
+// @version         1.0.0
+// @author          aubymori
+// @github          https://github.com/aubymori
+// @include         *
+// @exclude         windhawk.exe
+// @exclude         dwm.exe
+// @exclude         ShellExperienceHost.exe
+// @exclude         wlanext.exe
+// @exclude         svchost.exe
+// @exclude         SearchIndexer.exe
+// @exclude         consent.exe
+// @compilerOptions -luser32
+// ==/WindhawkMod==
+
+// ==WindhawkModReadme==
+/*
+# Eradicate Immersive Menus
+Gets rid of immersive menus system-wide. This mod should work on every Windows 10 version, from 1507
+all the way up to 22H2. However, **this mod will only work on Windhawk v1.4 and greater.**
+
+## Preview
+
+*From left to right: Explorer, Taskbar, Explorer (classic), Taskbar (classic)*
+
+**Before**:
+
+![Before](https://raw.githubusercontent.com/aubymori/images/main/eradicate-immersive-menus-before.png)
+
+**After**:
+![After](https://raw.githubusercontent.com/aubymori/images/main/eradicate-immersive-menus-after.png)
+
+# Common problems
+
+## Freezing
+
+This mod will take **a while** to load on the first run, and if you disable and then enable it while your system is all
+already up and running with many software open.
+
+## Stuck on "Initializing..."
+
+If ever a certain process gets stuck on "Initializing...", and the "Mod tasks in progress" window does not go away,
+restart Windhawk. If that does not work, simply add it to the exclusion list in the Advanced tab of this mod.
+
+## Failed to download symbols:
+
+Sometimes, Windhawk may fail to download the symbols for certain system files. This doesn't necessarily mean that your version
+of Windows is incompatible, but it means that whatever application had the mod hooked cannot connect to the internet (which is how Windhawk
+attempts to download symbols).
+
+You can work around this issue by manually downloading the symbols and dropping them in the Windhawk symbols
+folder. In order to do this, use a tool such as [PDB Downloader](https://github.com/rajkumar-rangaraj/PDB-Downloader/releases)
+and download the symbols for any of the missing files from the following list.
+
+- `ExplorerFrame.dll` (x86)
+- `shell32.dll` (x86)
+- `pnidui.dll`
+- `SndVolSSO.dll`
+- `SecurityHealthSSO.dll`
+- `twinui.dll`
+- `twinui.pcshell.dll`
+
+The DLLs themselves are located in `%SystemRoot%\System32`. Of these DLLs, the ones with (x86) at the end have a 32-bit
+variant in `%SystemRoot%\SysWOW64`, for open/save dialogs in 32-bit applications, so you may need to download symbols for
+these as well.
+
+The standard (non-portable) version of Windhawk stores symbols in `C:\ProgramData\Windhawk\Engine\Symbols`.
+
+If you encounter this error, a message box will be displayed warning you of the failure.
+*/
+// ==/WindhawkModReadme==
+
+// ==WindhawkModSettings==
+/*
+- explorer: true
+  $name: Apply to Explorer
+  $description: Applies to Explorer itself, including desktop, navigation pane, main Explorer view, open/save dialogs, and the main taskbar menu.
+- nosettingsicon: true
+  $name: No "Taskbar settings" icon
+  $description: Removes the icon from the "Taskbar settings" entry in the taskbar menu
+- sound: true
+  $name: Apply to Volume tray icon
+- network: true
+  $name: Apply to Network tray icon
+- defender: true
+  $name: Apply to Windows Security
+*/
+// ==/WindhawkModSettings==
+
+#include <windhawk_api.h>
+#include <windhawk_utils.h>
+
+struct {
+    BOOL explorer;
+    /* Tray icons are always 64-bit */
+    #ifdef _WIN64
+    BOOL sound;
+    BOOL network;
+    BOOL defender;
+    BOOL nosettingsicon;
+    #endif
+} settings;
+
+/* ImmersiveContextMenuHelper::CanApplyOwnerDrawToMenu */
+typedef bool (__fastcall *ICMH_CAODTM_t)(HMENU, HWND);
+
+ICMH_CAODTM_t ICMH_CAODTM_orig_shell32;
+ICMH_CAODTM_t ICMH_CAODTM_orig_ExplorerFrame;
+/* Explorer and tray icons are always 64-bit */
+#ifdef _WIN64
+ICMH_CAODTM_t ICMH_CAODTM_orig_explorer;
+ICMH_CAODTM_t ICMH_CAODTM_orig_twinui;
+ICMH_CAODTM_t ICMH_CAODTM_orig_twinui_pcshell;
+ICMH_CAODTM_t ICMH_CAODTM_orig_SndVolSSO;
+ICMH_CAODTM_t ICMH_CAODTM_orig_pnidui;
+ICMH_CAODTM_t ICMH_CAODTM_orig_SecurityHealthSSO;
+#endif
+
+/** 
+  * __fastcall is needed, or else stack corruption occurs
+  * in x86-32 applications.
+  */
+bool __fastcall ICMH_CAODTM_hook(
+    HMENU hMenu,
+    HWND hWnd
+)
+{
+    return false;
+}
+
+/* Prevent "Taskbar settings" icon from being set */
+#ifdef _WIN64
+using SetMenuItemInfoW_t = decltype(&SetMenuItemInfoW);
+SetMenuItemInfoW_t SetMenuItemInfoW_orig;
+BOOL SetMenuItemInfoW_hook(
+    HMENU            hMenu,
+    UINT             item,
+    BOOL             fByPosition,
+    LPCMENUITEMINFOW lpmii
+)
+{
+    /* homestuck reference??? */
+    if (settings.nosettingsicon && (!fByPosition && item == 413))
+    {
+        return TRUE;
+    }
+
+    return SetMenuItemInfoW_orig(
+        hMenu,
+        item,
+        fByPosition,
+        lpmii
+    );
+}
+#endif
+
+/**
+  * Hooks ImmersiveContextMenuHelper::CanApplyOwnerDrawToMenu for a given DLL.
+  * There are many DLLs that have this function, so each one needs to be accounted for.
+  */
+inline BOOL HookICMH_CAODTM(LPCWSTR lpDll, ICMH_CAODTM_t *pOrig)
+{
+    HMODULE hModule;
+
+    /* If lpDll is NULL, patch the current process instead */
+    if (lpDll == NULL)
+    {
+        hModule = GetModuleHandleW(NULL);
+    }
+    else
+    {
+        hModule = LoadLibraryW(lpDll);
+    }
+
+    if (!hModule)
+    {
+        Wh_Log(L"Failed to load %s", lpDll);
+        return FALSE;
+    }
+
+    WindhawkUtils::SYMBOL_HOOK hook = {
+        {
+            L"bool "
+#ifdef _WIN64
+            L"__cdecl"
+#else
+            L"__stdcall"
+#endif
+            L" ImmersiveContextMenuHelper::CanApplyOwnerDrawToMenu(struct HMENU__ *,struct HWND__ *)"
+        },
+        pOrig,
+        ICMH_CAODTM_hook,
+        true
+    };
+
+    bool bHookSuccess = WindhawkUtils::HookSymbols(
+        hModule,
+        &hook,
+        1
+    );
+
+    if (bHookSuccess)
+    {
+        return TRUE;
+    }
+
+    Wh_Log(
+        L"Failed to hook ImmersiveContextMenuHelper::CanApplyOwnerDrawToMenu in %s. "
+        L"This may be a result of symbols failing to download. "
+        L"Please see the \"Details\" page of the mod for instructions on how to download the symbols manually.",
+        lpDll ? lpDll : L"explorer.exe"
+    );
+    return FALSE;
+}
+
+void LoadSettings()
+{
+    settings.explorer = Wh_GetIntSetting(L"explorer", TRUE);
+    /* Tray icons are always 64-bit */
+#ifdef _WIN64
+    settings.sound          = Wh_GetIntSetting(L"sound",          TRUE);
+    settings.network        = Wh_GetIntSetting(L"network",        TRUE);
+    settings.defender       = Wh_GetIntSetting(L"defender",       TRUE);
+    settings.nosettingsicon = Wh_GetIntSetting(L"nosettingsicon", TRUE);
+#endif
+}
+
+BOOL Wh_ModInit()
+{
+    LoadSettings();
+
+    /* Explorer is always 64-bit */
+#ifdef _WIN64
+    WCHAR szProcessPath[MAX_PATH];
+    GetModuleFileNameW(NULL, szProcessPath, MAX_PATH);
+    BOOL bIsExplorer = (wcsicmp(szProcessPath, L"C:\\Windows\\explorer.exe") == 0);
+
+    if (settings.nosettingsicon)
+    {
+        Wh_SetFunctionHook(
+            (void *)SetMenuItemInfoW,
+            (void *)SetMenuItemInfoW_hook,
+            (void **)&SetMenuItemInfoW_orig
+        );
+    }
+#endif
+
+    if (settings.explorer
+#ifdef _WIN64
+    && wcsicmp(szProcessPath, L"C:\\Windows\\System32\\SecurityHealthSystray.exe")
+#endif
+    )
+    {
+        if (!HookICMH_CAODTM(
+            L"shell32.dll",
+            &ICMH_CAODTM_orig_shell32
+        ))
+        {
+            return FALSE;
+        }
+
+
+        if (!HookICMH_CAODTM(
+            L"ExplorerFrame.dll",
+            &ICMH_CAODTM_orig_ExplorerFrame
+        ))
+        {
+            return FALSE;
+        }
+
+        /* Explorer itself is always 64-bit */
+#ifdef _WIN64
+        if (bIsExplorer)
+        {
+            if (!HookICMH_CAODTM(
+                NULL,
+                &ICMH_CAODTM_orig_explorer
+            ))
+            {
+                return FALSE;
+            }
+
+            if (!HookICMH_CAODTM(
+                L"twinui.dll",
+                &ICMH_CAODTM_orig_twinui
+            ))
+            {
+                return FALSE;
+            }
+
+            if (!HookICMH_CAODTM(
+                L"twinui.pcshell.dll",
+                &ICMH_CAODTM_orig_twinui_pcshell
+            ))
+            {
+                return FALSE;
+            }
+        }
+#endif
+    }
+
+    /* Tray icons are always 64-bit */
+#ifdef _WIN64
+    if (bIsExplorer)
+    {
+        if (settings.sound)
+        {
+            if (!HookICMH_CAODTM(
+                L"SndVolSSO.dll",
+                &ICMH_CAODTM_orig_SndVolSSO
+            ))
+            {
+                return FALSE;
+            }
+        }
+
+        if (settings.network)
+        {
+            if (!HookICMH_CAODTM(
+                L"pnidui.dll",
+                &ICMH_CAODTM_orig_pnidui
+            ))
+            {
+                return FALSE;
+            }
+        }
+    }
+
+    if (settings.defender
+    &&  0 == wcscmp(wcslwr(szProcessPath), L"c:\\windows\\system32\\securityhealthsystray.exe"))
+    {
+        if (!HookICMH_CAODTM(
+            L"SecurityHealthSSO.dll",
+            &ICMH_CAODTM_orig_SecurityHealthSSO
+        ))
+        {
+            return FALSE;
+        }
+    }
+#endif
+
+    return TRUE;
+}
+
+BOOL Wh_ModSettingsChanged(BOOL *bReload)
+{
+    *bReload = TRUE;
+    LoadSettings();
+    return TRUE;
+}