msvcp90: Use streambuf sgetc/snextc in std::getline(istream<> &).
[wine.git] / dlls / jscript / dispex.c
blob2be474edf5bb41fe1de52668bb83e319da22730a
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_JSVAL,
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 jsval_t val;
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_JSVAL;
280 prop->flags = create_flags;
281 prop->u.val = jsval_undefined();
282 }else {
283 prop = alloc_prop(This, name, PROP_JSVAL, create_flags);
284 if(!prop)
285 return E_OUTOFMEMORY;
288 prop->u.val = jsval_undefined();
291 *ret = prop;
292 return hres;
295 static IDispatch *get_this(DISPPARAMS *dp)
297 DWORD i;
299 for(i=0; i < dp->cNamedArgs; i++) {
300 if(dp->rgdispidNamedArgs[i] == DISPID_THIS) {
301 if(V_VT(dp->rgvarg+i) == VT_DISPATCH)
302 return V_DISPATCH(dp->rgvarg+i);
304 WARN("This is not VT_DISPATCH\n");
305 return NULL;
309 TRACE("no this passed\n");
310 return NULL;
313 static HRESULT convert_params(const DISPPARAMS *dp, jsval_t *buf, unsigned *argc, jsval_t **ret)
315 jsval_t *argv;
316 unsigned cnt;
317 unsigned i;
318 HRESULT hres;
320 cnt = dp->cArgs - dp->cNamedArgs;
322 if(cnt > 6) {
323 argv = heap_alloc(cnt * sizeof(*argv));
324 if(!argv)
325 return E_OUTOFMEMORY;
326 }else {
327 argv = buf;
330 for(i = 0; i < cnt; i++) {
331 hres = variant_to_jsval(dp->rgvarg+dp->cArgs-i-1, argv+i);
332 if(FAILED(hres)) {
333 while(i--)
334 jsval_release(argv[i]);
335 if(argv != buf)
336 heap_free(argv);
337 return hres;
341 *argc = cnt;
342 *ret = argv;
343 return S_OK;
346 static HRESULT invoke_prop_func(jsdisp_t *This, IDispatch *jsthis, dispex_prop_t *prop, WORD flags,
347 unsigned argc, jsval_t *argv, jsval_t *r, IServiceProvider *caller)
349 HRESULT hres;
351 switch(prop->type) {
352 case PROP_BUILTIN: {
353 if(flags == DISPATCH_CONSTRUCT && (prop->flags & PROPF_METHOD)) {
354 WARN("%s is not a constructor\n", debugstr_w(prop->name));
355 return E_INVALIDARG;
358 if(prop->name || This->builtin_info->class != JSCLASS_FUNCTION) {
359 vdisp_t vthis;
361 if(jsthis)
362 set_disp(&vthis, jsthis);
363 else
364 set_jsdisp(&vthis, This);
365 hres = prop->u.p->invoke(This->ctx, &vthis, flags, argc, argv, r);
366 vdisp_release(&vthis);
367 }else {
368 /* Function object calls are special case */
369 hres = Function_invoke(This, jsthis, flags, argc, argv, r);
371 return hres;
373 case PROP_PROTREF:
374 return invoke_prop_func(This->prototype, jsthis, This->prototype->props+prop->u.ref,
375 flags, argc, argv, r, caller);
376 case PROP_JSVAL: {
377 if(!is_object_instance(prop->u.val)) {
378 FIXME("invoke %s\n", debugstr_jsval(prop->u.val));
379 return E_FAIL;
382 TRACE("call %s %p\n", debugstr_w(prop->name), get_object(prop->u.val));
384 return disp_call_value(This->ctx, get_object(prop->u.val), jsthis, flags, argc, argv, r);
386 default:
387 ERR("type %d\n", prop->type);
390 return E_FAIL;
393 static HRESULT prop_get(jsdisp_t *This, dispex_prop_t *prop, DISPPARAMS *dp,
394 jsval_t *r, IServiceProvider *caller)
396 HRESULT hres;
398 switch(prop->type) {
399 case PROP_BUILTIN:
400 if(prop->u.p->flags & PROPF_METHOD) {
401 jsdisp_t *obj;
402 hres = create_builtin_function(This->ctx, prop->u.p->invoke, prop->u.p->name, NULL,
403 prop->u.p->flags, NULL, &obj);
404 if(FAILED(hres))
405 break;
407 prop->type = PROP_JSVAL;
408 prop->u.val = jsval_obj(obj);
410 jsdisp_addref(obj);
411 *r = jsval_obj(obj);
412 }else {
413 vdisp_t vthis;
415 set_jsdisp(&vthis, This);
416 hres = prop->u.p->invoke(This->ctx, &vthis, DISPATCH_PROPERTYGET, 0, NULL, r);
417 vdisp_release(&vthis);
419 break;
420 case PROP_PROTREF:
421 hres = prop_get(This->prototype, This->prototype->props+prop->u.ref, dp, r, caller);
422 break;
423 case PROP_JSVAL:
424 hres = jsval_copy(prop->u.val, r);
425 break;
426 default:
427 ERR("type %d\n", prop->type);
428 return E_FAIL;
431 if(FAILED(hres)) {
432 TRACE("fail %08x\n", hres);
433 return hres;
436 TRACE("%s ret %s\n", debugstr_w(prop->name), debugstr_jsval(*r));
437 return hres;
440 static HRESULT prop_put(jsdisp_t *This, dispex_prop_t *prop, jsval_t val, IServiceProvider *caller)
442 HRESULT hres;
444 if(prop->flags & PROPF_CONST)
445 return S_OK;
447 switch(prop->type) {
448 case PROP_BUILTIN:
449 if(!(prop->flags & PROPF_METHOD)) {
450 vdisp_t vthis;
452 set_jsdisp(&vthis, This);
453 hres = prop->u.p->invoke(This->ctx, &vthis, DISPATCH_PROPERTYPUT, 1, &val, NULL);
454 vdisp_release(&vthis);
455 return hres;
457 case PROP_PROTREF:
458 prop->type = PROP_JSVAL;
459 prop->flags = PROPF_ENUM;
460 prop->u.val = jsval_undefined();
461 break;
462 case PROP_JSVAL:
463 jsval_release(prop->u.val);
464 break;
465 default:
466 ERR("type %d\n", prop->type);
467 return E_FAIL;
470 TRACE("%s = %s\n", debugstr_w(prop->name), debugstr_jsval(val));
472 hres = jsval_copy(val, &prop->u.val);
473 if(FAILED(hres)) {
474 prop->u.val = jsval_undefined();
475 return hres;
478 if(This->builtin_info->on_put)
479 This->builtin_info->on_put(This, prop->name);
481 TRACE("%s = %s\n", debugstr_w(prop->name), debugstr_jsval(val));
482 return S_OK;
485 static HRESULT fill_protrefs(jsdisp_t *This)
487 dispex_prop_t *iter, *prop;
488 HRESULT hres;
490 if(!This->prototype)
491 return S_OK;
493 fill_protrefs(This->prototype);
495 for(iter = This->prototype->props; iter < This->prototype->props+This->prototype->prop_cnt; iter++) {
496 if(!iter->name)
497 continue;
498 hres = find_prop_name(This, iter->hash, iter->name, &prop);
499 if(FAILED(hres))
500 return hres;
501 if(!prop || prop->type==PROP_DELETED) {
502 if(prop) {
503 prop->type = PROP_PROTREF;
504 prop->flags = 0;
505 prop->u.ref = iter - This->prototype->props;
506 }else {
507 prop = alloc_protref(This, iter->name, iter - This->prototype->props);
508 if(!prop)
509 return E_OUTOFMEMORY;
514 return S_OK;
517 static inline jsdisp_t *impl_from_IDispatchEx(IDispatchEx *iface)
519 return CONTAINING_RECORD(iface, jsdisp_t, IDispatchEx_iface);
522 static HRESULT WINAPI DispatchEx_QueryInterface(IDispatchEx *iface, REFIID riid, void **ppv)
524 jsdisp_t *This = impl_from_IDispatchEx(iface);
526 if(IsEqualGUID(&IID_IUnknown, riid)) {
527 TRACE("(%p)->(IID_IUnknown %p)\n", This, ppv);
528 *ppv = &This->IDispatchEx_iface;
529 }else if(IsEqualGUID(&IID_IDispatch, riid)) {
530 TRACE("(%p)->(IID_IDispatch %p)\n", This, ppv);
531 *ppv = &This->IDispatchEx_iface;
532 }else if(IsEqualGUID(&IID_IDispatchEx, riid)) {
533 TRACE("(%p)->(IID_IDispatchEx %p)\n", This, ppv);
534 *ppv = &This->IDispatchEx_iface;
535 }else if(IsEqualGUID(&IID_IDispatchJS, riid)) {
536 TRACE("(%p)->(IID_IDispatchJS %p)\n", This, ppv);
537 jsdisp_addref(This);
538 *ppv = This;
539 return S_OK;
540 }else {
541 WARN("(%p)->(%s %p)\n", This, debugstr_guid(riid), ppv);
542 *ppv = NULL;
543 return E_NOINTERFACE;
546 jsdisp_addref(This);
547 return S_OK;
550 static ULONG WINAPI DispatchEx_AddRef(IDispatchEx *iface)
552 jsdisp_t *This = impl_from_IDispatchEx(iface);
553 jsdisp_addref(This);
554 return This->ref;
557 static ULONG WINAPI DispatchEx_Release(IDispatchEx *iface)
559 jsdisp_t *This = impl_from_IDispatchEx(iface);
560 ULONG ref = --This->ref;
561 if(!ref)
562 jsdisp_free(This);
563 return ref;
566 static HRESULT WINAPI DispatchEx_GetTypeInfoCount(IDispatchEx *iface, UINT *pctinfo)
568 jsdisp_t *This = impl_from_IDispatchEx(iface);
570 TRACE("(%p)->(%p)\n", This, pctinfo);
572 *pctinfo = 1;
573 return S_OK;
576 static HRESULT WINAPI DispatchEx_GetTypeInfo(IDispatchEx *iface, UINT iTInfo, LCID lcid,
577 ITypeInfo **ppTInfo)
579 jsdisp_t *This = impl_from_IDispatchEx(iface);
580 FIXME("(%p)->(%u %u %p)\n", This, iTInfo, lcid, ppTInfo);
581 return E_NOTIMPL;
584 static HRESULT WINAPI DispatchEx_GetIDsOfNames(IDispatchEx *iface, REFIID riid,
585 LPOLESTR *rgszNames, UINT cNames, LCID lcid,
586 DISPID *rgDispId)
588 jsdisp_t *This = impl_from_IDispatchEx(iface);
589 UINT i;
590 HRESULT hres;
592 TRACE("(%p)->(%s %p %u %u %p)\n", This, debugstr_guid(riid), rgszNames, cNames,
593 lcid, rgDispId);
595 for(i=0; i < cNames; i++) {
596 hres = IDispatchEx_GetDispID(&This->IDispatchEx_iface, rgszNames[i], 0, rgDispId+i);
597 if(FAILED(hres))
598 return hres;
601 return S_OK;
604 static HRESULT WINAPI DispatchEx_Invoke(IDispatchEx *iface, DISPID dispIdMember,
605 REFIID riid, LCID lcid, WORD wFlags, DISPPARAMS *pDispParams,
606 VARIANT *pVarResult, EXCEPINFO *pExcepInfo, UINT *puArgErr)
608 jsdisp_t *This = impl_from_IDispatchEx(iface);
610 TRACE("(%p)->(%d %s %d %d %p %p %p %p)\n", This, dispIdMember, debugstr_guid(riid),
611 lcid, wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr);
613 return IDispatchEx_InvokeEx(&This->IDispatchEx_iface, dispIdMember, lcid, wFlags,
614 pDispParams, pVarResult, pExcepInfo, NULL);
617 static HRESULT WINAPI DispatchEx_GetDispID(IDispatchEx *iface, BSTR bstrName, DWORD grfdex, DISPID *pid)
619 jsdisp_t *This = impl_from_IDispatchEx(iface);
621 TRACE("(%p)->(%s %x %p)\n", This, debugstr_w(bstrName), grfdex, pid);
623 if(grfdex & ~(fdexNameCaseSensitive|fdexNameEnsure|fdexNameImplicit|FDEX_VERSION_MASK)) {
624 FIXME("Unsupported grfdex %x\n", grfdex);
625 return E_NOTIMPL;
628 return jsdisp_get_id(This, bstrName, grfdex, pid);
631 static HRESULT WINAPI DispatchEx_InvokeEx(IDispatchEx *iface, DISPID id, LCID lcid, WORD wFlags, DISPPARAMS *pdp,
632 VARIANT *pvarRes, EXCEPINFO *pei, IServiceProvider *pspCaller)
634 jsdisp_t *This = impl_from_IDispatchEx(iface);
635 dispex_prop_t *prop;
636 HRESULT hres;
638 TRACE("(%p)->(%x %x %x %p %p %p %p)\n", This, id, lcid, wFlags, pdp, pvarRes, pei, pspCaller);
640 if(pvarRes)
641 V_VT(pvarRes) = VT_EMPTY;
643 prop = get_prop(This, id);
644 if(!prop || prop->type == PROP_DELETED) {
645 TRACE("invalid id\n");
646 return DISP_E_MEMBERNOTFOUND;
649 clear_ei(This->ctx);
651 switch(wFlags) {
652 case DISPATCH_METHOD|DISPATCH_PROPERTYGET:
653 wFlags = DISPATCH_METHOD;
654 /* fall through */
655 case DISPATCH_METHOD:
656 case DISPATCH_CONSTRUCT: {
657 jsval_t *argv, buf[6], r;
658 unsigned argc;
660 hres = convert_params(pdp, buf, &argc, &argv);
661 if(FAILED(hres))
662 return hres;
664 hres = invoke_prop_func(This, get_this(pdp), prop, wFlags, argc, argv, pvarRes ? &r : NULL, pspCaller);
665 if(argv != buf)
666 heap_free(argv);
667 if(SUCCEEDED(hres) && pvarRes) {
668 hres = jsval_to_variant(r, pvarRes);
669 jsval_release(r);
671 break;
673 case DISPATCH_PROPERTYGET: {
674 jsval_t r;
676 hres = prop_get(This, prop, pdp, &r, pspCaller);
677 if(SUCCEEDED(hres)) {
678 hres = jsval_to_variant(r, pvarRes);
679 jsval_release(r);
681 break;
683 case DISPATCH_PROPERTYPUT: {
684 jsval_t val;
685 DWORD i;
687 for(i=0; i < pdp->cNamedArgs; i++) {
688 if(pdp->rgdispidNamedArgs[i] == DISPID_PROPERTYPUT)
689 break;
692 if(i == pdp->cNamedArgs) {
693 TRACE("no value to set\n");
694 return DISP_E_PARAMNOTOPTIONAL;
697 hres = variant_to_jsval(pdp->rgvarg+i, &val);
698 if(FAILED(hres))
699 return hres;
701 hres = prop_put(This, prop, val, pspCaller);
702 jsval_release(val);
703 break;
705 default:
706 FIXME("Unimplemented flags %x\n", wFlags);
707 return E_INVALIDARG;
710 if(pei)
711 *pei = This->ctx->ei.ei;
712 return hres;
715 static HRESULT delete_prop(dispex_prop_t *prop)
717 if(prop->type == PROP_JSVAL) {
718 jsval_release(prop->u.val);
719 prop->type = PROP_DELETED;
721 return S_OK;
724 static HRESULT WINAPI DispatchEx_DeleteMemberByName(IDispatchEx *iface, BSTR bstrName, DWORD grfdex)
726 jsdisp_t *This = impl_from_IDispatchEx(iface);
727 dispex_prop_t *prop;
728 HRESULT hres;
730 TRACE("(%p)->(%s %x)\n", This, debugstr_w(bstrName), grfdex);
732 if(grfdex & ~(fdexNameCaseSensitive|fdexNameEnsure|fdexNameImplicit|FDEX_VERSION_MASK))
733 FIXME("Unsupported grfdex %x\n", grfdex);
735 hres = find_prop_name(This, string_hash(bstrName), bstrName, &prop);
736 if(FAILED(hres))
737 return hres;
738 if(!prop) {
739 TRACE("not found\n");
740 return S_OK;
743 return delete_prop(prop);
746 static HRESULT WINAPI DispatchEx_DeleteMemberByDispID(IDispatchEx *iface, DISPID id)
748 jsdisp_t *This = impl_from_IDispatchEx(iface);
749 dispex_prop_t *prop;
751 TRACE("(%p)->(%x)\n", This, id);
753 prop = get_prop(This, id);
754 if(!prop) {
755 WARN("invalid id\n");
756 return DISP_E_MEMBERNOTFOUND;
759 return delete_prop(prop);
762 static HRESULT WINAPI DispatchEx_GetMemberProperties(IDispatchEx *iface, DISPID id, DWORD grfdexFetch, DWORD *pgrfdex)
764 jsdisp_t *This = impl_from_IDispatchEx(iface);
765 FIXME("(%p)->(%x %x %p)\n", This, id, grfdexFetch, pgrfdex);
766 return E_NOTIMPL;
769 static HRESULT WINAPI DispatchEx_GetMemberName(IDispatchEx *iface, DISPID id, BSTR *pbstrName)
771 jsdisp_t *This = impl_from_IDispatchEx(iface);
772 dispex_prop_t *prop;
774 TRACE("(%p)->(%x %p)\n", This, id, pbstrName);
776 prop = get_prop(This, id);
777 if(!prop || !prop->name || prop->type == PROP_DELETED)
778 return DISP_E_MEMBERNOTFOUND;
780 *pbstrName = SysAllocString(prop->name);
781 if(!*pbstrName)
782 return E_OUTOFMEMORY;
784 return S_OK;
787 static HRESULT WINAPI DispatchEx_GetNextDispID(IDispatchEx *iface, DWORD grfdex, DISPID id, DISPID *pid)
789 jsdisp_t *This = impl_from_IDispatchEx(iface);
790 dispex_prop_t *iter;
791 HRESULT hres;
793 TRACE("(%p)->(%x %x %p)\n", This, grfdex, id, pid);
795 if(id == DISPID_STARTENUM) {
796 hres = fill_protrefs(This);
797 if(FAILED(hres))
798 return hres;
801 if(id+1>=0 && id+1<This->prop_cnt) {
802 iter = &This->props[id+1];
803 }else {
804 *pid = DISPID_STARTENUM;
805 return S_FALSE;
808 while(iter < This->props + This->prop_cnt) {
809 if(iter->name && (get_flags(This, iter) & PROPF_ENUM) && iter->type!=PROP_DELETED) {
810 *pid = prop_to_id(This, iter);
811 return S_OK;
813 iter++;
816 *pid = DISPID_STARTENUM;
817 return S_FALSE;
820 static HRESULT WINAPI DispatchEx_GetNameSpaceParent(IDispatchEx *iface, IUnknown **ppunk)
822 jsdisp_t *This = impl_from_IDispatchEx(iface);
823 FIXME("(%p)->(%p)\n", This, ppunk);
824 return E_NOTIMPL;
827 static IDispatchExVtbl DispatchExVtbl = {
828 DispatchEx_QueryInterface,
829 DispatchEx_AddRef,
830 DispatchEx_Release,
831 DispatchEx_GetTypeInfoCount,
832 DispatchEx_GetTypeInfo,
833 DispatchEx_GetIDsOfNames,
834 DispatchEx_Invoke,
835 DispatchEx_GetDispID,
836 DispatchEx_InvokeEx,
837 DispatchEx_DeleteMemberByName,
838 DispatchEx_DeleteMemberByDispID,
839 DispatchEx_GetMemberProperties,
840 DispatchEx_GetMemberName,
841 DispatchEx_GetNextDispID,
842 DispatchEx_GetNameSpaceParent
845 jsdisp_t *as_jsdisp(IDispatch *disp)
847 assert(disp->lpVtbl == (IDispatchVtbl*)&DispatchExVtbl);
848 return impl_from_IDispatchEx((IDispatchEx*)disp);
851 jsdisp_t *to_jsdisp(IDispatch *disp)
853 return disp->lpVtbl == (IDispatchVtbl*)&DispatchExVtbl ? impl_from_IDispatchEx((IDispatchEx*)disp) : NULL;
856 HRESULT init_dispex(jsdisp_t *dispex, script_ctx_t *ctx, const builtin_info_t *builtin_info, jsdisp_t *prototype)
858 TRACE("%p (%p)\n", dispex, prototype);
860 dispex->IDispatchEx_iface.lpVtbl = &DispatchExVtbl;
861 dispex->ref = 1;
862 dispex->builtin_info = builtin_info;
864 dispex->props = heap_alloc_zero(sizeof(dispex_prop_t)*(dispex->buf_size=4));
865 if(!dispex->props)
866 return E_OUTOFMEMORY;
868 dispex->prototype = prototype;
869 if(prototype)
870 jsdisp_addref(prototype);
872 dispex->prop_cnt = 1;
873 if(builtin_info->value_prop.invoke) {
874 dispex->props[0].type = PROP_BUILTIN;
875 dispex->props[0].u.p = &builtin_info->value_prop;
876 }else {
877 dispex->props[0].type = PROP_DELETED;
880 script_addref(ctx);
881 dispex->ctx = ctx;
883 return S_OK;
886 static const builtin_info_t dispex_info = {
887 JSCLASS_NONE,
888 {NULL, NULL, 0},
889 0, NULL,
890 NULL,
891 NULL
894 HRESULT create_dispex(script_ctx_t *ctx, const builtin_info_t *builtin_info, jsdisp_t *prototype, jsdisp_t **dispex)
896 jsdisp_t *ret;
897 HRESULT hres;
899 ret = heap_alloc_zero(sizeof(jsdisp_t));
900 if(!ret)
901 return E_OUTOFMEMORY;
903 hres = init_dispex(ret, ctx, builtin_info ? builtin_info : &dispex_info, prototype);
904 if(FAILED(hres)) {
905 heap_free(ret);
906 return hres;
909 *dispex = ret;
910 return S_OK;
913 void jsdisp_free(jsdisp_t *obj)
915 dispex_prop_t *prop;
917 TRACE("(%p)\n", obj);
919 for(prop = obj->props; prop < obj->props+obj->prop_cnt; prop++) {
920 if(prop->type == PROP_JSVAL)
921 jsval_release(prop->u.val);
922 heap_free(prop->name);
924 heap_free(obj->props);
925 script_release(obj->ctx);
926 if(obj->prototype)
927 jsdisp_release(obj->prototype);
929 if(obj->builtin_info->destructor)
930 obj->builtin_info->destructor(obj);
931 else
932 heap_free(obj);
935 #ifdef TRACE_REFCNT
937 jsdisp_t *jsdisp_addref(jsdisp_t *jsdisp)
939 ULONG ref = ++jsdisp->ref;
940 TRACE("(%p) ref=%d\n", jsdisp, ref);
941 return jsdisp;
944 void jsdisp_release(jsdisp_t *jsdisp)
946 ULONG ref = --jsdisp->ref;
948 TRACE("(%p) ref=%d\n", jsdisp, ref);
950 if(!ref)
951 jsdisp_free(jsdisp);
954 #endif
956 HRESULT init_dispex_from_constr(jsdisp_t *dispex, script_ctx_t *ctx, const builtin_info_t *builtin_info, jsdisp_t *constr)
958 jsdisp_t *prot = NULL;
959 dispex_prop_t *prop;
960 HRESULT hres;
962 static const WCHAR prototypeW[] = {'p','r','o','t','o','t','y','p','e',0};
964 hres = find_prop_name_prot(constr, string_hash(prototypeW), prototypeW, &prop);
965 if(SUCCEEDED(hres) && prop && prop->type!=PROP_DELETED) {
966 jsval_t val;
968 hres = prop_get(constr, prop, NULL, &val, NULL);
969 if(FAILED(hres)) {
970 ERR("Could not get prototype\n");
971 return hres;
974 if(is_object_instance(val))
975 prot = iface_to_jsdisp((IUnknown*)get_object(val));
976 jsval_release(val);
979 hres = init_dispex(dispex, ctx, builtin_info, prot);
981 if(prot)
982 jsdisp_release(prot);
983 return hres;
986 jsdisp_t *iface_to_jsdisp(IUnknown *iface)
988 jsdisp_t *ret;
989 HRESULT hres;
991 hres = IUnknown_QueryInterface(iface, &IID_IDispatchJS, (void**)&ret);
992 if(FAILED(hres))
993 return NULL;
995 return ret;
998 HRESULT jsdisp_get_id(jsdisp_t *jsdisp, const WCHAR *name, DWORD flags, DISPID *id)
1000 dispex_prop_t *prop;
1001 HRESULT hres;
1003 if(flags & fdexNameEnsure)
1004 hres = ensure_prop_name(jsdisp, name, TRUE, PROPF_ENUM, &prop);
1005 else
1006 hres = find_prop_name_prot(jsdisp, string_hash(name), name, &prop);
1007 if(FAILED(hres))
1008 return hres;
1010 if(prop && prop->type!=PROP_DELETED) {
1011 *id = prop_to_id(jsdisp, prop);
1012 return S_OK;
1015 TRACE("not found %s\n", debugstr_w(name));
1016 return DISP_E_UNKNOWNNAME;
1019 HRESULT jsdisp_call_value(jsdisp_t *jsfunc, IDispatch *jsthis, WORD flags, unsigned argc, jsval_t *argv, jsval_t *r)
1021 HRESULT hres;
1023 if(is_class(jsfunc, JSCLASS_FUNCTION)) {
1024 hres = Function_invoke(jsfunc, jsthis, flags, argc, argv, r);
1025 }else {
1026 vdisp_t vdisp;
1028 set_disp(&vdisp, jsthis);
1029 hres = jsfunc->builtin_info->value_prop.invoke(jsfunc->ctx, &vdisp, flags, argc, argv, r);
1030 vdisp_release(&vdisp);
1032 return hres;
1035 HRESULT jsdisp_call(jsdisp_t *disp, DISPID id, WORD flags, unsigned argc, jsval_t *argv, jsval_t *r)
1037 dispex_prop_t *prop;
1039 prop = get_prop(disp, id);
1040 if(!prop)
1041 return DISP_E_MEMBERNOTFOUND;
1043 return invoke_prop_func(disp, to_disp(disp), prop, flags, argc, argv, r, NULL);
1046 HRESULT jsdisp_call_name(jsdisp_t *disp, const WCHAR *name, WORD flags, unsigned argc, jsval_t *argv, jsval_t *r)
1048 dispex_prop_t *prop;
1049 HRESULT hres;
1051 hres = find_prop_name_prot(disp, string_hash(name), name, &prop);
1052 if(FAILED(hres))
1053 return hres;
1055 return invoke_prop_func(disp, to_disp(disp), prop, flags, argc, argv, r, NULL);
1058 HRESULT disp_call(script_ctx_t *ctx, IDispatch *disp, DISPID id, WORD flags, unsigned argc, jsval_t *argv, jsval_t *ret)
1060 IDispatchEx *dispex;
1061 jsdisp_t *jsdisp;
1062 VARIANT buf[6], retv;
1063 DISPPARAMS dp;
1064 unsigned i;
1065 HRESULT hres;
1067 jsdisp = iface_to_jsdisp((IUnknown*)disp);
1068 if(jsdisp) {
1069 if(flags & DISPATCH_PROPERTYPUT) {
1070 FIXME("disp_call(propput) on builtin object\n");
1071 return E_FAIL;
1074 hres = jsdisp_call(jsdisp, id, flags, argc, argv, ret);
1075 jsdisp_release(jsdisp);
1076 return hres;
1079 if(ret && argc)
1080 flags |= DISPATCH_PROPERTYGET;
1082 dp.cArgs = argc;
1084 if(flags & DISPATCH_PROPERTYPUT) {
1085 static DISPID propput_dispid = DISPID_PROPERTYPUT;
1087 dp.cNamedArgs = 1;
1088 dp.rgdispidNamedArgs = &propput_dispid;
1089 }else {
1090 dp.cNamedArgs = 0;
1091 dp.rgdispidNamedArgs = NULL;
1094 if(argc > 6) {
1095 dp.rgvarg = heap_alloc(argc*sizeof(VARIANT));
1096 if(!dp.rgvarg)
1097 return E_OUTOFMEMORY;
1098 }else {
1099 dp.rgvarg = buf;
1102 for(i=0; i<argc; i++) {
1103 hres = jsval_to_variant(argv[i], dp.rgvarg+argc-i-1);
1104 if(FAILED(hres)) {
1105 while(i--)
1106 VariantClear(dp.rgvarg+argc-i-1);
1107 if(dp.rgvarg != buf)
1108 heap_free(dp.rgvarg);
1109 return hres;
1113 V_VT(&retv) = VT_EMPTY;
1114 clear_ei(ctx);
1115 hres = IDispatch_QueryInterface(disp, &IID_IDispatchEx, (void**)&dispex);
1116 if(SUCCEEDED(hres)) {
1117 hres = IDispatchEx_InvokeEx(dispex, id, ctx->lcid, flags, &dp, ret ? &retv : NULL, &ctx->ei.ei,
1118 &ctx->jscaller->IServiceProvider_iface);
1119 IDispatchEx_Release(dispex);
1120 }else {
1121 UINT err = 0;
1123 if(flags == DISPATCH_CONSTRUCT) {
1124 WARN("IDispatch cannot be constructor\n");
1125 return DISP_E_MEMBERNOTFOUND;
1128 TRACE("using IDispatch\n");
1129 hres = IDispatch_Invoke(disp, id, &IID_NULL, ctx->lcid, flags, &dp, ret ? &retv : NULL, &ctx->ei.ei, &err);
1132 for(i=0; i<argc; i++)
1133 VariantClear(dp.rgvarg+argc-i-1);
1134 if(dp.rgvarg != buf)
1135 heap_free(dp.rgvarg);
1136 if(FAILED(hres))
1137 return hres;
1139 if(ret) {
1140 hres = variant_to_jsval(&retv, ret);
1141 VariantClear(&retv);
1143 return hres;
1146 HRESULT disp_call_value(script_ctx_t *ctx, IDispatch *disp, IDispatch *jsthis, WORD flags, unsigned argc, jsval_t *argv,
1147 jsval_t *r)
1149 jsdisp_t *jsdisp;
1150 IDispatchEx *dispex;
1151 VARIANT buf[6], retv;
1152 DISPPARAMS dp;
1153 unsigned i;
1154 HRESULT hres;
1156 jsdisp = iface_to_jsdisp((IUnknown*)disp);
1157 if(jsdisp) {
1158 if(flags & DISPATCH_PROPERTYPUT) {
1159 FIXME("disp_call(propput) on builtin object\n");
1160 return E_FAIL;
1163 hres = jsdisp_call_value(jsdisp, jsthis, flags, argc, argv, r);
1164 jsdisp_release(jsdisp);
1165 return hres;
1168 if(r && argc)
1169 flags |= DISPATCH_PROPERTYGET;
1172 if(jsthis) {
1173 static DISPID this_id = DISPID_THIS;
1175 dp.cArgs = argc+1;
1176 dp.cNamedArgs = 1;
1177 dp.rgdispidNamedArgs = &this_id;
1178 }else {
1179 dp.cArgs = argc;
1180 dp.cNamedArgs = 0;
1181 dp.rgdispidNamedArgs = NULL;
1184 if(dp.cArgs > sizeof(buf)/sizeof(*buf)) {
1185 dp.rgvarg = heap_alloc(dp.cArgs*sizeof(VARIANT));
1186 if(!dp.rgvarg)
1187 return E_OUTOFMEMORY;
1188 }else {
1189 dp.rgvarg = buf;
1192 for(i=0; i<argc; i++) {
1193 hres = jsval_to_variant(argv[i], dp.rgvarg+dp.cArgs-i-1);
1194 if(FAILED(hres)) {
1195 while(i--)
1196 VariantClear(dp.rgvarg+dp.cArgs-i-1);
1197 if(dp.rgvarg != buf)
1198 heap_free(dp.rgvarg);
1199 return hres;
1202 if(jsthis) {
1203 V_VT(dp.rgvarg) = VT_DISPATCH;
1204 V_DISPATCH(dp.rgvarg) = jsthis;
1207 V_VT(&retv) = VT_EMPTY;
1208 clear_ei(ctx);
1209 hres = IDispatch_QueryInterface(disp, &IID_IDispatchEx, (void**)&dispex);
1210 if(SUCCEEDED(hres)) {
1211 hres = IDispatchEx_InvokeEx(dispex, DISPID_VALUE, ctx->lcid, flags, &dp, r ? &retv : NULL, &ctx->ei.ei,
1212 &ctx->jscaller->IServiceProvider_iface);
1213 IDispatchEx_Release(dispex);
1214 }else {
1215 UINT err = 0;
1217 if(flags == DISPATCH_CONSTRUCT) {
1218 WARN("IDispatch cannot be constructor\n");
1219 return DISP_E_MEMBERNOTFOUND;
1222 TRACE("using IDispatch\n");
1223 hres = IDispatch_Invoke(disp, DISPID_VALUE, &IID_NULL, ctx->lcid, flags, &dp, r ? &retv : NULL, &ctx->ei.ei, &err);
1226 for(i=0; i<argc; i++)
1227 VariantClear(dp.rgvarg+dp.cArgs-i-1);
1228 if(dp.rgvarg != buf)
1229 heap_free(dp.rgvarg);
1230 if(FAILED(hres))
1231 return hres;
1233 if(!r)
1234 return S_OK;
1236 hres = variant_to_jsval(&retv, r);
1237 VariantClear(&retv);
1238 return hres;
1241 HRESULT jsdisp_propput_name(jsdisp_t *obj, const WCHAR *name, jsval_t val)
1243 dispex_prop_t *prop;
1244 HRESULT hres;
1246 hres = ensure_prop_name(obj, name, FALSE, PROPF_ENUM, &prop);
1247 if(FAILED(hres))
1248 return hres;
1250 return prop_put(obj, prop, val, NULL);
1253 HRESULT jsdisp_propput_const(jsdisp_t *obj, const WCHAR *name, jsval_t val)
1255 dispex_prop_t *prop;
1256 HRESULT hres;
1258 hres = ensure_prop_name(obj, name, FALSE, PROPF_ENUM|PROPF_CONST, &prop);
1259 if(FAILED(hres))
1260 return hres;
1262 return jsval_copy(val, &prop->u.val);
1265 HRESULT jsdisp_propput_dontenum(jsdisp_t *obj, const WCHAR *name, jsval_t val)
1267 dispex_prop_t *prop;
1268 HRESULT hres;
1270 hres = ensure_prop_name(obj, name, FALSE, 0, &prop);
1271 if(FAILED(hres))
1272 return hres;
1274 return jsval_copy(val, &prop->u.val);
1277 HRESULT jsdisp_propput_idx(jsdisp_t *obj, DWORD idx, jsval_t val)
1279 WCHAR buf[12];
1281 static const WCHAR formatW[] = {'%','d',0};
1283 sprintfW(buf, formatW, idx);
1284 return jsdisp_propput_name(obj, buf, val);
1287 HRESULT disp_propput(script_ctx_t *ctx, IDispatch *disp, DISPID id, jsval_t val)
1289 jsdisp_t *jsdisp;
1290 HRESULT hres;
1292 jsdisp = iface_to_jsdisp((IUnknown*)disp);
1293 if(jsdisp) {
1294 dispex_prop_t *prop;
1296 prop = get_prop(jsdisp, id);
1297 if(prop)
1298 hres = prop_put(jsdisp, prop, val, NULL);
1299 else
1300 hres = DISP_E_MEMBERNOTFOUND;
1302 jsdisp_release(jsdisp);
1303 }else {
1304 DISPID dispid = DISPID_PROPERTYPUT;
1305 VARIANT var;
1306 DISPPARAMS dp = {&var, &dispid, 1, 1};
1307 IDispatchEx *dispex;
1309 hres = jsval_to_variant(val, &var);
1310 if(FAILED(hres))
1311 return hres;
1313 clear_ei(ctx);
1314 hres = IDispatch_QueryInterface(disp, &IID_IDispatchEx, (void**)&dispex);
1315 if(SUCCEEDED(hres)) {
1316 hres = IDispatchEx_InvokeEx(dispex, id, ctx->lcid, DISPATCH_PROPERTYPUT, &dp, NULL, &ctx->ei.ei,
1317 &ctx->jscaller->IServiceProvider_iface);
1318 IDispatchEx_Release(dispex);
1319 }else {
1320 ULONG err = 0;
1322 TRACE("using IDispatch\n");
1323 hres = IDispatch_Invoke(disp, id, &IID_NULL, ctx->lcid, DISPATCH_PROPERTYPUT, &dp, NULL, &ctx->ei.ei, &err);
1326 VariantClear(&var);
1329 return hres;
1332 HRESULT jsdisp_propget_name(jsdisp_t *obj, const WCHAR *name, jsval_t *val)
1334 DISPPARAMS dp = {NULL, NULL, 0, 0};
1335 dispex_prop_t *prop;
1336 HRESULT hres;
1338 hres = find_prop_name_prot(obj, string_hash(name), name, &prop);
1339 if(FAILED(hres))
1340 return hres;
1342 if(!prop || prop->type==PROP_DELETED) {
1343 *val = jsval_undefined();
1344 return S_OK;
1347 return prop_get(obj, prop, &dp, val, NULL);
1350 HRESULT jsdisp_get_idx(jsdisp_t *obj, DWORD idx, jsval_t *r)
1352 WCHAR name[12];
1353 DISPPARAMS dp = {NULL, NULL, 0, 0};
1354 dispex_prop_t *prop;
1355 HRESULT hres;
1357 static const WCHAR formatW[] = {'%','d',0};
1359 sprintfW(name, formatW, idx);
1361 hres = find_prop_name_prot(obj, string_hash(name), name, &prop);
1362 if(FAILED(hres))
1363 return hres;
1365 if(!prop || prop->type==PROP_DELETED) {
1366 *r = jsval_undefined();
1367 return DISP_E_UNKNOWNNAME;
1370 return prop_get(obj, prop, &dp, r, NULL);
1373 HRESULT jsdisp_propget(jsdisp_t *jsdisp, DISPID id, jsval_t *val)
1375 DISPPARAMS dp = {NULL,NULL,0,0};
1376 dispex_prop_t *prop;
1378 prop = get_prop(jsdisp, id);
1379 if(!prop)
1380 return DISP_E_MEMBERNOTFOUND;
1382 return prop_get(jsdisp, prop, &dp, val, NULL);
1385 HRESULT disp_propget(script_ctx_t *ctx, IDispatch *disp, DISPID id, jsval_t *val)
1387 DISPPARAMS dp = {NULL,NULL,0,0};
1388 IDispatchEx *dispex;
1389 jsdisp_t *jsdisp;
1390 VARIANT var;
1391 HRESULT hres;
1393 jsdisp = iface_to_jsdisp((IUnknown*)disp);
1394 if(jsdisp) {
1395 hres = jsdisp_propget(jsdisp, id, val);
1396 jsdisp_release(jsdisp);
1397 return hres;
1400 V_VT(&var) = VT_EMPTY;
1401 clear_ei(ctx);
1402 hres = IDispatch_QueryInterface(disp, &IID_IDispatchEx, (void**)&dispex);
1403 if(SUCCEEDED(hres)) {
1404 hres = IDispatchEx_InvokeEx(dispex, id, ctx->lcid, INVOKE_PROPERTYGET, &dp, &var, &ctx->ei.ei,
1405 &ctx->jscaller->IServiceProvider_iface);
1406 IDispatchEx_Release(dispex);
1407 }else {
1408 ULONG err = 0;
1410 TRACE("using IDispatch\n");
1411 hres = IDispatch_Invoke(disp, id, &IID_NULL, ctx->lcid, INVOKE_PROPERTYGET, &dp, &var, &ctx->ei.ei, &err);
1413 if(FAILED(hres))
1414 return hres;
1416 return variant_to_jsval(&var, val);
1419 HRESULT jsdisp_delete_idx(jsdisp_t *obj, DWORD idx)
1421 static const WCHAR formatW[] = {'%','d',0};
1422 WCHAR buf[12];
1423 dispex_prop_t *prop;
1424 HRESULT hres;
1426 sprintfW(buf, formatW, idx);
1428 hres = find_prop_name(obj, string_hash(buf), buf, &prop);
1429 if(FAILED(hres) || !prop)
1430 return hres;
1432 return delete_prop(prop);
1435 HRESULT jsdisp_is_own_prop(jsdisp_t *obj, const WCHAR *name, BOOL *ret)
1437 dispex_prop_t *prop;
1438 HRESULT hres;
1440 hres = find_prop_name(obj, string_hash(name), name, &prop);
1441 if(FAILED(hres))
1442 return hres;
1444 *ret = prop && (prop->type == PROP_JSVAL || prop->type == PROP_BUILTIN);
1445 return S_OK;