msi: Optimize MsiSourceListAddSourceEx for adding to the end of the list.
[wine.git] / dlls / msi / source.c
blob36fcc80322a218f2969b8837d509dcc1daed4f16
1 /*
2 * Implementation of the Microsoft Installer (msi.dll)
4 * Copyright 2005 Aric Stewart for CodeWeavers
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.
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.
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
21 #include <stdarg.h>
23 #define COBJMACROS
24 #define NONAMELESSUNION
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"
41 WINE_DEFAULT_DEBUG_CHANNEL(msi);
44 * These apis are defined in MSI 3.0
47 typedef struct tagMediaInfo
49 struct list entry;
50 LPWSTR path;
51 WCHAR szIndex[10];
52 DWORD index;
53 } media_info;
55 static UINT OpenSourceKey(LPCWSTR szProduct, HKEY* key, DWORD dwOptions,
56 MSIINSTALLCONTEXT context, BOOL create)
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};
62 if (context == MSIINSTALLCONTEXT_USERUNMANAGED)
64 if (dwOptions & MSICODE_PATCH)
65 rc = MSIREG_OpenUserPatchesKey(szProduct, &rootkey, create);
66 else
67 rc = MSIREG_OpenUserProductsKey(szProduct, &rootkey, create);
69 else if (context == MSIINSTALLCONTEXT_USERMANAGED)
71 if (dwOptions & MSICODE_PATCH)
72 rc = MSIREG_OpenUserPatchesKey(szProduct, &rootkey, create);
73 else
74 rc = MSIREG_OpenLocalManagedProductKey(szProduct, &rootkey, create);
76 else if (context == MSIINSTALLCONTEXT_MACHINE)
78 if (dwOptions & MSICODE_PATCH)
79 rc = MSIREG_OpenPatchesKey(szProduct, &rootkey, create);
80 else
81 rc = MSIREG_OpenLocalClassesProductKey(szProduct, &rootkey, create);
84 if (rc != ERROR_SUCCESS)
86 if (dwOptions & MSICODE_PATCH)
87 return ERROR_UNKNOWN_PATCH;
88 else
89 return ERROR_UNKNOWN_PRODUCT;
92 if (create)
93 rc = RegCreateKeyW(rootkey, szSourceList, key);
94 else
96 rc = RegOpenKeyW(rootkey,szSourceList, key);
97 if (rc != ERROR_SUCCESS)
98 rc = ERROR_BAD_CONFIGURATION;
101 return rc;
104 static UINT OpenMediaSubkey(HKEY rootkey, HKEY *key, BOOL create)
106 UINT rc;
107 static const WCHAR media[] = {'M','e','d','i','a',0};
109 if (create)
110 rc = RegCreateKeyW(rootkey, media, key);
111 else
112 rc = RegOpenKeyW(rootkey,media, key);
114 return rc;
117 static UINT OpenNetworkSubkey(HKEY rootkey, HKEY *key, BOOL create)
119 UINT rc;
120 static const WCHAR net[] = {'N','e','t',0};
122 if (create)
123 rc = RegCreateKeyW(rootkey, net, key);
124 else
125 rc = RegOpenKeyW(rootkey, net, key);
127 return rc;
130 static UINT OpenURLSubkey(HKEY rootkey, HKEY *key, BOOL create)
132 UINT rc;
133 static const WCHAR URL[] = {'U','R','L',0};
135 if (create)
136 rc = RegCreateKeyW(rootkey, URL, key);
137 else
138 rc = RegOpenKeyW(rootkey, URL, key);
140 return rc;
143 /******************************************************************
144 * MsiSourceListEnumMediaDisksA (MSI.@)
146 UINT WINAPI MsiSourceListEnumMediaDisksA(LPCSTR szProductCodeOrPatchCode,
147 LPCSTR szUserSid, MSIINSTALLCONTEXT dwContext,
148 DWORD dwOptions, DWORD dwIndex, LPWORD pdwDiskId,
149 LPSTR szVolumeLabel, LPDWORD pcchVolumeLabel,
150 LPSTR szDiskPrompt, LPDWORD pcchDiskPrompt)
152 LPWSTR product = NULL;
153 LPWSTR usersid = NULL;
154 LPWSTR volume = NULL;
155 LPWSTR prompt = NULL;
156 DWORD volumesz, promptsz;
157 UINT r = ERROR_INVALID_PARAMETER;
159 TRACE("(%s, %s, %d, %d, %d, %p, %p, %p, %p, %p)\n", debugstr_a(szProductCodeOrPatchCode),
160 debugstr_a(szUserSid), dwContext, dwOptions, dwIndex, pdwDiskId,
161 szVolumeLabel, pcchVolumeLabel, szDiskPrompt, pcchDiskPrompt);
163 if (szDiskPrompt && !pcchDiskPrompt)
164 return ERROR_INVALID_PARAMETER;
166 if (szProductCodeOrPatchCode) product = strdupAtoW(szProductCodeOrPatchCode);
167 if (szUserSid) usersid = strdupAtoW(szUserSid);
169 /* FIXME: add tests for an invalid format */
171 if (pcchVolumeLabel)
172 volume = msi_alloc(*pcchVolumeLabel * sizeof(WCHAR));
174 if (pcchDiskPrompt)
175 prompt = msi_alloc(*pcchDiskPrompt * sizeof(WCHAR));
177 if (volume) *volume = '\0';
178 if (prompt) *prompt = '\0';
179 r = MsiSourceListEnumMediaDisksW(product, usersid, dwContext, dwOptions,
180 dwIndex, pdwDiskId, volume, pcchVolumeLabel,
181 prompt, pcchDiskPrompt);
182 if (r != ERROR_SUCCESS)
183 goto done;
185 if (szVolumeLabel && pcchVolumeLabel)
186 volumesz = WideCharToMultiByte(CP_ACP, 0, volume, -1, szVolumeLabel,
187 *pcchVolumeLabel + 1, NULL, NULL);
189 if (szDiskPrompt)
190 promptsz = WideCharToMultiByte(CP_ACP, 0, prompt, -1, szDiskPrompt,
191 *pcchDiskPrompt + 1, NULL, NULL);
193 done:
194 msi_free(product);
195 msi_free(usersid);
196 msi_free(volume);
197 msi_free(prompt);
199 return r;
202 /******************************************************************
203 * MsiSourceListEnumMediaDisksW (MSI.@)
205 UINT WINAPI MsiSourceListEnumMediaDisksW(LPCWSTR szProductCodeOrPatchCode,
206 LPCWSTR szUserSid, MSIINSTALLCONTEXT dwContext,
207 DWORD dwOptions, DWORD dwIndex, LPWORD pdwDiskId,
208 LPWSTR szVolumeLabel, LPDWORD pcchVolumeLabel,
209 LPWSTR szDiskPrompt, LPDWORD pcchDiskPrompt)
211 WCHAR squished_pc[GUID_SIZE];
212 LPWSTR value = NULL;
213 LPWSTR data = NULL;
214 LPWSTR ptr;
215 HKEY source, media;
216 DWORD valuesz, datasz = 0;
217 DWORD type;
218 DWORD numvals, size;
219 LONG res;
220 UINT r;
221 static int index = 0;
223 TRACE("(%s, %s, %d, %d, %d, %p, %p, %p, %p)\n", debugstr_w(szProductCodeOrPatchCode),
224 debugstr_w(szUserSid), dwContext, dwOptions, dwIndex, szVolumeLabel,
225 pcchVolumeLabel, szDiskPrompt, pcchDiskPrompt);
227 if (!szProductCodeOrPatchCode ||
228 !squash_guid(szProductCodeOrPatchCode, squished_pc))
229 return ERROR_INVALID_PARAMETER;
231 if (dwContext == MSIINSTALLCONTEXT_MACHINE && szUserSid)
232 return ERROR_INVALID_PARAMETER;
234 if (dwOptions != MSICODE_PRODUCT && dwOptions != MSICODE_PATCH)
235 return ERROR_INVALID_PARAMETER;
237 if (szDiskPrompt && !pcchDiskPrompt)
238 return ERROR_INVALID_PARAMETER;
240 if (dwIndex == 0)
241 index = 0;
243 if (dwIndex != index)
244 return ERROR_INVALID_PARAMETER;
246 r = OpenSourceKey(szProductCodeOrPatchCode, &source,
247 dwOptions, dwContext, FALSE);
248 if (r != ERROR_SUCCESS)
249 return r;
251 r = OpenMediaSubkey(source, &media, FALSE);
252 if (r != ERROR_SUCCESS)
254 RegCloseKey(source);
255 return ERROR_NO_MORE_ITEMS;
258 if (!pcchVolumeLabel && !pcchDiskPrompt)
260 r = RegEnumValueW(media, dwIndex, NULL, NULL, NULL,
261 &type, NULL, NULL);
262 goto done;
265 res = RegQueryInfoKeyW(media, NULL, NULL, NULL, NULL, NULL,
266 NULL, &numvals, &valuesz, &datasz, NULL, NULL);
267 if (res != ERROR_SUCCESS)
269 r = ERROR_BAD_CONFIGURATION;
270 goto done;
273 value = msi_alloc(++valuesz * sizeof(WCHAR));
274 data = msi_alloc(++datasz * sizeof(WCHAR));
275 if (!value || !data)
277 r = ERROR_OUTOFMEMORY;
278 goto done;
281 r = RegEnumValueW(media, dwIndex, value, &valuesz,
282 NULL, &type, (LPBYTE)data, &datasz);
283 if (r != ERROR_SUCCESS)
284 goto done;
286 if (pdwDiskId)
287 *pdwDiskId = atolW(value);
289 ptr = strchrW(data, ';');
290 if (!ptr)
291 ptr = data;
292 else
293 *ptr = '\0';
295 if (pcchVolumeLabel)
297 size = lstrlenW(data);
298 if (size >= *pcchVolumeLabel)
299 r = ERROR_MORE_DATA;
300 else if (szVolumeLabel)
301 lstrcpyW(szVolumeLabel, data);
303 *pcchVolumeLabel = size;
306 if (pcchDiskPrompt)
308 data = ptr;
309 if (!*data)
310 data++;
312 size = lstrlenW(data);
313 if (size >= *pcchDiskPrompt)
314 r = ERROR_MORE_DATA;
315 else if (szDiskPrompt)
316 lstrcpyW(szDiskPrompt, data);
318 *pcchDiskPrompt = size;
321 index++;
323 done:
324 msi_free(value);
325 msi_free(data);
326 RegCloseKey(source);
328 return r;
331 /******************************************************************
332 * MsiSourceListEnumSourcesA (MSI.@)
334 UINT WINAPI MsiSourceListEnumSourcesA(LPCSTR szProductCodeOrPatch, LPCSTR szUserSid,
335 MSIINSTALLCONTEXT dwContext,
336 DWORD dwOptions, DWORD dwIndex,
337 LPSTR szSource, LPDWORD pcchSource)
339 LPWSTR product = NULL;
340 LPWSTR usersid = NULL;
341 LPWSTR source = NULL;
342 DWORD len = 0;
343 UINT r = ERROR_INVALID_PARAMETER;
344 static int index = 0;
346 TRACE("(%s, %s, %d, %d, %d, %p, %p)\n", debugstr_a(szProductCodeOrPatch),
347 debugstr_a(szUserSid), dwContext, dwOptions, dwIndex, szSource, pcchSource);
349 if (dwIndex == 0)
350 index = 0;
352 if (szSource && !pcchSource)
353 goto done;
355 if (dwIndex != index)
356 goto done;
358 if (szProductCodeOrPatch) product = strdupAtoW(szProductCodeOrPatch);
359 if (szUserSid) usersid = strdupAtoW(szUserSid);
361 r = MsiSourceListEnumSourcesW(product, usersid, dwContext, dwOptions,
362 dwIndex, NULL, &len);
363 if (r != ERROR_SUCCESS)
364 goto done;
366 source = msi_alloc(++len * sizeof(WCHAR));
367 if (!source)
369 r = ERROR_OUTOFMEMORY;
370 goto done;
373 *source = '\0';
374 r = MsiSourceListEnumSourcesW(product, usersid, dwContext, dwOptions,
375 dwIndex, source, &len);
376 if (r != ERROR_SUCCESS)
377 goto done;
379 len = WideCharToMultiByte(CP_ACP, 0, source, -1, NULL, 0, NULL, NULL);
380 if (pcchSource && *pcchSource >= len)
381 WideCharToMultiByte(CP_ACP, 0, source, -1, szSource, len, NULL, NULL);
382 else if (szSource)
383 r = ERROR_MORE_DATA;
385 if (pcchSource)
386 *pcchSource = len - 1;
388 done:
389 msi_free(product);
390 msi_free(usersid);
391 msi_free(source);
393 if (r == ERROR_SUCCESS)
395 if (szSource || !pcchSource) index++;
397 else if (dwIndex > index)
398 index = 0;
400 return r;
403 /******************************************************************
404 * MsiSourceListEnumSourcesW (MSI.@)
406 UINT WINAPI MsiSourceListEnumSourcesW(LPCWSTR szProductCodeOrPatch, LPCWSTR szUserSid,
407 MSIINSTALLCONTEXT dwContext,
408 DWORD dwOptions, DWORD dwIndex,
409 LPWSTR szSource, LPDWORD pcchSource)
411 WCHAR squished_pc[GUID_SIZE];
412 WCHAR name[32];
413 HKEY source = NULL;
414 HKEY subkey = NULL;
415 LONG res;
416 UINT r = ERROR_INVALID_PARAMETER;
417 static int index = 0;
419 static const WCHAR format[] = {'%','d',0};
421 TRACE("(%s, %s, %d, %d, %d, %p, %p)\n", debugstr_w(szProductCodeOrPatch),
422 debugstr_w(szUserSid), dwContext, dwOptions, dwIndex, szSource, pcchSource);
424 if (dwIndex == 0)
425 index = 0;
427 if (!szProductCodeOrPatch || !squash_guid(szProductCodeOrPatch, squished_pc))
428 goto done;
430 if (szSource && !pcchSource)
431 goto done;
433 if (!(dwOptions & (MSISOURCETYPE_NETWORK | MSISOURCETYPE_URL)))
434 goto done;
436 if ((dwOptions & MSISOURCETYPE_NETWORK) && (dwOptions & MSISOURCETYPE_URL))
437 goto done;
439 if (dwContext == MSIINSTALLCONTEXT_MACHINE && szUserSid)
440 goto done;
442 if (dwIndex != index)
443 goto done;
445 r = OpenSourceKey(szProductCodeOrPatch, &source,
446 dwOptions, dwContext, FALSE);
447 if (r != ERROR_SUCCESS)
448 goto done;
450 if (dwOptions & MSISOURCETYPE_NETWORK)
451 r = OpenNetworkSubkey(source, &subkey, FALSE);
452 else if (dwOptions & MSISOURCETYPE_URL)
453 r = OpenURLSubkey(source, &subkey, FALSE);
455 if (r != ERROR_SUCCESS)
457 r = ERROR_NO_MORE_ITEMS;
458 goto done;
461 sprintfW(name, format, dwIndex + 1);
463 res = RegQueryValueExW(subkey, name, 0, 0, (LPBYTE)szSource, pcchSource);
464 if (res != ERROR_SUCCESS && res != ERROR_MORE_DATA)
465 r = ERROR_NO_MORE_ITEMS;
467 done:
468 RegCloseKey(subkey);
469 RegCloseKey(source);
471 if (r == ERROR_SUCCESS)
473 if (szSource || !pcchSource) index++;
475 else if (dwIndex > index)
476 index = 0;
478 return r;
481 /******************************************************************
482 * MsiSourceListGetInfoA (MSI.@)
484 UINT WINAPI MsiSourceListGetInfoA( LPCSTR szProduct, LPCSTR szUserSid,
485 MSIINSTALLCONTEXT dwContext, DWORD dwOptions,
486 LPCSTR szProperty, LPSTR szValue,
487 LPDWORD pcchValue)
489 UINT ret;
490 LPWSTR product = NULL;
491 LPWSTR usersid = NULL;
492 LPWSTR property = NULL;
493 LPWSTR value = NULL;
494 DWORD len = 0;
496 if (szValue && !pcchValue)
497 return ERROR_INVALID_PARAMETER;
499 if (szProduct) product = strdupAtoW(szProduct);
500 if (szUserSid) usersid = strdupAtoW(szUserSid);
501 if (szProperty) property = strdupAtoW(szProperty);
503 ret = MsiSourceListGetInfoW(product, usersid, dwContext, dwOptions,
504 property, NULL, &len);
505 if (ret != ERROR_SUCCESS)
506 goto done;
508 value = msi_alloc(++len * sizeof(WCHAR));
509 if (!value)
510 return ERROR_OUTOFMEMORY;
512 *value = '\0';
513 ret = MsiSourceListGetInfoW(product, usersid, dwContext, dwOptions,
514 property, value, &len);
515 if (ret != ERROR_SUCCESS)
516 goto done;
518 len = WideCharToMultiByte(CP_ACP, 0, value, -1, NULL, 0, NULL, NULL);
519 if (*pcchValue >= len)
520 WideCharToMultiByte(CP_ACP, 0, value, -1, szValue, len, NULL, NULL);
521 else if (szValue)
522 ret = ERROR_MORE_DATA;
524 *pcchValue = len - 1;
526 done:
527 msi_free(product);
528 msi_free(usersid);
529 msi_free(property);
530 msi_free(value);
531 return ret;
534 /******************************************************************
535 * MsiSourceListGetInfoW (MSI.@)
537 UINT WINAPI MsiSourceListGetInfoW( LPCWSTR szProduct, LPCWSTR szUserSid,
538 MSIINSTALLCONTEXT dwContext, DWORD dwOptions,
539 LPCWSTR szProperty, LPWSTR szValue,
540 LPDWORD pcchValue)
542 WCHAR squished_pc[GUID_SIZE];
543 HKEY sourcekey, media;
544 LPWSTR source, ptr;
545 DWORD size;
546 UINT rc;
548 static const WCHAR mediapack[] = {
549 'M','e','d','i','a','P','a','c','k','a','g','e',0};
551 TRACE("%s %s\n", debugstr_w(szProduct), debugstr_w(szProperty));
553 if (!szProduct || !squash_guid(szProduct, squished_pc))
554 return ERROR_INVALID_PARAMETER;
556 if (szValue && !pcchValue)
557 return ERROR_INVALID_PARAMETER;
559 if (dwContext != MSIINSTALLCONTEXT_USERMANAGED &&
560 dwContext != MSIINSTALLCONTEXT_USERUNMANAGED &&
561 dwContext != MSIINSTALLCONTEXT_MACHINE)
562 return ERROR_INVALID_PARAMETER;
564 if (!szProperty)
565 return ERROR_INVALID_PARAMETER;
567 if (szUserSid)
568 FIXME("Unhandled UserSid %s\n",debugstr_w(szUserSid));
570 if (dwContext != MSIINSTALLCONTEXT_USERUNMANAGED)
571 FIXME("Unhandled context %d\n", dwContext);
573 rc = OpenSourceKey(szProduct, &sourcekey, dwOptions, dwContext, FALSE);
574 if (rc != ERROR_SUCCESS)
575 return rc;
577 if (!lstrcmpW(szProperty, INSTALLPROPERTY_MEDIAPACKAGEPATHW) ||
578 !lstrcmpW(szProperty, INSTALLPROPERTY_DISKPROMPTW))
580 rc = OpenMediaSubkey(sourcekey, &media, FALSE);
581 if (rc != ERROR_SUCCESS)
583 RegCloseKey(sourcekey);
584 return ERROR_SUCCESS;
587 if (!lstrcmpW(szProperty, INSTALLPROPERTY_MEDIAPACKAGEPATHW))
588 szProperty = mediapack;
590 RegQueryValueExW(media, szProperty, 0, 0, (LPBYTE)szValue, pcchValue);
591 RegCloseKey(media);
593 else if (!lstrcmpW(szProperty, INSTALLPROPERTY_LASTUSEDSOURCEW) ||
594 !lstrcmpW(szProperty, INSTALLPROPERTY_LASTUSEDTYPEW))
596 rc = RegQueryValueExW(sourcekey, INSTALLPROPERTY_LASTUSEDSOURCEW,
597 0, 0, NULL, &size);
598 if (rc != ERROR_SUCCESS)
600 RegCloseKey(sourcekey);
601 return ERROR_SUCCESS;
604 source = msi_alloc(size);
605 RegQueryValueExW(sourcekey, INSTALLPROPERTY_LASTUSEDSOURCEW,
606 0, 0, (LPBYTE)source, &size);
608 if (!*source)
610 msi_free(source);
611 RegCloseKey(sourcekey);
612 return ERROR_SUCCESS;
615 if (!lstrcmpW(szProperty, INSTALLPROPERTY_LASTUSEDTYPEW))
617 if (*source != 'n' && *source != 'u' && *source != 'm')
619 msi_free(source);
620 RegCloseKey(sourcekey);
621 return ERROR_SUCCESS;
624 ptr = source;
625 source[1] = '\0';
627 else
629 ptr = strrchrW(source, ';');
630 if (!ptr)
631 ptr = source;
632 else
633 ptr++;
636 if (szValue)
638 if (lstrlenW(ptr) < *pcchValue)
639 lstrcpyW(szValue, ptr);
640 else
641 rc = ERROR_MORE_DATA;
644 *pcchValue = lstrlenW(ptr);
645 msi_free(source);
647 else if (strcmpW(INSTALLPROPERTY_PACKAGENAMEW, szProperty)==0)
649 *pcchValue = *pcchValue * sizeof(WCHAR);
650 rc = RegQueryValueExW(sourcekey, INSTALLPROPERTY_PACKAGENAMEW, 0, 0,
651 (LPBYTE)szValue, pcchValue);
652 if (rc != ERROR_SUCCESS && rc != ERROR_MORE_DATA)
654 *pcchValue = 0;
655 rc = ERROR_SUCCESS;
657 else
659 if (*pcchValue)
660 *pcchValue = (*pcchValue - 1) / sizeof(WCHAR);
661 if (szValue)
662 szValue[*pcchValue] = '\0';
665 else
667 FIXME("Unknown property %s\n",debugstr_w(szProperty));
668 rc = ERROR_UNKNOWN_PROPERTY;
671 RegCloseKey(sourcekey);
672 return rc;
675 /******************************************************************
676 * MsiSourceListSetInfoA (MSI.@)
678 UINT WINAPI MsiSourceListSetInfoA(LPCSTR szProduct, LPCSTR szUserSid,
679 MSIINSTALLCONTEXT dwContext, DWORD dwOptions,
680 LPCSTR szProperty, LPCSTR szValue)
682 UINT ret;
683 LPWSTR product = NULL;
684 LPWSTR usersid = NULL;
685 LPWSTR property = NULL;
686 LPWSTR value = NULL;
688 if (szProduct) product = strdupAtoW(szProduct);
689 if (szUserSid) usersid = strdupAtoW(szUserSid);
690 if (szProperty) property = strdupAtoW(szProperty);
691 if (szValue) value = strdupAtoW(szValue);
693 ret = MsiSourceListSetInfoW(product, usersid, dwContext, dwOptions,
694 property, value);
696 msi_free(product);
697 msi_free(usersid);
698 msi_free(property);
699 msi_free(value);
701 return ret;
704 UINT msi_set_last_used_source(LPCWSTR product, LPCWSTR usersid,
705 MSIINSTALLCONTEXT context, DWORD options,
706 LPCWSTR value)
708 HKEY source;
709 LPWSTR buffer;
710 WCHAR typechar;
711 DWORD size;
712 UINT r;
713 int index = 1;
715 static const WCHAR format[] = {'%','c',';','%','i',';','%','s',0};
717 if (options & MSISOURCETYPE_NETWORK)
718 typechar = 'n';
719 else if (options & MSISOURCETYPE_URL)
720 typechar = 'u';
721 else if (options & MSISOURCETYPE_MEDIA)
722 typechar = 'm';
723 else
724 return ERROR_INVALID_PARAMETER;
726 if (!(options & MSISOURCETYPE_MEDIA))
728 r = MsiSourceListAddSourceExW(product, usersid, context,
729 options, value, 0);
730 if (r != ERROR_SUCCESS)
731 return r;
733 index = 0;
734 while ((r = MsiSourceListEnumSourcesW(product, usersid, context, options,
735 index, NULL, NULL)) == ERROR_SUCCESS)
736 index++;
738 if (r != ERROR_NO_MORE_ITEMS)
739 return r;
742 size = (lstrlenW(format) + lstrlenW(value) + 7) * sizeof(WCHAR);
743 buffer = msi_alloc(size);
744 if (!buffer)
745 return ERROR_OUTOFMEMORY;
747 r = OpenSourceKey(product, &source, MSICODE_PRODUCT, context, FALSE);
748 if (r != ERROR_SUCCESS)
749 return r;
751 sprintfW(buffer, format, typechar, index, value);
753 size = (lstrlenW(buffer) + 1) * sizeof(WCHAR);
754 r = RegSetValueExW(source, INSTALLPROPERTY_LASTUSEDSOURCEW, 0,
755 REG_SZ, (LPBYTE)buffer, size);
756 msi_free(buffer);
758 RegCloseKey(source);
759 return r;
762 /******************************************************************
763 * MsiSourceListSetInfoW (MSI.@)
765 UINT WINAPI MsiSourceListSetInfoW( LPCWSTR szProduct, LPCWSTR szUserSid,
766 MSIINSTALLCONTEXT dwContext, DWORD dwOptions,
767 LPCWSTR szProperty, LPCWSTR szValue)
769 WCHAR squished_pc[GUID_SIZE];
770 HKEY sourcekey, media;
771 LPCWSTR property;
772 UINT rc;
774 static const WCHAR media_package[] = {
775 'M','e','d','i','a','P','a','c','k','a','g','e',0
778 TRACE("%s %s %x %x %s %s\n", debugstr_w(szProduct), debugstr_w(szUserSid),
779 dwContext, dwOptions, debugstr_w(szProperty), debugstr_w(szValue));
781 if (!szProduct || !squash_guid(szProduct, squished_pc))
782 return ERROR_INVALID_PARAMETER;
784 if (!szProperty)
785 return ERROR_INVALID_PARAMETER;
787 if (!szValue)
788 return ERROR_UNKNOWN_PROPERTY;
790 if (dwContext == MSIINSTALLCONTEXT_MACHINE && szUserSid)
791 return ERROR_INVALID_PARAMETER;
793 if (dwOptions & MSICODE_PATCH)
795 FIXME("Unhandled options MSICODE_PATCH\n");
796 return ERROR_UNKNOWN_PATCH;
799 property = szProperty;
800 if (!lstrcmpW(szProperty, INSTALLPROPERTY_MEDIAPACKAGEPATHW))
801 property = media_package;
803 rc = OpenSourceKey(szProduct, &sourcekey, MSICODE_PRODUCT, dwContext, FALSE);
804 if (rc != ERROR_SUCCESS)
805 return rc;
807 if (lstrcmpW(szProperty, INSTALLPROPERTY_LASTUSEDSOURCEW) &&
808 dwOptions & (MSISOURCETYPE_NETWORK | MSISOURCETYPE_URL))
810 RegCloseKey(sourcekey);
811 return ERROR_INVALID_PARAMETER;
814 if (!lstrcmpW(szProperty, INSTALLPROPERTY_MEDIAPACKAGEPATHW) ||
815 !lstrcmpW(szProperty, INSTALLPROPERTY_DISKPROMPTW))
817 rc = OpenMediaSubkey(sourcekey, &media, TRUE);
818 if (rc == ERROR_SUCCESS)
820 rc = msi_reg_set_val_str(media, property, szValue);
821 RegCloseKey(media);
824 else if (strcmpW(INSTALLPROPERTY_PACKAGENAMEW, szProperty)==0)
826 DWORD size = lstrlenW(szValue)*sizeof(WCHAR);
827 rc = RegSetValueExW(sourcekey, INSTALLPROPERTY_PACKAGENAMEW, 0,
828 REG_SZ, (const BYTE *)szValue, size);
829 if (rc != ERROR_SUCCESS)
830 rc = ERROR_UNKNOWN_PROPERTY;
832 else if (!lstrcmpW(szProperty, INSTALLPROPERTY_LASTUSEDSOURCEW))
834 if (!(dwOptions & (MSISOURCETYPE_NETWORK | MSISOURCETYPE_URL)))
835 rc = ERROR_INVALID_PARAMETER;
836 else
837 rc = msi_set_last_used_source(szProduct, szUserSid, dwContext,
838 dwOptions, szValue);
840 else
841 rc = ERROR_UNKNOWN_PROPERTY;
843 RegCloseKey(sourcekey);
844 return rc;
847 /******************************************************************
848 * MsiSourceListAddSourceW (MSI.@)
850 UINT WINAPI MsiSourceListAddSourceW( LPCWSTR szProduct, LPCWSTR szUserName,
851 DWORD dwReserved, LPCWSTR szSource)
853 INT ret;
854 LPWSTR sidstr = NULL;
855 DWORD sidsize = 0;
856 DWORD domsize = 0;
858 TRACE("%s %s %s\n", debugstr_w(szProduct), debugstr_w(szUserName), debugstr_w(szSource));
860 if (LookupAccountNameW(NULL, szUserName, NULL, &sidsize, NULL, &domsize, NULL))
862 PSID psid = msi_alloc(sidsize);
864 if (LookupAccountNameW(NULL, szUserName, psid, &sidsize, NULL, &domsize, NULL))
865 ConvertSidToStringSidW(psid, &sidstr);
867 msi_free(psid);
870 ret = MsiSourceListAddSourceExW(szProduct, sidstr,
871 MSIINSTALLCONTEXT_USERMANAGED, MSISOURCETYPE_NETWORK, szSource, 0);
873 if (sidstr)
874 LocalFree(sidstr);
876 return ret;
879 /******************************************************************
880 * MsiSourceListAddSourceA (MSI.@)
882 UINT WINAPI MsiSourceListAddSourceA( LPCSTR szProduct, LPCSTR szUserName,
883 DWORD dwReserved, LPCSTR szSource)
885 INT ret;
886 LPWSTR szwproduct;
887 LPWSTR szwusername;
888 LPWSTR szwsource;
890 szwproduct = strdupAtoW( szProduct );
891 szwusername = strdupAtoW( szUserName );
892 szwsource = strdupAtoW( szSource );
894 ret = MsiSourceListAddSourceW(szwproduct, szwusername, 0, szwsource);
896 msi_free(szwproduct);
897 msi_free(szwusername);
898 msi_free(szwsource);
900 return ret;
903 /******************************************************************
904 * MsiSourceListAddSourceExA (MSI.@)
906 UINT WINAPI MsiSourceListAddSourceExA(LPCSTR szProduct, LPCSTR szUserSid,
907 MSIINSTALLCONTEXT dwContext, DWORD dwOptions, LPCSTR szSource, DWORD dwIndex)
909 UINT ret;
910 LPWSTR product, usersid, source;
912 product = strdupAtoW(szProduct);
913 usersid = strdupAtoW(szUserSid);
914 source = strdupAtoW(szSource);
916 ret = MsiSourceListAddSourceExW(product, usersid, dwContext,
917 dwOptions, source, dwIndex);
919 msi_free(product);
920 msi_free(usersid);
921 msi_free(source);
923 return ret;
926 static void free_source_list(struct list *sourcelist)
928 while (!list_empty(sourcelist))
930 media_info *info = LIST_ENTRY(list_head(sourcelist), media_info, entry);
931 list_remove(&info->entry);
932 msi_free(info->path);
933 msi_free(info);
937 static void add_source_to_list(struct list *sourcelist, media_info *info,
938 DWORD *index)
940 media_info *iter;
941 BOOL found = FALSE;
942 static const WCHAR fmt[] = {'%','i',0};
944 if (index) *index = 0;
946 if (list_empty(sourcelist))
948 list_add_head(sourcelist, &info->entry);
949 return;
952 LIST_FOR_EACH_ENTRY(iter, sourcelist, media_info, entry)
954 if (!found && info->index < iter->index)
956 found = TRUE;
957 list_add_before(&iter->entry, &info->entry);
960 /* update the rest of the list */
961 if (found)
962 sprintfW(iter->szIndex, fmt, ++iter->index);
963 else if (index)
964 (*index)++;
967 if (!found)
968 list_add_after(&iter->entry, &info->entry);
971 static UINT fill_source_list(struct list *sourcelist, HKEY sourcekey, DWORD *count)
973 UINT r = ERROR_SUCCESS;
974 DWORD index = 0;
975 WCHAR name[10];
976 DWORD size, val_size;
977 media_info *entry;
979 *count = 0;
981 while (r == ERROR_SUCCESS)
983 size = sizeof(name) / sizeof(name[0]);
984 r = RegEnumValueW(sourcekey, index, name, &size, NULL, NULL, NULL, &val_size);
985 if (r != ERROR_SUCCESS)
986 return r;
988 entry = msi_alloc(sizeof(media_info));
989 if (!entry)
990 goto error;
992 entry->path = msi_alloc(val_size);
993 if (!entry->path)
995 msi_free(entry);
996 goto error;
999 lstrcpyW(entry->szIndex, name);
1000 entry->index = atoiW(name);
1002 size++;
1003 r = RegEnumValueW(sourcekey, index, name, &size, NULL,
1004 NULL, (LPBYTE)entry->path, &val_size);
1005 if (r != ERROR_SUCCESS)
1007 msi_free(entry->path);
1008 msi_free(entry);
1009 goto error;
1012 index = ++(*count);
1013 add_source_to_list(sourcelist, entry, NULL);
1016 error:
1017 *count = -1;
1018 free_source_list(sourcelist);
1019 return ERROR_OUTOFMEMORY;
1022 /******************************************************************
1023 * MsiSourceListAddSourceExW (MSI.@)
1025 UINT WINAPI MsiSourceListAddSourceExW( LPCWSTR szProduct, LPCWSTR szUserSid,
1026 MSIINSTALLCONTEXT dwContext, DWORD dwOptions, LPCWSTR szSource,
1027 DWORD dwIndex)
1029 HKEY sourcekey;
1030 HKEY typekey;
1031 UINT rc;
1032 struct list sourcelist;
1033 media_info *info;
1034 WCHAR squished_pc[GUID_SIZE];
1035 WCHAR name[10];
1036 LPWSTR source;
1037 LPCWSTR postfix;
1038 DWORD size, count;
1039 DWORD index;
1041 static const WCHAR fmt[] = {'%','i',0};
1042 static const WCHAR one[] = {'1',0};
1043 static const WCHAR backslash[] = {'\\',0};
1044 static const WCHAR forwardslash[] = {'/',0};
1046 TRACE("%s %s %x %x %s %i\n", debugstr_w(szProduct), debugstr_w(szUserSid),
1047 dwContext, dwOptions, debugstr_w(szSource), dwIndex);
1049 if (!szProduct || !squash_guid(szProduct, squished_pc))
1050 return ERROR_INVALID_PARAMETER;
1052 if (!szSource || !*szSource)
1053 return ERROR_INVALID_PARAMETER;
1055 if (!(dwOptions & (MSISOURCETYPE_NETWORK | MSISOURCETYPE_URL)))
1056 return ERROR_INVALID_PARAMETER;
1058 if (dwOptions & MSICODE_PATCH)
1060 FIXME("Unhandled options MSICODE_PATCH\n");
1061 return ERROR_FUNCTION_FAILED;
1064 if (szUserSid && (dwContext & MSIINSTALLCONTEXT_MACHINE))
1065 return ERROR_INVALID_PARAMETER;
1067 rc = OpenSourceKey(szProduct, &sourcekey, MSICODE_PRODUCT, dwContext, FALSE);
1068 if (rc != ERROR_SUCCESS)
1069 return rc;
1071 if (dwOptions & MSISOURCETYPE_NETWORK)
1072 rc = OpenNetworkSubkey(sourcekey, &typekey, TRUE);
1073 else if (dwOptions & MSISOURCETYPE_URL)
1074 rc = OpenURLSubkey(sourcekey, &typekey, TRUE);
1075 else if (dwOptions & MSISOURCETYPE_MEDIA)
1076 rc = OpenMediaSubkey(sourcekey, &typekey, TRUE);
1077 else
1079 ERR("unknown media type: %08x\n", dwOptions);
1080 RegCloseKey(sourcekey);
1081 return ERROR_FUNCTION_FAILED;
1084 postfix = (dwOptions & MSISOURCETYPE_NETWORK) ? backslash : forwardslash;
1085 if (szSource[lstrlenW(szSource) - 1] == *postfix)
1086 source = strdupW(szSource);
1087 else
1089 size = lstrlenW(szSource) + 2;
1090 source = msi_alloc(size * sizeof(WCHAR));
1091 lstrcpyW(source, szSource);
1092 lstrcatW(source, postfix);
1095 list_init(&sourcelist);
1096 rc = fill_source_list(&sourcelist, typekey, &count);
1097 if (rc != ERROR_NO_MORE_ITEMS)
1098 return rc;
1100 size = (lstrlenW(source) + 1) * sizeof(WCHAR);
1102 if (count == 0)
1104 rc = RegSetValueExW(typekey, one, 0, REG_EXPAND_SZ, (LPBYTE)source, size);
1105 goto done;
1107 else if (dwIndex > count || dwIndex == 0)
1109 sprintfW(name, fmt, count + 1);
1110 rc = RegSetValueExW(typekey, name, 0, REG_EXPAND_SZ, (LPBYTE)source, size);
1111 goto done;
1113 else
1115 sprintfW(name, fmt, dwIndex);
1116 info = msi_alloc(sizeof(media_info));
1117 if (!info)
1119 rc = ERROR_OUTOFMEMORY;
1120 goto done;
1123 info->path = strdupW(source);
1124 lstrcpyW(info->szIndex, name);
1125 info->index = dwIndex;
1126 add_source_to_list(&sourcelist, info, &index);
1128 LIST_FOR_EACH_ENTRY(info, &sourcelist, media_info, entry)
1130 if (info->index < index)
1131 continue;
1133 size = (lstrlenW(info->path) + 1) * sizeof(WCHAR);
1134 rc = RegSetValueExW(typekey, info->szIndex, 0,
1135 REG_EXPAND_SZ, (LPBYTE)info->path, size);
1136 if (rc != ERROR_SUCCESS)
1137 goto done;
1141 done:
1142 free_source_list(&sourcelist);
1143 msi_free(source);
1144 RegCloseKey(typekey);
1145 RegCloseKey(sourcekey);
1146 return rc;
1149 /******************************************************************
1150 * MsiSourceListAddMediaDiskA (MSI.@)
1152 UINT WINAPI MsiSourceListAddMediaDiskA(LPCSTR szProduct, LPCSTR szUserSid,
1153 MSIINSTALLCONTEXT dwContext, DWORD dwOptions, DWORD dwDiskId,
1154 LPCSTR szVolumeLabel, LPCSTR szDiskPrompt)
1156 UINT r;
1157 LPWSTR product = NULL;
1158 LPWSTR usersid = NULL;
1159 LPWSTR volume = NULL;
1160 LPWSTR prompt = NULL;
1162 if (szProduct) product = strdupAtoW(szProduct);
1163 if (szUserSid) usersid = strdupAtoW(szUserSid);
1164 if (szVolumeLabel) volume = strdupAtoW(szVolumeLabel);
1165 if (szDiskPrompt) prompt = strdupAtoW(szDiskPrompt);
1167 r = MsiSourceListAddMediaDiskW(product, usersid, dwContext, dwOptions,
1168 dwDiskId, volume, prompt);
1170 msi_free(product);
1171 msi_free(usersid);
1172 msi_free(volume);
1173 msi_free(prompt);
1175 return r;
1178 /******************************************************************
1179 * MsiSourceListAddMediaDiskW (MSI.@)
1181 UINT WINAPI MsiSourceListAddMediaDiskW(LPCWSTR szProduct, LPCWSTR szUserSid,
1182 MSIINSTALLCONTEXT dwContext, DWORD dwOptions, DWORD dwDiskId,
1183 LPCWSTR szVolumeLabel, LPCWSTR szDiskPrompt)
1185 HKEY sourcekey;
1186 HKEY mediakey;
1187 UINT rc;
1188 WCHAR szIndex[10];
1189 WCHAR squished_pc[GUID_SIZE];
1190 LPWSTR buffer;
1191 DWORD size;
1193 static const WCHAR fmt[] = {'%','i',0};
1194 static const WCHAR semicolon[] = {';',0};
1196 TRACE("%s %s %x %x %i %s %s\n", debugstr_w(szProduct),
1197 debugstr_w(szUserSid), dwContext, dwOptions, dwDiskId,
1198 debugstr_w(szVolumeLabel), debugstr_w(szDiskPrompt));
1200 if (!szProduct || !squash_guid(szProduct, squished_pc))
1201 return ERROR_INVALID_PARAMETER;
1203 if (dwOptions != MSICODE_PRODUCT && dwOptions != MSICODE_PATCH)
1204 return ERROR_INVALID_PARAMETER;
1206 if ((szVolumeLabel && !*szVolumeLabel) || (szDiskPrompt && !*szDiskPrompt))
1207 return ERROR_INVALID_PARAMETER;
1209 if ((dwContext & MSIINSTALLCONTEXT_MACHINE) && szUserSid)
1210 return ERROR_INVALID_PARAMETER;
1212 if (dwOptions & MSICODE_PATCH)
1214 FIXME("Unhandled options MSICODE_PATCH\n");
1215 return ERROR_FUNCTION_FAILED;
1218 rc = OpenSourceKey(szProduct, &sourcekey, MSICODE_PRODUCT, dwContext, FALSE);
1219 if (rc != ERROR_SUCCESS)
1220 return rc;
1222 OpenMediaSubkey(sourcekey, &mediakey, TRUE);
1224 sprintfW(szIndex, fmt, dwDiskId);
1226 size = 2;
1227 if (szVolumeLabel) size += lstrlenW(szVolumeLabel);
1228 if (szDiskPrompt) size += lstrlenW(szDiskPrompt);
1230 size *= sizeof(WCHAR);
1231 buffer = msi_alloc(size);
1232 *buffer = '\0';
1234 if (szVolumeLabel) lstrcpyW(buffer, szVolumeLabel);
1235 lstrcatW(buffer, semicolon);
1236 if (szDiskPrompt) lstrcatW(buffer, szDiskPrompt);
1238 RegSetValueExW(mediakey, szIndex, 0, REG_SZ, (LPBYTE)buffer, size);
1239 msi_free(buffer);
1241 RegCloseKey(sourcekey);
1242 RegCloseKey(mediakey);
1244 return ERROR_SUCCESS;
1247 /******************************************************************
1248 * MsiSourceListClearAllA (MSI.@)
1250 UINT WINAPI MsiSourceListClearAllA( LPCSTR szProduct, LPCSTR szUserName, DWORD dwReserved )
1252 FIXME("(%s %s %d)\n", debugstr_a(szProduct), debugstr_a(szUserName), dwReserved);
1253 return ERROR_SUCCESS;
1256 /******************************************************************
1257 * MsiSourceListClearAllW (MSI.@)
1259 UINT WINAPI MsiSourceListClearAllW( LPCWSTR szProduct, LPCWSTR szUserName, DWORD dwReserved )
1261 FIXME("(%s %s %d)\n", debugstr_w(szProduct), debugstr_w(szUserName), dwReserved);
1262 return ERROR_SUCCESS;