mirror of
https://github.com/reactos/reactos.git
synced 2024-09-04 17:49:15 +00:00
265 lines
7.3 KiB
C
265 lines
7.3 KiB
C
/*
|
|
* PROJECT: ReactOS Spooler Router
|
|
* LICENSE: GPL-2.0+ (https://spdx.org/licenses/GPL-2.0+)
|
|
* PURPOSE: Main functions
|
|
* COPYRIGHT: Copyright 2015 Colin Finck (colin@reactos.org)
|
|
*/
|
|
|
|
#include "precomp.h"
|
|
|
|
// Global Variables
|
|
HANDLE hProcessHeap;
|
|
LIST_ENTRY PrintProviderList;
|
|
|
|
|
|
static DWORD
|
|
_AddPrintProviderToList(PCWSTR pwszFileName)
|
|
{
|
|
DWORD dwErrorCode;
|
|
HINSTANCE hinstPrintProvider;
|
|
PInitializePrintProvidor pfnInitializePrintProvidor;
|
|
PSPOOLSS_PRINT_PROVIDER pPrintProvider = NULL;
|
|
|
|
// Try to load it.
|
|
hinstPrintProvider = LoadLibraryW(pwszFileName);
|
|
if (!hinstPrintProvider)
|
|
{
|
|
dwErrorCode = GetLastError();
|
|
ERR("LoadLibraryW failed for \"%S\" with error %lu!\n", pwszFileName, dwErrorCode);
|
|
goto Cleanup;
|
|
}
|
|
|
|
// Get the initialization routine.
|
|
pfnInitializePrintProvidor = (PInitializePrintProvidor)GetProcAddress(hinstPrintProvider, "InitializePrintProvidor");
|
|
if (!pfnInitializePrintProvidor)
|
|
{
|
|
dwErrorCode = GetLastError();
|
|
ERR("GetProcAddress failed for \"%S\" with error %lu!\n", pwszFileName, dwErrorCode);
|
|
goto Cleanup;
|
|
}
|
|
|
|
// Create a new SPOOLSS_PRINT_PROVIDER structure for it.
|
|
pPrintProvider = DllAllocSplMem(sizeof(SPOOLSS_PRINT_PROVIDER));
|
|
if (!pPrintProvider)
|
|
{
|
|
dwErrorCode = ERROR_NOT_ENOUGH_MEMORY;
|
|
ERR("DllAllocSplMem failed!\n");
|
|
goto Cleanup;
|
|
}
|
|
|
|
// Call the Print Provider initialization function.
|
|
if (!pfnInitializePrintProvidor(&pPrintProvider->PrintProvider, sizeof(PRINTPROVIDOR), NULL))
|
|
{
|
|
dwErrorCode = GetLastError();
|
|
ERR("InitializePrintProvidor failed for \"%S\" with error %lu!\n", pwszFileName, dwErrorCode);
|
|
goto Cleanup;
|
|
}
|
|
|
|
// Add this Print Provider to the list.
|
|
InsertTailList(&PrintProviderList, &pPrintProvider->Entry);
|
|
|
|
// Don't let the cleanup routine free this.
|
|
pPrintProvider = NULL;
|
|
dwErrorCode = ERROR_SUCCESS;
|
|
|
|
Cleanup:
|
|
if (pPrintProvider)
|
|
DllFreeSplMem(pPrintProvider);
|
|
|
|
return dwErrorCode;
|
|
}
|
|
|
|
static BOOL
|
|
_InitializePrintProviderList(VOID)
|
|
{
|
|
DWORD cbFileName;
|
|
DWORD cchMaxSubKey;
|
|
DWORD cchPrintProviderName;
|
|
DWORD dwErrorCode;
|
|
DWORD dwSubKeys;
|
|
DWORD i;
|
|
HKEY hKey = NULL;
|
|
HKEY hSubKey = NULL;
|
|
PWSTR pwszPrintProviderName = NULL;
|
|
WCHAR wszFileName[MAX_PATH];
|
|
|
|
// Initialize an empty list for our Print Providers.
|
|
InitializeListHead(&PrintProviderList);
|
|
|
|
// First add the Local Spooler.
|
|
// This one must exist and must be the first one in the list.
|
|
dwErrorCode = _AddPrintProviderToList(L"localspl");
|
|
if (dwErrorCode != ERROR_SUCCESS)
|
|
{
|
|
ERR("The Local Spooler could not be loaded!\n");
|
|
goto Cleanup;
|
|
}
|
|
|
|
// Now add additional Print Providers from the registry.
|
|
// First of all, open the key containing print providers.
|
|
dwErrorCode = (DWORD)RegOpenKeyExW(HKEY_LOCAL_MACHINE, L"SYSTEM\\CurrentControlSet\\Control\\Print\\Providers", 0, KEY_READ, &hKey);
|
|
if (dwErrorCode != ERROR_SUCCESS)
|
|
{
|
|
ERR("RegOpenKeyExW failed with status %lu!\n", dwErrorCode);
|
|
goto Cleanup;
|
|
}
|
|
|
|
// Get the number of Print Providers and maximum sub key length.
|
|
dwErrorCode = (DWORD)RegQueryInfoKeyW(hKey, NULL, NULL, NULL, &dwSubKeys, &cchMaxSubKey, NULL, NULL, NULL, NULL, NULL, NULL);
|
|
if (dwErrorCode != ERROR_SUCCESS)
|
|
{
|
|
ERR("RegQueryInfoKeyW failed with status %lu!\n", dwErrorCode);
|
|
goto Cleanup;
|
|
}
|
|
|
|
// Allocate a temporary buffer for the Print Provider names.
|
|
pwszPrintProviderName = DllAllocSplMem((cchMaxSubKey + 1) * sizeof(WCHAR));
|
|
if (!pwszPrintProviderName)
|
|
{
|
|
dwErrorCode = ERROR_NOT_ENOUGH_MEMORY;
|
|
ERR("DllAllocSplMem failed!\n");
|
|
goto Cleanup;
|
|
}
|
|
|
|
// Loop through all available Print Providers.
|
|
for (i = 0; i < dwSubKeys; i++)
|
|
{
|
|
// Cleanup tasks from the previous run
|
|
if (hSubKey)
|
|
{
|
|
RegCloseKey(hSubKey);
|
|
hSubKey = NULL;
|
|
}
|
|
|
|
// Get the name of this Print Provider.
|
|
cchPrintProviderName = cchMaxSubKey + 1;
|
|
dwErrorCode = (DWORD)RegEnumKeyExW(hKey, i, pwszPrintProviderName, &cchPrintProviderName, NULL, NULL, NULL, NULL);
|
|
if (dwErrorCode != ERROR_SUCCESS)
|
|
{
|
|
ERR("RegEnumKeyExW failed for iteration %lu with status %lu!\n", i, dwErrorCode);
|
|
continue;
|
|
}
|
|
|
|
// Open this Print Provider's registry key.
|
|
dwErrorCode = (DWORD)RegOpenKeyExW(hKey, pwszPrintProviderName, 0, KEY_READ, &hSubKey);
|
|
if (dwErrorCode != ERROR_SUCCESS)
|
|
{
|
|
ERR("RegOpenKeyExW failed for Print Provider \"%S\" with status %lu!\n", pwszPrintProviderName, dwErrorCode);
|
|
continue;
|
|
}
|
|
|
|
// Get the file name of the Print Provider.
|
|
cbFileName = MAX_PATH * sizeof(WCHAR);
|
|
dwErrorCode = (DWORD)RegQueryValueExW(hKey, L"Driver", NULL, NULL, (PBYTE)wszFileName, &cbFileName);
|
|
if (dwErrorCode != ERROR_SUCCESS)
|
|
{
|
|
ERR("RegQueryValueExW failed with status %lu!\n", dwErrorCode);
|
|
continue;
|
|
}
|
|
|
|
// Load and add it to the list.
|
|
dwErrorCode = _AddPrintProviderToList(wszFileName);
|
|
if (dwErrorCode != ERROR_SUCCESS)
|
|
continue;
|
|
}
|
|
|
|
dwErrorCode = ERROR_SUCCESS;
|
|
|
|
Cleanup:
|
|
// Inside the loop
|
|
if (hSubKey)
|
|
RegCloseKey(hSubKey);
|
|
|
|
// Outside the loop
|
|
if (pwszPrintProviderName)
|
|
DllFreeSplMem(pwszPrintProviderName);
|
|
|
|
if (hKey)
|
|
RegCloseKey(hKey);
|
|
|
|
SetLastError(dwErrorCode);
|
|
return (dwErrorCode == ERROR_SUCCESS);
|
|
}
|
|
|
|
BOOL WINAPI
|
|
DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpvReserved)
|
|
{
|
|
switch (fdwReason)
|
|
{
|
|
case DLL_PROCESS_ATTACH:
|
|
DisableThreadLibraryCalls(hinstDLL);
|
|
hProcessHeap = GetProcessHeap();
|
|
break;
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
BOOL WINAPI
|
|
InitializeRouter(HANDLE SpoolerStatusHandle)
|
|
{
|
|
return _InitializePrintProviderList();
|
|
}
|
|
|
|
BOOL WINAPI
|
|
SplInitializeWinSpoolDrv(PVOID* pTable)
|
|
{
|
|
HINSTANCE hWinspool;
|
|
int i;
|
|
|
|
hWinspool = LoadLibraryW(L"winspool.drv");
|
|
if (!hWinspool)
|
|
{
|
|
ERR("Could not load winspool.drv, last error is %lu!\n", GetLastError());
|
|
return FALSE;
|
|
}
|
|
|
|
// Get the function pointers which are meant to be returned by this function.
|
|
pTable[0] = GetProcAddress(hWinspool, "OpenPrinterW");
|
|
pTable[1] = GetProcAddress(hWinspool, "ClosePrinter");
|
|
pTable[2] = GetProcAddress(hWinspool, "SpoolerDevQueryPrintW");
|
|
pTable[3] = GetProcAddress(hWinspool, "SpoolerPrinterEvent");
|
|
pTable[4] = GetProcAddress(hWinspool, "DocumentPropertiesW");
|
|
pTable[5] = GetProcAddress(hWinspool, (LPSTR)212);
|
|
pTable[6] = GetProcAddress(hWinspool, (LPSTR)213);
|
|
pTable[7] = GetProcAddress(hWinspool, (LPSTR)214);
|
|
pTable[8] = GetProcAddress(hWinspool, (LPSTR)215);
|
|
|
|
// Verify that all calls succeeded.
|
|
for (i = 0; i < 9; i++)
|
|
{
|
|
if (!pTable[i])
|
|
{
|
|
FreeLibrary(hWinspool);
|
|
return FALSE;
|
|
}
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
BOOL WINAPI
|
|
SplIsUpgrade(VOID)
|
|
{
|
|
return FALSE;
|
|
}
|
|
|
|
BOOL WINAPI
|
|
SpoolerInit(VOID)
|
|
{
|
|
// Nothing to do here yet
|
|
SetLastError(ERROR_SUCCESS);
|
|
return TRUE;
|
|
}
|
|
|
|
BOOL WINAPI
|
|
BuildOtherNamesFromMachineName(LPVOID * ptr1, ULONG * ptr2)
|
|
{
|
|
FIXME("(%p, %p) stub\n", ptr1, ptr2);
|
|
|
|
*ptr1 = NULL;
|
|
*ptr2 = 0;
|
|
return FALSE;
|
|
}
|
|
|