Microsoft/WindowsXP-SP1
Microsoft/WindowsXP-SP1
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity
master
WindowsXP-SP1/windows/core/termspl/umpdhook.c
Microsoft f7ca35dddd First commit
2020-09-30 16:53:49 +02:00

510 lines
15 KiB
C
Raw Permalink Blame History

This file contains invisible Unicode characters

This file contains invisible Unicode characters that are indistinguishable to humans but may be processed differently by a computer. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

/*++
Copyright (c) 1999 Microsoft Corporation
All rights reserved
Module Name:
umpdhook.c
Abstract:
This module is to redirect SPOOLSS.DLL functions to the WINSPOOL.DRV
functions for the User Mode Printer Device (UMPD) DLLs.
Author:
Min-Chih Lu Earl (v-mearl) 8-March-99 (Get ideas and implementation from
\private\tsext\admtools\tsappcmp\register.c
by v-johnjr)
Environment:
User Mode - Win32 (in WINSRV.DLL running in the CSRSS.EXE process)
Revision History:
--*/
#define _USER_
#include "precomp.h"
#pragma hdrstop
#include <ntddrdr.h>
#include <stdio.h>
#include <windows.h>
#include <winspool.h>
#if(WINVER >= 0x0500)
#include <winspl.h>
#else
#include "..\..\..\..\..\windows\spooler\spoolss\winspl.h"
#endif
#include <data.h>
#include "wingdip.h"
#if(WINVER >= 0x0500)
#include "musspl.h"
#else
#include "ctxspl.h"
#endif
//*====================================================================*//
//* Local definitions *//
//*====================================================================*//
#define SPOOLSS_DLL_W L"SPOOLSS.DLL"
#define SPOOLSS_DLL_A "SPOOLSS.DLL"
//*====================================================================*//
//* Local function prototypes
//*====================================================================*//
BOOL PlaceHooks(HMODULE hUMPD, HMODULE hSpoolss);
PVOID PlaceHookEntry(HMODULE hSpoolss, PVOID * pProcAddress);
//*====================================================================*//
//* Hook function prototypes *//
//*====================================================================*//
PVOID TSsplHookGetProcAddress(IN HMODULE hModule,IN LPCSTR lpProcName);
//*====================================================================*//
//* Public functions Implementations *//
//*====================================================================*//
BOOL
TSsplHookSplssToWinspool(
IN HMODULE hUMPD
)
/*++
Routine Description:
This routine redirect the statically linked SPOOLSS.DLL address
to the winspool.drv equivalent functions
Arguments:
hUMPD - Supplies the user mode printer driver DLL handle which uses
the SPOOLSS.DLL
Return Value:
TRUE - Success
FAIL - Error. Use GetLastError() to get the error status
--*/
{
BOOL bStatus = TRUE;
HMODULE hSpoolss;
/*
* Load SPOOLSS.DLL
*/
hSpoolss = LoadLibrary(SPOOLSS_DLL_W);
if (!hSpoolss) {
DBGMSG(DBG_WARNING,("TSsplHookSplssToWinspool - Cannot load SPOOLSS.DLL\n"));
return FALSE;
}
/*
* Redirect the spoolss.dll call to the winspool.drv call in the
* UMPD.DLL
*/
bStatus = PlaceHooks(hUMPD, hSpoolss);
FreeLibrary(hSpoolss);
DBGMSG(DBG_TRACE,("TSsplHookSplssToWinspool - Redirect UMPD.DLL Spoolss.dll to Winspool.dll\n"));
return bStatus;
}
//*====================================================================*//
//* Hook functions Implementations *//
//* These function hooks to the UMPD.DLL *//
//*====================================================================*//
PVOID
TSsplHookGetProcAddress(
IN HMODULE hModule,
IN LPCSTR lpProcName
)
/*++
Routine Description:
Redirect Spoolss.dll function in hUMPD to winspool.drv for dynamic load
--*/
{
PVOID p;
DWORD dllNameCount;
WCHAR dllName[MAX_PATH];
p = GetProcAddress(hModule, lpProcName);
if (p &&
(dllNameCount = GetModuleFileName(hModule, dllName, sizeof(dllName)/sizeof(WCHAR))) &&
(wcsstr(_wcsupr(dllName), SPOOLSS_DLL_W) )
) {
/*
*This is SPOOLSS.DLL GetProcAddres. We need to redirect the p
*/
DBGMSG(DBG_TRACE,("TSsplHookGetProcAddress - Redirect UMPD.DLL GetProcAddress %s\n",lpProcName));
p = PlaceHookEntry(hModule, &p);
}
return p;
}
//*==========================================================================*//
//* Local functions *//
//*==========================================================================*//
BOOL
PlaceHooks(
HMODULE hUMPD,
HMODULE hSpoolss
)
/*++
Routine Description:
Redirect Spoolss.dll function in hUMPD to winspool.drv.
--*/
{
NTSTATUS st;
PVOID IATBase;
SIZE_T BigIATSize;
ULONG LittleIATSize = 0;
PVOID *ProcAddresses;
ULONG NumberOfProcAddresses;
ULONG OldProtect;
IATBase = RtlImageDirectoryEntryToData( hUMPD,
TRUE,
IMAGE_DIRECTORY_ENTRY_IAT,
&LittleIATSize
);
BigIATSize = LittleIATSize;
if (IATBase != NULL) {
st = NtProtectVirtualMemory( NtCurrentProcess(),
&IATBase,
&BigIATSize,
PAGE_READWRITE,
&OldProtect
);
if (!NT_SUCCESS(st)) {
return FALSE;
} else {
ProcAddresses = (PVOID *)IATBase;
NumberOfProcAddresses = (ULONG)(BigIATSize / sizeof(PVOID));
while (NumberOfProcAddresses--) {
/*
* Redirect the LoadLibrary and GetProcAddress function. We will
* have a chance to replace the address when the UMPD is a
* dynamically load
* DLL
*/
if (*ProcAddresses == GetProcAddress) {
*ProcAddresses = TSsplHookGetProcAddress;
} else {
/* Replace the (static linked) spoolss.dll entry */
*ProcAddresses = PlaceHookEntry(hSpoolss, ProcAddresses);
}
ProcAddresses += 1;
}
NtProtectVirtualMemory( NtCurrentProcess(),
&IATBase,
&BigIATSize,
OldProtect,
&OldProtect
);
}
}
return TRUE;
}
PVOID
PlaceHookEntry(HMODULE hSpoolss,
PVOID * pProcAddress
)
/*--
Routine Description:
This routine redirect the pProcAddress SPOOLSS.DLL function to the
corresponding winspool.drv function.
Arguments:
Return Value:
The corresponding winspool.drv function if found. Else, the original
function
--*/
{
/*
* Print jobs functions
*/
if ((GetProcAddress(hSpoolss,"SetJobW")) == *pProcAddress) {
return (&SetJobW);
}
if ((GetProcAddress(hSpoolss,"GetJobW")) == *pProcAddress) {
return(&GetJobW);
}
if ((GetProcAddress(hSpoolss,"WritePrinter")) == *pProcAddress) {
return(&WritePrinter);
}
if ((GetProcAddress(hSpoolss,"EnumJobsW")) == *pProcAddress) {
return(&EnumJobsW);
}
if ((GetProcAddress(hSpoolss,"AddJobW")) == *pProcAddress) {
return(&AddJobW);
}
if ((GetProcAddress(hSpoolss,"ScheduleJob")) == *pProcAddress) {
return(&ScheduleJob);
}
/*
* Manage Printers
*/
if ((GetProcAddress(hSpoolss,"EnumPrintersW")) == *pProcAddress) {
return(&EnumPrintersW);
}
if (((BOOL (*)())GetProcAddress(hSpoolss,"AddPrinterW")) == *pProcAddress) {
return(&AddPrinterW);
}
if ((GetProcAddress(hSpoolss,"DeletePrinter")) == *pProcAddress) {
return(&DeletePrinter);
}
if ((GetProcAddress(hSpoolss,"SetPrinterW")) == *pProcAddress) {
return(&SetPrinterW);
}
if ((GetProcAddress(hSpoolss,"GetPrinterW")) == *pProcAddress) {
return(&GetPrinterW);
}
/*
* Printer Data functions
*/
if ((GetProcAddress(hSpoolss,"GetPrinterDataW")) == *pProcAddress) {
return(&GetPrinterDataW);
}
#if(WINVER >= 0x0500)
if ((GetProcAddress(hSpoolss,"GetPrinterDataExW")) == *pProcAddress) {
return(&GetPrinterDataExW);
}
#endif
if ((GetProcAddress(hSpoolss,"EnumPrinterDataW")) == *pProcAddress) {
return(&EnumPrinterDataW);
}
#if(WINVER >= 0x0500)
if ((GetProcAddress(hSpoolss,"EnumPrinterDataExW")) == *pProcAddress) {
return(&EnumPrinterDataExW);
}
if ((GetProcAddress(hSpoolss,"EnumPrinterKeyW")) == *pProcAddress) {
return(&EnumPrinterKeyW);
}
#endif
if ((GetProcAddress(hSpoolss,"DeletePrinterDataW")) == *pProcAddress) {
return(&DeletePrinterDataW);
}
#if(WINVER >= 0x0500)
if ((GetProcAddress(hSpoolss,"DeletePrinterDataExW")) == *pProcAddress) {
return(&DeletePrinterDataExW);
}
if ((GetProcAddress(hSpoolss,"DeletePrinterKeyW")) == *pProcAddress) {
return(&DeletePrinterKeyW);
}
#endif
if ((GetProcAddress(hSpoolss,"SetPrinterDataW")) == *pProcAddress) {
return(&SetPrinterDataW);
}
#if(WINVER >= 0x0500)
if ((GetProcAddress(hSpoolss,"SetPrinterDataExW")) == *pProcAddress) {
return(&SetPrinterDataExW);
}
#endif
/*
* PrinterConnection functions
*/
if ((GetProcAddress(hSpoolss,"AddPrinterConnectionW")) == *pProcAddress) {
return(&AddPrinterConnectionW);
}
if ((GetProcAddress(hSpoolss,"DeletePrinterConnectionW")) == *pProcAddress) {
return(&DeletePrinterConnectionW);
}
/*
* Driver functions
*/
if ((GetProcAddress(hSpoolss,"GetPrinterDriverDirectoryW")) == *pProcAddress) {
return(&GetPrinterDriverDirectoryW);
}
if ((GetProcAddress(hSpoolss,"GetPrinterDriverW")) == *pProcAddress) {
return(&GetPrinterDriverW);
}
if ((GetProcAddress(hSpoolss,"AddPrinterDriverW")) == *pProcAddress) {
return(&AddPrinterDriverW);
}
#if(WINVER >= 0x0500)
if ((GetProcAddress(hSpoolss,"AddPrinterDriverExW")) == *pProcAddress) {
return(&AddPrinterDriverExW);
}
#endif
if ((GetProcAddress(hSpoolss,"EnumPrinterDriversW")) == *pProcAddress) {
return(&EnumPrinterDriversW);
}
if ((GetProcAddress(hSpoolss,"DeletePrinterDriverW")) == *pProcAddress) {
return(&DeletePrinterDriverW);
}
#if(WINVER >= 0x0500)
if ((GetProcAddress(hSpoolss,"DeletePrinterDriverExW")) == *pProcAddress) {
return(&DeletePrinterDriverExW);
}
#endif
/*
* Print Processors
*/
if ((GetProcAddress(hSpoolss,"AddPrintProcessorW")) == *pProcAddress) {
return(&AddPrintProcessorW);
}
if ((GetProcAddress(hSpoolss,"EnumPrintProcessorsW")) == *pProcAddress) {
return(&EnumPrintProcessorsW);
}
if ((GetProcAddress(hSpoolss,"GetPrintProcessorDirectoryW")) == *pProcAddress) {
return(&GetPrintProcessorDirectoryW);
}
if ((GetProcAddress(hSpoolss,"DeletePrintProcessorW")) == *pProcAddress) {
return(&DeletePrintProcessorW);
}
if ((GetProcAddress(hSpoolss,"EnumPrintProcessorDatatypesW")) == *pProcAddress) {
return(&EnumPrintProcessorDatatypesW);
}
if ((GetProcAddress(hSpoolss,"OpenPrinterW")) == *pProcAddress) {
return(&OpenPrinterW);
}
if ((GetProcAddress(hSpoolss,"ResetPrinterW")) == *pProcAddress) {
return(&ResetPrinterW);
}
if ((GetProcAddress(hSpoolss,"ClosePrinter")) == *pProcAddress) {
return(&ClosePrinter);
}
if ((GetProcAddress(hSpoolss,"AddPrintProcessorW")) == *pProcAddress) {
return(&AddPrintProcessorW);
}
/*
* Doc Printer
*/
if ((GetProcAddress(hSpoolss,"StartDocPrinterW")) == *pProcAddress) {
return(&StartDocPrinterW);
}
if ((GetProcAddress(hSpoolss,"StartPagePrinter")) == *pProcAddress) {
return(&StartPagePrinter);
}
if ((GetProcAddress(hSpoolss,"EndPagePrinter")) == *pProcAddress) {
return(&EndPagePrinter);
}
if ((GetProcAddress(hSpoolss,"WritePrinter")) == *pProcAddress) {
return(&WritePrinter);
}
#if(WINVER >= 0x0500)
if ((GetProcAddress(hSpoolss,"FlushPrinter")) == *pProcAddress) {
return(&FlushPrinter);
}
#endif
if ((GetProcAddress(hSpoolss,"AbortPrinter")) == *pProcAddress) {
return(&AbortPrinter);
}
if ((GetProcAddress(hSpoolss,"ReadPrinter")) == *pProcAddress) {
return(&ReadPrinter);
}
if ((GetProcAddress(hSpoolss,"EndDocPrinter")) == *pProcAddress) {
return(&EndDocPrinter);
}
/*
* Change functions
*/
if ((GetProcAddress(hSpoolss,"WaitForPrinterChange")) == *pProcAddress) {
return(&WaitForPrinterChange);
}
if ((GetProcAddress(hSpoolss,"FindClosePrinterChangeNotification")) == *pProcAddress) {
return(&FindClosePrinterChangeNotification);
}
/*
* Forms and port
*/
if ((GetProcAddress(hSpoolss,"AddFormW")) == *pProcAddress) {
return(&AddFormW);
}
if ((GetProcAddress(hSpoolss,"DeleteFormW")) == *pProcAddress) {
return(&DeleteFormW);
}
if ((GetProcAddress(hSpoolss,"GetFormW")) == *pProcAddress) {
return(&GetFormW);
}
if ((GetProcAddress(hSpoolss,"SetFormW")) == *pProcAddress) {
return(&SetFormW);
}
if ((GetProcAddress(hSpoolss,"EnumFormsW")) == *pProcAddress) {
return(&EnumFormsW);
}
if ((GetProcAddress(hSpoolss,"EnumPortsW")) == *pProcAddress) {
return(&EnumPortsW);
}
if ((GetProcAddress(hSpoolss,"EnumMonitorsW")) == *pProcAddress) {
return(&EnumMonitorsW);
}
if ((GetProcAddress(hSpoolss,"AddPortW")) == *pProcAddress) {
return(&AddPortW);
}
if ((GetProcAddress(hSpoolss,"ConfigurePortW")) == *pProcAddress) {
return(&ConfigurePortW);
}
if ((GetProcAddress(hSpoolss,"DeletePortW")) == *pProcAddress) {
return(&DeletePortW);
}
if ((GetProcAddress(hSpoolss,"SetPortW")) == *pProcAddress) {
return(&SetPortW);
}
if ((GetProcAddress(hSpoolss,"AddMonitorW")) == *pProcAddress) {
return(&AddMonitorW);
}
if ((GetProcAddress(hSpoolss,"DeleteMonitorW")) == *pProcAddress) {
return(&DeleteMonitorW);
}
if ((GetProcAddress(hSpoolss,"AddPrintProvidorW")) == *pProcAddress) {
return(&AddPrintProvidorW);
}
return *pProcAddress;
}

Reference in New Issue View Git Blame Copy Permalink