windows-nt/Source/XPSP1/NT/printscan/print/spooler/inetsrv/gencab.cxx
2020-09-26 16:20:57 +08:00

910 lines
25 KiB
C++

/*****************************************************************************\
* MODULE: gencab.c
*
* The module contains routines for generating a self-extracting CAB file
* for the IExpress utility. This relies on the cdfCreate and infCreate
* to create the objects for the IExpress process.
*
*
* Copyright (C) 1996-1997 Microsoft Corporation
* Copyright (C) 1996-1997 Hewlett Packard
*
* History:
* 22-Nov-1996 <chriswil> Created.
*
\*****************************************************************************/
#include "pch.h"
#include "spool.h"
#include <initguid.h> // To define the IIS Metabase GUIDs used in this file. Every other file defines GUIDs as extern, except here where we actually define them in initguid.h.
#include <iadmw.h> // Interface header
#include <iiscnfg.h> // MD_ & IIS_MD_ defines
#define CAB_TIMEOUT 1000
/*****************************************************************************\
* Critical Section Handling
*
\*****************************************************************************/
DWORD gdwCritOwner = 0;
VOID cab_enter_crit(VOID)
{
EnterCriticalSection(&g_csGenCab);
gdwCritOwner = GetCurrentThreadId();
}
VOID cab_leave_crit(VOID)
{
gdwCritOwner = 0;
LeaveCriticalSection(&g_csGenCab);
}
VOID cab_check_crit(VOID)
{
DWORD dwCurrent = GetCurrentThreadId();
SPLASSERT((dwCurrent == gdwCritOwner));
}
#define EnterCABCrit() cab_enter_crit()
#define LeaveCABCrit() cab_leave_crit()
#define CheckCABCrit() cab_check_crit()
/*****************************************************************************\
* cab_SetClientSecurity (Local Routine)
*
* This routine sets the thread security so that it has max-privileges.
*
\*****************************************************************************/
HANDLE cab_SetClientSecurity(VOID)
{
HANDLE hToken;
HANDLE hThread = GetCurrentThread();
if (OpenThreadToken(hThread, TOKEN_IMPERSONATE, TRUE, &hToken)) {
if (SetThreadToken(&hThread, NULL)) {
return hToken;
}
CloseHandle(hToken);
}
return NULL;
}
/*****************************************************************************\
* cab_ResetClientSecurity (Local Routine)
*
* This routine resets the client-security to the token passed in. This
* is called in order to set the thread to the original security attributes.
*
\*****************************************************************************/
BOOL cab_ResetClientSecurity(
HANDLE hToken)
{
BOOL bRet;
bRet = SetThreadToken(NULL, hToken);
CloseHandle(hToken);
return bRet;
}
/*****************************************************************************\
* cab_iexpress_sync (Local Routine)
*
* This routine sychronizes the process and won't return until the process
* is done generating the CAB executable.
*
\*****************************************************************************/
BOOL cab_iexpress_sync(
HANDLE hProcess)
{
DWORD dwObj;
DWORD dwExitCode;
while (TRUE) {
dwObj = WaitForSingleObject(hProcess, INFINITE);
// Look for the exit type.
//
switch (dwObj) {
// The process handle triggered the wait. Let's get the
// exit-code and return whether the success. Otherwise,
// drop through and return the failure.
//
case WAIT_OBJECT_0:
GetExitCodeProcess(hProcess, &dwExitCode);
if (dwExitCode == 0)
return TRUE;
// Something failed in the call. We failed.
//
case WAIT_FAILED:
return FALSE;
}
}
}
/*****************************************************************************\
* cab_force_delete_file
*
* Reset a files attributes before deleting it. We copied them to the temp dir
* so we can delete them no matter what.
*
\*****************************************************************************/
VOID cab_force_delete_file(
LPCTSTR lpszFileName)
{
SetFileAttributes( lpszFileName, FILE_ATTRIBUTE_NORMAL );
DeleteFile( lpszFileName );
}
/*****************************************************************************\
* cab_cleanup_files
*
* Cleanup any files left-over in our generation process. Particularly,
* this will remove cab-files.
*
\*****************************************************************************/
VOID cab_cleanup_files(
LPCTSTR lpszDstPath)
{
TCHAR szAllExt[MIN_CAB_BUFFER];
HANDLE hFind;
DWORD idx;
DWORD cItems;
LPTSTR lpszCurDir;
WIN32_FIND_DATA wfd;
static LPCTSTR s_szFiles[] = {
g_szDotInf,
g_szDotBin,
g_szDotCat
};
// Put us into the destination directory where we
// want to cleanup the files.
//
if (lpszCurDir = genGetCurDir()) {
SetCurrentDirectory(lpszDstPath);
// Delete the temporary dat-file used in building
// the cdf-file
//
cab_force_delete_file(g_szDatFile);
// Delete certain extension files.
//
cItems = sizeof(s_szFiles) / sizeof(s_szFiles[0]);
for (idx = 0; idx < cItems; idx++) {
wsprintf(szAllExt, TEXT("*%s"), s_szFiles[idx]);
hFind = FindFirstFile(szAllExt, &wfd);
if (hFind && (hFind != INVALID_HANDLE_VALUE)) {
do {
cab_force_delete_file(wfd.cFileName);
} while (FindNextFile(hFind, &wfd));
FindClose(hFind);
}
}
SetCurrentDirectory(lpszCurDir);
genGFree(lpszCurDir, genGSize(lpszCurDir));
}
return;
}
/*****************************************************************************\
* cab_rename_cab
*
* Renames the .CAB to the dstname.
*
\*****************************************************************************/
BOOL cab_rename_cab(
LPCTSTR lpszDstName)
{
LPTSTR lpszSrc;
LPTSTR lpszDst;
BOOL bRet = FALSE;
if (lpszSrc = genBuildFileName(NULL, lpszDstName, g_szDotCab)) {
if (lpszDst = genBuildFileName(NULL, lpszDstName, g_szDotIpp)) {
bRet = MoveFileEx(lpszSrc, lpszDst, MOVEFILE_REPLACE_EXISTING);
genGFree(lpszDst, genGSize(lpszDst));
}
genGFree(lpszSrc, genGSize(lpszSrc));
}
return bRet;
}
/*****************************************************************************\
* cab_iexpress_exec (Local Routine)
*
* This exec's the IExpress utility to generate the self-extracting CAB
* file.
*
\*****************************************************************************/
BOOL cab_iexpress_exec(
LPCTSTR lpszCabDir,
LPCTSTR lpszDstName,
LPCTSTR lpszSedFile)
{
PROCESS_INFORMATION pi;
STARTUPINFO sti;
LPTSTR lpszCmd;
DWORD cbSize;
LPTSTR lpszOldDir;
BOOL bSuccess = FALSE;
TCHAR szWindowDir[MAX_PATH];
DWORD dwWinDirLen = 0;
if (dwWinDirLen = GetSystemWindowsDirectory (szWindowDir, MAX_PATH))
{
if (szWindowDir[dwWinDirLen - 1] == TEXT ('\\'))
{
szWindowDir[dwWinDirLen - 1] = 0;
}
// Calculate enough space to hold the command-line arguments.
//
cbSize = (dwWinDirLen + lstrlen(lpszSedFile) + lstrlen(g_szCabCmd) + 1) * sizeof(TCHAR);
// Allocate the command-line for the create-process call.
//
if (lpszCmd = (LPTSTR)genGAlloc(cbSize)) {
// Initialize startup-info fields.
//
memset(&sti, 0, sizeof(STARTUPINFO));
sti.cb = sizeof(STARTUPINFO);
// Build the command-line string that exec's IExpress.
//
wsprintf(lpszCmd, g_szCabCmd, szWindowDir, lpszSedFile);
// Change the directory to the cab/sed directory. It
// appears that IExpress requires this to generate.
//
if (lpszOldDir = genGetCurDir()) {
SetCurrentDirectory(lpszCabDir);
// Exec the process.
//
if (EXEC_PROCESS(lpszCmd, &sti, &pi)) {
CloseHandle(pi.hThread);
// This will wait until the process if finished generating
// the file. The return from this indicates whether the
// generation succeeded or not.
//
if (cab_iexpress_sync(pi.hProcess))
bSuccess = cab_rename_cab(lpszDstName);
CloseHandle(pi.hProcess);
}
// Restore our current-directory and free up any strings.
//
SetCurrentDirectory(lpszOldDir);
genGFree(lpszOldDir, genGSize(lpszOldDir));
}
genGFree(lpszCmd, cbSize);
}
}
return bSuccess;
}
/*****************************************************************************\
* cab_get_modtime
*
* Returns the modified-time of the CAB file. This can be used to determine
* whether this is out of date with other files.
*
\*****************************************************************************/
BOOL cab_get_modtime(
LPTSTR lpszOutPath,
LPFILETIME lpft)
{
HANDLE hFile;
BOOL bRet = FALSE;
// Open the file for reading.
//
hFile = gen_OpenFileRead(lpszOutPath);
// If the file exists and was opened, go get the time.
//
if (hFile && (hFile != INVALID_HANDLE_VALUE)) {
bRet = GetFileTime(hFile, NULL, NULL, lpft);
CloseHandle(hFile);
}
return bRet;
}
/*****************************************************************************\
* cab_get_drvname (Local Routine)
*
* This routine attempts to open the printer, and retrieve the printer
* driver-name associated with the FriendlyName. This returns to pointers
* to a driver and a share-name.
*
\*****************************************************************************/
BOOL cab_get_drvname(
LPCTSTR lpszFriendlyName,
LPTSTR* lpszDrvName,
LPTSTR* lpszShrName,
DWORD idxPlt
)
{
HANDLE hPrinter;
LPPRINTER_INFO_2 lppi;
DWORD cbBuf;
DWORD cbNeed;
BOOL bRet = FALSE;
// Initialize the pointers for the fail-case.
//
*lpszShrName = NULL;
*lpszDrvName = NULL;
// Open the printer and use the handle to query the printer for
// the driver-name.
//
if (OpenPrinter((LPTSTR)lpszFriendlyName, &hPrinter, NULL)) {
// First let's see how big our buffer will need to
// be in order to hold the PRINTER_INFO_2.
//
cbBuf = 0;
GetPrinter(hPrinter, PRT_LEV_2, NULL, 0, &cbBuf);
// Allocate storage for holding the print-info structure.
//
if (cbBuf && (lppi = (LPPRINTER_INFO_2)genGAlloc(cbBuf))) {
if (GetPrinter(hPrinter, PRT_LEV_2, (LPBYTE)lppi, cbBuf, &cbNeed)) {
//If this is a Win9X client make sure the driver name is correct
if ( genIsWin9X(idxPlt) )
{
// Call get printer driver to get the actual driver name for Win9X
DWORD rc, cbNeeded;
GetPrinterDriver( hPrinter, (LPTSTR) genStrCliEnvironment(idxPlt), 3,
NULL, 0, &cbNeeded);
rc = GetLastError();
if ( rc == ERROR_INSUFFICIENT_BUFFER )
{
LPBYTE pData;
DWORD dwSize = cbNeeded;
if ( pData = (LPBYTE) genGAlloc(cbNeeded) )
{
if ( GetPrinterDriver( hPrinter, (LPTSTR) genStrCliEnvironment(idxPlt), 3,
pData, dwSize, &cbNeeded) )
{
PDRIVER_INFO_3 pDriver = (PDRIVER_INFO_3) pData;
*lpszDrvName = genGAllocStr(pDriver->pName);
}
genGFree( pData, dwSize );
}
}
else
*lpszDrvName = genGAllocStr(lppi->pDriverName);
}
else
*lpszDrvName = genGAllocStr(lppi->pDriverName);
if ( *lpszDrvName ) {
if (*lpszShrName = genGAllocStr(lppi->pShareName)) {
bRet = TRUE;
} else {
genGFree(*lpszDrvName, genGSize(*lpszDrvName));
*lpszDrvName = NULL;
}
}
}
genGFree(lppi, cbBuf);
}
ClosePrinter(hPrinter);
}
return bRet;
}
/*****************************************************************************\
* cab_get_dstname
*
* Returns the then name of the destination-files. The name built
* incorporates the platform/version so that it is not duplicated with
* any other files.
*
* <Share ChkSum><Arch><Ver>.webpnp
*
* i.e. <share chksum>AXP2 - NT Alpha 4.0
* <share chksum>A863 - NT x86 5.0
* <share chksum>AXP3 - NT Alpha 5.0
* <share chksum>W9X0 - Win9X
*
\*****************************************************************************/
LPTSTR cab_get_dstname(
DWORD idxPlt,
DWORD idxVer,
LPCTSTR lpszShr)
{
int cch;
int cchBuf;
LPCTSTR lpszPlt;
LPTSTR lpszName = NULL;
// Get the platform requested for the client.
//
if (lpszPlt = genStrCliCab(idxPlt)) {
cchBuf = lstrlen(lpszPlt) + MIN_CAB_BUFFER;
// Build the cabname according to platform and version.
//
if (lpszName = (LPTSTR)genGAlloc(cchBuf * sizeof(TCHAR))) {
cch = wsprintf(lpszName, g_szCabName, genChkSum(lpszShr), lpszPlt, idxVer);
SPLASSERT((cch < cchBuf));
}
}
return lpszName;
}
/*****************************************************************************\
* cab_get_name
*
* Returns the name of the returned-file.
*
\*****************************************************************************/
VOID cab_get_name(
LPCTSTR lpszDstName,
LPCTSTR lpszShrName,
LPTSTR lpszName)
{
DWORD cch;
cch = wsprintf(lpszName, TEXT("/printers/%s/%s%s"), g_szPrtCabs, lpszDstName, g_szDotIpp);
SPLASSERT((cch < MAX_PATH));
return;
}
/*****************************************************************************\
* cab_get_webdir
*
* Returns the virtual-root-directory where the cab-files are to be generated
* and stored.
*
\*****************************************************************************/
LPTSTR cab_get_webdir(VOID)
{
BOOL bRet = FALSE;
HRESULT hr; // com error status
METADATA_HANDLE hMeta = NULL; // handle to metabase
CComPtr<IMSAdminBase> pIMeta; // ATL smart ptr
DWORD dwMDRequiredDataLen;
METADATA_RECORD mr;
HANDLE hToken = NULL;
LPTSTR lpszDir = NULL;
TCHAR szBuf[MAX_CAB_BUFFER];
// Need to revert to our service credential to be able to read IIS Metabase.
hToken = RevertToPrinterSelf();
// Create a instance of the metabase object
hr=::CoCreateInstance(CLSID_MSAdminBase,
NULL,
CLSCTX_ALL,
IID_IMSAdminBase,
(void **)&pIMeta);
if( SUCCEEDED( hr )) {
// open key to ROOT on website #1 (default)
hr = pIMeta->OpenKey(METADATA_MASTER_ROOT_HANDLE,
g_szMetabasePath,
METADATA_PERMISSION_READ,
CAB_TIMEOUT,
&hMeta);
if( SUCCEEDED( hr )) {
mr.dwMDIdentifier = MD_VR_PATH;
mr.dwMDAttributes = 0;
mr.dwMDUserType = IIS_MD_UT_FILE;
mr.dwMDDataType = STRING_METADATA;
mr.dwMDDataLen = sizeof(szBuf);
mr.pbMDData = reinterpret_cast<unsigned char *>(szBuf);
hr = pIMeta->GetData( hMeta, L"", &mr, &dwMDRequiredDataLen );
pIMeta->CloseKey( hMeta );
if( SUCCEEDED( hr ))
{
lpszDir = genBuildFileName(szBuf, g_szPrtCabs, NULL);
}
}
}
if (hToken)
{
ImpersonatePrinterClient(hToken);
}
return lpszDir;
}
/*****************************************************************************\
* cab_get_dstpath
*
* Returns a directory string where the cabinet files are to be generated.
*
\*****************************************************************************/
LPTSTR cab_get_dstpath(VOID)
{
HANDLE hDir;
LPTSTR lpszDir;
// First, we need to find the virtual-directory-root the cab files
// are located.
//
if (lpszDir = cab_get_webdir()) {
// Make sure the directory exists, or we can create it.
//
hDir = gen_OpenDirectory(lpszDir);
if (hDir && (hDir != INVALID_HANDLE_VALUE)) {
CloseHandle(hDir);
} else {
if (CreateDirectory(lpszDir, NULL) == FALSE) {
genGFree(lpszDir, genGSize(lpszDir));
lpszDir = NULL;
}
}
}
return lpszDir;
}
/*****************************************************************************\
* GenerateCAB
*
* Main entry-point for generating the CAB file.
*
* Parameters
* ----------
* lpszFriendlyName - Name of printer (shared-name).
* lpszPortName - Name of output port.
* dwCliInfo - Client platform/architecture/version information.
* lpszOutName - This is returned to the caller specifying the output.
*
\*****************************************************************************/
DWORD GenerateCAB(
LPCTSTR lpszFriendlyName,
LPCTSTR lpszPortName,
DWORD dwCliInfo,
LPTSTR lpszOutName,
BOOL bSecure)
{
INFGENPARM infParm;
HANDLE hToken;
HANDLE hsed;
HANDLE hinf;
FILETIME ftSED;
FILETIME ftCAB;
DWORD idxVer;
DWORD idxPlt;
BOOL bSed;
BOOL bCab;
LPTSTR lpszFullName;
LPTSTR lpszDrvName = NULL;
LPTSTR lpszShrName = NULL;
LPTSTR lpszDstName = NULL;
LPTSTR lpszDstPath = NULL;
LPTSTR lpszFrnName = NULL;
LPCTSTR lpszSedFile;
DWORD dwRet = HSE_STATUS_ERROR;
DWORD dwFailure = ERROR_SUCCESS;
// Initialize the security-token so that we will have
// max-privileges during the file-creation.
//
if ((hToken = cab_SetClientSecurity()) == NULL) {
DBGMSG(DBG_ERROR, ("GenerateCab : Access Denied"));
goto RetCabDone;
}
// Get the platform and version of the client (map to an index
// into tables). Validate the indexes to assure the information
// is valid.
//
idxPlt = genIdxCliPlatform(dwCliInfo);
idxVer = genIdxCliVersion(dwCliInfo);
#ifdef WIN95TST
// WORK : ChrisWil
if (idxPlt == IDX_W9X) {
lstrcpy(lpszOutName, TEXT("/hplaserj/cab_w9x0.webpnp"));
return HSE_STATUS_SUCCESS;
}
#endif
if ((idxPlt == IDX_UNKNOWN) || (idxVer == IDX_UNKNOWN)) {
dwFailure = ERROR_BAD_ENVIRONMENT;
DBGMSG(DBG_ERROR, ("GenerateCab : Unsupported client architecture/version"));
goto RetCabDone;
}
// Get the directory where the cab-files will be placed.
//
if ((lpszDstPath = cab_get_dstpath()) == NULL)
goto RetCabDone;
// Build a cluster-capable friendly-name.
//
if ((lpszFrnName = genFrnName(lpszFriendlyName)) == NULL)
goto RetCabDone;
// Get the driver information about the friendly-name.
//
if (cab_get_drvname(lpszFrnName, &lpszDrvName, &lpszShrName, idxPlt) == FALSE)
goto RetCabDone;
// Get the destination-name.
//
if ((lpszDstName = cab_get_dstname(idxPlt, idxVer, lpszShrName)) == NULL)
goto RetCabDone;
// Fill in the inf-input-parameters. These values should be
// validated at this point.
//
infParm.lpszFriendlyName = lpszFrnName;
infParm.lpszShareName = lpszShrName;
infParm.lpszPortName = lpszPortName;
infParm.idxPlt = idxPlt;
infParm.idxVer = idxVer;
infParm.dwCliInfo = dwCliInfo;
infParm.lpszDrvName = lpszDrvName;
infParm.lpszDstName = lpszDstName;
infParm.lpszDstPath = lpszDstPath;
// Grab the critical-section while we deal with generating
// the files for the driver-install.
//
EnterCABCrit();
// Call to have the INF/CDF files generated. If
// all goes well, then the SED file can be generated
// using the INF as input.
//
if (hinf = infCreate(&infParm)) {
// We created an INF object. Now process the INF
if ( infProcess(hinf) ) {
// Got good INF so now work on CDF
if (hsed = cdfCreate(hinf, bSecure)) {
// We created a CDF object. Now process the CDF
if ( cdfProcess(hsed) ) {
// Allocate a string representing the full-path-name of
// the generated executable.
//
lpszFullName = genBuildFileName(lpszDstPath, lpszDstName, g_szDotIpp);
if (lpszFullName) {
// Get the name of the directive file that we'll be using
// to lauch the iexpress-package with. Do not delete this
// pointer as it is handled by the SED object.
//
if (lpszSedFile = cdfGetName(hsed)) {
// Retrieve modified dates so that we can determine whether
// to generate a new-CAB or return an existing one. If the
// calls return FALSE, the we can assume a file doesn't
// exist, so we'll generate an new one anyway.
//
bSed = cdfGetModTime(hsed, &ftSED);
bCab = cab_get_modtime(lpszFullName, &ftCAB);
// Get the name of the Cab-File that will be
// generated (or returned). This is relative to the
// wwwroot path and is in the form /share/printer.
//
cab_get_name(lpszDstName, lpszShrName, lpszOutName);
// If the bCabMod is TRUE (meaning we have a CAB), and the
// SED is not newer than the CAB, then we really only need
// to return the existing CAB. Otherwise, some files
// must have changed.
//
if ((bSed && bCab) && (CompareFileTime(&ftSED, &ftCAB) <= 0)) {
goto RetCabName;
} else {
if (cab_iexpress_exec(lpszDstPath, lpszDstName, lpszSedFile)) {
RetCabName:
dwRet = HSE_STATUS_SUCCESS;
}
else
dwFailure = ERROR_CANNOT_MAKE;
}
}
else // If cdfGetName
dwFailure = GetLastError();
genGFree(lpszFullName, genGSize(lpszFullName));
}
else // If lpszFullName
dwFailure = GetLastError();
}
else // If cdfProcess()
dwFailure = cdfGetError(hsed); // Get saved error
cdfCleanUpSourceFiles(hinf); // Even if cdfProcess fails it might have
// partialy generated some temporary files
cdfDestroy(hsed);
}
else // If cdfCreate
dwFailure = GetLastError();
}
else // If infProcess()
dwFailure = infGetError(hinf); // Get saved error
infDestroy(hinf);
}
else // If infCreate
dwFailure = GetLastError();
// Cleanup the files generated during processing.
//
cab_cleanup_files(lpszDstPath);
RetCabDone:
// If something failed but we don't know what yet.. Get the error
if ( (dwRet == HSE_STATUS_ERROR) && ( dwFailure == ERROR_SUCCESS ) )
dwFailure = GetLastError();
// Free our strings allocated through this scope.
//
if (lpszFrnName)
genGFree(lpszFrnName, genGSize(lpszFrnName));
if (lpszDrvName)
genGFree(lpszDrvName, genGSize(lpszDrvName));
if (lpszDstName)
genGFree(lpszDstName, genGSize(lpszDstName));
if (lpszShrName)
genGFree(lpszShrName, genGSize(lpszShrName));
if (lpszDstPath)
genGFree(lpszDstPath, genGSize(lpszDstPath));
if (hToken)
cab_ResetClientSecurity(hToken);
LeaveCABCrit();
if (dwFailure != ERROR_SUCCESS)
SetLastError(dwFailure);
return dwRet;
}