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