2 * Copyright 2011 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
23 #include "wine/debug.h"
25 WINE_DEFAULT_DEBUG_CHANNEL(vbscript
);
27 static inline BOOL
is_func_id(vbdisp_t
*This
, DISPID id
)
29 return id
< This
->desc
->func_cnt
;
32 static BOOL
get_func_id(vbdisp_t
*This
, const WCHAR
*name
, vbdisp_invoke_type_t invoke_type
, BOOL search_private
, DISPID
*id
)
36 for(i
= invoke_type
== VBDISP_ANY
? 0 : 1; i
< This
->desc
->func_cnt
; i
++) {
37 if(invoke_type
== VBDISP_ANY
) {
38 if(!search_private
&& !This
->desc
->funcs
[i
].is_public
)
40 if(!i
&& !This
->desc
->funcs
[0].name
) /* default value may not exist */
43 if(!This
->desc
->funcs
[i
].entries
[invoke_type
]
44 || (!search_private
&& !This
->desc
->funcs
[i
].entries
[invoke_type
]->is_public
))
48 if(!strcmpiW(This
->desc
->funcs
[i
].name
, name
)) {
57 HRESULT
vbdisp_get_id(vbdisp_t
*This
, BSTR name
, vbdisp_invoke_type_t invoke_type
, BOOL search_private
, DISPID
*id
)
61 if(get_func_id(This
, name
, invoke_type
, search_private
, id
))
64 for(i
=0; i
< This
->desc
->prop_cnt
; i
++) {
65 if(!search_private
&& !This
->desc
->props
[i
].is_public
)
68 if(!strcmpiW(This
->desc
->props
[i
].name
, name
)) {
69 *id
= i
+ This
->desc
->func_cnt
;
74 if(This
->desc
->typeinfo
) {
77 hres
= ITypeInfo_GetIDsOfNames(This
->desc
->typeinfo
, &name
, 1, id
);
83 return DISP_E_UNKNOWNNAME
;
86 static VARIANT
*get_propput_arg(const DISPPARAMS
*dp
)
90 for(i
=0; i
< dp
->cNamedArgs
; i
++) {
91 if(dp
->rgdispidNamedArgs
[i
] == DISPID_PROPERTYPUT
)
98 static HRESULT
invoke_variant_prop(vbdisp_t
*This
, VARIANT
*v
, WORD flags
, DISPPARAMS
*dp
, VARIANT
*res
)
103 case DISPATCH_PROPERTYGET
|DISPATCH_METHOD
:
105 WARN("called with arguments\n");
106 return DISP_E_MEMBERNOTFOUND
; /* That's what tests show */
109 hres
= VariantCopy(res
, v
);
112 case DISPATCH_PROPERTYPUT
: {
115 put_val
= get_propput_arg(dp
);
117 WARN("no value to set\n");
118 return DISP_E_PARAMNOTOPTIONAL
;
122 V_VT(res
) = VT_EMPTY
;
124 hres
= VariantCopy(v
, put_val
);
129 FIXME("unimplemented flags %x\n", flags
);
136 static HRESULT
invoke_builtin(vbdisp_t
*This
, const builtin_prop_t
*prop
, WORD flags
, DISPPARAMS
*dp
, VARIANT
*res
)
138 VARIANT
*args
, arg_buf
[8];
142 case DISPATCH_PROPERTYGET
:
143 if(!(prop
->flags
& (BP_GET
|BP_GETPUT
))) {
144 FIXME("property does not support DISPATCH_PROPERTYGET\n");
148 case DISPATCH_PROPERTYGET
|DISPATCH_METHOD
:
150 case DISPATCH_METHOD
:
151 if(prop
->flags
& (BP_GET
|BP_GETPUT
)) {
152 FIXME("Call on property\n");
156 case DISPATCH_PROPERTYPUT
:
157 if(!(prop
->flags
& (BP_GET
|BP_GETPUT
))) {
158 FIXME("property does not support DISPATCH_PROPERTYPUT\n");
165 FIXME("unsupported flags %x\n", flags
);
171 if(argn
< prop
->min_args
|| argn
> (prop
->max_args
? prop
->max_args
: prop
->min_args
)) {
172 FIXME("invalid number of arguments\n");
179 if(V_VT(dp
->rgvarg
) == (VT_BYREF
|VT_VARIANT
))
180 args
= V_VARIANTREF(dp
->rgvarg
);
184 assert(argn
< sizeof(arg_buf
)/sizeof(*arg_buf
));
186 for(i
=0; i
< argn
; i
++) {
187 if(V_VT(dp
->rgvarg
+i
) == (VT_BYREF
|VT_VARIANT
)) {
188 for(i
=0; i
< argn
; i
++) {
189 if(V_VT(dp
->rgvarg
+i
) == (VT_BYREF
|VT_VARIANT
))
190 arg_buf
[i
] = *V_VARIANTREF(dp
->rgvarg
+i
);
192 arg_buf
[i
] = dp
->rgvarg
[i
];
200 return prop
->proc(This
, args
, dp
->cArgs
, res
);
203 static BOOL
run_terminator(vbdisp_t
*This
)
207 if(This
->terminator_ran
)
209 This
->terminator_ran
= TRUE
;
211 if(!This
->desc
->class_terminate_id
)
215 exec_script(This
->desc
->ctx
, This
->desc
->funcs
[This
->desc
->class_terminate_id
].entries
[VBDISP_CALLGET
],
216 (IDispatch
*)&This
->IDispatchEx_iface
, &dp
, NULL
);
220 static void clean_props(vbdisp_t
*This
)
227 for(i
=0; i
< This
->desc
->prop_cnt
; i
++)
228 VariantClear(This
->props
+i
);
231 static inline vbdisp_t
*impl_from_IDispatchEx(IDispatchEx
*iface
)
233 return CONTAINING_RECORD(iface
, vbdisp_t
, IDispatchEx_iface
);
236 static HRESULT WINAPI
DispatchEx_QueryInterface(IDispatchEx
*iface
, REFIID riid
, void **ppv
)
238 vbdisp_t
*This
= impl_from_IDispatchEx(iface
);
240 if(IsEqualGUID(&IID_IUnknown
, riid
)) {
241 TRACE("(%p)->(IID_IUnknown %p)\n", This
, ppv
);
242 *ppv
= &This
->IDispatchEx_iface
;
243 }else if(IsEqualGUID(&IID_IDispatch
, riid
)) {
244 TRACE("(%p)->(IID_IDispatch %p)\n", This
, ppv
);
245 *ppv
= &This
->IDispatchEx_iface
;
246 }else if(IsEqualGUID(&IID_IDispatchEx
, riid
)) {
247 TRACE("(%p)->(IID_IDispatchEx %p)\n", This
, ppv
);
248 *ppv
= &This
->IDispatchEx_iface
;
250 WARN("(%p)->(%s %p)\n", This
, debugstr_guid(riid
), ppv
);
252 return E_NOINTERFACE
;
255 IUnknown_AddRef((IUnknown
*)*ppv
);
259 static ULONG WINAPI
DispatchEx_AddRef(IDispatchEx
*iface
)
261 vbdisp_t
*This
= impl_from_IDispatchEx(iface
);
262 LONG ref
= InterlockedIncrement(&This
->ref
);
264 TRACE("(%p) ref=%d\n", This
, ref
);
269 static ULONG WINAPI
DispatchEx_Release(IDispatchEx
*iface
)
271 vbdisp_t
*This
= impl_from_IDispatchEx(iface
);
272 LONG ref
= InterlockedDecrement(&This
->ref
);
274 TRACE("(%p) ref=%d\n", This
, ref
);
276 if(!ref
&& run_terminator(This
)) {
278 list_remove(&This
->entry
);
285 static HRESULT WINAPI
DispatchEx_GetTypeInfoCount(IDispatchEx
*iface
, UINT
*pctinfo
)
287 vbdisp_t
*This
= impl_from_IDispatchEx(iface
);
289 TRACE("(%p)->(%p)\n", This
, pctinfo
);
295 static HRESULT WINAPI
DispatchEx_GetTypeInfo(IDispatchEx
*iface
, UINT iTInfo
, LCID lcid
,
298 vbdisp_t
*This
= impl_from_IDispatchEx(iface
);
299 FIXME("(%p)->(%u %u %p)\n", This
, iTInfo
, lcid
, ppTInfo
);
303 static HRESULT WINAPI
DispatchEx_GetIDsOfNames(IDispatchEx
*iface
, REFIID riid
,
304 LPOLESTR
*rgszNames
, UINT cNames
, LCID lcid
,
307 vbdisp_t
*This
= impl_from_IDispatchEx(iface
);
308 FIXME("(%p)->(%s %p %u %u %p)\n", This
, debugstr_guid(riid
), rgszNames
, cNames
,
313 static HRESULT WINAPI
DispatchEx_Invoke(IDispatchEx
*iface
, DISPID dispIdMember
,
314 REFIID riid
, LCID lcid
, WORD wFlags
, DISPPARAMS
*pDispParams
,
315 VARIANT
*pVarResult
, EXCEPINFO
*pExcepInfo
, UINT
*puArgErr
)
317 vbdisp_t
*This
= impl_from_IDispatchEx(iface
);
318 FIXME("(%p)->(%d %s %d %d %p %p %p %p)\n", This
, dispIdMember
, debugstr_guid(riid
),
319 lcid
, wFlags
, pDispParams
, pVarResult
, pExcepInfo
, puArgErr
);
323 static HRESULT WINAPI
DispatchEx_GetDispID(IDispatchEx
*iface
, BSTR bstrName
, DWORD grfdex
, DISPID
*pid
)
325 vbdisp_t
*This
= impl_from_IDispatchEx(iface
);
327 TRACE("(%p)->(%s %x %p)\n", This
, debugstr_w(bstrName
), grfdex
, pid
);
332 if(grfdex
& ~(fdexNameEnsure
|fdexNameCaseInsensitive
)) {
333 FIXME("unsupported flags %x\n", grfdex
);
337 return vbdisp_get_id(This
, bstrName
, VBDISP_ANY
, FALSE
, pid
);
340 static HRESULT WINAPI
DispatchEx_InvokeEx(IDispatchEx
*iface
, DISPID id
, LCID lcid
, WORD wFlags
, DISPPARAMS
*pdp
,
341 VARIANT
*pvarRes
, EXCEPINFO
*pei
, IServiceProvider
*pspCaller
)
343 vbdisp_t
*This
= impl_from_IDispatchEx(iface
);
345 TRACE("(%p)->(%x %x %x %p %p %p %p)\n", This
, id
, lcid
, wFlags
, pdp
, pvarRes
, pei
, pspCaller
);
351 V_VT(pvarRes
) = VT_EMPTY
;
354 return DISP_E_MEMBERNOTFOUND
;
356 if(is_func_id(This
, id
)) {
360 case DISPATCH_METHOD
:
361 case DISPATCH_METHOD
|DISPATCH_PROPERTYGET
:
362 func
= This
->desc
->funcs
[id
].entries
[VBDISP_CALLGET
];
364 FIXME("no invoke/getter\n");
365 return DISP_E_MEMBERNOTFOUND
;
368 return exec_script(This
->desc
->ctx
, func
, (IDispatch
*)&This
->IDispatchEx_iface
, pdp
, pvarRes
);
369 case DISPATCH_PROPERTYPUT
: {
371 DISPPARAMS dp
= {NULL
, NULL
, 1, 0};
374 FIXME("arguments not implemented\n");
378 put_val
= get_propput_arg(pdp
);
380 WARN("no value to set\n");
381 return DISP_E_PARAMNOTOPTIONAL
;
385 func
= This
->desc
->funcs
[id
].entries
[V_VT(put_val
) == VT_DISPATCH
? VBDISP_SET
: VBDISP_LET
];
387 FIXME("no letter/setter\n");
388 return DISP_E_MEMBERNOTFOUND
;
391 return exec_script(This
->desc
->ctx
, func
, (IDispatch
*)&This
->IDispatchEx_iface
, &dp
, NULL
);
394 FIXME("flags %x\n", wFlags
);
395 return DISP_E_MEMBERNOTFOUND
;
399 if(id
< This
->desc
->prop_cnt
+ This
->desc
->func_cnt
)
400 return invoke_variant_prop(This
, This
->props
+(id
-This
->desc
->func_cnt
), wFlags
, pdp
, pvarRes
);
402 if(This
->desc
->builtin_prop_cnt
) {
403 unsigned min
= 0, max
= This
->desc
->builtin_prop_cnt
-1, i
;
407 if(This
->desc
->builtin_props
[i
].id
== id
)
408 return invoke_builtin(This
, This
->desc
->builtin_props
+i
, wFlags
, pdp
, pvarRes
);
409 if(This
->desc
->builtin_props
[i
].id
< id
)
416 return DISP_E_MEMBERNOTFOUND
;
419 static HRESULT WINAPI
DispatchEx_DeleteMemberByName(IDispatchEx
*iface
, BSTR bstrName
, DWORD grfdex
)
421 vbdisp_t
*This
= impl_from_IDispatchEx(iface
);
422 FIXME("(%p)->(%s %x)\n", This
, debugstr_w(bstrName
), grfdex
);
426 static HRESULT WINAPI
DispatchEx_DeleteMemberByDispID(IDispatchEx
*iface
, DISPID id
)
428 vbdisp_t
*This
= impl_from_IDispatchEx(iface
);
429 FIXME("(%p)->(%x)\n", This
, id
);
433 static HRESULT WINAPI
DispatchEx_GetMemberProperties(IDispatchEx
*iface
, DISPID id
, DWORD grfdexFetch
, DWORD
*pgrfdex
)
435 vbdisp_t
*This
= impl_from_IDispatchEx(iface
);
436 FIXME("(%p)->(%x %x %p)\n", This
, id
, grfdexFetch
, pgrfdex
);
440 static HRESULT WINAPI
DispatchEx_GetMemberName(IDispatchEx
*iface
, DISPID id
, BSTR
*pbstrName
)
442 vbdisp_t
*This
= impl_from_IDispatchEx(iface
);
443 FIXME("(%p)->(%x %p)\n", This
, id
, pbstrName
);
447 static HRESULT WINAPI
DispatchEx_GetNextDispID(IDispatchEx
*iface
, DWORD grfdex
, DISPID id
, DISPID
*pid
)
449 vbdisp_t
*This
= impl_from_IDispatchEx(iface
);
450 FIXME("(%p)->(%x %x %p)\n", This
, grfdex
, id
, pid
);
454 static HRESULT WINAPI
DispatchEx_GetNameSpaceParent(IDispatchEx
*iface
, IUnknown
**ppunk
)
456 vbdisp_t
*This
= impl_from_IDispatchEx(iface
);
457 FIXME("(%p)->(%p)\n", This
, ppunk
);
461 static IDispatchExVtbl DispatchExVtbl
= {
462 DispatchEx_QueryInterface
,
465 DispatchEx_GetTypeInfoCount
,
466 DispatchEx_GetTypeInfo
,
467 DispatchEx_GetIDsOfNames
,
469 DispatchEx_GetDispID
,
471 DispatchEx_DeleteMemberByName
,
472 DispatchEx_DeleteMemberByDispID
,
473 DispatchEx_GetMemberProperties
,
474 DispatchEx_GetMemberName
,
475 DispatchEx_GetNextDispID
,
476 DispatchEx_GetNameSpaceParent
479 static inline vbdisp_t
*unsafe_impl_from_IDispatch(IDispatch
*iface
)
481 return iface
->lpVtbl
== (IDispatchVtbl
*)&DispatchExVtbl
482 ? CONTAINING_RECORD(iface
, vbdisp_t
, IDispatchEx_iface
)
486 HRESULT
create_vbdisp(const class_desc_t
*desc
, vbdisp_t
**ret
)
490 vbdisp
= heap_alloc_zero( FIELD_OFFSET( vbdisp_t
, props
[desc
->prop_cnt
] ));
492 return E_OUTOFMEMORY
;
494 vbdisp
->IDispatchEx_iface
.lpVtbl
= &DispatchExVtbl
;
498 if(desc
->class_initialize_id
) {
502 hres
= exec_script(desc
->ctx
, desc
->funcs
[desc
->class_initialize_id
].entries
[VBDISP_CALLGET
],
503 (IDispatch
*)&vbdisp
->IDispatchEx_iface
, &dp
, NULL
);
505 IDispatchEx_Release(&vbdisp
->IDispatchEx_iface
);
510 list_add_tail(&desc
->ctx
->objects
, &vbdisp
->entry
);
515 void collect_objects(script_ctx_t
*ctx
)
517 vbdisp_t
*iter
, *iter2
;
519 LIST_FOR_EACH_ENTRY_SAFE(iter
, iter2
, &ctx
->objects
, vbdisp_t
, entry
)
520 run_terminator(iter
);
522 while(!list_empty(&ctx
->objects
)) {
523 iter
= LIST_ENTRY(list_head(&ctx
->objects
), vbdisp_t
, entry
);
525 IDispatchEx_AddRef(&iter
->IDispatchEx_iface
);
528 list_remove(&iter
->entry
);
529 list_init(&iter
->entry
);
530 IDispatchEx_Release(&iter
->IDispatchEx_iface
);
534 HRESULT
disp_get_id(IDispatch
*disp
, BSTR name
, vbdisp_invoke_type_t invoke_type
, BOOL search_private
, DISPID
*id
)
540 vbdisp
= unsafe_impl_from_IDispatch(disp
);
542 return vbdisp_get_id(vbdisp
, name
, invoke_type
, search_private
, id
);
544 hres
= IDispatch_QueryInterface(disp
, &IID_IDispatchEx
, (void**)&dispex
);
546 TRACE("using IDispatch\n");
547 return IDispatch_GetIDsOfNames(disp
, &IID_NULL
, &name
, 1, 0, id
);
550 hres
= IDispatchEx_GetDispID(dispex
, name
, fdexNameCaseInsensitive
, id
);
551 IDispatchEx_Release(dispex
);
555 HRESULT
disp_call(script_ctx_t
*ctx
, IDispatch
*disp
, DISPID id
, DISPPARAMS
*dp
, VARIANT
*retv
)
557 const WORD flags
= DISPATCH_METHOD
|(retv
? DISPATCH_PROPERTYGET
: 0);
562 memset(&ei
, 0, sizeof(ei
));
564 V_VT(retv
) = VT_EMPTY
;
566 hres
= IDispatch_QueryInterface(disp
, &IID_IDispatchEx
, (void**)&dispex
);
570 TRACE("using IDispatch\n");
571 return IDispatch_Invoke(disp
, id
, &IID_NULL
, ctx
->lcid
, flags
, dp
, retv
, &ei
, &err
);
574 hres
= IDispatchEx_InvokeEx(dispex
, id
, ctx
->lcid
, flags
, dp
, retv
, &ei
, NULL
/* CALLER_FIXME */);
575 IDispatchEx_Release(dispex
);
579 HRESULT
disp_propput(script_ctx_t
*ctx
, IDispatch
*disp
, DISPID id
, DISPPARAMS
*dp
)
585 hres
= IDispatch_QueryInterface(disp
, &IID_IDispatchEx
, (void**)&dispex
);
586 if(SUCCEEDED(hres
)) {
587 hres
= IDispatchEx_InvokeEx(dispex
, id
, ctx
->lcid
, DISPATCH_PROPERTYPUT
, dp
, NULL
, &ei
, NULL
/* FIXME! */);
588 IDispatchEx_Release(dispex
);
592 TRACE("using IDispatch\n");
593 hres
= IDispatch_Invoke(disp
, id
, &IID_NULL
, ctx
->lcid
, DISPATCH_PROPERTYPUT
, dp
, NULL
, &ei
, &err
);