msi/tests: Fix test failures on W2K with Windows Installer 3.0.
[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, 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     return r;
725 }
726
727 /***********************************************************************
728  *           MsiSetMode    (MSI.@)
729  */
730 UINT WINAPI MsiSetMode(MSIHANDLE hInstall, MSIRUNMODE iRunMode, BOOL fState)
731 {
732     MSIPACKAGE *package;
733     UINT r;
734
735     TRACE("%d %d %d\n", hInstall, iRunMode, fState);
736
737     package = msihandle2msiinfo( hInstall, MSIHANDLETYPE_PACKAGE );
738     if (!package)
739     {
740         HRESULT hr;
741         IWineMsiRemotePackage *remote_package;
742
743         remote_package = (IWineMsiRemotePackage *)msi_get_remote( hInstall );
744         if (!remote_package)
745             return FALSE;
746
747         hr = IWineMsiRemotePackage_SetMode( remote_package, iRunMode, fState );
748         IWineMsiRemotePackage_Release( remote_package );
749
750         if (FAILED(hr))
751         {
752             if (HRESULT_FACILITY(hr) == FACILITY_WIN32)
753                 return HRESULT_CODE(hr);
754
755             return ERROR_FUNCTION_FAILED;
756         }
757
758         return ERROR_SUCCESS;
759     }
760
761     switch (iRunMode)
762     {
763     case MSIRUNMODE_REBOOTATEND:
764         package->need_reboot = 1;
765         r = ERROR_SUCCESS;
766         break;
767
768     case MSIRUNMODE_REBOOTNOW:
769         FIXME("unimplemented run mode: %d\n", iRunMode);
770         r = ERROR_FUNCTION_FAILED;
771         break;
772
773     default:
774         r = ERROR_ACCESS_DENIED;
775     }
776
777     return r;
778 }
779
780 /***********************************************************************
781  * MsiSetFeatureStateA (MSI.@)
782  *
783  * According to the docs, when this is called it immediately recalculates
784  * all the component states as well
785  */
786 UINT WINAPI MsiSetFeatureStateA(MSIHANDLE hInstall, LPCSTR szFeature,
787                                 INSTALLSTATE iState)
788 {
789     LPWSTR szwFeature = NULL;
790     UINT rc;
791
792     szwFeature = strdupAtoW(szFeature);
793
794     if (!szwFeature)
795         return ERROR_FUNCTION_FAILED;
796    
797     rc = MsiSetFeatureStateW(hInstall,szwFeature, iState); 
798
799     msi_free(szwFeature);
800
801     return rc;
802 }
803
804
805
806 UINT WINAPI MSI_SetFeatureStateW(MSIPACKAGE* package, LPCWSTR szFeature,
807                                 INSTALLSTATE iState)
808 {
809     UINT rc = ERROR_SUCCESS;
810     MSIFEATURE *feature, *child;
811
812     TRACE("%s %i\n", debugstr_w(szFeature), iState);
813
814     feature = get_loaded_feature(package,szFeature);
815     if (!feature)
816         return ERROR_UNKNOWN_FEATURE;
817
818     if (iState == INSTALLSTATE_ADVERTISED && 
819         feature->Attributes & msidbFeatureAttributesDisallowAdvertise)
820         return ERROR_FUNCTION_FAILED;
821
822     msi_feature_set_state(package, feature, iState);
823
824     ACTION_UpdateComponentStates(package,szFeature);
825
826     /* update all the features that are children of this feature */
827     LIST_FOR_EACH_ENTRY( child, &package->features, MSIFEATURE, entry )
828     {
829         if (lstrcmpW(szFeature, child->Feature_Parent) == 0)
830             MSI_SetFeatureStateW(package, child->Feature, iState);
831     }
832     
833     return rc;
834 }
835
836 /***********************************************************************
837  * MsiSetFeatureStateW (MSI.@)
838  */
839 UINT WINAPI MsiSetFeatureStateW(MSIHANDLE hInstall, LPCWSTR szFeature,
840                                 INSTALLSTATE iState)
841 {
842     MSIPACKAGE* package;
843     UINT rc = ERROR_SUCCESS;
844
845     TRACE("%s %i\n",debugstr_w(szFeature), iState);
846
847     package = msihandle2msiinfo(hInstall, MSIHANDLETYPE_PACKAGE);
848     if (!package)
849     {
850         HRESULT hr;
851         BSTR feature;
852         IWineMsiRemotePackage *remote_package;
853
854         remote_package = (IWineMsiRemotePackage *)msi_get_remote(hInstall);
855         if (!remote_package)
856             return ERROR_INVALID_HANDLE;
857
858         feature = SysAllocString(szFeature);
859         if (!feature)
860         {
861             IWineMsiRemotePackage_Release(remote_package);
862             return ERROR_OUTOFMEMORY;
863         }
864
865         hr = IWineMsiRemotePackage_SetFeatureState(remote_package, feature, iState);
866
867         SysFreeString(feature);
868         IWineMsiRemotePackage_Release(remote_package);
869
870         if (FAILED(hr))
871         {
872             if (HRESULT_FACILITY(hr) == FACILITY_WIN32)
873                 return HRESULT_CODE(hr);
874
875             return ERROR_FUNCTION_FAILED;
876         }
877
878         return ERROR_SUCCESS;
879     }
880
881     rc = MSI_SetFeatureStateW(package,szFeature,iState);
882
883     msiobj_release( &package->hdr );
884     return rc;
885 }
886
887 /***********************************************************************
888 * MsiGetFeatureStateA   (MSI.@)
889 */
890 UINT WINAPI MsiGetFeatureStateA(MSIHANDLE hInstall, LPCSTR szFeature,
891                   INSTALLSTATE *piInstalled, INSTALLSTATE *piAction)
892 {
893     LPWSTR szwFeature = NULL;
894     UINT rc;
895     
896     szwFeature = strdupAtoW(szFeature);
897
898     rc = MsiGetFeatureStateW(hInstall,szwFeature,piInstalled, piAction);
899
900     msi_free( szwFeature);
901
902     return rc;
903 }
904
905 UINT MSI_GetFeatureStateW(MSIPACKAGE *package, LPCWSTR szFeature,
906                   INSTALLSTATE *piInstalled, INSTALLSTATE *piAction)
907 {
908     MSIFEATURE *feature;
909
910     feature = get_loaded_feature(package,szFeature);
911     if (!feature)
912         return ERROR_UNKNOWN_FEATURE;
913
914     if (piInstalled)
915         *piInstalled = feature->Installed;
916
917     if (piAction)
918         *piAction = feature->Action;
919
920     TRACE("returning %i %i\n", feature->Installed, feature->Action);
921
922     return ERROR_SUCCESS;
923 }
924
925 /***********************************************************************
926 * MsiGetFeatureStateW   (MSI.@)
927 */
928 UINT WINAPI MsiGetFeatureStateW(MSIHANDLE hInstall, LPCWSTR szFeature,
929                   INSTALLSTATE *piInstalled, INSTALLSTATE *piAction)
930 {
931     MSIPACKAGE* package;
932     UINT ret;
933
934     TRACE("%d %s %p %p\n", hInstall, debugstr_w(szFeature), piInstalled, piAction);
935
936     package = msihandle2msiinfo(hInstall, MSIHANDLETYPE_PACKAGE);
937     if (!package)
938     {
939         HRESULT hr;
940         BSTR feature;
941         IWineMsiRemotePackage *remote_package;
942
943         remote_package = (IWineMsiRemotePackage *)msi_get_remote(hInstall);
944         if (!remote_package)
945             return ERROR_INVALID_HANDLE;
946
947         feature = SysAllocString(szFeature);
948         if (!feature)
949         {
950             IWineMsiRemotePackage_Release(remote_package);
951             return ERROR_OUTOFMEMORY;
952         }
953
954         hr = IWineMsiRemotePackage_GetFeatureState(remote_package, feature,
955                                                    piInstalled, piAction);
956
957         SysFreeString(feature);
958         IWineMsiRemotePackage_Release(remote_package);
959
960         if (FAILED(hr))
961         {
962             if (HRESULT_FACILITY(hr) == FACILITY_WIN32)
963                 return HRESULT_CODE(hr);
964
965             return ERROR_FUNCTION_FAILED;
966         }
967
968         return ERROR_SUCCESS;
969     }
970
971     ret = MSI_GetFeatureStateW(package, szFeature, piInstalled, piAction);
972     msiobj_release( &package->hdr );
973     return ret;
974 }
975
976 /***********************************************************************
977 * MsiGetFeatureCostA   (MSI.@)
978 */
979 UINT WINAPI MsiGetFeatureCostA(MSIHANDLE hInstall, LPCSTR szFeature,
980                   MSICOSTTREE iCostTree, INSTALLSTATE iState, LPINT piCost)
981 {
982     LPWSTR szwFeature = NULL;
983     UINT rc;
984
985     szwFeature = strdupAtoW(szFeature);
986
987     rc = MsiGetFeatureCostW(hInstall, szwFeature, iCostTree, iState, piCost);
988
989     msi_free(szwFeature);
990
991     return rc;
992 }
993
994 UINT MSI_GetFeatureCost(MSIPACKAGE *package, MSIFEATURE *feature,
995                         MSICOSTTREE iCostTree, INSTALLSTATE iState,
996                         LPINT piCost)
997 {
998     FIXME("(%s %i %i %p): not implemented yet\n",
999         debugstr_w(feature->Feature), iCostTree, iState, piCost);
1000     if (piCost) *piCost = 0;
1001     return ERROR_SUCCESS;
1002 }
1003
1004 /***********************************************************************
1005 * MsiGetFeatureCostW   (MSI.@)
1006 */
1007 UINT WINAPI MsiGetFeatureCostW(MSIHANDLE hInstall, LPCWSTR szFeature,
1008                   MSICOSTTREE iCostTree, INSTALLSTATE iState, LPINT piCost)
1009 {
1010     MSIPACKAGE *package;
1011     MSIFEATURE *feature;
1012     UINT ret;
1013
1014     TRACE("(%d %s %i %i %p)\n", hInstall, debugstr_w(szFeature),
1015           iCostTree, iState, piCost);
1016
1017     package = msihandle2msiinfo(hInstall, MSIHANDLETYPE_PACKAGE);
1018     if (!package)
1019     {
1020         HRESULT hr;
1021         BSTR feature;
1022         IWineMsiRemotePackage *remote_package;
1023
1024         remote_package = (IWineMsiRemotePackage *)msi_get_remote(hInstall);
1025         if (!remote_package)
1026             return ERROR_INVALID_HANDLE;
1027
1028         feature = SysAllocString(szFeature);
1029         if (!feature)
1030         {
1031             IWineMsiRemotePackage_Release(remote_package);
1032             return ERROR_OUTOFMEMORY;
1033         }
1034
1035         hr = IWineMsiRemotePackage_GetFeatureCost(remote_package, feature,
1036                                                   iCostTree, iState, piCost);
1037
1038         SysFreeString(feature);
1039         IWineMsiRemotePackage_Release(remote_package);
1040
1041         if (FAILED(hr))
1042         {
1043             if (HRESULT_FACILITY(hr) == FACILITY_WIN32)
1044                 return HRESULT_CODE(hr);
1045
1046             return ERROR_FUNCTION_FAILED;
1047         }
1048
1049         return ERROR_SUCCESS;
1050     }
1051
1052     feature = get_loaded_feature(package, szFeature);
1053
1054     if (feature)
1055         ret = MSI_GetFeatureCost(package, feature, iCostTree, iState, piCost);
1056     else
1057         ret = ERROR_UNKNOWN_FEATURE;
1058
1059     msiobj_release( &package->hdr );
1060     return ret;
1061 }
1062
1063 /***********************************************************************
1064  * MsiSetComponentStateA (MSI.@)
1065  */
1066 UINT WINAPI MsiSetComponentStateA(MSIHANDLE hInstall, LPCSTR szComponent,
1067                                   INSTALLSTATE iState)
1068 {
1069     UINT rc;
1070     LPWSTR szwComponent = strdupAtoW(szComponent);
1071
1072     rc = MsiSetComponentStateW(hInstall, szwComponent, iState);
1073
1074     msi_free(szwComponent);
1075
1076     return rc;
1077 }
1078
1079 /***********************************************************************
1080  * MsiGetComponentStateA (MSI.@)
1081  */
1082 UINT WINAPI MsiGetComponentStateA(MSIHANDLE hInstall, LPCSTR szComponent,
1083                   INSTALLSTATE *piInstalled, INSTALLSTATE *piAction)
1084 {
1085     LPWSTR szwComponent= NULL;
1086     UINT rc;
1087     
1088     szwComponent= strdupAtoW(szComponent);
1089
1090     rc = MsiGetComponentStateW(hInstall,szwComponent,piInstalled, piAction);
1091
1092     msi_free( szwComponent);
1093
1094     return rc;
1095 }
1096
1097 static UINT MSI_SetComponentStateW(MSIPACKAGE *package, LPCWSTR szComponent,
1098                                    INSTALLSTATE iState)
1099 {
1100     MSICOMPONENT *comp;
1101
1102     TRACE("%p %s %d\n", package, debugstr_w(szComponent), iState);
1103
1104     comp = get_loaded_component(package, szComponent);
1105     if (!comp)
1106         return ERROR_UNKNOWN_COMPONENT;
1107
1108     comp->Installed = iState;
1109
1110     return ERROR_SUCCESS;
1111 }
1112
1113 UINT MSI_GetComponentStateW(MSIPACKAGE *package, LPCWSTR szComponent,
1114                   INSTALLSTATE *piInstalled, INSTALLSTATE *piAction)
1115 {
1116     MSICOMPONENT *comp;
1117
1118     TRACE("%p %s %p %p\n", package, debugstr_w(szComponent),
1119            piInstalled, piAction);
1120
1121     comp = get_loaded_component(package,szComponent);
1122     if (!comp)
1123         return ERROR_UNKNOWN_COMPONENT;
1124
1125     if (piInstalled)
1126         *piInstalled = comp->Installed;
1127
1128     if (piAction)
1129         *piAction = comp->Action;
1130
1131     TRACE("states (%i, %i)\n", comp->Installed, comp->Action );
1132
1133     return ERROR_SUCCESS;
1134 }
1135
1136 /***********************************************************************
1137  * MsiSetComponentStateW (MSI.@)
1138  */
1139 UINT WINAPI MsiSetComponentStateW(MSIHANDLE hInstall, LPCWSTR szComponent,
1140                                   INSTALLSTATE iState)
1141 {
1142     MSIPACKAGE* package;
1143     UINT ret;
1144
1145     package = msihandle2msiinfo(hInstall, MSIHANDLETYPE_PACKAGE);
1146     if (!package)
1147     {
1148         HRESULT hr;
1149         BSTR component;
1150         IWineMsiRemotePackage *remote_package;
1151
1152         remote_package = (IWineMsiRemotePackage *)msi_get_remote(hInstall);
1153         if (!remote_package)
1154             return ERROR_INVALID_HANDLE;
1155
1156         component = SysAllocString(szComponent);
1157         if (!component)
1158         {
1159             IWineMsiRemotePackage_Release(remote_package);
1160             return ERROR_OUTOFMEMORY;
1161         }
1162
1163         hr = IWineMsiRemotePackage_SetComponentState(remote_package, component, iState);
1164
1165         SysFreeString(component);
1166         IWineMsiRemotePackage_Release(remote_package);
1167
1168         if (FAILED(hr))
1169         {
1170             if (HRESULT_FACILITY(hr) == FACILITY_WIN32)
1171                 return HRESULT_CODE(hr);
1172
1173             return ERROR_FUNCTION_FAILED;
1174         }
1175
1176         return ERROR_SUCCESS;
1177     }
1178
1179     ret = MSI_SetComponentStateW(package, szComponent, iState);
1180     msiobj_release(&package->hdr);
1181     return ret;
1182 }
1183
1184 /***********************************************************************
1185  * MsiGetComponentStateW (MSI.@)
1186  */
1187 UINT WINAPI MsiGetComponentStateW(MSIHANDLE hInstall, LPCWSTR szComponent,
1188                   INSTALLSTATE *piInstalled, INSTALLSTATE *piAction)
1189 {
1190     MSIPACKAGE* package;
1191     UINT ret;
1192
1193     TRACE("%d %s %p %p\n", hInstall, debugstr_w(szComponent),
1194            piInstalled, piAction);
1195
1196     package = msihandle2msiinfo(hInstall, MSIHANDLETYPE_PACKAGE);
1197     if (!package)
1198     {
1199         HRESULT hr;
1200         BSTR component;
1201         IWineMsiRemotePackage *remote_package;
1202
1203         remote_package = (IWineMsiRemotePackage *)msi_get_remote(hInstall);
1204         if (!remote_package)
1205             return ERROR_INVALID_HANDLE;
1206
1207         component = SysAllocString(szComponent);
1208         if (!component)
1209         {
1210             IWineMsiRemotePackage_Release(remote_package);
1211             return ERROR_OUTOFMEMORY;
1212         }
1213
1214         hr = IWineMsiRemotePackage_GetComponentState(remote_package, component,
1215                                                      piInstalled, piAction);
1216
1217         SysFreeString(component);
1218         IWineMsiRemotePackage_Release(remote_package);
1219
1220         if (FAILED(hr))
1221         {
1222             if (HRESULT_FACILITY(hr) == FACILITY_WIN32)
1223                 return HRESULT_CODE(hr);
1224
1225             return ERROR_FUNCTION_FAILED;
1226         }
1227
1228         return ERROR_SUCCESS;
1229     }
1230
1231     ret = MSI_GetComponentStateW( package, szComponent, piInstalled, piAction);
1232     msiobj_release( &package->hdr );
1233     return ret;
1234 }
1235
1236 /***********************************************************************
1237  * MsiGetLanguage (MSI.@)
1238  */
1239 LANGID WINAPI MsiGetLanguage(MSIHANDLE hInstall)
1240 {
1241     MSIPACKAGE* package;
1242     LANGID langid;
1243     static const WCHAR szProductLanguage[] =
1244         {'P','r','o','d','u','c','t','L','a','n','g','u','a','g','e',0};
1245     
1246     package = msihandle2msiinfo(hInstall, MSIHANDLETYPE_PACKAGE);
1247     if (!package)
1248     {
1249         HRESULT hr;
1250         LANGID lang;
1251         IWineMsiRemotePackage *remote_package;
1252
1253         remote_package = (IWineMsiRemotePackage *)msi_get_remote(hInstall);
1254         if (!remote_package)
1255             return ERROR_INVALID_HANDLE;
1256
1257         hr = IWineMsiRemotePackage_GetLanguage(remote_package, &lang);
1258
1259         if (SUCCEEDED(hr))
1260             return lang;
1261
1262         return 0;
1263     }
1264
1265     langid = msi_get_property_int( package, szProductLanguage, 0 );
1266     msiobj_release( &package->hdr );
1267     return langid;
1268 }
1269
1270 UINT MSI_SetInstallLevel( MSIPACKAGE *package, int iInstallLevel )
1271 {
1272     static const WCHAR szInstallLevel[] = {
1273         'I','N','S','T','A','L','L','L','E','V','E','L',0 };
1274     static const WCHAR fmt[] = { '%','d',0 };
1275     WCHAR level[6];
1276     UINT r;
1277
1278     TRACE("%p %i\n", package, iInstallLevel);
1279
1280     if (iInstallLevel > 32767)
1281         return ERROR_INVALID_PARAMETER;
1282
1283     if (iInstallLevel < 1)
1284         return MSI_SetFeatureStates( package );
1285
1286     sprintfW( level, fmt, iInstallLevel );
1287     r = MSI_SetPropertyW( package, szInstallLevel, level );
1288     if ( r == ERROR_SUCCESS )
1289         r = MSI_SetFeatureStates( package );
1290
1291     return r;
1292 }
1293
1294 /***********************************************************************
1295  * MsiSetInstallLevel (MSI.@)
1296  */
1297 UINT WINAPI MsiSetInstallLevel(MSIHANDLE hInstall, int iInstallLevel)
1298 {
1299     MSIPACKAGE* package;
1300     UINT r;
1301
1302     TRACE("%d %i\n", hInstall, iInstallLevel);
1303
1304     package = msihandle2msiinfo(hInstall, MSIHANDLETYPE_PACKAGE);
1305     if (!package)
1306     {
1307         HRESULT hr;
1308         IWineMsiRemotePackage *remote_package;
1309
1310         remote_package = (IWineMsiRemotePackage *)msi_get_remote(hInstall);
1311         if (!remote_package)
1312             return ERROR_INVALID_HANDLE;
1313
1314         hr = IWineMsiRemotePackage_SetInstallLevel(remote_package, iInstallLevel);
1315
1316         IWineMsiRemotePackage_Release(remote_package);
1317
1318         if (FAILED(hr))
1319         {
1320             if (HRESULT_FACILITY(hr) == FACILITY_WIN32)
1321                 return HRESULT_CODE(hr);
1322
1323             return ERROR_FUNCTION_FAILED;
1324         }
1325
1326         return ERROR_SUCCESS;
1327     }
1328
1329     r = MSI_SetInstallLevel( package, iInstallLevel );
1330
1331     msiobj_release( &package->hdr );
1332
1333     return r;
1334 }
1335
1336 /***********************************************************************
1337  * MsiGetFeatureValidStatesW (MSI.@)
1338  */
1339 UINT WINAPI MsiGetFeatureValidStatesW(MSIHANDLE hInstall, LPCWSTR szFeature,
1340                   LPDWORD pInstallState)
1341 {
1342     if(pInstallState) *pInstallState = 1<<INSTALLSTATE_LOCAL;
1343     FIXME("%d %s %p stub returning %d\n",
1344         hInstall, debugstr_w(szFeature), pInstallState, pInstallState ? *pInstallState : 0);
1345
1346     return ERROR_SUCCESS;
1347 }
1348
1349 /***********************************************************************
1350  * MsiGetFeatureValidStatesA (MSI.@)
1351  */
1352 UINT WINAPI MsiGetFeatureValidStatesA(MSIHANDLE hInstall, LPCSTR szFeature,
1353                   LPDWORD pInstallState)
1354 {
1355     UINT ret;
1356     LPWSTR szwFeature = strdupAtoW(szFeature);
1357
1358     ret = MsiGetFeatureValidStatesW(hInstall, szwFeature, pInstallState);
1359
1360     msi_free(szwFeature);
1361
1362     return ret;
1363 }