able/windows-nt
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity
windows-nt/Source/XPSP1/NT/printscan/print/spooler/localspl/inet.cxx

1377 lines
37 KiB
C++
Raw Normal View History

Add source files
2020-09-26 03:20:57 -05:00
/*****************************************************************************\
* MODULE: inet.cxx
*
* The module contains routines for the setting up the WWW Printer Service during spooler start up.
*
* The entry point here should be called by localspl\init.c\InitializePrintProvidor() once it is done
* with its work.
*
* Copyright (C) 1996 Microsoft Corporation
*
* History:
* Dec-1996 BabakJ Wrote it for IIS 2.0.
* June-1997 BabakJ Rewrote to use IIS 4.0's new Metabase interface
* Feb-1998 Weihaic Modify the URL in default.htm
* Feb 1999 BabakJ Made metabase interface a global to avoi calling too many CoCreateInstance() for perfrmance.
\*****************************************************************************/
//
//
// Note: We cannot use precomp.h here since we requrie ATL which can only be included in C++ source files.
//
//
#define INITGUID // babakj: (see comments in objbase.h) Needed to do it to get GUID_NULL defined.
#include "precomp.h"
#pragma hdrstop
#include <iadmw.h> // Interface header
#include <iiscnfg.h> // MD_ & IIS_MD_ defines
#include "..\spllib\webutil.hxx"
#define MY_META_TIMEOUT 1000
PWCHAR szW3SvcRootPath = L"/LM/W3svc/1/Root";
BOOL fW3SvcInstalled = FALSE; // Gobal flag telling if IIS or "Peer eb Server" is installed on the local machine.
PWCHAR szW3Root = NULL; // The WWWRoot dir, e.g. d:\inetpub\wwwroot
static CRITICAL_SECTION ClientCS;
static CRITICAL_SECTION ServerCS;
static HANDLE hMetaBaseThdReady;
static IMSAdminBase *pIMeta = NULL; // Metabase interface pointer
class CWebShareData {
public:
LPWSTR m_pszShareName;
BOOL m_bValid;
public:
CWebShareData (LPWSTR pszShareName);
~CWebShareData ();
int Compare (CWebShareData *pSecond) {return 0;};
};
class CWebShareList :
public CSingleList<CWebShareData*>
{
public:
CWebShareList () {};
~CWebShareList () {};
void WebSharePrinterList (void);
};
LPWSTR
mystrstrni(
LPWSTR pSrc,
LPWSTR pSearch
);
BOOL
CopyWebPrnFile(
VOID
);
BOOL
SetupWebPrnSvc(
IMSAdminBase *pIMSAdminBase,
BOOL fWebPrnDesired,
BOOL *pfW3SvcInstalled
);
BOOL
AddWebPrnSvc(
IMSAdminBase *pIMSAdminBase,
BOOL *pfW3SvcInstalled
);
BOOL
RemoveWebPrnSvc(
IMSAdminBase *pIMSAdminBase
);
BOOL
RemoveScript(
IMSAdminBase *pIMSAdminBase
);
BOOL
RemoveVirtualDir(
IMSAdminBase *pIMSAdminBase
);
BOOL
InstallWebPrnSvcWorker(
void
);
void
InstallWebPrnSvcWorkerThread(
PINISPOOLER pIniSpooler
);
BOOL
AddScriptAtPrinterVDir(
IMSAdminBase *pIMSAdminBase
);
BOOL
AddVirtualDir(
IMSAdminBase *pIMSAdminBase
);
void
WebShareWorker(
LPWSTR pShareName
);
BOOL
CreateVirtualDirForPrinterShare(
IMSAdminBase *pIMSAdminBase,
LPWSTR pShareName
);
void
WebUnShareWorker(
LPWSTR pShareName
);
BOOL
RemoveVirtualDirForPrinterShare(
IMSAdminBase *pIMSAdminBase,
LPWSTR pShareName
);
void
WebShareAllPrinters(
PINISPOOLER pIniSpooler
);
//
// This routine is called from init.c at localspl init time to kick start the whole thing.
//
// Make the COM activation on a separate thread in order not to slow down LocalSpl init process
//
void
InstallWebPrnSvc(
PINISPOOLER pIniSpooler
)
{
HANDLE ThreadHandle;
DWORD ThreadId;
HRESULT hr; // com error status
// Init the sync objects needed for device arrival thread management
InitializeCriticalSection( &ClientCS );
InitializeCriticalSection( &ServerCS );
hMetaBaseThdReady = CreateEvent( NULL, FALSE, FALSE, NULL); // Auto reset, non-signaled state
if (ThreadHandle = CreateThread( NULL,
INITIAL_STACK_COMMIT,
(LPTHREAD_START_ROUTINE)InstallWebPrnSvcWorkerThread,
pIniSpooler, 0, &ThreadId )) {
CloseHandle( ThreadHandle );
}
}
//////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
//////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
//
// We cannot free our interface pointer becasue loading and unloading ADMWPROX.DLL is very slow.
// So we have to keep the interface pointer around. But due to COM Apt limitation, the thread that creates the interface has
// to be alive for other threads to be able to use the pointer. So here we are with this fancy thread management code:
//
// The thread stays alive for a while. Then it goes away. Then future WebShare/Unshare would invoke it again.
// So This thread could be invoked 2 ways: First by spooler init code (only once!), then by WebShare/Unshare code.
//
//////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
//////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
#define EnterClient() EnterCriticalSection( &ClientCS );
#define EnterServer() EnterCriticalSection( &ServerCS );
#define LeaveClient() LeaveCriticalSection( &ClientCS );
#define LeaveServer() LeaveCriticalSection( &ServerCS );
static BOOL ThdAlive = FALSE;
///
/// Begin threading work
///
void
InstallWebPrnSvcWorkerThread(
PINISPOOLER pIniSpooler
)
{
IMSAdminBase *pILocalMetaBase = NULL;
SPLASSERT( (pIniSpooler == NULL) || (pIniSpooler == pLocalIniSpooler) ); // We only expect local pIniSpooler here!
EnterServer();
if( ThdAlive ) {
SPLASSERT( FALSE );
LeaveServer();
return;
}
if( FAILED (CoInitializeEx( NULL, COINIT_MULTITHREADED )) ||
FAILED (::CoCreateInstance(CLSID_MSAdminBase,
NULL,
CLSCTX_ALL,
IID_IMSAdminBase,
(void **)&pIMeta))) {
if( fW3SvcInstalled )
SetEvent( hMetaBaseThdReady ); // We must have a client thread waiting in WebShare/UnShare, signal it!
LeaveServer();
return;
}
if( !fW3SvcInstalled ) { // must be the first time we are being called.
if (InstallWebPrnSvcWorker()) {
WebShareAllPrinters( pIniSpooler );
fW3SvcInstalled = TRUE; // Once this is set, we never unset it. It is an indication that the WEb init code has succeeded.
}
}
else
SetEvent( hMetaBaseThdReady ); // event reset after a waiting thread is released.
ThdAlive = TRUE;
LeaveServer();
Sleep( 15 * 60 * 1000 ); // Allow other threads to use the COM pointer for 15 minutes
//
// Now tear down the IISADMIN object and self terminate thread. Ensure that
// we do not release the pointer inside the CS since this could take a long
// time, this could potentially cause a large number of WorkerThreads queueing
// up and doing the Release(), but that cannot be helped.
//
EnterServer();
pILocalMetaBase = pIMeta;
//
// The client thread expects valid pointers to be non-NULL!
//
pIMeta = NULL;
ThdAlive = FALSE;
LeaveServer();
pILocalMetaBase->Release();
CoUninitialize();
return;
}
void
WebShareManagement(
LPWSTR pShareName,
BOOL bShare // If TRUE, will share it, else unshare it.
) {
HANDLE ThreadHandle;
DWORD ThreadId;
if( !fW3SvcInstalled ) {
return;
}
if(FAILED (CoInitializeEx( NULL, COINIT_MULTITHREADED )))
return;
EnterClient();
EnterServer();
if( !ThdAlive ) {
LeaveServer();
if (ThreadHandle = CreateThread(NULL,
INITIAL_STACK_COMMIT,
(LPTHREAD_START_ROUTINE)InstallWebPrnSvcWorkerThread,
NULL,
0,
&ThreadId)) // sending NULL pIniSpooler since there is no need for it.
CloseHandle( ThreadHandle );
else {
LeaveClient();
return;
}
WaitForSingleObject( hMetaBaseThdReady, INFINITE ); // automatic reset event, so it is reset after a waiting thd released.
EnterServer();
}
// Now do the real work
if (pIMeta) {
if( bShare)
WebShareWorker( pShareName );
else
WebUnShareWorker( pShareName );
}
LeaveServer();
LeaveClient();
// No need to CoUninitialize();
}
///
/// End threading work
///
void
WebShareAllPrinters(
PINISPOOLER pIniSpooler
)
{
CWebShareList *pWebShareList = NULL;
BOOL bRet = TRUE;
PINIPRINTER pIniPrinter;
if (pWebShareList = new CWebShareList ()) {
// Go down the list of printers and share it out
EnterSplSem();
//
// Re-share all shared printers.
//
for( pIniPrinter = pIniSpooler->pIniPrinter;
pIniPrinter;
pIniPrinter = pIniPrinter->pNext ) {
if ( pIniPrinter->Attributes & PRINTER_ATTRIBUTE_SHARED ) {
CWebShareData *pData = new CWebShareData (pIniPrinter->pShareName);
if (pData && pData->m_bValid &&
pWebShareList->Insert (pData)) {
continue;
}
else {
if (pData) {
delete pData;
}
bRet = FALSE;
break;
}
}
}
LeaveSplSem ();
if (bRet) {
pWebShareList->WebSharePrinterList ();
}
delete pWebShareList;
}
}
PWSTR
GetPrinterUrl(
PSPOOL pSpool
)
{
PINIPRINTER pIniPrinter = pSpool->pIniPrinter;
DWORD cb;
PWSTR pszURL = NULL;
PWSTR pszServerName = NULL;
HRESULT hr;
SplInSem();
// http://machine/share
if (!pIniPrinter->pShareName)
goto error;
// Get FQDN of this machine
hr = GetDNSMachineName(pIniPrinter->pIniSpooler->pMachineName + 2, &pszServerName);
if (FAILED(hr)) {
SetLastError(HRESULT_CODE(hr));
goto error;
}
cb = 7 + wcslen(pszServerName); // http://machine
cb += 1 + wcslen(pIniPrinter->pShareName) + 1; // /share + NULL
cb *= sizeof(WCHAR);
if (pszURL = (PWSTR) AllocSplMem(cb)) {
wsprintf(pszURL, L"http://%ws/%ws", pszServerName, pIniPrinter->pShareName);
}
error:
FreeSplStr(pszServerName);
return pszURL;
}
//
//
// Reads the policy bit. Returns TRUE if Web Printing wanted, FASLE if not.
//
//
BOOL
IsWebPrintingDesired(
VOID
)
{
static PWCHAR szRegPolicyBitForDisableWebPrinting = L"Software\\Policies\\Microsoft\\Windows NT\\Printers";
HKEY hKey;
DWORD dwType;
DWORD cbData;
DWORD dwDataValue;
BOOL fRet = TRUE;
if( ERROR_SUCCESS == RegOpenKeyEx(HKEY_LOCAL_MACHINE,
szRegPolicyBitForDisableWebPrinting, 0, KEY_READ, &hKey)) {
cbData = sizeof(dwDataValue);
if( RegQueryValueEx(hKey, L"DisableWebPrinting" , NULL, &dwType, (LPBYTE)&dwDataValue, &cbData) == ERROR_SUCCESS ) {
// As long as the value does not exist, it means that web printing is enabled
// If the value is a DWORD, then if it is TRUE, Web Printing is Disabled
if (dwType == REG_DWORD && dwDataValue) fRet = FALSE;
}
RegCloseKey( hKey );
}
return fRet;
}
BOOL
InstallWebPrnSvcWorker(
VOID
)
{
WCHAR szTmpData[MAX_PATH];
HRESULT hr; // com error status
METADATA_HANDLE hMeta = NULL; // handle to metabase
BOOL fW3Svc = FALSE;
DWORD dwMDRequiredDataLen;
METADATA_RECORD mr;
// open key to ROOT on website #1 (default)
hr = pIMeta->OpenKey(METADATA_MASTER_ROOT_HANDLE,
L"/LM/W3svc/1",
METADATA_PERMISSION_READ,
MY_META_TIMEOUT,
&hMeta);
if( SUCCEEDED( hr )) {
// Get the physical path for the WWWROOT
mr.dwMDIdentifier = MD_VR_PATH;
mr.dwMDAttributes = 0;
mr.dwMDUserType = IIS_MD_UT_FILE;
mr.dwMDDataType = STRING_METADATA;
mr.dwMDDataLen = sizeof( szTmpData );
mr.pbMDData = reinterpret_cast<unsigned char *>(szTmpData);
hr = pIMeta->GetData( hMeta, L"/ROOT", &mr, &dwMDRequiredDataLen );
pIMeta->CloseKey( hMeta );
if( SUCCEEDED( hr )) {
szW3Root = AllocSplStr( szTmpData );
// Pass the inner dumb pointer for the callee to use
if( SetupWebPrnSvc( pIMeta, IsWebPrintingDesired(), &fW3Svc ))
DBGMSG(DBG_INFO, ("Setup of WWW Print Service successful.\n"));
else
DBGMSG(DBG_INFO, ("Setup of WWW Print Service failed.\n"));
}
}
return fW3Svc;
}
//
// Given that the WWW server is installed on local machine, it installs/removes the Web Printing service.
//
// Won't reinstall if it is already installed.
//
// Installation involves:
//
// - Add msw3prt as .printer. Just need to get the Win\sys dir.
// - Add the virtual dis .printer
//
// Uninstall means the removal of the above two.
//
//
BOOL
SetupWebPrnSvc(
IMSAdminBase *pIMSAdminBase,
BOOL fWebPrnDesired, // whether Web Printing is desired by the caller
BOOL *pfW3SvcInstalled // Whether Web Printing actually got installed by this routine.
)
{
if( fWebPrnDesired ) {
return AddWebPrnSvc( pIMSAdminBase, pfW3SvcInstalled );
}
else {
*pfW3SvcInstalled = FALSE;
return RemoveWebPrnSvc( pIMSAdminBase );
}
}
BOOL
AddWebPrnSvc(
IMSAdminBase *pIMSAdminBase,
BOOL *pfW3SvcInstalled // Whether Web Printing actually got installed by this routine.
)
{
HRESULT hr; // com error status
*pfW3SvcInstalled = FALSE; // Assume failure
//
// This is to remove the .printer script from the root
//
if( !RemoveScript( pIMSAdminBase ))
return FALSE;
if( !AddVirtualDir( pIMSAdminBase ))
return FALSE;
//
// Add ".printer" as a script map to the printers virtual directory
//
if( !AddScriptAtPrinterVDir( pIMSAdminBase ))
return FALSE;
// Flush out the changes and close
// Call SaveData() after making bulk changes, do not call it on each update
hr = pIMSAdminBase->SaveData();
if( FAILED( hr ))
return FALSE;
return( *pfW3SvcInstalled = TRUE ); // Web Printing installed.
}
BOOL
RemoveWebPrnSvc(
IMSAdminBase *pIMSAdminBase
)
{
HRESULT hr; // com error status
// Remove ".printer" as a script map from the website #1 (default)
if( !RemoveScript( pIMSAdminBase ))
return FALSE;
if( !RemoveVirtualDir( pIMSAdminBase ))
return FALSE;
// Flush out the changes and close
// Call SaveData() after making bulk changes, do not call it on each update
hr = pIMSAdminBase->SaveData();
if( FAILED( hr ))
return FALSE;
return( TRUE ); // Web Printing removed
}
//
//
// Finds the string pSearch in pSrc buffer and returns a ptr to the occurance of pSearch in pSrc.
//
//
LPWSTR mystrstrni( LPWSTR pSrc, LPWSTR pSearch )
{
UINT uSearchSize = wcslen( pSearch );
UINT uSrcSize = wcslen( pSrc );
LPCTSTR pEnd;
if( uSrcSize < uSearchSize )
return(NULL);
pEnd = pSrc + uSrcSize - uSearchSize;
for( ; pSrc <= pEnd; ++pSrc ) {
if( !_wcsnicmp( pSrc, pSearch, uSearchSize ))
return((LPWSTR)pSrc);
}
return(NULL);
}
//
// Determines if the string pSearch can be found inside of a MULTI_SZ string. If it can, it retunrs a
// pointer to the beginning of the string in multi-sz that contains pSearch.
//
LPWSTR IsStrInMultiSZ( LPWSTR pMultiSzSrc, LPWSTR pSearch )
{
LPWSTR pTmp = pMultiSzSrc;
while( TRUE ) {
if( mystrstrni( pTmp, pSearch )) // See pSearch (i.e. ".printer" appears anywhere within this string. If it does, it must be ours.
return pTmp;
pTmp = pTmp + (wcslen(pTmp) + 1); // Point to the beginning of the next string in the MULTI_SZ
if( !*pTmp )
return FALSE; // reached the end of the MULTI_SZ string.
}
}
#define W3SVC L"w3svc"
/*++
Routine Name:
AddScriptAtPrinterVDir
Description:
Add the .printer and .asp script mapping at printers virtual directory
Arguments:
pIMSAdminBase - Pointer to the IIS Admin base
Returns:
An HRESULT
--*/
BOOL
AddScriptAtPrinterVDir(
IMSAdminBase *pIMSAdminBase
)
{
static WCHAR szScritMapFmt[] = L"%ws%c.printer,%ws\\msw3prt.dll,1,GET,POST%c";
static WCHAR szPrinterVDir[] = L"w3svc/1/root/printers";
METADATA_HANDLE hMeta = NULL; // handle to metabase
PWCHAR szFullFormat = NULL;
DWORD dwMDRequiredDataLen;
METADATA_RECORD mr;
HRESULT hr = S_OK;
DWORD nLen;
WCHAR szSystemDir[MAX_PATH];
PWCHAR pAspMapping = NULL;
DWORD dwMapingLen = 0;
PWCHAR pScriptMap = NULL;
//
// Read any script map set at the top, on LM\w3svc where all other default ones are (e.g. .asp, etc.)
//
hr = pIMSAdminBase->OpenKey( METADATA_MASTER_ROOT_HANDLE,
L"/LM",
METADATA_PERMISSION_READ | METADATA_PERMISSION_WRITE,
MY_META_TIMEOUT,
&hMeta);
if(SUCCEEDED(hr))
{
mr.dwMDIdentifier = MD_SCRIPT_MAPS;
mr.dwMDAttributes = 0;
mr.dwMDUserType = IIS_MD_UT_FILE;
mr.dwMDDataType = MULTISZ_METADATA;
mr.dwMDDataLen = 0;
mr.pbMDData = NULL;
hr = pIMSAdminBase->GetData( hMeta, W3SVC, &mr, &dwMDRequiredDataLen );
hr = hr == HRESULT_FROM_WIN32 (ERROR_INSUFFICIENT_BUFFER) ? S_OK : E_FAIL;
}
if(SUCCEEDED(hr))
{
//
// allocate for existing stuff plus our new script map.
//
szFullFormat = new WCHAR[dwMDRequiredDataLen];
hr = szFullFormat? S_OK : E_OUTOFMEMORY;
}
if(SUCCEEDED(hr))
{
mr.dwMDIdentifier = MD_SCRIPT_MAPS;
mr.dwMDAttributes = 0;
mr.dwMDUserType = IIS_MD_UT_FILE;
mr.dwMDDataType = MULTISZ_METADATA;
mr.dwMDDataLen = dwMDRequiredDataLen * sizeof (WCHAR);
mr.pbMDData = reinterpret_cast<unsigned char *>(szFullFormat);
hr = pIMSAdminBase->GetData( hMeta, W3SVC, &mr, &dwMDRequiredDataLen );
}
if(SUCCEEDED(hr))
{
pAspMapping = IsStrInMultiSZ( szFullFormat, L".asp" );
hr = pAspMapping? S_OK: E_FAIL;
}
if(SUCCEEDED(hr))
{
nLen = COUNTOF (szScritMapFmt) + MAX_PATH + lstrlen (pAspMapping);
pScriptMap = new WCHAR[nLen];
hr = pScriptMap ? S_OK : E_OUTOFMEMORY;
}
if(SUCCEEDED(hr))
{
//
// Return value is the length in chars w/o null char.
//
hr = GetSystemDirectory( szSystemDir, COUNTOF (szSystemDir)) > 0 ? S_OK : E_FAIL;
}
if(SUCCEEDED(hr))
{
dwMapingLen = wsprintf( pScriptMap, szScritMapFmt, pAspMapping, L'\0', szSystemDir, L'\0');
hr = dwMapingLen > COUNTOF (szScritMapFmt) ? S_OK : E_FAIL;
}
if (SUCCEEDED(hr))
{
//
// Write the new SCRIPT value
//
mr.dwMDIdentifier = MD_SCRIPT_MAPS;
mr.dwMDAttributes = METADATA_INHERIT;
mr.dwMDUserType = IIS_MD_UT_FILE;
mr.dwMDDataType = MULTISZ_METADATA;
mr.dwMDDataLen = sizeof (WCHAR) * (dwMapingLen + 1) ;
mr.pbMDData = reinterpret_cast<unsigned char *>(pScriptMap);
hr = pIMSAdminBase->SetData( hMeta, szPrinterVDir, &mr );
}
if (hMeta)
{
pIMSAdminBase->CloseKey( hMeta );
hMeta = NULL;
}
delete [] pScriptMap;
delete [] szFullFormat;
return SUCCEEDED (hr);
}
//
//
// Finds and removed our script map from the multi_sz, and writes it back to the metabase.
//
//
BOOL
WriteStrippedScriptValue(
IMSAdminBase *pIMSAdminBase,
METADATA_HANDLE hMeta, // Handle to /LM tree
PWCHAR szFullFormat // MULTI_SZ string already there
)
{
LPWSTR pStrToKill, pNextStr;
HRESULT hr;
DBGMSG(DBG_INFO, ("Removing our script if already added.\n"));
// See if our script map is already there.
if( !(pStrToKill = IsStrInMultiSZ( szFullFormat, L".printer" )))
return TRUE;
// Find the next string (could be the final NULL char)
pNextStr = pStrToKill + (wcslen(pStrToKill) + 1);
if( !*pNextStr )
*pStrToKill = 0; // Our scipt map was at the end of multi_sz. Write the 2nd NULL char and we are done.
else
CopyMemory( pStrToKill, // Remove our script map by copying the remainder of the multi_sz on top of the string containing the map.
pNextStr,
GetMultiSZLen(pNextStr) * sizeof(WCHAR));
// Write the new SCRIPT value
METADATA_RECORD mr;
mr.dwMDIdentifier = MD_SCRIPT_MAPS;
mr.dwMDAttributes = METADATA_INHERIT;
mr.dwMDUserType = IIS_MD_UT_FILE;
mr.dwMDDataType = MULTISZ_METADATA;
mr.dwMDDataLen = GetMultiSZLen(szFullFormat) * sizeof(WCHAR);
mr.pbMDData = reinterpret_cast<unsigned char *>(szFullFormat);
hr = pIMSAdminBase->SetData( hMeta, W3SVC, &mr );
return( SUCCEEDED( hr ));
}
//
//
// Removes ".printer" as a script map from the website #1 (default) if alreaedy there
//
//
BOOL
RemoveScript(
IMSAdminBase *pIMSAdminBase
)
{
METADATA_HANDLE hMeta = NULL; // handle to metabase
PWCHAR szFullFormat;
DWORD dwMDRequiredDataLen;
METADATA_RECORD mr;
HRESULT hr;
BOOL fRet = FALSE;
// Read any script map set at the top, on LM\w3svc where all other default ones are (e.g. .asp, etc.)
hr = pIMSAdminBase->OpenKey( METADATA_MASTER_ROOT_HANDLE,
L"/LM",
METADATA_PERMISSION_READ | METADATA_PERMISSION_WRITE,
MY_META_TIMEOUT,
&hMeta);
if( SUCCEEDED( hr )) {
mr.dwMDIdentifier = MD_SCRIPT_MAPS;
mr.dwMDAttributes = 0;
mr.dwMDUserType = IIS_MD_UT_FILE;
mr.dwMDDataType = MULTISZ_METADATA;
mr.dwMDDataLen = 0;
mr.pbMDData = NULL;
hr = pIMSAdminBase->GetData( hMeta, W3SVC, &mr, &dwMDRequiredDataLen );
if( FAILED( hr )) {
if( HRESULT_CODE(hr) == ERROR_INSUFFICIENT_BUFFER ) {
if( szFullFormat = (PWCHAR)AllocSplMem( dwMDRequiredDataLen )) { // allocate for existing stuff plus our new script map.
mr.dwMDIdentifier = MD_SCRIPT_MAPS;
mr.dwMDAttributes = 0;
mr.dwMDUserType = IIS_MD_UT_FILE;
mr.dwMDDataType = MULTISZ_METADATA;
mr.dwMDDataLen = dwMDRequiredDataLen;
mr.pbMDData = reinterpret_cast<unsigned char *>(szFullFormat);
hr = pIMSAdminBase->GetData( hMeta, W3SVC, &mr, &dwMDRequiredDataLen );
if( SUCCEEDED( hr ))
fRet = WriteStrippedScriptValue( pIMSAdminBase, hMeta, szFullFormat ); // Remove the .printer map from the multi_sz if there;
FreeSplMem( szFullFormat );
}
}
}
pIMSAdminBase->CloseKey( hMeta );
}
return fRet;
}
/////////////////////////////////////////////////////////////////////////////////////////////////
static PWCHAR szPrinters = L"Printers";
#define PRINTERS szPrinters // Name of Printers virtual dir.
BOOL
AddVirtualDir(
IMSAdminBase *pIMSAdminBase
)
{
METADATA_HANDLE hMeta = NULL; // handle to metabase
WCHAR szVirPath[MAX_PATH];
WCHAR szPath[MAX_PATH];
DWORD dwMDRequiredDataLen;
DWORD dwAccessPerm;
METADATA_RECORD mr;
HRESULT hr;
BOOL fRet;
// Attempt to open the virtual dir set on Web server #1 (default server)
hr = pIMSAdminBase->OpenKey( METADATA_MASTER_ROOT_HANDLE,
szW3SvcRootPath,
METADATA_PERMISSION_READ | METADATA_PERMISSION_WRITE,
MY_META_TIMEOUT,
&hMeta );
// Create the key if it does not exist.
if( FAILED( hr ))
return FALSE;
fRet = TRUE;
mr.dwMDIdentifier = MD_VR_PATH;
mr.dwMDAttributes = 0;
mr.dwMDUserType = IIS_MD_UT_FILE;
mr.dwMDDataType = STRING_METADATA;
mr.dwMDDataLen = sizeof( szVirPath );
mr.pbMDData = reinterpret_cast<unsigned char *>(szVirPath);
// Read LM/W3Svc/1/Root/Printers see if MD_VR_PATH exists.
hr = pIMSAdminBase->GetData( hMeta, PRINTERS, &mr, &dwMDRequiredDataLen );
if( FAILED( hr )) {
fRet = FALSE;
if( hr == MD_ERROR_DATA_NOT_FOUND ||
HRESULT_CODE(hr) == ERROR_PATH_NOT_FOUND ) {
// Write both the key and the values if GetData() failed with any of the two errors.
pIMSAdminBase->AddKey( hMeta, PRINTERS );
if( GetWindowsDirectory( szPath, sizeof(szPath) / sizeof (TCHAR))) { // Return value is the length in chars w/o null char.
DBGMSG(DBG_INFO, ("Writing our virtual dir.\n"));
wsprintf( szVirPath, L"%ws\\web\\printers", szPath );
mr.dwMDIdentifier = MD_VR_PATH;
mr.dwMDAttributes = METADATA_INHERIT;
mr.dwMDUserType = IIS_MD_UT_FILE;
mr.dwMDDataType = STRING_METADATA;
mr.dwMDDataLen = (wcslen(szVirPath) + 1) * sizeof(WCHAR);
mr.pbMDData = reinterpret_cast<unsigned char *>(szVirPath);
// Write MD_VR_PATH value
hr = pIMSAdminBase->SetData( hMeta, PRINTERS, &mr );
fRet = SUCCEEDED( hr );
// Set the default authentication method
if( fRet ) {
DWORD dwAuthorization = MD_AUTH_NT; // NTLM only.
mr.dwMDIdentifier = MD_AUTHORIZATION;
mr.dwMDAttributes = METADATA_INHERIT; // need to inherit so that all subdirs are also protected.
mr.dwMDUserType = IIS_MD_UT_FILE;
mr.dwMDDataType = DWORD_METADATA;
mr.dwMDDataLen = sizeof(DWORD);
mr.pbMDData = reinterpret_cast<unsigned char *>(&dwAuthorization);
// Write MD_AUTHORIZATION value
hr = pIMSAdminBase->SetData( hMeta, PRINTERS, &mr );
fRet = SUCCEEDED( hr );
}
}
}
}
// In the following, do the stuff that we always want to do to the virtual dir, regardless of Admin's setting.
if( fRet ) {
dwAccessPerm = MD_ACCESS_READ | MD_ACCESS_SCRIPT;
mr.dwMDIdentifier = MD_ACCESS_PERM;
mr.dwMDAttributes = METADATA_INHERIT; // Make it inheritable so all subdirectories will have the same rights.
mr.dwMDUserType = IIS_MD_UT_FILE;
mr.dwMDDataType = DWORD_METADATA;
mr.dwMDDataLen = sizeof(DWORD);
mr.pbMDData = reinterpret_cast<unsigned char *>(&dwAccessPerm);
// Write MD_ACCESS_PERM value
hr = pIMSAdminBase->SetData( hMeta, PRINTERS, &mr );
fRet = SUCCEEDED( hr );
}
if( fRet ) {
PWCHAR szDefLoadFile = L"ipp_0001.asp";
mr.dwMDIdentifier = MD_DEFAULT_LOAD_FILE;
mr.dwMDAttributes = METADATA_INHERIT;
mr.dwMDUserType = IIS_MD_UT_FILE;
mr.dwMDDataType = STRING_METADATA;
mr.dwMDDataLen = (wcslen(szDefLoadFile) + 1) * sizeof(WCHAR);
mr.pbMDData = reinterpret_cast<unsigned char *>(szDefLoadFile);
// Write MD_DEFAULT_LOAD_FILE value
hr = pIMSAdminBase->SetData( hMeta, PRINTERS, &mr );
fRet = SUCCEEDED( hr );
}
if( fRet ) {
PWCHAR szKeyType = IIS_CLASS_WEB_VDIR_W;
mr.dwMDIdentifier = MD_KEY_TYPE;
mr.dwMDAttributes = METADATA_INHERIT;
mr.dwMDUserType = IIS_MD_UT_SERVER;
mr.dwMDDataType = STRING_METADATA;
mr.dwMDDataLen = (wcslen(szKeyType) + 1) * sizeof(WCHAR);
mr.pbMDData = reinterpret_cast<unsigned char *>(szKeyType);
// Write MD_DEFAULT_LOAD_FILE value
hr = pIMSAdminBase->SetData( hMeta, PRINTERS, &mr );
fRet = SUCCEEDED( hr );
}
pIMSAdminBase->CloseKey( hMeta );
return fRet;
}
//
//
// Removes "printers" virtual dir from IIS metabase ws3vc\1\root\printers
//
//
BOOL
RemoveVirtualDir(
IMSAdminBase *pIMSAdminBase
)
{
METADATA_HANDLE hMeta = NULL; // handle to metabase
HRESULT hr;
// Attempt to open the virtual dir set on Web server #1 (default server)
hr = pIMSAdminBase->OpenKey( METADATA_MASTER_ROOT_HANDLE,
szW3SvcRootPath,
METADATA_PERMISSION_READ | METADATA_PERMISSION_WRITE,
MY_META_TIMEOUT,
&hMeta );
// Create the key if it does not exist.
if( FAILED( hr ))
return FALSE;
pIMSAdminBase->DeleteKey( hMeta, PRINTERS ); // We don't check the retrun value since the key may already not exist and we could get an error for that reason.
pIMSAdminBase->CloseKey( hMeta );
return TRUE;
}
//=====================================================================================================
//=====================================================================================================
//=====================================================================================================
//
// Adding printer shares:
//
// To support http://<server>/<share>, we create a virtual directory with a redirect property.
//
//
//=====================================================================================================
//=====================================================================================================
//=====================================================================================================
//=====================================================================================================
//
// This function may be called during initialization time after fW3SvcInstalled is set,
// which means, a printer maybe webshared twice (once in InstallWebPrnSvcWorkerThread,
// and the other time when WebShare () is calleb by ShareThisPrinter() in net.c by
// FinalInitAfterRouterInitCompleteThread () during localspl initialization time.
//
//=====================================================================================================
void
WebShare(
LPWSTR pShareName
)
{
WebShareManagement( pShareName, TRUE );
}
void
WebShareWorker(
LPWSTR pShareName
)
{
HRESULT hr; // com error status
SPLASSERT( pIMeta != NULL );
// Pass the inner dumb pointer for the callee to use
if( CreateVirtualDirForPrinterShare( pIMeta, pShareName ))
// Flush out the changes and close
// Call SaveData() after making bulk changes, do not call it on each update
hr = pIMeta->SaveData();
}
BOOL
CreateVirtualDirForPrinterShare(
IMSAdminBase *pIMSAdminBase,
LPWSTR pShareName
)
{
METADATA_HANDLE hMeta = NULL; // handle to metabase
WCHAR szOldURL[MAX_PATH];
WCHAR szPath[MAX_PATH];
DWORD dwMDRequiredDataLen;
DWORD dwAccessPerm;
METADATA_RECORD mr;
HRESULT hr;
BOOL fRet;
// Attempt to open the virtual dir set on Web server #1 (default server)
hr = pIMSAdminBase->OpenKey( METADATA_MASTER_ROOT_HANDLE,
szW3SvcRootPath,
METADATA_PERMISSION_READ | METADATA_PERMISSION_WRITE,
MY_META_TIMEOUT,
&hMeta );
// Create the key if it does not exist.
if( FAILED( hr ))
return FALSE;
fRet = TRUE;
mr.dwMDIdentifier = MD_HTTP_REDIRECT;
mr.dwMDAttributes = 0;
mr.dwMDUserType = IIS_MD_UT_FILE;
mr.dwMDDataType = STRING_METADATA;
mr.dwMDDataLen = sizeof( szOldURL );
mr.pbMDData = reinterpret_cast<unsigned char *>(szOldURL);
// Read LM/W3Svc/1/Root/Printers to see if MD_HTTP_REDIRECT exists.
// Note that we are only concerned with the presence of the vir dir,
// not any properties it might have.
//
hr = pIMSAdminBase->GetData( hMeta, pShareName, &mr, &dwMDRequiredDataLen );
if( FAILED( hr )) {
fRet = FALSE;
// Notice if the virtual dir exists, we won't touch it. One scenario is
// if there is a name collision between a printer sharename and an existing,
// unrelated virtual dir.
if( HRESULT_CODE(hr) == ERROR_PATH_NOT_FOUND ) {
// Write both the key and the values if GetData() failed with any of the two errors.
pIMSAdminBase->AddKey( hMeta, pShareName );
dwAccessPerm = MD_ACCESS_READ;
mr.dwMDIdentifier = MD_ACCESS_PERM;
mr.dwMDAttributes = 0; // no need for inheritence
mr.dwMDUserType = IIS_MD_UT_FILE;
mr.dwMDDataType = DWORD_METADATA;
mr.dwMDDataLen = sizeof(DWORD);
mr.pbMDData = reinterpret_cast<unsigned char *>(&dwAccessPerm);
// Write MD_ACCESS_PERM value
hr = pIMSAdminBase->SetData( hMeta, pShareName, &mr );
fRet = SUCCEEDED( hr );
if( fRet ) {
PWCHAR szKeyType = IIS_CLASS_WEB_VDIR_W;
mr.dwMDIdentifier = MD_KEY_TYPE;
mr.dwMDAttributes = 0; // no need for inheritence
mr.dwMDUserType = IIS_MD_UT_FILE;
mr.dwMDDataType = STRING_METADATA;
mr.dwMDDataLen = (wcslen(szKeyType) + 1) * sizeof(WCHAR);
mr.pbMDData = reinterpret_cast<unsigned char *>(szKeyType);
// Write MD_DEFAULT_LOAD_FILE value
hr = pIMSAdminBase->SetData( hMeta, pShareName, &mr );
fRet = SUCCEEDED( hr );
}
if( fRet ) {
WCHAR szURL[MAX_PATH];
wsprintf( szURL, L"/printers/%ws/.printer", pShareName );
mr.dwMDIdentifier = MD_HTTP_REDIRECT;
mr.dwMDAttributes = 0; // no need for inheritence
mr.dwMDUserType = IIS_MD_UT_FILE;
mr.dwMDDataType = STRING_METADATA;
mr.dwMDDataLen = (wcslen(szURL) + 1) * sizeof(WCHAR);
mr.pbMDData = reinterpret_cast<unsigned char *>(szURL);
// Write MD_DEFAULT_LOAD_FILE value
hr = pIMSAdminBase->SetData( hMeta, pShareName, &mr );
fRet = SUCCEEDED( hr );
}
}
}
pIMSAdminBase->CloseKey( hMeta );
return fRet;
}
//=====================================================================================================
//=====================================================================================================
//=====================================================================================================
//
// Removing printer shares
//
//
//=====================================================================================================
//=====================================================================================================
//=====================================================================================================
void
WebUnShare(
LPWSTR pShareName
)
{
WebShareManagement( pShareName, FALSE );
}
void
WebUnShareWorker(
LPWSTR pShareName
)
{
HRESULT hr; // com error status
SPLASSERT( pIMeta != NULL );
// Pass the inner dumb pointer for the callee to use
if( RemoveVirtualDirForPrinterShare( pIMeta, pShareName ))
// Flush out the changes and close
// Call SaveData() after making bulk changes, do not call it on each update
hr = pIMeta->SaveData();
}
BOOL
RemoveVirtualDirForPrinterShare(
IMSAdminBase *pIMSAdminBase,
LPWSTR pShareName
)
{
METADATA_HANDLE hMeta = NULL; // handle to metabase
HRESULT hr;
// Attempt to open the virtual dir set on Web server #1 (default server)
hr = pIMSAdminBase->OpenKey( METADATA_MASTER_ROOT_HANDLE,
szW3SvcRootPath,
METADATA_PERMISSION_READ | METADATA_PERMISSION_WRITE,
MY_META_TIMEOUT,
&hMeta );
// Create the key if it does not exist.
if( FAILED( hr ))
return FALSE;
pIMSAdminBase->DeleteKey( hMeta, pShareName ); // We don't check the retrun value since the key may already not exist and we could get an error for that reason.
pIMSAdminBase->CloseKey( hMeta );
return TRUE;
}
CWebShareData::CWebShareData (LPWSTR pszShareName)
{
m_bValid = FALSE;
m_pszShareName = NULL;
if (m_pszShareName = new WCHAR[lstrlen (pszShareName) +1]) {
lstrcpy (m_pszShareName, pszShareName);
m_bValid = TRUE;
}
}
CWebShareData::~CWebShareData ()
{
if (m_pszShareName) {
delete [] m_pszShareName;
}
}
void CWebShareList::WebSharePrinterList ()
{
CSingleItem<CWebShareData*> * pItem = m_Dummy.GetNext();
CWebShareData * pData = NULL;
while (pItem && (pData = pItem->GetData ()) && pData->m_bValid) {
WebShareWorker (pData->m_pszShareName);
pItem = pItem->GetNext ();
}
}
Reference in a new issue Copy permalink