Windows: fix system shutdown handling

awawa-dev · Jan 4, 2025 · 94f754e · 94f754e
1 parent 993bf9d
commit 94f754e
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/sources/suspend-handler/SuspendHandlerWindows.cpp b/sources/suspend-handler/SuspendHandlerWindows.cpp
@@ -38,6 +38,7 @@
 #include <windows.h>
 #include <wtsapi32.h>
 #include <systray/Systray.h>
+#include <QCoreApplication>
 
 #pragma comment (lib, "WtsApi32.Lib")
 
@@ -49,7 +50,12 @@ namespace
 
 static void SuspendHandlerQueueHandler(WPARAM wparam)
 {
-	if (instance != nullptr)
+	if (wparam == 0)
+	{
+		auto instance = QCoreApplication::instance();
+		QUEUE_CALL_0(instance, quit);
+	}
+	else if (instance != nullptr)
 	{
 		MSG message{};
 		QByteArray eventType;

diff --git a/sources/systray/SystrayWindows.cpp b/sources/systray/SystrayWindows.cpp
@@ -71,6 +71,13 @@ static LRESULT CALLBACK _tray_wnd_proc(HWND hwnd, UINT msg, WPARAM wparam,
 			}
 			return true;
 
+		case WM_QUERYENDSESSION:
+			if (queueHandler != nullptr && lparam == 0)
+			{
+				queueHandler(0);
+			}
+			return false;
+
 		case WM_CLOSE:
 			DestroyWindow(hwnd);
 			hwnd = nullptr;