main.c File Reference

#include <apitest.h>
#include <io.h>
#include <windef.h>
#include <winbase.h>
#include <wingdi.h>
#include <winreg.h>
#include <winspool.h>
#include <winsplp.h>
#include "../localspl_apitest.h"
#include <debug.h>

Include dependency graph for main.c:

Go to the source code of this file.

Macros
#define	__ROS_LONG64__
#define	STANDALONE
#define	WIN32_NO_STATUS

Functions
void	func_fpEnumPrinters (void)
void	func_fpGetPrintProcessorDirectory (void)
void	func_fpSetJob (void)
PWSTR	GetDefaultPrinterFromRegistry (VOID)
BOOL	GetLocalsplFuncs (LPPRINTPROVIDOR pp)
PVOID	GetSpoolssFunc (const char *FunctionName)
BOOL WINAPI	DllMain (HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpvReserved)

Variables
const struct test	winetest_testlist []

Macro Definition Documentation

__ROS_LONG64__

#define __ROS_LONG64__

Definition at line 8 of file main.c.

STANDALONE

#define STANDALONE

Definition at line 10 of file main.c.

WIN32_NO_STATUS

#define WIN32_NO_STATUS

Definition at line 13 of file main.c.

Function Documentation

DllMain()

BOOL WINAPI DllMain	(	HINSTANCE	hinstDLL,
		DWORD	fdwReason,
		LPVOID	lpvReserved
	)

Definition at line 159 of file main.c.

{
    char szTestName[150];
    DWORD cbRead;
    FILE* fpStdout;
    HANDLE hCommandPipe;
    int iOldStdout;
 
    // We only want to run our test once when the DLL is injected to the process.
    if (fdwReason != DLL_PROCESS_ATTACH)
        return TRUE;
 
    // Read the test to run from the command pipe.
    hCommandPipe = CreateFileW(COMMAND_PIPE_NAME, GENERIC_READ, 0, NULL, OPEN_EXISTING, 0, NULL);
    if (hCommandPipe == INVALID_HANDLE_VALUE)
    {
        DPRINT("DLL: CreateFileW failed for the command pipe with error %lu!\n", GetLastError());
        return FALSE;
    }
 
    if (!ReadFile(hCommandPipe, szTestName, sizeof(szTestName), &cbRead, NULL))
    {
        DPRINT("DLL: ReadFile failed for the command pipe with error %lu!\n", GetLastError());
        return FALSE;
    }
 
    CloseHandle(hCommandPipe);
 
    // Check if the test name is valid.
    if (!find_test(szTestName))
    {
        DPRINT("DLL: Got invalid test name \"%s\"!\n", szTestName);
        return FALSE;
    }
 
    // Backup our current stdout and set it to the output pipe.
    iOldStdout = _dup(_fileno(stdout));
    fpStdout = _wfreopen(OUTPUT_PIPE_NAME, L"w", stdout);
    setbuf(stdout, NULL);
 
    // Run the test.
    run_test(szTestName);
 
    // Restore stdout to the previous value.
    fclose(fpStdout);
    _dup2(iOldStdout, _fileno(stdout));
 
    // Return FALSE so that our DLL is immediately unloaded.
    return FALSE;
}

func_fpEnumPrinters()

void func_fpEnumPrinters

(

void

)

func_fpGetPrintProcessorDirectory()

void func_fpGetPrintProcessorDirectory

(

void

)

func_fpSetJob()

void func_fpSetJob

(

void

)

GetDefaultPrinterFromRegistry()

PWSTR GetDefaultPrinterFromRegistry

(

VOID

)

We don't link against winspool, so we don't have GetDefaultPrinterW. We can easily implement a similar function ourselves though.

Definition at line 45 of file main.c.

{
    static const WCHAR wszWindowsKey[] = L"Software\\Microsoft\\Windows NT\\CurrentVersion\\Windows";
    static const WCHAR wszDeviceValue[] = L"Device";
 
    DWORD cbNeeded;
    DWORD dwErrorCode;
    HKEY hWindowsKey = NULL;
    PWSTR pwszDevice;
    PWSTR pwszComma;
    PWSTR pwszReturnValue = NULL;
 
    // Open the registry key where the default printer for the current user is stored.
    dwErrorCode = (DWORD)RegOpenKeyExW(HKEY_CURRENT_USER, wszWindowsKey, 0, KEY_READ, &hWindowsKey);
    if (dwErrorCode != ERROR_SUCCESS)
    {
        skip("RegOpenKeyExW failed with status %u!\n", dwErrorCode);
        goto Cleanup;
    }
 
    // Determine the size of the required buffer.
    dwErrorCode = (DWORD)RegQueryValueExW(hWindowsKey, wszDeviceValue, NULL, NULL, NULL, &cbNeeded);
    if (dwErrorCode != ERROR_SUCCESS)
    {
        skip("RegQueryValueExW failed with status %u!\n", dwErrorCode);
        goto Cleanup;
    }
 
    // Allocate it.
    pwszDevice = HeapAlloc(GetProcessHeap(), 0, cbNeeded);
    if (!pwszDevice)
    {
        skip("HeapAlloc failed!\n");
        goto Cleanup;
    }
 
    // Now get the actual value.
    dwErrorCode = RegQueryValueExW(hWindowsKey, wszDeviceValue, NULL, NULL, (PBYTE)pwszDevice, &cbNeeded);
    if (dwErrorCode != ERROR_SUCCESS)
    {
        skip("RegQueryValueExW failed with status %u!\n", dwErrorCode);
        goto Cleanup;
    }
 
    // We get a string "<Printer Name>,winspool,<Port>:".
    // Extract the printer name from it.
    pwszComma = wcschr(pwszDevice, L',');
    if (!pwszComma)
    {
        skip("Found no or invalid default printer: %S!\n", pwszDevice);
        goto Cleanup;
    }
 
    // Return the default printer.
    *pwszComma = 0;
    pwszReturnValue = pwszDevice;
 
Cleanup:
    if (hWindowsKey)
        RegCloseKey(hWindowsKey);
 
    return pwszReturnValue;
}

Referenced by START_TEST().

GetLocalsplFuncs()

BOOL GetLocalsplFuncs

(

LPPRINTPROVIDOR

pp

)

Definition at line 110 of file main.c.

{
    HMODULE hLocalspl;
    PInitializePrintProvidor pfnInitializePrintProvidor;
 
    // Get us a handle to the loaded localspl.dll.
    hLocalspl = GetModuleHandleW(L"localspl");
    if (!hLocalspl)
    {
        skip("GetModuleHandleW failed with error %u!\n", GetLastError());
        return FALSE;
    }
 
    // Get a pointer to its InitializePrintProvidor function.
    pfnInitializePrintProvidor = (PInitializePrintProvidor)GetProcAddress(hLocalspl, "InitializePrintProvidor");
    if (!pfnInitializePrintProvidor)
    {
        skip("GetProcAddress failed with error %u!\n", GetLastError());
        return FALSE;
    }
 
    // Get localspl's function pointers.
    if (!pfnInitializePrintProvidor(pp, sizeof(PRINTPROVIDOR), NULL))
    {
        skip("pfnInitializePrintProvidor failed with error %u!\n", GetLastError());
        return FALSE;
    }
 
    return TRUE;
}

Referenced by START_TEST().

GetSpoolssFunc()

PVOID GetSpoolssFunc

(

const char *

FunctionName

)

Definition at line 142 of file main.c.

{
    HMODULE hSpoolss;
 
    // Get us a handle to the loaded spoolss.dll.
    hSpoolss = GetModuleHandleW(L"spoolss");
    if (!hSpoolss)
    {
        skip("GetModuleHandleW failed with error %u!\n", GetLastError());
        return FALSE;
    }
 
    return GetProcAddress(hSpoolss, FunctionName);
}

Referenced by START_TEST().

Variable Documentation

winetest_testlist

const struct test winetest_testlist[]

Initial value:

=
{
    { "fpEnumPrinters", func_fpEnumPrinters },
    { "fpGetPrintProcessorDirectory", func_fpGetPrintProcessorDirectory },
    { "fpSetJob", func_fpSetJob },
    { 0, 0 }
}

Definition at line 32 of file main.c.