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