mshtml: Added support for calling functions returned by builtin getters.
[wine/multimedia.git] / dlls / mshtml / dispex.c
blobbbe6098eaf076b27b81881705a00d0addd4e03d2
1 /*
2 * Copyright 2008-2009 Jacek Caban for CodeWeavers
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
19 #include <stdarg.h>
20 #include <assert.h>
22 #define COBJMACROS
24 #include "windef.h"
25 #include "winbase.h"
26 #include "winuser.h"
27 #include "ole2.h"
29 #include "wine/debug.h"
31 #include "mshtml_private.h"
33 WINE_DEFAULT_DEBUG_CHANNEL(mshtml);
35 static CRITICAL_SECTION cs_dispex_static_data;
36 static CRITICAL_SECTION_DEBUG cs_dispex_static_data_dbg =
38 0, 0, &cs_dispex_static_data,
39 { &cs_dispex_static_data_dbg.ProcessLocksList, &cs_dispex_static_data_dbg.ProcessLocksList },
40 0, 0, { (DWORD_PTR)(__FILE__ ": dispex_static_data") }
42 static CRITICAL_SECTION cs_dispex_static_data = { &cs_dispex_static_data_dbg, -1, 0, 0, 0, 0 };
45 static const WCHAR objectW[] = {'[','o','b','j','e','c','t',']',0};
47 typedef struct {
48 DISPID id;
49 BSTR name;
50 tid_t tid;
51 SHORT put_vtbl_off;
52 SHORT get_vtbl_off;
53 SHORT func_disp_idx;
54 VARTYPE prop_vt;
55 } func_info_t;
57 struct dispex_data_t {
58 DWORD func_cnt;
59 func_info_t *funcs;
60 func_info_t **name_table;
61 DWORD func_disp_cnt;
63 struct list entry;
66 typedef struct {
67 VARIANT var;
68 LPWSTR name;
69 DWORD flags;
70 } dynamic_prop_t;
72 #define DYNPROP_DELETED 0x01
74 typedef struct {
75 DispatchEx dispex;
76 IUnknown IUnknown_iface;
77 DispatchEx *obj;
78 func_info_t *info;
79 } func_disp_t;
81 struct dispex_dynamic_data_t {
82 DWORD buf_size;
83 DWORD prop_cnt;
84 dynamic_prop_t *props;
85 func_disp_t **func_disps;
88 #define DISPID_DYNPROP_0 0x50000000
89 #define DISPID_DYNPROP_MAX 0x5fffffff
91 #define FDEX_VERSION_MASK 0xf0000000
93 static ITypeLib *typelib;
94 static ITypeInfo *typeinfos[LAST_tid];
95 static struct list dispex_data_list = LIST_INIT(dispex_data_list);
97 static REFIID tid_ids[] = {
98 #define XIID(iface) &IID_ ## iface,
99 #define XDIID(iface) &DIID_ ## iface,
100 TID_LIST
101 #undef XIID
102 #undef XDIID
105 static HRESULT load_typelib(void)
107 HRESULT hres;
108 ITypeLib *tl;
110 hres = LoadRegTypeLib(&LIBID_MSHTML, 4, 0, LOCALE_SYSTEM_DEFAULT, &tl);
111 if(FAILED(hres)) {
112 ERR("LoadRegTypeLib failed: %08x\n", hres);
113 return hres;
116 if(InterlockedCompareExchangePointer((void**)&typelib, tl, NULL))
117 ITypeLib_Release(tl);
118 return hres;
121 static HRESULT get_typeinfo(tid_t tid, ITypeInfo **typeinfo)
123 HRESULT hres;
125 if (!typelib)
126 hres = load_typelib();
127 if (!typelib)
128 return hres;
130 if(!typeinfos[tid]) {
131 ITypeInfo *ti;
133 hres = ITypeLib_GetTypeInfoOfGuid(typelib, tid_ids[tid], &ti);
134 if(FAILED(hres)) {
135 ERR("GetTypeInfoOfGuid(%s) failed: %08x\n", debugstr_guid(tid_ids[tid]), hres);
136 return hres;
139 if(InterlockedCompareExchangePointer((void**)(typeinfos+tid), ti, NULL))
140 ITypeInfo_Release(ti);
143 *typeinfo = typeinfos[tid];
144 return S_OK;
147 void release_typelib(void)
149 dispex_data_t *iter;
150 unsigned i;
152 while(!list_empty(&dispex_data_list)) {
153 iter = LIST_ENTRY(list_head(&dispex_data_list), dispex_data_t, entry);
154 list_remove(&iter->entry);
156 for(i=0; i < iter->func_cnt; i++)
157 SysFreeString(iter->funcs[i].name);
159 heap_free(iter->funcs);
160 heap_free(iter->name_table);
161 heap_free(iter);
164 if(!typelib)
165 return;
167 for(i=0; i < sizeof(typeinfos)/sizeof(*typeinfos); i++)
168 if(typeinfos[i])
169 ITypeInfo_Release(typeinfos[i]);
171 ITypeLib_Release(typelib);
172 DeleteCriticalSection(&cs_dispex_static_data);
175 HRESULT get_htmldoc_classinfo(ITypeInfo **typeinfo)
177 HRESULT hres;
179 if (!typelib)
180 hres = load_typelib();
181 if (!typelib)
182 return hres;
184 hres = ITypeLib_GetTypeInfoOfGuid(typelib, &CLSID_HTMLDocument, typeinfo);
185 if(FAILED(hres))
186 ERR("GetTypeInfoOfGuid failed: %08x\n", hres);
187 return hres;
190 static void add_func_info(dispex_data_t *data, DWORD *size, tid_t tid, const FUNCDESC *desc, ITypeInfo *dti)
192 func_info_t *info;
193 HRESULT hres;
195 if(data->func_cnt && data->funcs[data->func_cnt-1].id == desc->memid) {
196 info = data->funcs+data->func_cnt-1;
197 }else {
198 if(data->func_cnt == *size)
199 data->funcs = heap_realloc(data->funcs, (*size <<= 1)*sizeof(func_info_t));
201 info = data->funcs+data->func_cnt;
202 hres = ITypeInfo_GetDocumentation(dti, desc->memid, &info->name, NULL, NULL, NULL);
203 if(FAILED(hres))
204 return;
206 data->func_cnt++;
208 info->id = desc->memid;
209 info->tid = tid;
210 info->func_disp_idx = -1;
211 info->prop_vt = VT_EMPTY;
212 info->put_vtbl_off = 0;
213 info->get_vtbl_off = 0;
216 if(desc->invkind & DISPATCH_METHOD) {
217 info->func_disp_idx = data->func_disp_cnt++;
218 }else if(desc->invkind & (DISPATCH_PROPERTYPUT|DISPATCH_PROPERTYGET)) {
219 VARTYPE vt = VT_EMPTY;
221 if(desc->invkind & DISPATCH_PROPERTYGET) {
222 vt = desc->elemdescFunc.tdesc.vt;
223 info->get_vtbl_off = desc->oVft/sizeof(void*);
225 if(desc->invkind & DISPATCH_PROPERTYPUT) {
226 assert(desc->cParams == 1);
227 vt = desc->lprgelemdescParam->tdesc.vt;
228 info->put_vtbl_off = desc->oVft/sizeof(void*);
231 assert(info->prop_vt == VT_EMPTY || vt == info->prop_vt);
232 info->prop_vt = vt;
236 static int dispid_cmp(const void *p1, const void *p2)
238 return ((const func_info_t*)p1)->id - ((const func_info_t*)p2)->id;
241 static int func_name_cmp(const void *p1, const void *p2)
243 return strcmpiW((*(func_info_t* const*)p1)->name, (*(func_info_t* const*)p2)->name);
246 static dispex_data_t *preprocess_dispex_data(DispatchEx *This)
248 const tid_t *tid = This->data->iface_tids;
249 FUNCDESC *funcdesc;
250 dispex_data_t *data;
251 DWORD size = 16, i;
252 ITypeInfo *ti, *dti;
253 HRESULT hres;
255 TRACE("(%p)\n", This);
257 if(This->data->disp_tid) {
258 hres = get_typeinfo(This->data->disp_tid, &dti);
259 if(FAILED(hres)) {
260 ERR("Could not get disp type info: %08x\n", hres);
261 return NULL;
265 data = heap_alloc(sizeof(dispex_data_t));
266 data->func_cnt = 0;
267 data->func_disp_cnt = 0;
268 data->funcs = heap_alloc(size*sizeof(func_info_t));
269 list_add_tail(&dispex_data_list, &data->entry);
271 while(*tid) {
272 hres = get_typeinfo(*tid, &ti);
273 if(FAILED(hres))
274 break;
276 i=7;
277 while(1) {
278 hres = ITypeInfo_GetFuncDesc(ti, i++, &funcdesc);
279 if(FAILED(hres))
280 break;
282 add_func_info(data, &size, *tid, funcdesc, dti);
283 ITypeInfo_ReleaseFuncDesc(ti, funcdesc);
286 tid++;
289 if(!data->func_cnt) {
290 heap_free(data->funcs);
291 data->name_table = NULL;
292 data->funcs = NULL;
293 return data;
297 data->funcs = heap_realloc(data->funcs, data->func_cnt * sizeof(func_info_t));
298 qsort(data->funcs, data->func_cnt, sizeof(func_info_t), dispid_cmp);
300 data->name_table = heap_alloc(data->func_cnt * sizeof(func_info_t*));
301 for(i=0; i < data->func_cnt; i++)
302 data->name_table[i] = data->funcs+i;
303 qsort(data->name_table, data->func_cnt, sizeof(func_info_t*), func_name_cmp);
305 return data;
308 static int id_cmp(const void *p1, const void *p2)
310 return *(const DISPID*)p1 - *(const DISPID*)p2;
313 HRESULT get_dispids(tid_t tid, DWORD *ret_size, DISPID **ret)
315 unsigned i, func_cnt;
316 FUNCDESC *funcdesc;
317 ITypeInfo *ti;
318 TYPEATTR *attr;
319 DISPID *ids;
320 HRESULT hres;
322 hres = get_typeinfo(tid, &ti);
323 if(FAILED(hres))
324 return hres;
326 hres = ITypeInfo_GetTypeAttr(ti, &attr);
327 if(FAILED(hres)) {
328 ITypeInfo_Release(ti);
329 return hres;
332 func_cnt = attr->cFuncs;
333 ITypeInfo_ReleaseTypeAttr(ti, attr);
335 ids = heap_alloc(func_cnt*sizeof(DISPID));
336 if(!ids) {
337 ITypeInfo_Release(ti);
338 return E_OUTOFMEMORY;
341 for(i=0; i < func_cnt; i++) {
342 hres = ITypeInfo_GetFuncDesc(ti, i, &funcdesc);
343 if(FAILED(hres))
344 break;
346 ids[i] = funcdesc->memid;
347 ITypeInfo_ReleaseFuncDesc(ti, funcdesc);
350 ITypeInfo_Release(ti);
351 if(FAILED(hres)) {
352 heap_free(ids);
353 return hres;
356 qsort(ids, func_cnt, sizeof(DISPID), id_cmp);
358 *ret_size = func_cnt;
359 *ret = ids;
360 return S_OK;
363 static dispex_data_t *get_dispex_data(DispatchEx *This)
365 if(This->data->data)
366 return This->data->data;
368 EnterCriticalSection(&cs_dispex_static_data);
370 if(!This->data->data)
371 This->data->data = preprocess_dispex_data(This);
373 LeaveCriticalSection(&cs_dispex_static_data);
375 return This->data->data;
378 static inline BOOL is_custom_dispid(DISPID id)
380 return MSHTML_DISPID_CUSTOM_MIN <= id && id <= MSHTML_DISPID_CUSTOM_MAX;
383 static inline BOOL is_dynamic_dispid(DISPID id)
385 return DISPID_DYNPROP_0 <= id && id <= DISPID_DYNPROP_MAX;
388 static HRESULT variant_copy(VARIANT *dest, VARIANT *src)
390 if(V_VT(src) == VT_BSTR && !V_BSTR(src)) {
391 V_VT(dest) = VT_BSTR;
392 V_BSTR(dest) = NULL;
393 return S_OK;
396 return VariantCopy(dest, src);
399 static inline dispex_dynamic_data_t *get_dynamic_data(DispatchEx *This)
401 if(This->dynamic_data)
402 return This->dynamic_data;
404 This->dynamic_data = heap_alloc_zero(sizeof(dispex_dynamic_data_t));
405 if(!This->dynamic_data)
406 return NULL;
408 if(This->data->vtbl && This->data->vtbl->populate_props)
409 This->data->vtbl->populate_props(This);
411 return This->dynamic_data;
414 static HRESULT get_dynamic_prop(DispatchEx *This, const WCHAR *name, DWORD flags, dynamic_prop_t **ret)
416 const BOOL alloc = flags & fdexNameEnsure;
417 dispex_dynamic_data_t *data;
418 dynamic_prop_t *prop;
420 data = get_dynamic_data(This);
421 if(!data)
422 return E_OUTOFMEMORY;
424 for(prop = data->props; prop < data->props+data->prop_cnt; prop++) {
425 if(flags & fdexNameCaseInsensitive ? !strcmpiW(prop->name, name) : !strcmpW(prop->name, name)) {
426 if(prop->flags & DYNPROP_DELETED) {
427 if(!alloc)
428 return DISP_E_UNKNOWNNAME;
429 prop->flags &= ~DYNPROP_DELETED;
431 *ret = prop;
432 return S_OK;
436 if(!alloc)
437 return DISP_E_UNKNOWNNAME;
439 TRACE("creating dynamic prop %s\n", debugstr_w(name));
441 if(!data->buf_size) {
442 data->props = heap_alloc(sizeof(dynamic_prop_t)*4);
443 if(!data->props)
444 return E_OUTOFMEMORY;
445 data->buf_size = 4;
446 }else if(data->buf_size == data->prop_cnt) {
447 dynamic_prop_t *new_props;
449 new_props = heap_realloc(data->props, sizeof(dynamic_prop_t)*(data->buf_size<<1));
450 if(!new_props)
451 return E_OUTOFMEMORY;
453 data->props = new_props;
454 data->buf_size <<= 1;
457 prop = data->props + data->prop_cnt;
459 prop->name = heap_strdupW(name);
460 if(!prop->name)
461 return E_OUTOFMEMORY;
463 VariantInit(&prop->var);
464 prop->flags = 0;
465 data->prop_cnt++;
466 *ret = prop;
467 return S_OK;
470 HRESULT dispex_get_dprop_ref(DispatchEx *This, const WCHAR *name, BOOL alloc, VARIANT **ret)
472 dynamic_prop_t *prop;
473 HRESULT hres;
475 hres = get_dynamic_prop(This, name, alloc ? fdexNameEnsure : 0, &prop);
476 if(FAILED(hres))
477 return hres;
479 *ret = &prop->var;
480 return S_OK;
483 HRESULT dispex_get_dynid(DispatchEx *This, const WCHAR *name, DISPID *id)
485 dynamic_prop_t *prop;
486 HRESULT hres;
488 hres = get_dynamic_prop(This, name, fdexNameEnsure, &prop);
489 if(FAILED(hres))
490 return hres;
492 *id = DISPID_DYNPROP_0 + (prop - This->dynamic_data->props);
493 return S_OK;
496 static HRESULT dispex_value(DispatchEx *This, LCID lcid, WORD flags, DISPPARAMS *params,
497 VARIANT *res, EXCEPINFO *ei, IServiceProvider *caller)
499 if(This->data->vtbl && This->data->vtbl->value)
500 return This->data->vtbl->value(This, lcid, flags, params, res, ei, caller);
502 switch(flags) {
503 case DISPATCH_PROPERTYGET:
504 V_VT(res) = VT_BSTR;
505 V_BSTR(res) = SysAllocString(objectW);
506 if(!V_BSTR(res))
507 return E_OUTOFMEMORY;
508 break;
509 default:
510 FIXME("Unimplemented flags %x\n", flags);
511 return E_NOTIMPL;
514 return S_OK;
517 static HRESULT typeinfo_invoke(DispatchEx *This, func_info_t *func, WORD flags, DISPPARAMS *dp, VARIANT *res,
518 EXCEPINFO *ei)
520 ITypeInfo *ti;
521 IUnknown *unk;
522 UINT argerr=0;
523 HRESULT hres;
525 hres = get_typeinfo(func->tid, &ti);
526 if(FAILED(hres)) {
527 ERR("Could not get type info: %08x\n", hres);
528 return hres;
531 hres = IUnknown_QueryInterface(This->outer, tid_ids[func->tid], (void**)&unk);
532 if(FAILED(hres)) {
533 ERR("Could not get iface %s: %08x\n", debugstr_guid(tid_ids[func->tid]), hres);
534 return E_FAIL;
537 hres = ITypeInfo_Invoke(ti, unk, func->id, flags, dp, res, ei, &argerr);
539 IUnknown_Release(unk);
540 return hres;
543 static inline func_disp_t *impl_from_IUnknown(IUnknown *iface)
545 return CONTAINING_RECORD(iface, func_disp_t, IUnknown_iface);
548 static HRESULT WINAPI Function_QueryInterface(IUnknown *iface, REFIID riid, void **ppv)
550 func_disp_t *This = impl_from_IUnknown(iface);
552 if(IsEqualGUID(&IID_IUnknown, riid)) {
553 TRACE("(%p)->(IID_IUnknown %p)\n", This, ppv);
554 *ppv = &This->IUnknown_iface;
555 }else if(dispex_query_interface(&This->dispex, riid, ppv)) {
556 return *ppv ? S_OK : E_NOINTERFACE;
557 }else {
558 *ppv = NULL;
559 return E_NOINTERFACE;
562 IUnknown_AddRef((IUnknown*)*ppv);
563 return S_OK;
566 static ULONG WINAPI Function_AddRef(IUnknown *iface)
568 func_disp_t *This = impl_from_IUnknown(iface);
570 TRACE("(%p)\n", This);
572 return IDispatchEx_AddRef(&This->obj->IDispatchEx_iface);
575 static ULONG WINAPI Function_Release(IUnknown *iface)
577 func_disp_t *This = impl_from_IUnknown(iface);
579 TRACE("(%p)\n", This);
581 return IDispatchEx_Release(&This->obj->IDispatchEx_iface);
584 static const IUnknownVtbl FunctionUnkVtbl = {
585 Function_QueryInterface,
586 Function_AddRef,
587 Function_Release
590 static inline func_disp_t *impl_from_DispatchEx(DispatchEx *iface)
592 return CONTAINING_RECORD(iface, func_disp_t, dispex);
595 static HRESULT function_value(DispatchEx *dispex, LCID lcid, WORD flags, DISPPARAMS *params,
596 VARIANT *res, EXCEPINFO *ei, IServiceProvider *caller)
598 func_disp_t *This = impl_from_DispatchEx(dispex);
599 HRESULT hres;
601 switch(flags) {
602 case DISPATCH_METHOD|DISPATCH_PROPERTYGET:
603 if(!res)
604 return E_INVALIDARG;
605 /* fall through */
606 case DISPATCH_METHOD:
607 hres = typeinfo_invoke(This->obj, This->info, flags, params, res, ei);
608 break;
609 default:
610 FIXME("Unimplemented flags %x\n", flags);
611 hres = E_NOTIMPL;
614 return hres;
617 static const dispex_static_data_vtbl_t function_dispex_vtbl = {
618 function_value,
619 NULL,
620 NULL,
621 NULL
624 static const tid_t function_iface_tids[] = {0};
626 static dispex_static_data_t function_dispex = {
627 &function_dispex_vtbl,
628 NULL_tid,
629 NULL,
630 function_iface_tids
633 static func_disp_t *create_func_disp(DispatchEx *obj, func_info_t *info)
635 func_disp_t *ret;
637 ret = heap_alloc_zero(sizeof(func_disp_t));
638 if(!ret)
639 return NULL;
641 ret->IUnknown_iface.lpVtbl = &FunctionUnkVtbl;
642 init_dispex(&ret->dispex, &ret->IUnknown_iface, &function_dispex);
643 ret->obj = obj;
644 ret->info = info;
646 return ret;
649 static HRESULT invoke_disp_value(DispatchEx *This, IDispatch *func_disp, LCID lcid, WORD flags, DISPPARAMS *dp,
650 VARIANT *res, EXCEPINFO *ei, IServiceProvider *caller)
652 DISPID named_arg = DISPID_THIS;
653 DISPPARAMS new_dp = {NULL, &named_arg, 0, 1};
654 IDispatchEx *dispex;
655 HRESULT hres;
657 if(dp->cNamedArgs) {
658 FIXME("named args not supported\n");
659 return E_NOTIMPL;
662 new_dp.rgvarg = heap_alloc((dp->cArgs+1)*sizeof(VARIANTARG));
663 if(!new_dp.rgvarg)
664 return E_OUTOFMEMORY;
666 new_dp.cArgs = dp->cArgs+1;
667 memcpy(new_dp.rgvarg+1, dp->rgvarg, dp->cArgs*sizeof(VARIANTARG));
669 V_VT(new_dp.rgvarg) = VT_DISPATCH;
670 V_DISPATCH(new_dp.rgvarg) = (IDispatch*)&This->IDispatchEx_iface;
672 hres = IDispatch_QueryInterface(func_disp, &IID_IDispatchEx, (void**)&dispex);
673 TRACE(">>>\n");
674 if(SUCCEEDED(hres)) {
675 hres = IDispatchEx_InvokeEx(dispex, DISPID_VALUE, lcid, flags, &new_dp, res, ei, caller);
676 IDispatchEx_Release(dispex);
677 }else {
678 ULONG err = 0;
679 hres = IDispatch_Invoke(func_disp, DISPID_VALUE, &IID_NULL, lcid, flags, &new_dp, res, ei, &err);
681 if(SUCCEEDED(hres))
682 TRACE("<<< %s\n", debugstr_variant(res));
683 else
684 WARN("<<< %08x\n", hres);
686 heap_free(new_dp.rgvarg);
687 return hres;
690 static HRESULT function_invoke(DispatchEx *This, func_info_t *func, WORD flags, DISPPARAMS *dp, VARIANT *res,
691 EXCEPINFO *ei)
693 HRESULT hres;
695 switch(flags) {
696 case DISPATCH_METHOD|DISPATCH_PROPERTYGET:
697 if(!res)
698 return E_INVALIDARG;
699 /* fall through */
700 case DISPATCH_METHOD:
701 hres = typeinfo_invoke(This, func, flags, dp, res, ei);
702 break;
703 case DISPATCH_PROPERTYGET: {
704 dispex_dynamic_data_t *dynamic_data;
706 if(func->id == DISPID_VALUE) {
707 BSTR ret;
709 ret = SysAllocString(objectW);
710 if(!ret)
711 return E_OUTOFMEMORY;
713 V_VT(res) = VT_BSTR;
714 V_BSTR(res) = ret;
715 return S_OK;
718 dynamic_data = get_dynamic_data(This);
719 if(!dynamic_data)
720 return E_OUTOFMEMORY;
722 if(!dynamic_data->func_disps) {
723 dynamic_data->func_disps = heap_alloc_zero(This->data->data->func_disp_cnt * sizeof(func_disp_t*));
724 if(!dynamic_data->func_disps)
725 return E_OUTOFMEMORY;
728 if(!dynamic_data->func_disps[func->func_disp_idx]) {
729 dynamic_data->func_disps[func->func_disp_idx] = create_func_disp(This, func);
730 if(!dynamic_data->func_disps[func->func_disp_idx])
731 return E_OUTOFMEMORY;
734 V_VT(res) = VT_DISPATCH;
735 V_DISPATCH(res) = (IDispatch*)&dynamic_data->func_disps[func->func_disp_idx]->dispex.IDispatchEx_iface;
736 IDispatch_AddRef(V_DISPATCH(res));
737 hres = S_OK;
738 break;
740 default:
741 FIXME("Unimplemented flags %x\n", flags);
742 /* fall through */
743 case DISPATCH_PROPERTYPUT:
744 hres = E_NOTIMPL;
747 return hres;
750 static HRESULT get_builtin_func(dispex_data_t *data, DISPID id, func_info_t **ret)
752 int min, max, n;
754 min = 0;
755 max = data->func_cnt-1;
757 while(min <= max) {
758 n = (min+max)/2;
760 if(data->funcs[n].id == id) {
761 *ret = data->funcs+n;
762 return S_OK;
765 if(data->funcs[n].id < id)
766 min = n+1;
767 else
768 max = n-1;
771 WARN("invalid id %x\n", id);
772 return DISP_E_UNKNOWNNAME;
775 static HRESULT get_builtin_id(DispatchEx *This, BSTR name, DWORD grfdex, DISPID *ret)
777 dispex_data_t *data;
778 int min, max, n, c;
780 data = get_dispex_data(This);
781 if(!data)
782 return E_FAIL;
784 min = 0;
785 max = data->func_cnt-1;
787 while(min <= max) {
788 n = (min+max)/2;
790 c = strcmpiW(data->name_table[n]->name, name);
791 if(!c) {
792 if((grfdex & fdexNameCaseSensitive) && strcmpW(data->name_table[n]->name, name))
793 break;
795 *ret = data->name_table[n]->id;
796 return S_OK;
799 if(c > 0)
800 max = n-1;
801 else
802 min = n+1;
805 if(This->data->vtbl && This->data->vtbl->get_dispid) {
806 HRESULT hres;
808 hres = This->data->vtbl->get_dispid(This, name, grfdex, ret);
809 if(hres != DISP_E_UNKNOWNNAME)
810 return hres;
813 return DISP_E_UNKNOWNNAME;
816 /* List all types used by IDispatchEx-based properties */
817 #define BUILTIN_TYPES_SWITCH \
818 CASE_VT(VT_I2, INT16, V_I2); \
819 CASE_VT(VT_I4, INT32, V_I4); \
820 CASE_VT(VT_R4, float, V_R4); \
821 CASE_VT(VT_BSTR, BSTR, V_BSTR); \
822 CASE_VT(VT_BOOL, VARIANT_BOOL, V_BOOL); \
823 CASE_VT(VT_VARIANT, VARIANT, *); \
824 CASE_VT(VT_PTR, void*, V_BYREF); \
825 CASE_VT(VT_UNKNOWN, IUnknown*, V_UNKNOWN); \
826 CASE_VT(VT_DISPATCH, IDispatch*, V_DISPATCH)
828 static HRESULT change_type(VARIANT *dst, VARIANT *src, VARTYPE vt, IServiceProvider *caller)
830 V_VT(dst) = VT_EMPTY;
832 if(caller) {
833 IVariantChangeType *change_type = NULL;
834 HRESULT hres;
836 hres = IServiceProvider_QueryService(caller, &SID_VariantConversion, &IID_IVariantChangeType, (void**)&change_type);
837 if(SUCCEEDED(hres)) {
838 hres = IVariantChangeType_ChangeType(change_type, dst, src, LOCALE_NEUTRAL, vt);
839 IVariantChangeType_Release(change_type);
840 return hres;
844 return VariantChangeType(dst, src, 0, vt);
847 static HRESULT builtin_propget(DispatchEx *This, func_info_t *func, DISPPARAMS *dp, VARIANT *res)
849 IUnknown *iface;
850 HRESULT hres;
852 if(dp && dp->cArgs) {
853 FIXME("cArgs %d\n", dp->cArgs);
854 return E_NOTIMPL;
857 assert(func->get_vtbl_off);
859 hres = IUnknown_QueryInterface(This->outer, tid_ids[func->tid], (void**)&iface);
860 if(SUCCEEDED(hres)) {
861 switch(func->prop_vt) {
862 #define CASE_VT(vt,type,access) \
863 case vt: { \
864 type val; \
865 hres = ((HRESULT (WINAPI*)(IUnknown*,type*))((void**)iface->lpVtbl)[func->get_vtbl_off])(iface,&val); \
866 if(SUCCEEDED(hres)) \
867 access(res) = val; \
869 break
870 BUILTIN_TYPES_SWITCH;
871 #undef CASE_VT
872 default:
873 FIXME("Unhandled vt %d\n", func->prop_vt);
874 hres = E_NOTIMPL;
876 IUnknown_Release(iface);
879 if(FAILED(hres))
880 return hres;
882 if(func->prop_vt != VT_VARIANT)
883 V_VT(res) = func->prop_vt == VT_PTR ? VT_DISPATCH : func->prop_vt;
884 return S_OK;
887 static HRESULT builtin_propput(DispatchEx *This, func_info_t *func, DISPPARAMS *dp, IServiceProvider *caller)
889 VARIANT *v, tmpv;
890 IUnknown *iface;
891 HRESULT hres;
893 if(dp->cArgs != 1 || (dp->cNamedArgs == 1 && *dp->rgdispidNamedArgs != DISPID_PROPERTYPUT)
894 || dp->cNamedArgs > 1) {
895 FIXME("invalid args\n");
896 return E_INVALIDARG;
899 if(!func->put_vtbl_off) {
900 FIXME("No setter\n");
901 return E_FAIL;
904 v = dp->rgvarg;
905 if(func->prop_vt != VT_VARIANT && V_VT(v) != func->prop_vt) {
906 hres = change_type(&tmpv, v, func->prop_vt, caller);
907 if(FAILED(hres))
908 return hres;
909 v = &tmpv;
912 hres = IUnknown_QueryInterface(This->outer, tid_ids[func->tid], (void**)&iface);
913 if(SUCCEEDED(hres)) {
914 switch(func->prop_vt) {
915 #define CASE_VT(vt,type,access) \
916 case vt: \
917 hres = ((HRESULT (WINAPI*)(IUnknown*,type))((void**)iface->lpVtbl)[func->put_vtbl_off])(iface,access(v)); \
918 break
919 BUILTIN_TYPES_SWITCH;
920 #undef CASE_VT
921 default:
922 FIXME("Unimplemented vt %d\n", func->prop_vt);
923 hres = E_NOTIMPL;
926 IUnknown_Release(iface);
929 if(v == &tmpv)
930 VariantClear(v);
931 return hres;
934 static HRESULT invoke_builtin_prop(DispatchEx *This, DISPID id, LCID lcid, WORD flags, DISPPARAMS *dp,
935 VARIANT *res, EXCEPINFO *ei, IServiceProvider *caller)
937 dispex_data_t *data;
938 func_info_t *func;
939 HRESULT hres;
941 data = get_dispex_data(This);
942 if(!data)
943 return E_FAIL;
945 hres = get_builtin_func(data, id, &func);
946 if(id == DISPID_VALUE && hres == DISP_E_UNKNOWNNAME)
947 return dispex_value(This, lcid, flags, dp, res, ei, caller);
948 if(FAILED(hres))
949 return hres;
951 if(func->func_disp_idx != -1)
952 return function_invoke(This, func, flags, dp, res, ei);
954 switch(flags) {
955 case DISPATCH_PROPERTYPUT:
956 if(res)
957 V_VT(res) = VT_EMPTY;
958 hres = builtin_propput(This, func, dp, caller);
959 break;
960 case DISPATCH_PROPERTYGET:
961 hres = builtin_propget(This, func, dp, res);
962 break;
963 default:
964 if(!func->get_vtbl_off) {
965 hres = typeinfo_invoke(This, func, flags, dp, res, ei);
966 }else {
967 VARIANT v;
969 hres = builtin_propget(This, func, NULL, &v);
970 if(FAILED(hres))
971 return hres;
973 if(V_VT(&v) != VT_DISPATCH) {
974 FIXME("Not a function %s\n", debugstr_variant(&v));
975 VariantClear(&v);
976 return E_FAIL;
979 hres = invoke_disp_value(This, V_DISPATCH(&v), lcid, flags, dp, res, ei, caller);
981 IDispatch_Release(V_DISPATCH(&v));
985 return hres;
988 HRESULT remove_prop(DispatchEx *This, BSTR name, VARIANT_BOOL *success)
990 dynamic_prop_t *prop;
991 DISPID id;
992 HRESULT hres;
994 hres = get_builtin_id(This, name, 0, &id);
995 if(hres == S_OK) {
996 DISPID named_id = DISPID_PROPERTYPUT;
997 VARIANT var;
998 DISPPARAMS dp = {&var,&named_id,1,1};
999 EXCEPINFO ei;
1001 V_VT(&var) = VT_EMPTY;
1002 memset(&ei, 0, sizeof(ei));
1003 hres = invoke_builtin_prop(This, id, 0, DISPATCH_PROPERTYPUT, &dp, NULL, &ei, NULL);
1004 if(FAILED(hres))
1005 return hres;
1007 *success = VARIANT_TRUE;
1008 return S_OK;
1011 hres = get_dynamic_prop(This, name, 0, &prop);
1012 if(FAILED(hres)) {
1013 if(hres != DISP_E_UNKNOWNNAME)
1014 return hres;
1015 *success = VARIANT_FALSE;
1016 return S_OK;
1019 VariantClear(&prop->var);
1020 prop->flags |= DYNPROP_DELETED;
1021 *success = VARIANT_TRUE;
1022 return S_OK;
1025 static inline DispatchEx *impl_from_IDispatchEx(IDispatchEx *iface)
1027 return CONTAINING_RECORD(iface, DispatchEx, IDispatchEx_iface);
1030 static HRESULT WINAPI DispatchEx_QueryInterface(IDispatchEx *iface, REFIID riid, void **ppv)
1032 DispatchEx *This = impl_from_IDispatchEx(iface);
1034 return IUnknown_QueryInterface(This->outer, riid, ppv);
1037 static ULONG WINAPI DispatchEx_AddRef(IDispatchEx *iface)
1039 DispatchEx *This = impl_from_IDispatchEx(iface);
1041 return IUnknown_AddRef(This->outer);
1044 static ULONG WINAPI DispatchEx_Release(IDispatchEx *iface)
1046 DispatchEx *This = impl_from_IDispatchEx(iface);
1048 return IUnknown_Release(This->outer);
1051 static HRESULT WINAPI DispatchEx_GetTypeInfoCount(IDispatchEx *iface, UINT *pctinfo)
1053 DispatchEx *This = impl_from_IDispatchEx(iface);
1055 TRACE("(%p)->(%p)\n", This, pctinfo);
1057 *pctinfo = 1;
1058 return S_OK;
1061 static HRESULT WINAPI DispatchEx_GetTypeInfo(IDispatchEx *iface, UINT iTInfo,
1062 LCID lcid, ITypeInfo **ppTInfo)
1064 DispatchEx *This = impl_from_IDispatchEx(iface);
1065 HRESULT hres;
1067 TRACE("(%p)->(%u %u %p)\n", This, iTInfo, lcid, ppTInfo);
1069 hres = get_typeinfo(This->data->disp_tid, ppTInfo);
1070 if(FAILED(hres))
1071 return hres;
1073 ITypeInfo_AddRef(*ppTInfo);
1074 return S_OK;
1077 static HRESULT WINAPI DispatchEx_GetIDsOfNames(IDispatchEx *iface, REFIID riid,
1078 LPOLESTR *rgszNames, UINT cNames,
1079 LCID lcid, DISPID *rgDispId)
1081 DispatchEx *This = impl_from_IDispatchEx(iface);
1082 UINT i;
1083 HRESULT hres;
1085 TRACE("(%p)->(%s %p %u %u %p)\n", This, debugstr_guid(riid), rgszNames, cNames,
1086 lcid, rgDispId);
1088 for(i=0; i < cNames; i++) {
1089 hres = IDispatchEx_GetDispID(&This->IDispatchEx_iface, rgszNames[i], 0, rgDispId+i);
1090 if(FAILED(hres))
1091 return hres;
1094 return S_OK;
1097 static HRESULT WINAPI DispatchEx_Invoke(IDispatchEx *iface, DISPID dispIdMember,
1098 REFIID riid, LCID lcid, WORD wFlags, DISPPARAMS *pDispParams,
1099 VARIANT *pVarResult, EXCEPINFO *pExcepInfo, UINT *puArgErr)
1101 DispatchEx *This = impl_from_IDispatchEx(iface);
1103 TRACE("(%p)->(%d %s %d %d %p %p %p %p)\n", This, dispIdMember, debugstr_guid(riid),
1104 lcid, wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr);
1106 return IDispatchEx_InvokeEx(&This->IDispatchEx_iface, dispIdMember, lcid, wFlags, pDispParams,
1107 pVarResult, pExcepInfo, NULL);
1110 static HRESULT WINAPI DispatchEx_GetDispID(IDispatchEx *iface, BSTR bstrName, DWORD grfdex, DISPID *pid)
1112 DispatchEx *This = impl_from_IDispatchEx(iface);
1113 dynamic_prop_t *dprop;
1114 HRESULT hres;
1116 TRACE("(%p)->(%s %x %p)\n", This, debugstr_w(bstrName), grfdex, pid);
1118 if(grfdex & ~(fdexNameCaseSensitive|fdexNameCaseInsensitive|fdexNameEnsure|fdexNameImplicit|FDEX_VERSION_MASK))
1119 FIXME("Unsupported grfdex %x\n", grfdex);
1121 hres = get_builtin_id(This, bstrName, grfdex, pid);
1122 if(hres != DISP_E_UNKNOWNNAME)
1123 return hres;
1125 hres = get_dynamic_prop(This, bstrName, grfdex, &dprop);
1126 if(FAILED(hres))
1127 return hres;
1129 *pid = DISPID_DYNPROP_0 + (dprop - This->dynamic_data->props);
1130 return S_OK;
1133 static HRESULT WINAPI DispatchEx_InvokeEx(IDispatchEx *iface, DISPID id, LCID lcid, WORD wFlags, DISPPARAMS *pdp,
1134 VARIANT *pvarRes, EXCEPINFO *pei, IServiceProvider *pspCaller)
1136 DispatchEx *This = impl_from_IDispatchEx(iface);
1137 HRESULT hres;
1139 TRACE("(%p)->(%x %x %x %p %p %p %p)\n", This, id, lcid, wFlags, pdp, pvarRes, pei, pspCaller);
1141 if(is_custom_dispid(id) && This->data->vtbl && This->data->vtbl->invoke)
1142 return This->data->vtbl->invoke(This, id, lcid, wFlags, pdp, pvarRes, pei, pspCaller);
1144 if(wFlags == DISPATCH_CONSTRUCT) {
1145 if(id == DISPID_VALUE) {
1146 if(This->data->vtbl && This->data->vtbl->value) {
1147 return This->data->vtbl->value(This, lcid, wFlags, pdp, pvarRes, pei, pspCaller);
1149 FIXME("DISPATCH_CONSTRUCT flag but missing value function\n");
1150 return E_FAIL;
1152 FIXME("DISPATCH_CONSTRUCT flag without DISPID_VALUE\n");
1153 return E_FAIL;
1156 if(is_dynamic_dispid(id)) {
1157 DWORD idx = id - DISPID_DYNPROP_0;
1158 dynamic_prop_t *prop;
1160 if(!get_dynamic_data(This) || This->dynamic_data->prop_cnt <= idx)
1161 return DISP_E_UNKNOWNNAME;
1163 prop = This->dynamic_data->props+idx;
1165 switch(wFlags) {
1166 case DISPATCH_METHOD|DISPATCH_PROPERTYGET:
1167 if(!pvarRes)
1168 return E_INVALIDARG;
1169 /* fall through */
1170 case DISPATCH_METHOD:
1171 if(V_VT(&prop->var) != VT_DISPATCH) {
1172 FIXME("invoke %s\n", debugstr_variant(&prop->var));
1173 return E_NOTIMPL;
1176 return invoke_disp_value(This, V_DISPATCH(&prop->var), lcid, wFlags, pdp, pvarRes, pei, pspCaller);
1177 case DISPATCH_PROPERTYGET:
1178 if(prop->flags & DYNPROP_DELETED)
1179 return DISP_E_UNKNOWNNAME;
1180 V_VT(pvarRes) = VT_EMPTY;
1181 return variant_copy(pvarRes, &prop->var);
1182 case DISPATCH_PROPERTYPUT|DISPATCH_PROPERTYPUTREF:
1183 case DISPATCH_PROPERTYPUT:
1184 if(pdp->cArgs != 1 || (pdp->cNamedArgs == 1 && *pdp->rgdispidNamedArgs != DISPID_PROPERTYPUT)
1185 || pdp->cNamedArgs > 1) {
1186 FIXME("invalid args\n");
1187 return E_INVALIDARG;
1190 TRACE("put %s\n", debugstr_variant(pdp->rgvarg));
1191 VariantClear(&prop->var);
1192 hres = variant_copy(&prop->var, pdp->rgvarg);
1193 if(FAILED(hres))
1194 return hres;
1196 prop->flags &= ~DYNPROP_DELETED;
1197 return S_OK;
1198 default:
1199 FIXME("unhandled wFlags %x\n", wFlags);
1200 return E_NOTIMPL;
1204 return invoke_builtin_prop(This, id, lcid, wFlags, pdp, pvarRes, pei, pspCaller);
1207 static HRESULT WINAPI DispatchEx_DeleteMemberByName(IDispatchEx *iface, BSTR bstrName, DWORD grfdex)
1209 DispatchEx *This = impl_from_IDispatchEx(iface);
1211 TRACE("(%p)->(%s %x)\n", This, debugstr_w(bstrName), grfdex);
1213 /* Not implemented by IE */
1214 return E_NOTIMPL;
1217 static HRESULT WINAPI DispatchEx_DeleteMemberByDispID(IDispatchEx *iface, DISPID id)
1219 DispatchEx *This = impl_from_IDispatchEx(iface);
1221 TRACE("(%p)->(%x)\n", This, id);
1223 /* Not implemented by IE */
1224 return E_NOTIMPL;
1227 static HRESULT WINAPI DispatchEx_GetMemberProperties(IDispatchEx *iface, DISPID id, DWORD grfdexFetch, DWORD *pgrfdex)
1229 DispatchEx *This = impl_from_IDispatchEx(iface);
1230 FIXME("(%p)->(%x %x %p)\n", This, id, grfdexFetch, pgrfdex);
1231 return E_NOTIMPL;
1234 static HRESULT WINAPI DispatchEx_GetMemberName(IDispatchEx *iface, DISPID id, BSTR *pbstrName)
1236 DispatchEx *This = impl_from_IDispatchEx(iface);
1237 dispex_data_t *data;
1238 func_info_t *func;
1239 HRESULT hres;
1241 TRACE("(%p)->(%x %p)\n", This, id, pbstrName);
1243 if(is_dynamic_dispid(id)) {
1244 DWORD idx = id - DISPID_DYNPROP_0;
1246 if(!get_dynamic_data(This) || This->dynamic_data->prop_cnt <= idx)
1247 return DISP_E_UNKNOWNNAME;
1249 *pbstrName = SysAllocString(This->dynamic_data->props[idx].name);
1250 if(!*pbstrName)
1251 return E_OUTOFMEMORY;
1253 return S_OK;
1256 data = get_dispex_data(This);
1257 if(!data)
1258 return E_FAIL;
1260 hres = get_builtin_func(data, id, &func);
1261 if(FAILED(hres))
1262 return hres;
1264 *pbstrName = SysAllocString(func->name);
1265 if(!*pbstrName)
1266 return E_OUTOFMEMORY;
1267 return S_OK;
1270 static HRESULT WINAPI DispatchEx_GetNextDispID(IDispatchEx *iface, DWORD grfdex, DISPID id, DISPID *pid)
1272 DispatchEx *This = impl_from_IDispatchEx(iface);
1273 dispex_data_t *data;
1274 func_info_t *func;
1275 HRESULT hres;
1277 TRACE("(%p)->(%x %x %p)\n", This, grfdex, id, pid);
1279 if(is_dynamic_dispid(id)) {
1280 DWORD idx = id - DISPID_DYNPROP_0;
1282 if(!get_dynamic_data(This) || This->dynamic_data->prop_cnt <= idx)
1283 return DISP_E_UNKNOWNNAME;
1285 while(++idx < This->dynamic_data->prop_cnt && This->dynamic_data->props[idx].flags & DYNPROP_DELETED);
1287 if(idx == This->dynamic_data->prop_cnt) {
1288 *pid = DISPID_STARTENUM;
1289 return S_FALSE;
1292 *pid = DISPID_DYNPROP_0+idx;
1293 return S_OK;
1296 data = get_dispex_data(This);
1297 if(!data)
1298 return E_FAIL;
1300 if(id == DISPID_STARTENUM) {
1301 func = data->funcs;
1302 }else {
1303 hres = get_builtin_func(data, id, &func);
1304 if(FAILED(hres))
1305 return hres;
1306 func++;
1309 while(func < data->funcs+data->func_cnt) {
1310 /* FIXME: Skip hidden properties */
1311 if(func->func_disp_idx == -1) {
1312 *pid = func->id;
1313 return S_OK;
1315 func++;
1318 if(get_dynamic_data(This) && This->dynamic_data->prop_cnt) {
1319 *pid = DISPID_DYNPROP_0;
1320 return S_OK;
1323 *pid = DISPID_STARTENUM;
1324 return S_FALSE;
1327 static HRESULT WINAPI DispatchEx_GetNameSpaceParent(IDispatchEx *iface, IUnknown **ppunk)
1329 DispatchEx *This = impl_from_IDispatchEx(iface);
1330 FIXME("(%p)->(%p)\n", This, ppunk);
1331 return E_NOTIMPL;
1334 static IDispatchExVtbl DispatchExVtbl = {
1335 DispatchEx_QueryInterface,
1336 DispatchEx_AddRef,
1337 DispatchEx_Release,
1338 DispatchEx_GetTypeInfoCount,
1339 DispatchEx_GetTypeInfo,
1340 DispatchEx_GetIDsOfNames,
1341 DispatchEx_Invoke,
1342 DispatchEx_GetDispID,
1343 DispatchEx_InvokeEx,
1344 DispatchEx_DeleteMemberByName,
1345 DispatchEx_DeleteMemberByDispID,
1346 DispatchEx_GetMemberProperties,
1347 DispatchEx_GetMemberName,
1348 DispatchEx_GetNextDispID,
1349 DispatchEx_GetNameSpaceParent
1352 BOOL dispex_query_interface(DispatchEx *This, REFIID riid, void **ppv)
1354 static const IID IID_UndocumentedScriptIface =
1355 {0x719c3050,0xf9d3,0x11cf,{0xa4,0x93,0x00,0x40,0x05,0x23,0xa8,0xa0}};
1356 static const IID IID_IDispatchJS =
1357 {0x719c3050,0xf9d3,0x11cf,{0xa4,0x93,0x00,0x40,0x05,0x23,0xa8,0xa6}};
1359 if(IsEqualGUID(&IID_IDispatch, riid)) {
1360 TRACE("(%p)->(IID_IDispatch %p)\n", This, ppv);
1361 *ppv = &This->IDispatchEx_iface;
1362 }else if(IsEqualGUID(&IID_IDispatchEx, riid)) {
1363 TRACE("(%p)->(IID_IDispatchEx %p)\n", This, ppv);
1364 *ppv = &This->IDispatchEx_iface;
1365 }else if(IsEqualGUID(&IID_IDispatchJS, riid)) {
1366 TRACE("(%p)->(IID_IDispatchJS %p) returning NULL\n", This, ppv);
1367 *ppv = NULL;
1368 }else if(IsEqualGUID(&IID_UndocumentedScriptIface, riid)) {
1369 TRACE("(%p)->(IID_UndocumentedScriptIface %p) returning NULL\n", This, ppv);
1370 *ppv = NULL;
1371 }else {
1372 return FALSE;
1375 if(*ppv)
1376 IUnknown_AddRef((IUnknown*)*ppv);
1377 return TRUE;
1380 void release_dispex(DispatchEx *This)
1382 dynamic_prop_t *prop;
1384 if(!This->dynamic_data)
1385 return;
1387 for(prop = This->dynamic_data->props; prop < This->dynamic_data->props + This->dynamic_data->prop_cnt; prop++) {
1388 VariantClear(&prop->var);
1389 heap_free(prop->name);
1392 heap_free(This->dynamic_data->props);
1394 if(This->dynamic_data->func_disps) {
1395 unsigned i;
1397 for(i=0; i < This->data->data->func_disp_cnt; i++) {
1398 if(This->dynamic_data->func_disps[i]) {
1399 release_dispex(&This->dynamic_data->func_disps[i]->dispex);
1400 heap_free(This->dynamic_data->func_disps[i]);
1404 heap_free(This->dynamic_data->func_disps);
1407 heap_free(This->dynamic_data);
1410 void init_dispex(DispatchEx *dispex, IUnknown *outer, dispex_static_data_t *data)
1412 dispex->IDispatchEx_iface.lpVtbl = &DispatchExVtbl;
1413 dispex->outer = outer;
1414 dispex->data = data;
1415 dispex->dynamic_data = NULL;