?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
msi: Implement the UnregisterTypeLibraries standard action.
[wine/multimedia.git] / dlls / msi / action.c
blob0f1877e82f84d31b560419a2747e316b1860e5ec
1 /*
2 * Implementation of the Microsoft Installer (msi.dll)
3 *
4 * Copyright 2004,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 #include <stdarg.h>
22
23 #define COBJMACROS
24
25 #include "windef.h"
26 #include "winbase.h"
27 #include "winerror.h"
28 #include "winreg.h"
29 #include "winsvc.h"
30 #include "odbcinst.h"
31 #include "wine/debug.h"
32 #include "msidefs.h"
33 #include "msipriv.h"
34 #include "winuser.h"
35 #include "shlobj.h"
36 #include "objbase.h"
37 #include "mscoree.h"
38 #include "fusion.h"
39 #include "shlwapi.h"
40 #include "wine/unicode.h"
41 #include "winver.h"
42
43 #define REG_PROGRESS_VALUE 13200
44 #define COMPONENT_PROGRESS_VALUE 24000
45
46 WINE_DEFAULT_DEBUG_CHANNEL(msi);
47
48 /*
49 * consts and values used
50 */
51 static const WCHAR c_colon[] = {'C',':','\\',0};
52
53 static const WCHAR szCreateFolders[] =
54 {'C','r','e','a','t','e','F','o','l','d','e','r','s',0};
55 static const WCHAR szCostFinalize[] =
56 {'C','o','s','t','F','i','n','a','l','i','z','e',0};
57 static const WCHAR szWriteRegistryValues[] =
58 {'W','r','i','t','e','R','e','g','i','s','t','r','y','V','a','l','u','e','s',0};
59 static const WCHAR szCostInitialize[] =
60 {'C','o','s','t','I','n','i','t','i','a','l','i','z','e',0};
61 static const WCHAR szFileCost[] =
62 {'F','i','l','e','C','o','s','t',0};
63 static const WCHAR szInstallInitialize[] =
64 {'I','n','s','t','a','l','l','I','n','i','t','i','a','l','i','z','e',0};
65 static const WCHAR szInstallValidate[] =
66 {'I','n','s','t','a','l','l','V','a','l','i','d','a','t','e',0};
67 static const WCHAR szLaunchConditions[] =
68 {'L','a','u','n','c','h','C','o','n','d','i','t','i','o','n','s',0};
69 static const WCHAR szProcessComponents[] =
70 {'P','r','o','c','e','s','s','C','o','m','p','o','n','e','n','t','s',0};
71 static const WCHAR szRegisterTypeLibraries[] =
72 {'R','e','g','i','s','t','e','r','T','y','p','e','L','i','b','r','a','r','i','e','s',0};
73 static const WCHAR szCreateShortcuts[] =
74 {'C','r','e','a','t','e','S','h','o','r','t','c','u','t','s',0};
75 static const WCHAR szPublishProduct[] =
76 {'P','u','b','l','i','s','h','P','r','o','d','u','c','t',0};
77 static const WCHAR szWriteIniValues[] =
78 {'W','r','i','t','e','I','n','i','V','a','l','u','e','s',0};
79 static const WCHAR szSelfRegModules[] =
80 {'S','e','l','f','R','e','g','M','o','d','u','l','e','s',0};
81 static const WCHAR szPublishFeatures[] =
82 {'P','u','b','l','i','s','h','F','e','a','t','u','r','e','s',0};
83 static const WCHAR szRegisterProduct[] =
84 {'R','e','g','i','s','t','e','r','P','r','o','d','u','c','t',0};
85 static const WCHAR szInstallExecute[] =
86 {'I','n','s','t','a','l','l','E','x','e','c','u','t','e',0};
87 static const WCHAR szInstallExecuteAgain[] =
88 {'I','n','s','t','a','l','l','E','x','e','c','u','t','e','A','g','a','i','n',0};
89 static const WCHAR szInstallFinalize[] =
90 {'I','n','s','t','a','l','l','F','i','n','a','l','i','z','e',0};
91 static const WCHAR szForceReboot[] =
92 {'F','o','r','c','e','R','e','b','o','o','t',0};
93 static const WCHAR szResolveSource[] =
94 {'R','e','s','o','l','v','e','S','o','u','r','c','e',0};
95 static const WCHAR szAppSearch[] =
96 {'A','p','p','S','e','a','r','c','h',0};
97 static const WCHAR szAllocateRegistrySpace[] =
98 {'A','l','l','o','c','a','t','e','R','e','g','i','s','t','r','y','S','p','a','c','e',0};
99 static const WCHAR szBindImage[] =
100 {'B','i','n','d','I','m','a','g','e',0};
101 static const WCHAR szCCPSearch[] =
102 {'C','C','P','S','e','a','r','c','h',0};
103 static const WCHAR szDeleteServices[] =
104 {'D','e','l','e','t','e','S','e','r','v','i','c','e','s',0};
105 static const WCHAR szDisableRollback[] =
106 {'D','i','s','a','b','l','e','R','o','l','l','b','a','c','k',0};
107 static const WCHAR szExecuteAction[] =
108 {'E','x','e','c','u','t','e','A','c','t','i','o','n',0};
109 static const WCHAR szInstallAdminPackage[] =
110 {'I','n','s','t','a','l','l','A','d','m','i','n','P','a','c','k','a','g','e',0};
111 static const WCHAR szInstallSFPCatalogFile[] =
112 {'I','n','s','t','a','l','l','S','F','P','C','a','t','a','l','o','g','F','i','l','e',0};
113 static const WCHAR szIsolateComponents[] =
114 {'I','s','o','l','a','t','e','C','o','m','p','o','n','e','n','t','s',0};
115 static const WCHAR szMigrateFeatureStates[] =
116 {'M','i','g','r','a','t','e','F','e','a','t','u','r','e','S','t','a','t','e','s',0};
117 static const WCHAR szMoveFiles[] =
118 {'M','o','v','e','F','i','l','e','s',0};
119 static const WCHAR szMsiPublishAssemblies[] =
120 {'M','s','i','P','u','b','l','i','s','h','A','s','s','e','m','b','l','i','e','s',0};
121 static const WCHAR szMsiUnpublishAssemblies[] =
122 {'M','s','i','U','n','p','u','b','l','i','s','h','A','s','s','e','m','b','l','i','e','s',0};
123 static const WCHAR szInstallODBC[] =
124 {'I','n','s','t','a','l','l','O','D','B','C',0};
125 static const WCHAR szInstallServices[] =
126 {'I','n','s','t','a','l','l','S','e','r','v','i','c','e','s',0};
127 static const WCHAR szPatchFiles[] =
128 {'P','a','t','c','h','F','i','l','e','s',0};
129 static const WCHAR szPublishComponents[] =
130 {'P','u','b','l','i','s','h','C','o','m','p','o','n','e','n','t','s',0};
131 static const WCHAR szRegisterComPlus[] =
132 {'R','e','g','i','s','t','e','r','C','o','m','P','l','u','s',0};
133 static const WCHAR szRegisterFonts[] =
134 {'R','e','g','i','s','t','e','r','F','o','n','t','s',0};
135 static const WCHAR szRegisterUser[] =
136 {'R','e','g','i','s','t','e','r','U','s','e','r',0};
137 static const WCHAR szRemoveDuplicateFiles[] =
138 {'R','e','m','o','v','e','D','u','p','l','i','c','a','t','e','F','i','l','e','s',0};
139 static const WCHAR szRemoveEnvironmentStrings[] =
140 {'R','e','m','o','v','e','E','n','v','i','r','o','n','m','e','n','t','S','t','r','i','n','g','s',0};
141 static const WCHAR szRemoveExistingProducts[] =
142 {'R','e','m','o','v','e','E','x','i','s','t','i','n','g','P','r','o','d','u','c','t','s',0};
143 static const WCHAR szRemoveFolders[] =
144 {'R','e','m','o','v','e','F','o','l','d','e','r','s',0};
145 static const WCHAR szRemoveIniValues[] =
146 {'R','e','m','o','v','e','I','n','i','V','a','l','u','e','s',0};
147 static const WCHAR szRemoveODBC[] =
148 {'R','e','m','o','v','e','O','D','B','C',0};
149 static const WCHAR szRemoveRegistryValues[] =
150 {'R','e','m','o','v','e','R','e','g','i','s','t','r','y','V','a','l','u','e','s',0};
151 static const WCHAR szRemoveShortcuts[] =
152 {'R','e','m','o','v','e','S','h','o','r','t','c','u','t','s',0};
153 static const WCHAR szRMCCPSearch[] =
154 {'R','M','C','C','P','S','e','a','r','c','h',0};
155 static const WCHAR szScheduleReboot[] =
156 {'S','c','h','e','d','u','l','e','R','e','b','o','o','t',0};
157 static const WCHAR szSelfUnregModules[] =
158 {'S','e','l','f','U','n','r','e','g','M','o','d','u','l','e','s',0};
159 static const WCHAR szSetODBCFolders[] =
160 {'S','e','t','O','D','B','C','F','o','l','d','e','r','s',0};
161 static const WCHAR szStartServices[] =
162 {'S','t','a','r','t','S','e','r','v','i','c','e','s',0};
163 static const WCHAR szStopServices[] =
164 {'S','t','o','p','S','e','r','v','i','c','e','s',0};
165 static const WCHAR szUnpublishComponents[] =
166 {'U','n','p','u','b','l','i','s','h', 'C','o','m','p','o','n','e','n','t','s',0};
167 static const WCHAR szUnpublishFeatures[] =
168 {'U','n','p','u','b','l','i','s','h','F','e','a','t','u','r','e','s',0};
169 static const WCHAR szUnregisterClassInfo[] =
170 {'U','n','r','e','g','i','s','t','e','r','C','l','a','s','s','I','n','f','o',0};
171 static const WCHAR szUnregisterComPlus[] =
172 {'U','n','r','e','g','i','s','t','e','r','C','o','m','P','l','u','s',0};
173 static const WCHAR szUnregisterExtensionInfo[] =
174 {'U','n','r','e','g','i','s','t','e','r','E','x','t','e','n','s','i','o','n','I','n','f','o',0};
175 static const WCHAR szUnregisterFonts[] =
176 {'U','n','r','e','g','i','s','t','e','r','F','o','n','t','s',0};
177 static const WCHAR szUnregisterMIMEInfo[] =
178 {'U','n','r','e','g','i','s','t','e','r','M','I','M','E','I','n','f','o',0};
179 static const WCHAR szUnregisterProgIdInfo[] =
180 {'U','n','r','e','g','i','s','t','e','r','P','r','o','g','I','d','I','n','f','o',0};
181 static const WCHAR szUnregisterTypeLibraries[] =
182 {'U','n','r','e','g','i','s','t','e','r','T','y','p','e','L','i','b','r','a','r','i','e','s',0};
183 static const WCHAR szValidateProductID[] =
184 {'V','a','l','i','d','a','t','e','P','r','o','d','u','c','t','I','D',0};
185 static const WCHAR szWriteEnvironmentStrings[] =
186 {'W','r','i','t','e','E','n','v','i','r','o','n','m','e','n','t','S','t','r','i','n','g','s',0};
187
188 /********************************************************
189 * helper functions
190 ********************************************************/
191
192 static void ui_actionstart(MSIPACKAGE *package, LPCWSTR action)
193 {
194 static const WCHAR Query_t[] =
195 {'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
196 '`','A','c','t','i','o', 'n','T','e','x','t','`',' ',
197 'W','H','E','R','E', ' ','`','A','c','t','i','o','n','`',' ','=',
198 ' ','\'','%','s','\'',0};
199 MSIRECORD * row;
200
201 row = MSI_QueryGetRecord( package->db, Query_t, action );
202 if (!row)
203 return;
204 MSI_ProcessMessage(package, INSTALLMESSAGE_ACTIONSTART, row);
205 msiobj_release(&row->hdr);
206 }
207
208 static void ui_actioninfo(MSIPACKAGE *package, LPCWSTR action, BOOL start,
209 UINT rc)
210 {
211 MSIRECORD * row;
212 static const WCHAR template_s[]=
213 {'A','c','t','i','o','n',' ','s','t','a','r','t',' ','%','s',':',' ',
214 '%','s', '.',0};
215 static const WCHAR template_e[]=
216 {'A','c','t','i','o','n',' ','e','n','d','e','d',' ','%','s',':',' ',
217 '%','s', '.',' ','R','e','t','u','r','n',' ','v','a','l','u','e',' ',
218 '%','i','.',0};
219 static const WCHAR format[] =
220 {'H','H','\'',':','\'','m','m','\'',':','\'','s','s',0};
221 WCHAR message[1024];
222 WCHAR timet[0x100];
223
224 GetTimeFormatW(LOCALE_USER_DEFAULT, 0, NULL, format, timet, 0x100);
225 if (start)
226 sprintfW(message,template_s,timet,action);
227 else
228 sprintfW(message,template_e,timet,action,rc);
229
230 row = MSI_CreateRecord(1);
231 MSI_RecordSetStringW(row,1,message);
232
233 MSI_ProcessMessage(package, INSTALLMESSAGE_INFO, row);
234 msiobj_release(&row->hdr);
235 }
236
237 UINT msi_parse_command_line( MSIPACKAGE *package, LPCWSTR szCommandLine,
238 BOOL preserve_case )
239 {
240 LPCWSTR ptr,ptr2;
241 BOOL quote;
242 DWORD len;
243 LPWSTR prop = NULL, val = NULL;
244
245 if (!szCommandLine)
246 return ERROR_SUCCESS;
247
248 ptr = szCommandLine;
249
250 while (*ptr)
251 {
252 if (*ptr==' ')
253 {
254 ptr++;
255 continue;
256 }
257
258 TRACE("Looking at %s\n",debugstr_w(ptr));
259
260 ptr2 = strchrW(ptr,'=');
261 if (!ptr2)
262 {
263 ERR("command line contains unknown string : %s\n", debugstr_w(ptr));
264 break;
265 }
266
267 quote = FALSE;
268
269 len = ptr2-ptr;
270 prop = msi_alloc((len+1)*sizeof(WCHAR));
271 memcpy(prop,ptr,len*sizeof(WCHAR));
272 prop[len]=0;
273
274 if (!preserve_case)
275 struprW(prop);
276
277 ptr2++;
278
279 len = 0;
280 ptr = ptr2;
281 while (*ptr && (quote || (!quote && *ptr!=' ')))
282 {
283 if (*ptr == '"')
284 quote = !quote;
285 ptr++;
286 len++;
287 }
288
289 if (*ptr2=='"')
290 {
291 ptr2++;
292 len -= 2;
293 }
294 val = msi_alloc((len+1)*sizeof(WCHAR));
295 memcpy(val,ptr2,len*sizeof(WCHAR));
296 val[len] = 0;
297
298 if (lstrlenW(prop) > 0)
299 {
300 TRACE("Found commandline property (%s) = (%s)\n",
301 debugstr_w(prop), debugstr_w(val));
302 MSI_SetPropertyW(package,prop,val);
303 }
304 msi_free(val);
305 msi_free(prop);
306 }
307
308 return ERROR_SUCCESS;
309 }
310
311
312 static LPWSTR* msi_split_string( LPCWSTR str, WCHAR sep )
313 {
314 LPCWSTR pc;
315 LPWSTR p, *ret = NULL;
316 UINT count = 0;
317
318 if (!str)
319 return ret;
320
321 /* count the number of substrings */
322 for ( pc = str, count = 0; pc; count++ )
323 {
324 pc = strchrW( pc, sep );
325 if (pc)
326 pc++;
327 }
328
329 /* allocate space for an array of substring pointers and the substrings */
330 ret = msi_alloc( (count+1) * sizeof (LPWSTR) +
331 (lstrlenW(str)+1) * sizeof(WCHAR) );
332 if (!ret)
333 return ret;
334
335 /* copy the string and set the pointers */
336 p = (LPWSTR) &ret[count+1];
337 lstrcpyW( p, str );
338 for( count = 0; (ret[count] = p); count++ )
339 {
340 p = strchrW( p, sep );
341 if (p)
342 *p++ = 0;
343 }
344
345 return ret;
346 }
347
348 static UINT msi_check_transform_applicable( MSIPACKAGE *package, IStorage *patch )
349 {
350 static const WCHAR szSystemLanguageID[] =
351 { 'S','y','s','t','e','m','L','a','n','g','u','a','g','e','I','D',0 };
352
353 LPWSTR prod_code, patch_product, langid = NULL, template = NULL;
354 UINT ret = ERROR_FUNCTION_FAILED;
355
356 prod_code = msi_dup_property( package, szProductCode );
357 patch_product = msi_get_suminfo_product( patch );
358
359 TRACE("db = %s patch = %s\n", debugstr_w(prod_code), debugstr_w(patch_product));
360
361 if ( strstrW( patch_product, prod_code ) )
362 {
363 MSISUMMARYINFO *si;
364 const WCHAR *p;
365
366 si = MSI_GetSummaryInformationW( patch, 0 );
367 if (!si)
368 {
369 ERR("no summary information!\n");
370 goto end;
371 }
372
373 template = msi_suminfo_dup_string( si, PID_TEMPLATE );
374 if (!template)
375 {
376 ERR("no template property!\n");
377 msiobj_release( &si->hdr );
378 goto end;
379 }
380
381 if (!template[0])
382 {
383 ret = ERROR_SUCCESS;
384 msiobj_release( &si->hdr );
385 goto end;
386 }
387
388 langid = msi_dup_property( package, szSystemLanguageID );
389 if (!langid)
390 {
391 msiobj_release( &si->hdr );
392 goto end;
393 }
394
395 p = strchrW( template, ';' );
396 if (p && (!strcmpW( p + 1, langid ) || !strcmpW( p + 1, szZero )))
397 {
398 TRACE("applicable transform\n");
399 ret = ERROR_SUCCESS;
400 }
401
402 /* FIXME: check platform */
403
404 msiobj_release( &si->hdr );
405 }
406
407 end:
408 msi_free( patch_product );
409 msi_free( prod_code );
410 msi_free( template );
411 msi_free( langid );
412
413 return ret;
414 }
415
416 static UINT msi_apply_substorage_transform( MSIPACKAGE *package,
417 MSIDATABASE *patch_db, LPCWSTR name )
418 {
419 UINT ret = ERROR_FUNCTION_FAILED;
420 IStorage *stg = NULL;
421 HRESULT r;
422
423 TRACE("%p %s\n", package, debugstr_w(name) );
424
425 if (*name++ != ':')
426 {
427 ERR("expected a colon in %s\n", debugstr_w(name));
428 return ERROR_FUNCTION_FAILED;
429 }
430
431 r = IStorage_OpenStorage( patch_db->storage, name, NULL, STGM_SHARE_EXCLUSIVE, NULL, 0, &stg );
432 if (SUCCEEDED(r))
433 {
434 ret = msi_check_transform_applicable( package, stg );
435 if (ret == ERROR_SUCCESS)
436 msi_table_apply_transform( package->db, stg );
437 else
438 TRACE("substorage transform %s wasn't applicable\n", debugstr_w(name));
439 IStorage_Release( stg );
440 }
441 else
442 ERR("failed to open substorage %s\n", debugstr_w(name));
443
444 return ERROR_SUCCESS;
445 }
446
447 UINT msi_check_patch_applicable( MSIPACKAGE *package, MSISUMMARYINFO *si )
448 {
449 LPWSTR guid_list, *guids, product_code;
450 UINT i, ret = ERROR_FUNCTION_FAILED;
451
452 product_code = msi_dup_property( package, szProductCode );
453 if (!product_code)
454 {
455 /* FIXME: the property ProductCode should be written into the DB somewhere */
456 ERR("no product code to check\n");
457 return ERROR_SUCCESS;
458 }
459
460 guid_list = msi_suminfo_dup_string( si, PID_TEMPLATE );
461 guids = msi_split_string( guid_list, ';' );
462 for ( i = 0; guids[i] && ret != ERROR_SUCCESS; i++ )
463 {
464 if (!lstrcmpW( guids[i], product_code ))
465 ret = ERROR_SUCCESS;
466 }
467 msi_free( guids );
468 msi_free( guid_list );
469 msi_free( product_code );
470
471 return ret;
472 }
473
474 static UINT msi_set_media_source_prop(MSIPACKAGE *package)
475 {
476 MSIQUERY *view;
477 MSIRECORD *rec = NULL;
478 LPWSTR patch;
479 LPCWSTR prop;
480 UINT r;
481
482 static const WCHAR query[] = {'S','E','L','E','C','T',' ',
483 '`','S','o','u','r','c','e','`',' ','F','R','O','M',' ',
484 '`','M','e','d','i','a','`',' ','W','H','E','R','E',' ',
485 '`','S','o','u','r','c','e','`',' ','I','S',' ',
486 'N','O','T',' ','N','U','L','L',0};
487
488 r = MSI_DatabaseOpenViewW(package->db, query, &view);
489 if (r != ERROR_SUCCESS)
490 return r;
491
492 r = MSI_ViewExecute(view, 0);
493 if (r != ERROR_SUCCESS)
494 goto done;
495
496 if (MSI_ViewFetch(view, &rec) == ERROR_SUCCESS)
497 {
498 prop = MSI_RecordGetString(rec, 1);
499 patch = msi_dup_property(package, szPatch);
500 MSI_SetPropertyW(package, prop, patch);
501 msi_free(patch);
502 }
503
504 done:
505 if (rec) msiobj_release(&rec->hdr);
506 msiobj_release(&view->hdr);
507
508 return r;
509 }
510
511 static UINT msi_parse_patch_summary( MSIPACKAGE *package, MSIDATABASE *patch_db )
512 {
513 MSISUMMARYINFO *si;
514 LPWSTR str, *substorage;
515 UINT i, r = ERROR_SUCCESS;
516
517 si = MSI_GetSummaryInformationW( patch_db->storage, 0 );
518 if (!si)
519 return ERROR_FUNCTION_FAILED;
520
521 if (msi_check_patch_applicable( package, si ) != ERROR_SUCCESS)
522 {
523 TRACE("Patch not applicable\n");
524 return ERROR_SUCCESS;
525 }
526
527 package->patch = msi_alloc(sizeof(MSIPATCHINFO));
528 if (!package->patch)
529 return ERROR_OUTOFMEMORY;
530
531 package->patch->patchcode = msi_suminfo_dup_string(si, PID_REVNUMBER);
532 if (!package->patch->patchcode)
533 return ERROR_OUTOFMEMORY;
534
535 /* enumerate the substorage */
536 str = msi_suminfo_dup_string( si, PID_LASTAUTHOR );
537 package->patch->transforms = str;
538
539 substorage = msi_split_string( str, ';' );
540 for ( i = 0; substorage && substorage[i] && r == ERROR_SUCCESS; i++ )
541 r = msi_apply_substorage_transform( package, patch_db, substorage[i] );
542
543 msi_free( substorage );
544 msiobj_release( &si->hdr );
545
546 msi_set_media_source_prop(package);
547
548 return r;
549 }
550
551 static UINT msi_apply_patch_package( MSIPACKAGE *package, LPCWSTR file )
552 {
553 MSIDATABASE *patch_db = NULL;
554 UINT r;
555
556 TRACE("%p %s\n", package, debugstr_w( file ) );
557
558 /* FIXME:
559 * We probably want to make sure we only open a patch collection here.
560 * Patch collections (.msp) and databases (.msi) have different GUIDs
561 * but currently MSI_OpenDatabaseW will accept both.
562 */
563 r = MSI_OpenDatabaseW( file, MSIDBOPEN_READONLY, &patch_db );
564 if ( r != ERROR_SUCCESS )
565 {
566 ERR("failed to open patch collection %s\n", debugstr_w( file ) );
567 return r;
568 }
569
570 msi_parse_patch_summary( package, patch_db );
571
572 /*
573 * There might be a CAB file in the patch package,
574 * so append it to the list of storage to search for streams.
575 */
576 append_storage_to_db( package->db, patch_db->storage );
577
578 msiobj_release( &patch_db->hdr );
579
580 return ERROR_SUCCESS;
581 }
582
583 /* get the PATCH property, and apply all the patches it specifies */
584 static UINT msi_apply_patches( MSIPACKAGE *package )
585 {
586 LPWSTR patch_list, *patches;
587 UINT i, r = ERROR_SUCCESS;
588
589 patch_list = msi_dup_property( package, szPatch );
590
591 TRACE("patches to be applied: %s\n", debugstr_w( patch_list ) );
592
593 patches = msi_split_string( patch_list, ';' );
594 for( i=0; patches && patches[i] && r == ERROR_SUCCESS; i++ )
595 r = msi_apply_patch_package( package, patches[i] );
596
597 msi_free( patches );
598 msi_free( patch_list );
599
600 return r;
601 }
602
603 static UINT msi_apply_transforms( MSIPACKAGE *package )
604 {
605 static const WCHAR szTransforms[] = {
606 'T','R','A','N','S','F','O','R','M','S',0 };
607 LPWSTR xform_list, *xforms;
608 UINT i, r = ERROR_SUCCESS;
609
610 xform_list = msi_dup_property( package, szTransforms );
611 xforms = msi_split_string( xform_list, ';' );
612
613 for( i=0; xforms && xforms[i] && r == ERROR_SUCCESS; i++ )
614 {
615 if (xforms[i][0] == ':')
616 r = msi_apply_substorage_transform( package, package->db, xforms[i] );
617 else
618 r = MSI_DatabaseApplyTransformW( package->db, xforms[i], 0 );
619 }
620
621 msi_free( xforms );
622 msi_free( xform_list );
623
624 return r;
625 }
626
627 static BOOL ui_sequence_exists( MSIPACKAGE *package )
628 {
629 MSIQUERY *view;
630 UINT rc;
631
632 static const WCHAR ExecSeqQuery [] =
633 {'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
634 '`','I','n','s','t','a','l','l',
635 'U','I','S','e','q','u','e','n','c','e','`',
636 ' ','W','H','E','R','E',' ',
637 '`','S','e','q','u','e','n','c','e','`',' ',
638 '>',' ','0',' ','O','R','D','E','R',' ','B','Y',' ',
639 '`','S','e','q','u','e','n','c','e','`',0};
640
641 rc = MSI_DatabaseOpenViewW(package->db, ExecSeqQuery, &view);
642 if (rc == ERROR_SUCCESS)
643 {
644 msiobj_release(&view->hdr);
645 return TRUE;
646 }
647
648 return FALSE;
649 }
650
651 static UINT msi_set_sourcedir_props(MSIPACKAGE *package, BOOL replace)
652 {
653 LPWSTR p, db;
654 LPWSTR source, check;
655 DWORD len;
656
657 static const WCHAR szOriginalDatabase[] =
658 {'O','r','i','g','i','n','a','l','D','a','t','a','b','a','s','e',0};
659
660 db = msi_dup_property( package, szOriginalDatabase );
661 if (!db)
662 return ERROR_OUTOFMEMORY;
663
664 p = strrchrW( db, '\\' );
665 if (!p)
666 {
667 p = strrchrW( db, '/' );
668 if (!p)
669 {
670 msi_free(db);
671 return ERROR_SUCCESS;
672 }
673 }
674
675 len = p - db + 2;
676 source = msi_alloc( len * sizeof(WCHAR) );
677 lstrcpynW( source, db, len );
678
679 check = msi_dup_property( package, cszSourceDir );
680 if (!check || replace)
681 MSI_SetPropertyW( package, cszSourceDir, source );
682
683 msi_free( check );
684
685 check = msi_dup_property( package, cszSOURCEDIR );
686 if (!check || replace)
687 MSI_SetPropertyW( package, cszSOURCEDIR, source );
688
689 msi_free( check );
690 msi_free( source );
691 msi_free( db );
692
693 return ERROR_SUCCESS;
694 }
695
696 static BOOL needs_ui_sequence(MSIPACKAGE *package)
697 {
698 INT level = msi_get_property_int(package, szUILevel, 0);
699 return (level & INSTALLUILEVEL_MASK) >= INSTALLUILEVEL_REDUCED;
700 }
701
702 static UINT msi_set_context(MSIPACKAGE *package)
703 {
704 WCHAR val[10];
705 DWORD sz = 10;
706 DWORD num;
707 UINT r;
708
709 package->Context = MSIINSTALLCONTEXT_USERUNMANAGED;
710
711 r = MSI_GetPropertyW(package, szAllUsers, val, &sz);
712 if (r == ERROR_SUCCESS)
713 {
714 num = atolW(val);
715 if (num == 1 || num == 2)
716 package->Context = MSIINSTALLCONTEXT_MACHINE;
717 }
718
719 return ERROR_SUCCESS;
720 }
721
722 static UINT ITERATE_Actions(MSIRECORD *row, LPVOID param)
723 {
724 UINT rc;
725 LPCWSTR cond, action;
726 MSIPACKAGE *package = param;
727
728 action = MSI_RecordGetString(row,1);
729 if (!action)
730 {
731 ERR("Error is retrieving action name\n");
732 return ERROR_FUNCTION_FAILED;
733 }
734
735 /* check conditions */
736 cond = MSI_RecordGetString(row,2);
737
738 /* this is a hack to skip errors in the condition code */
739 if (MSI_EvaluateConditionW(package, cond) == MSICONDITION_FALSE)
740 {
741 TRACE("Skipping action: %s (condition is false)\n", debugstr_w(action));
742 return ERROR_SUCCESS;
743 }
744
745 if (needs_ui_sequence(package))
746 rc = ACTION_PerformUIAction(package, action, -1);
747 else
748 rc = ACTION_PerformAction(package, action, -1, FALSE);
749
750 msi_dialog_check_messages( NULL );
751
752 if (package->CurrentInstallState != ERROR_SUCCESS)
753 rc = package->CurrentInstallState;
754
755 if (rc == ERROR_FUNCTION_NOT_CALLED)
756 rc = ERROR_SUCCESS;
757
758 if (rc != ERROR_SUCCESS)
759 ERR("Execution halted, action %s returned %i\n", debugstr_w(action), rc);
760
761 return rc;
762 }
763
764 UINT MSI_Sequence( MSIPACKAGE *package, LPCWSTR szTable, INT iSequenceMode )
765 {
766 MSIQUERY * view;
767 UINT r;
768 static const WCHAR query[] =
769 {'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
770 '`','%','s','`',
771 ' ','W','H','E','R','E',' ',
772 '`','S','e','q','u','e','n','c','e','`',' ',
773 '>',' ','0',' ','O','R','D','E','R',' ','B','Y',' ',
774 '`','S','e','q','u','e','n','c','e','`',0};
775
776 TRACE("%p %s %i\n", package, debugstr_w(szTable), iSequenceMode );
777
778 r = MSI_OpenQuery( package->db, &view, query, szTable );
779 if (r == ERROR_SUCCESS)
780 {
781 r = MSI_IterateRecords( view, NULL, ITERATE_Actions, package );
782 msiobj_release(&view->hdr);
783 }
784
785 return r;
786 }
787
788 static UINT ACTION_ProcessExecSequence(MSIPACKAGE *package, BOOL UIran)
789 {
790 MSIQUERY * view;
791 UINT rc;
792 static const WCHAR ExecSeqQuery[] =
793 {'S','E','L','E','C','T',' ','*',' ', 'F','R','O','M',' ',
794 '`','I','n','s','t','a','l','l','E','x','e','c','u','t','e',
795 'S','e','q','u','e','n','c','e','`',' ', 'W','H','E','R','E',' ',
796 '`','S','e','q','u','e','n','c','e','`',' ', '>',' ','%','i',' ',
797 'O','R','D','E','R',' ', 'B','Y',' ',
798 '`','S','e','q','u','e','n','c','e','`',0 };
799 static const WCHAR IVQuery[] =
800 {'S','E','L','E','C','T',' ','`','S','e','q','u','e','n','c','e','`',
801 ' ', 'F','R','O','M',' ','`','I','n','s','t','a','l','l',
802 'E','x','e','c','u','t','e','S','e','q','u','e','n','c','e','`',' ',
803 'W','H','E','R','E',' ','`','A','c','t','i','o','n','`',' ','=',
804 ' ','\'', 'I','n','s','t','a','l','l',
805 'V','a','l','i','d','a','t','e','\'', 0};
806 INT seq = 0;
807
808 if (package->script->ExecuteSequenceRun)
809 {
810 TRACE("Execute Sequence already Run\n");
811 return ERROR_SUCCESS;
812 }
813
814 package->script->ExecuteSequenceRun = TRUE;
815
816 /* get the sequence number */
817 if (UIran)
818 {
819 MSIRECORD *row = MSI_QueryGetRecord(package->db, IVQuery);
820 if( !row )
821 return ERROR_FUNCTION_FAILED;
822 seq = MSI_RecordGetInteger(row,1);
823 msiobj_release(&row->hdr);
824 }
825
826 rc = MSI_OpenQuery(package->db, &view, ExecSeqQuery, seq);
827 if (rc == ERROR_SUCCESS)
828 {
829 TRACE("Running the actions\n");
830
831 rc = MSI_IterateRecords(view, NULL, ITERATE_Actions, package);
832 msiobj_release(&view->hdr);
833 }
834
835 return rc;
836 }
837
838 static UINT ACTION_ProcessUISequence(MSIPACKAGE *package)
839 {
840 MSIQUERY * view;
841 UINT rc;
842 static const WCHAR ExecSeqQuery [] =
843 {'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
844 '`','I','n','s','t','a','l','l',
845 'U','I','S','e','q','u','e','n','c','e','`',
846 ' ','W','H','E','R','E',' ',
847 '`','S','e','q','u','e','n','c','e','`',' ',
848 '>',' ','0',' ','O','R','D','E','R',' ','B','Y',' ',
849 '`','S','e','q','u','e','n','c','e','`',0};
850
851 rc = MSI_DatabaseOpenViewW(package->db, ExecSeqQuery, &view);
852 if (rc == ERROR_SUCCESS)
853 {
854 TRACE("Running the actions\n");
855
856 rc = MSI_IterateRecords(view, NULL, ITERATE_Actions, package);
857 msiobj_release(&view->hdr);
858 }
859
860 return rc;
861 }
862
863 /********************************************************
864 * ACTION helper functions and functions that perform the actions
865 *******************************************************/
866 static BOOL ACTION_HandleCustomAction( MSIPACKAGE* package, LPCWSTR action,
867 UINT* rc, UINT script, BOOL force )
868 {
869 BOOL ret=FALSE;
870 UINT arc;
871
872 arc = ACTION_CustomAction(package, action, script, force);
873
874 if (arc != ERROR_CALL_NOT_IMPLEMENTED)
875 {
876 *rc = arc;
877 ret = TRUE;
878 }
879 return ret;
880 }
881
882 /*
883 * Actual Action Handlers
884 */
885
886 static UINT ITERATE_CreateFolders(MSIRECORD *row, LPVOID param)
887 {
888 MSIPACKAGE *package = param;
889 LPCWSTR dir;
890 LPWSTR full_path;
891 MSIRECORD *uirow;
892 MSIFOLDER *folder;
893
894 dir = MSI_RecordGetString(row,1);
895 if (!dir)
896 {
897 ERR("Unable to get folder id\n");
898 return ERROR_SUCCESS;
899 }
900
901 full_path = resolve_folder(package,dir,FALSE,FALSE,TRUE,&folder);
902 if (!full_path)
903 {
904 ERR("Unable to resolve folder id %s\n",debugstr_w(dir));
905 return ERROR_SUCCESS;
906 }
907
908 TRACE("Folder is %s\n",debugstr_w(full_path));
909
910 /* UI stuff */
911 uirow = MSI_CreateRecord(1);
912 MSI_RecordSetStringW(uirow,1,full_path);
913 ui_actiondata(package,szCreateFolders,uirow);
914 msiobj_release( &uirow->hdr );
915
916 if (folder->State == 0)
917 create_full_pathW(full_path);
918
919 folder->State = 3;
920
921 msi_free(full_path);
922 return ERROR_SUCCESS;
923 }
924
925 /* FIXME: probably should merge this with the above function */
926 static UINT msi_create_directory( MSIPACKAGE* package, LPCWSTR dir )
927 {
928 UINT rc = ERROR_SUCCESS;
929 MSIFOLDER *folder;
930 LPWSTR install_path;
931
932 install_path = resolve_folder(package, dir, FALSE, FALSE, TRUE, &folder);
933 if (!install_path)
934 return ERROR_FUNCTION_FAILED;
935
936 /* create the path */
937 if (folder->State == 0)
938 {
939 create_full_pathW(install_path);
940 folder->State = 2;
941 }
942 msi_free(install_path);
943
944 return rc;
945 }
946
947 UINT msi_create_component_directories( MSIPACKAGE *package )
948 {
949 MSICOMPONENT *comp;
950
951 /* create all the folders required by the components are going to install */
952 LIST_FOR_EACH_ENTRY( comp, &package->components, MSICOMPONENT, entry )
953 {
954 if (!ACTION_VerifyComponentForAction( comp, INSTALLSTATE_LOCAL))
955 continue;
956 msi_create_directory( package, comp->Directory );
957 }
958
959 return ERROR_SUCCESS;
960 }
961
962 static UINT ACTION_CreateFolders(MSIPACKAGE *package)
963 {
964 static const WCHAR ExecSeqQuery[] =
965 {'S','E','L','E','C','T',' ',
966 '`','D','i','r','e','c','t','o','r','y','_','`',
967 ' ','F','R','O','M',' ',
968 '`','C','r','e','a','t','e','F','o','l','d','e','r','`',0 };
969 UINT rc;
970 MSIQUERY *view;
971
972 /* create all the empty folders specified in the CreateFolder table */
973 rc = MSI_DatabaseOpenViewW(package->db, ExecSeqQuery, &view );
974 if (rc != ERROR_SUCCESS)
975 return ERROR_SUCCESS;
976
977 rc = MSI_IterateRecords(view, NULL, ITERATE_CreateFolders, package);
978 msiobj_release(&view->hdr);
979
980 return rc;
981 }
982
983 static UINT ITERATE_RemoveFolders( MSIRECORD *row, LPVOID param )
984 {
985 MSIPACKAGE *package = param;
986 LPCWSTR dir;
987 LPWSTR full_path;
988 MSIRECORD *uirow;
989 MSIFOLDER *folder;
990
991 dir = MSI_RecordGetString( row, 1 );
992 if (!dir)
993 {
994 ERR("Unable to get folder id\n");
995 return ERROR_SUCCESS;
996 }
997
998 full_path = resolve_folder( package, dir, FALSE, FALSE, TRUE, &folder );
999 if (!full_path)
1000 {
1001 ERR("Unable to resolve folder id %s\n", debugstr_w(dir));
1002 return ERROR_SUCCESS;
1003 }
1004
1005 TRACE("folder is %s\n", debugstr_w(full_path));
1006
1007 uirow = MSI_CreateRecord( 1 );
1008 MSI_RecordSetStringW( uirow, 1, full_path );
1009 ui_actiondata( package, szRemoveFolders, uirow );
1010 msiobj_release( &uirow->hdr );
1011
1012 RemoveDirectoryW( full_path );
1013 folder->State = 0;
1014
1015 msi_free( full_path );
1016 return ERROR_SUCCESS;
1017 }
1018
1019 static UINT ACTION_RemoveFolders( MSIPACKAGE *package )
1020 {
1021 static const WCHAR query[] =
1022 {'S','E','L','E','C','T',' ', '`','D','i','r','e','c','t','o','r','y','_','`',
1023 ' ','F','R','O','M',' ', '`','C','r','e','a','t','e','F','o','l','d','e','r','`',0};
1024
1025 MSIQUERY *view;
1026 UINT rc;
1027
1028 rc = MSI_DatabaseOpenViewW( package->db, query, &view );
1029 if (rc != ERROR_SUCCESS)
1030 return ERROR_SUCCESS;
1031
1032 rc = MSI_IterateRecords( view, NULL, ITERATE_RemoveFolders, package );
1033 msiobj_release( &view->hdr );
1034
1035 return rc;
1036 }
1037
1038 static UINT load_component( MSIRECORD *row, LPVOID param )
1039 {
1040 MSIPACKAGE *package = param;
1041 MSICOMPONENT *comp;
1042
1043 comp = msi_alloc_zero( sizeof(MSICOMPONENT) );
1044 if (!comp)
1045 return ERROR_FUNCTION_FAILED;
1046
1047 list_add_tail( &package->components, &comp->entry );
1048
1049 /* fill in the data */
1050 comp->Component = msi_dup_record_field( row, 1 );
1051
1052 TRACE("Loading Component %s\n", debugstr_w(comp->Component));
1053
1054 comp->ComponentId = msi_dup_record_field( row, 2 );
1055 comp->Directory = msi_dup_record_field( row, 3 );
1056 comp->Attributes = MSI_RecordGetInteger(row,4);
1057 comp->Condition = msi_dup_record_field( row, 5 );
1058 comp->KeyPath = msi_dup_record_field( row, 6 );
1059
1060 comp->Installed = INSTALLSTATE_UNKNOWN;
1061 msi_component_set_state(package, comp, INSTALLSTATE_UNKNOWN);
1062
1063 return ERROR_SUCCESS;
1064 }
1065
1066 static UINT load_all_components( MSIPACKAGE *package )
1067 {
1068 static const WCHAR query[] = {
1069 'S','E','L','E','C','T',' ','*',' ','F','R', 'O','M',' ',
1070 '`','C','o','m','p','o','n','e','n','t','`',0 };
1071 MSIQUERY *view;
1072 UINT r;
1073
1074 if (!list_empty(&package->components))
1075 return ERROR_SUCCESS;
1076
1077 r = MSI_DatabaseOpenViewW( package->db, query, &view );
1078 if (r != ERROR_SUCCESS)
1079 return r;
1080
1081 r = MSI_IterateRecords(view, NULL, load_component, package);
1082 msiobj_release(&view->hdr);
1083 return r;
1084 }
1085
1086 typedef struct {
1087 MSIPACKAGE *package;
1088 MSIFEATURE *feature;
1089 } _ilfs;
1090
1091 static UINT add_feature_component( MSIFEATURE *feature, MSICOMPONENT *comp )
1092 {
1093 ComponentList *cl;
1094
1095 cl = msi_alloc( sizeof (*cl) );
1096 if ( !cl )
1097 return ERROR_NOT_ENOUGH_MEMORY;
1098 cl->component = comp;
1099 list_add_tail( &feature->Components, &cl->entry );
1100
1101 return ERROR_SUCCESS;
1102 }
1103
1104 static UINT add_feature_child( MSIFEATURE *parent, MSIFEATURE *child )
1105 {
1106 FeatureList *fl;
1107
1108 fl = msi_alloc( sizeof(*fl) );
1109 if ( !fl )
1110 return ERROR_NOT_ENOUGH_MEMORY;
1111 fl->feature = child;
1112 list_add_tail( &parent->Children, &fl->entry );
1113
1114 return ERROR_SUCCESS;
1115 }
1116
1117 static UINT iterate_load_featurecomponents(MSIRECORD *row, LPVOID param)
1118 {
1119 _ilfs* ilfs = param;
1120 LPCWSTR component;
1121 MSICOMPONENT *comp;
1122
1123 component = MSI_RecordGetString(row,1);
1124
1125 /* check to see if the component is already loaded */
1126 comp = get_loaded_component( ilfs->package, component );
1127 if (!comp)
1128 {
1129 ERR("unknown component %s\n", debugstr_w(component));
1130 return ERROR_FUNCTION_FAILED;
1131 }
1132
1133 add_feature_component( ilfs->feature, comp );
1134 comp->Enabled = TRUE;
1135
1136 return ERROR_SUCCESS;
1137 }
1138
1139 static MSIFEATURE *find_feature_by_name( MSIPACKAGE *package, LPCWSTR name )
1140 {
1141 MSIFEATURE *feature;
1142
1143 if ( !name )
1144 return NULL;
1145
1146 LIST_FOR_EACH_ENTRY( feature, &package->features, MSIFEATURE, entry )
1147 {
1148 if ( !lstrcmpW( feature->Feature, name ) )
1149 return feature;
1150 }
1151
1152 return NULL;
1153 }
1154
1155 static UINT load_feature(MSIRECORD * row, LPVOID param)
1156 {
1157 MSIPACKAGE* package = param;
1158 MSIFEATURE* feature;
1159 static const WCHAR Query1[] =
1160 {'S','E','L','E','C','T',' ',
1161 '`','C','o','m','p','o','n','e','n','t','_','`',
1162 ' ','F','R','O','M',' ','`','F','e','a','t','u','r','e',
1163 'C','o','m','p','o','n','e','n','t','s','`',' ',
1164 'W','H','E','R','E',' ',
1165 '`','F','e', 'a','t','u','r','e','_','`',' ','=','\'','%','s','\'',0};
1166 MSIQUERY * view;
1167 UINT rc;
1168 _ilfs ilfs;
1169
1170 /* fill in the data */
1171
1172 feature = msi_alloc_zero( sizeof (MSIFEATURE) );
1173 if (!feature)
1174 return ERROR_NOT_ENOUGH_MEMORY;
1175
1176 list_init( &feature->Children );
1177 list_init( &feature->Components );
1178
1179 feature->Feature = msi_dup_record_field( row, 1 );
1180
1181 TRACE("Loading feature %s\n",debugstr_w(feature->Feature));
1182
1183 feature->Feature_Parent = msi_dup_record_field( row, 2 );
1184 feature->Title = msi_dup_record_field( row, 3 );
1185 feature->Description = msi_dup_record_field( row, 4 );
1186
1187 if (!MSI_RecordIsNull(row,5))
1188 feature->Display = MSI_RecordGetInteger(row,5);
1189
1190 feature->Level= MSI_RecordGetInteger(row,6);
1191 feature->Directory = msi_dup_record_field( row, 7 );
1192 feature->Attributes = MSI_RecordGetInteger(row,8);
1193
1194 feature->Installed = INSTALLSTATE_UNKNOWN;
1195 msi_feature_set_state(package, feature, INSTALLSTATE_UNKNOWN);
1196
1197 list_add_tail( &package->features, &feature->entry );
1198
1199 /* load feature components */
1200
1201 rc = MSI_OpenQuery( package->db, &view, Query1, feature->Feature );
1202 if (rc != ERROR_SUCCESS)
1203 return ERROR_SUCCESS;
1204
1205 ilfs.package = package;
1206 ilfs.feature = feature;
1207
1208 MSI_IterateRecords(view, NULL, iterate_load_featurecomponents , &ilfs);
1209 msiobj_release(&view->hdr);
1210
1211 return ERROR_SUCCESS;
1212 }
1213
1214 static UINT find_feature_children(MSIRECORD * row, LPVOID param)
1215 {
1216 MSIPACKAGE* package = param;
1217 MSIFEATURE *parent, *child;
1218
1219 child = find_feature_by_name( package, MSI_RecordGetString( row, 1 ) );
1220 if (!child)
1221 return ERROR_FUNCTION_FAILED;
1222
1223 if (!child->Feature_Parent)
1224 return ERROR_SUCCESS;
1225
1226 parent = find_feature_by_name( package, child->Feature_Parent );
1227 if (!parent)
1228 return ERROR_FUNCTION_FAILED;
1229
1230 add_feature_child( parent, child );
1231 return ERROR_SUCCESS;
1232 }
1233
1234 static UINT load_all_features( MSIPACKAGE *package )
1235 {
1236 static const WCHAR query[] = {
1237 'S','E','L','E','C','T',' ','*',' ', 'F','R','O','M',' ',
1238 '`','F','e','a','t','u','r','e','`',' ','O','R','D','E','R',
1239 ' ','B','Y',' ','`','D','i','s','p','l','a','y','`',0};
1240 MSIQUERY *view;
1241 UINT r;
1242
1243 if (!list_empty(&package->features))
1244 return ERROR_SUCCESS;
1245
1246 r = MSI_DatabaseOpenViewW( package->db, query, &view );
1247 if (r != ERROR_SUCCESS)
1248 return r;
1249
1250 r = MSI_IterateRecords( view, NULL, load_feature, package );
1251 if (r != ERROR_SUCCESS)
1252 return r;
1253
1254 r = MSI_IterateRecords( view, NULL, find_feature_children, package );
1255 msiobj_release( &view->hdr );
1256
1257 return r;
1258 }
1259
1260 static LPWSTR folder_split_path(LPWSTR p, WCHAR ch)
1261 {
1262 if (!p)
1263 return p;
1264 p = strchrW(p, ch);
1265 if (!p)
1266 return p;
1267 *p = 0;
1268 return p+1;
1269 }
1270
1271 static UINT load_file_hash(MSIPACKAGE *package, MSIFILE *file)
1272 {
1273 static const WCHAR query[] = {
1274 'S','E','L','E','C','T',' ','*',' ', 'F','R','O','M',' ',
1275 '`','M','s','i','F','i','l','e','H','a','s','h','`',' ',
1276 'W','H','E','R','E',' ','`','F','i','l','e','_','`',' ','=',' ','\'','%','s','\'',0};
1277 MSIQUERY *view = NULL;
1278 MSIRECORD *row = NULL;
1279 UINT r;
1280
1281 TRACE("%s\n", debugstr_w(file->File));
1282
1283 r = MSI_OpenQuery(package->db, &view, query, file->File);
1284 if (r != ERROR_SUCCESS)
1285 goto done;
1286
1287 r = MSI_ViewExecute(view, NULL);
1288 if (r != ERROR_SUCCESS)
1289 goto done;
1290
1291 r = MSI_ViewFetch(view, &row);
1292 if (r != ERROR_SUCCESS)
1293 goto done;
1294
1295 file->hash.dwFileHashInfoSize = sizeof(MSIFILEHASHINFO);
1296 file->hash.dwData[0] = MSI_RecordGetInteger(row, 3);
1297 file->hash.dwData[1] = MSI_RecordGetInteger(row, 4);
1298 file->hash.dwData[2] = MSI_RecordGetInteger(row, 5);
1299 file->hash.dwData[3] = MSI_RecordGetInteger(row, 6);
1300
1301 done:
1302 if (view) msiobj_release(&view->hdr);
1303 if (row) msiobj_release(&row->hdr);
1304 return r;
1305 }
1306
1307 static UINT load_file(MSIRECORD *row, LPVOID param)
1308 {
1309 MSIPACKAGE* package = param;
1310 LPCWSTR component;
1311 MSIFILE *file;
1312
1313 /* fill in the data */
1314
1315 file = msi_alloc_zero( sizeof (MSIFILE) );
1316 if (!file)
1317 return ERROR_NOT_ENOUGH_MEMORY;
1318
1319 file->File = msi_dup_record_field( row, 1 );
1320
1321 component = MSI_RecordGetString( row, 2 );
1322 file->Component = get_loaded_component( package, component );
1323
1324 if (!file->Component)
1325 {
1326 WARN("Component not found: %s\n", debugstr_w(component));
1327 msi_free(file->File);
1328 msi_free(file);
1329 return ERROR_SUCCESS;
1330 }
1331
1332 file->FileName = msi_dup_record_field( row, 3 );
1333 reduce_to_longfilename( file->FileName );
1334
1335 file->ShortName = msi_dup_record_field( row, 3 );
1336 file->LongName = strdupW( folder_split_path(file->ShortName, '|'));
1337
1338 file->FileSize = MSI_RecordGetInteger( row, 4 );
1339 file->Version = msi_dup_record_field( row, 5 );
1340 file->Language = msi_dup_record_field( row, 6 );
1341 file->Attributes = MSI_RecordGetInteger( row, 7 );
1342 file->Sequence = MSI_RecordGetInteger( row, 8 );
1343
1344 file->state = msifs_invalid;
1345
1346 /* if the compressed bits are not set in the file attributes,
1347 * then read the information from the package word count property
1348 */
1349 if (package->WordCount & msidbSumInfoSourceTypeAdminImage)
1350 {
1351 file->IsCompressed = FALSE;
1352 }
1353 else if (file->Attributes &
1354 (msidbFileAttributesCompressed | msidbFileAttributesPatchAdded))
1355 {
1356 file->IsCompressed = TRUE;
1357 }
1358 else if (file->Attributes & msidbFileAttributesNoncompressed)
1359 {
1360 file->IsCompressed = FALSE;
1361 }
1362 else
1363 {
1364 file->IsCompressed = package->WordCount & msidbSumInfoSourceTypeCompressed;
1365 }
1366
1367 load_file_hash(package, file);
1368
1369 TRACE("File Loaded (%s)\n",debugstr_w(file->File));
1370
1371 list_add_tail( &package->files, &file->entry );
1372
1373 return ERROR_SUCCESS;
1374 }
1375
1376 static UINT load_all_files(MSIPACKAGE *package)
1377 {
1378 MSIQUERY * view;
1379 UINT rc;
1380 static const WCHAR Query[] =
1381 {'S','E','L','E','C','T',' ','*',' ', 'F','R','O','M',' ',
1382 '`','F','i','l','e','`',' ', 'O','R','D','E','R',' ','B','Y',' ',
1383 '`','S','e','q','u','e','n','c','e','`', 0};
1384
1385 if (!list_empty(&package->files))
1386 return ERROR_SUCCESS;
1387
1388 rc = MSI_DatabaseOpenViewW(package->db, Query, &view);
1389 if (rc != ERROR_SUCCESS)
1390 return ERROR_SUCCESS;
1391
1392 rc = MSI_IterateRecords(view, NULL, load_file, package);
1393 msiobj_release(&view->hdr);
1394
1395 return ERROR_SUCCESS;
1396 }
1397
1398 static UINT load_folder( MSIRECORD *row, LPVOID param )
1399 {
1400 MSIPACKAGE *package = param;
1401 static WCHAR szEmpty[] = { 0 };
1402 LPWSTR p, tgt_short, tgt_long, src_short, src_long;
1403 MSIFOLDER *folder;
1404
1405 folder = msi_alloc_zero( sizeof (MSIFOLDER) );
1406 if (!folder)
1407 return ERROR_NOT_ENOUGH_MEMORY;
1408
1409 folder->Directory = msi_dup_record_field( row, 1 );
1410
1411 TRACE("%s\n", debugstr_w(folder->Directory));
1412
1413 p = msi_dup_record_field(row, 3);
1414
1415 /* split src and target dir */
1416 tgt_short = p;
1417 src_short = folder_split_path( p, ':' );
1418
1419 /* split the long and short paths */
1420 tgt_long = folder_split_path( tgt_short, '|' );
1421 src_long = folder_split_path( src_short, '|' );
1422
1423 /* check for no-op dirs */
1424 if (!lstrcmpW(szDot, tgt_short))
1425 tgt_short = szEmpty;
1426 if (!lstrcmpW(szDot, src_short))
1427 src_short = szEmpty;
1428
1429 if (!tgt_long)
1430 tgt_long = tgt_short;
1431
1432 if (!src_short) {
1433 src_short = tgt_short;
1434 src_long = tgt_long;
1435 }
1436
1437 if (!src_long)
1438 src_long = src_short;
1439
1440 /* FIXME: use the target short path too */
1441 folder->TargetDefault = strdupW(tgt_long);
1442 folder->SourceShortPath = strdupW(src_short);
1443 folder->SourceLongPath = strdupW(src_long);
1444 msi_free(p);
1445
1446 TRACE("TargetDefault = %s\n",debugstr_w( folder->TargetDefault ));
1447 TRACE("SourceLong = %s\n", debugstr_w( folder->SourceLongPath ));
1448 TRACE("SourceShort = %s\n", debugstr_w( folder->SourceShortPath ));
1449
1450 folder->Parent = msi_dup_record_field( row, 2 );
1451
1452 folder->Property = msi_dup_property( package, folder->Directory );
1453
1454 list_add_tail( &package->folders, &folder->entry );
1455
1456 TRACE("returning %p\n", folder);
1457
1458 return ERROR_SUCCESS;
1459 }
1460
1461 static UINT load_all_folders( MSIPACKAGE *package )
1462 {
1463 static const WCHAR query[] = {
1464 'S','E','L','E','C','T',' ','*',' ','F','R', 'O','M',' ',
1465 '`','D','i','r','e','c','t','o','r','y','`',0 };
1466 MSIQUERY *view;
1467 UINT r;
1468
1469 if (!list_empty(&package->folders))
1470 return ERROR_SUCCESS;
1471
1472 r = MSI_DatabaseOpenViewW( package->db, query, &view );
1473 if (r != ERROR_SUCCESS)
1474 return r;
1475
1476 r = MSI_IterateRecords(view, NULL, load_folder, package);
1477 msiobj_release(&view->hdr);
1478 return r;
1479 }
1480
1481 /*
1482 * I am not doing any of the costing functionality yet.
1483 * Mostly looking at doing the Component and Feature loading
1484 *
1485 * The native MSI does A LOT of modification to tables here. Mostly adding
1486 * a lot of temporary columns to the Feature and Component tables.
1487 *
1488 * note: Native msi also tracks the short filename. But I am only going to
1489 * track the long ones. Also looking at this directory table
1490 * it appears that the directory table does not get the parents
1491 * resolved base on property only based on their entries in the
1492 * directory table.
1493 */
1494 static UINT ACTION_CostInitialize(MSIPACKAGE *package)
1495 {
1496 static const WCHAR szCosting[] =
1497 {'C','o','s','t','i','n','g','C','o','m','p','l','e','t','e',0 };
1498
1499 MSI_SetPropertyW(package, szCosting, szZero);
1500 MSI_SetPropertyW(package, cszRootDrive, c_colon);
1501
1502 load_all_folders( package );
1503 load_all_components( package );
1504 load_all_features( package );
1505 load_all_files( package );
1506
1507 return ERROR_SUCCESS;
1508 }
1509
1510 static UINT execute_script(MSIPACKAGE *package, UINT script )
1511 {
1512 UINT i;
1513 UINT rc = ERROR_SUCCESS;
1514
1515 TRACE("Executing Script %i\n",script);
1516
1517 if (!package->script)
1518 {
1519 ERR("no script!\n");
1520 return ERROR_FUNCTION_FAILED;
1521 }
1522
1523 for (i = 0; i < package->script->ActionCount[script]; i++)
1524 {
1525 LPWSTR action;
1526 action = package->script->Actions[script][i];
1527 ui_actionstart(package, action);
1528 TRACE("Executing Action (%s)\n",debugstr_w(action));
1529 rc = ACTION_PerformAction(package, action, script, TRUE);
1530 if (rc != ERROR_SUCCESS)
1531 break;
1532 }
1533 msi_free_action_script(package, script);
1534 return rc;
1535 }
1536
1537 static UINT ACTION_FileCost(MSIPACKAGE *package)
1538 {
1539 return ERROR_SUCCESS;
1540 }
1541
1542 static void ACTION_GetComponentInstallStates(MSIPACKAGE *package)
1543 {
1544 MSICOMPONENT *comp;
1545 INSTALLSTATE state;
1546 UINT r;
1547
1548 state = MsiQueryProductStateW(package->ProductCode);
1549
1550 LIST_FOR_EACH_ENTRY(comp, &package->components, MSICOMPONENT, entry)
1551 {
1552 if (!comp->ComponentId)
1553 continue;
1554
1555 if (state != INSTALLSTATE_LOCAL && state != INSTALLSTATE_DEFAULT)
1556 comp->Installed = INSTALLSTATE_ABSENT;
1557 else
1558 {
1559 r = MsiQueryComponentStateW(package->ProductCode, NULL,
1560 package->Context, comp->ComponentId,
1561 &comp->Installed);
1562 if (r != ERROR_SUCCESS)
1563 comp->Installed = INSTALLSTATE_ABSENT;
1564 }
1565 }
1566 }
1567
1568 static void ACTION_GetFeatureInstallStates(MSIPACKAGE *package)
1569 {
1570 MSIFEATURE *feature;
1571 INSTALLSTATE state;
1572
1573 state = MsiQueryProductStateW(package->ProductCode);
1574
1575 LIST_FOR_EACH_ENTRY( feature, &package->features, MSIFEATURE, entry )
1576 {
1577 if (state != INSTALLSTATE_LOCAL && state != INSTALLSTATE_DEFAULT)
1578 feature->Installed = INSTALLSTATE_ABSENT;
1579 else
1580 {
1581 feature->Installed = MsiQueryFeatureStateW(package->ProductCode,
1582 feature->Feature);
1583 }
1584 }
1585 }
1586
1587 static BOOL process_state_property(MSIPACKAGE* package, int level,
1588 LPCWSTR property, INSTALLSTATE state)
1589 {
1590 LPWSTR override;
1591 MSIFEATURE *feature;
1592
1593 override = msi_dup_property( package, property );
1594 if (!override)
1595 return FALSE;
1596
1597 LIST_FOR_EACH_ENTRY( feature, &package->features, MSIFEATURE, entry )
1598 {
1599 if (lstrcmpW(property, szRemove) &&
1600 (feature->Level <= 0 || feature->Level > level))
1601 continue;
1602
1603 if (!strcmpW(property, szReinstall)) state = feature->Installed;
1604
1605 if (strcmpiW(override, szAll)==0)
1606 msi_feature_set_state(package, feature, state);
1607 else
1608 {
1609 LPWSTR ptr = override;
1610 LPWSTR ptr2 = strchrW(override,',');
1611
1612 while (ptr)
1613 {
1614 int len = ptr2 - ptr;
1615
1616 if ((ptr2 && strlenW(feature->Feature) == len && !strncmpW(ptr, feature->Feature, len))
1617 || (!ptr2 && !strcmpW(ptr, feature->Feature)))
1618 {
1619 msi_feature_set_state(package, feature, state);
1620 break;
1621 }
1622 if (ptr2)
1623 {
1624 ptr=ptr2+1;
1625 ptr2 = strchrW(ptr,',');
1626 }
1627 else
1628 break;
1629 }
1630 }
1631 }
1632 msi_free(override);
1633
1634 return TRUE;
1635 }
1636
1637 static BOOL process_overrides( MSIPACKAGE *package, int level )
1638 {
1639 static const WCHAR szAddLocal[] =
1640 {'A','D','D','L','O','C','A','L',0};
1641 static const WCHAR szAddSource[] =
1642 {'A','D','D','S','O','U','R','C','E',0};
1643 static const WCHAR szAdvertise[] =
1644 {'A','D','V','E','R','T','I','S','E',0};
1645 BOOL ret = FALSE;
1646
1647 /* all these activation/deactivation things happen in order and things
1648 * later on the list override things earlier on the list.
1649 *
1650 * 0 INSTALLLEVEL processing
1651 * 1 ADDLOCAL
1652 * 2 REMOVE
1653 * 3 ADDSOURCE
1654 * 4 ADDDEFAULT
1655 * 5 REINSTALL
1656 * 6 ADVERTISE
1657 * 7 COMPADDLOCAL
1658 * 8 COMPADDSOURCE
1659 * 9 FILEADDLOCAL
1660 * 10 FILEADDSOURCE
1661 * 11 FILEADDDEFAULT
1662 */
1663 ret |= process_state_property( package, level, szAddLocal, INSTALLSTATE_LOCAL );
1664 ret |= process_state_property( package, level, szRemove, INSTALLSTATE_ABSENT );
1665 ret |= process_state_property( package, level, szAddSource, INSTALLSTATE_SOURCE );
1666 ret |= process_state_property( package, level, szReinstall, INSTALLSTATE_UNKNOWN );
1667 ret |= process_state_property( package, level, szAdvertise, INSTALLSTATE_ADVERTISED );
1668
1669 if (ret)
1670 MSI_SetPropertyW( package, szPreselected, szOne );
1671
1672 return ret;
1673 }
1674
1675 UINT MSI_SetFeatureStates(MSIPACKAGE *package)
1676 {
1677 int level;
1678 static const WCHAR szlevel[] =
1679 {'I','N','S','T','A','L','L','L','E','V','E','L',0};
1680 MSICOMPONENT* component;
1681 MSIFEATURE *feature;
1682
1683 TRACE("Checking Install Level\n");
1684
1685 level = msi_get_property_int(package, szlevel, 1);
1686
1687 if (!msi_get_property_int( package, szPreselected, 0 ))
1688 {
1689 LIST_FOR_EACH_ENTRY( feature, &package->features, MSIFEATURE, entry )
1690 {
1691 BOOL feature_state = ((feature->Level > 0) &&
1692 (feature->Level <= level));
1693
1694 if ((feature_state) && (feature->Action == INSTALLSTATE_UNKNOWN))
1695 {
1696 if (feature->Attributes & msidbFeatureAttributesFavorSource)
1697 msi_feature_set_state(package, feature, INSTALLSTATE_SOURCE);
1698 else if (feature->Attributes & msidbFeatureAttributesFavorAdvertise)
1699 msi_feature_set_state(package, feature, INSTALLSTATE_ADVERTISED);
1700 else
1701 msi_feature_set_state(package, feature, INSTALLSTATE_LOCAL);
1702 }
1703 }
1704
1705 /* disable child features of unselected parent features */
1706 LIST_FOR_EACH_ENTRY( feature, &package->features, MSIFEATURE, entry )
1707 {
1708 FeatureList *fl;
1709
1710 if (feature->Level > 0 && feature->Level <= level)
1711 continue;
1712
1713 LIST_FOR_EACH_ENTRY( fl, &feature->Children, FeatureList, entry )
1714 msi_feature_set_state(package, fl->feature, INSTALLSTATE_UNKNOWN);
1715 }
1716 }
1717
1718 /*
1719 * now we want to enable or disable components base on feature
1720 */
1721
1722 LIST_FOR_EACH_ENTRY( feature, &package->features, MSIFEATURE, entry )
1723 {
1724 ComponentList *cl;
1725
1726 TRACE("Examining Feature %s (Level %i, Installed %i, Action %i)\n",
1727 debugstr_w(feature->Feature), feature->Level, feature->Installed, feature->Action);
1728
1729 if (!feature->Level)
1730 continue;
1731
1732 /* features with components that have compressed files are made local */
1733 LIST_FOR_EACH_ENTRY( cl, &feature->Components, ComponentList, entry )
1734 {
1735 if (cl->component->Enabled &&
1736 cl->component->ForceLocalState &&
1737 feature->Action == INSTALLSTATE_SOURCE)
1738 {
1739 msi_feature_set_state(package, feature, INSTALLSTATE_LOCAL);
1740 break;
1741 }
1742 }
1743
1744 LIST_FOR_EACH_ENTRY( cl, &feature->Components, ComponentList, entry )
1745 {
1746 component = cl->component;
1747
1748 if (!component->Enabled)
1749 continue;
1750
1751 switch (feature->Action)
1752 {
1753 case INSTALLSTATE_ABSENT:
1754 component->anyAbsent = 1;
1755 break;
1756 case INSTALLSTATE_ADVERTISED:
1757 component->hasAdvertiseFeature = 1;
1758 break;
1759 case INSTALLSTATE_SOURCE:
1760 component->hasSourceFeature = 1;
1761 break;
1762 case INSTALLSTATE_LOCAL:
1763 component->hasLocalFeature = 1;
1764 break;
1765 case INSTALLSTATE_DEFAULT:
1766 if (feature->Attributes & msidbFeatureAttributesFavorAdvertise)
1767 component->hasAdvertiseFeature = 1;
1768 else if (feature->Attributes & msidbFeatureAttributesFavorSource)
1769 component->hasSourceFeature = 1;
1770 else
1771 component->hasLocalFeature = 1;
1772 break;
1773 default:
1774 break;
1775 }
1776 }
1777 }
1778
1779 LIST_FOR_EACH_ENTRY( component, &package->components, MSICOMPONENT, entry )
1780 {
1781 /* if the component isn't enabled, leave it alone */
1782 if (!component->Enabled)
1783 continue;
1784
1785 /* check if it's local or source */
1786 if (!(component->Attributes & msidbComponentAttributesOptional) &&
1787 (component->hasLocalFeature || component->hasSourceFeature))
1788 {
1789 if ((component->Attributes & msidbComponentAttributesSourceOnly) &&
1790 !component->ForceLocalState)
1791 msi_component_set_state(package, component, INSTALLSTATE_SOURCE);
1792 else
1793 msi_component_set_state(package, component, INSTALLSTATE_LOCAL);
1794 continue;
1795 }
1796
1797 /* if any feature is local, the component must be local too */
1798 if (component->hasLocalFeature)
1799 {
1800 msi_component_set_state(package, component, INSTALLSTATE_LOCAL);
1801 continue;
1802 }
1803
1804 if (component->hasSourceFeature)
1805 {
1806 msi_component_set_state(package, component, INSTALLSTATE_SOURCE);
1807 continue;
1808 }
1809
1810 if (component->hasAdvertiseFeature)
1811 {
1812 msi_component_set_state(package, component, INSTALLSTATE_ADVERTISED);
1813 continue;
1814 }
1815
1816 TRACE("nobody wants component %s\n", debugstr_w(component->Component));
1817 if (component->anyAbsent)
1818 msi_component_set_state(package, component, INSTALLSTATE_ABSENT);
1819 }
1820
1821 LIST_FOR_EACH_ENTRY( component, &package->components, MSICOMPONENT, entry )
1822 {
1823 if (component->Action == INSTALLSTATE_DEFAULT)
1824 {
1825 TRACE("%s was default, setting to local\n", debugstr_w(component->Component));
1826 msi_component_set_state(package, component, INSTALLSTATE_LOCAL);
1827 }
1828
1829 TRACE("Result: Component %s (Installed %i, Action %i)\n",
1830 debugstr_w(component->Component), component->Installed, component->Action);
1831 }
1832
1833
1834 return ERROR_SUCCESS;
1835 }
1836
1837 static UINT ITERATE_CostFinalizeDirectories(MSIRECORD *row, LPVOID param)
1838 {
1839 MSIPACKAGE *package = param;
1840 LPCWSTR name;
1841 LPWSTR path;
1842 MSIFOLDER *f;
1843
1844 name = MSI_RecordGetString(row,1);
1845
1846 f = get_loaded_folder(package, name);
1847 if (!f) return ERROR_SUCCESS;
1848
1849 /* reset the ResolvedTarget */
1850 msi_free(f->ResolvedTarget);
1851 f->ResolvedTarget = NULL;
1852
1853 /* This helper function now does ALL the work */
1854 TRACE("Dir %s ...\n",debugstr_w(name));
1855 path = resolve_folder(package,name,FALSE,TRUE,TRUE,NULL);
1856 TRACE("resolves to %s\n",debugstr_w(path));
1857 msi_free(path);
1858
1859 return ERROR_SUCCESS;
1860 }
1861
1862 static UINT ITERATE_CostFinalizeConditions(MSIRECORD *row, LPVOID param)
1863 {
1864 MSIPACKAGE *package = param;
1865 LPCWSTR name;
1866 MSIFEATURE *feature;
1867
1868 name = MSI_RecordGetString( row, 1 );
1869
1870 feature = get_loaded_feature( package, name );
1871 if (!feature)
1872 ERR("FAILED to find loaded feature %s\n",debugstr_w(name));
1873 else
1874 {
1875 LPCWSTR Condition;
1876 Condition = MSI_RecordGetString(row,3);
1877
1878 if (MSI_EvaluateConditionW(package,Condition) == MSICONDITION_TRUE)
1879 {
1880 int level = MSI_RecordGetInteger(row,2);
1881 TRACE("Resetting feature %s to level %i\n", debugstr_w(name), level);
1882 feature->Level = level;
1883 }
1884 }
1885 return ERROR_SUCCESS;
1886 }
1887
1888 static LPWSTR msi_get_disk_file_version( LPCWSTR filename )
1889 {
1890 static const WCHAR name_fmt[] =
1891 {'%','u','.','%','u','.','%','u','.','%','u',0};
1892 static const WCHAR name[] = {'\\',0};
1893 VS_FIXEDFILEINFO *lpVer;
1894 WCHAR filever[0x100];
1895 LPVOID version;
1896 DWORD versize;
1897 DWORD handle;
1898 UINT sz;
1899
1900 TRACE("%s\n", debugstr_w(filename));
1901
1902 versize = GetFileVersionInfoSizeW( filename, &handle );
1903 if (!versize)
1904 return NULL;
1905
1906 version = msi_alloc( versize );
1907 GetFileVersionInfoW( filename, 0, versize, version );
1908
1909 if (!VerQueryValueW( version, name, (LPVOID*)&lpVer, &sz ))
1910 {
1911 msi_free( version );
1912 return NULL;
1913 }
1914
1915 sprintfW( filever, name_fmt,
1916 HIWORD(lpVer->dwFileVersionMS),
1917 LOWORD(lpVer->dwFileVersionMS),
1918 HIWORD(lpVer->dwFileVersionLS),
1919 LOWORD(lpVer->dwFileVersionLS));
1920
1921 msi_free( version );
1922
1923 return strdupW( filever );
1924 }
1925
1926 static UINT msi_check_file_install_states( MSIPACKAGE *package )
1927 {
1928 LPWSTR file_version;
1929 MSIFILE *file;
1930
1931 LIST_FOR_EACH_ENTRY( file, &package->files, MSIFILE, entry )
1932 {
1933 MSICOMPONENT* comp = file->Component;
1934 LPWSTR p;
1935
1936 if (!comp)
1937 continue;
1938
1939 if (file->IsCompressed)
1940 comp->ForceLocalState = TRUE;
1941
1942 /* calculate target */
1943 p = resolve_folder(package, comp->Directory, FALSE, FALSE, TRUE, NULL);
1944
1945 msi_free(file->TargetPath);
1946
1947 TRACE("file %s is named %s\n",
1948 debugstr_w(file->File), debugstr_w(file->FileName));
1949
1950 file->TargetPath = build_directory_name(2, p, file->FileName);
1951
1952 msi_free(p);
1953
1954 TRACE("file %s resolves to %s\n",
1955 debugstr_w(file->File), debugstr_w(file->TargetPath));
1956
1957 /* don't check files of components that aren't installed */
1958 if (comp->Installed == INSTALLSTATE_UNKNOWN ||
1959 comp->Installed == INSTALLSTATE_ABSENT)
1960 {
1961 file->state = msifs_missing; /* assume files are missing */
1962 continue;
1963 }
1964
1965 if (GetFileAttributesW(file->TargetPath) == INVALID_FILE_ATTRIBUTES)
1966 {
1967 file->state = msifs_missing;
1968 comp->Cost += file->FileSize;
1969 continue;
1970 }
1971
1972 if (file->Version &&
1973 (file_version = msi_get_disk_file_version( file->TargetPath )))
1974 {
1975 TRACE("new %s old %s\n", debugstr_w(file->Version),
1976 debugstr_w(file_version));
1977 /* FIXME: seems like a bad way to compare version numbers */
1978 if (lstrcmpiW(file_version, file->Version)<0)
1979 {
1980 file->state = msifs_overwrite;
1981 comp->Cost += file->FileSize;
1982 }
1983 else
1984 file->state = msifs_present;
1985 msi_free( file_version );
1986 }
1987 else
1988 file->state = msifs_present;
1989 }
1990
1991 return ERROR_SUCCESS;
1992 }
1993
1994 /*
1995 * A lot is done in this function aside from just the costing.
1996 * The costing needs to be implemented at some point but for now I am going
1997 * to focus on the directory building
1998 *
1999 */
2000 static UINT ACTION_CostFinalize(MSIPACKAGE *package)
2001 {
2002 static const WCHAR ExecSeqQuery[] =
2003 {'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
2004 '`','D','i','r','e','c','t','o','r','y','`',0};
2005 static const WCHAR ConditionQuery[] =
2006 {'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
2007 '`','C','o','n','d','i','t','i','o','n','`',0};
2008 static const WCHAR szCosting[] =
2009 {'C','o','s','t','i','n','g','C','o','m','p','l','e','t','e',0 };
2010 static const WCHAR szlevel[] =
2011 {'I','N','S','T','A','L','L','L','E','V','E','L',0};
2012 static const WCHAR szOutOfDiskSpace[] =
2013 {'O','u','t','O','f','D','i','s','k','S','p','a','c','e',0};
2014 MSICOMPONENT *comp;
2015 UINT rc = ERROR_SUCCESS;
2016 MSIQUERY * view;
2017 LPWSTR level;
2018
2019 TRACE("Building Directory properties\n");
2020
2021 rc = MSI_DatabaseOpenViewW(package->db, ExecSeqQuery, &view);
2022 if (rc == ERROR_SUCCESS)
2023 {
2024 rc = MSI_IterateRecords(view, NULL, ITERATE_CostFinalizeDirectories,
2025 package);
2026 msiobj_release(&view->hdr);
2027 }
2028
2029 /* read components states from the registry */
2030 ACTION_GetComponentInstallStates(package);
2031 ACTION_GetFeatureInstallStates(package);
2032
2033 TRACE("File calculations\n");
2034 msi_check_file_install_states( package );
2035
2036 if (!process_overrides( package, msi_get_property_int( package, szlevel, 1 ) ))
2037 {
2038 TRACE("Evaluating Condition Table\n");
2039
2040 rc = MSI_DatabaseOpenViewW( package->db, ConditionQuery, &view );
2041 if (rc == ERROR_SUCCESS)
2042 {
2043 rc = MSI_IterateRecords( view, NULL, ITERATE_CostFinalizeConditions, package );
2044 msiobj_release( &view->hdr );
2045 }
2046
2047 TRACE("Enabling or Disabling Components\n");
2048 LIST_FOR_EACH_ENTRY( comp, &package->components, MSICOMPONENT, entry )
2049 {
2050 if (MSI_EvaluateConditionW( package, comp->Condition ) == MSICONDITION_FALSE)
2051 {
2052 TRACE("Disabling component %s\n", debugstr_w(comp->Component));
2053 comp->Enabled = FALSE;
2054 }
2055 else
2056 comp->Enabled = TRUE;
2057 }
2058 }
2059
2060 MSI_SetPropertyW(package,szCosting,szOne);
2061 /* set default run level if not set */
2062 level = msi_dup_property( package, szlevel );
2063 if (!level)
2064 MSI_SetPropertyW(package,szlevel, szOne);
2065 msi_free(level);
2066
2067 /* FIXME: check volume disk space */
2068 MSI_SetPropertyW(package, szOutOfDiskSpace, szZero);
2069
2070 return MSI_SetFeatureStates(package);
2071 }
2072
2073 /* OK this value is "interpreted" and then formatted based on the
2074 first few characters */
2075 static LPSTR parse_value(MSIPACKAGE *package, LPCWSTR value, DWORD *type,
2076 DWORD *size)
2077 {
2078 LPSTR data = NULL;
2079
2080 if (value[0]=='#' && value[1]!='#' && value[1]!='%')
2081 {
2082 if (value[1]=='x')
2083 {
2084 LPWSTR ptr;
2085 CHAR byte[5];
2086 LPWSTR deformated = NULL;
2087 int count;
2088
2089 deformat_string(package, &value[2], &deformated);
2090
2091 /* binary value type */
2092 ptr = deformated;
2093 *type = REG_BINARY;
2094 if (strlenW(ptr)%2)
2095 *size = (strlenW(ptr)/2)+1;
2096 else
2097 *size = strlenW(ptr)/2;
2098
2099 data = msi_alloc(*size);
2100
2101 byte[0] = '0';
2102 byte[1] = 'x';
2103 byte[4] = 0;
2104 count = 0;
2105 /* if uneven pad with a zero in front */
2106 if (strlenW(ptr)%2)
2107 {
2108 byte[2]= '0';
2109 byte[3]= *ptr;
2110 ptr++;
2111 data[count] = (BYTE)strtol(byte,NULL,0);
2112 count ++;
2113 TRACE("Uneven byte count\n");
2114 }
2115 while (*ptr)
2116 {
2117 byte[2]= *ptr;
2118 ptr++;
2119 byte[3]= *ptr;
2120 ptr++;
2121 data[count] = (BYTE)strtol(byte,NULL,0);
2122 count ++;
2123 }
2124 msi_free(deformated);
2125
2126 TRACE("Data %i bytes(%i)\n",*size,count);
2127 }
2128 else
2129 {
2130 LPWSTR deformated;
2131 LPWSTR p;
2132 DWORD d = 0;
2133 deformat_string(package, &value[1], &deformated);
2134
2135 *type=REG_DWORD;
2136 *size = sizeof(DWORD);
2137 data = msi_alloc(*size);
2138 p = deformated;
2139 if (*p == '-')
2140 p++;
2141 while (*p)
2142 {
2143 if ( (*p < '0') || (*p > '9') )
2144 break;
2145 d *= 10;
2146 d += (*p - '0');
2147 p++;
2148 }
2149 if (deformated[0] == '-')
2150 d = -d;
2151 *(LPDWORD)data = d;
2152 TRACE("DWORD %i\n",*(LPDWORD)data);
2153
2154 msi_free(deformated);
2155 }
2156 }
2157 else
2158 {
2159 static const WCHAR szMulti[] = {'[','~',']',0};
2160 LPCWSTR ptr;
2161 *type=REG_SZ;
2162
2163 if (value[0]=='#')
2164 {
2165 if (value[1]=='%')
2166 {
2167 ptr = &value[2];
2168 *type=REG_EXPAND_SZ;
2169 }
2170 else
2171 ptr = &value[1];
2172 }
2173 else
2174 ptr=value;
2175
2176 if (strstrW(value,szMulti))
2177 *type = REG_MULTI_SZ;
2178
2179 /* remove initial delimiter */
2180 if (!strncmpW(value, szMulti, 3))
2181 ptr = value + 3;
2182
2183 *size = deformat_string(package, ptr,(LPWSTR*)&data);
2184
2185 /* add double NULL terminator */
2186 if (*type == REG_MULTI_SZ)
2187 {
2188 *size += 2 * sizeof(WCHAR); /* two NULL terminators */
2189 data = msi_realloc_zero(data, *size);
2190 }
2191 }
2192 return data;
2193 }
2194
2195 static UINT ITERATE_WriteRegistryValues(MSIRECORD *row, LPVOID param)
2196 {
2197 MSIPACKAGE *package = param;
2198 static const WCHAR szHCR[] =
2199 {'H','K','E','Y','_','C','L','A','S','S','E','S','_',
2200 'R','O','O','T','\\',0};
2201 static const WCHAR szHCU[] =
2202 {'H','K','E','Y','_','C','U','R','R','E','N','T','_',
2203 'U','S','E','R','\\',0};
2204 static const WCHAR szHLM[] =
2205 {'H','K','E','Y','_','L','O','C','A','L','_',
2206 'M','A','C','H','I','N','E','\\',0};
2207 static const WCHAR szHU[] =
2208 {'H','K','E','Y','_','U','S','E','R','S','\\',0};
2209
2210 LPSTR value_data = NULL;
2211 HKEY root_key, hkey;
2212 DWORD type,size;
2213 LPWSTR deformated;
2214 LPCWSTR szRoot, component, name, key, value;
2215 MSICOMPONENT *comp;
2216 MSIRECORD * uirow;
2217 LPWSTR uikey;
2218 INT root;
2219 BOOL check_first = FALSE;
2220 UINT rc;
2221
2222 ui_progress(package,2,0,0,0);
2223
2224 value = NULL;
2225 key = NULL;
2226 uikey = NULL;
2227 name = NULL;
2228
2229 component = MSI_RecordGetString(row, 6);
2230 comp = get_loaded_component(package,component);
2231 if (!comp)
2232 return ERROR_SUCCESS;
2233
2234 if (!ACTION_VerifyComponentForAction( comp, INSTALLSTATE_LOCAL))
2235 {
2236 TRACE("Skipping write due to disabled component %s\n",
2237 debugstr_w(component));
2238
2239 comp->Action = comp->Installed;
2240
2241 return ERROR_SUCCESS;
2242 }
2243
2244 comp->Action = INSTALLSTATE_LOCAL;
2245
2246 name = MSI_RecordGetString(row, 4);
2247 if( MSI_RecordIsNull(row,5) && name )
2248 {
2249 /* null values can have special meanings */
2250 if (name[0]=='-' && name[1] == 0)
2251 return ERROR_SUCCESS;
2252 else if ((name[0]=='+' && name[1] == 0) ||
2253 (name[0] == '*' && name[1] == 0))
2254 name = NULL;
2255 check_first = TRUE;
2256 }
2257
2258 root = MSI_RecordGetInteger(row,2);
2259 key = MSI_RecordGetString(row, 3);
2260
2261 /* get the root key */
2262 switch (root)
2263 {
2264 case -1:
2265 {
2266 LPWSTR all_users = msi_dup_property( package, szAllUsers );
2267 if (all_users && all_users[0] == '1')
2268 {
2269 root_key = HKEY_LOCAL_MACHINE;
2270 szRoot = szHLM;
2271 }
2272 else
2273 {
2274 root_key = HKEY_CURRENT_USER;
2275 szRoot = szHCU;
2276 }
2277 msi_free(all_users);
2278 }
2279 break;
2280 case 0: root_key = HKEY_CLASSES_ROOT;
2281 szRoot = szHCR;
2282 break;
2283 case 1: root_key = HKEY_CURRENT_USER;
2284 szRoot = szHCU;
2285 break;
2286 case 2: root_key = HKEY_LOCAL_MACHINE;
2287 szRoot = szHLM;
2288 break;
2289 case 3: root_key = HKEY_USERS;
2290 szRoot = szHU;
2291 break;
2292 default:
2293 ERR("Unknown root %i\n",root);
2294 root_key=NULL;
2295 szRoot = NULL;
2296 break;
2297 }
2298 if (!root_key)
2299 return ERROR_SUCCESS;
2300
2301 deformat_string(package, key , &deformated);
2302 size = strlenW(deformated) + strlenW(szRoot) + 1;
2303 uikey = msi_alloc(size*sizeof(WCHAR));
2304 strcpyW(uikey,szRoot);
2305 strcatW(uikey,deformated);
2306
2307 if (RegCreateKeyW( root_key, deformated, &hkey))
2308 {
2309 ERR("Could not create key %s\n",debugstr_w(deformated));
2310 msi_free(deformated);
2311 msi_free(uikey);
2312 return ERROR_SUCCESS;
2313 }
2314 msi_free(deformated);
2315
2316 value = MSI_RecordGetString(row,5);
2317 if (value)
2318 value_data = parse_value(package, value, &type, &size);
2319 else
2320 {
2321 value_data = (LPSTR)strdupW(szEmpty);
2322 size = sizeof(szEmpty);
2323 type = REG_SZ;
2324 }
2325
2326 deformat_string(package, name, &deformated);
2327
2328 if (!check_first)
2329 {
2330 TRACE("Setting value %s of %s\n",debugstr_w(deformated),
2331 debugstr_w(uikey));
2332 RegSetValueExW(hkey, deformated, 0, type, (LPBYTE)value_data, size);
2333 }
2334 else
2335 {
2336 DWORD sz = 0;
2337 rc = RegQueryValueExW(hkey, deformated, NULL, NULL, NULL, &sz);
2338 if (rc == ERROR_SUCCESS || rc == ERROR_MORE_DATA)
2339 {
2340 TRACE("value %s of %s checked already exists\n",
2341 debugstr_w(deformated), debugstr_w(uikey));
2342 }
2343 else
2344 {
2345 TRACE("Checked and setting value %s of %s\n",
2346 debugstr_w(deformated), debugstr_w(uikey));
2347 if (deformated || size)
2348 RegSetValueExW(hkey, deformated, 0, type, (LPBYTE) value_data, size);
2349 }
2350 }
2351 RegCloseKey(hkey);
2352
2353 uirow = MSI_CreateRecord(3);
2354 MSI_RecordSetStringW(uirow,2,deformated);
2355 MSI_RecordSetStringW(uirow,1,uikey);
2356
2357 if (type == REG_SZ)
2358 MSI_RecordSetStringW(uirow,3,(LPWSTR)value_data);
2359 else
2360 MSI_RecordSetStringW(uirow,3,value);
2361
2362 ui_actiondata(package,szWriteRegistryValues,uirow);
2363 msiobj_release( &uirow->hdr );
2364
2365 msi_free(value_data);
2366 msi_free(deformated);
2367 msi_free(uikey);
2368
2369 return ERROR_SUCCESS;
2370 }
2371
2372 static UINT ACTION_WriteRegistryValues(MSIPACKAGE *package)
2373 {
2374 UINT rc;
2375 MSIQUERY * view;
2376 static const WCHAR ExecSeqQuery[] =
2377 {'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
2378 '`','R','e','g','i','s','t','r','y','`',0 };
2379
2380 rc = MSI_DatabaseOpenViewW(package->db, ExecSeqQuery, &view);
2381 if (rc != ERROR_SUCCESS)
2382 return ERROR_SUCCESS;
2383
2384 /* increment progress bar each time action data is sent */
2385 ui_progress(package,1,REG_PROGRESS_VALUE,1,0);
2386
2387 rc = MSI_IterateRecords(view, NULL, ITERATE_WriteRegistryValues, package);
2388
2389 msiobj_release(&view->hdr);
2390 return rc;
2391 }
2392
2393 static UINT ACTION_InstallInitialize(MSIPACKAGE *package)
2394 {
2395 package->script->CurrentlyScripting = TRUE;
2396
2397 return ERROR_SUCCESS;
2398 }
2399
2400
2401 static UINT ACTION_InstallValidate(MSIPACKAGE *package)
2402 {
2403 MSICOMPONENT *comp;
2404 DWORD progress = 0;
2405 DWORD total = 0;
2406 static const WCHAR q1[]=
2407 {'S','E','L','E','C','T',' ','*',' ', 'F','R','O','M',' ',
2408 '`','R','e','g','i','s','t','r','y','`',0};
2409 UINT rc;
2410 MSIQUERY * view;
2411 MSIFEATURE *feature;
2412 MSIFILE *file;
2413
2414 TRACE("InstallValidate\n");
2415
2416 rc = MSI_DatabaseOpenViewW(package->db, q1, &view);
2417 if (rc == ERROR_SUCCESS)
2418 {
2419 MSI_IterateRecords( view, &progress, NULL, package );
2420 msiobj_release( &view->hdr );
2421 total += progress * REG_PROGRESS_VALUE;
2422 }
2423
2424 LIST_FOR_EACH_ENTRY( comp, &package->components, MSICOMPONENT, entry )
2425 total += COMPONENT_PROGRESS_VALUE;
2426
2427 LIST_FOR_EACH_ENTRY( file, &package->files, MSIFILE, entry )
2428 total += file->FileSize;
2429
2430 ui_progress(package,0,total,0,0);
2431
2432 LIST_FOR_EACH_ENTRY( feature, &package->features, MSIFEATURE, entry )
2433 {
2434 TRACE("Feature: %s; Installed: %i; Action %i; Request %i\n",
2435 debugstr_w(feature->Feature), feature->Installed, feature->Action,
2436 feature->ActionRequest);
2437 }
2438
2439 return ERROR_SUCCESS;
2440 }
2441
2442 static UINT ITERATE_LaunchConditions(MSIRECORD *row, LPVOID param)
2443 {
2444 MSIPACKAGE* package = param;
2445 LPCWSTR cond = NULL;
2446 LPCWSTR message = NULL;
2447 UINT r;
2448
2449 static const WCHAR title[]=
2450 {'I','n','s','t','a','l','l',' ','F','a', 'i','l','e','d',0};
2451
2452 cond = MSI_RecordGetString(row,1);
2453
2454 r = MSI_EvaluateConditionW(package,cond);
2455 if (r == MSICONDITION_FALSE)
2456 {
2457 if ((gUILevel & INSTALLUILEVEL_MASK) != INSTALLUILEVEL_NONE)
2458 {
2459 LPWSTR deformated;
2460 message = MSI_RecordGetString(row,2);
2461 deformat_string(package,message,&deformated);
2462 MessageBoxW(NULL,deformated,title,MB_OK);
2463 msi_free(deformated);
2464 }
2465
2466 return ERROR_INSTALL_FAILURE;
2467 }
2468
2469 return ERROR_SUCCESS;
2470 }
2471
2472 static UINT ACTION_LaunchConditions(MSIPACKAGE *package)
2473 {
2474 UINT rc;
2475 MSIQUERY * view = NULL;
2476 static const WCHAR ExecSeqQuery[] =
2477 {'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
2478 '`','L','a','u','n','c','h','C','o','n','d','i','t','i','o','n','`',0};
2479
2480 TRACE("Checking launch conditions\n");
2481
2482 rc = MSI_DatabaseOpenViewW(package->db, ExecSeqQuery, &view);
2483 if (rc != ERROR_SUCCESS)
2484 return ERROR_SUCCESS;
2485
2486 rc = MSI_IterateRecords(view, NULL, ITERATE_LaunchConditions, package);
2487 msiobj_release(&view->hdr);
2488
2489 return rc;
2490 }
2491
2492 static LPWSTR resolve_keypath( MSIPACKAGE* package, MSICOMPONENT *cmp )
2493 {
2494
2495 if (!cmp->KeyPath)
2496 return resolve_folder(package,cmp->Directory,FALSE,FALSE,TRUE,NULL);
2497
2498 if (cmp->Attributes & msidbComponentAttributesRegistryKeyPath)
2499 {
2500 MSIRECORD * row = 0;
2501 UINT root,len;
2502 LPWSTR deformated,buffer,deformated_name;
2503 LPCWSTR key,name;
2504 static const WCHAR ExecSeqQuery[] =
2505 {'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
2506 '`','R','e','g','i','s','t','r','y','`',' ',
2507 'W','H','E','R','E',' ', '`','R','e','g','i','s','t','r','y','`',
2508 ' ','=',' ' ,'\'','%','s','\'',0 };
2509 static const WCHAR fmt[]={'%','0','2','i',':','\\','%','s','\\',0};
2510 static const WCHAR fmt2[]=
2511 {'%','0','2','i',':','\\','%','s','\\','%','s',0};
2512
2513 row = MSI_QueryGetRecord(package->db, ExecSeqQuery,cmp->KeyPath);
2514 if (!row)
2515 return NULL;
2516
2517 root = MSI_RecordGetInteger(row,2);
2518 key = MSI_RecordGetString(row, 3);
2519 name = MSI_RecordGetString(row, 4);
2520 deformat_string(package, key , &deformated);
2521 deformat_string(package, name, &deformated_name);
2522
2523 len = strlenW(deformated) + 6;
2524 if (deformated_name)
2525 len+=strlenW(deformated_name);
2526
2527 buffer = msi_alloc( len *sizeof(WCHAR));
2528
2529 if (deformated_name)
2530 sprintfW(buffer,fmt2,root,deformated,deformated_name);
2531 else
2532 sprintfW(buffer,fmt,root,deformated);
2533
2534 msi_free(deformated);
2535 msi_free(deformated_name);
2536 msiobj_release(&row->hdr);
2537
2538 return buffer;
2539 }
2540 else if (cmp->Attributes & msidbComponentAttributesODBCDataSource)
2541 {
2542 FIXME("UNIMPLEMENTED keypath as ODBC Source\n");
2543 return NULL;
2544 }
2545 else
2546 {
2547 MSIFILE *file = get_loaded_file( package, cmp->KeyPath );
2548
2549 if (file)
2550 return strdupW( file->TargetPath );
2551 }
2552 return NULL;
2553 }
2554
2555 static HKEY openSharedDLLsKey(void)
2556 {
2557 HKEY hkey=0;
2558 static const WCHAR path[] =
2559 {'S','o','f','t','w','a','r','e','\\',
2560 'M','i','c','r','o','s','o','f','t','\\',
2561 'W','i','n','d','o','w','s','\\',
2562 'C','u','r','r','e','n','t','V','e','r','s','i','o','n','\\',
2563 'S','h','a','r','e','d','D','L','L','s',0};
2564
2565 RegCreateKeyW(HKEY_LOCAL_MACHINE,path,&hkey);
2566 return hkey;
2567 }
2568
2569 static UINT ACTION_GetSharedDLLsCount(LPCWSTR dll)
2570 {
2571 HKEY hkey;
2572 DWORD count=0;
2573 DWORD type;
2574 DWORD sz = sizeof(count);
2575 DWORD rc;
2576
2577 hkey = openSharedDLLsKey();
2578 rc = RegQueryValueExW(hkey, dll, NULL, &type, (LPBYTE)&count, &sz);
2579 if (rc != ERROR_SUCCESS)
2580 count = 0;
2581 RegCloseKey(hkey);
2582 return count;
2583 }
2584
2585 static UINT ACTION_WriteSharedDLLsCount(LPCWSTR path, UINT count)
2586 {
2587 HKEY hkey;
2588
2589 hkey = openSharedDLLsKey();
2590 if (count > 0)
2591 msi_reg_set_val_dword( hkey, path, count );
2592 else
2593 RegDeleteValueW(hkey,path);
2594 RegCloseKey(hkey);
2595 return count;
2596 }
2597
2598 /*
2599 * Return TRUE if the count should be written out and FALSE if not
2600 */
2601 static void ACTION_RefCountComponent( MSIPACKAGE* package, MSICOMPONENT *comp )
2602 {
2603 MSIFEATURE *feature;
2604 INT count = 0;
2605 BOOL write = FALSE;
2606
2607 /* only refcount DLLs */
2608 if (comp->KeyPath == NULL ||
2609 comp->Attributes & msidbComponentAttributesRegistryKeyPath ||
2610 comp->Attributes & msidbComponentAttributesODBCDataSource)
2611 write = FALSE;
2612 else
2613 {
2614 count = ACTION_GetSharedDLLsCount( comp->FullKeypath);
2615 write = (count > 0);
2616
2617 if (comp->Attributes & msidbComponentAttributesSharedDllRefCount)
2618 write = TRUE;
2619 }
2620
2621 /* increment counts */
2622 LIST_FOR_EACH_ENTRY( feature, &package->features, MSIFEATURE, entry )
2623 {
2624 ComponentList *cl;
2625
2626 if (!ACTION_VerifyFeatureForAction( feature, INSTALLSTATE_LOCAL ))
2627 continue;
2628
2629 LIST_FOR_EACH_ENTRY( cl, &feature->Components, ComponentList, entry )
2630 {
2631 if ( cl->component == comp )
2632 count++;
2633 }
2634 }
2635
2636 /* decrement counts */
2637 LIST_FOR_EACH_ENTRY( feature, &package->features, MSIFEATURE, entry )
2638 {
2639 ComponentList *cl;
2640
2641 if (!ACTION_VerifyFeatureForAction( feature, INSTALLSTATE_ABSENT ))
2642 continue;
2643
2644 LIST_FOR_EACH_ENTRY( cl, &feature->Components, ComponentList, entry )
2645 {
2646 if ( cl->component == comp )
2647 count--;
2648 }
2649 }
2650
2651 /* ref count all the files in the component */
2652 if (write)
2653 {
2654 MSIFILE *file;
2655
2656 LIST_FOR_EACH_ENTRY( file, &package->files, MSIFILE, entry )
2657 {
2658 if (file->Component == comp)
2659 ACTION_WriteSharedDLLsCount( file->TargetPath, count );
2660 }
2661 }
2662
2663 /* add a count for permanent */
2664 if (comp->Attributes & msidbComponentAttributesPermanent)
2665 count ++;
2666
2667 comp->RefCount = count;
2668
2669 if (write)
2670 ACTION_WriteSharedDLLsCount( comp->FullKeypath, comp->RefCount );
2671 }
2672
2673 static UINT ACTION_ProcessComponents(MSIPACKAGE *package)
2674 {
2675 WCHAR squished_pc[GUID_SIZE];
2676 WCHAR squished_cc[GUID_SIZE];
2677 UINT rc;
2678 MSICOMPONENT *comp;
2679 HKEY hkey;
2680
2681 TRACE("\n");
2682
2683 squash_guid(package->ProductCode,squished_pc);
2684 ui_progress(package,1,COMPONENT_PROGRESS_VALUE,1,0);
2685
2686 LIST_FOR_EACH_ENTRY( comp, &package->components, MSICOMPONENT, entry )
2687 {
2688 MSIRECORD * uirow;
2689
2690 ui_progress(package,2,0,0,0);
2691 if (!comp->ComponentId)
2692 continue;
2693
2694 squash_guid(comp->ComponentId,squished_cc);
2695
2696 msi_free(comp->FullKeypath);
2697 comp->FullKeypath = resolve_keypath( package, comp );
2698
2699 ACTION_RefCountComponent( package, comp );
2700
2701 TRACE("Component %s (%s), Keypath=%s, RefCount=%i\n",
2702 debugstr_w(comp->Component),
2703 debugstr_w(squished_cc),
2704 debugstr_w(comp->FullKeypath),
2705 comp->RefCount);
2706
2707 if (ACTION_VerifyComponentForAction( comp, INSTALLSTATE_LOCAL) ||
2708 ACTION_VerifyComponentForAction( comp, INSTALLSTATE_SOURCE))
2709 {
2710 if (!comp->FullKeypath)
2711 continue;
2712
2713 if (package->Context == MSIINSTALLCONTEXT_MACHINE)
2714 rc = MSIREG_OpenUserDataComponentKey(comp->ComponentId, szLocalSid,
2715 &hkey, TRUE);
2716 else
2717 rc = MSIREG_OpenUserDataComponentKey(comp->ComponentId, NULL,
2718 &hkey, TRUE);
2719
2720 if (rc != ERROR_SUCCESS)
2721 continue;
2722
2723 if (comp->Attributes & msidbComponentAttributesPermanent)
2724 {
2725 static const WCHAR szPermKey[] =
2726 { '0','0','0','0','0','0','0','0','0','0','0','0',
2727 '0','0','0','0','0','0','0','0','0','0','0','0',
2728 '0','0','0','0','0','0','0','0',0 };
2729
2730 msi_reg_set_val_str(hkey, szPermKey, comp->FullKeypath);
2731 }
2732
2733 if (comp->Action == INSTALLSTATE_LOCAL)
2734 msi_reg_set_val_str(hkey, squished_pc, comp->FullKeypath);
2735 else
2736 {
2737 MSIFILE *file;
2738 MSIRECORD *row;
2739 LPWSTR ptr, ptr2;
2740 WCHAR source[MAX_PATH];
2741 WCHAR base[MAX_PATH];
2742 LPWSTR sourcepath;
2743
2744 static const WCHAR fmt[] = {'%','0','2','d','\\',0};
2745 static const WCHAR query[] = {
2746 'S','E','L','E','C','T',' ','*',' ', 'F','R','O','M',' ',
2747 '`','M','e','d','i','a','`',' ','W','H','E','R','E',' ',
2748 '`','L','a','s','t','S','e','q','u','e','n','c','e','`',' ',
2749 '>','=',' ','%','i',' ','O','R','D','E','R',' ','B','Y',' ',
2750 '`','D','i','s','k','I','d','`',0};
2751
2752 file = get_loaded_file(package, comp->KeyPath);
2753 if (!file)
2754 continue;
2755
2756 row = MSI_QueryGetRecord(package->db, query, file->Sequence);
2757 sprintfW(source, fmt, MSI_RecordGetInteger(row, 1));
2758 ptr2 = strrchrW(source, '\\') + 1;
2759 msiobj_release(&row->hdr);
2760
2761 lstrcpyW(base, package->PackagePath);
2762 ptr = strrchrW(base, '\\');
2763 *(ptr + 1) = '\0';
2764
2765 sourcepath = resolve_file_source(package, file);
2766 ptr = sourcepath + lstrlenW(base);
2767 lstrcpyW(ptr2, ptr);
2768 msi_free(sourcepath);
2769
2770 msi_reg_set_val_str(hkey, squished_pc, source);
2771 }
2772 RegCloseKey(hkey);
2773 }
2774 else if (ACTION_VerifyComponentForAction(comp, INSTALLSTATE_ABSENT))
2775 {
2776 if (package->Context == MSIINSTALLCONTEXT_MACHINE)
2777 MSIREG_DeleteUserDataComponentKey(comp->ComponentId, szLocalSid);
2778 else
2779 MSIREG_DeleteUserDataComponentKey(comp->ComponentId, NULL);
2780 }
2781
2782 /* UI stuff */
2783 uirow = MSI_CreateRecord(3);
2784 MSI_RecordSetStringW(uirow,1,package->ProductCode);
2785 MSI_RecordSetStringW(uirow,2,comp->ComponentId);
2786 MSI_RecordSetStringW(uirow,3,comp->FullKeypath);
2787 ui_actiondata(package,szProcessComponents,uirow);
2788 msiobj_release( &uirow->hdr );
2789 }
2790
2791 return ERROR_SUCCESS;
2792 }
2793
2794 typedef struct {
2795 CLSID clsid;
2796 LPWSTR source;
2797
2798 LPWSTR path;
2799 ITypeLib *ptLib;
2800 } typelib_struct;
2801
2802 static BOOL CALLBACK Typelib_EnumResNameProc( HMODULE hModule, LPCWSTR lpszType,
2803 LPWSTR lpszName, LONG_PTR lParam)
2804 {
2805 TLIBATTR *attr;
2806 typelib_struct *tl_struct = (typelib_struct*) lParam;
2807 static const WCHAR fmt[] = {'%','s','\\','%','i',0};
2808 int sz;
2809 HRESULT res;
2810
2811 if (!IS_INTRESOURCE(lpszName))
2812 {
2813 ERR("Not Int Resource Name %s\n",debugstr_w(lpszName));
2814 return TRUE;
2815 }
2816
2817 sz = strlenW(tl_struct->source)+4;
2818 sz *= sizeof(WCHAR);
2819
2820 if ((INT_PTR)lpszName == 1)
2821 tl_struct->path = strdupW(tl_struct->source);
2822 else
2823 {
2824 tl_struct->path = msi_alloc(sz);
2825 sprintfW(tl_struct->path,fmt,tl_struct->source, lpszName);
2826 }
2827
2828 TRACE("trying %s\n", debugstr_w(tl_struct->path));
2829 res = LoadTypeLib(tl_struct->path,&tl_struct->ptLib);
2830 if (FAILED(res))
2831 {
2832 msi_free(tl_struct->path);
2833 tl_struct->path = NULL;
2834
2835 return TRUE;
2836 }
2837
2838 ITypeLib_GetLibAttr(tl_struct->ptLib, &attr);
2839 if (IsEqualGUID(&(tl_struct->clsid),&(attr->guid)))
2840 {
2841 ITypeLib_ReleaseTLibAttr(tl_struct->ptLib, attr);
2842 return FALSE;
2843 }
2844
2845 msi_free(tl_struct->path);
2846 tl_struct->path = NULL;
2847
2848 ITypeLib_ReleaseTLibAttr(tl_struct->ptLib, attr);
2849 ITypeLib_Release(tl_struct->ptLib);
2850
2851 return TRUE;
2852 }
2853
2854 static UINT ITERATE_RegisterTypeLibraries(MSIRECORD *row, LPVOID param)
2855 {
2856 MSIPACKAGE* package = param;
2857 LPCWSTR component;
2858 MSICOMPONENT *comp;
2859 MSIFILE *file;
2860 typelib_struct tl_struct;
2861 ITypeLib *tlib;
2862 HMODULE module;
2863 HRESULT hr;
2864
2865 component = MSI_RecordGetString(row,3);
2866 comp = get_loaded_component(package,component);
2867 if (!comp)
2868 return ERROR_SUCCESS;
2869
2870 if (!ACTION_VerifyComponentForAction( comp, INSTALLSTATE_LOCAL))
2871 {
2872 TRACE("Skipping typelib reg due to disabled component\n");
2873
2874 comp->Action = comp->Installed;
2875
2876 return ERROR_SUCCESS;
2877 }
2878
2879 comp->Action = INSTALLSTATE_LOCAL;
2880
2881 file = get_loaded_file( package, comp->KeyPath );
2882 if (!file)
2883 return ERROR_SUCCESS;
2884
2885 module = LoadLibraryExW( file->TargetPath, NULL, LOAD_LIBRARY_AS_DATAFILE );
2886 if (module)
2887 {
2888 LPCWSTR guid;
2889 guid = MSI_RecordGetString(row,1);
2890 CLSIDFromString((LPWSTR)guid, &tl_struct.clsid);
2891 tl_struct.source = strdupW( file->TargetPath );
2892 tl_struct.path = NULL;
2893
2894 EnumResourceNamesW(module, szTYPELIB, Typelib_EnumResNameProc,
2895 (LONG_PTR)&tl_struct);
2896
2897 if (tl_struct.path)
2898 {
2899 LPWSTR help = NULL;
2900 LPCWSTR helpid;
2901 HRESULT res;
2902
2903 helpid = MSI_RecordGetString(row,6);
2904
2905 if (helpid)
2906 help = resolve_folder(package,helpid,FALSE,FALSE,TRUE,NULL);
2907 res = RegisterTypeLib(tl_struct.ptLib,tl_struct.path,help);
2908 msi_free(help);
2909
2910 if (FAILED(res))
2911 ERR("Failed to register type library %s\n",
2912 debugstr_w(tl_struct.path));
2913 else
2914 {
2915 ui_actiondata(package,szRegisterTypeLibraries,row);
2916
2917 TRACE("Registered %s\n", debugstr_w(tl_struct.path));
2918 }
2919
2920 ITypeLib_Release(tl_struct.ptLib);
2921 msi_free(tl_struct.path);
2922 }
2923 else
2924 ERR("Failed to load type library %s\n",
2925 debugstr_w(tl_struct.source));
2926
2927 FreeLibrary(module);
2928 msi_free(tl_struct.source);
2929 }
2930 else
2931 {
2932 hr = LoadTypeLibEx(file->TargetPath, REGKIND_REGISTER, &tlib);
2933 if (FAILED(hr))
2934 {
2935 ERR("Failed to load type library: %08x\n", hr);
2936 return ERROR_FUNCTION_FAILED;
2937 }
2938
2939 ITypeLib_Release(tlib);
2940 }
2941
2942 return ERROR_SUCCESS;
2943 }
2944
2945 static UINT ACTION_RegisterTypeLibraries(MSIPACKAGE *package)
2946 {
2947 /*
2948 * OK this is a bit confusing.. I am given a _Component key and I believe
2949 * that the file that is being registered as a type library is the "key file
2950 * of that component" which I interpret to mean "The file in the KeyPath of
2951 * that component".
2952 */
2953 UINT rc;
2954 MSIQUERY * view;
2955 static const WCHAR Query[] =
2956 {'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
2957 '`','T','y','p','e','L','i','b','`',0};
2958
2959 rc = MSI_DatabaseOpenViewW(package->db, Query, &view);
2960 if (rc != ERROR_SUCCESS)
2961 return ERROR_SUCCESS;
2962
2963 rc = MSI_IterateRecords(view, NULL, ITERATE_RegisterTypeLibraries, package);
2964 msiobj_release(&view->hdr);
2965 return rc;
2966 }
2967
2968 static UINT ITERATE_UnregisterTypeLibraries( MSIRECORD *row, LPVOID param )
2969 {
2970 MSIPACKAGE *package = param;
2971 LPCWSTR component, guid;
2972 MSICOMPONENT *comp;
2973 GUID libid;
2974 UINT version;
2975 LCID language;
2976 SYSKIND syskind;
2977 HRESULT hr;
2978
2979 component = MSI_RecordGetString( row, 3 );
2980 comp = get_loaded_component( package, component );
2981 if (!comp)
2982 return ERROR_SUCCESS;
2983
2984 if (!ACTION_VerifyComponentForAction( comp, INSTALLSTATE_ABSENT ))
2985 {
2986 TRACE("Skipping, component is not scheduled for uninstall\n");
2987
2988 comp->Action = comp->Installed;
2989 return ERROR_SUCCESS;
2990 }
2991 comp->Action = INSTALLSTATE_ABSENT;
2992
2993 guid = MSI_RecordGetString( row, 1 );
2994 CLSIDFromString( (LPWSTR)guid, &libid );
2995 version = MSI_RecordGetInteger( row, 4 );
2996 language = MSI_RecordGetInteger( row, 2 );
2997
2998 #ifdef _WIN64
2999 syskind = SYS_WIN64;
3000 #else
3001 syskind = SYS_WIN32;
3002 #endif
3003
3004 hr = UnRegisterTypeLib( &libid, (version >> 8) & 0xffff, version & 0xff, language, syskind );
3005 if (FAILED(hr))
3006 {
3007 WARN("Failed to unregister typelib: %08x\n", hr);
3008 }
3009
3010 return ERROR_SUCCESS;
3011 }
3012
3013 static UINT ACTION_UnregisterTypeLibraries( MSIPACKAGE *package )
3014 {
3015 UINT rc;
3016 MSIQUERY *view;
3017 static const WCHAR query[] =
3018 {'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
3019 '`','T','y','p','e','L','i','b','`',0};
3020
3021 rc = MSI_DatabaseOpenViewW( package->db, query, &view );
3022 if (rc != ERROR_SUCCESS)
3023 return ERROR_SUCCESS;
3024
3025 rc = MSI_IterateRecords( view, NULL, ITERATE_UnregisterTypeLibraries, package );
3026 msiobj_release( &view->hdr );
3027 return rc;
3028 }
3029
3030 static UINT ITERATE_CreateShortcuts(MSIRECORD *row, LPVOID param)
3031 {
3032 MSIPACKAGE *package = param;
3033 LPWSTR target_file, target_folder, filename;
3034 LPCWSTR buffer, extension;
3035 MSICOMPONENT *comp;
3036 static const WCHAR szlnk[]={'.','l','n','k',0};
3037 IShellLinkW *sl = NULL;
3038 IPersistFile *pf = NULL;
3039 HRESULT res;
3040
3041 buffer = MSI_RecordGetString(row,4);
3042 comp = get_loaded_component(package,buffer);
3043 if (!comp)
3044 return ERROR_SUCCESS;
3045
3046 if (!ACTION_VerifyComponentForAction( comp, INSTALLSTATE_LOCAL ))
3047 {
3048 TRACE("Skipping shortcut creation due to disabled component\n");
3049
3050 comp->Action = comp->Installed;
3051
3052 return ERROR_SUCCESS;
3053 }
3054
3055 comp->Action = INSTALLSTATE_LOCAL;
3056
3057 ui_actiondata(package,szCreateShortcuts,row);
3058
3059 res = CoCreateInstance( &CLSID_ShellLink, NULL, CLSCTX_INPROC_SERVER,
3060 &IID_IShellLinkW, (LPVOID *) &sl );
3061
3062 if (FAILED( res ))
3063 {
3064 ERR("CLSID_ShellLink not available\n");
3065 goto err;
3066 }
3067
3068 res = IShellLinkW_QueryInterface( sl, &IID_IPersistFile,(LPVOID*) &pf );
3069 if (FAILED( res ))
3070 {
3071 ERR("QueryInterface(IID_IPersistFile) failed\n");
3072 goto err;
3073 }
3074
3075 buffer = MSI_RecordGetString(row,2);
3076 target_folder = resolve_folder(package, buffer,FALSE,FALSE,TRUE,NULL);
3077
3078 /* may be needed because of a bug somewhere else */
3079 create_full_pathW(target_folder);
3080
3081 filename = msi_dup_record_field( row, 3 );
3082 reduce_to_longfilename(filename);
3083
3084 extension = strchrW(filename,'.');
3085 if (!extension || strcmpiW(extension,szlnk))
3086 {
3087 int len = strlenW(filename);
3088 filename = msi_realloc(filename, len * sizeof(WCHAR) + sizeof(szlnk));
3089 memcpy(filename + len, szlnk, sizeof(szlnk));
3090 }
3091 target_file = build_directory_name(2, target_folder, filename);
3092 msi_free(target_folder);
3093 msi_free(filename);
3094
3095 buffer = MSI_RecordGetString(row,5);
3096 if (strchrW(buffer,'['))
3097 {
3098 LPWSTR deformated;
3099 deformat_string(package,buffer,&deformated);
3100 IShellLinkW_SetPath(sl,deformated);
3101 msi_free(deformated);
3102 }
3103 else
3104 {
3105 FIXME("poorly handled shortcut format, advertised shortcut\n");
3106 IShellLinkW_SetPath(sl,comp->FullKeypath);
3107 }
3108
3109 if (!MSI_RecordIsNull(row,6))
3110 {
3111 LPWSTR deformated;
3112 buffer = MSI_RecordGetString(row,6);
3113 deformat_string(package,buffer,&deformated);
3114 IShellLinkW_SetArguments(sl,deformated);
3115 msi_free(deformated);
3116 }
3117
3118 if (!MSI_RecordIsNull(row,7))
3119 {
3120 buffer = MSI_RecordGetString(row,7);
3121 IShellLinkW_SetDescription(sl,buffer);
3122 }
3123
3124 if (!MSI_RecordIsNull(row,8))
3125 IShellLinkW_SetHotkey(sl,MSI_RecordGetInteger(row,8));
3126
3127 if (!MSI_RecordIsNull(row,9))
3128 {
3129 LPWSTR Path;
3130 INT index;
3131
3132 buffer = MSI_RecordGetString(row,9);
3133
3134 Path = build_icon_path(package,buffer);
3135 index = MSI_RecordGetInteger(row,10);
3136
3137 /* no value means 0 */
3138 if (index == MSI_NULL_INTEGER)
3139 index = 0;
3140
3141 IShellLinkW_SetIconLocation(sl,Path,index);
3142 msi_free(Path);
3143 }
3144
3145 if (!MSI_RecordIsNull(row,11))
3146 IShellLinkW_SetShowCmd(sl,MSI_RecordGetInteger(row,11));
3147
3148 if (!MSI_RecordIsNull(row,12))
3149 {
3150 LPWSTR Path;
3151 buffer = MSI_RecordGetString(row,12);
3152 Path = resolve_folder(package, buffer, FALSE, FALSE, TRUE, NULL);
3153 if (Path)
3154 IShellLinkW_SetWorkingDirectory(sl,Path);
3155 msi_free(Path);
3156 }
3157
3158 TRACE("Writing shortcut to %s\n",debugstr_w(target_file));
3159 IPersistFile_Save(pf,target_file,FALSE);
3160
3161 msi_free(target_file);
3162
3163 err:
3164 if (pf)
3165 IPersistFile_Release( pf );
3166 if (sl)
3167 IShellLinkW_Release( sl );
3168
3169 return ERROR_SUCCESS;
3170 }
3171
3172 static UINT ACTION_CreateShortcuts(MSIPACKAGE *package)
3173 {
3174 UINT rc;
3175 HRESULT res;
3176 MSIQUERY * view;
3177 static const WCHAR Query[] =
3178 {'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
3179 '`','S','h','o','r','t','c','u','t','`',0};
3180
3181 rc = MSI_DatabaseOpenViewW(package->db, Query, &view);
3182 if (rc != ERROR_SUCCESS)
3183 return ERROR_SUCCESS;
3184
3185 res = CoInitialize( NULL );
3186
3187 rc = MSI_IterateRecords(view, NULL, ITERATE_CreateShortcuts, package);
3188 msiobj_release(&view->hdr);
3189
3190 if (SUCCEEDED(res))
3191 CoUninitialize();
3192
3193 return rc;
3194 }
3195
3196 static UINT ITERATE_PublishIcon(MSIRECORD *row, LPVOID param)
3197 {
3198 MSIPACKAGE* package = param;
3199 HANDLE the_file;
3200 LPWSTR FilePath;
3201 LPCWSTR FileName;
3202 CHAR buffer[1024];
3203 DWORD sz;
3204 UINT rc;
3205 MSIRECORD *uirow;
3206
3207 FileName = MSI_RecordGetString(row,1);
3208 if (!FileName)
3209 {
3210 ERR("Unable to get FileName\n");
3211 return ERROR_SUCCESS;
3212 }
3213
3214 FilePath = build_icon_path(package,FileName);
3215
3216 TRACE("Creating icon file at %s\n",debugstr_w(FilePath));
3217
3218 the_file = CreateFileW(FilePath, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS,
3219 FILE_ATTRIBUTE_NORMAL, NULL);
3220
3221 if (the_file == INVALID_HANDLE_VALUE)
3222 {
3223 ERR("Unable to create file %s\n",debugstr_w(FilePath));
3224 msi_free(FilePath);
3225 return ERROR_SUCCESS;
3226 }
3227
3228 do
3229 {
3230 DWORD write;
3231 sz = 1024;
3232 rc = MSI_RecordReadStream(row,2,buffer,&sz);
3233 if (rc != ERROR_SUCCESS)
3234 {
3235 ERR("Failed to get stream\n");
3236 CloseHandle(the_file);
3237 DeleteFileW(FilePath);
3238 break;
3239 }
3240 WriteFile(the_file,buffer,sz,&write,NULL);
3241 } while (sz == 1024);
3242
3243 msi_free(FilePath);
3244
3245 CloseHandle(the_file);
3246
3247 uirow = MSI_CreateRecord(1);
3248 MSI_RecordSetStringW(uirow,1,FileName);
3249 ui_actiondata(package,szPublishProduct,uirow);
3250 msiobj_release( &uirow->hdr );
3251
3252 return ERROR_SUCCESS;
3253 }
3254
3255 static UINT msi_publish_icons(MSIPACKAGE *package)
3256 {
3257 UINT r;
3258 MSIQUERY *view;
3259
3260 static const WCHAR query[]= {
3261 'S','E','L','E','C','T',' ','*',' ',
3262 'F','R','O','M',' ','`','I','c','o','n','`',0};
3263
3264 r = MSI_DatabaseOpenViewW(package->db, query, &view);
3265 if (r == ERROR_SUCCESS)
3266 {
3267 MSI_IterateRecords(view, NULL, ITERATE_PublishIcon, package);
3268 msiobj_release(&view->hdr);
3269 }
3270
3271 return ERROR_SUCCESS;
3272 }
3273
3274 static UINT msi_publish_sourcelist(MSIPACKAGE *package, HKEY hkey)
3275 {
3276 UINT r;
3277 HKEY source;
3278 LPWSTR buffer;
3279 MSIMEDIADISK *disk;
3280 MSISOURCELISTINFO *info;
3281
3282 r = RegCreateKeyW(hkey, szSourceList, &source);
3283 if (r != ERROR_SUCCESS)
3284 return r;
3285
3286 RegCloseKey(source);
3287
3288 buffer = strrchrW(package->PackagePath, '\\') + 1;
3289 r = MsiSourceListSetInfoW(package->ProductCode, NULL,
3290 package->Context, MSICODE_PRODUCT,
3291 INSTALLPROPERTY_PACKAGENAMEW, buffer);
3292 if (r != ERROR_SUCCESS)
3293 return r;
3294
3295 r = MsiSourceListSetInfoW(package->ProductCode, NULL,
3296 package->Context, MSICODE_PRODUCT,
3297 INSTALLPROPERTY_MEDIAPACKAGEPATHW, szEmpty);
3298 if (r != ERROR_SUCCESS)
3299 return r;
3300
3301 r = MsiSourceListSetInfoW(package->ProductCode, NULL,
3302 package->Context, MSICODE_PRODUCT,
3303 INSTALLPROPERTY_DISKPROMPTW, szEmpty);
3304 if (r != ERROR_SUCCESS)
3305 return r;
3306
3307 LIST_FOR_EACH_ENTRY(info, &package->sourcelist_info, MSISOURCELISTINFO, entry)
3308 {
3309 if (!lstrcmpW(info->property, INSTALLPROPERTY_LASTUSEDSOURCEW))
3310 msi_set_last_used_source(package->ProductCode, NULL, info->context,
3311 info->options, info->value);
3312 else
3313 MsiSourceListSetInfoW(package->ProductCode, NULL,
3314 info->context, info->options,
3315 info->property, info->value);
3316 }
3317
3318 LIST_FOR_EACH_ENTRY(disk, &package->sourcelist_media, MSIMEDIADISK, entry)
3319 {
3320 MsiSourceListAddMediaDiskW(package->ProductCode, NULL,
3321 disk->context, disk->options,
3322 disk->disk_id, disk->volume_label, disk->disk_prompt);
3323 }
3324
3325 return ERROR_SUCCESS;
3326 }
3327
3328 static UINT msi_publish_product_properties(MSIPACKAGE *package, HKEY hkey)
3329 {
3330 MSIHANDLE hdb, suminfo;
3331 WCHAR guids[MAX_PATH];
3332 WCHAR packcode[SQUISH_GUID_SIZE];
3333 LPWSTR buffer;
3334 LPWSTR ptr;
3335 DWORD langid;
3336 DWORD size;
3337 UINT r;
3338
3339 static const WCHAR szProductLanguage[] =
3340 {'P','r','o','d','u','c','t','L','a','n','g','u','a','g','e',0};
3341 static const WCHAR szARPProductIcon[] =
3342 {'A','R','P','P','R','O','D','U','C','T','I','C','O','N',0};
3343 static const WCHAR szProductVersion[] =
3344 {'P','r','o','d','u','c','t','V','e','r','s','i','o','n',0};
3345 static const WCHAR szAssignment[] =
3346 {'A','s','s','i','g','n','m','e','n','t',0};
3347 static const WCHAR szAdvertiseFlags[] =
3348 {'A','d','v','e','r','t','i','s','e','F','l','a','g','s',0};
3349 static const WCHAR szClients[] =
3350 {'C','l','i','e','n','t','s',0};
3351 static const WCHAR szColon[] = {':',0};
3352
3353 buffer = msi_dup_property(package, INSTALLPROPERTY_PRODUCTNAMEW);
3354 msi_reg_set_val_str(hkey, INSTALLPROPERTY_PRODUCTNAMEW, buffer);
3355 msi_free(buffer);
3356
3357 langid = msi_get_property_int(package, szProductLanguage, 0);
3358 msi_reg_set_val_dword(hkey, INSTALLPROPERTY_LANGUAGEW, langid);
3359
3360 /* FIXME */
3361 msi_reg_set_val_dword(hkey, INSTALLPROPERTY_AUTHORIZED_LUA_APPW, 0);
3362
3363 buffer = msi_dup_property(package, szARPProductIcon);
3364 if (buffer)
3365 {
3366 LPWSTR path = build_icon_path(package,buffer);
3367 msi_reg_set_val_str(hkey, INSTALLPROPERTY_PRODUCTICONW, path);
3368 msi_free(path);
3369 msi_free(buffer);
3370 }
3371
3372 buffer = msi_dup_property(package, szProductVersion);
3373 if (buffer)
3374 {
3375 DWORD verdword = msi_version_str_to_dword(buffer);
3376 msi_reg_set_val_dword(hkey, INSTALLPROPERTY_VERSIONW, verdword);
3377 msi_free(buffer);
3378 }
3379
3380 msi_reg_set_val_dword(hkey, szAssignment, 0);
3381 msi_reg_set_val_dword(hkey, szAdvertiseFlags, 0x184);
3382 msi_reg_set_val_dword(hkey, INSTALLPROPERTY_INSTANCETYPEW, 0);
3383 msi_reg_set_val_str(hkey, szClients, szColon);
3384
3385 hdb = alloc_msihandle(&package->db->hdr);
3386 if (!hdb)
3387 return ERROR_NOT_ENOUGH_MEMORY;
3388
3389 r = MsiGetSummaryInformationW(hdb, NULL, 0, &suminfo);
3390 MsiCloseHandle(hdb);
3391 if (r != ERROR_SUCCESS)
3392 goto done;
3393
3394 size = MAX_PATH;
3395 r = MsiSummaryInfoGetPropertyW(suminfo, PID_REVNUMBER, NULL, NULL,
3396 NULL, guids, &size);
3397 if (r != ERROR_SUCCESS)
3398 goto done;
3399
3400 ptr = strchrW(guids, ';');
3401 if (ptr) *ptr = 0;
3402 squash_guid(guids, packcode);
3403 msi_reg_set_val_str(hkey, INSTALLPROPERTY_PACKAGECODEW, packcode);
3404
3405 done:
3406 MsiCloseHandle(suminfo);
3407 return ERROR_SUCCESS;
3408 }
3409
3410 static UINT msi_publish_upgrade_code(MSIPACKAGE *package)
3411 {
3412 UINT r;
3413 HKEY hkey;
3414 LPWSTR upgrade;
3415 WCHAR squashed_pc[SQUISH_GUID_SIZE];
3416
3417 static const WCHAR szUpgradeCode[] =
3418 {'U','p','g','r','a','d','e','C','o','d','e',0};
3419
3420 upgrade = msi_dup_property(package, szUpgradeCode);
3421 if (!upgrade)
3422 return ERROR_SUCCESS;
3423
3424 if (package->Context == MSIINSTALLCONTEXT_MACHINE)
3425 {
3426 r = MSIREG_OpenClassesUpgradeCodesKey(upgrade, &hkey, TRUE);
3427 if (r != ERROR_SUCCESS)
3428 goto done;
3429 }
3430 else
3431 {
3432 r = MSIREG_OpenUserUpgradeCodesKey(upgrade, &hkey, TRUE);
3433 if (r != ERROR_SUCCESS)
3434 goto done;
3435 }
3436
3437 squash_guid(package->ProductCode, squashed_pc);
3438 msi_reg_set_val_str(hkey, squashed_pc, NULL);
3439
3440 RegCloseKey(hkey);
3441
3442 done:
3443 msi_free(upgrade);
3444 return r;
3445 }
3446
3447 static BOOL msi_check_publish(MSIPACKAGE *package)
3448 {
3449 MSIFEATURE *feature;
3450
3451 LIST_FOR_EACH_ENTRY(feature, &package->features, MSIFEATURE, entry)
3452 {
3453 if (feature->ActionRequest == INSTALLSTATE_LOCAL)
3454 return TRUE;
3455 }
3456
3457 return FALSE;
3458 }
3459
3460 static BOOL msi_check_unpublish(MSIPACKAGE *package)
3461 {
3462 MSIFEATURE *feature;
3463
3464 LIST_FOR_EACH_ENTRY(feature, &package->features, MSIFEATURE, entry)
3465 {
3466 if (feature->ActionRequest != INSTALLSTATE_ABSENT)
3467 return FALSE;
3468 }
3469
3470 return TRUE;
3471 }
3472
3473 static UINT msi_publish_patch(MSIPACKAGE *package, HKEY prodkey, HKEY hudkey)
3474 {
3475 WCHAR patch_squashed[GUID_SIZE];
3476 HKEY patches;
3477 LONG res;
3478 UINT r = ERROR_FUNCTION_FAILED;
3479
3480 res = RegCreateKeyExW(prodkey, szPatches, 0, NULL, 0, KEY_ALL_ACCESS, NULL,
3481 &patches, NULL);
3482 if (res != ERROR_SUCCESS)
3483 return ERROR_FUNCTION_FAILED;
3484
3485 squash_guid(package->patch->patchcode, patch_squashed);
3486
3487 res = RegSetValueExW(patches, szPatches, 0, REG_MULTI_SZ,
3488 (const BYTE *)patch_squashed,
3489 (lstrlenW(patch_squashed) + 1) * sizeof(WCHAR));
3490 if (res != ERROR_SUCCESS)
3491 goto done;
3492
3493 res = RegSetValueExW(patches, patch_squashed, 0, REG_SZ,
3494 (const BYTE *)package->patch->transforms,
3495 (lstrlenW(package->patch->transforms) + 1) * sizeof(WCHAR));
3496 if (res == ERROR_SUCCESS)
3497 r = ERROR_SUCCESS;
3498
3499 done:
3500 RegCloseKey(patches);
3501 return r;
3502 }
3503
3504 /*
3505 * 99% of the work done here is only done for
3506 * advertised installs. However this is where the
3507 * Icon table is processed and written out
3508 * so that is what I am going to do here.
3509 */
3510 static UINT ACTION_PublishProduct(MSIPACKAGE *package)
3511 {
3512 UINT rc;
3513 HKEY hukey=0;
3514 HKEY hudkey=0;
3515
3516 /* FIXME: also need to publish if the product is in advertise mode */
3517 if (!msi_check_publish(package))
3518 return ERROR_SUCCESS;
3519
3520 rc = MSIREG_OpenProductKey(package->ProductCode, NULL, package->Context,
3521 &hukey, TRUE);
3522 if (rc != ERROR_SUCCESS)
3523 goto end;
3524
3525 rc = MSIREG_OpenUserDataProductKey(package->ProductCode, package->Context,
3526 NULL, &hudkey, TRUE);
3527 if (rc != ERROR_SUCCESS)
3528 goto end;
3529
3530 rc = msi_publish_upgrade_code(package);
3531 if (rc != ERROR_SUCCESS)
3532 goto end;
3533
3534 if (package->patch)
3535 {
3536 rc = msi_publish_patch(package, hukey, hudkey);
3537 if (rc != ERROR_SUCCESS)
3538 goto end;
3539 }
3540
3541 rc = msi_publish_product_properties(package, hukey);
3542 if (rc != ERROR_SUCCESS)
3543 goto end;
3544
3545 rc = msi_publish_sourcelist(package, hukey);
3546 if (rc != ERROR_SUCCESS)
3547 goto end;
3548
3549 rc = msi_publish_icons(package);
3550
3551 end:
3552 RegCloseKey(hukey);
3553 RegCloseKey(hudkey);
3554
3555 return rc;
3556 }
3557
3558 static UINT ITERATE_WriteIniValues(MSIRECORD *row, LPVOID param)
3559 {
3560 MSIPACKAGE *package = param;
3561 LPCWSTR component, section, key, value, identifier, dirproperty;
3562 LPWSTR deformated_section, deformated_key, deformated_value;
3563 LPWSTR folder, filename, fullname = NULL;
3564 LPCWSTR filenameptr;
3565 MSIRECORD * uirow;
3566 INT action;
3567 MSICOMPONENT *comp;
3568 static const WCHAR szWindowsFolder[] =
3569 {'W','i','n','d','o','w','s','F','o','l','d','e','r',0};
3570
3571 component = MSI_RecordGetString(row, 8);
3572 comp = get_loaded_component(package,component);
3573
3574 if (!ACTION_VerifyComponentForAction( comp, INSTALLSTATE_LOCAL))
3575 {
3576 TRACE("Skipping ini file due to disabled component %s\n",
3577 debugstr_w(component));
3578
3579 comp->Action = comp->Installed;
3580
3581 return ERROR_SUCCESS;
3582 }
3583
3584 comp->Action = INSTALLSTATE_LOCAL;
3585
3586 identifier = MSI_RecordGetString(row,1);
3587 dirproperty = MSI_RecordGetString(row,3);
3588 section = MSI_RecordGetString(row,4);
3589 key = MSI_RecordGetString(row,5);
3590 value = MSI_RecordGetString(row,6);
3591 action = MSI_RecordGetInteger(row,7);
3592
3593 deformat_string(package,section,&deformated_section);
3594 deformat_string(package,key,&deformated_key);
3595 deformat_string(package,value,&deformated_value);
3596
3597 filename = msi_dup_record_field(row, 2);
3598 if (filename && (filenameptr = strchrW(filename, '|')))
3599 filenameptr++;
3600 else
3601 filenameptr = filename;
3602
3603 if (dirproperty)
3604 {
3605 folder = resolve_folder(package, dirproperty, FALSE, FALSE, TRUE, NULL);
3606 if (!folder)
3607 folder = msi_dup_property( package, dirproperty );
3608 }
3609 else
3610 folder = msi_dup_property( package, szWindowsFolder );
3611
3612 if (!folder)
3613 {
3614 ERR("Unable to resolve folder! (%s)\n",debugstr_w(dirproperty));
3615 goto cleanup;
3616 }
3617
3618 fullname = build_directory_name(2, folder, filenameptr);
3619
3620 if (action == 0)
3621 {
3622 TRACE("Adding value %s to section %s in %s\n",
3623 debugstr_w(deformated_key), debugstr_w(deformated_section),
3624 debugstr_w(fullname));
3625 WritePrivateProfileStringW(deformated_section, deformated_key,
3626 deformated_value, fullname);
3627 }
3628 else if (action == 1)
3629 {
3630 WCHAR returned[10];
3631 GetPrivateProfileStringW(deformated_section, deformated_key, NULL,
3632 returned, 10, fullname);
3633 if (returned[0] == 0)
3634 {
3635 TRACE("Adding value %s to section %s in %s\n",
3636 debugstr_w(deformated_key), debugstr_w(deformated_section),
3637 debugstr_w(fullname));
3638
3639 WritePrivateProfileStringW(deformated_section, deformated_key,
3640 deformated_value, fullname);
3641 }
3642 }
3643 else if (action == 3)
3644 FIXME("Append to existing section not yet implemented\n");
3645
3646 uirow = MSI_CreateRecord(4);
3647 MSI_RecordSetStringW(uirow,1,identifier);
3648 MSI_RecordSetStringW(uirow,2,deformated_section);
3649 MSI_RecordSetStringW(uirow,3,deformated_key);
3650 MSI_RecordSetStringW(uirow,4,deformated_value);
3651 ui_actiondata(package,szWriteIniValues,uirow);
3652 msiobj_release( &uirow->hdr );
3653
3654 cleanup:
3655 msi_free(filename);
3656 msi_free(fullname);
3657 msi_free(folder);
3658 msi_free(deformated_key);
3659 msi_free(deformated_value);
3660 msi_free(deformated_section);
3661 return ERROR_SUCCESS;
3662 }
3663
3664 static UINT ACTION_WriteIniValues(MSIPACKAGE *package)
3665 {
3666 UINT rc;
3667 MSIQUERY * view;
3668 static const WCHAR ExecSeqQuery[] =
3669 {'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
3670 '`','I','n','i','F','i','l','e','`',0};
3671
3672 rc = MSI_DatabaseOpenViewW(package->db, ExecSeqQuery, &view);
3673 if (rc != ERROR_SUCCESS)
3674 {
3675 TRACE("no IniFile table\n");
3676 return ERROR_SUCCESS;
3677 }
3678
3679 rc = MSI_IterateRecords(view, NULL, ITERATE_WriteIniValues, package);
3680 msiobj_release(&view->hdr);
3681 return rc;
3682 }
3683
3684 static UINT ITERATE_SelfRegModules(MSIRECORD *row, LPVOID param)
3685 {
3686 MSIPACKAGE *package = param;
3687 LPCWSTR filename;
3688 LPWSTR FullName;
3689 MSIFILE *file;
3690 DWORD len;
3691 static const WCHAR ExeStr[] =
3692 {'r','e','g','s','v','r','3','2','.','e','x','e',' ','\"',0};
3693 static const WCHAR close[] = {'\"',0};
3694 STARTUPINFOW si;
3695 PROCESS_INFORMATION info;
3696 BOOL brc;
3697 MSIRECORD *uirow;
3698 LPWSTR uipath, p;
3699
3700 memset(&si,0,sizeof(STARTUPINFOW));
3701
3702 filename = MSI_RecordGetString(row,1);
3703 file = get_loaded_file( package, filename );
3704
3705 if (!file)
3706 {
3707 ERR("Unable to find file id %s\n",debugstr_w(filename));
3708 return ERROR_SUCCESS;
3709 }
3710
3711 len = strlenW(ExeStr) + strlenW( file->TargetPath ) + 2;
3712
3713 FullName = msi_alloc(len*sizeof(WCHAR));
3714 strcpyW(FullName,ExeStr);
3715 strcatW( FullName, file->TargetPath );
3716 strcatW(FullName,close);
3717
3718 TRACE("Registering %s\n",debugstr_w(FullName));
3719 brc = CreateProcessW(NULL, FullName, NULL, NULL, FALSE, 0, NULL, c_colon,
3720 &si, &info);
3721
3722 if (brc)
3723 {
3724 CloseHandle(info.hThread);
3725 msi_dialog_check_messages(info.hProcess);
3726 CloseHandle(info.hProcess);
3727 }
3728
3729 msi_free(FullName);
3730
3731 /* the UI chunk */
3732 uirow = MSI_CreateRecord( 2 );
3733 uipath = strdupW( file->TargetPath );
3734 p = strrchrW(uipath,'\\');
3735 if (p)
3736 p[0]=0;
3737 MSI_RecordSetStringW( uirow, 1, &p[1] );
3738 MSI_RecordSetStringW( uirow, 2, uipath);
3739 ui_actiondata( package, szSelfRegModules, uirow);
3740 msiobj_release( &uirow->hdr );
3741 msi_free( uipath );
3742 /* FIXME: call ui_progress? */
3743
3744 return ERROR_SUCCESS;
3745 }
3746
3747 static UINT ACTION_SelfRegModules(MSIPACKAGE *package)
3748 {
3749 UINT rc;
3750 MSIQUERY * view;
3751 static const WCHAR ExecSeqQuery[] =
3752 {'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
3753 '`','S','e','l','f','R','e','g','`',0};
3754
3755 rc = MSI_DatabaseOpenViewW(package->db, ExecSeqQuery, &view);
3756 if (rc != ERROR_SUCCESS)
3757 {
3758 TRACE("no SelfReg table\n");
3759 return ERROR_SUCCESS;
3760 }
3761
3762 MSI_IterateRecords(view, NULL, ITERATE_SelfRegModules, package);
3763 msiobj_release(&view->hdr);
3764
3765 return ERROR_SUCCESS;
3766 }
3767
3768 static UINT ITERATE_SelfUnregModules( MSIRECORD *row, LPVOID param )
3769 {
3770 static const WCHAR regsvr32[] =
3771 {'r','e','g','s','v','r','3','2','.','e','x','e',' ','/','u',' ','\"',0};
3772 static const WCHAR close[] = {'\"',0};
3773 MSIPACKAGE *package = param;
3774 LPCWSTR filename;
3775 LPWSTR cmdline;
3776 MSIFILE *file;
3777 DWORD len;
3778 STARTUPINFOW si;
3779 PROCESS_INFORMATION pi;
3780 BOOL ret;
3781 MSIRECORD *uirow;
3782 LPWSTR uipath, p;
3783
3784 memset( &si, 0, sizeof(STARTUPINFOW) );
3785
3786 filename = MSI_RecordGetString( row, 1 );
3787 file = get_loaded_file( package, filename );
3788
3789 if (!file)
3790 {
3791 ERR("Unable to find file id %s\n", debugstr_w(filename));
3792 return ERROR_SUCCESS;
3793 }
3794
3795 len = strlenW( regsvr32 ) + strlenW( file->TargetPath ) + 2;
3796
3797 cmdline = msi_alloc( len * sizeof(WCHAR) );
3798 strcpyW( cmdline, regsvr32 );
3799 strcatW( cmdline, file->TargetPath );
3800 strcatW( cmdline, close );
3801
3802 TRACE("Unregistering %s\n", debugstr_w(cmdline));
3803
3804 ret = CreateProcessW( NULL, cmdline, NULL, NULL, FALSE, 0, NULL, c_colon, &si, &pi );
3805 if (ret)
3806 {
3807 CloseHandle( pi.hThread );
3808 msi_dialog_check_messages( pi.hProcess );
3809 CloseHandle( pi.hProcess );
3810 }
3811
3812 msi_free( cmdline );
3813
3814 uirow = MSI_CreateRecord( 2 );
3815 uipath = strdupW( file->TargetPath );
3816 if ((p = strrchrW( uipath, '\\' )))
3817 {
3818 *p = 0;
3819 MSI_RecordSetStringW( uirow, 1, ++p );
3820 }
3821 MSI_RecordSetStringW( uirow, 2, uipath );
3822 ui_actiondata( package, szSelfUnregModules, uirow );
3823 msiobj_release( &uirow->hdr );
3824 msi_free( uipath );
3825 /* FIXME call ui_progress? */
3826
3827 return ERROR_SUCCESS;
3828 }
3829
3830 static UINT ACTION_SelfUnregModules( MSIPACKAGE *package )
3831 {
3832 UINT rc;
3833 MSIQUERY *view;
3834 static const WCHAR query[] =
3835 {'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
3836 '`','S','e','l','f','R','e','g','`',0};
3837
3838 rc = MSI_DatabaseOpenViewW( package->db, query, &view );
3839 if (rc != ERROR_SUCCESS)
3840 {
3841 TRACE("no SelfReg table\n");
3842 return ERROR_SUCCESS;
3843 }
3844
3845 MSI_IterateRecords( view, NULL, ITERATE_SelfUnregModules, package );
3846 msiobj_release( &view->hdr );
3847
3848 return ERROR_SUCCESS;
3849 }
3850
3851 static UINT ACTION_PublishFeatures(MSIPACKAGE *package)
3852 {
3853 MSIFEATURE *feature;
3854 UINT rc;
3855 HKEY hkey;
3856 HKEY userdata = NULL;
3857
3858 if (!msi_check_publish(package))
3859 return ERROR_SUCCESS;
3860
3861 rc = MSIREG_OpenFeaturesKey(package->ProductCode, package->Context,
3862 &hkey, TRUE);
3863 if (rc != ERROR_SUCCESS)
3864 goto end;
3865
3866 rc = MSIREG_OpenUserDataFeaturesKey(package->ProductCode, package->Context,
3867 &userdata, TRUE);
3868 if (rc != ERROR_SUCCESS)
3869 goto end;
3870
3871 /* here the guids are base 85 encoded */
3872 LIST_FOR_EACH_ENTRY( feature, &package->features, MSIFEATURE, entry )
3873 {
3874 ComponentList *cl;
3875 LPWSTR data = NULL;
3876 GUID clsid;
3877 INT size;
3878 BOOL absent = FALSE;
3879 MSIRECORD *uirow;
3880
3881 if (!ACTION_VerifyFeatureForAction( feature, INSTALLSTATE_LOCAL ) &&
3882 !ACTION_VerifyFeatureForAction( feature, INSTALLSTATE_SOURCE ) &&
3883 !ACTION_VerifyFeatureForAction( feature, INSTALLSTATE_ADVERTISED ))
3884 absent = TRUE;
3885
3886 size = 1;
3887 LIST_FOR_EACH_ENTRY( cl, &feature->Components, ComponentList, entry )
3888 {
3889 size += 21;
3890 }
3891 if (feature->Feature_Parent)
3892 size += strlenW( feature->Feature_Parent )+2;
3893
3894 data = msi_alloc(size * sizeof(WCHAR));
3895
3896 data[0] = 0;
3897 LIST_FOR_EACH_ENTRY( cl, &feature->Components, ComponentList, entry )
3898 {
3899 MSICOMPONENT* component = cl->component;
3900 WCHAR buf[21];
3901
3902 buf[0] = 0;
3903 if (component->ComponentId)
3904 {
3905 TRACE("From %s\n",debugstr_w(component->ComponentId));
3906 CLSIDFromString(component->ComponentId, &clsid);
3907 encode_base85_guid(&clsid,buf);
3908 TRACE("to %s\n",debugstr_w(buf));
3909 strcatW(data,buf);
3910 }
3911 }
3912
3913 if (feature->Feature_Parent)
3914 {
3915 static const WCHAR sep[] = {'\2',0};
3916 strcatW(data,sep);
3917 strcatW(data,feature->Feature_Parent);
3918 }
3919
3920 msi_reg_set_val_str( userdata, feature->Feature, data );
3921 msi_free(data);
3922
3923 size = 0;
3924 if (feature->Feature_Parent)
3925 size = strlenW(feature->Feature_Parent)*sizeof(WCHAR);
3926 if (!absent)
3927 {
3928 size += sizeof(WCHAR);
3929 RegSetValueExW(hkey,feature->Feature,0,REG_SZ,
3930 (LPBYTE)(feature->Feature_Parent ? feature->Feature_Parent : szEmpty),size);
3931 }
3932 else
3933 {
3934 size += 2*sizeof(WCHAR);
3935 data = msi_alloc(size);
3936 data[0] = 0x6;
3937 data[1] = 0;
3938 if (feature->Feature_Parent)
3939 strcpyW( &data[1], feature->Feature_Parent );
3940 RegSetValueExW(hkey,feature->Feature,0,REG_SZ,
3941 (LPBYTE)data,size);
3942 msi_free(data);
3943 }
3944
3945 /* the UI chunk */
3946 uirow = MSI_CreateRecord( 1 );
3947 MSI_RecordSetStringW( uirow, 1, feature->Feature );
3948 ui_actiondata( package, szPublishFeatures, uirow);
3949 msiobj_release( &uirow->hdr );
3950 /* FIXME: call ui_progress? */
3951 }
3952
3953 end:
3954 RegCloseKey(hkey);
3955 RegCloseKey(userdata);
3956 return rc;
3957 }
3958
3959 static UINT msi_unpublish_feature(MSIPACKAGE *package, MSIFEATURE *feature)
3960 {
3961 UINT r;
3962 HKEY hkey;
3963
3964 TRACE("unpublishing feature %s\n", debugstr_w(feature->Feature));
3965
3966 r = MSIREG_OpenFeaturesKey(package->ProductCode, package->Context,
3967 &hkey, FALSE);
3968 if (r == ERROR_SUCCESS)
3969 {
3970 RegDeleteValueW(hkey, feature->Feature);
3971 RegCloseKey(hkey);
3972 }
3973
3974 r = MSIREG_OpenUserDataFeaturesKey(package->ProductCode, package->Context,
3975 &hkey, FALSE);
3976 if (r == ERROR_SUCCESS)
3977 {
3978 RegDeleteValueW(hkey, feature->Feature);
3979 RegCloseKey(hkey);
3980 }
3981
3982 return ERROR_SUCCESS;
3983 }
3984
3985 static UINT ACTION_UnpublishFeatures(MSIPACKAGE *package)
3986 {
3987 MSIFEATURE *feature;
3988
3989 if (!msi_check_unpublish(package))
3990 return ERROR_SUCCESS;
3991
3992 LIST_FOR_EACH_ENTRY(feature, &package->features, MSIFEATURE, entry)
3993 {
3994 msi_unpublish_feature(package, feature);
3995 }
3996
3997 return ERROR_SUCCESS;
3998 }
3999
4000 static UINT msi_publish_install_properties(MSIPACKAGE *package, HKEY hkey)
4001 {
4002 LPWSTR prop, val, key;
4003 SYSTEMTIME systime;
4004 DWORD size, langid;
4005 WCHAR date[9];
4006 LPWSTR buffer;
4007
4008 static const WCHAR date_fmt[] = {'%','i','%','0','2','i','%','0','2','i',0};
4009 static const WCHAR szWindowsInstaller[] =
4010 {'W','i','n','d','o','w','s','I','n','s','t','a','l','l','e','r',0};
4011 static const WCHAR modpath_fmt[] =
4012 {'M','s','i','E','x','e','c','.','e','x','e',' ',
4013 '/','I','[','P','r','o','d','u','c','t','C','o','d','e',']',0};
4014 static const WCHAR szModifyPath[] =
4015 {'M','o','d','i','f','y','P','a','t','h',0};
4016 static const WCHAR szUninstallString[] =
4017 {'U','n','i','n','s','t','a','l','l','S','t','r','i','n','g',0};
4018 static const WCHAR szEstimatedSize[] =
4019 {'E','s','t','i','m','a','t','e','d','S','i','z','e',0};
4020 static const WCHAR szProductLanguage[] =
4021 {'P','r','o','d','u','c','t','L','a','n','g','u','a','g','e',0};
4022 static const WCHAR szProductVersion[] =
4023 {'P','r','o','d','u','c','t','V','e','r','s','i','o','n',0};
4024 static const WCHAR szProductName[] =
4025 {'P','r','o','d','u','c','t','N','a','m','e',0};
4026 static const WCHAR szDisplayName[] =
4027 {'D','i','s','p','l','a','y','N','a','m','e',0};
4028 static const WCHAR szDisplayVersion[] =
4029 {'D','i','s','p','l','a','y','V','e','r','s','i','o','n',0};
4030 static const WCHAR szManufacturer[] =
4031 {'M','a','n','u','f','a','c','t','u','r','e','r',0};
4032
4033 static const LPCSTR propval[] = {
4034 "ARPAUTHORIZEDCDFPREFIX", "AuthorizedCDFPrefix",
4035 "ARPCONTACT", "Contact",
4036 "ARPCOMMENTS", "Comments",
4037 "ProductName", "DisplayName",
4038 "ProductVersion", "DisplayVersion",
4039 "ARPHELPLINK", "HelpLink",
4040 "ARPHELPTELEPHONE", "HelpTelephone",
4041 "ARPINSTALLLOCATION", "InstallLocation",
4042 "SourceDir", "InstallSource",
4043 "Manufacturer", "Publisher",
4044 "ARPREADME", "Readme",
4045 "ARPSIZE", "Size",
4046 "ARPURLINFOABOUT", "URLInfoAbout",
4047 "ARPURLUPDATEINFO", "URLUpdateInfo",
4048 NULL,
4049 };
4050 const LPCSTR *p = propval;
4051
4052 while (*p)
4053 {
4054 prop = strdupAtoW(*p++);
4055 key = strdupAtoW(*p++);
4056 val = msi_dup_property(package, prop);
4057 msi_reg_set_val_str(hkey, key, val);
4058 msi_free(val);
4059 msi_free(key);
4060 msi_free(prop);
4061 }
4062
4063 msi_reg_set_val_dword(hkey, szWindowsInstaller, 1);
4064
4065 size = deformat_string(package, modpath_fmt, &buffer);
4066 RegSetValueExW(hkey, szModifyPath, 0, REG_EXPAND_SZ, (LPBYTE)buffer, size);
4067 RegSetValueExW(hkey, szUninstallString, 0, REG_EXPAND_SZ, (LPBYTE)buffer, size);
4068 msi_free(buffer);
4069
4070 /* FIXME: Write real Estimated Size when we have it */
4071 msi_reg_set_val_dword(hkey, szEstimatedSize, 0);
4072
4073 buffer = msi_dup_property(package, szProductName);
4074 msi_reg_set_val_str(hkey, szDisplayName, buffer);
4075 msi_free(buffer);
4076
4077 buffer = msi_dup_property(package, cszSourceDir);
4078 msi_reg_set_val_str(hkey, INSTALLPROPERTY_INSTALLSOURCEW, buffer);
4079 msi_free(buffer);
4080
4081 buffer = msi_dup_property(package, szManufacturer);
4082 msi_reg_set_val_str(hkey, INSTALLPROPERTY_PUBLISHERW, buffer);
4083 msi_free(buffer);
4084
4085 GetLocalTime(&systime);
4086 sprintfW(date, date_fmt, systime.wYear, systime.wMonth, systime.wDay);
4087 msi_reg_set_val_str(hkey, INSTALLPROPERTY_INSTALLDATEW, date);
4088
4089 langid = msi_get_property_int(package, szProductLanguage, 0);
4090 msi_reg_set_val_dword(hkey, INSTALLPROPERTY_LANGUAGEW, langid);
4091
4092 buffer = msi_dup_property(package, szProductVersion);
4093 msi_reg_set_val_str(hkey, szDisplayVersion, buffer);
4094 if (buffer)
4095 {
4096 DWORD verdword = msi_version_str_to_dword(buffer);
4097
4098 msi_reg_set_val_dword(hkey, INSTALLPROPERTY_VERSIONW, verdword);
4099 msi_reg_set_val_dword(hkey, INSTALLPROPERTY_VERSIONMAJORW, verdword >> 24);
4100 msi_reg_set_val_dword(hkey, INSTALLPROPERTY_VERSIONMINORW, (verdword >> 16) & 0xFF);
4101 msi_free(buffer);
4102 }
4103
4104 return ERROR_SUCCESS;
4105 }
4106
4107 static UINT ACTION_RegisterProduct(MSIPACKAGE *package)
4108 {
4109 WCHAR squashed_pc[SQUISH_GUID_SIZE];
4110 LPWSTR upgrade_code;
4111 HKEY hkey, props;
4112 HKEY upgrade;
4113 UINT rc;
4114
4115 static const WCHAR szUpgradeCode[] = {
4116 'U','p','g','r','a','d','e','C','o','d','e',0};
4117
4118 /* FIXME: also need to publish if the product is in advertise mode */
4119 if (!msi_check_publish(package))
4120 return ERROR_SUCCESS;
4121
4122 rc = MSIREG_OpenUninstallKey(package->ProductCode, &hkey, TRUE);
4123 if (rc != ERROR_SUCCESS)
4124 return rc;
4125
4126 rc = MSIREG_OpenInstallProps(package->ProductCode, package->Context,
4127 NULL, &props, TRUE);
4128 if (rc != ERROR_SUCCESS)
4129 goto done;
4130
4131 msi_reg_set_val_str( props, INSTALLPROPERTY_LOCALPACKAGEW, package->db->localfile );
4132 msi_free( package->db->localfile );
4133 package->db->localfile = NULL;
4134
4135 rc = msi_publish_install_properties(package, hkey);
4136 if (rc != ERROR_SUCCESS)
4137 goto done;
4138
4139 rc = msi_publish_install_properties(package, props);
4140 if (rc != ERROR_SUCCESS)
4141 goto done;
4142
4143 upgrade_code = msi_dup_property(package, szUpgradeCode);
4144 if (upgrade_code)
4145 {
4146 MSIREG_OpenUpgradeCodesKey(upgrade_code, &upgrade, TRUE);
4147 squash_guid(package->ProductCode, squashed_pc);
4148 msi_reg_set_val_str(upgrade, squashed_pc, NULL);
4149 RegCloseKey(upgrade);
4150 msi_free(upgrade_code);
4151 }
4152
4153 done:
4154 RegCloseKey(hkey);
4155
4156 return ERROR_SUCCESS;
4157 }
4158
4159 static UINT ACTION_InstallExecute(MSIPACKAGE *package)
4160 {
4161 return execute_script(package,INSTALL_SCRIPT);
4162 }
4163
4164 static UINT msi_unpublish_product(MSIPACKAGE *package)
4165 {
4166 LPWSTR upgrade;
4167 LPWSTR remove = NULL;
4168 LPWSTR *features = NULL;
4169 BOOL full_uninstall = TRUE;
4170 MSIFEATURE *feature;
4171
4172 static const WCHAR szUpgradeCode[] =
4173 {'U','p','g','r','a','d','e','C','o','d','e',0};
4174
4175 remove = msi_dup_property(package, szRemove);
4176 if (!remove)
4177 return ERROR_SUCCESS;
4178
4179 features = msi_split_string(remove, ',');
4180 if (!features)
4181 {
4182 msi_free(remove);
4183 ERR("REMOVE feature list is empty!\n");
4184 return ERROR_FUNCTION_FAILED;
4185 }
4186
4187 if (!lstrcmpW(features[0], szAll))
4188 full_uninstall = TRUE;
4189 else
4190 {
4191 LIST_FOR_EACH_ENTRY(feature, &package->features, MSIFEATURE, entry)
4192 {
4193 if (feature->Action != INSTALLSTATE_ABSENT)
4194 full_uninstall = FALSE;
4195 }
4196 }
4197
4198 if (!full_uninstall)
4199 goto done;
4200
4201 MSIREG_DeleteProductKey(package->ProductCode);
4202 MSIREG_DeleteUserDataProductKey(package->ProductCode);
4203 MSIREG_DeleteUninstallKey(package->ProductCode);
4204
4205 if (package->Context == MSIINSTALLCONTEXT_MACHINE)
4206 {
4207 MSIREG_DeleteLocalClassesProductKey(package->ProductCode);
4208 MSIREG_DeleteLocalClassesFeaturesKey(package->ProductCode);
4209 }
4210 else
4211 {
4212 MSIREG_DeleteUserProductKey(package->ProductCode);
4213 MSIREG_DeleteUserFeaturesKey(package->ProductCode);
4214 }
4215
4216 upgrade = msi_dup_property(package, szUpgradeCode);
4217 if (upgrade)
4218 {
4219 MSIREG_DeleteUserUpgradeCodesKey(upgrade);
4220 msi_free(upgrade);
4221 }
4222
4223 done:
4224 msi_free(remove);
4225 msi_free(features);
4226 return ERROR_SUCCESS;
4227 }
4228
4229 static UINT ACTION_InstallFinalize(MSIPACKAGE *package)
4230 {
4231 UINT rc;
4232
4233 rc = msi_unpublish_product(package);
4234 if (rc != ERROR_SUCCESS)
4235 return rc;
4236
4237 /* turn off scheduling */
4238 package->script->CurrentlyScripting= FALSE;
4239
4240 /* first do the same as an InstallExecute */
4241 rc = ACTION_InstallExecute(package);
4242 if (rc != ERROR_SUCCESS)
4243 return rc;
4244
4245 /* then handle Commit Actions */
4246 rc = execute_script(package,COMMIT_SCRIPT);
4247
4248 return rc;
4249 }
4250
4251 UINT ACTION_ForceReboot(MSIPACKAGE *package)
4252 {
4253 static const WCHAR RunOnce[] = {
4254 'S','o','f','t','w','a','r','e','\\',
4255 'M','i','c','r','o','s','o','f','t','\\',
4256 'W','i','n','d','o','w','s','\\',
4257 'C','u','r','r','e','n','t','V','e','r','s','i','o','n','\\',
4258 'R','u','n','O','n','c','e',0};
4259 static const WCHAR InstallRunOnce[] = {
4260 'S','o','f','t','w','a','r','e','\\',
4261 'M','i','c','r','o','s','o','f','t','\\',
4262 'W','i','n','d','o','w','s','\\',
4263 'C','u','r','r','e','n','t','V','e','r','s','i','o','n','\\',
4264 'I','n','s','t','a','l','l','e','r','\\',
4265 'R','u','n','O','n','c','e','E','n','t','r','i','e','s',0};
4266
4267 static const WCHAR msiexec_fmt[] = {
4268 '%','s',
4269 '\\','M','s','i','E','x','e','c','.','e','x','e',' ','/','@',' ',
4270 '\"','%','s','\"',0};
4271 static const WCHAR install_fmt[] = {
4272 '/','I',' ','\"','%','s','\"',' ',
4273 'A','F','T','E','R','R','E','B','O','O','T','=','1',' ',
4274 'R','U','N','O','N','C','E','E','N','T','R','Y','=','\"','%','s','\"',0};
4275 WCHAR buffer[256], sysdir[MAX_PATH];
4276 HKEY hkey;
4277 WCHAR squished_pc[100];
4278
4279 squash_guid(package->ProductCode,squished_pc);
4280
4281 GetSystemDirectoryW(sysdir, sizeof(sysdir)/sizeof(sysdir[0]));
4282 RegCreateKeyW(HKEY_LOCAL_MACHINE,RunOnce,&hkey);
4283 snprintfW(buffer,sizeof(buffer)/sizeof(buffer[0]),msiexec_fmt,sysdir,
4284 squished_pc);
4285
4286 msi_reg_set_val_str( hkey, squished_pc, buffer );
4287 RegCloseKey(hkey);
4288
4289 TRACE("Reboot command %s\n",debugstr_w(buffer));
4290
4291 RegCreateKeyW(HKEY_LOCAL_MACHINE,InstallRunOnce,&hkey);
4292 sprintfW(buffer,install_fmt,package->ProductCode,squished_pc);
4293
4294 msi_reg_set_val_str( hkey, squished_pc, buffer );
4295 RegCloseKey(hkey);
4296
4297 return ERROR_INSTALL_SUSPEND;
4298 }
4299
4300 static UINT ACTION_ResolveSource(MSIPACKAGE* package)
4301 {
4302 DWORD attrib;
4303 UINT rc;
4304
4305 /*
4306 * We are currently doing what should be done here in the top level Install
4307 * however for Administrative and uninstalls this step will be needed
4308 */
4309 if (!package->PackagePath)
4310 return ERROR_SUCCESS;
4311
4312 msi_set_sourcedir_props(package, TRUE);
4313
4314 attrib = GetFileAttributesW(package->db->path);
4315 if (attrib == INVALID_FILE_ATTRIBUTES)
4316 {
4317 LPWSTR prompt;
4318 LPWSTR msg;
4319 DWORD size = 0;
4320
4321 rc = MsiSourceListGetInfoW(package->ProductCode, NULL,
4322 package->Context, MSICODE_PRODUCT,
4323 INSTALLPROPERTY_DISKPROMPTW,NULL,&size);
4324 if (rc == ERROR_MORE_DATA)
4325 {
4326 prompt = msi_alloc(size * sizeof(WCHAR));
4327 MsiSourceListGetInfoW(package->ProductCode, NULL,
4328 package->Context, MSICODE_PRODUCT,
4329 INSTALLPROPERTY_DISKPROMPTW,prompt,&size);
4330 }
4331 else
4332 prompt = strdupW(package->db->path);
4333
4334 msg = generate_error_string(package,1302,1,prompt);
4335 while(attrib == INVALID_FILE_ATTRIBUTES)
4336 {
4337 rc = MessageBoxW(NULL,msg,NULL,MB_OKCANCEL);
4338 if (rc == IDCANCEL)
4339 {
4340 rc = ERROR_INSTALL_USEREXIT;
4341 break;
4342 }
4343 attrib = GetFileAttributesW(package->db->path);
4344 }
4345 msi_free(prompt);
4346 rc = ERROR_SUCCESS;
4347 }
4348 else
4349 return ERROR_SUCCESS;
4350
4351 return rc;
4352 }
4353
4354 static UINT ACTION_RegisterUser(MSIPACKAGE *package)
4355 {
4356 HKEY hkey=0;
4357 LPWSTR buffer;
4358 LPWSTR productid;
4359 UINT rc,i;
4360
4361 static const WCHAR szPropKeys[][80] =
4362 {
4363 {'P','r','o','d','u','c','t','I','D',0},
4364 {'U','S','E','R','N','A','M','E',0},
4365 {'C','O','M','P','A','N','Y','N','A','M','E',0},
4366 {0},
4367 };
4368
4369 static const WCHAR szRegKeys[][80] =
4370 {
4371 {'P','r','o','d','u','c','t','I','D',0},
4372 {'R','e','g','O','w','n','e','r',0},
4373 {'R','e','g','C','o','m','p','a','n','y',0},
4374 {0},
4375 };
4376
4377 if (msi_check_unpublish(package))
4378 {
4379 MSIREG_DeleteUserDataProductKey(package->ProductCode);
4380 return ERROR_SUCCESS;
4381 }
4382
4383 productid = msi_dup_property( package, INSTALLPROPERTY_PRODUCTIDW );
4384 if (!productid)
4385 return ERROR_SUCCESS;
4386
4387 rc = MSIREG_OpenInstallProps(package->ProductCode, package->Context,
4388 NULL, &hkey, TRUE);
4389 if (rc != ERROR_SUCCESS)
4390 goto end;
4391
4392 for( i = 0; szPropKeys[i][0]; i++ )
4393 {
4394 buffer = msi_dup_property( package, szPropKeys[i] );
4395 msi_reg_set_val_str( hkey, szRegKeys[i], buffer );
4396 msi_free( buffer );
4397 }
4398
4399 end:
4400 msi_free(productid);
4401 RegCloseKey(hkey);
4402
4403 /* FIXME: call ui_actiondata */
4404
4405 return rc;
4406 }
4407
4408
4409 static UINT ACTION_ExecuteAction(MSIPACKAGE *package)
4410 {
4411 UINT rc;
4412
4413 package->script->InWhatSequence |= SEQUENCE_EXEC;
4414 rc = ACTION_ProcessExecSequence(package,FALSE);
4415 return rc;
4416 }
4417
4418
4419 static UINT ITERATE_PublishComponent(MSIRECORD *rec, LPVOID param)
4420 {
4421 MSIPACKAGE *package = param;
4422 LPCWSTR compgroupid=NULL;
4423 LPCWSTR feature=NULL;
4424 LPCWSTR text = NULL;
4425 LPCWSTR qualifier = NULL;
4426 LPCWSTR component = NULL;
4427 LPWSTR advertise = NULL;
4428 LPWSTR output = NULL;
4429 HKEY hkey;
4430 UINT rc = ERROR_SUCCESS;
4431 MSICOMPONENT *comp;
4432 DWORD sz = 0;
4433 MSIRECORD *uirow;
4434
4435 component = MSI_RecordGetString(rec,3);
4436 comp = get_loaded_component(package,component);
4437
4438 if (!ACTION_VerifyComponentForAction( comp, INSTALLSTATE_LOCAL ) &&
4439 !ACTION_VerifyComponentForAction( comp, INSTALLSTATE_SOURCE ) &&
4440 !ACTION_VerifyComponentForAction( comp, INSTALLSTATE_ADVERTISED ))
4441 {
4442 TRACE("Skipping: Component %s not scheduled for install\n",
4443 debugstr_w(component));
4444
4445 return ERROR_SUCCESS;
4446 }
4447
4448 compgroupid = MSI_RecordGetString(rec,1);
4449 qualifier = MSI_RecordGetString(rec,2);
4450
4451 rc = MSIREG_OpenUserComponentsKey(compgroupid, &hkey, TRUE);
4452 if (rc != ERROR_SUCCESS)
4453 goto end;
4454
4455 text = MSI_RecordGetString(rec,4);
4456 feature = MSI_RecordGetString(rec,5);
4457
4458 advertise = create_component_advertise_string(package, comp, feature);
4459
4460 sz = strlenW(advertise);
4461
4462 if (text)
4463 sz += lstrlenW(text);
4464
4465 sz+=3;
4466 sz *= sizeof(WCHAR);
4467
4468 output = msi_alloc_zero(sz);
4469 strcpyW(output,advertise);
4470 msi_free(advertise);
4471
4472 if (text)
4473 strcatW(output,text);
4474
4475 msi_reg_set_val_multi_str( hkey, qualifier, output );
4476
4477 end:
4478 RegCloseKey(hkey);
4479 msi_free(output);
4480
4481 /* the UI chunk */
4482 uirow = MSI_CreateRecord( 2 );
4483 MSI_RecordSetStringW( uirow, 1, compgroupid );
4484 MSI_RecordSetStringW( uirow, 2, qualifier);
4485 ui_actiondata( package, szPublishComponents, uirow);
4486 msiobj_release( &uirow->hdr );
4487 /* FIXME: call ui_progress? */
4488
4489 return rc;
4490 }
4491
4492 /*
4493 * At present I am ignorning the advertised components part of this and only
4494 * focusing on the qualified component sets
4495 */
4496 static UINT ACTION_PublishComponents(MSIPACKAGE *package)
4497 {
4498 UINT rc;
4499 MSIQUERY * view;
4500 static const WCHAR ExecSeqQuery[] =
4501 {'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
4502 '`','P','u','b','l','i','s','h',
4503 'C','o','m','p','o','n','e','n','t','`',0};
4504
4505 rc = MSI_DatabaseOpenViewW(package->db, ExecSeqQuery, &view);
4506 if (rc != ERROR_SUCCESS)
4507 return ERROR_SUCCESS;
4508
4509 rc = MSI_IterateRecords(view, NULL, ITERATE_PublishComponent, package);
4510 msiobj_release(&view->hdr);
4511
4512 return rc;
4513 }
4514
4515 static UINT ITERATE_InstallService(MSIRECORD *rec, LPVOID param)
4516 {
4517 MSIPACKAGE *package = param;
4518 MSIRECORD *row;
4519 MSIFILE *file;
4520 SC_HANDLE hscm, service = NULL;
4521 LPCWSTR comp, depends, pass;
4522 LPWSTR name = NULL, disp = NULL;
4523 LPCWSTR load_order, serv_name, key;
4524 DWORD serv_type, start_type;
4525 DWORD err_control;
4526
4527 static const WCHAR query[] =
4528 {'S','E','L','E','C','T',' ','*',' ','F','R', 'O','M',' ',
4529 '`','C','o','m','p','o','n','e','n','t','`',' ',
4530 'W','H','E','R','E',' ',
4531 '`','C','o','m','p','o','n','e','n','t','`',' ',
4532 '=','\'','%','s','\'',0};
4533
4534 hscm = OpenSCManagerW(NULL, SERVICES_ACTIVE_DATABASEW, GENERIC_WRITE);
4535 if (!hscm)
4536 {
4537 ERR("Failed to open the SC Manager!\n");
4538 goto done;
4539 }
4540
4541 start_type = MSI_RecordGetInteger(rec, 5);
4542 if (start_type == SERVICE_BOOT_START || start_type == SERVICE_SYSTEM_START)
4543 goto done;
4544
4545 depends = MSI_RecordGetString(rec, 8);
4546 if (depends && *depends)
4547 FIXME("Dependency list unhandled!\n");
4548
4549 deformat_string(package, MSI_RecordGetString(rec, 2), &name);
4550 deformat_string(package, MSI_RecordGetString(rec, 3), &disp);
4551 serv_type = MSI_RecordGetInteger(rec, 4);
4552 err_control = MSI_RecordGetInteger(rec, 6);
4553 load_order = MSI_RecordGetString(rec, 7);
4554 serv_name = MSI_RecordGetString(rec, 9);
4555 pass = MSI_RecordGetString(rec, 10);
4556 comp = MSI_RecordGetString(rec, 12);
4557
4558 /* fetch the service path */
4559 row = MSI_QueryGetRecord(package->db, query, comp);
4560 if (!row)
4561 {
4562 ERR("Control query failed!\n");
4563 goto done;
4564 }
4565
4566 key = MSI_RecordGetString(row, 6);
4567
4568 file = get_loaded_file(package, key);
4569 msiobj_release(&row->hdr);
4570 if (!file)
4571 {
4572 ERR("Failed to load the service file\n");
4573 goto done;
4574 }
4575
4576 service = CreateServiceW(hscm, name, disp, GENERIC_ALL, serv_type,
4577 start_type, err_control, file->TargetPath,
4578 load_order, NULL, NULL, serv_name, pass);
4579 if (!service)
4580 {
4581 if (GetLastError() != ERROR_SERVICE_EXISTS)
4582 ERR("Failed to create service %s: %d\n", debugstr_w(name), GetLastError());
4583 }
4584
4585 done:
4586 CloseServiceHandle(service);
4587 CloseServiceHandle(hscm);
4588 msi_free(name);
4589 msi_free(disp);
4590
4591 return ERROR_SUCCESS;
4592 }
4593
4594 static UINT ACTION_InstallServices( MSIPACKAGE *package )
4595 {
4596 UINT rc;
4597 MSIQUERY * view;
4598 static const WCHAR ExecSeqQuery[] =
4599 {'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
4600 'S','e','r','v','i','c','e','I','n','s','t','a','l','l',0};
4601
4602 rc = MSI_DatabaseOpenViewW(package->db, ExecSeqQuery, &view);
4603 if (rc != ERROR_SUCCESS)
4604 return ERROR_SUCCESS;
4605
4606 rc = MSI_IterateRecords(view, NULL, ITERATE_InstallService, package);
4607 msiobj_release(&view->hdr);
4608
4609 return rc;
4610 }
4611
4612 /* converts arg1[~]arg2[~]arg3 to a list of ptrs to the strings */
4613 static LPCWSTR *msi_service_args_to_vector(LPWSTR args, DWORD *numargs)
4614 {
4615 LPCWSTR *vector, *temp_vector;
4616 LPWSTR p, q;
4617 DWORD sep_len;
4618
4619 static const WCHAR separator[] = {'[','~',']',0};
4620
4621 *numargs = 0;
4622 sep_len = sizeof(separator) / sizeof(WCHAR) - 1;
4623
4624 if (!args)
4625 return NULL;
4626
4627 vector = msi_alloc(sizeof(LPWSTR));
4628 if (!vector)
4629 return NULL;
4630
4631 p = args;
4632 do
4633 {
4634 (*numargs)++;
4635 vector[*numargs - 1] = p;
4636
4637 if ((q = strstrW(p, separator)))
4638 {
4639 *q = '\0';
4640
4641 temp_vector = msi_realloc(vector, (*numargs + 1) * sizeof(LPWSTR));
4642 if (!temp_vector)
4643 {
4644 msi_free(vector);
4645 return NULL;
4646 }
4647 vector = temp_vector;
4648
4649 p = q + sep_len;
4650 }
4651 } while (q);
4652
4653 return vector;
4654 }
4655
4656 static UINT ITERATE_StartService(MSIRECORD *rec, LPVOID param)
4657 {
4658 MSIPACKAGE *package = param;
4659 MSICOMPONENT *comp;
4660 SC_HANDLE scm, service = NULL;
4661 LPCWSTR *vector = NULL;
4662 LPWSTR name, args;
4663 DWORD event, numargs;
4664 UINT r = ERROR_FUNCTION_FAILED;
4665
4666 comp = get_loaded_component(package, MSI_RecordGetString(rec, 6));
4667 if (!comp || comp->Action == INSTALLSTATE_UNKNOWN || comp->Action == INSTALLSTATE_ABSENT)
4668 return ERROR_SUCCESS;
4669
4670 deformat_string(package, MSI_RecordGetString(rec, 2), &name);
4671 deformat_string(package, MSI_RecordGetString(rec, 4), &args);
4672 event = MSI_RecordGetInteger(rec, 3);
4673
4674 if (!(event & msidbServiceControlEventStart))
4675 return ERROR_SUCCESS;
4676
4677 scm = OpenSCManagerW(NULL, NULL, SC_MANAGER_CONNECT);
4678 if (!scm)
4679 {
4680 ERR("Failed to open the service control manager\n");
4681 goto done;
4682 }
4683
4684 service = OpenServiceW(scm, name, SERVICE_START);
4685 if (!service)
4686 {
4687 ERR("Failed to open service %s (%u)\n", debugstr_w(name), GetLastError());
4688 goto done;
4689 }
4690
4691 vector = msi_service_args_to_vector(args, &numargs);
4692
4693 if (!StartServiceW(service, numargs, vector) &&
4694 GetLastError() != ERROR_SERVICE_ALREADY_RUNNING)
4695 {
4696 ERR("Failed to start service %s (%u)\n", debugstr_w(name), GetLastError());
4697 goto done;
4698 }
4699
4700 r = ERROR_SUCCESS;
4701
4702 done:
4703 CloseServiceHandle(service);
4704 CloseServiceHandle(scm);
4705
4706 msi_free(name);
4707 msi_free(args);
4708 msi_free(vector);
4709 return r;
4710 }
4711
4712 static UINT ACTION_StartServices( MSIPACKAGE *package )
4713 {
4714 UINT rc;
4715 MSIQUERY *view;
4716
4717 static const WCHAR query[] = {
4718 'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
4719 'S','e','r','v','i','c','e','C','o','n','t','r','o','l',0 };
4720
4721 rc = MSI_DatabaseOpenViewW(package->db, query, &view);
4722 if (rc != ERROR_SUCCESS)
4723 return ERROR_SUCCESS;
4724
4725 rc = MSI_IterateRecords(view, NULL, ITERATE_StartService, package);
4726 msiobj_release(&view->hdr);
4727
4728 return rc;
4729 }
4730
4731 static BOOL stop_service_dependents(SC_HANDLE scm, SC_HANDLE service)
4732 {
4733 DWORD i, needed, count;
4734 ENUM_SERVICE_STATUSW *dependencies;
4735 SERVICE_STATUS ss;
4736 SC_HANDLE depserv;
4737
4738 if (EnumDependentServicesW(service, SERVICE_ACTIVE, NULL,
4739 0, &needed, &count))
4740 return TRUE;
4741
4742 if (GetLastError() != ERROR_MORE_DATA)
4743 return FALSE;
4744
4745 dependencies = msi_alloc(needed);
4746 if (!dependencies)
4747 return FALSE;
4748
4749 if (!EnumDependentServicesW(service, SERVICE_ACTIVE, dependencies,
4750 needed, &needed, &count))
4751 goto error;
4752
4753 for (i = 0; i < count; i++)
4754 {
4755 depserv = OpenServiceW(scm, dependencies[i].lpServiceName,
4756 SERVICE_STOP | SERVICE_QUERY_STATUS);
4757 if (!depserv)
4758 goto error;
4759
4760 if (!ControlService(depserv, SERVICE_CONTROL_STOP, &ss))
4761 goto error;
4762 }
4763
4764 return TRUE;
4765
4766 error:
4767 msi_free(dependencies);
4768 return FALSE;
4769 }
4770
4771 static UINT stop_service( LPCWSTR name )
4772 {
4773 SC_HANDLE scm = NULL, service = NULL;
4774 SERVICE_STATUS status;
4775 SERVICE_STATUS_PROCESS ssp;
4776 DWORD needed;
4777
4778 scm = OpenSCManagerW(NULL, NULL, SC_MANAGER_ALL_ACCESS);
4779 if (!scm)
4780 {
4781 WARN("Failed to open the SCM: %d\n", GetLastError());
4782 goto done;
4783 }
4784
4785 service = OpenServiceW(scm, name,
4786 SERVICE_STOP |
4787 SERVICE_QUERY_STATUS |
4788 SERVICE_ENUMERATE_DEPENDENTS);
4789 if (!service)
4790 {
4791 WARN("Failed to open service (%s): %d\n", debugstr_w(name), GetLastError());
4792 goto done;
4793 }
4794
4795 if (!QueryServiceStatusEx(service, SC_STATUS_PROCESS_INFO, (LPBYTE)&ssp,
4796 sizeof(SERVICE_STATUS_PROCESS), &needed))
4797 {
4798 WARN("Failed to query service status (%s): %d\n", debugstr_w(name), GetLastError());
4799 goto done;
4800 }
4801
4802 if (ssp.dwCurrentState == SERVICE_STOPPED)
4803 goto done;
4804
4805 stop_service_dependents(scm, service);
4806
4807 if (!ControlService(service, SERVICE_CONTROL_STOP, &status))
4808 WARN("Failed to stop service (%s): %d\n", debugstr_w(name), GetLastError());
4809
4810 done:
4811 CloseServiceHandle(service);
4812 CloseServiceHandle(scm);
4813
4814 return ERROR_SUCCESS;
4815 }
4816
4817 static UINT ITERATE_StopService( MSIRECORD *rec, LPVOID param )
4818 {
4819 MSIPACKAGE *package = param;
4820 MSICOMPONENT *comp;
4821 LPWSTR name;
4822 DWORD event;
4823
4824 event = MSI_RecordGetInteger( rec, 3 );
4825 if (!(event & msidbServiceControlEventStop))
4826 return ERROR_SUCCESS;
4827
4828 comp = get_loaded_component( package, MSI_RecordGetString( rec, 6 ) );
4829 if (!comp || comp->Action == INSTALLSTATE_UNKNOWN || comp->Action == INSTALLSTATE_ABSENT)
4830 return ERROR_SUCCESS;
4831
4832 deformat_string( package, MSI_RecordGetString( rec, 2 ), &name );
4833 stop_service( name );
4834 msi_free( name );
4835
4836 return ERROR_SUCCESS;
4837 }
4838
4839 static UINT ACTION_StopServices( MSIPACKAGE *package )
4840 {
4841 UINT rc;
4842 MSIQUERY *view;
4843
4844 static const WCHAR query[] = {
4845 'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
4846 'S','e','r','v','i','c','e','C','o','n','t','r','o','l',0 };
4847
4848 rc = MSI_DatabaseOpenViewW(package->db, query, &view);
4849 if (rc != ERROR_SUCCESS)
4850 return ERROR_SUCCESS;
4851
4852 rc = MSI_IterateRecords(view, NULL, ITERATE_StopService, package);
4853 msiobj_release(&view->hdr);
4854
4855 return rc;
4856 }
4857
4858 static UINT ITERATE_DeleteService( MSIRECORD *rec, LPVOID param )
4859 {
4860 MSIPACKAGE *package = param;
4861 MSICOMPONENT *comp;
4862 LPWSTR name = NULL;
4863 DWORD event;
4864 SC_HANDLE scm = NULL, service = NULL;
4865
4866 event = MSI_RecordGetInteger( rec, 3 );
4867 if (!(event & msidbServiceControlEventDelete))
4868 return ERROR_SUCCESS;
4869
4870 comp = get_loaded_component( package, MSI_RecordGetString(rec, 6) );
4871 if (!comp || comp->Action == INSTALLSTATE_UNKNOWN || comp->Action == INSTALLSTATE_ABSENT)
4872 return ERROR_SUCCESS;
4873
4874 deformat_string( package, MSI_RecordGetString(rec, 2), &name );
4875 stop_service( name );
4876
4877 scm = OpenSCManagerW( NULL, NULL, SC_MANAGER_ALL_ACCESS );
4878 if (!scm)
4879 {
4880 WARN("Failed to open the SCM: %d\n", GetLastError());
4881 goto done;
4882 }
4883
4884 service = OpenServiceW( scm, name, DELETE );
4885 if (!service)
4886 {
4887 WARN("Failed to open service (%s): %u\n", debugstr_w(name), GetLastError());
4888 goto done;
4889 }
4890
4891 if (!DeleteService( service ))
4892 WARN("Failed to delete service (%s): %u\n", debugstr_w(name), GetLastError());
4893
4894 done:
4895 CloseServiceHandle( service );
4896 CloseServiceHandle( scm );
4897 msi_free( name );
4898
4899 return ERROR_SUCCESS;
4900 }
4901
4902 static UINT ACTION_DeleteServices( MSIPACKAGE *package )
4903 {
4904 UINT rc;
4905 MSIQUERY *view;
4906
4907 static const WCHAR query[] = {
4908 'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
4909 'S','e','r','v','i','c','e','C','o','n','t','r','o','l',0 };
4910
4911 rc = MSI_DatabaseOpenViewW( package->db, query, &view );
4912 if (rc != ERROR_SUCCESS)
4913 return ERROR_SUCCESS;
4914
4915 rc = MSI_IterateRecords( view, NULL, ITERATE_DeleteService, package );
4916 msiobj_release( &view->hdr );
4917
4918 return rc;
4919 }
4920
4921 static MSIFILE *msi_find_file( MSIPACKAGE *package, LPCWSTR filename )
4922 {
4923 MSIFILE *file;
4924
4925 LIST_FOR_EACH_ENTRY(file, &package->files, MSIFILE, entry)
4926 {
4927 if (!lstrcmpW(file->File, filename))
4928 return file;
4929 }
4930
4931 return NULL;
4932 }
4933
4934 static UINT ITERATE_InstallODBCDriver( MSIRECORD *rec, LPVOID param )
4935 {
4936 MSIPACKAGE *package = param;
4937 LPWSTR driver, driver_path, ptr;
4938 WCHAR outpath[MAX_PATH];
4939 MSIFILE *driver_file, *setup_file;
4940 LPCWSTR desc;
4941 DWORD len, usage;
4942 UINT r = ERROR_SUCCESS;
4943
4944 static const WCHAR driver_fmt[] = {
4945 'D','r','i','v','e','r','=','%','s',0};
4946 static const WCHAR setup_fmt[] = {
4947 'S','e','t','u','p','=','%','s',0};
4948 static const WCHAR usage_fmt[] = {
4949 'F','i','l','e','U','s','a','g','e','=','1',0};
4950
4951 desc = MSI_RecordGetString(rec, 3);
4952
4953 driver_file = msi_find_file(package, MSI_RecordGetString(rec, 4));
4954 setup_file = msi_find_file(package, MSI_RecordGetString(rec, 5));
4955
4956 if (!driver_file)
4957 {
4958 ERR("ODBC Driver entry not found!\n");
4959 return ERROR_FUNCTION_FAILED;
4960 }
4961
4962 len = lstrlenW(desc) + lstrlenW(driver_fmt) + lstrlenW(driver_file->FileName);
4963 if (setup_file)
4964 len += lstrlenW(setup_fmt) + lstrlenW(setup_file->FileName);
4965 len += lstrlenW(usage_fmt) + 1;
4966
4967 driver = msi_alloc(len * sizeof(WCHAR));
4968 if (!driver)
4969 return ERROR_OUTOFMEMORY;
4970
4971 ptr = driver;
4972 lstrcpyW(ptr, desc);
4973 ptr += lstrlenW(ptr) + 1;
4974
4975 sprintfW(ptr, driver_fmt, driver_file->FileName);
4976 ptr += lstrlenW(ptr) + 1;
4977
4978 if (setup_file)
4979 {
4980 sprintfW(ptr, setup_fmt, setup_file->FileName);
4981 ptr += lstrlenW(ptr) + 1;
4982 }
4983
4984 lstrcpyW(ptr, usage_fmt);
4985 ptr += lstrlenW(ptr) + 1;
4986 *ptr = '\0';
4987
4988 driver_path = strdupW(driver_file->TargetPath);
4989 ptr = strrchrW(driver_path, '\\');
4990 if (ptr) *ptr = '\0';
4991
4992 if (!SQLInstallDriverExW(driver, driver_path, outpath, MAX_PATH,
4993 NULL, ODBC_INSTALL_COMPLETE, &usage))
4994 {
4995 ERR("Failed to install SQL driver!\n");
4996 r = ERROR_FUNCTION_FAILED;
4997 }
4998
4999 msi_free(driver);
5000 msi_free(driver_path);
5001
5002 return r;
5003 }
5004
5005 static UINT ITERATE_InstallODBCTranslator( MSIRECORD *rec, LPVOID param )
5006 {
5007 MSIPACKAGE *package = param;
5008 LPWSTR translator, translator_path, ptr;
5009 WCHAR outpath[MAX_PATH];
5010 MSIFILE *translator_file, *setup_file;
5011 LPCWSTR desc;
5012 DWORD len, usage;
5013 UINT r = ERROR_SUCCESS;
5014
5015 static const WCHAR translator_fmt[] = {
5016 'T','r','a','n','s','l','a','t','o','r','=','%','s',0};
5017 static const WCHAR setup_fmt[] = {
5018 'S','e','t','u','p','=','%','s',0};
5019
5020 desc = MSI_RecordGetString(rec, 3);
5021
5022 translator_file = msi_find_file(package, MSI_RecordGetString(rec, 4));
5023 setup_file = msi_find_file(package, MSI_RecordGetString(rec, 5));
5024
5025 if (!translator_file)
5026 {
5027 ERR("ODBC Translator entry not found!\n");
5028 return ERROR_FUNCTION_FAILED;
5029 }
5030
5031 len = lstrlenW(desc) + lstrlenW(translator_fmt) + lstrlenW(translator_file->FileName) + 1;
5032 if (setup_file)
5033 len += lstrlenW(setup_fmt) + lstrlenW(setup_file->FileName);
5034
5035 translator = msi_alloc(len * sizeof(WCHAR));
5036 if (!translator)
5037 return ERROR_OUTOFMEMORY;
5038
5039 ptr = translator;
5040 lstrcpyW(ptr, desc);
5041 ptr += lstrlenW(ptr) + 1;
5042
5043 sprintfW(ptr, translator_fmt, translator_file->FileName);
5044 ptr += lstrlenW(ptr) + 1;
5045
5046 if (setup_file)
5047 {
5048 sprintfW(ptr, setup_fmt, setup_file->FileName);
5049 ptr += lstrlenW(ptr) + 1;
5050 }
5051 *ptr = '\0';
5052
5053 translator_path = strdupW(translator_file->TargetPath);
5054 ptr = strrchrW(translator_path, '\\');
5055 if (ptr) *ptr = '\0';
5056
5057 if (!SQLInstallTranslatorExW(translator, translator_path, outpath, MAX_PATH,
5058 NULL, ODBC_INSTALL_COMPLETE, &usage))
5059 {
5060 ERR("Failed to install SQL translator!\n");
5061 r = ERROR_FUNCTION_FAILED;
5062 }
5063
5064 msi_free(translator);
5065 msi_free(translator_path);
5066
5067 return r;
5068 }
5069
5070 static UINT ITERATE_InstallODBCDataSource( MSIRECORD *rec, LPVOID param )
5071 {
5072 LPWSTR attrs;
5073 LPCWSTR desc, driver;
5074 WORD request = ODBC_ADD_SYS_DSN;
5075 INT registration;
5076 DWORD len;
5077 UINT r = ERROR_SUCCESS;
5078
5079 static const WCHAR attrs_fmt[] = {
5080 'D','S','N','=','%','s',0 };
5081
5082 desc = MSI_RecordGetString(rec, 3);
5083 driver = MSI_RecordGetString(rec, 4);
5084 registration = MSI_RecordGetInteger(rec, 5);
5085
5086 if (registration == msidbODBCDataSourceRegistrationPerMachine) request = ODBC_ADD_SYS_DSN;
5087 else if (registration == msidbODBCDataSourceRegistrationPerUser) request = ODBC_ADD_DSN;
5088
5089 len = lstrlenW(attrs_fmt) + lstrlenW(desc) + 1 + 1;
5090 attrs = msi_alloc(len * sizeof(WCHAR));
5091 if (!attrs)
5092 return ERROR_OUTOFMEMORY;
5093
5094 len = sprintfW(attrs, attrs_fmt, desc);
5095 attrs[len + 1] = 0;
5096
5097 if (!SQLConfigDataSourceW(NULL, request, driver, attrs))
5098 {
5099 ERR("Failed to install SQL data source!\n");
5100 r = ERROR_FUNCTION_FAILED;
5101 }
5102
5103 msi_free(attrs);
5104
5105 return r;
5106 }
5107
5108 static UINT ACTION_InstallODBC( MSIPACKAGE *package )
5109 {
5110 UINT rc;
5111 MSIQUERY *view;
5112
5113 static const WCHAR driver_query[] = {
5114 'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
5115 'O','D','B','C','D','r','i','v','e','r',0 };
5116
5117 static const WCHAR translator_query[] = {
5118 'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
5119 'O','D','B','C','T','r','a','n','s','l','a','t','o','r',0 };
5120
5121 static const WCHAR source_query[] = {
5122 'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
5123 'O','D','B','C','D','a','t','a','S','o','u','r','c','e',0 };
5124
5125 rc = MSI_DatabaseOpenViewW(package->db, driver_query, &view);
5126 if (rc != ERROR_SUCCESS)
5127 return ERROR_SUCCESS;
5128
5129 rc = MSI_IterateRecords(view, NULL, ITERATE_InstallODBCDriver, package);
5130 msiobj_release(&view->hdr);
5131
5132 rc = MSI_DatabaseOpenViewW(package->db, translator_query, &view);
5133 if (rc != ERROR_SUCCESS)
5134 return ERROR_SUCCESS;
5135
5136 rc = MSI_IterateRecords(view, NULL, ITERATE_InstallODBCTranslator, package);
5137 msiobj_release(&view->hdr);
5138
5139 rc = MSI_DatabaseOpenViewW(package->db, source_query, &view);
5140 if (rc != ERROR_SUCCESS)
5141 return ERROR_SUCCESS;
5142
5143 rc = MSI_IterateRecords(view, NULL, ITERATE_InstallODBCDataSource, package);
5144 msiobj_release(&view->hdr);
5145
5146 return rc;
5147 }
5148
5149 static UINT ITERATE_RemoveODBCDriver( MSIRECORD *rec, LPVOID param )
5150 {
5151 DWORD usage;
5152 LPCWSTR desc;
5153
5154 desc = MSI_RecordGetString( rec, 3 );
5155 if (!SQLRemoveDriverW( desc, FALSE, &usage ))
5156 {
5157 WARN("Failed to remove ODBC driver\n");
5158 }
5159 else if (!usage)
5160 {
5161 FIXME("Usage count reached 0\n");
5162 }
5163
5164 return ERROR_SUCCESS;
5165 }
5166
5167 static UINT ITERATE_RemoveODBCTranslator( MSIRECORD *rec, LPVOID param )
5168 {
5169 DWORD usage;
5170 LPCWSTR desc;
5171
5172 desc = MSI_RecordGetString( rec, 3 );
5173 if (!SQLRemoveTranslatorW( desc, &usage ))
5174 {
5175 WARN("Failed to remove ODBC translator\n");
5176 }
5177 else if (!usage)
5178 {
5179 FIXME("Usage count reached 0\n");
5180 }
5181
5182 return ERROR_SUCCESS;
5183 }
5184
5185 static UINT ITERATE_RemoveODBCDataSource( MSIRECORD *rec, LPVOID param )
5186 {
5187 LPWSTR attrs;
5188 LPCWSTR desc, driver;
5189 WORD request = ODBC_REMOVE_SYS_DSN;
5190 INT registration;
5191 DWORD len;
5192
5193 static const WCHAR attrs_fmt[] = {
5194 'D','S','N','=','%','s',0 };
5195
5196 desc = MSI_RecordGetString( rec, 3 );
5197 driver = MSI_RecordGetString( rec, 4 );
5198 registration = MSI_RecordGetInteger( rec, 5 );
5199
5200 if (registration == msidbODBCDataSourceRegistrationPerMachine) request = ODBC_REMOVE_SYS_DSN;
5201 else if (registration == msidbODBCDataSourceRegistrationPerUser) request = ODBC_REMOVE_DSN;
5202
5203 len = strlenW( attrs_fmt ) + strlenW( desc ) + 1 + 1;
5204 attrs = msi_alloc( len * sizeof(WCHAR) );
5205 if (!attrs)
5206 return ERROR_OUTOFMEMORY;
5207
5208 FIXME("Use ODBCSourceAttribute table\n");
5209
5210 len = sprintfW( attrs, attrs_fmt, desc );
5211 attrs[len + 1] = 0;
5212
5213 if (!SQLConfigDataSourceW( NULL, request, driver, attrs ))
5214 {
5215 WARN("Failed to remove ODBC data source\n");
5216 }
5217 msi_free( attrs );
5218
5219 return ERROR_SUCCESS;
5220 }
5221
5222 static UINT ACTION_RemoveODBC( MSIPACKAGE *package )
5223 {
5224 UINT rc;
5225 MSIQUERY *view;
5226
5227 static const WCHAR driver_query[] = {
5228 'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
5229 'O','D','B','C','D','r','i','v','e','r',0 };
5230
5231 static const WCHAR translator_query[] = {
5232 'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
5233 'O','D','B','C','T','r','a','n','s','l','a','t','o','r',0 };
5234
5235 static const WCHAR source_query[] = {
5236 'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
5237 'O','D','B','C','D','a','t','a','S','o','u','r','c','e',0 };
5238
5239 rc = MSI_DatabaseOpenViewW( package->db, driver_query, &view );
5240 if (rc != ERROR_SUCCESS)
5241 return ERROR_SUCCESS;
5242
5243 rc = MSI_IterateRecords( view, NULL, ITERATE_RemoveODBCDriver, package );
5244 msiobj_release( &view->hdr );
5245
5246 rc = MSI_DatabaseOpenViewW( package->db, translator_query, &view );
5247 if (rc != ERROR_SUCCESS)
5248 return ERROR_SUCCESS;
5249
5250 rc = MSI_IterateRecords( view, NULL, ITERATE_RemoveODBCTranslator, package );
5251 msiobj_release( &view->hdr );
5252
5253 rc = MSI_DatabaseOpenViewW( package->db, source_query, &view );
5254 if (rc != ERROR_SUCCESS)
5255 return ERROR_SUCCESS;
5256
5257 rc = MSI_IterateRecords( view, NULL, ITERATE_RemoveODBCDataSource, package );
5258 msiobj_release( &view->hdr );
5259
5260 return rc;
5261 }
5262
5263 #define ENV_ACT_SETALWAYS 0x1
5264 #define ENV_ACT_SETABSENT 0x2
5265 #define ENV_ACT_REMOVE 0x4
5266 #define ENV_ACT_REMOVEMATCH 0x8
5267
5268 #define ENV_MOD_MACHINE 0x20000000
5269 #define ENV_MOD_APPEND 0x40000000
5270 #define ENV_MOD_PREFIX 0x80000000
5271 #define ENV_MOD_MASK 0xC0000000
5272
5273 #define check_flag_combo(x, y) ((x) & ~(y)) == (y)
5274
5275 static LONG env_set_flags( LPCWSTR *name, LPCWSTR *value, DWORD *flags )
5276 {
5277 LPCWSTR cptr = *name;
5278
5279 static const WCHAR prefix[] = {'[','~',']',0};
5280 static const int prefix_len = 3;
5281
5282 *flags = 0;
5283 while (*cptr)
5284 {
5285 if (*cptr == '=')
5286 *flags |= ENV_ACT_SETALWAYS;
5287 else if (*cptr == '+')
5288 *flags |= ENV_ACT_SETABSENT;
5289 else if (*cptr == '-')
5290 *flags |= ENV_ACT_REMOVE;
5291 else if (*cptr == '!')
5292 *flags |= ENV_ACT_REMOVEMATCH;
5293 else if (*cptr == '*')
5294 *flags |= ENV_MOD_MACHINE;
5295 else
5296 break;
5297
5298 cptr++;
5299 (*name)++;
5300 }
5301
5302 if (!*cptr)
5303 {
5304 ERR("Missing environment variable\n");
5305 return ERROR_FUNCTION_FAILED;
5306 }
5307
5308 if (*value)
5309 {
5310 LPCWSTR ptr = *value;
5311 if (!strncmpW(ptr, prefix, prefix_len))
5312 {
5313 if (ptr[prefix_len] == szSemiColon[0])
5314 {
5315 *flags |= ENV_MOD_APPEND;
5316 *value += lstrlenW(prefix);
5317 }
5318 else
5319 {
5320 *value = NULL;
5321 }
5322 }
5323 else if (lstrlenW(*value) >= prefix_len)
5324 {
5325 ptr += lstrlenW(ptr) - prefix_len;
5326 if (!lstrcmpW(ptr, prefix))
5327 {
5328 if ((ptr-1) > *value && *(ptr-1) == szSemiColon[0])
5329 {
5330 *flags |= ENV_MOD_PREFIX;
5331 /* the "[~]" will be removed by deformat_string */;
5332 }
5333 else
5334 {
5335 *value = NULL;
5336 }
5337 }
5338 }
5339 }
5340
5341 if (check_flag_combo(*flags, ENV_ACT_SETALWAYS | ENV_ACT_SETABSENT) ||
5342 check_flag_combo(*flags, ENV_ACT_REMOVEMATCH | ENV_ACT_SETABSENT) ||
5343 check_flag_combo(*flags, ENV_ACT_REMOVEMATCH | ENV_ACT_SETALWAYS) ||
5344 check_flag_combo(*flags, ENV_ACT_SETABSENT | ENV_MOD_MASK))
5345 {
5346 ERR("Invalid flags: %08x\n", *flags);
5347 return ERROR_FUNCTION_FAILED;
5348 }
5349
5350 if (!*flags)
5351 *flags = ENV_ACT_SETALWAYS | ENV_ACT_REMOVE;
5352
5353 return ERROR_SUCCESS;
5354 }
5355
5356 static UINT ITERATE_WriteEnvironmentString( MSIRECORD *rec, LPVOID param )
5357 {
5358 MSIPACKAGE *package = param;
5359 LPCWSTR name, value;
5360 LPWSTR data = NULL, newval = NULL;
5361 LPWSTR deformatted = NULL, ptr;
5362 DWORD flags, type, size;
5363 LONG res;
5364 HKEY env = NULL, root;
5365 LPCWSTR environment;
5366
5367 static const WCHAR user_env[] =
5368 {'E','n','v','i','r','o','n','m','e','n','t',0};
5369 static const WCHAR machine_env[] =
5370 {'S','y','s','t','e','m','\\',
5371 'C','u','r','r','e','n','t','C','o','n','t','r','o','l','S','e','t','\\',
5372 'C','o','n','t','r','o','l','\\',
5373 'S','e','s','s','i','o','n',' ','M','a','n','a','g','e','r','\\',
5374 'E','n','v','i','r','o','n','m','e','n','t',0};
5375
5376 name = MSI_RecordGetString(rec, 2);
5377 value = MSI_RecordGetString(rec, 3);
5378
5379 TRACE("name %s value %s\n", debugstr_w(name), debugstr_w(value));
5380
5381 res = env_set_flags(&name, &value, &flags);
5382 if (res != ERROR_SUCCESS || !value)
5383 goto done;
5384
5385 if (value && !deformat_string(package, value, &deformatted))
5386 {
5387 res = ERROR_OUTOFMEMORY;
5388 goto done;
5389 }
5390
5391 value = deformatted;
5392
5393 if (flags & ENV_MOD_MACHINE)
5394 {
5395 environment = machine_env;
5396 root = HKEY_LOCAL_MACHINE;
5397 }
5398 else
5399 {
5400 environment = user_env;
5401 root = HKEY_CURRENT_USER;
5402 }
5403
5404 res = RegCreateKeyExW(root, environment, 0, NULL, 0,
5405 KEY_ALL_ACCESS, NULL, &env, NULL);
5406 if (res != ERROR_SUCCESS)
5407 goto done;
5408
5409 if (flags & ENV_ACT_REMOVE)
5410 FIXME("Not removing environment variable on uninstall!\n");
5411
5412 size = 0;
5413 type = REG_SZ;
5414 res = RegQueryValueExW(env, name, NULL, &type, NULL, &size);
5415 if ((res != ERROR_SUCCESS && res != ERROR_FILE_NOT_FOUND) ||
5416 (res == ERROR_SUCCESS && type != REG_SZ && type != REG_EXPAND_SZ))
5417 goto done;
5418
5419 if ((res == ERROR_FILE_NOT_FOUND || !(flags & ENV_MOD_MASK)))
5420 {
5421 /* Nothing to do. */
5422 if (!value)
5423 {
5424 res = ERROR_SUCCESS;
5425 goto done;
5426 }
5427
5428 /* If we are appending but the string was empty, strip ; */
5429 if ((flags & ENV_MOD_APPEND) && (value[0] == szSemiColon[0])) value++;
5430
5431 size = (lstrlenW(value) + 1) * sizeof(WCHAR);
5432 newval = strdupW(value);
5433 if (!newval)
5434 {
5435 res = ERROR_OUTOFMEMORY;
5436 goto done;
5437 }
5438 }
5439 else
5440 {
5441 /* Contrary to MSDN, +-variable to [~];path works */
5442 if (flags & ENV_ACT_SETABSENT && !(flags & ENV_MOD_MASK))
5443 {
5444 res = ERROR_SUCCESS;
5445 goto done;
5446 }
5447
5448 data = msi_alloc(size);
5449 if (!data)
5450 {
5451 RegCloseKey(env);
5452 return ERROR_OUTOFMEMORY;
5453 }
5454
5455 res = RegQueryValueExW(env, name, NULL, &type, (LPVOID)data, &size);
5456 if (res != ERROR_SUCCESS)
5457 goto done;
5458
5459 if (flags & ENV_ACT_REMOVEMATCH && (!value || !lstrcmpW(data, value)))
5460 {
5461 res = RegDeleteKeyW(env, name);
5462 goto done;
5463 }
5464
5465 size = (lstrlenW(data) + 1) * sizeof(WCHAR);
5466 if (flags & ENV_MOD_MASK)
5467 {
5468 DWORD mod_size;
5469 int multiplier = 0;
5470 if (flags & ENV_MOD_APPEND) multiplier++;
5471 if (flags & ENV_MOD_PREFIX) multiplier++;
5472 mod_size = lstrlenW(value) * multiplier;
5473 size += mod_size * sizeof(WCHAR);
5474 }
5475
5476 newval = msi_alloc(size);
5477 ptr = newval;
5478 if (!newval)
5479 {
5480 res = ERROR_OUTOFMEMORY;
5481 goto done;
5482 }
5483
5484 if (flags & ENV_MOD_PREFIX)
5485 {
5486 lstrcpyW(newval, value);
5487 ptr = newval + lstrlenW(value);
5488 }
5489
5490 lstrcpyW(ptr, data);
5491
5492 if (flags & ENV_MOD_APPEND)
5493 {
5494 lstrcatW(newval, value);
5495 }
5496 }
5497 TRACE("setting %s to %s\n", debugstr_w(name), debugstr_w(newval));
5498 res = RegSetValueExW(env, name, 0, type, (LPVOID)newval, size);
5499
5500 done:
5501 if (env) RegCloseKey(env);
5502 msi_free(deformatted);
5503 msi_free(data);
5504 msi_free(newval);
5505 return res;
5506 }
5507
5508 static UINT ACTION_WriteEnvironmentStrings( MSIPACKAGE *package )
5509 {
5510 UINT rc;
5511 MSIQUERY * view;
5512 static const WCHAR ExecSeqQuery[] =
5513 {'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
5514 '`','E','n','v','i','r','o','n','m','e','n','t','`',0};
5515 rc = MSI_DatabaseOpenViewW(package->db, ExecSeqQuery, &view);
5516 if (rc != ERROR_SUCCESS)
5517 return ERROR_SUCCESS;
5518
5519 rc = MSI_IterateRecords(view, NULL, ITERATE_WriteEnvironmentString, package);
5520 msiobj_release(&view->hdr);
5521
5522 return rc;
5523 }
5524
5525 #define is_dot_dir(x) ((x[0] == '.') && ((x[1] == 0) || ((x[1] == '.') && (x[2] == 0))))
5526
5527 typedef struct
5528 {
5529 struct list entry;
5530 LPWSTR sourcename;
5531 LPWSTR destname;
5532 LPWSTR source;
5533 LPWSTR dest;
5534 } FILE_LIST;
5535
5536 static BOOL msi_move_file(LPCWSTR source, LPCWSTR dest, int options)
5537 {
5538 BOOL ret;
5539
5540 if (GetFileAttributesW(source) == FILE_ATTRIBUTE_DIRECTORY ||
5541 GetFileAttributesW(dest) == FILE_ATTRIBUTE_DIRECTORY)
5542 {
5543 WARN("Source or dest is directory, not moving\n");
5544 return FALSE;
5545 }
5546
5547 if (options == msidbMoveFileOptionsMove)
5548 {
5549 TRACE("moving %s -> %s\n", debugstr_w(source), debugstr_w(dest));
5550 ret = MoveFileExW(source, dest, MOVEFILE_REPLACE_EXISTING);
5551 if (!ret)
5552 {
5553 WARN("MoveFile failed: %d\n", GetLastError());
5554 return FALSE;
5555 }
5556 }
5557 else
5558 {
5559 TRACE("copying %s -> %s\n", debugstr_w(source), debugstr_w(dest));
5560 ret = CopyFileW(source, dest, FALSE);
5561 if (!ret)
5562 {
5563 WARN("CopyFile failed: %d\n", GetLastError());
5564 return FALSE;
5565 }
5566 }
5567
5568 return TRUE;
5569 }
5570
5571 static LPWSTR wildcard_to_file(LPWSTR wildcard, LPWSTR filename)
5572 {
5573 LPWSTR path, ptr;
5574 DWORD dirlen, pathlen;
5575
5576 ptr = strrchrW(wildcard, '\\');
5577 dirlen = ptr - wildcard + 1;
5578
5579 pathlen = dirlen + lstrlenW(filename) + 1;
5580 path = msi_alloc(pathlen * sizeof(WCHAR));
5581
5582 lstrcpynW(path, wildcard, dirlen + 1);
5583 lstrcatW(path, filename);
5584
5585 return path;
5586 }
5587
5588 static void free_file_entry(FILE_LIST *file)
5589 {
5590 msi_free(file->source);
5591 msi_free(file->dest);
5592 msi_free(file);
5593 }
5594
5595 static void free_list(FILE_LIST *list)
5596 {
5597 while (!list_empty(&list->entry))
5598 {
5599 FILE_LIST *file = LIST_ENTRY(list_head(&list->entry), FILE_LIST, entry);
5600
5601 list_remove(&file->entry);
5602 free_file_entry(file);
5603 }
5604 }
5605
5606 static BOOL add_wildcard(FILE_LIST *files, LPWSTR source, LPWSTR dest)
5607 {
5608 FILE_LIST *new, *file;
5609 LPWSTR ptr, filename;
5610 DWORD size;
5611
5612 new = msi_alloc_zero(sizeof(FILE_LIST));
5613 if (!new)
5614 return FALSE;
5615
5616 new->source = strdupW(source);
5617 ptr = strrchrW(dest, '\\') + 1;
5618 filename = strrchrW(new->source, '\\') + 1;
5619
5620 new->sourcename = filename;
5621
5622 if (*ptr)
5623 new->destname = ptr;
5624 else
5625 new->destname = new->sourcename;
5626
5627 size = (ptr - dest) + lstrlenW(filename) + 1;
5628 new->dest = msi_alloc(size * sizeof(WCHAR));
5629 if (!new->dest)
5630 {
5631 free_file_entry(new);
5632 return FALSE;
5633 }
5634
5635 lstrcpynW(new->dest, dest, ptr - dest + 1);
5636 lstrcatW(new->dest, filename);
5637
5638 if (list_empty(&files->entry))
5639 {
5640 list_add_head(&files->entry, &new->entry);
5641 return TRUE;
5642 }
5643
5644 LIST_FOR_EACH_ENTRY(file, &files->entry, FILE_LIST, entry)
5645 {
5646 if (lstrcmpW(source, file->source) < 0)
5647 {
5648 list_add_before(&file->entry, &new->entry);
5649 return TRUE;
5650 }
5651 }
5652
5653 list_add_after(&file->entry, &new->entry);
5654 return TRUE;
5655 }
5656
5657 static BOOL move_files_wildcard(LPWSTR source, LPWSTR dest, int options)
5658 {
5659 WIN32_FIND_DATAW wfd;
5660 HANDLE hfile;
5661 LPWSTR path;
5662 BOOL res;
5663 FILE_LIST files, *file;
5664 DWORD size;
5665
5666 hfile = FindFirstFileW(source, &wfd);
5667 if (hfile == INVALID_HANDLE_VALUE) return FALSE;
5668
5669 list_init(&files.entry);
5670
5671 for (res = TRUE; res; res = FindNextFileW(hfile, &wfd))
5672 {
5673 if (is_dot_dir(wfd.cFileName)) continue;
5674
5675 path = wildcard_to_file(source, wfd.cFileName);
5676 if (!path)
5677 {
5678 res = FALSE;
5679 goto done;
5680 }
5681
5682 add_wildcard(&files, path, dest);
5683 msi_free(path);
5684 }
5685
5686 /* no files match the wildcard */
5687 if (list_empty(&files.entry))
5688 goto done;
5689
5690 /* only the first wildcard match gets renamed to dest */
5691 file = LIST_ENTRY(list_head(&files.entry), FILE_LIST, entry);
5692 size = (strrchrW(file->dest, '\\') - file->dest) + lstrlenW(file->destname) + 2;
5693 file->dest = msi_realloc(file->dest, size * sizeof(WCHAR));
5694 if (!file->dest)
5695 {
5696 res = FALSE;
5697 goto done;
5698 }
5699
5700 /* file->dest may be shorter after the reallocation, so add a NULL
5701 * terminator. This is needed for the call to strrchrW, as there will no
5702 * longer be a NULL terminator within the bounds of the allocation in this case.
5703 */
5704 file->dest[size - 1] = '\0';
5705 lstrcpyW(strrchrW(file->dest, '\\') + 1, file->destname);
5706
5707 while (!list_empty(&files.entry))
5708 {
5709 file = LIST_ENTRY(list_head(&files.entry), FILE_LIST, entry);
5710
5711 msi_move_file(file->source, file->dest, options);
5712
5713 list_remove(&file->entry);
5714 free_file_entry(file);
5715 }
5716
5717 res = TRUE;
5718
5719 done:
5720 free_list(&files);
5721 FindClose(hfile);
5722 return res;
5723 }
5724
5725 static UINT ITERATE_MoveFiles( MSIRECORD *rec, LPVOID param )
5726 {
5727 MSIPACKAGE *package = param;
5728 MSICOMPONENT *comp;
5729 LPCWSTR sourcename;
5730 LPWSTR destname = NULL;
5731 LPWSTR sourcedir = NULL, destdir = NULL;
5732 LPWSTR source = NULL, dest = NULL;
5733 int options;
5734 DWORD size;
5735 BOOL ret, wildcards;
5736
5737 comp = get_loaded_component(package, MSI_RecordGetString(rec, 2));
5738 if (!comp || !comp->Enabled ||
5739 !(comp->Action & (INSTALLSTATE_LOCAL | INSTALLSTATE_SOURCE)))
5740 {
5741 TRACE("Component not set for install, not moving file\n");
5742 return ERROR_SUCCESS;
5743 }
5744
5745 sourcename = MSI_RecordGetString(rec, 3);
5746 options = MSI_RecordGetInteger(rec, 7);
5747
5748 sourcedir = msi_dup_property(package, MSI_RecordGetString(rec, 5));
5749 if (!sourcedir)
5750 goto done;
5751
5752 destdir = msi_dup_property(package, MSI_RecordGetString(rec, 6));
5753 if (!destdir)
5754 goto done;
5755
5756 if (!sourcename)
5757 {
5758 if (GetFileAttributesW(sourcedir) == INVALID_FILE_ATTRIBUTES)
5759 goto done;
5760
5761 source = strdupW(sourcedir);
5762 if (!source)
5763 goto done;
5764 }
5765 else
5766 {
5767 size = lstrlenW(sourcedir) + lstrlenW(sourcename) + 2;
5768 source = msi_alloc(size * sizeof(WCHAR));
5769 if (!source)
5770 goto done;
5771
5772 lstrcpyW(source, sourcedir);
5773 if (source[lstrlenW(source) - 1] != '\\')
5774 lstrcatW(source, szBackSlash);
5775 lstrcatW(source, sourcename);
5776 }
5777
5778 wildcards = strchrW(source, '*') || strchrW(source, '?');
5779
5780 if (MSI_RecordIsNull(rec, 4))
5781 {
5782 if (!wildcards)
5783 {
5784 destname = strdupW(sourcename);
5785 if (!destname)
5786 goto done;
5787 }
5788 }
5789 else
5790 {
5791 destname = strdupW(MSI_RecordGetString(rec, 4));
5792 if (destname)
5793 reduce_to_longfilename(destname);
5794 }
5795
5796 size = 0;
5797 if (destname)
5798 size = lstrlenW(destname);
5799
5800 size += lstrlenW(destdir) + 2;
5801 dest = msi_alloc(size * sizeof(WCHAR));
5802 if (!dest)
5803 goto done;
5804
5805 lstrcpyW(dest, destdir);
5806 if (dest[lstrlenW(dest) - 1] != '\\')
5807 lstrcatW(dest, szBackSlash);
5808
5809 if (destname)
5810 lstrcatW(dest, destname);
5811
5812 if (GetFileAttributesW(destdir) == INVALID_FILE_ATTRIBUTES)
5813 {
5814 ret = CreateDirectoryW(destdir, NULL);
5815 if (!ret)
5816 {
5817 WARN("CreateDirectory failed: %d\n", GetLastError());
5818 return ERROR_SUCCESS;
5819 }
5820 }
5821
5822 if (!wildcards)
5823 msi_move_file(source, dest, options);
5824 else
5825 move_files_wildcard(source, dest, options);
5826
5827 done:
5828 msi_free(sourcedir);
5829 msi_free(destdir);
5830 msi_free(destname);
5831 msi_free(source);
5832 msi_free(dest);
5833
5834 return ERROR_SUCCESS;
5835 }
5836
5837 static UINT ACTION_MoveFiles( MSIPACKAGE *package )
5838 {
5839 UINT rc;
5840 MSIQUERY *view;
5841
5842 static const WCHAR ExecSeqQuery[] =
5843 {'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
5844 '`','M','o','v','e','F','i','l','e','`',0};
5845
5846 rc = MSI_DatabaseOpenViewW(package->db, ExecSeqQuery, &view);
5847 if (rc != ERROR_SUCCESS)
5848 return ERROR_SUCCESS;
5849
5850 rc = MSI_IterateRecords(view, NULL, ITERATE_MoveFiles, package);
5851 msiobj_release(&view->hdr);
5852
5853 return rc;
5854 }
5855
5856 typedef struct tagMSIASSEMBLY
5857 {
5858 struct list entry;
5859 MSICOMPONENT *component;
5860 MSIFEATURE *feature;
5861 MSIFILE *file;
5862 LPWSTR manifest;
5863 LPWSTR application;
5864 DWORD attributes;
5865 BOOL installed;
5866 } MSIASSEMBLY;
5867
5868 static HRESULT (WINAPI *pCreateAssemblyCache)(IAssemblyCache **ppAsmCache,
5869 DWORD dwReserved);
5870 static HRESULT (WINAPI *pLoadLibraryShim)(LPCWSTR szDllName, LPCWSTR szVersion,
5871 LPVOID pvReserved, HMODULE *phModDll);
5872
5873 static BOOL init_functionpointers(void)
5874 {
5875 HRESULT hr;
5876 HMODULE hfusion;
5877 HMODULE hmscoree;
5878
5879 static const WCHAR szFusion[] = {'f','u','s','i','o','n','.','d','l','l',0};
5880
5881 hmscoree = LoadLibraryA("mscoree.dll");
5882 if (!hmscoree)
5883 {
5884 WARN("mscoree.dll not available\n");
5885 return FALSE;
5886 }
5887
5888 pLoadLibraryShim = (void *)GetProcAddress(hmscoree, "LoadLibraryShim");
5889 if (!pLoadLibraryShim)
5890 {
5891 WARN("LoadLibraryShim not available\n");
5892 FreeLibrary(hmscoree);
5893 return FALSE;
5894 }
5895
5896 hr = pLoadLibraryShim(szFusion, NULL, NULL, &hfusion);
5897 if (FAILED(hr))
5898 {
5899 WARN("fusion.dll not available\n");
5900 FreeLibrary(hmscoree);
5901 return FALSE;
5902 }
5903
5904 pCreateAssemblyCache = (void *)GetProcAddress(hfusion, "CreateAssemblyCache");
5905
5906 FreeLibrary(hmscoree);
5907 return TRUE;
5908 }
5909
5910 static UINT install_assembly(MSIPACKAGE *package, MSIASSEMBLY *assembly,
5911 LPWSTR path)
5912 {
5913 IAssemblyCache *cache;
5914 HRESULT hr;
5915 UINT r = ERROR_FUNCTION_FAILED;
5916
5917 TRACE("installing assembly: %s\n", debugstr_w(path));
5918
5919 if (assembly->feature)
5920 msi_feature_set_state(package, assembly->feature, INSTALLSTATE_LOCAL);
5921
5922 if (assembly->manifest)
5923 FIXME("Manifest unhandled\n");
5924
5925 if (assembly->application)
5926 {
5927 FIXME("Assembly should be privately installed\n");
5928 return ERROR_SUCCESS;
5929 }
5930
5931 if (assembly->attributes == msidbAssemblyAttributesWin32)
5932 {
5933 FIXME("Win32 assemblies not handled\n");
5934 return ERROR_SUCCESS;
5935 }
5936
5937 hr = pCreateAssemblyCache(&cache, 0);
5938 if (FAILED(hr))
5939 goto done;
5940
5941 hr = IAssemblyCache_InstallAssembly(cache, 0, path, NULL);
5942 if (FAILED(hr))
5943 ERR("Failed to install assembly: %s %08x\n", debugstr_w(path), hr);
5944
5945 r = ERROR_SUCCESS;
5946
5947 done:
5948 IAssemblyCache_Release(cache);
5949 return r;
5950 }
5951
5952 typedef struct tagASSEMBLY_LIST
5953 {
5954 MSIPACKAGE *package;
5955 IAssemblyCache *cache;
5956 struct list *assemblies;
5957 } ASSEMBLY_LIST;
5958
5959 typedef struct tagASSEMBLY_NAME
5960 {
5961 LPWSTR name;
5962 LPWSTR version;
5963 LPWSTR culture;
5964 LPWSTR pubkeytoken;
5965 } ASSEMBLY_NAME;
5966
5967 static UINT parse_assembly_name(MSIRECORD *rec, LPVOID param)
5968 {
5969 ASSEMBLY_NAME *asmname = param;
5970 LPCWSTR name = MSI_RecordGetString(rec, 2);
5971 LPWSTR val = msi_dup_record_field(rec, 3);
5972
5973 static const WCHAR Name[] = {'N','a','m','e',0};
5974 static const WCHAR Version[] = {'V','e','r','s','i','o','n',0};
5975 static const WCHAR Culture[] = {'C','u','l','t','u','r','e',0};
5976 static const WCHAR PublicKeyToken[] = {
5977 'P','u','b','l','i','c','K','e','y','T','o','k','e','n',0};
5978
5979 if (!strcmpiW(name, Name))
5980 asmname->name = val;
5981 else if (!strcmpiW(name, Version))
5982 asmname->version = val;
5983 else if (!strcmpiW(name, Culture))
5984 asmname->culture = val;
5985 else if (!strcmpiW(name, PublicKeyToken))
5986 asmname->pubkeytoken = val;
5987 else
5988 msi_free(val);
5989
5990 return ERROR_SUCCESS;
5991 }
5992
5993 static void append_str(LPWSTR *str, DWORD *size, LPCWSTR append)
5994 {
5995 if (!*str)
5996 {
5997 *size = lstrlenW(append) + 1;
5998 *str = msi_alloc((*size) * sizeof(WCHAR));
5999 lstrcpyW(*str, append);
6000 return;
6001 }
6002
6003 (*size) += lstrlenW(append);
6004 *str = msi_realloc(*str, (*size) * sizeof(WCHAR));
6005 lstrcatW(*str, append);
6006 }
6007
6008 static BOOL check_assembly_installed(MSIDATABASE *db, IAssemblyCache *cache,
6009 MSICOMPONENT *comp)
6010 {
6011 ASSEMBLY_INFO asminfo;
6012 ASSEMBLY_NAME name;
6013 MSIQUERY *view;
6014 LPWSTR disp;
6015 DWORD size;
6016 BOOL found;
6017 UINT r;
6018
6019 static const WCHAR separator[] = {',',' ',0};
6020 static const WCHAR Version[] = {'V','e','r','s','i','o','n','=',0};
6021 static const WCHAR Culture[] = {'C','u','l','t','u','r','e','=',0};
6022 static const WCHAR PublicKeyToken[] = {
6023 'P','u','b','l','i','c','K','e','y','T','o','k','e','n','=',0};
6024 static const WCHAR query[] = {
6025 'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
6026 '`','M','s','i','A','s','s','e','m','b','l','y','N','a','m','e','`',' ',
6027 'W','H','E','R','E',' ','`','C','o','m','p','o','n','e','n','t','_','`',
6028 '=','\'','%','s','\'',0};
6029
6030 disp = NULL;
6031 found = FALSE;
6032 ZeroMemory(&name, sizeof(ASSEMBLY_NAME));
6033 ZeroMemory(&asminfo, sizeof(ASSEMBLY_INFO));
6034
6035 r = MSI_OpenQuery(db, &view, query, comp->Component);
6036 if (r != ERROR_SUCCESS)
6037 return ERROR_SUCCESS;
6038
6039 MSI_IterateRecords(view, NULL, parse_assembly_name, &name);
6040 msiobj_release(&view->hdr);
6041
6042 if (!name.name)
6043 {
6044 ERR("No assembly name specified!\n");
6045 goto done;
6046 }
6047
6048 append_str(&disp, &size, name.name);
6049
6050 if (name.version)
6051 {
6052 append_str(&disp, &size, separator);
6053 append_str(&disp, &size, Version);
6054 append_str(&disp, &size, name.version);
6055 }
6056
6057 if (name.culture)
6058 {
6059 append_str(&disp, &size, separator);
6060 append_str(&disp, &size, Culture);
6061 append_str(&disp, &size, name.culture);
6062 }
6063
6064 if (name.pubkeytoken)
6065 {
6066 append_str(&disp, &size, separator);
6067 append_str(&disp, &size, PublicKeyToken);
6068 append_str(&disp, &size, name.pubkeytoken);
6069 }
6070
6071 asminfo.cbAssemblyInfo = sizeof(ASSEMBLY_INFO);
6072 IAssemblyCache_QueryAssemblyInfo(cache, QUERYASMINFO_FLAG_VALIDATE,
6073 disp, &asminfo);
6074 found = (asminfo.dwAssemblyFlags == ASSEMBLYINFO_FLAG_INSTALLED);
6075
6076 done:
6077 msi_free(disp);
6078 msi_free(name.name);
6079 msi_free(name.version);
6080 msi_free(name.culture);
6081 msi_free(name.pubkeytoken);
6082
6083 return found;
6084 }
6085
6086 static UINT load_assembly(MSIRECORD *rec, LPVOID param)
6087 {
6088 ASSEMBLY_LIST *list = param;
6089 MSIASSEMBLY *assembly;
6090
6091 assembly = msi_alloc_zero(sizeof(MSIASSEMBLY));
6092 if (!assembly)
6093 return ERROR_OUTOFMEMORY;
6094
6095 assembly->component = get_loaded_component(list->package, MSI_RecordGetString(rec, 1));
6096
6097 if (!assembly->component || !assembly->component->Enabled ||
6098 !(assembly->component->Action & (INSTALLSTATE_LOCAL | INSTALLSTATE_SOURCE)))
6099 {
6100 TRACE("Component not set for install, not publishing assembly\n");
6101 msi_free(assembly);
6102 return ERROR_SUCCESS;
6103 }
6104
6105 assembly->feature = find_feature_by_name(list->package, MSI_RecordGetString(rec, 2));
6106 assembly->file = msi_find_file(list->package, assembly->component->KeyPath);
6107
6108 if (!assembly->file)
6109 {
6110 ERR("File %s not found\n", debugstr_w(assembly->component->KeyPath));
6111 return ERROR_FUNCTION_FAILED;
6112 }
6113
6114 assembly->manifest = strdupW(MSI_RecordGetString(rec, 3));
6115 assembly->application = strdupW(MSI_RecordGetString(rec, 4));
6116 assembly->attributes = MSI_RecordGetInteger(rec, 5);
6117
6118 if (assembly->application)
6119 {
6120 WCHAR version[24];
6121 DWORD size = sizeof(version)/sizeof(WCHAR);
6122
6123 /* FIXME: we should probably check the manifest file here */
6124
6125 if (!MsiGetFileVersionW(assembly->file->TargetPath, version, &size, NULL, NULL) &&
6126 (!assembly->file->Version || strcmpW(version, assembly->file->Version) >= 0))
6127 {
6128 assembly->installed = TRUE;
6129 }
6130 }
6131 else
6132 assembly->installed = check_assembly_installed(list->package->db,
6133 list->cache,
6134 assembly->component);
6135
6136 list_add_head(list->assemblies, &assembly->entry);
6137 return ERROR_SUCCESS;
6138 }
6139
6140 static UINT load_assemblies(MSIPACKAGE *package, struct list *assemblies)
6141 {
6142 IAssemblyCache *cache = NULL;
6143 ASSEMBLY_LIST list;
6144 MSIQUERY *view;
6145 HRESULT hr;
6146 UINT r;
6147
6148 static const WCHAR query[] =
6149 {'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
6150 '`','M','s','i','A','s','s','e','m','b','l','y','`',0};
6151
6152 r = MSI_DatabaseOpenViewW(package->db, query, &view);
6153 if (r != ERROR_SUCCESS)
6154 return ERROR_SUCCESS;
6155
6156 hr = pCreateAssemblyCache(&cache, 0);
6157 if (FAILED(hr))
6158 return ERROR_FUNCTION_FAILED;
6159
6160 list.package = package;
6161 list.cache = cache;
6162 list.assemblies = assemblies;
6163
6164 r = MSI_IterateRecords(view, NULL, load_assembly, &list);
6165 msiobj_release(&view->hdr);
6166
6167 IAssemblyCache_Release(cache);
6168
6169 return r;
6170 }
6171
6172 static void free_assemblies(struct list *assemblies)
6173 {
6174 struct list *item, *cursor;
6175
6176 LIST_FOR_EACH_SAFE(item, cursor, assemblies)
6177 {
6178 MSIASSEMBLY *assembly = LIST_ENTRY(item, MSIASSEMBLY, entry);
6179
6180 list_remove(&assembly->entry);
6181 msi_free(assembly->application);
6182 msi_free(assembly->manifest);
6183 msi_free(assembly);
6184 }
6185 }
6186
6187 static BOOL find_assembly(struct list *assemblies, LPCWSTR file, MSIASSEMBLY **out)
6188 {
6189 MSIASSEMBLY *assembly;
6190
6191 LIST_FOR_EACH_ENTRY(assembly, assemblies, MSIASSEMBLY, entry)
6192 {
6193 if (!lstrcmpW(assembly->file->File, file))
6194 {
6195 *out = assembly;
6196 return TRUE;
6197 }
6198 }
6199
6200 return FALSE;
6201 }
6202
6203 static BOOL installassembly_cb(MSIPACKAGE *package, LPCWSTR file, DWORD action,
6204 LPWSTR *path, DWORD *attrs, PVOID user)
6205 {
6206 MSIASSEMBLY *assembly;
6207 WCHAR temppath[MAX_PATH];
6208 struct list *assemblies = user;
6209 UINT r;
6210
6211 if (!find_assembly(assemblies, file, &assembly))
6212 return FALSE;
6213
6214 GetTempPathW(MAX_PATH, temppath);
6215 PathAddBackslashW(temppath);
6216 lstrcatW(temppath, assembly->file->FileName);
6217
6218 if (action == MSICABEXTRACT_BEGINEXTRACT)
6219 {
6220 if (assembly->installed)
6221 return FALSE;
6222
6223 *path = strdupW(temppath);
6224 *attrs = assembly->file->Attributes;
6225 }
6226 else if (action == MSICABEXTRACT_FILEEXTRACTED)
6227 {
6228 assembly->installed = TRUE;
6229
6230 r = install_assembly(package, assembly, temppath);
6231 if (r != ERROR_SUCCESS)
6232 ERR("Failed to install assembly\n");
6233 }
6234
6235 return TRUE;
6236 }
6237
6238 static UINT ACTION_MsiPublishAssemblies( MSIPACKAGE *package )
6239 {
6240 UINT r;
6241 struct list assemblies = LIST_INIT(assemblies);
6242 MSIASSEMBLY *assembly;
6243 MSIMEDIAINFO *mi;
6244
6245 if (!init_functionpointers() || !pCreateAssemblyCache)
6246 return ERROR_FUNCTION_FAILED;
6247
6248 r = load_assemblies(package, &assemblies);
6249 if (r != ERROR_SUCCESS)
6250 goto done;
6251
6252 if (list_empty(&assemblies))
6253 goto done;
6254
6255 mi = msi_alloc_zero(sizeof(MSIMEDIAINFO));
6256 if (!mi)
6257 {
6258 r = ERROR_OUTOFMEMORY;
6259 goto done;
6260 }
6261
6262 LIST_FOR_EACH_ENTRY(assembly, &assemblies, MSIASSEMBLY, entry)
6263 {
6264 if (assembly->installed && !mi->is_continuous)
6265 continue;
6266
6267 if (assembly->file->Sequence > mi->last_sequence || mi->is_continuous ||
6268 (assembly->file->IsCompressed && !mi->is_extracted))
6269 {
6270 MSICABDATA data;
6271
6272 r = ready_media(package, assembly->file, mi);
6273 if (r != ERROR_SUCCESS)
6274 {
6275 ERR("Failed to ready media\n");
6276 break;
6277 }
6278
6279 data.mi = mi;
6280 data.package = package;
6281 data.cb = installassembly_cb;
6282 data.user = &assemblies;
6283
6284 if (assembly->file->IsCompressed &&
6285 !msi_cabextract(package, mi, &data))
6286 {
6287 ERR("Failed to extract cabinet: %s\n", debugstr_w(mi->cabinet));
6288 r = ERROR_FUNCTION_FAILED;
6289 break;
6290 }
6291 }
6292
6293 if (!assembly->file->IsCompressed)
6294 {
6295 LPWSTR source = resolve_file_source(package, assembly->file);
6296
6297 r = install_assembly(package, assembly, source);
6298 if (r != ERROR_SUCCESS)
6299 ERR("Failed to install assembly\n");
6300
6301 msi_free(source);
6302 }
6303
6304 /* FIXME: write Installer assembly reg values */
6305 }
6306
6307 done:
6308 free_assemblies(&assemblies);
6309 return r;
6310 }
6311
6312 static UINT ACTION_ValidateProductID( MSIPACKAGE *package )
6313 {
6314 LPWSTR key, template, id;
6315 UINT r = ERROR_SUCCESS;
6316
6317 id = msi_dup_property( package, szProductID );
6318 if (id)
6319 {
6320 msi_free( id );
6321 return ERROR_SUCCESS;
6322 }
6323 template = msi_dup_property( package, szPIDTemplate );
6324 key = msi_dup_property( package, szPIDKEY );
6325
6326 if (key && template)
6327 {
6328 FIXME( "partial stub: template %s key %s\n", debugstr_w(template), debugstr_w(key) );
6329 r = MSI_SetPropertyW( package, szProductID, key );
6330 }
6331 msi_free( template );
6332 msi_free( key );
6333 return r;
6334 }
6335
6336 static UINT ACTION_ScheduleReboot( MSIPACKAGE *package )
6337 {
6338 TRACE("\n");
6339 package->need_reboot = 1;
6340 return ERROR_SUCCESS;
6341 }
6342
6343 static UINT ACTION_AllocateRegistrySpace( MSIPACKAGE *package )
6344 {
6345 TRACE("%p\n", package);
6346 return ERROR_SUCCESS;
6347 }
6348
6349 static UINT ACTION_DisableRollback( MSIPACKAGE *package )
6350 {
6351 FIXME("%p\n", package);
6352 return ERROR_SUCCESS;
6353 }
6354
6355 static UINT ACTION_InstallAdminPackage( MSIPACKAGE *package )
6356 {
6357 FIXME("%p\n", package);
6358 return ERROR_SUCCESS;
6359 }
6360
6361 static UINT msi_unimplemented_action_stub( MSIPACKAGE *package,
6362 LPCSTR action, LPCWSTR table )
6363 {
6364 static const WCHAR query[] = {
6365 'S','E','L','E','C','T',' ','*',' ',
6366 'F','R','O','M',' ','`','%','s','`',0 };
6367 MSIQUERY *view = NULL;
6368 DWORD count = 0;
6369 UINT r;
6370
6371 r = MSI_OpenQuery( package->db, &view, query, table );
6372 if (r == ERROR_SUCCESS)
6373 {
6374 r = MSI_IterateRecords(view, &count, NULL, package);
6375 msiobj_release(&view->hdr);
6376 }
6377
6378 if (count)
6379 FIXME("%s -> %u ignored %s table values\n",
6380 action, count, debugstr_w(table));
6381
6382 return ERROR_SUCCESS;
6383 }
6384
6385 static UINT ACTION_RemoveIniValues( MSIPACKAGE *package )
6386 {
6387 static const WCHAR table[] =
6388 {'R','e','m','o','v','e','I','n','i','F','i','l','e',0 };
6389 return msi_unimplemented_action_stub( package, "RemoveIniValues", table );
6390 }
6391
6392 static UINT ACTION_PatchFiles( MSIPACKAGE *package )
6393 {
6394 static const WCHAR table[] = { 'P','a','t','c','h',0 };
6395 return msi_unimplemented_action_stub( package, "PatchFiles", table );
6396 }
6397
6398 static UINT ACTION_BindImage( MSIPACKAGE *package )
6399 {
6400 static const WCHAR table[] = { 'B','i','n','d','I','m','a','g','e',0 };
6401 return msi_unimplemented_action_stub( package, "BindImage", table );
6402 }
6403
6404 static UINT ACTION_IsolateComponents( MSIPACKAGE *package )
6405 {
6406 static const WCHAR table[] = {
6407 'I','s','o','l','a','t','e','C','o','m','p','o','n','e','n','t',0 };
6408 return msi_unimplemented_action_stub( package, "IsolateComponents", table );
6409 }
6410
6411 static UINT ACTION_MigrateFeatureStates( MSIPACKAGE *package )
6412 {
6413 static const WCHAR table[] = { 'U','p','g','r','a','d','e',0 };
6414 return msi_unimplemented_action_stub( package, "MigrateFeatureStates", table );
6415 }
6416
6417 static UINT ACTION_RemoveEnvironmentStrings( MSIPACKAGE *package )
6418 {
6419 static const WCHAR table[] = {
6420 'E','n','v','i','r','o','n','m','e','n','t',0 };
6421 return msi_unimplemented_action_stub( package, "RemoveEnvironmentStrings", table );
6422 }
6423
6424 static UINT ACTION_MsiUnpublishAssemblies( MSIPACKAGE *package )
6425 {
6426 static const WCHAR table[] = {
6427 'M','s','i','A','s','s','e','m','b','l','y',0 };
6428 return msi_unimplemented_action_stub( package, "MsiUnpublishAssemblies", table );
6429 }
6430
6431 static UINT ACTION_RMCCPSearch( MSIPACKAGE *package )
6432 {
6433 static const WCHAR table[] = { 'C','C','P','S','e','a','r','c','h',0 };
6434 return msi_unimplemented_action_stub( package, "RMCCPSearch", table );
6435 }
6436
6437 static UINT ACTION_RegisterComPlus( MSIPACKAGE *package )
6438 {
6439 static const WCHAR table[] = { 'C','o','m','p','l','u','s',0 };
6440 return msi_unimplemented_action_stub( package, "RegisterComPlus", table );
6441 }
6442
6443 static UINT ACTION_UnregisterComPlus( MSIPACKAGE *package )
6444 {
6445 static const WCHAR table[] = { 'C','o','m','p','l','u','s',0 };
6446 return msi_unimplemented_action_stub( package, "UnregisterComPlus", table );
6447 }
6448
6449 static UINT ACTION_InstallSFPCatalogFile( MSIPACKAGE *package )
6450 {
6451 static const WCHAR table[] = { 'S','F','P','C','a','t','a','l','o','g',0 };
6452 return msi_unimplemented_action_stub( package, "InstallSFPCatalogFile", table );
6453 }
6454
6455 static UINT ACTION_RemoveDuplicateFiles( MSIPACKAGE *package )
6456 {
6457 static const WCHAR table[] = { 'D','u','p','l','i','c','a','t','e','F','i','l','e',0 };
6458 return msi_unimplemented_action_stub( package, "RemoveDuplicateFiles", table );
6459 }
6460
6461 static UINT ACTION_RemoveExistingProducts( MSIPACKAGE *package )
6462 {
6463 static const WCHAR table[] = { 'U','p','g','r','a','d','e',0 };
6464 return msi_unimplemented_action_stub( package, "RemoveExistingProducts", table );
6465 }
6466
6467 static UINT ACTION_RemoveRegistryValues( MSIPACKAGE *package )
6468 {
6469 static const WCHAR table[] = { 'R','e','m','o','v','e','R','e','g','i','s','t','r','y',0 };
6470 return msi_unimplemented_action_stub( package, "RemoveRegistryValues", table );
6471 }
6472
6473 static UINT ACTION_RemoveShortcuts( MSIPACKAGE *package )
6474 {
6475 static const WCHAR table[] = { 'S','h','o','r','t','c','u','t',0 };
6476 return msi_unimplemented_action_stub( package, "RemoveShortcuts", table );
6477 }
6478
6479 static UINT ACTION_SetODBCFolders( MSIPACKAGE *package )
6480 {
6481 static const WCHAR table[] = { 'D','i','r','e','c','t','o','r','y',0 };
6482 return msi_unimplemented_action_stub( package, "SetODBCFolders", table );
6483 }
6484
6485 static UINT ACTION_UnpublishComponents( MSIPACKAGE *package )
6486 {
6487 static const WCHAR table[] = { 'P','u','b','l','i','s','h','C','o','m','p','o','n','e','n','t',0 };
6488 return msi_unimplemented_action_stub( package, "UnpublishComponents", table );
6489 }
6490
6491 static UINT ACTION_UnregisterClassInfo( MSIPACKAGE *package )
6492 {
6493 static const WCHAR table[] = { 'A','p','p','I','d',0 };
6494 return msi_unimplemented_action_stub( package, "UnregisterClassInfo", table );
6495 }
6496
6497 static UINT ACTION_UnregisterExtensionInfo( MSIPACKAGE *package )
6498 {
6499 static const WCHAR table[] = { 'E','x','t','e','n','s','i','o','n',0 };
6500 return msi_unimplemented_action_stub( package, "UnregisterExtensionInfo", table );
6501 }
6502
6503 static UINT ACTION_UnregisterMIMEInfo( MSIPACKAGE *package )
6504 {
6505 static const WCHAR table[] = { 'M','I','M','E',0 };
6506 return msi_unimplemented_action_stub( package, "UnregisterMIMEInfo", table );
6507 }
6508
6509 static UINT ACTION_UnregisterProgIdInfo( MSIPACKAGE *package )
6510 {
6511 static const WCHAR table[] = { 'P','r','o','g','I','d',0 };
6512 return msi_unimplemented_action_stub( package, "UnregisterProgIdInfo", table );
6513 }
6514
6515 typedef UINT (*STANDARDACTIONHANDLER)(MSIPACKAGE*);
6516
6517 static const struct
6518 {
6519 const WCHAR *action;
6520 UINT (*handler)(MSIPACKAGE *);
6521 }
6522 StandardActions[] =
6523 {
6524 { szAllocateRegistrySpace, ACTION_AllocateRegistrySpace },
6525 { szAppSearch, ACTION_AppSearch },
6526 { szBindImage, ACTION_BindImage },
6527 { szCCPSearch, ACTION_CCPSearch },
6528 { szCostFinalize, ACTION_CostFinalize },
6529 { szCostInitialize, ACTION_CostInitialize },
6530 { szCreateFolders, ACTION_CreateFolders },
6531 { szCreateShortcuts, ACTION_CreateShortcuts },
6532 { szDeleteServices, ACTION_DeleteServices },
6533 { szDisableRollback, ACTION_DisableRollback },
6534 { szDuplicateFiles, ACTION_DuplicateFiles },
6535 { szExecuteAction, ACTION_ExecuteAction },
6536 { szFileCost, ACTION_FileCost },
6537 { szFindRelatedProducts, ACTION_FindRelatedProducts },
6538 { szForceReboot, ACTION_ForceReboot },
6539 { szInstallAdminPackage, ACTION_InstallAdminPackage },
6540 { szInstallExecute, ACTION_InstallExecute },
6541 { szInstallExecuteAgain, ACTION_InstallExecute },
6542 { szInstallFiles, ACTION_InstallFiles},
6543 { szInstallFinalize, ACTION_InstallFinalize },
6544 { szInstallInitialize, ACTION_InstallInitialize },
6545 { szInstallSFPCatalogFile, ACTION_InstallSFPCatalogFile },
6546 { szInstallValidate, ACTION_InstallValidate },
6547 { szIsolateComponents, ACTION_IsolateComponents },
6548 { szLaunchConditions, ACTION_LaunchConditions },
6549 { szMigrateFeatureStates, ACTION_MigrateFeatureStates },
6550 { szMoveFiles, ACTION_MoveFiles },
6551 { szMsiPublishAssemblies, ACTION_MsiPublishAssemblies },
6552 { szMsiUnpublishAssemblies, ACTION_MsiUnpublishAssemblies },
6553 { szInstallODBC, ACTION_InstallODBC },
6554 { szInstallServices, ACTION_InstallServices },
6555 { szPatchFiles, ACTION_PatchFiles },
6556 { szProcessComponents, ACTION_ProcessComponents },
6557 { szPublishComponents, ACTION_PublishComponents },
6558 { szPublishFeatures, ACTION_PublishFeatures },
6559 { szPublishProduct, ACTION_PublishProduct },
6560 { szRegisterClassInfo, ACTION_RegisterClassInfo },
6561 { szRegisterComPlus, ACTION_RegisterComPlus},
6562 { szRegisterExtensionInfo, ACTION_RegisterExtensionInfo },
6563 { szRegisterFonts, ACTION_RegisterFonts },
6564 { szRegisterMIMEInfo, ACTION_RegisterMIMEInfo },
6565 { szRegisterProduct, ACTION_RegisterProduct },
6566 { szRegisterProgIdInfo, ACTION_RegisterProgIdInfo },
6567 { szRegisterTypeLibraries, ACTION_RegisterTypeLibraries },
6568 { szRegisterUser, ACTION_RegisterUser },
6569 { szRemoveDuplicateFiles, ACTION_RemoveDuplicateFiles },
6570 { szRemoveEnvironmentStrings, ACTION_RemoveEnvironmentStrings },
6571 { szRemoveExistingProducts, ACTION_RemoveExistingProducts },
6572 { szRemoveFiles, ACTION_RemoveFiles },
6573 { szRemoveFolders, ACTION_RemoveFolders },
6574 { szRemoveIniValues, ACTION_RemoveIniValues },
6575 { szRemoveODBC, ACTION_RemoveODBC },
6576 { szRemoveRegistryValues, ACTION_RemoveRegistryValues },
6577 { szRemoveShortcuts, ACTION_RemoveShortcuts },
6578 { szResolveSource, ACTION_ResolveSource },
6579 { szRMCCPSearch, ACTION_RMCCPSearch },
6580 { szScheduleReboot, ACTION_ScheduleReboot },
6581 { szSelfRegModules, ACTION_SelfRegModules },
6582 { szSelfUnregModules, ACTION_SelfUnregModules },
6583 { szSetODBCFolders, ACTION_SetODBCFolders },
6584 { szStartServices, ACTION_StartServices },
6585 { szStopServices, ACTION_StopServices },
6586 { szUnpublishComponents, ACTION_UnpublishComponents },
6587 { szUnpublishFeatures, ACTION_UnpublishFeatures },
6588 { szUnregisterClassInfo, ACTION_UnregisterClassInfo },
6589 { szUnregisterComPlus, ACTION_UnregisterComPlus },
6590 { szUnregisterExtensionInfo, ACTION_UnregisterExtensionInfo },
6591 { szUnregisterFonts, ACTION_UnregisterFonts },
6592 { szUnregisterMIMEInfo, ACTION_UnregisterMIMEInfo },
6593 { szUnregisterProgIdInfo, ACTION_UnregisterProgIdInfo },
6594 { szUnregisterTypeLibraries, ACTION_UnregisterTypeLibraries },
6595 { szValidateProductID, ACTION_ValidateProductID },
6596 { szWriteEnvironmentStrings, ACTION_WriteEnvironmentStrings },
6597 { szWriteIniValues, ACTION_WriteIniValues },
6598 { szWriteRegistryValues, ACTION_WriteRegistryValues },
6599 { NULL, NULL },
6600 };
6601
6602 static BOOL ACTION_HandleStandardAction(MSIPACKAGE *package, LPCWSTR action,
6603 UINT* rc, BOOL force )
6604 {
6605 BOOL ret = FALSE;
6606 BOOL run = force;
6607 int i;
6608
6609 if (!run && !package->script->CurrentlyScripting)
6610 run = TRUE;
6611
6612 if (!run)
6613 {
6614 if (strcmpW(action,szInstallFinalize) == 0 ||
6615 strcmpW(action,szInstallExecute) == 0 ||
6616 strcmpW(action,szInstallExecuteAgain) == 0)
6617 run = TRUE;
6618 }
6619
6620 i = 0;
6621 while (StandardActions[i].action != NULL)
6622 {
6623 if (strcmpW(StandardActions[i].action, action)==0)
6624 {
6625 if (!run)
6626 {
6627 ui_actioninfo(package, action, TRUE, 0);
6628 *rc = schedule_action(package,INSTALL_SCRIPT,action);
6629 ui_actioninfo(package, action, FALSE, *rc);
6630 }
6631 else
6632 {
6633 ui_actionstart(package, action);
6634 if (StandardActions[i].handler)
6635 {
6636 *rc = StandardActions[i].handler(package);
6637 }
6638 else
6639 {
6640 FIXME("unhandled standard action %s\n",debugstr_w(action));
6641 *rc = ERROR_SUCCESS;
6642 }
6643 }
6644 ret = TRUE;
6645 break;
6646 }
6647 i++;
6648 }
6649 return ret;
6650 }
6651
6652 UINT ACTION_PerformAction(MSIPACKAGE *package, const WCHAR *action, UINT script, BOOL force)
6653 {
6654 UINT rc = ERROR_SUCCESS;
6655 BOOL handled;
6656
6657 TRACE("Performing action (%s)\n", debugstr_w(action));
6658
6659 handled = ACTION_HandleStandardAction(package, action, &rc, force);
6660
6661 if (!handled)
6662 handled = ACTION_HandleCustomAction(package, action, &rc, script, force);
6663
6664 if (!handled)
6665 {
6666 WARN("unhandled msi action %s\n", debugstr_w(action));
6667 rc = ERROR_FUNCTION_NOT_CALLED;
6668 }
6669
6670 return rc;
6671 }
6672
6673 UINT ACTION_PerformUIAction(MSIPACKAGE *package, const WCHAR *action, UINT script)
6674 {
6675 UINT rc = ERROR_SUCCESS;
6676 BOOL handled = FALSE;
6677
6678 TRACE("Performing action (%s)\n", debugstr_w(action));
6679
6680 handled = ACTION_HandleStandardAction(package, action, &rc,TRUE);
6681
6682 if (!handled)
6683 handled = ACTION_HandleCustomAction(package, action, &rc, script, FALSE);
6684
6685 if( !handled && ACTION_DialogBox(package, action) == ERROR_SUCCESS )
6686 handled = TRUE;
6687
6688 if (!handled)
6689 {
6690 WARN("unhandled msi action %s\n", debugstr_w(action));
6691 rc = ERROR_FUNCTION_NOT_CALLED;
6692 }
6693
6694 return rc;
6695 }
6696
6697 static UINT ACTION_PerformActionSequence(MSIPACKAGE *package, UINT seq)
6698 {
6699 UINT rc = ERROR_SUCCESS;
6700 MSIRECORD *row;
6701
6702 static const WCHAR ExecSeqQuery[] =
6703 {'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
6704 '`','I','n','s','t','a','l','l','E','x','e','c','u','t','e',
6705 'S','e','q','u','e','n','c','e','`',' ', 'W','H','E','R','E',' ',
6706 '`','S','e','q','u','e','n','c','e','`',' ', '=',' ','%','i',0};
6707 static const WCHAR UISeqQuery[] =
6708 {'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
6709 '`','I','n','s','t','a','l','l','U','I','S','e','q','u','e','n','c','e',
6710 '`', ' ', 'W','H','E','R','E',' ','`','S','e','q','u','e','n','c','e','`',
6711 ' ', '=',' ','%','i',0};
6712
6713 if (needs_ui_sequence(package))
6714 row = MSI_QueryGetRecord(package->db, UISeqQuery, seq);
6715 else
6716 row = MSI_QueryGetRecord(package->db, ExecSeqQuery, seq);
6717
6718 if (row)
6719 {
6720 LPCWSTR action, cond;
6721
6722 TRACE("Running the actions\n");
6723
6724 /* check conditions */
6725 cond = MSI_RecordGetString(row, 2);
6726
6727 /* this is a hack to skip errors in the condition code */
6728 if (MSI_EvaluateConditionW(package, cond) == MSICONDITION_FALSE)
6729 {
6730 msiobj_release(&row->hdr);
6731 return ERROR_SUCCESS;
6732 }
6733
6734 action = MSI_RecordGetString(row, 1);
6735 if (!action)
6736 {
6737 ERR("failed to fetch action\n");
6738 msiobj_release(&row->hdr);
6739 return ERROR_FUNCTION_FAILED;
6740 }
6741
6742 if (needs_ui_sequence(package))
6743 rc = ACTION_PerformUIAction(package, action, -1);
6744 else
6745 rc = ACTION_PerformAction(package, action, -1, FALSE);
6746
6747 msiobj_release(&row->hdr);
6748 }
6749
6750 return rc;
6751 }
6752
6753 /****************************************************
6754 * TOP level entry points
6755 *****************************************************/
6756
6757 UINT MSI_InstallPackage( MSIPACKAGE *package, LPCWSTR szPackagePath,
6758 LPCWSTR szCommandLine )
6759 {
6760 UINT rc;
6761 BOOL ui_exists;
6762
6763 static const WCHAR szAction[] = {'A','C','T','I','O','N',0};
6764 static const WCHAR szInstall[] = {'I','N','S','T','A','L','L',0};
6765
6766 MSI_SetPropertyW(package, szAction, szInstall);
6767
6768 package->script->InWhatSequence = SEQUENCE_INSTALL;
6769
6770 if (szPackagePath)
6771 {
6772 LPWSTR p, dir;
6773 LPCWSTR file;
6774
6775 dir = strdupW(szPackagePath);
6776 p = strrchrW(dir, '\\');
6777 if (p)
6778 {
6779 *(++p) = 0;
6780 file = szPackagePath + (p - dir);
6781 }
6782 else
6783 {
6784 msi_free(dir);
6785 dir = msi_alloc(MAX_PATH * sizeof(WCHAR));
6786 GetCurrentDirectoryW(MAX_PATH, dir);
6787 lstrcatW(dir, szBackSlash);
6788 file = szPackagePath;
6789 }
6790
6791 msi_free( package->PackagePath );
6792 package->PackagePath = msi_alloc((lstrlenW(dir) + lstrlenW(file) + 1) * sizeof(WCHAR));
6793 if (!package->PackagePath)
6794 {
6795 msi_free(dir);
6796 return ERROR_OUTOFMEMORY;
6797 }
6798
6799 lstrcpyW(package->PackagePath, dir);
6800 lstrcatW(package->PackagePath, file);
6801 msi_free(dir);
6802
6803 msi_set_sourcedir_props(package, FALSE);
6804 }
6805
6806 msi_parse_command_line( package, szCommandLine, FALSE );
6807
6808 msi_apply_transforms( package );
6809 msi_apply_patches( package );
6810
6811 if (!szCommandLine && msi_get_property_int( package, szInstalled, 0 ))
6812 {
6813 TRACE("setting reinstall property\n");
6814 MSI_SetPropertyW( package, szReinstall, szAll );
6815 }
6816
6817 /* properties may have been added by a transform */
6818 msi_clone_properties( package );
6819 msi_set_context( package );
6820
6821 if (needs_ui_sequence( package))
6822 {
6823 package->script->InWhatSequence |= SEQUENCE_UI;
6824 rc = ACTION_ProcessUISequence(package);
6825 ui_exists = ui_sequence_exists(package);
6826 if (rc == ERROR_SUCCESS || !ui_exists)
6827 {
6828 package->script->InWhatSequence |= SEQUENCE_EXEC;
6829 rc = ACTION_ProcessExecSequence(package, ui_exists);
6830 }
6831 }
6832 else
6833 rc = ACTION_ProcessExecSequence(package, FALSE);
6834
6835 package->script->CurrentlyScripting = FALSE;
6836
6837 /* process the ending type action */
6838 if (rc == ERROR_SUCCESS)
6839 ACTION_PerformActionSequence(package, -1);
6840 else if (rc == ERROR_INSTALL_USEREXIT)
6841 ACTION_PerformActionSequence(package, -2);
6842 else if (rc == ERROR_INSTALL_SUSPEND)
6843 ACTION_PerformActionSequence(package, -4);
6844 else /* failed */
6845 ACTION_PerformActionSequence(package, -3);
6846
6847 /* finish up running custom actions */
6848 ACTION_FinishCustomActions(package);
6849
6850 if (rc == ERROR_SUCCESS && package->need_reboot)
6851 return ERROR_SUCCESS_REBOOT_REQUIRED;
6852
6853 return rc;
6854 }
Maarten's multimedia branch