Custom Shutdown Dialog 1.1.1 (#1231)

- Remove unnecessary symbol hook, `LogoffWindowsDialog` is exported by ordinal
ramensoftware · Nov 10, 2024 · 0e44640 · 0e44640
1 parent f396f78
commit 0e44640
Showing 1 changed file with 12 additions and 16 deletions.
diff --git a/mods/custom-shutdown-dialog.wh.cpp b/mods/custom-shutdown-dialog.wh.cpp
@@ -2,7 +2,7 @@
 // @id              custom-shutdown-dialog
 // @name            Custom Shutdown Dialog
 // @description     Override the classic shutdown dialog in Explorer with your own
-// @version         1.1.0
+// @version         1.1.1
 // @author          aubymori
 // @github          https://github.com/aubymori
 // @include         explorer.exe
@@ -84,17 +84,6 @@ WindhawkUtils::SYMBOL_HOOK shutdownuxDllHooks[] = {
         _ShutdownDialogEx_hook,
         false
     }
-};\
-
-WindhawkUtils::SYMBOL_HOOK shell32DllHooks[] = {
-    {
-        {
-            L"LogoffWindowsDialog"
-        },
-        &LogoffWindowsDialog_orig,
-        LogoffWindowsDialog_hook,
-        false
-    }
 };
 
 void LoadSettings(void)
@@ -133,10 +122,17 @@ BOOL Wh_ModInit(void)
         return FALSE;
     }
 
-    if (!WindhawkUtils::HookSymbols(
-        hShell32,
-        shell32DllHooks,
-        ARRAYSIZE(shell32DllHooks)
+    FARPROC LogoffWindowsDialog = GetProcAddress(hShell32, (LPCSTR)54);
+    if (!LogoffWindowsDialog)
+    {
+        Wh_Log(L"Failed to get address of LogoffWindowsDialog");
+        return FALSE;
+    }
+
+    if (!Wh_SetFunctionHook(
+        (void *)LogoffWindowsDialog,
+        (void *)LogoffWindowsDialog_hook,
+        (void **)&LogoffWindowsDialog_orig
     ))
     {
         Wh_Log(L"Failed to hook LogoffWindowsDialog");