1 files changed, 62 insertions, 0 deletions
diff --git a/winpr/libwinpr/wtsapi/test/TestWtsApiSessionNotification.c b/winpr/libwinpr/wtsapi/test/TestWtsApiSessionNotification.c
new file mode 100644
index 0000000..e83c27b
--- /dev/null
+++ b/winpr/libwinpr/wtsapi/test/TestWtsApiSessionNotification.c
@@ -0,0 +1,62 @@
+
+#include <winpr/crt.h>
+#include <winpr/error.h>
+#include <winpr/wtsapi.h>
+#include <winpr/environment.h>
+
+int TestWtsApiSessionNotification(int argc, char* argv[])
+{
+	HWND hWnd = NULL;
+	BOOL bSuccess = 0;
+	DWORD dwFlags = 0;
+
+	WINPR_UNUSED(argc);
+	WINPR_UNUSED(argv);
+
+#ifndef _WIN32
+	if (!GetEnvironmentVariableA("WTSAPI_LIBRARY", NULL, 0))
+	{
+		printf("%s: No RDS environment detected, skipping test\n", __func__);
+		return 0;
+	}
+#else
+	/* We create a message-only window and use the predefined class name "STATIC" for simplicity */
+	hWnd = CreateWindowA("STATIC", "TestWtsApiSessionNotification", 0, 0, 0, 0, 0, HWND_MESSAGE,
+	                     NULL, NULL, NULL);
+	if (!hWnd)
+	{
+		printf("%s: error creating message-only window: %" PRIu32 "\n", __func__, GetLastError());
+		return -1;
+	}
+#endif
+
+	dwFlags = NOTIFY_FOR_ALL_SESSIONS;
+
+	bSuccess = WTSRegisterSessionNotification(hWnd, dwFlags);
+
+	if (!bSuccess)
+	{
+		printf("%s: WTSRegisterSessionNotification failed: %" PRIu32 "\n", __func__,
+		       GetLastError());
+		return -1;
+	}
+
+	bSuccess = WTSUnRegisterSessionNotification(hWnd);
+
+#ifdef _WIN32
+	if (hWnd)
+	{
+		DestroyWindow(hWnd);
+		hWnd = NULL;
+	}
+#endif
+
+	if (!bSuccess)
+	{
+		printf("%s: WTSUnRegisterSessionNotification failed: %" PRIu32 "\n", __func__,
+		       GetLastError());
+		return -1;
+	}
+
+	return 0;
+}