mirror of https://github.com/lianthony/NT4.0
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
119 lines
2.8 KiB
119 lines
2.8 KiB
/*++
|
|
Module Name:
|
|
|
|
windows\spooler\prtprocs\winprint\support.c
|
|
|
|
Abstract:
|
|
|
|
Support routines for WinPrint.
|
|
|
|
Author:
|
|
|
|
Tommy Evans (vtommye) 10-22-1993
|
|
|
|
Revision History:
|
|
|
|
--*/
|
|
#include <windows.h>
|
|
#include <winspool.h>
|
|
#include <winsplp.h>
|
|
#include <wchar.h>
|
|
|
|
#include "winprint.h"
|
|
|
|
|
|
|
|
/*++
|
|
*******************************************************************
|
|
G e t P r i n t e r I n f o
|
|
|
|
Routine Description:
|
|
This routine allocates the required memory for a
|
|
PRINTER_INFO_? structure and retrieves the information
|
|
from NT. This returns a pointer to the structure, which
|
|
must be freed by the calling routine.
|
|
|
|
Arguments:
|
|
hPrinter HANDLE to the printer the job is in
|
|
StructLevel The structure level to get
|
|
pErrorCode => field to place error, if one
|
|
|
|
Return Value:
|
|
PUCHAR => buffer where devmode info is if okay
|
|
NULL if error - pErrorCode returns error
|
|
*******************************************************************
|
|
--*/
|
|
PUCHAR
|
|
GetPrinterInfo(IN HANDLE hPrinter,
|
|
IN ULONG StructLevel,
|
|
OUT PULONG pErrorCode)
|
|
{
|
|
ULONG reqbytes, alloc_size;
|
|
PUCHAR ptr_info;
|
|
USHORT retry = 2;
|
|
|
|
alloc_size = BASE_PRINTER_BUFFER_SIZE;
|
|
|
|
/** Allocate a buffer. **/
|
|
|
|
ptr_info = AllocSplMem(alloc_size);
|
|
|
|
/** If the buffer isn't big enough, try once more **/
|
|
|
|
while (retry--) {
|
|
|
|
/** If the alloc / realloc failed, return error **/
|
|
|
|
if (!ptr_info) {
|
|
*pErrorCode = ERROR_NOT_ENOUGH_MEMORY;
|
|
return NULL;
|
|
}
|
|
|
|
/** Go get the printer information **/
|
|
|
|
if (GetPrinter(
|
|
hPrinter,
|
|
StructLevel,
|
|
(PUCHAR)ptr_info,
|
|
alloc_size,
|
|
&reqbytes) == TRUE) {
|
|
|
|
/** Got the info - return it **/
|
|
|
|
*pErrorCode = 0;
|
|
return (PUCHAR)ptr_info;
|
|
}
|
|
|
|
/**
|
|
GetPrinter failed - if not because of insufficient buffer, fail
|
|
the call. Otherwise, up our hint, re-allocate and try again.
|
|
**/
|
|
|
|
*pErrorCode = GetLastError();
|
|
|
|
if (*pErrorCode != ERROR_INSUFFICIENT_BUFFER) {
|
|
FreeSplMem(ptr_info);
|
|
return NULL;
|
|
}
|
|
|
|
/**
|
|
Reallocate the buffer and re-try (note that, because we
|
|
allocated the buffer as LMEM_FIXED, the LMEM_MOVABLE does
|
|
not return a movable allocation, it just allows realloc
|
|
to return a different pointer.
|
|
**/
|
|
|
|
alloc_size = reqbytes + 10;
|
|
ptr_info = ReallocSplMem(ptr_info, alloc_size, 0);
|
|
|
|
} /* While re-trying */
|
|
|
|
if (ptr_info) {
|
|
FreeSplMem(ptr_info);
|
|
}
|
|
|
|
*pErrorCode = ERROR_NOT_ENOUGH_MEMORY;
|
|
return NULL;
|
|
}
|
|
|
|
|