James Hawkins : msi: Implement MsiSourceListEnumMediaDisks.
Alexandre Julliard
julliard at winehq.org
Tue Feb 26 05:45:14 CST 2008
Module: wine
Branch: master
Commit: 1bd49bcf7486ece38d8e804108cb28d04defe0f6
URL: http://source.winehq.org/git/wine.git/?a=commit;h=1bd49bcf7486ece38d8e804108cb28d04defe0f6
Author: James Hawkins <jhawkins at codeweavers.com>
Date: Mon Feb 25 22:07:42 2008 -0600
msi: Implement MsiSourceListEnumMediaDisks.
---
dlls/msi/msi.spec | 4 +-
dlls/msi/source.c | 188 +++++++++++++++++++++++++++++++++++++++++++++++++++++
include/msi.h | 6 ++
3 files changed, 196 insertions(+), 2 deletions(-)
diff --git a/dlls/msi/msi.spec b/dlls/msi/msi.spec
index b510de7..598e2f3 100644
--- a/dlls/msi/msi.spec
+++ b/dlls/msi/msi.spec
@@ -264,8 +264,8 @@
268 stdcall MsiSourceListSetInfoW(wstr wstr long long wstr wstr)
269 stub MsiEnumPatchesExA
270 stub MsiEnumPatchesExW
-271 stub MsiSourceListEnumMediaDisksA
-272 stub MsiSourceListEnumMediaDisksW
+271 stdcall MsiSourceListEnumMediaDisksA(str str long long long ptr ptr ptr ptr ptr)
+272 stdcall MsiSourceListEnumMediaDisksW(wstr wstr long long long ptr ptr ptr ptr ptr)
273 stdcall MsiSourceListAddMediaDiskA(str str long long long str str)
274 stdcall MsiSourceListAddMediaDiskW(wstr wstr long long long wstr wstr)
275 stub MsiSourceListClearMediaDiskA
diff --git a/dlls/msi/source.c b/dlls/msi/source.c
index 716414f..b508569 100644
--- a/dlls/msi/source.c
+++ b/dlls/msi/source.c
@@ -141,6 +141,194 @@ static UINT OpenURLSubkey(HKEY rootkey, HKEY *key, BOOL create)
}
/******************************************************************
+ * MsiSourceListEnumMediaDisksA (MSI.@)
+ */
+UINT WINAPI MsiSourceListEnumMediaDisksA(LPCSTR szProductCodeOrPatchCode,
+ LPCSTR szUserSid, MSIINSTALLCONTEXT dwContext,
+ DWORD dwOptions, DWORD dwIndex, LPWORD pdwDiskId,
+ LPSTR szVolumeLabel, LPDWORD pcchVolumeLabel,
+ LPSTR szDiskPrompt, LPDWORD pcchDiskPrompt)
+{
+ LPWSTR product = NULL;
+ LPWSTR usersid = NULL;
+ LPWSTR volume = NULL;
+ LPWSTR prompt = NULL;
+ DWORD volumesz, promptsz;
+ UINT r = ERROR_INVALID_PARAMETER;
+
+ TRACE("(%s, %s, %d, %d, %d, %p, %p, %p, %p, %p)\n", debugstr_a(szProductCodeOrPatchCode),
+ debugstr_a(szUserSid), dwContext, dwOptions, dwIndex, pdwDiskId,
+ szVolumeLabel, pcchVolumeLabel, szDiskPrompt, pcchDiskPrompt);
+
+ if (szDiskPrompt && !pcchDiskPrompt)
+ return ERROR_INVALID_PARAMETER;
+
+ if (szProductCodeOrPatchCode) product = strdupAtoW(szProductCodeOrPatchCode);
+ if (szUserSid) usersid = strdupAtoW(szUserSid);
+
+ /* FIXME: add tests for an invalid format */
+
+ if (pcchVolumeLabel)
+ volume = msi_alloc(*pcchVolumeLabel * sizeof(WCHAR));
+
+ if (pcchDiskPrompt)
+ prompt = msi_alloc(*pcchDiskPrompt * sizeof(WCHAR));
+
+ if (volume) *volume = '\0';
+ if (prompt) *prompt = '\0';
+ r = MsiSourceListEnumMediaDisksW(product, usersid, dwContext, dwOptions,
+ dwIndex, pdwDiskId, volume, pcchVolumeLabel,
+ prompt, pcchDiskPrompt);
+ if (r != ERROR_SUCCESS)
+ goto done;
+
+ if (szVolumeLabel && pcchVolumeLabel)
+ volumesz = WideCharToMultiByte(CP_ACP, 0, volume, -1, szVolumeLabel,
+ *pcchVolumeLabel + 1, NULL, NULL);
+
+ if (szDiskPrompt)
+ promptsz = WideCharToMultiByte(CP_ACP, 0, prompt, -1, szDiskPrompt,
+ *pcchDiskPrompt + 1, NULL, NULL);
+
+done:
+ msi_free(product);
+ msi_free(usersid);
+ msi_free(volume);
+ msi_free(prompt);
+
+ return r;
+}
+
+/******************************************************************
+ * MsiSourceListEnumMediaDisksW (MSI.@)
+ */
+UINT WINAPI MsiSourceListEnumMediaDisksW(LPCWSTR szProductCodeOrPatchCode,
+ LPCWSTR szUserSid, MSIINSTALLCONTEXT dwContext,
+ DWORD dwOptions, DWORD dwIndex, LPWORD pdwDiskId,
+ LPWSTR szVolumeLabel, LPDWORD pcchVolumeLabel,
+ LPWSTR szDiskPrompt, LPDWORD pcchDiskPrompt)
+{
+ WCHAR squished_pc[GUID_SIZE];
+ LPWSTR value = NULL;
+ LPWSTR data = NULL;
+ LPWSTR ptr;
+ HKEY source, media;
+ DWORD valuesz, datasz = 0;
+ DWORD type;
+ DWORD numvals, size;
+ LONG res;
+ UINT r;
+ static int index = 0;
+
+ TRACE("(%s, %s, %d, %d, %d, %p, %p, %p, %p)\n", debugstr_w(szProductCodeOrPatchCode),
+ debugstr_w(szUserSid), dwContext, dwOptions, dwIndex, szVolumeLabel,
+ pcchVolumeLabel, szDiskPrompt, pcchDiskPrompt);
+
+ if (!szProductCodeOrPatchCode ||
+ !squash_guid(szProductCodeOrPatchCode, squished_pc))
+ return ERROR_INVALID_PARAMETER;
+
+ if (dwContext == MSIINSTALLCONTEXT_MACHINE && szUserSid)
+ return ERROR_INVALID_PARAMETER;
+
+ if (dwOptions != MSICODE_PRODUCT && dwOptions != MSICODE_PATCH)
+ return ERROR_INVALID_PARAMETER;
+
+ if (szDiskPrompt && !pcchDiskPrompt)
+ return ERROR_INVALID_PARAMETER;
+
+ if (dwIndex == 0)
+ index = 0;
+
+ if (dwIndex != index)
+ return ERROR_INVALID_PARAMETER;
+
+ r = OpenSourceKey(szProductCodeOrPatchCode, &source,
+ dwOptions, dwContext, FALSE);
+ if (r != ERROR_SUCCESS)
+ return r;
+
+ r = OpenMediaSubkey(source, &media, FALSE);
+ if (r != ERROR_SUCCESS)
+ {
+ RegCloseKey(source);
+ return ERROR_NO_MORE_ITEMS;
+ }
+
+ if (!pcchVolumeLabel && !pcchDiskPrompt)
+ {
+ r = RegEnumValueW(media, dwIndex, NULL, NULL, NULL,
+ &type, NULL, NULL);
+ goto done;
+ }
+
+ res = RegQueryInfoKeyW(media, NULL, NULL, NULL, NULL, NULL,
+ NULL, &numvals, &valuesz, &datasz, NULL, NULL);
+ if (res != ERROR_SUCCESS)
+ {
+ r = ERROR_BAD_CONFIGURATION;
+ goto done;
+ }
+
+ value = msi_alloc(++valuesz * sizeof(WCHAR));
+ data = msi_alloc(++datasz * sizeof(WCHAR));
+ if (!value || !data)
+ {
+ r = ERROR_OUTOFMEMORY;
+ goto done;
+ }
+
+ r = RegEnumValueW(media, dwIndex, value, &valuesz,
+ NULL, &type, (LPBYTE)data, &datasz);
+ if (r != ERROR_SUCCESS)
+ goto done;
+
+ if (pdwDiskId)
+ *pdwDiskId = atolW(value);
+
+ ptr = strchrW(data, ';');
+ if (!ptr)
+ ptr = data;
+ else
+ *ptr = '\0';
+
+ if (pcchVolumeLabel)
+ {
+ size = lstrlenW(data);
+ if (size >= *pcchVolumeLabel)
+ r = ERROR_MORE_DATA;
+ else if (szVolumeLabel)
+ lstrcpyW(szVolumeLabel, data);
+
+ *pcchVolumeLabel = size;
+ }
+
+ if (pcchDiskPrompt)
+ {
+ data = ptr;
+ if (!*data)
+ data++;
+
+ size = lstrlenW(data);
+ if (size >= *pcchDiskPrompt)
+ r = ERROR_MORE_DATA;
+ else if (szDiskPrompt)
+ lstrcpyW(szDiskPrompt, data);
+
+ *pcchDiskPrompt = size;
+ }
+
+ index++;
+
+done:
+ msi_free(value);
+ msi_free(data);
+ RegCloseKey(source);
+
+ return r;
+}
+
+/******************************************************************
* MsiSourceListEnumSourcesA (MSI.@)
*/
UINT WINAPI MsiSourceListEnumSourcesA(LPCSTR szProductCodeOrPatch, LPCSTR szUserSid,
diff --git a/include/msi.h b/include/msi.h
index 26d07fb..93c4c28 100644
--- a/include/msi.h
+++ b/include/msi.h
@@ -537,6 +537,12 @@ UINT WINAPI MsiSourceListAddSourceA(LPCSTR, LPCSTR, DWORD, LPCSTR);
UINT WINAPI MsiSourceListAddSourceW(LPCWSTR, LPCWSTR, DWORD, LPCWSTR);
#define MsiSourceListAddSource WINELIB_NAME_AW(MsiSourceListAddSource)
+UINT WINAPI MsiSourceListEnumMediaDisksA(LPCSTR, LPCSTR, MSIINSTALLCONTEXT, DWORD, DWORD, LPWORD,
+ LPSTR, LPDWORD, LPSTR, LPDWORD);
+UINT WINAPI MsiSourceListEnumMediaDisksW(LPCWSTR, LPCWSTR, MSIINSTALLCONTEXT, DWORD, DWORD, LPWORD,
+ LPWSTR, LPDWORD, LPWSTR, LPDWORD);
+#define MsiSourceListEnumMediaDisks WINELIB_NAME_AW(MsiSourceListEnumMediaDisks)
+
UINT WINAPI MsiSourceListEnumSourcesA(LPCSTR, LPCSTR, MSIINSTALLCONTEXT, DWORD, DWORD, LPSTR, LPDWORD);
UINT WINAPI MsiSourceListEnumSourcesW(LPCWSTR, LPCWSTR, MSIINSTALLCONTEXT, DWORD, DWORD, LPWSTR, LPDWORD);
#define MsiSourceListEnumSources WINELIB_NAME_AW(MsiSourceListEnumSources)
More information about the wine-cvs
mailing list