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