Zebediah Figura : odbccp32: Clean up SQLGetInstalledDrivers().
Alexandre Julliard
julliard at winehq.org
Tue May 22 15:37:14 CDT 2018
Module: wine
Branch: master
Commit: 169960b676e6b6a36f8a69a626072f3f5a846bd8
URL: https://source.winehq.org/git/wine.git/?a=commit;h=169960b676e6b6a36f8a69a626072f3f5a846bd8
Author: Zebediah Figura <z.figura12 at gmail.com>
Date: Fri May 18 22:00:55 2018 -0500
odbccp32: Clean up SQLGetInstalledDrivers().
Signed-off-by: Zebediah Figura <z.figura12 at gmail.com>
Signed-off-by: Hans Leidekker <hans at codeweavers.com>
Signed-off-by: Alexandre Julliard <julliard at winehq.org>
---
dlls/odbccp32/odbccp32.c | 138 +++++++++++++++++++++++------------------------
1 file changed, 68 insertions(+), 70 deletions(-)
diff --git a/dlls/odbccp32/odbccp32.c b/dlls/odbccp32/odbccp32.c
index f9a2269..8788330 100644
--- a/dlls/odbccp32/odbccp32.c
+++ b/dlls/odbccp32/odbccp32.c
@@ -482,99 +482,97 @@ BOOL WINAPI SQLGetConfigMode(UWORD *pwConfigMode)
return TRUE;
}
-/* This is implemented sensibly rather than according to exact conformance to Microsoft's buggy implementations
- * e.g. The Microsoft one occasionally actually adds a third nul character (possibly beyond the buffer).
- * e.g. If the key has no drivers then version 3.525.1117.0 does not modify the buffer at all, not even a nul character.
- */
-BOOL WINAPI SQLGetInstalledDriversW(LPWSTR lpszBuf, WORD cbBufMax,
- WORD *pcbBufOut)
+BOOL WINAPI SQLGetInstalledDriversW(WCHAR *buf, WORD size, WORD *sizeout)
{
- HKEY hDrivers; /* Registry handle to the Drivers key */
- LONG reg_ret; /* Return code from registry functions */
- BOOL success = FALSE; /* The value we will return */
+ WORD written = 0;
+ DWORD index = 0;
+ BOOL ret = TRUE;
+ DWORD valuelen;
+ WCHAR *value;
+ HKEY drivers;
+ DWORD len;
+ LONG res;
clear_errors();
- TRACE("%p %d %p\n", lpszBuf, cbBufMax, pcbBufOut);
+ TRACE("%p %d %p\n", buf, size, sizeout);
- if (!lpszBuf || cbBufMax == 0)
+ if (!buf || !size)
{
push_error(ODBC_ERROR_INVALID_BUFF_LEN, odbc_error_invalid_buff_len);
+ return FALSE;
}
- else if ((reg_ret = RegOpenKeyExW (HKEY_LOCAL_MACHINE /* The drivers does not depend on the config mode */,
- drivers_key, 0, KEY_READ /* Maybe overkill */,
- &hDrivers)) == ERROR_SUCCESS)
+
+ res = RegOpenKeyExW(HKEY_LOCAL_MACHINE, drivers_key, 0, KEY_QUERY_VALUE, &drivers);
+ if (res)
{
- DWORD index = 0;
- cbBufMax--;
- success = TRUE;
- while (cbBufMax > 0)
- {
- DWORD size_name;
- size_name = cbBufMax;
- if ((reg_ret = RegEnumValueW(hDrivers, index, lpszBuf, &size_name, NULL, NULL, NULL, NULL)) == ERROR_SUCCESS)
- {
- index++;
- assert (size_name < cbBufMax && *(lpszBuf + size_name) == 0);
- size_name++;
- cbBufMax-= size_name;
- lpszBuf+=size_name;
- }
- else
- {
- if (reg_ret != ERROR_NO_MORE_ITEMS)
- {
- success = FALSE;
- push_error(ODBC_ERROR_GENERAL_ERR, odbc_error_general_err);
- }
- break;
- }
- }
- *lpszBuf = 0;
- if ((reg_ret = RegCloseKey (hDrivers)) != ERROR_SUCCESS)
- TRACE ("Error %d closing ODBC Drivers key\n", reg_ret);
+ push_error(ODBC_ERROR_COMPONENT_NOT_FOUND, odbc_error_component_not_found);
+ return FALSE;
}
- else
+
+ RegQueryInfoKeyW(drivers, 0, 0, 0, 0, 0, 0, 0, &valuelen, 0, 0, 0);
+ value = heap_alloc(++valuelen * sizeof(WCHAR));
+
+ size--;
+
+ while (1)
{
- /* MSDN states that it returns failure with COMPONENT_NOT_FOUND in this case.
- * Version 3.525.1117.0 (Windows 2000) does not; it actually returns success.
- * I doubt if it will actually be an issue.
- */
- push_error(ODBC_ERROR_COMPONENT_NOT_FOUND, odbc_error_component_not_found);
+ len = valuelen;
+ res = RegEnumValueW(drivers, index++, value, &len, NULL, NULL, NULL, NULL);
+ if (res == ERROR_SUCCESS)
+ {
+ lstrcpynW(buf + written, value, size - written);
+ written += min(len + 1, size - written);
+ }
+ else if (res == ERROR_NO_MORE_ITEMS)
+ break;
+ else
+ {
+ push_error(ODBC_ERROR_GENERAL_ERR, odbc_error_general_err);
+ ret = FALSE;
+ break;
+ }
}
- return success;
+
+ buf[written++] = 0;
+
+ heap_free(value);
+ RegCloseKey(drivers);
+ if (sizeout)
+ *sizeout = written;
+ return ret;
}
-BOOL WINAPI SQLGetInstalledDrivers(LPSTR lpszBuf, WORD cbBufMax,
- WORD *pcbBufOut)
+BOOL WINAPI SQLGetInstalledDrivers(char *buf, WORD size, WORD *sizeout)
{
+ WORD written;
+ WCHAR *wbuf;
BOOL ret;
- int size_wbuf = cbBufMax;
- LPWSTR wbuf;
- WORD size_used;
- TRACE("%p %d %p\n", lpszBuf, cbBufMax, pcbBufOut);
+ TRACE("%p %d %p\n", buf, size, sizeout);
- wbuf = HeapAlloc(GetProcessHeap(), 0, size_wbuf*sizeof(WCHAR));
- if (wbuf)
+ if (!buf || !size)
{
- ret = SQLGetInstalledDriversW(wbuf, size_wbuf, &size_used);
- if (ret)
- {
- if (!(ret = SQLInstall_narrow(2, lpszBuf, wbuf, size_used, cbBufMax, pcbBufOut)))
- {
- push_error(ODBC_ERROR_GENERAL_ERR, odbc_error_general_err);
- }
- }
- HeapFree(GetProcessHeap(), 0, wbuf);
- /* ignore failure; we have achieved the aim */
+ push_error(ODBC_ERROR_INVALID_BUFF_LEN, odbc_error_invalid_buff_len);
+ return FALSE;
}
- else
+
+ wbuf = heap_alloc(size * sizeof(WCHAR));
+ if (!wbuf)
{
push_error(ODBC_ERROR_OUT_OF_MEM, odbc_error_out_of_mem);
- ret = FALSE;
+ return FALSE;
}
- return ret;
+
+ ret = SQLGetInstalledDriversW(wbuf, size, &written);
+ if (!ret)
+ return FALSE;
+
+ *sizeout = WideCharToMultiByte(CP_ACP, 0, wbuf, written, NULL, 0, NULL, NULL);
+ WideCharToMultiByte(CP_ACP, 0, wbuf, written, buf, size, NULL, NULL);
+
+ heap_free(wbuf);
+ return TRUE;
}
static HKEY get_privateprofile_sectionkey(LPCWSTR section, LPCWSTR filename)
More information about the wine-cvs
mailing list