fpGetPrintProcessorDirectory.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 fpGetPrintProcessorDirectory.c:

Go to the source code of this file.

Macros
#define	WIN32_NO_STATUS

Typedefs
typedef BOOL(WINAPI *	PGetPrintProcessorDirectoryW) (LPWSTR, LPWSTR, DWORD, LPBYTE, DWORD, LPDWORD)

Functions
BOOL	GetLocalsplFuncs (LPPRINTPROVIDOR pp)
PVOID	GetSpoolssFunc (const char *FunctionName)
	START_TEST (fpGetPrintProcessorDirectory)

Macro Definition Documentation

WIN32_NO_STATUS

#define WIN32_NO_STATUS

Definition at line 10 of file fpGetPrintProcessorDirectory.c.

Typedef Documentation

PGetPrintProcessorDirectoryW

typedef BOOL(WINAPI * PGetPrintProcessorDirectoryW) (LPWSTR, LPWSTR, DWORD, LPBYTE, DWORD, LPDWORD)

Definition at line 21 of file fpGetPrintProcessorDirectory.c.

Function Documentation

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

START_TEST()

START_TEST

(

fpGetPrintProcessorDirectory

)

Definition at line 25 of file fpGetPrintProcessorDirectory.c.

{
    DWORD cbNeeded;
    DWORD cbTemp;
    //DWORD dwReturned;
    PGetPrintProcessorDirectoryW pGetPrintProcessorDirectoryW;
    PRINTPROVIDOR pp;
    PWSTR pwszBuffer;

    if (!GetLocalsplFuncs(&pp))
        return;

    pGetPrintProcessorDirectoryW = GetSpoolssFunc("GetPrintProcessorDirectoryW");
    if (!pGetPrintProcessorDirectoryW)
        return;

    // In contrast to GetPrintProcessorDirectoryW, fpGetPrintProcessorDirectory needs an environment and doesn't just accept NULL.
    SetLastError(0xDEADBEEF);
    ok(!pp.fpGetPrintProcessorDirectory(NULL, NULL, 0, NULL, 0, NULL), "fpGetPrintProcessorDirectory returns TRUE!\n");
    ok(GetLastError() == ERROR_INVALID_ENVIRONMENT, "fpGetPrintProcessorDirectory returns error %lu!\n", GetLastError());

    // Try with an invalid environment as well.
    SetLastError(0xDEADBEEF);
    ok(!pp.fpGetPrintProcessorDirectory(NULL, L"invalid", 0, NULL, 0, NULL), "fpGetPrintProcessorDirectory returns TRUE!\n");
    ok(GetLastError() == ERROR_INVALID_ENVIRONMENT, "fpGetPrintProcessorDirectory returns error %lu!\n", GetLastError());

    // This test corrupts something inside spoolsv so that it's only runnable once without restarting spoolsv. Therefore it's disabled.
#if 0
    // Now provide a valid environment and prove that it is checked case-insensitively.
    // In contrast to GetPrintProcessorDirectoryW, the level isn't the next thing checked here, but fpGetPrintProcessorDirectory
    // already tries to access the non-supplied pcbNeeded variable.
    _SEH2_TRY
    {
        dwReturned = 0;
        pp.fpGetPrintProcessorDirectory(NULL, L"wIndows nt x86", 0, NULL, 0, NULL);
    }
    _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
    {
        dwReturned = _SEH2_GetExceptionCode();
    }
    _SEH2_END;

    ok(dwReturned == EXCEPTION_ACCESS_VIOLATION, "dwReturned is %lu!\n", dwReturned);
#endif

    // fpGetPrintProcessorDirectory doesn't care about the supplied level at all. Prove this here.
    // With no buffer given, this needs to fail with ERROR_INSUFFICIENT_BUFFER.
    SetLastError(0xDEADBEEF);
    cbNeeded = 0;
    ok(!pp.fpGetPrintProcessorDirectory(NULL, L"wIndows nt x86", 1337, NULL, 0, &cbNeeded), "fpGetPrintProcessorDirectory returns TRUE!\n");
    ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER, "fpGetPrintProcessorDirectory returns error %lu!\n", GetLastError());
    ok(cbNeeded > 0, "cbNeeded is %lu!\n", cbNeeded);

    // This test corrupts something inside spoolsv so that it's only runnable once without restarting spoolsv. Therefore it's disabled.
#if 0
    // Now provide the demanded size, but no buffer.
    // Unlike GetPrintProcessorDirectoryW, fpGetPrintProcessorDirectory doesn't check for this case and tries to access the buffer.
    _SEH2_TRY
    {
        dwReturned = 0;
        pp.fpGetPrintProcessorDirectory(NULL, L"wIndows nt x86", 1, NULL, cbNeeded, &cbTemp);
    }
    _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
    {
        dwReturned = _SEH2_GetExceptionCode();
    }
    _SEH2_END;

    ok(dwReturned == EXCEPTION_ACCESS_VIOLATION, "dwReturned is %lu!\n", dwReturned);
#endif

    // Prove that this check is implemented in spoolss' GetPrintProcessorDirectoryW instead.
    // In contrast to winspool's GetPrintProcessorDirectoryW, cbTemp is left untouched though. This comes from the fact that RPC isn't involved here.
    SetLastError(0xDEADBEEF);
    cbTemp = 0xDEADBEEF;
    ok(!pGetPrintProcessorDirectoryW(NULL, L"wIndows nt x86", 1, NULL, cbNeeded, &cbTemp), "pGetPrintProcessorDirectoryW returns TRUE!\n");
    ok(GetLastError() == ERROR_INVALID_USER_BUFFER, "pGetPrintProcessorDirectoryW returns error %lu!\n", GetLastError());
    ok(cbTemp == 0xDEADBEEF, "cbTemp is %lu!\n", cbTemp);

    // Finally use the function as intended and aim for success!
    // We only check success by the boolean return value though. GetLastError doesn't return anything meaningful here.
    pwszBuffer = DllAllocSplMem(cbNeeded);
    SetLastError(0xDEADBEEF);
    ok(pp.fpGetPrintProcessorDirectory(NULL, L"wIndows nt x86", 1, (PBYTE)pwszBuffer, cbNeeded, &cbTemp), "fpGetPrintProcessorDirectory returns FALSE!\n");
    ok(wcslen(pwszBuffer) == cbNeeded / sizeof(WCHAR) - 1, "fpGetPrintProcessorDirectory string is %Iu characters long, but %lu characters expected!\n", wcslen(pwszBuffer), cbNeeded / sizeof(WCHAR) - 1);
    DllFreeSplMem(pwszBuffer);
}