msi: Remove unused variables.
[wine] / dlls / msi / source.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 #include <stdarg.h>
22
23 #define COBJMACROS
24 #define NONAMELESSUNION
25
26 #include "windef.h"
27 #include "winbase.h"
28 #include "winreg.h"
29 #include "winnls.h"
30 #include "shlwapi.h"
31 #include "wine/debug.h"
32 #include "msi.h"
33 #include "msiquery.h"
34 #include "msipriv.h"
35 #include "wincrypt.h"
36 #include "winver.h"
37 #include "winuser.h"
38 #include "wine/unicode.h"
39 #include "sddl.h"
40
41 WINE_DEFAULT_DEBUG_CHANNEL(msi);
42
43 /*
44  * These apis are defined in MSI 3.0
45  */
46
47 typedef struct tagMediaInfo
48 {
49     struct list entry;
50     LPWSTR  path;
51     WCHAR   szIndex[10];
52     DWORD   index;
53 } media_info;
54
55 static UINT OpenSourceKey(LPCWSTR szProduct, HKEY* key, DWORD dwOptions,
56                           MSIINSTALLCONTEXT context, BOOL create)
57 {
58     HKEY rootkey = 0; 
59     UINT rc = ERROR_FUNCTION_FAILED;
60     static const WCHAR szSourceList[] = {'S','o','u','r','c','e','L','i','s','t',0};
61
62     if (context == MSIINSTALLCONTEXT_USERUNMANAGED)
63     {
64         if (dwOptions & MSICODE_PATCH)
65             rc = MSIREG_OpenUserPatchesKey(szProduct, &rootkey, create);
66         else
67             rc = MSIREG_OpenUserProductsKey(szProduct, &rootkey, create);
68     }
69     else if (context == MSIINSTALLCONTEXT_USERMANAGED)
70     {
71         if (dwOptions & MSICODE_PATCH)
72             rc = MSIREG_OpenUserPatchesKey(szProduct, &rootkey, create);
73         else
74             rc = MSIREG_OpenLocalManagedProductKey(szProduct, &rootkey, create);
75     }
76     else if (context == MSIINSTALLCONTEXT_MACHINE)
77     {
78         if (dwOptions & MSICODE_PATCH)
79             rc = MSIREG_OpenPatchesKey(szProduct, &rootkey, create);
80         else
81             rc = MSIREG_OpenLocalClassesProductKey(szProduct, &rootkey, create);
82     }
83
84     if (rc != ERROR_SUCCESS)
85     {
86         if (dwOptions & MSICODE_PATCH)
87             return ERROR_UNKNOWN_PATCH;
88         else
89             return ERROR_UNKNOWN_PRODUCT;
90     }
91
92     if (create)
93         rc = RegCreateKeyW(rootkey, szSourceList, key);
94     else
95     {
96         rc = RegOpenKeyW(rootkey,szSourceList, key);
97         if (rc != ERROR_SUCCESS)
98             rc = ERROR_BAD_CONFIGURATION;
99     }
100
101     return rc;
102 }
103
104 static UINT OpenMediaSubkey(HKEY rootkey, HKEY *key, BOOL create)
105 {
106     UINT rc;
107     static const WCHAR media[] = {'M','e','d','i','a',0};
108
109     if (create)
110         rc = RegCreateKeyW(rootkey, media, key);
111     else
112         rc = RegOpenKeyW(rootkey,media, key); 
113
114     return rc;
115 }
116
117 static UINT OpenNetworkSubkey(HKEY rootkey, HKEY *key, BOOL create)
118 {
119     UINT rc;
120     static const WCHAR net[] = {'N','e','t',0};
121
122     if (create)
123         rc = RegCreateKeyW(rootkey, net, key);
124     else
125         rc = RegOpenKeyW(rootkey, net, key); 
126
127     return rc;
128 }
129
130 static UINT OpenURLSubkey(HKEY rootkey, HKEY *key, BOOL create)
131 {
132     UINT rc;
133     static const WCHAR URL[] = {'U','R','L',0};
134
135     if (create)
136         rc = RegCreateKeyW(rootkey, URL, key);
137     else
138         rc = RegOpenKeyW(rootkey, URL, key); 
139
140     return rc;
141 }
142
143 /******************************************************************
144  *  MsiSourceListEnumMediaDisksA   (MSI.@)
145  */
146 UINT WINAPI MsiSourceListEnumMediaDisksA(LPCSTR szProductCodeOrPatchCode,
147                                          LPCSTR szUserSid, MSIINSTALLCONTEXT dwContext,
148                                          DWORD dwOptions, DWORD dwIndex, LPDWORD pdwDiskId,
149                                          LPSTR szVolumeLabel, LPDWORD pcchVolumeLabel,
150                                          LPSTR szDiskPrompt, LPDWORD pcchDiskPrompt)
151 {
152     LPWSTR product = NULL;
153     LPWSTR usersid = NULL;
154     LPWSTR volume = NULL;
155     LPWSTR prompt = NULL;
156     UINT r = ERROR_INVALID_PARAMETER;
157
158     TRACE("(%s, %s, %d, %d, %d, %p, %p, %p, %p, %p)\n", debugstr_a(szProductCodeOrPatchCode),
159           debugstr_a(szUserSid), dwContext, dwOptions, dwIndex, pdwDiskId,
160           szVolumeLabel, pcchVolumeLabel, szDiskPrompt, pcchDiskPrompt);
161
162     if (szDiskPrompt && !pcchDiskPrompt)
163         return ERROR_INVALID_PARAMETER;
164
165     if (szProductCodeOrPatchCode) product = strdupAtoW(szProductCodeOrPatchCode);
166     if (szUserSid) usersid = strdupAtoW(szUserSid);
167
168     /* FIXME: add tests for an invalid format */
169
170     if (pcchVolumeLabel)
171         volume = msi_alloc(*pcchVolumeLabel * sizeof(WCHAR));
172
173     if (pcchDiskPrompt)
174         prompt = msi_alloc(*pcchDiskPrompt * sizeof(WCHAR));
175
176     if (volume) *volume = '\0';
177     if (prompt) *prompt = '\0';
178     r = MsiSourceListEnumMediaDisksW(product, usersid, dwContext, dwOptions,
179                                      dwIndex, pdwDiskId, volume, pcchVolumeLabel,
180                                      prompt, pcchDiskPrompt);
181     if (r != ERROR_SUCCESS)
182         goto done;
183
184     if (szVolumeLabel && pcchVolumeLabel)
185         WideCharToMultiByte(CP_ACP, 0, volume, -1, szVolumeLabel,
186                             *pcchVolumeLabel + 1, NULL, NULL);
187
188     if (szDiskPrompt)
189         WideCharToMultiByte(CP_ACP, 0, prompt, -1, szDiskPrompt,
190                             *pcchDiskPrompt + 1, NULL, NULL);
191
192 done:
193     msi_free(product);
194     msi_free(usersid);
195     msi_free(volume);
196     msi_free(prompt);
197
198     return r;
199 }
200
201 /******************************************************************
202  *  MsiSourceListEnumMediaDisksW   (MSI.@)
203  */
204 UINT WINAPI MsiSourceListEnumMediaDisksW(LPCWSTR szProductCodeOrPatchCode,
205                                          LPCWSTR szUserSid, MSIINSTALLCONTEXT dwContext,
206                                          DWORD dwOptions, DWORD dwIndex, LPDWORD pdwDiskId,
207                                          LPWSTR szVolumeLabel, LPDWORD pcchVolumeLabel,
208                                          LPWSTR szDiskPrompt, LPDWORD pcchDiskPrompt)
209 {
210     WCHAR squished_pc[GUID_SIZE];
211     LPWSTR value = NULL;
212     LPWSTR data = NULL;
213     LPWSTR ptr;
214     HKEY source, media;
215     DWORD valuesz, datasz = 0;
216     DWORD type;
217     DWORD numvals, size;
218     LONG res;
219     UINT r;
220     static int index = 0;
221
222     TRACE("(%s, %s, %d, %d, %d, %p, %p, %p, %p)\n", debugstr_w(szProductCodeOrPatchCode),
223           debugstr_w(szUserSid), dwContext, dwOptions, dwIndex, szVolumeLabel,
224           pcchVolumeLabel, szDiskPrompt, pcchDiskPrompt);
225
226     if (!szProductCodeOrPatchCode ||
227         !squash_guid(szProductCodeOrPatchCode, squished_pc))
228         return ERROR_INVALID_PARAMETER;
229
230     if (dwContext == MSIINSTALLCONTEXT_MACHINE && szUserSid)
231         return ERROR_INVALID_PARAMETER;
232
233     if (dwOptions != MSICODE_PRODUCT && dwOptions != MSICODE_PATCH)
234         return ERROR_INVALID_PARAMETER;
235
236     if (szDiskPrompt && !pcchDiskPrompt)
237         return ERROR_INVALID_PARAMETER;
238
239     if (dwIndex == 0)
240         index = 0;
241
242     if (dwIndex != index)
243         return ERROR_INVALID_PARAMETER;
244
245     r = OpenSourceKey(szProductCodeOrPatchCode, &source,
246                       dwOptions, dwContext, FALSE);
247     if (r != ERROR_SUCCESS)
248         return r;
249
250     r = OpenMediaSubkey(source, &media, FALSE);
251     if (r != ERROR_SUCCESS)
252     {
253         RegCloseKey(source);
254         return ERROR_NO_MORE_ITEMS;
255     }
256
257     if (!pcchVolumeLabel && !pcchDiskPrompt)
258     {
259         r = RegEnumValueW(media, dwIndex, NULL, NULL, NULL,
260                           &type, NULL, NULL);
261         goto done;
262     }
263
264     res = RegQueryInfoKeyW(media, NULL, NULL, NULL, NULL, NULL,
265                            NULL, &numvals, &valuesz, &datasz, NULL, NULL);
266     if (res != ERROR_SUCCESS)
267     {
268         r = ERROR_BAD_CONFIGURATION;
269         goto done;
270     }
271
272     value = msi_alloc(++valuesz * sizeof(WCHAR));
273     data = msi_alloc(++datasz * sizeof(WCHAR));
274     if (!value || !data)
275     {
276         r = ERROR_OUTOFMEMORY;
277         goto done;
278     }
279
280     r = RegEnumValueW(media, dwIndex, value, &valuesz,
281                       NULL, &type, (LPBYTE)data, &datasz);
282     if (r != ERROR_SUCCESS)
283         goto done;
284
285     if (pdwDiskId)
286         *pdwDiskId = atolW(value);
287
288     ptr = strchrW(data, ';');
289     if (!ptr)
290         ptr = data;
291     else
292         *ptr = '\0';
293
294     if (pcchVolumeLabel)
295     {
296         size = lstrlenW(data);
297         if (size >= *pcchVolumeLabel)
298             r = ERROR_MORE_DATA;
299         else if (szVolumeLabel)
300             lstrcpyW(szVolumeLabel, data);
301
302         *pcchVolumeLabel = size;
303     }
304
305     if (pcchDiskPrompt)
306     {
307         data = ptr;
308         if (!*data)
309             data++;
310
311         size = lstrlenW(data);
312         if (size >= *pcchDiskPrompt)
313             r = ERROR_MORE_DATA;
314         else if (szDiskPrompt)
315             lstrcpyW(szDiskPrompt, data);
316
317         *pcchDiskPrompt = size;
318     }
319
320     index++;
321
322 done:
323     msi_free(value);
324     msi_free(data);
325     RegCloseKey(source);
326
327     return r;
328 }
329
330 /******************************************************************
331  *  MsiSourceListEnumSourcesA   (MSI.@)
332  */
333 UINT WINAPI MsiSourceListEnumSourcesA(LPCSTR szProductCodeOrPatch, LPCSTR szUserSid,
334                                       MSIINSTALLCONTEXT dwContext,
335                                       DWORD dwOptions, DWORD dwIndex,
336                                       LPSTR szSource, LPDWORD pcchSource)
337 {
338     LPWSTR product = NULL;
339     LPWSTR usersid = NULL;
340     LPWSTR source = NULL;
341     DWORD len = 0;
342     UINT r = ERROR_INVALID_PARAMETER;
343     static int index = 0;
344
345     TRACE("(%s, %s, %d, %d, %d, %p, %p)\n", debugstr_a(szProductCodeOrPatch),
346           debugstr_a(szUserSid), dwContext, dwOptions, dwIndex, szSource, pcchSource);
347
348     if (dwIndex == 0)
349         index = 0;
350
351     if (szSource && !pcchSource)
352         goto done;
353
354     if (dwIndex != index)
355         goto done;
356
357     if (szProductCodeOrPatch) product = strdupAtoW(szProductCodeOrPatch);
358     if (szUserSid) usersid = strdupAtoW(szUserSid);
359
360     r = MsiSourceListEnumSourcesW(product, usersid, dwContext, dwOptions,
361                                   dwIndex, NULL, &len);
362     if (r != ERROR_SUCCESS)
363         goto done;
364
365     source = msi_alloc(++len * sizeof(WCHAR));
366     if (!source)
367     {
368         r = ERROR_OUTOFMEMORY;
369         goto done;
370     }
371
372     *source = '\0';
373     r = MsiSourceListEnumSourcesW(product, usersid, dwContext, dwOptions,
374                                   dwIndex, source, &len);
375     if (r != ERROR_SUCCESS)
376         goto done;
377
378     len = WideCharToMultiByte(CP_ACP, 0, source, -1, NULL, 0, NULL, NULL);
379     if (pcchSource && *pcchSource >= len)
380         WideCharToMultiByte(CP_ACP, 0, source, -1, szSource, len, NULL, NULL);
381     else if (szSource)
382         r = ERROR_MORE_DATA;
383
384     if (pcchSource)
385         *pcchSource = len - 1;
386
387 done:
388     msi_free(product);
389     msi_free(usersid);
390     msi_free(source);
391
392     if (r == ERROR_SUCCESS)
393     {
394         if (szSource || !pcchSource) index++;
395     }
396     else if (dwIndex > index)
397         index = 0;
398
399     return r;
400 }
401
402 /******************************************************************
403  *  MsiSourceListEnumSourcesW   (MSI.@)
404  */
405 UINT WINAPI MsiSourceListEnumSourcesW(LPCWSTR szProductCodeOrPatch, LPCWSTR szUserSid,
406                                       MSIINSTALLCONTEXT dwContext,
407                                       DWORD dwOptions, DWORD dwIndex,
408                                       LPWSTR szSource, LPDWORD pcchSource)
409 {
410     WCHAR squished_pc[GUID_SIZE];
411     WCHAR name[32];
412     HKEY source = NULL;
413     HKEY subkey = NULL;
414     LONG res;
415     UINT r = ERROR_INVALID_PARAMETER;
416     static int index = 0;
417
418     static const WCHAR format[] = {'%','d',0};
419
420     TRACE("(%s, %s, %d, %d, %d, %p, %p)\n", debugstr_w(szProductCodeOrPatch),
421           debugstr_w(szUserSid), dwContext, dwOptions, dwIndex, szSource, pcchSource);
422
423     if (dwIndex == 0)
424         index = 0;
425
426     if (!szProductCodeOrPatch || !squash_guid(szProductCodeOrPatch, squished_pc))
427         goto done;
428
429     if (szSource && !pcchSource)
430         goto done;
431
432     if (!(dwOptions & (MSISOURCETYPE_NETWORK | MSISOURCETYPE_URL)))
433         goto done;
434
435     if ((dwOptions & MSISOURCETYPE_NETWORK) && (dwOptions & MSISOURCETYPE_URL))
436         goto done;
437
438     if (dwContext == MSIINSTALLCONTEXT_MACHINE && szUserSid)
439         goto done;
440
441     if (dwIndex != index)
442         goto done;
443
444     r = OpenSourceKey(szProductCodeOrPatch, &source,
445                       dwOptions, dwContext, FALSE);
446     if (r != ERROR_SUCCESS)
447         goto done;
448
449     if (dwOptions & MSISOURCETYPE_NETWORK)
450         r = OpenNetworkSubkey(source, &subkey, FALSE);
451     else if (dwOptions & MSISOURCETYPE_URL)
452         r = OpenURLSubkey(source, &subkey, FALSE);
453
454     if (r != ERROR_SUCCESS)
455     {
456         r = ERROR_NO_MORE_ITEMS;
457         goto done;
458     }
459
460     sprintfW(name, format, dwIndex + 1);
461
462     res = RegQueryValueExW(subkey, name, 0, 0, (LPBYTE)szSource, pcchSource);
463     if (res != ERROR_SUCCESS && res != ERROR_MORE_DATA)
464         r = ERROR_NO_MORE_ITEMS;
465
466 done:
467     RegCloseKey(subkey);
468     RegCloseKey(source);
469
470     if (r == ERROR_SUCCESS)
471     {
472         if (szSource || !pcchSource) index++;
473     }
474     else if (dwIndex > index)
475         index = 0;
476
477     return r;
478 }
479
480 /******************************************************************
481  *  MsiSourceListGetInfoA   (MSI.@)
482  */
483 UINT WINAPI MsiSourceListGetInfoA( LPCSTR szProduct, LPCSTR szUserSid,
484                                    MSIINSTALLCONTEXT dwContext, DWORD dwOptions,
485                                    LPCSTR szProperty, LPSTR szValue,
486                                    LPDWORD pcchValue)
487 {
488     UINT ret;
489     LPWSTR product = NULL;
490     LPWSTR usersid = NULL;
491     LPWSTR property = NULL;
492     LPWSTR value = NULL;
493     DWORD len = 0;
494
495     if (szValue && !pcchValue)
496         return ERROR_INVALID_PARAMETER;
497
498     if (szProduct) product = strdupAtoW(szProduct);
499     if (szUserSid) usersid = strdupAtoW(szUserSid);
500     if (szProperty) property = strdupAtoW(szProperty);
501
502     ret = MsiSourceListGetInfoW(product, usersid, dwContext, dwOptions,
503                                 property, NULL, &len);
504     if (ret != ERROR_SUCCESS)
505         goto done;
506
507     value = msi_alloc(++len * sizeof(WCHAR));
508     if (!value)
509         return ERROR_OUTOFMEMORY;
510
511     *value = '\0';
512     ret = MsiSourceListGetInfoW(product, usersid, dwContext, dwOptions,
513                                 property, value, &len);
514     if (ret != ERROR_SUCCESS)
515         goto done;
516
517     len = WideCharToMultiByte(CP_ACP, 0, value, -1, NULL, 0, NULL, NULL);
518     if (*pcchValue >= len)
519         WideCharToMultiByte(CP_ACP, 0, value, -1, szValue, len, NULL, NULL);
520     else if (szValue)
521         ret = ERROR_MORE_DATA;
522
523     *pcchValue = len - 1;
524
525 done:
526     msi_free(product);
527     msi_free(usersid);
528     msi_free(property);
529     msi_free(value);
530     return ret;
531 }
532
533 /******************************************************************
534  *  MsiSourceListGetInfoW   (MSI.@)
535  */
536 UINT WINAPI MsiSourceListGetInfoW( LPCWSTR szProduct, LPCWSTR szUserSid,
537                                    MSIINSTALLCONTEXT dwContext, DWORD dwOptions,
538                                    LPCWSTR szProperty, LPWSTR szValue, 
539                                    LPDWORD pcchValue) 
540 {
541     WCHAR squished_pc[GUID_SIZE];
542     HKEY sourcekey, media;
543     LPWSTR source, ptr;
544     DWORD size;
545     UINT rc;
546
547     static const WCHAR mediapack[] = {
548         'M','e','d','i','a','P','a','c','k','a','g','e',0};
549
550     TRACE("%s %s\n", debugstr_w(szProduct), debugstr_w(szProperty));
551
552     if (!szProduct || !squash_guid(szProduct, squished_pc))
553         return ERROR_INVALID_PARAMETER;
554
555     if (szValue && !pcchValue)
556         return ERROR_INVALID_PARAMETER;
557
558     if (dwContext != MSIINSTALLCONTEXT_USERMANAGED &&
559         dwContext != MSIINSTALLCONTEXT_USERUNMANAGED &&
560         dwContext != MSIINSTALLCONTEXT_MACHINE)
561         return ERROR_INVALID_PARAMETER;
562
563     if (!szProperty)
564         return ERROR_INVALID_PARAMETER;
565
566     if (szUserSid)
567         FIXME("Unhandled UserSid %s\n",debugstr_w(szUserSid));
568
569     if (dwContext != MSIINSTALLCONTEXT_USERUNMANAGED)
570         FIXME("Unhandled context %d\n", dwContext);
571
572     rc = OpenSourceKey(szProduct, &sourcekey, dwOptions, dwContext, FALSE);
573     if (rc != ERROR_SUCCESS)
574         return rc;
575
576     if (!lstrcmpW(szProperty, INSTALLPROPERTY_MEDIAPACKAGEPATHW) ||
577         !lstrcmpW(szProperty, INSTALLPROPERTY_DISKPROMPTW))
578     {
579         rc = OpenMediaSubkey(sourcekey, &media, FALSE);
580         if (rc != ERROR_SUCCESS)
581         {
582             RegCloseKey(sourcekey);
583             return ERROR_SUCCESS;
584         }
585
586         if (!lstrcmpW(szProperty, INSTALLPROPERTY_MEDIAPACKAGEPATHW))
587             szProperty = mediapack;
588
589         RegQueryValueExW(media, szProperty, 0, 0, (LPBYTE)szValue, pcchValue);
590         RegCloseKey(media);
591     }
592     else if (!lstrcmpW(szProperty, INSTALLPROPERTY_LASTUSEDSOURCEW) ||
593              !lstrcmpW(szProperty, INSTALLPROPERTY_LASTUSEDTYPEW))
594     {
595         rc = RegQueryValueExW(sourcekey, INSTALLPROPERTY_LASTUSEDSOURCEW,
596                               0, 0, NULL, &size);
597         if (rc != ERROR_SUCCESS)
598         {
599             RegCloseKey(sourcekey);
600             return ERROR_SUCCESS;
601         }
602
603         source = msi_alloc(size);
604         RegQueryValueExW(sourcekey, INSTALLPROPERTY_LASTUSEDSOURCEW,
605                          0, 0, (LPBYTE)source, &size);
606
607         if (!*source)
608         {
609             msi_free(source);
610             RegCloseKey(sourcekey);
611             return ERROR_SUCCESS;
612         }
613
614         if (!lstrcmpW(szProperty, INSTALLPROPERTY_LASTUSEDTYPEW))
615         {
616             if (*source != 'n' && *source != 'u' && *source != 'm')
617             {
618                 msi_free(source);
619                 RegCloseKey(sourcekey);
620                 return ERROR_SUCCESS;
621             }
622
623             ptr = source;
624             source[1] = '\0';
625         }
626         else
627         {
628             ptr = strrchrW(source, ';');
629             if (!ptr)
630                 ptr = source;
631             else
632                 ptr++;
633         }
634
635         if (szValue)
636         {
637             if (lstrlenW(ptr) < *pcchValue)
638                 lstrcpyW(szValue, ptr);
639             else
640                 rc = ERROR_MORE_DATA;
641         }
642
643         *pcchValue = lstrlenW(ptr);
644         msi_free(source);
645     }
646     else if (strcmpW(INSTALLPROPERTY_PACKAGENAMEW, szProperty)==0)
647     {
648         *pcchValue = *pcchValue * sizeof(WCHAR);
649         rc = RegQueryValueExW(sourcekey, INSTALLPROPERTY_PACKAGENAMEW, 0, 0,
650                               (LPBYTE)szValue, pcchValue);
651         if (rc != ERROR_SUCCESS && rc != ERROR_MORE_DATA)
652         {
653             *pcchValue = 0;
654             rc = ERROR_SUCCESS;
655         }
656         else
657         {
658             if (*pcchValue)
659                 *pcchValue = (*pcchValue - 1) / sizeof(WCHAR);
660             if (szValue)
661                 szValue[*pcchValue] = '\0';
662         }
663     }
664     else
665     {
666         FIXME("Unknown property %s\n",debugstr_w(szProperty));
667         rc = ERROR_UNKNOWN_PROPERTY;
668     }
669
670     RegCloseKey(sourcekey);
671     return rc;
672 }
673
674 /******************************************************************
675  *  MsiSourceListSetInfoA   (MSI.@)
676  */
677 UINT WINAPI MsiSourceListSetInfoA(LPCSTR szProduct, LPCSTR szUserSid,
678                                   MSIINSTALLCONTEXT dwContext, DWORD dwOptions,
679                                   LPCSTR szProperty, LPCSTR szValue)
680 {
681     UINT ret;
682     LPWSTR product = NULL;
683     LPWSTR usersid = NULL;
684     LPWSTR property = NULL;
685     LPWSTR value = NULL;
686
687     if (szProduct) product = strdupAtoW(szProduct);
688     if (szUserSid) usersid = strdupAtoW(szUserSid);
689     if (szProperty) property = strdupAtoW(szProperty);
690     if (szValue) value = strdupAtoW(szValue);
691
692     ret = MsiSourceListSetInfoW(product, usersid, dwContext, dwOptions,
693                                 property, value);
694
695     msi_free(product);
696     msi_free(usersid);
697     msi_free(property);
698     msi_free(value);
699
700     return ret;
701 }
702
703 UINT msi_set_last_used_source(LPCWSTR product, LPCWSTR usersid,
704                               MSIINSTALLCONTEXT context, DWORD options,
705                               LPCWSTR value)
706 {
707     HKEY source;
708     LPWSTR buffer;
709     WCHAR typechar;
710     DWORD size;
711     UINT r;
712     int index = 1;
713
714     static const WCHAR format[] = {'%','c',';','%','i',';','%','s',0};
715
716     if (options & MSISOURCETYPE_NETWORK)
717         typechar = 'n';
718     else if (options & MSISOURCETYPE_URL)
719         typechar = 'u';
720     else if (options & MSISOURCETYPE_MEDIA)
721         typechar = 'm';
722     else
723         return ERROR_INVALID_PARAMETER;
724
725     if (!(options & MSISOURCETYPE_MEDIA))
726     {
727         r = MsiSourceListAddSourceExW(product, usersid, context,
728                                       options, value, 0);
729         if (r != ERROR_SUCCESS)
730             return r;
731
732         index = 0;
733         while ((r = MsiSourceListEnumSourcesW(product, usersid, context, options,
734                                               index, NULL, NULL)) == ERROR_SUCCESS)
735             index++;
736
737         if (r != ERROR_NO_MORE_ITEMS)
738             return r;
739     }
740
741     size = (lstrlenW(format) + lstrlenW(value) + 7) * sizeof(WCHAR);
742     buffer = msi_alloc(size);
743     if (!buffer)
744         return ERROR_OUTOFMEMORY;
745
746     r = OpenSourceKey(product, &source, MSICODE_PRODUCT, context, FALSE);
747     if (r != ERROR_SUCCESS)
748         return r;
749
750     sprintfW(buffer, format, typechar, index, value);
751
752     size = (lstrlenW(buffer) + 1) * sizeof(WCHAR);
753     r = RegSetValueExW(source, INSTALLPROPERTY_LASTUSEDSOURCEW, 0,
754                        REG_SZ, (LPBYTE)buffer, size);
755     msi_free(buffer);
756
757     RegCloseKey(source);
758     return r;
759 }
760
761 /******************************************************************
762  *  MsiSourceListSetInfoW   (MSI.@)
763  */
764 UINT WINAPI MsiSourceListSetInfoW( LPCWSTR szProduct, LPCWSTR szUserSid,
765                                    MSIINSTALLCONTEXT dwContext, DWORD dwOptions,
766                                    LPCWSTR szProperty, LPCWSTR szValue)
767 {
768     WCHAR squished_pc[GUID_SIZE];
769     HKEY sourcekey, media;
770     LPCWSTR property;
771     UINT rc;
772
773     static const WCHAR media_package[] = {
774         'M','e','d','i','a','P','a','c','k','a','g','e',0
775     };
776
777     TRACE("%s %s %x %x %s %s\n", debugstr_w(szProduct), debugstr_w(szUserSid),
778             dwContext, dwOptions, debugstr_w(szProperty), debugstr_w(szValue));
779
780     if (!szProduct || !squash_guid(szProduct, squished_pc))
781         return ERROR_INVALID_PARAMETER;
782
783     if (!szProperty)
784         return ERROR_INVALID_PARAMETER;
785
786     if (!szValue)
787         return ERROR_UNKNOWN_PROPERTY;
788
789     if (dwContext == MSIINSTALLCONTEXT_MACHINE && szUserSid)
790         return ERROR_INVALID_PARAMETER;
791
792     if (dwOptions & MSICODE_PATCH)
793     {
794         FIXME("Unhandled options MSICODE_PATCH\n");
795         return ERROR_UNKNOWN_PATCH;
796     }
797
798     property = szProperty;
799     if (!lstrcmpW(szProperty, INSTALLPROPERTY_MEDIAPACKAGEPATHW))
800         property = media_package;
801
802     rc = OpenSourceKey(szProduct, &sourcekey, MSICODE_PRODUCT, dwContext, FALSE);
803     if (rc != ERROR_SUCCESS)
804         return rc;
805
806     if (lstrcmpW(szProperty, INSTALLPROPERTY_LASTUSEDSOURCEW) &&
807         dwOptions & (MSISOURCETYPE_NETWORK | MSISOURCETYPE_URL))
808     {
809         RegCloseKey(sourcekey);
810         return ERROR_INVALID_PARAMETER;
811     }
812
813     if (!lstrcmpW(szProperty, INSTALLPROPERTY_MEDIAPACKAGEPATHW) ||
814         !lstrcmpW(szProperty, INSTALLPROPERTY_DISKPROMPTW))
815     {
816         rc = OpenMediaSubkey(sourcekey, &media, TRUE);
817         if (rc == ERROR_SUCCESS)
818         {
819             rc = msi_reg_set_val_str(media, property, szValue);
820             RegCloseKey(media);
821         }
822     }
823     else if (strcmpW(INSTALLPROPERTY_PACKAGENAMEW, szProperty)==0)
824     {
825         DWORD size = lstrlenW(szValue)*sizeof(WCHAR);
826         rc = RegSetValueExW(sourcekey, INSTALLPROPERTY_PACKAGENAMEW, 0,
827                 REG_SZ, (const BYTE *)szValue, size);
828         if (rc != ERROR_SUCCESS)
829             rc = ERROR_UNKNOWN_PROPERTY;
830     }
831     else if (!lstrcmpW(szProperty, INSTALLPROPERTY_LASTUSEDSOURCEW))
832     {
833         if (!(dwOptions & (MSISOURCETYPE_NETWORK | MSISOURCETYPE_URL)))
834             rc = ERROR_INVALID_PARAMETER;
835         else
836             rc = msi_set_last_used_source(szProduct, szUserSid, dwContext,
837                                           dwOptions, szValue);
838     }
839     else
840         rc = ERROR_UNKNOWN_PROPERTY;
841
842     RegCloseKey(sourcekey);
843     return rc;
844 }
845
846 /******************************************************************
847  *  MsiSourceListAddSourceW (MSI.@)
848  */
849 UINT WINAPI MsiSourceListAddSourceW( LPCWSTR szProduct, LPCWSTR szUserName,
850         DWORD dwReserved, LPCWSTR szSource)
851 {
852     WCHAR squished_pc[GUID_SIZE];
853     INT ret;
854     LPWSTR sidstr = NULL;
855     DWORD sidsize = 0;
856     DWORD domsize = 0;
857     DWORD context;
858     HKEY hkey = 0;
859     UINT r;
860
861     TRACE("%s %s %s\n", debugstr_w(szProduct), debugstr_w(szUserName), debugstr_w(szSource));
862
863     if (!szSource || !*szSource)
864         return ERROR_INVALID_PARAMETER;
865
866     if (dwReserved != 0)
867         return ERROR_INVALID_PARAMETER;
868
869     if (!szProduct || !squash_guid(szProduct, squished_pc))
870         return ERROR_INVALID_PARAMETER;
871
872     if (!szUserName || !*szUserName)
873         context = MSIINSTALLCONTEXT_MACHINE;
874     else
875     {
876         if (LookupAccountNameW(NULL, szUserName, NULL, &sidsize, NULL, &domsize, NULL))
877         {
878             PSID psid = msi_alloc(sidsize);
879
880             if (LookupAccountNameW(NULL, szUserName, psid, &sidsize, NULL, &domsize, NULL))
881                 ConvertSidToStringSidW(psid, &sidstr);
882
883             msi_free(psid);
884         }
885
886         r = MSIREG_OpenLocalManagedProductKey(szProduct, &hkey, FALSE);
887         if (r == ERROR_SUCCESS)
888             context = MSIINSTALLCONTEXT_USERMANAGED;
889         else
890         {
891             r = MSIREG_OpenUserProductsKey(szProduct, &hkey, FALSE);
892             if (r != ERROR_SUCCESS)
893                 return ERROR_UNKNOWN_PRODUCT;
894
895             context = MSIINSTALLCONTEXT_USERUNMANAGED;
896         }
897
898         RegCloseKey(hkey);
899     }
900
901     ret = MsiSourceListAddSourceExW(szProduct, sidstr, 
902         context, MSISOURCETYPE_NETWORK, szSource, 0);
903
904     if (sidstr)
905         LocalFree(sidstr);
906
907     return ret;
908 }
909
910 /******************************************************************
911  *  MsiSourceListAddSourceA (MSI.@)
912  */
913 UINT WINAPI MsiSourceListAddSourceA( LPCSTR szProduct, LPCSTR szUserName,
914         DWORD dwReserved, LPCSTR szSource)
915 {
916     INT ret;
917     LPWSTR szwproduct;
918     LPWSTR szwusername;
919     LPWSTR szwsource;
920
921     szwproduct = strdupAtoW( szProduct );
922     szwusername = strdupAtoW( szUserName );
923     szwsource = strdupAtoW( szSource );
924
925     ret = MsiSourceListAddSourceW(szwproduct, szwusername, dwReserved, szwsource);
926
927     msi_free(szwproduct);
928     msi_free(szwusername);
929     msi_free(szwsource);
930
931     return ret;
932 }
933
934 /******************************************************************
935  *  MsiSourceListAddSourceExA (MSI.@)
936  */
937 UINT WINAPI MsiSourceListAddSourceExA(LPCSTR szProduct, LPCSTR szUserSid,
938         MSIINSTALLCONTEXT dwContext, DWORD dwOptions, LPCSTR szSource, DWORD dwIndex)
939 {
940     UINT ret;
941     LPWSTR product, usersid, source;
942
943     product = strdupAtoW(szProduct);
944     usersid = strdupAtoW(szUserSid);
945     source = strdupAtoW(szSource);
946
947     ret = MsiSourceListAddSourceExW(product, usersid, dwContext,
948                                     dwOptions, source, dwIndex);
949
950     msi_free(product);
951     msi_free(usersid);
952     msi_free(source);
953
954     return ret;
955 }
956
957 static void free_source_list(struct list *sourcelist)
958 {
959     while (!list_empty(sourcelist))
960     {
961         media_info *info = LIST_ENTRY(list_head(sourcelist), media_info, entry);
962         list_remove(&info->entry);
963         msi_free(info->path);
964         msi_free(info);
965     }
966 }
967
968 static void add_source_to_list(struct list *sourcelist, media_info *info,
969                                DWORD *index)
970 {
971     media_info *iter;
972     BOOL found = FALSE;
973     static const WCHAR fmt[] = {'%','i',0};
974
975     if (index) *index = 0;
976
977     if (list_empty(sourcelist))
978     {
979         list_add_head(sourcelist, &info->entry);
980         return;
981     }
982
983     LIST_FOR_EACH_ENTRY(iter, sourcelist, media_info, entry)
984     {
985         if (!found && info->index < iter->index)
986         {
987             found = TRUE;
988             list_add_before(&iter->entry, &info->entry);
989         }
990
991         /* update the rest of the list */
992         if (found)
993             sprintfW(iter->szIndex, fmt, ++iter->index);
994         else if (index)
995             (*index)++;
996     }
997
998     if (!found)
999         list_add_after(&iter->entry, &info->entry);
1000 }
1001
1002 static UINT fill_source_list(struct list *sourcelist, HKEY sourcekey, DWORD *count)
1003 {
1004     UINT r = ERROR_SUCCESS;
1005     DWORD index = 0;
1006     WCHAR name[10];
1007     DWORD size, val_size;
1008     media_info *entry;
1009
1010     *count = 0;
1011
1012     while (r == ERROR_SUCCESS)
1013     {
1014         size = sizeof(name) / sizeof(name[0]);
1015         r = RegEnumValueW(sourcekey, index, name, &size, NULL, NULL, NULL, &val_size);
1016         if (r != ERROR_SUCCESS)
1017             return r;
1018
1019         entry = msi_alloc(sizeof(media_info));
1020         if (!entry)
1021             goto error;
1022
1023         entry->path = msi_alloc(val_size);
1024         if (!entry->path)
1025         {
1026             msi_free(entry);
1027             goto error;
1028         }
1029
1030         lstrcpyW(entry->szIndex, name);
1031         entry->index = atoiW(name);
1032
1033         size++;
1034         r = RegEnumValueW(sourcekey, index, name, &size, NULL,
1035                           NULL, (LPBYTE)entry->path, &val_size);
1036         if (r != ERROR_SUCCESS)
1037         {
1038             msi_free(entry->path);
1039             msi_free(entry);
1040             goto error;
1041         }
1042
1043         index = ++(*count);
1044         add_source_to_list(sourcelist, entry, NULL);
1045     }
1046
1047 error:
1048     *count = -1;
1049     free_source_list(sourcelist);
1050     return ERROR_OUTOFMEMORY;
1051 }
1052
1053 /******************************************************************
1054  *  MsiSourceListAddSourceExW (MSI.@)
1055  */
1056 UINT WINAPI MsiSourceListAddSourceExW( LPCWSTR szProduct, LPCWSTR szUserSid,
1057         MSIINSTALLCONTEXT dwContext, DWORD dwOptions, LPCWSTR szSource, 
1058         DWORD dwIndex)
1059 {
1060     HKEY sourcekey;
1061     HKEY typekey;
1062     UINT rc;
1063     struct list sourcelist;
1064     media_info *info;
1065     WCHAR squished_pc[GUID_SIZE];
1066     WCHAR name[10];
1067     LPWSTR source;
1068     LPCWSTR postfix;
1069     DWORD size, count;
1070     DWORD index;
1071
1072     static const WCHAR fmt[] = {'%','i',0};
1073     static const WCHAR one[] = {'1',0};
1074     static const WCHAR backslash[] = {'\\',0};
1075     static const WCHAR forwardslash[] = {'/',0};
1076
1077     TRACE("%s %s %x %x %s %i\n", debugstr_w(szProduct), debugstr_w(szUserSid),
1078           dwContext, dwOptions, debugstr_w(szSource), dwIndex);
1079
1080     if (!szProduct || !squash_guid(szProduct, squished_pc))
1081         return ERROR_INVALID_PARAMETER;
1082
1083     if (!szSource || !*szSource)
1084         return ERROR_INVALID_PARAMETER;
1085
1086     if (!(dwOptions & (MSISOURCETYPE_NETWORK | MSISOURCETYPE_URL)))
1087         return ERROR_INVALID_PARAMETER;
1088
1089     if (dwOptions & MSICODE_PATCH)
1090     {
1091         FIXME("Unhandled options MSICODE_PATCH\n");
1092         return ERROR_FUNCTION_FAILED;
1093     }
1094
1095     if (szUserSid && (dwContext & MSIINSTALLCONTEXT_MACHINE))
1096         return ERROR_INVALID_PARAMETER;
1097
1098     rc = OpenSourceKey(szProduct, &sourcekey, MSICODE_PRODUCT, dwContext, FALSE);
1099     if (rc != ERROR_SUCCESS)
1100         return rc;
1101
1102     if (dwOptions & MSISOURCETYPE_NETWORK)
1103         rc = OpenNetworkSubkey(sourcekey, &typekey, TRUE);
1104     else if (dwOptions & MSISOURCETYPE_URL)
1105         rc = OpenURLSubkey(sourcekey, &typekey, TRUE);
1106     else if (dwOptions & MSISOURCETYPE_MEDIA)
1107         rc = OpenMediaSubkey(sourcekey, &typekey, TRUE);
1108     else
1109     {
1110         ERR("unknown media type: %08x\n", dwOptions);
1111         RegCloseKey(sourcekey);
1112         return ERROR_FUNCTION_FAILED;
1113     }
1114
1115     postfix = (dwOptions & MSISOURCETYPE_NETWORK) ? backslash : forwardslash;
1116     if (szSource[lstrlenW(szSource) - 1] == *postfix)
1117         source = strdupW(szSource);
1118     else
1119     {
1120         size = lstrlenW(szSource) + 2;
1121         source = msi_alloc(size * sizeof(WCHAR));
1122         lstrcpyW(source, szSource);
1123         lstrcatW(source, postfix);
1124     }
1125
1126     list_init(&sourcelist);
1127     rc = fill_source_list(&sourcelist, typekey, &count);
1128     if (rc != ERROR_NO_MORE_ITEMS)
1129         return rc;
1130
1131     size = (lstrlenW(source) + 1) * sizeof(WCHAR);
1132
1133     if (count == 0)
1134     {
1135         rc = RegSetValueExW(typekey, one, 0, REG_EXPAND_SZ, (LPBYTE)source, size);
1136         goto done;
1137     }
1138     else if (dwIndex > count || dwIndex == 0)
1139     {
1140         sprintfW(name, fmt, count + 1);
1141         rc = RegSetValueExW(typekey, name, 0, REG_EXPAND_SZ, (LPBYTE)source, size);
1142         goto done;
1143     }
1144     else
1145     {
1146         sprintfW(name, fmt, dwIndex);
1147         info = msi_alloc(sizeof(media_info));
1148         if (!info)
1149         {
1150             rc = ERROR_OUTOFMEMORY;
1151             goto done;
1152         }
1153
1154         info->path = strdupW(source);
1155         lstrcpyW(info->szIndex, name);
1156         info->index = dwIndex;
1157         add_source_to_list(&sourcelist, info, &index);
1158
1159         LIST_FOR_EACH_ENTRY(info, &sourcelist, media_info, entry)
1160         {
1161             if (info->index < index)
1162                 continue;
1163
1164             size = (lstrlenW(info->path) + 1) * sizeof(WCHAR);
1165             rc = RegSetValueExW(typekey, info->szIndex, 0,
1166                                 REG_EXPAND_SZ, (LPBYTE)info->path, size);
1167             if (rc != ERROR_SUCCESS)
1168                 goto done;
1169         }
1170     }
1171
1172 done:
1173     free_source_list(&sourcelist);
1174     msi_free(source);
1175     RegCloseKey(typekey);
1176     RegCloseKey(sourcekey);
1177     return rc;
1178 }
1179
1180 /******************************************************************
1181  *  MsiSourceListAddMediaDiskA (MSI.@)
1182  */
1183 UINT WINAPI MsiSourceListAddMediaDiskA(LPCSTR szProduct, LPCSTR szUserSid,
1184         MSIINSTALLCONTEXT dwContext, DWORD dwOptions, DWORD dwDiskId,
1185         LPCSTR szVolumeLabel, LPCSTR szDiskPrompt)
1186 {
1187     UINT r;
1188     LPWSTR product = NULL;
1189     LPWSTR usersid = NULL;
1190     LPWSTR volume = NULL;
1191     LPWSTR prompt = NULL;
1192
1193     if (szProduct) product = strdupAtoW(szProduct);
1194     if (szUserSid) usersid = strdupAtoW(szUserSid);
1195     if (szVolumeLabel) volume = strdupAtoW(szVolumeLabel);
1196     if (szDiskPrompt) prompt = strdupAtoW(szDiskPrompt);
1197
1198     r = MsiSourceListAddMediaDiskW(product, usersid, dwContext, dwOptions,
1199                                      dwDiskId, volume, prompt);
1200
1201     msi_free(product);
1202     msi_free(usersid);
1203     msi_free(volume);
1204     msi_free(prompt);
1205
1206     return r;
1207 }
1208
1209 /******************************************************************
1210  *  MsiSourceListAddMediaDiskW (MSI.@)
1211  */
1212 UINT WINAPI MsiSourceListAddMediaDiskW(LPCWSTR szProduct, LPCWSTR szUserSid, 
1213         MSIINSTALLCONTEXT dwContext, DWORD dwOptions, DWORD dwDiskId, 
1214         LPCWSTR szVolumeLabel, LPCWSTR szDiskPrompt)
1215 {
1216     HKEY sourcekey;
1217     HKEY mediakey;
1218     UINT rc;
1219     WCHAR szIndex[10];
1220     WCHAR squished_pc[GUID_SIZE];
1221     LPWSTR buffer;
1222     DWORD size;
1223
1224     static const WCHAR fmt[] = {'%','i',0};
1225     static const WCHAR semicolon[] = {';',0};
1226
1227     TRACE("%s %s %x %x %i %s %s\n", debugstr_w(szProduct),
1228             debugstr_w(szUserSid), dwContext, dwOptions, dwDiskId,
1229             debugstr_w(szVolumeLabel), debugstr_w(szDiskPrompt));
1230
1231     if (!szProduct || !squash_guid(szProduct, squished_pc))
1232         return ERROR_INVALID_PARAMETER;
1233
1234     if (dwOptions != MSICODE_PRODUCT && dwOptions != MSICODE_PATCH)
1235         return ERROR_INVALID_PARAMETER;
1236
1237     if ((szVolumeLabel && !*szVolumeLabel) || (szDiskPrompt && !*szDiskPrompt))
1238         return ERROR_INVALID_PARAMETER;
1239
1240     if ((dwContext & MSIINSTALLCONTEXT_MACHINE) && szUserSid)
1241         return ERROR_INVALID_PARAMETER;
1242
1243     if (dwOptions & MSICODE_PATCH)
1244     {
1245         FIXME("Unhandled options MSICODE_PATCH\n");
1246         return ERROR_FUNCTION_FAILED;
1247     }
1248
1249     rc = OpenSourceKey(szProduct, &sourcekey, MSICODE_PRODUCT, dwContext, FALSE);
1250     if (rc != ERROR_SUCCESS)
1251         return rc;
1252
1253     OpenMediaSubkey(sourcekey, &mediakey, TRUE);
1254
1255     sprintfW(szIndex, fmt, dwDiskId);
1256
1257     size = 2;
1258     if (szVolumeLabel) size += lstrlenW(szVolumeLabel);
1259     if (szDiskPrompt) size += lstrlenW(szDiskPrompt);
1260
1261     size *= sizeof(WCHAR);
1262     buffer = msi_alloc(size);
1263     *buffer = '\0';
1264
1265     if (szVolumeLabel) lstrcpyW(buffer, szVolumeLabel);
1266     lstrcatW(buffer, semicolon);
1267     if (szDiskPrompt) lstrcatW(buffer, szDiskPrompt);
1268
1269     RegSetValueExW(mediakey, szIndex, 0, REG_SZ, (LPBYTE)buffer, size);
1270     msi_free(buffer);
1271
1272     RegCloseKey(sourcekey);
1273     RegCloseKey(mediakey);
1274
1275     return ERROR_SUCCESS;
1276 }
1277
1278 /******************************************************************
1279  *  MsiSourceListClearAllA (MSI.@)
1280  */
1281 UINT WINAPI MsiSourceListClearAllA( LPCSTR szProduct, LPCSTR szUserName, DWORD dwReserved )
1282 {
1283     FIXME("(%s %s %d)\n", debugstr_a(szProduct), debugstr_a(szUserName), dwReserved);
1284     return ERROR_SUCCESS;
1285 }
1286
1287 /******************************************************************
1288  *  MsiSourceListClearAllW (MSI.@)
1289  */
1290 UINT WINAPI MsiSourceListClearAllW( LPCWSTR szProduct, LPCWSTR szUserName, DWORD dwReserved )
1291 {
1292     FIXME("(%s %s %d)\n", debugstr_w(szProduct), debugstr_w(szUserName), dwReserved);
1293     return ERROR_SUCCESS;
1294 }