msi/tests: Opening package with relative filename sometimes succeeds on Windows.
[wine] / dlls / 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 extern const WCHAR szFindRelatedProducts[];
44 extern const WCHAR szMigrateFeatureStates[];
45 extern const WCHAR szRemoveExistingProducts[];
46
47 static BOOL check_language(DWORD lang1, LPCWSTR lang2, DWORD attributes)
48 {
49     DWORD langdword;
50
51     if (!lang2 || lang2[0]==0)
52         return TRUE;
53
54     langdword = atoiW(lang2);
55
56     if (attributes & msidbUpgradeAttributesLanguagesExclusive)
57         return (lang1 != langdword);
58     else
59         return (lang1 == langdword);
60 }
61
62 static void append_productcode(MSIPACKAGE* package, LPCWSTR action_property,
63                                LPCWSTR productid)
64 {
65     LPWSTR prop;
66     LPWSTR newprop;
67     DWORD len;
68     static const WCHAR separator[] = {';',0};
69
70     prop = msi_dup_property(package, action_property );
71     if (prop)
72         len = strlenW(prop);
73     else
74         len = 0;
75
76     /*separator*/
77     len ++;
78
79     len += strlenW(productid);
80
81     /*null*/
82     len++;
83
84     newprop = msi_alloc( len*sizeof(WCHAR) );
85
86     if (prop)
87     {
88         strcpyW(newprop,prop);
89         strcatW(newprop,separator);
90     }
91     else
92         newprop[0] = 0;
93     strcatW(newprop,productid);
94
95     MSI_SetPropertyW(package, action_property, newprop);
96     TRACE("Found Related Product... %s now %s\n",debugstr_w(action_property),
97                     debugstr_w(newprop));
98     msi_free( prop );
99     msi_free( newprop );
100 }
101
102 static UINT ITERATE_FindRelatedProducts(MSIRECORD *rec, LPVOID param)
103 {
104     MSIPACKAGE *package = (MSIPACKAGE*)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         TRACE("Looking at (%i) %s\n",index,debugstr_w(product));
127         if (rc == ERROR_SUCCESS)
128         {
129             WCHAR productid[GUID_SIZE];
130             LPCWSTR ver;
131             LPCWSTR language;
132             LPCWSTR action_property;
133             DWORD check = 0x00000000;
134             DWORD comp_ver = 0x00000000;
135             DWORD sz = 0x100;
136             HKEY hukey;
137             INT r;
138
139             unsquash_guid(product, productid);
140             rc = MSIREG_OpenProductKey(productid, package->Context,
141                                        &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 = msi_version_str_to_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 = msi_version_str_to_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 %x 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 }