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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  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 "action.h"
  40 #include "wine/unicode.h"
  41
  42 WINE_DEFAULT_DEBUG_CHANNEL(msi);
  43
  44 extern const WCHAR szFindRelatedProducts[];
  45 extern const WCHAR szMigrateFeatureStates[];
  46 extern const WCHAR szRemoveExistingProducts[];
  47
  48 static BOOL check_language(DWORD lang1, LPCWSTR lang2, DWORD attributes)
  49 {
  50     DWORD langdword;
  51
  52     if (!lang2 || lang2[0]==0)
  53         return TRUE;
  54
  55     langdword = atoiW(lang2);
  56
  57     if (attributes & msidbUpgradeAttributesLanguagesExclusive)
  58         return (lang1 != langdword);
  59     else
  60         return (lang1 == langdword);
  61 }
  62
  63 static void append_productcode(MSIPACKAGE* package, LPCWSTR action_property,
  64                                LPCWSTR productid)
  65 {
  66     LPWSTR prop;
  67     LPWSTR newprop;
  68     DWORD len;
  69     static const WCHAR separator[] = {';',0};
  70
  71     prop = msi_dup_property(package, action_property );
  72     if (prop)
  73         len = strlenW(prop);
  74     else
  75         len = 0;
  76
  77     /*separator*/
  78     len ++;
  79
  80     len += strlenW(productid);
  81
  82     /*null*/
  83     len++;
  84
  85     newprop = msi_alloc( len*sizeof(WCHAR) );
  86
  87     if (prop)
  88     {
  89         strcpyW(newprop,prop);
  90         strcatW(newprop,separator);
  91     }
  92     else
  93         newprop[0] = 0;
  94     strcatW(newprop,productid);
  95
  96     MSI_SetPropertyW(package, action_property, newprop);
  97     TRACE("Found Related Product... %s now %s\n",debugstr_w(action_property),
  98                     debugstr_w(newprop));
  99     msi_free( prop );
 100     msi_free( newprop );
 101 }
 102
 103 static UINT ITERATE_FindRelatedProducts(MSIRECORD *rec, LPVOID param)
 104 {
 105     MSIPACKAGE *package = (MSIPACKAGE*)param;
 106     WCHAR product[GUID_SIZE];
 107     DWORD index = 0;
 108     DWORD attributes = 0;
 109     DWORD sz = GUID_SIZE;
 110     LPCWSTR upgrade_code;
 111     HKEY hkey = 0;
 112     UINT rc = ERROR_SUCCESS;
 113     MSIRECORD *uirow;
 114
 115     upgrade_code = MSI_RecordGetString(rec,1);
 116
 117     rc = MSIREG_OpenUpgradeCodesKey(upgrade_code, &hkey, FALSE);
 118     if (rc != ERROR_SUCCESS)
 119         return ERROR_SUCCESS;
 120
 121     uirow = MSI_CreateRecord(1);
 122     attributes = MSI_RecordGetInteger(rec,5);
 123
 124     while (rc == ERROR_SUCCESS)
 125     {
 126         rc = RegEnumValueW(hkey, index, product, &sz, NULL, NULL, NULL, NULL);
 127         TRACE("Looking at (%li) %s\n",index,debugstr_w(product));
 128         if (rc == ERROR_SUCCESS)
 129         {
 130             WCHAR productid[GUID_SIZE];
 131             LPCWSTR ver;
 132             LPCWSTR language;
 133             LPCWSTR action_property;
 134             DWORD check = 0x00000000;
 135             DWORD comp_ver = 0x00000000;
 136             DWORD sz = 0x100;
 137             HKEY hukey;
 138             INT r;
 139
 140             unsquash_guid(product,productid);
 141             rc = MSIREG_OpenUserProductsKey(productid, &hukey, FALSE);
 142             if (rc != ERROR_SUCCESS)
 143             {
 144                 rc = ERROR_SUCCESS;
 145                 index ++;
 146                 continue;
 147             }
 148
 149             sz = sizeof(DWORD);
 150             RegQueryValueExW(hukey, INSTALLPROPERTY_VERSIONW, NULL, NULL,
 151                     (LPBYTE)&check, &sz);
 152             /* check min */
 153             ver = MSI_RecordGetString(rec,2);
 154             comp_ver = build_version_dword(ver);
 155             r = check - comp_ver;
 156             if (r < 0 || (r == 0 && !(attributes &
 157                                     msidbUpgradeAttributesVersionMinInclusive)))
 158             {
 159                 RegCloseKey(hukey);
 160                 index ++;
 161                 continue;
 162             }
 163
 164             /* check max */
 165             ver = MSI_RecordGetString(rec,3);
 166             comp_ver = build_version_dword(ver);
 167             r = check - comp_ver;
 168             if (r > 0 || (r == 0 && !(attributes &
 169                                     msidbUpgradeAttributesVersionMaxInclusive)))
 170             {
 171                 RegCloseKey(hukey);
 172                 index ++;
 173                 continue;
 174             }
 175
 176             /* check language*/
 177             sz = sizeof(DWORD);
 178             RegQueryValueExW(hukey, INSTALLPROPERTY_LANGUAGEW, NULL, NULL,
 179                     (LPBYTE)&check, &sz);
 180             RegCloseKey(hukey);
 181             language = MSI_RecordGetString(rec,4);
 182             TRACE("Checking languages 0x%lx and %s\n", check,
 183                             debugstr_w(language));
 184             if (!check_language(check, language, attributes))
 185             {
 186                 index ++;
 187                 continue;
 188             }
 189
 190             action_property = MSI_RecordGetString(rec,7);
 191             append_productcode(package,action_property,productid);
 192             ui_actiondata(package,szFindRelatedProducts,uirow);
 193         }
 194         index ++;
 195     }
 196     RegCloseKey(hkey);
 197     msiobj_release( &uirow->hdr);
 198
 199     return ERROR_SUCCESS;
 200 }
 201
 202 UINT ACTION_FindRelatedProducts(MSIPACKAGE *package)
 203 {
 204     static const WCHAR Query[] =
 205         {'S','E','L','E','C','T',' ','*',' ','F','R','O','M',
 206          ' ','`','U','p','g','r','a','d','e','`',0};
 207     UINT rc = ERROR_SUCCESS;
 208     MSIQUERY *view;
 209
 210     if (check_unique_action(package,szFindRelatedProducts))
 211     {
 212         TRACE("Skipping FindRelatedProducts action: already done on client side\n");
 213         return ERROR_SUCCESS;
 214     }
 215     else
 216         register_unique_action(package,szFindRelatedProducts);
 217
 218     rc = MSI_DatabaseOpenViewW(package->db, Query, &view);
 219     if (rc != ERROR_SUCCESS)
 220         return ERROR_SUCCESS;
 221
 222     rc = MSI_IterateRecords(view, NULL, ITERATE_FindRelatedProducts, package);
 223     msiobj_release(&view->hdr);
 224
 225     return rc;
 226 }