1330 lines
42 KiB
C++
1330 lines
42 KiB
C++
|
#include "ParseInf.h"
|
||
|
#include "general.h"
|
||
|
#include <shlwapi.h>
|
||
|
#include <wininet.h>
|
||
|
|
||
|
//#define USE_SHORT_PATH_NAME 1
|
||
|
|
||
|
#define REG_PATH_IE_CACHE_LIST TEXT("SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Internet Settings\\ActiveX Cache")
|
||
|
|
||
|
#define cCachePathsMax 5 // maximum number of legacy caches + the current cache du jour
|
||
|
|
||
|
|
||
|
struct OCCFindData
|
||
|
{
|
||
|
LPCLSIDLIST_ITEM m_pcliHead;
|
||
|
LPCLSIDLIST_ITEM m_pcliTail;
|
||
|
struct {
|
||
|
TCHAR m_sz[MAX_PATH];
|
||
|
DWORD m_cch;
|
||
|
} m_aCachePath[cCachePathsMax];
|
||
|
|
||
|
OCCFindData();
|
||
|
~OCCFindData();
|
||
|
|
||
|
BOOL IsCachePath( LPCTSTR szPath );
|
||
|
|
||
|
// Control List operations
|
||
|
HRESULT AddListItem( LPCTSTR szFileName, LPCTSTR szCLSID, DWORD dwIsDistUnit );
|
||
|
LPCLSIDLIST_ITEM TakeFirstItem(void);
|
||
|
};
|
||
|
|
||
|
DWORD CCacheLegacyControl::s_dwType = 1;
|
||
|
DWORD CCacheDistUnit::s_dwType = 2;
|
||
|
|
||
|
HRESULT CCacheLegacyControl::Init( HKEY hkeyCLSID, LPCTSTR szFile, LPCTSTR szCLSID )
|
||
|
{
|
||
|
HRESULT hr = S_OK;
|
||
|
|
||
|
lstrcpyn(m_szFile, szFile, ARRAYSIZE(m_szFile));
|
||
|
lstrcpyn(m_szCLSID, szCLSID, ARRAYSIZE(m_szCLSID));
|
||
|
|
||
|
// Get full user type name
|
||
|
m_szName[0] = '\0';
|
||
|
DWORD dw = sizeof(m_szName);
|
||
|
LRESULT lResult = RegQueryValue(hkeyCLSID, m_szCLSID, m_szName, (LONG*)&dw);
|
||
|
// if the fails, we should get a resource string (seanf 5/9/97 )
|
||
|
// Get type lib id
|
||
|
TCHAR szTypeLibValName[MAX_PATH];
|
||
|
CatPathStrN( szTypeLibValName, szCLSID, HKCR_TYPELIB, ARRAYSIZE(szTypeLibValName) );
|
||
|
|
||
|
dw = sizeof(m_szTypeLibID);
|
||
|
lResult = RegQueryValue( hkeyCLSID, szTypeLibValName, m_szTypeLibID, (LONG*)&dw);
|
||
|
if (lResult != ERROR_SUCCESS)
|
||
|
m_szTypeLibID[0] = TEXT('\0');
|
||
|
|
||
|
// Set Codebase
|
||
|
m_szCodeBase[0] = '\0';
|
||
|
m_szVersion[0] = '\0';
|
||
|
hr = DoParse( m_szFile, m_szCLSID );
|
||
|
|
||
|
return hr;
|
||
|
}
|
||
|
|
||
|
HRESULT CCacheDistUnit::Init( HKEY hkeyCLSID, LPCTSTR szFile, LPCTSTR szCLSID, HKEY hkeyDist, LPCTSTR szDU )
|
||
|
{
|
||
|
HRESULT hr = S_OK;
|
||
|
HKEY hkeyDU;
|
||
|
HKEY hkeyDLInfo; // DownloadInformation subkey
|
||
|
HKEY hkeyVers; // InstalledVersion subkey
|
||
|
HKEY hkeyCOM; // subkey of HKCR\CLSID, used if outside of cache dir
|
||
|
LRESULT lResult = ERROR_SUCCESS;
|
||
|
DWORD dw;
|
||
|
TCHAR szNameT[MAX_PATH];
|
||
|
UINT uiVerSize = 0;
|
||
|
DWORD dwVerSize = 0;
|
||
|
DWORD dwHandle = 0;
|
||
|
BYTE *pbBuffer = NULL;
|
||
|
HANDLE hFile;
|
||
|
FILETIME ftLastAccess;
|
||
|
BOOL bRunOnNT5 = FALSE;
|
||
|
OSVERSIONINFO osvi;
|
||
|
VS_FIXEDFILEINFO *lpVSInfo = NULL;
|
||
|
|
||
|
if ( szFile[0] == '\0' &&
|
||
|
RegOpenKeyEx( hkeyCLSID, szCLSID, 0, KEY_READ, &hkeyCOM ) == ERROR_SUCCESS )
|
||
|
{
|
||
|
LONG lcb = sizeof(szNameT);
|
||
|
lResult = RegQueryValue( hkeyCOM, INPROCSERVER, szNameT, &lcb );
|
||
|
|
||
|
if ( lResult != ERROR_SUCCESS )
|
||
|
{
|
||
|
lcb = sizeof(szNameT);
|
||
|
lResult = RegQueryValue( hkeyCOM, INPROCSERVER32, szNameT, &lcb );
|
||
|
}
|
||
|
|
||
|
if ( lResult != ERROR_SUCCESS )
|
||
|
{
|
||
|
lcb = sizeof(szNameT);
|
||
|
lResult = RegQueryValue( hkeyCOM, INPROCSERVERX86, szNameT, &lcb );
|
||
|
}
|
||
|
|
||
|
if ( lResult != ERROR_SUCCESS )
|
||
|
{
|
||
|
lcb = sizeof(szNameT);
|
||
|
lResult = RegQueryValue( hkeyCOM, LOCALSERVER, szNameT, &lcb );
|
||
|
}
|
||
|
|
||
|
if ( lResult != ERROR_SUCCESS )
|
||
|
{
|
||
|
lcb = sizeof(szNameT);
|
||
|
lResult = RegQueryValue( hkeyCOM, LOCALSERVER32, szNameT, &lcb );
|
||
|
}
|
||
|
|
||
|
if ( lResult != ERROR_SUCCESS )
|
||
|
{
|
||
|
lcb = sizeof(szNameT);
|
||
|
lResult = RegQueryValue( hkeyCOM, LOCALSERVERX86, szNameT, &lcb );
|
||
|
}
|
||
|
|
||
|
RegCloseKey( hkeyCOM );
|
||
|
}
|
||
|
else
|
||
|
lstrcpyn( szNameT, szFile, ARRAYSIZE(szNameT));
|
||
|
|
||
|
if ( lResult != ERROR_SUCCESS ) // needed to find file path but couldn't
|
||
|
szNameT[0] = '\0';
|
||
|
|
||
|
hr = CCacheLegacyControl::Init( hkeyCLSID, szNameT, szCLSID );
|
||
|
|
||
|
if ( FAILED(hr) )
|
||
|
return hr;
|
||
|
|
||
|
lResult = RegOpenKeyEx(hkeyDist, szDU, 0, KEY_READ, &hkeyDU);
|
||
|
if (lResult != ERROR_SUCCESS)
|
||
|
return E_FAIL;
|
||
|
|
||
|
// Get CLSID
|
||
|
lstrcpyn(m_szCLSID, szDU, MAX_DIST_UNIT_NAME_LEN);
|
||
|
|
||
|
// Get full user type name - only override the control name if DU name is not empty
|
||
|
dw = sizeof(szNameT);
|
||
|
lResult = RegQueryValue(hkeyDU, NULL, szNameT, (LONG*)&dw);
|
||
|
if ( lResult == ERROR_SUCCESS && szNameT[0] != '\0' )
|
||
|
{
|
||
|
lstrcpyn( m_szName, szNameT, ARRAYSIZE(m_szName) );
|
||
|
}
|
||
|
else if ( *m_szName == '\0' ) // worst case, if we still don't have a name, a GUID will suffice
|
||
|
lstrcpyn( m_szName, szDU, ARRAYSIZE(m_szName) );
|
||
|
|
||
|
// Get type lib id
|
||
|
// Get type lib id
|
||
|
TCHAR szTypeLibValName[MAX_PATH];
|
||
|
CatPathStrN(szTypeLibValName, m_szCLSID, HKCR_TYPELIB, ARRAYSIZE(szTypeLibValName));
|
||
|
dw = sizeof(m_szTypeLibID);
|
||
|
lResult = RegQueryValue( hkeyCLSID, szTypeLibValName, m_szTypeLibID, (LONG*)&dw);
|
||
|
if (lResult != ERROR_SUCCESS)
|
||
|
(m_szTypeLibID)[0] = TEXT('\0');
|
||
|
|
||
|
m_szCodeBase[0] ='\0';
|
||
|
lResult = RegOpenKeyEx(hkeyDU, REGSTR_DOWNLOAD_INFORMATION, 0, KEY_READ, &hkeyDLInfo);
|
||
|
if (lResult == ERROR_SUCCESS)
|
||
|
{
|
||
|
dw = sizeof(m_szCodeBase);
|
||
|
HRESULT hrErr = RegQueryValueEx(hkeyDLInfo, REGSTR_DLINFO_CODEBASE, NULL, NULL,
|
||
|
(unsigned char *)m_szCodeBase, &dw);
|
||
|
RegCloseKey( hkeyDLInfo );
|
||
|
}
|
||
|
|
||
|
// Get Version from DU branch
|
||
|
|
||
|
m_szVersion[0] ='\0';
|
||
|
lResult = RegOpenKeyEx(hkeyDU, REGSTR_INSTALLED_VERSION, 0,
|
||
|
KEY_READ, &hkeyVers);
|
||
|
if (lResult == ERROR_SUCCESS)
|
||
|
{
|
||
|
dw = sizeof(m_szVersion);
|
||
|
RegQueryValueEx(hkeyVers, NULL, NULL, NULL, (LPBYTE)m_szVersion, &dw);
|
||
|
RegCloseKey(hkeyVers);
|
||
|
}
|
||
|
|
||
|
// The version specified in the COM branch is the definitive word on
|
||
|
// what the version is. If a key exists in the COM branch, use the version
|
||
|
// that is found inside the InProcServer/LocalServer.
|
||
|
|
||
|
if (RegOpenKeyEx( hkeyCLSID, szCLSID, 0, KEY_READ, &hkeyCOM ) == ERROR_SUCCESS)
|
||
|
{
|
||
|
LONG lcb = sizeof(szNameT);
|
||
|
lResult = RegQueryValue( hkeyCOM, INPROCSERVER32, szNameT, &lcb );
|
||
|
|
||
|
if ( lResult != ERROR_SUCCESS )
|
||
|
{
|
||
|
lcb = sizeof(szNameT);
|
||
|
lResult = RegQueryValue( hkeyCOM, INPROCSERVER, szNameT, &lcb );
|
||
|
}
|
||
|
|
||
|
if ( lResult != ERROR_SUCCESS )
|
||
|
{
|
||
|
lcb = sizeof(szNameT);
|
||
|
lResult = RegQueryValue( hkeyCOM, INPROCSERVERX86, szNameT, &lcb );
|
||
|
}
|
||
|
|
||
|
if ( lResult != ERROR_SUCCESS )
|
||
|
{
|
||
|
lcb = sizeof(szNameT);
|
||
|
lResult = RegQueryValue( hkeyCOM, LOCALSERVER32, szNameT, &lcb );
|
||
|
}
|
||
|
|
||
|
if ( lResult != ERROR_SUCCESS )
|
||
|
{
|
||
|
lcb = sizeof(szNameT);
|
||
|
lResult = RegQueryValue( hkeyCOM, LOCALSERVER, szNameT, &lcb );
|
||
|
}
|
||
|
|
||
|
if ( lResult != ERROR_SUCCESS )
|
||
|
{
|
||
|
lcb = sizeof(szNameT);
|
||
|
lResult = RegQueryValue( hkeyCOM, LOCALSERVERX86, szNameT, &lcb );
|
||
|
}
|
||
|
|
||
|
RegCloseKey( hkeyCOM );
|
||
|
|
||
|
// HACK! GetFileVersionInfoSize and GetFileVersionInfo modify
|
||
|
// the last access time of the file under NT5! This causes us
|
||
|
// to retrieve the wrong last access time when removing expired
|
||
|
// controls. This hack gets the last access time before the
|
||
|
// GetFileVersionInfo calls, and sets it back afterwards.
|
||
|
// See IE5 RAID #56927 for details. This code should be removed
|
||
|
// when NT5 fixes this bug.
|
||
|
|
||
|
osvi.dwOSVersionInfoSize = sizeof(osvi);
|
||
|
GetVersionEx(&osvi);
|
||
|
|
||
|
if (osvi.dwPlatformId == VER_PLATFORM_WIN32_NT && osvi.dwMajorVersion == 5) {
|
||
|
bRunOnNT5 = TRUE;
|
||
|
}
|
||
|
|
||
|
if (bRunOnNT5) {
|
||
|
hFile = CreateFile(szNameT, GENERIC_READ, FILE_SHARE_READ, NULL,
|
||
|
OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
|
||
|
|
||
|
if (hFile != INVALID_HANDLE_VALUE) {
|
||
|
GetFileTime(hFile, NULL, &ftLastAccess, NULL);
|
||
|
CloseHandle(hFile);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
dwVerSize = GetFileVersionInfoSize((char *)szNameT, &dwHandle);
|
||
|
pbBuffer = new BYTE[dwVerSize];
|
||
|
if (!pbBuffer)
|
||
|
{
|
||
|
return E_OUTOFMEMORY;
|
||
|
}
|
||
|
if (GetFileVersionInfo((char *)szNameT, 0, dwVerSize, pbBuffer))
|
||
|
{
|
||
|
if (VerQueryValue(pbBuffer, "\\", (void **)&lpVSInfo, &uiVerSize))
|
||
|
{
|
||
|
wsprintf(m_szVersion, "%d,%d,%d,%d", (lpVSInfo->dwFileVersionMS >> 16) & 0xFFFF
|
||
|
, lpVSInfo->dwFileVersionMS & 0xFFFF
|
||
|
, (lpVSInfo->dwFileVersionLS >> 16) & 0xFFFF
|
||
|
, lpVSInfo->dwFileVersionLS & 0xFFFF);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
delete [] pbBuffer;
|
||
|
|
||
|
if (bRunOnNT5) {
|
||
|
hFile = CreateFile(szNameT, GENERIC_WRITE, FILE_SHARE_READ, NULL,
|
||
|
OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
|
||
|
if (hFile != INVALID_HANDLE_VALUE) {
|
||
|
SetFileTime(hFile, NULL, &ftLastAccess, NULL);
|
||
|
CloseHandle(hFile);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
}
|
||
|
|
||
|
RegCloseKey( hkeyDU );
|
||
|
|
||
|
return DoParseDU( m_szFile, m_szCLSID);
|
||
|
}
|
||
|
|
||
|
HRESULT MakeCacheItemFromControlList( HKEY hkeyClass, // HKCR\CLSID
|
||
|
HKEY hkeyDist, // HKLM\SOFTWARE\MICROSOFT\Code Store Database\Distribution Units
|
||
|
LPCLSIDLIST_ITEM pcli,
|
||
|
CCacheItem **ppci )
|
||
|
{
|
||
|
HRESULT hr = E_FAIL;
|
||
|
|
||
|
*ppci = NULL;
|
||
|
if ( pcli->bIsDistUnit )
|
||
|
{
|
||
|
CCacheDistUnit *pcdu = new CCacheDistUnit();
|
||
|
if ( pcdu != NULL &&
|
||
|
SUCCEEDED(hr = pcdu->Init( hkeyClass,
|
||
|
pcli->szFile,
|
||
|
pcli->szCLSID,
|
||
|
hkeyDist,
|
||
|
pcli->szCLSID)) )
|
||
|
*ppci = pcdu;
|
||
|
else
|
||
|
hr = E_OUTOFMEMORY;
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
CCacheLegacyControl *pclc = new CCacheLegacyControl();
|
||
|
if ( pclc != NULL &&
|
||
|
SUCCEEDED(hr = pclc->Init( hkeyClass,
|
||
|
pcli->szFile,
|
||
|
pcli->szCLSID )) )
|
||
|
*ppci = pclc;
|
||
|
else
|
||
|
hr = E_OUTOFMEMORY;
|
||
|
|
||
|
}
|
||
|
|
||
|
return hr;
|
||
|
}
|
||
|
|
||
|
OCCFindData::OCCFindData() : m_pcliHead(NULL), m_pcliTail(NULL)
|
||
|
{
|
||
|
LONG lResult;
|
||
|
HKEY hkeyCacheList;
|
||
|
|
||
|
for ( int i = 0; i < cCachePathsMax; i++ )
|
||
|
{
|
||
|
m_aCachePath[i].m_cch = 0;
|
||
|
m_aCachePath[i].m_sz[0] = '\0';
|
||
|
}
|
||
|
|
||
|
// Unhook occache as a shell extension for the cache folders.
|
||
|
lResult = RegOpenKeyEx( HKEY_LOCAL_MACHINE,
|
||
|
REG_PATH_IE_CACHE_LIST,
|
||
|
0x0,
|
||
|
KEY_READ,
|
||
|
&hkeyCacheList );
|
||
|
|
||
|
if ( lResult == ERROR_SUCCESS ) {
|
||
|
DWORD dwIndex;
|
||
|
TCHAR szName[MAX_PATH];
|
||
|
DWORD cbName;
|
||
|
DWORD cbValue;
|
||
|
|
||
|
for ( dwIndex = 0, cbName = sizeof(szName), cbValue = MAX_PATH * sizeof(TCHAR);
|
||
|
dwIndex < cCachePathsMax;
|
||
|
dwIndex++, cbName = sizeof(szName), cbValue = MAX_PATH * sizeof(TCHAR) )
|
||
|
{
|
||
|
lResult = RegEnumValue( hkeyCacheList, dwIndex,
|
||
|
szName, &cbName,
|
||
|
NULL, NULL,
|
||
|
(LPBYTE)m_aCachePath[dwIndex].m_sz, &cbValue );
|
||
|
m_aCachePath[dwIndex].m_cch = lstrlen( m_aCachePath[dwIndex].m_sz );
|
||
|
}
|
||
|
// We leave this key in place because it is the only record we have of the
|
||
|
// cache folders and would be useful to future installations of IE
|
||
|
RegCloseKey( hkeyCacheList );
|
||
|
}
|
||
|
}
|
||
|
|
||
|
OCCFindData::~OCCFindData()
|
||
|
{
|
||
|
if ( m_pcliHead )
|
||
|
RemoveList(m_pcliHead);
|
||
|
}
|
||
|
|
||
|
BOOL OCCFindData::IsCachePath( LPCTSTR szPath )
|
||
|
{
|
||
|
BOOL fMatch = FALSE;
|
||
|
|
||
|
for ( int i = 0; i < cCachePathsMax && !fMatch; i++ )
|
||
|
fMatch = m_aCachePath[i].m_cch != 0 &&
|
||
|
LStrNICmp( szPath, m_aCachePath[i].m_sz, m_aCachePath[i].m_cch ) == 0;
|
||
|
return fMatch;
|
||
|
}
|
||
|
|
||
|
HRESULT OCCFindData::AddListItem( LPCTSTR szFile, LPCTSTR szCLSID, DWORD dwIsDistUnit )
|
||
|
{
|
||
|
HRESULT hr = S_OK;
|
||
|
|
||
|
if ( m_pcliTail == NULL )
|
||
|
{
|
||
|
m_pcliTail = new CLSIDLIST_ITEM;
|
||
|
if (m_pcliHead == NULL)
|
||
|
m_pcliHead = m_pcliTail;
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
m_pcliTail->pNext = new CLSIDLIST_ITEM;
|
||
|
m_pcliTail = m_pcliTail->pNext;
|
||
|
}
|
||
|
|
||
|
if ( m_pcliTail != NULL )
|
||
|
{
|
||
|
m_pcliTail->pNext = NULL;
|
||
|
lstrcpyn(m_pcliTail->szFile, szFile, MAX_PATH);
|
||
|
lstrcpyn(m_pcliTail->szCLSID, szCLSID, MAX_DIST_UNIT_NAME_LEN);
|
||
|
m_pcliTail->bIsDistUnit = dwIsDistUnit;
|
||
|
}
|
||
|
else
|
||
|
hr = E_OUTOFMEMORY;
|
||
|
|
||
|
return hr;
|
||
|
}
|
||
|
|
||
|
LPCLSIDLIST_ITEM OCCFindData::TakeFirstItem(void)
|
||
|
{
|
||
|
LPCLSIDLIST_ITEM pcli = m_pcliHead;
|
||
|
|
||
|
if (m_pcliHead != NULL)
|
||
|
{
|
||
|
m_pcliHead = m_pcliHead;
|
||
|
m_pcliHead = m_pcliHead->pNext;
|
||
|
if ( m_pcliHead == NULL )
|
||
|
m_pcliTail = NULL;
|
||
|
}
|
||
|
|
||
|
return pcli;
|
||
|
}
|
||
|
|
||
|
BOOL IsDUDisplayable(HKEY hkeyDU)
|
||
|
{
|
||
|
BOOL bRet = FALSE;
|
||
|
|
||
|
if (hkeyDU)
|
||
|
{
|
||
|
if (IsShowAllFilesEnabled())
|
||
|
{
|
||
|
bRet = TRUE;
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
DWORD dwType = 0, dwSystem = 0, dwSize = sizeof(dwSystem);
|
||
|
long lResult = RegQueryValueEx(hkeyDU, VALUE_SYSTEM, NULL, &dwType, (LPBYTE)&dwSystem, &dwSize);
|
||
|
bRet = (lResult == ERROR_SUCCESS && dwSystem == TRUE) ? (FALSE) : (TRUE);
|
||
|
}
|
||
|
}
|
||
|
return bRet;
|
||
|
}
|
||
|
|
||
|
BOOL IsShowAllFilesEnabled()
|
||
|
{
|
||
|
HKEY hkey = 0;
|
||
|
BOOL bRet = FALSE;
|
||
|
DWORD dwShowAll = 0;
|
||
|
|
||
|
DWORD lResult = RegOpenKeyEx(HKEY_LOCAL_MACHINE, REGSTR_PATH_DIST_UNITS, 0, KEY_READ, &hkey);
|
||
|
if (lResult == ERROR_SUCCESS)
|
||
|
{
|
||
|
DWORD dwType, dwSize = sizeof(dwShowAll);
|
||
|
lResult = RegQueryValueEx(hkey, REGSTR_SHOW_ALL_FILES, NULL, &dwType, (LPBYTE)&dwShowAll, &dwSize);
|
||
|
if (lResult == ERROR_SUCCESS)
|
||
|
{
|
||
|
bRet = (dwShowAll != 0);
|
||
|
}
|
||
|
RegCloseKey(hkey);
|
||
|
}
|
||
|
|
||
|
return bRet;
|
||
|
}
|
||
|
|
||
|
void ToggleShowAllFiles()
|
||
|
{
|
||
|
DWORD dwShowAll = !IsShowAllFilesEnabled();
|
||
|
HKEY hkey = 0;
|
||
|
DWORD lResult = RegOpenKeyEx(HKEY_LOCAL_MACHINE, REGSTR_PATH_DIST_UNITS, 0, KEY_ALL_ACCESS, &hkey);
|
||
|
|
||
|
if (lResult == ERROR_SUCCESS)
|
||
|
{
|
||
|
RegSetValueEx(hkey, REGSTR_SHOW_ALL_FILES, 0, REG_DWORD, (CONST BYTE *)&dwShowAll, sizeof(dwShowAll));
|
||
|
RegCloseKey(hkey);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
LONG WINAPI FindFirstControl(HANDLE& hFindHandle, HANDLE& hControlHandle, LPCTSTR pszCachePath)
|
||
|
{
|
||
|
LONG lResult = ERROR_SUCCESS;
|
||
|
HRESULT hr = S_OK;
|
||
|
DWORD dw = 0;
|
||
|
HKEY hKeyClass = NULL;
|
||
|
HKEY hKeyMod = NULL;
|
||
|
HKEY hKeyDist = NULL;
|
||
|
TCHAR szT[MAX_PATH]; // scratch buffer
|
||
|
int cEnum = 0;
|
||
|
CCacheItem *pci = NULL;
|
||
|
LPCLSIDLIST_ITEM pcli = NULL;
|
||
|
TCHAR szDUName[MAX_DIST_UNIT_NAME_LEN];
|
||
|
|
||
|
OCCFindData *poccfd = new OCCFindData();
|
||
|
if ( poccfd == NULL )
|
||
|
{
|
||
|
lResult = ERROR_NOT_ENOUGH_MEMORY;
|
||
|
goto EXIT_FINDFIRSTCONTROL;
|
||
|
}
|
||
|
|
||
|
// Open up the HKCR\CLSID key.
|
||
|
lResult = RegOpenKeyEx(HKEY_CLASSES_ROOT, HKCR_CLSID, 0, KEY_READ, &hKeyClass);
|
||
|
if (ERROR_SUCCESS != lResult)
|
||
|
goto EXIT_FINDFIRSTCONTROL;
|
||
|
|
||
|
// Search for legacy controls found in the COM branch
|
||
|
lResult = RegOpenKeyEx(HKEY_LOCAL_MACHINE, REGSTR_PATH_MODULE_USAGE, 0, KEY_READ, &hKeyMod);
|
||
|
if (ERROR_SUCCESS != lResult)
|
||
|
goto EXIT_FINDFIRSTCONTROL;
|
||
|
|
||
|
// Enumerate the known modules and build up a list of the owners.
|
||
|
// This is a search for legacy controls.
|
||
|
while ((lResult = RegEnumKey(hKeyMod, cEnum++, szT, ARRAYSIZE(szT))) == ERROR_SUCCESS)
|
||
|
{
|
||
|
TCHAR szClient[MAX_CLIENT_LEN];
|
||
|
HKEY hKeyClsid = NULL;
|
||
|
HKEY hkeyMUEntry = NULL;
|
||
|
|
||
|
lResult = RegOpenKeyEx( hKeyMod, szT, 0, KEY_READ, &hkeyMUEntry );
|
||
|
if (ERROR_SUCCESS != lResult)
|
||
|
continue;
|
||
|
|
||
|
// Fetch the module owner.
|
||
|
// If the module owner is in the COM branch AND
|
||
|
// ( the owner lives in the cache OR it has an INF in the cache )
|
||
|
// Then add the _owner_ to our list of legacy controls.
|
||
|
// In the INF case, we may be looking at a control that was re-registered
|
||
|
// outside of the cache.
|
||
|
// If it doesn't have these properties, then it is either a DU module or
|
||
|
// was installed by something other than MSICD. In either case, we'll skip it
|
||
|
// at least for now.
|
||
|
dw = sizeof(szClient);
|
||
|
lResult = RegQueryValueEx(hkeyMUEntry, VALUE_OWNER, NULL, NULL, (LPBYTE)szClient, &dw);
|
||
|
if (ERROR_SUCCESS != lResult)
|
||
|
continue;
|
||
|
|
||
|
lResult = RegOpenKeyEx(hKeyClass, szClient, 0, KEY_READ, &hKeyClsid);
|
||
|
if (ERROR_SUCCESS == lResult)
|
||
|
{
|
||
|
TCHAR szCLocation[MAX_PATH]; // Canonical path of control
|
||
|
TCHAR szLocation[MAX_PATH]; // Location in COM CLSID reg tree.
|
||
|
|
||
|
// Look for InprocServer[32] or LocalServer[32] key
|
||
|
dw = sizeof(szLocation);
|
||
|
lResult = RegQueryValue(hKeyClsid, INPROCSERVER32, szLocation, (PLONG)&dw);
|
||
|
if (lResult != ERROR_SUCCESS)
|
||
|
{
|
||
|
dw = sizeof(szLocation);
|
||
|
lResult = RegQueryValue(hKeyClsid, LOCALSERVER32, szLocation, (PLONG)&dw);
|
||
|
}
|
||
|
|
||
|
RegCloseKey(hKeyClsid);
|
||
|
hKeyClsid = NULL;
|
||
|
|
||
|
if ( lResult == ERROR_SUCCESS )
|
||
|
{
|
||
|
BOOL bAddOwner;
|
||
|
|
||
|
// see if we've already got an entry for this one.
|
||
|
for ( pcli = poccfd->m_pcliHead;
|
||
|
pcli != NULL && lstrcmp( szClient, pcli->szCLSID ) != 0;
|
||
|
pcli = pcli->pNext );
|
||
|
|
||
|
if ( pcli == NULL ) // not found - possibly add new item
|
||
|
{
|
||
|
// Canonicalize the path for use in comparisons with cache dirs
|
||
|
if ( OCCGetLongPathName(szCLocation, szLocation, MAX_PATH) == 0 )
|
||
|
lstrcpyn( szCLocation, szLocation, MAX_PATH );
|
||
|
|
||
|
// Is the owner in our cache?
|
||
|
bAddOwner = poccfd->IsCachePath( szCLocation );
|
||
|
|
||
|
if ( !bAddOwner )
|
||
|
{
|
||
|
// does it have an INF in our cache(s)?
|
||
|
// We'll appropriate szDCachePath
|
||
|
for ( int i = 0; i < cCachePathsMax && !bAddOwner; i++ )
|
||
|
{
|
||
|
if ( poccfd->m_aCachePath[i].m_sz != '\0' )
|
||
|
{
|
||
|
CatPathStrN( szT, poccfd->m_aCachePath[i].m_sz, PathFindFileName( szCLocation ), MAX_PATH);
|
||
|
|
||
|
// Note if another copy of the owner exists within the cache(s).
|
||
|
// This would be a case of re-registration.
|
||
|
if ( PathFileExists( szT ) )
|
||
|
{
|
||
|
// add our version of the control.
|
||
|
lstrcpyn( szCLocation, szT, MAX_PATH );
|
||
|
bAddOwner = TRUE;
|
||
|
}
|
||
|
else
|
||
|
bAddOwner = PathRenameExtension( szT, INF_EXTENSION ) &&
|
||
|
PathFileExists( szT );
|
||
|
} // if cache path
|
||
|
} // for each cache directory
|
||
|
} // if check for cached INF
|
||
|
|
||
|
if ( bAddOwner )
|
||
|
{
|
||
|
HKEY hkeyDUCheck = 0;
|
||
|
char achBuf[MAX_REGPATH_LEN];
|
||
|
|
||
|
wnsprintfA(achBuf, MAX_REGPATH_LEN, "%s\\%s", REGSTR_PATH_DIST_UNITS, szClient);
|
||
|
|
||
|
lResult = RegOpenKeyEx(HKEY_LOCAL_MACHINE, achBuf, 0, KEY_READ, &hkeyDUCheck);
|
||
|
|
||
|
if (lResult != ERROR_SUCCESS)
|
||
|
{
|
||
|
// This is a legacy control with no corresponding DU
|
||
|
poccfd->AddListItem( szCLocation, szClient, FALSE );
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
if (IsDUDisplayable(hkeyDUCheck))
|
||
|
{
|
||
|
// Legacy control w/ DU keys that is displayable
|
||
|
poccfd->AddListItem( szCLocation, szClient, FALSE );
|
||
|
}
|
||
|
RegCloseKey(hkeyDUCheck);
|
||
|
}
|
||
|
}
|
||
|
} // if owner we haven't seen before
|
||
|
} // if owner has local or inproc server
|
||
|
} // if owner has COM entry
|
||
|
RegCloseKey( hkeyMUEntry );
|
||
|
} // while enumerating Module Usage
|
||
|
|
||
|
// we're finished with module usage
|
||
|
RegCloseKey(hKeyMod);
|
||
|
|
||
|
// Now search distribution units
|
||
|
|
||
|
// Check for duplicates - distribution units for controls we detected above
|
||
|
|
||
|
lResult = RegOpenKeyEx(HKEY_LOCAL_MACHINE, REGSTR_PATH_DIST_UNITS, 0, KEY_READ, &hKeyDist);
|
||
|
if (lResult == ERROR_SUCCESS)
|
||
|
{
|
||
|
cEnum = 0;
|
||
|
// Enumerate distribution units and queue them up in the list
|
||
|
while ((lResult = RegEnumKey(hKeyDist, cEnum++, szDUName, ARRAYSIZE(szDUName))) == ERROR_SUCCESS)
|
||
|
{
|
||
|
// We should only display DU's installed by code download.
|
||
|
HKEY hkeyDU;
|
||
|
DWORD dwType;
|
||
|
|
||
|
lResult = RegOpenKeyEx( hKeyDist, szDUName, 0, KEY_READ, &hkeyDU );
|
||
|
Assert( lResult == ERROR_SUCCESS );
|
||
|
|
||
|
if ((ERROR_SUCCESS != lResult) ||
|
||
|
!IsDUDisplayable(hkeyDU))
|
||
|
{
|
||
|
continue;
|
||
|
}
|
||
|
|
||
|
szT[0] = '\0';
|
||
|
DWORD cb = sizeof(szT);
|
||
|
lResult = RegQueryValueEx( hkeyDU, DU_INSTALLER_VALUE, NULL, &dwType, (LPBYTE)szT, &cb );
|
||
|
|
||
|
Assert( lResult == ERROR_SUCCESS ); // properly-formed DU will have this
|
||
|
Assert( dwType == REG_SZ ); // properly-formed DU's have a string here
|
||
|
|
||
|
// Check for an installed version. We might just have a DU that has an AvailableVersion
|
||
|
// but hasn't been installed yet.
|
||
|
lResult = RegQueryValue( hkeyDU, REGSTR_INSTALLED_VERSION, NULL, NULL );
|
||
|
|
||
|
RegCloseKey( hkeyDU );
|
||
|
|
||
|
if ( lstrcmpi( szT, CDL_INSTALLER ) == 0 &&
|
||
|
lResult == ERROR_SUCCESS ) // from InstalledVersion RegQueryValue
|
||
|
{
|
||
|
// If we can convert the unique name to a GUID, then this DU
|
||
|
// may have already been added on the first pass through the
|
||
|
// COM branch.
|
||
|
CLSID clsidDummy = CLSID_NULL;
|
||
|
WORD szDummyStr[MAX_CTRL_NAME_SIZE];
|
||
|
BOOL bFoundDuplicate = FALSE;
|
||
|
|
||
|
MultiByteToWideChar(CP_ACP, 0, szDUName, -1, szDummyStr, ARRAYSIZE(szDummyStr));
|
||
|
if ((CLSIDFromString(szDummyStr, &clsidDummy) == S_OK))
|
||
|
{
|
||
|
for (pcli = poccfd->m_pcliHead; pcli; pcli = pcli->pNext)
|
||
|
{
|
||
|
if (!lstrcmpi(szDUName, pcli->szCLSID))
|
||
|
{
|
||
|
// Duplicate found. Use dist unit information to
|
||
|
// fill in additional fields if it is the first
|
||
|
// entry in the list
|
||
|
bFoundDuplicate = TRUE;
|
||
|
pcli->bIsDistUnit = TRUE;
|
||
|
break;
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if (!bFoundDuplicate)
|
||
|
{
|
||
|
// Okay we're looking at some sort of Java scenario. We have a distribution unit, but
|
||
|
// no corresponding entry in the COM branch. This generally means we've got a DU that
|
||
|
// consists of java packages. It can also mean that we're dealing with a java/code download
|
||
|
// backdoor introduced in IE3. In this case, an Object tag gets a CAB downloaded that
|
||
|
// installs Java classes and sets of a CLSID that invokes MSJava.dll on the class ( ESPN's
|
||
|
// sportszone control/applet works this way ). In the first case, we get the name
|
||
|
// squared-away when we parse the DU. In the latter case, we need to try and pick the name
|
||
|
// up from the COM branch.
|
||
|
hr = poccfd->AddListItem( "", szDUName, TRUE );
|
||
|
if ( FAILED(hr) )
|
||
|
{
|
||
|
lResult = ERROR_NOT_ENOUGH_MEMORY;
|
||
|
goto EXIT_FINDFIRSTCONTROL;
|
||
|
}
|
||
|
} // if no duplicate - add DU to the list
|
||
|
} // if installed by MSICD
|
||
|
} // while enumerating DU's
|
||
|
} // if we can open the DU key.
|
||
|
else
|
||
|
lResult = ERROR_NO_MORE_ITEMS; // if no DU's then make due with our legacy controls, if any
|
||
|
|
||
|
pcli = poccfd->TakeFirstItem();
|
||
|
if (pcli)
|
||
|
{
|
||
|
hr = MakeCacheItemFromControlList(hKeyClass, hKeyDist, pcli, &pci);
|
||
|
delete pcli;
|
||
|
if ( FAILED(hr) )
|
||
|
lResult = hr;
|
||
|
}
|
||
|
|
||
|
if (hKeyDist)
|
||
|
{
|
||
|
RegCloseKey(hKeyDist);
|
||
|
hKeyDist = 0;
|
||
|
}
|
||
|
|
||
|
|
||
|
// Clean up
|
||
|
|
||
|
if (lResult != ERROR_NO_MORE_ITEMS)
|
||
|
goto EXIT_FINDFIRSTCONTROL;
|
||
|
|
||
|
if (pci == NULL)
|
||
|
lResult = ERROR_NO_MORE_ITEMS;
|
||
|
else
|
||
|
{
|
||
|
lResult = ERROR_SUCCESS;
|
||
|
}
|
||
|
|
||
|
hFindHandle = (HANDLE)poccfd;
|
||
|
hControlHandle = (HANDLE)pci;
|
||
|
|
||
|
EXIT_FINDFIRSTCONTROL:
|
||
|
|
||
|
if (hKeyDist)
|
||
|
RegCloseKey(hKeyDist);
|
||
|
|
||
|
if (hKeyClass)
|
||
|
RegCloseKey(hKeyClass);
|
||
|
|
||
|
if (lResult != ERROR_SUCCESS)
|
||
|
{
|
||
|
if ( pci != NULL )
|
||
|
delete pci;
|
||
|
if ( poccfd != NULL )
|
||
|
delete poccfd;
|
||
|
hFindHandle = INVALID_HANDLE_VALUE;
|
||
|
hControlHandle = INVALID_HANDLE_VALUE;
|
||
|
}
|
||
|
|
||
|
return lResult;
|
||
|
}
|
||
|
|
||
|
LONG WINAPI FindNextControl(HANDLE& hFindHandle, HANDLE& hControlHandle)
|
||
|
{
|
||
|
LONG lResult = ERROR_SUCCESS;
|
||
|
HRESULT hr = S_OK;
|
||
|
HKEY hKeyClass = NULL;
|
||
|
|
||
|
CCacheItem *pci = NULL;
|
||
|
OCCFindData *poccfd = (OCCFindData *)hFindHandle;
|
||
|
|
||
|
LPCLSIDLIST_ITEM pcli = poccfd->TakeFirstItem();
|
||
|
hControlHandle = INVALID_HANDLE_VALUE;
|
||
|
|
||
|
if (pcli == NULL)
|
||
|
{
|
||
|
lResult = ERROR_NO_MORE_ITEMS;
|
||
|
goto EXIT_FINDNEXTCONTROL;
|
||
|
}
|
||
|
|
||
|
if ((lResult = RegOpenKeyEx(HKEY_CLASSES_ROOT, HKCR_CLSID, 0, KEY_READ, &hKeyClass)) != ERROR_SUCCESS)
|
||
|
goto EXIT_FINDNEXTCONTROL;
|
||
|
|
||
|
if ( pcli->bIsDistUnit )
|
||
|
{
|
||
|
HKEY hKeyDist;
|
||
|
|
||
|
lResult = RegOpenKeyEx( HKEY_LOCAL_MACHINE, REGSTR_PATH_DIST_UNITS, 0,
|
||
|
KEY_READ, &hKeyDist);
|
||
|
|
||
|
if ( lResult == ERROR_SUCCESS )
|
||
|
{
|
||
|
hr = MakeCacheItemFromControlList( hKeyClass,
|
||
|
hKeyDist,
|
||
|
pcli,
|
||
|
&pci );
|
||
|
if ( FAILED(hr) )
|
||
|
lResult = hr;
|
||
|
|
||
|
RegCloseKey( hKeyDist );
|
||
|
}
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
// This is not a distribution unit. Fill in CCachItem information
|
||
|
// from the COM branch.
|
||
|
hr = MakeCacheItemFromControlList(hKeyClass, NULL, pcli, &pci );
|
||
|
if ( FAILED(hr) )
|
||
|
lResult = hr;
|
||
|
}
|
||
|
|
||
|
hControlHandle = (HANDLE)pci;
|
||
|
|
||
|
EXIT_FINDNEXTCONTROL:
|
||
|
|
||
|
if (hKeyClass)
|
||
|
RegCloseKey(hKeyClass);
|
||
|
|
||
|
if (pcli != NULL)
|
||
|
{
|
||
|
delete pcli;
|
||
|
}
|
||
|
|
||
|
return lResult;
|
||
|
}
|
||
|
|
||
|
void WINAPI FindControlClose(HANDLE hFindHandle)
|
||
|
{
|
||
|
if (hFindHandle == INVALID_HANDLE_VALUE ||
|
||
|
hFindHandle == (HANDLE)0)
|
||
|
return;
|
||
|
|
||
|
delete (OCCFindData*)hFindHandle;
|
||
|
}
|
||
|
|
||
|
void WINAPI ReleaseControlHandle(HANDLE hControlHandle)
|
||
|
{
|
||
|
if (hControlHandle == INVALID_HANDLE_VALUE ||
|
||
|
hControlHandle == (HANDLE)0)
|
||
|
return;
|
||
|
|
||
|
delete (CCacheItem *)hControlHandle;
|
||
|
}
|
||
|
|
||
|
HRESULT WINAPI RemoveControlByHandle(HANDLE hControlHandle, BOOL bForceRemove /* = FALSE */)
|
||
|
{
|
||
|
return RemoveControlByHandle2( hControlHandle, bForceRemove, FALSE );
|
||
|
}
|
||
|
|
||
|
|
||
|
HRESULT WINAPI RemoveControlByName(LPCTSTR lpszFile, LPCTSTR lpszCLSID, LPCTSTR lpszTypeLibID, BOOL bForceRemove, /* = FALSE */ DWORD dwIsDistUnit /* = FALSE */)
|
||
|
{
|
||
|
return RemoveControlByName2( lpszFile, lpszCLSID, lpszTypeLibID, bForceRemove, dwIsDistUnit, FALSE);
|
||
|
}
|
||
|
|
||
|
LONG WINAPI GetControlDependentFile(int iFile, HANDLE hControlHandle, LPTSTR lpszFile, LPDWORD lpdwSize, BOOL bToUpper /* = FALSE */)
|
||
|
{
|
||
|
CCacheItem *pci = (CCacheItem *)hControlHandle;
|
||
|
|
||
|
if (iFile < 0 || lpszFile == NULL || lpdwSize == NULL)
|
||
|
return ERROR_BAD_ARGUMENTS;
|
||
|
|
||
|
// loop through the list of files to find the one indicated
|
||
|
// by the given index.
|
||
|
// this way is dumb but since a control does not depend on
|
||
|
// too many files, it's ok
|
||
|
CFileNode *pFileNode = pci->GetFirstFile();
|
||
|
for (int i = 0; i < iFile && pFileNode != NULL; i++)
|
||
|
pFileNode = pci->GetNextFile();
|
||
|
|
||
|
if (pFileNode == NULL)
|
||
|
{
|
||
|
lpszFile[0] = TEXT('\0');
|
||
|
lpdwSize = 0;
|
||
|
return ERROR_NO_MORE_FILES;
|
||
|
}
|
||
|
|
||
|
// Make a fully qualified filename
|
||
|
if (pFileNode->GetPath() != NULL)
|
||
|
{
|
||
|
CatPathStrN( lpszFile, pFileNode->GetPath(), pFileNode->GetName(), MAX_PATH);
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
lstrcpy(lpszFile, pFileNode->GetName());
|
||
|
}
|
||
|
|
||
|
if (FAILED(GetSizeOfFile(lpszFile, lpdwSize)))
|
||
|
*lpdwSize = 0;
|
||
|
|
||
|
// to upper case if required
|
||
|
if (bToUpper)
|
||
|
CharUpper(lpszFile);
|
||
|
|
||
|
return ERROR_SUCCESS;
|
||
|
}
|
||
|
|
||
|
// determine if a control or one of its associated files can be removed
|
||
|
// by reading its SharedDlls count
|
||
|
BOOL WINAPI IsModuleRemovable(LPCTSTR lpszFile)
|
||
|
{
|
||
|
TCHAR szFile[MAX_PATH];
|
||
|
TCHAR szT[MAX_PATH];
|
||
|
|
||
|
if (lpszFile == NULL)
|
||
|
return FALSE;
|
||
|
|
||
|
if ( OCCGetLongPathName(szFile, lpszFile, MAX_PATH) == 0 )
|
||
|
lstrcpyn( szFile, lpszFile, MAX_PATH );
|
||
|
|
||
|
// Don't ever pull something out of the system directory.
|
||
|
// This is a "safe" course of action because it is not reasonable
|
||
|
// to expect the user to judge whether yanking this file damage other
|
||
|
// software installations or the system itself.
|
||
|
GetSystemDirectory(szT, MAX_PATH);
|
||
|
if (StrStrI(szFile, szT))
|
||
|
return FALSE;
|
||
|
|
||
|
// check moduleusage if a control is safe to remove
|
||
|
if (LookUpModuleUsage(szFile, NULL, szT, MAX_PATH) != S_OK)
|
||
|
return FALSE;
|
||
|
|
||
|
// if we don't know who the owner of the module is, it's not
|
||
|
// safe to remove
|
||
|
if (lstrcmpi(szT, UNKNOWNOWNER) == 0)
|
||
|
return FALSE;
|
||
|
else
|
||
|
{
|
||
|
// check shareddlls if a control is safe to remove
|
||
|
LONG cRef;
|
||
|
|
||
|
HRESULT hr = SetSharedDllsCount( szFile, -1, &cRef );
|
||
|
|
||
|
return cRef == 1;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
BOOL WINAPI GetControlInfo(HANDLE hControlHandle, UINT nFlag,
|
||
|
DWORD *pdwData, LPTSTR pszData, int nBufLen)
|
||
|
{
|
||
|
if (hControlHandle == 0 || hControlHandle == INVALID_HANDLE_VALUE)
|
||
|
return FALSE;
|
||
|
|
||
|
BOOL bResult = TRUE;
|
||
|
LPCTSTR pStr = NULL;
|
||
|
DWORD dw = 0;
|
||
|
|
||
|
switch (nFlag)
|
||
|
{
|
||
|
case GCI_NAME: // get friend name of control
|
||
|
pStr = ((CCacheItem *)hControlHandle)->m_szName;
|
||
|
break;
|
||
|
|
||
|
case GCI_FILE: // get filename of control (with full path)
|
||
|
pStr = ((CCacheItem *)hControlHandle)->m_szFile;
|
||
|
// if there is no file, but there is a package list, fake it
|
||
|
// with the path to the first package's ZIP file.
|
||
|
if ( *pStr == '\0' )
|
||
|
{
|
||
|
CPNode *ppn = ((CCacheItem *)hControlHandle)->GetFirstPackage();
|
||
|
if (ppn)
|
||
|
{
|
||
|
pStr = ppn->GetPath();
|
||
|
if (!pStr)
|
||
|
{
|
||
|
return FALSE; // this means hControlHandle is an invalid arg
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if ( pStr && *pStr == TEXT('\0') )
|
||
|
{
|
||
|
CPNode *pfn = ((CCacheItem *)hControlHandle)->GetFirstFile();
|
||
|
if ( pfn != NULL )
|
||
|
pStr = pfn->GetPath();
|
||
|
}
|
||
|
break;
|
||
|
|
||
|
case GCI_DIST_UNIT_VERSION:
|
||
|
pStr = ((CCacheItem *)hControlHandle)->m_szVersion;
|
||
|
break;
|
||
|
|
||
|
case GCI_CLSID: // get CLSID of control
|
||
|
pStr = ((CCacheItem *)hControlHandle)->m_szCLSID;
|
||
|
break;
|
||
|
|
||
|
case GCI_TYPELIBID: // get TYPELIB id of control
|
||
|
pStr = ((CCacheItem *)hControlHandle)->m_szTypeLibID;
|
||
|
break;
|
||
|
|
||
|
case GCI_TOTALSIZE: // get total size in bytes
|
||
|
dw = ((CCacheItem *)hControlHandle)->GetTotalFileSize();
|
||
|
break;
|
||
|
|
||
|
case GCI_SIZESAVED: // get total size restored if control is removed
|
||
|
dw = ((CCacheItem *)hControlHandle)->GetTotalSizeSaved();
|
||
|
break;
|
||
|
|
||
|
case GCI_TOTALFILES: // get total number of files related to control
|
||
|
dw = (DWORD)(((CCacheItem *)hControlHandle)->GetTotalFiles());
|
||
|
break;
|
||
|
|
||
|
case GCI_CODEBASE: // get CodeBase for control
|
||
|
pStr = ((CCacheItem *)hControlHandle)->m_szCodeBase;
|
||
|
break;
|
||
|
|
||
|
case GCI_ISDISTUNIT:
|
||
|
dw = ((CCacheItem *)hControlHandle)->ItemType() == CCacheDistUnit::s_dwType;
|
||
|
break;
|
||
|
|
||
|
case GCI_STATUS:
|
||
|
dw = ((CCacheItem *)hControlHandle)->GetStatus();
|
||
|
break;
|
||
|
|
||
|
case GCI_HAS_ACTIVEX:
|
||
|
dw = ((CCacheItem *)hControlHandle)->GetHasActiveX();
|
||
|
break;
|
||
|
|
||
|
case GCI_HAS_JAVA:
|
||
|
dw = ((CCacheItem *)hControlHandle)->GetHasJava();
|
||
|
break;
|
||
|
}
|
||
|
|
||
|
if (nFlag == GCI_TOTALSIZE ||
|
||
|
nFlag == GCI_SIZESAVED ||
|
||
|
nFlag == GCI_TOTALFILES ||
|
||
|
nFlag == GCI_ISDISTUNIT ||
|
||
|
nFlag == GCI_STATUS ||
|
||
|
nFlag == GCI_HAS_ACTIVEX ||
|
||
|
nFlag == GCI_HAS_JAVA)
|
||
|
{
|
||
|
bResult = pdwData != NULL;
|
||
|
if (bResult)
|
||
|
*pdwData = dw;
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
bResult = pszData && pStr;
|
||
|
if (bResult)
|
||
|
lstrcpyn(pszData, pStr, nBufLen);
|
||
|
}
|
||
|
|
||
|
return bResult;
|
||
|
}
|
||
|
|
||
|
///////////////////////////////////////////////////////////////////////////////
|
||
|
// API to be called by Advpack.dll
|
||
|
|
||
|
// Define list node to be used in a linked list of control
|
||
|
struct tagHANDLENODE;
|
||
|
typedef struct tagHANDLENODE HANDLENODE;
|
||
|
typedef HANDLENODE* LPHANDLENODE;
|
||
|
struct tagHANDLENODE
|
||
|
{
|
||
|
HANDLE hControl;
|
||
|
struct tagHANDLENODE* pNext;
|
||
|
};
|
||
|
|
||
|
// Given a handle to a control, get the control's last access time
|
||
|
// Result is stored in a FILETIME struct
|
||
|
HRESULT GetLastAccessTime(HANDLE hControl, FILETIME *pLastAccess)
|
||
|
{
|
||
|
Assert(hControl != NULL && hControl != INVALID_HANDLE_VALUE);
|
||
|
Assert(pLastAccess != NULL);
|
||
|
|
||
|
HRESULT hr = S_OK;
|
||
|
WIN32_FIND_DATA fdata;
|
||
|
HANDLE h = INVALID_HANDLE_VALUE;
|
||
|
LPCTSTR lpszFile = NULL;
|
||
|
CCacheItem *pci = (CCacheItem *)hControl;
|
||
|
CPNode *ppn;
|
||
|
|
||
|
if (pci->m_szFile[0] != 0)
|
||
|
lpszFile = pci->m_szFile;
|
||
|
else if ( (ppn = pci->GetFirstPackage()) != NULL )
|
||
|
lpszFile = ppn->GetPath();
|
||
|
else if ( (ppn = pci->GetFirstFile()) != NULL )
|
||
|
lpszFile = ppn->GetPath();
|
||
|
|
||
|
if ( lpszFile )
|
||
|
h = FindFirstFile(lpszFile, &fdata);
|
||
|
|
||
|
if (h == INVALID_HANDLE_VALUE)
|
||
|
{
|
||
|
SYSTEMTIME stNow;
|
||
|
GetLocalTime(&stNow);
|
||
|
SystemTimeToFileTime(&stNow, pLastAccess);
|
||
|
hr = HRESULT_FROM_WIN32(GetLastError());
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
// Convert file time to local file time, then file time to
|
||
|
// system time. Set those fields to be ignored to 0, and
|
||
|
// set system time back to file time.
|
||
|
// FILETIME struct is used because API for time comparison
|
||
|
// only works on FILETIME.
|
||
|
|
||
|
// SYSTEMTIME sysTime;
|
||
|
|
||
|
FindClose(h);
|
||
|
FileTimeToLocalFileTime(&(fdata.ftLastAccessTime), pLastAccess);
|
||
|
}
|
||
|
|
||
|
return hr;
|
||
|
}
|
||
|
|
||
|
HRESULT WINAPI SweepControlsByLastAccessDate(
|
||
|
SYSTEMTIME *pLastAccessTime /* = NULL */,
|
||
|
PFNDOBEFOREREMOVAL pfnDoBefore /* = NULL */,
|
||
|
PFNDOAFTERREMOVAL pfnDoAfter /* = NULL */,
|
||
|
DWORD dwSizeLimit /* = 0 */
|
||
|
)
|
||
|
{
|
||
|
LONG lResult = ERROR_SUCCESS;
|
||
|
HRESULT hr = S_FALSE;
|
||
|
DWORD dwSize = 0, dwTotalSize = 0;
|
||
|
HANDLE hFind = NULL, hControl = NULL;
|
||
|
LPHANDLENODE pHead = NULL, pCur = NULL;
|
||
|
FILETIME timeLastAccess, timeRemovePrior;
|
||
|
UINT cCnt = 0;
|
||
|
TCHAR szFile[MAX_PATH];
|
||
|
|
||
|
// ignore all fields except wYear, wMonth and wDay
|
||
|
if (pLastAccessTime != NULL)
|
||
|
{
|
||
|
pLastAccessTime->wDayOfWeek = 0;
|
||
|
pLastAccessTime->wHour = 0;
|
||
|
pLastAccessTime->wMinute = 0;
|
||
|
pLastAccessTime->wSecond = 0;
|
||
|
pLastAccessTime->wMilliseconds = 0;
|
||
|
}
|
||
|
|
||
|
// loop through all controls and put in a list the
|
||
|
// ones that are accessed before the given date and
|
||
|
// are safe to uninstall
|
||
|
lResult = FindFirstControl(hFind, hControl);
|
||
|
for (;lResult == ERROR_SUCCESS;
|
||
|
lResult = FindNextControl(hFind, hControl))
|
||
|
{
|
||
|
// check last access time
|
||
|
if (pLastAccessTime != NULL)
|
||
|
{
|
||
|
GetLastAccessTime(hControl, &timeLastAccess);
|
||
|
SystemTimeToFileTime(pLastAccessTime, &timeRemovePrior);
|
||
|
if (CompareFileTime(&timeLastAccess, &timeRemovePrior) > 0)
|
||
|
{
|
||
|
ReleaseControlHandle(hControl);
|
||
|
continue;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// check if control is safe to remove
|
||
|
GetControlInfo(hControl, GCI_FILE, NULL, szFile, MAX_PATH);
|
||
|
if (!IsModuleRemovable(szFile))
|
||
|
{
|
||
|
ReleaseControlHandle(hControl);
|
||
|
continue;
|
||
|
}
|
||
|
|
||
|
// put control in a list
|
||
|
if (pHead == NULL)
|
||
|
{
|
||
|
pHead = new HANDLENODE;
|
||
|
pCur = pHead;
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
pCur->pNext = new HANDLENODE;
|
||
|
pCur = pCur->pNext;
|
||
|
}
|
||
|
|
||
|
if (pCur == NULL)
|
||
|
{
|
||
|
hr = E_OUTOFMEMORY;
|
||
|
goto EXIT_REMOVECONTROLBYLASTACCESSDATE;
|
||
|
}
|
||
|
|
||
|
pCur->pNext = NULL;
|
||
|
pCur->hControl = hControl;
|
||
|
cCnt += 1;
|
||
|
|
||
|
// calculate total size
|
||
|
GetControlInfo(pCur->hControl, GCI_SIZESAVED, &dwSize, NULL, NULL);
|
||
|
dwTotalSize += dwSize;
|
||
|
}
|
||
|
|
||
|
// quit if total size restored is less than the given amount
|
||
|
if (dwTotalSize < dwSizeLimit)
|
||
|
goto EXIT_REMOVECONTROLBYLASTACCESSDATE;
|
||
|
|
||
|
// traverse the list and remove each control
|
||
|
for (pCur = pHead; pCur != NULL; cCnt--)
|
||
|
{
|
||
|
hr = S_OK;
|
||
|
pHead = pHead->pNext;
|
||
|
|
||
|
// call callback function before removing a control
|
||
|
if (pfnDoBefore == NULL || SUCCEEDED(pfnDoBefore(pCur->hControl, cCnt)))
|
||
|
{
|
||
|
hr = RemoveControlByHandle(pCur->hControl);
|
||
|
|
||
|
// call callback function after removing a control, passing it the
|
||
|
// result of the removal
|
||
|
if (pfnDoAfter != NULL && FAILED(pfnDoAfter(hr, cCnt - 1)))
|
||
|
{
|
||
|
pHead = pCur; // set pHead back to head of list
|
||
|
goto EXIT_REMOVECONTROLBYLASTACCESSDATE;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// release memory used by the control handle
|
||
|
ReleaseControlHandle(pCur->hControl);
|
||
|
delete pCur;
|
||
|
pCur = pHead;
|
||
|
}
|
||
|
|
||
|
EXIT_REMOVECONTROLBYLASTACCESSDATE:
|
||
|
|
||
|
FindControlClose(hFind);
|
||
|
|
||
|
// release memory taken up by the list
|
||
|
for (pCur = pHead; pCur != NULL; pCur = pHead)
|
||
|
{
|
||
|
pHead = pHead->pNext;
|
||
|
ReleaseControlHandle(pCur->hControl);
|
||
|
delete pCur;
|
||
|
}
|
||
|
|
||
|
return hr;
|
||
|
}
|
||
|
|
||
|
HRESULT WINAPI RemoveExpiredControls(DWORD dwFlags, DWORD dwReserved)
|
||
|
{
|
||
|
LONG lResult = ERROR_SUCCESS;
|
||
|
HRESULT hr = S_FALSE;
|
||
|
HANDLE hFind = NULL, hControl = NULL;
|
||
|
LPHANDLENODE pHead = NULL, pCur = NULL;
|
||
|
FILETIME ftNow, ftMinLastAccess, ftLastAccess;
|
||
|
LARGE_INTEGER liMinLastAccess;
|
||
|
SYSTEMTIME stNow;
|
||
|
UINT cCnt = 0;
|
||
|
|
||
|
GetLocalTime( &stNow );
|
||
|
SystemTimeToFileTime(&stNow, &ftNow);
|
||
|
|
||
|
// loop through all controls and put in a list the
|
||
|
// ones that are accessed before the given date and
|
||
|
// are safe to uninstall
|
||
|
lResult = FindFirstControl(hFind, hControl);
|
||
|
for (;lResult == ERROR_SUCCESS;
|
||
|
lResult = FindNextControl(hFind, hControl))
|
||
|
{
|
||
|
CCacheItem *pci = (CCacheItem *)hControl;
|
||
|
|
||
|
// Controls must have a last access time of at least ftMinLastAccess or they will
|
||
|
// expire by default. If they have the Office Auto-expire set, then they may
|
||
|
// have to pass a higher bar.
|
||
|
|
||
|
liMinLastAccess.LowPart = ftNow.dwLowDateTime;
|
||
|
liMinLastAccess.HighPart = ftNow.dwHighDateTime;
|
||
|
// We add one to GetExpireDays to deal with bug 17151. The last access time
|
||
|
// returned by the file system is truncated down to 12AM, so we need to
|
||
|
// expand the expire interval to ensure that this truncation does not cause
|
||
|
// the control to expire prematurely.
|
||
|
liMinLastAccess.QuadPart -= ((pci->GetExpireDays()+1) * 864000000000L); //24*3600*10^7
|
||
|
ftMinLastAccess.dwLowDateTime = liMinLastAccess.LowPart;
|
||
|
ftMinLastAccess.dwHighDateTime = liMinLastAccess.HighPart;
|
||
|
|
||
|
GetLastAccessTime(hControl, &ftLastAccess); // ftLastAccess is a local file time
|
||
|
|
||
|
if (CompareFileTime(&ftLastAccess, &ftMinLastAccess) >= 0)
|
||
|
{
|
||
|
ReleaseControlHandle(hControl);
|
||
|
continue;
|
||
|
}
|
||
|
|
||
|
|
||
|
// put control in a list
|
||
|
if (pHead == NULL)
|
||
|
{
|
||
|
pHead = new HANDLENODE;
|
||
|
pCur = pHead;
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
pCur->pNext = new HANDLENODE;
|
||
|
pCur = pCur->pNext;
|
||
|
}
|
||
|
|
||
|
if (pCur == NULL)
|
||
|
{
|
||
|
hr = E_OUTOFMEMORY;
|
||
|
goto cleanup;
|
||
|
}
|
||
|
|
||
|
pCur->pNext = NULL;
|
||
|
pCur->hControl = hControl;
|
||
|
cCnt += 1;
|
||
|
}
|
||
|
|
||
|
// traverse the list and remove each control
|
||
|
for (pCur = pHead; pCur != NULL; cCnt--)
|
||
|
{
|
||
|
hr = S_OK;
|
||
|
pHead = pHead->pNext;
|
||
|
|
||
|
hr = RemoveControlByHandle2(pCur->hControl, FALSE, TRUE);
|
||
|
|
||
|
// release memory used by the control handle
|
||
|
ReleaseControlHandle(pCur->hControl);
|
||
|
delete pCur;
|
||
|
pCur = pHead;
|
||
|
}
|
||
|
|
||
|
cleanup:
|
||
|
|
||
|
FindControlClose(hFind);
|
||
|
|
||
|
// release memory taken up by the list, if any left
|
||
|
for (pCur = pHead; pCur != NULL; pCur = pHead)
|
||
|
{
|
||
|
pHead = pHead->pNext;
|
||
|
ReleaseControlHandle(pCur->hControl);
|
||
|
delete pCur;
|
||
|
}
|
||
|
|
||
|
return hr;
|
||
|
}
|
||
|
|