ws2_32: Set SO_REUSEPORT in addition to SO_REUSEADDR on MacOS.
[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 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 }