INTEGRATION: CWS native36 (1.7.4); FILE MERGED
2006/02/20 11:31:13 is 1.7.4.3: RESYNC: (1.7-1.8); FILE MERGED 2006/02/20 10:30:01 is 1.7.4.2: #131293# correct text encoding 2006/02/16 14:25:06 is 1.7.4.1: #131293# new CustomAction to set feature states in patch
This commit is contained in:
Rüdiger Timm
parent
8caf6d961b
commit
2521869690
1 changed files with 245 additions and 26 deletions
|
|
@ -4,9 +4,9 @@
|
||||||
*
|
*
|
||||||
* $RCSfile: swappatchfiles.cxx,v $
|
* $RCSfile: swappatchfiles.cxx,v $
|
||||||
*
|
*
|
||||||
* $Revision: 1.8 $
|
* $Revision: 1.9 $
|
||||||
*
|
*
|
||||||
* last change: $Author: hr $ $Date: 2006-02-17 13:28:18 $
|
* last change: $Author: rt $ $Date: 2006-03-06 14:03:30 $
|
||||||
*
|
*
|
||||||
* The Contents of this file are made available subject to
|
* The Contents of this file are made available subject to
|
||||||
* the terms of GNU Lesser General Public License Version 2.1.
|
* the terms of GNU Lesser General Public License Version 2.1.
|
||||||
|
|
@ -37,6 +37,7 @@
|
||||||
#include <windows.h>
|
#include <windows.h>
|
||||||
#include <msiquery.h>
|
#include <msiquery.h>
|
||||||
#include <malloc.h>
|
#include <malloc.h>
|
||||||
|
#include <assert.h>
|
||||||
|
|
||||||
#ifdef UNICODE
|
#ifdef UNICODE
|
||||||
#define _UNICODE
|
#define _UNICODE
|
||||||
|
|
@ -54,11 +55,137 @@
|
||||||
|
|
||||||
#include <systools/win32/uwinapi.h>
|
#include <systools/win32/uwinapi.h>
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#define WININIT_FILENAME "wininit.ini"
|
#define WININIT_FILENAME "wininit.ini"
|
||||||
#define RENAME_SECTION "rename"
|
#define RENAME_SECTION "rename"
|
||||||
|
|
||||||
|
|
||||||
|
static std::_tstring GetMsiProperty( MSIHANDLE handle, const std::_tstring& sProperty )
|
||||||
|
{
|
||||||
|
std::_tstring result;
|
||||||
|
TCHAR szDummy[1] = TEXT("");
|
||||||
|
DWORD nChars = 0;
|
||||||
|
|
||||||
|
if ( MsiGetProperty( handle, sProperty.c_str(), szDummy, &nChars ) == ERROR_MORE_DATA )
|
||||||
|
{
|
||||||
|
DWORD nBytes = ++nChars * sizeof(TCHAR);
|
||||||
|
LPTSTR buffer = reinterpret_cast<LPTSTR>(_alloca(nBytes));
|
||||||
|
ZeroMemory( buffer, nBytes );
|
||||||
|
MsiGetProperty(handle, sProperty.c_str(), buffer, &nChars);
|
||||||
|
result = buffer;
|
||||||
|
}
|
||||||
|
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
|
||||||
|
// The provided GUID must be without surounding '{}'
|
||||||
|
static std::_tstring GetGuidPart(const std::_tstring& guid, int index)
|
||||||
|
{
|
||||||
|
assert((guid.length() == 36) && "No GUID or wrong format!");
|
||||||
|
assert(((index > -1) && (index < 5)) && "Out of range!");
|
||||||
|
|
||||||
|
if (index == 0) return std::_tstring(guid.c_str(), 8);
|
||||||
|
if (index == 1) return std::_tstring(guid.c_str() + 9, 4);
|
||||||
|
if (index == 2) return std::_tstring(guid.c_str() + 14, 4);
|
||||||
|
if (index == 3) return std::_tstring(guid.c_str() + 19, 4);
|
||||||
|
if (index == 4) return std::_tstring(guid.c_str() + 24, 12);
|
||||||
|
|
||||||
|
return std::_tstring();
|
||||||
|
}
|
||||||
|
|
||||||
|
static void Swap(char* p1, char* p2)
|
||||||
|
{
|
||||||
|
char tmp = *p1;
|
||||||
|
*p1 = *p2;
|
||||||
|
*p2 = tmp;
|
||||||
|
}
|
||||||
|
|
||||||
|
static std::_tstring Invert(const std::_tstring& str)
|
||||||
|
{
|
||||||
|
char* buff = reinterpret_cast<char*>(_alloca(str.length()));
|
||||||
|
strncpy(buff, str.c_str(), str.length());
|
||||||
|
|
||||||
|
char* front = buff;
|
||||||
|
char* back = buff + str.length() - 1;
|
||||||
|
|
||||||
|
while (front < back)
|
||||||
|
Swap(front++, back--);
|
||||||
|
|
||||||
|
return std::_tstring(buff, str.length());
|
||||||
|
}
|
||||||
|
|
||||||
|
// Convert the upgrade code (which is a GUID) according
|
||||||
|
// to the way the windows installer does when writing it
|
||||||
|
// to the registry
|
||||||
|
// The first 8 bytes will be inverted, from the the last
|
||||||
|
// 8 bytes always the nibbles will be inverted for further
|
||||||
|
// details look in the MSDN under compressed registry keys
|
||||||
|
static std::_tstring ConvertGuid(const std::_tstring& guid)
|
||||||
|
{
|
||||||
|
std::_tstring convertedGuid;
|
||||||
|
|
||||||
|
std::_tstring part = GetGuidPart(guid, 0);
|
||||||
|
convertedGuid = Invert(part);
|
||||||
|
|
||||||
|
part = GetGuidPart(guid, 1);
|
||||||
|
convertedGuid += Invert(part);
|
||||||
|
|
||||||
|
part = GetGuidPart(guid, 2);
|
||||||
|
convertedGuid += Invert(part);
|
||||||
|
|
||||||
|
part = GetGuidPart(guid, 3);
|
||||||
|
convertedGuid += Invert(std::_tstring(part.c_str(), 2));
|
||||||
|
convertedGuid += Invert(std::_tstring(part.c_str() + 2, 2));
|
||||||
|
|
||||||
|
part = GetGuidPart(guid, 4);
|
||||||
|
int pos = 0;
|
||||||
|
for (int i = 0; i < 6; i++)
|
||||||
|
{
|
||||||
|
convertedGuid += Invert(std::_tstring(part.c_str() + pos, 2));
|
||||||
|
pos += 2;
|
||||||
|
}
|
||||||
|
return convertedGuid;
|
||||||
|
}
|
||||||
|
|
||||||
|
static inline bool IsSetMsiProperty(MSIHANDLE handle, const std::_tstring& sProperty)
|
||||||
|
{
|
||||||
|
std::_tstring value = GetMsiProperty(handle, sProperty);
|
||||||
|
return (value.length() > 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
static inline void UnsetMsiProperty(MSIHANDLE handle, const std::_tstring& sProperty)
|
||||||
|
{
|
||||||
|
MsiSetProperty(handle, sProperty.c_str(), NULL);
|
||||||
|
}
|
||||||
|
|
||||||
|
static inline void SetMsiProperty(MSIHANDLE handle, const std::_tstring& sProperty)
|
||||||
|
{
|
||||||
|
MsiSetProperty(handle, sProperty.c_str(), TEXT("1"));
|
||||||
|
}
|
||||||
|
|
||||||
|
static bool RegistryKeyHasUpgradeSubKey(HKEY hRootKey, const std::_tstring& regKey, const std::_tstring& upgradeKey)
|
||||||
|
{
|
||||||
|
HKEY hKey;
|
||||||
|
if (RegOpenKey(hRootKey, regKey.c_str(), &hKey) == ERROR_SUCCESS)
|
||||||
|
{
|
||||||
|
DWORD nSubKeys;
|
||||||
|
DWORD lLongestSubKey;
|
||||||
|
|
||||||
|
if (RegQueryInfoKey(
|
||||||
|
hKey, NULL, NULL, NULL, &nSubKeys, &lLongestSubKey, NULL, NULL, NULL, NULL, NULL, NULL) == ERROR_SUCCESS)
|
||||||
|
{
|
||||||
|
LPTSTR buffer = reinterpret_cast<LPTSTR>(_alloca(lLongestSubKey + 1));
|
||||||
|
|
||||||
|
for (DWORD i = 0; i < nSubKeys; i++)
|
||||||
|
{
|
||||||
|
LONG ret = RegEnumKey(hKey, i, buffer, lLongestSubKey + 1);
|
||||||
|
if ((ret == ERROR_SUCCESS) && (buffer == upgradeKey))
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
static BOOL MoveFileEx9x( LPCSTR lpExistingFileNameA, LPCSTR lpNewFileNameA, DWORD dwFlags )
|
static BOOL MoveFileEx9x( LPCSTR lpExistingFileNameA, LPCSTR lpNewFileNameA, DWORD dwFlags )
|
||||||
{
|
{
|
||||||
BOOL fSuccess = FALSE; // assume failure
|
BOOL fSuccess = FALSE; // assume failure
|
||||||
|
|
@ -127,25 +254,6 @@ static BOOL MoveFileExImpl( LPCSTR lpExistingFileNameA, LPCSTR lpNewFileNameA, D
|
||||||
return MoveFileExA( lpExistingFileNameA, lpNewFileNameA, dwFlags );
|
return MoveFileExA( lpExistingFileNameA, lpNewFileNameA, dwFlags );
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
static std::_tstring GetMsiProperty( MSIHANDLE handle, const std::_tstring& sProperty )
|
|
||||||
{
|
|
||||||
std::_tstring result;
|
|
||||||
TCHAR szDummy[1] = TEXT("");
|
|
||||||
DWORD nChars = 0;
|
|
||||||
|
|
||||||
if ( MsiGetProperty( handle, sProperty.c_str(), szDummy, &nChars ) == ERROR_MORE_DATA )
|
|
||||||
{
|
|
||||||
DWORD nBytes = ++nChars * sizeof(TCHAR);
|
|
||||||
LPTSTR buffer = reinterpret_cast<LPTSTR>(_alloca(nBytes));
|
|
||||||
ZeroMemory( buffer, nBytes );
|
|
||||||
MsiGetProperty(handle, sProperty.c_str(), buffer, &nChars);
|
|
||||||
result = buffer;
|
|
||||||
}
|
|
||||||
|
|
||||||
return result;
|
|
||||||
}
|
|
||||||
|
|
||||||
static bool SwapFiles( const std::_tstring& sFileName1, const std::_tstring& sFileName2 )
|
static bool SwapFiles( const std::_tstring& sFileName1, const std::_tstring& sFileName2 )
|
||||||
{
|
{
|
||||||
std::_tstring sTempFileName = sFileName1 + TEXT(".tmp");
|
std::_tstring sTempFileName = sFileName1 + TEXT(".tmp");
|
||||||
|
|
@ -153,8 +261,7 @@ static bool SwapFiles( const std::_tstring& sFileName1, const std::_tstring& sFi
|
||||||
bool fSuccess = true;
|
bool fSuccess = true;
|
||||||
|
|
||||||
//Try to move the original file to a temp file
|
//Try to move the original file to a temp file
|
||||||
fSuccess = MoveFileExImpl( sFileName1.c_str(), sTempFileName.c_str(),
|
fSuccess = MoveFileExImpl( sFileName1.c_str(), sTempFileName.c_str(), MOVEFILE_REPLACE_EXISTING);
|
||||||
MOVEFILE_REPLACE_EXISTING );
|
|
||||||
|
|
||||||
std::_tstring mystr;
|
std::_tstring mystr;
|
||||||
|
|
||||||
|
|
@ -165,7 +272,7 @@ static bool SwapFiles( const std::_tstring& sFileName1, const std::_tstring& sFi
|
||||||
if ( fSuccess )
|
if ( fSuccess )
|
||||||
{
|
{
|
||||||
fSuccess = MoveFileExImpl( sTempFileName.c_str(), sFileName2.c_str(),
|
fSuccess = MoveFileExImpl( sTempFileName.c_str(), sFileName2.c_str(),
|
||||||
MOVEFILE_REPLACE_EXISTING );
|
MOVEFILE_REPLACE_EXISTING );
|
||||||
if ( !fSuccess )
|
if ( !fSuccess )
|
||||||
{
|
{
|
||||||
MoveFileExImpl( sFileName1.c_str(), sFileName2.c_str(), MOVEFILE_REPLACE_EXISTING );
|
MoveFileExImpl( sFileName1.c_str(), sFileName2.c_str(), MOVEFILE_REPLACE_EXISTING );
|
||||||
|
|
@ -409,3 +516,115 @@ extern "C" UINT __stdcall IsOfficeRunning( MSIHANDLE handle )
|
||||||
return ERROR_SUCCESS;
|
return ERROR_SUCCESS;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
extern "C" UINT __stdcall SetFeatureState( MSIHANDLE handle )
|
||||||
|
{
|
||||||
|
std::_tstring mystr;
|
||||||
|
|
||||||
|
// 1. Reading Product Code from setup.ini of installed Office
|
||||||
|
|
||||||
|
std::_tstring sInstallPath = GetMsiProperty(handle, TEXT("INSTALLLOCATION"));
|
||||||
|
// MessageBox(NULL, sInstallPath.c_str(), "INSTALLLOCATION", MB_OK);
|
||||||
|
std::_tstring sSetupiniPath = sInstallPath + TEXT("program\\setup.ini");
|
||||||
|
|
||||||
|
TCHAR szProductCode[32767];
|
||||||
|
|
||||||
|
GetPrivateProfileString(
|
||||||
|
TEXT("Bootstrap"),
|
||||||
|
TEXT("ProductCode"),
|
||||||
|
TEXT("NOTFOUND"),
|
||||||
|
szProductCode,
|
||||||
|
elementsof(szProductCode),
|
||||||
|
sSetupiniPath.c_str()
|
||||||
|
);
|
||||||
|
|
||||||
|
if ( !_tcsicmp( szProductCode, TEXT("NOTFOUND") ) )
|
||||||
|
{
|
||||||
|
// No setup.ini or no "ProductCode" in setup.ini. This is an invalid directory.
|
||||||
|
// MessageBox(NULL, "NOTFOUND set", "DEBUG", MB_OK);
|
||||||
|
return ERROR_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 2. Converting Product code
|
||||||
|
|
||||||
|
std::_tstring productCode = TEXT(szProductCode);
|
||||||
|
productCode = ConvertGuid(std::_tstring(productCode.c_str() + 1, productCode.length() - 2));
|
||||||
|
mystr = TEXT("Changed product code: ") + productCode;
|
||||||
|
// MessageBox(NULL, mystr.c_str(), "ProductCode", MB_OK);
|
||||||
|
|
||||||
|
// 3. Setting path in the Windows registry to find installed features
|
||||||
|
|
||||||
|
std::_tstring registryKey;
|
||||||
|
HKEY registryRoot;
|
||||||
|
|
||||||
|
if ( IsSetMsiProperty(handle, TEXT("ALLUSERS")) )
|
||||||
|
{
|
||||||
|
registryRoot = HKEY_LOCAL_MACHINE;
|
||||||
|
registryKey = TEXT("Software\\Classes\\Installer\\Features\\") + productCode;
|
||||||
|
mystr = registryKey;
|
||||||
|
// MessageBox( NULL, mystr.c_str(), "ALLUSERS", MB_OK );
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
registryRoot = HKEY_CURRENT_USER;
|
||||||
|
registryKey = TEXT("Software\\Microsoft\\Installer\\Features\\") + productCode;
|
||||||
|
mystr = registryKey;
|
||||||
|
// MessageBox( NULL, mystr.c_str(), "ALLUSERS", MB_OK );
|
||||||
|
}
|
||||||
|
|
||||||
|
// 4. Collecting all installed features from Windows registry
|
||||||
|
|
||||||
|
HKEY hKey;
|
||||||
|
if (RegOpenKey(registryRoot, registryKey.c_str(), &hKey) == ERROR_SUCCESS)
|
||||||
|
{
|
||||||
|
int counter = 0;
|
||||||
|
// DWORD counter = 0;
|
||||||
|
LONG lEnumResult;
|
||||||
|
|
||||||
|
do
|
||||||
|
{
|
||||||
|
TCHAR szValueName[8192];
|
||||||
|
DWORD nValueNameSize = sizeof(szValueName);
|
||||||
|
LPDWORD pValueNameSize = &nValueNameSize;
|
||||||
|
TCHAR szValueData[8192];
|
||||||
|
DWORD nValueDataSize = sizeof(szValueData);
|
||||||
|
|
||||||
|
lEnumResult = RegEnumValue( hKey, counter, szValueName, pValueNameSize, NULL, NULL, (LPBYTE)szValueData, &nValueDataSize);
|
||||||
|
|
||||||
|
if ( ERROR_SUCCESS == lEnumResult )
|
||||||
|
{
|
||||||
|
std::_tstring sValueName = szValueName;
|
||||||
|
std::_tstring sValueData = szValueData;
|
||||||
|
|
||||||
|
// mystr = sValueName;
|
||||||
|
// MessageBox( NULL, mystr.c_str(), "ValueName", MB_OK );
|
||||||
|
// mystr = sValueData;
|
||||||
|
// MessageBox( NULL, mystr.c_str(), "ValueData", MB_OK );
|
||||||
|
|
||||||
|
// Does this feature exist in this patch?
|
||||||
|
if ( IsSetMsiProperty(handle, sValueName) )
|
||||||
|
{
|
||||||
|
// Feature is not installed, if szValueData starts with a "square" (ascii 6)
|
||||||
|
if ( 6 == szValueData[0] )
|
||||||
|
{
|
||||||
|
MsiSetFeatureState(handle,sValueName.c_str(),INSTALLSTATE_ABSENT); // do not install this feature
|
||||||
|
// mystr = TEXT("Do NOT install: ") + sValueName;
|
||||||
|
// MessageBox( NULL, mystr.c_str(), "ValueName", MB_OK );
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
MsiSetFeatureState(handle,sValueName.c_str(),INSTALLSTATE_LOCAL); // do install this feature
|
||||||
|
// mystr = TEXT("Do install: ") + sValueName;
|
||||||
|
// MessageBox( NULL, mystr.c_str(), "ValueName", MB_OK );
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
counter = counter + 1;
|
||||||
|
|
||||||
|
} while ( ERROR_SUCCESS == lEnumResult );
|
||||||
|
|
||||||
|
RegCloseKey( hKey );
|
||||||
|
}
|
||||||
|
|
||||||
|
return ERROR_SUCCESS;
|
||||||
|
}
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue