msi/tests: Fix copy-paste errors in tests.
[wine] / dlls / msi / install.c
1 /*
2  * Implementation of the Microsoft Installer (msi.dll)
3  *
4  * Copyright 2005 Aric Stewart for CodeWeavers
5  *
6  * This library is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * This library is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with this library; if not, write to the Free Software
18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
19  */
20
21 /* Msi top level apis directly related to installs */
22
23 #define COBJMACROS
24
25 #include <stdarg.h>
26
27 #include "windef.h"
28 #include "winbase.h"
29 #include "winerror.h"
30 #include "wine/debug.h"
31 #include "msi.h"
32 #include "msidefs.h"
33 #include "objbase.h"
34 #include "oleauto.h"
35
36 #include "msipriv.h"
37 #include "msiserver.h"
38 #include "wine/unicode.h"
39
40 WINE_DEFAULT_DEBUG_CHANNEL(msi);
41
42 /***********************************************************************
43  * MsiDoActionA       (MSI.@)
44  */
45 UINT WINAPI MsiDoActionA( MSIHANDLE hInstall, LPCSTR szAction )
46 {
47     LPWSTR szwAction;
48     UINT ret;
49
50     TRACE("%s\n", debugstr_a(szAction));
51
52     szwAction = strdupAtoW(szAction);
53     if (szAction && !szwAction)
54         return ERROR_FUNCTION_FAILED; 
55
56     ret = MsiDoActionW( hInstall, szwAction );
57     msi_free( szwAction );
58     return ret;
59 }
60
61 /***********************************************************************
62  * MsiDoActionW       (MSI.@)
63  */
64 UINT WINAPI MsiDoActionW( MSIHANDLE hInstall, LPCWSTR szAction )
65 {
66     MSIPACKAGE *package;
67     UINT ret;
68
69     TRACE("%s\n",debugstr_w(szAction));
70
71     if (!szAction)
72         return ERROR_INVALID_PARAMETER;
73
74     package = msihandle2msiinfo( hInstall, MSIHANDLETYPE_PACKAGE );
75     if (!package)
76     {
77         HRESULT hr;
78         BSTR action;
79         IWineMsiRemotePackage *remote_package;
80
81         remote_package = (IWineMsiRemotePackage *)msi_get_remote( hInstall );
82         if (!remote_package)
83             return ERROR_INVALID_HANDLE;
84
85         action = SysAllocString( szAction );
86         if (!action)
87         {
88             IWineMsiRemotePackage_Release( remote_package );
89             return ERROR_OUTOFMEMORY;
90         }
91
92         hr = IWineMsiRemotePackage_DoAction( remote_package, action );
93
94         SysFreeString( action );
95         IWineMsiRemotePackage_Release( remote_package );
96
97         if (FAILED(hr))
98         {
99             if (HRESULT_FACILITY(hr) == FACILITY_WIN32)
100                 return HRESULT_CODE(hr);
101
102             return ERROR_FUNCTION_FAILED;
103         }
104
105         return ERROR_SUCCESS;
106     }
107  
108     ret = ACTION_PerformUIAction( package, szAction, -1 );
109     msiobj_release( &package->hdr );
110
111     return ret;
112 }
113
114 /***********************************************************************
115  * MsiSequenceA       (MSI.@)
116  */
117 UINT WINAPI MsiSequenceA( MSIHANDLE hInstall, LPCSTR szTable, INT iSequenceMode )
118 {
119     LPWSTR szwTable;
120     UINT ret;
121
122     TRACE("%s\n", debugstr_a(szTable));
123
124     szwTable = strdupAtoW(szTable);
125     if (szTable && !szwTable)
126         return ERROR_FUNCTION_FAILED; 
127
128     ret = MsiSequenceW( hInstall, szwTable, iSequenceMode );
129     msi_free( szwTable );
130     return ret;
131 }
132
133 /***********************************************************************
134  * MsiSequenceW       (MSI.@)
135  */
136 UINT WINAPI MsiSequenceW( MSIHANDLE hInstall, LPCWSTR szTable, INT iSequenceMode )
137 {
138     MSIPACKAGE *package;
139     UINT ret;
140
141     TRACE("%s\n", debugstr_w(szTable));
142
143     package = msihandle2msiinfo( hInstall, MSIHANDLETYPE_PACKAGE );
144     if (!package)
145     {
146         HRESULT hr;
147         BSTR table;
148         IWineMsiRemotePackage *remote_package;
149
150         remote_package = (IWineMsiRemotePackage *)msi_get_remote( hInstall );
151         if (!remote_package)
152             return ERROR_INVALID_HANDLE;
153
154         table = SysAllocString( szTable );
155         if (!table)
156         {
157             IWineMsiRemotePackage_Release( remote_package );
158             return ERROR_OUTOFMEMORY;
159         }
160
161         hr = IWineMsiRemotePackage_Sequence( remote_package, table, iSequenceMode );
162
163         SysFreeString( table );
164         IWineMsiRemotePackage_Release( remote_package );
165
166         if (FAILED(hr))
167         {
168             if (HRESULT_FACILITY(hr) == FACILITY_WIN32)
169                 return HRESULT_CODE(hr);
170
171             return ERROR_FUNCTION_FAILED;
172         }
173
174         return ERROR_SUCCESS;
175     }
176
177     ret = MSI_Sequence( package, szTable, iSequenceMode );
178     msiobj_release( &package->hdr );
179  
180     return ret;
181 }
182
183 UINT msi_strcpy_to_awstring( LPCWSTR str, awstring *awbuf, DWORD *sz )
184 {
185     UINT len, r = ERROR_SUCCESS;
186
187     if (awbuf->str.w && !sz )
188         return ERROR_INVALID_PARAMETER;
189
190     if (!sz)
191         return r;
192  
193     if (awbuf->unicode)
194     {
195         len = lstrlenW( str );
196         if (awbuf->str.w) 
197             lstrcpynW( awbuf->str.w, str, *sz );
198     }
199     else
200     {
201         len = WideCharToMultiByte( CP_ACP, 0, str, -1, NULL, 0, NULL, NULL );
202         if (len)
203             len--;
204         WideCharToMultiByte( CP_ACP, 0, str, -1, awbuf->str.a, *sz, NULL, NULL );
205         if ( awbuf->str.a && *sz && (len >= *sz) )
206             awbuf->str.a[*sz - 1] = 0;
207     }
208
209     if (awbuf->str.w && len >= *sz)
210         r = ERROR_MORE_DATA;
211     *sz = len;
212     return r;
213 }
214
215 /***********************************************************************
216  * MsiGetTargetPath   (internal)
217  */
218 static UINT MSI_GetTargetPath( MSIHANDLE hInstall, LPCWSTR szFolder,
219                                awstring *szPathBuf, LPDWORD pcchPathBuf )
220 {
221     MSIPACKAGE *package;
222     LPWSTR path;
223     UINT r = ERROR_FUNCTION_FAILED;
224
225     if (!szFolder)
226         return ERROR_INVALID_PARAMETER;
227
228     package = msihandle2msiinfo( hInstall, MSIHANDLETYPE_PACKAGE );
229     if (!package)
230     {
231         HRESULT hr;
232         IWineMsiRemotePackage *remote_package;
233         LPWSTR value = NULL;
234         BSTR folder;
235         DWORD len;
236
237         remote_package = (IWineMsiRemotePackage *)msi_get_remote( hInstall );
238         if (!remote_package)
239             return ERROR_INVALID_HANDLE;
240
241         folder = SysAllocString( szFolder );
242         if (!folder)
243         {
244             IWineMsiRemotePackage_Release( remote_package );
245             return ERROR_OUTOFMEMORY;
246         }
247
248         len = 0;
249         hr = IWineMsiRemotePackage_GetTargetPath( remote_package, folder,
250                                                   NULL, &len );
251         if (FAILED(hr))
252             goto done;
253
254         len++;
255         value = msi_alloc(len * sizeof(WCHAR));
256         if (!value)
257         {
258             r = ERROR_OUTOFMEMORY;
259             goto done;
260         }
261
262         hr = IWineMsiRemotePackage_GetTargetPath( remote_package, folder,
263                                                   (BSTR *)value, &len);
264         if (FAILED(hr))
265             goto done;
266
267         r = msi_strcpy_to_awstring( value, szPathBuf, pcchPathBuf );
268
269 done:
270         IWineMsiRemotePackage_Release( remote_package );
271         SysFreeString( folder );
272         msi_free( value );
273
274         if (FAILED(hr))
275         {
276             if (HRESULT_FACILITY(hr) == FACILITY_WIN32)
277                 return HRESULT_CODE(hr);
278
279             return ERROR_FUNCTION_FAILED;
280         }
281
282         return r;
283     }
284
285     path = resolve_folder( package, szFolder, FALSE, FALSE, TRUE, NULL );
286     msiobj_release( &package->hdr );
287
288     if (!path)
289         return ERROR_DIRECTORY;
290
291     r = msi_strcpy_to_awstring( path, szPathBuf, pcchPathBuf );
292     msi_free( path );
293     return r;
294 }
295
296 /***********************************************************************
297  * MsiGetTargetPathA        (MSI.@)
298  */
299 UINT WINAPI MsiGetTargetPathA( MSIHANDLE hInstall, LPCSTR szFolder, 
300                                LPSTR szPathBuf, LPDWORD pcchPathBuf )
301 {
302     LPWSTR szwFolder;
303     awstring path;
304     UINT r;
305
306     TRACE("%s %p %p\n", debugstr_a(szFolder), szPathBuf, pcchPathBuf);
307
308     szwFolder = strdupAtoW(szFolder);
309     if (szFolder && !szwFolder)
310         return ERROR_FUNCTION_FAILED; 
311
312     path.unicode = FALSE;
313     path.str.a = szPathBuf;
314
315     r = MSI_GetTargetPath( hInstall, szwFolder, &path, pcchPathBuf );
316
317     msi_free( szwFolder );
318
319     return r;
320 }
321
322 /***********************************************************************
323  * MsiGetTargetPathW        (MSI.@)
324  */
325 UINT WINAPI MsiGetTargetPathW( MSIHANDLE hInstall, LPCWSTR szFolder,
326                                LPWSTR szPathBuf, LPDWORD pcchPathBuf )
327 {
328     awstring path;
329
330     TRACE("%s %p %p\n", debugstr_w(szFolder), szPathBuf, pcchPathBuf);
331
332     path.unicode = TRUE;
333     path.str.w = szPathBuf;
334
335     return MSI_GetTargetPath( hInstall, szFolder, &path, pcchPathBuf );
336 }
337
338 /***********************************************************************
339  * MsiGetSourcePath   (internal)
340  */
341 static UINT MSI_GetSourcePath( MSIHANDLE hInstall, LPCWSTR szFolder,
342                                awstring *szPathBuf, LPDWORD pcchPathBuf )
343 {
344     MSIPACKAGE *package;
345     LPWSTR path;
346     UINT r = ERROR_FUNCTION_FAILED;
347
348     TRACE("%s %p %p\n", debugstr_w(szFolder), szPathBuf, pcchPathBuf );
349
350     if (!szFolder)
351         return ERROR_INVALID_PARAMETER;
352
353     package = msihandle2msiinfo( hInstall, MSIHANDLETYPE_PACKAGE );
354     if (!package)
355     {
356         HRESULT hr;
357         IWineMsiRemotePackage *remote_package;
358         LPWSTR value = NULL;
359         BSTR folder;
360         DWORD len;
361
362         remote_package = (IWineMsiRemotePackage *)msi_get_remote( hInstall );
363         if (!remote_package)
364             return ERROR_INVALID_HANDLE;
365
366         folder = SysAllocString( szFolder );
367         if (!folder)
368         {
369             IWineMsiRemotePackage_Release( remote_package );
370             return ERROR_OUTOFMEMORY;
371         }
372
373         len = 0;
374         hr = IWineMsiRemotePackage_GetSourcePath( remote_package, folder,
375                                                   NULL, &len );
376         if (FAILED(hr))
377             goto done;
378
379         len++;
380         value = msi_alloc(len * sizeof(WCHAR));
381         if (!value)
382         {
383             r = ERROR_OUTOFMEMORY;
384             goto done;
385         }
386
387         hr = IWineMsiRemotePackage_GetSourcePath( remote_package, folder,
388                                                   (BSTR *)value, &len);
389         if (FAILED(hr))
390             goto done;
391
392         r = msi_strcpy_to_awstring( value, szPathBuf, pcchPathBuf );
393
394 done:
395         IWineMsiRemotePackage_Release( remote_package );
396         SysFreeString( folder );
397         msi_free( value );
398
399         if (FAILED(hr))
400         {
401             if (HRESULT_FACILITY(hr) == FACILITY_WIN32)
402                 return HRESULT_CODE(hr);
403
404             return ERROR_FUNCTION_FAILED;
405         }
406
407         return r;
408     }
409
410     if (szPathBuf->str.w && !pcchPathBuf )
411     {
412         msiobj_release( &package->hdr );
413         return ERROR_INVALID_PARAMETER;
414     }
415
416     path = resolve_folder(package, szFolder, TRUE, FALSE, TRUE, NULL);
417     msiobj_release( &package->hdr );
418
419     TRACE("path = %s\n",debugstr_w(path));
420     if (!path)
421         return ERROR_DIRECTORY;
422
423     r = msi_strcpy_to_awstring( path, szPathBuf, pcchPathBuf );
424     msi_free( path );
425     return r;
426 }
427
428 /***********************************************************************
429  * MsiGetSourcePathA     (MSI.@)
430  */
431 UINT WINAPI MsiGetSourcePathA( MSIHANDLE hInstall, LPCSTR szFolder, 
432                                LPSTR szPathBuf, LPDWORD pcchPathBuf )
433 {
434     LPWSTR folder;
435     awstring str;
436     UINT r;
437
438     TRACE("%s %p %p\n", szFolder, debugstr_a(szPathBuf), pcchPathBuf);
439
440     str.unicode = FALSE;
441     str.str.a = szPathBuf;
442
443     folder = strdupAtoW( szFolder );
444     r = MSI_GetSourcePath( hInstall, folder, &str, pcchPathBuf );
445     msi_free( folder );
446
447     return r;
448 }
449
450 /***********************************************************************
451  * MsiGetSourcePathW     (MSI.@)
452  */
453 UINT WINAPI MsiGetSourcePathW( MSIHANDLE hInstall, LPCWSTR szFolder,
454                                LPWSTR szPathBuf, LPDWORD pcchPathBuf )
455 {
456     awstring str;
457
458     TRACE("%s %p %p\n", debugstr_w(szFolder), szPathBuf, pcchPathBuf );
459
460     str.unicode = TRUE;
461     str.str.w = szPathBuf;
462
463     return MSI_GetSourcePath( hInstall, szFolder, &str, pcchPathBuf );
464 }
465
466 /***********************************************************************
467  * MsiSetTargetPathA  (MSI.@)
468  */
469 UINT WINAPI MsiSetTargetPathA( MSIHANDLE hInstall, LPCSTR szFolder,
470                                LPCSTR szFolderPath )
471 {
472     LPWSTR szwFolder = NULL, szwFolderPath = NULL;
473     UINT rc = ERROR_OUTOFMEMORY;
474
475     if ( !szFolder || !szFolderPath )
476         return ERROR_INVALID_PARAMETER;
477
478     szwFolder = strdupAtoW(szFolder);
479     szwFolderPath = strdupAtoW(szFolderPath);
480     if (!szwFolder || !szwFolderPath)
481         goto end;
482
483     rc = MsiSetTargetPathW( hInstall, szwFolder, szwFolderPath );
484
485 end:
486     msi_free(szwFolder);
487     msi_free(szwFolderPath);
488
489     return rc;
490 }
491
492 /*
493  * Ok my original interpretation of this was wrong. And it looks like msdn has
494  * changed a bit also. The given folder path does not have to actually already
495  * exist, it just cannot be read only and must be a legal folder path.
496  */
497 UINT MSI_SetTargetPathW(MSIPACKAGE *package, LPCWSTR szFolder, 
498                              LPCWSTR szFolderPath)
499 {
500     DWORD attrib;
501     LPWSTR path = NULL;
502     LPWSTR path2 = NULL;
503     MSIFOLDER *folder;
504     MSIFILE *file;
505
506     TRACE("%p %s %s\n",package, debugstr_w(szFolder),debugstr_w(szFolderPath));
507
508     attrib = GetFileAttributesW(szFolderPath);
509     /* native MSI tests writeability by making temporary files at each drive */
510     if ( attrib != INVALID_FILE_ATTRIBUTES &&
511           (attrib & FILE_ATTRIBUTE_OFFLINE ||
512            attrib & FILE_ATTRIBUTE_READONLY))
513         return ERROR_FUNCTION_FAILED;
514
515     path = resolve_folder(package,szFolder,FALSE,FALSE,FALSE,&folder);
516     if (!path)
517         return ERROR_DIRECTORY;
518
519     msi_free(folder->Property);
520     folder->Property = build_directory_name(2, szFolderPath, NULL);
521
522     if (lstrcmpiW(path, folder->Property) == 0)
523     {
524         /*
525          *  Resolved Target has not really changed, so just 
526          *  set this folder and do not recalculate everything.
527          */
528         msi_free(folder->ResolvedTarget);
529         folder->ResolvedTarget = NULL;
530         path2 = resolve_folder(package,szFolder,FALSE,TRUE,FALSE,NULL);
531         msi_free(path2);
532     }
533     else
534     {
535         MSIFOLDER *f;
536
537         LIST_FOR_EACH_ENTRY( f, &package->folders, MSIFOLDER, entry )
538         {
539             msi_free(f->ResolvedTarget);
540             f->ResolvedTarget=NULL;
541         }
542
543         LIST_FOR_EACH_ENTRY( f, &package->folders, MSIFOLDER, entry )
544         {
545             path2 = resolve_folder(package, f->Directory, FALSE, TRUE, FALSE, NULL);
546             msi_free(path2);
547         }
548
549         LIST_FOR_EACH_ENTRY( file, &package->files, MSIFILE, entry )
550         {
551             MSICOMPONENT *comp = file->Component;
552             LPWSTR p;
553
554             if (!comp)
555                 continue;
556
557             p = resolve_folder(package, comp->Directory, FALSE, FALSE, FALSE, NULL);
558             msi_free(file->TargetPath);
559
560             file->TargetPath = build_directory_name(2, p, file->FileName);
561             msi_free(p);
562         }
563     }
564     msi_free(path);
565
566     return ERROR_SUCCESS;
567 }
568
569 /***********************************************************************
570  * MsiSetTargetPathW  (MSI.@)
571  */
572 UINT WINAPI MsiSetTargetPathW(MSIHANDLE hInstall, LPCWSTR szFolder, 
573                              LPCWSTR szFolderPath)
574 {
575     MSIPACKAGE *package;
576     UINT ret;
577
578     TRACE("%s %s\n",debugstr_w(szFolder),debugstr_w(szFolderPath));
579
580     if ( !szFolder || !szFolderPath )
581         return ERROR_INVALID_PARAMETER;
582
583     package = msihandle2msiinfo(hInstall, MSIHANDLETYPE_PACKAGE);
584     if (!package)
585     {
586         HRESULT hr;
587         BSTR folder, path;
588         IWineMsiRemotePackage *remote_package;
589
590         remote_package = (IWineMsiRemotePackage *)msi_get_remote( hInstall );
591         if (!remote_package)
592             return ERROR_INVALID_HANDLE;
593
594         folder = SysAllocString( szFolder );
595         path = SysAllocString( szFolderPath );
596         if (!folder || !path)
597         {
598             SysFreeString(folder);
599             SysFreeString(path);
600             IWineMsiRemotePackage_Release( remote_package );
601             return ERROR_OUTOFMEMORY;
602         }
603
604         hr = IWineMsiRemotePackage_SetTargetPath( remote_package, folder, path );
605
606         SysFreeString(folder);
607         SysFreeString(path);
608         IWineMsiRemotePackage_Release( remote_package );
609
610         if (FAILED(hr))
611         {
612             if (HRESULT_FACILITY(hr) == FACILITY_WIN32)
613                 return HRESULT_CODE(hr);
614
615             return ERROR_FUNCTION_FAILED;
616         }
617
618         return ERROR_SUCCESS;
619     }
620
621     ret = MSI_SetTargetPathW( package, szFolder, szFolderPath );
622     msiobj_release( &package->hdr );
623     return ret;
624 }
625
626 /***********************************************************************
627  *           MsiGetMode    (MSI.@)
628  *
629  * Returns an internal installer state (if it is running in a mode iRunMode)
630  *
631  * PARAMS
632  *   hInstall    [I]  Handle to the installation
633  *   hRunMode    [I]  Checking run mode
634  *        MSIRUNMODE_ADMIN             Administrative mode
635  *        MSIRUNMODE_ADVERTISE         Advertisement mode
636  *        MSIRUNMODE_MAINTENANCE       Maintenance mode
637  *        MSIRUNMODE_ROLLBACKENABLED   Rollback is enabled
638  *        MSIRUNMODE_LOGENABLED        Log file is writing
639  *        MSIRUNMODE_OPERATIONS        Operations in progress??
640  *        MSIRUNMODE_REBOOTATEND       We need to reboot after installation completed
641  *        MSIRUNMODE_REBOOTNOW         We need to reboot to continue the installation
642  *        MSIRUNMODE_CABINET           Files from cabinet are installed
643  *        MSIRUNMODE_SOURCESHORTNAMES  Long names in source files is suppressed
644  *        MSIRUNMODE_TARGETSHORTNAMES  Long names in destination files is suppressed
645  *        MSIRUNMODE_RESERVED11        Reserved
646  *        MSIRUNMODE_WINDOWS9X         Running under Windows95/98
647  *        MSIRUNMODE_ZAWENABLED        Demand installation is supported
648  *        MSIRUNMODE_RESERVED14        Reserved
649  *        MSIRUNMODE_RESERVED15        Reserved
650  *        MSIRUNMODE_SCHEDULED         called from install script
651  *        MSIRUNMODE_ROLLBACK          called from rollback script
652  *        MSIRUNMODE_COMMIT            called from commit script
653  *
654  * RETURNS
655  *    In the state: TRUE
656  *    Not in the state: FALSE
657  *
658  */
659 BOOL WINAPI MsiGetMode(MSIHANDLE hInstall, MSIRUNMODE iRunMode)
660 {
661     MSIPACKAGE *package;
662     BOOL r = FALSE;
663
664     TRACE("%d %d\n", hInstall, iRunMode);
665
666     package = msihandle2msiinfo(hInstall, MSIHANDLETYPE_PACKAGE);
667     if (!package)
668     {
669         BOOL ret;
670         HRESULT hr;
671         IWineMsiRemotePackage *remote_package;
672
673         remote_package = (IWineMsiRemotePackage *)msi_get_remote(hInstall);
674         if (!remote_package)
675             return FALSE;
676
677         hr = IWineMsiRemotePackage_GetMode(remote_package, iRunMode, &ret);
678         IWineMsiRemotePackage_Release(remote_package);
679
680         if (hr == S_OK)
681             return ret;
682
683         return FALSE;
684     }
685
686     switch (iRunMode)
687     {
688     case MSIRUNMODE_WINDOWS9X:
689         if (GetVersion() & 0x80000000)
690             r = TRUE;
691         break;
692
693     case MSIRUNMODE_OPERATIONS:
694     case MSIRUNMODE_RESERVED11:
695     case MSIRUNMODE_RESERVED14:
696     case MSIRUNMODE_RESERVED15:
697         break;
698
699     case MSIRUNMODE_SCHEDULED:
700         r = package->scheduled_action_running;
701         break;
702
703     case MSIRUNMODE_ROLLBACK:
704         r = package->rollback_action_running;
705         break;
706
707     case MSIRUNMODE_COMMIT:
708         r = package->commit_action_running;
709         break;
710
711     case MSIRUNMODE_MAINTENANCE:
712         r = msi_get_property_int( package->db, szInstalled, 0 ) != 0;
713         break;
714
715     case MSIRUNMODE_REBOOTATEND:
716         r = package->need_reboot;
717         break;
718
719     default:
720         FIXME("unimplemented run mode: %d\n", iRunMode);
721         r = TRUE;
722     }
723
724     msiobj_release( &package->hdr );
725     return r;
726 }
727
728 /***********************************************************************
729  *           MsiSetMode    (MSI.@)
730  */
731 UINT WINAPI MsiSetMode(MSIHANDLE hInstall, MSIRUNMODE iRunMode, BOOL fState)
732 {
733     MSIPACKAGE *package;
734     UINT r;
735
736     TRACE("%d %d %d\n", hInstall, iRunMode, fState);
737
738     package = msihandle2msiinfo( hInstall, MSIHANDLETYPE_PACKAGE );
739     if (!package)
740     {
741         HRESULT hr;
742         IWineMsiRemotePackage *remote_package;
743
744         remote_package = (IWineMsiRemotePackage *)msi_get_remote( hInstall );
745         if (!remote_package)
746             return FALSE;
747
748         hr = IWineMsiRemotePackage_SetMode( remote_package, iRunMode, fState );
749         IWineMsiRemotePackage_Release( remote_package );
750
751         if (FAILED(hr))
752         {
753             if (HRESULT_FACILITY(hr) == FACILITY_WIN32)
754                 return HRESULT_CODE(hr);
755
756             return ERROR_FUNCTION_FAILED;
757         }
758
759         return ERROR_SUCCESS;
760     }
761
762     switch (iRunMode)
763     {
764     case MSIRUNMODE_REBOOTATEND:
765         package->need_reboot = 1;
766         r = ERROR_SUCCESS;
767         break;
768
769     case MSIRUNMODE_REBOOTNOW:
770         FIXME("unimplemented run mode: %d\n", iRunMode);
771         r = ERROR_FUNCTION_FAILED;
772         break;
773
774     default:
775         r = ERROR_ACCESS_DENIED;
776     }
777
778     msiobj_release( &package->hdr );
779     return r;
780 }
781
782 /***********************************************************************
783  * MsiSetFeatureStateA (MSI.@)
784  *
785  * According to the docs, when this is called it immediately recalculates
786  * all the component states as well
787  */
788 UINT WINAPI MsiSetFeatureStateA(MSIHANDLE hInstall, LPCSTR szFeature,
789                                 INSTALLSTATE iState)
790 {
791     LPWSTR szwFeature = NULL;
792     UINT rc;
793
794     szwFeature = strdupAtoW(szFeature);
795
796     if (!szwFeature)
797         return ERROR_FUNCTION_FAILED;
798    
799     rc = MsiSetFeatureStateW(hInstall,szwFeature, iState); 
800
801     msi_free(szwFeature);
802
803     return rc;
804 }
805
806
807
808 UINT WINAPI MSI_SetFeatureStateW(MSIPACKAGE* package, LPCWSTR szFeature,
809                                 INSTALLSTATE iState)
810 {
811     UINT rc = ERROR_SUCCESS;
812     MSIFEATURE *feature, *child;
813
814     TRACE("%s %i\n", debugstr_w(szFeature), iState);
815
816     feature = get_loaded_feature(package,szFeature);
817     if (!feature)
818         return ERROR_UNKNOWN_FEATURE;
819
820     if (iState == INSTALLSTATE_ADVERTISED && 
821         feature->Attributes & msidbFeatureAttributesDisallowAdvertise)
822         return ERROR_FUNCTION_FAILED;
823
824     msi_feature_set_state(package, feature, iState);
825
826     ACTION_UpdateComponentStates(package,szFeature);
827
828     /* update all the features that are children of this feature */
829     LIST_FOR_EACH_ENTRY( child, &package->features, MSIFEATURE, entry )
830     {
831         if (lstrcmpW(szFeature, child->Feature_Parent) == 0)
832             MSI_SetFeatureStateW(package, child->Feature, iState);
833     }
834     
835     return rc;
836 }
837
838 /***********************************************************************
839  * MsiSetFeatureStateW (MSI.@)
840  */
841 UINT WINAPI MsiSetFeatureStateW(MSIHANDLE hInstall, LPCWSTR szFeature,
842                                 INSTALLSTATE iState)
843 {
844     MSIPACKAGE* package;
845     UINT rc = ERROR_SUCCESS;
846
847     TRACE("%s %i\n",debugstr_w(szFeature), iState);
848
849     package = msihandle2msiinfo(hInstall, MSIHANDLETYPE_PACKAGE);
850     if (!package)
851     {
852         HRESULT hr;
853         BSTR feature;
854         IWineMsiRemotePackage *remote_package;
855
856         remote_package = (IWineMsiRemotePackage *)msi_get_remote(hInstall);
857         if (!remote_package)
858             return ERROR_INVALID_HANDLE;
859
860         feature = SysAllocString(szFeature);
861         if (!feature)
862         {
863             IWineMsiRemotePackage_Release(remote_package);
864             return ERROR_OUTOFMEMORY;
865         }
866
867         hr = IWineMsiRemotePackage_SetFeatureState(remote_package, feature, iState);
868
869         SysFreeString(feature);
870         IWineMsiRemotePackage_Release(remote_package);
871
872         if (FAILED(hr))
873         {
874             if (HRESULT_FACILITY(hr) == FACILITY_WIN32)
875                 return HRESULT_CODE(hr);
876
877             return ERROR_FUNCTION_FAILED;
878         }
879
880         return ERROR_SUCCESS;
881     }
882
883     rc = MSI_SetFeatureStateW(package,szFeature,iState);
884
885     msiobj_release( &package->hdr );
886     return rc;
887 }
888
889 /***********************************************************************
890 * MsiGetFeatureStateA   (MSI.@)
891 */
892 UINT WINAPI MsiGetFeatureStateA(MSIHANDLE hInstall, LPCSTR szFeature,
893                   INSTALLSTATE *piInstalled, INSTALLSTATE *piAction)
894 {
895     LPWSTR szwFeature = NULL;
896     UINT rc;
897     
898     szwFeature = strdupAtoW(szFeature);
899
900     rc = MsiGetFeatureStateW(hInstall,szwFeature,piInstalled, piAction);
901
902     msi_free( szwFeature);
903
904     return rc;
905 }
906
907 UINT MSI_GetFeatureStateW(MSIPACKAGE *package, LPCWSTR szFeature,
908                   INSTALLSTATE *piInstalled, INSTALLSTATE *piAction)
909 {
910     MSIFEATURE *feature;
911
912     feature = get_loaded_feature(package,szFeature);
913     if (!feature)
914         return ERROR_UNKNOWN_FEATURE;
915
916     if (piInstalled)
917         *piInstalled = feature->Installed;
918
919     if (piAction)
920         *piAction = feature->Action;
921
922     TRACE("returning %i %i\n", feature->Installed, feature->Action);
923
924     return ERROR_SUCCESS;
925 }
926
927 /***********************************************************************
928 * MsiGetFeatureStateW   (MSI.@)
929 */
930 UINT WINAPI MsiGetFeatureStateW(MSIHANDLE hInstall, LPCWSTR szFeature,
931                   INSTALLSTATE *piInstalled, INSTALLSTATE *piAction)
932 {
933     MSIPACKAGE* package;
934     UINT ret;
935
936     TRACE("%d %s %p %p\n", hInstall, debugstr_w(szFeature), piInstalled, piAction);
937
938     package = msihandle2msiinfo(hInstall, MSIHANDLETYPE_PACKAGE);
939     if (!package)
940     {
941         HRESULT hr;
942         BSTR feature;
943         IWineMsiRemotePackage *remote_package;
944
945         remote_package = (IWineMsiRemotePackage *)msi_get_remote(hInstall);
946         if (!remote_package)
947             return ERROR_INVALID_HANDLE;
948
949         feature = SysAllocString(szFeature);
950         if (!feature)
951         {
952             IWineMsiRemotePackage_Release(remote_package);
953             return ERROR_OUTOFMEMORY;
954         }
955
956         hr = IWineMsiRemotePackage_GetFeatureState(remote_package, feature,
957                                                    piInstalled, piAction);
958
959         SysFreeString(feature);
960         IWineMsiRemotePackage_Release(remote_package);
961
962         if (FAILED(hr))
963         {
964             if (HRESULT_FACILITY(hr) == FACILITY_WIN32)
965                 return HRESULT_CODE(hr);
966
967             return ERROR_FUNCTION_FAILED;
968         }
969
970         return ERROR_SUCCESS;
971     }
972
973     ret = MSI_GetFeatureStateW(package, szFeature, piInstalled, piAction);
974     msiobj_release( &package->hdr );
975     return ret;
976 }
977
978 /***********************************************************************
979 * MsiGetFeatureCostA   (MSI.@)
980 */
981 UINT WINAPI MsiGetFeatureCostA(MSIHANDLE hInstall, LPCSTR szFeature,
982                   MSICOSTTREE iCostTree, INSTALLSTATE iState, LPINT piCost)
983 {
984     LPWSTR szwFeature = NULL;
985     UINT rc;
986
987     szwFeature = strdupAtoW(szFeature);
988
989     rc = MsiGetFeatureCostW(hInstall, szwFeature, iCostTree, iState, piCost);
990
991     msi_free(szwFeature);
992
993     return rc;
994 }
995
996 static INT feature_cost( MSIFEATURE *feature )
997 {
998     INT cost = 0;
999     MSICOMPONENT *comp;
1000
1001     LIST_FOR_EACH_ENTRY( comp, &feature->Components, MSICOMPONENT, entry )
1002     {
1003         cost += comp->Cost;
1004     }
1005     return cost;
1006 }
1007
1008 UINT MSI_GetFeatureCost( MSIPACKAGE *package, MSIFEATURE *feature, MSICOSTTREE tree,
1009                          INSTALLSTATE state, LPINT cost )
1010 {
1011     TRACE("%s, %u, %d, %p\n", debugstr_w(feature->Feature), tree, state, cost);
1012
1013     *cost = 0;
1014     switch (tree)
1015     {
1016     case MSICOSTTREE_CHILDREN:
1017     {
1018         MSIFEATURE *child;
1019
1020         LIST_FOR_EACH_ENTRY( child, &feature->Children, MSIFEATURE, entry )
1021         {
1022             if (child->ActionRequest == state)
1023                 *cost += feature_cost( child );
1024         }
1025         break;
1026     }
1027     case MSICOSTTREE_PARENTS:
1028     {
1029         const WCHAR *feature_parent = feature->Feature_Parent;
1030         for (;;)
1031         {
1032             MSIFEATURE *parent = get_loaded_feature( package, feature_parent );
1033             if (!parent)
1034                 break;
1035
1036             if (parent->ActionRequest == state)
1037                 *cost += feature_cost( parent );
1038
1039             feature_parent = parent->Feature_Parent;
1040         }
1041         break;
1042     }
1043     case MSICOSTTREE_SELFONLY:
1044         if (feature->ActionRequest == state)
1045             *cost = feature_cost( feature );
1046         break;
1047
1048     default:
1049         WARN("unhandled cost tree %u\n", tree);
1050         break;
1051     }
1052
1053     *cost /= 512;
1054     return ERROR_SUCCESS;
1055 }
1056
1057 /***********************************************************************
1058 * MsiGetFeatureCostW   (MSI.@)
1059 */
1060 UINT WINAPI MsiGetFeatureCostW(MSIHANDLE hInstall, LPCWSTR szFeature,
1061                   MSICOSTTREE iCostTree, INSTALLSTATE iState, LPINT piCost)
1062 {
1063     MSIPACKAGE *package;
1064     MSIFEATURE *feature;
1065     UINT ret;
1066
1067     TRACE("(%d %s %i %i %p)\n", hInstall, debugstr_w(szFeature),
1068           iCostTree, iState, piCost);
1069
1070     package = msihandle2msiinfo(hInstall, MSIHANDLETYPE_PACKAGE);
1071     if (!package)
1072     {
1073         HRESULT hr;
1074         BSTR feature;
1075         IWineMsiRemotePackage *remote_package;
1076
1077         remote_package = (IWineMsiRemotePackage *)msi_get_remote(hInstall);
1078         if (!remote_package)
1079             return ERROR_INVALID_HANDLE;
1080
1081         feature = SysAllocString(szFeature);
1082         if (!feature)
1083         {
1084             IWineMsiRemotePackage_Release(remote_package);
1085             return ERROR_OUTOFMEMORY;
1086         }
1087
1088         hr = IWineMsiRemotePackage_GetFeatureCost(remote_package, feature,
1089                                                   iCostTree, iState, piCost);
1090
1091         SysFreeString(feature);
1092         IWineMsiRemotePackage_Release(remote_package);
1093
1094         if (FAILED(hr))
1095         {
1096             if (HRESULT_FACILITY(hr) == FACILITY_WIN32)
1097                 return HRESULT_CODE(hr);
1098
1099             return ERROR_FUNCTION_FAILED;
1100         }
1101
1102         return ERROR_SUCCESS;
1103     }
1104
1105     feature = get_loaded_feature(package, szFeature);
1106
1107     if (feature)
1108         ret = MSI_GetFeatureCost(package, feature, iCostTree, iState, piCost);
1109     else
1110         ret = ERROR_UNKNOWN_FEATURE;
1111
1112     msiobj_release( &package->hdr );
1113     return ret;
1114 }
1115
1116 /***********************************************************************
1117  * MsiSetComponentStateA (MSI.@)
1118  */
1119 UINT WINAPI MsiSetComponentStateA(MSIHANDLE hInstall, LPCSTR szComponent,
1120                                   INSTALLSTATE iState)
1121 {
1122     UINT rc;
1123     LPWSTR szwComponent = strdupAtoW(szComponent);
1124
1125     rc = MsiSetComponentStateW(hInstall, szwComponent, iState);
1126
1127     msi_free(szwComponent);
1128
1129     return rc;
1130 }
1131
1132 /***********************************************************************
1133  * MsiGetComponentStateA (MSI.@)
1134  */
1135 UINT WINAPI MsiGetComponentStateA(MSIHANDLE hInstall, LPCSTR szComponent,
1136                   INSTALLSTATE *piInstalled, INSTALLSTATE *piAction)
1137 {
1138     LPWSTR szwComponent= NULL;
1139     UINT rc;
1140     
1141     szwComponent= strdupAtoW(szComponent);
1142
1143     rc = MsiGetComponentStateW(hInstall,szwComponent,piInstalled, piAction);
1144
1145     msi_free( szwComponent);
1146
1147     return rc;
1148 }
1149
1150 static UINT MSI_SetComponentStateW(MSIPACKAGE *package, LPCWSTR szComponent,
1151                                    INSTALLSTATE iState)
1152 {
1153     MSICOMPONENT *comp;
1154
1155     TRACE("%p %s %d\n", package, debugstr_w(szComponent), iState);
1156
1157     comp = get_loaded_component(package, szComponent);
1158     if (!comp)
1159         return ERROR_UNKNOWN_COMPONENT;
1160
1161     comp->Installed = iState;
1162
1163     return ERROR_SUCCESS;
1164 }
1165
1166 UINT MSI_GetComponentStateW(MSIPACKAGE *package, LPCWSTR szComponent,
1167                   INSTALLSTATE *piInstalled, INSTALLSTATE *piAction)
1168 {
1169     MSICOMPONENT *comp;
1170
1171     TRACE("%p %s %p %p\n", package, debugstr_w(szComponent),
1172            piInstalled, piAction);
1173
1174     comp = get_loaded_component(package,szComponent);
1175     if (!comp)
1176         return ERROR_UNKNOWN_COMPONENT;
1177
1178     if (piInstalled)
1179         *piInstalled = comp->Installed;
1180
1181     if (piAction)
1182         *piAction = comp->Action;
1183
1184     TRACE("states (%i, %i)\n", comp->Installed, comp->Action );
1185
1186     return ERROR_SUCCESS;
1187 }
1188
1189 /***********************************************************************
1190  * MsiSetComponentStateW (MSI.@)
1191  */
1192 UINT WINAPI MsiSetComponentStateW(MSIHANDLE hInstall, LPCWSTR szComponent,
1193                                   INSTALLSTATE iState)
1194 {
1195     MSIPACKAGE* package;
1196     UINT ret;
1197
1198     package = msihandle2msiinfo(hInstall, MSIHANDLETYPE_PACKAGE);
1199     if (!package)
1200     {
1201         HRESULT hr;
1202         BSTR component;
1203         IWineMsiRemotePackage *remote_package;
1204
1205         remote_package = (IWineMsiRemotePackage *)msi_get_remote(hInstall);
1206         if (!remote_package)
1207             return ERROR_INVALID_HANDLE;
1208
1209         component = SysAllocString(szComponent);
1210         if (!component)
1211         {
1212             IWineMsiRemotePackage_Release(remote_package);
1213             return ERROR_OUTOFMEMORY;
1214         }
1215
1216         hr = IWineMsiRemotePackage_SetComponentState(remote_package, component, iState);
1217
1218         SysFreeString(component);
1219         IWineMsiRemotePackage_Release(remote_package);
1220
1221         if (FAILED(hr))
1222         {
1223             if (HRESULT_FACILITY(hr) == FACILITY_WIN32)
1224                 return HRESULT_CODE(hr);
1225
1226             return ERROR_FUNCTION_FAILED;
1227         }
1228
1229         return ERROR_SUCCESS;
1230     }
1231
1232     ret = MSI_SetComponentStateW(package, szComponent, iState);
1233     msiobj_release(&package->hdr);
1234     return ret;
1235 }
1236
1237 /***********************************************************************
1238  * MsiGetComponentStateW (MSI.@)
1239  */
1240 UINT WINAPI MsiGetComponentStateW(MSIHANDLE hInstall, LPCWSTR szComponent,
1241                   INSTALLSTATE *piInstalled, INSTALLSTATE *piAction)
1242 {
1243     MSIPACKAGE* package;
1244     UINT ret;
1245
1246     TRACE("%d %s %p %p\n", hInstall, debugstr_w(szComponent),
1247            piInstalled, piAction);
1248
1249     package = msihandle2msiinfo(hInstall, MSIHANDLETYPE_PACKAGE);
1250     if (!package)
1251     {
1252         HRESULT hr;
1253         BSTR component;
1254         IWineMsiRemotePackage *remote_package;
1255
1256         remote_package = (IWineMsiRemotePackage *)msi_get_remote(hInstall);
1257         if (!remote_package)
1258             return ERROR_INVALID_HANDLE;
1259
1260         component = SysAllocString(szComponent);
1261         if (!component)
1262         {
1263             IWineMsiRemotePackage_Release(remote_package);
1264             return ERROR_OUTOFMEMORY;
1265         }
1266
1267         hr = IWineMsiRemotePackage_GetComponentState(remote_package, component,
1268                                                      piInstalled, piAction);
1269
1270         SysFreeString(component);
1271         IWineMsiRemotePackage_Release(remote_package);
1272
1273         if (FAILED(hr))
1274         {
1275             if (HRESULT_FACILITY(hr) == FACILITY_WIN32)
1276                 return HRESULT_CODE(hr);
1277
1278             return ERROR_FUNCTION_FAILED;
1279         }
1280
1281         return ERROR_SUCCESS;
1282     }
1283
1284     ret = MSI_GetComponentStateW( package, szComponent, piInstalled, piAction);
1285     msiobj_release( &package->hdr );
1286     return ret;
1287 }
1288
1289 /***********************************************************************
1290  * MsiGetLanguage (MSI.@)
1291  */
1292 LANGID WINAPI MsiGetLanguage(MSIHANDLE hInstall)
1293 {
1294     MSIPACKAGE* package;
1295     LANGID langid;
1296     static const WCHAR szProductLanguage[] =
1297         {'P','r','o','d','u','c','t','L','a','n','g','u','a','g','e',0};
1298     
1299     package = msihandle2msiinfo(hInstall, MSIHANDLETYPE_PACKAGE);
1300     if (!package)
1301     {
1302         HRESULT hr;
1303         LANGID lang;
1304         IWineMsiRemotePackage *remote_package;
1305
1306         remote_package = (IWineMsiRemotePackage *)msi_get_remote(hInstall);
1307         if (!remote_package)
1308             return ERROR_INVALID_HANDLE;
1309
1310         hr = IWineMsiRemotePackage_GetLanguage(remote_package, &lang);
1311
1312         if (SUCCEEDED(hr))
1313             return lang;
1314
1315         return 0;
1316     }
1317
1318     langid = msi_get_property_int( package->db, szProductLanguage, 0 );
1319     msiobj_release( &package->hdr );
1320     return langid;
1321 }
1322
1323 UINT MSI_SetInstallLevel( MSIPACKAGE *package, int iInstallLevel )
1324 {
1325     static const WCHAR szInstallLevel[] = {
1326         'I','N','S','T','A','L','L','L','E','V','E','L',0 };
1327     static const WCHAR fmt[] = { '%','d',0 };
1328     WCHAR level[6];
1329     UINT r;
1330
1331     TRACE("%p %i\n", package, iInstallLevel);
1332
1333     if (iInstallLevel > 32767)
1334         return ERROR_INVALID_PARAMETER;
1335
1336     if (iInstallLevel < 1)
1337         return MSI_SetFeatureStates( package );
1338
1339     sprintfW( level, fmt, iInstallLevel );
1340     r = msi_set_property( package->db, szInstallLevel, level );
1341     if ( r == ERROR_SUCCESS )
1342         r = MSI_SetFeatureStates( package );
1343
1344     return r;
1345 }
1346
1347 /***********************************************************************
1348  * MsiSetInstallLevel (MSI.@)
1349  */
1350 UINT WINAPI MsiSetInstallLevel(MSIHANDLE hInstall, int iInstallLevel)
1351 {
1352     MSIPACKAGE* package;
1353     UINT r;
1354
1355     TRACE("%d %i\n", hInstall, iInstallLevel);
1356
1357     package = msihandle2msiinfo(hInstall, MSIHANDLETYPE_PACKAGE);
1358     if (!package)
1359     {
1360         HRESULT hr;
1361         IWineMsiRemotePackage *remote_package;
1362
1363         remote_package = (IWineMsiRemotePackage *)msi_get_remote(hInstall);
1364         if (!remote_package)
1365             return ERROR_INVALID_HANDLE;
1366
1367         hr = IWineMsiRemotePackage_SetInstallLevel(remote_package, iInstallLevel);
1368
1369         IWineMsiRemotePackage_Release(remote_package);
1370
1371         if (FAILED(hr))
1372         {
1373             if (HRESULT_FACILITY(hr) == FACILITY_WIN32)
1374                 return HRESULT_CODE(hr);
1375
1376             return ERROR_FUNCTION_FAILED;
1377         }
1378
1379         return ERROR_SUCCESS;
1380     }
1381
1382     r = MSI_SetInstallLevel( package, iInstallLevel );
1383
1384     msiobj_release( &package->hdr );
1385
1386     return r;
1387 }
1388
1389 /***********************************************************************
1390  * MsiGetFeatureValidStatesW (MSI.@)
1391  */
1392 UINT WINAPI MsiGetFeatureValidStatesW(MSIHANDLE hInstall, LPCWSTR szFeature,
1393                   LPDWORD pInstallState)
1394 {
1395     if(pInstallState) *pInstallState = 1<<INSTALLSTATE_LOCAL;
1396     FIXME("%d %s %p stub returning %d\n",
1397         hInstall, debugstr_w(szFeature), pInstallState, pInstallState ? *pInstallState : 0);
1398
1399     return ERROR_SUCCESS;
1400 }
1401
1402 /***********************************************************************
1403  * MsiGetFeatureValidStatesA (MSI.@)
1404  */
1405 UINT WINAPI MsiGetFeatureValidStatesA(MSIHANDLE hInstall, LPCSTR szFeature,
1406                   LPDWORD pInstallState)
1407 {
1408     UINT ret;
1409     LPWSTR szwFeature = strdupAtoW(szFeature);
1410
1411     ret = MsiGetFeatureValidStatesW(hInstall, szwFeature, pInstallState);
1412
1413     msi_free(szwFeature);
1414
1415     return ret;
1416 }