From 6186e8891e98f75a0e3f4b019ed928a947fad092 Mon Sep 17 00:00:00 2001
From: aubymori <aubyomori@gmail.com>
Date: Sat, 4 Nov 2023 23:06:54 -0500
Subject: [PATCH 1/3] Custom Shutdown Dialog 1.0.0

---
 mods/custom-shutdown-dialog.wh.cpp | 107 +++++++++++++++++++++++++++++
 1 file changed, 107 insertions(+)
 create mode 100644 mods/custom-shutdown-dialog.wh.cpp

diff --git a/mods/custom-shutdown-dialog.wh.cpp b/mods/custom-shutdown-dialog.wh.cpp
new file mode 100644
index 000000000..e9922b5fb
--- /dev/null
+++ b/mods/custom-shutdown-dialog.wh.cpp
@@ -0,0 +1,107 @@
+// ==WindhawkMod==
+// @id              custom-shutdown-dialog
+// @name            Custom Shutdown Dialog
+// @description     Override the classic shutdown dialog in Explorer with your own
+// @version         1.0.0
+// @author          aubymori
+// @github          https://github.com/aubymori
+// @include         explorer.exe
+// @architecture    x86-64
+// ==/WindhawkMod==
+
+// ==WindhawkModReadme==
+/*
+# Custom Shutdown Dialog
+Override the classic shutdown dialog in Explorer
+which is invoked with `ALT`+`F4` with your own program.
+
+**This mod will only work on Windows 10 or greater and Windhawk v1.4 or greater.**
+*/
+// ==/WindhawkModReadme==
+
+// ==WindhawkModSettings==
+/*
+- exe: C:\Classic\ClassicShutdown\ClassicShutdown.exe
+  $name: Executable
+  $description: Path to executable to open instead of dialog.
+- args: /style classic
+  $name: Arguments
+  $description: Arguments to pass to the executable, if any.
+*/
+// ==/WindhawkModSettings==
+
+#include <windhawk_utils.h>
+#include <versionhelpers.h>
+
+LPCWSTR g_szExe, g_szArgs;
+
+typedef __int64 (* _ShutdownDialogEx_t)(HWND, int, int, UINT);
+_ShutdownDialogEx_t _ShutdownDialogEx_orig;
+__int64 _ShutdownDialogEx_hook(
+    HWND hWndParent,
+    int  i1,
+    int  i2,
+    UINT i3
+)
+{
+    ShellExecuteW(
+        hWndParent,
+        L"open",
+        g_szExe,
+        g_szArgs,
+        NULL,
+        SW_NORMAL
+    );
+    return 0;
+}
+
+void LoadSettings(void)
+{
+    g_szExe = Wh_GetStringSetting(L"exe");
+    g_szArgs = Wh_GetStringSetting(L"args");
+}
+
+BOOL Wh_ModInit(void)
+{
+    if (!IsWindows10OrGreater())
+    {
+        Wh_Log(L"This mod was designed for Windows 10 and up.");
+        return FALSE;
+    }
+
+    LoadSettings();
+
+    HMODULE hShutdownUx = LoadLibraryW(L"shutdownux.dll");
+    if (!hShutdownUx)
+    {
+        Wh_Log(L"Failed to load shutdownux.dll");
+        return FALSE;
+    }
+
+    WindhawkUtils::SYMBOL_HOOK hook = {
+        {
+            L"static  _ShutdownDialogEx()"
+        },
+        &_ShutdownDialogEx_orig,
+        _ShutdownDialogEx_hook
+    };
+
+    if (!HookSymbols(hShutdownUx, &hook, 1))
+    {
+        Wh_Log(L"Failed to hook _ShutdownDialogEx");
+        return FALSE;
+    }
+
+    return TRUE;
+}
+
+void Wh_ModSettingsChanged(void)
+{
+    LoadSettings();
+}
+
+void Wh_ModUninit(void)
+{
+    Wh_FreeStringSetting(g_szExe);
+    Wh_FreeStringSetting(g_szArgs);
+}
\ No newline at end of file

From 98bcd7d3ee00b8a82cf1bf1a75b3e47f8e5d2f41 Mon Sep 17 00:00:00 2001
From: aubymori <aubyomori@gmail.com>
Date: Sat, 4 Nov 2023 23:23:54 -0500
Subject: [PATCH 2/3] Custom Shutdown Dialog: Free settings strsings upon
 settings change

---
 mods/custom-shutdown-dialog.wh.cpp | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/mods/custom-shutdown-dialog.wh.cpp b/mods/custom-shutdown-dialog.wh.cpp
index e9922b5fb..b7f32fe8b 100644
--- a/mods/custom-shutdown-dialog.wh.cpp
+++ b/mods/custom-shutdown-dialog.wh.cpp
@@ -97,6 +97,8 @@ BOOL Wh_ModInit(void)
 
 void Wh_ModSettingsChanged(void)
 {
+    Wh_FreeStringSetting(g_szExe);
+    Wh_FreeStringSetting(g_szArgs);
     LoadSettings();
 }
 

From cb8f909b6edb1d448ea79f8735fe95bb08f9b6a7 Mon Sep 17 00:00:00 2001
From: aubymori <aubyomori@gmail.com>
Date: Sat, 4 Nov 2023 23:40:53 -0500
Subject: [PATCH 3/3] Custom Shutdown Dialog: Use StringSetting class

---
 mods/custom-shutdown-dialog.wh.cpp | 14 +++-----------
 1 file changed, 3 insertions(+), 11 deletions(-)

diff --git a/mods/custom-shutdown-dialog.wh.cpp b/mods/custom-shutdown-dialog.wh.cpp
index b7f32fe8b..a68e80b4f 100644
--- a/mods/custom-shutdown-dialog.wh.cpp
+++ b/mods/custom-shutdown-dialog.wh.cpp
@@ -33,7 +33,7 @@ which is invoked with `ALT`+`F4` with your own program.
 #include <windhawk_utils.h>
 #include <versionhelpers.h>
 
-LPCWSTR g_szExe, g_szArgs;
+WindhawkUtils::StringSetting g_szExe, g_szArgs;
 
 typedef __int64 (* _ShutdownDialogEx_t)(HWND, int, int, UINT);
 _ShutdownDialogEx_t _ShutdownDialogEx_orig;
@@ -57,8 +57,8 @@ __int64 _ShutdownDialogEx_hook(
 
 void LoadSettings(void)
 {
-    g_szExe = Wh_GetStringSetting(L"exe");
-    g_szArgs = Wh_GetStringSetting(L"args");
+    g_szExe = WindhawkUtils::StringSetting::make(L"exe");
+    g_szArgs = WindhawkUtils::StringSetting::make(L"args");
 }
 
 BOOL Wh_ModInit(void)
@@ -97,13 +97,5 @@ BOOL Wh_ModInit(void)
 
 void Wh_ModSettingsChanged(void)
 {
-    Wh_FreeStringSetting(g_szExe);
-    Wh_FreeStringSetting(g_szArgs);
     LoadSettings();
-}
-
-void Wh_ModUninit(void)
-{
-    Wh_FreeStringSetting(g_szExe);
-    Wh_FreeStringSetting(g_szArgs);
 }
\ No newline at end of file