910 lines
25 KiB
C++
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;
|
||
|
|
}
|