?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
mshtml: Wine Gecko 2.34-beta2 release.
[wine/wine-gecko.git] / dlls / mshtml / dispex.c
blob4a872a8bebb43b04fb89e2a2707740b80b5d02c9
1 /*
2 * Copyright 2008-2009 Jacek Caban for CodeWeavers
3 *
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.
8 *
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.
13 *
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
17 */
18
19 #include <stdarg.h>
20 #include <assert.h>
21
22 #define COBJMACROS
23 #define NONAMELESSUNION
24
25 #include "windef.h"
26 #include "winbase.h"
27 #include "winuser.h"
28 #include "ole2.h"
29
30 #include "wine/debug.h"
31
32 #include "mshtml_private.h"
33
34 WINE_DEFAULT_DEBUG_CHANNEL(mshtml);
35
36 #define MAX_ARGS 16
37
38 static CRITICAL_SECTION cs_dispex_static_data;
39 static CRITICAL_SECTION_DEBUG cs_dispex_static_data_dbg =
40 {
41 0, 0, &cs_dispex_static_data,
42 { &cs_dispex_static_data_dbg.ProcessLocksList, &cs_dispex_static_data_dbg.ProcessLocksList },
43 0, 0, { (DWORD_PTR)(__FILE__ ": dispex_static_data") }
44 };
45 static CRITICAL_SECTION cs_dispex_static_data = { &cs_dispex_static_data_dbg, -1, 0, 0, 0, 0 };
46
47
48 static const WCHAR objectW[] = {'[','o','b','j','e','c','t',']',0};
49
50 typedef struct {
51 DISPID id;
52 BSTR name;
53 tid_t tid;
54 SHORT call_vtbl_off;
55 SHORT put_vtbl_off;
56 SHORT get_vtbl_off;
57 SHORT func_disp_idx;
58 USHORT argc;
59 VARTYPE prop_vt;
60 VARTYPE *arg_types;
61 } func_info_t;
62
63 struct dispex_data_t {
64 DWORD func_cnt;
65 func_info_t *funcs;
66 func_info_t **name_table;
67 DWORD func_disp_cnt;
68
69 struct list entry;
70 };
71
72 typedef struct {
73 VARIANT var;
74 LPWSTR name;
75 DWORD flags;
76 } dynamic_prop_t;
77
78 #define DYNPROP_DELETED 0x01
79
80 typedef struct {
81 DispatchEx dispex;
82 IUnknown IUnknown_iface;
83 LONG ref;
84 DispatchEx *obj;
85 func_info_t *info;
86 } func_disp_t;
87
88 typedef struct {
89 func_disp_t *func_obj;
90 IDispatch *val;
91 } func_obj_entry_t;
92
93 struct dispex_dynamic_data_t {
94 DWORD buf_size;
95 DWORD prop_cnt;
96 dynamic_prop_t *props;
97 func_obj_entry_t *func_disps;
98 };
99
100 #define DISPID_DYNPROP_0 0x50000000
101 #define DISPID_DYNPROP_MAX 0x5fffffff
102
103 #define FDEX_VERSION_MASK 0xf0000000
104
105 static ITypeLib *typelib;
106 static ITypeInfo *typeinfos[LAST_tid];
107 static struct list dispex_data_list = LIST_INIT(dispex_data_list);
108
109 static REFIID tid_ids[] = {
110 #define XIID(iface) &IID_ ## iface,
111 #define XDIID(iface) &DIID_ ## iface,
112 TID_LIST
113 #undef XIID
114 #undef XDIID
115 };
116
117 static HRESULT load_typelib(void)
118 {
119 HRESULT hres;
120 ITypeLib *tl;
121
122 hres = LoadRegTypeLib(&LIBID_MSHTML, 4, 0, LOCALE_SYSTEM_DEFAULT, &tl);
123 if(FAILED(hres)) {
124 ERR("LoadRegTypeLib failed: %08x\n", hres);
125 return hres;
126 }
127
128 if(InterlockedCompareExchangePointer((void**)&typelib, tl, NULL))
129 ITypeLib_Release(tl);
130 return hres;
131 }
132
133 static HRESULT get_typeinfo(tid_t tid, ITypeInfo **typeinfo)
134 {
135 HRESULT hres;
136
137 if (!typelib)
138 hres = load_typelib();
139 if (!typelib)
140 return hres;
141
142 if(!typeinfos[tid]) {
143 ITypeInfo *ti;
144
145 hres = ITypeLib_GetTypeInfoOfGuid(typelib, tid_ids[tid], &ti);
146 if(FAILED(hres)) {
147 ERR("GetTypeInfoOfGuid(%s) failed: %08x\n", debugstr_mshtml_guid(tid_ids[tid]), hres);
148 return hres;
149 }
150
151 if(InterlockedCompareExchangePointer((void**)(typeinfos+tid), ti, NULL))
152 ITypeInfo_Release(ti);
153 }
154
155 *typeinfo = typeinfos[tid];
156 return S_OK;
157 }
158
159 void release_typelib(void)
160 {
161 dispex_data_t *iter;
162 unsigned i;
163
164 while(!list_empty(&dispex_data_list)) {
165 iter = LIST_ENTRY(list_head(&dispex_data_list), dispex_data_t, entry);
166 list_remove(&iter->entry);
167
168 for(i=0; i < iter->func_cnt; i++)
169 SysFreeString(iter->funcs[i].name);
170
171 heap_free(iter->funcs);
172 heap_free(iter->name_table);
173 heap_free(iter);
174 }
175
176 if(!typelib)
177 return;
178
179 for(i=0; i < sizeof(typeinfos)/sizeof(*typeinfos); i++)
180 if(typeinfos[i])
181 ITypeInfo_Release(typeinfos[i]);
182
183 ITypeLib_Release(typelib);
184 DeleteCriticalSection(&cs_dispex_static_data);
185 }
186
187 HRESULT get_htmldoc_classinfo(ITypeInfo **typeinfo)
188 {
189 HRESULT hres;
190
191 if (!typelib)
192 hres = load_typelib();
193 if (!typelib)
194 return hres;
195
196 hres = ITypeLib_GetTypeInfoOfGuid(typelib, &CLSID_HTMLDocument, typeinfo);
197 if(FAILED(hres))
198 ERR("GetTypeInfoOfGuid failed: %08x\n", hres);
199 return hres;
200 }
201
202 /* Not all argument types are supported yet */
203 #define BUILTIN_ARG_TYPES_SWITCH \
204 CASE_VT(VT_I2, INT16, V_I2); \
205 CASE_VT(VT_I4, INT32, V_I4); \
206 CASE_VT(VT_R4, float, V_R4); \
207 CASE_VT(VT_BSTR, BSTR, V_BSTR); \
208 CASE_VT(VT_BOOL, VARIANT_BOOL, V_BOOL)
209
210 /* List all types used by IDispatchEx-based properties */
211 #define BUILTIN_TYPES_SWITCH \
212 BUILTIN_ARG_TYPES_SWITCH; \
213 CASE_VT(VT_VARIANT, VARIANT, *); \
214 CASE_VT(VT_PTR, void*, V_BYREF); \
215 CASE_VT(VT_UNKNOWN, IUnknown*, V_UNKNOWN); \
216 CASE_VT(VT_DISPATCH, IDispatch*, V_DISPATCH)
217
218 static BOOL is_arg_type_supported(VARTYPE vt)
219 {
220 switch(vt) {
221 #define CASE_VT(x,a,b) case x: return TRUE
222 BUILTIN_ARG_TYPES_SWITCH;
223 #undef CASE_VT
224 }
225 return FALSE;
226 }
227
228 static void add_func_info(dispex_data_t *data, DWORD *size, tid_t tid, const FUNCDESC *desc, ITypeInfo *dti)
229 {
230 func_info_t *info;
231 HRESULT hres;
232
233 if(data->func_cnt && data->funcs[data->func_cnt-1].id == desc->memid) {
234 info = data->funcs+data->func_cnt-1;
235 }else {
236 if(data->func_cnt == *size)
237 data->funcs = heap_realloc_zero(data->funcs, (*size <<= 1)*sizeof(func_info_t));
238
239 info = data->funcs+data->func_cnt;
240 hres = ITypeInfo_GetDocumentation(dti, desc->memid, &info->name, NULL, NULL, NULL);
241 if(FAILED(hres))
242 return;
243
244 data->func_cnt++;
245
246 info->id = desc->memid;
247 info->tid = tid;
248 info->func_disp_idx = -1;
249 info->prop_vt = VT_EMPTY;
250 }
251
252 if(desc->invkind & DISPATCH_METHOD) {
253 unsigned i;
254
255 info->func_disp_idx = data->func_disp_cnt++;
256 info->argc = desc->cParams;
257
258 assert(info->argc < MAX_ARGS);
259 assert(desc->funckind == FUNC_DISPATCH);
260
261 info->arg_types = heap_alloc(sizeof(*info->arg_types) * info->argc);
262 if(!info->arg_types)
263 return; /* FIXME: real error instead of fallback */
264
265 for(i=0; i < info->argc; i++)
266 info->arg_types[i] = desc->lprgelemdescParam[i].tdesc.vt;
267
268 info->prop_vt = desc->elemdescFunc.tdesc.vt;
269 if(info->prop_vt != VT_VOID && !is_arg_type_supported(info->prop_vt)) {
270 TRACE("%s: return type %d\n", debugstr_w(info->name), info->prop_vt);
271 return; /* Fallback to ITypeInfo::Invoke */
272 }
273
274 if(desc->cParamsOpt) {
275 TRACE("%s: optional params\n", debugstr_w(info->name));
276 return; /* Fallback to ITypeInfo::Invoke */
277 }
278
279 for(i=0; i < info->argc; i++) {
280 if(!is_arg_type_supported(info->arg_types[i])) {
281 return; /* Fallback to ITypeInfo for unsupported arg types */
282 }
283
284 if(desc->lprgelemdescParam[i].u.paramdesc.wParamFlags & PARAMFLAG_FHASDEFAULT) {
285 TRACE("%s param %d: default value\n", debugstr_w(info->name), i);
286 return; /* Fallback to ITypeInfo::Invoke */
287 }
288 }
289
290 assert(info->argc <= MAX_ARGS);
291 assert(desc->callconv == CC_STDCALL);
292
293 info->call_vtbl_off = desc->oVft/sizeof(void*);
294 }else if(desc->invkind & (DISPATCH_PROPERTYPUT|DISPATCH_PROPERTYGET)) {
295 VARTYPE vt = VT_EMPTY;
296
297 if(desc->invkind & DISPATCH_PROPERTYGET) {
298 vt = desc->elemdescFunc.tdesc.vt;
299 info->get_vtbl_off = desc->oVft/sizeof(void*);
300 }
301 if(desc->invkind & DISPATCH_PROPERTYPUT) {
302 assert(desc->cParams == 1);
303 vt = desc->lprgelemdescParam->tdesc.vt;
304 info->put_vtbl_off = desc->oVft/sizeof(void*);
305 }
306
307 assert(info->prop_vt == VT_EMPTY || vt == info->prop_vt);
308 info->prop_vt = vt;
309 }
310 }
311
312 static int dispid_cmp(const void *p1, const void *p2)
313 {
314 return ((const func_info_t*)p1)->id - ((const func_info_t*)p2)->id;
315 }
316
317 static int func_name_cmp(const void *p1, const void *p2)
318 {
319 return strcmpiW((*(func_info_t* const*)p1)->name, (*(func_info_t* const*)p2)->name);
320 }
321
322 static dispex_data_t *preprocess_dispex_data(DispatchEx *This)
323 {
324 const tid_t *tid = This->data->iface_tids;
325 FUNCDESC *funcdesc;
326 dispex_data_t *data;
327 DWORD size = 16, i;
328 ITypeInfo *ti, *dti;
329 HRESULT hres;
330
331 TRACE("(%p)\n", This);
332
333 if(This->data->disp_tid) {
334 hres = get_typeinfo(This->data->disp_tid, &dti);
335 if(FAILED(hres)) {
336 ERR("Could not get disp type info: %08x\n", hres);
337 return NULL;
338 }
339 }
340
341 data = heap_alloc(sizeof(dispex_data_t));
342 data->func_cnt = 0;
343 data->func_disp_cnt = 0;
344 data->funcs = heap_alloc_zero(size*sizeof(func_info_t));
345 list_add_tail(&dispex_data_list, &data->entry);
346
347 while(*tid) {
348 hres = get_typeinfo(*tid, &ti);
349 if(FAILED(hres))
350 break;
351
352 i=7;
353 while(1) {
354 hres = ITypeInfo_GetFuncDesc(ti, i++, &funcdesc);
355 if(FAILED(hres))
356 break;
357
358 add_func_info(data, &size, *tid, funcdesc, dti);
359 ITypeInfo_ReleaseFuncDesc(ti, funcdesc);
360 }
361
362 tid++;
363 }
364
365 if(!data->func_cnt) {
366 heap_free(data->funcs);
367 data->name_table = NULL;
368 data->funcs = NULL;
369 return data;
370 }
371
372
373 data->funcs = heap_realloc(data->funcs, data->func_cnt * sizeof(func_info_t));
374 qsort(data->funcs, data->func_cnt, sizeof(func_info_t), dispid_cmp);
375
376 for(i = 1; i < data->func_cnt && data->funcs[i-1].id != data->funcs[i].id; i++);
377 if(i < data->func_cnt) {
378 unsigned j = i--;
379
380 /* We have at least one duplicated property. This may happen if more than one
381 * interface implements the same property. We have to remove these duplicated
382 * entries. */
383
384 while(j < data->func_cnt) {
385 while(j+1 < data->func_cnt && data->funcs[j+1].id == data->funcs[j].id)
386 j++;
387 data->funcs[i++] = data->funcs[j++];
388 }
389 data->func_cnt = i;
390 }
391
392 data->name_table = heap_alloc(data->func_cnt * sizeof(func_info_t*));
393 for(i=0; i < data->func_cnt; i++)
394 data->name_table[i] = data->funcs+i;
395 qsort(data->name_table, data->func_cnt, sizeof(func_info_t*), func_name_cmp);
396
397 return data;
398 }
399
400 static int id_cmp(const void *p1, const void *p2)
401 {
402 return *(const DISPID*)p1 - *(const DISPID*)p2;
403 }
404
405 HRESULT get_dispids(tid_t tid, DWORD *ret_size, DISPID **ret)
406 {
407 unsigned i, func_cnt;
408 FUNCDESC *funcdesc;
409 ITypeInfo *ti;
410 TYPEATTR *attr;
411 DISPID *ids;
412 HRESULT hres;
413
414 hres = get_typeinfo(tid, &ti);
415 if(FAILED(hres))
416 return hres;
417
418 hres = ITypeInfo_GetTypeAttr(ti, &attr);
419 if(FAILED(hres)) {
420 ITypeInfo_Release(ti);
421 return hres;
422 }
423
424 func_cnt = attr->cFuncs;
425 ITypeInfo_ReleaseTypeAttr(ti, attr);
426
427 ids = heap_alloc(func_cnt*sizeof(DISPID));
428 if(!ids) {
429 ITypeInfo_Release(ti);
430 return E_OUTOFMEMORY;
431 }
432
433 for(i=0; i < func_cnt; i++) {
434 hres = ITypeInfo_GetFuncDesc(ti, i, &funcdesc);
435 if(FAILED(hres))
436 break;
437
438 ids[i] = funcdesc->memid;
439 ITypeInfo_ReleaseFuncDesc(ti, funcdesc);
440 }
441
442 ITypeInfo_Release(ti);
443 if(FAILED(hres)) {
444 heap_free(ids);
445 return hres;
446 }
447
448 qsort(ids, func_cnt, sizeof(DISPID), id_cmp);
449
450 *ret_size = func_cnt;
451 *ret = ids;
452 return S_OK;
453 }
454
455 static dispex_data_t *get_dispex_data(DispatchEx *This)
456 {
457 if(This->data->data)
458 return This->data->data;
459
460 EnterCriticalSection(&cs_dispex_static_data);
461
462 if(!This->data->data)
463 This->data->data = preprocess_dispex_data(This);
464
465 LeaveCriticalSection(&cs_dispex_static_data);
466
467 return This->data->data;
468 }
469
470 static inline BOOL is_custom_dispid(DISPID id)
471 {
472 return MSHTML_DISPID_CUSTOM_MIN <= id && id <= MSHTML_DISPID_CUSTOM_MAX;
473 }
474
475 static inline BOOL is_dynamic_dispid(DISPID id)
476 {
477 return DISPID_DYNPROP_0 <= id && id <= DISPID_DYNPROP_MAX;
478 }
479
480 dispex_prop_type_t get_dispid_type(DISPID id)
481 {
482 if(is_dynamic_dispid(id))
483 return DISPEXPROP_DYNAMIC;
484 if(is_custom_dispid(id))
485 return DISPEXPROP_CUSTOM;
486 return DISPEXPROP_BUILTIN;
487 }
488
489 static HRESULT variant_copy(VARIANT *dest, VARIANT *src)
490 {
491 if(V_VT(src) == VT_BSTR && !V_BSTR(src)) {
492 V_VT(dest) = VT_BSTR;
493 V_BSTR(dest) = NULL;
494 return S_OK;
495 }
496
497 return VariantCopy(dest, src);
498 }
499
500 static inline dispex_dynamic_data_t *get_dynamic_data(DispatchEx *This)
501 {
502 if(This->dynamic_data)
503 return This->dynamic_data;
504
505 This->dynamic_data = heap_alloc_zero(sizeof(dispex_dynamic_data_t));
506 if(!This->dynamic_data)
507 return NULL;
508
509 if(This->data->vtbl && This->data->vtbl->populate_props)
510 This->data->vtbl->populate_props(This);
511
512 return This->dynamic_data;
513 }
514
515 static HRESULT get_dynamic_prop(DispatchEx *This, const WCHAR *name, DWORD flags, dynamic_prop_t **ret)
516 {
517 const BOOL alloc = flags & fdexNameEnsure;
518 dispex_dynamic_data_t *data;
519 dynamic_prop_t *prop;
520
521 data = get_dynamic_data(This);
522 if(!data)
523 return E_OUTOFMEMORY;
524
525 for(prop = data->props; prop < data->props+data->prop_cnt; prop++) {
526 if(flags & fdexNameCaseInsensitive ? !strcmpiW(prop->name, name) : !strcmpW(prop->name, name)) {
527 if(prop->flags & DYNPROP_DELETED) {
528 if(!alloc)
529 return DISP_E_UNKNOWNNAME;
530 prop->flags &= ~DYNPROP_DELETED;
531 }
532 *ret = prop;
533 return S_OK;
534 }
535 }
536
537 if(!alloc)
538 return DISP_E_UNKNOWNNAME;
539
540 TRACE("creating dynamic prop %s\n", debugstr_w(name));
541
542 if(!data->buf_size) {
543 data->props = heap_alloc(sizeof(dynamic_prop_t)*4);
544 if(!data->props)
545 return E_OUTOFMEMORY;
546 data->buf_size = 4;
547 }else if(data->buf_size == data->prop_cnt) {
548 dynamic_prop_t *new_props;
549
550 new_props = heap_realloc(data->props, sizeof(dynamic_prop_t)*(data->buf_size<<1));
551 if(!new_props)
552 return E_OUTOFMEMORY;
553
554 data->props = new_props;
555 data->buf_size <<= 1;
556 }
557
558 prop = data->props + data->prop_cnt;
559
560 prop->name = heap_strdupW(name);
561 if(!prop->name)
562 return E_OUTOFMEMORY;
563
564 VariantInit(&prop->var);
565 prop->flags = 0;
566 data->prop_cnt++;
567 *ret = prop;
568 return S_OK;
569 }
570
571 HRESULT dispex_get_dprop_ref(DispatchEx *This, const WCHAR *name, BOOL alloc, VARIANT **ret)
572 {
573 dynamic_prop_t *prop;
574 HRESULT hres;
575
576 hres = get_dynamic_prop(This, name, alloc ? fdexNameEnsure : 0, &prop);
577 if(FAILED(hres))
578 return hres;
579
580 *ret = &prop->var;
581 return S_OK;
582 }
583
584 HRESULT dispex_get_dynid(DispatchEx *This, const WCHAR *name, DISPID *id)
585 {
586 dynamic_prop_t *prop;
587 HRESULT hres;
588
589 hres = get_dynamic_prop(This, name, fdexNameEnsure, &prop);
590 if(FAILED(hres))
591 return hres;
592
593 *id = DISPID_DYNPROP_0 + (prop - This->dynamic_data->props);
594 return S_OK;
595 }
596
597 static HRESULT dispex_value(DispatchEx *This, LCID lcid, WORD flags, DISPPARAMS *params,
598 VARIANT *res, EXCEPINFO *ei, IServiceProvider *caller)
599 {
600 if(This->data->vtbl && This->data->vtbl->value)
601 return This->data->vtbl->value(This, lcid, flags, params, res, ei, caller);
602
603 switch(flags) {
604 case DISPATCH_PROPERTYGET:
605 V_VT(res) = VT_BSTR;
606 V_BSTR(res) = SysAllocString(objectW);
607 if(!V_BSTR(res))
608 return E_OUTOFMEMORY;
609 break;
610 default:
611 FIXME("Unimplemented flags %x\n", flags);
612 return E_NOTIMPL;
613 }
614
615 return S_OK;
616 }
617
618 static HRESULT typeinfo_invoke(DispatchEx *This, func_info_t *func, WORD flags, DISPPARAMS *dp, VARIANT *res,
619 EXCEPINFO *ei)
620 {
621 ITypeInfo *ti;
622 IUnknown *unk;
623 UINT argerr=0;
624 HRESULT hres;
625
626 hres = get_typeinfo(func->tid, &ti);
627 if(FAILED(hres)) {
628 ERR("Could not get type info: %08x\n", hres);
629 return hres;
630 }
631
632 hres = IUnknown_QueryInterface(This->outer, tid_ids[func->tid], (void**)&unk);
633 if(FAILED(hres)) {
634 ERR("Could not get iface %s: %08x\n", debugstr_mshtml_guid(tid_ids[func->tid]), hres);
635 return E_FAIL;
636 }
637
638 hres = ITypeInfo_Invoke(ti, unk, func->id, flags, dp, res, ei, &argerr);
639
640 IUnknown_Release(unk);
641 return hres;
642 }
643
644 static inline func_disp_t *impl_from_IUnknown(IUnknown *iface)
645 {
646 return CONTAINING_RECORD(iface, func_disp_t, IUnknown_iface);
647 }
648
649 static HRESULT WINAPI Function_QueryInterface(IUnknown *iface, REFIID riid, void **ppv)
650 {
651 func_disp_t *This = impl_from_IUnknown(iface);
652
653 if(IsEqualGUID(&IID_IUnknown, riid)) {
654 TRACE("(%p)->(IID_IUnknown %p)\n", This, ppv);
655 *ppv = &This->IUnknown_iface;
656 }else if(dispex_query_interface(&This->dispex, riid, ppv)) {
657 return *ppv ? S_OK : E_NOINTERFACE;
658 }else {
659 *ppv = NULL;
660 return E_NOINTERFACE;
661 }
662
663 IUnknown_AddRef((IUnknown*)*ppv);
664 return S_OK;
665 }
666
667 static ULONG WINAPI Function_AddRef(IUnknown *iface)
668 {
669 func_disp_t *This = impl_from_IUnknown(iface);
670 LONG ref = InterlockedIncrement(&This->ref);
671
672 TRACE("(%p) ref=%d\n", This, ref);
673
674 return ref;
675 }
676
677 static ULONG WINAPI Function_Release(IUnknown *iface)
678 {
679 func_disp_t *This = impl_from_IUnknown(iface);
680 LONG ref = InterlockedDecrement(&This->ref);
681
682 TRACE("(%p) ref=%d\n", This, ref);
683
684 if(!ref) {
685 assert(!This->obj);
686 release_dispex(&This->dispex);
687 heap_free(This);
688 }
689
690 return ref;
691 }
692
693 static const IUnknownVtbl FunctionUnkVtbl = {
694 Function_QueryInterface,
695 Function_AddRef,
696 Function_Release
697 };
698
699 static inline func_disp_t *impl_from_DispatchEx(DispatchEx *iface)
700 {
701 return CONTAINING_RECORD(iface, func_disp_t, dispex);
702 }
703
704 static HRESULT function_value(DispatchEx *dispex, LCID lcid, WORD flags, DISPPARAMS *params,
705 VARIANT *res, EXCEPINFO *ei, IServiceProvider *caller)
706 {
707 func_disp_t *This = impl_from_DispatchEx(dispex);
708 HRESULT hres;
709
710 switch(flags) {
711 case DISPATCH_METHOD|DISPATCH_PROPERTYGET:
712 if(!res)
713 return E_INVALIDARG;
714 /* fall through */
715 case DISPATCH_METHOD:
716 if(!This->obj)
717 return E_UNEXPECTED;
718 hres = typeinfo_invoke(This->obj, This->info, flags, params, res, ei);
719 break;
720 case DISPATCH_PROPERTYGET: {
721 unsigned name_len;
722 WCHAR *ptr;
723 BSTR str;
724
725 static const WCHAR func_prefixW[] =
726 {'\n','f','u','n','c','t','i','o','n',' '};
727 static const WCHAR func_suffixW[] =
728 {'(',')',' ','{','\n',' ',' ',' ',' ','[','n','a','t','i','v','e',' ','c','o','d','e',']','\n','}','\n'};
729
730 /* FIXME: This probably should be more generic. Also we should try to get IID_IActiveScriptSite and SID_GetCaller. */
731 if(!caller)
732 return E_ACCESSDENIED;
733
734 name_len = SysStringLen(This->info->name);
735 ptr = str = SysAllocStringLen(NULL, name_len + (sizeof(func_prefixW)+sizeof(func_suffixW))/sizeof(WCHAR));
736 if(!str)
737 return E_OUTOFMEMORY;
738
739 memcpy(ptr, func_prefixW, sizeof(func_prefixW));
740 ptr += sizeof(func_prefixW)/sizeof(WCHAR);
741
742 memcpy(ptr, This->info->name, name_len*sizeof(WCHAR));
743 ptr += name_len;
744
745 memcpy(ptr, func_suffixW, sizeof(func_suffixW));
746
747 V_VT(res) = VT_BSTR;
748 V_BSTR(res) = str;
749 return S_OK;
750 }
751 default:
752 FIXME("Unimplemented flags %x\n", flags);
753 hres = E_NOTIMPL;
754 }
755
756 return hres;
757 }
758
759 static const dispex_static_data_vtbl_t function_dispex_vtbl = {
760 function_value,
761 NULL,
762 NULL,
763 NULL
764 };
765
766 static const tid_t function_iface_tids[] = {0};
767
768 static dispex_static_data_t function_dispex = {
769 &function_dispex_vtbl,
770 NULL_tid,
771 NULL,
772 function_iface_tids
773 };
774
775 static func_disp_t *create_func_disp(DispatchEx *obj, func_info_t *info)
776 {
777 func_disp_t *ret;
778
779 ret = heap_alloc_zero(sizeof(func_disp_t));
780 if(!ret)
781 return NULL;
782
783 ret->IUnknown_iface.lpVtbl = &FunctionUnkVtbl;
784 init_dispex(&ret->dispex, &ret->IUnknown_iface, &function_dispex);
785 ret->ref = 1;
786 ret->obj = obj;
787 ret->info = info;
788
789 return ret;
790 }
791
792 static HRESULT invoke_disp_value(DispatchEx *This, IDispatch *func_disp, LCID lcid, WORD flags, DISPPARAMS *dp,
793 VARIANT *res, EXCEPINFO *ei, IServiceProvider *caller)
794 {
795 DISPID named_arg = DISPID_THIS;
796 DISPPARAMS new_dp = {NULL, &named_arg, 0, 1};
797 IDispatchEx *dispex;
798 HRESULT hres;
799
800 if(dp->cNamedArgs) {
801 FIXME("named args not supported\n");
802 return E_NOTIMPL;
803 }
804
805 new_dp.rgvarg = heap_alloc((dp->cArgs+1)*sizeof(VARIANTARG));
806 if(!new_dp.rgvarg)
807 return E_OUTOFMEMORY;
808
809 new_dp.cArgs = dp->cArgs+1;
810 memcpy(new_dp.rgvarg+1, dp->rgvarg, dp->cArgs*sizeof(VARIANTARG));
811
812 V_VT(new_dp.rgvarg) = VT_DISPATCH;
813 V_DISPATCH(new_dp.rgvarg) = (IDispatch*)&This->IDispatchEx_iface;
814
815 hres = IDispatch_QueryInterface(func_disp, &IID_IDispatchEx, (void**)&dispex);
816 TRACE(">>>\n");
817 if(SUCCEEDED(hres)) {
818 hres = IDispatchEx_InvokeEx(dispex, DISPID_VALUE, lcid, flags, &new_dp, res, ei, caller);
819 IDispatchEx_Release(dispex);
820 }else {
821 ULONG err = 0;
822 hres = IDispatch_Invoke(func_disp, DISPID_VALUE, &IID_NULL, lcid, flags, &new_dp, res, ei, &err);
823 }
824 if(SUCCEEDED(hres))
825 TRACE("<<< %s\n", debugstr_variant(res));
826 else
827 WARN("<<< %08x\n", hres);
828
829 heap_free(new_dp.rgvarg);
830 return hres;
831 }
832
833 static HRESULT get_func_obj_entry(DispatchEx *This, func_info_t *func, func_obj_entry_t **ret)
834 {
835 dispex_dynamic_data_t *dynamic_data;
836
837 dynamic_data = get_dynamic_data(This);
838 if(!dynamic_data)
839 return E_OUTOFMEMORY;
840
841 if(!dynamic_data->func_disps) {
842 dynamic_data->func_disps = heap_alloc_zero(This->data->data->func_disp_cnt * sizeof(*dynamic_data->func_disps));
843 if(!dynamic_data->func_disps)
844 return E_OUTOFMEMORY;
845 }
846
847 if(!dynamic_data->func_disps[func->func_disp_idx].func_obj) {
848 func_disp_t *func_obj;
849
850 func_obj = create_func_disp(This, func);
851 if(!func_obj)
852 return E_OUTOFMEMORY;
853
854 dynamic_data->func_disps[func->func_disp_idx].func_obj = func_obj;
855
856 IDispatchEx_AddRef(&func_obj->dispex.IDispatchEx_iface);
857 dynamic_data->func_disps[func->func_disp_idx].val = (IDispatch*)&func_obj->dispex.IDispatchEx_iface;
858 }
859
860 *ret = dynamic_data->func_disps+func->func_disp_idx;
861 return S_OK;
862 }
863
864 static HRESULT get_builtin_func(dispex_data_t *data, DISPID id, func_info_t **ret)
865 {
866 int min, max, n;
867
868 min = 0;
869 max = data->func_cnt-1;
870
871 while(min <= max) {
872 n = (min+max)/2;
873
874 if(data->funcs[n].id == id) {
875 *ret = data->funcs+n;
876 return S_OK;
877 }
878
879 if(data->funcs[n].id < id)
880 min = n+1;
881 else
882 max = n-1;
883 }
884
885 WARN("invalid id %x\n", id);
886 return DISP_E_UNKNOWNNAME;
887 }
888
889 static HRESULT get_builtin_id(DispatchEx *This, BSTR name, DWORD grfdex, DISPID *ret)
890 {
891 dispex_data_t *data;
892 int min, max, n, c;
893
894 data = get_dispex_data(This);
895 if(!data)
896 return E_FAIL;
897
898 min = 0;
899 max = data->func_cnt-1;
900
901 while(min <= max) {
902 n = (min+max)/2;
903
904 c = strcmpiW(data->name_table[n]->name, name);
905 if(!c) {
906 if((grfdex & fdexNameCaseSensitive) && strcmpW(data->name_table[n]->name, name))
907 break;
908
909 *ret = data->name_table[n]->id;
910 return S_OK;
911 }
912
913 if(c > 0)
914 max = n-1;
915 else
916 min = n+1;
917 }
918
919 if(This->data->vtbl && This->data->vtbl->get_dispid) {
920 HRESULT hres;
921
922 hres = This->data->vtbl->get_dispid(This, name, grfdex, ret);
923 if(hres != DISP_E_UNKNOWNNAME)
924 return hres;
925 }
926
927 return DISP_E_UNKNOWNNAME;
928 }
929
930 static HRESULT change_type(VARIANT *dst, VARIANT *src, VARTYPE vt, IServiceProvider *caller)
931 {
932 V_VT(dst) = VT_EMPTY;
933
934 if(caller) {
935 IVariantChangeType *change_type = NULL;
936 HRESULT hres;
937
938 hres = IServiceProvider_QueryService(caller, &SID_VariantConversion, &IID_IVariantChangeType, (void**)&change_type);
939 if(SUCCEEDED(hres)) {
940 hres = IVariantChangeType_ChangeType(change_type, dst, src, LOCALE_NEUTRAL, vt);
941 IVariantChangeType_Release(change_type);
942 return hres;
943 }
944 }
945
946 switch(vt) {
947 case VT_BOOL:
948 if(V_VT(src) == VT_BSTR) {
949 V_VT(dst) = VT_BOOL;
950 V_BOOL(dst) = V_BSTR(src) && *V_BSTR(src) ? VARIANT_TRUE : VARIANT_FALSE;
951 return S_OK;
952 }
953 break;
954 }
955
956 return VariantChangeType(dst, src, 0, vt);
957 }
958
959 static HRESULT builtin_propget(DispatchEx *This, func_info_t *func, DISPPARAMS *dp, VARIANT *res)
960 {
961 IUnknown *iface;
962 HRESULT hres;
963
964 if(dp && dp->cArgs) {
965 FIXME("cArgs %d\n", dp->cArgs);
966 return E_NOTIMPL;
967 }
968
969 assert(func->get_vtbl_off);
970
971 hres = IUnknown_QueryInterface(This->outer, tid_ids[func->tid], (void**)&iface);
972 if(SUCCEEDED(hres)) {
973 switch(func->prop_vt) {
974 #define CASE_VT(vt,type,access) \
975 case vt: { \
976 type val; \
977 hres = ((HRESULT (WINAPI*)(IUnknown*,type*))((void**)iface->lpVtbl)[func->get_vtbl_off])(iface,&val); \
978 if(SUCCEEDED(hres)) \
979 access(res) = val; \
980 } \
981 break
982 BUILTIN_TYPES_SWITCH;
983 #undef CASE_VT
984 default:
985 FIXME("Unhandled vt %d\n", func->prop_vt);
986 hres = E_NOTIMPL;
987 }
988 IUnknown_Release(iface);
989 }
990
991 if(FAILED(hres))
992 return hres;
993
994 if(func->prop_vt != VT_VARIANT)
995 V_VT(res) = func->prop_vt == VT_PTR ? VT_DISPATCH : func->prop_vt;
996 return S_OK;
997 }
998
999 static HRESULT builtin_propput(DispatchEx *This, func_info_t *func, DISPPARAMS *dp, IServiceProvider *caller)
1000 {
1001 VARIANT *v, tmpv;
1002 IUnknown *iface;
1003 HRESULT hres;
1004
1005 if(dp->cArgs != 1 || (dp->cNamedArgs == 1 && *dp->rgdispidNamedArgs != DISPID_PROPERTYPUT)
1006 || dp->cNamedArgs > 1) {
1007 FIXME("invalid args\n");
1008 return E_INVALIDARG;
1009 }
1010
1011 if(!func->put_vtbl_off) {
1012 FIXME("No setter\n");
1013 return E_FAIL;
1014 }
1015
1016 v = dp->rgvarg;
1017 if(func->prop_vt != VT_VARIANT && V_VT(v) != func->prop_vt) {
1018 hres = change_type(&tmpv, v, func->prop_vt, caller);
1019 if(FAILED(hres))
1020 return hres;
1021 v = &tmpv;
1022 }
1023
1024 hres = IUnknown_QueryInterface(This->outer, tid_ids[func->tid], (void**)&iface);
1025 if(SUCCEEDED(hres)) {
1026 switch(func->prop_vt) {
1027 #define CASE_VT(vt,type,access) \
1028 case vt: \
1029 hres = ((HRESULT (WINAPI*)(IUnknown*,type))((void**)iface->lpVtbl)[func->put_vtbl_off])(iface,access(v)); \
1030 break
1031 BUILTIN_TYPES_SWITCH;
1032 #undef CASE_VT
1033 default:
1034 FIXME("Unimplemented vt %d\n", func->prop_vt);
1035 hres = E_NOTIMPL;
1036 }
1037
1038 IUnknown_Release(iface);
1039 }
1040
1041 if(v == &tmpv)
1042 VariantClear(v);
1043 return hres;
1044 }
1045
1046 static HRESULT invoke_builtin_function(DispatchEx *This, func_info_t *func, DISPPARAMS *dp, VARIANT *res, IServiceProvider *caller)
1047 {
1048 VARIANT arg_buf[MAX_ARGS], *arg_ptrs[MAX_ARGS], *arg, retv, ret_ref, vhres;
1049 unsigned i, nconv = 0;
1050 IUnknown *iface;
1051 HRESULT hres;
1052
1053 if(dp->cNamedArgs) {
1054 FIXME("Named arguments not supported\n");
1055 return E_NOTIMPL;
1056 }
1057
1058 if(dp->cArgs != func->argc) {
1059 FIXME("Invalid argument count (expected %u, got %u)\n", func->argc, dp->cArgs);
1060 return E_INVALIDARG;
1061 }
1062
1063 hres = IUnknown_QueryInterface(This->outer, tid_ids[func->tid], (void**)&iface);
1064 if(FAILED(hres))
1065 return hres;
1066
1067 for(i=0; i < func->argc; i++) {
1068 arg = dp->rgvarg+dp->cArgs-i-1;
1069 if(func->arg_types[i] == V_VT(arg)) {
1070 arg_ptrs[i] = arg;
1071 }else {
1072 hres = change_type(arg_buf+nconv, arg, func->arg_types[i], caller);
1073 if(FAILED(hres))
1074 break;
1075 arg_ptrs[i] = arg_buf + nconv++;
1076 }
1077 }
1078
1079 if(SUCCEEDED(hres)) {
1080 if(func->prop_vt == VT_VOID) {
1081 V_VT(&retv) = VT_EMPTY;
1082 }else {
1083 V_VT(&retv) = func->prop_vt;
1084 arg_ptrs[func->argc] = &ret_ref;
1085 V_VT(&ret_ref) = VT_BYREF|func->prop_vt;
1086
1087 switch(func->prop_vt) {
1088 #define CASE_VT(vt,type,access) \
1089 case vt: \
1090 V_BYREF(&ret_ref) = &access(&retv); \
1091 break
1092 BUILTIN_TYPES_SWITCH;
1093 #undef CASE_VT
1094 default:
1095 assert(0);
1096 }
1097 }
1098
1099 V_VT(&vhres) = VT_ERROR;
1100 hres = DispCallFunc(iface, func->call_vtbl_off*sizeof(void*), CC_STDCALL, VT_ERROR,
1101 func->argc + (func->prop_vt == VT_VOID ? 0 : 1), func->arg_types, arg_ptrs, &vhres);
1102 }
1103
1104 while(nconv--)
1105 VariantClear(arg_buf+nconv);
1106 IUnknown_Release(iface);
1107 if(FAILED(hres))
1108 return hres;
1109 if(FAILED(V_ERROR(&vhres)))
1110 return V_ERROR(&vhres);
1111
1112 if(res)
1113 *res = retv;
1114 else
1115 VariantClear(&retv);
1116 return V_ERROR(&vhres);
1117 }
1118
1119 static HRESULT function_invoke(DispatchEx *This, func_info_t *func, WORD flags, DISPPARAMS *dp, VARIANT *res,
1120 EXCEPINFO *ei, IServiceProvider *caller)
1121 {
1122 HRESULT hres;
1123
1124 switch(flags) {
1125 case DISPATCH_METHOD|DISPATCH_PROPERTYGET:
1126 if(!res)
1127 return E_INVALIDARG;
1128 /* fall through */
1129 case DISPATCH_METHOD:
1130 if(This->dynamic_data && This->dynamic_data->func_disps
1131 && This->dynamic_data->func_disps[func->func_disp_idx].func_obj) {
1132 func_obj_entry_t *entry = This->dynamic_data->func_disps + func->func_disp_idx;
1133
1134 if((IDispatch*)&entry->func_obj->dispex.IDispatchEx_iface != entry->val) {
1135 if(!entry->val) {
1136 FIXME("Calling null\n");
1137 return E_FAIL;
1138 }
1139
1140 hres = invoke_disp_value(This, entry->val, 0, flags, dp, res, ei, NULL);
1141 break;
1142 }
1143 }
1144
1145 if(func->call_vtbl_off)
1146 hres = invoke_builtin_function(This, func, dp, res, caller);
1147 else
1148 hres = typeinfo_invoke(This, func, flags, dp, res, ei);
1149 break;
1150 case DISPATCH_PROPERTYGET: {
1151 func_obj_entry_t *entry;
1152
1153 if(func->id == DISPID_VALUE) {
1154 BSTR ret;
1155
1156 ret = SysAllocString(objectW);
1157 if(!ret)
1158 return E_OUTOFMEMORY;
1159
1160 V_VT(res) = VT_BSTR;
1161 V_BSTR(res) = ret;
1162 return S_OK;
1163 }
1164
1165 hres = get_func_obj_entry(This, func, &entry);
1166 if(FAILED(hres))
1167 return hres;
1168
1169 V_VT(res) = VT_DISPATCH;
1170 V_DISPATCH(res) = entry->val;
1171 if(V_DISPATCH(res))
1172 IDispatch_AddRef(V_DISPATCH(res));
1173 hres = S_OK;
1174 break;
1175 }
1176 case DISPATCH_PROPERTYPUT: {
1177 func_obj_entry_t *entry;
1178 VARIANT *v;
1179
1180 if(dp->cArgs != 1 || (dp->cNamedArgs == 1 && *dp->rgdispidNamedArgs != DISPID_PROPERTYPUT)
1181 || dp->cNamedArgs > 1) {
1182 FIXME("invalid args\n");
1183 return E_INVALIDARG;
1184 }
1185
1186 v = dp->rgvarg;
1187 /* FIXME: not exactly right */
1188 if(V_VT(v) != VT_DISPATCH)
1189 return E_NOTIMPL;
1190
1191 hres = get_func_obj_entry(This, func, &entry);
1192 if(FAILED(hres))
1193 return hres;
1194
1195 if(entry->val)
1196 IDispatch_Release(entry->val);
1197 entry->val = V_DISPATCH(v);
1198 if(entry->val)
1199 IDispatch_AddRef(entry->val);
1200 hres = S_OK;
1201 break;
1202 }
1203 default:
1204 FIXME("Unimplemented flags %x\n", flags);
1205 hres = E_NOTIMPL;
1206 }
1207
1208 return hres;
1209 }
1210
1211 static HRESULT invoke_builtin_prop(DispatchEx *This, DISPID id, LCID lcid, WORD flags, DISPPARAMS *dp,
1212 VARIANT *res, EXCEPINFO *ei, IServiceProvider *caller)
1213 {
1214 dispex_data_t *data;
1215 func_info_t *func;
1216 HRESULT hres;
1217
1218 data = get_dispex_data(This);
1219 if(!data)
1220 return E_FAIL;
1221
1222 hres = get_builtin_func(data, id, &func);
1223 if(id == DISPID_VALUE && hres == DISP_E_UNKNOWNNAME)
1224 return dispex_value(This, lcid, flags, dp, res, ei, caller);
1225 if(FAILED(hres))
1226 return hres;
1227
1228 if(func->func_disp_idx != -1)
1229 return function_invoke(This, func, flags, dp, res, ei, caller);
1230
1231 switch(flags) {
1232 case DISPATCH_PROPERTYPUT:
1233 case DISPATCH_PROPERTYPUT | DISPATCH_PROPERTYPUTREF:
1234 if(res)
1235 V_VT(res) = VT_EMPTY;
1236 hres = builtin_propput(This, func, dp, caller);
1237 break;
1238 case DISPATCH_PROPERTYGET:
1239 hres = builtin_propget(This, func, dp, res);
1240 break;
1241 default:
1242 if(!func->get_vtbl_off) {
1243 hres = typeinfo_invoke(This, func, flags, dp, res, ei);
1244 }else {
1245 VARIANT v;
1246
1247 hres = builtin_propget(This, func, NULL, &v);
1248 if(FAILED(hres))
1249 return hres;
1250
1251 if(flags != (DISPATCH_PROPERTYGET|DISPATCH_METHOD) || dp->cArgs) {
1252 if(V_VT(&v) != VT_DISPATCH) {
1253 FIXME("Not a function %s flags %08x\n", debugstr_variant(&v), flags);
1254 VariantClear(&v);
1255 return E_FAIL;
1256 }
1257
1258 hres = invoke_disp_value(This, V_DISPATCH(&v), lcid, flags, dp, res, ei, caller);
1259 IDispatch_Release(V_DISPATCH(&v));
1260 }else if(res) {
1261 *res = v;
1262 }else {
1263 VariantClear(&v);
1264 }
1265 }
1266 }
1267
1268 return hres;
1269 }
1270
1271 HRESULT remove_prop(DispatchEx *This, BSTR name, VARIANT_BOOL *success)
1272 {
1273 dynamic_prop_t *prop;
1274 DISPID id;
1275 HRESULT hres;
1276
1277 hres = get_builtin_id(This, name, 0, &id);
1278 if(hres == S_OK) {
1279 DISPID named_id = DISPID_PROPERTYPUT;
1280 VARIANT var;
1281 DISPPARAMS dp = {&var,&named_id,1,1};
1282 EXCEPINFO ei;
1283
1284 V_VT(&var) = VT_EMPTY;
1285 memset(&ei, 0, sizeof(ei));
1286 hres = invoke_builtin_prop(This, id, 0, DISPATCH_PROPERTYPUT, &dp, NULL, &ei, NULL);
1287 if(FAILED(hres))
1288 return hres;
1289
1290 *success = VARIANT_TRUE;
1291 return S_OK;
1292 }
1293
1294 hres = get_dynamic_prop(This, name, 0, &prop);
1295 if(FAILED(hres)) {
1296 if(hres != DISP_E_UNKNOWNNAME)
1297 return hres;
1298 *success = VARIANT_FALSE;
1299 return S_OK;
1300 }
1301
1302 VariantClear(&prop->var);
1303 prop->flags |= DYNPROP_DELETED;
1304 *success = VARIANT_TRUE;
1305 return S_OK;
1306 }
1307
1308 static inline DispatchEx *impl_from_IDispatchEx(IDispatchEx *iface)
1309 {
1310 return CONTAINING_RECORD(iface, DispatchEx, IDispatchEx_iface);
1311 }
1312
1313 static HRESULT WINAPI DispatchEx_QueryInterface(IDispatchEx *iface, REFIID riid, void **ppv)
1314 {
1315 DispatchEx *This = impl_from_IDispatchEx(iface);
1316
1317 return IUnknown_QueryInterface(This->outer, riid, ppv);
1318 }
1319
1320 static ULONG WINAPI DispatchEx_AddRef(IDispatchEx *iface)
1321 {
1322 DispatchEx *This = impl_from_IDispatchEx(iface);
1323
1324 return IUnknown_AddRef(This->outer);
1325 }
1326
1327 static ULONG WINAPI DispatchEx_Release(IDispatchEx *iface)
1328 {
1329 DispatchEx *This = impl_from_IDispatchEx(iface);
1330
1331 return IUnknown_Release(This->outer);
1332 }
1333
1334 static HRESULT WINAPI DispatchEx_GetTypeInfoCount(IDispatchEx *iface, UINT *pctinfo)
1335 {
1336 DispatchEx *This = impl_from_IDispatchEx(iface);
1337
1338 TRACE("(%p)->(%p)\n", This, pctinfo);
1339
1340 *pctinfo = 1;
1341 return S_OK;
1342 }
1343
1344 static HRESULT WINAPI DispatchEx_GetTypeInfo(IDispatchEx *iface, UINT iTInfo,
1345 LCID lcid, ITypeInfo **ppTInfo)
1346 {
1347 DispatchEx *This = impl_from_IDispatchEx(iface);
1348 HRESULT hres;
1349
1350 TRACE("(%p)->(%u %u %p)\n", This, iTInfo, lcid, ppTInfo);
1351
1352 hres = get_typeinfo(This->data->disp_tid, ppTInfo);
1353 if(FAILED(hres))
1354 return hres;
1355
1356 ITypeInfo_AddRef(*ppTInfo);
1357 return S_OK;
1358 }
1359
1360 static HRESULT WINAPI DispatchEx_GetIDsOfNames(IDispatchEx *iface, REFIID riid,
1361 LPOLESTR *rgszNames, UINT cNames,
1362 LCID lcid, DISPID *rgDispId)
1363 {
1364 DispatchEx *This = impl_from_IDispatchEx(iface);
1365 UINT i;
1366 HRESULT hres;
1367
1368 TRACE("(%p)->(%s %p %u %u %p)\n", This, debugstr_guid(riid), rgszNames, cNames,
1369 lcid, rgDispId);
1370
1371 for(i=0; i < cNames; i++) {
1372 hres = IDispatchEx_GetDispID(&This->IDispatchEx_iface, rgszNames[i], 0, rgDispId+i);
1373 if(FAILED(hres))
1374 return hres;
1375 }
1376
1377 return S_OK;
1378 }
1379
1380 static HRESULT WINAPI DispatchEx_Invoke(IDispatchEx *iface, DISPID dispIdMember,
1381 REFIID riid, LCID lcid, WORD wFlags, DISPPARAMS *pDispParams,
1382 VARIANT *pVarResult, EXCEPINFO *pExcepInfo, UINT *puArgErr)
1383 {
1384 DispatchEx *This = impl_from_IDispatchEx(iface);
1385
1386 TRACE("(%p)->(%d %s %d %d %p %p %p %p)\n", This, dispIdMember, debugstr_guid(riid),
1387 lcid, wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr);
1388
1389 return IDispatchEx_InvokeEx(&This->IDispatchEx_iface, dispIdMember, lcid, wFlags, pDispParams,
1390 pVarResult, pExcepInfo, NULL);
1391 }
1392
1393 static HRESULT WINAPI DispatchEx_GetDispID(IDispatchEx *iface, BSTR bstrName, DWORD grfdex, DISPID *pid)
1394 {
1395 DispatchEx *This = impl_from_IDispatchEx(iface);
1396 dynamic_prop_t *dprop;
1397 HRESULT hres;
1398
1399 TRACE("(%p)->(%s %x %p)\n", This, debugstr_w(bstrName), grfdex, pid);
1400
1401 if(grfdex & ~(fdexNameCaseSensitive|fdexNameCaseInsensitive|fdexNameEnsure|fdexNameImplicit|FDEX_VERSION_MASK))
1402 FIXME("Unsupported grfdex %x\n", grfdex);
1403
1404 hres = get_builtin_id(This, bstrName, grfdex, pid);
1405 if(hres != DISP_E_UNKNOWNNAME)
1406 return hres;
1407
1408 hres = get_dynamic_prop(This, bstrName, grfdex, &dprop);
1409 if(FAILED(hres))
1410 return hres;
1411
1412 *pid = DISPID_DYNPROP_0 + (dprop - This->dynamic_data->props);
1413 return S_OK;
1414 }
1415
1416 static HRESULT WINAPI DispatchEx_InvokeEx(IDispatchEx *iface, DISPID id, LCID lcid, WORD wFlags, DISPPARAMS *pdp,
1417 VARIANT *pvarRes, EXCEPINFO *pei, IServiceProvider *pspCaller)
1418 {
1419 DispatchEx *This = impl_from_IDispatchEx(iface);
1420 HRESULT hres;
1421
1422 TRACE("(%p)->(%x %x %x %p %p %p %p)\n", This, id, lcid, wFlags, pdp, pvarRes, pei, pspCaller);
1423
1424 switch(get_dispid_type(id)) {
1425 case DISPEXPROP_CUSTOM:
1426 if(!This->data->vtbl || !This->data->vtbl->invoke)
1427 return DISP_E_UNKNOWNNAME;
1428 return This->data->vtbl->invoke(This, id, lcid, wFlags, pdp, pvarRes, pei, pspCaller);
1429
1430 case DISPEXPROP_DYNAMIC: {
1431 DWORD idx = id - DISPID_DYNPROP_0;
1432 dynamic_prop_t *prop;
1433
1434 if(!get_dynamic_data(This) || This->dynamic_data->prop_cnt <= idx)
1435 return DISP_E_UNKNOWNNAME;
1436
1437 prop = This->dynamic_data->props+idx;
1438
1439 switch(wFlags) {
1440 case DISPATCH_METHOD|DISPATCH_PROPERTYGET:
1441 if(!pvarRes)
1442 return E_INVALIDARG;
1443 /* fall through */
1444 case DISPATCH_METHOD:
1445 if(V_VT(&prop->var) != VT_DISPATCH) {
1446 FIXME("invoke %s\n", debugstr_variant(&prop->var));
1447 return E_NOTIMPL;
1448 }
1449
1450 return invoke_disp_value(This, V_DISPATCH(&prop->var), lcid, wFlags, pdp, pvarRes, pei, pspCaller);
1451 case DISPATCH_PROPERTYGET:
1452 if(prop->flags & DYNPROP_DELETED)
1453 return DISP_E_UNKNOWNNAME;
1454 V_VT(pvarRes) = VT_EMPTY;
1455 return variant_copy(pvarRes, &prop->var);
1456 case DISPATCH_PROPERTYPUT|DISPATCH_PROPERTYPUTREF:
1457 case DISPATCH_PROPERTYPUT:
1458 if(pdp->cArgs != 1 || (pdp->cNamedArgs == 1 && *pdp->rgdispidNamedArgs != DISPID_PROPERTYPUT)
1459 || pdp->cNamedArgs > 1) {
1460 FIXME("invalid args\n");
1461 return E_INVALIDARG;
1462 }
1463
1464 TRACE("put %s\n", debugstr_variant(pdp->rgvarg));
1465 VariantClear(&prop->var);
1466 hres = variant_copy(&prop->var, pdp->rgvarg);
1467 if(FAILED(hres))
1468 return hres;
1469
1470 prop->flags &= ~DYNPROP_DELETED;
1471 return S_OK;
1472 default:
1473 FIXME("unhandled wFlags %x\n", wFlags);
1474 return E_NOTIMPL;
1475 }
1476 }
1477 case DISPEXPROP_BUILTIN:
1478 if(wFlags == DISPATCH_CONSTRUCT) {
1479 if(id == DISPID_VALUE) {
1480 if(This->data->vtbl && This->data->vtbl->value) {
1481 return This->data->vtbl->value(This, lcid, wFlags, pdp, pvarRes, pei, pspCaller);
1482 }
1483 FIXME("DISPATCH_CONSTRUCT flag but missing value function\n");
1484 return E_FAIL;
1485 }
1486 FIXME("DISPATCH_CONSTRUCT flag without DISPID_VALUE\n");
1487 return E_FAIL;
1488 }
1489
1490 return invoke_builtin_prop(This, id, lcid, wFlags, pdp, pvarRes, pei, pspCaller);
1491 default:
1492 assert(0);
1493 return E_FAIL;
1494 }
1495 }
1496
1497 static HRESULT WINAPI DispatchEx_DeleteMemberByName(IDispatchEx *iface, BSTR bstrName, DWORD grfdex)
1498 {
1499 DispatchEx *This = impl_from_IDispatchEx(iface);
1500
1501 TRACE("(%p)->(%s %x)\n", This, debugstr_w(bstrName), grfdex);
1502
1503 /* Not implemented by IE */
1504 return E_NOTIMPL;
1505 }
1506
1507 static HRESULT WINAPI DispatchEx_DeleteMemberByDispID(IDispatchEx *iface, DISPID id)
1508 {
1509 DispatchEx *This = impl_from_IDispatchEx(iface);
1510
1511 TRACE("(%p)->(%x)\n", This, id);
1512
1513 /* Not implemented by IE */
1514 return E_NOTIMPL;
1515 }
1516
1517 static HRESULT WINAPI DispatchEx_GetMemberProperties(IDispatchEx *iface, DISPID id, DWORD grfdexFetch, DWORD *pgrfdex)
1518 {
1519 DispatchEx *This = impl_from_IDispatchEx(iface);
1520 FIXME("(%p)->(%x %x %p)\n", This, id, grfdexFetch, pgrfdex);
1521 return E_NOTIMPL;
1522 }
1523
1524 static HRESULT WINAPI DispatchEx_GetMemberName(IDispatchEx *iface, DISPID id, BSTR *pbstrName)
1525 {
1526 DispatchEx *This = impl_from_IDispatchEx(iface);
1527 dispex_data_t *data;
1528 func_info_t *func;
1529 HRESULT hres;
1530
1531 TRACE("(%p)->(%x %p)\n", This, id, pbstrName);
1532
1533 if(is_dynamic_dispid(id)) {
1534 DWORD idx = id - DISPID_DYNPROP_0;
1535
1536 if(!get_dynamic_data(This) || This->dynamic_data->prop_cnt <= idx)
1537 return DISP_E_UNKNOWNNAME;
1538
1539 *pbstrName = SysAllocString(This->dynamic_data->props[idx].name);
1540 if(!*pbstrName)
1541 return E_OUTOFMEMORY;
1542
1543 return S_OK;
1544 }
1545
1546 data = get_dispex_data(This);
1547 if(!data)
1548 return E_FAIL;
1549
1550 hres = get_builtin_func(data, id, &func);
1551 if(FAILED(hres))
1552 return hres;
1553
1554 *pbstrName = SysAllocString(func->name);
1555 if(!*pbstrName)
1556 return E_OUTOFMEMORY;
1557 return S_OK;
1558 }
1559
1560 static HRESULT WINAPI DispatchEx_GetNextDispID(IDispatchEx *iface, DWORD grfdex, DISPID id, DISPID *pid)
1561 {
1562 DispatchEx *This = impl_from_IDispatchEx(iface);
1563 dispex_data_t *data;
1564 func_info_t *func;
1565 HRESULT hres;
1566
1567 TRACE("(%p)->(%x %x %p)\n", This, grfdex, id, pid);
1568
1569 if(is_dynamic_dispid(id)) {
1570 DWORD idx = id - DISPID_DYNPROP_0;
1571
1572 if(!get_dynamic_data(This) || This->dynamic_data->prop_cnt <= idx)
1573 return DISP_E_UNKNOWNNAME;
1574
1575 while(++idx < This->dynamic_data->prop_cnt && This->dynamic_data->props[idx].flags & DYNPROP_DELETED);
1576
1577 if(idx == This->dynamic_data->prop_cnt) {
1578 *pid = DISPID_STARTENUM;
1579 return S_FALSE;
1580 }
1581
1582 *pid = DISPID_DYNPROP_0+idx;
1583 return S_OK;
1584 }
1585
1586 data = get_dispex_data(This);
1587 if(!data)
1588 return E_FAIL;
1589
1590 if(id == DISPID_STARTENUM) {
1591 func = data->funcs;
1592 }else {
1593 hres = get_builtin_func(data, id, &func);
1594 if(FAILED(hres))
1595 return hres;
1596 func++;
1597 }
1598
1599 while(func < data->funcs+data->func_cnt) {
1600 /* FIXME: Skip hidden properties */
1601 if(func->func_disp_idx == -1) {
1602 *pid = func->id;
1603 return S_OK;
1604 }
1605 func++;
1606 }
1607
1608 if(get_dynamic_data(This) && This->dynamic_data->prop_cnt) {
1609 *pid = DISPID_DYNPROP_0;
1610 return S_OK;
1611 }
1612
1613 *pid = DISPID_STARTENUM;
1614 return S_FALSE;
1615 }
1616
1617 static HRESULT WINAPI DispatchEx_GetNameSpaceParent(IDispatchEx *iface, IUnknown **ppunk)
1618 {
1619 DispatchEx *This = impl_from_IDispatchEx(iface);
1620 FIXME("(%p)->(%p)\n", This, ppunk);
1621 return E_NOTIMPL;
1622 }
1623
1624 static IDispatchExVtbl DispatchExVtbl = {
1625 DispatchEx_QueryInterface,
1626 DispatchEx_AddRef,
1627 DispatchEx_Release,
1628 DispatchEx_GetTypeInfoCount,
1629 DispatchEx_GetTypeInfo,
1630 DispatchEx_GetIDsOfNames,
1631 DispatchEx_Invoke,
1632 DispatchEx_GetDispID,
1633 DispatchEx_InvokeEx,
1634 DispatchEx_DeleteMemberByName,
1635 DispatchEx_DeleteMemberByDispID,
1636 DispatchEx_GetMemberProperties,
1637 DispatchEx_GetMemberName,
1638 DispatchEx_GetNextDispID,
1639 DispatchEx_GetNameSpaceParent
1640 };
1641
1642 BOOL dispex_query_interface(DispatchEx *This, REFIID riid, void **ppv)
1643 {
1644 static const IID IID_UndocumentedScriptIface =
1645 {0x719c3050,0xf9d3,0x11cf,{0xa4,0x93,0x00,0x40,0x05,0x23,0xa8,0xa0}};
1646 static const IID IID_IDispatchJS =
1647 {0x719c3050,0xf9d3,0x11cf,{0xa4,0x93,0x00,0x40,0x05,0x23,0xa8,0xa6}};
1648
1649 if(IsEqualGUID(&IID_IDispatch, riid)) {
1650 TRACE("(%p)->(IID_IDispatch %p)\n", This, ppv);
1651 *ppv = &This->IDispatchEx_iface;
1652 }else if(IsEqualGUID(&IID_IDispatchEx, riid)) {
1653 TRACE("(%p)->(IID_IDispatchEx %p)\n", This, ppv);
1654 *ppv = &This->IDispatchEx_iface;
1655 }else if(IsEqualGUID(&IID_IDispatchJS, riid)) {
1656 TRACE("(%p)->(IID_IDispatchJS %p) returning NULL\n", This, ppv);
1657 *ppv = NULL;
1658 }else if(IsEqualGUID(&IID_UndocumentedScriptIface, riid)) {
1659 TRACE("(%p)->(IID_UndocumentedScriptIface %p) returning NULL\n", This, ppv);
1660 *ppv = NULL;
1661 }else {
1662 return FALSE;
1663 }
1664
1665 if(*ppv)
1666 IUnknown_AddRef((IUnknown*)*ppv);
1667 return TRUE;
1668 }
1669
1670 void dispex_traverse(DispatchEx *This, nsCycleCollectionTraversalCallback *cb)
1671 {
1672 dynamic_prop_t *prop;
1673
1674 if(!This->dynamic_data)
1675 return;
1676
1677 for(prop = This->dynamic_data->props; prop < This->dynamic_data->props + This->dynamic_data->prop_cnt; prop++) {
1678 if(V_VT(&prop->var) == VT_DISPATCH)
1679 note_cc_edge((nsISupports*)V_DISPATCH(&prop->var), "dispex_data", cb);
1680 }
1681
1682 /* FIXME: Traverse func_disps */
1683 }
1684
1685 void dispex_unlink(DispatchEx *This)
1686 {
1687 dynamic_prop_t *prop;
1688
1689 if(!This->dynamic_data)
1690 return;
1691
1692 for(prop = This->dynamic_data->props; prop < This->dynamic_data->props + This->dynamic_data->prop_cnt; prop++) {
1693 if(V_VT(&prop->var) == VT_DISPATCH) {
1694 V_VT(&prop->var) = VT_EMPTY;
1695 IDispatch_Release(V_DISPATCH(&prop->var));
1696 }else {
1697 VariantClear(&prop->var);
1698 }
1699 }
1700 }
1701
1702 void release_dispex(DispatchEx *This)
1703 {
1704 dynamic_prop_t *prop;
1705
1706 if(!This->dynamic_data)
1707 return;
1708
1709 for(prop = This->dynamic_data->props; prop < This->dynamic_data->props + This->dynamic_data->prop_cnt; prop++) {
1710 VariantClear(&prop->var);
1711 heap_free(prop->name);
1712 }
1713
1714 heap_free(This->dynamic_data->props);
1715
1716 if(This->dynamic_data->func_disps) {
1717 func_obj_entry_t *iter;
1718
1719 for(iter = This->dynamic_data->func_disps; iter < This->dynamic_data->func_disps+This->data->data->func_disp_cnt; iter++) {
1720 if(iter->func_obj) {
1721 iter->func_obj->obj = NULL;
1722 IDispatchEx_Release(&iter->func_obj->dispex.IDispatchEx_iface);
1723 }
1724 if(iter->val)
1725 IDispatch_Release(iter->val);
1726 }
1727
1728 heap_free(This->dynamic_data->func_disps);
1729 }
1730
1731 heap_free(This->dynamic_data);
1732 }
1733
1734 void init_dispex(DispatchEx *dispex, IUnknown *outer, dispex_static_data_t *data)
1735 {
1736 dispex->IDispatchEx_iface.lpVtbl = &DispatchExVtbl;
1737 dispex->outer = outer;
1738 dispex->data = data;
1739 dispex->dynamic_data = NULL;
1740 }
Wine tree for Wine Gecko development