diff options
Diffstat (limited to 'setup_native')
| -rw-r--r-- | setup_native/source/win32/customactions/shellextensions/upgrade.cxx | 233 |
1 files changed, 233 insertions, 0 deletions
diff --git a/setup_native/source/win32/customactions/shellextensions/upgrade.cxx b/setup_native/source/win32/customactions/shellextensions/upgrade.cxx new file mode 100644 index 000000000000..a321f39bdc98 --- /dev/null +++ b/setup_native/source/win32/customactions/shellextensions/upgrade.cxx @@ -0,0 +1,233 @@ +/************************************************************************* + * + * $RCSfile: upgrade.cxx,v $ + * + * $Revision: 1.2 $ + * + * last change: $Author: rt $ $Date: 2005-04-04 10:12:29 $ + * + * The Contents of this file are made available subject to the terms of + * either of the following licenses + * + * - GNU Lesser General Public License Version 2.1 + * - Sun Industry Standards Source License Version 1.1 + * + * Sun Microsystems Inc., October, 2000 + * + * GNU Lesser General Public License Version 2.1 + * ============================================= + * Copyright 2000 by Sun Microsystems, Inc. + * 901 San Antonio Road, Palo Alto, CA 94303, USA + * + * This library is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License version 2.1, as published by the Free Software Foundation. + * + * This library is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * Lesser General Public License for more details. + * + * You should have received a copy of the GNU Lesser General Public + * License along with this library; if not, write to the Free Software + * Foundation, Inc., 59 Temple Place, Suite 330, Boston, + * MA 02111-1307 USA + * + * + * Sun Industry Standards Source License Version 1.1 + * ================================================= + * The contents of this file are subject to the Sun Industry Standards + * Source License Version 1.1 (the "License"); You may not use this file + * except in compliance with the License. You may obtain a copy of the + * License at http://www.openoffice.org/license.html. + * + * Software provided under this License is provided on an "AS IS" basis, + * WITHOUT WARRANTY OF ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING, + * WITHOUT LIMITATION, WARRANTIES THAT THE SOFTWARE IS FREE OF DEFECTS, + * MERCHANTABLE, FIT FOR A PARTICULAR PURPOSE, OR NON-INFRINGING. + * See the License for the specific provisions governing your rights and + * obligations concerning the Software. + * + * The Initial Developer of the Original Code is: Sun Microsystems, Inc. + * + * Copyright: 2000 by Sun Microsystems, Inc. + * + * All Rights Reserved. + * + * Contributor(s): _______________________________________ + * + * + ************************************************************************/ + +#undef UNICODE +#undef _UNICODE + +#define WIN32_LEAN_AND_MEAN +#include <windows.h> +#include <msiquery.h> +#include <malloc.h> +#include <assert.h> + +#include <tchar.h> +#include <string> + +#ifdef _WIN32_WINNT +#error YES +#endif + +using namespace std; + +namespace +{ + // The provided GUID must be without surounding '{}' + string GetGuidPart(const string& guid, int index) + { + assert((guid.length() == 36) && "No GUID or wrong format!"); + assert(((index > -1) && (index < 5)) && "Out of range!"); + + if (index == 0) return string(guid.c_str(), 8); + if (index == 1) return string(guid.c_str() + 9, 4); + if (index == 2) return string(guid.c_str() + 14, 4); + if (index == 3) return string(guid.c_str() + 19, 4); + if (index == 4) return string(guid.c_str() + 24, 12); + + return string(); + } + + void Swap(char* p1, char* p2) + { + char tmp = *p1; + *p1 = *p2; + *p2 = tmp; + } + + string Invert(const string& 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 string(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 + string ConvertGuid(const string& guid) + { + string convertedGuid; + + string 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(string(part.c_str(), 2)); + convertedGuid += Invert(string(part.c_str() + 2, 2)); + + part = GetGuidPart(guid, 4); + int pos = 0; + for (int i = 0; i < 6; i++) + { + convertedGuid += Invert(string(part.c_str() + pos, 2)); + pos += 2; + } + return convertedGuid; + } + + string GetMsiProperty(MSIHANDLE handle, const string& sProperty) + { + string 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; + } + + inline bool IsSetMsiProperty(MSIHANDLE handle, const string& sProperty) + { + return (GetMsiProperty(handle, sProperty).length() > 0); + } + + inline void UnsetMsiProperty(MSIHANDLE handle, const string& sProperty) + { + MsiSetProperty(handle, sProperty.c_str(), NULL); + } + + inline void SetMsiProperty(MSIHANDLE handle, const string& sProperty) + { + MsiSetProperty(handle, sProperty.c_str(), TEXT("1")); + } + + bool RegistryKeyHasUpgradeSubKey( + HKEY hRootKey, const string& regKey, const string& 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; + } +} // namespace + +extern "C" UINT __stdcall SetProductInstallMode(MSIHANDLE handle) +{ + string upgradeCode = GetMsiProperty(handle, TEXT("UpgradeCode")); + upgradeCode = ConvertGuid(string(upgradeCode.c_str() + 1, upgradeCode.length() - 2)); + + //MessageBox(NULL, upgradeCode.c_str(), TEXT("Debug"), MB_OK); + + if (RegistryKeyHasUpgradeSubKey( + HKEY_CURRENT_USER, + TEXT("Software\\Microsoft\\Installer\\UpgradeCodes"), + upgradeCode) && IsSetMsiProperty(handle, TEXT("ALLUSERS"))) + { + UnsetMsiProperty(handle, TEXT("ALLUSERS")); + //MessageBox(NULL, "ALLUSERS removed", "DEBUG", MB_OK); + } + else if (RegistryKeyHasUpgradeSubKey( + HKEY_LOCAL_MACHINE, + TEXT("Software\\Classes\\Installer\\UpgradeCodes"), + upgradeCode) && !IsSetMsiProperty(handle, TEXT("ALLUSERS"))) + { + SetMsiProperty(handle, TEXT("ALLUSERS")); + //MessageBox(NULL, "ALLUSERS set", "DEBUG", MB_OK); + } + return ERROR_SUCCESS; +} |