mshtml: Don't use ITypeInfo for InvokeEx(DISPATCH_PROPERTYGET) implementation.
[wine] / dlls / mshtml / dispex.c
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
24 #include "windef.h"
25 #include "winbase.h"
26 #include "winuser.h"
27 #include "ole2.h"
28
29 #include "wine/debug.h"
30
31 #include "mshtml_private.h"
32
33 WINE_DEFAULT_DEBUG_CHANNEL(mshtml);
34
35 static CRITICAL_SECTION cs_dispex_static_data;
36 static CRITICAL_SECTION_DEBUG cs_dispex_static_data_dbg =
37 {
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") }
41 };
42 static CRITICAL_SECTION cs_dispex_static_data = { &cs_dispex_static_data_dbg, -1, 0, 0, 0, 0 };
43
44
45 static const WCHAR objectW[] = {'[','o','b','j','e','c','t',']',0};
46
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;
56
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;
62
63     struct list entry;
64 };
65
66 typedef struct {
67     VARIANT var;
68     LPWSTR name;
69     DWORD flags;
70 } dynamic_prop_t;
71
72 #define DYNPROP_DELETED    0x01
73
74 typedef struct {
75     DispatchEx dispex;
76     IUnknown IUnknown_iface;
77     DispatchEx *obj;
78     func_info_t *info;
79 } func_disp_t;
80
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;
86 };
87
88 #define DISPID_DYNPROP_0    0x50000000
89 #define DISPID_DYNPROP_MAX  0x5fffffff
90
91 #define FDEX_VERSION_MASK 0xf0000000
92
93 static ITypeLib *typelib;
94 static ITypeInfo *typeinfos[LAST_tid];
95 static struct list dispex_data_list = LIST_INIT(dispex_data_list);
96
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
103 };
104
105 static HRESULT load_typelib(void)
106 {
107     HRESULT hres;
108     ITypeLib *tl;
109
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;
114     }
115
116     if(InterlockedCompareExchangePointer((void**)&typelib, tl, NULL))
117         ITypeLib_Release(tl);
118     return hres;
119 }
120
121 static HRESULT get_typeinfo(tid_t tid, ITypeInfo **typeinfo)
122 {
123     HRESULT hres;
124
125     if (!typelib)
126         hres = load_typelib();
127     if (!typelib)
128         return hres;
129
130     if(!typeinfos[tid]) {
131         ITypeInfo *ti;
132
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;
137         }
138
139         if(InterlockedCompareExchangePointer((void**)(typeinfos+tid), ti, NULL))
140             ITypeInfo_Release(ti);
141     }
142
143     *typeinfo = typeinfos[tid];
144     return S_OK;
145 }
146
147 void release_typelib(void)
148 {
149     dispex_data_t *iter;
150     unsigned i;
151
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);
155
156         for(i=0; i < iter->func_cnt; i++)
157             SysFreeString(iter->funcs[i].name);
158
159         heap_free(iter->funcs);
160         heap_free(iter->name_table);
161         heap_free(iter);
162     }
163
164     if(!typelib)
165         return;
166
167     for(i=0; i < sizeof(typeinfos)/sizeof(*typeinfos); i++)
168         if(typeinfos[i])
169             ITypeInfo_Release(typeinfos[i]);
170
171     ITypeLib_Release(typelib);
172     DeleteCriticalSection(&cs_dispex_static_data);
173 }
174
175 HRESULT get_htmldoc_classinfo(ITypeInfo **typeinfo)
176 {
177     HRESULT hres;
178
179     if (!typelib)
180         hres = load_typelib();
181     if (!typelib)
182         return hres;
183
184     hres = ITypeLib_GetTypeInfoOfGuid(typelib, &CLSID_HTMLDocument, typeinfo);
185     if(FAILED(hres))
186         ERR("GetTypeInfoOfGuid failed: %08x\n", hres);
187     return hres;
188 }
189
190 static void add_func_info(dispex_data_t *data, DWORD *size, tid_t tid, const FUNCDESC *desc, ITypeInfo *dti)
191 {
192     func_info_t *info;
193     HRESULT hres;
194
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));
200
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;
205
206         data->func_cnt++;
207
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;
214     }
215
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;
220
221         if(desc->invkind & DISPATCH_PROPERTYGET) {
222             vt = desc->elemdescFunc.tdesc.vt;
223             info->get_vtbl_off = desc->oVft/sizeof(void*);
224         }
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*);
229         }
230
231         assert(info->prop_vt == VT_EMPTY || vt == info->prop_vt);
232         info->prop_vt = vt;
233     }
234 }
235
236 static int dispid_cmp(const void *p1, const void *p2)
237 {
238     return ((const func_info_t*)p1)->id - ((const func_info_t*)p2)->id;
239 }
240
241 static int func_name_cmp(const void *p1, const void *p2)
242 {
243     return strcmpiW((*(func_info_t* const*)p1)->name, (*(func_info_t* const*)p2)->name);
244 }
245
246 static dispex_data_t *preprocess_dispex_data(DispatchEx *This)
247 {
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;
254
255     TRACE("(%p)\n", This);
256
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;
262         }
263     }
264
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);
270
271     while(*tid) {
272         hres = get_typeinfo(*tid, &ti);
273         if(FAILED(hres))
274             break;
275
276         i=7;
277         while(1) {
278             hres = ITypeInfo_GetFuncDesc(ti, i++, &funcdesc);
279             if(FAILED(hres))
280                 break;
281
282             add_func_info(data, &size, *tid, funcdesc, dti);
283             ITypeInfo_ReleaseFuncDesc(ti, funcdesc);
284         }
285
286         tid++;
287     }
288
289     if(!data->func_cnt) {
290         heap_free(data->funcs);
291         data->name_table = NULL;
292         data->funcs = NULL;
293         return data;
294     }
295
296
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);
299
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);
304
305     return data;
306 }
307
308 static int id_cmp(const void *p1, const void *p2)
309 {
310     return *(const DISPID*)p1 - *(const DISPID*)p2;
311 }
312
313 HRESULT get_dispids(tid_t tid, DWORD *ret_size, DISPID **ret)
314 {
315     unsigned i, func_cnt;
316     FUNCDESC *funcdesc;
317     ITypeInfo *ti;
318     TYPEATTR *attr;
319     DISPID *ids;
320     HRESULT hres;
321
322     hres = get_typeinfo(tid, &ti);
323     if(FAILED(hres))
324         return hres;
325
326     hres = ITypeInfo_GetTypeAttr(ti, &attr);
327     if(FAILED(hres)) {
328         ITypeInfo_Release(ti);
329         return hres;
330     }
331
332     func_cnt = attr->cFuncs;
333     ITypeInfo_ReleaseTypeAttr(ti, attr);
334
335     ids = heap_alloc(func_cnt*sizeof(DISPID));
336     if(!ids) {
337         ITypeInfo_Release(ti);
338         return E_OUTOFMEMORY;
339     }
340
341     for(i=0; i < func_cnt; i++) {
342         hres = ITypeInfo_GetFuncDesc(ti, i, &funcdesc);
343         if(FAILED(hres))
344             break;
345
346         ids[i] = funcdesc->memid;
347         ITypeInfo_ReleaseFuncDesc(ti, funcdesc);
348     }
349
350     ITypeInfo_Release(ti);
351     if(FAILED(hres)) {
352         heap_free(ids);
353         return hres;
354     }
355
356     qsort(ids, func_cnt, sizeof(DISPID), id_cmp);
357
358     *ret_size = func_cnt;
359     *ret = ids;
360     return S_OK;
361 }
362
363 static dispex_data_t *get_dispex_data(DispatchEx *This)
364 {
365     if(This->data->data)
366         return This->data->data;
367
368     EnterCriticalSection(&cs_dispex_static_data);
369
370     if(!This->data->data)
371         This->data->data = preprocess_dispex_data(This);
372
373     LeaveCriticalSection(&cs_dispex_static_data);
374
375     return This->data->data;
376 }
377
378 static inline BOOL is_custom_dispid(DISPID id)
379 {
380     return MSHTML_DISPID_CUSTOM_MIN <= id && id <= MSHTML_DISPID_CUSTOM_MAX;
381 }
382
383 static inline BOOL is_dynamic_dispid(DISPID id)
384 {
385     return DISPID_DYNPROP_0 <= id && id <= DISPID_DYNPROP_MAX;
386 }
387
388 static HRESULT variant_copy(VARIANT *dest, VARIANT *src)
389 {
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;
394     }
395
396     return VariantCopy(dest, src);
397 }
398
399 static inline dispex_dynamic_data_t *get_dynamic_data(DispatchEx *This)
400 {
401     if(This->dynamic_data)
402         return This->dynamic_data;
403
404     This->dynamic_data = heap_alloc_zero(sizeof(dispex_dynamic_data_t));
405     if(!This->dynamic_data)
406         return NULL;
407
408     if(This->data->vtbl && This->data->vtbl->populate_props)
409         This->data->vtbl->populate_props(This);
410
411     return This->dynamic_data;
412 }
413
414 static HRESULT get_dynamic_prop(DispatchEx *This, const WCHAR *name, DWORD flags, dynamic_prop_t **ret)
415 {
416     const BOOL alloc = flags & fdexNameEnsure;
417     dispex_dynamic_data_t *data;
418     dynamic_prop_t *prop;
419
420     data = get_dynamic_data(This);
421     if(!data)
422         return E_OUTOFMEMORY;
423
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;
430             }
431             *ret = prop;
432             return S_OK;
433         }
434     }
435
436     if(!alloc)
437         return DISP_E_UNKNOWNNAME;
438
439     TRACE("creating dynamic prop %s\n", debugstr_w(name));
440
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;
448
449         new_props = heap_realloc(data->props, sizeof(dynamic_prop_t)*(data->buf_size<<1));
450         if(!new_props)
451             return E_OUTOFMEMORY;
452
453         data->props = new_props;
454         data->buf_size <<= 1;
455     }
456
457     prop = data->props + data->prop_cnt;
458
459     prop->name = heap_strdupW(name);
460     if(!prop->name)
461         return E_OUTOFMEMORY;
462
463     VariantInit(&prop->var);
464     prop->flags = 0;
465     data->prop_cnt++;
466     *ret = prop;
467     return S_OK;
468 }
469
470 HRESULT dispex_get_dprop_ref(DispatchEx *This, const WCHAR *name, BOOL alloc, VARIANT **ret)
471 {
472     dynamic_prop_t *prop;
473     HRESULT hres;
474
475     hres = get_dynamic_prop(This, name, alloc ? fdexNameEnsure : 0, &prop);
476     if(FAILED(hres))
477         return hres;
478
479     *ret = &prop->var;
480     return S_OK;
481 }
482
483 static HRESULT dispex_value(DispatchEx *This, LCID lcid, WORD flags, DISPPARAMS *params,
484         VARIANT *res, EXCEPINFO *ei, IServiceProvider *caller)
485 {
486     if(This->data->vtbl && This->data->vtbl->value)
487         return This->data->vtbl->value(This, lcid, flags, params, res, ei, caller);
488
489     switch(flags) {
490     case DISPATCH_PROPERTYGET:
491         V_VT(res) = VT_BSTR;
492         V_BSTR(res) = SysAllocString(objectW);
493         if(!V_BSTR(res))
494             return E_OUTOFMEMORY;
495         break;
496     default:
497         FIXME("Unimplemented flags %x\n", flags);
498         return E_NOTIMPL;
499     }
500
501     return S_OK;
502 }
503
504 static HRESULT typeinfo_invoke(DispatchEx *This, func_info_t *func, WORD flags, DISPPARAMS *dp, VARIANT *res,
505         EXCEPINFO *ei)
506 {
507     ITypeInfo *ti;
508     IUnknown *unk;
509     UINT argerr=0;
510     HRESULT hres;
511
512     hres = get_typeinfo(func->tid, &ti);
513     if(FAILED(hres)) {
514         ERR("Could not get type info: %08x\n", hres);
515         return hres;
516     }
517
518     hres = IUnknown_QueryInterface(This->outer, tid_ids[func->tid], (void**)&unk);
519     if(FAILED(hres)) {
520         ERR("Could not get iface %s: %08x\n", debugstr_guid(tid_ids[func->tid]), hres);
521         return E_FAIL;
522     }
523
524     hres = ITypeInfo_Invoke(ti, unk, func->id, flags, dp, res, ei, &argerr);
525
526     IUnknown_Release(unk);
527     return hres;
528 }
529
530 static inline func_disp_t *impl_from_IUnknown(IUnknown *iface)
531 {
532     return CONTAINING_RECORD(iface, func_disp_t, IUnknown_iface);
533 }
534
535 static HRESULT WINAPI Function_QueryInterface(IUnknown *iface, REFIID riid, void **ppv)
536 {
537     func_disp_t *This = impl_from_IUnknown(iface);
538
539     if(IsEqualGUID(&IID_IUnknown, riid)) {
540         TRACE("(%p)->(IID_IUnknown %p)\n", This, ppv);
541         *ppv = &This->IUnknown_iface;
542     }else if(dispex_query_interface(&This->dispex, riid, ppv)) {
543         return *ppv ? S_OK : E_NOINTERFACE;
544     }else {
545         *ppv = NULL;
546         return E_NOINTERFACE;
547     }
548
549     IUnknown_AddRef((IUnknown*)*ppv);
550     return S_OK;
551 }
552
553 static ULONG WINAPI Function_AddRef(IUnknown *iface)
554 {
555     func_disp_t *This = impl_from_IUnknown(iface);
556
557     TRACE("(%p)\n", This);
558
559     return IDispatchEx_AddRef(&This->obj->IDispatchEx_iface);
560 }
561
562 static ULONG WINAPI Function_Release(IUnknown *iface)
563 {
564     func_disp_t *This = impl_from_IUnknown(iface);
565
566     TRACE("(%p)\n", This);
567
568     return IDispatchEx_Release(&This->obj->IDispatchEx_iface);
569 }
570
571 static const IUnknownVtbl FunctionUnkVtbl = {
572     Function_QueryInterface,
573     Function_AddRef,
574     Function_Release
575 };
576
577 static inline func_disp_t *impl_from_DispatchEx(DispatchEx *iface)
578 {
579     return CONTAINING_RECORD(iface, func_disp_t, dispex);
580 }
581
582 static HRESULT function_value(DispatchEx *dispex, LCID lcid, WORD flags, DISPPARAMS *params,
583         VARIANT *res, EXCEPINFO *ei, IServiceProvider *caller)
584 {
585     func_disp_t *This = impl_from_DispatchEx(dispex);
586     HRESULT hres;
587
588     switch(flags) {
589     case DISPATCH_METHOD|DISPATCH_PROPERTYGET:
590         if(!res)
591             return E_INVALIDARG;
592         /* fall through */
593     case DISPATCH_METHOD:
594         hres = typeinfo_invoke(This->obj, This->info, flags, params, res, ei);
595         break;
596     default:
597         FIXME("Unimplemented flags %x\n", flags);
598         hres = E_NOTIMPL;
599     }
600
601     return hres;
602 }
603
604 static const dispex_static_data_vtbl_t function_dispex_vtbl = {
605     function_value,
606     NULL,
607     NULL,
608     NULL
609 };
610
611 static const tid_t function_iface_tids[] = {0};
612
613 static dispex_static_data_t function_dispex = {
614     &function_dispex_vtbl,
615     NULL_tid,
616     NULL,
617     function_iface_tids
618 };
619
620 static func_disp_t *create_func_disp(DispatchEx *obj, func_info_t *info)
621 {
622     func_disp_t *ret;
623
624     ret = heap_alloc_zero(sizeof(func_disp_t));
625     if(!ret)
626         return NULL;
627
628     ret->IUnknown_iface.lpVtbl = &FunctionUnkVtbl;
629     init_dispex(&ret->dispex, &ret->IUnknown_iface,  &function_dispex);
630     ret->obj = obj;
631     ret->info = info;
632
633     return ret;
634 }
635
636 static HRESULT function_invoke(DispatchEx *This, func_info_t *func, WORD flags, DISPPARAMS *dp, VARIANT *res,
637         EXCEPINFO *ei)
638 {
639     HRESULT hres;
640
641     switch(flags) {
642     case DISPATCH_METHOD|DISPATCH_PROPERTYGET:
643         if(!res)
644             return E_INVALIDARG;
645         /* fall through */
646     case DISPATCH_METHOD:
647         hres = typeinfo_invoke(This, func, flags, dp, res, ei);
648         break;
649     case DISPATCH_PROPERTYGET: {
650         dispex_dynamic_data_t *dynamic_data;
651
652         if(func->id == DISPID_VALUE) {
653             BSTR ret;
654
655             ret = SysAllocString(objectW);
656             if(!ret)
657                 return E_OUTOFMEMORY;
658
659             V_VT(res) = VT_BSTR;
660             V_BSTR(res) = ret;
661             return S_OK;
662         }
663
664         dynamic_data = get_dynamic_data(This);
665         if(!dynamic_data)
666             return E_OUTOFMEMORY;
667
668         if(!dynamic_data->func_disps) {
669             dynamic_data->func_disps = heap_alloc_zero(This->data->data->func_disp_cnt * sizeof(func_disp_t*));
670             if(!dynamic_data->func_disps)
671                 return E_OUTOFMEMORY;
672         }
673
674         if(!dynamic_data->func_disps[func->func_disp_idx]) {
675             dynamic_data->func_disps[func->func_disp_idx] = create_func_disp(This, func);
676             if(!dynamic_data->func_disps[func->func_disp_idx])
677                 return E_OUTOFMEMORY;
678         }
679
680         V_VT(res) = VT_DISPATCH;
681         V_DISPATCH(res) = (IDispatch*)&dynamic_data->func_disps[func->func_disp_idx]->dispex.IDispatchEx_iface;
682         IDispatch_AddRef(V_DISPATCH(res));
683         hres = S_OK;
684         break;
685     }
686     default:
687         FIXME("Unimplemented flags %x\n", flags);
688         /* fall through */
689     case DISPATCH_PROPERTYPUT:
690         hres = E_NOTIMPL;
691     }
692
693     return hres;
694 }
695
696 static HRESULT get_builtin_func(dispex_data_t *data, DISPID id, func_info_t **ret)
697 {
698     int min, max, n;
699
700     min = 0;
701     max = data->func_cnt-1;
702
703     while(min <= max) {
704         n = (min+max)/2;
705
706         if(data->funcs[n].id == id) {
707             *ret = data->funcs+n;
708             return S_OK;
709         }
710
711         if(data->funcs[n].id < id)
712             min = n+1;
713         else
714             max = n-1;
715     }
716
717     WARN("invalid id %x\n", id);
718     return DISP_E_UNKNOWNNAME;
719 }
720
721 static HRESULT get_builtin_id(DispatchEx *This, BSTR name, DWORD grfdex, DISPID *ret)
722 {
723     dispex_data_t *data;
724     int min, max, n, c;
725
726     data = get_dispex_data(This);
727     if(!data)
728         return E_FAIL;
729
730     min = 0;
731     max = data->func_cnt-1;
732
733     while(min <= max) {
734         n = (min+max)/2;
735
736         c = strcmpiW(data->name_table[n]->name, name);
737         if(!c) {
738             if((grfdex & fdexNameCaseSensitive) && strcmpW(data->name_table[n]->name, name))
739                 break;
740
741             *ret = data->name_table[n]->id;
742             return S_OK;
743         }
744
745         if(c > 0)
746             max = n-1;
747         else
748             min = n+1;
749     }
750
751     if(This->data->vtbl && This->data->vtbl->get_dispid) {
752         HRESULT hres;
753
754         hres = This->data->vtbl->get_dispid(This, name, grfdex, ret);
755         if(hres != DISP_E_UNKNOWNNAME)
756             return hres;
757     }
758
759     return DISP_E_UNKNOWNNAME;
760 }
761
762 /* List all types used by IDispatchEx-based properties */
763 #define BUILTIN_TYPES_SWITCH                            \
764     CASE_VT(VT_I2, INT16, V_I2);                        \
765     CASE_VT(VT_I4, INT32, V_I4);                        \
766     CASE_VT(VT_R4, float, V_R4);                        \
767     CASE_VT(VT_BSTR, BSTR, V_BSTR);                     \
768     CASE_VT(VT_BOOL, VARIANT_BOOL, V_BOOL);             \
769     CASE_VT(VT_VARIANT, VARIANT, *);                    \
770     CASE_VT(VT_PTR, void*, V_BYREF);                    \
771     CASE_VT(VT_UNKNOWN, IUnknown*, V_UNKNOWN);          \
772     CASE_VT(VT_DISPATCH, IDispatch*, V_DISPATCH)
773
774 static HRESULT builtin_propget(DispatchEx *This, func_info_t *func, DISPPARAMS *dp, VARIANT *res)
775 {
776     IUnknown *iface;
777     HRESULT hres;
778
779     if(dp->cArgs) {
780         FIXME("cArgs %d\n", dp->cArgs);
781         return E_NOTIMPL;
782     }
783
784     assert(func->get_vtbl_off);
785
786     hres = IUnknown_QueryInterface(This->outer, tid_ids[func->tid], (void**)&iface);
787     if(SUCCEEDED(hres)) {
788         switch(func->prop_vt) {
789 #define CASE_VT(vt,type,access) \
790         case vt: { \
791             type val; \
792             hres = ((HRESULT (WINAPI*)(IUnknown*,type*))((void**)iface->lpVtbl)[func->get_vtbl_off])(iface,&val); \
793             if(SUCCEEDED(hres)) \
794                 access(res) = val; \
795             } \
796             break
797         BUILTIN_TYPES_SWITCH;
798 #undef CASE_VT
799         default:
800             FIXME("Unhandled vt %d\n", func->prop_vt);
801             hres = E_NOTIMPL;
802         }
803         IUnknown_Release(iface);
804     }
805
806     if(FAILED(hres))
807         return hres;
808
809     V_VT(res) = func->prop_vt == VT_PTR ? VT_DISPATCH : func->prop_vt;
810     return S_OK;
811 }
812
813 static HRESULT invoke_builtin_prop(DispatchEx *This, DISPID id, LCID lcid, WORD flags, DISPPARAMS *dp,
814         VARIANT *res, EXCEPINFO *ei, IServiceProvider *caller)
815 {
816     dispex_data_t *data;
817     func_info_t *func;
818     HRESULT hres;
819
820     data = get_dispex_data(This);
821     if(!data)
822         return E_FAIL;
823
824     hres = get_builtin_func(data, id, &func);
825     if(id == DISPID_VALUE && hres == DISP_E_UNKNOWNNAME)
826         return dispex_value(This, lcid, flags, dp, res, ei, caller);
827     if(FAILED(hres))
828         return hres;
829
830     if(func->func_disp_idx != -1)
831         return function_invoke(This, func, flags, dp, res, ei);
832
833     switch(flags) {
834     case DISPATCH_PROPERTYGET:
835         hres = builtin_propget(This, func, dp, res);
836         break;
837     default:
838         hres = typeinfo_invoke(This, func, flags, dp, res, ei);
839     }
840
841     return hres;
842 }
843
844 HRESULT remove_prop(DispatchEx *This, BSTR name, VARIANT_BOOL *success)
845 {
846     dynamic_prop_t *prop;
847     DISPID id;
848     HRESULT hres;
849
850     hres = get_builtin_id(This, name, 0, &id);
851     if(hres == S_OK) {
852         DISPID named_id = DISPID_PROPERTYPUT;
853         VARIANT var;
854         DISPPARAMS dp = {&var,&named_id,1,1};
855         EXCEPINFO ei;
856
857         V_VT(&var) = VT_EMPTY;
858         memset(&ei, 0, sizeof(ei));
859         hres = invoke_builtin_prop(This, id, 0, DISPATCH_PROPERTYPUT, &dp, NULL, &ei, NULL);
860         if(FAILED(hres))
861             return hres;
862
863         *success = VARIANT_TRUE;
864         return S_OK;
865     }
866
867     hres = get_dynamic_prop(This, name, 0, &prop);
868     if(FAILED(hres)) {
869         if(hres != DISP_E_UNKNOWNNAME)
870             return hres;
871         *success = VARIANT_FALSE;
872         return S_OK;
873     }
874
875     VariantClear(&prop->var);
876     prop->flags |= DYNPROP_DELETED;
877     *success = VARIANT_TRUE;
878     return S_OK;
879 }
880
881 static inline DispatchEx *impl_from_IDispatchEx(IDispatchEx *iface)
882 {
883     return CONTAINING_RECORD(iface, DispatchEx, IDispatchEx_iface);
884 }
885
886 static HRESULT WINAPI DispatchEx_QueryInterface(IDispatchEx *iface, REFIID riid, void **ppv)
887 {
888     DispatchEx *This = impl_from_IDispatchEx(iface);
889
890     return IUnknown_QueryInterface(This->outer, riid, ppv);
891 }
892
893 static ULONG WINAPI DispatchEx_AddRef(IDispatchEx *iface)
894 {
895     DispatchEx *This = impl_from_IDispatchEx(iface);
896
897     return IUnknown_AddRef(This->outer);
898 }
899
900 static ULONG WINAPI DispatchEx_Release(IDispatchEx *iface)
901 {
902     DispatchEx *This = impl_from_IDispatchEx(iface);
903
904     return IUnknown_Release(This->outer);
905 }
906
907 static HRESULT WINAPI DispatchEx_GetTypeInfoCount(IDispatchEx *iface, UINT *pctinfo)
908 {
909     DispatchEx *This = impl_from_IDispatchEx(iface);
910
911     TRACE("(%p)->(%p)\n", This, pctinfo);
912
913     *pctinfo = 1;
914     return S_OK;
915 }
916
917 static HRESULT WINAPI DispatchEx_GetTypeInfo(IDispatchEx *iface, UINT iTInfo,
918                                               LCID lcid, ITypeInfo **ppTInfo)
919 {
920     DispatchEx *This = impl_from_IDispatchEx(iface);
921     HRESULT hres;
922
923     TRACE("(%p)->(%u %u %p)\n", This, iTInfo, lcid, ppTInfo);
924
925     hres = get_typeinfo(This->data->disp_tid, ppTInfo);
926     if(FAILED(hres))
927         return hres;
928
929     ITypeInfo_AddRef(*ppTInfo);
930     return S_OK;
931 }
932
933 static HRESULT WINAPI DispatchEx_GetIDsOfNames(IDispatchEx *iface, REFIID riid,
934                                                 LPOLESTR *rgszNames, UINT cNames,
935                                                 LCID lcid, DISPID *rgDispId)
936 {
937     DispatchEx *This = impl_from_IDispatchEx(iface);
938     UINT i;
939     HRESULT hres;
940
941     TRACE("(%p)->(%s %p %u %u %p)\n", This, debugstr_guid(riid), rgszNames, cNames,
942           lcid, rgDispId);
943
944     for(i=0; i < cNames; i++) {
945         hres = IDispatchEx_GetDispID(&This->IDispatchEx_iface, rgszNames[i], 0, rgDispId+i);
946         if(FAILED(hres))
947             return hres;
948     }
949
950     return S_OK;
951 }
952
953 static HRESULT WINAPI DispatchEx_Invoke(IDispatchEx *iface, DISPID dispIdMember,
954                             REFIID riid, LCID lcid, WORD wFlags, DISPPARAMS *pDispParams,
955                             VARIANT *pVarResult, EXCEPINFO *pExcepInfo, UINT *puArgErr)
956 {
957     DispatchEx *This = impl_from_IDispatchEx(iface);
958
959     TRACE("(%p)->(%d %s %d %d %p %p %p %p)\n", This, dispIdMember, debugstr_guid(riid),
960           lcid, wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr);
961
962     return IDispatchEx_InvokeEx(&This->IDispatchEx_iface, dispIdMember, lcid, wFlags, pDispParams,
963             pVarResult, pExcepInfo, NULL);
964 }
965
966 static HRESULT WINAPI DispatchEx_GetDispID(IDispatchEx *iface, BSTR bstrName, DWORD grfdex, DISPID *pid)
967 {
968     DispatchEx *This = impl_from_IDispatchEx(iface);
969     dynamic_prop_t *dprop;
970     HRESULT hres;
971
972     TRACE("(%p)->(%s %x %p)\n", This, debugstr_w(bstrName), grfdex, pid);
973
974     if(grfdex & ~(fdexNameCaseSensitive|fdexNameCaseInsensitive|fdexNameEnsure|fdexNameImplicit|FDEX_VERSION_MASK))
975         FIXME("Unsupported grfdex %x\n", grfdex);
976
977     hres = get_builtin_id(This, bstrName, grfdex, pid);
978     if(hres != DISP_E_UNKNOWNNAME)
979         return hres;
980
981     hres = get_dynamic_prop(This, bstrName, grfdex, &dprop);
982     if(FAILED(hres))
983         return hres;
984
985     *pid = DISPID_DYNPROP_0 + (dprop - This->dynamic_data->props);
986     return S_OK;
987 }
988
989 static HRESULT WINAPI DispatchEx_InvokeEx(IDispatchEx *iface, DISPID id, LCID lcid, WORD wFlags, DISPPARAMS *pdp,
990         VARIANT *pvarRes, EXCEPINFO *pei, IServiceProvider *pspCaller)
991 {
992     DispatchEx *This = impl_from_IDispatchEx(iface);
993     HRESULT hres;
994
995     TRACE("(%p)->(%x %x %x %p %p %p %p)\n", This, id, lcid, wFlags, pdp, pvarRes, pei, pspCaller);
996
997     if(is_custom_dispid(id) && This->data->vtbl && This->data->vtbl->invoke)
998         return This->data->vtbl->invoke(This, id, lcid, wFlags, pdp, pvarRes, pei, pspCaller);
999
1000     if(wFlags == DISPATCH_CONSTRUCT) {
1001         if(id == DISPID_VALUE) {
1002             if(This->data->vtbl && This->data->vtbl->value) {
1003                 return This->data->vtbl->value(This, lcid, wFlags, pdp, pvarRes, pei, pspCaller);
1004             }
1005             FIXME("DISPATCH_CONSTRUCT flag but missing value function\n");
1006             return E_FAIL;
1007         }
1008         FIXME("DISPATCH_CONSTRUCT flag without DISPID_VALUE\n");
1009         return E_FAIL;
1010     }
1011
1012     if(is_dynamic_dispid(id)) {
1013         DWORD idx = id - DISPID_DYNPROP_0;
1014         dynamic_prop_t *prop;
1015
1016         if(!get_dynamic_data(This) || This->dynamic_data->prop_cnt <= idx)
1017             return DISP_E_UNKNOWNNAME;
1018
1019         prop = This->dynamic_data->props+idx;
1020
1021         switch(wFlags) {
1022         case DISPATCH_METHOD|DISPATCH_PROPERTYGET:
1023             if(!pvarRes)
1024                 return E_INVALIDARG;
1025             /* fall through */
1026         case DISPATCH_METHOD: {
1027             DISPID named_arg = DISPID_THIS;
1028             DISPPARAMS dp = {NULL, &named_arg, 0, 1};
1029             IDispatchEx *dispex;
1030
1031             if(V_VT(&prop->var) != VT_DISPATCH) {
1032                 FIXME("invoke %s\n", debugstr_variant(&prop->var));
1033                 return E_NOTIMPL;
1034             }
1035
1036             if(pdp->cNamedArgs) {
1037                 FIXME("named args not supported\n");
1038                 return E_NOTIMPL;
1039             }
1040
1041             dp.rgvarg = heap_alloc((pdp->cArgs+1)*sizeof(VARIANTARG));
1042             if(!dp.rgvarg)
1043                 return E_OUTOFMEMORY;
1044
1045             dp.cArgs = pdp->cArgs+1;
1046             memcpy(dp.rgvarg+1, pdp->rgvarg, pdp->cArgs*sizeof(VARIANTARG));
1047
1048             V_VT(dp.rgvarg) = VT_DISPATCH;
1049             V_DISPATCH(dp.rgvarg) = (IDispatch*)&This->IDispatchEx_iface;
1050
1051             hres = IDispatch_QueryInterface(V_DISPATCH(&prop->var), &IID_IDispatchEx, (void**)&dispex);
1052             TRACE("%s call\n", debugstr_w(This->dynamic_data->props[idx].name));
1053             if(SUCCEEDED(hres)) {
1054                 hres = IDispatchEx_InvokeEx(dispex, DISPID_VALUE, lcid, wFlags, &dp, pvarRes, pei, pspCaller);
1055                 IDispatchEx_Release(dispex);
1056             }else {
1057                 ULONG err = 0;
1058                 hres = IDispatch_Invoke(V_DISPATCH(&prop->var), DISPID_VALUE, &IID_NULL, lcid, wFlags, pdp, pvarRes, pei, &err);
1059             }
1060             TRACE("%s ret %08x\n", debugstr_w(This->dynamic_data->props[idx].name), hres);
1061
1062             heap_free(dp.rgvarg);
1063             return hres;
1064         }
1065         case DISPATCH_PROPERTYGET:
1066             if(prop->flags & DYNPROP_DELETED)
1067                 return DISP_E_UNKNOWNNAME;
1068             V_VT(pvarRes) = VT_EMPTY;
1069             return variant_copy(pvarRes, &prop->var);
1070         case DISPATCH_PROPERTYPUT|DISPATCH_PROPERTYPUTREF:
1071         case DISPATCH_PROPERTYPUT:
1072             if(pdp->cArgs != 1 || (pdp->cNamedArgs == 1 && *pdp->rgdispidNamedArgs != DISPID_PROPERTYPUT)
1073                || pdp->cNamedArgs > 1) {
1074                 FIXME("invalid args\n");
1075                 return E_INVALIDARG;
1076             }
1077
1078             TRACE("put %s\n", debugstr_variant(pdp->rgvarg));
1079             VariantClear(&prop->var);
1080             hres = variant_copy(&prop->var, pdp->rgvarg);
1081             if(FAILED(hres))
1082                 return hres;
1083
1084             prop->flags &= ~DYNPROP_DELETED;
1085             return S_OK;
1086         default:
1087             FIXME("unhandled wFlags %x\n", wFlags);
1088             return E_NOTIMPL;
1089         }
1090     }
1091
1092     return invoke_builtin_prop(This, id, lcid, wFlags, pdp, pvarRes, pei, pspCaller);
1093 }
1094
1095 static HRESULT WINAPI DispatchEx_DeleteMemberByName(IDispatchEx *iface, BSTR bstrName, DWORD grfdex)
1096 {
1097     DispatchEx *This = impl_from_IDispatchEx(iface);
1098
1099     TRACE("(%p)->(%s %x)\n", This, debugstr_w(bstrName), grfdex);
1100
1101     /* Not implemented by IE */
1102     return E_NOTIMPL;
1103 }
1104
1105 static HRESULT WINAPI DispatchEx_DeleteMemberByDispID(IDispatchEx *iface, DISPID id)
1106 {
1107     DispatchEx *This = impl_from_IDispatchEx(iface);
1108
1109     TRACE("(%p)->(%x)\n", This, id);
1110
1111     /* Not implemented by IE */
1112     return E_NOTIMPL;
1113 }
1114
1115 static HRESULT WINAPI DispatchEx_GetMemberProperties(IDispatchEx *iface, DISPID id, DWORD grfdexFetch, DWORD *pgrfdex)
1116 {
1117     DispatchEx *This = impl_from_IDispatchEx(iface);
1118     FIXME("(%p)->(%x %x %p)\n", This, id, grfdexFetch, pgrfdex);
1119     return E_NOTIMPL;
1120 }
1121
1122 static HRESULT WINAPI DispatchEx_GetMemberName(IDispatchEx *iface, DISPID id, BSTR *pbstrName)
1123 {
1124     DispatchEx *This = impl_from_IDispatchEx(iface);
1125     dispex_data_t *data;
1126     func_info_t *func;
1127     HRESULT hres;
1128
1129     TRACE("(%p)->(%x %p)\n", This, id, pbstrName);
1130
1131     if(is_dynamic_dispid(id)) {
1132         DWORD idx = id - DISPID_DYNPROP_0;
1133
1134         if(!get_dynamic_data(This) || This->dynamic_data->prop_cnt <= idx)
1135             return DISP_E_UNKNOWNNAME;
1136
1137         *pbstrName = SysAllocString(This->dynamic_data->props[idx].name);
1138         if(!*pbstrName)
1139             return E_OUTOFMEMORY;
1140
1141         return S_OK;
1142     }
1143
1144     data = get_dispex_data(This);
1145     if(!data)
1146         return E_FAIL;
1147
1148     hres = get_builtin_func(data, id, &func);
1149     if(FAILED(hres))
1150         return hres;
1151
1152     *pbstrName = SysAllocString(func->name);
1153     if(!*pbstrName)
1154         return E_OUTOFMEMORY;
1155     return S_OK;
1156 }
1157
1158 static HRESULT WINAPI DispatchEx_GetNextDispID(IDispatchEx *iface, DWORD grfdex, DISPID id, DISPID *pid)
1159 {
1160     DispatchEx *This = impl_from_IDispatchEx(iface);
1161     dispex_data_t *data;
1162     func_info_t *func;
1163     HRESULT hres;
1164
1165     TRACE("(%p)->(%x %x %p)\n", This, grfdex, id, pid);
1166
1167     if(is_dynamic_dispid(id)) {
1168         DWORD idx = id - DISPID_DYNPROP_0;
1169
1170         if(!get_dynamic_data(This) || This->dynamic_data->prop_cnt <= idx)
1171             return DISP_E_UNKNOWNNAME;
1172
1173         while(++idx < This->dynamic_data->prop_cnt && This->dynamic_data->props[idx].flags & DYNPROP_DELETED);
1174
1175         if(idx == This->dynamic_data->prop_cnt) {
1176             *pid = DISPID_STARTENUM;
1177             return S_FALSE;
1178         }
1179
1180         *pid = DISPID_DYNPROP_0+idx;
1181         return S_OK;
1182     }
1183
1184     data = get_dispex_data(This);
1185     if(!data)
1186         return E_FAIL;
1187
1188     if(id == DISPID_STARTENUM) {
1189         func = data->funcs;
1190     }else {
1191         hres = get_builtin_func(data, id, &func);
1192         if(FAILED(hres))
1193             return hres;
1194         func++;
1195     }
1196
1197     while(func < data->funcs+data->func_cnt) {
1198         /* FIXME: Skip hidden properties */
1199         if(func->func_disp_idx == -1) {
1200             *pid = func->id;
1201             return S_OK;
1202         }
1203         func++;
1204     }
1205
1206     if(get_dynamic_data(This) && This->dynamic_data->prop_cnt) {
1207         *pid = DISPID_DYNPROP_0;
1208         return S_OK;
1209     }
1210
1211     *pid = DISPID_STARTENUM;
1212     return S_FALSE;
1213 }
1214
1215 static HRESULT WINAPI DispatchEx_GetNameSpaceParent(IDispatchEx *iface, IUnknown **ppunk)
1216 {
1217     DispatchEx *This = impl_from_IDispatchEx(iface);
1218     FIXME("(%p)->(%p)\n", This, ppunk);
1219     return E_NOTIMPL;
1220 }
1221
1222 static IDispatchExVtbl DispatchExVtbl = {
1223     DispatchEx_QueryInterface,
1224     DispatchEx_AddRef,
1225     DispatchEx_Release,
1226     DispatchEx_GetTypeInfoCount,
1227     DispatchEx_GetTypeInfo,
1228     DispatchEx_GetIDsOfNames,
1229     DispatchEx_Invoke,
1230     DispatchEx_GetDispID,
1231     DispatchEx_InvokeEx,
1232     DispatchEx_DeleteMemberByName,
1233     DispatchEx_DeleteMemberByDispID,
1234     DispatchEx_GetMemberProperties,
1235     DispatchEx_GetMemberName,
1236     DispatchEx_GetNextDispID,
1237     DispatchEx_GetNameSpaceParent
1238 };
1239
1240 BOOL dispex_query_interface(DispatchEx *This, REFIID riid, void **ppv)
1241 {
1242     static const IID IID_UndocumentedScriptIface =
1243         {0x719c3050,0xf9d3,0x11cf,{0xa4,0x93,0x00,0x40,0x05,0x23,0xa8,0xa0}};
1244     static const IID IID_IDispatchJS =
1245         {0x719c3050,0xf9d3,0x11cf,{0xa4,0x93,0x00,0x40,0x05,0x23,0xa8,0xa6}};
1246
1247     if(IsEqualGUID(&IID_IDispatch, riid)) {
1248         TRACE("(%p)->(IID_IDispatch %p)\n", This, ppv);
1249         *ppv = &This->IDispatchEx_iface;
1250     }else if(IsEqualGUID(&IID_IDispatchEx, riid)) {
1251         TRACE("(%p)->(IID_IDispatchEx %p)\n", This, ppv);
1252         *ppv = &This->IDispatchEx_iface;
1253     }else if(IsEqualGUID(&IID_IDispatchJS, riid)) {
1254         TRACE("(%p)->(IID_IDispatchJS %p) returning NULL\n", This, ppv);
1255         *ppv = NULL;
1256     }else if(IsEqualGUID(&IID_UndocumentedScriptIface, riid)) {
1257         TRACE("(%p)->(IID_UndocumentedScriptIface %p) returning NULL\n", This, ppv);
1258         *ppv = NULL;
1259     }else {
1260         return FALSE;
1261     }
1262
1263     if(*ppv)
1264         IUnknown_AddRef((IUnknown*)*ppv);
1265     return TRUE;
1266 }
1267
1268 void release_dispex(DispatchEx *This)
1269 {
1270     dynamic_prop_t *prop;
1271
1272     if(!This->dynamic_data)
1273         return;
1274
1275     for(prop = This->dynamic_data->props; prop < This->dynamic_data->props + This->dynamic_data->prop_cnt; prop++) {
1276         VariantClear(&prop->var);
1277         heap_free(prop->name);
1278     }
1279
1280     heap_free(This->dynamic_data->props);
1281
1282     if(This->dynamic_data->func_disps) {
1283         unsigned i;
1284
1285         for(i=0; i < This->data->data->func_disp_cnt; i++) {
1286             if(This->dynamic_data->func_disps[i]) {
1287                 release_dispex(&This->dynamic_data->func_disps[i]->dispex);
1288                 heap_free(This->dynamic_data->func_disps[i]);
1289             }
1290         }
1291
1292         heap_free(This->dynamic_data->func_disps);
1293     }
1294
1295     heap_free(This->dynamic_data);
1296 }
1297
1298 void init_dispex(DispatchEx *dispex, IUnknown *outer, dispex_static_data_t *data)
1299 {
1300     dispex->IDispatchEx_iface.lpVtbl = &DispatchExVtbl;
1301     dispex->outer = outer;
1302     dispex->data = data;
1303     dispex->dynamic_data = NULL;
1304 }