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