1 /* OLE DB Conversion library
3 * Copyright 2009 Huw Davies
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.
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.
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
23 #define NONAMELESSUNION
24 #define NONAMELESSSTRUCT
33 #include "oledb_private.h"
35 #include "wine/debug.h"
37 WINE_DEFAULT_DEBUG_CHANNEL(oledb);
41 const struct IDataConvertVtbl *lpVtbl;
42 const struct IDCInfoVtbl *lpDCInfoVtbl;
46 UINT version; /* Set by IDCInfo_SetInfo */
49 static inline convert *impl_from_IDataConvert(IDataConvert *iface)
51 return (convert *)((char*)iface - FIELD_OFFSET(convert, lpVtbl));
54 static inline convert *impl_from_IDCInfo(IDCInfo *iface)
56 return (convert *)((char*)iface - FIELD_OFFSET(convert, lpDCInfoVtbl));
59 static HRESULT WINAPI convert_QueryInterface(IDataConvert* iface,
63 convert *This = impl_from_IDataConvert(iface);
64 TRACE("(%p)->(%s, %p)\n", This, debugstr_guid(riid), obj);
68 if(IsEqualIID(riid, &IID_IUnknown) ||
69 IsEqualIID(riid, &IID_IDataConvert))
73 else if(IsEqualIID(riid, &IID_IDCInfo))
75 *obj = &This->lpDCInfoVtbl;
79 FIXME("interface %s not implemented\n", debugstr_guid(riid));
83 IDataConvert_AddRef(iface);
88 static ULONG WINAPI convert_AddRef(IDataConvert* iface)
90 convert *This = impl_from_IDataConvert(iface);
91 TRACE("(%p)\n", This);
93 return InterlockedIncrement(&This->ref);
97 static ULONG WINAPI convert_Release(IDataConvert* iface)
99 convert *This = impl_from_IDataConvert(iface);
102 TRACE("(%p)\n", This);
104 ref = InterlockedDecrement(&This->ref);
107 HeapFree(GetProcessHeap(), 0, This);
113 static int get_length(DBTYPE type)
131 case DBTYPE_FILETIME:
132 return sizeof(FILETIME);
134 case DBTYPE_BYREF | DBTYPE_WSTR:
137 FIXME("Unhandled type %04x\n", type);
142 static HRESULT WINAPI convert_DataConvert(IDataConvert* iface,
143 DBTYPE src_type, DBTYPE dst_type,
144 DBLENGTH src_len, DBLENGTH *dst_len,
145 void *src, void *dst,
146 DBLENGTH dst_max_len,
147 DBSTATUS src_status, DBSTATUS *dst_status,
148 BYTE precision, BYTE scale,
151 convert *This = impl_from_IDataConvert(iface);
154 TRACE("(%p)->(%d, %d, %d, %p, %p, %p, %d, %d, %p, %d, %d, %x)\n", This,
155 src_type, dst_type, src_len, dst_len, src, dst, dst_max_len,
156 src_status, dst_status, precision, scale, flags);
158 *dst_len = get_length(dst_type);
159 *dst_status = DBSTATUS_E_BADACCESSOR;
161 if(IDataConvert_CanConvert(iface, src_type, dst_type) != S_OK)
163 return DB_E_UNSUPPORTEDCONVERSION;
166 if(src_type == DBTYPE_STR)
171 if(flags & DBDATACONVERT_LENGTHFROMNTS)
172 len = MultiByteToWideChar(CP_ACP, 0, src, -1, NULL, 0) - 1;
174 len = MultiByteToWideChar(CP_ACP, 0, src, src_len, NULL, 0);
175 b = SysAllocStringLen(NULL, len);
176 if(!b) return E_OUTOFMEMORY;
177 if(flags & DBDATACONVERT_LENGTHFROMNTS)
178 MultiByteToWideChar(CP_ACP, 0, src, -1, b, len + 1);
180 MultiByteToWideChar(CP_ACP, 0, src, src_len, b, len);
182 hr = IDataConvert_DataConvert(iface, DBTYPE_BSTR, dst_type, 0, dst_len,
183 &b, dst, dst_max_len, src_status, dst_status,
184 precision, scale, flags);
190 if(src_type == DBTYPE_WSTR)
194 if(flags & DBDATACONVERT_LENGTHFROMNTS)
195 b = SysAllocString(src);
197 b = SysAllocStringLen(src, src_len / 2);
198 if(!b) return E_OUTOFMEMORY;
199 hr = IDataConvert_DataConvert(iface, DBTYPE_BSTR, dst_type, 0, dst_len,
200 &b, dst, dst_max_len, src_status, dst_status,
201 precision, scale, flags);
210 signed short *d = dst;
213 case DBTYPE_EMPTY: *d = 0; hr = S_OK; break;
214 case DBTYPE_I2: *d = *(signed short*)src; hr = S_OK; break;
215 case DBTYPE_I4: hr = VarI2FromI4(*(signed int*)src, d); break;
216 case DBTYPE_R4: hr = VarI2FromR4(*(FLOAT*)src, d); break;
217 case DBTYPE_R8: hr = VarI2FromR8(*(double*)src, d); break;
218 case DBTYPE_CY: hr = VarI2FromCy(*(CY*)src, d); break;
219 case DBTYPE_DATE: hr = VarI2FromDate(*(DATE*)src, d); break;
220 case DBTYPE_BSTR: hr = VarI2FromStr(*(WCHAR**)src, LOCALE_USER_DEFAULT, 0, d); break;
221 case DBTYPE_BOOL: hr = VarI2FromBool(*(VARIANT_BOOL*)src, d); break;
222 case DBTYPE_DECIMAL: hr = VarI2FromDec((DECIMAL*)src, d); break;
223 case DBTYPE_I1: hr = VarI2FromI1(*(signed char*)src, d); break;
224 case DBTYPE_UI1: hr = VarI2FromUI1(*(BYTE*)src, d); break;
225 case DBTYPE_UI2: hr = VarI2FromUI2(*(WORD*)src, d); break;
226 case DBTYPE_UI4: hr = VarI2FromUI4(*(DWORD*)src, d); break;
227 case DBTYPE_I8: hr = VarI2FromI8(*(LONGLONG*)src, d); break;
228 case DBTYPE_UI8: hr = VarI2FromUI8(*(ULONGLONG*)src, d); break;
229 default: FIXME("Unimplemented conversion %04x -> I2\n", src_type); return E_NOTIMPL;
239 case DBTYPE_EMPTY: *d = 0; hr = S_OK; break;
240 case DBTYPE_I2: hr = VarI4FromI2(*(signed short*)src, d); break;
241 case DBTYPE_I4: *d = *(signed int*)src; hr = S_OK; break;
242 case DBTYPE_R4: hr = VarI4FromR4(*(FLOAT*)src, d); break;
243 case DBTYPE_R8: hr = VarI4FromR8(*(double*)src, d); break;
244 case DBTYPE_CY: hr = VarI4FromCy(*(CY*)src, d); break;
245 case DBTYPE_DATE: hr = VarI4FromDate(*(DATE*)src, d); break;
246 case DBTYPE_BSTR: hr = VarI4FromStr(*(WCHAR**)src, LOCALE_USER_DEFAULT, 0, d); break;
247 case DBTYPE_BOOL: hr = VarI4FromBool(*(VARIANT_BOOL*)src, d); break;
248 case DBTYPE_DECIMAL: hr = VarI4FromDec((DECIMAL*)src, d); break;
249 case DBTYPE_I1: hr = VarI4FromI1(*(signed char*)src, d); break;
250 case DBTYPE_UI1: hr = VarI4FromUI1(*(BYTE*)src, d); break;
251 case DBTYPE_UI2: hr = VarI4FromUI2(*(WORD*)src, d); break;
252 case DBTYPE_UI4: hr = VarI4FromUI4(*(DWORD*)src, d); break;
253 case DBTYPE_I8: hr = VarI4FromI8(*(LONGLONG*)src, d); break;
254 case DBTYPE_UI8: hr = VarI4FromUI8(*(ULONGLONG*)src, d); break;
255 default: FIXME("Unimplemented conversion %04x -> I4\n", src_type); return E_NOTIMPL;
265 case DBTYPE_EMPTY: *d = SysAllocStringLen(NULL, 0); hr = *d ? S_OK : E_OUTOFMEMORY; break;
266 case DBTYPE_I2: hr = VarBstrFromI2(*(signed short*)src, LOCALE_USER_DEFAULT, 0, d); break;
267 case DBTYPE_I4: hr = VarBstrFromI4(*(signed int*)src, LOCALE_USER_DEFAULT, 0, d); break;
268 case DBTYPE_R4: hr = VarBstrFromR4(*(FLOAT*)src, LOCALE_USER_DEFAULT, 0, d); break;
269 case DBTYPE_R8: hr = VarBstrFromR8(*(double*)src, LOCALE_USER_DEFAULT, 0, d); break;
270 case DBTYPE_CY: hr = VarBstrFromCy(*(CY*)src, LOCALE_USER_DEFAULT, 0, d); break;
271 case DBTYPE_DATE: hr = VarBstrFromDate(*(DATE*)src, LOCALE_USER_DEFAULT, 0, d); break;
272 case DBTYPE_BSTR: *d = SysAllocStringLen(*(BSTR*)src, SysStringLen(*(BSTR*)src)); hr = *d ? S_OK : E_OUTOFMEMORY; break;
273 case DBTYPE_BOOL: hr = VarBstrFromBool(*(VARIANT_BOOL*)src, LOCALE_USER_DEFAULT, 0, d); break;
274 case DBTYPE_DECIMAL: hr = VarBstrFromDec((DECIMAL*)src, LOCALE_USER_DEFAULT, 0, d); break;
275 case DBTYPE_I1: hr = VarBstrFromI1(*(signed char*)src, LOCALE_USER_DEFAULT, 0, d); break;
276 case DBTYPE_UI1: hr = VarBstrFromUI1(*(BYTE*)src, LOCALE_USER_DEFAULT, 0, d); break;
277 case DBTYPE_UI2: hr = VarBstrFromUI2(*(WORD*)src, LOCALE_USER_DEFAULT, 0, d); break;
278 case DBTYPE_UI4: hr = VarBstrFromUI4(*(DWORD*)src, LOCALE_USER_DEFAULT, 0, d); break;
279 case DBTYPE_I8: hr = VarBstrFromI8(*(LONGLONG*)src, LOCALE_USER_DEFAULT, 0, d); break;
280 case DBTYPE_UI8: hr = VarBstrFromUI8(*(ULONGLONG*)src, LOCALE_USER_DEFAULT, 0, d); break;
281 default: FIXME("Unimplemented conversion %04x -> BSTR\n", src_type); return E_NOTIMPL;
286 case DBTYPE_FILETIME:
291 case DBTYPE_EMPTY: d->dwLowDateTime = d->dwHighDateTime = 0; hr = S_OK; break;
292 case DBTYPE_FILETIME: *d = *(FILETIME*)src; hr = S_OK; break;
293 default: FIXME("Unimplemented conversion %04x -> FILETIME\n", src_type); return E_NOTIMPL;
303 hr = IDataConvert_DataConvert(iface, src_type, DBTYPE_BSTR, src_len, &bstr_len,
304 src, &b, sizeof(BSTR), src_status, dst_status,
305 precision, scale, flags);
306 if(hr != S_OK) return hr;
307 bstr_len = SysStringLen(b);
308 *dst_len = bstr_len * sizeof(WCHAR); /* Doesn't include size for '\0' */
309 *dst_status = DBSTATUS_S_OK;
310 bytes_to_copy = min(*dst_len + sizeof(WCHAR), dst_max_len);
313 if(bytes_to_copy >= sizeof(WCHAR))
315 memcpy(dst, b, bytes_to_copy - sizeof(WCHAR));
316 *((WCHAR*)dst + bytes_to_copy / sizeof(WCHAR) - 1) = 0;
317 if(bytes_to_copy < *dst_len + sizeof(WCHAR))
318 *dst_status = DBSTATUS_S_TRUNCATED;
322 *dst_status = DBSTATUS_E_DATAOVERFLOW;
323 hr = DB_E_ERRORSOCCURRED;
330 case DBTYPE_BYREF | DBTYPE_WSTR:
335 hr = IDataConvert_DataConvert(iface, src_type, DBTYPE_BSTR, src_len, &bstr_len,
336 src, &b, sizeof(BSTR), src_status, dst_status,
337 precision, scale, flags);
338 if(hr != S_OK) return hr;
340 bstr_len = SysStringLen(b) * sizeof(WCHAR);
341 *dst_len = bstr_len; /* Doesn't include size for '\0' */
343 *d = CoTaskMemAlloc(bstr_len + sizeof(WCHAR));
344 if(*d) memcpy(*d, b, bstr_len + sizeof(WCHAR));
345 else hr = E_OUTOFMEMORY;
351 FIXME("Unimplemented conversion %04x -> %04x\n", src_type, dst_type);
356 if(hr == DISP_E_OVERFLOW)
358 *dst_status = DBSTATUS_E_DATAOVERFLOW;
359 hr = DB_E_ERRORSOCCURRED;
362 *dst_status = DBSTATUS_S_OK;
367 static inline WORD get_dbtype_class(DBTYPE type)
394 case DBTYPE_DBTIMESTAMP:
395 return DBTYPE_DBDATE;
400 /* Many src types will convert to this group of dst types */
401 static inline BOOL common_class(WORD dst_class)
421 static inline BOOL array_type(DBTYPE type)
423 return (type >= DBTYPE_I2 && type <= DBTYPE_UI4);
426 static HRESULT WINAPI convert_CanConvert(IDataConvert* iface,
427 DBTYPE src_type, DBTYPE dst_type)
429 convert *This = impl_from_IDataConvert(iface);
430 DBTYPE src_base_type = src_type & 0x1ff;
431 DBTYPE dst_base_type = dst_type & 0x1ff;
432 WORD dst_class = get_dbtype_class(dst_base_type);
434 TRACE("(%p)->(%d, %d)\n", This, src_type, dst_type);
436 if(src_type & DBTYPE_VECTOR || dst_type & DBTYPE_VECTOR) return S_FALSE;
438 if(src_type & DBTYPE_ARRAY)
440 if(!array_type(src_base_type)) return S_FALSE;
441 if(dst_type & DBTYPE_ARRAY)
443 if(src_type == dst_type) return S_OK;
446 if(dst_type == DBTYPE_VARIANT) return S_OK;
450 if(dst_type & DBTYPE_ARRAY)
452 if(!array_type(dst_base_type)) return S_FALSE;
453 if(src_type == DBTYPE_IDISPATCH || src_type == DBTYPE_VARIANT) return S_OK;
457 if(dst_type & DBTYPE_BYREF)
458 if(dst_base_type != DBTYPE_BYTES && dst_base_type != DBTYPE_STR && dst_base_type != DBTYPE_WSTR)
461 switch(get_dbtype_class(src_base_type))
464 if(common_class(dst_class)) return S_OK;
469 case DBTYPE_FILETIME:
472 if(dst_base_type == DBTYPE_DBTIMESTAMP) return S_OK;
477 switch(dst_base_type)
481 case DBTYPE_FILETIME: return S_OK;
482 default: return S_FALSE;
486 if(dst_base_type == DBTYPE_BYTES) return S_OK;
489 if(dst_base_type == DBTYPE_DATE) return S_OK;
492 if(common_class(dst_class)) return S_OK;
493 if(dst_class == DBTYPE_DBDATE) return S_OK;
497 if(dst_base_type == DBTYPE_DATE) return S_OK;
500 if(common_class(dst_class)) return S_OK;
504 if(common_class(dst_class)) return S_OK;
505 switch(dst_base_type)
508 case DBTYPE_FILETIME: return S_OK;
509 default: return S_FALSE;
525 case DBTYPE_FILETIME:
527 default: return S_FALSE;
530 case DBTYPE_IDISPATCH:
532 switch(dst_base_type)
534 case DBTYPE_IDISPATCH:
536 case DBTYPE_IUNKNOWN:
541 if(common_class(dst_class)) return S_OK;
548 case DBTYPE_FILETIME:
550 default: return S_FALSE;
554 switch(dst_base_type)
561 default: return S_FALSE;
564 case DBTYPE_IUNKNOWN:
565 switch(dst_base_type)
569 case DBTYPE_IDISPATCH:
571 case DBTYPE_IUNKNOWN:
573 default: return S_FALSE;
577 if(dst_class == DBTYPE_I4 || dst_class == DBTYPE_I8) return S_OK;
589 default: return S_FALSE;
592 case DBTYPE_FILETIME:
593 if(dst_class == DBTYPE_I8) return S_OK;
604 case DBTYPE_FILETIME:
606 default: return S_FALSE;
613 static HRESULT WINAPI convert_GetConversionSize(IDataConvert* iface,
614 DBTYPE wSrcType, DBTYPE wDstType,
615 DBLENGTH *pcbSrcLength, DBLENGTH *pcbDstLength,
618 convert *This = impl_from_IDataConvert(iface);
619 FIXME("(%p)->(%d, %d, %p, %p, %p): stub\n", This, wSrcType, wDstType, pcbSrcLength, pcbDstLength, pSrc);
624 static const struct IDataConvertVtbl convert_vtbl =
626 convert_QueryInterface,
631 convert_GetConversionSize
634 static HRESULT WINAPI dcinfo_QueryInterface(IDCInfo* iface, REFIID riid, void **obj)
636 convert *This = impl_from_IDCInfo(iface);
638 return IDataConvert_QueryInterface((IDataConvert *)This, riid, obj);
641 static ULONG WINAPI dcinfo_AddRef(IDCInfo* iface)
643 convert *This = impl_from_IDCInfo(iface);
645 return IDataConvert_AddRef((IDataConvert *)This);
648 static ULONG WINAPI dcinfo_Release(IDCInfo* iface)
650 convert *This = impl_from_IDCInfo(iface);
652 return IDataConvert_Release((IDataConvert *)This);
655 static HRESULT WINAPI dcinfo_GetInfo(IDCInfo *iface, ULONG num, DCINFOTYPE types[], DCINFO **info_ptr)
657 convert *This = impl_from_IDCInfo(iface);
661 TRACE("(%p)->(%d, %p, %p)\n", This, num, types, info_ptr);
663 *info_ptr = infos = CoTaskMemAlloc(num * sizeof(*infos));
664 if(!infos) return E_OUTOFMEMORY;
666 for(i = 0; i < num; i++)
668 infos[i].eInfoType = types[i];
669 VariantInit(&infos[i].vData);
673 case DCINFOTYPE_VERSION:
674 V_VT(&infos[i].vData) = VT_UI4;
675 V_UI4(&infos[i].vData) = This->version;
683 static HRESULT WINAPI dcinfo_SetInfo(IDCInfo* iface, ULONG num, DCINFO info[])
685 convert *This = impl_from_IDCInfo(iface);
689 TRACE("(%p)->(%d, %p)\n", This, num, info);
691 for(i = 0; i < num; i++)
693 switch(info[i].eInfoType)
695 case DCINFOTYPE_VERSION:
696 if(V_VT(&info[i].vData) != VT_UI4)
698 FIXME("VERSION with vt %x\n", V_VT(&info[i].vData));
699 hr = DB_S_ERRORSOCCURRED;
702 This->version = V_UI4(&info[i].vData);
706 FIXME("Unhandled info type %d (vt %x)\n", info[i].eInfoType, V_VT(&info[i].vData));
712 static const struct IDCInfoVtbl dcinfo_vtbl =
714 dcinfo_QueryInterface,
721 HRESULT create_oledb_convert(IUnknown *outer, void **obj)
725 TRACE("(%p, %p)\n", outer, obj);
729 if(outer) return CLASS_E_NOAGGREGATION;
731 This = HeapAlloc(GetProcessHeap(), 0, sizeof(*This));
732 if(!This) return E_OUTOFMEMORY;
734 This->lpVtbl = &convert_vtbl;
735 This->lpDCInfoVtbl = &dcinfo_vtbl;
737 This->version = 0x110;
739 *obj = &This->lpVtbl;