?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
oledb32: Fix dst_len in DataConvert.
[wine.git] / dlls / oledb32 / convert.c
blob01c304c339150cf85d4659b8952cf42cfe2c5508
1 /* OLE DB Conversion library
2 *
3 * Copyright 2009 Huw Davies
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
20 #include <stdarg.h>
21
22 #define COBJMACROS
23 #define NONAMELESSUNION
24 #define NONAMELESSSTRUCT
25
26 #include "windef.h"
27 #include "winbase.h"
28 #include "winnls.h"
29 #include "ole2.h"
30 #include "msdadc.h"
31 #include "oledberr.h"
32
33 #include "oledb_private.h"
34
35 #include "wine/debug.h"
36
37 WINE_DEFAULT_DEBUG_CHANNEL(oledb);
38
39 typedef struct
40 {
41 const struct IDataConvertVtbl *lpVtbl;
42 const struct IDCInfoVtbl *lpDCInfoVtbl;
43
44 LONG ref;
45
46 UINT version; /* Set by IDCInfo_SetInfo */
47 } convert;
48
49 static inline convert *impl_from_IDataConvert(IDataConvert *iface)
50 {
51 return (convert *)((char*)iface - FIELD_OFFSET(convert, lpVtbl));
52 }
53
54 static inline convert *impl_from_IDCInfo(IDCInfo *iface)
55 {
56 return (convert *)((char*)iface - FIELD_OFFSET(convert, lpDCInfoVtbl));
57 }
58
59 static HRESULT WINAPI convert_QueryInterface(IDataConvert* iface,
60 REFIID riid,
61 void **obj)
62 {
63 convert *This = impl_from_IDataConvert(iface);
64 TRACE("(%p)->(%s, %p)\n", This, debugstr_guid(riid), obj);
65
66 *obj = NULL;
67
68 if(IsEqualIID(riid, &IID_IUnknown) ||
69 IsEqualIID(riid, &IID_IDataConvert))
70 {
71 *obj = iface;
72 }
73 else if(IsEqualIID(riid, &IID_IDCInfo))
74 {
75 *obj = &This->lpDCInfoVtbl;
76 }
77 else
78 {
79 FIXME("interface %s not implemented\n", debugstr_guid(riid));
80 return E_NOINTERFACE;
81 }
82
83 IDataConvert_AddRef(iface);
84 return S_OK;
85 }
86
87
88 static ULONG WINAPI convert_AddRef(IDataConvert* iface)
89 {
90 convert *This = impl_from_IDataConvert(iface);
91 TRACE("(%p)\n", This);
92
93 return InterlockedIncrement(&This->ref);
94 }
95
96
97 static ULONG WINAPI convert_Release(IDataConvert* iface)
98 {
99 convert *This = impl_from_IDataConvert(iface);
100 LONG ref;
101
102 TRACE("(%p)\n", This);
103
104 ref = InterlockedDecrement(&This->ref);
105 if(ref == 0)
106 {
107 HeapFree(GetProcessHeap(), 0, This);
108 }
109
110 return ref;
111 }
112
113 static int get_length(DBTYPE type)
114 {
115 switch(type)
116 {
117 case DBTYPE_I1:
118 case DBTYPE_UI1:
119 return 1;
120 case DBTYPE_I2:
121 case DBTYPE_UI2:
122 return 2;
123 case DBTYPE_I4:
124 case DBTYPE_UI4:
125 case DBTYPE_R4:
126 return 4;
127 case DBTYPE_I8:
128 case DBTYPE_UI8:
129 return 8;
130 case DBTYPE_CY:
131 return sizeof(CY);
132 case DBTYPE_BSTR:
133 return sizeof(BSTR);
134 case DBTYPE_FILETIME:
135 return sizeof(FILETIME);
136 case DBTYPE_GUID:
137 return sizeof(GUID);
138 case DBTYPE_WSTR:
139 case DBTYPE_STR:
140 case DBTYPE_BYREF | DBTYPE_WSTR:
141 return 0;
142 default:
143 FIXME("Unhandled type %04x\n", type);
144 return 0;
145 }
146 }
147
148 static HRESULT WINAPI convert_DataConvert(IDataConvert* iface,
149 DBTYPE src_type, DBTYPE dst_type,
150 DBLENGTH src_len, DBLENGTH *dst_len,
151 void *src, void *dst,
152 DBLENGTH dst_max_len,
153 DBSTATUS src_status, DBSTATUS *dst_status,
154 BYTE precision, BYTE scale,
155 DBDATACONVERT flags)
156 {
157 convert *This = impl_from_IDataConvert(iface);
158 HRESULT hr;
159
160 TRACE("(%p)->(%d, %d, %d, %p, %p, %p, %d, %d, %p, %d, %d, %x)\n", This,
161 src_type, dst_type, src_len, dst_len, src, dst, dst_max_len,
162 src_status, dst_status, precision, scale, flags);
163
164 *dst_status = DBSTATUS_E_BADACCESSOR;
165
166 if(IDataConvert_CanConvert(iface, src_type, dst_type) != S_OK)
167 {
168 return DB_E_UNSUPPORTEDCONVERSION;
169 }
170
171 if(src_type == DBTYPE_STR)
172 {
173 BSTR b;
174 DWORD len;
175
176 if(flags & DBDATACONVERT_LENGTHFROMNTS)
177 len = MultiByteToWideChar(CP_ACP, 0, src, -1, NULL, 0) - 1;
178 else
179 len = MultiByteToWideChar(CP_ACP, 0, src, src_len, NULL, 0);
180 b = SysAllocStringLen(NULL, len);
181 if(!b) return E_OUTOFMEMORY;
182 if(flags & DBDATACONVERT_LENGTHFROMNTS)
183 MultiByteToWideChar(CP_ACP, 0, src, -1, b, len + 1);
184 else
185 MultiByteToWideChar(CP_ACP, 0, src, src_len, b, len);
186
187 hr = IDataConvert_DataConvert(iface, DBTYPE_BSTR, dst_type, 0, dst_len,
188 &b, dst, dst_max_len, src_status, dst_status,
189 precision, scale, flags);
190
191 SysFreeString(b);
192 return hr;
193 }
194
195 if(src_type == DBTYPE_WSTR)
196 {
197 BSTR b;
198
199 if(flags & DBDATACONVERT_LENGTHFROMNTS)
200 b = SysAllocString(src);
201 else
202 b = SysAllocStringLen(src, src_len / 2);
203 if(!b) return E_OUTOFMEMORY;
204 hr = IDataConvert_DataConvert(iface, DBTYPE_BSTR, dst_type, 0, dst_len,
205 &b, dst, dst_max_len, src_status, dst_status,
206 precision, scale, flags);
207 SysFreeString(b);
208 return hr;
209 }
210
211 switch(dst_type)
212 {
213 case DBTYPE_I2:
214 {
215 signed short *d = dst;
216 switch(src_type)
217 {
218 case DBTYPE_EMPTY: *d = 0; hr = S_OK; break;
219 case DBTYPE_I2: *d = *(signed short*)src; hr = S_OK; break;
220 case DBTYPE_I4: hr = VarI2FromI4(*(signed int*)src, d); break;
221 case DBTYPE_R4: hr = VarI2FromR4(*(FLOAT*)src, d); break;
222 case DBTYPE_R8: hr = VarI2FromR8(*(double*)src, d); break;
223 case DBTYPE_CY: hr = VarI2FromCy(*(CY*)src, d); break;
224 case DBTYPE_DATE: hr = VarI2FromDate(*(DATE*)src, d); break;
225 case DBTYPE_BSTR: hr = VarI2FromStr(*(WCHAR**)src, LOCALE_USER_DEFAULT, 0, d); break;
226 case DBTYPE_BOOL: hr = VarI2FromBool(*(VARIANT_BOOL*)src, d); break;
227 case DBTYPE_DECIMAL: hr = VarI2FromDec((DECIMAL*)src, d); break;
228 case DBTYPE_I1: hr = VarI2FromI1(*(signed char*)src, d); break;
229 case DBTYPE_UI1: hr = VarI2FromUI1(*(BYTE*)src, d); break;
230 case DBTYPE_UI2: hr = VarI2FromUI2(*(WORD*)src, d); break;
231 case DBTYPE_UI4: hr = VarI2FromUI4(*(DWORD*)src, d); break;
232 case DBTYPE_I8: hr = VarI2FromI8(*(LONGLONG*)src, d); break;
233 case DBTYPE_UI8: hr = VarI2FromUI8(*(ULONGLONG*)src, d); break;
234 default: FIXME("Unimplemented conversion %04x -> I2\n", src_type); return E_NOTIMPL;
235 }
236 break;
237 }
238
239 case DBTYPE_I4:
240 {
241 signed int *d = dst;
242 switch(src_type)
243 {
244 case DBTYPE_EMPTY: *d = 0; hr = S_OK; break;
245 case DBTYPE_I2: hr = VarI4FromI2(*(signed short*)src, d); break;
246 case DBTYPE_I4: *d = *(signed int*)src; hr = S_OK; break;
247 case DBTYPE_R4: hr = VarI4FromR4(*(FLOAT*)src, d); break;
248 case DBTYPE_R8: hr = VarI4FromR8(*(double*)src, d); break;
249 case DBTYPE_CY: hr = VarI4FromCy(*(CY*)src, d); break;
250 case DBTYPE_DATE: hr = VarI4FromDate(*(DATE*)src, d); break;
251 case DBTYPE_BSTR: hr = VarI4FromStr(*(WCHAR**)src, LOCALE_USER_DEFAULT, 0, d); break;
252 case DBTYPE_BOOL: hr = VarI4FromBool(*(VARIANT_BOOL*)src, d); break;
253 case DBTYPE_DECIMAL: hr = VarI4FromDec((DECIMAL*)src, d); break;
254 case DBTYPE_I1: hr = VarI4FromI1(*(signed char*)src, d); break;
255 case DBTYPE_UI1: hr = VarI4FromUI1(*(BYTE*)src, d); break;
256 case DBTYPE_UI2: hr = VarI4FromUI2(*(WORD*)src, d); break;
257 case DBTYPE_UI4: hr = VarI4FromUI4(*(DWORD*)src, d); break;
258 case DBTYPE_I8: hr = VarI4FromI8(*(LONGLONG*)src, d); break;
259 case DBTYPE_UI8: hr = VarI4FromUI8(*(ULONGLONG*)src, d); break;
260 default: FIXME("Unimplemented conversion %04x -> I4\n", src_type); return E_NOTIMPL;
261 }
262 break;
263 }
264
265 case DBTYPE_R4:
266 {
267 FLOAT *d = dst;
268 switch(src_type)
269 {
270 case DBTYPE_EMPTY: *d = 0; hr = S_OK; break;
271 case DBTYPE_I2: hr = VarR4FromI2(*(signed short*)src, d); break;
272 case DBTYPE_I4: hr = VarR4FromI4(*(signed int*)src, d); break;
273 case DBTYPE_R4: *d = *(FLOAT*)src; hr = S_OK; break;
274 case DBTYPE_R8: hr = VarR4FromR8(*(double*)src, d); break;
275 case DBTYPE_CY: hr = VarR4FromCy(*(CY*)src, d); break;
276 case DBTYPE_DATE: hr = VarR4FromDate(*(DATE*)src, d); break;
277 case DBTYPE_BSTR: hr = VarR4FromStr(*(WCHAR**)src, LOCALE_USER_DEFAULT, 0, d); break;
278 case DBTYPE_BOOL: hr = VarR4FromBool(*(VARIANT_BOOL*)src, d); break;
279 case DBTYPE_DECIMAL: hr = VarR4FromDec((DECIMAL*)src, d); break;
280 case DBTYPE_I1: hr = VarR4FromI1(*(signed char*)src, d); break;
281 case DBTYPE_UI1: hr = VarR4FromUI1(*(BYTE*)src, d); break;
282 case DBTYPE_UI2: hr = VarR4FromUI2(*(WORD*)src, d); break;
283 case DBTYPE_UI4: hr = VarR4FromUI4(*(DWORD*)src, d); break;
284 case DBTYPE_I8: hr = VarR4FromI8(*(LONGLONG*)src, d); break;
285 case DBTYPE_UI8: hr = VarR4FromUI8(*(ULONGLONG*)src, d); break;
286 default: FIXME("Unimplemented conversion %04x -> R4\n", src_type); return E_NOTIMPL;
287 }
288 break;
289 }
290
291 case DBTYPE_CY:
292 {
293 CY *d = dst;
294 switch(src_type)
295 {
296 case DBTYPE_EMPTY: d->int64 = 0; hr = S_OK; break;
297 case DBTYPE_I2: hr = VarCyFromI2(*(signed short*)src, d); break;
298 case DBTYPE_I4: hr = VarCyFromI4(*(signed int*)src, d); break;
299 case DBTYPE_R4: hr = VarCyFromR4(*(FLOAT*)src, d); break;
300 case DBTYPE_R8: hr = VarCyFromR8(*(double*)src, d); break;
301 case DBTYPE_CY: *d = *(CY*)src; hr = S_OK; break;
302 case DBTYPE_DATE: hr = VarCyFromDate(*(DATE*)src, d); break;
303 case DBTYPE_BSTR: hr = VarCyFromStr(*(WCHAR**)src, LOCALE_USER_DEFAULT, 0, d); break;
304 case DBTYPE_BOOL: hr = VarCyFromBool(*(VARIANT_BOOL*)src, d); break;
305 case DBTYPE_DECIMAL: hr = VarCyFromDec((DECIMAL*)src, d); break;
306 case DBTYPE_I1: hr = VarCyFromI1(*(signed char*)src, d); break;
307 case DBTYPE_UI1: hr = VarCyFromUI1(*(BYTE*)src, d); break;
308 case DBTYPE_UI2: hr = VarCyFromUI2(*(WORD*)src, d); break;
309 case DBTYPE_UI4: hr = VarCyFromUI4(*(DWORD*)src, d); break;
310 case DBTYPE_I8: hr = VarCyFromI8(*(LONGLONG*)src, d); break;
311 case DBTYPE_UI8: hr = VarCyFromUI8(*(ULONGLONG*)src, d); break;
312 default: FIXME("Unimplemented conversion %04x -> CY\n", src_type); return E_NOTIMPL;
313 }
314 break;
315 }
316
317 case DBTYPE_BSTR:
318 {
319 BSTR *d = dst;
320 switch(src_type)
321 {
322 case DBTYPE_EMPTY: *d = SysAllocStringLen(NULL, 0); hr = *d ? S_OK : E_OUTOFMEMORY; break;
323 case DBTYPE_I2: hr = VarBstrFromI2(*(signed short*)src, LOCALE_USER_DEFAULT, 0, d); break;
324 case DBTYPE_I4: hr = VarBstrFromI4(*(signed int*)src, LOCALE_USER_DEFAULT, 0, d); break;
325 case DBTYPE_R4: hr = VarBstrFromR4(*(FLOAT*)src, LOCALE_USER_DEFAULT, 0, d); break;
326 case DBTYPE_R8: hr = VarBstrFromR8(*(double*)src, LOCALE_USER_DEFAULT, 0, d); break;
327 case DBTYPE_CY: hr = VarBstrFromCy(*(CY*)src, LOCALE_USER_DEFAULT, 0, d); break;
328 case DBTYPE_DATE: hr = VarBstrFromDate(*(DATE*)src, LOCALE_USER_DEFAULT, 0, d); break;
329 case DBTYPE_BSTR: *d = SysAllocStringLen(*(BSTR*)src, SysStringLen(*(BSTR*)src)); hr = *d ? S_OK : E_OUTOFMEMORY; break;
330 case DBTYPE_BOOL: hr = VarBstrFromBool(*(VARIANT_BOOL*)src, LOCALE_USER_DEFAULT, 0, d); break;
331 case DBTYPE_DECIMAL: hr = VarBstrFromDec((DECIMAL*)src, LOCALE_USER_DEFAULT, 0, d); break;
332 case DBTYPE_I1: hr = VarBstrFromI1(*(signed char*)src, LOCALE_USER_DEFAULT, 0, d); break;
333 case DBTYPE_UI1: hr = VarBstrFromUI1(*(BYTE*)src, LOCALE_USER_DEFAULT, 0, d); break;
334 case DBTYPE_UI2: hr = VarBstrFromUI2(*(WORD*)src, LOCALE_USER_DEFAULT, 0, d); break;
335 case DBTYPE_UI4: hr = VarBstrFromUI4(*(DWORD*)src, LOCALE_USER_DEFAULT, 0, d); break;
336 case DBTYPE_I8: hr = VarBstrFromI8(*(LONGLONG*)src, LOCALE_USER_DEFAULT, 0, d); break;
337 case DBTYPE_UI8: hr = VarBstrFromUI8(*(ULONGLONG*)src, LOCALE_USER_DEFAULT, 0, d); break;
338 case DBTYPE_GUID:
339 {
340 WCHAR szBuff[39];
341 const GUID *id = (const GUID *)src;
342 WCHAR format[] = {
343 '{','%','0','8','X','-','%','0','4','X','-','%','0','4','X','-',
344 '%','0','2','X','%','0','2','X','-',
345 '%','0','2','X','%','0','2','X','%','0','2','X','%','0','2','X','%','0','2','X','%','0','2','X','}',0};
346 wsprintfW(szBuff, format,
347 id->Data1, id->Data2, id->Data3,
348 id->Data4[0], id->Data4[1], id->Data4[2], id->Data4[3],
349 id->Data4[4], id->Data4[5], id->Data4[6], id->Data4[7] );
350 *d = SysAllocString(szBuff);
351 hr = *d ? S_OK : E_OUTOFMEMORY;
352 }
353 break;
354 case DBTYPE_BYTES:
355 {
356 *d = SysAllocStringLen(NULL, 2 * src_len);
357 if (*d == NULL)
358 hr = E_OUTOFMEMORY;
359 else
360 {
361 const char hexchars[] = "0123456789ABCDEF";
362 WCHAR *s = *d;
363 unsigned char *p = src;
364 while (src_len > 0)
365 {
366 *s++ = hexchars[(*p >> 4) & 0x0F];
367 *s++ = hexchars[(*p) & 0x0F];
368 src_len--; p++;
369 }
370 hr = S_OK;
371 }
372 }
373 break;
374 default: FIXME("Unimplemented conversion %04x -> BSTR\n", src_type); return E_NOTIMPL;
375 }
376 break;
377 }
378
379 case DBTYPE_UI1:
380 {
381 BYTE *d = dst;
382 switch(src_type)
383 {
384 case DBTYPE_EMPTY: *d = 0; hr = S_OK; break;
385 case DBTYPE_I2: hr = VarUI1FromI2(*(signed short*)src, d); break;
386 case DBTYPE_I4: hr = VarUI1FromI4(*(signed int*)src, d); break;
387 case DBTYPE_R4: hr = VarUI1FromR4(*(FLOAT*)src, d); break;
388 case DBTYPE_R8: hr = VarUI1FromR8(*(double*)src, d); break;
389 case DBTYPE_CY: hr = VarUI1FromCy(*(CY*)src, d); break;
390 case DBTYPE_DATE: hr = VarUI1FromDate(*(DATE*)src, d); break;
391 case DBTYPE_BSTR: hr = VarUI1FromStr(*(WCHAR**)src, LOCALE_USER_DEFAULT, 0, d); break;
392 case DBTYPE_BOOL: hr = VarUI1FromBool(*(VARIANT_BOOL*)src, d); break;
393 case DBTYPE_DECIMAL: hr = VarUI1FromDec((DECIMAL*)src, d); break;
394 case DBTYPE_I1: hr = VarUI1FromI1(*(signed char*)src, d); break;
395 case DBTYPE_UI1: *d = *(BYTE*)src; hr = S_OK; break;
396 case DBTYPE_UI2: hr = VarUI1FromUI2(*(WORD*)src, d); break;
397 case DBTYPE_UI4: hr = VarUI1FromUI4(*(DWORD*)src, d); break;
398 case DBTYPE_I8: hr = VarUI1FromI8(*(LONGLONG*)src, d); break;
399 case DBTYPE_UI8: hr = VarUI1FromUI8(*(ULONGLONG*)src, d); break;
400 default: FIXME("Unimplemented conversion %04x -> UI1\n", src_type); return E_NOTIMPL;
401 }
402 break;
403 }
404
405 case DBTYPE_UI4:
406 {
407 DWORD *d = dst;
408 switch(src_type)
409 {
410 case DBTYPE_EMPTY: *d = 0; hr = S_OK; break;
411 case DBTYPE_I2: hr = VarUI4FromI2(*(signed short*)src, d); break;
412 case DBTYPE_I4: hr = VarUI4FromI4(*(signed int*)src, d); break;
413 case DBTYPE_R4: hr = VarUI4FromR4(*(FLOAT*)src, d); break;
414 case DBTYPE_R8: hr = VarUI4FromR8(*(double*)src, d); break;
415 case DBTYPE_CY: hr = VarUI4FromCy(*(CY*)src, d); break;
416 case DBTYPE_DATE: hr = VarUI4FromDate(*(DATE*)src, d); break;
417 case DBTYPE_BSTR: hr = VarUI4FromStr(*(WCHAR**)src, LOCALE_USER_DEFAULT, 0, d); break;
418 case DBTYPE_BOOL: hr = VarUI4FromBool(*(VARIANT_BOOL*)src, d); break;
419 case DBTYPE_DECIMAL: hr = VarUI4FromDec((DECIMAL*)src, d); break;
420 case DBTYPE_I1: hr = VarUI4FromI1(*(signed char*)src, d); break;
421 case DBTYPE_UI1: hr = VarUI4FromUI1(*(BYTE*)src, d); break;
422 case DBTYPE_UI2: hr = VarUI4FromUI2(*(WORD*)src, d); break;
423 case DBTYPE_UI4: *d = *(DWORD*)src; hr = S_OK; break;
424 case DBTYPE_I8: hr = VarUI4FromI8(*(LONGLONG*)src, d); break;
425 case DBTYPE_UI8: hr = VarUI4FromUI8(*(ULONGLONG*)src, d); break;
426 default: FIXME("Unimplemented conversion %04x -> UI4\n", src_type); return E_NOTIMPL;
427 }
428 break;
429 }
430
431 case DBTYPE_UI8:
432 {
433 ULONGLONG *d = dst;
434 switch(src_type)
435 {
436 case DBTYPE_EMPTY: *d = 0; hr = S_OK; break;
437 case DBTYPE_I2: hr = VarUI8FromI2(*(signed short*)src, d); break;
438 case DBTYPE_I4: {LONGLONG s = *(signed int*)src; hr = VarUI8FromI8(s, d); break;}
439 case DBTYPE_R4: hr = VarUI8FromR4(*(FLOAT*)src, d); break;
440 case DBTYPE_R8: hr = VarUI8FromR8(*(double*)src, d); break;
441 case DBTYPE_CY: hr = VarUI8FromCy(*(CY*)src, d); break;
442 case DBTYPE_DATE: hr = VarUI8FromDate(*(DATE*)src, d); break;
443 case DBTYPE_BSTR: hr = VarUI8FromStr(*(WCHAR**)src, LOCALE_USER_DEFAULT, 0, d); break;
444 case DBTYPE_BOOL: hr = VarUI8FromBool(*(VARIANT_BOOL*)src, d); break;
445 case DBTYPE_DECIMAL: hr = VarUI8FromDec((DECIMAL*)src, d); break;
446 case DBTYPE_I1: hr = VarUI8FromI1(*(signed char*)src, d); break;
447 case DBTYPE_UI1: hr = VarUI8FromUI1(*(BYTE*)src, d); break;
448 case DBTYPE_UI2: hr = VarUI8FromUI2(*(WORD*)src, d); break;
449 case DBTYPE_UI4: hr = VarUI8FromUI4(*(DWORD*)src, d); break;
450 case DBTYPE_I8: hr = VarUI8FromI8(*(LONGLONG*)src, d); break;
451 case DBTYPE_UI8: *d = *(ULONGLONG*)src; hr = S_OK; break;
452 default: FIXME("Unimplemented conversion %04x -> UI8\n", src_type); return E_NOTIMPL;
453 }
454 break;
455 }
456
457 case DBTYPE_FILETIME:
458 {
459 FILETIME *d = dst;
460 switch(src_type)
461 {
462 case DBTYPE_EMPTY: d->dwLowDateTime = d->dwHighDateTime = 0; hr = S_OK; break;
463 case DBTYPE_FILETIME: *d = *(FILETIME*)src; hr = S_OK; break;
464 default: FIXME("Unimplemented conversion %04x -> FILETIME\n", src_type); return E_NOTIMPL;
465 }
466 break;
467 }
468
469 case DBTYPE_GUID:
470 {
471 GUID *d = dst;
472 switch(src_type)
473 {
474 case DBTYPE_EMPTY: *d = GUID_NULL; hr = S_OK; break;
475 case DBTYPE_GUID: *d = *(GUID*)src; hr = S_OK; break;
476 default: FIXME("Unimplemented conversion %04x -> GUID\n", src_type); return E_NOTIMPL;
477 }
478 break;
479 }
480
481 case DBTYPE_WSTR:
482 {
483 BSTR b;
484 DBLENGTH bstr_len;
485 INT bytes_to_copy;
486 hr = IDataConvert_DataConvert(iface, src_type, DBTYPE_BSTR, src_len, &bstr_len,
487 src, &b, sizeof(BSTR), src_status, dst_status,
488 precision, scale, flags);
489 if(hr != S_OK) return hr;
490 bstr_len = SysStringLen(b);
491 *dst_len = bstr_len * sizeof(WCHAR); /* Doesn't include size for '\0' */
492 *dst_status = DBSTATUS_S_OK;
493 bytes_to_copy = min(*dst_len + sizeof(WCHAR), dst_max_len);
494 if(dst)
495 {
496 if(bytes_to_copy >= sizeof(WCHAR))
497 {
498 memcpy(dst, b, bytes_to_copy - sizeof(WCHAR));
499 *((WCHAR*)dst + bytes_to_copy / sizeof(WCHAR) - 1) = 0;
500 if(bytes_to_copy < *dst_len + sizeof(WCHAR))
501 *dst_status = DBSTATUS_S_TRUNCATED;
502 }
503 else
504 {
505 *dst_status = DBSTATUS_E_DATAOVERFLOW;
506 hr = DB_E_ERRORSOCCURRED;
507 }
508 }
509 SysFreeString(b);
510 return hr;
511 }
512 case DBTYPE_STR:
513 {
514 BSTR b;
515 DBLENGTH bstr_len;
516 INT bytes_to_copy;
517 hr = IDataConvert_DataConvert(iface, src_type, DBTYPE_BSTR, src_len, &bstr_len,
518 src, &b, sizeof(BSTR), src_status, dst_status,
519 precision, scale, flags);
520 if(hr != S_OK) return hr;
521 bstr_len = SysStringLen(b);
522 *dst_len = bstr_len * sizeof(char); /* Doesn't include size for '\0' */
523 *dst_status = DBSTATUS_S_OK;
524 bytes_to_copy = min(*dst_len + sizeof(char), dst_max_len);
525 if(dst)
526 {
527 if(bytes_to_copy >= sizeof(char))
528 {
529 WideCharToMultiByte(CP_ACP, 0, b, bytes_to_copy - sizeof(char), dst, dst_max_len, NULL, NULL);
530 *((char *)dst + bytes_to_copy / sizeof(char) - 1) = 0;
531 if(bytes_to_copy < *dst_len + sizeof(char))
532 *dst_status = DBSTATUS_S_TRUNCATED;
533 }
534 else
535 {
536 *dst_status = DBSTATUS_E_DATAOVERFLOW;
537 hr = DB_E_ERRORSOCCURRED;
538 }
539 }
540 SysFreeString(b);
541 return hr;
542 }
543
544 case DBTYPE_BYREF | DBTYPE_WSTR:
545 {
546 BSTR b;
547 WCHAR **d = dst;
548 DBLENGTH bstr_len;
549 hr = IDataConvert_DataConvert(iface, src_type, DBTYPE_BSTR, src_len, &bstr_len,
550 src, &b, sizeof(BSTR), src_status, dst_status,
551 precision, scale, flags);
552 if(hr != S_OK) return hr;
553
554 bstr_len = SysStringLen(b) * sizeof(WCHAR);
555 *dst_len = bstr_len; /* Doesn't include size for '\0' */
556
557 *d = CoTaskMemAlloc(bstr_len + sizeof(WCHAR));
558 if(*d) memcpy(*d, b, bstr_len + sizeof(WCHAR));
559 else hr = E_OUTOFMEMORY;
560 SysFreeString(b);
561 return hr;
562 }
563
564 default:
565 FIXME("Unimplemented conversion %04x -> %04x\n", src_type, dst_type);
566 return E_NOTIMPL;
567
568 }
569
570 if(hr == DISP_E_OVERFLOW)
571 {
572 *dst_status = DBSTATUS_E_DATAOVERFLOW;
573 *dst_len = get_length(dst_type);
574 hr = DB_E_ERRORSOCCURRED;
575 }
576 else if(hr == S_OK)
577 {
578 *dst_status = DBSTATUS_S_OK;
579 *dst_len = get_length(dst_type);
580 }
581
582 return hr;
583 }
584
585 static inline WORD get_dbtype_class(DBTYPE type)
586 {
587 switch(type)
588 {
589 case DBTYPE_I2:
590 case DBTYPE_R4:
591 case DBTYPE_R8:
592 case DBTYPE_I1:
593 case DBTYPE_UI1:
594 case DBTYPE_UI2:
595 return DBTYPE_I2;
596
597 case DBTYPE_I4:
598 case DBTYPE_UI4:
599 return DBTYPE_I4;
600
601 case DBTYPE_I8:
602 case DBTYPE_UI8:
603 return DBTYPE_I8;
604
605 case DBTYPE_BSTR:
606 case DBTYPE_STR:
607 case DBTYPE_WSTR:
608 return DBTYPE_BSTR;
609
610 case DBTYPE_DBDATE:
611 case DBTYPE_DBTIME:
612 case DBTYPE_DBTIMESTAMP:
613 return DBTYPE_DBDATE;
614 }
615 return type;
616 }
617
618 /* Many src types will convert to this group of dst types */
619 static inline BOOL common_class(WORD dst_class)
620 {
621 switch(dst_class)
622 {
623 case DBTYPE_EMPTY:
624 case DBTYPE_NULL:
625 case DBTYPE_I2:
626 case DBTYPE_I4:
627 case DBTYPE_BSTR:
628 case DBTYPE_BOOL:
629 case DBTYPE_VARIANT:
630 case DBTYPE_I8:
631 case DBTYPE_CY:
632 case DBTYPE_DECIMAL:
633 case DBTYPE_NUMERIC:
634 return TRUE;
635 }
636 return FALSE;
637 }
638
639 static inline BOOL array_type(DBTYPE type)
640 {
641 return (type >= DBTYPE_I2 && type <= DBTYPE_UI4);
642 }
643
644 static HRESULT WINAPI convert_CanConvert(IDataConvert* iface,
645 DBTYPE src_type, DBTYPE dst_type)
646 {
647 convert *This = impl_from_IDataConvert(iface);
648 DBTYPE src_base_type = src_type & 0x1ff;
649 DBTYPE dst_base_type = dst_type & 0x1ff;
650 WORD dst_class = get_dbtype_class(dst_base_type);
651
652 TRACE("(%p)->(%d, %d)\n", This, src_type, dst_type);
653
654 if(src_type & DBTYPE_VECTOR || dst_type & DBTYPE_VECTOR) return S_FALSE;
655
656 if(src_type & DBTYPE_ARRAY)
657 {
658 if(!array_type(src_base_type)) return S_FALSE;
659 if(dst_type & DBTYPE_ARRAY)
660 {
661 if(src_type == dst_type) return S_OK;
662 return S_FALSE;
663 }
664 if(dst_type == DBTYPE_VARIANT) return S_OK;
665 return S_FALSE;
666 }
667
668 if(dst_type & DBTYPE_ARRAY)
669 {
670 if(!array_type(dst_base_type)) return S_FALSE;
671 if(src_type == DBTYPE_IDISPATCH || src_type == DBTYPE_VARIANT) return S_OK;
672 return S_FALSE;
673 }
674
675 if(dst_type & DBTYPE_BYREF)
676 if(dst_base_type != DBTYPE_BYTES && dst_base_type != DBTYPE_STR && dst_base_type != DBTYPE_WSTR)
677 return S_FALSE;
678
679 switch(get_dbtype_class(src_base_type))
680 {
681 case DBTYPE_EMPTY:
682 if(common_class(dst_class)) return S_OK;
683 switch(dst_class)
684 {
685 case DBTYPE_DATE:
686 case DBTYPE_GUID:
687 case DBTYPE_FILETIME:
688 return S_OK;
689 default:
690 if(dst_base_type == DBTYPE_DBTIMESTAMP) return S_OK;
691 return S_FALSE;
692 }
693
694 case DBTYPE_NULL:
695 switch(dst_base_type)
696 {
697 case DBTYPE_NULL:
698 case DBTYPE_VARIANT:
699 case DBTYPE_FILETIME: return S_OK;
700 default: return S_FALSE;
701 }
702
703 case DBTYPE_I4:
704 if(dst_base_type == DBTYPE_BYTES) return S_OK;
705 /* fall through */
706 case DBTYPE_I2:
707 if(dst_base_type == DBTYPE_DATE) return S_OK;
708 /* fall through */
709 case DBTYPE_DECIMAL:
710 if(common_class(dst_class)) return S_OK;
711 if(dst_class == DBTYPE_DBDATE) return S_OK;
712 return S_FALSE;
713
714 case DBTYPE_BOOL:
715 if(dst_base_type == DBTYPE_DATE) return S_OK;
716 case DBTYPE_NUMERIC:
717 case DBTYPE_CY:
718 if(common_class(dst_class)) return S_OK;
719 return S_FALSE;
720
721 case DBTYPE_I8:
722 if(common_class(dst_class)) return S_OK;
723 switch(dst_base_type)
724 {
725 case DBTYPE_BYTES:
726 case DBTYPE_FILETIME: return S_OK;
727 default: return S_FALSE;
728 }
729
730 case DBTYPE_DATE:
731 switch(dst_class)
732 {
733 case DBTYPE_EMPTY:
734 case DBTYPE_NULL:
735 case DBTYPE_I2:
736 case DBTYPE_I4:
737 case DBTYPE_BSTR:
738 case DBTYPE_BOOL:
739 case DBTYPE_VARIANT:
740 case DBTYPE_I8:
741 case DBTYPE_DATE:
742 case DBTYPE_DBDATE:
743 case DBTYPE_FILETIME:
744 return S_OK;
745 default: return S_FALSE;
746 }
747
748 case DBTYPE_IDISPATCH:
749 case DBTYPE_VARIANT:
750 switch(dst_base_type)
751 {
752 case DBTYPE_IDISPATCH:
753 case DBTYPE_ERROR:
754 case DBTYPE_IUNKNOWN:
755 return S_OK;
756 }
757 /* fall through */
758 case DBTYPE_BSTR:
759 if(common_class(dst_class)) return S_OK;
760 switch(dst_class)
761 {
762 case DBTYPE_DATE:
763 case DBTYPE_GUID:
764 case DBTYPE_BYTES:
765 case DBTYPE_DBDATE:
766 case DBTYPE_FILETIME:
767 return S_OK;
768 default: return S_FALSE;
769 }
770
771 case DBTYPE_ERROR:
772 switch(dst_base_type)
773 {
774 case DBTYPE_BSTR:
775 case DBTYPE_ERROR:
776 case DBTYPE_VARIANT:
777 case DBTYPE_WSTR:
778 return S_OK;
779 default: return S_FALSE;
780 }
781
782 case DBTYPE_IUNKNOWN:
783 switch(dst_base_type)
784 {
785 case DBTYPE_EMPTY:
786 case DBTYPE_NULL:
787 case DBTYPE_IDISPATCH:
788 case DBTYPE_VARIANT:
789 case DBTYPE_IUNKNOWN:
790 return S_OK;
791 default: return S_FALSE;
792 }
793
794 case DBTYPE_BYTES:
795 if(dst_class == DBTYPE_I4 || dst_class == DBTYPE_I8) return S_OK;
796 /* fall through */
797 case DBTYPE_GUID:
798 switch(dst_class)
799 {
800 case DBTYPE_EMPTY:
801 case DBTYPE_NULL:
802 case DBTYPE_BSTR:
803 case DBTYPE_VARIANT:
804 case DBTYPE_GUID:
805 case DBTYPE_BYTES:
806 return S_OK;
807 default: return S_FALSE;
808 }
809
810 case DBTYPE_FILETIME:
811 if(dst_class == DBTYPE_I8) return S_OK;
812 /* fall through */
813 case DBTYPE_DBDATE:
814 switch(dst_class)
815 {
816 case DBTYPE_EMPTY:
817 case DBTYPE_NULL:
818 case DBTYPE_DATE:
819 case DBTYPE_BSTR:
820 case DBTYPE_VARIANT:
821 case DBTYPE_DBDATE:
822 case DBTYPE_FILETIME:
823 return S_OK;
824 default: return S_FALSE;
825 }
826
827 }
828 return S_FALSE;
829 }
830
831 static HRESULT WINAPI convert_GetConversionSize(IDataConvert* iface,
832 DBTYPE wSrcType, DBTYPE wDstType,
833 DBLENGTH *pcbSrcLength, DBLENGTH *pcbDstLength,
834 void *pSrc)
835 {
836 convert *This = impl_from_IDataConvert(iface);
837 FIXME("(%p)->(%d, %d, %p, %p, %p): stub\n", This, wSrcType, wDstType, pcbSrcLength, pcbDstLength, pSrc);
838
839 return E_NOTIMPL;
840 }
841
842 static const struct IDataConvertVtbl convert_vtbl =
843 {
844 convert_QueryInterface,
845 convert_AddRef,
846 convert_Release,
847 convert_DataConvert,
848 convert_CanConvert,
849 convert_GetConversionSize
850 };
851
852 static HRESULT WINAPI dcinfo_QueryInterface(IDCInfo* iface, REFIID riid, void **obj)
853 {
854 convert *This = impl_from_IDCInfo(iface);
855
856 return IDataConvert_QueryInterface((IDataConvert *)This, riid, obj);
857 }
858
859 static ULONG WINAPI dcinfo_AddRef(IDCInfo* iface)
860 {
861 convert *This = impl_from_IDCInfo(iface);
862
863 return IDataConvert_AddRef((IDataConvert *)This);
864 }
865
866 static ULONG WINAPI dcinfo_Release(IDCInfo* iface)
867 {
868 convert *This = impl_from_IDCInfo(iface);
869
870 return IDataConvert_Release((IDataConvert *)This);
871 }
872
873 static HRESULT WINAPI dcinfo_GetInfo(IDCInfo *iface, ULONG num, DCINFOTYPE types[], DCINFO **info_ptr)
874 {
875 convert *This = impl_from_IDCInfo(iface);
876 ULONG i;
877 DCINFO *infos;
878
879 TRACE("(%p)->(%d, %p, %p)\n", This, num, types, info_ptr);
880
881 *info_ptr = infos = CoTaskMemAlloc(num * sizeof(*infos));
882 if(!infos) return E_OUTOFMEMORY;
883
884 for(i = 0; i < num; i++)
885 {
886 infos[i].eInfoType = types[i];
887 VariantInit(&infos[i].vData);
888
889 switch(types[i])
890 {
891 case DCINFOTYPE_VERSION:
892 V_VT(&infos[i].vData) = VT_UI4;
893 V_UI4(&infos[i].vData) = This->version;
894 break;
895 }
896 }
897
898 return S_OK;
899 }
900
901 static HRESULT WINAPI dcinfo_SetInfo(IDCInfo* iface, ULONG num, DCINFO info[])
902 {
903 convert *This = impl_from_IDCInfo(iface);
904 ULONG i;
905 HRESULT hr = S_OK;
906
907 TRACE("(%p)->(%d, %p)\n", This, num, info);
908
909 for(i = 0; i < num; i++)
910 {
911 switch(info[i].eInfoType)
912 {
913 case DCINFOTYPE_VERSION:
914 if(V_VT(&info[i].vData) != VT_UI4)
915 {
916 FIXME("VERSION with vt %x\n", V_VT(&info[i].vData));
917 hr = DB_S_ERRORSOCCURRED;
918 break;
919 }
920 This->version = V_UI4(&info[i].vData);
921 break;
922
923 default:
924 FIXME("Unhandled info type %d (vt %x)\n", info[i].eInfoType, V_VT(&info[i].vData));
925 }
926 }
927 return hr;
928 }
929
930 static const struct IDCInfoVtbl dcinfo_vtbl =
931 {
932 dcinfo_QueryInterface,
933 dcinfo_AddRef,
934 dcinfo_Release,
935 dcinfo_GetInfo,
936 dcinfo_SetInfo
937 };
938
939 HRESULT create_oledb_convert(IUnknown *outer, void **obj)
940 {
941 convert *This;
942
943 TRACE("(%p, %p)\n", outer, obj);
944
945 *obj = NULL;
946
947 if(outer) return CLASS_E_NOAGGREGATION;
948
949 This = HeapAlloc(GetProcessHeap(), 0, sizeof(*This));
950 if(!This) return E_OUTOFMEMORY;
951
952 This->lpVtbl = &convert_vtbl;
953 This->lpDCInfoVtbl = &dcinfo_vtbl;
954 This->ref = 1;
955 This->version = 0x110;
956
957 *obj = &This->lpVtbl;
958
959 return S_OK;
960 }
Windows API implementation