jscript: Added assign expression implementation.
[wine/wine64.git] / dlls / mshtml / dispex.c
blob587d95305814ee75f0daef51cf9c4f44fc5367df
1 /*
2 * Copyright 2008 Jacek Caban for CodeWeavers
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.
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.
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
19 #include <stdarg.h>
21 #define COBJMACROS
23 #include "windef.h"
24 #include "winbase.h"
25 #include "winuser.h"
26 #include "ole2.h"
28 #include "wine/debug.h"
30 #include "mshtml_private.h"
32 WINE_DEFAULT_DEBUG_CHANNEL(mshtml);
34 typedef struct {
35 DISPID id;
36 BSTR name;
37 tid_t tid;
38 } func_info_t;
40 struct dispex_data_t {
41 DWORD func_cnt;
42 func_info_t *funcs;
43 func_info_t **name_table;
45 struct list entry;
48 typedef struct {
49 VARIANT var;
50 LPWSTR name;
51 } dynamic_prop_t;
53 struct dispex_dynamic_data_t {
54 DWORD buf_size;
55 DWORD prop_cnt;
56 dynamic_prop_t *props;
59 #define DISPID_DYNPROP_0 0x50000000
60 #define DISPID_DYNPROP_MAX 0x5fffffff
62 static ITypeLib *typelib;
63 static ITypeInfo *typeinfos[LAST_tid];
64 static struct list dispex_data_list = LIST_INIT(dispex_data_list);
66 static REFIID tid_ids[] = {
67 &IID_NULL,
68 &DIID_DispDOMChildrenCollection,
69 &DIID_DispHTMLBody,
70 &DIID_DispHTMLCommentElement,
71 &DIID_DispHTMLDocument,
72 &DIID_DispHTMLDOMTextNode,
73 &DIID_DispHTMLElementCollection,
74 &DIID_DispHTMLGenericElement,
75 &DIID_DispHTMLImg,
76 &DIID_DispHTMLInputElement,
77 &DIID_DispHTMLOptionElement,
78 &DIID_DispHTMLSelectElement,
79 &DIID_DispHTMLStyle,
80 &DIID_DispHTMLUnknownElement,
81 &DIID_DispHTMLWindow2,
82 &IID_IHTMLBodyElement,
83 &IID_IHTMLBodyElement2,
84 &IID_IHTMLCommentElement,
85 &IID_IHTMLControlElement,
86 &IID_IHTMLDocument2,
87 &IID_IHTMLDocument3,
88 &IID_IHTMLDocument4,
89 &IID_IHTMLDocument5,
90 &IID_IHTMLDOMChildrenCollection,
91 &IID_IHTMLDOMNode,
92 &IID_IHTMLDOMNode2,
93 &IID_IHTMLDOMTextNode,
94 &IID_IHTMLElement,
95 &IID_IHTMLElement2,
96 &IID_IHTMLElement3,
97 &IID_IHTMLElement4,
98 &IID_IHTMLElementCollection,
99 &IID_IHTMLGenericElement,
100 &IID_IHTMLImgElement,
101 &IID_IHTMLInputElement,
102 &IID_IHTMLOptionElement,
103 &IID_IHTMLSelectElement,
104 &IID_IHTMLStyle,
105 &IID_IHTMLTextContainer,
106 &IID_IHTMLUniqueName,
107 &IID_IHTMLWindow2,
108 &IID_IHTMLWindow3,
109 &IID_IOmNavigator
112 HRESULT get_typeinfo(tid_t tid, ITypeInfo **typeinfo)
114 HRESULT hres;
116 if(!typelib) {
117 ITypeLib *tl;
119 hres = LoadRegTypeLib(&LIBID_MSHTML, 4, 0, LOCALE_SYSTEM_DEFAULT, &tl);
120 if(FAILED(hres)) {
121 ERR("LoadRegTypeLib failed: %08x\n", hres);
122 return hres;
125 if(InterlockedCompareExchangePointer((void**)&typelib, tl, NULL))
126 ITypeLib_Release(tl);
129 if(!typeinfos[tid]) {
130 ITypeInfo *typeinfo;
132 hres = ITypeLib_GetTypeInfoOfGuid(typelib, tid_ids[tid], &typeinfo);
133 if(FAILED(hres)) {
134 ERR("GetTypeInfoOfGuid(%s) failed: %08x\n", debugstr_guid(tid_ids[tid]), hres);
135 return hres;
138 if(InterlockedCompareExchangePointer((void**)(typeinfos+tid), typeinfo, NULL))
139 ITypeInfo_Release(typeinfo);
142 *typeinfo = typeinfos[tid];
143 return S_OK;
146 void release_typelib(void)
148 dispex_data_t *iter;
149 unsigned i;
151 while(!list_empty(&dispex_data_list)) {
152 iter = LIST_ENTRY(list_head(&dispex_data_list), dispex_data_t, entry);
153 list_remove(&iter->entry);
155 for(i=0; i < iter->func_cnt; i++)
156 SysFreeString(iter->funcs[i].name);
158 heap_free(iter->funcs);
159 heap_free(iter->name_table);
160 heap_free(iter);
163 if(!typelib)
164 return;
166 for(i=0; i < sizeof(typeinfos)/sizeof(*typeinfos); i++)
167 if(typeinfos[i])
168 ITypeInfo_Release(typeinfos[i]);
170 ITypeLib_Release(typelib);
173 static void add_func_info(dispex_data_t *data, DWORD *size, tid_t tid, DISPID id, ITypeInfo *dti)
175 HRESULT hres;
177 if(data->func_cnt && data->funcs[data->func_cnt-1].id == id)
178 return;
180 if(data->func_cnt == *size)
181 data->funcs = heap_realloc(data->funcs, (*size <<= 1)*sizeof(func_info_t));
183 hres = ITypeInfo_GetDocumentation(dti, id, &data->funcs[data->func_cnt].name, NULL, NULL, NULL);
184 if(FAILED(hres))
185 return;
187 data->funcs[data->func_cnt].id = id;
188 data->funcs[data->func_cnt].tid = tid;
190 data->func_cnt++;
193 static int dispid_cmp(const void *p1, const void *p2)
195 return ((func_info_t*)p1)->id - ((func_info_t*)p2)->id;
198 static int func_name_cmp(const void *p1, const void *p2)
200 return strcmpiW((*(func_info_t**)p1)->name, (*(func_info_t**)p2)->name);
203 static dispex_data_t *preprocess_dispex_data(DispatchEx *This)
205 const tid_t *tid = This->data->iface_tids;
206 FUNCDESC *funcdesc;
207 dispex_data_t *data;
208 DWORD size = 16, i;
209 ITypeInfo *ti, *dti;
210 HRESULT hres;
212 TRACE("(%p)\n", This);
214 hres = get_typeinfo(This->data->disp_tid, &dti);
215 if(FAILED(hres)) {
216 ERR("Could not get disp type info: %08x\n", hres);
217 return NULL;
220 data = heap_alloc(sizeof(dispex_data_t));
221 data->func_cnt = 0;
222 data->funcs = heap_alloc(size*sizeof(func_info_t));
223 list_add_tail(&dispex_data_list, &data->entry);
225 while(*tid) {
226 hres = get_typeinfo(*tid, &ti);
227 if(FAILED(hres))
228 break;
230 i=7;
231 while(1) {
232 hres = ITypeInfo_GetFuncDesc(ti, i++, &funcdesc);
233 if(FAILED(hres))
234 break;
236 add_func_info(data, &size, *tid, funcdesc->memid, dti);
237 ITypeInfo_ReleaseFuncDesc(ti, funcdesc);
240 tid++;
243 if(!data->func_cnt) {
244 heap_free(data->funcs);
245 data->funcs = NULL;
246 }else if(data->func_cnt != size) {
247 data->funcs = heap_realloc(data->funcs, data->func_cnt * sizeof(func_info_t));
250 qsort(data->funcs, data->func_cnt, sizeof(func_info_t), dispid_cmp);
252 if(data->funcs) {
253 data->name_table = heap_alloc(data->func_cnt * sizeof(func_info_t*));
254 for(i=0; i < data->func_cnt; i++)
255 data->name_table[i] = data->funcs+i;
256 qsort(data->name_table, data->func_cnt, sizeof(func_info_t*), func_name_cmp);
257 }else {
258 data->name_table = NULL;
261 return data;
264 static CRITICAL_SECTION cs_dispex_static_data;
265 static CRITICAL_SECTION_DEBUG cs_dispex_static_data_dbg =
267 0, 0, &cs_dispex_static_data,
268 { &cs_dispex_static_data_dbg.ProcessLocksList, &cs_dispex_static_data_dbg.ProcessLocksList },
269 0, 0, { (DWORD_PTR)(__FILE__ ": dispex_static_data") }
271 static CRITICAL_SECTION cs_dispex_static_data = { &cs_dispex_static_data_dbg, -1, 0, 0, 0, 0 };
274 static dispex_data_t *get_dispex_data(DispatchEx *This)
276 if(This->data->data)
277 return This->data->data;
279 EnterCriticalSection(&cs_dispex_static_data);
281 if(!This->data->data)
282 This->data->data = preprocess_dispex_data(This);
284 LeaveCriticalSection(&cs_dispex_static_data);
286 return This->data->data;
289 void call_disp_func(HTMLDocument *doc, IDispatch *disp)
291 DISPID named_arg = DISPID_THIS;
292 VARIANTARG arg;
293 DISPPARAMS params = {&arg, &named_arg, 1, 1};
294 EXCEPINFO ei;
295 IDispatchEx *dispex;
296 VARIANT res;
297 HRESULT hres;
299 hres = IDispatch_QueryInterface(disp, &IID_IDispatchEx, (void**)&dispex);
300 if(FAILED(hres)) {
301 FIXME("Could not get IDispatchEx interface: %08x\n", hres);
302 return;
305 V_VT(&arg) = VT_DISPATCH;
306 V_DISPATCH(&arg) = (IDispatch*)HTMLWINDOW2(doc->window);
307 VariantInit(&res);
308 memset(&ei, 0, sizeof(ei));
310 hres = IDispatchEx_InvokeEx(dispex, 0, GetUserDefaultLCID(), DISPATCH_METHOD, &params, &res, &ei, NULL);
311 IDispatchEx_Release(dispex);
313 TRACE("%p returned %08x\n", disp, hres);
315 VariantClear(&res);
318 static inline BOOL is_custom_dispid(DISPID id)
320 return MSHTML_DISPID_CUSTOM_MIN <= id && id <= MSHTML_DISPID_CUSTOM_MAX;
323 static inline BOOL is_dynamic_dispid(DISPID id)
325 return DISPID_DYNPROP_0 <= id && id <= DISPID_DYNPROP_MAX;
328 #define DISPATCHEX_THIS(iface) DEFINE_THIS(DispatchEx, IDispatchEx, iface)
330 static HRESULT WINAPI DispatchEx_QueryInterface(IDispatchEx *iface, REFIID riid, void **ppv)
332 DispatchEx *This = DISPATCHEX_THIS(iface);
334 return IUnknown_QueryInterface(This->outer, riid, ppv);
337 static ULONG WINAPI DispatchEx_AddRef(IDispatchEx *iface)
339 DispatchEx *This = DISPATCHEX_THIS(iface);
341 return IUnknown_AddRef(This->outer);
344 static ULONG WINAPI DispatchEx_Release(IDispatchEx *iface)
346 DispatchEx *This = DISPATCHEX_THIS(iface);
348 return IUnknown_Release(This->outer);
351 static HRESULT WINAPI DispatchEx_GetTypeInfoCount(IDispatchEx *iface, UINT *pctinfo)
353 DispatchEx *This = DISPATCHEX_THIS(iface);
355 TRACE("(%p)->(%p)\n", This, pctinfo);
357 *pctinfo = 1;
358 return S_OK;
361 static HRESULT WINAPI DispatchEx_GetTypeInfo(IDispatchEx *iface, UINT iTInfo,
362 LCID lcid, ITypeInfo **ppTInfo)
364 DispatchEx *This = DISPATCHEX_THIS(iface);
365 HRESULT hres;
367 TRACE("(%p)->(%u %u %p)\n", This, iTInfo, lcid, ppTInfo);
369 hres = get_typeinfo(This->data->disp_tid, ppTInfo);
370 if(FAILED(hres))
371 return hres;
373 ITypeInfo_AddRef(*ppTInfo);
374 return S_OK;
377 static HRESULT WINAPI DispatchEx_GetIDsOfNames(IDispatchEx *iface, REFIID riid,
378 LPOLESTR *rgszNames, UINT cNames,
379 LCID lcid, DISPID *rgDispId)
381 DispatchEx *This = DISPATCHEX_THIS(iface);
382 UINT i;
383 HRESULT hres;
385 TRACE("(%p)->(%s %p %u %u %p)\n", This, debugstr_guid(riid), rgszNames, cNames,
386 lcid, rgDispId);
388 for(i=0; i < cNames; i++) {
389 hres = IDispatchEx_GetDispID(DISPATCHEX(This), rgszNames[i], 0, rgDispId+i);
390 if(FAILED(hres))
391 return hres;
394 return S_OK;
397 static HRESULT WINAPI DispatchEx_Invoke(IDispatchEx *iface, DISPID dispIdMember,
398 REFIID riid, LCID lcid, WORD wFlags, DISPPARAMS *pDispParams,
399 VARIANT *pVarResult, EXCEPINFO *pExcepInfo, UINT *puArgErr)
401 DispatchEx *This = DISPATCHEX_THIS(iface);
403 TRACE("(%p)->(%d %s %d %d %p %p %p %p)\n", This, dispIdMember, debugstr_guid(riid),
404 lcid, wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr);
406 return IDispatchEx_InvokeEx(DISPATCHEX(This), dispIdMember, lcid, wFlags,
407 pDispParams, pVarResult, pExcepInfo, NULL);
410 static HRESULT WINAPI DispatchEx_GetDispID(IDispatchEx *iface, BSTR bstrName, DWORD grfdex, DISPID *pid)
412 DispatchEx *This = DISPATCHEX_THIS(iface);
413 dispex_data_t *data;
414 int min, max, n, c;
416 TRACE("(%p)->(%s %x %p)\n", This, debugstr_w(bstrName), grfdex, pid);
418 if(grfdex & ~(fdexNameCaseSensitive|fdexNameEnsure|fdexNameImplicit))
419 FIXME("Unsupported grfdex %x\n", grfdex);
421 data = get_dispex_data(This);
422 if(!data)
423 return E_FAIL;
425 min = 0;
426 max = data->func_cnt-1;
428 while(min <= max) {
429 n = (min+max)/2;
431 c = strcmpiW(data->name_table[n]->name, bstrName);
432 if(!c) {
433 if((grfdex & fdexNameCaseSensitive) && strcmpW(data->name_table[n]->name, bstrName))
434 break;
436 *pid = data->name_table[n]->id;
437 return S_OK;
440 if(c > 0)
441 max = n-1;
442 else
443 min = n+1;
446 if(This->dynamic_data) {
447 unsigned i;
449 for(i=0; i < This->dynamic_data->prop_cnt; i++) {
450 if(!strcmpW(This->dynamic_data->props[i].name, bstrName)) {
451 *pid = DISPID_DYNPROP_0 + i;
452 return S_OK;
457 if(This->data->vtbl && This->data->vtbl->get_dispid) {
458 HRESULT hres;
460 hres = This->data->vtbl->get_dispid(This->outer, bstrName, grfdex, pid);
461 if(hres != DISP_E_UNKNOWNNAME)
462 return hres;
465 if(grfdex & fdexNameEnsure) {
466 TRACE("creating dynamic prop %s\n", debugstr_w(bstrName));
468 if(!This->dynamic_data) {
469 This->dynamic_data = heap_alloc_zero(sizeof(dispex_dynamic_data_t));
470 This->dynamic_data->props = heap_alloc(This->dynamic_data->buf_size = 4);
471 }else if(This->dynamic_data->buf_size == This->dynamic_data->prop_cnt) {
472 This->dynamic_data->props = heap_realloc(This->dynamic_data->props, This->dynamic_data->buf_size<<=1);
475 This->dynamic_data->props[This->dynamic_data->prop_cnt].name = heap_strdupW(bstrName);
476 VariantInit(&This->dynamic_data->props[This->dynamic_data->prop_cnt].var);
477 *pid = DISPID_DYNPROP_0 + This->dynamic_data->prop_cnt++;
479 return S_OK;
482 TRACE("not found %s\n", debugstr_w(bstrName));
483 return DISP_E_UNKNOWNNAME;
486 static HRESULT WINAPI DispatchEx_InvokeEx(IDispatchEx *iface, DISPID id, LCID lcid, WORD wFlags, DISPPARAMS *pdp,
487 VARIANT *pvarRes, EXCEPINFO *pei, IServiceProvider *pspCaller)
489 DispatchEx *This = DISPATCHEX_THIS(iface);
490 IUnknown *unk;
491 ITypeInfo *ti;
492 dispex_data_t *data;
493 UINT argerr=0;
494 int min, max, n;
495 HRESULT hres;
497 TRACE("(%p)->(%x %x %x %p %p %p %p)\n", This, id, lcid, wFlags, pdp, pvarRes, pei, pspCaller);
499 if(is_custom_dispid(id) && This->data->vtbl && This->data->vtbl->invoke)
500 return This->data->vtbl->invoke(This->outer, id, lcid, wFlags, pdp, pvarRes, pei, pspCaller);
502 if(wFlags == DISPATCH_CONSTRUCT) {
503 FIXME("DISPATCH_CONSTRUCT not implemented\n");
504 return E_NOTIMPL;
507 if(is_dynamic_dispid(id)) {
508 DWORD idx = id - DISPID_DYNPROP_0;
509 VARIANT *var;
511 if(!This->dynamic_data || This->dynamic_data->prop_cnt <= idx)
512 return DISP_E_UNKNOWNNAME;
514 var = &This->dynamic_data->props[idx].var;
516 switch(wFlags) {
517 case INVOKE_PROPERTYGET:
518 return VariantCopy(pvarRes, var);
519 case INVOKE_PROPERTYPUT:
520 VariantClear(var);
521 return VariantCopy(var, pdp->rgvarg);
522 default:
523 FIXME("unhandled wFlags %x\n", wFlags);
524 return E_NOTIMPL;
528 data = get_dispex_data(This);
529 if(!data)
530 return E_FAIL;
532 min = 0;
533 max = data->func_cnt-1;
535 while(min <= max) {
536 n = (min+max)/2;
538 if(data->funcs[n].id == id)
539 break;
541 if(data->funcs[n].id < id)
542 min = n+1;
543 else
544 max = n-1;
547 if(min > max) {
548 WARN("invalid id %x\n", id);
549 return DISP_E_UNKNOWNNAME;
552 hres = get_typeinfo(data->funcs[n].tid, &ti);
553 if(FAILED(hres)) {
554 ERR("Could not get type info: %08x\n", hres);
555 return hres;
558 hres = IUnknown_QueryInterface(This->outer, tid_ids[data->funcs[n].tid], (void**)&unk);
559 if(FAILED(hres)) {
560 ERR("Could not get iface: %08x\n", hres);
561 return E_FAIL;
564 hres = ITypeInfo_Invoke(ti, unk, id, wFlags, pdp, pvarRes, pei, &argerr);
566 IUnknown_Release(unk);
567 return hres;
570 static HRESULT WINAPI DispatchEx_DeleteMemberByName(IDispatchEx *iface, BSTR bstrName, DWORD grfdex)
572 DispatchEx *This = DISPATCHEX_THIS(iface);
573 FIXME("(%p)->(%s %x)\n", This, debugstr_w(bstrName), grfdex);
574 return E_NOTIMPL;
577 static HRESULT WINAPI DispatchEx_DeleteMemberByDispID(IDispatchEx *iface, DISPID id)
579 DispatchEx *This = DISPATCHEX_THIS(iface);
580 FIXME("(%p)->(%x)\n", This, id);
581 return E_NOTIMPL;
584 static HRESULT WINAPI DispatchEx_GetMemberProperties(IDispatchEx *iface, DISPID id, DWORD grfdexFetch, DWORD *pgrfdex)
586 DispatchEx *This = DISPATCHEX_THIS(iface);
587 FIXME("(%p)->(%x %x %p)\n", This, id, grfdexFetch, pgrfdex);
588 return E_NOTIMPL;
591 static HRESULT WINAPI DispatchEx_GetMemberName(IDispatchEx *iface, DISPID id, BSTR *pbstrName)
593 DispatchEx *This = DISPATCHEX_THIS(iface);
594 FIXME("(%p)->(%x %p)\n", This, id, pbstrName);
595 return E_NOTIMPL;
598 static HRESULT WINAPI DispatchEx_GetNextDispID(IDispatchEx *iface, DWORD grfdex, DISPID id, DISPID *pid)
600 DispatchEx *This = DISPATCHEX_THIS(iface);
601 FIXME("(%p)->(%x %x %p)\n", This, grfdex, id, pid);
602 return E_NOTIMPL;
605 static HRESULT WINAPI DispatchEx_GetNameSpaceParent(IDispatchEx *iface, IUnknown **ppunk)
607 DispatchEx *This = DISPATCHEX_THIS(iface);
608 FIXME("(%p)->(%p)\n", This, ppunk);
609 return E_NOTIMPL;
612 #undef DISPATCHEX_THIS
614 static IDispatchExVtbl DispatchExVtbl = {
615 DispatchEx_QueryInterface,
616 DispatchEx_AddRef,
617 DispatchEx_Release,
618 DispatchEx_GetTypeInfoCount,
619 DispatchEx_GetTypeInfo,
620 DispatchEx_GetIDsOfNames,
621 DispatchEx_Invoke,
622 DispatchEx_GetDispID,
623 DispatchEx_InvokeEx,
624 DispatchEx_DeleteMemberByName,
625 DispatchEx_DeleteMemberByDispID,
626 DispatchEx_GetMemberProperties,
627 DispatchEx_GetMemberName,
628 DispatchEx_GetNextDispID,
629 DispatchEx_GetNameSpaceParent
632 BOOL dispex_query_interface(DispatchEx *This, REFIID riid, void **ppv)
634 static const IID IID_UndocumentedScriptIface =
635 {0x719c3050,0xf9d3,0x11cf,{0xa4,0x93,0x00,0x40,0x05,0x23,0xa8,0xa0}};
637 if(IsEqualGUID(&IID_IDispatch, riid)) {
638 TRACE("(%p)->(IID_IDispatch %p)\n", This, ppv);
639 *ppv = DISPATCHEX(This);
640 }else if(IsEqualGUID(&IID_IDispatchEx, riid)) {
641 TRACE("(%p)->(IID_IDispatchEx %p)\n", This, ppv);
642 *ppv = DISPATCHEX(This);
643 }else if(IsEqualGUID(&IID_UndocumentedScriptIface, riid)) {
644 TRACE("(%p)->(IID_UndocumentedScriptIface %p) returning NULL\n", This, ppv);
645 *ppv = NULL;
646 }else {
647 return FALSE;
650 if(*ppv)
651 IUnknown_AddRef((IUnknown*)*ppv);
652 return TRUE;
655 void init_dispex(DispatchEx *dispex, IUnknown *outer, dispex_static_data_t *data)
657 dispex->lpIDispatchExVtbl = &DispatchExVtbl;
658 dispex->outer = outer;
659 dispex->data = data;