setup_native/source/win32/customactions/shellextensions/migrateinstallpath.cxx

   1 /*************************************************************************
   2  *
   3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   4  *
   5  * Copyright 2008 by Sun Microsystems, Inc.
   6  *
   7  * OpenOffice.org - a multi-platform office productivity suite
   8  *
   9  * $RCSfile: migrateinstallpath.cxx,v $
  10  * $Revision: 1.8.108.1 $
  11  *
  12  * This file is part of OpenOffice.org.
  13  *
  14  * OpenOffice.org is free software: you can redistribute it and/or modify
  15  * it under the terms of the GNU Lesser General Public License version 3
  16  * only, as published by the Free Software Foundation.
  17  *
  18  * OpenOffice.org is distributed in the hope that it will be useful,
  19  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  20  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  21  * GNU Lesser General Public License version 3 for more details
  22  * (a copy is included in the LICENSE file that accompanied this code).
  23  *
  24  * You should have received a copy of the GNU Lesser General Public License
  25  * version 3 along with OpenOffice.org.  If not, see
  26  * <http://www.openoffice.org/license.html>
  27  * for a copy of the LGPLv3 License.
  28  *
  29  ************************************************************************/
  30
  31 #define _WIN32_WINDOWS 0x0410
  32 #ifdef _MSC_VER
  33 #pragma warning(push, 1) /* disable warnings within system headers */
  34 #endif
  35 #define WIN32_LEAN_AND_MEAN
  36 #include <windows.h>
  37 #include <msiquery.h>
  38 #ifdef _MSC_VER
  39 #pragma warning(pop)
  40 #endif
  41
  42 #include <malloc.h>
  43
  44 #ifdef UNICODE
  45 #define _UNICODE
  46 #define _tstring        wstring
  47 #else
  48 #define _tstring        string
  49 #endif
  50 #include <tchar.h>
  51 #include <string>
  52
  53 using namespace std;
  54
  55 namespace
  56 {
  57     std::_tstring GetMsiProperty( MSIHANDLE handle, const std::_tstring& sProperty )
  58     {
  59         std::_tstring   result;
  60         TCHAR   szDummy[1] = TEXT("");
  61         DWORD   nChars = 0;
  62
  63         if ( MsiGetProperty( handle, sProperty.c_str(), szDummy, &nChars ) == ERROR_MORE_DATA )
  64         {
  65             DWORD nBytes = ++nChars * sizeof(TCHAR);
  66             LPTSTR buffer = reinterpret_cast<LPTSTR>(_alloca(nBytes));
  67             ZeroMemory( buffer, nBytes );
  68             MsiGetProperty(handle, sProperty.c_str(), buffer, &nChars);
  69             result = buffer;
  70         }
  71
  72         return  result;
  73     }
  74 } // namespace
  75
  76 extern "C" UINT __stdcall MigrateInstallPath( MSIHANDLE handle )
  77 {
  78     TCHAR       szValue[8192];
  79     DWORD       nValueSize = sizeof(szValue);
  80     HKEY        hKey;
  81     std::_tstring       sInstDir;
  82
  83     std::_tstring       sManufacturer = GetMsiProperty( handle, TEXT("Manufacturer") );
  84     std::_tstring       sDefinedName = GetMsiProperty( handle, TEXT("DEFINEDPRODUCT") );
  85     std::_tstring       sUpdateVersion = GetMsiProperty( handle, TEXT("DEFINEDVERSION") );
  86     std::_tstring       sUpgradeCode = GetMsiProperty( handle, TEXT("UpgradeCode") );
  87
  88     std::_tstring       sProductKey = "Software\\" + sManufacturer + "\\" + sDefinedName +
  89                                         "\\" + sUpdateVersion + "\\" + sUpgradeCode;
  90
  91     std::_tstring       mystr;
  92     mystr = "ProductKey: " + sProductKey;
  93     // MessageBox( NULL, mystr.c_str(), "ProductKey", MB_OK );
  94
  95     if ( ERROR_SUCCESS == RegOpenKey( HKEY_CURRENT_USER,  sProductKey.c_str(), &hKey ) )
  96     {
  97         if ( ERROR_SUCCESS == RegQueryValueEx( hKey, TEXT("OFFICEINSTALLLOCATION"), NULL, NULL, (LPBYTE)szValue, &nValueSize ) )
  98         {
  99             sInstDir = szValue;
 100             MsiSetProperty(handle, TEXT("OFFICEINSTALLLOCATION"), sInstDir.c_str());
 101             // MessageBox( NULL, sInstDir.c_str(), "Found in HKEY_CURRENT_USER", MB_OK );
 102         }
 103
 104         if ( ERROR_SUCCESS == RegQueryValueEx( hKey, TEXT("BASISINSTALLLOCATION"), NULL, NULL, (LPBYTE)szValue, &nValueSize ) )
 105         {
 106             sInstDir = szValue;
 107             MsiSetProperty(handle, TEXT("BASISINSTALLLOCATION"), sInstDir.c_str());
 108             // MessageBox( NULL, sInstDir.c_str(), "Found in HKEY_CURRENT_USER", MB_OK );
 109         }
 110
 111         if ( ERROR_SUCCESS == RegQueryValueEx( hKey, TEXT("UREINSTALLLOCATION"), NULL, NULL, (LPBYTE)szValue, &nValueSize ) )
 112         {
 113             sInstDir = szValue;
 114             MsiSetProperty(handle, TEXT("UREINSTALLLOCATION"), sInstDir.c_str());
 115             // MessageBox( NULL, sInstDir.c_str(), "Found in HKEY_CURRENT_USER", MB_OK );
 116         }
 117
 118         if ( ERROR_SUCCESS == RegQueryValueEx( hKey, TEXT("INSTALLLOCATION"), NULL, NULL, (LPBYTE)szValue, &nValueSize ) )
 119         {
 120             sInstDir = szValue;
 121             MsiSetProperty(handle, TEXT("INSTALLLOCATION"), sInstDir.c_str());
 122             // MessageBox( NULL, sInstDir.c_str(), "Found in HKEY_CURRENT_USER", MB_OK );
 123         }
 124
 125         RegCloseKey( hKey );
 126     }
 127     else if ( ERROR_SUCCESS == RegOpenKey( HKEY_LOCAL_MACHINE,  sProductKey.c_str(), &hKey ) )
 128     {
 129         if ( ERROR_SUCCESS == RegQueryValueEx( hKey, TEXT("OFFICEINSTALLLOCATION"), NULL, NULL, (LPBYTE)szValue, &nValueSize ) )
 130         {
 131             sInstDir = szValue;
 132             MsiSetProperty(handle, TEXT("OFFICEINSTALLLOCATION"), sInstDir.c_str());
 133             // MessageBox( NULL, sInstDir.c_str(), "Found in HKEY_LOCAL_MACHINE", MB_OK );
 134         }
 135
 136         if ( ERROR_SUCCESS == RegQueryValueEx( hKey, TEXT("BASISINSTALLLOCATION"), NULL, NULL, (LPBYTE)szValue, &nValueSize ) )
 137         {
 138             sInstDir = szValue;
 139             MsiSetProperty(handle, TEXT("BASISINSTALLLOCATION"), sInstDir.c_str());
 140             // MessageBox( NULL, sInstDir.c_str(), "Found in HKEY_LOCAL_MACHINE", MB_OK );
 141         }
 142
 143         if ( ERROR_SUCCESS == RegQueryValueEx( hKey, TEXT("UREINSTALLLOCATION"), NULL, NULL, (LPBYTE)szValue, &nValueSize ) )
 144         {
 145             sInstDir = szValue;
 146             MsiSetProperty(handle, TEXT("UREINSTALLLOCATION"), sInstDir.c_str());
 147             // MessageBox( NULL, sInstDir.c_str(), "Found in HKEY_LOCAL_MACHINE", MB_OK );
 148         }
 149
 150         if ( ERROR_SUCCESS == RegQueryValueEx( hKey, TEXT("INSTALLLOCATION"), NULL, NULL, (LPBYTE)szValue, &nValueSize ) )
 151         {
 152             sInstDir = szValue;
 153             MsiSetProperty(handle, TEXT("INSTALLLOCATION"), sInstDir.c_str());
 154             // MessageBox( NULL, sInstDir.c_str(), "Found in HKEY_LOCAL_MACHINE", MB_OK );
 155         }
 156
 157         RegCloseKey( hKey );
 158     }
 159
 160     return ERROR_SUCCESS;
 161
 162 }