spoolss: Move routing functions to a seperate file.

This commit is contained in:
Detlef Riekenberg 2009-02-06 09:35:13 +01:00 committed by Alexandre Julliard
parent 0639904cc2
commit 9056d4d2d4
4 changed files with 317 additions and 84 deletions

View file

@ -7,7 +7,8 @@ IMPORTLIB = spoolss
IMPORTS = winspool kernel32
C_SRCS = \
spoolss_main.c
router.c \
spoolss_main.c
@MAKE_DLL_RULES@

270
dlls/spoolss/router.c Normal file
View file

@ -0,0 +1,270 @@
/*
* Routing for Spooler-Service helper DLL
*
* Copyright 2006-2009 Detlef Riekenberg
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this library; if not, write to the Free Software
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
*/
#include <stdarg.h>
#include "windef.h"
#include "winbase.h"
#include "winerror.h"
#include "winreg.h"
#include "wingdi.h"
#include "winspool.h"
#include "ddk/winsplp.h"
#include "spoolss.h"
#include "wine/debug.h"
WINE_DEFAULT_DEBUG_CHANNEL(spoolss);
/* ################################ */
#define MAX_BACKEND 3
typedef struct {
/* PRINTPROVIDOR functions */
DWORD (WINAPI *fpOpenPrinter)(LPWSTR, HANDLE *, LPPRINTER_DEFAULTSW);
DWORD (WINAPI *fpSetJob)(HANDLE, DWORD, DWORD, LPBYTE, DWORD);
DWORD (WINAPI *fpGetJob)(HANDLE, DWORD, DWORD, LPBYTE, DWORD, LPDWORD);
DWORD (WINAPI *fpEnumJobs)(HANDLE, DWORD, DWORD, DWORD, LPBYTE, DWORD, LPDWORD, LPDWORD);
HANDLE (WINAPI *fpAddPrinter)(LPWSTR, DWORD, LPBYTE);
DWORD (WINAPI *fpDeletePrinter)(HANDLE);
DWORD (WINAPI *fpSetPrinter)(HANDLE, DWORD, LPBYTE, DWORD);
DWORD (WINAPI *fpGetPrinter)(HANDLE, DWORD, LPBYTE, DWORD, LPDWORD);
DWORD (WINAPI *fpEnumPrinters)(DWORD, LPWSTR, DWORD, LPBYTE, DWORD, LPDWORD, LPDWORD);
DWORD (WINAPI *fpAddPrinterDriver)(LPWSTR, DWORD, LPBYTE);
DWORD (WINAPI *fpEnumPrinterDrivers)(LPWSTR, LPWSTR, DWORD, LPBYTE, DWORD, LPDWORD, LPDWORD);
DWORD (WINAPI *fpGetPrinterDriver)(HANDLE, LPWSTR, DWORD, LPBYTE, DWORD, LPDWORD);
DWORD (WINAPI *fpGetPrinterDriverDirectory)(LPWSTR, LPWSTR, DWORD, LPBYTE, DWORD, LPDWORD);
DWORD (WINAPI *fpDeletePrinterDriver)(LPWSTR, LPWSTR, LPWSTR);
DWORD (WINAPI *fpAddPrintProcessor)(LPWSTR, LPWSTR, LPWSTR, LPWSTR);
DWORD (WINAPI *fpEnumPrintProcessors)(LPWSTR, LPWSTR, DWORD, LPBYTE, DWORD, LPDWORD, LPDWORD);
DWORD (WINAPI *fpGetPrintProcessorDirectory)(LPWSTR, LPWSTR, DWORD, LPBYTE, DWORD, LPDWORD);
DWORD (WINAPI *fpDeletePrintProcessor)(LPWSTR, LPWSTR, LPWSTR);
DWORD (WINAPI *fpEnumPrintProcessorDatatypes)(LPWSTR, LPWSTR, DWORD, LPBYTE, DWORD, LPDWORD, LPDWORD);
DWORD (WINAPI *fpStartDocPrinter)(HANDLE, DWORD, LPBYTE);
DWORD (WINAPI *fpStartPagePrinter)(HANDLE);
DWORD (WINAPI *fpWritePrinter)(HANDLE, LPVOID, DWORD, LPDWORD);
DWORD (WINAPI *fpEndPagePrinter)(HANDLE);
DWORD (WINAPI *fpAbortPrinter)(HANDLE);
DWORD (WINAPI *fpReadPrinter)(HANDLE, LPVOID, DWORD, LPDWORD);
DWORD (WINAPI *fpEndDocPrinter)(HANDLE);
DWORD (WINAPI *fpAddJob)(HANDLE, DWORD, LPBYTE, DWORD, LPDWORD);
DWORD (WINAPI *fpScheduleJob)(HANDLE, DWORD);
DWORD (WINAPI *fpGetPrinterData)(HANDLE, LPWSTR, LPDWORD, LPBYTE, DWORD, LPDWORD);
DWORD (WINAPI *fpSetPrinterData)(HANDLE, LPWSTR, DWORD, LPBYTE, DWORD);
DWORD (WINAPI *fpWaitForPrinterChange)(HANDLE, DWORD);
DWORD (WINAPI *fpClosePrinter)(HANDLE);
DWORD (WINAPI *fpAddForm)(HANDLE, DWORD, LPBYTE);
DWORD (WINAPI *fpDeleteForm)(HANDLE, LPWSTR);
DWORD (WINAPI *fpGetForm)(HANDLE, LPWSTR, DWORD, LPBYTE, DWORD, LPDWORD);
DWORD (WINAPI *fpSetForm)(HANDLE, LPWSTR, DWORD, LPBYTE);
DWORD (WINAPI *fpEnumForms)(HANDLE, DWORD, LPBYTE, DWORD, LPDWORD, LPDWORD);
DWORD (WINAPI *fpEnumMonitors)(LPWSTR, DWORD, LPBYTE, DWORD, LPDWORD, LPDWORD);
DWORD (WINAPI *fpEnumPorts)(LPWSTR, DWORD, LPBYTE, DWORD, LPDWORD, LPDWORD);
DWORD (WINAPI *fpAddPort)(LPWSTR, HWND, LPWSTR);
DWORD (WINAPI *fpConfigurePort)(LPWSTR, HWND, LPWSTR);
DWORD (WINAPI *fpDeletePort)(LPWSTR, HWND, LPWSTR);
HANDLE (WINAPI *fpCreatePrinterIC)(HANDLE, LPDEVMODEW);
DWORD (WINAPI *fpPlayGdiScriptOnPrinterIC)(HANDLE, LPBYTE, DWORD, LPBYTE, DWORD, DWORD);
DWORD (WINAPI *fpDeletePrinterIC)(HANDLE);
DWORD (WINAPI *fpAddPrinterConnection)(LPWSTR);
DWORD (WINAPI *fpDeletePrinterConnection)(LPWSTR);
DWORD (WINAPI *fpPrinterMessageBox)(HANDLE, DWORD, HWND, LPWSTR, LPWSTR, DWORD);
DWORD (WINAPI *fpAddMonitor)(LPWSTR, DWORD, LPBYTE);
DWORD (WINAPI *fpDeleteMonitor)(LPWSTR, LPWSTR, LPWSTR);
DWORD (WINAPI *fpResetPrinter)(HANDLE, LPPRINTER_DEFAULTSW);
DWORD (WINAPI *fpGetPrinterDriverEx)(HANDLE, LPWSTR, DWORD, LPBYTE, DWORD, LPDWORD, DWORD, DWORD, PDWORD, PDWORD);
HANDLE (WINAPI *fpFindFirstPrinterChangeNotification)(HANDLE, DWORD, DWORD, LPVOID);
DWORD (WINAPI *fpFindClosePrinterChangeNotification)(HANDLE);
DWORD (WINAPI *fpAddPortEx)(HANDLE, LPWSTR, DWORD, LPBYTE, LPWSTR);
DWORD (WINAPI *fpShutDown)(LPVOID);
DWORD (WINAPI *fpRefreshPrinterChangeNotification)(HANDLE, DWORD, PVOID, PVOID);
DWORD (WINAPI *fpOpenPrinterEx)(LPWSTR, LPHANDLE, LPPRINTER_DEFAULTSW, LPBYTE, DWORD);
HANDLE (WINAPI *fpAddPrinterEx)(LPWSTR, DWORD, LPBYTE, LPBYTE, DWORD);
DWORD (WINAPI *fpSetPort)(LPWSTR, LPWSTR, DWORD, LPBYTE);
DWORD (WINAPI *fpEnumPrinterData)(HANDLE, DWORD, LPWSTR, DWORD, LPDWORD, LPDWORD, LPBYTE, DWORD, LPDWORD);
DWORD (WINAPI *fpDeletePrinterData)(HANDLE, LPWSTR);
DWORD (WINAPI *fpClusterSplOpen)(LPCWSTR, LPCWSTR, PHANDLE, LPCWSTR, LPCWSTR);
DWORD (WINAPI *fpClusterSplClose)(HANDLE);
DWORD (WINAPI *fpClusterSplIsAlive)(HANDLE);
DWORD (WINAPI *fpSetPrinterDataEx)(HANDLE, LPCWSTR, LPCWSTR, DWORD, LPBYTE, DWORD);
DWORD (WINAPI *fpGetPrinterDataEx)(HANDLE, LPCWSTR, LPCWSTR, LPDWORD, LPBYTE, DWORD, LPDWORD);
DWORD (WINAPI *fpEnumPrinterDataEx)(HANDLE, LPCWSTR, LPBYTE, DWORD, LPDWORD, LPDWORD);
DWORD (WINAPI *fpEnumPrinterKey)(HANDLE, LPCWSTR, LPWSTR, DWORD, LPDWORD);
DWORD (WINAPI *fpDeletePrinterDataEx)(HANDLE, LPCWSTR, LPCWSTR);
DWORD (WINAPI *fpDeletePrinterKey)(HANDLE hPrinter, LPCWSTR pKeyName);
DWORD (WINAPI *fpSeekPrinter)(HANDLE, LARGE_INTEGER, PLARGE_INTEGER, DWORD, BOOL);
DWORD (WINAPI *fpDeletePrinterDriverEx)(LPWSTR, LPWSTR, LPWSTR, DWORD, DWORD);
DWORD (WINAPI *fpAddPerMachineConnection)(LPCWSTR, LPCWSTR, LPCWSTR, LPCWSTR);
DWORD (WINAPI *fpDeletePerMachineConnection)(LPCWSTR, LPCWSTR);
DWORD (WINAPI *fpEnumPerMachineConnections)(LPCWSTR, LPBYTE, DWORD, LPDWORD, LPDWORD);
DWORD (WINAPI *fpXcvData)(HANDLE, LPCWSTR, PBYTE, DWORD, PBYTE, DWORD, PDWORD, PDWORD);
DWORD (WINAPI *fpAddPrinterDriverEx)(LPWSTR, DWORD, LPBYTE, DWORD);
DWORD (WINAPI *fpSplReadPrinter)(HANDLE, LPBYTE *, DWORD);
DWORD (WINAPI *fpDriverUnloadComplete)(LPWSTR);
DWORD (WINAPI *fpGetSpoolFileInfo)(HANDLE, LPWSTR *, LPHANDLE, HANDLE, HANDLE);
DWORD (WINAPI *fpCommitSpoolData)(HANDLE, DWORD);
DWORD (WINAPI *fpCloseSpoolFileHandle)(HANDLE);
DWORD (WINAPI *fpFlushPrinter)(HANDLE, LPBYTE, DWORD, LPDWORD, DWORD);
DWORD (WINAPI *fpSendRecvBidiData)(HANDLE, LPCWSTR, LPBIDI_REQUEST_CONTAINER, LPBIDI_RESPONSE_CONTAINER *);
DWORD (WINAPI *fpAddDriverCatalog)(HANDLE, DWORD, VOID *, DWORD);
/* Private Data */
HMODULE dll;
LPWSTR dllname;
LPWSTR name;
LPWSTR regroot;
DWORD index;
} backend_t;
/* ################################ */
static backend_t *backend[MAX_BACKEND];
static DWORD used_backends = 0;
static CRITICAL_SECTION backend_cs;
static CRITICAL_SECTION_DEBUG backend_cs_debug =
{
0, 0, &backend_cs,
{ &backend_cs_debug.ProcessLocksList, &backend_cs_debug.ProcessLocksList },
0, 0, { (DWORD_PTR)(__FILE__ ": backend_cs") }
};
static CRITICAL_SECTION backend_cs = { &backend_cs_debug, -1, 0, 0, 0, 0 };
/* ################################ */
static WCHAR localsplW[] = {'l','o','c','a','l','s','p','l','.','d','l','l',0};
/******************************************************************
* strdupW [internal]
*
* create a copy of a unicode-string
*
*/
static LPWSTR strdupW(LPCWSTR p)
{
LPWSTR ret;
DWORD len;
if(!p) return NULL;
len = (lstrlenW(p) + 1) * sizeof(WCHAR);
ret = heap_alloc(len);
memcpy(ret, p, len);
return ret;
}
/******************************************************************
* backend_unload_all [internal]
*
* unload all backends
*/
void backend_unload_all(void)
{
EnterCriticalSection(&backend_cs);
while (used_backends > 0) {
used_backends--;
FreeLibrary(backend[used_backends]->dll);
heap_free(backend[used_backends]->dllname);
heap_free(backend[used_backends]->name);
heap_free(backend[used_backends]->regroot);
heap_free(backend[used_backends]);
backend[used_backends] = NULL;
}
LeaveCriticalSection(&backend_cs);
}
/******************************************************************************
* backend_load [internal]
*
* load and init a backend
*
* PARAMS
* name [I] Printprovider to use for the backend. NULL for the local print provider
*
* RETURNS
* Success: PTR to the backend
* Failure: NULL
*
*/
static backend_t * backend_load(LPWSTR dllname, LPWSTR name, LPWSTR regroot)
{
BOOL (WINAPI *pInitializePrintProvidor)(LPPRINTPROVIDOR, DWORD, LPWSTR);
DWORD id;
DWORD res;
EnterCriticalSection(&backend_cs);
id = used_backends;
backend[id] = heap_alloc_zero(sizeof(backend_t));
if (!backend[id]) return NULL;
backend[id]->dllname = strdupW(dllname);
backend[id]->name = strdupW(name);
backend[id]->regroot = strdupW(regroot);
backend[id]->dll = LoadLibraryW(dllname);
if (backend[id]->dll) {
pInitializePrintProvidor = (void *) GetProcAddress(backend[id]->dll, "InitializePrintProvidor");
if (pInitializePrintProvidor) {
/* native localspl does not clear unused entries */
res = pInitializePrintProvidor((PRINTPROVIDOR *) backend[id], sizeof(PRINTPROVIDOR), regroot);
if (res) {
used_backends++;
backend[id]->index = used_backends;
LeaveCriticalSection(&backend_cs);
TRACE("--> backend #%d: %p (%s)\n", id, backend[id], debugstr_w(dllname));
return backend[id];
}
}
FreeLibrary(backend[id]->dll);
}
heap_free(backend[id]->dllname);
heap_free(backend[id]->name);
heap_free(backend[id]->regroot);
heap_free(backend[id]);
backend[id] = NULL;
LeaveCriticalSection(&backend_cs);
WARN("failed to init %s: %u\n", debugstr_w(dllname), GetLastError());
return NULL;
}
/******************************************************************************
* backend_load_all [internal]
*
* load and init all backends
*
* RETURNS
* Success: TRUE
* Failure: FALSE
*
*/
BOOL backend_load_all(void)
{
backend_t * pb;
pb = backend_load(localsplW, NULL, NULL);
/* ToDo: parse the registry and load all other backends */
return (pb != NULL);
}

42
dlls/spoolss/spoolss.h Normal file
View file

@ -0,0 +1,42 @@
/*
* Copyright 2006-2009 Detlef Riekenberg
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this library; if not, write to the Free Software
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
*/
#include "windef.h"
#include "winbase.h"
/* ################################ */
BOOL backend_load_all(void);
void backend_unload_all(void);
/* ## Memory allocation functions ## */
static inline void * __WINE_ALLOC_SIZE(1) heap_alloc( size_t len )
{
return HeapAlloc( GetProcessHeap(), 0, len );
}
static inline void * __WINE_ALLOC_SIZE(1) heap_alloc_zero( size_t len )
{
return HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, len );
}
static inline BOOL heap_free( void *mem )
{
return HeapFree( GetProcessHeap(), 0, mem );
}

View file

@ -28,6 +28,7 @@
#include "wingdi.h"
#include "winspool.h"
#include "ddk/winsplp.h"
#include "spoolss.h"
#include "wine/debug.h"
@ -35,91 +36,10 @@ WINE_DEFAULT_DEBUG_CHANNEL(spoolss);
/* ################################ */
static CRITICAL_SECTION backend_cs;
static CRITICAL_SECTION_DEBUG backend_cs_debug =
{
0, 0, &backend_cs,
{ &backend_cs_debug.ProcessLocksList, &backend_cs_debug.ProcessLocksList },
0, 0, { (DWORD_PTR)(__FILE__ ": backend_cs") }
};
static CRITICAL_SECTION backend_cs = { &backend_cs_debug, -1, 0, 0, 0, 0 };
/* ################################ */
static HMODULE hwinspool;
static HMODULE hlocalspl;
static BOOL (WINAPI *pInitializePrintProvidor)(LPPRINTPROVIDOR, DWORD, LPWSTR);
static PRINTPROVIDOR * backend;
/* ################################ */
static const WCHAR localspldllW[] = {'l','o','c','a','l','s','p','l','.','d','l','l',0};
static const WCHAR winspooldrvW[] = {'w','i','n','s','p','o','o','l','.','d','r','v',0};
/******************************************************************************
* backend_load [internal]
*
* load and init our backend (the local printprovider: "localspl.dll")
*
* PARAMS
*
* RETURNS
* Success: TRUE
* Failure: FALSE and RPC_S_SERVER_UNAVAILABLE
*
* NOTES
* In windows, the spooler router (spoolss.dll) support multiple
* printprovider (localspl.dll for the local system)
*
*/
static BOOL backend_load(void)
{
static PRINTPROVIDOR mybackend;
DWORD res;
if (backend) return TRUE;
EnterCriticalSection(&backend_cs);
hlocalspl = LoadLibraryW(localspldllW);
if (hlocalspl) {
pInitializePrintProvidor = (void *) GetProcAddress(hlocalspl, "InitializePrintProvidor");
if (pInitializePrintProvidor) {
/* native localspl does not clear unused entries */
memset(&mybackend, 0, sizeof(mybackend));
res = pInitializePrintProvidor(&mybackend, sizeof(mybackend), NULL);
if (res) {
backend = &mybackend;
LeaveCriticalSection(&backend_cs);
TRACE("backend: %p (%p)\n", backend, hlocalspl);
return TRUE;
}
}
FreeLibrary(hlocalspl);
}
LeaveCriticalSection(&backend_cs);
WARN("failed to load the backend: %u\n", GetLastError());
SetLastError(RPC_S_SERVER_UNAVAILABLE);
return FALSE;
}
/******************************************************************
* unload_backend [internal]
*
*/
static void backend_unload(void)
{
EnterCriticalSection(&backend_cs);
if (backend) {
backend = NULL;
FreeLibrary(hlocalspl);
}
LeaveCriticalSection(&backend_cs);
}
/******************************************************************************
*
*/
@ -135,7 +55,7 @@ BOOL WINAPI DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpvReserved)
break;
case DLL_PROCESS_DETACH:
backend_unload();
backend_unload_all();
break;
}
}
@ -267,7 +187,7 @@ BOOL WINAPI ImpersonatePrinterClient(HANDLE hToken)
BOOL WINAPI InitializeRouter(void)
{
TRACE("()\n");
return backend_load();
return backend_load_all();
}
/******************************************************************