[PATCH v2 1/1] odbccp32: SQLGet/WritePrivateProfileStringW support driver config

[Alistair Leslie-Hughes]

From: Alistair Leslie-Hughes

---
 dlls/odbccp32/odbccp32.c   | 14 ++++++++------
 dlls/odbccp32/tests/misc.c |  5 +++++
 2 files changed, 13 insertions(+), 6 deletions(-)

diff --git a/dlls/odbccp32/odbccp32.c b/dlls/odbccp32/odbccp32.c
index 90a7517b957..fd02391a384 100644
--- a/dlls/odbccp32/odbccp32.c
+++ b/dlls/odbccp32/odbccp32.c
@@ -734,10 +734,11 @@ int WINAPI SQLGetPrivateProfileStringW(LPCWSTR section, LPCWSTR entry,
     if (!defvalue || !buff)
         return 0;

-    if (config_mode == ODBC_USER_DSN)
-        sectionkey = get_privateprofile_sectionkey(HKEY_CURRENT_USER, section, filename);
-    else if (config_mode == ODBC_SYSTEM_DSN)
+    /* odbcinit.ini is only for drivers, so default to local Machine */
+    if (!wcsicmp(filename, L"ODBCINST.INI") || config_mode == ODBC_SYSTEM_DSN)
         sectionkey = get_privateprofile_sectionkey(HKEY_LOCAL_MACHINE, section, filename);
+    else if (config_mode == ODBC_USER_DSN)
+        sectionkey = get_privateprofile_sectionkey(HKEY_CURRENT_USER, section, filename);
     else
     {
         sectionkey = get_privateprofile_sectionkey(HKEY_CURRENT_USER, section, filename);
@@ -1812,10 +1813,11 @@ BOOL WINAPI SQLWritePrivateProfileStringW(LPCWSTR lpszSection, LPCWSTR lpszEntry
         return FALSE;
     }

-    if (config_mode == ODBC_USER_DSN)
-        ret = RegCreateKeyW(HKEY_CURRENT_USER, L"Software\ODBC", &hkey);
-    else if (config_mode == ODBC_SYSTEM_DSN)
+    /* odbcinit.ini is only for drivers, so default to local Machine */
+    if (!wcsicmp(lpszFilename, L"ODBCINST.INI") || config_mode == ODBC_SYSTEM_DSN)
         ret = RegCreateKeyW(HKEY_LOCAL_MACHINE, L"Software\ODBC", &hkey);
+    else if (config_mode == ODBC_USER_DSN)
+        ret = RegCreateKeyW(HKEY_CURRENT_USER, L"Software\ODBC", &hkey);
     else
     {
         ret = RegCreateKeyW(HKEY_CURRENT_USER, L"Software\ODBC", &hkey);
diff --git a/dlls/odbccp32/tests/misc.c b/dlls/odbccp32/tests/misc.c
index 81c66a44a24..0b982642177 100644
--- a/dlls/odbccp32/tests/misc.c
+++ b/dlls/odbccp32/tests/misc.c
@@ -317,6 +317,11 @@ static void test_SQLGetPrivateProfileStringW(void)
     WCHAR buffer[256] = {0};
     LONG reg_ret;

+    /* Get Driver Information - Currently under wine we dont have any default drivers. */
+    ret = SQLGetPrivateProfileStringW(L"SQL Server", L"SQLlevel" , L"99", buffer, 256, L"ODBCINST.INI");
+    todo_wine ok(ret == 1, "SQLGetPrivateProfileStringW returned %d\n", ret);
+    todo_wine ok(!lstrcmpW(buffer, L"1"), "incorrect string '%s'\n", wine_dbgstr_w(buffer));
+
     lstrcpyW(buffer, L"wine");
     ret = SQLGetPrivateProfileStringW(NULL, L"testing", L"default", buffer, 256, L"ODBC.INI");
     ok(ret == 0, "SQLGetPrivateProfileStringW returned %d\n", ret);