Sweden-Number/dlls/oledb32/dslocator.c

447 lines
13 KiB
C

/* Data Links
*
* Copyright 2013 Alistair Leslie-Hughes
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this library; if not, write to the Free Software
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
*/
#include <stdarg.h>
#include <string.h>
#define COBJMACROS
#define NONAMELESSUNION
#include "windef.h"
#include "winbase.h"
#include "objbase.h"
#include "oleauto.h"
#include "winerror.h"
#include "oledb.h"
#include "oledberr.h"
#include "msdasc.h"
#include "prsht.h"
#include "commctrl.h"
#include "oledb_private.h"
#include "resource.h"
#include "wine/debug.h"
#include "wine/heap.h"
WINE_DEFAULT_DEBUG_CHANNEL(oledb);
typedef struct DSLocatorImpl
{
IDataSourceLocator IDataSourceLocator_iface;
IDataInitialize IDataInitialize_iface;
LONG ref;
HWND hwnd;
} DSLocatorImpl;
static inline DSLocatorImpl *impl_from_IDataSourceLocator( IDataSourceLocator *iface )
{
return CONTAINING_RECORD(iface, DSLocatorImpl, IDataSourceLocator_iface);
}
static inline DSLocatorImpl *impl_from_IDataInitialize(IDataInitialize *iface)
{
return CONTAINING_RECORD(iface, DSLocatorImpl, IDataInitialize_iface);
}
static HRESULT WINAPI dslocator_QueryInterface(IDataSourceLocator *iface, REFIID riid, void **ppvoid)
{
DSLocatorImpl *This = impl_from_IDataSourceLocator(iface);
TRACE("(%p)->(%s, %p)\n", This, debugstr_guid(riid),ppvoid);
*ppvoid = NULL;
if (IsEqualIID(riid, &IID_IUnknown) ||
IsEqualIID(riid, &IID_IDispatch) ||
IsEqualIID(riid, &IID_IDataSourceLocator))
{
*ppvoid = &This->IDataSourceLocator_iface;
}
else if (IsEqualIID(riid, &IID_IDataInitialize))
{
*ppvoid = &This->IDataInitialize_iface;
}
else if (IsEqualIID(riid, &IID_IRunnableObject))
{
TRACE("IID_IRunnableObject returning NULL\n");
return E_NOINTERFACE;
}
else if (IsEqualIID(riid, &IID_IProvideClassInfo))
{
TRACE("IID_IProvideClassInfo returning NULL\n");
return E_NOINTERFACE;
}
else if (IsEqualIID(riid, &IID_IMarshal))
{
TRACE("IID_IMarshal returning NULL\n");
return E_NOINTERFACE;
}
else if (IsEqualIID(riid, &IID_IRpcOptions))
{
TRACE("IID_IRpcOptions returning NULL\n");
return E_NOINTERFACE;
}
if(*ppvoid)
{
IUnknown_AddRef( (IUnknown*)*ppvoid );
return S_OK;
}
FIXME("interface %s not implemented\n", debugstr_guid(riid));
return E_NOINTERFACE;
}
static ULONG WINAPI dslocator_AddRef(IDataSourceLocator *iface)
{
DSLocatorImpl *This = impl_from_IDataSourceLocator(iface);
TRACE("(%p)->%u\n",This,This->ref);
return InterlockedIncrement(&This->ref);
}
static ULONG WINAPI dslocator_Release(IDataSourceLocator *iface)
{
DSLocatorImpl *This = impl_from_IDataSourceLocator(iface);
ULONG ref = InterlockedDecrement(&This->ref);
TRACE("(%p)->%u\n",This,ref+1);
if (!ref)
{
heap_free(This);
}
return ref;
}
static HRESULT WINAPI dslocator_GetTypeInfoCount(IDataSourceLocator *iface, UINT *pctinfo)
{
DSLocatorImpl *This = impl_from_IDataSourceLocator(iface);
FIXME("(%p)->()\n", This);
return E_NOTIMPL;
}
static HRESULT WINAPI dslocator_GetTypeInfo(IDataSourceLocator *iface, UINT iTInfo, LCID lcid, ITypeInfo **ppTInfo)
{
DSLocatorImpl *This = impl_from_IDataSourceLocator(iface);
FIXME("(%p)->(%u %u %p)\n", This, iTInfo, lcid, ppTInfo);
return E_NOTIMPL;
}
static HRESULT WINAPI dslocator_GetIDsOfNames(IDataSourceLocator *iface, REFIID riid, LPOLESTR *rgszNames,
UINT cNames, LCID lcid, DISPID *rgDispId)
{
DSLocatorImpl *This = impl_from_IDataSourceLocator(iface);
FIXME("(%p)->(%s %p %u %u %p)\n", This, debugstr_guid(riid), rgszNames, cNames, lcid, rgDispId);
return E_NOTIMPL;
}
static HRESULT WINAPI dslocator_Invoke(IDataSourceLocator *iface, DISPID dispIdMember, REFIID riid,
LCID lcid, WORD wFlags, DISPPARAMS *pDispParams, VARIANT *pVarResult, EXCEPINFO *pExcepInfo, UINT *puArgErr)
{
DSLocatorImpl *This = impl_from_IDataSourceLocator(iface);
FIXME("(%p)->(%d %s %d %d %p %p %p %p)\n", This, dispIdMember, debugstr_guid(riid),
lcid, wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr);
return E_NOTIMPL;
}
static HRESULT WINAPI dslocator_get_hWnd(IDataSourceLocator *iface, COMPATIBLE_LONG *phwndParent)
{
DSLocatorImpl *This = impl_from_IDataSourceLocator(iface);
TRACE("(%p)->(%p)\n",This, phwndParent);
*phwndParent = (COMPATIBLE_LONG)This->hwnd;
return S_OK;
}
static HRESULT WINAPI dslocator_put_hWnd(IDataSourceLocator *iface, COMPATIBLE_LONG hwndParent)
{
DSLocatorImpl *This = impl_from_IDataSourceLocator(iface);
TRACE("(%p)->(%p)\n",This, (HWND)hwndParent);
This->hwnd = (HWND)hwndParent;
return S_OK;
}
static void create_connections_columns(HWND lv)
{
RECT rc;
WCHAR buf[256];
LVCOLUMNW column;
SendMessageW(lv, LVM_SETEXTENDEDLISTVIEWSTYLE, 0, LVS_EX_FULLROWSELECT);
GetWindowRect(lv, &rc);
LoadStringW(instance, IDS_COL_PROVIDER, buf, ARRAY_SIZE(buf));
column.mask = LVCF_WIDTH | LVCF_TEXT;
column.cx = (rc.right - rc.left) - 5;
column.pszText = buf;
SendMessageW(lv, LVM_INSERTCOLUMNW, 0, (LPARAM)&column);
}
static void add_connections_providers(HWND lv)
{
static const WCHAR oledbprov[] = {'\\','O','L','E',' ','D','B',' ','P','r','o','v','i','d','e','r',0};
LONG res;
HKEY key = NULL, subkey;
DWORD index = 0;
LONG next_key;
WCHAR provider[MAX_PATH];
WCHAR guidkey[MAX_PATH];
LONG size;
res = RegOpenKeyExW(HKEY_CLASSES_ROOT, L"CLSID", 0, KEY_READ, &key);
if (res == ERROR_FILE_NOT_FOUND)
return;
next_key = RegEnumKeyW(key, index, provider, MAX_PATH);
while (next_key == ERROR_SUCCESS)
{
WCHAR description[MAX_PATH];
lstrcpyW(guidkey, provider);
lstrcatW(guidkey, oledbprov);
res = RegOpenKeyW(key, guidkey, &subkey);
if (res == ERROR_SUCCESS)
{
TRACE("Found %s\n", debugstr_w(guidkey));
size = MAX_PATH;
res = RegQueryValueW(subkey, NULL, description, &size);
if (res == ERROR_SUCCESS)
{
LVITEMW item;
item.mask = LVIF_TEXT;
item.iItem = SendMessageW(lv, LVM_GETITEMCOUNT, 0, 0);
item.iSubItem = 0;
item.pszText = description;
SendMessageW(lv, LVM_INSERTITEMW, 0, (LPARAM)&item);
/* TODO - Add ProgID to item data */
}
RegCloseKey(subkey);
}
index++;
next_key = RegEnumKeyW(key, index, provider, MAX_PATH);
}
RegCloseKey(key);
}
static LRESULT CALLBACK data_link_properties_dlg_proc(HWND hwnd, UINT msg, WPARAM wp, LPARAM lp)
{
TRACE("(%p, %08x, %08lx, %08lx)\n", hwnd, msg, wp, lp);
switch (msg)
{
case WM_INITDIALOG:
{
HWND btn, lv = GetDlgItem(hwnd, IDC_LST_CONNECTIONS);
create_connections_columns(lv);
add_connections_providers(lv);
btn = GetDlgItem(GetParent(hwnd), IDOK);
EnableWindow(btn, FALSE);
break;
}
case WM_COMMAND:
{
if (LOWORD(wp) == IDC_BTN_NEXT)
{
/* TODO: Implement Connection dialog */
MessageBoxA(hwnd, "Not implemented yet.", "Error", MB_OK | MB_ICONEXCLAMATION);
}
break;
}
default:
break;
}
return 0;
}
static HRESULT WINAPI dslocator_PromptNew(IDataSourceLocator *iface, IDispatch **connection)
{
DSLocatorImpl *This = impl_from_IDataSourceLocator(iface);
PROPSHEETHEADERW hdr;
PROPSHEETPAGEW page;
INT_PTR ret;
FIXME("(%p, %p) Semi-stub\n", iface, connection);
if(!connection)
return E_INVALIDARG;
*connection = NULL;
memset(&page, 0, sizeof(PROPSHEETPAGEW));
page.dwSize = sizeof(page);
page.hInstance = instance;
page.u.pszTemplate = MAKEINTRESOURCEW(IDD_PROVIDER);
page.pfnDlgProc = data_link_properties_dlg_proc;
memset(&hdr, 0, sizeof(hdr));
hdr.dwSize = sizeof(hdr);
hdr.hwndParent = This->hwnd;
hdr.dwFlags = PSH_NOAPPLYNOW | PSH_PROPSHEETPAGE;
hdr.hInstance = instance;
hdr.pszCaption = MAKEINTRESOURCEW(IDS_PROPSHEET_TITLE);
hdr.u3.ppsp = &page;
hdr.nPages = 1;
ret = PropertySheetW(&hdr);
return ret ? S_OK : S_FALSE;
}
static HRESULT WINAPI dslocator_PromptEdit(IDataSourceLocator *iface, IDispatch **ppADOConnection, VARIANT_BOOL *success)
{
DSLocatorImpl *This = impl_from_IDataSourceLocator(iface);
FIXME("(%p)->(%p %p)\n",This, ppADOConnection, success);
return E_NOTIMPL;
}
static const IDataSourceLocatorVtbl DSLocatorVtbl =
{
dslocator_QueryInterface,
dslocator_AddRef,
dslocator_Release,
dslocator_GetTypeInfoCount,
dslocator_GetTypeInfo,
dslocator_GetIDsOfNames,
dslocator_Invoke,
dslocator_get_hWnd,
dslocator_put_hWnd,
dslocator_PromptNew,
dslocator_PromptEdit
};
static HRESULT WINAPI datainitialize_QueryInterface(IDataInitialize *iface, REFIID riid, void **obj)
{
DSLocatorImpl *This = impl_from_IDataInitialize(iface);
return IDataSourceLocator_QueryInterface(&This->IDataSourceLocator_iface, riid, obj);
}
static ULONG WINAPI datainitialize_AddRef(IDataInitialize *iface)
{
DSLocatorImpl *This = impl_from_IDataInitialize(iface);
return IDataSourceLocator_AddRef(&This->IDataSourceLocator_iface);
}
static ULONG WINAPI datainitialize_Release(IDataInitialize *iface)
{
DSLocatorImpl *This = impl_from_IDataInitialize(iface);
return IDataSourceLocator_Release(&This->IDataSourceLocator_iface);
}
static HRESULT WINAPI datainitialize_GetDataSource(IDataInitialize *iface,
IUnknown *outer, DWORD context, LPWSTR initstring, REFIID riid, IUnknown **datasource)
{
TRACE("(%p)->(%p %#x %s %s %p)\n", iface, outer, context, debugstr_w(initstring), debugstr_guid(riid),
datasource);
return get_data_source(outer, context, initstring, riid, datasource);
}
static HRESULT WINAPI datainitialize_GetInitializationString(IDataInitialize *iface, IUnknown *datasource,
boolean include_password, LPWSTR *initstring)
{
FIXME("(%p)->(%d %p): stub\n", iface, include_password, initstring);
return E_NOTIMPL;
}
static HRESULT WINAPI datainitialize_CreateDBInstance(IDataInitialize *iface, REFCLSID prov, IUnknown *outer,
DWORD clsctx, LPWSTR reserved, REFIID riid, IUnknown **datasource)
{
FIXME("(%p)->(%s %p %#x %p %s %p): stub\n", iface, debugstr_guid(prov), outer, clsctx, reserved,
debugstr_guid(riid), datasource);
return E_NOTIMPL;
}
static HRESULT WINAPI datainitialize_CreateDBInstanceEx(IDataInitialize *iface, REFCLSID prov, IUnknown *outer,
DWORD clsctx, LPWSTR reserved, COSERVERINFO *server_info, DWORD cmq, MULTI_QI *results)
{
FIXME("(%p)->(%s %p %#x %p %p %u %p): stub\n", iface, debugstr_guid(prov), outer, clsctx, reserved,
server_info, cmq, results);
return E_NOTIMPL;
}
static HRESULT WINAPI datainitialize_LoadStringFromStorage(IDataInitialize *iface, LPWSTR filename, LPWSTR *initstring)
{
FIXME("(%p)->(%s %p): stub\n", iface, debugstr_w(filename), initstring);
return E_NOTIMPL;
}
static HRESULT WINAPI datainitialize_WriteStringToStorage(IDataInitialize *iface, LPWSTR filename, LPWSTR initstring,
DWORD disposition)
{
FIXME("(%p)->(%s %s %#x): stub\n", iface, debugstr_w(filename), debugstr_w(initstring), disposition);
return E_NOTIMPL;
}
static const IDataInitializeVtbl ds_datainitialize_vtbl =
{
datainitialize_QueryInterface,
datainitialize_AddRef,
datainitialize_Release,
datainitialize_GetDataSource,
datainitialize_GetInitializationString,
datainitialize_CreateDBInstance,
datainitialize_CreateDBInstanceEx,
datainitialize_LoadStringFromStorage,
datainitialize_WriteStringToStorage,
};
HRESULT create_dslocator(IUnknown *outer, void **obj)
{
DSLocatorImpl *This;
TRACE("(%p, %p)\n", outer, obj);
*obj = NULL;
if(outer) return CLASS_E_NOAGGREGATION;
This = heap_alloc(sizeof(*This));
if(!This) return E_OUTOFMEMORY;
This->IDataSourceLocator_iface.lpVtbl = &DSLocatorVtbl;
This->IDataInitialize_iface.lpVtbl = &ds_datainitialize_vtbl;
This->ref = 1;
This->hwnd = 0;
*obj = &This->IDataSourceLocator_iface;
return S_OK;
}