fpSetJob.c File Reference

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

Include dependency graph for fpSetJob.c:

Go to the source code of this file.

Macros
#define	WIN32_NO_STATUS
#define	MAX_PRINTER_NAME   220

Functions
PWSTR	GetDefaultPrinterFromRegistry (VOID)
BOOL	GetLocalsplFuncs (LPPRINTPROVIDOR pp)
	START_TEST (fpSetJob)

Macro Definition Documentation

MAX_PRINTER_NAME

#define MAX_PRINTER_NAME   220

Definition at line 25 of file fpSetJob.c.

WIN32_NO_STATUS

#define WIN32_NO_STATUS

Definition at line 10 of file fpSetJob.c.

Function Documentation

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 43 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 108 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().

START_TEST()

START_TEST

(

fpSetJob

)

Definition at line 27 of file fpSetJob.c.

{
    HANDLE hPrinter = NULL;
    PRINTPROVIDOR pp;
    PWSTR pwszDefaultPrinter = NULL;
 
    if (!GetLocalsplFuncs(&pp))
        goto Cleanup;
 
    // Verify that fpSetJob returns ERROR_INVALID_HANDLE when nothing is provided.
    ok(!pp.fpSetJob(NULL, 0, 0, NULL, 0), "fpSetJob returns TRUE\n");
    ok(GetLastError() == ERROR_INVALID_HANDLE, "fpSetJob returns error %lu!\n", GetLastError());
 
    // Get the default printer.
    pwszDefaultPrinter = GetDefaultPrinterFromRegistry();
    if (!pwszDefaultPrinter)
    {
        skip("Could not determine the default printer!\n");
        goto Cleanup;
    }
 
    if (!pp.fpOpenPrinter(pwszDefaultPrinter, &hPrinter, NULL))
    {
        skip("Could not open a handle to the default printer, last error is %lu!\n", GetLastError());
        goto Cleanup;
    }
 
    // Verify that fpSetJob returns ERROR_INVALID_PARAMETER if only a printer handle is provided.
    ok(!pp.fpSetJob(hPrinter, 0, 0, NULL, 0), "fpSetJob returns TRUE\n");
    ok(GetLastError() == ERROR_INVALID_PARAMETER, "fpSetJob returns error %lu!\n", GetLastError());
 
Cleanup:
    if (pwszDefaultPrinter)
        HeapFree(GetProcessHeap(), 0, pwszDefaultPrinter);
 
    if (hPrinter)
        pp.fpClosePrinter(hPrinter);
}