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
21 #include "wine/debug.h"
23 WINE_DEFAULT_DEBUG_CHANNEL(vbscript);
25 static inline BOOL is_func_id(vbdisp_t *This, DISPID id)
27 return id < This->desc->func_cnt;
30 static BOOL get_func_id(vbdisp_t *This, const WCHAR *name, vbdisp_invoke_type_t invoke_type, BOOL search_private, DISPID *id)
34 for(i = invoke_type == VBDISP_ANY ? 0 : 1; i < This->desc->func_cnt; i++) {
35 if(invoke_type == VBDISP_ANY) {
36 if(!search_private && !This->desc->funcs[i].is_public)
38 if(!i && !This->desc->funcs[0].name) /* default value may not exist */
41 if(!This->desc->funcs[i].entries[invoke_type]
42 || (!search_private && !This->desc->funcs[i].entries[invoke_type]->is_public))
46 if(!strcmpiW(This->desc->funcs[i].name, name)) {
55 HRESULT vbdisp_get_id(vbdisp_t *This, BSTR name, vbdisp_invoke_type_t invoke_type, BOOL search_private, DISPID *id)
59 if(get_func_id(This, name, invoke_type, search_private, id))
62 for(i=0; i < This->desc->prop_cnt; i++) {
63 if(!search_private && !This->desc->props[i].is_public)
66 if(!strcmpiW(This->desc->props[i].name, name)) {
67 *id = i + This->desc->func_cnt;
72 if(This->desc->typeinfo) {
75 hres = ITypeInfo_GetIDsOfNames(This->desc->typeinfo, &name, 1, id);
81 return DISP_E_UNKNOWNNAME;
84 static VARIANT *get_propput_arg(const DISPPARAMS *dp)
88 for(i=0; i < dp->cNamedArgs; i++) {
89 if(dp->rgdispidNamedArgs[i] == DISPID_PROPERTYPUT)
96 static HRESULT invoke_variant_prop(vbdisp_t *This, VARIANT *v, WORD flags, DISPPARAMS *dp, VARIANT *res)
101 case DISPATCH_PROPERTYGET|DISPATCH_METHOD:
103 WARN("called with arguments\n");
104 return DISP_E_MEMBERNOTFOUND; /* That's what tests show */
107 hres = VariantCopy(res, v);
110 case DISPATCH_PROPERTYPUT: {
113 put_val = get_propput_arg(dp);
115 WARN("no value to set\n");
116 return DISP_E_PARAMNOTOPTIONAL;
120 V_VT(res) = VT_EMPTY;
122 hres = VariantCopy(v, put_val);
127 FIXME("unimplemented flags %x\n", flags);
134 static HRESULT invoke_builtin(vbdisp_t *This, const builtin_prop_t *prop, WORD flags, DISPPARAMS *dp, VARIANT *res)
137 case DISPATCH_PROPERTYGET:
138 if(!(prop->flags & (BP_GET|BP_GETPUT))) {
139 FIXME("property does not support DISPATCH_PROPERTYGET\n");
143 case DISPATCH_PROPERTYGET|DISPATCH_METHOD:
145 case DISPATCH_METHOD:
146 if(prop->flags & (BP_GET|BP_GETPUT)) {
147 FIXME("Call on property\n");
151 case DISPATCH_PROPERTYPUT:
152 if(!(prop->flags & (BP_GET|BP_GETPUT))) {
153 FIXME("property does not support DISPATCH_PROPERTYPUT\n");
160 FIXME("unsupported flags %x\n", flags);
164 if(arg_cnt(dp) < prop->min_args || arg_cnt(dp) > (prop->max_args ? prop->max_args : prop->min_args)) {
165 FIXME("invalid number of arguments\n");
169 return prop->proc(This, dp->rgvarg, dp->cArgs, res);
172 static BOOL run_terminator(vbdisp_t *This)
176 if(This->terminator_ran)
178 This->terminator_ran = TRUE;
180 if(!This->desc->class_terminate_id)
184 exec_script(This->desc->ctx, This->desc->funcs[This->desc->class_terminate_id].entries[VBDISP_CALLGET],
185 (IDispatch*)&This->IDispatchEx_iface, &dp, NULL);
189 static void clean_props(vbdisp_t *This)
196 for(i=0; i < This->desc->prop_cnt; i++)
197 VariantClear(This->props+i);
200 static inline vbdisp_t *impl_from_IDispatchEx(IDispatchEx *iface)
202 return CONTAINING_RECORD(iface, vbdisp_t, IDispatchEx_iface);
205 static HRESULT WINAPI DispatchEx_QueryInterface(IDispatchEx *iface, REFIID riid, void **ppv)
207 vbdisp_t *This = impl_from_IDispatchEx(iface);
209 if(IsEqualGUID(&IID_IUnknown, riid)) {
210 TRACE("(%p)->(IID_IUnknown %p)\n", This, ppv);
211 *ppv = &This->IDispatchEx_iface;
212 }else if(IsEqualGUID(&IID_IDispatch, riid)) {
213 TRACE("(%p)->(IID_IDispatch %p)\n", This, ppv);
214 *ppv = &This->IDispatchEx_iface;
215 }else if(IsEqualGUID(&IID_IDispatchEx, riid)) {
216 TRACE("(%p)->(IID_IDispatchEx %p)\n", This, ppv);
217 *ppv = &This->IDispatchEx_iface;
219 WARN("(%p)->(%s %p)\n", This, debugstr_guid(riid), ppv);
221 return E_NOINTERFACE;
224 IUnknown_AddRef((IUnknown*)*ppv);
228 static ULONG WINAPI DispatchEx_AddRef(IDispatchEx *iface)
230 vbdisp_t *This = impl_from_IDispatchEx(iface);
231 LONG ref = InterlockedIncrement(&This->ref);
233 TRACE("(%p) ref=%d\n", This, ref);
238 static ULONG WINAPI DispatchEx_Release(IDispatchEx *iface)
240 vbdisp_t *This = impl_from_IDispatchEx(iface);
241 LONG ref = InterlockedDecrement(&This->ref);
243 TRACE("(%p) ref=%d\n", This, ref);
245 if(!ref && run_terminator(This)) {
247 list_remove(&This->entry);
254 static HRESULT WINAPI DispatchEx_GetTypeInfoCount(IDispatchEx *iface, UINT *pctinfo)
256 vbdisp_t *This = impl_from_IDispatchEx(iface);
258 TRACE("(%p)->(%p)\n", This, pctinfo);
264 static HRESULT WINAPI DispatchEx_GetTypeInfo(IDispatchEx *iface, UINT iTInfo, LCID lcid,
267 vbdisp_t *This = impl_from_IDispatchEx(iface);
268 FIXME("(%p)->(%u %u %p)\n", This, iTInfo, lcid, ppTInfo);
272 static HRESULT WINAPI DispatchEx_GetIDsOfNames(IDispatchEx *iface, REFIID riid,
273 LPOLESTR *rgszNames, UINT cNames, LCID lcid,
276 vbdisp_t *This = impl_from_IDispatchEx(iface);
277 FIXME("(%p)->(%s %p %u %u %p)\n", This, debugstr_guid(riid), rgszNames, cNames,
282 static HRESULT WINAPI DispatchEx_Invoke(IDispatchEx *iface, DISPID dispIdMember,
283 REFIID riid, LCID lcid, WORD wFlags, DISPPARAMS *pDispParams,
284 VARIANT *pVarResult, EXCEPINFO *pExcepInfo, UINT *puArgErr)
286 vbdisp_t *This = impl_from_IDispatchEx(iface);
287 FIXME("(%p)->(%d %s %d %d %p %p %p %p)\n", This, dispIdMember, debugstr_guid(riid),
288 lcid, wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr);
292 static HRESULT WINAPI DispatchEx_GetDispID(IDispatchEx *iface, BSTR bstrName, DWORD grfdex, DISPID *pid)
294 vbdisp_t *This = impl_from_IDispatchEx(iface);
296 TRACE("(%p)->(%s %x %p)\n", This, debugstr_w(bstrName), grfdex, pid);
301 if(grfdex & ~(fdexNameEnsure|fdexNameCaseInsensitive)) {
302 FIXME("unsupported flags %x\n", grfdex);
306 return vbdisp_get_id(This, bstrName, VBDISP_ANY, FALSE, pid);
309 static HRESULT WINAPI DispatchEx_InvokeEx(IDispatchEx *iface, DISPID id, LCID lcid, WORD wFlags, DISPPARAMS *pdp,
310 VARIANT *pvarRes, EXCEPINFO *pei, IServiceProvider *pspCaller)
312 vbdisp_t *This = impl_from_IDispatchEx(iface);
314 TRACE("(%p)->(%x %x %x %p %p %p %p)\n", This, id, lcid, wFlags, pdp, pvarRes, pei, pspCaller);
320 V_VT(pvarRes) = VT_EMPTY;
323 return DISP_E_MEMBERNOTFOUND;
325 if(is_func_id(This, id)) {
329 case DISPATCH_METHOD:
330 case DISPATCH_METHOD|DISPATCH_PROPERTYGET:
331 func = This->desc->funcs[id].entries[VBDISP_CALLGET];
333 FIXME("no invoke/getter\n");
334 return DISP_E_MEMBERNOTFOUND;
337 return exec_script(This->desc->ctx, func, (IDispatch*)&This->IDispatchEx_iface, pdp, pvarRes);
338 case DISPATCH_PROPERTYPUT: {
340 DISPPARAMS dp = {NULL, NULL, 1, 0};
343 FIXME("arguments not implemented\n");
347 put_val = get_propput_arg(pdp);
349 WARN("no value to set\n");
350 return DISP_E_PARAMNOTOPTIONAL;
354 func = This->desc->funcs[id].entries[V_VT(put_val) == VT_DISPATCH ? VBDISP_SET : VBDISP_LET];
356 FIXME("no letter/setter\n");
357 return DISP_E_MEMBERNOTFOUND;
360 return exec_script(This->desc->ctx, func, (IDispatch*)&This->IDispatchEx_iface, &dp, NULL);
363 FIXME("flags %x\n", wFlags);
364 return DISP_E_MEMBERNOTFOUND;
368 if(id < This->desc->prop_cnt + This->desc->func_cnt)
369 return invoke_variant_prop(This, This->props+(id-This->desc->func_cnt), wFlags, pdp, pvarRes);
371 if(This->desc->builtin_prop_cnt) {
372 unsigned min = 0, max = This->desc->builtin_prop_cnt-1, i;
376 if(This->desc->builtin_props[i].id == id)
377 return invoke_builtin(This, This->desc->builtin_props+i, wFlags, pdp, pvarRes);
378 if(This->desc->builtin_props[i].id < id)
385 return DISP_E_MEMBERNOTFOUND;
388 static HRESULT WINAPI DispatchEx_DeleteMemberByName(IDispatchEx *iface, BSTR bstrName, DWORD grfdex)
390 vbdisp_t *This = impl_from_IDispatchEx(iface);
391 FIXME("(%p)->(%s %x)\n", This, debugstr_w(bstrName), grfdex);
395 static HRESULT WINAPI DispatchEx_DeleteMemberByDispID(IDispatchEx *iface, DISPID id)
397 vbdisp_t *This = impl_from_IDispatchEx(iface);
398 FIXME("(%p)->(%x)\n", This, id);
402 static HRESULT WINAPI DispatchEx_GetMemberProperties(IDispatchEx *iface, DISPID id, DWORD grfdexFetch, DWORD *pgrfdex)
404 vbdisp_t *This = impl_from_IDispatchEx(iface);
405 FIXME("(%p)->(%x %x %p)\n", This, id, grfdexFetch, pgrfdex);
409 static HRESULT WINAPI DispatchEx_GetMemberName(IDispatchEx *iface, DISPID id, BSTR *pbstrName)
411 vbdisp_t *This = impl_from_IDispatchEx(iface);
412 FIXME("(%p)->(%x %p)\n", This, id, pbstrName);
416 static HRESULT WINAPI DispatchEx_GetNextDispID(IDispatchEx *iface, DWORD grfdex, DISPID id, DISPID *pid)
418 vbdisp_t *This = impl_from_IDispatchEx(iface);
419 FIXME("(%p)->(%x %x %p)\n", This, grfdex, id, pid);
423 static HRESULT WINAPI DispatchEx_GetNameSpaceParent(IDispatchEx *iface, IUnknown **ppunk)
425 vbdisp_t *This = impl_from_IDispatchEx(iface);
426 FIXME("(%p)->(%p)\n", This, ppunk);
430 static IDispatchExVtbl DispatchExVtbl = {
431 DispatchEx_QueryInterface,
434 DispatchEx_GetTypeInfoCount,
435 DispatchEx_GetTypeInfo,
436 DispatchEx_GetIDsOfNames,
438 DispatchEx_GetDispID,
440 DispatchEx_DeleteMemberByName,
441 DispatchEx_DeleteMemberByDispID,
442 DispatchEx_GetMemberProperties,
443 DispatchEx_GetMemberName,
444 DispatchEx_GetNextDispID,
445 DispatchEx_GetNameSpaceParent
448 static inline vbdisp_t *unsafe_impl_from_IDispatch(IDispatch *iface)
450 return iface->lpVtbl == (IDispatchVtbl*)&DispatchExVtbl
451 ? CONTAINING_RECORD(iface, vbdisp_t, IDispatchEx_iface)
455 HRESULT create_vbdisp(const class_desc_t *desc, vbdisp_t **ret)
459 vbdisp = heap_alloc_zero( FIELD_OFFSET( vbdisp_t, props[desc->prop_cnt] ));
461 return E_OUTOFMEMORY;
463 vbdisp->IDispatchEx_iface.lpVtbl = &DispatchExVtbl;
467 if(desc->class_initialize_id) {
471 hres = exec_script(desc->ctx, desc->funcs[desc->class_initialize_id].entries[VBDISP_CALLGET],
472 (IDispatch*)&vbdisp->IDispatchEx_iface, &dp, NULL);
474 IDispatchEx_Release(&vbdisp->IDispatchEx_iface);
479 list_add_tail(&desc->ctx->objects, &vbdisp->entry);
484 void collect_objects(script_ctx_t *ctx)
486 vbdisp_t *iter, *iter2;
488 LIST_FOR_EACH_ENTRY_SAFE(iter, iter2, &ctx->objects, vbdisp_t, entry)
489 run_terminator(iter);
491 LIST_FOR_EACH_ENTRY_SAFE(iter, iter2, &ctx->objects, vbdisp_t, entry) {
492 IDispatchEx_AddRef(&iter->IDispatchEx_iface);
495 IDispatchEx_Release(&iter->IDispatchEx_iface);
499 HRESULT disp_get_id(IDispatch *disp, BSTR name, vbdisp_invoke_type_t invoke_type, BOOL search_private, DISPID *id)
505 vbdisp = unsafe_impl_from_IDispatch(disp);
507 return vbdisp_get_id(vbdisp, name, invoke_type, search_private, id);
509 hres = IDispatch_QueryInterface(disp, &IID_IDispatchEx, (void**)&dispex);
511 TRACE("unsing IDispatch\n");
512 return IDispatch_GetIDsOfNames(disp, &IID_NULL, &name, 1, 0, id);
515 hres = IDispatchEx_GetDispID(dispex, name, fdexNameCaseInsensitive, id);
516 IDispatchEx_Release(dispex);
520 HRESULT disp_call(script_ctx_t *ctx, IDispatch *disp, DISPID id, DISPPARAMS *dp, VARIANT *retv)
522 const WORD flags = DISPATCH_METHOD|(retv ? DISPATCH_PROPERTYGET : 0);
527 memset(&ei, 0, sizeof(ei));
529 V_VT(retv) = VT_EMPTY;
531 hres = IDispatch_QueryInterface(disp, &IID_IDispatchEx, (void**)&dispex);
535 TRACE("using IDispatch\n");
536 return IDispatch_Invoke(disp, id, &IID_NULL, ctx->lcid, flags, dp, retv, &ei, &err);
539 hres = IDispatchEx_InvokeEx(dispex, id, ctx->lcid, flags, dp, retv, &ei, NULL /* CALLER_FIXME */);
540 IDispatchEx_Release(dispex);
544 HRESULT disp_propput(script_ctx_t *ctx, IDispatch *disp, DISPID id, VARIANT *val)
546 DISPID propput_dispid = DISPID_PROPERTYPUT;
547 DISPPARAMS dp = {val, &propput_dispid, 1, 1};
552 hres = IDispatch_QueryInterface(disp, &IID_IDispatchEx, (void**)&dispex);
553 if(SUCCEEDED(hres)) {
554 hres = IDispatchEx_InvokeEx(dispex, id, ctx->lcid, DISPATCH_PROPERTYPUT, &dp, NULL, &ei, NULL /* FIXME! */);
555 IDispatchEx_Release(dispex);
559 TRACE("using IDispatch\n");
560 hres = IDispatch_Invoke(disp, id, &IID_NULL, ctx->lcid, DISPATCH_PROPERTYPUT, &dp, NULL, &ei, &err);