windows-nt/Source/XPSP1/NT/printscan/print/spooler/localspl/addjob.c

561 lines
14 KiB
C
Raw Normal View History

2020-09-26 03:20:57 -05:00
/*++
Copyright (c) 1990-1994 Microsoft Corporation
Module Name:
addjob.c
Abstract:
This module provides all the public exported APIs relating to Printer
and Job management for the Local Print Providor. This module contains
LocalSpl's implementation of the following spooler apis
LocalAddJob
LocalScheduleJob
Author:
Dave Snipp (DaveSn) 15-Mar-1991
Revision History:
Rewritten both apis -- Krishna Ganugapati (KrishnaG) 5-Apr-1994
RapidPrint -- Matthew A Felton (mattfe) June 1994
--*/
#include <precomp.h>
#pragma hdrstop
#include "jobid.h"
#include "winsprlp.h"
#include "filepool.hxx"
VOID
AddJobEntry(
PINIPRINTER pIniPrinter,
PINIJOB pIniJob
);
BOOL
LocalAddJob(
HANDLE hPrinter,
DWORD Level,
LPBYTE pData,
DWORD cbBuf,
LPDWORD pcbNeeded
)
{
PINIPRINTER pIniPrinter;
PINIJOB pIniJob;
PSPOOL pSpool=(PSPOOL)hPrinter;
DWORD cb;
WCHAR szFileName[MAX_PATH];
LPBYTE pEnd;
DWORD LastError=0;
LPADDJOB_INFO_1 pAddJob = (LPADDJOB_INFO_1)pData;
DWORD NextId;
BOOL bRemote = FALSE;
DOC_INFO_1 DocInfo1;
BOOL bRet;
DWORD dwStatus = 0;
HANDLE hFile = INVALID_HANDLE_VALUE;
LPWSTR pMachineName = NULL;
LPWSTR pszSpoolFile = NULL;
PMAPPED_JOB pMappedJob = NULL;
SplOutSem();
switch( Level ){
case 1:
break;
case 2:
case 3:
pMachineName = (LPWSTR)( ((PBYTE)pData) +
(ULONG_PTR)((PADDJOB_INFO_2W)pData)->pData );
//
// Validate string.
//
if( pMachineName > (LPWSTR)( ((PBYTE)pData)+cbBuf )){
SetLastError( ERROR_INVALID_LEVEL );
return FALSE;
}
//
// Ensure NULL termination.
//
*(PWCHAR)(((ULONG_PTR)(pData + cbBuf - sizeof( WCHAR ))&~1)) = 0;
break;
default:
SetLastError( ERROR_INVALID_LEVEL );
return FALSE;
}
//
// memset docinfo
//
memset((LPBYTE)&DocInfo1, 0, sizeof(DOC_INFO_1));
//
// Figure out whether the job is a remote or local job
//
if (!IsLocalCall()) {
bRemote = TRUE;
}
//
// Get the name of the user
//
if (bRemote) {
DocInfo1.pDocName = szRemoteDoc;
} else{
DocInfo1.pDocName = szLocalDoc;
}
EnterSplSem();
if (!ValidateSpoolHandle(pSpool, PRINTER_HANDLE_SERVER )) {
LeaveSplSem();
return(FALSE);
}
//
// We're interested if this is a remote call (not if it was opened
// via \\server\remote). The server process does this.
//
if (pSpool->TypeofHandle & PRINTER_HANDLE_REMOTE_CALL) {
LeaveSplSem();
SetLastError(ERROR_INVALID_PARAMETER);
return(FALSE);
}
if (pSpool->TypeofHandle & PRINTER_HANDLE_PORT) {
if (pSpool->pIniPort->Status & PP_MONITOR) {
LeaveSplSem();
SetLastError(ERROR_INVALID_PARAMETER);
return(FALSE);
} else {
//
// If we had level == 2 (passing in the computer name), then
// convert back down to level 1 for old print providers.
// We don't need to fix up the structure since level 1 and 2
// are identical; it's just that level 2 is an in-out buffer.
//
//
if (Level == 2 || Level == 3) {
Level = 1;
}
//
// This is the "Local Printer masquerading as a Remote Printer"
//
LeaveSplSem();
bRet = AddJob(pSpool->hPort, Level, pData, cbBuf, pcbNeeded);
return(bRet);
}
}
pIniPrinter = pSpool->pIniPrinter;
SPLASSERT(pIniPrinter);
if (pIniPrinter->Attributes & PRINTER_ATTRIBUTE_DIRECT) {
LeaveSplSem();
SetLastError(ERROR_INVALID_ACCESS);
return(FALSE);
}
//
// Disallow EMF if PRINTER_ATTRIBUTE_RAW_ONLY is set.
//
if( pIniPrinter->Attributes & PRINTER_ATTRIBUTE_RAW_ONLY ){
LPWSTR pszDatatype = pSpool->pDatatype ?
pSpool->pDatatype :
pIniPrinter->pDatatype;
if( !ValidRawDatatype( pszDatatype )){
LeaveSplSem();
SetLastError( ERROR_INVALID_DATATYPE );
return FALSE;
}
}
NextId = GetNextId( pIniPrinter->pIniSpooler->hJobIdMap );
GetFullNameFromId(pIniPrinter, NextId, TRUE, szFileName,
pSpool->TypeofHandle & PRINTER_HANDLE_REMOTE_CALL);
cb = wcslen(szFileName)*sizeof(WCHAR) + sizeof(WCHAR) +
sizeof(ADDJOB_INFO_1);
*pcbNeeded = cb;
if (cb > cbBuf) {
// Freeup the JobId.
vMarkOff( pIniPrinter->pIniSpooler->hJobIdMap, NextId);
LeaveSplSem();
SetLastError(ERROR_INSUFFICIENT_BUFFER);
return(FALSE);
}
//
// WMI Trace Event
//
LeaveSplSem();
LogWmiTraceEvent(NextId, EVENT_TRACE_TYPE_SPL_SPOOLJOB, NULL);
EnterSplSem();
SplInSem();
dwStatus = JOB_SPOOLING | JOB_TYPE_ADDJOB;
if (Level == 2 || Level ==3) {
dwStatus |= JOB_DOWNLEVEL;
}
if ((pIniJob = CreateJobEntry(pSpool,
1,
(LPBYTE)&DocInfo1,
NextId,
bRemote,
dwStatus,
pMachineName)) == NULL) {
// Freeup the JobId.
vMarkOff( pIniPrinter->pIniSpooler->hJobIdMap, NextId);
DBGMSG(DBG_WARNING,("Error: CreateJobEntry failed in LocalAddJob\n"));
LeaveSplSem();
return(FALSE);
}
//
// Level 3 is called only by RDR/SRV. For details see LocalScheduleJob
//
pIniJob->AddJobLevel = Level;
pIniPrinter->cSpooling++;
if (pIniPrinter->cSpooling > pIniPrinter->cMaxSpooling) {
pIniPrinter->cMaxSpooling = pIniPrinter->cSpooling;
}
AddJobEntry(pIniPrinter, pIniJob);
pEnd = (LPBYTE)pAddJob+cbBuf;
pEnd -= wcslen(szFileName)*sizeof(WCHAR)+sizeof(WCHAR);
WORD_ALIGN_DOWN(pEnd);
wcscpy((LPWSTR)pEnd, szFileName);
pAddJob->Path = (LPWSTR)pEnd;
pAddJob->JobId = pIniJob->JobId;
//
// Now we want to add the job into the spools list of current jobs.
// This is so that the spool file can be deleted correctly at the end
// of the job, even if we have aborted.
//
pMappedJob = AllocSplMem(sizeof( MAPPED_JOB ));
pszSpoolFile = AllocSplMem(MAX_PATH * sizeof( WCHAR ));
if (pMappedJob && pszSpoolFile)
{
BOOL bDuplicate = FALSE;
DWORD TempJobId = pIniJob->JobId;
PMAPPED_JOB pTempMappedJob;
wcscpy(pszSpoolFile, szFileName);
//
// Run through the list and make sure we have no duplicates
//
for (pTempMappedJob = pSpool->pMappedJob;
pTempMappedJob;
pTempMappedJob = pTempMappedJob->pNext) {
if (pTempMappedJob->JobId == TempJobId) {
bDuplicate = TRUE;
break;
}
}
//
// No duplicates, add this job to the linked list.
//
if (!bDuplicate) {
pMappedJob->pszSpoolFile = pszSpoolFile;
pMappedJob->JobId = TempJobId;
pMappedJob->pNext = pSpool->pMappedJob;
pSpool->pMappedJob = pMappedJob;
} else {
FreeSplMem(pszSpoolFile);
FreeSplMem(pMappedJob);
}
}
else
{
FreeSplMem(pMappedJob);
FreeSplMem(pszSpoolFile);
}
//
//
// Storing pIniJob in pSpool is bogus since you can call AddJob multiple
// times. We should have a linked list here. If the client calls AddJob
// two times then closes the handle (no ScheduleJob), then only the last
// job is rundown and eliminated.
//
// This bug has been here since 3.1, and probably isn't worth fixing.
//
pSpool->pIniJob = pIniJob;
pSpool->Status |= SPOOL_STATUS_ADDJOB;
SetPrinterChange(pSpool->pIniPrinter,
pIniJob,
NVAddJob,
PRINTER_CHANGE_ADD_JOB | PRINTER_CHANGE_SET_PRINTER,
pSpool->pIniSpooler );
//
// If necessary Start Downlevel Size Detection thread
//
CheckSizeDetectionThread();
LeaveSplSem();
SplOutSem();
return TRUE;
}
BOOL
LocalScheduleJob(
HANDLE hPrinter,
DWORD JobId)
/*++
Routine Description:
Arguments:
Returns:
--*/
{
PSPOOL pSpool=(PSPOOL)hPrinter;
WCHAR szFileName[MAX_PATH];
PINIJOB pIniJob;
DWORD Position;
DWORD LastError = FALSE;
HANDLE hPort;
BOOL bRet;
NOTIFYVECTOR NotifyVector;
WIN32_FILE_ATTRIBUTE_DATA FileAttributeData;
COPYNV(NotifyVector, NVJobStatus);
//
// WMI Trace Event.
//
LogWmiTraceEvent(JobId, EVENT_TRACE_TYPE_SPL_TRACKTHREAD, NULL);
SplOutSem();
EnterSplSem();
if (!ValidateSpoolHandle(pSpool, PRINTER_HANDLE_SERVER )) {
LeaveSplSem();
return (FALSE);
}
if (pSpool->Status & SPOOL_STATUS_STARTDOC) {
SetLastError(ERROR_SPL_NO_ADDJOB);
LeaveSplSem();
return(FALSE);
}
if (pSpool->TypeofHandle & PRINTER_HANDLE_PORT) {
if (pSpool->pIniPort->Status & PP_MONITOR) {
SetLastError(ERROR_INVALID_ACCESS);
LeaveSplSem();
return(FALSE);
}
//
// This is the "Local Printer masquerading as the Network Printer"
//
hPort = pSpool->hPort;
LeaveSplSem();
bRet = ScheduleJob(hPort, JobId);
return(bRet);
}
if ((pIniJob = FindJob(pSpool->pIniPrinter, JobId, &Position)) == NULL) {
SetLastError(ERROR_INVALID_PARAMETER);
LeaveSplSem();
return(FALSE);
}
if (pIniJob->Status & JOB_SCHEDULE_JOB) {
DBGMSG(DBG_WARNING, ("ScheduleJob: job 0x%x (id = %d) already scheduled\n",
pIniJob, pIniJob->JobId));
SetLastError(ERROR_INVALID_PARAMETER);
LeaveSplSem();
return FALSE;
}
if (!(pIniJob->Status & JOB_TYPE_ADDJOB)) {
DBGMSG(DBG_WARNING, ("ScheduleJob: job 0x%x (id = %d) no addjob\n",
pIniJob, pIniJob->JobId));
SetLastError(ERROR_SPL_NO_ADDJOB);
LeaveSplSem();
return(FALSE);
}
pIniJob->Status |= JOB_SCHEDULE_JOB;
if (pIniJob->Status & JOB_SPOOLING) {
pIniJob->Status &= ~JOB_SPOOLING;
pIniJob->pIniPrinter->cSpooling--;
}
if ( pIniJob->Status & JOB_TIMEOUT ) {
pIniJob->Status &= ~( JOB_TIMEOUT | JOB_ABANDON );
FreeSplStr(pIniJob->pStatus);
pIniJob->pStatus = NULL;
}
SplInSem();
// Despooling whilst spooling requires us to wake the writing
// thread if it is waiting.
if ( pIniJob->WaitForWrite != NULL )
SetEvent(pIniJob->WaitForWrite);
// Release any thread waiting on SeekPrinter for this job.
SeekPrinterSetEvent(pIniJob, NULL, TRUE);
SPLASSERT(pIniJob->cRef != 0);
DECJOBREF(pIniJob);
DBGMSG(DBG_TRACE, ("ScheduleJob:cRef = %d\n", pIniJob->cRef));
//
// FP Change
// For File pools, we know the Filename of the spool file, so
// we can just copy it in.
//
if ( pIniJob->pszSplFileName )
{
wcsncpy(szFileName, pIniJob->pszSplFileName, COUNTOF(szFileName));
}
else
{
GetFullNameFromId(pSpool->pIniPrinter, pIniJob->JobId, TRUE,
szFileName, FALSE);
}
bRet = GetFileAttributesEx(szFileName,
GetFileExInfoStandard,
&FileAttributeData);
//
// According to MSDN: The ScheduleJob function checks for a valid spool file.
// If there is an invalid spool file, or if it is empty, ScheduleJob deletes
// both the spool file and the corresponding print job entry in the print spooler.
//
// The RDR/SRV will call AddJob even if the caller of CreateFile did noy request
// WRITE access. This will cause us at add a job, but nobody will ever write to
// the spooler file. In this case, we delete the job. For this reason we have
// level 3 for AddJob. Level 3 is meant to be used only by RDR/SRV.
//
if (!bRet ||
!(FileAttributeData.nFileSizeLow || FileAttributeData.nFileSizeHigh) && pIniJob->AddJobLevel == 3) {
DBGMSG(DBG_WARNING, ("Could not GetFileAttributesEx %ws in ScheduleJob or file size is 0\n", szFileName));
DeleteJob(pIniJob, BROADCAST);
pSpool->pIniJob = NULL;
pSpool->Status &= ~SPOOL_STATUS_ADDJOB;
LeaveSplSem();
//
// If we deleted the job because the spool file was empty and the job came via RDR/SRV
// In this case we return success.
//
if (bRet)
{
return TRUE;
}
//
// We delete the job because the spool is not found
//
SetLastError(ERROR_SPOOL_FILE_NOT_FOUND);
return(FALSE);
}
//
// If size changed, we must update our size
// and potentially notify people.
//
if (pIniJob->Size != FileAttributeData.nFileSizeLow) {
ADDNV(NotifyVector, NVSpoolJob);
pIniJob->Size = FileAttributeData.nFileSizeLow;
}
WriteShadowJob(pIniJob, FALSE);
if (pIniJob->Status & JOB_PENDING_DELETION) {
DBGMSG(DBG_TRACE, ("LocalScheduleJob: Deleting Job because its pending deletion\n"));
DeleteJob(pIniJob, BROADCAST);
} else {
CHECK_SCHEDULER();
SetPrinterChange(pIniJob->pIniPrinter,
pIniJob,
NotifyVector,
PRINTER_CHANGE_SET_JOB,
pIniJob->pIniPrinter->pIniSpooler );
}
pSpool->pIniJob = NULL;
pSpool->Status &= ~SPOOL_STATUS_ADDJOB;
LeaveSplSem();
SplOutSem();
return(TRUE);
}