dlls/oledb32/dslocator.c

   1 /* Data Links
   2  *
   3  * Copyright 2013 Alistair Leslie-Hughes
   4  *
   5  * This library is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU Lesser General Public
   7  * License as published by the Free Software Foundation; either
   8  * version 2.1 of the License, or (at your option) any later version.
   9  *
  10  * This library is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * Lesser General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU Lesser General Public
  16  * License along with this library; if not, write to the Free Software
  17  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  18  */
  19 #include <stdarg.h>
  20 #include <string.h>
  21
  22 #define COBJMACROS
  23 #define NONAMELESSUNION
  24
  25 #include "windef.h"
  26 #include "winbase.h"
  27 #include "objbase.h"
  28 #include "oleauto.h"
  29 #include "winerror.h"
  30 #include "oledb.h"
  31 #include "oledberr.h"
  32 #include "msdasc.h"
  33 #include "prsht.h"
  34 #include "commctrl.h"
  35
  36 #include "oledb_private.h"
  37 #include "resource.h"
  38
  39 #include "wine/debug.h"
  40 #include "wine/heap.h"
  41
  42 WINE_DEFAULT_DEBUG_CHANNEL(oledb);
  43
  44 struct datasource
  45 {
  46     CLSID clsid;
  47     IDBProperties *provider;
  48     DBPROPINFOSET *propinfoset;
  49     WCHAR *description;
  50 };
  51
  52 static struct datasource *create_datasource(WCHAR *guid)
  53 {
  54     struct datasource *data = heap_alloc_zero(sizeof(struct datasource));
  55     if (data)
  56     {
  57         CLSIDFromString(guid, &data->clsid);
  58     }
  59
  60     return data;
  61 }
  62
  63 static void destroy_datasource(struct datasource *data)
  64 {
  65     if (data->propinfoset)
  66     {
  67         ULONG i;
  68
  69         for (i = 0; i < data->propinfoset->cPropertyInfos; i++)
  70             VariantClear(&data->propinfoset->rgPropertyInfos[i].vValues);
  71
  72         CoTaskMemFree(data->propinfoset->rgPropertyInfos);
  73         CoTaskMemFree(data->propinfoset);
  74     }
  75     CoTaskMemFree(data->description);
  76
  77     if (data->provider)
  78         IDBProperties_Release(data->provider);
  79
  80     heap_free(data);
  81 }
  82
  83 static BOOL initialize_datasource(struct datasource *data)
  84 {
  85     HRESULT hr;
  86     DBPROPIDSET propidset;
  87     ULONG infocount;
  88
  89     hr = CoCreateInstance(&data->clsid, NULL, CLSCTX_INPROC_SERVER, &IID_IDBProperties, (void**)&data->provider);
  90     if (FAILED(hr))
  91     {
  92         WARN("Datasource cannot be created (0x%08x)\n", hr);
  93         return FALSE;
  94     }
  95
  96     propidset.rgPropertyIDs = NULL;
  97     propidset.cPropertyIDs  = 0;
  98     propidset.guidPropertySet = DBPROPSET_DBINITALL;
  99
 100     hr = IDBProperties_GetPropertyInfo(data->provider, 1, &propidset, &infocount, &data->propinfoset, &data->description);
 101     if (FAILED(hr))
 102     {
 103         WARN("Failed to get DB Properties (0x%08x)\n", hr);
 104
 105         IDBProperties_Release(data->provider);
 106         data->provider = NULL;
 107         return FALSE;
 108     }
 109
 110     return TRUE;
 111 }
 112
 113 typedef struct DSLocatorImpl
 114 {
 115     IDataSourceLocator IDataSourceLocator_iface;
 116     IDataInitialize IDataInitialize_iface;
 117     LONG ref;
 118
 119     HWND hwnd;
 120 } DSLocatorImpl;
 121
 122 static inline DSLocatorImpl *impl_from_IDataSourceLocator( IDataSourceLocator *iface )
 123 {
 124     return CONTAINING_RECORD(iface, DSLocatorImpl, IDataSourceLocator_iface);
 125 }
 126
 127 static inline DSLocatorImpl *impl_from_IDataInitialize(IDataInitialize *iface)
 128 {
 129     return CONTAINING_RECORD(iface, DSLocatorImpl, IDataInitialize_iface);
 130 }
 131
 132 static HRESULT WINAPI dslocator_QueryInterface(IDataSourceLocator *iface, REFIID riid, void **ppvoid)
 133 {
 134     DSLocatorImpl *This = impl_from_IDataSourceLocator(iface);
 135     TRACE("(%p)->(%s, %p)\n", This, debugstr_guid(riid),ppvoid);
 136
 137     *ppvoid = NULL;
 138
 139     if (IsEqualIID(riid, &IID_IUnknown) ||
 140         IsEqualIID(riid, &IID_IDispatch) ||
 141         IsEqualIID(riid, &IID_IDataSourceLocator))
 142     {
 143       *ppvoid = &This->IDataSourceLocator_iface;
 144     }
 145     else if (IsEqualIID(riid, &IID_IDataInitialize))
 146     {
 147       *ppvoid = &This->IDataInitialize_iface;
 148     }
 149     else if (IsEqualIID(riid, &IID_IRunnableObject))
 150     {
 151       TRACE("IID_IRunnableObject returning NULL\n");
 152       return E_NOINTERFACE;
 153     }
 154     else if (IsEqualIID(riid, &IID_IProvideClassInfo))
 155     {
 156       TRACE("IID_IProvideClassInfo returning NULL\n");
 157       return E_NOINTERFACE;
 158     }
 159     else if (IsEqualIID(riid, &IID_IMarshal))
 160     {
 161       TRACE("IID_IMarshal returning NULL\n");
 162       return E_NOINTERFACE;
 163     }
 164     else if (IsEqualIID(riid, &IID_IRpcOptions))
 165     {
 166       TRACE("IID_IRpcOptions returning NULL\n");
 167       return E_NOINTERFACE;
 168     }
 169
 170     if(*ppvoid)
 171     {
 172       IUnknown_AddRef( (IUnknown*)*ppvoid );
 173       return S_OK;
 174     }
 175
 176     FIXME("interface %s not implemented\n", debugstr_guid(riid));
 177     return E_NOINTERFACE;
 178 }
 179
 180 static ULONG WINAPI dslocator_AddRef(IDataSourceLocator *iface)
 181 {
 182     DSLocatorImpl *This = impl_from_IDataSourceLocator(iface);
 183     TRACE("(%p)->%u\n",This,This->ref);
 184     return InterlockedIncrement(&This->ref);
 185 }
 186
 187 static ULONG WINAPI dslocator_Release(IDataSourceLocator *iface)
 188 {
 189     DSLocatorImpl *This = impl_from_IDataSourceLocator(iface);
 190     ULONG ref = InterlockedDecrement(&This->ref);
 191
 192     TRACE("(%p)->%u\n",This,ref+1);
 193
 194     if (!ref)
 195     {
 196         heap_free(This);
 197     }
 198
 199     return ref;
 200 }
 201
 202 static HRESULT WINAPI dslocator_GetTypeInfoCount(IDataSourceLocator *iface, UINT *pctinfo)
 203 {
 204     DSLocatorImpl *This = impl_from_IDataSourceLocator(iface);
 205
 206     FIXME("(%p)->()\n", This);
 207
 208     return E_NOTIMPL;
 209 }
 210
 211 static HRESULT WINAPI dslocator_GetTypeInfo(IDataSourceLocator *iface, UINT iTInfo, LCID lcid, ITypeInfo **ppTInfo)
 212 {
 213     DSLocatorImpl *This = impl_from_IDataSourceLocator(iface);
 214
 215     FIXME("(%p)->(%u %u %p)\n", This, iTInfo, lcid, ppTInfo);
 216
 217     return E_NOTIMPL;
 218 }
 219
 220 static HRESULT WINAPI dslocator_GetIDsOfNames(IDataSourceLocator *iface, REFIID riid, LPOLESTR *rgszNames,
 221     UINT cNames, LCID lcid, DISPID *rgDispId)
 222 {
 223     DSLocatorImpl *This = impl_from_IDataSourceLocator(iface);
 224
 225     FIXME("(%p)->(%s %p %u %u %p)\n", This, debugstr_guid(riid), rgszNames, cNames, lcid, rgDispId);
 226
 227     return E_NOTIMPL;
 228 }
 229
 230 static HRESULT WINAPI dslocator_Invoke(IDataSourceLocator *iface, DISPID dispIdMember, REFIID riid,
 231     LCID lcid, WORD wFlags, DISPPARAMS *pDispParams, VARIANT *pVarResult, EXCEPINFO *pExcepInfo, UINT *puArgErr)
 232 {
 233     DSLocatorImpl *This = impl_from_IDataSourceLocator(iface);
 234
 235     FIXME("(%p)->(%d %s %d %d %p %p %p %p)\n", This, dispIdMember, debugstr_guid(riid),
 236            lcid, wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr);
 237
 238     return E_NOTIMPL;
 239 }
 240
 241 static HRESULT WINAPI dslocator_get_hWnd(IDataSourceLocator *iface, COMPATIBLE_LONG *phwndParent)
 242 {
 243     DSLocatorImpl *This = impl_from_IDataSourceLocator(iface);
 244
 245     TRACE("(%p)->(%p)\n",This, phwndParent);
 246
 247     *phwndParent = (COMPATIBLE_LONG)This->hwnd;
 248
 249     return S_OK;
 250 }
 251
 252 static HRESULT WINAPI dslocator_put_hWnd(IDataSourceLocator *iface, COMPATIBLE_LONG hwndParent)
 253 {
 254     DSLocatorImpl *This = impl_from_IDataSourceLocator(iface);
 255
 256     TRACE("(%p)->(%p)\n",This, (HWND)hwndParent);
 257
 258     This->hwnd = (HWND)hwndParent;
 259
 260     return S_OK;
 261 }
 262
 263 static void create_connections_columns(HWND lv)
 264 {
 265     RECT rc;
 266     WCHAR buf[256];
 267     LVCOLUMNW column;
 268
 269     SendMessageW(lv, LVM_SETEXTENDEDLISTVIEWSTYLE, 0, LVS_EX_FULLROWSELECT);
 270     GetWindowRect(lv, &rc);
 271     LoadStringW(instance, IDS_COL_PROVIDER, buf, ARRAY_SIZE(buf));
 272     column.mask = LVCF_WIDTH | LVCF_TEXT;
 273     column.cx = (rc.right - rc.left) - 5;
 274     column.pszText = buf;
 275     SendMessageW(lv, LVM_INSERTCOLUMNW, 0, (LPARAM)&column);
 276 }
 277
 278 static void add_connections_providers(HWND lv)
 279 {
 280     LONG res;
 281     HKEY key = NULL, subkey;
 282     DWORD index = 0;
 283     LONG next_key;
 284     WCHAR provider[MAX_PATH];
 285     WCHAR guidkey[MAX_PATH];
 286     LONG size;
 287
 288     res = RegOpenKeyExW(HKEY_CLASSES_ROOT, L"CLSID", 0, KEY_READ, &key);
 289     if (res == ERROR_FILE_NOT_FOUND)
 290         return;
 291
 292     next_key = RegEnumKeyW(key, index, provider, MAX_PATH);
 293     while (next_key == ERROR_SUCCESS)
 294     {
 295         WCHAR description[MAX_PATH];
 296
 297         lstrcpyW(guidkey, provider);
 298         lstrcatW(guidkey, L"\\OLE DB Provider");
 299
 300         res = RegOpenKeyW(key, guidkey, &subkey);
 301         if (res == ERROR_SUCCESS)
 302         {
 303             TRACE("Found %s\n", debugstr_w(guidkey));
 304
 305             size = MAX_PATH;
 306             res = RegQueryValueW(subkey, NULL, description, &size);
 307             if (res == ERROR_SUCCESS)
 308             {
 309                 LVITEMW item;
 310                 struct datasource *data;
 311
 312                 data = create_datasource(guidkey);
 313
 314                 item.mask = LVIF_TEXT | LVIF_PARAM;
 315                 item.iItem = SendMessageW(lv, LVM_GETITEMCOUNT, 0, 0);
 316                 item.iSubItem = 0;
 317                 item.pszText = description;
 318                 item.lParam = (LPARAM)data;
 319
 320                 SendMessageW(lv, LVM_INSERTITEMW, 0, (LPARAM)&item);
 321             }
 322             RegCloseKey(subkey);
 323         }
 324
 325         index++;
 326         next_key = RegEnumKeyW(key, index, provider, MAX_PATH);
 327     }
 328
 329     RegCloseKey(key);
 330 }
 331
 332 static INT_PTR CALLBACK data_link_properties_dlg_proc(HWND hwnd, UINT msg, WPARAM wp, LPARAM lp)
 333 {
 334     TRACE("(%p, %08x, %08lx, %08lx)\n", hwnd, msg, wp, lp);
 335
 336     switch (msg)
 337     {
 338         case WM_INITDIALOG:
 339         {
 340             HWND btn, lv = GetDlgItem(hwnd, IDC_LST_CONNECTIONS);
 341             create_connections_columns(lv);
 342             add_connections_providers(lv);
 343
 344             btn = GetDlgItem(GetParent(hwnd), IDOK);
 345             EnableWindow(btn, FALSE);
 346
 347             break;
 348         }
 349         case WM_DESTROY:
 350         {
 351             HWND lv = GetDlgItem(hwnd, IDC_LST_CONNECTIONS);
 352             LVITEMA item;
 353
 354             item.iItem = 0;
 355             item.iSubItem = 0;
 356             item.mask = LVIF_PARAM;
 357
 358             while(ListView_GetItemA(lv, &item))
 359             {
 360                 destroy_datasource( (struct datasource *)item.lParam);
 361                 item.iItem++;
 362             }
 363             break;
 364         }
 365         case WM_NOTIFY:
 366         {
 367             NMHDR *hdr = ((LPNMHDR)lp);
 368             switch(hdr->code)
 369             {
 370                 case PSN_KILLACTIVE:
 371                 {
 372                     LVITEMA item;
 373                     /*
 374                      * FIXME: This needs to replace the connection page based off the selection.
 375                      *   We only care about the ODBC for now which is the default.
 376                      */
 377
 378                     HWND lv = GetDlgItem(hwnd, IDC_LST_CONNECTIONS);
 379                     if (!SendMessageW(lv, LVM_GETSELECTEDCOUNT, 0, 0))
 380                     {
 381                         WCHAR title[256], msg[256];
 382
 383                         LoadStringW(instance, IDS_PROVIDER_TITLE, title, ARRAY_SIZE(title));
 384                         LoadStringW(instance, IDS_PROVIDER_ERROR, msg, ARRAY_SIZE(msg));
 385                         MessageBoxW(hwnd, msg, title, MB_OK | MB_ICONEXCLAMATION);
 386                         SetWindowLongPtrW(hwnd, DWLP_MSGRESULT, TRUE);
 387                         return TRUE;
 388                     }
 389
 390                     item.iItem = 0;
 391                     item.iSubItem = 0;
 392                     item.stateMask = LVIS_SELECTED;
 393                     item.mask = LVIF_PARAM | LVIF_STATE;
 394
 395                     if(ListView_GetItemA(lv, &item))
 396                     {
 397                         if(!initialize_datasource( (struct datasource*)item.lParam))
 398                         {
 399                             WCHAR title[256], msg[256];
 400                             LoadStringW(instance, IDS_PROVIDER_TITLE, title, ARRAY_SIZE(title));
 401                             LoadStringW(instance, IDS_PROVIDER_NOT_AVAIL, msg, ARRAY_SIZE(msg));
 402                             MessageBoxW(hwnd, msg, title, MB_OK | MB_ICONEXCLAMATION);
 403                             SetWindowLongPtrW(hwnd, DWLP_MSGRESULT, TRUE);
 404                             return TRUE;
 405                         }
 406                     }
 407                     else
 408                         ERR("Failed to get selected item\n");
 409
 410                     return FALSE;
 411                 }
 412             }
 413
 414             break;
 415         }
 416         case WM_COMMAND:
 417         {
 418             if (LOWORD(wp) == IDC_BTN_NEXT)
 419                 SendMessageW(GetParent(hwnd), PSM_SETCURSEL, 1, 0);
 420             break;
 421         }
 422         default:
 423             break;
 424     }
 425     return 0;
 426 }
 427
 428 static void connection_fill_odbc_list(HWND parent)
 429 {
 430     LONG res;
 431     HKEY key;
 432     DWORD index = 0;
 433     WCHAR name[MAX_PATH];
 434     DWORD nameLen;
 435
 436     HWND combo = GetDlgItem(parent, IDC_CBO_NAMES);
 437     if (!combo)
 438         return;
 439
 440     res = RegOpenKeyExW(HKEY_CURRENT_USER, L"Software\\ODBC\\ODBC.INI\\ODBC Data Sources", 0, KEY_READ, &key);
 441     if (res == ERROR_FILE_NOT_FOUND)
 442         return;
 443
 444     SendMessageW (combo, CB_RESETCONTENT, 0, 0);
 445
 446     for(;; index++)
 447     {
 448         nameLen = MAX_PATH;
 449         if (RegEnumValueW(key, index, name, &nameLen, NULL, NULL, NULL, NULL) != ERROR_SUCCESS)
 450             break;
 451
 452         SendMessageW(combo, CB_ADDSTRING, 0, (LPARAM)name);
 453     }
 454
 455     RegCloseKey(key);
 456 }
 457
 458 static void connection_initialize_controls(HWND parent)
 459 {
 460     HWND hwnd = GetDlgItem(parent, IDC_RDO_SRC_NAME);
 461     if (hwnd)
 462         SendMessageA(hwnd, BM_SETCHECK, BST_CHECKED, 0);
 463 }
 464
 465 static void connection_toggle_controls(HWND parent)
 466 {
 467     BOOL checked = TRUE;
 468     HWND hwnd = GetDlgItem(parent, IDC_RDO_SRC_NAME);
 469     if (hwnd)
 470         checked = SendMessageA(hwnd, BM_GETCHECK, 0, 0);
 471
 472     EnableWindow(GetDlgItem(parent, IDC_CBO_NAMES), checked);
 473     EnableWindow(GetDlgItem(parent, IDC_BTN_REFRESH), checked);
 474
 475     EnableWindow(GetDlgItem(parent, IDC_LBL_CONNECTION), !checked);
 476     EnableWindow(GetDlgItem(parent, IDC_EDT_CONNECTION), !checked);
 477     EnableWindow(GetDlgItem(parent, IDC_BTN_BUILD), !checked);
 478 }
 479
 480 static INT_PTR CALLBACK data_link_connection_dlg_proc(HWND hwnd, UINT msg, WPARAM wp, LPARAM lp)
 481 {
 482     TRACE("(%p, %08x, %08lx, %08lx)\n", hwnd, msg, wp, lp);
 483
 484     switch (msg)
 485     {
 486         case WM_INITDIALOG:
 487         {
 488             connection_initialize_controls(hwnd);
 489             connection_fill_odbc_list(hwnd);
 490             connection_toggle_controls(hwnd);
 491             break;
 492         }
 493         case WM_COMMAND:
 494         {
 495             switch (LOWORD(wp))
 496             {
 497                 case IDC_RDO_SRC_NAME:
 498                 case IDC_BTN_CONNECTION:
 499                     connection_toggle_controls(hwnd);
 500                     break;
 501                 case IDC_BTN_REFRESH:
 502                     connection_fill_odbc_list(hwnd);
 503                     break;
 504                 case IDC_BTN_BUILD:
 505                 case IDC_BTN_TEST:
 506                     /* TODO: Implement dialogs */
 507                     MessageBoxA(hwnd, "Not implemented yet.", "Error", MB_OK | MB_ICONEXCLAMATION);
 508                     break;
 509             }
 510
 511             break;
 512         }
 513         default:
 514             break;
 515     }
 516     return 0;
 517 }
 518
 519 static void advanced_fill_permission_list(HWND parent)
 520 {
 521     LVITEMW item;
 522     LVCOLUMNW column;
 523     RECT rc;
 524     int resources[] = {IDS_PERM_READ, IDS_PERM_READWRITE, IDS_PERM_SHAREDENYNONE,
 525                         IDS_PERM_SHAREDENYREAD, IDS_PERM_SHAREDENYWRITE, IDS_PERM_SHAREEXCLUSIVE,
 526                         IDS_PERM_WRITE};
 527     int i;
 528     WCHAR buf[256];
 529     HWND lv = GetDlgItem(parent, IDC_LST_PERMISSIONS);
 530     if (!lv)
 531         return;
 532
 533     SendMessageW(lv, LVM_SETEXTENDEDLISTVIEWSTYLE, 0, LVS_EX_CHECKBOXES);
 534     GetWindowRect(lv, &rc);
 535     column.mask = LVCF_WIDTH | LVCF_FMT;
 536     column.fmt = LVCFMT_FIXED_WIDTH;
 537     column.cx = (rc.right - rc.left) - 25;
 538     column.pszText = buf;
 539     SendMessageW(lv, LVM_INSERTCOLUMNW, 0, (LPARAM)&column);
 540
 541     for(i =0; i < ARRAY_SIZE(resources); i++)
 542     {
 543         item.mask = LVIF_TEXT;
 544         item.iItem = SendMessageW(lv, LVM_GETITEMCOUNT, 0, 0);
 545         item.iSubItem = 0;
 546         LoadStringW(instance, resources[i], buf, ARRAY_SIZE(buf));
 547         item.pszText = buf;
 548         SendMessageW(lv, LVM_INSERTITEMW, 0, (LPARAM)&item);
 549     }
 550 }
 551
 552 static INT_PTR CALLBACK data_link_advanced_dlg_proc(HWND hwnd, UINT msg, WPARAM wp, LPARAM lp)
 553 {
 554     TRACE("(%p, %08x, %08lx, %08lx)\n", hwnd, msg, wp, lp);
 555
 556     switch (msg)
 557     {
 558         case WM_INITDIALOG:
 559         {
 560             EnableWindow(GetDlgItem(hwnd, IDC_LBL_LEVEL), FALSE);
 561             EnableWindow(GetDlgItem(hwnd, IDC_CBO_LEVEL), FALSE);
 562             EnableWindow(GetDlgItem(hwnd, IDC_LBL_PROTECTION), FALSE);
 563             EnableWindow(GetDlgItem(hwnd, IDC_CBO_PROTECTION), FALSE);
 564
 565             advanced_fill_permission_list(hwnd);
 566
 567             break;
 568         }
 569         default:
 570             break;
 571     }
 572     return 0;
 573 }
 574
 575 static void create_page_all_columns(HWND lv)
 576 {
 577     RECT rc;
 578     WCHAR buf[256];
 579     LVCOLUMNW column;
 580
 581     SendMessageW(lv, LVM_SETEXTENDEDLISTVIEWSTYLE, 0, LVS_EX_FULLROWSELECT);
 582     GetWindowRect(lv, &rc);
 583     LoadStringW(instance, IDS_COL_NAME, buf, ARRAY_SIZE(buf));
 584     column.mask = LVCF_WIDTH | LVCF_TEXT;
 585     column.cx = (rc.right / 2);
 586     column.pszText = buf;
 587     SendMessageW(lv, LVM_INSERTCOLUMNW, 0, (LPARAM)&column);
 588
 589     LoadStringW(instance, IDS_COL_VALUE, buf, ARRAY_SIZE(buf));
 590     column.mask = LVCF_WIDTH | LVCF_TEXT;
 591     column.cx = (rc.right / 2);
 592     column.pszText = buf;
 593     SendMessageW(lv, LVM_INSERTCOLUMNW, 0, (LPARAM)&column);
 594 }
 595
 596 static INT_PTR CALLBACK data_link_all_dlg_proc(HWND hwnd, UINT msg, WPARAM wp, LPARAM lp)
 597 {
 598     TRACE("(%p, %08x, %08lx, %08lx)\n", hwnd, msg, wp, lp);
 599
 600     switch (msg)
 601     {
 602         case WM_INITDIALOG:
 603         {
 604             HWND lv = GetDlgItem(hwnd, IDC_LST_PROPERTIES);
 605             create_page_all_columns(lv);
 606             break;
 607         }
 608         case WM_COMMAND:
 609         {
 610             if (LOWORD(wp) == IDC_BTN_EDIT)
 611             {
 612                  /* TODO: Implement Connection dialog */
 613                  MessageBoxA(hwnd, "Not implemented yet.", "Error", MB_OK | MB_ICONEXCLAMATION);
 614             }
 615          }
 616      }
 617
 618      return 0;
 619  }
 620
 621 static HRESULT WINAPI dslocator_PromptNew(IDataSourceLocator *iface, IDispatch **connection)
 622 {
 623     DSLocatorImpl *This = impl_from_IDataSourceLocator(iface);
 624     PROPSHEETHEADERW hdr;
 625     PROPSHEETPAGEW pages[4];
 626     INT_PTR ret;
 627
 628     FIXME("(%p, %p) Semi-stub\n", iface, connection);
 629
 630     if(!connection)
 631         return E_INVALIDARG;
 632
 633     *connection = NULL;
 634
 635     memset(&pages, 0, sizeof(pages));
 636
 637     pages[0].dwSize = sizeof(PROPSHEETPAGEW);
 638     pages[0].hInstance = instance;
 639     pages[0].u.pszTemplate = MAKEINTRESOURCEW(IDD_PROVIDER);
 640     pages[0].pfnDlgProc = data_link_properties_dlg_proc;
 641
 642     pages[1].dwSize = sizeof(PROPSHEETPAGEW);
 643     pages[1].hInstance = instance;
 644     pages[1].u.pszTemplate = MAKEINTRESOURCEW(IDD_CONNECTION);
 645     pages[1].pfnDlgProc = data_link_connection_dlg_proc;
 646
 647     pages[2].dwSize = sizeof(PROPSHEETPAGEW);
 648     pages[2].hInstance = instance;
 649     pages[2].u.pszTemplate = MAKEINTRESOURCEW(IDD_ADVANCED);
 650     pages[2].pfnDlgProc = data_link_advanced_dlg_proc;
 651
 652     pages[3].dwSize = sizeof(pages[0]);
 653     pages[3].hInstance = instance;
 654     pages[3].u.pszTemplate = MAKEINTRESOURCEW(IDD_ALL);
 655     pages[3].pfnDlgProc = data_link_all_dlg_proc;
 656
 657     memset(&hdr, 0, sizeof(hdr));
 658     hdr.dwSize = sizeof(hdr);
 659     hdr.hwndParent = This->hwnd;
 660     hdr.dwFlags = PSH_NOAPPLYNOW | PSH_PROPSHEETPAGE;
 661     hdr.hInstance = instance;
 662     hdr.pszCaption = MAKEINTRESOURCEW(IDS_PROPSHEET_TITLE);
 663     hdr.u3.ppsp = pages;
 664     hdr.nPages = ARRAY_SIZE(pages);
 665     ret = PropertySheetW(&hdr);
 666
 667     return ret ? S_OK : S_FALSE;
 668 }
 669
 670 static HRESULT WINAPI dslocator_PromptEdit(IDataSourceLocator *iface, IDispatch **ppADOConnection, VARIANT_BOOL *success)
 671 {
 672     DSLocatorImpl *This = impl_from_IDataSourceLocator(iface);
 673
 674     FIXME("(%p)->(%p %p)\n",This, ppADOConnection, success);
 675
 676     return E_NOTIMPL;
 677 }
 678
 679 static const IDataSourceLocatorVtbl DSLocatorVtbl =
 680 {
 681     dslocator_QueryInterface,
 682     dslocator_AddRef,
 683     dslocator_Release,
 684     dslocator_GetTypeInfoCount,
 685     dslocator_GetTypeInfo,
 686     dslocator_GetIDsOfNames,
 687     dslocator_Invoke,
 688     dslocator_get_hWnd,
 689     dslocator_put_hWnd,
 690     dslocator_PromptNew,
 691     dslocator_PromptEdit
 692 };
 693
 694 static HRESULT WINAPI datainitialize_QueryInterface(IDataInitialize *iface, REFIID riid, void **obj)
 695 {
 696     DSLocatorImpl *This = impl_from_IDataInitialize(iface);
 697     return IDataSourceLocator_QueryInterface(&This->IDataSourceLocator_iface, riid, obj);
 698 }
 699
 700 static ULONG WINAPI datainitialize_AddRef(IDataInitialize *iface)
 701 {
 702     DSLocatorImpl *This = impl_from_IDataInitialize(iface);
 703     return IDataSourceLocator_AddRef(&This->IDataSourceLocator_iface);
 704 }
 705
 706 static ULONG WINAPI datainitialize_Release(IDataInitialize *iface)
 707 {
 708     DSLocatorImpl *This = impl_from_IDataInitialize(iface);
 709     return IDataSourceLocator_Release(&This->IDataSourceLocator_iface);
 710 }
 711
 712 static HRESULT WINAPI datainitialize_GetDataSource(IDataInitialize *iface,
 713     IUnknown *outer, DWORD context, LPWSTR initstring, REFIID riid, IUnknown **datasource)
 714 {
 715     TRACE("(%p)->(%p %#x %s %s %p)\n", iface, outer, context, debugstr_w(initstring), debugstr_guid(riid),
 716         datasource);
 717
 718     return get_data_source(outer, context, initstring, riid, datasource);
 719 }
 720
 721 static HRESULT WINAPI datainitialize_GetInitializationString(IDataInitialize *iface, IUnknown *datasource,
 722     boolean include_password, LPWSTR *initstring)
 723 {
 724     FIXME("(%p)->(%d %p): stub\n", iface, include_password, initstring);
 725     return E_NOTIMPL;
 726 }
 727
 728 static HRESULT WINAPI datainitialize_CreateDBInstance(IDataInitialize *iface, REFCLSID prov, IUnknown *outer,
 729     DWORD clsctx, LPWSTR reserved, REFIID riid, IUnknown **datasource)
 730 {
 731     FIXME("(%p)->(%s %p %#x %p %s %p): stub\n", iface, debugstr_guid(prov), outer, clsctx, reserved,
 732         debugstr_guid(riid), datasource);
 733     return E_NOTIMPL;
 734 }
 735
 736 static HRESULT WINAPI datainitialize_CreateDBInstanceEx(IDataInitialize *iface, REFCLSID prov, IUnknown *outer,
 737     DWORD clsctx, LPWSTR reserved, COSERVERINFO *server_info, DWORD cmq, MULTI_QI *results)
 738 {
 739     FIXME("(%p)->(%s %p %#x %p %p %u %p): stub\n", iface, debugstr_guid(prov), outer, clsctx, reserved,
 740         server_info, cmq, results);
 741     return E_NOTIMPL;
 742 }
 743
 744 static HRESULT WINAPI datainitialize_LoadStringFromStorage(IDataInitialize *iface, LPWSTR filename, LPWSTR *initstring)
 745 {
 746     FIXME("(%p)->(%s %p): stub\n", iface, debugstr_w(filename), initstring);
 747     return E_NOTIMPL;
 748 }
 749
 750 static HRESULT WINAPI datainitialize_WriteStringToStorage(IDataInitialize *iface, LPWSTR filename, LPWSTR initstring,
 751     DWORD disposition)
 752 {
 753     FIXME("(%p)->(%s %s %#x): stub\n", iface, debugstr_w(filename), debugstr_w(initstring), disposition);
 754     return E_NOTIMPL;
 755 }
 756
 757 static const IDataInitializeVtbl ds_datainitialize_vtbl =
 758 {
 759     datainitialize_QueryInterface,
 760     datainitialize_AddRef,
 761     datainitialize_Release,
 762     datainitialize_GetDataSource,
 763     datainitialize_GetInitializationString,
 764     datainitialize_CreateDBInstance,
 765     datainitialize_CreateDBInstanceEx,
 766     datainitialize_LoadStringFromStorage,
 767     datainitialize_WriteStringToStorage,
 768 };
 769
 770 HRESULT create_dslocator(IUnknown *outer, void **obj)
 771 {
 772     DSLocatorImpl *This;
 773
 774     TRACE("(%p, %p)\n", outer, obj);
 775
 776     *obj = NULL;
 777
 778     if(outer) return CLASS_E_NOAGGREGATION;
 779
 780     This = heap_alloc(sizeof(*This));
 781     if(!This) return E_OUTOFMEMORY;
 782
 783     This->IDataSourceLocator_iface.lpVtbl = &DSLocatorVtbl;
 784     This->IDataInitialize_iface.lpVtbl = &ds_datainitialize_vtbl;
 785     This->ref = 1;
 786     This->hwnd = 0;
 787
 788     *obj = &This->IDataSourceLocator_iface;
 789
 790     return S_OK;
 791 }