https://git.reactos.org/?p=reactos.git;a=commitdiff;h=67adf7ab786bc436168027...
commit 67adf7ab786bc43616802799ce4f3b49034150ab Author: Hermès Bélusca-Maïto hermes.belusca-maito@reactos.org AuthorDate: Sun Jun 17 21:39:44 2018 +0200 Commit: Hermès Bélusca-Maïto hermes.belusca-maito@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/7b9bfb4136f23655e243bab89fb62b04bdbacc... + */ +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
-
Hermès Bélusca-Maïto