#include "precomp.h"
#include <prtprocenv.h>

Include dependency graph for main.c:

Functions
static BOOL	_InitializeLocalSpooler (void)

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

BOOL WINAPI	InitializePrintProvidor (LPPRINTPROVIDOR pPrintProvidor, DWORD cbPrintProvidor, LPWSTR pFullRegistryPath)

Variables
HKEY	hPrintKey = NULL

HKEY	hPrintersKey = NULL

WCHAR	wszJobDirectory [MAX_PATH]

DWORD	cchJobDirectory

WCHAR	wszSpoolDirectory [MAX_PATH]

DWORD	cchSpoolDirectory

const DWORD	dwSpoolerMajorVersion = 3

const DWORD	dwSpoolerMinorVersion = 0

const WCHAR	wszDefaultDocumentName [] = L"Local Downlevel Document"

PCWSTR	wszPrintProviderInfo [3]

static const PRINTPROVIDOR	_PrintProviderFunctions

Function Documentation

◆ _InitializeLocalSpooler()

static BOOL _InitializeLocalSpooler ( void )

static

Definition at line 124 of file main.c.

{
    const WCHAR wszPrintersPath[] = L"\\PRINTERS";
    const DWORD cchPrintersPath = _countof(wszPrintersPath) - 1;
    const WCHAR wszSpoolPath[] = L"\\spool";
    const DWORD cchSpoolPath = _countof(wszSpoolPath) - 1;
    const WCHAR wszSymbolicLinkValue[] = L"\\REGISTRY\\MACHINE\\SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion\\Print\\Printers";
    const DWORD cbSymbolicLinkValue = sizeof(wszSymbolicLinkValue) - sizeof(WCHAR);
 
    BOOL bReturnValue = FALSE;
    DWORD cbData;
    DWORD dwErrorCode;
    HKEY hKey = NULL;
 
    // On startup, always create a volatile symbolic link in the registry if it doesn't exist yet.
    //   "SYSTEM\CurrentControlSet\Control\Print\Printers" -> "SOFTWARE\Microsoft\Windows NT\CurrentVersion\Print\Printers"
    //
    // According to https://social.technet.microsoft.com/Forums/windowsserver/en-US/a683ab54-c43c-4ebe-af8f-1f7a65af2a51
    // this is needed when having >900 printers to work around a size limit of the SYSTEM registry hive.
    dwErrorCode = (DWORD)RegCreateKeyExW(HKEY_LOCAL_MACHINE, L"SYSTEM\\CurrentControlSet\\Control\\Print\\Printers", 0, NULL, REG_OPTION_VOLATILE | REG_OPTION_CREATE_LINK, KEY_CREATE_LINK | KEY_SET_VALUE, NULL, &hKey, NULL);
    if (dwErrorCode == ERROR_SUCCESS)
    {
        // Note that wszSymbolicLink has to be stored WITHOUT the terminating null character for the symbolic link to work!
        // See cbSymbolicLinkValue above.
        dwErrorCode = (DWORD)RegSetValueExW(hKey, L"SymbolicLinkValue", 0, REG_LINK, (PBYTE)wszSymbolicLinkValue, cbSymbolicLinkValue);
        if (dwErrorCode != ERROR_SUCCESS)
        {
            ERR("RegSetValueExW failed for the Printers symlink with error %lu!\n", dwErrorCode);
            goto Cleanup;
        }
    }
    else if (dwErrorCode != ERROR_ALREADY_EXISTS)
    {
        ERR("RegCreateKeyExW failed for the Printers symlink with error %lu!\n", dwErrorCode);
        goto Cleanup;
    }
 
    // Open some registry keys and leave them open. We need them multiple times throughout the Local Spooler.
    dwErrorCode = (DWORD)RegOpenKeyExW(HKEY_LOCAL_MACHINE, L"SYSTEM\\CurrentControlSet\\Control\\Print", 0, KEY_ALL_ACCESS, &hPrintKey);
    if (dwErrorCode != ERROR_SUCCESS)
    {
        ERR("RegOpenKeyExW failed for \"Print\" with error %lu!\n", dwErrorCode);
        goto Cleanup;
    }
 
    dwErrorCode = (DWORD)RegOpenKeyExW(HKEY_LOCAL_MACHINE, L"SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion\\Print\\Printers", 0, KEY_ALL_ACCESS, &hPrintersKey);
    if (dwErrorCode != ERROR_SUCCESS)
    {
        ERR("RegOpenKeyExW failed for \"Printers\" with error %lu!\n", dwErrorCode);
        goto Cleanup;
    }
 
    // Construct the path to "%SystemRoot%\system32\spool".
    // Forget about length checks here. If this doesn't fit into MAX_PATH, our OS has more serious problems...
    cchSpoolDirectory = GetSystemDirectoryW(wszSpoolDirectory, MAX_PATH);
    CopyMemory(&wszSpoolDirectory[cchSpoolDirectory], wszSpoolPath, (cchSpoolPath + 1) * sizeof(WCHAR));
    cchSpoolDirectory += cchSpoolPath;
 
    // Query the job directory.
    cbData = sizeof(wszJobDirectory);
    dwErrorCode = (DWORD)RegQueryValueExW(hPrintersKey, SPLREG_DEFAULT_SPOOL_DIRECTORY, NULL, NULL, (PBYTE)wszJobDirectory, &cbData);
    if (dwErrorCode == ERROR_SUCCESS)
    {
        cchJobDirectory = cbData / sizeof(WCHAR) - 1;
    }
    else if (dwErrorCode == ERROR_FILE_NOT_FOUND)
    {
        // Use the default "%SystemRoot%\system32\spool\PRINTERS".
        CopyMemory(wszJobDirectory, wszSpoolDirectory, cchSpoolDirectory * sizeof(WCHAR));
        CopyMemory(&wszJobDirectory[cchSpoolDirectory], wszPrintersPath, (cchPrintersPath + 1) * sizeof(WCHAR));
        cchJobDirectory = cchSpoolDirectory + cchPrintersPath;
 
        // Save this for next time.
        RegSetValueExW(hPrintersKey, SPLREG_DEFAULT_SPOOL_DIRECTORY, 0, REG_SZ, (PBYTE)wszJobDirectory, (cchJobDirectory + 1) * sizeof(WCHAR));
    }
    else
    {
        ERR("RegQueryValueExW failed for \"%S\" with error %lu!\n", SPLREG_DEFAULT_SPOOL_DIRECTORY, dwErrorCode);
        goto Cleanup;
    }
 
    // Initialize all lists.
    if (!InitializePrintMonitorList())
        goto Cleanup;
 
    if (!InitializePortList())
        goto Cleanup;
 
    if (!InitializePrintProcessorList())
        goto Cleanup;
 
    if (!InitializePrinterList())
        goto Cleanup;
 
    if (!InitializeGlobalJobList())
        goto Cleanup;
 
    if (!InitializeFormList())
        goto Cleanup;
 
    if (!InitializePrinterDrivers())
        goto Cleanup;
 
    // Local Spooler Initialization finished successfully!
    bReturnValue = TRUE;
 
Cleanup:
    if (hKey)
        RegCloseKey(hKey);
 
    return bReturnValue;
}

Referenced by DllMain().

◆ DllMain()

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

Definition at line 238 of file main.c.

{
    switch (fdwReason)
    {
        case DLL_PROCESS_ATTACH:
            DisableThreadLibraryCalls(hinstDLL);
            return _InitializeLocalSpooler();
 
        default:
            return TRUE;
    }
}

◆ InitializePrintProvidor()

BOOL WINAPI InitializePrintProvidor	(	LPPRINTPROVIDOR	pPrintProvidor,
		DWORD	cbPrintProvidor,
		LPWSTR	pFullRegistryPath
	)

Definition at line 252 of file main.c.

{
    TRACE("InitializePrintProvidor(%p, %lu, %S)\n", pPrintProvidor, cbPrintProvidor, pFullRegistryPath);
 
    CopyMemory(pPrintProvidor, &_PrintProviderFunctions, min(cbPrintProvidor, sizeof(PRINTPROVIDOR)));
 
    SetLastError(ERROR_SUCCESS);
    return TRUE;
}

Variable Documentation

◆ _PrintProviderFunctions

const PRINTPROVIDOR _PrintProviderFunctions

static

Definition at line 34 of file main.c.

Referenced by InitializePrintProvidor().

◆ cchJobDirectory

DWORD cchJobDirectory

Definition at line 14 of file main.c.

Referenced by _InitializeLocalSpooler(), GetJobFilePath(), and InitializeGlobalJobList().

◆ cchSpoolDirectory

DWORD cchSpoolDirectory

Definition at line 16 of file main.c.

Referenced by _InitializeLocalSpooler(), and LocalGetPrintProcessorDirectory().

◆ dwSpoolerMajorVersion

const DWORD dwSpoolerMajorVersion = 3

This is what the Spooler of Windows Server 2003 returns (for example using GetPrinterDataExW, SPLREG_MAJOR_VERSION/SPLREG_MINOR_VERSION)

Definition at line 22 of file main.c.

Referenced by _LocalGetPrintServerHandleData().

◆ dwSpoolerMinorVersion

const DWORD dwSpoolerMinorVersion = 0

Definition at line 23 of file main.c.

Referenced by _LocalGetPrintServerHandleData().

◆ hPrintersKey

HKEY hPrintersKey = NULL

Definition at line 12 of file main.c.

Referenced by _InitializeLocalSpooler(), _LocalGetPrinterHandleData(), _LocalGetPrintServerHandleData(), _LocalSetPrinterHandleData(), _LocalSetPrintServerHandleData(), and InitializePrinterList().

◆ hPrintKey

HKEY hPrintKey = NULL

Definition at line 11 of file main.c.

Referenced by _InitializeLocalSpooler(), _LocalGetPrintServerHandleData(), and _LocalSetPrintServerHandleData().

◆ wszDefaultDocumentName

const WCHAR wszDefaultDocumentName[] = L"Local Downlevel Document"

Definition at line 25 of file main.c.

Referenced by CreateJob().

◆ wszJobDirectory

WCHAR wszJobDirectory[MAX_PATH]

Definition at line 13 of file main.c.

Referenced by _InitializeLocalSpooler(), GetJobFilePath(), and InitializeGlobalJobList().

◆ wszPrintProviderInfo

PCWSTR wszPrintProviderInfo[3]

Initial value:

= {
    L"Windows NT Local Print Providor",     
    L"Locally connected Printers",          
    L"Windows NT Local Printers"            
}

Definition at line 27 of file main.c.

Referenced by _DumpLevel1PrintProviderInformation(), _LocalEnumPrintersCheckName(), _LocalGetPrinterDriverLevel6(), and _LocalGetPrinterDriverLevel8().

◆ wszSpoolDirectory

WCHAR wszSpoolDirectory[MAX_PATH]

Definition at line 15 of file main.c.