windows-nt/Source/XPSP1/NT/printscan/print/spooler/localspl/monitor.c
2020-09-26 16:20:57 +08:00

1784 lines
42 KiB
C

/*++
Copyright (c) 1990 - 1996 Microsoft Corporation
Module Name:
monitor.c
Abstract:
This module contains all code for Monitor-based Spooler apis
LocalEnumPorts
LocalAddMonitor
LocalDeleteMonitor
LocalEnumMonitors
LocalAddPort
LocalConfigurePort
LocalDeletePort
Support Functions in monitor.c - (Warning! Do Not Add to this list!!)
CopyIniMonitorToMonitor -- KrishnaG
GetMonitorSize -- KrishnaG
Author:
Dave Snipp (DaveSn) 15-Mar-1991
Revision History:
Khaled Sedky (khaleds) 15-March-2000
- Added LocalSendRecvBidiData
Muhunthan Sivapragasam (MuhuntS) 15-Jun-1995
- Port info 2 changes
Krishna Ganugapati (KrishnaG) 2-Feb-1994
- reorganized the entire source file
Matthew Felton (mattfe) June 1994 pIniSpooler
--*/
#include <precomp.h>
#include <offsets.h>
#include <clusspl.h>
//
// Private declarations
//
HDESK ghdeskServer = NULL;
//
// Function declarations
//
LPBYTE
CopyIniMonitorToMonitor(
PINIMONITOR pIniMonitor,
DWORD Level,
LPBYTE pMonitorInfo,
LPBYTE pEnd
);
DWORD
GetMonitorSize(
PINIMONITOR pIniMonitor,
DWORD Level
);
BOOL
LocalEnumPorts(
LPWSTR pName,
DWORD Level,
LPBYTE pPorts,
DWORD cbBuf,
LPDWORD pcbNeeded,
LPDWORD pcReturned
)
{
PINISPOOLER pIniSpooler;
BOOL bReturn;
pIniSpooler = FindSpoolerByNameIncRef( pName, NULL );
if( !pIniSpooler ){
return ROUTER_UNKNOWN;
}
bReturn = SplEnumPorts( pName,
Level,
pPorts,
cbBuf,
pcbNeeded,
pcReturned,
pIniSpooler );
FindSpoolerByNameDecRef( pIniSpooler );
return bReturn;
}
VOID
SplReenumeratePorts(
)
{
LPVOID pBuf;
DWORD cbNeeded, dwDontCare;
//
// EnumPorts checks for new ports enumerated by port monitors and updates
// localspl pIniPorts list
//
if ( !SplEnumPorts(NULL, 1, NULL, 0, &cbNeeded,
&dwDontCare, pLocalIniSpooler) &&
GetLastError() == ERROR_INSUFFICIENT_BUFFER &&
(pBuf = AllocSplMem(cbNeeded)) ) {
SplEnumPorts(NULL, 1, pBuf, cbNeeded, &cbNeeded,
&dwDontCare, pLocalIniSpooler);
FreeSplMem(pBuf);
}
}
BOOL
GetPortInfo2UsingPortInfo1(
PINIMONITOR pIniMonitor,
LPWSTR pName,
LPBYTE pPorts,
DWORD cbBuf,
LPDWORD pcbNeeded,
LPDWORD pcReturned
)
{
BOOL bRet;
LPPORT_INFO_1 pPortInfo1;
LPPORT_INFO_2 pPortInfo2;
DWORD cReturned;
bRet = (*pIniMonitor->Monitor2.pfnEnumPorts)(
pIniMonitor->hMonitor,
pName,
1,
pPorts,
cbBuf,
pcbNeeded,
pcReturned );
if ( !bRet ) {
//
// This is the upperbound
//
if ( GetLastError() == ERROR_INSUFFICIENT_BUFFER )
*pcbNeeded += (*pcbNeeded / sizeof(PORT_INFO_1)) *
(sizeof(PORT_INFO_2) - sizeof(PORT_INFO_1));
} else {
*pcbNeeded += *pcReturned * (sizeof(PORT_INFO_2) - sizeof(PORT_INFO_1));
if ( *pcbNeeded <= cbBuf ) {
cReturned = *pcReturned;
while ( cReturned-- ) {
pPortInfo1 = (LPPORT_INFO_1) (pPorts + cReturned * sizeof(PORT_INFO_1));
pPortInfo2 = (LPPORT_INFO_2) (pPorts + cReturned * sizeof(PORT_INFO_2));
pPortInfo2->pPortName = pPortInfo1->pName;
pPortInfo2->pMonitorName = NULL;
pPortInfo2->pDescription = NULL;
pPortInfo2->fPortType = 0;
pPortInfo2->Reserved = 0;
}
} else {
*pcReturned = 0;
bRet = FALSE;
SetLastError(ERROR_INSUFFICIENT_BUFFER);
}
}
return bRet;
}
BOOL
SplEnumPorts(
LPWSTR pName,
DWORD Level,
LPBYTE pPorts,
DWORD cbBuf,
LPDWORD pcbNeeded,
LPDWORD pcReturned,
PINISPOOLER pIniSpooler
)
{
PINIMONITOR pIniMonitor;
DWORD dwIndex, cReturned=0, cbStruct, TotalcbNeeded=0;
LPBYTE pBuffer = pPorts, pTemp;
DWORD Error=0, TempError = 0;
DWORD BufferSize=cbBuf;
LPWSTR pPortName;
PINIPORT pIniPort;
HANDLE hToken;
BOOL bRemoteCall;
if (!MyName( pName, pIniSpooler )) {
return FALSE;
}
//
// HACK: Some monitors choke if pName is non-NULL. We can make
// it NULL at this point since we know that we're using the same
// ports on the local machine.
//
pName = NULL;
if ( !ValidateObjectAccess(SPOOLER_OBJECT_SERVER,
SERVER_ACCESS_ENUMERATE,
NULL, NULL, pIniSpooler )) {
return FALSE;
}
switch (Level) {
case 1:
cbStruct = sizeof(PORT_INFO_1);
break;
case 2:
cbStruct = sizeof(PORT_INFO_2);
break;
default:
return ERROR_INVALID_LEVEL;
}
bRemoteCall = !IsLocalCall();
//
// We revert to local system context only when the caller is remote.
// The monitors may load dlls from system32 and remote users like
// guest or anonymous logon do not have sufficient privileges for that.
// This is safe to revert to self since we do not support delegation, so
// we will never use the credentials of the remote user to go remote again.
// If the caller is logged on interactively, then we do not switch the
// context. Thus, a monitor may be able to go on the network for the port
// enumeration.
//
if (bRemoteCall && !(hToken = RevertToPrinterSelf()))
{
return FALSE;
}
for ( pIniMonitor = pIniSpooler->pIniMonitor ;
pIniMonitor ;
pIniMonitor = pIniMonitor->pNext ) {
//
// Lang monitor does not have to define this
//
if ( !pIniMonitor->Monitor2.pfnEnumPorts )
continue;
*pcReturned = 0;
*pcbNeeded = 0;
if (!(*pIniMonitor->Monitor2.pfnEnumPorts)(
pIniMonitor->hMonitor,
pName,
Level,
pPorts,
BufferSize,
pcbNeeded,
pcReturned)) {
TempError = GetLastError();
//
// Level 2 is a superset of level 1. So we can make a level 1
// call if the monitor does not support it
//
if ( Level == 2 && TempError == ERROR_INVALID_LEVEL ) {
TempError = 0;
if ( !GetPortInfo2UsingPortInfo1(pIniMonitor,
pName,
pPorts,
BufferSize,
pcbNeeded,
pcReturned) )
TempError = GetLastError();
}
if ( TempError ) {
Error = TempError;
*pcReturned = 0;
if ( TempError != ERROR_INSUFFICIENT_BUFFER ) {
*pcbNeeded = 0;
break;
}
}
} else {
//
// Now we look for new ports not in pIniPort list and add them
//
EnterSplSem();
for ( dwIndex = 0, pTemp = pPorts ;
dwIndex < *pcReturned ;
++dwIndex ) {
switch ( Level ) {
case 1:
pPortName = ((LPPORT_INFO_1)pTemp)->pName;
pTemp += sizeof(PORT_INFO_1);
break;
case 2:
pPortName = ((LPPORT_INFO_2)pTemp)->pPortName;
pTemp += sizeof(PORT_INFO_2);
break;
default:
SPLASSERT(Level == 1 || Level == 2);
}
pIniPort = FindPort(pPortName, pIniSpooler);
if ( !pIniPort ) {
CreatePortEntry(pPortName, pIniMonitor, pIniSpooler);
} else if ( !pIniPort->pIniMonitor ) {
//
// If a fake port gets eventually enumerated by a monitor,
// update the pIniPort structure (USB monitor). It is no
// longer a placeholder port at this point.
//
pIniPort->pIniMonitor = pIniMonitor;
pIniPort->Status |= PP_MONITOR;
pIniPort->Status &= ~PP_PLACEHOLDER;
}
}
LeaveSplSem();
}
cReturned += *pcReturned;
pPorts += *pcReturned * cbStruct;
if (*pcbNeeded <= BufferSize)
BufferSize -= *pcbNeeded;
else
BufferSize = 0;
TotalcbNeeded += *pcbNeeded;
}
if (bRemoteCall)
{
ImpersonatePrinterClient(hToken);
}
*pcbNeeded = TotalcbNeeded;
*pcReturned = cReturned;
if (Error) {
SetLastError(Error);
return FALSE;
} else if (TotalcbNeeded > cbBuf ) {
SetLastError(ERROR_INSUFFICIENT_BUFFER);
return FALSE;
} else {
//
// Stop routing if this is a cluster'd spooler. Otherwise,
// we'll talk to win32spl, which RPCs to us again.
//
if( pIniSpooler->SpoolerFlags & SPL_TYPE_CLUSTER ){
return ROUTER_STOP_ROUTING;
}
return TRUE;
}
}
BOOL
LocalEnumMonitors(
LPWSTR pName,
DWORD Level,
LPBYTE pMonitors,
DWORD cbBuf,
LPDWORD pcbNeeded,
LPDWORD pcReturned
)
{
PINISPOOLER pIniSpooler;
BOOL bReturn;
pIniSpooler = FindSpoolerByNameIncRef( pName, NULL );
if( !pIniSpooler ){
return ROUTER_UNKNOWN;
}
bReturn = SplEnumMonitors( pName, Level, pMonitors, cbBuf,
pcbNeeded, pcReturned, pIniSpooler );
FindSpoolerByNameDecRef( pIniSpooler );
return bReturn;
}
BOOL
SplEnumMonitors(
LPWSTR pName,
DWORD Level,
LPBYTE pMonitors,
DWORD cbBuf,
LPDWORD pcbNeeded,
LPDWORD pcReturned,
PINISPOOLER pIniSpooler
)
{
PINIMONITOR pIniMonitor;
DWORD cReturned=0, cbStruct, cb;
LPBYTE pBuffer = pMonitors;
DWORD BufferSize=cbBuf, rc;
LPBYTE pEnd;
if (!MyName( pName, pIniSpooler )) {
return ROUTER_UNKNOWN;
}
if ( !ValidateObjectAccess(SPOOLER_OBJECT_SERVER,
SERVER_ACCESS_ENUMERATE,
NULL, NULL, pIniSpooler )) {
return ROUTER_UNKNOWN;
}
switch (Level) {
case 1:
cbStruct = sizeof(MONITOR_INFO_1);
break;
case 2:
cbStruct = sizeof(MONITOR_INFO_2);
break;
default:
SetLastError(ERROR_INVALID_LEVEL);
return ROUTER_UNKNOWN;
}
EnterSplSem();
for ( cb = 0, pIniMonitor = pIniSpooler->pIniMonitor ;
pIniMonitor ;
pIniMonitor = pIniMonitor->pNext ) {
//
// We'll not enumerate monitors which do not support AddPort
//
if ( pIniMonitor->Monitor2.pfnAddPort ||
pIniMonitor->Monitor2.pfnXcvOpenPort)
cb+=GetMonitorSize(pIniMonitor, Level);
}
*pcbNeeded = cb;
*pcReturned = 0;
if (cb <= cbBuf) {
pEnd=pMonitors + cbBuf;
for ( pIniMonitor = pIniSpooler->pIniMonitor ;
pIniMonitor ;
pIniMonitor = pIniMonitor->pNext ) {
//
// We'll not enumerate monitors which do not support AddPort
//
if ( !pIniMonitor->Monitor2.pfnAddPort &&
!pIniMonitor->Monitor2.pfnXcvOpenPort )
continue;
pEnd = CopyIniMonitorToMonitor(pIniMonitor, Level, pMonitors, pEnd);
switch (Level) {
case 1:
pMonitors+=sizeof(MONITOR_INFO_1);
break;
case 2:
pMonitors+=sizeof(MONITOR_INFO_2);
break;
}
(*pcReturned)++;
}
if( pIniSpooler->SpoolerFlags & SPL_TYPE_CLUSTER ){
//
// Stop routing, since we don't want any one else to report
// back monitors. If we're on the local machine now and
// we continue routing, win32spl will RPC back to ourself
// and re-enumerate the same ports.
//
rc = ROUTER_STOP_ROUTING;
} else {
rc = ROUTER_SUCCESS;
}
} else {
rc = ROUTER_UNKNOWN;
SetLastError(ERROR_INSUFFICIENT_BUFFER);
}
LeaveSplSem();
return rc;
}
BOOL
LocalAddPort(
LPWSTR pName,
HWND hWnd,
LPWSTR pMonitorName
)
{
PINISPOOLER pIniSpooler;
BOOL bReturn;
pIniSpooler = FindSpoolerByNameIncRef( pName, NULL );
if( !pIniSpooler ){
return ROUTER_UNKNOWN;
}
bReturn = SplAddPort( pName, hWnd, pMonitorName, pIniSpooler );
FindSpoolerByNameDecRef( pIniSpooler );
return bReturn;
}
BOOL
SplAddPort(
LPWSTR pName,
HWND hWnd,
LPWSTR pMonitorName,
PINISPOOLER pIniSpooler
)
{
PINIMONITOR pIniMonitor;
BOOL rc=FALSE;
if (!MyName( pName, pIniSpooler )) {
return FALSE;
}
if ( !ValidateObjectAccess(SPOOLER_OBJECT_SERVER,
SERVER_ACCESS_ADMINISTER,
NULL, NULL, pIniSpooler )) {
return FALSE;
}
EnterSplSem();
SPLASSERT( pIniSpooler->signature == ISP_SIGNATURE );
pIniMonitor = FindMonitor(pMonitorName, pIniSpooler);
LeaveSplSem();
if ( pIniMonitor ) {
if ( pIniMonitor->Monitor2.pfnAddPort )
rc = (*pIniMonitor->Monitor2.pfnAddPort)(
pIniMonitor->hMonitor,
pName,
hWnd,
pMonitorName );
else
SetLastError(ERROR_INVALID_PARAMETER);
}
else {
SetLastError(ERROR_INVALID_NAME);
}
if (rc)
rc = AddPortToSpooler(pName, pIniMonitor, pIniSpooler);
return rc;
}
BOOL
AddPortToSpooler(
PCWSTR pName,
PINIMONITOR pIniMonitor,
PINISPOOLER pIniSpooler
)
{
DWORD i, cbNeeded, cbDummy, cReturned;
PPORT_INFO_1 pPorts;
PINIPORT pIniPort;
/* If we don't already have the port in our local cache, add it:
*/
if (!(*pIniMonitor->Monitor2.pfnEnumPorts)(
pIniMonitor->hMonitor,
(PWSTR)pName,
1,
NULL,
0,
&cbNeeded,
&cReturned)) {
pPorts = AllocSplMem(cbNeeded);
if (pPorts) {
if ((*pIniMonitor->Monitor2.pfnEnumPorts)(
pIniMonitor->hMonitor,
(PWSTR)pName,
1,
(LPBYTE)pPorts,
cbNeeded,
&cbDummy,
&cReturned)) {
EnterSplSem();
for (i = 0 ; i < cReturned ; ++i) {
pIniPort = FindPort(pPorts[i].pName, pIniSpooler);
if ( !pIniPort ) {
CreatePortEntry(pPorts[i].pName, pIniMonitor, pIniSpooler);
//
// If we have a port without a monitor and it gets added at
// this time. Remove the placeholder status from it.
//
} else if ( !pIniPort->pIniMonitor ) {
pIniPort->pIniMonitor = pIniMonitor;
pIniPort->Status |= PP_MONITOR;
pIniPort->Status &= ~PP_PLACEHOLDER;
}
}
LeaveSplSem();
}
FreeSplMem(pPorts);
}
}
EnterSplSem();
SetPrinterChange(NULL,
NULL,
NULL,
PRINTER_CHANGE_ADD_PORT,
pIniSpooler);
LeaveSplSem();
return TRUE;
}
BOOL
LocalConfigurePort(
LPWSTR pName,
HWND hWnd,
LPWSTR pPortName
)
{
PINISPOOLER pIniSpooler;
BOOL bReturn;
pIniSpooler = FindSpoolerByNameIncRef( pName, NULL );
if( !pIniSpooler ){
return ROUTER_UNKNOWN;
}
bReturn = SplConfigurePort( pName, hWnd, pPortName, pIniSpooler );
FindSpoolerByNameDecRef( pIniSpooler );
return bReturn;
}
BOOL
SplConfigurePort(
LPWSTR pName,
HWND hWnd,
LPWSTR pPortName,
PINISPOOLER pIniSpooler
)
{
PINIPORT pIniPort;
BOOL rc;
if (!MyName( pName, pIniSpooler )) {
return FALSE;
}
if ( !ValidateObjectAccess(SPOOLER_OBJECT_SERVER,
SERVER_ACCESS_ADMINISTER,
NULL, NULL, pIniSpooler )) {
return FALSE;
}
EnterSplSem();
pIniPort = FindPort(pPortName, pIniSpooler);
LeaveSplSem();
if ((pIniPort) && (pIniPort->Status & PP_MONITOR)) {
if ( !pIniPort->pIniMonitor->Monitor2.pfnConfigurePort ) {
SetLastError(ERROR_NOT_SUPPORTED);
return FALSE;
}
if (rc = (*pIniPort->pIniMonitor->Monitor2.pfnConfigurePort)(
pIniPort->pIniMonitor->hMonitor,
pName,
hWnd,
pPortName)) {
EnterSplSem();
SetPrinterChange(NULL,
NULL,
NULL,
PRINTER_CHANGE_CONFIGURE_PORT,
pIniSpooler);
LeaveSplSem();
}
return rc;
}
SetLastError(ERROR_UNKNOWN_PORT);
return FALSE;
}
BOOL
LocalDeletePort(
LPWSTR pName,
HWND hWnd,
LPWSTR pPortName
)
{
PINISPOOLER pIniSpooler;
BOOL bReturn;
pIniSpooler = FindSpoolerByNameIncRef( pName, NULL );
if( !pIniSpooler ){
return ROUTER_UNKNOWN;
}
bReturn = SplDeletePort( pName,
hWnd,
pPortName,
pIniSpooler );
FindSpoolerByNameDecRef( pIniSpooler );
return bReturn;
}
BOOL
SplDeletePort(
LPWSTR pName,
HWND hWnd,
LPWSTR pPortName,
PINISPOOLER pIniSpooler
)
{
PINIPORT pIniPort;
BOOL rc=FALSE;
if (!MyName( pName, pIniSpooler )) {
return FALSE;
}
if ( !ValidateObjectAccess(SPOOLER_OBJECT_SERVER,
SERVER_ACCESS_ADMINISTER,
NULL, NULL, pIniSpooler )) {
return FALSE;
}
EnterSplSem();
pIniPort = FindPort(pPortName, pIniSpooler);
if ( !pIniPort || !(pIniPort->Status & PP_MONITOR) ) {
LeaveSplSem();
SetLastError(ERROR_UNKNOWN_PORT);
return FALSE;
}
if( !pIniPort->pIniMonitor->Monitor2.pfnDeletePort ){
LeaveSplSem();
SetLastError( ERROR_NOT_SUPPORTED );
return FALSE;
}
rc = DeletePortFromSpoolerStart( pIniPort );
LeaveSplSem();
if (!rc)
goto Cleanup;
rc = (*pIniPort->pIniMonitor->Monitor2.pfnDeletePort)(
pIniPort->pIniMonitor->hMonitor,
pName,
hWnd,
pPortName);
rc = DeletePortFromSpoolerEnd(pIniPort, pIniSpooler, rc);
Cleanup:
SplOutSem();
return rc;
}
BOOL
DeletePortFromSpoolerEnd(
PINIPORT pIniPort,
PINISPOOLER pIniSpooler,
BOOL bSuccess
)
{
EnterSplSem();
if(bSuccess) {
DeletePortEntry( pIniPort );
//
// Success, delete the port data and send a notification.
//
SetPrinterChange( NULL,
NULL,
NULL,
PRINTER_CHANGE_DELETE_PORT,
pIniSpooler );
} else {
//
// Add it back. If the name is already used (e.g., just added
// while we were out of the critical section), we're in trouble,
// but there's not much we can do about it. (When we restart,
// we'll re-enumerate the duplicate name from the monitors
// anyway.)
//
DBGMSG( DBG_WARN, ( "SplDeletePort: port.DeletePort failed %d\n", GetLastError()));
LinkPortToSpooler( pIniPort, pIniSpooler );
}
LeaveSplSem();
SplOutSem();
return bSuccess;
}
BOOL
DeletePortFromSpoolerStart(
PINIPORT pIniPort
)
{
BOOL rc = FALSE;
PINISPOOLER pIniSpooler = pIniPort->pIniSpooler;
SplInSem();
if ( pIniPort->cPrinters || pIniPort->cRef || pIniPort->cJobs ) {
SetLastError(ERROR_BUSY);
goto Cleanup;
}
//
// Remove it from the linked list so that no one will try to grab
// a reference to while we're deleting it.
//
DelinkPortFromSpooler( pIniPort, pIniSpooler );
rc = TRUE;
Cleanup:
return rc;
}
BOOL
LocalAddMonitor(
LPWSTR pName,
DWORD Level,
LPBYTE pMonitorInfo
)
{
PINISPOOLER pIniSpooler;
BOOL bReturn;
pIniSpooler = FindSpoolerByNameIncRef( pName, NULL );
if( !pIniSpooler ){
return ROUTER_UNKNOWN;
}
bReturn = SplAddMonitor( pName,
Level,
pMonitorInfo,
pIniSpooler );
FindSpoolerByNameDecRef( pIniSpooler );
return bReturn;
}
BOOL
SplAddMonitor(
LPWSTR pName,
DWORD Level,
LPBYTE pMonitorInfo,
PINISPOOLER pIniSpooler
)
{
PINIMONITOR pIniMonitor;
PMONITOR_INFO_2 pMonitor = (PMONITOR_INFO_2)pMonitorInfo;
HANDLE hToken;
HKEY hKey;
LONG Status;
BOOL rc = FALSE;
DWORD dwPathLen = 0;
if (!MyName( pName, pIniSpooler )) {
return FALSE;
}
if ( !ValidateObjectAccess(SPOOLER_OBJECT_SERVER,
SERVER_ACCESS_ADMINISTER,
NULL, NULL, pIniSpooler )) {
return FALSE;
}
if (Level != 2) {
SetLastError( ERROR_INVALID_LEVEL );
return FALSE;
}
if (!pMonitor ||
!pMonitor->pName ||
!*pMonitor->pName) {
SetLastError( ERROR_INVALID_PARAMETER );
return FALSE;
}
if (!pMonitor->pEnvironment ||
!*pMonitor->pEnvironment ||
lstrcmpi(pMonitor->pEnvironment, szEnvironment)) {
SetLastError( ERROR_INVALID_ENVIRONMENT );
return FALSE;
}
if (!pMonitor->pDLLName ||
!*pMonitor->pDLLName ){
SetLastError( ERROR_INVALID_PARAMETER );
return FALSE;
}
EnterSplSem();
if (FindMonitor(pMonitor->pName, pIniSpooler)) {
LeaveSplSem();
SetLastError(ERROR_PRINT_MONITOR_ALREADY_INSTALLED);
return FALSE;
}
hToken = RevertToPrinterSelf();
pIniMonitor = CreateMonitorEntry(pMonitor->pDLLName,
pMonitor->pName,
pIniSpooler);
if (pIniMonitor != (PINIMONITOR)-1) {
WCHAR szRegistryRoot[MAX_PATH];
PINISPOOLER pIniSpoolerMonitor;
HANDLE hKeyOut;
LPCWSTR pszPathOut;
//
// Note that even though this is built once per pIniSpooler, the
// list of monitors is the same for all spoolers. However, the
// ports that the monitor returns from EnumPorts is different for
// each pIniSpooler (for clustering).
//
// If it's not local, then it could be a cached win32 monitor.
//
if( pIniSpooler->SpoolerFlags & SPL_TYPE_LOCAL ){
pIniSpoolerMonitor = pLocalIniSpooler;
} else {
pIniSpoolerMonitor = pIniSpooler;
}
//
// Build the registry path. In some cases it's a relative
// path from hckRoot; other times it's a hard coded path from
// HKLM (e.g., win32spl).
//
GetRegistryLocation( pIniSpoolerMonitor->hckRoot,
pIniSpoolerMonitor->pszRegistryMonitors,
&hKeyOut,
&pszPathOut );
Status = StrNCatBuff( szRegistryRoot,
COUNTOF(szRegistryRoot),
pszPathOut,
L"\\",
pMonitor->pName,
NULL );
if (Status == ERROR_SUCCESS)
{
Status = RegCreateKeyEx( hKeyOut,
szRegistryRoot,
0,
NULL,
0,
KEY_WRITE,
NULL,
&hKey,
NULL );
if (Status == ERROR_SUCCESS) {
Status = RegSetValueEx( hKey,
L"Driver",
0,
REG_SZ,
(LPBYTE)pMonitor->pDLLName,
(wcslen(pMonitor->pDLLName) + 1)*sizeof(WCHAR));
if (Status == ERROR_SUCCESS) {
rc = TRUE;
} else {
SetLastError( Status );
}
RegCloseKey(hKey);
} else {
SetLastError( Status );
}
}
else
{
SetLastError(Status);
}
}
ImpersonatePrinterClient(hToken);
//
// Bug 54843 if this fails we could still have a IniMonitor on the linked list that
// is BAD, it should be removed. MattFe 19th Jan 95
// Note *maybe* we do this because a monitor might fail to initialize
// but will correctly function next time you reboot, like hpmon ( dlc doesn't become active until
// the next reboot. Please Verify.
LeaveSplSem();
if ( !rc ) {
DBGMSG( DBG_WARNING, ("SplAddMonitor failed %d\n", GetLastError() ));
}
return rc;
}
BOOL
LocalDeleteMonitor(
LPWSTR pName,
LPWSTR pEnvironment,
LPWSTR pMonitorName
)
{
PINISPOOLER pIniSpooler;
BOOL bReturn;
pIniSpooler = FindSpoolerByNameIncRef( pName, NULL );
if( !pIniSpooler ){
return ROUTER_UNKNOWN;
}
bReturn = SplDeleteMonitor( pName,
pEnvironment,
pMonitorName,
pIniSpooler );
FindSpoolerByNameDecRef( pIniSpooler );
return bReturn;
}
BOOL
SplDeleteMonitor(
LPWSTR pName,
LPWSTR pEnvironment,
LPWSTR pMonitorName,
PINISPOOLER pIniSpooler
)
{
BOOL Remote=FALSE;
PINIMONITOR pIniMonitor;
PINIPORT pIniPort, pIniPortNext;
HKEY hKeyMonitors, hKey;
LONG Status;
BOOL rc = FALSE;
HANDLE hToken;
HANDLE hKeyOut;
LPCWSTR pszPathOut;
if (pName && *pName) {
if (!MyName( pName, pIniSpooler )) {
return FALSE;
} else {
Remote=TRUE;
}
}
if ((pMonitorName == NULL) || (*pMonitorName == L'\0')) {
SetLastError(ERROR_INVALID_PARAMETER);
return FALSE;
}
if ( !ValidateObjectAccess(SPOOLER_OBJECT_SERVER,
SERVER_ACCESS_ADMINISTER,
NULL, NULL, pIniSpooler )) {
return FALSE;
}
EnterSplSem();
if (!(pIniMonitor=(PINIMONITOR)FindMonitor(pMonitorName,
pIniSpooler))) {
SetLastError(ERROR_UNKNOWN_PRINT_MONITOR);
LeaveSplSem();
return FALSE;
}
if ( pIniMonitor->cRef ) {
SetLastError(ERROR_PRINT_MONITOR_IN_USE);
LeaveSplSem();
return FALSE;
}
pIniPort = pIniSpooler->pIniPort;
while (pIniPort) {
if ((pIniPort->pIniMonitor == pIniMonitor) &&
(pIniPort->cPrinters || pIniPort->cRef)) {
SetLastError(ERROR_BUSY);
LeaveSplSem();
return FALSE;
}
pIniPort = pIniPort->pNext;
}
hToken = RevertToPrinterSelf();
GetRegistryLocation( pIniSpooler->hckRoot,
pIniSpooler->pszRegistryMonitors,
&hKeyOut,
&pszPathOut );
Status = SplRegOpenKey(hKeyOut,
pszPathOut,
KEY_READ | KEY_WRITE,
&hKeyMonitors,
pIniSpooler);
if (Status == ERROR_SUCCESS)
{
Status = SplRegOpenKey(hKeyMonitors,
pMonitorName,
KEY_READ | KEY_WRITE,
&hKey,
pIniSpooler);
if (Status == ERROR_SUCCESS)
{
Status = DeleteSubkeys(hKey, pIniSpooler);
SplRegCloseKey(hKey, pIniSpooler);
if (Status == ERROR_SUCCESS)
Status = SplRegDeleteKey(hKeyMonitors, pMonitorName, pIniSpooler);
}
SplRegCloseKey(hKeyMonitors, pIniSpooler);
}
if (Status == ERROR_SUCCESS) {
pIniPort = pIniSpooler->pIniPort;
while (pIniPort) {
pIniPortNext = pIniPort->pNext;
if (pIniPort->pIniMonitor == pIniMonitor)
DeletePortEntry(pIniPort);
pIniPort = pIniPortNext;
}
RemoveFromList((PINIENTRY *)&pIniSpooler->pIniMonitor,
(PINIENTRY)pIniMonitor);
FreeIniMonitor( pIniMonitor );
rc = TRUE;
}
if (Status != ERROR_SUCCESS)
SetLastError(Status);
ImpersonatePrinterClient(hToken);
LeaveSplSem();
return rc;
}
LPBYTE
CopyIniMonitorToMonitor(
PINIMONITOR pIniMonitor,
DWORD Level,
LPBYTE pMonitorInfo,
LPBYTE pEnd
)
{
LPWSTR *pSourceStrings, *SourceStrings;
DWORD j;
DWORD *pOffsets;
switch (Level) {
case 1:
pOffsets = MonitorInfo1Strings;
break;
case 2:
pOffsets = MonitorInfo2Strings;
break;
default:
return pEnd;
}
for (j=0; pOffsets[j] != -1; j++) {
}
SourceStrings = pSourceStrings = AllocSplMem(j * sizeof(LPWSTR));
if (!SourceStrings) {
DBGMSG(DBG_WARNING, ("Failed to alloc Port source strings.\n"));
return pEnd;
}
switch (Level) {
case 1:
*pSourceStrings++=pIniMonitor->pName;
break;
case 2:
*pSourceStrings++=pIniMonitor->pName;
*pSourceStrings++=szEnvironment;
*pSourceStrings++=pIniMonitor->pMonitorDll;
break;
}
pEnd = PackStrings(SourceStrings, pMonitorInfo, pOffsets, pEnd);
FreeSplMem(SourceStrings);
return pEnd;
}
DWORD
GetMonitorSize(
PINIMONITOR pIniMonitor,
DWORD Level
)
{
DWORD cb=0;
switch (Level) {
case 1:
cb=sizeof(MONITOR_INFO_1) + wcslen(pIniMonitor->pName)*sizeof(WCHAR) +
sizeof(WCHAR);
break;
case 2:
cb = wcslen(pIniMonitor->pName) + 1 + wcslen(pIniMonitor->pMonitorDll) + 1
+ wcslen(szEnvironment) + 1;
cb *= sizeof(WCHAR);
cb += sizeof(MONITOR_INFO_2);
break;
default:
cb = 0;
break;
}
return cb;
}
BOOL
LocalAddPortEx(
LPWSTR pName,
DWORD Level,
LPBYTE pBuffer,
LPWSTR pMonitorName
)
{
return ( SplAddPortEx( pName,
Level,
pBuffer,
pMonitorName,
pLocalIniSpooler ));
}
BOOL
SplAddPortEx(
LPWSTR pName,
DWORD Level,
LPBYTE pBuffer,
LPWSTR pMonitorName,
PINISPOOLER pIniSpooler
)
{
PINIMONITOR pIniMonitor;
BOOL rc=FALSE;
DWORD i, cbNeeded, cReturned, cbDummy;
PPORT_INFO_1 pPorts = NULL;
PINIPORT pIniPort;
if (!MyName( pName, pIniSpooler )) {
return FALSE;
}
if ( !ValidateObjectAccess(SPOOLER_OBJECT_SERVER,
SERVER_ACCESS_ADMINISTER,
NULL, NULL, pIniSpooler )) {
return FALSE;
}
EnterSplSem();
pIniMonitor = FindMonitor(pMonitorName, pIniSpooler);
LeaveSplSem();
if (!pIniMonitor) {
SetLastError(ERROR_INVALID_NAME);
return(FALSE);
}
if (pIniMonitor->Monitor2.pfnAddPortEx) {
rc = (*pIniMonitor->Monitor2.pfnAddPortEx)(
pIniMonitor->hMonitor,
pName,
Level,
pBuffer,
pMonitorName);
}
if (!rc) {
return(FALSE);
}
if (!(*pIniMonitor->Monitor2.pfnEnumPorts)(
pIniMonitor->hMonitor,
pName,
1,
NULL,
0,
&cbNeeded,
&cReturned)) {
pPorts = AllocSplMem(cbNeeded);
}
if (pPorts) {
if ((*pIniMonitor->Monitor2.pfnEnumPorts)(
pIniMonitor->hMonitor,
pName,
1,
(LPBYTE)pPorts,
cbNeeded,
&cbDummy,
&cReturned)) {
EnterSplSem();
for (i = 0; i < cReturned; i++) {
pIniPort = FindPort(pPorts[i].pName, pIniSpooler);
if ( !pIniPort ) {
CreatePortEntry(pPorts[i].pName, pIniMonitor, pIniSpooler);
} else if ( !pIniPort->pIniMonitor ) {
pIniPort->pIniMonitor = pIniMonitor;
pIniPort->Status |= PP_MONITOR;
}
}
LeaveSplSem();
}
FreeSplMem(pPorts);
}
EnterSplSem();
SetPrinterChange(NULL,
NULL,
NULL,
PRINTER_CHANGE_ADD_PORT,
pIniSpooler);
LeaveSplSem();
return rc;
}
VOID
LinkPortToSpooler(
PINIPORT pIniPort,
PINISPOOLER pIniSpooler
)
/*++
Routine Description:
Links a pIniPort onto the pIniSpooler.
Arguments:
pIniPort - Port to link; must not already be on a ll.
pIniSpooler - Provides ll for pIniPort.
Return Value:
--*/
{
SplInSem();
SPLASSERT( !pIniPort->pIniSpooler );
pIniPort->pNext = pIniSpooler->pIniPort;
pIniPort->pIniSpooler = pIniSpooler;
pIniSpooler->pIniPort = pIniPort;
}
VOID
DelinkPortFromSpooler(
PINIPORT pIniPort,
PINISPOOLER pIniSpooler
)
/*++
Routine Description:
Remove a pIniPort from a pIniSpooler->pIniPort linked list. The
pIniPort may or may not be on the list; if it isn't, then this
routine does nothing.
Generic delink code ripped out into a subroutine.
The refcount on pIniPort must be zero. Anyone that uses pIniPort
must hold a reference, since it may be deleted outside the
SplSem when cRef==0.
Arguments:
pIniPort - Port to delink from the list. May or may not be on
pIniSpooler->pIniPort.
pIniSpooler - Linked list from which the pIniPort will be removed.
Return Value:
--*/
{
PINIPORT *ppCurPort;
SplInSem();
SPLASSERT( !pIniPort->cRef );
//
// Keep searching for pIniPort until we hit the end of the
// list or we've found it.
//
for( ppCurPort = &pIniSpooler->pIniPort;
*ppCurPort && *ppCurPort != pIniPort;
ppCurPort = &((*ppCurPort)->pNext )){
; // Don't do anything.
}
//
// If we found it, delink it.
//
if( *ppCurPort ){
*ppCurPort = (*ppCurPort)->pNext;
//
// Null out the back pointer since we have removed it from
// the pIniSpooler.
//
pIniPort->pIniSpooler = NULL;
}
}
/*++
Function Name:
LocalSendRecvBidiData
Description:
This function is the providor point of communicating with
Monitors supporting BIDI data. It allows the providor to
set data in the printer and query data from the printer
Parameters:
hPrinter : This could be a Printer/Port Handle
dwAccessBit : Priverledges allowed to the accessing thread
pAction :
pReqData : Request encapsulatig the queries in an array
ppResData : Response returned to client in an array of Data
Return Value:
Win32 Error Code
--*/
DWORD
LocalSendRecvBidiData(
IN HANDLE hPrinter,
IN LPCTSTR pAction,
IN PBIDI_REQUEST_CONTAINER pReqData,
OUT PBIDI_RESPONSE_CONTAINER* ppResData
)
{
DWORD dwRet = ERROR_SUCCESS;
PSPOOL pSpool = (PSPOOL)hPrinter;
PINIPORT pIniPort = NULL;
PINIMONITOR pIniMonitor = NULL;
EnterSplSem();
{
//
// Process of validating the parameters
//
if((!pAction || !*pAction) ||
(!pReqData && !ppResData))
{
dwRet = ERROR_INVALID_PARAMETER;
}
else
{
if (!ValidateSpoolHandle( pSpool, PRINTER_HANDLE_SERVER ))
{
dwRet = ERROR_INVALID_HANDLE;
}
else
{
if(pSpool->TypeofHandle & PRINTER_HANDLE_PRINTER)
{
PINIPRINTER pIniPrinter;
PINIMONITOR pIniLangMonitor = NULL;
if(pIniPrinter = pSpool->pIniPrinter)
{
pIniPort = FindIniPortFromIniPrinter( pIniPrinter );
if (pIniPort)
{
pIniLangMonitor = pIniPrinter->pIniDriver->pIniLangMonitor;
if ( pIniLangMonitor &&
!pIniLangMonitor->Monitor2.pfnSendRecvBidiDataFromPort )
pIniLangMonitor = NULL;
//
// Port needs to be opened?
//
if ( pIniPort->pIniLangMonitor != pIniLangMonitor ||
!pIniPort->hPort )
{
LPTSTR pszPrinter;
TCHAR szFullPrinter[ MAX_UNC_PRINTER_NAME ];
if( pIniPrinter->pIniSpooler->SpoolerFlags & SPL_TYPE_CLUSTER )
{
pszPrinter = szFullPrinter;
wsprintf( szFullPrinter,
L"%ws\\%ws",
pIniPrinter->pIniSpooler->pMachineName,
pIniPrinter->pName );
} else {
pszPrinter = pIniPrinter->pName;
}
if ( !OpenMonitorPort(pIniPort,
& pIniLangMonitor,
pszPrinter,
FALSE) ) {
dwRet = ERROR_INVALID_HANDLE;
}
}
}
else
dwRet = ERROR_INVALID_HANDLE;
}
}
else if(pSpool->TypeofHandle & PRINTER_HANDLE_PORT)
{
pIniPort = pSpool->pIniPort;
}
if(dwRet == ERROR_SUCCESS && pIniPort && (pIniPort->Status & PP_MONITOR))
{
pIniMonitor = pIniPort->pIniLangMonitor ?
pIniPort->pIniLangMonitor :
pIniPort->pIniMonitor;
if(pIniMonitor)
{
//
// Calling into the monitor
//
if(pIniMonitor->Monitor2.pfnSendRecvBidiDataFromPort)
{
INCPORTREF(pIniPort);
INCMONITORREF(pIniMonitor);
LeaveSplSem();
SplOutSem();
dwRet = (*pIniMonitor->Monitor2.pfnSendRecvBidiDataFromPort)(pIniPort->hPort,
pSpool->GrantedAccess,
pAction,
pReqData,
ppResData);
EnterSplSem();
DECMONITORREF(pIniMonitor);
DECPORTREF(pIniPort);
}
else
{
//
// Here we could use a simulation code;
//
dwRet = ERROR_NOT_SUPPORTED;
}
}
else
{
dwRet = ERROR_INVALID_HANDLE;
}
}
else
{
dwRet = ERROR_INVALID_HANDLE;
}
}
}
}
LeaveSplSem();
return(dwRet);
}