Windows: fix suspend & power off handling (#1039)

* Windows: fix system suspend handling * Windows: fix system shutdown handling * Windows: fix system sleep & shutdown handling * Windows: fix system sleep & shutdown handling (part 2) * Revert "Windows: fix system sleep & shutdown handling (part 2)" This reverts commit 57d366d.
awawa-dev · Jan 11, 2025 · ea1c59e · ea1c59e
1 parent e4a7f7d
commit ea1c59e
Show file tree

Hide file tree

Showing 7 changed files with 59 additions and 0 deletions.
diff --git a/include/systray/Systray.h b/include/systray/Systray.h
@@ -72,6 +72,7 @@ struct SystrayMenu
 
 #ifdef _WIN32
 	HWND SystrayGetWindow();
+	void SystrayAssignQueueHandler(std::function<void(WPARAM wparam)> _queueHandler);
 #endif
 
 
diff --git a/sources/base/AccessManager.cpp b/sources/base/AccessManager.cpp
@@ -6,6 +6,7 @@
 #include <base/AccessManager.h>
 #include <db/AuthTable.h>
 #include <db/MetaTable.h>
+#include <QUuid>
 
 using namespace hyperhdr;
 

diff --git a/sources/db/AuthTable.cpp b/sources/db/AuthTable.cpp
@@ -1,4 +1,5 @@
 #include <QCryptographicHash>
+#include <QUuid>
 
 #include <db/AuthTable.h>
 using namespace hyperhdr;

diff --git a/sources/db/MetaTable.cpp b/sources/db/MetaTable.cpp
@@ -1,5 +1,6 @@
 #include <QCryptographicHash>
 #include <QNetworkInterface>
+#include <QUuid>
 
 #include <db/MetaTable.h>
 

diff --git a/sources/led-drivers/net/DriverNetUdpE131.cpp b/sources/led-drivers/net/DriverNetUdpE131.cpp
@@ -5,6 +5,7 @@
 #endif
 
 #include <QHostInfo>
+#include <QUuid>
 
 // hyperhdr local includes
 #include <led-drivers/net/DriverNetUdpE131.h>

diff --git a/sources/suspend-handler/SuspendHandlerWindows.cpp b/sources/suspend-handler/SuspendHandlerWindows.cpp
@@ -38,20 +38,49 @@
 #include <windows.h>
 #include <wtsapi32.h>
 #include <systray/Systray.h>
+#include <QCoreApplication>
 
 #pragma comment (lib, "WtsApi32.Lib")
 
 namespace
 {
 	HWND handle = nullptr;
+	SuspendHandler* instance = nullptr;
+}
+
+static void SuspendHandlerQueueHandler(WPARAM wparam)
+{
+	if (wparam == 0)
+	{
+		auto instance = QCoreApplication::instance();
+		QUEUE_CALL_0(instance, quit);
+	}
+	else if (instance != nullptr)
+	{
+		MSG message{};
+		QByteArray eventType;
+
+		message.message = WM_POWERBROADCAST;
+		message.wParam = wparam;
+
+		#if (QT_VERSION < QT_VERSION_CHECK(6, 0, 0))
+			long result = 0;
+			BLOCK_CALL_3(instance, nativeEventFilter, QByteArray, eventType, void*, &message, long*, &result);
+		#else
+			qintptr result = 0;
+			BLOCK_CALL_3(instance, nativeEventFilter, QByteArray, eventType, void*, &message, qintptr*, &result);
+		#endif
+	}
 }
 
 SuspendHandler::SuspendHandler(bool sessionLocker):
 	_notifyHandle(NULL),
 	_notifyMonitorHandle(NULL),
 	_sessionLocker(sessionLocker)
 {
+	instance = this;
 	handle = SystrayGetWindow();
+	SystrayAssignQueueHandler(SuspendHandlerQueueHandler);
 	_notifyHandle = RegisterSuspendResumeNotification(handle, DEVICE_NOTIFY_WINDOW_HANDLE);
 
 	if (_notifyHandle == NULL)
@@ -78,6 +107,9 @@ SuspendHandler::SuspendHandler(bool sessionLocker):
 
 SuspendHandler::~SuspendHandler()
 {
+	SystrayAssignQueueHandler(nullptr);
+	instance = nullptr;
+
 	if (_notifyHandle != NULL)
 	{
 		UnregisterSuspendResumeNotification(_notifyHandle);

diff --git a/sources/systray/SystrayWindows.cpp b/sources/systray/SystrayWindows.cpp
@@ -46,18 +46,38 @@ namespace
 	std::list<HICON> icons;
 	std::list<HBITMAP> bitmaps;
 	ULONG_PTR gdiToken = 0;
+	std::function<void(WPARAM suspend)> queueHandler = nullptr;
 }
 
 HWND SystrayGetWindow()
 {
 	return window;
 }
 
+void SystrayAssignQueueHandler(std::function<void(WPARAM wparam)> _queueHandler)
+{
+	queueHandler = _queueHandler;
+}
+
 static LRESULT CALLBACK _tray_wnd_proc(HWND hwnd, UINT msg, WPARAM wparam,
                                        LPARAM lparam)
 {
 	switch (msg)
 	{
+		case WM_POWERBROADCAST:
+			if (queueHandler != nullptr && (wparam == PBT_APMSUSPEND || wparam == PBT_APMRESUMESUSPEND || wparam == PBT_POWERSETTINGCHANGE))
+			{				
+				queueHandler(wparam);
+			}
+			return true;
+
+		case WM_QUERYENDSESSION:
+			if (queueHandler != nullptr && lparam == 0)
+			{
+				queueHandler(0);
+			}
+			return false;
+
 		case WM_CLOSE:
 			DestroyWindow(hwnd);
 			hwnd = nullptr;
@@ -290,6 +310,8 @@ void SystrayUpdate(SystrayMenu *tray)
 
 void SystrayClose()
 {
+	queueHandler = nullptr;
+
 	if (systrayIcon.hIcon != nullptr)
 	{
 		Shell_NotifyIcon(NIM_DELETE, &systrayIcon);