22 Jul
2018
22 Jul
'18
5:03 p.m.
https://git.reactos.org/?p=reactos.git;a=commitdiff;h=67adf7ab786bc43616802…
commit 67adf7ab786bc43616802799ce4f3b49034150ab
Author: Hermès Bélusca-Maïto <hermes.belusca-maito(a)reactos.org>
AuthorDate: Sun Jun 17 21:39:44 2018 +0200
Commit: Hermès Bélusca-Maïto <hermes.belusca-maito(a)reactos.org>
CommitDate: Sun Jul 22 19:03:17 2018 +0200
[ROSTESTS] Add an interactive test for window stations. See also PR #621.
---
modules/rostests/win32/user32/CMakeLists.txt | 1 +
.../win32/user32/winstation/CMakeLists.txt | 5 +
.../rostests/win32/user32/winstation/resource.h | 2 +
.../rostests/win32/user32/winstation/winstation.c | 239 +++++++++++++++++++++
.../rostests/win32/user32/winstation/winstation.rc | 15 ++
5 files changed, 262 insertions(+)
diff --git a/modules/rostests/win32/user32/CMakeLists.txt
b/modules/rostests/win32/user32/CMakeLists.txt
index 8298a919d3..d1d83974bd 100644
--- a/modules/rostests/win32/user32/CMakeLists.txt
+++ b/modules/rostests/win32/user32/CMakeLists.txt
@@ -1,3 +1,4 @@
add_subdirectory(biditext)
add_subdirectory(paintdesktop)
add_subdirectory(sysicon)
+add_subdirectory(winstation)
diff --git a/modules/rostests/win32/user32/winstation/CMakeLists.txt
b/modules/rostests/win32/user32/winstation/CMakeLists.txt
new file mode 100644
index 0000000000..88d6ea8173
--- /dev/null
+++ b/modules/rostests/win32/user32/winstation/CMakeLists.txt
@@ -0,0 +1,5 @@
+
+add_executable(winstation winstation.c winstation.rc)
+set_module_type(winstation win32gui UNICODE)
+add_importlibs(winstation user32 msvcrt kernel32)
+add_rostests_file(TARGET winstation SUBDIR suppl)
diff --git a/modules/rostests/win32/user32/winstation/resource.h
b/modules/rostests/win32/user32/winstation/resource.h
new file mode 100644
index 0000000000..795deabfbd
--- /dev/null
+++ b/modules/rostests/win32/user32/winstation/resource.h
@@ -0,0 +1,2 @@
+#define IDD_ABOUTBOX 101
+#define IDC_STATIC -1
diff --git a/modules/rostests/win32/user32/winstation/winstation.c
b/modules/rostests/win32/user32/winstation/winstation.c
new file mode 100644
index 0000000000..f2434dbad0
--- /dev/null
+++ b/modules/rostests/win32/user32/winstation/winstation.c
@@ -0,0 +1,239 @@
+/*
+ * PROJECT: ReactOS Tests
+ * LICENSE: GPL-2.0+ (https://spdx.org/licenses/GPL-2.0+)
+ * PURPOSE: Tests some aspects of the Window Station API.
+ * COPYRIGHT: Copyright 2018 Hermes Belusca-Maito
+ */
+
+#include <windef.h>
+#include <winbase.h>
+#include <wincon.h>
+#include <winuser.h>
+
+#include <strsafe.h>
+
+#include "resource.h"
+
+
+/*
+ * Logging facilities
+ */
+
+typedef struct _LOG_FILE
+{
+ HANDLE hLogFile;
+ PWCHAR pBuffer;
+ size_t cbBufferSize;
+} LOG_FILE, *PLOG_FILE;
+
+BOOL
+InitLog(
+ OUT PLOG_FILE LogFile,
+ IN LPCWSTR LogFileName,
+ IN PWCHAR pBuffer,
+ IN size_t cbBufferSize)
+{
+ HANDLE hLogFile;
+
+ ZeroMemory(LogFile, sizeof(*LogFile));
+
+ hLogFile = CreateFileW(LogFileName,
+ GENERIC_WRITE,
+ FILE_SHARE_READ,
+ NULL,
+ OPEN_ALWAYS,
+ FILE_ATTRIBUTE_NORMAL,
+ NULL);
+ if (hLogFile == INVALID_HANDLE_VALUE)
+ return FALSE;
+
+ LogFile->hLogFile = hLogFile;
+ LogFile->pBuffer = pBuffer;
+ LogFile->cbBufferSize = cbBufferSize;
+
+ return TRUE;
+}
+
+VOID
+CloseLog(
+ IN PLOG_FILE LogFile)
+{
+ CloseHandle(LogFile->hLogFile);
+ ZeroMemory(LogFile, sizeof(*LogFile));
+}
+
+BOOL
+WriteToLog(
+ IN PLOG_FILE LogFile,
+ IN LPCVOID Buffer,
+ IN DWORD dwBufferSize)
+{
+ return WriteFile(LogFile->hLogFile,
+ Buffer,
+ dwBufferSize,
+ &dwBufferSize, NULL);
+}
+
+BOOL
+WriteToLogPuts(
+ IN PLOG_FILE LogFile,
+ IN LPCWSTR String)
+{
+ return WriteToLog(LogFile,
+ String,
+ wcslen(String) * sizeof(WCHAR));
+}
+
+BOOL
+WriteToLogPrintfV(
+ IN PLOG_FILE LogFile,
+ IN LPCWSTR Format,
+ IN va_list args)
+{
+ StringCbVPrintfW(LogFile->pBuffer,
+ LogFile->cbBufferSize,
+ Format, args);
+
+ return WriteToLog(LogFile,
+ LogFile->pBuffer,
+ wcslen(LogFile->pBuffer) * sizeof(WCHAR));
+}
+
+BOOL
+WriteToLogPrintf(
+ IN PLOG_FILE LogFile,
+ IN LPCWSTR Format,
+ ...)
+{
+ BOOL bRet;
+ va_list args;
+
+ va_start(args, Format);
+ bRet = WriteToLogPrintfV(LogFile, Format, args);
+ va_end(args);
+
+ return bRet;
+}
+
+
+/*
+ * Window Station tests
+ */
+
+BOOL
+CALLBACK
+EnumDesktopProc(
+ IN LPWSTR lpszDesktop,
+ IN LPARAM lParam)
+{
+ PLOG_FILE LogFile = (PLOG_FILE)lParam;
+
+ WriteToLogPrintf(LogFile, L" :: Found desktop '%s'\r\n",
lpszDesktop);
+
+ /* Continue the enumeration */
+ return TRUE;
+}
+
+/*
+ * This test inspects the same window station aspects that are used in the
+ * Cygwin fhandler_console.cc!fhandler_console::create_invisible_console()
+ * function, see:
+ *
https://github.com/cygwin/cygwin/blob/7b9bfb4136f23655e243bab89fb62b04bdbac…
+ */
+VOID DoTest(HWND hWnd)
+{
+ HWINSTA hWinSta;
+ LPCWSTR lpszWinSta = L"Test-WinSta";
+ BOOL bIsItOk;
+ LOG_FILE LogFile;
+ WCHAR szBuffer[2048];
+
+ bIsItOk = InitLog(&LogFile, L"test_winsta.log", szBuffer,
sizeof(szBuffer));
+ if (!bIsItOk)
+ {
+ MessageBoxW(hWnd, L"Could not create the log file, stopping test
now...", L"Error", MB_ICONERROR | MB_OK);
+ return;
+ }
+
+ /* Switch output to UTF-16 (little endian) */
+ WriteToLog(&LogFile, "\xFF\xFE", 2);
+
+ WriteToLogPrintf(&LogFile, L"Creating Window Station '%s'\r\n",
lpszWinSta);
+ hWinSta = CreateWindowStationW(lpszWinSta, 0, WINSTA_ALL_ACCESS, NULL);
+ WriteToLogPrintf(&LogFile, L"--> Returned handle 0x%p ; last error:
%lu\r\n", hWinSta, GetLastError());
+
+ if (!hWinSta)
+ {
+ WriteToLogPuts(&LogFile, L"\r\nHandle is NULL, cannot proceed further,
stopping the test!\r\n\r\n");
+ return;
+ }
+
+ WriteToLogPrintf(&LogFile, L"Enumerate desktops on Window Station
'%s' (0x%p) (before process attach)\r\n", lpszWinSta, hWinSta);
+ bIsItOk = EnumDesktopsW(hWinSta, EnumDesktopProc, (LPARAM)&LogFile);
+ WriteToLogPrintf(&LogFile, L"--> Returned %s ; last error:
%lu\r\n",
+ (bIsItOk ? L"success" : L"failure"),
GetLastError());
+
+ WriteToLogPrintf(&LogFile, L"Setting current process to Window Station
'%s' (0x%p)\r\n", lpszWinSta, hWinSta);
+ bIsItOk = SetProcessWindowStation(hWinSta);
+ WriteToLogPrintf(&LogFile, L"--> Returned %s ; last error:
%lu\r\n",
+ (bIsItOk ? L"success" : L"failure"),
GetLastError());
+
+ WriteToLogPrintf(&LogFile, L"Enumerate desktops on Window Station
'%s' (0x%p) (after process attach, before allocating console)\r\n",
lpszWinSta, hWinSta);
+ bIsItOk = EnumDesktopsW(hWinSta, EnumDesktopProc, (LPARAM)&LogFile);
+ WriteToLogPrintf(&LogFile, L"--> Returned %s ; last error:
%lu\r\n",
+ (bIsItOk ? L"success" : L"failure"),
GetLastError());
+
+ WriteToLogPrintf(&LogFile, L"Allocating a new console on Window Station
'%s' (0x%p)\r\n", lpszWinSta, hWinSta);
+ bIsItOk = AllocConsole();
+ WriteToLogPrintf(&LogFile, L"--> Returned %s ; last error:
%lu\r\n",
+ (bIsItOk ? L"success" : L"failure"),
GetLastError());
+
+ WriteToLogPrintf(&LogFile, L"Enumerate desktops on Window Station
'%s' (0x%p) (after allocating console)\r\n", lpszWinSta, hWinSta);
+ bIsItOk = EnumDesktopsW(hWinSta, EnumDesktopProc, (LPARAM)&LogFile);
+ WriteToLogPrintf(&LogFile, L"--> Returned %s ; last error:
%lu\r\n",
+ (bIsItOk ? L"success" : L"failure"),
GetLastError());
+
+ WriteToLogPrintf(&LogFile, L"Now closing Window Station '%s'
(0x%p)\r\n", lpszWinSta, hWinSta);
+ bIsItOk = CloseWindowStation(hWinSta);
+ WriteToLogPrintf(&LogFile, L"--> Returned %s ; last error:
%lu\r\n\r\n",
+ (bIsItOk ? L"success" : L"failure"),
GetLastError());
+
+ CloseLog(&LogFile);
+}
+
+INT_PTR CALLBACK About(HWND hDlg, UINT message, WPARAM wParam, LPARAM lParam)
+{
+ UNREFERENCED_PARAMETER(lParam);
+ switch (message)
+ {
+ case WM_INITDIALOG:
+ return (INT_PTR)TRUE;
+
+ case WM_COMMAND:
+ switch (LOWORD(wParam))
+ {
+ case IDOK:
+ DoTest(hDlg);
+ EndDialog(hDlg, LOWORD(wParam));
+ break;
+
+ case IDCANCEL:
+ default:
+ EndDialog(hDlg, LOWORD(wParam));
+ break;
+ }
+ return (INT_PTR)TRUE;
+ }
+ return (INT_PTR)FALSE;
+}
+
+int APIENTRY wWinMain(HINSTANCE hInstance,
+ HINSTANCE hPrevInstance,
+ LPWSTR lpCmdLine,
+ int nCmdShow)
+{
+ UNREFERENCED_PARAMETER(hPrevInstance);
+ UNREFERENCED_PARAMETER(lpCmdLine);
+
+ return DialogBoxW(hInstance, MAKEINTRESOURCEW(IDD_ABOUTBOX), NULL, About);
+}
diff --git a/modules/rostests/win32/user32/winstation/winstation.rc
b/modules/rostests/win32/user32/winstation/winstation.rc
new file mode 100644
index 0000000000..0e9e05ffe0
--- /dev/null
+++ b/modules/rostests/win32/user32/winstation/winstation.rc
@@ -0,0 +1,15 @@
+
+#include <windef.h>
+#include <winuser.h>
+#include "resource.h"
+
+LANGUAGE LANG_NEUTRAL, SUBLANG_NEUTRAL
+
+IDD_ABOUTBOX DIALOGEX 0, 0, 170, 62
+STYLE DS_SETFONT | DS_MODALFRAME | DS_FIXEDSYS | DS_CENTER | WS_POPUP | WS_CAPTION |
WS_SYSMENU
+CAPTION "Test WindowStation"
+FONT 8, "MS Shell Dlg", 0, 0, 0x1
+BEGIN
+ CTEXT "Test for WindowStation\nTo start the test, press on the button
below.",IDC_STATIC,7,7,156,20,SS_NOPREFIX
+ DEFPUSHBUTTON "Start!",IDOK,7,32,156,23,WS_GROUP
+END