1585 lines
38 KiB
C
1585 lines
38 KiB
C
/*++
|
|
|
|
Copyright (c) 1990 Microsoft Corporation
|
|
|
|
Module Name:
|
|
|
|
util.c
|
|
|
|
Abstract:
|
|
|
|
This module provides all the utility functions for the Routing Layer and
|
|
the local Print Providor
|
|
|
|
Author:
|
|
|
|
Dave Snipp (DaveSn) 15-Mar-1991
|
|
|
|
Revision History:
|
|
|
|
--*/
|
|
|
|
#include "precomp.h"
|
|
#pragma hdrstop
|
|
|
|
#include "local.h"
|
|
|
|
#include <winddiui.h>
|
|
#include <winsock2.h>
|
|
#include <wininet.h>
|
|
|
|
|
|
LPWSTR *ppszOtherNames = NULL; // Contains szMachineName, DNS name, and all other machine name forms
|
|
DWORD cOtherNames = 0; // Number of other names
|
|
|
|
WCHAR *gszDrvConvert = L",DrvConvert";
|
|
|
|
|
|
//
|
|
// Lowercase, just like win31 for WM_WININICHANGE
|
|
//
|
|
WCHAR *szDevices=L"devices";
|
|
WCHAR *szWindows=L"windows";
|
|
|
|
#define NUM_INTERACTIVE_RIDS 1
|
|
|
|
extern DWORD RouterCacheSize;
|
|
extern PROUTERCACHE RouterCacheTable;
|
|
|
|
typedef struct _DEVMODECHG_INFO {
|
|
DWORD signature;
|
|
HANDLE hDrvModule;
|
|
FARPROC pfnConvertDevMode;
|
|
} DEVMODECHG_INFO, *PDEVMODECHG_INFO;
|
|
|
|
#define DMC_SIGNATURE 'DMC' /* 'DMC' is the signature value */
|
|
|
|
DWORD
|
|
RouterIsOlderThan(
|
|
DWORD i,
|
|
DWORD j
|
|
);
|
|
|
|
VOID
|
|
FreeOtherNames(LPWSTR **ppszMyOtherNames, DWORD *cOtherNames);
|
|
|
|
LPWSTR
|
|
AnsiToUnicodeStringWithAlloc(LPSTR pAnsi);
|
|
|
|
|
|
BOOL
|
|
DeleteSubKeyTree(
|
|
HKEY ParentHandle,
|
|
WCHAR SubKeyName[]
|
|
)
|
|
{
|
|
LONG Error;
|
|
DWORD Index;
|
|
HKEY KeyHandle;
|
|
BOOL RetValue;
|
|
|
|
WCHAR ChildKeyName[ MAX_PATH ];
|
|
DWORD ChildKeyNameLength;
|
|
|
|
Error = RegOpenKeyEx(
|
|
ParentHandle,
|
|
SubKeyName,
|
|
0,
|
|
KEY_READ | KEY_WRITE,
|
|
&KeyHandle
|
|
);
|
|
if (Error != ERROR_SUCCESS) {
|
|
SetLastError(Error);
|
|
return(FALSE);
|
|
}
|
|
|
|
ChildKeyNameLength = MAX_PATH;
|
|
Index = 0; // Don't increment this Index
|
|
|
|
while ((Error = RegEnumKeyEx(
|
|
KeyHandle,
|
|
Index,
|
|
ChildKeyName,
|
|
&ChildKeyNameLength,
|
|
NULL,
|
|
NULL,
|
|
NULL,
|
|
NULL
|
|
)) == ERROR_SUCCESS) {
|
|
|
|
RetValue = DeleteSubKeyTree( KeyHandle, ChildKeyName );
|
|
|
|
if (RetValue == FALSE) {
|
|
|
|
// Error -- couldn't delete the sub key
|
|
|
|
RegCloseKey(KeyHandle);
|
|
return(FALSE);
|
|
|
|
}
|
|
|
|
ChildKeyNameLength = MAX_PATH;
|
|
|
|
}
|
|
|
|
Error = RegCloseKey(
|
|
KeyHandle
|
|
);
|
|
if (Error != ERROR_SUCCESS) {
|
|
return(FALSE);
|
|
}
|
|
|
|
Error = RegDeleteKey(
|
|
ParentHandle,
|
|
SubKeyName
|
|
);
|
|
if (Error != ERROR_SUCCESS) {
|
|
return(FALSE);
|
|
}
|
|
|
|
// Return Success - the key has successfully been deleted
|
|
|
|
return(TRUE);
|
|
}
|
|
|
|
LPWSTR RemoveOrderEntry(
|
|
LPWSTR szOrderString,
|
|
DWORD cbStringSize,
|
|
LPWSTR szOrderEntry,
|
|
LPDWORD pcbBytesReturned
|
|
)
|
|
{
|
|
LPWSTR lpMem, psz, temp;
|
|
|
|
if (szOrderString == NULL) {
|
|
*pcbBytesReturned = 0;
|
|
return(NULL);
|
|
}
|
|
if (lpMem = AllocSplMem( cbStringSize)) {
|
|
temp = szOrderString;
|
|
psz = lpMem;
|
|
while (*temp) {
|
|
if (!lstrcmpi(temp, szOrderEntry)) { // we need to remove
|
|
temp += lstrlen(temp)+1; // this entry in Order
|
|
continue;
|
|
}
|
|
lstrcpy(psz,temp);
|
|
psz += lstrlen(temp)+1;
|
|
temp += lstrlen(temp)+1;
|
|
}
|
|
*psz = L'\0';
|
|
*pcbBytesReturned = (DWORD) ((psz - lpMem)+1)*sizeof(WCHAR);
|
|
return(lpMem);
|
|
}
|
|
*pcbBytesReturned = 0;
|
|
return(lpMem);
|
|
}
|
|
|
|
|
|
|
|
LPWSTR AppendOrderEntry(
|
|
LPWSTR szOrderString,
|
|
DWORD cbStringSize,
|
|
LPWSTR szOrderEntry,
|
|
LPDWORD pcbBytesReturned
|
|
)
|
|
{
|
|
LPWSTR lpMem, temp, psz;
|
|
DWORD cb = 0;
|
|
BOOL bExists = FALSE;
|
|
|
|
if ((szOrderString == NULL) && (szOrderEntry == NULL)) {
|
|
*pcbBytesReturned = 0;
|
|
return(NULL);
|
|
}
|
|
if (szOrderString == NULL) {
|
|
cb = wcslen(szOrderEntry)*sizeof(WCHAR)+ sizeof(WCHAR) + sizeof(WCHAR);
|
|
if (lpMem = AllocSplMem(cb)){
|
|
wcscpy(lpMem, szOrderEntry);
|
|
*pcbBytesReturned = cb;
|
|
} else {
|
|
*pcbBytesReturned = 0;
|
|
}
|
|
return lpMem;
|
|
}
|
|
|
|
if (lpMem = AllocSplMem( cbStringSize + wcslen(szOrderEntry)*sizeof(WCHAR)
|
|
+ sizeof(WCHAR))){
|
|
|
|
|
|
temp = szOrderString;
|
|
psz = lpMem;
|
|
while (*temp) {
|
|
if (!lstrcmpi(temp, szOrderEntry)) { // Make sure we don't
|
|
bExists = TRUE; // duplicate entries
|
|
}
|
|
lstrcpy(psz, temp);
|
|
psz += lstrlen(temp)+ 1;
|
|
temp += lstrlen(temp)+1;
|
|
}
|
|
if (!bExists) { // if it doesn't exist
|
|
lstrcpy(psz, szOrderEntry); // add the entry
|
|
psz += lstrlen(szOrderEntry)+1;
|
|
}
|
|
*psz = L'\0'; // the second null character
|
|
|
|
*pcbBytesReturned = (DWORD) ((psz - lpMem) + 1)* sizeof(WCHAR);
|
|
}
|
|
return(lpMem);
|
|
|
|
}
|
|
|
|
|
|
typedef struct {
|
|
DWORD dwType;
|
|
DWORD dwMessage;
|
|
WPARAM wParam;
|
|
LPARAM lParam;
|
|
} MESSAGE, *PMESSAGE;
|
|
|
|
VOID
|
|
SendMessageThread(
|
|
PMESSAGE pMessage);
|
|
|
|
|
|
BOOL
|
|
BroadcastMessage(
|
|
DWORD dwType,
|
|
DWORD dwMessage,
|
|
WPARAM wParam,
|
|
LPARAM lParam)
|
|
{
|
|
HANDLE hThread;
|
|
DWORD ThreadId;
|
|
PMESSAGE pMessage;
|
|
BOOL bReturn = FALSE;
|
|
|
|
pMessage = AllocSplMem(sizeof(MESSAGE));
|
|
|
|
if (pMessage) {
|
|
|
|
pMessage->dwType = dwType;
|
|
pMessage->dwMessage = dwMessage;
|
|
pMessage->wParam = wParam;
|
|
pMessage->lParam = lParam;
|
|
|
|
//
|
|
// We should have a queue of events to broadcast and then have a
|
|
// single thread pulling them off the queue until there is nothing
|
|
// left and then that thread could go away.
|
|
//
|
|
// The current design can lead to a huge number of threads being
|
|
// created and torn down in both this and csrss process.
|
|
//
|
|
hThread = CreateThread(NULL, 0,
|
|
(LPTHREAD_START_ROUTINE)SendMessageThread,
|
|
(LPVOID)pMessage,
|
|
0,
|
|
&ThreadId);
|
|
|
|
if (hThread) {
|
|
|
|
CloseHandle(hThread);
|
|
bReturn = TRUE;
|
|
|
|
} else {
|
|
|
|
FreeSplMem(pMessage);
|
|
}
|
|
}
|
|
|
|
return bReturn;
|
|
}
|
|
|
|
|
|
// The Broadcasts are done on a separate thread, the reason it CSRSS
|
|
// will create a server side thread when we call user and we don't want
|
|
// that to be pared up with the RPC thread which is in the spooss server.
|
|
// We want it to go away the moment we have completed the SendMessage.
|
|
// We also call SendNotifyMessage since we don't care if the broadcasts
|
|
// are syncronous this uses less resources since usually we don't have more
|
|
// than one broadcast.
|
|
|
|
VOID
|
|
SendMessageThread(
|
|
PMESSAGE pMessage)
|
|
{
|
|
switch (pMessage->dwType) {
|
|
|
|
case BROADCAST_TYPE_MESSAGE:
|
|
|
|
SendNotifyMessage(HWND_BROADCAST,
|
|
pMessage->dwMessage,
|
|
pMessage->wParam,
|
|
pMessage->lParam);
|
|
break;
|
|
|
|
case BROADCAST_TYPE_CHANGEDEFAULT:
|
|
|
|
//
|
|
// Same order and strings as win31.
|
|
//
|
|
SendNotifyMessage(HWND_BROADCAST,
|
|
WM_WININICHANGE,
|
|
0,
|
|
(LPARAM)szDevices);
|
|
|
|
SendNotifyMessage(HWND_BROADCAST,
|
|
WM_WININICHANGE,
|
|
0,
|
|
(LPARAM)szWindows);
|
|
break;
|
|
}
|
|
|
|
FreeSplMem(pMessage);
|
|
|
|
ExitThread(0);
|
|
}
|
|
|
|
|
|
BOOL
|
|
IsNamedPipeRpcCall(
|
|
VOID
|
|
)
|
|
{
|
|
unsigned int uType;
|
|
|
|
//
|
|
//
|
|
//
|
|
return ERROR_SUCCESS == I_RpcBindingInqTransportType(NULL, &uType) &&
|
|
uType != TRANSPORT_TYPE_LPC;
|
|
|
|
}
|
|
|
|
|
|
BOOL
|
|
IsLocalCall(
|
|
VOID
|
|
)
|
|
{
|
|
HANDLE hToken;
|
|
BOOL bStatus;
|
|
DWORD dwError;
|
|
PSID pTestSid = NULL;
|
|
PSID pCurSid;
|
|
DWORD cbSize = 0;
|
|
DWORD cbRequired = 0;
|
|
DWORD i;
|
|
BOOL bRet = FALSE;
|
|
BOOL bMember;
|
|
|
|
DWORD dwSaveLastError = GetLastError();
|
|
|
|
SID_IDENTIFIER_AUTHORITY sia = SECURITY_NT_AUTHORITY;
|
|
unsigned int uType;
|
|
|
|
dwError = I_RpcBindingInqTransportType(NULL, &uType);
|
|
|
|
if ( dwError == RPC_S_NO_CALL_ACTIVE ) {
|
|
|
|
//
|
|
// KM call
|
|
//
|
|
return TRUE;
|
|
}
|
|
|
|
if ( dwError == ERROR_SUCCESS ) {
|
|
|
|
if ( uType != TRANSPORT_TYPE_LPC ) {
|
|
|
|
//
|
|
// Not LRPC so call is remote
|
|
//
|
|
return FALSE;
|
|
}
|
|
|
|
} else {
|
|
|
|
//
|
|
// This should not fail. So we'll assert on chk bld and
|
|
// continue looking at SIDS on fre builds
|
|
//
|
|
SPLASSERT( dwError != ERROR_SUCCESS);
|
|
}
|
|
|
|
bStatus = OpenThreadToken(GetCurrentThread(), TOKEN_QUERY, TRUE, &hToken);
|
|
if (!bStatus) {
|
|
|
|
// Couldn't open the thread's token, nothing much we can do
|
|
DBGMSG(DBG_TRACE,("Error: couldn't open the thread's Access token %d\n", GetLastError()));
|
|
return FALSE;
|
|
}
|
|
|
|
if ( !AllocateAndInitializeSid(&sia,
|
|
1,
|
|
SECURITY_NETWORK_RID,
|
|
0, 0, 0, 0, 0, 0, 0,
|
|
&pTestSid) ) {
|
|
|
|
DBGMSG(DBG_TRACE,
|
|
("Error: could not AllocateAndInitializeSid -%d\n",
|
|
GetLastError()));
|
|
goto Done;
|
|
}
|
|
|
|
if( !CheckTokenMembership( hToken,
|
|
pTestSid,
|
|
&bMember )){
|
|
DBGMSG(DBG_TRACE,
|
|
("Error: CheckTokenMembership failed %d\n",
|
|
GetLastError()));
|
|
goto Done;
|
|
}
|
|
|
|
if( !bMember ){
|
|
|
|
//
|
|
// Not a member, so no match for network sid. Therefore local.
|
|
//
|
|
bRet = TRUE;
|
|
}
|
|
|
|
Done:
|
|
if ( pTestSid )
|
|
FreeSid(pTestSid);
|
|
|
|
CloseHandle(hToken);
|
|
|
|
SetLastError( dwSaveLastError );
|
|
return bRet;
|
|
}
|
|
|
|
|
|
LPPROVIDOR
|
|
FindEntryinRouterCache(
|
|
LPWSTR pPrinterName
|
|
)
|
|
{
|
|
DWORD i;
|
|
|
|
if (!pPrinterName)
|
|
return NULL;
|
|
|
|
DBGMSG(DBG_TRACE, ("FindEntryinRouterCache with %ws\n", pPrinterName));
|
|
|
|
if (RouterCacheSize == 0 ) {
|
|
DBGMSG(DBG_TRACE, ("FindEntryInRouterCache with %ws returning -1 (zero cache)\n", pPrinterName));
|
|
return NULL;
|
|
}
|
|
|
|
for (i = 0; i < RouterCacheSize; i++ ) {
|
|
|
|
if (RouterCacheTable[i].bAvailable) {
|
|
if (!_wcsicmp(RouterCacheTable[i].pPrinterName, pPrinterName)) {
|
|
|
|
//
|
|
// update the time stamp so that it is current and not old
|
|
//
|
|
GetSystemTime(&RouterCacheTable[i].st);
|
|
|
|
//
|
|
//
|
|
//
|
|
DBGMSG(DBG_TRACE, ("FindEntryinRouterCache returning with %d\n", i));
|
|
return RouterCacheTable[i].pProvidor;
|
|
}
|
|
}
|
|
}
|
|
DBGMSG(DBG_TRACE, ("FindEntryinRouterCache returning with -1\n"));
|
|
return NULL;
|
|
}
|
|
|
|
|
|
DWORD
|
|
AddEntrytoRouterCache(
|
|
LPWSTR pPrinterName,
|
|
LPPROVIDOR pProvidor
|
|
)
|
|
{
|
|
DWORD LRUEntry = (DWORD)-1;
|
|
DWORD i;
|
|
DBGMSG(DBG_TRACE, ("AddEntrytoRouterCache with %ws\n", pPrinterName));
|
|
|
|
if (RouterCacheSize == 0 ) {
|
|
DBGMSG(DBG_TRACE, ("AddEntrytoRouterCache with %ws returning -1 (zero cache)\n", pPrinterName));
|
|
return (DWORD)-1;
|
|
}
|
|
|
|
for (i = 0; i < RouterCacheSize; i++ ) {
|
|
|
|
if (!RouterCacheTable[i].bAvailable) {
|
|
|
|
//
|
|
// Found an available entry; use it
|
|
// fill in the name of the printer and the providor
|
|
// that supports this printer.
|
|
//
|
|
break;
|
|
|
|
} else {
|
|
|
|
if ((LRUEntry == -1) || (i == RouterIsOlderThan(i, LRUEntry))){
|
|
LRUEntry = i;
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
if (i == RouterCacheSize) {
|
|
|
|
//
|
|
// We have no available entries so we need to use
|
|
// the LRUEntry which is busy
|
|
//
|
|
FreeSplStr(RouterCacheTable[LRUEntry].pPrinterName);
|
|
RouterCacheTable[LRUEntry].bAvailable = FALSE;
|
|
|
|
i = LRUEntry;
|
|
}
|
|
|
|
|
|
if ((RouterCacheTable[i].pPrinterName = AllocSplStr(pPrinterName)) == NULL){
|
|
|
|
//
|
|
// Alloc failed so we're kinda hosed so return -1
|
|
//
|
|
return (DWORD)-1;
|
|
}
|
|
|
|
|
|
RouterCacheTable[i].bAvailable = TRUE;
|
|
RouterCacheTable[i].pProvidor = pProvidor;
|
|
|
|
//
|
|
// update the time stamp so that we know when this entry was made
|
|
//
|
|
GetSystemTime(&RouterCacheTable[i].st);
|
|
DBGMSG(DBG_TRACE, ("AddEntrytoRouterCache returning with %d\n", i));
|
|
return i;
|
|
}
|
|
|
|
|
|
VOID
|
|
DeleteEntryfromRouterCache(
|
|
LPWSTR pPrinterName
|
|
)
|
|
{
|
|
DWORD i;
|
|
|
|
if (RouterCacheSize == 0) {
|
|
DBGMSG(DBG_TRACE, ("DeleteEntryfromRouterCache with %ws returning -1 (zero cache)\n", pPrinterName));
|
|
return;
|
|
}
|
|
|
|
DBGMSG(DBG_TRACE, ("DeleteEntryFromRouterCache with %ws\n", pPrinterName));
|
|
for (i = 0; i < RouterCacheSize; i++ ) {
|
|
if (RouterCacheTable[i].bAvailable) {
|
|
if (!_wcsicmp(RouterCacheTable[i].pPrinterName, pPrinterName)) {
|
|
//
|
|
// reset the available flag on this node
|
|
//
|
|
FreeSplStr(RouterCacheTable[i].pPrinterName);
|
|
|
|
RouterCacheTable[i].pProvidor = NULL;
|
|
RouterCacheTable[i].bAvailable = FALSE;
|
|
|
|
DBGMSG(DBG_TRACE, ("DeleteEntryFromRouterCache returning after deleting the %d th entry\n", i));
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
DBGMSG(DBG_TRACE, ("DeleteEntryFromRouterCache returning after not finding an entry to delete\n"));
|
|
}
|
|
|
|
|
|
|
|
DWORD
|
|
RouterIsOlderThan(
|
|
DWORD i,
|
|
DWORD j
|
|
)
|
|
{
|
|
SYSTEMTIME *pi, *pj;
|
|
DWORD iMs, jMs;
|
|
DBGMSG(DBG_TRACE, ("RouterIsOlderThan entering with i %d j %d\n", i, j));
|
|
pi = &(RouterCacheTable[i].st);
|
|
pj = &(RouterCacheTable[j].st);
|
|
DBGMSG(DBG_TRACE, ("Index i %d - %d:%d:%d:%d:%d:%d:%d\n",
|
|
i, pi->wYear, pi->wMonth, pi->wDay, pi->wHour, pi->wMinute, pi->wSecond, pi->wMilliseconds));
|
|
|
|
|
|
DBGMSG(DBG_TRACE,("Index j %d - %d:%d:%d:%d:%d:%d:%d\n",
|
|
j, pj->wYear, pj->wMonth, pj->wDay, pj->wHour, pj->wMinute, pj->wSecond, pj->wMilliseconds));
|
|
|
|
if (pi->wYear < pj->wYear) {
|
|
DBGMSG(DBG_TRACE, ("RouterIsOlderThan returns %d\n", i));
|
|
return(i);
|
|
} else if (pi->wYear > pj->wYear) {
|
|
DBGMSG(DBG_TRACE, ("RouterIsOlderThan returns %d\n", j));
|
|
return(j);
|
|
} else if (pi->wMonth < pj->wMonth) {
|
|
DBGMSG(DBG_TRACE, ("RouterIsOlderThan returns %d\n", i));
|
|
return(i);
|
|
} else if (pi->wMonth > pj->wMonth) {
|
|
DBGMSG(DBG_TRACE, ("RouterIsOlderThan returns %d\n", j));
|
|
return(j);
|
|
} else if (pi->wDay < pj->wDay) {
|
|
DBGMSG(DBG_TRACE, ("RouterIsOlderThan returns %d\n", i));
|
|
return(i);
|
|
} else if (pi->wDay > pj->wDay) {
|
|
DBGMSG(DBG_TRACE, ("RouterIsOlderThan returns %d\n", j));
|
|
return(j);
|
|
} else {
|
|
iMs = ((((pi->wHour * 60) + pi->wMinute)*60) + pi->wSecond)* 1000 + pi->wMilliseconds;
|
|
jMs = ((((pj->wHour * 60) + pj->wMinute)*60) + pj->wSecond)* 1000 + pj->wMilliseconds;
|
|
|
|
if (iMs <= jMs) {
|
|
DBGMSG(DBG_TRACE, ("RouterIsOlderThan returns %d\n", i));
|
|
return(i);
|
|
} else {
|
|
DBGMSG(DBG_TRACE, ("RouterIsOlderThan returns %d\n", j));
|
|
return(j);
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
/*++
|
|
|
|
Routine Name
|
|
|
|
ImpersonationToken
|
|
|
|
Routine Description:
|
|
|
|
This routine checks if a token is a primary token or an impersonation
|
|
token.
|
|
|
|
Arguments:
|
|
|
|
hToken - impersonation token or primary token of the process
|
|
|
|
Return Value:
|
|
|
|
TRUE, if the token is an impersonation token
|
|
FALSE, otherwise.
|
|
|
|
--*/
|
|
BOOL
|
|
ImpersonationToken(
|
|
IN HANDLE hToken
|
|
)
|
|
{
|
|
BOOL bRet = TRUE;
|
|
TOKEN_TYPE eTokenType;
|
|
DWORD cbNeeded;
|
|
DWORD LastError;
|
|
|
|
//
|
|
// Preserve the last error. Some callers of ImpersonatePrinterClient (which
|
|
// calls ImpersonationToken) rely on the fact that ImpersonatePrinterClient
|
|
// does not alter the last error.
|
|
//
|
|
LastError = GetLastError();
|
|
|
|
//
|
|
// Get the token type from the thread token. The token comes
|
|
// from RevertToPrinterSelf. An impersonation token cannot be
|
|
// queried, because RevertToPRinterSelf doesn't open it with
|
|
// TOKEN_QUERY access. That's why we assume that hToken is
|
|
// an impersonation token by default
|
|
//
|
|
if (GetTokenInformation(hToken,
|
|
TokenType,
|
|
&eTokenType,
|
|
sizeof(eTokenType),
|
|
&cbNeeded))
|
|
{
|
|
bRet = eTokenType == TokenImpersonation;
|
|
}
|
|
|
|
SetLastError(LastError);
|
|
|
|
return bRet;
|
|
}
|
|
|
|
/*++
|
|
|
|
Routine Name
|
|
|
|
RevertToPrinterSelf
|
|
|
|
Routine Description:
|
|
|
|
This routine will revert to the local system. It returns the token that
|
|
ImpersonatePrinterClient then uses to imersonate the client again. If the
|
|
current thread doesn't impersonate, then the function merely returns the
|
|
primary token of the process. (instead of returning NULL) Thus we honor
|
|
a request for reverting to printer self, even if the thread is not impersonating.
|
|
|
|
Arguments:
|
|
|
|
None.
|
|
|
|
Return Value:
|
|
|
|
NULL, if the function failed
|
|
HANDLE to token, otherwise.
|
|
|
|
--*/
|
|
HANDLE
|
|
RevertToPrinterSelf(
|
|
VOID
|
|
)
|
|
{
|
|
HANDLE NewToken, OldToken;
|
|
NTSTATUS Status;
|
|
|
|
NewToken = NULL;
|
|
|
|
Status = NtOpenThreadToken(NtCurrentThread(),
|
|
TOKEN_IMPERSONATE,
|
|
TRUE,
|
|
&OldToken);
|
|
|
|
if (NT_SUCCESS(Status))
|
|
{
|
|
//
|
|
// We are currently impersonating
|
|
//
|
|
Status = NtSetInformationThread(NtCurrentThread(),
|
|
ThreadImpersonationToken,
|
|
(PVOID)&NewToken,
|
|
(ULONG)sizeof(HANDLE));
|
|
}
|
|
else if (Status == STATUS_NO_TOKEN)
|
|
{
|
|
//
|
|
// We are not impersonating
|
|
//
|
|
Status = NtOpenProcessToken(NtCurrentProcess(),
|
|
TOKEN_QUERY,
|
|
&OldToken);
|
|
|
|
}
|
|
|
|
if (!NT_SUCCESS(Status))
|
|
{
|
|
SetLastError(Status);
|
|
return NULL;
|
|
}
|
|
|
|
return OldToken;
|
|
}
|
|
|
|
/*++
|
|
|
|
Routine Name
|
|
|
|
ImpersonatePrinterClient
|
|
|
|
Routine Description:
|
|
|
|
This routine attempts to set the passed in hToken as the token for the
|
|
current thread. If hToken is not an impersonation token, then the routine
|
|
will simply close the token.
|
|
|
|
Arguments:
|
|
|
|
hToken - impersonation token or primary token of the process
|
|
|
|
Return Value:
|
|
|
|
TRUE, if the function succeeds in setting hToken
|
|
FALSE, otherwise.
|
|
|
|
--*/
|
|
BOOL
|
|
ImpersonatePrinterClient(
|
|
HANDLE hToken)
|
|
{
|
|
NTSTATUS Status;
|
|
|
|
//
|
|
// Check if we have an impersonation token
|
|
//
|
|
if (ImpersonationToken(hToken))
|
|
{
|
|
Status = NtSetInformationThread(NtCurrentThread(),
|
|
ThreadImpersonationToken,
|
|
(PVOID)&hToken,
|
|
(ULONG)sizeof(HANDLE));
|
|
if (!NT_SUCCESS(Status))
|
|
{
|
|
SetLastError(Status);
|
|
return FALSE;
|
|
}
|
|
}
|
|
|
|
NtClose(hToken);
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
HANDLE
|
|
LoadDriver(
|
|
LPWSTR pDriverFile)
|
|
{
|
|
UINT uOldErrorMode;
|
|
fnWinSpoolDrv fnList;
|
|
HANDLE hReturn = NULL;
|
|
|
|
if (!pDriverFile || !*pDriverFile) {
|
|
// Nothing to load
|
|
return hReturn;
|
|
}
|
|
|
|
uOldErrorMode = SetErrorMode(SEM_FAILCRITICALERRORS | SEM_NOOPENFILEERRORBOX);
|
|
|
|
if (SplInitializeWinSpoolDrv(&fnList)) {
|
|
|
|
hReturn = (*(fnList.pfnRefCntLoadDriver))(pDriverFile,
|
|
LOAD_WITH_ALTERED_SEARCH_PATH,
|
|
0, FALSE);
|
|
}
|
|
|
|
(VOID)SetErrorMode(uOldErrorMode);
|
|
|
|
return hReturn;
|
|
}
|
|
|
|
VOID
|
|
UnloadDriver(
|
|
HANDLE hModule
|
|
)
|
|
{
|
|
fnWinSpoolDrv fnList;
|
|
|
|
if (SplInitializeWinSpoolDrv(&fnList)) {
|
|
(* (fnList.pfnRefCntUnloadDriver))(hModule, TRUE);
|
|
}
|
|
}
|
|
|
|
VOID
|
|
UnloadDriverFile(
|
|
IN OUT HANDLE hDevModeChgInfo
|
|
)
|
|
/*++
|
|
|
|
Description:
|
|
Does a FreeLibrary on the driver file and frees memory
|
|
|
|
Arguments:
|
|
hDevModeChgInfo - A handle returned by LoadDriverFiletoConvertDevmode
|
|
|
|
Return Value:
|
|
None
|
|
|
|
--*/
|
|
{
|
|
PDEVMODECHG_INFO pDevModeChgInfo = (PDEVMODECHG_INFO) hDevModeChgInfo;
|
|
|
|
SPLASSERT(pDevModeChgInfo &&
|
|
pDevModeChgInfo->signature == DMC_SIGNATURE);
|
|
|
|
if ( pDevModeChgInfo && pDevModeChgInfo->signature == DMC_SIGNATURE ) {
|
|
|
|
if ( pDevModeChgInfo->hDrvModule ) {
|
|
UnloadDriver(pDevModeChgInfo->hDrvModule);
|
|
}
|
|
FreeSplMem((LPVOID)pDevModeChgInfo);
|
|
}
|
|
}
|
|
|
|
HANDLE
|
|
LoadDriverFiletoConvertDevmode(
|
|
IN LPWSTR pDriverFile
|
|
)
|
|
/*++
|
|
|
|
Description:
|
|
Does a LoadLibrary on the driver file given. This will give a handle
|
|
which can be used to do devmode conversion later using
|
|
CallDrvDevModeConversion.
|
|
|
|
Caller should call UnloadDriverFile to do a FreeLibrary and free memory
|
|
|
|
Note: Driver will call OpenPrinter to spooler
|
|
|
|
Arguments:
|
|
pDriverFile - Full path of the driver file to do a LoadLibrary
|
|
|
|
Return Value:
|
|
A handle value to be used to make calls to CallDrvDevModeConversion,
|
|
NULL on error
|
|
|
|
--*/
|
|
{
|
|
PDEVMODECHG_INFO pDevModeChgInfo = NULL;
|
|
BOOL bFail = TRUE;
|
|
DWORD dwNeeded;
|
|
|
|
SPLASSERT(pDriverFile != NULL);
|
|
|
|
pDevModeChgInfo = (PDEVMODECHG_INFO) AllocSplMem(sizeof(*pDevModeChgInfo));
|
|
|
|
if ( !pDevModeChgInfo ) {
|
|
|
|
DBGMSG(DBG_WARNING, ("printer.c: Memory allocation failed for DEVMODECHG_INFO\n"));
|
|
goto Cleanup;
|
|
}
|
|
|
|
pDevModeChgInfo->signature = DMC_SIGNATURE;
|
|
|
|
pDevModeChgInfo->hDrvModule = LoadDriver(pDriverFile);
|
|
|
|
if ( !pDevModeChgInfo->hDrvModule ) {
|
|
|
|
DBGMSG(DBG_WARNING,("LoadDriverFiletoConvertDevmode: Can't load driver file %ws\n", pDriverFile));
|
|
goto Cleanup;
|
|
}
|
|
|
|
//
|
|
// Some third party driver may not be providing DrvConvertDevMode
|
|
//
|
|
pDevModeChgInfo->pfnConvertDevMode = GetProcAddress(pDevModeChgInfo->hDrvModule,
|
|
"DrvConvertDevMode");
|
|
if ( !pDevModeChgInfo->pfnConvertDevMode )
|
|
goto Cleanup;
|
|
|
|
bFail = FALSE;
|
|
|
|
Cleanup:
|
|
|
|
if ( bFail ) {
|
|
|
|
if ( pDevModeChgInfo ) {
|
|
UnloadDriverFile((HANDLE)pDevModeChgInfo);
|
|
}
|
|
return (HANDLE) NULL;
|
|
|
|
} else {
|
|
|
|
return (HANDLE) pDevModeChgInfo;
|
|
}
|
|
}
|
|
|
|
DWORD
|
|
CallDrvDevModeConversion(
|
|
IN HANDLE hDevModeChgInfo,
|
|
IN LPWSTR pszPrinterName,
|
|
IN LPBYTE pDevMode1,
|
|
IN OUT LPBYTE *ppDevMode2,
|
|
IN OUT LPDWORD pdwOutDevModeSize,
|
|
IN DWORD dwConvertMode,
|
|
IN BOOL bAlloc
|
|
)
|
|
/*++
|
|
|
|
Description:
|
|
Does deve mode conversion by calling driver
|
|
|
|
If bAlloc is TRUE routine will do the allocation using AllocSplMem
|
|
|
|
Note: Driver is going to call OpenPrinter.
|
|
|
|
Arguments:
|
|
hDevModeChgInfo - Points to DEVMODECHG_INFO
|
|
|
|
pszPrinterName - Printer name
|
|
|
|
pInDevMode - Input devmode (will be NULL for CDM_DRIVER_DEFAULT)
|
|
|
|
*pOutDevMode - Points to output devmode
|
|
|
|
pdwOutDevModeSize - Output devmode size on succesful return
|
|
if !bAlloc this will give input buffer size
|
|
|
|
dwConvertMode - Devmode conversion mode to give to driver
|
|
|
|
bAllocate - Tells the routine to do allocation to *pOutPrinter
|
|
If bAllocate is TRUE and no devmode conversion is required
|
|
call will fail.
|
|
|
|
Return Value:
|
|
Returns last error
|
|
|
|
--*/
|
|
{
|
|
DWORD dwBufSize, dwSize, dwLastError = ERROR_SUCCESS;
|
|
LPDEVMODE pInDevMode = (LPDEVMODE)pDevMode1,
|
|
*ppOutDevMode = (LPDEVMODE *) ppDevMode2;
|
|
PDEVMODECHG_INFO pDevModeChgInfo = (PDEVMODECHG_INFO) hDevModeChgInfo;
|
|
PWSTR pszDrvConvert = pszPrinterName;
|
|
|
|
|
|
if ( !pDevModeChgInfo ||
|
|
pDevModeChgInfo->signature != DMC_SIGNATURE ||
|
|
!pDevModeChgInfo->pfnConvertDevMode ) {
|
|
|
|
SPLASSERT(pDevModeChgInfo &&
|
|
pDevModeChgInfo->signature == DMC_SIGNATURE &&
|
|
pDevModeChgInfo->pfnConvertDevMode);
|
|
|
|
return ERROR_INVALID_PARAMETER;
|
|
}
|
|
|
|
//
|
|
// We decorate the pszPrinterName with ",DrvConvert" to prevent drivers from
|
|
// infinitely recursing by calling GetPrinter inside ConvertDevMode
|
|
//
|
|
if (wcsstr(pszPrinterName, gszDrvConvert)) {
|
|
return ERROR_INVALID_PARAMETER;
|
|
}
|
|
|
|
if (!(pszDrvConvert = AutoCat(pszPrinterName, gszDrvConvert))) {
|
|
return GetLastError();
|
|
}
|
|
|
|
DBGMSG(DBG_INFO,("Convert DevMode %d\n", dwConvertMode));
|
|
|
|
|
|
#if DBG
|
|
#else
|
|
try {
|
|
#endif
|
|
|
|
if ( bAlloc ) {
|
|
|
|
//
|
|
// If we have to do allocation first find size neeeded
|
|
//
|
|
*pdwOutDevModeSize = 0;
|
|
*ppOutDevMode = NULL;
|
|
|
|
(*pDevModeChgInfo->pfnConvertDevMode)(pszDrvConvert,
|
|
pInDevMode,
|
|
NULL,
|
|
pdwOutDevModeSize,
|
|
dwConvertMode);
|
|
|
|
dwLastError = GetLastError();
|
|
if ( dwLastError != ERROR_INSUFFICIENT_BUFFER ) {
|
|
|
|
DBGMSG(DBG_WARNING,
|
|
("CallDrvDevModeConversion: Unexpected error %d\n",
|
|
GetLastError()));
|
|
|
|
if (dwLastError == ERROR_SUCCESS) {
|
|
|
|
SPLASSERT(dwLastError != ERROR_SUCCESS);
|
|
// if driver doesn't fail the above call, it is a broken driver and probably
|
|
// failed a HeapAlloc, which doesn't SetLastError()
|
|
SetLastError(dwLastError = ERROR_NOT_ENOUGH_MEMORY);
|
|
}
|
|
#if DBG
|
|
goto Cleanup;
|
|
#else
|
|
leave;
|
|
#endif
|
|
}
|
|
|
|
*ppOutDevMode = AllocSplMem(*pdwOutDevModeSize);
|
|
if ( !*ppOutDevMode ) {
|
|
|
|
dwLastError = GetLastError();
|
|
#if DBG
|
|
goto Cleanup;
|
|
#else
|
|
leave;
|
|
#endif
|
|
}
|
|
}
|
|
|
|
dwBufSize = *pdwOutDevModeSize;
|
|
|
|
if ( !(*pDevModeChgInfo->pfnConvertDevMode)(
|
|
pszDrvConvert,
|
|
pInDevMode,
|
|
ppOutDevMode ? *ppOutDevMode
|
|
: NULL,
|
|
pdwOutDevModeSize,
|
|
dwConvertMode) ) {
|
|
|
|
dwLastError = GetLastError();
|
|
if (dwLastError == ERROR_SUCCESS) {
|
|
|
|
SPLASSERT(dwLastError != ERROR_SUCCESS);
|
|
// if driver doesn't fail the above call, it is a broken driver and probably
|
|
// failed a HeapAlloc, which doesn't SetLastError()
|
|
SetLastError(dwLastError = ERROR_NOT_ENOUGH_MEMORY);
|
|
}
|
|
} else {
|
|
|
|
dwLastError = ERROR_SUCCESS;
|
|
}
|
|
|
|
#if DBG
|
|
Cleanup:
|
|
#else
|
|
} except(1) {
|
|
|
|
DBGMSG(DBG_ERROR,
|
|
("CallDrvDevModeConversion: Exception from driver\n"));
|
|
dwLastError = GetExceptionCode();
|
|
SetLastError(dwLastError);
|
|
}
|
|
#endif
|
|
|
|
//
|
|
// If we allocated mmeory free it and zero the pointer
|
|
//
|
|
if ( dwLastError != ERROR_SUCCESS && bAlloc && *ppOutDevMode ) {
|
|
|
|
FreeSplMem(*ppOutDevMode);
|
|
*ppOutDevMode = 0;
|
|
*pdwOutDevModeSize = 0;
|
|
}
|
|
|
|
if ( dwLastError != ERROR_SUCCESS &&
|
|
dwLastError != ERROR_INSUFFICIENT_BUFFER ) {
|
|
|
|
DBGMSG(DBG_WARNING, ("DevmodeConvert unexpected error %d\n", dwLastError));
|
|
}
|
|
|
|
if ( dwLastError == ERROR_SUCCESS ) {
|
|
|
|
dwSize = (*ppOutDevMode)->dmSize + (*ppOutDevMode)->dmDriverExtra;
|
|
|
|
//
|
|
// Did the driver return correct size as per the devmode?
|
|
//
|
|
if ( *pdwOutDevModeSize != dwSize ) {
|
|
|
|
DBGMSG(DBG_ERROR,
|
|
("Driver says outsize is %d, really %d\n",
|
|
*pdwOutDevModeSize, dwSize));
|
|
|
|
*pdwOutDevModeSize = dwSize;
|
|
}
|
|
|
|
//
|
|
// Is it a valid devmode which did not overwrite the buffer?
|
|
//
|
|
if ( *pdwOutDevModeSize < MIN_DEVMODE_SIZEW ||
|
|
*pdwOutDevModeSize > dwBufSize ) {
|
|
|
|
DBGMSG(DBG_ERROR,
|
|
("Bad devmode from the driver size %d, buffer %d",
|
|
*pdwOutDevModeSize, dwBufSize));
|
|
dwLastError = ERROR_INVALID_PARAMETER;
|
|
|
|
if ( bAlloc ) {
|
|
|
|
FreeSplMem(*ppOutDevMode);
|
|
*ppOutDevMode = NULL;
|
|
}
|
|
*pdwOutDevModeSize = 0;
|
|
}
|
|
}
|
|
|
|
FreeSplMem(pszDrvConvert);
|
|
|
|
return dwLastError;
|
|
}
|
|
|
|
BOOL
|
|
BuildOtherNamesFromMachineName(
|
|
LPWSTR **ppszMyOtherNames,
|
|
DWORD *cOtherNames
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
This routine builds list of names other than the machine name that
|
|
can be used to call spooler APIs.
|
|
|
|
--*/
|
|
{
|
|
HANDLE hModule;
|
|
struct hostent *HostEnt, *(*Fngethostbyname)(LPTSTR);
|
|
struct in_addr *ptr;
|
|
INT (*FnWSAStartup)(WORD, LPWSADATA);
|
|
DWORD Index, Count;
|
|
WSADATA WSAData;
|
|
VOID (*FnWSACleanup)();
|
|
LPSTR (*Fninet_ntoa)(struct in_addr);
|
|
WORD wVersion;
|
|
BOOL bRet = FALSE;
|
|
DWORD dwRet;
|
|
|
|
|
|
SPLASSERT(cOtherNames && ppszMyOtherNames);
|
|
|
|
*cOtherNames = 0;
|
|
|
|
wVersion = MAKEWORD(1, 1);
|
|
|
|
|
|
dwRet = WSAStartup(wVersion, &WSAData);
|
|
if (dwRet) {
|
|
DBGMSG(DBG_WARNING, ("BuildOtherNamesFromMachineName: WSAStartup failed\n"));
|
|
SetLastError(dwRet);
|
|
return FALSE;
|
|
}
|
|
|
|
HostEnt = gethostbyname(NULL);
|
|
|
|
if (HostEnt) {
|
|
|
|
for (*cOtherNames = 0 ; HostEnt->h_addr_list[*cOtherNames] ; ++(*cOtherNames))
|
|
;
|
|
|
|
*cOtherNames += 2; // Add one for DNS and one for machine name
|
|
|
|
*ppszMyOtherNames = (LPWSTR *) AllocSplMem(*cOtherNames*sizeof *ppszMyOtherNames);
|
|
if ( !*ppszMyOtherNames ) {
|
|
*cOtherNames = 0;
|
|
goto Cleanup;
|
|
}
|
|
|
|
(*ppszMyOtherNames)[0] = AllocSplStr(szMachineName + 2); // Exclude the leading double-backslash
|
|
(*ppszMyOtherNames)[1] = AnsiToUnicodeStringWithAlloc(HostEnt->h_name);
|
|
|
|
for (Index = 0 ; HostEnt->h_addr_list[Index] ; ++Index) {
|
|
ptr = (struct in_addr *) HostEnt->h_addr_list[Index];
|
|
(*ppszMyOtherNames)[Index+2] = AnsiToUnicodeStringWithAlloc(inet_ntoa(*ptr));
|
|
}
|
|
|
|
// check for allocation failures
|
|
for (Index = 0 ; Index < *cOtherNames ; ++Index) {
|
|
if ( !(*ppszMyOtherNames)[Index] ) {
|
|
FreeOtherNames(ppszMyOtherNames, cOtherNames);
|
|
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
|
|
goto Cleanup;
|
|
}
|
|
}
|
|
|
|
bRet = TRUE;
|
|
|
|
} else {
|
|
DBGMSG(DBG_WARNING, ("BuildOtherNamesFromMachineName: gethostbyname failed for with %d\n", GetLastError()));
|
|
}
|
|
|
|
|
|
Cleanup:
|
|
WSACleanup();
|
|
|
|
return bRet;
|
|
}
|
|
|
|
|
|
VOID
|
|
FreeOtherNames(LPWSTR **ppszMyOtherNames, DWORD *cOtherNames)
|
|
{
|
|
DWORD i;
|
|
|
|
for( i = 0 ; i < *cOtherNames ; ++i)
|
|
FreeSplMem((*ppszMyOtherNames)[i]);
|
|
|
|
FreeSplMem(*ppszMyOtherNames);
|
|
}
|
|
|
|
|
|
|
|
LPWSTR
|
|
AnsiToUnicodeStringWithAlloc(
|
|
LPSTR pAnsi
|
|
)
|
|
/*++
|
|
|
|
Description:
|
|
Convert ANSI string to UNICODE. Routine allocates memory from the heap
|
|
which should be freed by the caller.
|
|
|
|
Arguments:
|
|
pAnsi - Points to the ANSI string
|
|
|
|
Return Vlaue:
|
|
Pointer to UNICODE string
|
|
|
|
--*/
|
|
{
|
|
LPWSTR pUnicode;
|
|
DWORD rc;
|
|
|
|
rc = MultiByteToWideChar(CP_ACP,
|
|
MB_PRECOMPOSED,
|
|
pAnsi,
|
|
-1,
|
|
NULL,
|
|
0);
|
|
|
|
rc *= sizeof(WCHAR);
|
|
if ( !rc || !(pUnicode = (LPWSTR) AllocSplMem(rc)) )
|
|
return NULL;
|
|
|
|
rc = MultiByteToWideChar(CP_ACP,
|
|
MB_PRECOMPOSED,
|
|
pAnsi,
|
|
-1,
|
|
pUnicode,
|
|
rc);
|
|
|
|
if ( rc )
|
|
return pUnicode;
|
|
else {
|
|
FreeSplMem(pUnicode);
|
|
return NULL;
|
|
}
|
|
}
|
|
|
|
|
|
/*++
|
|
|
|
Routine Description
|
|
|
|
Determines whether or not a machine name contains the local machine name.
|
|
|
|
Localspl enum calls fail if pName != local machine name (\\Machine).
|
|
Remote enum provider is then called. The remote enum provider must check
|
|
if the UNC name refers to the local machine, and fail if it does to avoid
|
|
endless recursion.
|
|
|
|
Arguments:
|
|
|
|
LPWSTR pName - UNC name.
|
|
|
|
Return Value:
|
|
|
|
TRUE: pName == \\szMachineName\...
|
|
- or -
|
|
pName == \\szMachineName
|
|
|
|
FALSE: anything else
|
|
|
|
Author: swilson
|
|
|
|
--*/
|
|
|
|
BOOL
|
|
MyUNCName(
|
|
LPWSTR pNameStart
|
|
)
|
|
{
|
|
PWCHAR pMachine = szMachineName;
|
|
LPWSTR pName;
|
|
DWORD i;
|
|
extern LPWSTR *ppszOtherNames; // Contains szMachineName, DNS name, and all other machine name forms
|
|
extern DWORD cOtherNames;
|
|
|
|
|
|
if (!pNameStart || !*pNameStart) // This differs from MyName(), which returns TRUE
|
|
return FALSE;
|
|
|
|
if (*pNameStart == L'\\' && *(pNameStart + 1) == L'\\') {
|
|
for (i = 0 , pName = pNameStart + 2 ; i < cOtherNames ; ++i , pName = pNameStart + 2) {
|
|
for(pMachine = ppszOtherNames[i] ;
|
|
*pName && towupper(*pName) == towupper(*pMachine) ;
|
|
++pName, ++pMachine)
|
|
;
|
|
|
|
if(!*pMachine && (!*pName || *pName == L'\\'))
|
|
return TRUE;
|
|
}
|
|
}
|
|
|
|
return FALSE;
|
|
}
|
|
|
|
BOOL
|
|
SplIsUpgrade(
|
|
VOID
|
|
)
|
|
{
|
|
return dwUpgradeFlag;
|
|
|
|
}
|
|
|
|
|
|
PWSTR
|
|
AutoCat(
|
|
PCWSTR pszInput,
|
|
PCWSTR pszCat
|
|
)
|
|
{
|
|
PWSTR pszOut;
|
|
|
|
if (!pszCat) {
|
|
|
|
pszOut = AllocSplStr(pszInput);
|
|
|
|
} else if (pszInput) {
|
|
|
|
pszOut = AllocSplMem((wcslen(pszInput) + wcslen(pszCat) + 1)*sizeof(WCHAR));
|
|
if (pszOut) {
|
|
wcscpy(pszOut, pszInput);
|
|
wcscat(pszOut, pszCat);
|
|
}
|
|
|
|
} else {
|
|
|
|
pszOut = AllocSplStr(pszCat);
|
|
|
|
}
|
|
|
|
return pszOut;
|
|
}
|
|
|
|
PBIDI_RESPONSE_CONTAINER
|
|
RouterAllocBidiResponseContainer(
|
|
DWORD Count
|
|
)
|
|
{
|
|
DWORD MemSize = 0;
|
|
//
|
|
// Add the size of the container - the size of the first data element
|
|
//
|
|
MemSize += (sizeof(BIDI_RESPONSE_CONTAINER) - sizeof(BIDI_RESPONSE_DATA));
|
|
//
|
|
// Add the size of all the returned RESPONSE_DATA elements
|
|
//
|
|
MemSize += (Count * sizeof(BIDI_RESPONSE_DATA));
|
|
|
|
return((PBIDI_RESPONSE_CONTAINER) MIDL_user_allocate(MemSize));
|
|
}
|
|
|
|
/*++
|
|
|
|
Routine Name
|
|
|
|
GetAPDPolicy
|
|
|
|
Routine Description:
|
|
|
|
This function reads a DWORD value from the location
|
|
HKEY\pszRelPath\pszValueName. We use this function for
|
|
preserving the AddPrinterDrivers policy value when the
|
|
LanMan Print Services print provider is deleted from
|
|
the system.
|
|
|
|
Arguments:
|
|
|
|
hKey - key tree
|
|
pszRelPath - relative path of the value to be get
|
|
pszValueName - value name
|
|
pValue - pointer to memory to store a dword value
|
|
|
|
Return Value:
|
|
|
|
ERROR_SUCCESS - the value was retrieved
|
|
Win32 error code - an error occured
|
|
|
|
--*/
|
|
DWORD
|
|
GetAPDPolicy(
|
|
IN HKEY hKey,
|
|
IN LPCWSTR pszRelPath,
|
|
IN LPCWSTR pszValueName,
|
|
IN LPDWORD pValue
|
|
)
|
|
{
|
|
DWORD Error = ERROR_INVALID_PARAMETER;
|
|
|
|
if (hKey && pszRelPath && pszValueName && pValue)
|
|
{
|
|
HKEY hRelKey = NULL;
|
|
|
|
*pValue = 0;
|
|
|
|
//
|
|
// Check if we have a value in the new location already
|
|
//
|
|
if ((Error = RegOpenKeyEx(hKey,
|
|
pszRelPath,
|
|
0,
|
|
KEY_READ,
|
|
&hRelKey)) == ERROR_SUCCESS)
|
|
{
|
|
DWORD cbData = sizeof(DWORD);
|
|
|
|
Error = RegQueryValueEx(hRelKey,
|
|
pszValueName,
|
|
NULL,
|
|
NULL,
|
|
(LPBYTE)pValue,
|
|
&cbData);
|
|
|
|
RegCloseKey(hRelKey);
|
|
}
|
|
}
|
|
|
|
return Error;
|
|
}
|
|
|
|
/*++
|
|
|
|
Routine Name
|
|
|
|
SetAPDPolicy
|
|
|
|
Routine Description:
|
|
|
|
This function writes a DWORD value to the location
|
|
HKEY\pszRelPath\pszValueName. We use this function for
|
|
preserving the AddPrinterDrivers policy value when the
|
|
LanMan Print Services print provider is deleted from
|
|
the system.
|
|
|
|
Arguments:
|
|
|
|
hKey - key tree
|
|
pszRelPath - relative path of the value to be set
|
|
pszValueName - value name to be set
|
|
Value - dword value to be set
|
|
|
|
Return Value:
|
|
|
|
ERROR_SUCCESS - the value was set sucessfully
|
|
Win32 error code - an error occured and the value was not set
|
|
|
|
--*/
|
|
DWORD
|
|
SetAPDPolicy(
|
|
IN HKEY hKey,
|
|
IN LPCWSTR pszRelPath,
|
|
IN LPCWSTR pszValueName,
|
|
IN DWORD Value
|
|
)
|
|
{
|
|
DWORD Error = ERROR_INVALID_PARAMETER;
|
|
|
|
if (hKey && pszRelPath && pszValueName)
|
|
{
|
|
HKEY hRelKey = NULL;
|
|
|
|
//
|
|
// Check if we have a value in the new location already
|
|
//
|
|
if ((Error = RegCreateKeyEx(hKey,
|
|
pszRelPath,
|
|
0,
|
|
NULL,
|
|
0,
|
|
KEY_ALL_ACCESS,
|
|
NULL,
|
|
&hRelKey,
|
|
NULL)) == ERROR_SUCCESS)
|
|
{
|
|
Error = RegSetValueEx(hRelKey,
|
|
pszValueName,
|
|
0,
|
|
REG_DWORD,
|
|
(LPBYTE)&Value,
|
|
sizeof(DWORD));
|
|
|
|
RegCloseKey(hRelKey);
|
|
}
|
|
}
|
|
|
|
return Error;
|
|
}
|
|
|