jscript: Don't use DISPPARAMS named args for internal parameter passing.
[wine/multimedia.git] / dlls / jscript / dispex.c
blobc52ac7d431ebd6fb7dc482658a6809461ae30b05
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 <assert.h>
21 #include "jscript.h"
23 #include "wine/unicode.h"
24 #include "wine/debug.h"
26 WINE_DEFAULT_DEBUG_CHANNEL(jscript);
29 * This IID is used to get jsdisp_t objecto from interface.
30 * We might consider using private interface instead.
32 static const IID IID_IDispatchJS =
33 {0x719c3050,0xf9d3,0x11cf,{0xa4,0x93,0x00,0x40,0x05,0x23,0xa8,0xa6}};
35 #define FDEX_VERSION_MASK 0xf0000000
36 #define GOLDEN_RATIO 0x9E3779B9U
38 typedef enum {
39 PROP_VARIANT,
40 PROP_BUILTIN,
41 PROP_PROTREF,
42 PROP_DELETED
43 } prop_type_t;
45 struct _dispex_prop_t {
46 WCHAR *name;
47 unsigned hash;
48 prop_type_t type;
49 DWORD flags;
51 union {
52 VARIANT var;
53 const builtin_prop_t *p;
54 DWORD ref;
55 } u;
57 int bucket_head;
58 int bucket_next;
61 static inline DISPID prop_to_id(jsdisp_t *This, dispex_prop_t *prop)
63 return prop - This->props;
66 static inline dispex_prop_t *get_prop(jsdisp_t *This, DISPID id)
68 if(id < 0 || id >= This->prop_cnt || This->props[id].type == PROP_DELETED)
69 return NULL;
71 return This->props+id;
74 static DWORD get_flags(jsdisp_t *This, dispex_prop_t *prop)
76 if(prop->type == PROP_PROTREF) {
77 dispex_prop_t *parent = get_prop(This->prototype, prop->u.ref);
78 if(!parent) {
79 prop->type = PROP_DELETED;
80 return 0;
83 return get_flags(This->prototype, parent);
86 return prop->flags;
89 static const builtin_prop_t *find_builtin_prop(jsdisp_t *This, const WCHAR *name)
91 int min = 0, max, i, r;
93 max = This->builtin_info->props_cnt-1;
94 while(min <= max) {
95 i = (min+max)/2;
97 r = strcmpW(name, This->builtin_info->props[i].name);
98 if(!r)
99 return This->builtin_info->props + i;
101 if(r < 0)
102 max = i-1;
103 else
104 min = i+1;
107 return NULL;
110 static inline unsigned string_hash(const WCHAR *name)
112 unsigned h = 0;
113 for(; *name; name++)
114 h = (h>>(sizeof(unsigned)*8-4)) ^ (h<<4) ^ tolowerW(*name);
115 return h;
118 static inline unsigned get_props_idx(jsdisp_t *This, unsigned hash)
120 return (hash*GOLDEN_RATIO) & (This->buf_size-1);
123 static inline HRESULT resize_props(jsdisp_t *This)
125 dispex_prop_t *props;
126 int i, bucket;
128 if(This->buf_size != This->prop_cnt)
129 return S_FALSE;
131 props = heap_realloc(This->props, sizeof(dispex_prop_t)*This->buf_size*2);
132 if(!props)
133 return E_OUTOFMEMORY;
134 This->buf_size *= 2;
135 This->props = props;
137 for(i=0; i<This->buf_size; i++) {
138 This->props[i].bucket_head = 0;
139 This->props[i].bucket_next = 0;
142 for(i=1; i<This->prop_cnt; i++) {
143 props = This->props+i;
145 bucket = get_props_idx(This, props->hash);
146 props->bucket_next = This->props[bucket].bucket_head;
147 This->props[bucket].bucket_head = i;
150 return S_OK;
153 static inline dispex_prop_t* alloc_prop(jsdisp_t *This, const WCHAR *name, prop_type_t type, DWORD flags)
155 dispex_prop_t *prop;
156 unsigned bucket;
158 if(FAILED(resize_props(This)))
159 return NULL;
161 prop = &This->props[This->prop_cnt];
162 prop->name = heap_strdupW(name);
163 if(!prop->name)
164 return NULL;
165 prop->type = type;
166 prop->flags = flags;
167 prop->hash = string_hash(name);
169 bucket = get_props_idx(This, prop->hash);
170 prop->bucket_next = This->props[bucket].bucket_head;
171 This->props[bucket].bucket_head = This->prop_cnt++;
172 return prop;
175 static dispex_prop_t *alloc_protref(jsdisp_t *This, const WCHAR *name, DWORD ref)
177 dispex_prop_t *ret;
179 ret = alloc_prop(This, name, PROP_PROTREF, 0);
180 if(!ret)
181 return NULL;
183 ret->u.ref = ref;
184 return ret;
187 static HRESULT find_prop_name(jsdisp_t *This, unsigned hash, const WCHAR *name, dispex_prop_t **ret)
189 const builtin_prop_t *builtin;
190 unsigned bucket, pos, prev = 0;
191 dispex_prop_t *prop;
193 bucket = get_props_idx(This, hash);
194 pos = This->props[bucket].bucket_head;
195 while(pos != 0) {
196 if(!strcmpW(name, This->props[pos].name)) {
197 if(prev != 0) {
198 This->props[prev].bucket_next = This->props[pos].bucket_next;
199 This->props[pos].bucket_next = This->props[bucket].bucket_head;
200 This->props[bucket].bucket_head = pos;
203 *ret = &This->props[pos];
204 return S_OK;
207 prev = pos;
208 pos = This->props[pos].bucket_next;
211 builtin = find_builtin_prop(This, name);
212 if(builtin) {
213 prop = alloc_prop(This, name, PROP_BUILTIN, builtin->flags);
214 if(!prop)
215 return E_OUTOFMEMORY;
217 prop->u.p = builtin;
218 *ret = prop;
219 return S_OK;
222 *ret = NULL;
223 return S_OK;
226 static HRESULT find_prop_name_prot(jsdisp_t *This, unsigned hash, const WCHAR *name, dispex_prop_t **ret)
228 dispex_prop_t *prop, *del=NULL;
229 HRESULT hres;
231 hres = find_prop_name(This, hash, name, &prop);
232 if(FAILED(hres))
233 return hres;
234 if(prop && prop->type==PROP_DELETED) {
235 del = prop;
236 } else if(prop) {
237 *ret = prop;
238 return S_OK;
241 if(This->prototype) {
242 hres = find_prop_name_prot(This->prototype, hash, name, &prop);
243 if(FAILED(hres))
244 return hres;
245 if(prop) {
246 if(del) {
247 del->type = PROP_PROTREF;
248 del->flags = 0;
249 del->u.ref = prop - This->prototype->props;
250 prop = del;
251 }else {
252 prop = alloc_protref(This, prop->name, prop - This->prototype->props);
253 if(!prop)
254 return E_OUTOFMEMORY;
257 *ret = prop;
258 return S_OK;
262 *ret = del;
263 return S_OK;
266 static HRESULT ensure_prop_name(jsdisp_t *This, const WCHAR *name, BOOL search_prot, DWORD create_flags, dispex_prop_t **ret)
268 dispex_prop_t *prop;
269 HRESULT hres;
271 if(search_prot)
272 hres = find_prop_name_prot(This, string_hash(name), name, &prop);
273 else
274 hres = find_prop_name(This, string_hash(name), name, &prop);
275 if(SUCCEEDED(hres) && (!prop || prop->type==PROP_DELETED)) {
276 TRACE("creating prop %s\n", debugstr_w(name));
278 if(prop) {
279 prop->type = PROP_VARIANT;
280 prop->flags = create_flags;
281 }else {
282 prop = alloc_prop(This, name, PROP_VARIANT, create_flags);
283 if(!prop)
284 return E_OUTOFMEMORY;
287 VariantInit(&prop->u.var);
290 *ret = prop;
291 return hres;
294 static HRESULT set_this(DISPPARAMS *dp, const DISPPARAMS *olddp, IDispatch *jsthis)
296 VARIANTARG *oldargs;
297 int i;
299 static DISPID this_id = DISPID_THIS;
301 *dp = *olddp;
303 for(i = 0; i < dp->cNamedArgs; i++) {
304 if(dp->rgdispidNamedArgs[i] == DISPID_THIS)
305 return S_OK;
308 oldargs = dp->rgvarg;
309 dp->rgvarg = heap_alloc((dp->cArgs+1) * sizeof(VARIANTARG));
310 if(!dp->rgvarg)
311 return E_OUTOFMEMORY;
312 memcpy(dp->rgvarg+1, oldargs, dp->cArgs*sizeof(VARIANTARG));
313 V_VT(dp->rgvarg) = VT_DISPATCH;
314 V_DISPATCH(dp->rgvarg) = jsthis;
315 dp->cArgs++;
317 if(dp->cNamedArgs) {
318 DISPID *old = dp->rgdispidNamedArgs;
319 dp->rgdispidNamedArgs = heap_alloc((dp->cNamedArgs+1)*sizeof(DISPID));
320 if(!dp->rgdispidNamedArgs) {
321 heap_free(dp->rgvarg);
322 return E_OUTOFMEMORY;
325 memcpy(dp->rgdispidNamedArgs+1, old, dp->cNamedArgs*sizeof(DISPID));
326 dp->rgdispidNamedArgs[0] = DISPID_THIS;
327 dp->cNamedArgs++;
328 }else {
329 dp->rgdispidNamedArgs = &this_id;
330 dp->cNamedArgs = 1;
333 return S_OK;
336 static IDispatch *get_this(DISPPARAMS *dp)
338 DWORD i;
340 for(i=0; i < dp->cNamedArgs; i++) {
341 if(dp->rgdispidNamedArgs[i] == DISPID_THIS) {
342 if(V_VT(dp->rgvarg+i) == VT_DISPATCH)
343 return V_DISPATCH(dp->rgvarg+i);
345 WARN("This is not VT_DISPATCH\n");
346 return NULL;
350 TRACE("no this passed\n");
351 return NULL;
354 static HRESULT convert_params(const DISPPARAMS *dp, VARIANT *buf, DISPPARAMS *ret)
356 BOOL need_conversion = FALSE;
357 const VARIANT *s;
358 VARIANT *d;
359 unsigned i;
361 ret->cArgs = dp->cArgs - dp->cNamedArgs;
362 ret->cNamedArgs = 0;
364 for(i = 0; i < ret->cArgs && !need_conversion; i++) {
365 switch(V_VT(get_arg(dp, i))) {
366 case VT_I2:
367 case VT_INT:
368 need_conversion = TRUE;
369 break;
373 if(!need_conversion) {
374 ret->rgvarg = dp->rgvarg + dp->cNamedArgs;
375 return S_OK;
378 if(ret->cArgs > 6) {
379 ret->rgvarg = heap_alloc(ret->cArgs * sizeof(VARIANT));
380 if(!ret->rgvarg)
381 return E_OUTOFMEMORY;
382 }else {
383 ret->rgvarg = buf;
386 for(i = 0; i < ret->cArgs; i++) {
387 s = get_arg(dp, i);
388 d = get_arg(ret, i);
389 switch(V_VT(s)) {
390 case VT_I2:
391 V_VT(d) = VT_I4;
392 V_I4(d) = V_I2(s);
393 break;
394 case VT_INT:
395 V_VT(d) = VT_I4;
396 V_I4(d) = V_INT(s);
397 break;
398 default:
399 *d = *s;
403 return S_OK;
406 static HRESULT invoke_prop_func(jsdisp_t *This, IDispatch *jsthis, dispex_prop_t *prop, WORD flags,
407 DISPPARAMS *dp, VARIANT *retv, jsexcept_t *ei, IServiceProvider *caller)
409 HRESULT hres;
411 switch(prop->type) {
412 case PROP_BUILTIN: {
413 if(flags == DISPATCH_CONSTRUCT && (prop->flags & PROPF_METHOD)) {
414 WARN("%s is not a constructor\n", debugstr_w(prop->name));
415 return E_INVALIDARG;
418 if(prop->name || This->builtin_info->class != JSCLASS_FUNCTION) {
419 vdisp_t vthis;
421 if(jsthis)
422 set_disp(&vthis, jsthis);
423 else
424 set_jsdisp(&vthis, This);
425 hres = prop->u.p->invoke(This->ctx, &vthis, flags, dp, retv, ei);
426 vdisp_release(&vthis);
427 }else {
428 /* Function object calls are special case */
429 hres = Function_invoke(This, jsthis, flags, dp, retv, ei);
431 return hres;
433 case PROP_PROTREF:
434 return invoke_prop_func(This->prototype, jsthis, This->prototype->props+prop->u.ref, flags, dp, retv, ei, caller);
435 case PROP_VARIANT: {
436 if(V_VT(&prop->u.var) != VT_DISPATCH) {
437 FIXME("invoke vt %d\n", V_VT(&prop->u.var));
438 return E_FAIL;
441 TRACE("call %s %p\n", debugstr_w(prop->name), V_DISPATCH(&prop->u.var));
443 return disp_call_value(This->ctx, V_DISPATCH(&prop->u.var), jsthis, flags, dp, retv, ei);
445 default:
446 ERR("type %d\n", prop->type);
449 return E_FAIL;
452 static HRESULT prop_get(jsdisp_t *This, dispex_prop_t *prop, DISPPARAMS *dp,
453 VARIANT *retv, jsexcept_t *ei, IServiceProvider *caller)
455 HRESULT hres;
457 switch(prop->type) {
458 case PROP_BUILTIN:
459 if(prop->u.p->flags & PROPF_METHOD) {
460 jsdisp_t *obj;
461 hres = create_builtin_function(This->ctx, prop->u.p->invoke, prop->u.p->name, NULL,
462 prop->u.p->flags, NULL, &obj);
463 if(FAILED(hres))
464 break;
466 prop->type = PROP_VARIANT;
467 var_set_jsdisp(&prop->u.var, obj);
468 hres = VariantCopy(retv, &prop->u.var);
469 }else {
470 vdisp_t vthis;
472 set_jsdisp(&vthis, This);
473 hres = prop->u.p->invoke(This->ctx, &vthis, DISPATCH_PROPERTYGET, dp, retv, ei);
474 vdisp_release(&vthis);
476 break;
477 case PROP_PROTREF:
478 hres = prop_get(This->prototype, This->prototype->props+prop->u.ref, dp, retv, ei, caller);
479 break;
480 case PROP_VARIANT:
481 hres = VariantCopy(retv, &prop->u.var);
482 break;
483 default:
484 ERR("type %d\n", prop->type);
485 return E_FAIL;
488 if(FAILED(hres)) {
489 TRACE("fail %08x\n", hres);
490 return hres;
493 TRACE("%s ret %s\n", debugstr_w(prop->name), debugstr_variant(retv));
494 return hres;
497 static HRESULT prop_put(jsdisp_t *This, dispex_prop_t *prop, VARIANT *val,
498 jsexcept_t *ei, IServiceProvider *caller)
500 HRESULT hres;
502 if(prop->flags & PROPF_CONST)
503 return S_OK;
505 switch(prop->type) {
506 case PROP_BUILTIN:
507 if(!(prop->flags & PROPF_METHOD)) {
508 DISPPARAMS dp = {val, NULL, 1, 0};
509 vdisp_t vthis;
511 set_jsdisp(&vthis, This);
512 hres = prop->u.p->invoke(This->ctx, &vthis, DISPATCH_PROPERTYPUT, &dp, NULL, ei);
513 vdisp_release(&vthis);
514 return hres;
516 case PROP_PROTREF:
517 prop->type = PROP_VARIANT;
518 prop->flags = PROPF_ENUM;
519 V_VT(&prop->u.var) = VT_EMPTY;
520 break;
521 case PROP_VARIANT:
522 VariantClear(&prop->u.var);
523 break;
524 default:
525 ERR("type %d\n", prop->type);
526 return E_FAIL;
529 hres = VariantCopy(&prop->u.var, val);
530 if(FAILED(hres))
531 return hres;
533 if(This->builtin_info->on_put)
534 This->builtin_info->on_put(This, prop->name);
536 TRACE("%s = %s\n", debugstr_w(prop->name), debugstr_variant(val));
537 return S_OK;
540 static HRESULT fill_protrefs(jsdisp_t *This)
542 dispex_prop_t *iter, *prop;
543 HRESULT hres;
545 if(!This->prototype)
546 return S_OK;
548 fill_protrefs(This->prototype);
550 for(iter = This->prototype->props; iter < This->prototype->props+This->prototype->prop_cnt; iter++) {
551 if(!iter->name)
552 continue;
553 hres = find_prop_name(This, iter->hash, iter->name, &prop);
554 if(FAILED(hres))
555 return hres;
556 if(!prop || prop->type==PROP_DELETED) {
557 if(prop) {
558 prop->type = PROP_PROTREF;
559 prop->flags = 0;
560 prop->u.ref = iter - This->prototype->props;
561 }else {
562 prop = alloc_protref(This, iter->name, iter - This->prototype->props);
563 if(!prop)
564 return E_OUTOFMEMORY;
569 return S_OK;
572 static inline jsdisp_t *impl_from_IDispatchEx(IDispatchEx *iface)
574 return CONTAINING_RECORD(iface, jsdisp_t, IDispatchEx_iface);
577 static HRESULT WINAPI DispatchEx_QueryInterface(IDispatchEx *iface, REFIID riid, void **ppv)
579 jsdisp_t *This = impl_from_IDispatchEx(iface);
581 if(IsEqualGUID(&IID_IUnknown, riid)) {
582 TRACE("(%p)->(IID_IUnknown %p)\n", This, ppv);
583 *ppv = &This->IDispatchEx_iface;
584 }else if(IsEqualGUID(&IID_IDispatch, riid)) {
585 TRACE("(%p)->(IID_IDispatch %p)\n", This, ppv);
586 *ppv = &This->IDispatchEx_iface;
587 }else if(IsEqualGUID(&IID_IDispatchEx, riid)) {
588 TRACE("(%p)->(IID_IDispatchEx %p)\n", This, ppv);
589 *ppv = &This->IDispatchEx_iface;
590 }else if(IsEqualGUID(&IID_IDispatchJS, riid)) {
591 TRACE("(%p)->(IID_IDispatchJS %p)\n", This, ppv);
592 jsdisp_addref(This);
593 *ppv = This;
594 return S_OK;
595 }else {
596 WARN("(%p)->(%s %p)\n", This, debugstr_guid(riid), ppv);
597 *ppv = NULL;
598 return E_NOINTERFACE;
601 IUnknown_AddRef((IUnknown*)*ppv);
602 return S_OK;
605 static ULONG WINAPI DispatchEx_AddRef(IDispatchEx *iface)
607 jsdisp_t *This = impl_from_IDispatchEx(iface);
608 LONG ref = InterlockedIncrement(&This->ref);
610 TRACE("(%p) ref=%d\n", This, ref);
612 return ref;
615 static ULONG WINAPI DispatchEx_Release(IDispatchEx *iface)
617 jsdisp_t *This = impl_from_IDispatchEx(iface);
618 LONG ref = InterlockedDecrement(&This->ref);
620 TRACE("(%p) ref=%d\n", This, ref);
622 if(!ref) {
623 dispex_prop_t *prop;
625 for(prop = This->props; prop < This->props+This->prop_cnt; prop++) {
626 if(prop->type == PROP_VARIANT)
627 VariantClear(&prop->u.var);
628 heap_free(prop->name);
630 heap_free(This->props);
631 script_release(This->ctx);
632 if(This->prototype)
633 jsdisp_release(This->prototype);
635 if(This->builtin_info->destructor)
636 This->builtin_info->destructor(This);
637 else
638 heap_free(This);
641 return ref;
644 static HRESULT WINAPI DispatchEx_GetTypeInfoCount(IDispatchEx *iface, UINT *pctinfo)
646 jsdisp_t *This = impl_from_IDispatchEx(iface);
648 TRACE("(%p)->(%p)\n", This, pctinfo);
650 *pctinfo = 1;
651 return S_OK;
654 static HRESULT WINAPI DispatchEx_GetTypeInfo(IDispatchEx *iface, UINT iTInfo, LCID lcid,
655 ITypeInfo **ppTInfo)
657 jsdisp_t *This = impl_from_IDispatchEx(iface);
658 FIXME("(%p)->(%u %u %p)\n", This, iTInfo, lcid, ppTInfo);
659 return E_NOTIMPL;
662 static HRESULT WINAPI DispatchEx_GetIDsOfNames(IDispatchEx *iface, REFIID riid,
663 LPOLESTR *rgszNames, UINT cNames, LCID lcid,
664 DISPID *rgDispId)
666 jsdisp_t *This = impl_from_IDispatchEx(iface);
667 UINT i;
668 HRESULT hres;
670 TRACE("(%p)->(%s %p %u %u %p)\n", This, debugstr_guid(riid), rgszNames, cNames,
671 lcid, rgDispId);
673 for(i=0; i < cNames; i++) {
674 hres = IDispatchEx_GetDispID(&This->IDispatchEx_iface, rgszNames[i], 0, rgDispId+i);
675 if(FAILED(hres))
676 return hres;
679 return S_OK;
682 static HRESULT WINAPI DispatchEx_Invoke(IDispatchEx *iface, DISPID dispIdMember,
683 REFIID riid, LCID lcid, WORD wFlags, DISPPARAMS *pDispParams,
684 VARIANT *pVarResult, EXCEPINFO *pExcepInfo, UINT *puArgErr)
686 jsdisp_t *This = impl_from_IDispatchEx(iface);
688 TRACE("(%p)->(%d %s %d %d %p %p %p %p)\n", This, dispIdMember, debugstr_guid(riid),
689 lcid, wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr);
691 return IDispatchEx_InvokeEx(&This->IDispatchEx_iface, dispIdMember, lcid, wFlags,
692 pDispParams, pVarResult, pExcepInfo, NULL);
695 static HRESULT WINAPI DispatchEx_GetDispID(IDispatchEx *iface, BSTR bstrName, DWORD grfdex, DISPID *pid)
697 jsdisp_t *This = impl_from_IDispatchEx(iface);
699 TRACE("(%p)->(%s %x %p)\n", This, debugstr_w(bstrName), grfdex, pid);
701 if(grfdex & ~(fdexNameCaseSensitive|fdexNameEnsure|fdexNameImplicit|FDEX_VERSION_MASK)) {
702 FIXME("Unsupported grfdex %x\n", grfdex);
703 return E_NOTIMPL;
706 return jsdisp_get_id(This, bstrName, grfdex, pid);
709 static HRESULT WINAPI DispatchEx_InvokeEx(IDispatchEx *iface, DISPID id, LCID lcid, WORD wFlags, DISPPARAMS *pdp,
710 VARIANT *pvarRes, EXCEPINFO *pei, IServiceProvider *pspCaller)
712 jsdisp_t *This = impl_from_IDispatchEx(iface);
713 dispex_prop_t *prop;
714 jsexcept_t jsexcept;
715 HRESULT hres;
717 TRACE("(%p)->(%x %x %x %p %p %p %p)\n", This, id, lcid, wFlags, pdp, pvarRes, pei, pspCaller);
719 if(pvarRes)
720 V_VT(pvarRes) = VT_EMPTY;
722 prop = get_prop(This, id);
723 if(!prop || prop->type == PROP_DELETED) {
724 TRACE("invalid id\n");
725 return DISP_E_MEMBERNOTFOUND;
728 memset(&jsexcept, 0, sizeof(jsexcept));
730 switch(wFlags) {
731 case DISPATCH_METHOD|DISPATCH_PROPERTYGET:
732 wFlags = DISPATCH_METHOD;
733 /* fall through */
734 case DISPATCH_METHOD:
735 case DISPATCH_CONSTRUCT: {
736 DISPPARAMS params;
737 VARIANT buf[6];
739 hres = convert_params(pdp, buf, &params);
740 if(FAILED(hres))
741 return hres;
743 hres = invoke_prop_func(This, get_this(pdp), prop, wFlags, &params, pvarRes, &jsexcept, pspCaller);
744 if(params.rgvarg != buf && params.rgvarg != pdp->rgvarg)
745 heap_free(params.rgvarg);
746 break;
748 case DISPATCH_PROPERTYGET:
749 hres = prop_get(This, prop, pdp, pvarRes, &jsexcept, pspCaller);
750 break;
751 case DISPATCH_PROPERTYPUT: {
752 DWORD i;
754 for(i=0; i < pdp->cNamedArgs; i++) {
755 if(pdp->rgdispidNamedArgs[i] == DISPID_PROPERTYPUT)
756 break;
759 if(i == pdp->cNamedArgs) {
760 TRACE("no value to set\n");
761 return DISP_E_PARAMNOTOPTIONAL;
764 hres = prop_put(This, prop, pdp->rgvarg+i, &jsexcept, pspCaller);
765 break;
767 default:
768 FIXME("Unimplemented flags %x\n", wFlags);
769 return E_INVALIDARG;
772 if(pei)
773 *pei = jsexcept.ei;
775 return hres;
778 static HRESULT delete_prop(dispex_prop_t *prop)
780 if(prop->type == PROP_VARIANT) {
781 VariantClear(&prop->u.var);
782 prop->type = PROP_DELETED;
784 return S_OK;
787 static HRESULT WINAPI DispatchEx_DeleteMemberByName(IDispatchEx *iface, BSTR bstrName, DWORD grfdex)
789 jsdisp_t *This = impl_from_IDispatchEx(iface);
790 dispex_prop_t *prop;
791 HRESULT hres;
793 TRACE("(%p)->(%s %x)\n", This, debugstr_w(bstrName), grfdex);
795 if(grfdex & ~(fdexNameCaseSensitive|fdexNameEnsure|fdexNameImplicit|FDEX_VERSION_MASK))
796 FIXME("Unsupported grfdex %x\n", grfdex);
798 hres = find_prop_name(This, string_hash(bstrName), bstrName, &prop);
799 if(FAILED(hres))
800 return hres;
801 if(!prop) {
802 TRACE("not found\n");
803 return S_OK;
806 return delete_prop(prop);
809 static HRESULT WINAPI DispatchEx_DeleteMemberByDispID(IDispatchEx *iface, DISPID id)
811 jsdisp_t *This = impl_from_IDispatchEx(iface);
812 dispex_prop_t *prop;
814 TRACE("(%p)->(%x)\n", This, id);
816 prop = get_prop(This, id);
817 if(!prop) {
818 WARN("invalid id\n");
819 return DISP_E_MEMBERNOTFOUND;
822 return delete_prop(prop);
825 static HRESULT WINAPI DispatchEx_GetMemberProperties(IDispatchEx *iface, DISPID id, DWORD grfdexFetch, DWORD *pgrfdex)
827 jsdisp_t *This = impl_from_IDispatchEx(iface);
828 FIXME("(%p)->(%x %x %p)\n", This, id, grfdexFetch, pgrfdex);
829 return E_NOTIMPL;
832 static HRESULT WINAPI DispatchEx_GetMemberName(IDispatchEx *iface, DISPID id, BSTR *pbstrName)
834 jsdisp_t *This = impl_from_IDispatchEx(iface);
835 dispex_prop_t *prop;
837 TRACE("(%p)->(%x %p)\n", This, id, pbstrName);
839 prop = get_prop(This, id);
840 if(!prop || !prop->name || prop->type == PROP_DELETED)
841 return DISP_E_MEMBERNOTFOUND;
843 *pbstrName = SysAllocString(prop->name);
844 if(!*pbstrName)
845 return E_OUTOFMEMORY;
847 return S_OK;
850 static HRESULT WINAPI DispatchEx_GetNextDispID(IDispatchEx *iface, DWORD grfdex, DISPID id, DISPID *pid)
852 jsdisp_t *This = impl_from_IDispatchEx(iface);
853 dispex_prop_t *iter;
854 HRESULT hres;
856 TRACE("(%p)->(%x %x %p)\n", This, grfdex, id, pid);
858 if(id == DISPID_STARTENUM) {
859 hres = fill_protrefs(This);
860 if(FAILED(hres))
861 return hres;
864 if(id+1>=0 && id+1<This->prop_cnt) {
865 iter = &This->props[id+1];
866 }else {
867 *pid = DISPID_STARTENUM;
868 return S_FALSE;
871 while(iter < This->props + This->prop_cnt) {
872 if(iter->name && (get_flags(This, iter) & PROPF_ENUM) && iter->type!=PROP_DELETED) {
873 *pid = prop_to_id(This, iter);
874 return S_OK;
876 iter++;
879 *pid = DISPID_STARTENUM;
880 return S_FALSE;
883 static HRESULT WINAPI DispatchEx_GetNameSpaceParent(IDispatchEx *iface, IUnknown **ppunk)
885 jsdisp_t *This = impl_from_IDispatchEx(iface);
886 FIXME("(%p)->(%p)\n", This, ppunk);
887 return E_NOTIMPL;
890 static IDispatchExVtbl DispatchExVtbl = {
891 DispatchEx_QueryInterface,
892 DispatchEx_AddRef,
893 DispatchEx_Release,
894 DispatchEx_GetTypeInfoCount,
895 DispatchEx_GetTypeInfo,
896 DispatchEx_GetIDsOfNames,
897 DispatchEx_Invoke,
898 DispatchEx_GetDispID,
899 DispatchEx_InvokeEx,
900 DispatchEx_DeleteMemberByName,
901 DispatchEx_DeleteMemberByDispID,
902 DispatchEx_GetMemberProperties,
903 DispatchEx_GetMemberName,
904 DispatchEx_GetNextDispID,
905 DispatchEx_GetNameSpaceParent
908 jsdisp_t *as_jsdisp(IDispatch *disp)
910 assert(disp->lpVtbl == (IDispatchVtbl*)&DispatchExVtbl);
911 return impl_from_IDispatchEx((IDispatchEx*)disp);
914 jsdisp_t *to_jsdisp(IDispatch *disp)
916 return disp->lpVtbl == (IDispatchVtbl*)&DispatchExVtbl ? impl_from_IDispatchEx((IDispatchEx*)disp) : NULL;
919 HRESULT init_dispex(jsdisp_t *dispex, script_ctx_t *ctx, const builtin_info_t *builtin_info, jsdisp_t *prototype)
921 TRACE("%p (%p)\n", dispex, prototype);
923 dispex->IDispatchEx_iface.lpVtbl = &DispatchExVtbl;
924 dispex->ref = 1;
925 dispex->builtin_info = builtin_info;
927 dispex->props = heap_alloc_zero(sizeof(dispex_prop_t)*(dispex->buf_size=4));
928 if(!dispex->props)
929 return E_OUTOFMEMORY;
931 dispex->prototype = prototype;
932 if(prototype)
933 jsdisp_addref(prototype);
935 dispex->prop_cnt = 1;
936 if(builtin_info->value_prop.invoke) {
937 dispex->props[0].type = PROP_BUILTIN;
938 dispex->props[0].u.p = &builtin_info->value_prop;
939 }else {
940 dispex->props[0].type = PROP_DELETED;
943 script_addref(ctx);
944 dispex->ctx = ctx;
946 return S_OK;
949 static const builtin_info_t dispex_info = {
950 JSCLASS_NONE,
951 {NULL, NULL, 0},
952 0, NULL,
953 NULL,
954 NULL
957 HRESULT create_dispex(script_ctx_t *ctx, const builtin_info_t *builtin_info, jsdisp_t *prototype, jsdisp_t **dispex)
959 jsdisp_t *ret;
960 HRESULT hres;
962 ret = heap_alloc_zero(sizeof(jsdisp_t));
963 if(!ret)
964 return E_OUTOFMEMORY;
966 hres = init_dispex(ret, ctx, builtin_info ? builtin_info : &dispex_info, prototype);
967 if(FAILED(hres))
968 return hres;
970 *dispex = ret;
971 return S_OK;
974 HRESULT init_dispex_from_constr(jsdisp_t *dispex, script_ctx_t *ctx, const builtin_info_t *builtin_info, jsdisp_t *constr)
976 jsdisp_t *prot = NULL;
977 dispex_prop_t *prop;
978 HRESULT hres;
980 static const WCHAR constructorW[] = {'c','o','n','s','t','r','u','c','t','o','r',0};
981 static const WCHAR prototypeW[] = {'p','r','o','t','o','t','y','p','e',0};
983 hres = find_prop_name_prot(constr, string_hash(prototypeW), prototypeW, &prop);
984 if(SUCCEEDED(hres) && prop && prop->type!=PROP_DELETED) {
985 jsexcept_t jsexcept;
986 VARIANT var;
988 V_VT(&var) = VT_EMPTY;
989 memset(&jsexcept, 0, sizeof(jsexcept));
990 hres = prop_get(constr, prop, NULL, &var, &jsexcept, NULL/*FIXME*/);
991 if(FAILED(hres)) {
992 ERR("Could not get prototype\n");
993 return hres;
996 if(V_VT(&var) == VT_DISPATCH)
997 prot = iface_to_jsdisp((IUnknown*)V_DISPATCH(&var));
998 VariantClear(&var);
1001 hres = init_dispex(dispex, ctx, builtin_info, prot);
1003 if(prot)
1004 jsdisp_release(prot);
1005 if(FAILED(hres))
1006 return hres;
1008 hres = ensure_prop_name(dispex, constructorW, FALSE, 0, &prop);
1009 if(SUCCEEDED(hres)) {
1010 jsexcept_t jsexcept;
1011 VARIANT var;
1013 var_set_jsdisp(&var, constr);
1014 memset(&jsexcept, 0, sizeof(jsexcept));
1015 hres = prop_put(dispex, prop, &var, &jsexcept, NULL/*FIXME*/);
1017 if(FAILED(hres))
1018 jsdisp_release(dispex);
1020 return hres;
1023 jsdisp_t *iface_to_jsdisp(IUnknown *iface)
1025 jsdisp_t *ret;
1026 HRESULT hres;
1028 hres = IUnknown_QueryInterface(iface, &IID_IDispatchJS, (void**)&ret);
1029 if(FAILED(hres))
1030 return NULL;
1032 return ret;
1035 static void ensure_retval_type(VARIANT *v)
1037 switch(V_VT(v)) {
1038 case VT_I2:
1039 V_VT(v) = VT_I4;
1040 V_I4(v) = V_I2(v);
1041 break;
1042 case VT_INT:
1043 V_VT(v) = VT_I4;
1044 V_I4(v) = V_INT(v);
1048 HRESULT jsdisp_get_id(jsdisp_t *jsdisp, const WCHAR *name, DWORD flags, DISPID *id)
1050 dispex_prop_t *prop;
1051 HRESULT hres;
1053 if(flags & fdexNameEnsure)
1054 hres = ensure_prop_name(jsdisp, name, TRUE, PROPF_ENUM, &prop);
1055 else
1056 hres = find_prop_name_prot(jsdisp, string_hash(name), name, &prop);
1057 if(FAILED(hres))
1058 return hres;
1060 if(prop && prop->type!=PROP_DELETED) {
1061 *id = prop_to_id(jsdisp, prop);
1062 return S_OK;
1065 TRACE("not found %s\n", debugstr_w(name));
1066 return DISP_E_UNKNOWNNAME;
1069 HRESULT jsdisp_call_value(jsdisp_t *jsfunc, IDispatch *jsthis, WORD flags, DISPPARAMS *dp, VARIANT *retv, jsexcept_t *ei)
1071 HRESULT hres;
1073 TRACE("args %p\n", dp->rgvarg);
1075 if(is_class(jsfunc, JSCLASS_FUNCTION)) {
1076 hres = Function_invoke(jsfunc, jsthis, flags, dp, retv, ei);
1077 }else {
1078 vdisp_t vdisp;
1080 set_disp(&vdisp, jsthis);
1081 hres = jsfunc->builtin_info->value_prop.invoke(jsfunc->ctx, &vdisp, flags, dp, retv, ei);
1082 vdisp_release(&vdisp);
1084 return hres;
1087 HRESULT jsdisp_call(jsdisp_t *disp, DISPID id, WORD flags, DISPPARAMS *dp, VARIANT *retv, jsexcept_t *ei)
1089 dispex_prop_t *prop;
1091 memset(ei, 0, sizeof(*ei));
1092 if(retv)
1093 V_VT(retv) = VT_EMPTY;
1095 prop = get_prop(disp, id);
1096 if(!prop)
1097 return DISP_E_MEMBERNOTFOUND;
1099 return invoke_prop_func(disp, to_disp(disp), prop, flags, dp, retv, ei, NULL);
1102 HRESULT jsdisp_call_name(jsdisp_t *disp, const WCHAR *name, WORD flags, DISPPARAMS *dp, VARIANT *retv,
1103 jsexcept_t *ei)
1105 dispex_prop_t *prop;
1106 HRESULT hres;
1108 hres = find_prop_name_prot(disp, string_hash(name), name, &prop);
1109 if(FAILED(hres))
1110 return hres;
1112 memset(ei, 0, sizeof(*ei));
1113 if(retv)
1114 V_VT(retv) = VT_EMPTY;
1116 return invoke_prop_func(disp, to_disp(disp), prop, flags, dp, retv, ei, NULL);
1119 HRESULT disp_call(script_ctx_t *ctx, IDispatch *disp, DISPID id, WORD flags, DISPPARAMS *dp, VARIANT *retv, jsexcept_t *ei)
1121 jsdisp_t *jsdisp;
1122 IDispatchEx *dispex;
1123 HRESULT hres;
1125 jsdisp = iface_to_jsdisp((IUnknown*)disp);
1126 if(jsdisp) {
1127 if(flags & DISPATCH_PROPERTYPUT) {
1128 FIXME("disp_call(propput) on builtin object\n");
1129 return E_FAIL;
1132 hres = jsdisp_call(jsdisp, id, flags, dp, retv, ei);
1133 jsdisp_release(jsdisp);
1134 return hres;
1137 memset(ei, 0, sizeof(*ei));
1138 if(retv && arg_cnt(dp))
1139 flags |= DISPATCH_PROPERTYGET;
1141 if(retv)
1142 V_VT(retv) = VT_EMPTY;
1143 hres = IDispatch_QueryInterface(disp, &IID_IDispatchEx, (void**)&dispex);
1144 if(SUCCEEDED(hres)) {
1145 hres = IDispatchEx_InvokeEx(dispex, id, ctx->lcid, flags, dp, retv, &ei->ei, &ctx->jscaller->IServiceProvider_iface);
1146 IDispatchEx_Release(dispex);
1147 }else {
1148 UINT err = 0;
1150 if(flags == DISPATCH_CONSTRUCT) {
1151 WARN("IDispatch cannot be constructor\n");
1152 return DISP_E_MEMBERNOTFOUND;
1155 TRACE("using IDispatch\n");
1156 hres = IDispatch_Invoke(disp, id, &IID_NULL, ctx->lcid, flags, dp, retv, &ei->ei, &err);
1159 if(FAILED(hres))
1160 return hres;
1162 if(retv)
1163 ensure_retval_type(retv);
1164 return S_OK;
1167 HRESULT disp_call_value(script_ctx_t *ctx, IDispatch *disp, IDispatch *jsthis, WORD flags, DISPPARAMS *dp,
1168 VARIANT *retv, jsexcept_t *ei)
1170 DISPPARAMS params;
1171 jsdisp_t *jsdisp;
1172 IDispatchEx *dispex;
1173 HRESULT hres;
1175 jsdisp = iface_to_jsdisp((IUnknown*)disp);
1176 if(jsdisp) {
1177 if(flags & DISPATCH_PROPERTYPUT) {
1178 FIXME("disp_call(propput) on builtin object\n");
1179 return E_FAIL;
1182 hres = jsdisp_call_value(jsdisp, jsthis, flags, dp, retv, ei);
1183 jsdisp_release(jsdisp);
1184 return hres;
1187 memset(ei, 0, sizeof(*ei));
1188 if(retv && arg_cnt(dp))
1189 flags |= DISPATCH_PROPERTYGET;
1191 if(jsthis) {
1192 hres = set_this(&params, dp, jsthis);
1193 if(FAILED(hres))
1194 return hres;
1195 }else {
1196 params = *dp;
1199 if(retv)
1200 V_VT(retv) = VT_EMPTY;
1201 hres = IDispatch_QueryInterface(disp, &IID_IDispatchEx, (void**)&dispex);
1202 if(SUCCEEDED(hres)) {
1203 hres = IDispatchEx_InvokeEx(dispex, DISPID_VALUE, ctx->lcid, flags, &params, retv, &ei->ei,
1204 &ctx->jscaller->IServiceProvider_iface);
1205 IDispatchEx_Release(dispex);
1206 }else {
1207 UINT err = 0;
1209 if(flags == DISPATCH_CONSTRUCT) {
1210 WARN("IDispatch cannot be constructor\n");
1211 return DISP_E_MEMBERNOTFOUND;
1214 TRACE("using IDispatch\n");
1215 hres = IDispatch_Invoke(disp, DISPID_VALUE, &IID_NULL, ctx->lcid, flags, &params, retv, &ei->ei, &err);
1218 if(params.rgvarg != dp->rgvarg) {
1219 heap_free(params.rgvarg);
1220 if(params.cNamedArgs > 1)
1221 heap_free(params.rgdispidNamedArgs);
1224 if(FAILED(hres))
1225 return hres;
1227 if(retv)
1228 ensure_retval_type(retv);
1229 return S_OK;
1232 HRESULT jsdisp_propput_name(jsdisp_t *obj, const WCHAR *name, VARIANT *val, jsexcept_t *ei)
1234 dispex_prop_t *prop;
1235 HRESULT hres;
1237 hres = ensure_prop_name(obj, name, FALSE, PROPF_ENUM, &prop);
1238 if(FAILED(hres))
1239 return hres;
1241 return prop_put(obj, prop, val, ei, NULL);
1244 HRESULT jsdisp_propput_const(jsdisp_t *obj, const WCHAR *name, VARIANT *val)
1246 dispex_prop_t *prop;
1247 HRESULT hres;
1249 hres = ensure_prop_name(obj, name, FALSE, PROPF_ENUM|PROPF_CONST, &prop);
1250 if(FAILED(hres))
1251 return hres;
1253 return VariantCopy(&prop->u.var, val);
1256 HRESULT jsdisp_propput_idx(jsdisp_t *obj, DWORD idx, VARIANT *val, jsexcept_t *ei)
1258 WCHAR buf[12];
1260 static const WCHAR formatW[] = {'%','d',0};
1262 sprintfW(buf, formatW, idx);
1263 return jsdisp_propput_name(obj, buf, val, ei);
1266 HRESULT disp_propput(script_ctx_t *ctx, IDispatch *disp, DISPID id, VARIANT *val, jsexcept_t *ei)
1268 jsdisp_t *jsdisp;
1269 HRESULT hres;
1271 jsdisp = iface_to_jsdisp((IUnknown*)disp);
1272 if(jsdisp) {
1273 dispex_prop_t *prop;
1275 prop = get_prop(jsdisp, id);
1276 if(prop)
1277 hres = prop_put(jsdisp, prop, val, ei, NULL);
1278 else
1279 hres = DISP_E_MEMBERNOTFOUND;
1281 jsdisp_release(jsdisp);
1282 }else {
1283 DISPID dispid = DISPID_PROPERTYPUT;
1284 DISPPARAMS dp = {val, &dispid, 1, 1};
1285 IDispatchEx *dispex;
1287 hres = IDispatch_QueryInterface(disp, &IID_IDispatchEx, (void**)&dispex);
1288 if(SUCCEEDED(hres)) {
1289 hres = IDispatchEx_InvokeEx(dispex, id, ctx->lcid, DISPATCH_PROPERTYPUT, &dp, NULL, &ei->ei,
1290 &ctx->jscaller->IServiceProvider_iface);
1291 IDispatchEx_Release(dispex);
1292 }else {
1293 ULONG err = 0;
1295 TRACE("using IDispatch\n");
1296 hres = IDispatch_Invoke(disp, id, &IID_NULL, ctx->lcid, DISPATCH_PROPERTYPUT, &dp, NULL, &ei->ei, &err);
1300 return hres;
1303 HRESULT jsdisp_propget_name(jsdisp_t *obj, const WCHAR *name, VARIANT *var, jsexcept_t *ei)
1305 DISPPARAMS dp = {NULL, NULL, 0, 0};
1306 dispex_prop_t *prop;
1307 HRESULT hres;
1309 hres = find_prop_name_prot(obj, string_hash(name), name, &prop);
1310 if(FAILED(hres))
1311 return hres;
1313 V_VT(var) = VT_EMPTY;
1314 if(!prop || prop->type==PROP_DELETED)
1315 return S_OK;
1317 return prop_get(obj, prop, &dp, var, ei, NULL);
1320 HRESULT jsdisp_get_idx(jsdisp_t *obj, DWORD idx, VARIANT *var, jsexcept_t *ei)
1322 WCHAR name[12];
1323 DISPPARAMS dp = {NULL, NULL, 0, 0};
1324 dispex_prop_t *prop;
1325 HRESULT hres;
1327 static const WCHAR formatW[] = {'%','d',0};
1329 sprintfW(name, formatW, idx);
1331 hres = find_prop_name_prot(obj, string_hash(name), name, &prop);
1332 if(FAILED(hres))
1333 return hres;
1335 V_VT(var) = VT_EMPTY;
1336 if(!prop || prop->type==PROP_DELETED)
1337 return DISP_E_UNKNOWNNAME;
1339 return prop_get(obj, prop, &dp, var, ei, NULL);
1342 HRESULT jsdisp_propget(jsdisp_t *jsdisp, DISPID id, VARIANT *val, jsexcept_t *ei)
1344 DISPPARAMS dp = {NULL,NULL,0,0};
1345 dispex_prop_t *prop;
1347 prop = get_prop(jsdisp, id);
1348 if(!prop)
1349 return DISP_E_MEMBERNOTFOUND;
1351 V_VT(val) = VT_EMPTY;
1352 return prop_get(jsdisp, prop, &dp, val, ei, NULL);
1355 HRESULT disp_propget(script_ctx_t *ctx, IDispatch *disp, DISPID id, VARIANT *val, jsexcept_t *ei)
1357 DISPPARAMS dp = {NULL,NULL,0,0};
1358 IDispatchEx *dispex;
1359 jsdisp_t *jsdisp;
1360 HRESULT hres;
1362 jsdisp = iface_to_jsdisp((IUnknown*)disp);
1363 if(jsdisp) {
1364 hres = jsdisp_propget(jsdisp, id, val, ei);
1365 jsdisp_release(jsdisp);
1366 return hres;
1369 hres = IDispatch_QueryInterface(disp, &IID_IDispatchEx, (void**)&dispex);
1370 if(SUCCEEDED(hres)) {
1371 hres = IDispatchEx_InvokeEx(dispex, id, ctx->lcid, INVOKE_PROPERTYGET, &dp, val, &ei->ei,
1372 &ctx->jscaller->IServiceProvider_iface);
1373 IDispatchEx_Release(dispex);
1374 }else {
1375 ULONG err = 0;
1377 TRACE("using IDispatch\n");
1378 hres = IDispatch_Invoke(disp, id, &IID_NULL, ctx->lcid, INVOKE_PROPERTYGET, &dp, val, &ei->ei, &err);
1380 if(FAILED(hres))
1381 return hres;
1383 ensure_retval_type(val);
1384 return S_OK;
1387 HRESULT jsdisp_delete_idx(jsdisp_t *obj, DWORD idx)
1389 static const WCHAR formatW[] = {'%','d',0};
1390 WCHAR buf[12];
1391 dispex_prop_t *prop;
1392 HRESULT hres;
1394 sprintfW(buf, formatW, idx);
1396 hres = find_prop_name(obj, string_hash(buf), buf, &prop);
1397 if(FAILED(hres) || !prop)
1398 return hres;
1400 return delete_prop(prop);
1403 VARIANT_BOOL jsdisp_is_own_prop(jsdisp_t *obj, BSTR name)
1405 dispex_prop_t *prop;
1406 HRESULT hres;
1408 hres = find_prop_name(obj, string_hash(name), name, &prop);
1409 if(FAILED(hres))
1410 return VARIANT_FALSE;
1411 else if(!prop)
1412 return VARIANT_FALSE;
1414 return prop->type==PROP_VARIANT ? VARIANT_TRUE : VARIANT_FALSE;