msvcrt: ungetc must create a buffer, even if _IONBF has been set with setvbuf.
[wine/multimedia.git] / dlls / jscript / dispex.c
blob653857cb39763bfd3d990e5239554ca27a738e57
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 "jscript.h"
21 #include "wine/unicode.h"
22 #include "wine/debug.h"
24 WINE_DEFAULT_DEBUG_CHANNEL(jscript);
27 * This IID is used to get jsdisp_t objecto from interface.
28 * We might consider using private insteface instead.
30 static const IID IID_IDispatchJS =
31 {0x719c3050,0xf9d3,0x11cf,{0xa4,0x93,0x00,0x40,0x05,0x23,0xa8,0xa6}};
33 #define FDEX_VERSION_MASK 0xf0000000
35 typedef enum {
36 PROP_VARIANT,
37 PROP_BUILTIN,
38 PROP_PROTREF,
39 PROP_DELETED
40 } prop_type_t;
42 struct _dispex_prop_t {
43 WCHAR *name;
44 prop_type_t type;
45 DWORD flags;
47 union {
48 VARIANT var;
49 const builtin_prop_t *p;
50 DWORD ref;
51 } u;
54 static inline DISPID prop_to_id(jsdisp_t *This, dispex_prop_t *prop)
56 return prop - This->props;
59 static inline dispex_prop_t *get_prop(jsdisp_t *This, DISPID id)
61 if(id < 0 || id >= This->prop_cnt || This->props[id].type == PROP_DELETED)
62 return NULL;
64 return This->props+id;
67 static DWORD get_flags(jsdisp_t *This, dispex_prop_t *prop)
69 if(prop->type == PROP_PROTREF) {
70 dispex_prop_t *parent = get_prop(This->prototype, prop->u.ref);
71 if(!parent) {
72 prop->type = PROP_DELETED;
73 return 0;
76 return get_flags(This->prototype, parent);
79 return prop->flags;
82 static const builtin_prop_t *find_builtin_prop(jsdisp_t *This, const WCHAR *name)
84 int min = 0, max, i, r;
86 max = This->builtin_info->props_cnt-1;
87 while(min <= max) {
88 i = (min+max)/2;
90 r = strcmpW(name, This->builtin_info->props[i].name);
91 if(!r)
92 return This->builtin_info->props + i;
94 if(r < 0)
95 max = i-1;
96 else
97 min = i+1;
100 return NULL;
103 static dispex_prop_t *alloc_prop(jsdisp_t *This, const WCHAR *name, prop_type_t type, DWORD flags)
105 dispex_prop_t *ret;
107 if(This->buf_size == This->prop_cnt) {
108 dispex_prop_t *tmp = heap_realloc(This->props, (This->buf_size<<=1)*sizeof(*This->props));
109 if(!tmp)
110 return NULL;
111 This->props = tmp;
114 ret = This->props + This->prop_cnt++;
115 ret->type = type;
116 ret->flags = flags;
117 ret->name = heap_strdupW(name);
118 if(!ret->name)
119 return NULL;
121 return ret;
124 static dispex_prop_t *alloc_protref(jsdisp_t *This, const WCHAR *name, DWORD ref)
126 dispex_prop_t *ret;
128 ret = alloc_prop(This, name, PROP_PROTREF, 0);
129 if(!ret)
130 return NULL;
132 ret->u.ref = ref;
133 return ret;
136 static HRESULT find_prop_name(jsdisp_t *This, const WCHAR *name, dispex_prop_t **ret)
138 const builtin_prop_t *builtin;
139 dispex_prop_t *prop;
141 for(prop = This->props; prop < This->props+This->prop_cnt; prop++) {
142 if(prop->name && !strcmpW(prop->name, name)) {
143 *ret = prop;
144 return S_OK;
148 builtin = find_builtin_prop(This, name);
149 if(builtin) {
150 prop = alloc_prop(This, name, PROP_BUILTIN, builtin->flags);
151 if(!prop)
152 return E_OUTOFMEMORY;
154 prop->u.p = builtin;
155 *ret = prop;
156 return S_OK;
159 *ret = NULL;
160 return S_OK;
163 static HRESULT find_prop_name_prot(jsdisp_t *This, const WCHAR *name, dispex_prop_t **ret)
165 dispex_prop_t *prop;
166 HRESULT hres;
168 hres = find_prop_name(This, name, &prop);
169 if(FAILED(hres))
170 return hres;
171 if(prop) {
172 *ret = prop;
173 return S_OK;
176 if(This->prototype) {
177 hres = find_prop_name_prot(This->prototype, name, &prop);
178 if(FAILED(hres))
179 return hres;
180 if(prop) {
181 prop = alloc_protref(This, prop->name, prop - This->prototype->props);
182 if(!prop)
183 return E_OUTOFMEMORY;
184 *ret = prop;
185 return S_OK;
189 *ret = prop;
190 return S_OK;
193 static HRESULT ensure_prop_name(jsdisp_t *This, const WCHAR *name, BOOL search_prot, DWORD create_flags, dispex_prop_t **ret)
195 dispex_prop_t *prop;
196 HRESULT hres;
198 if(search_prot)
199 hres = find_prop_name_prot(This, name, &prop);
200 else
201 hres = find_prop_name(This, name, &prop);
202 if(SUCCEEDED(hres) && !prop) {
203 TRACE("creating prop %s\n", debugstr_w(name));
205 prop = alloc_prop(This, name, PROP_VARIANT, create_flags);
206 if(!prop)
207 return E_OUTOFMEMORY;
208 VariantInit(&prop->u.var);
211 *ret = prop;
212 return hres;
215 static HRESULT set_this(DISPPARAMS *dp, DISPPARAMS *olddp, IDispatch *jsthis)
217 VARIANTARG *oldargs;
218 int i;
220 static DISPID this_id = DISPID_THIS;
222 *dp = *olddp;
224 for(i = 0; i < dp->cNamedArgs; i++) {
225 if(dp->rgdispidNamedArgs[i] == DISPID_THIS)
226 return S_OK;
229 oldargs = dp->rgvarg;
230 dp->rgvarg = heap_alloc((dp->cArgs+1) * sizeof(VARIANTARG));
231 if(!dp->rgvarg)
232 return E_OUTOFMEMORY;
233 memcpy(dp->rgvarg+1, oldargs, dp->cArgs*sizeof(VARIANTARG));
234 V_VT(dp->rgvarg) = VT_DISPATCH;
235 V_DISPATCH(dp->rgvarg) = jsthis;
236 dp->cArgs++;
238 if(dp->cNamedArgs) {
239 DISPID *old = dp->rgdispidNamedArgs;
240 dp->rgdispidNamedArgs = heap_alloc((dp->cNamedArgs+1)*sizeof(DISPID));
241 if(!dp->rgdispidNamedArgs) {
242 heap_free(dp->rgvarg);
243 return E_OUTOFMEMORY;
246 memcpy(dp->rgdispidNamedArgs+1, old, dp->cNamedArgs*sizeof(DISPID));
247 dp->rgdispidNamedArgs[0] = DISPID_THIS;
248 dp->cNamedArgs++;
249 }else {
250 dp->rgdispidNamedArgs = &this_id;
251 dp->cNamedArgs = 1;
254 return S_OK;
257 static HRESULT invoke_prop_func(jsdisp_t *This, jsdisp_t *jsthis, dispex_prop_t *prop, WORD flags,
258 DISPPARAMS *dp, VARIANT *retv, jsexcept_t *ei, IServiceProvider *caller)
260 HRESULT hres;
262 switch(prop->type) {
263 case PROP_BUILTIN: {
264 vdisp_t vthis;
266 if(flags == DISPATCH_CONSTRUCT && (prop->flags & PROPF_METHOD)) {
267 WARN("%s is not a constructor\n", debugstr_w(prop->name));
268 return E_INVALIDARG;
271 set_jsdisp(&vthis, jsthis);
272 hres = prop->u.p->invoke(This->ctx, &vthis, flags, dp, retv, ei, caller);
273 vdisp_release(&vthis);
274 return hres;
276 case PROP_PROTREF:
277 return invoke_prop_func(This->prototype, jsthis, This->prototype->props+prop->u.ref, flags, dp, retv, ei, caller);
278 case PROP_VARIANT: {
279 DISPPARAMS new_dp;
281 if(V_VT(&prop->u.var) != VT_DISPATCH) {
282 FIXME("invoke vt %d\n", V_VT(&prop->u.var));
283 return E_FAIL;
286 TRACE("call %s %p\n", debugstr_w(prop->name), V_DISPATCH(&prop->u.var));
288 hres = set_this(&new_dp, dp, to_disp(jsthis));
289 if(FAILED(hres))
290 return hres;
292 hres = disp_call(This->ctx, V_DISPATCH(&prop->u.var), DISPID_VALUE, flags, &new_dp, retv, ei, caller);
294 if(new_dp.rgvarg != dp->rgvarg) {
295 heap_free(new_dp.rgvarg);
296 if(new_dp.cNamedArgs > 1)
297 heap_free(new_dp.rgdispidNamedArgs);
300 return hres;
302 default:
303 ERR("type %d\n", prop->type);
306 return E_FAIL;
309 static HRESULT prop_get(jsdisp_t *This, dispex_prop_t *prop, DISPPARAMS *dp,
310 VARIANT *retv, jsexcept_t *ei, IServiceProvider *caller)
312 HRESULT hres;
314 switch(prop->type) {
315 case PROP_BUILTIN:
316 if(prop->u.p->flags & PROPF_METHOD) {
317 jsdisp_t *obj;
318 hres = create_builtin_function(This->ctx, prop->u.p->invoke, prop->u.p->name, NULL,
319 prop->u.p->flags, NULL, &obj);
320 if(FAILED(hres))
321 break;
323 prop->type = PROP_VARIANT;
324 var_set_jsdisp(&prop->u.var, obj);
325 hres = VariantCopy(retv, &prop->u.var);
326 }else {
327 vdisp_t vthis;
329 set_jsdisp(&vthis, This);
330 hres = prop->u.p->invoke(This->ctx, &vthis, DISPATCH_PROPERTYGET, dp, retv, ei, caller);
331 vdisp_release(&vthis);
333 break;
334 case PROP_PROTREF:
335 hres = prop_get(This->prototype, This->prototype->props+prop->u.ref, dp, retv, ei, caller);
336 break;
337 case PROP_VARIANT:
338 hres = VariantCopy(retv, &prop->u.var);
339 break;
340 default:
341 ERR("type %d\n", prop->type);
342 return E_FAIL;
345 if(FAILED(hres)) {
346 TRACE("fail %08x\n", hres);
347 return hres;
350 TRACE("%s ret %s\n", debugstr_w(prop->name), debugstr_variant(retv));
351 return hres;
354 static HRESULT prop_put(jsdisp_t *This, dispex_prop_t *prop, VARIANT *val,
355 jsexcept_t *ei, IServiceProvider *caller)
357 HRESULT hres;
359 if(prop->flags & PROPF_CONST)
360 return S_OK;
362 switch(prop->type) {
363 case PROP_BUILTIN:
364 if(!(prop->flags & PROPF_METHOD)) {
365 DISPPARAMS dp = {val, NULL, 1, 0};
366 vdisp_t vthis;
368 set_jsdisp(&vthis, This);
369 hres = prop->u.p->invoke(This->ctx, &vthis, DISPATCH_PROPERTYPUT, &dp, NULL, ei, caller);
370 vdisp_release(&vthis);
371 return hres;
373 case PROP_PROTREF:
374 prop->type = PROP_VARIANT;
375 prop->flags = PROPF_ENUM;
376 V_VT(&prop->u.var) = VT_EMPTY;
377 break;
378 case PROP_VARIANT:
379 VariantClear(&prop->u.var);
380 break;
381 default:
382 ERR("type %d\n", prop->type);
383 return E_FAIL;
386 hres = VariantCopy(&prop->u.var, val);
387 if(FAILED(hres))
388 return hres;
390 if(This->builtin_info->on_put)
391 This->builtin_info->on_put(This, prop->name);
393 TRACE("%s = %s\n", debugstr_w(prop->name), debugstr_variant(val));
394 return S_OK;
397 static HRESULT fill_protrefs(jsdisp_t *This)
399 dispex_prop_t *iter, *prop;
400 HRESULT hres;
402 if(!This->prototype)
403 return S_OK;
405 fill_protrefs(This->prototype);
407 for(iter = This->prototype->props; iter < This->prototype->props+This->prototype->prop_cnt; iter++) {
408 if(!iter->name)
409 continue;
410 hres = find_prop_name(This, iter->name, &prop);
411 if(FAILED(hres))
412 return hres;
413 if(!prop) {
414 prop = alloc_protref(This, iter->name, iter - This->prototype->props);
415 if(!prop)
416 return E_OUTOFMEMORY;
420 return S_OK;
423 static inline jsdisp_t *impl_from_IDispatchEx(IDispatchEx *iface)
425 return CONTAINING_RECORD(iface, jsdisp_t, IDispatchEx_iface);
428 static HRESULT WINAPI DispatchEx_QueryInterface(IDispatchEx *iface, REFIID riid, void **ppv)
430 jsdisp_t *This = impl_from_IDispatchEx(iface);
432 if(IsEqualGUID(&IID_IUnknown, riid)) {
433 TRACE("(%p)->(IID_IUnknown %p)\n", This, ppv);
434 *ppv = &This->IDispatchEx_iface;
435 }else if(IsEqualGUID(&IID_IDispatch, riid)) {
436 TRACE("(%p)->(IID_IDispatch %p)\n", This, ppv);
437 *ppv = &This->IDispatchEx_iface;
438 }else if(IsEqualGUID(&IID_IDispatchEx, riid)) {
439 TRACE("(%p)->(IID_IDispatchEx %p)\n", This, ppv);
440 *ppv = &This->IDispatchEx_iface;
441 }else if(IsEqualGUID(&IID_IDispatchJS, riid)) {
442 TRACE("(%p)->(IID_IDispatchJS %p)\n", This, ppv);
443 jsdisp_addref(This);
444 *ppv = This;
445 return S_OK;
446 }else {
447 WARN("(%p)->(%s %p)\n", This, debugstr_guid(riid), ppv);
448 *ppv = NULL;
449 return E_NOINTERFACE;
452 IUnknown_AddRef((IUnknown*)*ppv);
453 return S_OK;
456 static ULONG WINAPI DispatchEx_AddRef(IDispatchEx *iface)
458 jsdisp_t *This = impl_from_IDispatchEx(iface);
459 LONG ref = InterlockedIncrement(&This->ref);
461 TRACE("(%p) ref=%d\n", This, ref);
463 return ref;
466 static ULONG WINAPI DispatchEx_Release(IDispatchEx *iface)
468 jsdisp_t *This = impl_from_IDispatchEx(iface);
469 LONG ref = InterlockedDecrement(&This->ref);
471 TRACE("(%p) ref=%d\n", This, ref);
473 if(!ref) {
474 dispex_prop_t *prop;
476 for(prop = This->props; prop < This->props+This->prop_cnt; prop++) {
477 if(prop->type == PROP_VARIANT)
478 VariantClear(&prop->u.var);
479 heap_free(prop->name);
481 heap_free(This->props);
482 script_release(This->ctx);
483 if(This->prototype)
484 jsdisp_release(This->prototype);
486 if(This->builtin_info->destructor)
487 This->builtin_info->destructor(This);
488 else
489 heap_free(This);
492 return ref;
495 static HRESULT WINAPI DispatchEx_GetTypeInfoCount(IDispatchEx *iface, UINT *pctinfo)
497 jsdisp_t *This = impl_from_IDispatchEx(iface);
499 TRACE("(%p)->(%p)\n", This, pctinfo);
501 *pctinfo = 1;
502 return S_OK;
505 static HRESULT WINAPI DispatchEx_GetTypeInfo(IDispatchEx *iface, UINT iTInfo, LCID lcid,
506 ITypeInfo **ppTInfo)
508 jsdisp_t *This = impl_from_IDispatchEx(iface);
509 FIXME("(%p)->(%u %u %p)\n", This, iTInfo, lcid, ppTInfo);
510 return E_NOTIMPL;
513 static HRESULT WINAPI DispatchEx_GetIDsOfNames(IDispatchEx *iface, REFIID riid,
514 LPOLESTR *rgszNames, UINT cNames, LCID lcid,
515 DISPID *rgDispId)
517 jsdisp_t *This = impl_from_IDispatchEx(iface);
518 UINT i;
519 HRESULT hres;
521 TRACE("(%p)->(%s %p %u %u %p)\n", This, debugstr_guid(riid), rgszNames, cNames,
522 lcid, rgDispId);
524 for(i=0; i < cNames; i++) {
525 hres = IDispatchEx_GetDispID(&This->IDispatchEx_iface, rgszNames[i], 0, rgDispId+i);
526 if(FAILED(hres))
527 return hres;
530 return S_OK;
533 static HRESULT WINAPI DispatchEx_Invoke(IDispatchEx *iface, DISPID dispIdMember,
534 REFIID riid, LCID lcid, WORD wFlags, DISPPARAMS *pDispParams,
535 VARIANT *pVarResult, EXCEPINFO *pExcepInfo, UINT *puArgErr)
537 jsdisp_t *This = impl_from_IDispatchEx(iface);
539 TRACE("(%p)->(%d %s %d %d %p %p %p %p)\n", This, dispIdMember, debugstr_guid(riid),
540 lcid, wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr);
542 return IDispatchEx_InvokeEx(&This->IDispatchEx_iface, dispIdMember, lcid, wFlags,
543 pDispParams, pVarResult, pExcepInfo, NULL);
546 static HRESULT WINAPI DispatchEx_GetDispID(IDispatchEx *iface, BSTR bstrName, DWORD grfdex, DISPID *pid)
548 jsdisp_t *This = impl_from_IDispatchEx(iface);
550 TRACE("(%p)->(%s %x %p)\n", This, debugstr_w(bstrName), grfdex, pid);
552 if(grfdex & ~(fdexNameCaseSensitive|fdexNameEnsure|fdexNameImplicit|FDEX_VERSION_MASK)) {
553 FIXME("Unsupported grfdex %x\n", grfdex);
554 return E_NOTIMPL;
557 return jsdisp_get_id(This, bstrName, grfdex, pid);
560 static HRESULT WINAPI DispatchEx_InvokeEx(IDispatchEx *iface, DISPID id, LCID lcid, WORD wFlags, DISPPARAMS *pdp,
561 VARIANT *pvarRes, EXCEPINFO *pei, IServiceProvider *pspCaller)
563 jsdisp_t *This = impl_from_IDispatchEx(iface);
564 dispex_prop_t *prop;
565 jsexcept_t jsexcept;
566 HRESULT hres;
568 TRACE("(%p)->(%x %x %x %p %p %p %p)\n", This, id, lcid, wFlags, pdp, pvarRes, pei, pspCaller);
570 if(pvarRes)
571 V_VT(pvarRes) = VT_EMPTY;
573 prop = get_prop(This, id);
574 if(!prop || prop->type == PROP_DELETED) {
575 TRACE("invalid id\n");
576 return DISP_E_MEMBERNOTFOUND;
579 memset(&jsexcept, 0, sizeof(jsexcept));
581 switch(wFlags) {
582 case DISPATCH_METHOD|DISPATCH_PROPERTYGET:
583 wFlags = DISPATCH_METHOD;
584 case DISPATCH_METHOD:
585 case DISPATCH_CONSTRUCT:
586 hres = invoke_prop_func(This, This, prop, wFlags, pdp, pvarRes, &jsexcept, pspCaller);
587 break;
588 case DISPATCH_PROPERTYGET:
589 hres = prop_get(This, prop, pdp, pvarRes, &jsexcept, pspCaller);
590 break;
591 case DISPATCH_PROPERTYPUT: {
592 DWORD i;
594 for(i=0; i < pdp->cNamedArgs; i++) {
595 if(pdp->rgdispidNamedArgs[i] == DISPID_PROPERTYPUT)
596 break;
599 if(i == pdp->cNamedArgs) {
600 TRACE("no value to set\n");
601 return DISP_E_PARAMNOTOPTIONAL;
604 hres = prop_put(This, prop, pdp->rgvarg+i, &jsexcept, pspCaller);
605 break;
607 default:
608 FIXME("Unimplemented flags %x\n", wFlags);
609 return E_INVALIDARG;
612 if(pei)
613 *pei = jsexcept.ei;
615 return hres;
618 static HRESULT delete_prop(dispex_prop_t *prop)
620 heap_free(prop->name);
621 prop->name = NULL;
622 prop->type = PROP_DELETED;
624 return S_OK;
627 static HRESULT WINAPI DispatchEx_DeleteMemberByName(IDispatchEx *iface, BSTR bstrName, DWORD grfdex)
629 jsdisp_t *This = impl_from_IDispatchEx(iface);
630 dispex_prop_t *prop;
631 HRESULT hres;
633 TRACE("(%p)->(%s %x)\n", This, debugstr_w(bstrName), grfdex);
635 if(grfdex & ~(fdexNameCaseSensitive|fdexNameEnsure|fdexNameImplicit|FDEX_VERSION_MASK))
636 FIXME("Unsupported grfdex %x\n", grfdex);
638 hres = find_prop_name(This, bstrName, &prop);
639 if(FAILED(hres))
640 return hres;
641 if(!prop) {
642 TRACE("not found\n");
643 return S_OK;
646 return delete_prop(prop);
649 static HRESULT WINAPI DispatchEx_DeleteMemberByDispID(IDispatchEx *iface, DISPID id)
651 jsdisp_t *This = impl_from_IDispatchEx(iface);
652 dispex_prop_t *prop;
654 TRACE("(%p)->(%x)\n", This, id);
656 prop = get_prop(This, id);
657 if(!prop) {
658 WARN("invalid id\n");
659 return DISP_E_MEMBERNOTFOUND;
662 return delete_prop(prop);
665 static HRESULT WINAPI DispatchEx_GetMemberProperties(IDispatchEx *iface, DISPID id, DWORD grfdexFetch, DWORD *pgrfdex)
667 jsdisp_t *This = impl_from_IDispatchEx(iface);
668 FIXME("(%p)->(%x %x %p)\n", This, id, grfdexFetch, pgrfdex);
669 return E_NOTIMPL;
672 static HRESULT WINAPI DispatchEx_GetMemberName(IDispatchEx *iface, DISPID id, BSTR *pbstrName)
674 jsdisp_t *This = impl_from_IDispatchEx(iface);
675 dispex_prop_t *prop;
677 TRACE("(%p)->(%x %p)\n", This, id, pbstrName);
679 prop = get_prop(This, id);
680 if(!prop || !prop->name || prop->type == PROP_DELETED)
681 return DISP_E_MEMBERNOTFOUND;
683 *pbstrName = SysAllocString(prop->name);
684 if(!*pbstrName)
685 return E_OUTOFMEMORY;
687 return S_OK;
690 static HRESULT WINAPI DispatchEx_GetNextDispID(IDispatchEx *iface, DWORD grfdex, DISPID id, DISPID *pid)
692 jsdisp_t *This = impl_from_IDispatchEx(iface);
693 dispex_prop_t *iter;
694 HRESULT hres;
696 TRACE("(%p)->(%x %x %p)\n", This, grfdex, id, pid);
698 if(id == DISPID_STARTENUM) {
699 hres = fill_protrefs(This);
700 if(FAILED(hres))
701 return hres;
704 iter = get_prop(This, id+1);
705 if(!iter) {
706 *pid = DISPID_STARTENUM;
707 return S_FALSE;
710 while(iter < This->props + This->prop_cnt) {
711 if(iter->name && (get_flags(This, iter) & PROPF_ENUM)) {
712 *pid = prop_to_id(This, iter);
713 return S_OK;
715 iter++;
718 *pid = DISPID_STARTENUM;
719 return S_FALSE;
722 static HRESULT WINAPI DispatchEx_GetNameSpaceParent(IDispatchEx *iface, IUnknown **ppunk)
724 jsdisp_t *This = impl_from_IDispatchEx(iface);
725 FIXME("(%p)->(%p)\n", This, ppunk);
726 return E_NOTIMPL;
729 static IDispatchExVtbl DispatchExVtbl = {
730 DispatchEx_QueryInterface,
731 DispatchEx_AddRef,
732 DispatchEx_Release,
733 DispatchEx_GetTypeInfoCount,
734 DispatchEx_GetTypeInfo,
735 DispatchEx_GetIDsOfNames,
736 DispatchEx_Invoke,
737 DispatchEx_GetDispID,
738 DispatchEx_InvokeEx,
739 DispatchEx_DeleteMemberByName,
740 DispatchEx_DeleteMemberByDispID,
741 DispatchEx_GetMemberProperties,
742 DispatchEx_GetMemberName,
743 DispatchEx_GetNextDispID,
744 DispatchEx_GetNameSpaceParent
747 HRESULT init_dispex(jsdisp_t *dispex, script_ctx_t *ctx, const builtin_info_t *builtin_info, jsdisp_t *prototype)
749 TRACE("%p (%p)\n", dispex, prototype);
751 dispex->IDispatchEx_iface.lpVtbl = &DispatchExVtbl;
752 dispex->ref = 1;
753 dispex->builtin_info = builtin_info;
755 dispex->props = heap_alloc((dispex->buf_size=4) * sizeof(dispex_prop_t));
756 if(!dispex->props)
757 return E_OUTOFMEMORY;
759 dispex->prototype = prototype;
760 if(prototype)
761 jsdisp_addref(prototype);
763 dispex->prop_cnt = 1;
764 dispex->props[0].name = NULL;
765 dispex->props[0].flags = 0;
766 if(builtin_info->value_prop.invoke) {
767 dispex->props[0].type = PROP_BUILTIN;
768 dispex->props[0].u.p = &builtin_info->value_prop;
769 }else {
770 dispex->props[0].type = PROP_DELETED;
773 script_addref(ctx);
774 dispex->ctx = ctx;
776 return S_OK;
779 static const builtin_info_t dispex_info = {
780 JSCLASS_NONE,
781 {NULL, NULL, 0},
782 0, NULL,
783 NULL,
784 NULL
787 HRESULT create_dispex(script_ctx_t *ctx, const builtin_info_t *builtin_info, jsdisp_t *prototype, jsdisp_t **dispex)
789 jsdisp_t *ret;
790 HRESULT hres;
792 ret = heap_alloc_zero(sizeof(jsdisp_t));
793 if(!ret)
794 return E_OUTOFMEMORY;
796 hres = init_dispex(ret, ctx, builtin_info ? builtin_info : &dispex_info, prototype);
797 if(FAILED(hres))
798 return hres;
800 *dispex = ret;
801 return S_OK;
804 HRESULT init_dispex_from_constr(jsdisp_t *dispex, script_ctx_t *ctx, const builtin_info_t *builtin_info, jsdisp_t *constr)
806 jsdisp_t *prot = NULL;
807 dispex_prop_t *prop;
808 HRESULT hres;
810 static const WCHAR constructorW[] = {'c','o','n','s','t','r','u','c','t','o','r',0};
811 static const WCHAR prototypeW[] = {'p','r','o','t','o','t','y','p','e',0};
813 hres = find_prop_name_prot(constr, prototypeW, &prop);
814 if(SUCCEEDED(hres) && prop) {
815 jsexcept_t jsexcept;
816 VARIANT var;
818 V_VT(&var) = VT_EMPTY;
819 memset(&jsexcept, 0, sizeof(jsexcept));
820 hres = prop_get(constr, prop, NULL, &var, &jsexcept, NULL/*FIXME*/);
821 if(FAILED(hres)) {
822 ERR("Could not get prototype\n");
823 return hres;
826 if(V_VT(&var) == VT_DISPATCH)
827 prot = iface_to_jsdisp((IUnknown*)V_DISPATCH(&var));
828 VariantClear(&var);
831 hres = init_dispex(dispex, ctx, builtin_info, prot);
833 if(prot)
834 jsdisp_release(prot);
835 if(FAILED(hres))
836 return hres;
838 hres = ensure_prop_name(dispex, constructorW, FALSE, 0, &prop);
839 if(SUCCEEDED(hres)) {
840 jsexcept_t jsexcept;
841 VARIANT var;
843 var_set_jsdisp(&var, constr);
844 memset(&jsexcept, 0, sizeof(jsexcept));
845 hres = prop_put(dispex, prop, &var, &jsexcept, NULL/*FIXME*/);
847 if(FAILED(hres))
848 jsdisp_release(dispex);
850 return hres;
853 jsdisp_t *iface_to_jsdisp(IUnknown *iface)
855 jsdisp_t *ret;
856 HRESULT hres;
858 hres = IUnknown_QueryInterface(iface, &IID_IDispatchJS, (void**)&ret);
859 if(FAILED(hres))
860 return NULL;
862 return ret;
865 HRESULT jsdisp_get_id(jsdisp_t *jsdisp, const WCHAR *name, DWORD flags, DISPID *id)
867 dispex_prop_t *prop;
868 HRESULT hres;
870 if(flags & fdexNameEnsure)
871 hres = ensure_prop_name(jsdisp, name, TRUE, PROPF_ENUM, &prop);
872 else
873 hres = find_prop_name_prot(jsdisp, name, &prop);
874 if(FAILED(hres))
875 return hres;
877 if(prop) {
878 *id = prop_to_id(jsdisp, prop);
879 return S_OK;
882 TRACE("not found %s\n", debugstr_w(name));
883 return DISP_E_UNKNOWNNAME;
886 HRESULT jsdisp_call_value(jsdisp_t *jsthis, WORD flags, DISPPARAMS *dp, VARIANT *retv,
887 jsexcept_t *ei, IServiceProvider *caller)
889 vdisp_t vdisp;
890 HRESULT hres;
892 set_jsdisp(&vdisp, jsthis);
893 hres = jsthis->builtin_info->value_prop.invoke(jsthis->ctx, &vdisp, flags, dp, retv, ei, caller);
894 vdisp_release(&vdisp);
895 return hres;
898 HRESULT jsdisp_call(jsdisp_t *disp, DISPID id, WORD flags, DISPPARAMS *dp, VARIANT *retv,
899 jsexcept_t *ei, IServiceProvider *caller)
901 dispex_prop_t *prop;
903 memset(ei, 0, sizeof(*ei));
904 if(retv)
905 V_VT(retv) = VT_EMPTY;
907 prop = get_prop(disp, id);
908 if(!prop)
909 return DISP_E_MEMBERNOTFOUND;
911 return invoke_prop_func(disp, disp, prop, flags, dp, retv, ei, caller);
914 HRESULT jsdisp_call_name(jsdisp_t *disp, const WCHAR *name, WORD flags, DISPPARAMS *dp, VARIANT *retv,
915 jsexcept_t *ei, IServiceProvider *caller)
917 dispex_prop_t *prop;
918 HRESULT hres;
920 hres = find_prop_name_prot(disp, name, &prop);
921 if(FAILED(hres))
922 return hres;
924 memset(ei, 0, sizeof(*ei));
925 if(retv)
926 V_VT(retv) = VT_EMPTY;
928 return invoke_prop_func(disp, disp, prop, flags, dp, retv, ei, caller);
931 HRESULT disp_call(script_ctx_t *ctx, IDispatch *disp, DISPID id, WORD flags, DISPPARAMS *dp, VARIANT *retv,
932 jsexcept_t *ei, IServiceProvider *caller)
934 jsdisp_t *jsdisp;
935 IDispatchEx *dispex;
936 HRESULT hres;
938 jsdisp = iface_to_jsdisp((IUnknown*)disp);
939 if(jsdisp) {
940 hres = jsdisp_call(jsdisp, id, flags, dp, retv, ei, caller);
941 jsdisp_release(jsdisp);
942 return hres;
945 memset(ei, 0, sizeof(*ei));
946 if(retv && arg_cnt(dp))
947 flags |= DISPATCH_PROPERTYGET;
949 if(retv)
950 V_VT(retv) = VT_EMPTY;
951 hres = IDispatch_QueryInterface(disp, &IID_IDispatchEx, (void**)&dispex);
952 if(FAILED(hres)) {
953 UINT err = 0;
955 if(flags == DISPATCH_CONSTRUCT) {
956 WARN("IDispatch cannot be constructor\n");
957 return DISP_E_MEMBERNOTFOUND;
960 TRACE("using IDispatch\n");
961 return IDispatch_Invoke(disp, id, &IID_NULL, ctx->lcid, flags, dp, retv, &ei->ei, &err);
964 hres = IDispatchEx_InvokeEx(dispex, id, ctx->lcid, flags, dp, retv, &ei->ei, caller);
965 IDispatchEx_Release(dispex);
967 return hres;
970 HRESULT jsdisp_propput_name(jsdisp_t *obj, const WCHAR *name, VARIANT *val, jsexcept_t *ei, IServiceProvider *caller)
972 dispex_prop_t *prop;
973 HRESULT hres;
975 hres = ensure_prop_name(obj, name, FALSE, PROPF_ENUM, &prop);
976 if(FAILED(hres))
977 return hres;
979 return prop_put(obj, prop, val, ei, caller);
982 HRESULT jsdisp_propput_const(jsdisp_t *obj, const WCHAR *name, VARIANT *val)
984 dispex_prop_t *prop;
985 HRESULT hres;
987 hres = ensure_prop_name(obj, name, FALSE, PROPF_ENUM|PROPF_CONST, &prop);
988 if(FAILED(hres))
989 return hres;
991 return VariantCopy(&prop->u.var, val);
994 HRESULT jsdisp_propput_idx(jsdisp_t *obj, DWORD idx, VARIANT *val, jsexcept_t *ei, IServiceProvider *caller)
996 WCHAR buf[12];
998 static const WCHAR formatW[] = {'%','d',0};
1000 sprintfW(buf, formatW, idx);
1001 return jsdisp_propput_name(obj, buf, val, ei, caller);
1004 HRESULT disp_propput(script_ctx_t *ctx, IDispatch *disp, DISPID id, VARIANT *val, jsexcept_t *ei, IServiceProvider *caller)
1006 jsdisp_t *jsdisp;
1007 HRESULT hres;
1009 jsdisp = iface_to_jsdisp((IUnknown*)disp);
1010 if(jsdisp) {
1011 dispex_prop_t *prop;
1013 prop = get_prop(jsdisp, id);
1014 if(prop)
1015 hres = prop_put(jsdisp, prop, val, ei, caller);
1016 else
1017 hres = DISP_E_MEMBERNOTFOUND;
1019 jsdisp_release(jsdisp);
1020 }else {
1021 DISPID dispid = DISPID_PROPERTYPUT;
1022 DISPPARAMS dp = {val, &dispid, 1, 1};
1023 IDispatchEx *dispex;
1025 hres = IDispatch_QueryInterface(disp, &IID_IDispatchEx, (void**)&dispex);
1026 if(SUCCEEDED(hres)) {
1027 hres = IDispatchEx_InvokeEx(dispex, id, ctx->lcid, DISPATCH_PROPERTYPUT, &dp, NULL, &ei->ei, caller);
1028 IDispatchEx_Release(dispex);
1029 }else {
1030 ULONG err = 0;
1032 TRACE("using IDispatch\n");
1033 hres = IDispatch_Invoke(disp, id, &IID_NULL, ctx->lcid, DISPATCH_PROPERTYPUT, &dp, NULL, &ei->ei, &err);
1037 return hres;
1040 HRESULT jsdisp_propget_name(jsdisp_t *obj, const WCHAR *name, VARIANT *var, jsexcept_t *ei, IServiceProvider *caller)
1042 DISPPARAMS dp = {NULL, NULL, 0, 0};
1043 dispex_prop_t *prop;
1044 HRESULT hres;
1046 hres = find_prop_name_prot(obj, name, &prop);
1047 if(FAILED(hres))
1048 return hres;
1050 V_VT(var) = VT_EMPTY;
1051 if(!prop)
1052 return S_OK;
1054 return prop_get(obj, prop, &dp, var, ei, caller);
1057 HRESULT jsdisp_get_idx(jsdisp_t *obj, DWORD idx, VARIANT *var, jsexcept_t *ei, IServiceProvider *caller)
1059 WCHAR name[12];
1060 DISPPARAMS dp = {NULL, NULL, 0, 0};
1061 dispex_prop_t *prop;
1062 HRESULT hres;
1064 static const WCHAR formatW[] = {'%','d',0};
1066 sprintfW(name, formatW, idx);
1068 hres = find_prop_name_prot(obj, name, &prop);
1069 if(FAILED(hres))
1070 return hres;
1072 V_VT(var) = VT_EMPTY;
1073 if(!prop)
1074 return DISP_E_UNKNOWNNAME;
1076 return prop_get(obj, prop, &dp, var, ei, caller);
1079 HRESULT jsdisp_propget(jsdisp_t *jsdisp, DISPID id, VARIANT *val, jsexcept_t *ei, IServiceProvider *caller)
1081 DISPPARAMS dp = {NULL,NULL,0,0};
1082 dispex_prop_t *prop;
1084 prop = get_prop(jsdisp, id);
1085 if(!prop)
1086 return DISP_E_MEMBERNOTFOUND;
1088 V_VT(val) = VT_EMPTY;
1089 return prop_get(jsdisp, prop, &dp, val, ei, caller);
1092 HRESULT disp_propget(script_ctx_t *ctx, IDispatch *disp, DISPID id, VARIANT *val, jsexcept_t *ei, IServiceProvider *caller)
1094 DISPPARAMS dp = {NULL,NULL,0,0};
1095 IDispatchEx *dispex;
1096 jsdisp_t *jsdisp;
1097 HRESULT hres;
1099 jsdisp = iface_to_jsdisp((IUnknown*)disp);
1100 if(jsdisp) {
1101 hres = jsdisp_propget(jsdisp, id, val, ei, caller);
1102 jsdisp_release(jsdisp);
1103 return hres;
1106 hres = IDispatch_QueryInterface(disp, &IID_IDispatchEx, (void**)&dispex);
1107 if(FAILED(hres)) {
1108 ULONG err = 0;
1110 TRACE("using IDispatch\n");
1111 return IDispatch_Invoke(disp, id, &IID_NULL, ctx->lcid, INVOKE_PROPERTYGET, &dp, val, &ei->ei, &err);
1114 hres = IDispatchEx_InvokeEx(dispex, id, ctx->lcid, INVOKE_PROPERTYGET, &dp, val, &ei->ei, caller);
1115 IDispatchEx_Release(dispex);
1117 return hres;
1120 HRESULT jsdisp_delete_idx(jsdisp_t *obj, DWORD idx)
1122 static const WCHAR formatW[] = {'%','d',0};
1123 WCHAR buf[12];
1124 dispex_prop_t *prop;
1125 HRESULT hres;
1127 sprintfW(buf, formatW, idx);
1129 hres = find_prop_name(obj, buf, &prop);
1130 if(FAILED(hres) || !prop)
1131 return hres;
1133 return delete_prop(prop);