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