reactos/dll/win32/msi/upgrade.c

   1 /*
   2  * Implementation of the Microsoft Installer (msi.dll)
   3  *
   4  * Copyright 2005 Aric Stewart for CodeWeavers
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  */
  20
  21 /*
  22  * Actions focused on in this module
  23  *
  24  * FindRelatedProducts
  25  * MigrateFeatureStates (TODO)
  26  * RemoveExistingProducts (TODO)
  27  */
  28
  29 #include <stdarg.h>
  30
  31 #include "windef.h"
  32 #include "winbase.h"
  33 #include "winerror.h"
  34 #include "winreg.h"
  35 #include "wine/debug.h"
  36 #include "msidefs.h"
  37 #include "msipriv.h"
  38 #include "winuser.h"
  39 #include "wine/unicode.h"
  40
  41 WINE_DEFAULT_DEBUG_CHANNEL(msi);
  42
  43 static BOOL check_language(DWORD lang1, LPCWSTR lang2, DWORD attributes)
  44 {
  45     DWORD langdword;
  46
  47     if (!lang2 || lang2[0]==0)
  48         return TRUE;
  49
  50     langdword = atoiW(lang2);
  51
  52     if (attributes & msidbUpgradeAttributesLanguagesExclusive)
  53         return (lang1 != langdword);
  54     else
  55         return (lang1 == langdword);
  56 }
  57
  58 static void append_productcode(MSIPACKAGE* package, LPCWSTR action_property,
  59                                LPCWSTR productid)
  60 {
  61     LPWSTR prop;
  62     LPWSTR newprop;
  63     DWORD len;
  64     UINT r;
  65
  66     prop = msi_dup_property(package->db, action_property );
  67     if (prop)
  68         len = strlenW(prop);
  69     else
  70         len = 0;
  71
  72     /*separator*/
  73     len ++;
  74
  75     len += strlenW(productid);
  76
  77     /*null*/
  78     len++;
  79
  80     newprop = msi_alloc( len*sizeof(WCHAR) );
  81
  82     if (prop)
  83     {
  84         strcpyW(newprop,prop);
  85         strcatW(newprop,szSemiColon);
  86     }
  87     else
  88         newprop[0] = 0;
  89     strcatW(newprop,productid);
  90
  91     r = msi_set_property( package->db, action_property, newprop, -1 );
  92     if (r == ERROR_SUCCESS && !strcmpW( action_property, szSourceDir ))
  93         msi_reset_folders( package, TRUE );
  94
  95     TRACE("Found Related Product... %s now %s\n",
  96           debugstr_w(action_property), debugstr_w(newprop));
  97
  98     msi_free( prop );
  99     msi_free( newprop );
 100 }
 101
 102 static UINT ITERATE_FindRelatedProducts(MSIRECORD *rec, LPVOID param)
 103 {
 104     MSIPACKAGE *package = param;
 105     WCHAR product[GUID_SIZE];
 106     DWORD index = 0;
 107     DWORD attributes = 0;
 108     DWORD sz = GUID_SIZE;
 109     LPCWSTR upgrade_code;
 110     HKEY hkey = 0;
 111     UINT rc = ERROR_SUCCESS;
 112     MSIRECORD *uirow;
 113
 114     upgrade_code = MSI_RecordGetString(rec,1);
 115
 116     rc = MSIREG_OpenUpgradeCodesKey(upgrade_code, &hkey, FALSE);
 117     if (rc != ERROR_SUCCESS)
 118         return ERROR_SUCCESS;
 119
 120     uirow = MSI_CreateRecord(1);
 121     attributes = MSI_RecordGetInteger(rec,5);
 122
 123     while (rc == ERROR_SUCCESS)
 124     {
 125         rc = RegEnumValueW(hkey, index, product, &sz, NULL, NULL, NULL, NULL);
 126         if (rc == ERROR_SUCCESS)
 127         {
 128             WCHAR productid[GUID_SIZE];
 129             LPCWSTR ver, language, action_property;
 130             DWORD check = 0, comp_ver, sz = 0x100;
 131             HKEY hukey;
 132             INT r;
 133
 134             TRACE("Looking at index %u product %s\n", index, debugstr_w(product));
 135
 136             unsquash_guid(product, productid);
 137             if (MSIREG_OpenProductKey(productid, NULL, MSIINSTALLCONTEXT_USERMANAGED, &hukey, FALSE) &&
 138                 MSIREG_OpenProductKey(productid, NULL, MSIINSTALLCONTEXT_USERUNMANAGED, &hukey, FALSE) &&
 139                 MSIREG_OpenProductKey(productid, NULL, MSIINSTALLCONTEXT_MACHINE, &hukey, FALSE))
 140             {
 141                 TRACE("product key not found\n");
 142                 rc = ERROR_SUCCESS;
 143                 index ++;
 144                 continue;
 145             }
 146
 147             sz = sizeof(DWORD);
 148             RegQueryValueExW(hukey, INSTALLPROPERTY_VERSIONW, NULL, NULL, (LPBYTE)&check, &sz);
 149
 150             /* check version minimum */
 151             ver = MSI_RecordGetString(rec,2);
 152             if (ver)
 153             {
 154                 comp_ver = msi_version_str_to_dword(ver);
 155                 r = check - comp_ver;
 156                 if (r < 0 || (r == 0 && !(attributes & msidbUpgradeAttributesVersionMinInclusive)))
 157                 {
 158                     TRACE("version below minimum\n");
 159                     RegCloseKey(hukey);
 160                     index ++;
 161                     continue;
 162                 }
 163             }
 164
 165             /* check version maximum */
 166             ver = MSI_RecordGetString(rec,3);
 167             if (ver)
 168             {
 169                 comp_ver = msi_version_str_to_dword(ver);
 170                 r = check - comp_ver;
 171                 if (r > 0 || (r == 0 && !(attributes & msidbUpgradeAttributesVersionMaxInclusive)))
 172                 {
 173                     RegCloseKey(hukey);
 174                     index ++;
 175                     continue;
 176                 }
 177                 TRACE("version above maximum\n");
 178             }
 179
 180             /* check language */
 181             sz = sizeof(DWORD);
 182             RegQueryValueExW(hukey, INSTALLPROPERTY_LANGUAGEW, NULL, NULL, (LPBYTE)&check, &sz);
 183             RegCloseKey(hukey);
 184             language = MSI_RecordGetString(rec,4);
 185             if (!check_language(check, language, attributes))
 186             {
 187                 index ++;
 188                 TRACE("language doesn't match\n");
 189                 continue;
 190             }
 191             TRACE("found related product\n");
 192
 193             action_property = MSI_RecordGetString(rec, 7);
 194             append_productcode(package, action_property, productid);
 195             MSI_RecordSetStringW(uirow, 1, productid);
 196             msi_ui_actiondata(package, szFindRelatedProducts, uirow);
 197         }
 198         index ++;
 199     }
 200     RegCloseKey(hkey);
 201     msiobj_release( &uirow->hdr);
 202
 203     return ERROR_SUCCESS;
 204 }
 205
 206 UINT ACTION_FindRelatedProducts(MSIPACKAGE *package)
 207 {
 208     static const WCHAR query[] = {
 209         'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
 210         '`','U','p','g','r','a','d','e','`',0};
 211     MSIQUERY *view;
 212     UINT rc;
 213
 214     if (msi_get_property_int(package->db, szInstalled, 0))
 215     {
 216         TRACE("Skipping FindRelatedProducts action: product already installed\n");
 217         return ERROR_SUCCESS;
 218     }
 219     if (msi_action_is_unique(package, szFindRelatedProducts))
 220     {
 221         TRACE("Skipping FindRelatedProducts action: already done in UI sequence\n");
 222         return ERROR_SUCCESS;
 223     }
 224     else
 225         msi_register_unique_action(package, szFindRelatedProducts);
 226
 227     rc = MSI_DatabaseOpenViewW(package->db, query, &view);
 228     if (rc != ERROR_SUCCESS)
 229         return ERROR_SUCCESS;
 230
 231     rc = MSI_IterateRecords(view, NULL, ITERATE_FindRelatedProducts, package);
 232     msiobj_release(&view->hdr);
 233     return rc;
 234 }