2 * Copyright (C) 2012 Alistair Leslie-Hughes
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.
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.
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
21 #define NONAMELESSUNION
22 #define NONAMELESSSTRUCT
31 #include "oledb_private.h"
33 #include "wine/debug.h"
34 #include "wine/unicode.h"
36 WINE_DEFAULT_DEBUG_CHANNEL(oledb
);
38 DEFINE_GUID(DBPROPSET_DBINIT
, 0xc8b522bc, 0x5cf3, 0x11ce, 0xad, 0xe5, 0x00, 0xaa, 0x00, 0x44, 0x77, 0x3d);
40 typedef struct datainit
42 IDataInitialize IDataInitialize_iface
;
47 static inline datainit
*impl_from_IDataInitialize(IDataInitialize
*iface
)
49 return CONTAINING_RECORD(iface
, datainit
, IDataInitialize_iface
);
54 IDBInitialize IDBInitialize_iface
;
55 IDBProperties IDBProperties_iface
;
60 static inline dbinit
*impl_from_IDBInitialize(IDBInitialize
*iface
)
62 return CONTAINING_RECORD(iface
, dbinit
, IDBInitialize_iface
);
65 static inline dbinit
*impl_from_IDBProperties(IDBProperties
*iface
)
67 return CONTAINING_RECORD(iface
, dbinit
, IDBProperties_iface
);
70 static HRESULT WINAPI
dbprops_QueryInterface(IDBProperties
*iface
, REFIID riid
, void **ppvObject
)
72 dbinit
*This
= impl_from_IDBProperties(iface
);
74 return IDBInitialize_QueryInterface(&This
->IDBInitialize_iface
, riid
, ppvObject
);
77 static ULONG WINAPI
dbprops_AddRef(IDBProperties
*iface
)
79 dbinit
*This
= impl_from_IDBProperties(iface
);
81 return IDBInitialize_AddRef(&This
->IDBInitialize_iface
);
84 static ULONG WINAPI
dbprops_Release(IDBProperties
*iface
)
86 dbinit
*This
= impl_from_IDBProperties(iface
);
88 return IDBInitialize_Release(&This
->IDBInitialize_iface
);
91 static HRESULT WINAPI
dbprops_GetProperties(IDBProperties
*iface
, ULONG cPropertyIDSets
,
92 const DBPROPIDSET rgPropertyIDSets
[], ULONG
*pcPropertySets
, DBPROPSET
**prgPropertySets
)
94 dbinit
*This
= impl_from_IDBProperties(iface
);
96 FIXME("(%p)->(%d %p %p %p)\n", This
, cPropertyIDSets
, rgPropertyIDSets
, pcPropertySets
, prgPropertySets
);
101 static HRESULT WINAPI
dbprops_GetPropertyInfo(IDBProperties
*iface
, ULONG cPropertyIDSets
,
102 const DBPROPIDSET rgPropertyIDSets
[], ULONG
*pcPropertyInfoSets
,
103 DBPROPINFOSET
**prgPropertyInfoSets
, OLECHAR
**ppDescBuffer
)
105 dbinit
*This
= impl_from_IDBProperties(iface
);
107 FIXME("(%p)->(%d %p %p %p %p)\n", This
, cPropertyIDSets
, rgPropertyIDSets
, pcPropertyInfoSets
,
108 prgPropertyInfoSets
, ppDescBuffer
);
113 static HRESULT WINAPI
dbprops_SetProperties(IDBProperties
*iface
, ULONG cPropertySets
,
114 DBPROPSET rgPropertySets
[])
116 dbinit
*This
= impl_from_IDBProperties(iface
);
118 FIXME("(%p)->(%d %p)\n", This
, cPropertySets
, rgPropertySets
);
123 static const struct IDBPropertiesVtbl dbprops_vtbl
=
125 dbprops_QueryInterface
,
128 dbprops_GetProperties
,
129 dbprops_GetPropertyInfo
,
130 dbprops_SetProperties
133 static HRESULT WINAPI
dbinit_QueryInterface(IDBInitialize
*iface
, REFIID riid
, void **obj
)
135 dbinit
*This
= impl_from_IDBInitialize(iface
);
136 TRACE("(%p)->(%s, %p)\n", This
, debugstr_guid(riid
), obj
);
140 if(IsEqualIID(riid
, &IID_IUnknown
) ||
141 IsEqualIID(riid
, &IID_IDBInitialize
))
145 else if(IsEqualIID(riid
, &IID_IDBProperties
))
147 *obj
= &This
->IDBProperties_iface
;
151 FIXME("interface %s not implemented\n", debugstr_guid(riid
));
152 return E_NOINTERFACE
;
155 IDBInitialize_AddRef(iface
);
159 static ULONG WINAPI
dbinit_AddRef(IDBInitialize
*iface
)
161 dbinit
*This
= impl_from_IDBInitialize(iface
);
162 TRACE("(%p)\n", This
);
164 return InterlockedIncrement(&This
->ref
);
167 static ULONG WINAPI
dbinit_Release(IDBInitialize
*iface
)
169 dbinit
*This
= impl_from_IDBInitialize(iface
);
172 TRACE("(%p)\n", This
);
174 ref
= InterlockedDecrement(&This
->ref
);
177 HeapFree(GetProcessHeap(), 0, This
);
183 static HRESULT WINAPI
dbinit_Initialize(IDBInitialize
*iface
)
185 dbinit
*This
= impl_from_IDBInitialize(iface
);
187 FIXME("(%p) stub\n", This
);
192 static HRESULT WINAPI
dbinit_Uninitialize(IDBInitialize
*iface
)
194 dbinit
*This
= impl_from_IDBInitialize(iface
);
196 FIXME("(%p) stub\n", This
);
201 static const IDBInitializeVtbl dbinit_vtbl
=
203 dbinit_QueryInterface
,
210 static HRESULT
create_db_init(void **obj
)
218 This
= HeapAlloc(GetProcessHeap(), 0, sizeof(*This
));
219 if(!This
) return E_OUTOFMEMORY
;
221 This
->IDBInitialize_iface
.lpVtbl
= &dbinit_vtbl
;
222 This
->IDBProperties_iface
.lpVtbl
= &dbprops_vtbl
;
225 *obj
= &This
->IDBInitialize_iface
;
230 static HRESULT WINAPI
datainit_QueryInterface(IDataInitialize
*iface
, REFIID riid
, void **obj
)
232 datainit
*This
= impl_from_IDataInitialize(iface
);
233 TRACE("(%p)->(%s, %p)\n", This
, debugstr_guid(riid
), obj
);
237 if(IsEqualIID(riid
, &IID_IUnknown
) ||
238 IsEqualIID(riid
, &IID_IDataInitialize
))
240 *obj
= &This
->IDataInitialize_iface
;
244 FIXME("interface %s not implemented\n", debugstr_guid(riid
));
245 return E_NOINTERFACE
;
248 IDataInitialize_AddRef(iface
);
252 static ULONG WINAPI
datainit_AddRef(IDataInitialize
*iface
)
254 datainit
*This
= impl_from_IDataInitialize(iface
);
255 TRACE("(%p)\n", This
);
257 return InterlockedIncrement(&This
->ref
);
260 static ULONG WINAPI
datainit_Release(IDataInitialize
*iface
)
262 datainit
*This
= impl_from_IDataInitialize(iface
);
265 TRACE("(%p)\n", This
);
267 ref
= InterlockedDecrement(&This
->ref
);
270 HeapFree(GetProcessHeap(), 0, This
);
276 static void free_dbpropset(ULONG count
, DBPROPSET
*propset
)
280 for (i
= 0; i
< count
; i
++)
284 for (p
= 0; p
< propset
[i
].cProperties
; p
++)
285 VariantClear(&propset
[i
].rgProperties
[p
].vValue
);
287 CoTaskMemFree(propset
[i
].rgProperties
);
289 CoTaskMemFree(propset
);
292 static HRESULT
set_dbpropset(BSTR name
, BSTR value
, DBPROPSET
**propset
)
294 static const WCHAR datasourceW
[] = {'D','a','t','a',' ','S','o','u','r','c','e',0};
296 if (!strcmpW(datasourceW
, name
))
298 *propset
= CoTaskMemAlloc(sizeof(DBPROPSET
));
299 (*propset
)->rgProperties
= CoTaskMemAlloc(sizeof(DBPROP
));
300 (*propset
)->cProperties
= 1;
301 (*propset
)->guidPropertySet
= DBPROPSET_DBINIT
;
302 (*propset
)->rgProperties
[0].dwPropertyID
= DBPROP_INIT_DATASOURCE
;
303 (*propset
)->rgProperties
[0].dwOptions
= DBPROPOPTIONS_REQUIRED
;
304 (*propset
)->rgProperties
[0].dwStatus
= 0;
305 memset(&(*propset
)->rgProperties
[0].colid
, 0, sizeof(DBID
));
306 V_VT(&(*propset
)->rgProperties
[0].vValue
) = VT_BSTR
;
307 V_BSTR(&(*propset
)->rgProperties
[0].vValue
) = SysAllocString(value
);
313 FIXME("unsupported property %s\n", debugstr_w(name
));
318 /*** IDataInitialize methods ***/
319 static HRESULT WINAPI
datainit_GetDataSource(IDataInitialize
*iface
, IUnknown
*outer
, DWORD clsctx
,
320 LPWSTR initstring
, REFIID riid
, IUnknown
**datasource
)
322 static const WCHAR providerW
[] = {'P','r','o','v','i','d','e','r','=',0};
323 static const WCHAR msdasqlW
[] = {'M','S','D','A','S','Q','L',0};
324 datainit
*This
= impl_from_IDataInitialize(iface
);
325 IDBProperties
*dbprops
;
331 TRACE("(%p)->(%p 0x%x %s %s %p)\n", This
, outer
, clsctx
, debugstr_w(initstring
), debugstr_guid(riid
), datasource
);
333 /* first get provider name */
334 provclsid
= IID_NULL
;
335 if (initstring
&& (prov
= strstrW(initstring
, providerW
)))
337 WCHAR
*start
, *progid
;
340 prov
+= sizeof(providerW
)/sizeof(WCHAR
)-1;
342 while (*prov
&& *prov
!= ';')
344 TRACE("got provider %s\n", debugstr_wn(start
, prov
-start
));
347 progid
= CoTaskMemAlloc((len
+1)*sizeof(WCHAR
));
348 if (!progid
) return E_OUTOFMEMORY
;
350 memcpy(progid
, start
, len
*sizeof(WCHAR
));
353 hr
= CLSIDFromProgID(progid
, &provclsid
);
354 CoTaskMemFree(progid
);
357 ERR("provider %s not registered\n", debugstr_wn(start
, prov
-start
));
363 hr
= CLSIDFromProgID(msdasqlW
, &provclsid
);
365 ERR("ODBC provider for OLE DB not registered\n");
368 /* check for provider mismatch if it was specified in init string */
369 if (*datasource
&& prov
)
371 DBPROPIDSET propidset
;
372 enum DBPROPENUM prop
;
376 hr
= IUnknown_QueryInterface(*datasource
, &IID_IDBProperties
, (void**)&dbprops
);
379 WARN("provider doesn't support IDBProperties\n");
383 prop
= DBPROP_INIT_DATASOURCE
;
384 propidset
.rgPropertyIDs
= &prop
;
385 propidset
.cPropertyIDs
= 1;
386 propidset
.guidPropertySet
= DBPROPSET_DBINIT
;
389 hr
= IDBProperties_GetProperties(dbprops
, 1, &propidset
, &count
, &propset
);
390 IDBProperties_Release(dbprops
);
393 WARN("GetProperties failed for datasource, 0x%08x\n", hr
);
397 TRACE("initial data source provider %s\n", debugstr_w(V_BSTR(&propset
->rgProperties
[0].vValue
)));
399 CLSIDFromProgID(V_BSTR(&propset
->rgProperties
[0].vValue
), &initprov
);
400 free_dbpropset(count
, propset
);
402 if (!IsEqualIID(&provclsid
, &initprov
)) return DB_E_MISMATCHEDPROVIDER
;
407 if (!IsEqualIID(&provclsid
, &IID_NULL
))
408 hr
= CoCreateInstance(&provclsid
, outer
, clsctx
, riid
, (void**)datasource
);
410 if (FAILED(hr
) && IsEqualIID(riid
, &IID_IDBInitialize
))
411 hr
= create_db_init((void**)datasource
);
414 /* now set properties */
419 hr
= IUnknown_QueryInterface(*datasource
, &IID_IDBProperties
, (void**)&dbprops
);
422 WARN("provider doesn't support IDBProperties\n");
427 while (start
&& (eq
= strchrW(start
, '=')))
429 static const WCHAR providerW
[] = {'P','r','o','v','i','d','e','r',0};
430 WCHAR
*scol
= strchrW(eq
+1, ';');
433 name
= SysAllocStringLen(start
, eq
- start
);
434 /* skip equal sign to get value */
436 value
= SysAllocStringLen(eq
, scol
? scol
- eq
: -1);
438 /* skip semicolon if present */
442 if (!strcmpW(name
, providerW
))
445 SysFreeString(value
);
449 TRACE("property (name=%s, value=%s)\n", debugstr_w(name
), debugstr_w(value
));
451 hr
= set_dbpropset(name
, value
, &propset
);
453 SysFreeString(value
);
454 if (FAILED(hr
)) return hr
;
456 hr
= IDBProperties_SetProperties(dbprops
, 1, propset
);
457 free_dbpropset(1, propset
);
458 TRACE("provider ret 0x%08x\n", hr
);
461 IDBProperties_Release(dbprops
);
467 /* returns character length of string representation */
468 static int get_propvalue_length(DBPROP
*prop
)
473 if (V_VT(&prop
->vValue
) == VT_BSTR
) return SysStringLen(V_BSTR(&prop
->vValue
));
476 hr
= VariantChangeType(&str
, &prop
->vValue
, 0, VT_BSTR
);
479 int len
= SysStringLen(V_BSTR(&str
));
487 static void write_propvalue_str(WCHAR
*str
, DBPROP
*prop
)
489 VARIANT
*v
= &prop
->vValue
;
493 if (V_VT(v
) == VT_BSTR
)
495 strcatW(str
, V_BSTR(v
));
500 hr
= VariantChangeType(&vstr
, v
, VARIANT_ALPHABOOL
, VT_BSTR
);
503 strcatW(str
, V_BSTR(&vstr
));
508 static WCHAR
*get_propinfo_descr(DBPROP
*prop
, DBPROPINFOSET
*propinfoset
)
512 for (i
= 0; i
< propinfoset
->cPropertyInfos
; i
++)
513 if (propinfoset
->rgPropertyInfos
[i
].dwPropertyID
== prop
->dwPropertyID
)
514 return propinfoset
->rgPropertyInfos
[i
].pwszDescription
;
519 static void free_dbpropinfoset(ULONG count
, DBPROPINFOSET
*propinfoset
)
523 for (i
= 0; i
< count
; i
++)
527 for (p
= 0; p
< propinfoset
[i
].cPropertyInfos
; p
++)
528 VariantClear(&propinfoset
[i
].rgPropertyInfos
[p
].vValues
);
530 CoTaskMemFree(propinfoset
[i
].rgPropertyInfos
);
532 CoTaskMemFree(propinfoset
);
535 static HRESULT WINAPI
datainit_GetInitializationString(IDataInitialize
*iface
, IUnknown
*datasource
,
536 boolean include_pass
, LPWSTR
*init_string
)
538 static const WCHAR provW
[] = {'P','r','o','v','i','d','e','r','=',0};
539 static const WCHAR colW
[] = {';',0};
540 datainit
*This
= impl_from_IDataInitialize(iface
);
541 DBPROPINFOSET
*propinfoset
;
542 IDBProperties
*props
;
543 DBPROPIDSET propidset
;
544 ULONG count
, infocount
;
545 WCHAR
*progid
, *desc
;
552 TRACE("(%p)->(%p %d %p)\n", This
, datasource
, include_pass
, init_string
);
554 /* IPersist support is mandatory for data sources */
555 hr
= IUnknown_QueryInterface(datasource
, &IID_IPersist
, (void**)&persist
);
556 if (FAILED(hr
)) return hr
;
558 memset(&clsid
, 0, sizeof(clsid
));
559 hr
= IPersist_GetClassID(persist
, &clsid
);
560 IPersist_Release(persist
);
561 if (FAILED(hr
)) return hr
;
564 ProgIDFromCLSID(&clsid
, &progid
);
565 TRACE("clsid=%s, progid=%s\n", debugstr_guid(&clsid
), debugstr_w(progid
));
567 /* now get initialization properties */
568 hr
= IUnknown_QueryInterface(datasource
, &IID_IDBProperties
, (void**)&props
);
571 WARN("IDBProperties not supported\n");
572 CoTaskMemFree(progid
);
576 propidset
.rgPropertyIDs
= NULL
;
577 propidset
.cPropertyIDs
= 0;
578 propidset
.guidPropertySet
= DBPROPSET_DBINIT
;
581 hr
= IDBProperties_GetProperties(props
, 1, &propidset
, &count
, &propset
);
584 WARN("failed to get data source properties, 0x%08x\n", hr
);
585 CoTaskMemFree(progid
);
590 IDBProperties_GetPropertyInfo(props
, 1, &propidset
, &infocount
, &propinfoset
, &desc
);
591 IDBProperties_Release(props
);
593 /* check if we need to skip password */
594 len
= strlenW(progid
) + strlenW(provW
) + 1; /* including ';' */
595 for (i
= 0; i
< count
; i
++)
597 WCHAR
*descr
= get_propinfo_descr(&propset
->rgProperties
[i
], propinfoset
);
600 /* include '=' and ';' */
601 len
+= strlenW(descr
) + 2;
602 len
+= get_propvalue_length(&propset
->rgProperties
[i
]);
605 if ((propset
->rgProperties
[i
].dwPropertyID
== DBPROP_AUTH_PERSIST_SENSITIVE_AUTHINFO
) &&
606 (V_BOOL(&propset
->rgProperties
[i
].vValue
) == VARIANT_FALSE
))
607 include_pass
= FALSE
;
610 len
*= sizeof(WCHAR
);
611 *init_string
= CoTaskMemAlloc(len
);
615 strcatW(*init_string
, provW
);
616 strcatW(*init_string
, progid
);
617 strcatW(*init_string
, colW
);
618 CoTaskMemFree(progid
);
620 for (i
= 0; i
< count
; i
++)
624 if (!include_pass
&& propset
->rgProperties
[i
].dwPropertyID
== DBPROP_AUTH_PASSWORD
) continue;
626 descr
= get_propinfo_descr(&propset
->rgProperties
[i
], propinfoset
);
629 static const WCHAR eqW
[] = {'=',0};
630 strcatW(*init_string
, descr
);
631 strcatW(*init_string
, eqW
);
632 write_propvalue_str(*init_string
, &propset
->rgProperties
[i
]);
633 strcatW(*init_string
, colW
);
637 free_dbpropset(count
, propset
);
638 free_dbpropinfoset(infocount
, propinfoset
);
642 TRACE("%s\n", debugstr_w(*init_string
));
646 static HRESULT WINAPI
datainit_CreateDBInstance(IDataInitialize
*iface
, REFCLSID provider
,
647 IUnknown
*outer
, DWORD clsctx
, LPWSTR reserved
, REFIID riid
,
648 IUnknown
**datasource
)
650 datainit
*This
= impl_from_IDataInitialize(iface
);
652 TRACE("(%p)->(%s %p 0x%08x %s %s %p)\n", This
, debugstr_guid(provider
), outer
, clsctx
, debugstr_w(reserved
),
653 debugstr_guid(riid
), datasource
);
655 return CoCreateInstance(provider
, outer
, clsctx
, riid
, (void**)datasource
);
658 static HRESULT WINAPI
datainit_RemoteCreateDBInstanceEx(IDataInitialize
*iface
, REFCLSID clsidProvider
,
659 IUnknown
*pUnkOuter
, DWORD dwClsCtx
, LPWSTR pwszReserved
, COSERVERINFO
*pServerInfo
,
660 DWORD cmq
, GUID
**rgpIID
, IUnknown
**rgpItf
, HRESULT
*rghr
)
662 datainit
*This
= impl_from_IDataInitialize(iface
);
664 FIXME("(%p)->()\n", This
);
669 static HRESULT WINAPI
datainit_LoadStringFromStorage(IDataInitialize
*iface
, LPWSTR pwszFileName
,
670 LPWSTR
*ppwszInitializationString
)
672 datainit
*This
= impl_from_IDataInitialize(iface
);
674 FIXME("(%p)->(%s %p)\n", This
, debugstr_w(pwszFileName
), ppwszInitializationString
);
679 static HRESULT WINAPI
datainit_WriteStringToStorage(IDataInitialize
*iface
, LPWSTR pwszFileName
,
680 LPWSTR pwszInitializationString
, DWORD dwCreationDisposition
)
682 datainit
*This
= impl_from_IDataInitialize(iface
);
684 FIXME("(%p)->(%s %s %d)\n", This
, debugstr_w(pwszFileName
), debugstr_w(pwszInitializationString
), dwCreationDisposition
);
690 static const struct IDataInitializeVtbl datainit_vtbl
=
692 datainit_QueryInterface
,
695 datainit_GetDataSource
,
696 datainit_GetInitializationString
,
697 datainit_CreateDBInstance
,
698 datainit_RemoteCreateDBInstanceEx
,
699 datainit_LoadStringFromStorage
,
700 datainit_WriteStringToStorage
703 HRESULT
create_data_init(IUnknown
*outer
, void **obj
)
707 TRACE("(%p)\n", obj
);
709 if(outer
) return CLASS_E_NOAGGREGATION
;
713 This
= HeapAlloc(GetProcessHeap(), 0, sizeof(*This
));
714 if(!This
) return E_OUTOFMEMORY
;
716 This
->IDataInitialize_iface
.lpVtbl
= &datainit_vtbl
;
719 *obj
= &This
->IDataInitialize_iface
;