[PATCH v2] odbccp32: Implement SQLConfigDataSource/W (resend)

[Alistair Leslie-Hughes]

Signed-off-by: Alistair Leslie-Hughes <leslie_alistair at hotmail.com>
---
 dlls/odbccp32/odbccp32.c | 155 +++++++++++++++++++++++++++++++++++++++--------
 1 file changed, 131 insertions(+), 24 deletions(-)

diff --git a/dlls/odbccp32/odbccp32.c b/dlls/odbccp32/odbccp32.c
index 694a90c..4ad64c4 100644
--- a/dlls/odbccp32/odbccp32.c
+++ b/dlls/odbccp32/odbccp32.c
@@ -102,6 +102,32 @@ static inline WCHAR *heap_strdupAtoW(const char *str)
     return ret;
 }
 
+static inline WCHAR *heap_attrdupAtoW(const char *str)
+{
+    WCHAR *ret = NULL;
+
+    if (str)
+    {
+        DWORD size = 0, len;
+        const char *p;
+        WCHAR *current;
+
+        for (p = str; *p; p += lstrlenA(p) + 1)
+        {
+            len = MultiByteToWideChar(CP_ACP, 0, p, -1, NULL, 0);
+            size += len;
+        }
+
+        current = ret = heap_alloc_zero( (size+1)*sizeof(WCHAR));
+        for (p = str; *p; p += lstrlenA(p) + 1)
+        {
+            len = MultiByteToWideChar(CP_ACP, 0, p, -1, current, size);
+            current += len;
+        }
+    }
+    return ret;
+}
+
 
 BOOL WINAPI ODBCCPlApplet( LONG i, LONG j, LONG * p1, LONG * p2)
 {
@@ -220,30 +246,6 @@ static BOOL SQLInstall_narrow(int mode, LPSTR buffer, LPCWSTR str, WORD str_leng
     return success;
 }
 
-BOOL WINAPI SQLConfigDataSourceW(HWND hwndParent, WORD fRequest,
-               LPCWSTR lpszDriver, LPCWSTR lpszAttributes)
-{
-    LPCWSTR p;
-
-    clear_errors();
-    FIXME("%p %d %s %s\n", hwndParent, fRequest, debugstr_w(lpszDriver),
-          debugstr_w(lpszAttributes));
-
-    for (p = lpszAttributes; *p; p += lstrlenW(p) + 1)
-        FIXME("%s\n", debugstr_w(p));
-
-    return TRUE;
-}
-
-BOOL WINAPI SQLConfigDataSource(HWND hwndParent, WORD fRequest,
-               LPCSTR lpszDriver, LPCSTR lpszAttributes)
-{
-    FIXME("%p %d %s %s\n", hwndParent, fRequest, debugstr_a(lpszDriver),
-          debugstr_a(lpszAttributes));
-    clear_errors();
-    return TRUE;
-}
-
 static HMODULE load_config_driver(const WCHAR *driver)
 {
     static WCHAR reg_driver[] = {'d','r','i','v','e','r',0};
@@ -360,6 +362,111 @@ fail:
     return FALSE;
 }
 
+static BOOL call_ConfigDSN(HANDLE mod, HWND hwnd, WORD request, const char *driverA, const char *attributesA, const WCHAR *driverW, const WCHAR *attributesW)
+{
+    BOOL (WINAPI *pConfigDSN)(HWND hwnd, WORD request, const char *driver, const char *attr);
+    BOOL (WINAPI *pConfigDSNW)(HWND hwnd, WORD request, const WCHAR *driver, const WCHAR *attr);
+    BOOL ret = FALSE;
+
+    TRACE("%p, %p, %d, %p, %p, %p %p\n", mod, hwnd, request, driverA, attributesA, driverW, attributesW);
+
+    pConfigDSN = (void*)GetProcAddress(mod, "ConfigDSN");
+    pConfigDSNW = (void*)GetProcAddress(mod, "ConfigDSNW");
+
+    if (pConfigDSNW)
+    {
+        const WCHAR *attr = attributesW;
+        if(!attr)
+            attr = heap_attrdupAtoW(attributesA);
+
+        TRACE("Calling ConfigDSNW (%p)\n", pConfigDSNW);
+
+        ret = pConfigDSNW(hwnd, request, driverW, attr);
+
+        if(attr != attributesW)
+            heap_free((WCHAR *)attr);
+    }
+    else if (pConfigDSN)
+    {
+        TRACE("Calling pConfigDSN (%p)\n", pConfigDSN);
+        ret = pConfigDSN(hwnd, request, driverA, attributesA);
+    }
+    else
+        ERR("Neither ConfigDSN or ConfigDSNW found.\n");
+
+    if (!ret)
+        push_error(ODBC_ERROR_REQUEST_FAILED, odbc_error_request_failed);
+
+    return ret;
+}
+
+BOOL WINAPI SQLConfigDataSourceW(HWND hwnd, WORD request,
+               LPCWSTR driver, LPCWSTR attributes)
+{
+    HMODULE hmod;
+    BOOL ret = FALSE;
+
+    TRACE("%p, %d, %s, %s\n", hwnd, request, debugstr_w(driver), debugstr_w(attributes));
+    if (TRACE_ON(odbc))
+    {
+        const WCHAR *p;
+        for (p = attributes; *p; p += lstrlenW(p) + 1)
+            TRACE("%s\n", debugstr_w(p));
+    }
+
+    clear_errors();
+
+    hmod = load_config_driver(driver);
+    if (!hmod)
+        return FALSE;
+
+    ret = call_ConfigDSN (hmod, hwnd, request, NULL, NULL, driver, attributes);
+
+    FreeLibrary(hmod);
+
+    return ret;
+}
+
+BOOL WINAPI SQLConfigDataSource(HWND hwnd, WORD request,
+               LPCSTR driver, LPCSTR attributes)
+{
+    HMODULE hmod;
+    BOOL ret = FALSE;
+    WCHAR *driverW;
+
+    TRACE("%p, %d, %s, %s\n", hwnd, request, debugstr_a(driver), debugstr_a(attributes));
+
+    if (TRACE_ON(odbc))
+    {
+        const char *p;
+        for (p = attributes; *p; p += lstrlenA(p) + 1)
+            TRACE("%s\n", debugstr_a(p));
+    }
+
+    clear_errors();
+
+    driverW = heap_strdupAtoW(driver);
+    if (!driverW)
+    {
+        push_error(ODBC_ERROR_OUT_OF_MEM, odbc_error_out_of_mem);
+        return FALSE;
+    }
+
+    hmod = load_config_driver(driverW);
+    if (!hmod)
+    {
+        heap_free(driverW);
+        return FALSE;
+    }
+
+    ret = call_ConfigDSN (hmod, hwnd, request, driver, attributes, driverW, NULL);
+    heap_free(driverW);
+
+    FreeLibrary(hmod);
+
+    return ret;
+}
+
 BOOL WINAPI SQLConfigDriverW(HWND hwnd, WORD request, LPCWSTR driver,
                LPCWSTR args, LPWSTR msg, WORD msgmax, WORD *msgout)
 {
-- 
1.9.1