dlls/msdasql/msdasql_main.c

   1 /*
   2  * Copyright 2019 Alistair Leslie-Hughes
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Lesser General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2.1 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Lesser General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser General Public
  15  * License along with this library; if not, write to the Free Software
  16  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  17  */
  18
  19 #define COBJMACROS
  20
  21 #include <stdarg.h>
  22
  23 #include "windef.h"
  24 #include "winbase.h"
  25 #include "objbase.h"
  26 #include "oledb.h"
  27 #include "rpcproxy.h"
  28 #include "wine/debug.h"
  29 #include "oledberr.h"
  30
  31 #include "initguid.h"
  32 #include "msdasql.h"
  33 #include "odbcinst.h"
  34 #include "sqlext.h"
  35 #include "sqlucode.h"
  36
  37 #include "msdasql_private.h"
  38
  39 WINE_DEFAULT_DEBUG_CHANNEL(msdasql);
  40
  41 DEFINE_GUID(DBPROPSET_DATASOURCEINFO, 0xc8b522bb, 0x5cf3, 0x11ce, 0xad, 0xe5, 0x00, 0xaa, 0x00, 0x44, 0x77, 0x3d);
  42 DEFINE_GUID(DBPROPSET_DBINIT,    0xc8b522bc, 0x5cf3, 0x11ce, 0xad, 0xe5, 0x00, 0xaa, 0x00, 0x44, 0x77, 0x3d);
  43 DEFINE_GUID(DBPROPSET_ROWSET,    0xc8b522be, 0x5cf3, 0x11ce, 0xad, 0xe5, 0x00, 0xaa, 0x00, 0x44, 0x77, 0x3d);
  44
  45 DEFINE_GUID(DBGUID_DEFAULT,      0xc8b521fb, 0x5cf3, 0x11ce, 0xad, 0xe5, 0x00, 0xaa, 0x00, 0x44, 0x77, 0x3d);
  46
  47 void dump_sql_diag_records(SQLSMALLINT type, SQLHANDLE handle)
  48 {
  49     SQLCHAR state[6], msg[SQL_MAX_MESSAGE_LENGTH];
  50     SQLINTEGER native;
  51     SQLSMALLINT i = 1, len;
  52
  53     if (!TRACE_ON(msdasql))
  54         return;
  55
  56     while(SQLGetDiagRec(type, handle, i, state, &native, msg, sizeof(msg), &len) == SQL_SUCCESS)
  57     {
  58         WARN("%d: %s: %s\n", i, state, msg);
  59         i++;
  60     }
  61 }
  62
  63 static HRESULT WINAPI ClassFactory_QueryInterface(IClassFactory *iface, REFIID riid, void **ppv)
  64 {
  65     *ppv = NULL;
  66
  67     if(IsEqualGUID(&IID_IUnknown, riid)) {
  68         TRACE("(%p)->(IID_IUnknown %p)\n", iface, ppv);
  69         *ppv = iface;
  70     }else if(IsEqualGUID(&IID_IClassFactory, riid)) {
  71         TRACE("(%p)->(IID_IClassFactory %p)\n", iface, ppv);
  72         *ppv = iface;
  73     }
  74
  75     if(*ppv) {
  76         IUnknown_AddRef((IUnknown*)*ppv);
  77         return S_OK;
  78     }
  79
  80     WARN("(%p)->(%s %p)\n", iface, debugstr_guid(riid), ppv);
  81     return E_NOINTERFACE;
  82 }
  83
  84 static ULONG WINAPI ClassFactory_AddRef(IClassFactory *iface)
  85 {
  86     TRACE("(%p)\n", iface);
  87     return 2;
  88 }
  89
  90 static ULONG WINAPI ClassFactory_Release(IClassFactory *iface)
  91 {
  92     TRACE("(%p)\n", iface);
  93     return 1;
  94 }
  95
  96 static HRESULT create_msdasql_provider(REFIID riid, void **ppv);
  97 static HRESULT create_msdasql_enumerator(REFIID riid, void **ppv);
  98
  99 HRESULT WINAPI msdasql_CreateInstance(IClassFactory *iface, IUnknown *outer, REFIID riid, void **ppv)
 100 {
 101     TRACE("(%p %s %p)\n", outer, debugstr_guid(riid), ppv);
 102
 103     return create_msdasql_provider(riid, ppv);
 104 }
 105
 106 static HRESULT WINAPI ClassFactory_LockServer(IClassFactory *iface, BOOL fLock)
 107 {
 108     TRACE("(%p)->(%x)\n", iface, fLock);
 109     return S_OK;
 110 }
 111
 112 static const IClassFactoryVtbl cfmsdasqlVtbl = {
 113     ClassFactory_QueryInterface,
 114     ClassFactory_AddRef,
 115     ClassFactory_Release,
 116     msdasql_CreateInstance,
 117     ClassFactory_LockServer
 118 };
 119
 120 HRESULT WINAPI enumerationcf_CreateInstance(IClassFactory *iface, IUnknown *outer, REFIID riid, void **ppv)
 121 {
 122     TRACE("(%p %s %p)\n", outer, debugstr_guid(riid), ppv);
 123
 124     return create_msdasql_enumerator(riid, ppv);
 125 }
 126
 127 static const IClassFactoryVtbl enumfactoryVtbl =
 128 {
 129     ClassFactory_QueryInterface,
 130     ClassFactory_AddRef,
 131     ClassFactory_Release,
 132     enumerationcf_CreateInstance,
 133     ClassFactory_LockServer
 134 };
 135
 136 static IClassFactory cfmsdasql = { &cfmsdasqlVtbl };
 137 static IClassFactory enumfactory = { &enumfactoryVtbl };
 138
 139 HRESULT WINAPI DllGetClassObject( REFCLSID rclsid, REFIID riid, void **ppv )
 140 {
 141     TRACE("%s %s %p\n", debugstr_guid(rclsid), debugstr_guid(riid), ppv);
 142
 143     if (IsEqualGUID(&CLSID_MSDASQL, rclsid))
 144         return IClassFactory_QueryInterface(&cfmsdasql, riid, ppv);
 145     else if (IsEqualGUID(&CLSID_MSDASQL_ENUMERATOR, rclsid))
 146         return IClassFactory_QueryInterface(&enumfactory, riid, ppv);
 147
 148     return CLASS_E_CLASSNOTAVAILABLE;
 149 }
 150
 151 struct dbproperty
 152 {
 153     const WCHAR *name;
 154     DBPROPID id;
 155     DBPROPOPTIONS options;
 156     VARTYPE type;
 157     HRESULT (*convert_dbproperty)(const WCHAR *src, VARIANT *dest);
 158 };
 159
 160 struct mode_propval
 161 {
 162     const WCHAR *name;
 163     DWORD value;
 164 };
 165
 166 static int __cdecl dbmodeprop_compare(const void *a, const void *b)
 167 {
 168     const WCHAR *src = a;
 169     const struct mode_propval *propval = b;
 170     return wcsicmp(src, propval->name);
 171 }
 172
 173 static HRESULT convert_dbproperty_mode(const WCHAR *src, VARIANT *dest)
 174 {
 175     struct mode_propval mode_propvals[] =
 176     {
 177         { L"Read", DB_MODE_READ },
 178         { L"ReadWrite", DB_MODE_READWRITE },
 179         { L"Share Deny None", DB_MODE_SHARE_DENY_NONE },
 180         { L"Share Deny Read", DB_MODE_SHARE_DENY_READ },
 181         { L"Share Deny Write", DB_MODE_SHARE_DENY_WRITE },
 182         { L"Share Exclusive", DB_MODE_SHARE_EXCLUSIVE },
 183         { L"Write", DB_MODE_WRITE },
 184     };
 185     struct mode_propval *prop;
 186
 187     if ((prop = bsearch(src, mode_propvals, ARRAY_SIZE(mode_propvals),
 188                         sizeof(struct mode_propval), dbmodeprop_compare)))
 189     {
 190         V_VT(dest) = VT_I4;
 191         V_I4(dest) = prop->value;
 192         TRACE("%s = %#lx\n", debugstr_w(src), prop->value);
 193         return S_OK;
 194     }
 195
 196     return E_FAIL;
 197 }
 198
 199 static const struct dbproperty dbproperties[] =
 200 {
 201     { L"Password",                 DBPROP_AUTH_PASSWORD,                   DBPROPOPTIONS_OPTIONAL, VT_BSTR },
 202     { L"Persist Security Info",    DBPROP_AUTH_PERSIST_SENSITIVE_AUTHINFO, DBPROPOPTIONS_OPTIONAL, VT_BOOL },
 203     { L"User ID",                  DBPROP_AUTH_USERID,                     DBPROPOPTIONS_OPTIONAL, VT_BSTR },
 204     { L"Data Source",              DBPROP_INIT_DATASOURCE,                 DBPROPOPTIONS_REQUIRED, VT_BSTR },
 205     { L"Window Handle",            DBPROP_INIT_HWND,                       DBPROPOPTIONS_OPTIONAL, sizeof(void *) == 8 ? VT_I8 : VT_I4 },
 206     { L"Location",                 DBPROP_INIT_LOCATION,                   DBPROPOPTIONS_OPTIONAL, VT_BSTR },
 207     { L"Mode",                     DBPROP_INIT_MODE,                       DBPROPOPTIONS_OPTIONAL, VT_I4, convert_dbproperty_mode },
 208     { L"Prompt",                   DBPROP_INIT_PROMPT,                     DBPROPOPTIONS_OPTIONAL, VT_I2 },
 209     { L"Connect Timeout",          DBPROP_INIT_TIMEOUT,                    DBPROPOPTIONS_OPTIONAL, VT_I4 },
 210     { L"Extended Properties",      DBPROP_INIT_PROVIDERSTRING,             DBPROPOPTIONS_REQUIRED, VT_BSTR },
 211     { L"Locale Identifier",        DBPROP_INIT_LCID,                       DBPROPOPTIONS_OPTIONAL, VT_I4 },
 212     { L"Initial Catalog",          DBPROP_INIT_CATALOG,                    DBPROPOPTIONS_OPTIONAL, VT_BSTR },
 213     { L"OLE DB Services",          DBPROP_INIT_OLEDBSERVICES,              DBPROPOPTIONS_OPTIONAL, VT_I4 },
 214     { L"General Timeout",          DBPROP_INIT_GENERALTIMEOUT,             DBPROPOPTIONS_OPTIONAL, VT_I4 },
 215 };
 216
 217 struct msdasql_prop
 218 {
 219     VARTYPE id;
 220     VARIANT value;
 221 };
 222
 223 struct msdasql
 224 {
 225     IUnknown         MSDASQL_iface;
 226     IDBProperties    IDBProperties_iface;
 227     IDBInitialize    IDBInitialize_iface;
 228     IDBCreateSession IDBCreateSession_iface;
 229     IPersist         IPersist_iface;
 230
 231     LONG     ref;
 232     struct msdasql_prop properties[14];
 233
 234     /* ODBC Support */
 235     HENV henv;
 236     HDBC hdbc;
 237 };
 238
 239 static inline struct msdasql *impl_from_IUnknown(IUnknown *iface)
 240 {
 241     return CONTAINING_RECORD(iface, struct msdasql, MSDASQL_iface);
 242 }
 243
 244 static inline struct msdasql *impl_from_IDBProperties(IDBProperties *iface)
 245 {
 246     return CONTAINING_RECORD(iface, struct msdasql, IDBProperties_iface);
 247 }
 248
 249 static inline struct msdasql *impl_from_IDBInitialize(IDBInitialize *iface)
 250 {
 251     return CONTAINING_RECORD(iface, struct msdasql, IDBInitialize_iface);
 252 }
 253
 254 static inline struct msdasql *impl_from_IDBCreateSession(IDBCreateSession *iface)
 255 {
 256     return CONTAINING_RECORD(iface, struct msdasql, IDBCreateSession_iface);
 257 }
 258
 259 static inline struct msdasql *impl_from_IPersist( IPersist *iface )
 260 {
 261     return CONTAINING_RECORD( iface, struct msdasql, IPersist_iface );
 262 }
 263
 264 static HRESULT WINAPI msdsql_QueryInterface(IUnknown *iface, REFIID riid, void **out)
 265 {
 266     struct msdasql *provider = impl_from_IUnknown(iface);
 267
 268     TRACE("(%p)->(%s %p)\n", provider, debugstr_guid(riid), out);
 269
 270     if(IsEqualGUID(riid, &IID_IUnknown) ||
 271        IsEqualGUID(riid, &CLSID_MSDASQL))
 272     {
 273         *out = &provider->MSDASQL_iface;
 274     }
 275     else if(IsEqualGUID(riid, &IID_IDBProperties))
 276     {
 277         *out = &provider->IDBProperties_iface;
 278     }
 279     else if ( IsEqualGUID(riid, &IID_IDBInitialize))
 280     {
 281         *out = &provider->IDBInitialize_iface;
 282     }
 283     else if (IsEqualGUID(riid, &IID_IDBCreateSession))
 284     {
 285         *out = &provider->IDBCreateSession_iface;
 286     }
 287     else if(IsEqualGUID(&IID_IPersist, riid))
 288     {
 289         *out = &provider->IPersist_iface;
 290     }
 291     else
 292     {
 293         FIXME("(%s, %p)\n", debugstr_guid(riid), out);
 294         *out = NULL;
 295         return E_NOINTERFACE;
 296     }
 297
 298     IUnknown_AddRef((IUnknown*)*out);
 299     return S_OK;
 300 }
 301
 302 static ULONG WINAPI msdsql_AddRef(IUnknown *iface)
 303 {
 304     struct msdasql *provider = impl_from_IUnknown(iface);
 305     ULONG ref = InterlockedIncrement(&provider->ref);
 306
 307     TRACE("(%p) ref=%lu\n", provider, ref);
 308
 309     return ref;
 310 }
 311
 312 static ULONG  WINAPI msdsql_Release(IUnknown *iface)
 313 {
 314     struct msdasql *provider = impl_from_IUnknown(iface);
 315     ULONG ref = InterlockedDecrement(&provider->ref);
 316
 317     TRACE("(%p) ref=%lu\n", provider, ref);
 318
 319     if (!ref)
 320     {
 321         SQLDisconnect(provider->hdbc);
 322
 323         SQLFreeHandle(SQL_HANDLE_DBC, provider->hdbc);
 324         SQLFreeHandle(SQL_HANDLE_ENV, provider->henv);
 325         free(provider);
 326     }
 327
 328     return ref;
 329 }
 330
 331 static const IUnknownVtbl msdsql_vtbl =
 332 {
 333     msdsql_QueryInterface,
 334     msdsql_AddRef,
 335     msdsql_Release
 336 };
 337
 338 static HRESULT WINAPI dbprops_QueryInterface(IDBProperties *iface, REFIID riid, void **ppvObject)
 339 {
 340     struct msdasql *provider = impl_from_IDBProperties(iface);
 341
 342     return IUnknown_QueryInterface(&provider->MSDASQL_iface, riid, ppvObject);
 343 }
 344
 345 static ULONG WINAPI dbprops_AddRef(IDBProperties *iface)
 346 {
 347     struct msdasql *provider = impl_from_IDBProperties(iface);
 348
 349     return IUnknown_AddRef(&provider->MSDASQL_iface);
 350 }
 351
 352 static ULONG WINAPI dbprops_Release(IDBProperties *iface)
 353 {
 354     struct msdasql *provider = impl_from_IDBProperties(iface);
 355
 356     return IUnknown_Release(&provider->MSDASQL_iface);
 357 }
 358
 359 static HRESULT WINAPI dbprops_GetProperties(IDBProperties *iface, ULONG cPropertyIDSets,
 360             const DBPROPIDSET rgPropertyIDSets[], ULONG *pcPropertySets, DBPROPSET **prgPropertySets)
 361 {
 362     struct msdasql *provider = impl_from_IDBProperties(iface);
 363     int i, j, k;
 364     DBPROPSET *propset;
 365
 366     TRACE("(%p)->(%ld %p %p %p)\n", provider, cPropertyIDSets, rgPropertyIDSets, pcPropertySets, prgPropertySets);
 367
 368     *pcPropertySets = 1;
 369
 370     if (cPropertyIDSets != 1)
 371     {
 372         FIXME("Currently only 1 property set supported.\n");
 373         cPropertyIDSets = 1;
 374     }
 375
 376     propset = CoTaskMemAlloc(cPropertyIDSets * sizeof(DBPROPSET));
 377
 378     if (IsEqualGUID(&rgPropertyIDSets[0].guidPropertySet, &DBPROPSET_DATASOURCEINFO))
 379     {
 380         TRACE("Propertyset DBPROPSET_DATASOURCEINFO not supported\n");
 381         propset->guidPropertySet = rgPropertyIDSets[0].guidPropertySet;
 382         propset->cProperties = rgPropertyIDSets[0].cPropertyIDs;
 383
 384         propset->rgProperties = CoTaskMemAlloc(propset->cProperties * sizeof(DBPROP));
 385
 386         for (j=0; j < propset->cProperties; j++)
 387         {
 388             propset->rgProperties[j].dwPropertyID = rgPropertyIDSets[0].rgPropertyIDs[j];
 389             propset->rgProperties[j].dwStatus = DBPROPSTATUS_NOTSUPPORTED;
 390         }
 391
 392         *prgPropertySets = propset;
 393
 394         return DB_E_ERRORSOCCURRED;
 395     }
 396
 397     propset->guidPropertySet = DBPROPSET_DBINIT;
 398
 399     for (i=0; i < cPropertyIDSets; i++)
 400     {
 401         TRACE("Property id %d (count %ld, set %s)\n", i, rgPropertyIDSets[i].cPropertyIDs,
 402                 debugstr_guid(&rgPropertyIDSets[i].guidPropertySet));
 403
 404         propset->cProperties = rgPropertyIDSets[i].cPropertyIDs;
 405         propset->rgProperties = CoTaskMemAlloc(propset->cProperties * sizeof(DBPROP));
 406
 407         for (j=0; j < propset->cProperties; j++)
 408         {
 409             propset->rgProperties[j].dwPropertyID = rgPropertyIDSets[i].rgPropertyIDs[j];
 410
 411             for(k = 0; k < ARRAY_SIZE(provider->properties); k++)
 412             {
 413                 if (provider->properties[k].id == rgPropertyIDSets[i].rgPropertyIDs[j])
 414                 {
 415                     V_VT(&propset->rgProperties[j].vValue) = VT_EMPTY;
 416                     VariantCopy(&propset->rgProperties[j].vValue, &provider->properties[k].value);
 417                     break;
 418                 }
 419             }
 420         }
 421     }
 422
 423     *prgPropertySets = propset;
 424
 425     return S_OK;
 426 }
 427
 428 static HRESULT WINAPI dbprops_GetPropertyInfo(IDBProperties *iface, ULONG cPropertyIDSets,
 429             const DBPROPIDSET rgPropertyIDSets[], ULONG *pcPropertyInfoSets,
 430             DBPROPINFOSET **prgPropertyInfoSets, OLECHAR **ppDescBuffer)
 431 {
 432     struct msdasql *provider = impl_from_IDBProperties(iface);
 433     int i;
 434     DBPROPINFOSET *infoset;
 435     int size = 1;
 436     OLECHAR *ptr;
 437
 438     TRACE("(%p)->(%ld %p %p %p %p)\n", provider, cPropertyIDSets, rgPropertyIDSets, pcPropertyInfoSets,
 439                 prgPropertyInfoSets, ppDescBuffer);
 440
 441     infoset = CoTaskMemAlloc(sizeof(DBPROPINFOSET));
 442     memcpy(&infoset->guidPropertySet, &DBPROPSET_DBINIT, sizeof(GUID));
 443     infoset->cPropertyInfos = ARRAY_SIZE(dbproperties);
 444     infoset->rgPropertyInfos = CoTaskMemAlloc(sizeof(DBPROPINFO) * ARRAY_SIZE(dbproperties));
 445
 446     for(i=0; i < ARRAY_SIZE(dbproperties); i++)
 447     {
 448         size += lstrlenW(dbproperties[i].name) + 1;
 449     }
 450
 451     ptr = *ppDescBuffer = CoTaskMemAlloc(size * sizeof(WCHAR));
 452     memset(*ppDescBuffer, 0, size * sizeof(WCHAR));
 453
 454     for(i=0; i < ARRAY_SIZE(dbproperties); i++)
 455     {
 456         lstrcpyW(ptr, dbproperties[i].name);
 457         infoset->rgPropertyInfos[i].pwszDescription = ptr;
 458         infoset->rgPropertyInfos[i].dwPropertyID =  dbproperties[i].id;
 459         infoset->rgPropertyInfos[i].dwFlags = DBPROPFLAGS_DBINIT | DBPROPFLAGS_READ | DBPROPFLAGS_WRITE;
 460         infoset->rgPropertyInfos[i].vtType =  dbproperties[i].type;
 461         V_VT(&infoset->rgPropertyInfos[i].vValues) =  VT_EMPTY;
 462
 463         ptr += lstrlenW(dbproperties[i].name) + 1;
 464     }
 465
 466     *pcPropertyInfoSets = 1;
 467     *prgPropertyInfoSets = infoset;
 468
 469     return S_OK;
 470 }
 471
 472 static HRESULT WINAPI dbprops_SetProperties(IDBProperties *iface, ULONG cPropertySets,
 473             DBPROPSET rgPropertySets[])
 474 {
 475     struct msdasql *provider = impl_from_IDBProperties(iface);
 476     int i, j, k;
 477
 478     TRACE("(%p)->(%ld %p)\n", provider, cPropertySets, rgPropertySets);
 479
 480     for (i=0; i < cPropertySets; i++)
 481     {
 482         for (j=0; j < rgPropertySets[i].cProperties; j++)
 483         {
 484             for(k=0; k < ARRAY_SIZE(provider->properties); k++)
 485             {
 486                 if (provider->properties[k].id == rgPropertySets[i].rgProperties[j].dwPropertyID)
 487                 {
 488                     TRACE("Found property %d\n", provider->properties[k].id);
 489                     VariantCopy(&provider->properties[k].value, &rgPropertySets[i].rgProperties[j].vValue);
 490                     break;
 491                 }
 492             }
 493         }
 494     }
 495
 496     return S_OK;
 497 }
 498
 499 static const struct IDBPropertiesVtbl dbprops_vtbl =
 500 {
 501     dbprops_QueryInterface,
 502     dbprops_AddRef,
 503     dbprops_Release,
 504     dbprops_GetProperties,
 505     dbprops_GetPropertyInfo,
 506     dbprops_SetProperties
 507 };
 508
 509 static HRESULT WINAPI dbinit_QueryInterface(IDBInitialize *iface, REFIID riid, void **ppvObject)
 510 {
 511     struct msdasql *provider = impl_from_IDBInitialize(iface);
 512
 513     return IUnknown_QueryInterface(&provider->MSDASQL_iface, riid, ppvObject);
 514 }
 515
 516 static ULONG WINAPI dbinit_AddRef(IDBInitialize *iface)
 517 {
 518     struct msdasql *provider = impl_from_IDBInitialize(iface);
 519
 520     return IUnknown_AddRef(&provider->MSDASQL_iface);
 521 }
 522
 523 static ULONG WINAPI dbinit_Release(IDBInitialize *iface)
 524 {
 525     struct msdasql *provider = impl_from_IDBInitialize(iface);
 526
 527     return IUnknown_Release(&provider->MSDASQL_iface);
 528 }
 529
 530 static HRESULT WINAPI dbinit_Initialize(IDBInitialize *iface)
 531 {
 532     struct msdasql *provider = impl_from_IDBInitialize(iface);
 533     int i, len;
 534     SQLRETURN ret;
 535     WCHAR connection[1024], *p = connection, outstr[1024];
 536
 537     FIXME("%p semi-stub\n", provider);
 538
 539     *p = 0;
 540     for(i=0; i < ARRAY_SIZE(provider->properties); i++)
 541     {
 542         if (provider->properties[i].id == DBPROP_INIT_DATASOURCE)
 543         {
 544             len = swprintf(p, ARRAY_SIZE(connection) - (p - connection),
 545                            L"DSN=%s;", V_BSTR(&provider->properties[i].value));
 546             p+= len;
 547         }
 548         else if (provider->properties[i].id == DBPROP_INIT_PROVIDERSTRING)
 549         {
 550             if (V_VT(&provider->properties[i].value) == VT_BSTR && SysStringLen(V_BSTR(&provider->properties[i].value)) )
 551             {
 552                 len = swprintf(p, ARRAY_SIZE(connection) - (p - connection),
 553                                L"%s;", V_BSTR(&provider->properties[i].value));
 554                 p+= len;
 555             }
 556         }
 557     }
 558
 559     ret = SQLDriverConnectW( provider->hdbc, NULL, connection, wcslen(connection),
 560                              outstr, ARRAY_SIZE(outstr), NULL, 0);
 561     TRACE("SQLDriverConnectW ret %d\n", ret);
 562     if (ret != SQL_SUCCESS)
 563     {
 564         dump_sql_diag_records(SQL_HANDLE_DBC, provider->hdbc);
 565
 566         if (ret != SQL_SUCCESS_WITH_INFO)
 567            return E_FAIL;
 568     }
 569
 570     return S_OK;
 571 }
 572
 573 static HRESULT WINAPI dbinit_Uninitialize(IDBInitialize *iface)
 574 {
 575     struct msdasql *provider = impl_from_IDBInitialize(iface);
 576
 577     FIXME("%p stub\n", provider);
 578
 579     return S_OK;
 580 }
 581
 582
 583 static const struct IDBInitializeVtbl dbinit_vtbl =
 584 {
 585     dbinit_QueryInterface,
 586     dbinit_AddRef,
 587     dbinit_Release,
 588     dbinit_Initialize,
 589     dbinit_Uninitialize
 590 };
 591
 592 static HRESULT WINAPI dbsess_QueryInterface(IDBCreateSession *iface, REFIID riid, void **ppvObject)
 593 {
 594     struct msdasql *provider = impl_from_IDBCreateSession(iface);
 595
 596     return IUnknown_QueryInterface(&provider->MSDASQL_iface, riid, ppvObject);
 597 }
 598
 599 static ULONG WINAPI dbsess_AddRef(IDBCreateSession *iface)
 600 {
 601     struct msdasql *provider = impl_from_IDBCreateSession(iface);
 602
 603     return IUnknown_AddRef(&provider->MSDASQL_iface);
 604 }
 605
 606 static ULONG WINAPI dbsess_Release(IDBCreateSession *iface)
 607 {
 608     struct msdasql *provider = impl_from_IDBCreateSession(iface);
 609
 610     return IUnknown_Release(&provider->MSDASQL_iface);
 611 }
 612
 613 static HRESULT WINAPI dbsess_CreateSession(IDBCreateSession *iface, IUnknown *outer, REFIID riid,
 614         IUnknown **session)
 615 {
 616     struct msdasql *provider = impl_from_IDBCreateSession(iface);
 617     HRESULT hr;
 618
 619     TRACE("%p, outer %p, riid %s, session %p stub\n", provider, outer, debugstr_guid(riid), session);
 620
 621     if (outer)
 622         FIXME("outer currently not supported.\n");
 623
 624     hr = create_db_session(riid, &provider->MSDASQL_iface, provider->hdbc, (void**)session);
 625
 626     return hr;
 627 }
 628
 629 static const struct IDBCreateSessionVtbl dbsess_vtbl =
 630 {
 631     dbsess_QueryInterface,
 632     dbsess_AddRef,
 633     dbsess_Release,
 634     dbsess_CreateSession
 635 };
 636
 637 static HRESULT WINAPI persist_QueryInterface(IPersist *iface, REFIID riid, void **ppv)
 638 {
 639     struct msdasql *provider = impl_from_IPersist( iface );
 640     return IUnknown_QueryInterface(&provider->MSDASQL_iface, riid, ppv);
 641 }
 642
 643 static ULONG WINAPI persist_AddRef(IPersist *iface)
 644 {
 645     struct msdasql *provider = impl_from_IPersist( iface );
 646     return IUnknown_AddRef(&provider->MSDASQL_iface);
 647 }
 648
 649 static ULONG WINAPI persist_Release(IPersist *iface)
 650 {
 651     struct msdasql *provider = impl_from_IPersist( iface );
 652     return IUnknown_Release(&provider->MSDASQL_iface);
 653 }
 654
 655 static HRESULT WINAPI persist_GetClassID(IPersist *iface, CLSID *classid)
 656 {
 657     struct msdasql *provider = impl_from_IPersist( iface );
 658
 659     TRACE("(%p)->(%p)\n", provider, classid);
 660
 661     if(!classid)
 662         return E_INVALIDARG;
 663
 664     *classid = CLSID_MSDASQL;
 665     return S_OK;
 666
 667 }
 668
 669 static const IPersistVtbl persistVtbl = {
 670     persist_QueryInterface,
 671     persist_AddRef,
 672     persist_Release,
 673     persist_GetClassID
 674 };
 675
 676 static HRESULT create_msdasql_provider(REFIID riid, void **ppv)
 677 {
 678     struct msdasql *provider;
 679     HRESULT hr;
 680     int i;
 681
 682     provider = malloc(sizeof(struct msdasql));
 683     if (!provider)
 684         return E_OUTOFMEMORY;
 685
 686     provider->MSDASQL_iface.lpVtbl = &msdsql_vtbl;
 687     provider->IDBProperties_iface.lpVtbl = &dbprops_vtbl;
 688     provider->IDBInitialize_iface.lpVtbl = &dbinit_vtbl;
 689     provider->IDBCreateSession_iface.lpVtbl = &dbsess_vtbl;
 690     provider->IPersist_iface.lpVtbl = &persistVtbl;
 691     provider->ref = 1;
 692
 693     for(i=0; i < ARRAY_SIZE(dbproperties); i++)
 694     {
 695         provider->properties[i].id = dbproperties[i].id;
 696         VariantInit(&provider->properties[i].value);
 697
 698         /* Only the follow are initialized to a value */
 699         switch(dbproperties[i].id)
 700         {
 701             case DBPROP_INIT_PROMPT:
 702                 V_VT(&provider->properties[i].value) = dbproperties[i].type;
 703                 V_I2(&provider->properties[i].value) = 4;
 704                 break;
 705             case DBPROP_INIT_LCID:
 706                 V_VT(&provider->properties[i].value) = dbproperties[i].type;
 707                 V_I4(&provider->properties[i].value) = GetUserDefaultLCID();
 708                 break;
 709             case DBPROP_INIT_OLEDBSERVICES:
 710                 V_VT(&provider->properties[i].value) = dbproperties[i].type;
 711                 V_I4(&provider->properties[i].value) = -1;
 712                 break;
 713             default:
 714                 V_VT(&provider->properties[i].value) = VT_EMPTY;
 715         }
 716     }
 717
 718     SQLAllocHandle(SQL_HANDLE_ENV, NULL, &provider->henv );
 719     if (SQLSetEnvAttr(provider->henv, SQL_ATTR_ODBC_VERSION, (void *)SQL_OV_ODBC3_80, 0) == SQL_ERROR)
 720     {
 721         WARN("Falling back to SQL_OV_ODBC3\n");
 722         SQLSetEnvAttr(provider->henv, SQL_ATTR_ODBC_VERSION, (void *)SQL_OV_ODBC3, 0);
 723     }
 724
 725     SQLAllocHandle(SQL_HANDLE_DBC, provider->henv, &provider->hdbc);
 726
 727     hr = IUnknown_QueryInterface(&provider->MSDASQL_iface, riid, ppv);
 728     IUnknown_Release(&provider->MSDASQL_iface);
 729     return hr;
 730 }
 731
 732 struct msdasql_enum
 733 {
 734     ISourcesRowset ISourcesRowset_iface;
 735     LONG ref;
 736 };
 737
 738 static inline struct msdasql_enum *msdasql_enum_from_ISourcesRowset(ISourcesRowset *iface)
 739 {
 740     return CONTAINING_RECORD(iface, struct msdasql_enum, ISourcesRowset_iface);
 741 }
 742
 743 static HRESULT WINAPI msdasql_enum_QueryInterface(ISourcesRowset *iface, REFIID riid, void **out)
 744 {
 745     struct msdasql_enum *enumerator = msdasql_enum_from_ISourcesRowset(iface);
 746
 747     TRACE("(%p)->(%s %p)\n", enumerator, debugstr_guid(riid), out);
 748
 749     if(IsEqualGUID(riid, &IID_IUnknown) ||
 750         IsEqualGUID(riid, &IID_ISourcesRowset ) )
 751     {
 752         *out = &enumerator->ISourcesRowset_iface;
 753     }
 754     else
 755     {
 756         FIXME("(%s, %p)\n", debugstr_guid(riid), out);
 757         *out = NULL;
 758         return E_NOINTERFACE;
 759     }
 760
 761     IUnknown_AddRef((IUnknown*)*out);
 762     return S_OK;
 763 }
 764
 765 static ULONG WINAPI msdasql_enum_AddRef(ISourcesRowset *iface)
 766 {
 767     struct msdasql_enum *enumerator = msdasql_enum_from_ISourcesRowset(iface);
 768     ULONG ref = InterlockedIncrement(&enumerator->ref);
 769
 770     TRACE("(%p) ref=%lu\n", enumerator, ref);
 771
 772     return ref;
 773 }
 774
 775 static ULONG WINAPI msdasql_enum_Release(ISourcesRowset *iface)
 776 {
 777     struct msdasql_enum *enumerator = msdasql_enum_from_ISourcesRowset(iface);
 778     ULONG ref = InterlockedDecrement(&enumerator->ref);
 779
 780     TRACE("(%p) ref=%lu\n", enumerator, ref);
 781
 782     if (!ref)
 783     {
 784         free(enumerator);
 785     }
 786
 787     return ref;
 788 }
 789
 790 struct msdasql_enum_rowset
 791 {
 792     IRowset IRowset_iface;
 793     IAccessor IAccessor_iface;
 794     LONG ref;
 795 };
 796
 797 static inline struct msdasql_enum_rowset *msdasql_rs_from_IRowset(IRowset *iface)
 798 {
 799     return CONTAINING_RECORD(iface, struct msdasql_enum_rowset, IRowset_iface);
 800 }
 801
 802 static inline struct msdasql_enum_rowset *msdasql_enum_from_IAccessor ( IAccessor *iface )
 803 {
 804     return CONTAINING_RECORD( iface, struct msdasql_enum_rowset, IAccessor_iface );
 805 }
 806
 807 static HRESULT WINAPI enum_rowset_QueryInterface(IRowset *iface, REFIID riid,  void **ppv)
 808 {
 809     struct msdasql_enum_rowset *rowset = msdasql_rs_from_IRowset( iface );
 810
 811     TRACE( "%p, %s, %p\n", rowset, debugstr_guid(riid), ppv );
 812     *ppv = NULL;
 813
 814     if(IsEqualGUID(&IID_IUnknown, riid) ||
 815        IsEqualGUID(&IID_IRowset, riid))
 816     {
 817         *ppv = &rowset->IRowset_iface;
 818     }
 819     else if(IsEqualGUID(&IID_IAccessor, riid))
 820     {
 821          *ppv = &rowset->IAccessor_iface;
 822     }
 823
 824     if(*ppv)
 825     {
 826         IUnknown_AddRef((IUnknown*)*ppv);
 827         return S_OK;
 828     }
 829
 830     FIXME("(%p)->(%s %p)\n", iface, debugstr_guid(riid), ppv);
 831     return E_NOINTERFACE;
 832 }
 833
 834 static ULONG WINAPI enum_rowset_AddRef(IRowset *iface)
 835 {
 836     struct msdasql_enum_rowset *rowset = msdasql_rs_from_IRowset( iface );
 837     LONG refs = InterlockedIncrement( &rowset->ref );
 838     TRACE( "%p new refcount %ld\n", rowset, refs );
 839     return refs;
 840 }
 841
 842 static ULONG WINAPI enum_rowset_Release(IRowset *iface)
 843 {
 844     struct msdasql_enum_rowset *rowset = msdasql_rs_from_IRowset( iface );
 845     LONG refs = InterlockedDecrement( &rowset->ref );
 846     TRACE( "%p new refcount %ld\n", rowset, refs );
 847     if (!refs)
 848     {
 849         TRACE( "destroying %p\n", rowset );
 850         free( rowset );
 851     }
 852     return refs;
 853 }
 854
 855 static HRESULT WINAPI enum_rowset_AddRefRows(IRowset *iface, DBCOUNTITEM count,
 856         const HROW rows[], DBREFCOUNT ref_counts[], DBROWSTATUS status[])
 857 {
 858     struct msdasql_enum_rowset *rowset = msdasql_rs_from_IRowset( iface );
 859
 860     FIXME("%p, %Iu, %p, %p, %p\n", rowset, count, rows, ref_counts, status);
 861
 862     return E_NOTIMPL;
 863 }
 864
 865 static HRESULT WINAPI enum_rowset_GetData(IRowset *iface, HROW row, HACCESSOR accessor, void *data)
 866 {
 867     struct msdasql_enum_rowset *rowset = msdasql_rs_from_IRowset( iface );
 868
 869     FIXME("%p, %Iu, %Iu, %p\n", rowset, row, accessor, data);
 870
 871     return E_NOTIMPL;
 872 }
 873
 874 static HRESULT WINAPI enum_rowset_GetNextRows(IRowset *iface, HCHAPTER reserved, DBROWOFFSET offset,
 875         DBROWCOUNT count, DBCOUNTITEM *obtained, HROW **rows)
 876 {
 877     struct msdasql_enum_rowset *rowset = msdasql_rs_from_IRowset( iface );
 878
 879     FIXME("%p, %Iu, %Iu, %Iu, %p, %p\n", rowset, reserved, offset, count, obtained, rows);
 880
 881     if (!obtained || !rows)
 882         return E_INVALIDARG;
 883
 884     *obtained = 0;
 885
 886     if (!count)
 887         return S_OK;
 888
 889     return DB_S_ENDOFROWSET;
 890 }
 891
 892 static HRESULT WINAPI enum_rowset_ReleaseRows(IRowset *iface, DBCOUNTITEM count,
 893         const HROW rows[], DBROWOPTIONS options[], DBREFCOUNT ref_counts[], DBROWSTATUS status[])
 894 {
 895     struct msdasql_enum_rowset *rowset = msdasql_rs_from_IRowset( iface );
 896
 897     FIXME("%p, %Iu, %p, %p, %p, %p\n", rowset, count, rows, options, ref_counts, status);
 898
 899     return S_OK;
 900 }
 901
 902 static HRESULT WINAPI enum_rowset_RestartPosition(IRowset *iface, HCHAPTER reserved)
 903 {
 904     struct msdasql_enum_rowset *rowset = msdasql_rs_from_IRowset( iface );
 905
 906     FIXME("%p, %Iu\n", rowset, reserved);
 907
 908     return S_OK;
 909 }
 910
 911 static const struct IRowsetVtbl enum_rowset_vtbl =
 912 {
 913     enum_rowset_QueryInterface,
 914     enum_rowset_AddRef,
 915     enum_rowset_Release,
 916     enum_rowset_AddRefRows,
 917     enum_rowset_GetData,
 918     enum_rowset_GetNextRows,
 919     enum_rowset_ReleaseRows,
 920     enum_rowset_RestartPosition
 921 };
 922
 923 static HRESULT WINAPI enum_rs_accessor_QueryInterface(IAccessor *iface, REFIID riid, void **out)
 924 {
 925     struct msdasql_enum_rowset *rowset = msdasql_enum_from_IAccessor( iface );
 926     return IRowset_QueryInterface(&rowset->IRowset_iface, riid, out);
 927 }
 928
 929 static ULONG WINAPI enum_rs_accessor_AddRef(IAccessor *iface)
 930 {
 931     struct msdasql_enum_rowset *rowset = msdasql_enum_from_IAccessor( iface );
 932     return IRowset_AddRef(&rowset->IRowset_iface);
 933 }
 934
 935 static ULONG  WINAPI enum_rs_accessor_Release(IAccessor *iface)
 936 {
 937     struct msdasql_enum_rowset *rowset = msdasql_enum_from_IAccessor( iface );
 938     return IRowset_Release(&rowset->IRowset_iface);
 939 }
 940
 941 static HRESULT WINAPI enum_rs_accessor_AddRefAccessor(IAccessor *iface, HACCESSOR accessor, DBREFCOUNT *count)
 942 {
 943     struct msdasql_enum_rowset *rowset = msdasql_enum_from_IAccessor( iface );
 944     FIXME("%p, %Iu, %p\n", rowset, accessor, count);
 945     return E_NOTIMPL;
 946 }
 947
 948 static HRESULT WINAPI enum_rs_accessor_CreateAccessor(IAccessor *iface, DBACCESSORFLAGS flags,
 949         DBCOUNTITEM count, const DBBINDING bindings[], DBLENGTH row_size, HACCESSOR *accessor,
 950         DBBINDSTATUS status[])
 951 {
 952     struct msdasql_enum_rowset *rowset = msdasql_enum_from_IAccessor( iface );
 953
 954     FIXME("%p 0x%08lx, %Iu, %p, %Iu, %p, %p\n", rowset, flags, count, bindings, row_size, accessor, status);
 955
 956     if (accessor)
 957         *accessor = 0xdeadbeef;
 958
 959     return S_OK;
 960 }
 961
 962 static HRESULT WINAPI enum_rs_accessor_GetBindings(IAccessor *iface, HACCESSOR accessor,
 963         DBACCESSORFLAGS *flags, DBCOUNTITEM *count, DBBINDING **bindings)
 964 {
 965     struct msdasql_enum_rowset *rowset = msdasql_enum_from_IAccessor( iface );
 966     FIXME("%p, %Iu, %p, %p, %p\n", rowset, accessor, flags, count, bindings);
 967     return E_NOTIMPL;
 968 }
 969
 970 static HRESULT WINAPI enum_rs_accessor_ReleaseAccessor(IAccessor *iface, HACCESSOR accessor, DBREFCOUNT *count)
 971 {
 972     struct msdasql_enum_rowset *rowset = msdasql_enum_from_IAccessor( iface );
 973
 974     FIXME("%p, %Iu, %p\n", rowset, accessor, count);
 975
 976     if (count)
 977         *count = 0;
 978
 979     return S_OK;
 980 }
 981
 982 struct IAccessorVtbl enum_accessor_vtbl =
 983 {
 984     enum_rs_accessor_QueryInterface,
 985     enum_rs_accessor_AddRef,
 986     enum_rs_accessor_Release,
 987     enum_rs_accessor_AddRefAccessor,
 988     enum_rs_accessor_CreateAccessor,
 989     enum_rs_accessor_GetBindings,
 990     enum_rs_accessor_ReleaseAccessor
 991 };
 992
 993 static HRESULT WINAPI msdasql_enum_GetSourcesRowset(ISourcesRowset *iface, IUnknown *outer, REFIID riid, ULONG sets,
 994                 DBPROPSET properties[], IUnknown **rowset)
 995 {
 996     struct msdasql_enum *enumerator = msdasql_enum_from_ISourcesRowset(iface);
 997     struct msdasql_enum_rowset *enum_rs;
 998     HRESULT hr;
 999
1000     TRACE("(%p) %p, %s, %lu, %p, %p\n", enumerator, outer, debugstr_guid(riid), sets, properties, rowset);
1001
1002     enum_rs = malloc(sizeof(*enum_rs));
1003     enum_rs->IRowset_iface.lpVtbl = &enum_rowset_vtbl;
1004     enum_rs->IAccessor_iface.lpVtbl = &enum_accessor_vtbl;
1005     enum_rs->ref = 1;
1006
1007     hr = IRowset_QueryInterface(&enum_rs->IRowset_iface, riid, (void**)rowset);
1008     IRowset_Release(&enum_rs->IRowset_iface);
1009     return hr;
1010 }
1011
1012 static const ISourcesRowsetVtbl msdsqlenum_vtbl =
1013 {
1014     msdasql_enum_QueryInterface,
1015     msdasql_enum_AddRef,
1016     msdasql_enum_Release,
1017     msdasql_enum_GetSourcesRowset
1018 };
1019
1020 static HRESULT create_msdasql_enumerator(REFIID riid, void **ppv)
1021 {
1022     struct msdasql_enum *enumerator;
1023     HRESULT hr;
1024
1025     enumerator = malloc(sizeof(*enumerator));
1026     if (!enumerator)
1027         return E_OUTOFMEMORY;
1028
1029     enumerator->ISourcesRowset_iface.lpVtbl = &msdsqlenum_vtbl;
1030     enumerator->ref = 1;
1031
1032     hr = IUnknown_QueryInterface(&enumerator->ISourcesRowset_iface, riid, ppv);
1033     IUnknown_Release(&enumerator->ISourcesRowset_iface);
1034     return hr;
1035 }