winstation.c File Reference

#include <windef.h>
#include <winbase.h>
#include <wincon.h>
#include <winuser.h>
#include <strsafe.h>
#include "resource.h"

Include dependency graph for winstation.c:

Go to the source code of this file.

Classes
struct	_LOG_FILE

Typedefs
typedef struct _LOG_FILE	LOG_FILE
typedef struct _LOG_FILE *	PLOG_FILE

Functions
BOOL	InitLog (OUT PLOG_FILE LogFile, IN LPCWSTR LogFileName, IN PWCHAR pBuffer, IN size_t cbBufferSize)
VOID	CloseLog (IN PLOG_FILE LogFile)
BOOL	WriteToLog (IN PLOG_FILE LogFile, IN LPCVOID Buffer, IN DWORD dwBufferSize)
BOOL	WriteToLogPuts (IN PLOG_FILE LogFile, IN LPCWSTR String)
BOOL	WriteToLogPrintfV (IN PLOG_FILE LogFile, IN LPCWSTR Format, IN va_list args)
BOOL	WriteToLogPrintf (IN PLOG_FILE LogFile, IN LPCWSTR Format,...)
BOOL CALLBACK	EnumDesktopProc (IN LPWSTR lpszDesktop, IN LPARAM lParam)
VOID	DoTest (HWND hWnd)
INT_PTR CALLBACK	About (HWND hDlg, UINT message, WPARAM wParam, LPARAM lParam)
int APIENTRY	wWinMain (HINSTANCE hInstance, HINSTANCE hPrevInstance, LPWSTR lpCmdLine, int nCmdShow)

Typedef Documentation

LOG_FILE

typedef struct _LOG_FILE LOG_FILE

PLOG_FILE

typedef struct _LOG_FILE * PLOG_FILE

Function Documentation

About()

INT_PTR CALLBACK About	(	HWND	hDlg,
		UINT	message,
		WPARAM	wParam,
		LPARAM	lParam
	)

Definition at line 204 of file winstation.c.

{
    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;
}

Referenced by wWinMain().

CloseLog()

VOID CloseLog

(

IN PLOG_FILE

LogFile

)

Definition at line 58 of file winstation.c.

{
    CloseHandle(LogFile->hLogFile);
    ZeroMemory(LogFile, sizeof(*LogFile));
}

Referenced by DoTest().

DoTest()

VOID DoTest

(

HWND

hWnd

)

Definition at line 143 of file winstation.c.

{
    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);
}

Referenced by About(), and ProcessCommandLine().

EnumDesktopProc()

BOOL CALLBACK EnumDesktopProc	(	IN LPWSTR	lpszDesktop,
		IN LPARAM	lParam
	)

Definition at line 125 of file winstation.c.

{
    PLOG_FILE LogFile = (PLOG_FILE)lParam;
 
    WriteToLogPrintf(LogFile, L" :: Found desktop '%s'\r\n", lpszDesktop);
 
    /* Continue the enumeration */
    return TRUE;
}

Referenced by DoTest().

InitLog()

BOOL InitLog	(	OUT PLOG_FILE	LogFile,
		IN LPCWSTR	LogFileName,
		IN PWCHAR	pBuffer,
		IN size_t	cbBufferSize
	)

Definition at line 30 of file winstation.c.

{
    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;
}

WriteToLog()

BOOL WriteToLog	(	IN PLOG_FILE	LogFile,
		IN LPCVOID	Buffer,
		IN DWORD	dwBufferSize
	)

Definition at line 66 of file winstation.c.

{
    return WriteFile(LogFile->hLogFile,
                     Buffer,
                     dwBufferSize,
                     &dwBufferSize, NULL);
}

Referenced by DoTest(), WriteToLogPrintfV(), and WriteToLogPuts().

WriteToLogPrintf()

BOOL WriteToLogPrintf	(	IN PLOG_FILE	LogFile,
		IN LPCWSTR	Format,
			...
	)

Definition at line 103 of file winstation.c.

{
    BOOL bRet;
    va_list args;
 
    va_start(args, Format);
    bRet = WriteToLogPrintfV(LogFile, Format, args);
    va_end(args);
 
    return bRet;
}

Referenced by DoTest(), and EnumDesktopProc().

WriteToLogPrintfV()

BOOL WriteToLogPrintfV	(	IN PLOG_FILE	LogFile,
		IN LPCWSTR	Format,
		IN va_list	args
	)

Definition at line 88 of file winstation.c.

{
    StringCbVPrintfW(LogFile->pBuffer,
                     LogFile->cbBufferSize,
                     Format, args);
 
    return WriteToLog(LogFile,
                      LogFile->pBuffer,
                      wcslen(LogFile->pBuffer) * sizeof(WCHAR));
}

Referenced by WriteToLogPrintf().

WriteToLogPuts()

BOOL WriteToLogPuts	(	IN PLOG_FILE	LogFile,
		IN LPCWSTR	String
	)

Definition at line 78 of file winstation.c.

{
    return WriteToLog(LogFile,
                      String,
                      wcslen(String) * sizeof(WCHAR));
}

Referenced by DoTest().

wWinMain()

int APIENTRY wWinMain	(	HINSTANCE	hInstance,
		HINSTANCE	hPrevInstance,
		LPWSTR	lpCmdLine,
		int	nShowCmd
	)

This file has no copyright assigned and is placed in the Public Domain. This file is part of the w64 mingw-runtime package. No warranty is given; refer to the file DISCLAIMER.PD within this package.

Definition at line 230 of file winstation.c.

{
    UNREFERENCED_PARAMETER(hPrevInstance);
    UNREFERENCED_PARAMETER(lpCmdLine);
 
    return DialogBoxW(hInstance, MAKEINTRESOURCEW(IDD_ABOUTBOX), NULL, About);
}